source: git/external/TrackCovariance/ObsTrk.h@ ebf40fd

//
#ifndef G__OBSTRK_H
#define G__OBSTRK_H
#include <TVector3.h>
#include <TVectorD.h>
#include <TMatrixDSym.h>
#include <TDecompChol.h>
#include "SolGeom.h"
#include "TrkUtil.h"
#include "SolGridCov.h"
//
// Class to handle smearing of generated charged particle tracks
//
// Author: F. Bedeschi
//         INFN - Sezione di Pisa, Italy
//
class ObsTrk: public TrkUtil
{
        //
        // Class to handle simulation of tracking resolution
        // Prefix Obs marks variables after resolution smearing
        // Prefix Gen marks variables before resolution smearing
        //
private:        
        Double_t fB;                                    // Solenoid magnetic field
        SolGridCov* fGC;                                // Covariance matrix grid
        SolGeom*    fG;                                 // Tracker geometry
        Double_t fGenQ;                                 // Generated track charge
        Double_t fObsQ;                                 // Observed  track charge
        TVector3 fGenX;                                 // Generated track origin (x,y,z)
        TVector3 fObsX;                                 // Observed  track origin (x,y,z) @ track min. approach 
        TVector3 fGenP;                                 // Generated track momentum at track origin
        TVector3 fObsP;                                 // Observed  track momentum @ track minimum approach
        TVectorD fGenPar;                               // Generated helix track parameters (D, phi0, C, z0, cot(th)) in meters
        TVectorD fGenParMm;                             // Generated helix track parameters (D, phi0, C, z0, cot(th)) in mm
        TVectorD fGenParACTS;                   // Generated helix track parameters (D, z0, phi0, th, q/p, time
        TVectorD fGenParILC;                    // Generated helix track parameters (w, phi0, d0, z0, tan(lambda))
        TVectorD fObsPar;                               // Observed  helix track parameters (D, phi0, C, z0, cot(th)) in meters
        TVectorD fObsParMm;                             // Observed  helix track parameters (D, phi0, C, z0, cot(th)) in mm
        TVectorD fObsParACTS;                   // Observed  helix track parameters (D, z0, phi0, th, q/p, time
        TVectorD fObsParILC;                    // Observed  helix track parameters (d0, phi0, w, z0, tan(lambda))
        TMatrixDSym fCov;                               // Interpolated covariance of track in meters
        TMatrixDSym fCovMm;                             // Interpolated covariance of track parameters in mm
        TMatrixDSym fCovACTS;                   // Covariance of track parameters in ACTS format
                                                                        // (D, z0, phi0, theta, q/p, time)
        TMatrixDSym fCovILC;                    // Covariance of track parameters in ILC format
                                                                        // (d0, phi0, w, z0, tan(lambda))
        //
        // Service routines
        //
        TVectorD GenToObsPar(TVectorD gPar);
        //
public:
        //
        // Constructors
        // x(3) track origin, p(3) track momentum at origin, Q charge, B magnetic field in Tesla
        ObsTrk(TVector3 x, TVector3 p, Double_t Q, SolGridCov *GC, SolGeom *G); // Initialize and generate smeared 
        ObsTrk(Double_t *x, Double_t *p, Double_t Q, SolGridCov* GC, SolGeom *G);       // Initialize and generate smeared track
        // Destructor
        ~ObsTrk();
        //
        // Accessors
        //
        // Generator level:
        // X, P, Q
        Double_t GetGenQ()      { return fGenQ; }
        TVector3 GetGenX()      { return fGenX; }
        TVector3 GetGenP()      { return fGenP; }
        // D, phi0, C, z0, cot(th)
        TVectorD GetGenPar()    { return fGenPar; }             // in meters
        TVectorD GetGenParMm()  { return fGenParMm; }           // in mm
        // D, z0, phi0, theta, q/p, time
        TVectorD GetGenParACTS()        { return fGenParACTS; }
        // d0, phi0, w, z0, tan(lambda)
        TVectorD GetGenParILC() { return fGenParILC; }
        // Observed level X, P, Q
        Double_t GetObsQ()      { return fObsQ; }
        TVector3 GetObsX()      { return fObsX; }
        TVector3 GetObsP()      { return fObsP; }
        // D, phi0, C, z0, cot(th)
        TVectorD GetObsPar()    { return fObsPar; }             // in meters
        TVectorD GetObsParMm()  { return fObsParMm; }   // In mm
        // D, z0, phi0, theta, q/p, time
        TVectorD GetObsParACTS()        { return fObsParACTS; }
        // d0, phi0, w, z0, tan(lambda)
        TVectorD GetObsParILC() { return fObsParILC; }
        // Covariances
        TMatrixDSym GetCov()    { return fCov; }        // in meters
        TMatrixDSym GetCovMm()  { return fCov; }        // in mm
        TMatrixDSym GetCovACTS(){ return fCovACTS; }
        TMatrixDSym GetCovILC(){ return fCovILC; }
        //
};

#endif