| 23efe5f1 |
1 | #ifndef ALIITSPID_H |
| 2 | #define ALIITSPID_H |
| 3 | #include <TObject.h> |
| 4 | #include <TClonesArray.h> |
| 5 | #include <TVector.h> |
| 79b921e1 |
6 | #include "../TPC/AliTPCtrack.h" |
| 7 | #include "AliITSIOTrack.h" |
| 8 | #include "AliITStrackV2.h" |
| 11b3ae13 |
9 | #include <TF1.h> |
| 23efe5f1 |
10 | #include <assert.h> |
| 23efe5f1 |
11 | //___________________________________________________________________________ |
| 12 | class AliITSPid : |
| 13 | public TObject { |
| 14 | |
| 15 | public: |
| 16 | AliITSPid(Int_t ntrs=1000); |
| 17 | virtual ~AliITSPid(){} |
| 18 | void SetEdep(Int_t track,Float_t Edep); |
| 19 | void SetPmom(Int_t track,Float_t Pmom); |
| 20 | void SetPcod(Int_t track,Int_t Pcod); |
| 21 | void Print(Int_t track); |
| 22 | void Tab(void); |
| 23 | void Reset(void); |
| 24 | void SetVec(Int_t track,TVector info); |
| 25 | TVector* GetVec(Int_t track); |
| 26 | Int_t GetPcode(TClonesArray*,Float_t); |
| 27 | Int_t GetPcode(Float_t,Float_t); |
| 38f7c306 |
28 | Int_t GetPcode(AliTPCtrack*); |
| 29 | Int_t GetPcode(AliITSIOTrack*); |
| 30 | Int_t GetPcode(AliITStrackV2*); |
| 23efe5f1 |
31 | void SetCut(Int_t,Float_t,Float_t,Float_t, |
| 32 | Float_t,Float_t,Float_t,Float_t); |
| 38f7c306 |
33 | void SetAProb(Int_t ivar,Int_t icut,Float_t apro){ aprob[ivar][icut]=apro; } |
| 34 | Float_t GetAProb(Int_t ivar,Int_t icut){ return aprob[ivar][icut]; } |
| 23efe5f1 |
35 | Float_t GetWpi(){return fWpi;} |
| 36 | Float_t GetWk(){return fWk;} |
| 37 | Float_t GetWp(){return fWp;} |
| 38 | Int_t GetPid(){return fPcode;}; |
| 39 | protected: |
| 40 | public: |
| 38f7c306 |
41 | Float_t cut[13][7],aprob[3][8]; |
| 23efe5f1 |
42 | Int_t mxtrs; |
| 43 | TClonesArray *trs; |
| 44 | Float_t qtot; |
| 45 | Float_t fWpi,fWk,fWp; |
| 46 | Float_t fRpik,fRppi,fRpka,fRp; |
| 47 | Int_t fPcode; |
| 48 | //private: |
| 49 | public: |
| 23efe5f1 |
50 | int qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;} |
| 51 | Float_t qtrm(Int_t track); |
| 79b921e1 |
52 | Float_t qtrm(Float_t qarr[6],Int_t narr); |
| 38f7c306 |
53 | Float_t fSigmin; |
| 11b3ae13 |
54 | Int_t wpik(Float_t,Float_t); |
| 55 | Int_t wpikp(Float_t,Float_t); |
| 23efe5f1 |
56 | Int_t pion(){return fWpi=1.,fPcode=211;} |
| 57 | Int_t kaon(){return fWk=1.,fPcode=321;} |
| 58 | Int_t proton(){return fWp=1.,fPcode=2212;} |
| 11b3ae13 |
59 | Int_t fSilent; |
| 60 | TF1* fCutKa; |
| 61 | TF1* fCutPr; |
| 62 | TF1* fGGpi[6]; |
| 63 | TF1* fGGka[3]; |
| 64 | TF1* fGGpr[3]; |
| 65 | TF1* fggpi; |
| 66 | TF1* fggka; |
| 67 | TF1* fggpr; |
| 23efe5f1 |
68 | public: |
| e6727e13 |
69 | ClassDef(AliITSPid,2) // Class for ITS PID |
| 23efe5f1 |
70 | }; |
| 71 | |
| 72 | #endif |
| 73 | |
| 74 | |
| 75 | |
| 76 | |
git://git.uio.no / u / mrichter / AliRoot.git / blame