]>
Commit | Line | Data |
---|---|---|
37dde34e | 1 | // $Id: AliJTrack.h,v 1.3 2008/01/21 11:56:39 djkim Exp $ |
2 | //////////////////////////////////////////////////// | |
3 | /*! | |
4 | \file AliJTrack.h | |
5 | \brief | |
6 | \author J. Rak, D.J.Kim, R.Diaz (University of Jyvaskyla) | |
7 | \email: djkim@jyu.fi | |
8 | \version $Revision: 1.1 $ | |
9 | \date $Date: 2008/05/02 11:56:39 $ | |
10 | */ | |
11 | //////////////////////////////////////////////////// | |
12 | ||
13 | #ifndef ALIJTRACK_H | |
14 | #define ALIJTRACK_H | |
15 | ||
16 | #ifndef ROOT_TObject | |
17 | #include <TObject.h> | |
18 | #endif | |
19 | ||
20 | #include "AliJBaseTrack.h" | |
21 | ||
22 | class AliJTrack : public AliJBaseTrack { | |
23 | ||
24 | public: | |
25 | enum AliJTrkPID { | |
26 | kMostProbableAliJ = -1, | |
27 | kElectronAliJ, | |
28 | kMuonAliJ, | |
29 | kPionAliJ, | |
30 | kKaonAliJ, | |
31 | kProtonAliJ, | |
32 | kNAliJTrkPID | |
33 | }; | |
34 | ||
35 | enum AliJTrkPIDmethod { | |
36 | kTOF=0, kTPC=1, kTPCTOF=2, kNAliJTrkPIDmethod | |
37 | }; | |
38 | ||
39 | AliJTrack(); // default constructor | |
40 | AliJTrack(const AliJTrack& a); // copy constructor | |
41 | ~AliJTrack(){;} //destructor | |
42 | ||
43 | Double32_t GetPID(AliJTrkPID p, AliJTrkPIDmethod m) const { return fTrkPID[p][m]; } | |
44 | UInt_t GetFilterMap() const { return fFilterMap; } | |
45 | Bool_t IsFiltered( int i ) const { return TESTBIT( fFilterMap, i); } | |
46 | Bool_t IsFilteredMask( UInt_t mask ) const { return ((Bool_t)(((fFilterMap) & mask) != 0)); }; | |
47 | int GetTPCnClust() const {return fTPCnClust;} | |
48 | Float_t GetTPCdEdx() const {return fTPCdEdx; } | |
49 | Float_t GetTOFbeta() const {return fTOFbeta;} | |
50 | Float_t GetExpectedTOFbeta( AliJTrkPID p ) const {return fExpTOFbeta[p];} | |
51 | Double_t GetExpectedTPCdEdx( AliJTrkPID p ) const {return fExpTPCdEdx[p];} | |
52 | Double_t GetTPCsigma( AliJTrkPID p) const {return fTPCsigma[p];} | |
53 | Double_t GetTOFsigma( AliJTrkPID p) const {return fTOFsigma[p];} | |
54 | ||
55 | void SetPID(AliJTrkPID p, double pro, AliJTrkPIDmethod m){ fTrkPID[p][m] = pro; } | |
56 | void SetFilterMap( UInt_t map ){ fFilterMap = map; } | |
57 | void SetFilterMap( int i, bool t ){ t?SETBIT(fFilterMap,i):CLRBIT(fFilterMap,i); } | |
58 | void SetTPCnClust(int ival) {fTPCnClust = ival;} | |
59 | void SetTPCdEdx( Float_t dedx ){ fTPCdEdx = dedx; } | |
60 | void SetTOFbeta(Float_t beta) {fTOFbeta = beta;} | |
61 | void SetExpectedTOFbeta(AliJTrkPID p, float exbeta) {fExpTOFbeta[p] = exbeta;} | |
62 | void SetExpectedTPCdEdx(AliJTrkPID p, Double_t exdEdx) {fExpTPCdEdx[p] = exdEdx;} | |
63 | void SetTPCsigma(AliJTrkPID p, Double_t TPCsigma) {fTPCsigma[p] = TPCsigma;} | |
64 | void SetTOFsigma(AliJTrkPID p, Double_t TOFsigma) {fTOFsigma[p] = TOFsigma;} // Return the expected sigma of the PID signal for the specified particle type. | |
65 | ||
66 | void SetTPCTrack( Float_t px, Float_t py, Float_t pz ){ fTPCTrack[0]=px;fTPCTrack[1]=py;fTPCTrack[2]=pz;} | |
67 | void SetUseTPCTrack(){ SetPxPyPzE( fTPCTrack[0], fTPCTrack[1], fTPCTrack[2], 0 ); } | |
68 | Float_t* GetTPCTrack(){ return fTPCTrack; } | |
69 | ||
70 | AliJTrack& operator=(const AliJTrack& trk); | |
71 | ||
72 | private: | |
73 | Double32_t fTrkPID[kNAliJTrkPID][kNAliJTrkPIDmethod]; //[0.,1.,8] Array for PID. | |
74 | UInt_t fFilterMap; // bit serious of cuts | |
75 | Short_t fTPCnClust; // track TPC nclusters | |
76 | Float_t fTPCdEdx; | |
77 | Float_t fTOFbeta; //! | |
78 | Float_t fExpTOFbeta[kNAliJTrkPID]; //! | |
79 | Double_t fExpTPCdEdx[kNAliJTrkPID]; //! | |
80 | Double_t fTPCsigma[kNAliJTrkPID]; //! | |
81 | Double_t fTOFsigma[kNAliJTrkPID]; //! | |
82 | Float_t fTPCTrack[3]; // px, py, pz for TPCTrack; | |
83 | ||
84 | ClassDef(AliJTrack,1) | |
85 | }; | |
86 | ||
87 | #endif |