added functions to get the mass according to AliHelperParticleSpecies_t and identity...
[u/mrichter/AliRoot.git] / PWG / Tools / AliHelperPID.h
1 #ifndef ALIHELPERPID_H
2 #define ALIHELPERPID_H
3
4 class AliAODEvent;
5 class AliVEvent;
6 class TH1F;
7 class TH2F;
8 class TList;
9 class AliVTrack;
10 class AliVParticle;
11 class AliStack;
12 class TParticle;
13 class AliPIDResponse;  
14 class AliPIDCombined;  
15
16 #include "TNamed.h"
17
18 namespace AliHelperPIDNameSpace {
19   
20   enum PIDType_t
21   {
22     kNSigmaTPC = 0,
23     kNSigmaTOF,
24     kNSigmaTPCTOF,  // squared sum
25     kBayes
26   };
27   
28   const Int_t kNSigmaPIDType=kNSigmaTPCTOF;//number of Nsigma PID types
29   
30   enum AliHelperDetectorType_t
31   {
32     kITS = 0,
33     kTPC,
34     kTOF,
35     kNDetectors
36   };
37   
38   
39   enum AliHelperParticleSpecies_t
40   {
41     kSpPion = 0,
42     kSpKaon,
43     kSpProton,
44     kNSpecies,
45     kSpUndefined=999
46   }; // Particle species used in plotting
47   
48   
49   enum AliHelperCharge_t
50   {
51     kChPos = 0,
52     kChNeg,
53     kNCharge
54   };
55 }
56
57 using namespace AliHelperPIDNameSpace;
58
59 class AliHelperPID : public TNamed
60 {
61  public:
62   
63   AliHelperPID();
64   virtual  ~AliHelperPID() {}
65   
66   //MC or data
67   Bool_t GetisMC(){return   fisMC;}
68   void SetisMC(Bool_t mc){fisMC=mc;}
69   //PID Type
70   void SetPIDType(PIDType_t PIDType) { fPIDType = PIDType; }
71   PIDType_t GetPIDType() {return fPIDType; }
72   //NSigma cut
73   void SetNSigmaCut(Double_t nsigma) { fNSigmaPID = nsigma; }
74   Double_t GetNSigmaCut() {return fNSigmaPID; }
75   //TOF PID
76   void SetfRequestTOFPID(Bool_t tof){fRequestTOFPID=tof;}//fRequestTOFPID
77   Bool_t GetfRequestTOFPID(){return   fRequestTOFPID;}//fRequestTOFPID
78   void SetfRemoveTracksT0Fill(Bool_t tof){fRemoveTracksT0Fill=tof;}//fRemoveTracksT0Fill
79   Bool_t GetfRemoveTracksT0Fill(){return   fRemoveTracksT0Fill;}//fRemoveTracksT0Fill
80   //Exclusive NSIgma
81   void SetfUseExclusiveNSigma(Bool_t nsigEx){fUseExclusiveNSigma=nsigEx;}//fUseExclusiveNSigma
82   Bool_t GetfUseExclusiveNSigma(){return   fUseExclusiveNSigma;}//fUseExclusiveNSigma
83   //lower pt fot TOF PID
84   Double_t SetPtTOFPID(){return   fPtTOFPID;}
85   void SetfPtTOFPID(Double_t pttof){fPtTOFPID=pttof;}
86   //set PID Combined
87   void SetPIDCombined(AliPIDCombined *obj){fPIDCombined=obj;}
88   AliPIDCombined *GetPIDCombined(){return fPIDCombined;}
89   //set cut on beyesian probability
90   void SetBayesCut(Double_t cut){fBayesCut=cut;}
91   Double_t GetBayesCut(){return fBayesCut;}
92   
93   //getters of the other data members
94   TList * GetOutputList() {return fOutputList;}//get the TList with histos
95   Double_t* GetNSigmas(AliHelperParticleSpecies_t species) {return fnsigmas[species];}//get nsigma[ipart][idet], calculated in CalculateNSigmas(trk)
96   Bool_t* GetfHasDoubleCounting() {return fHasDoubleCounting;}//get fHasDoubleCounting[ipart], calculated in GetDoubleCounting(trk)
97   //getter of histo "name" from fOutput
98   TH2F* GetHistogram2D(const char * name);//return histogram "name" from fOutputList
99   
100   //PID functions
101   // User should call ONLY the function GetParticleSpecies and set the PID strategy in the steering macro!
102   Int_t GetParticleSpecies(AliVTrack * trk, Bool_t FIllQAHistos);//calculate the PID according to the slected method.
103   Int_t GetParticleSpecies(AliVParticle * part);
104
105   Int_t GetIDBayes(AliVTrack * trk, Bool_t FIllQAHistos);//calculate the PID according to bayesian PID
106   UInt_t CalcPIDCombined(const AliVTrack *track,const AliPIDResponse *PIDResponse, Int_t detMask, Double_t* prob) const;
107   void CalculateNSigmas(AliVTrack * trk, Bool_t FIllQAHistos);//Calcuate nsigma[ipart][idet], fill NSigma histos
108   Int_t FindMinNSigma(AliVTrack * trk, Bool_t FIllQAHistos);//retun the minimum Nsigma
109   Bool_t* GetDoubleCounting(AliVTrack * trk, Bool_t FIllQAHistos);//if a particle has double counting set fHasDoubleCounting[ipart]=kTRUE (only for the second or third identity)
110   Bool_t* GetAllCompatibleIdentitiesNSigma(AliVTrack * trk, Bool_t FIllQAHistos);//All the identities are true
111   Int_t GetMCParticleSpecie(AliVEvent* event, AliVTrack * trk, Bool_t FIllQAHistos);//calculate the PID according to MC truth
112   void CheckTOF(AliVTrack * trk);//check the TOF matching and set fHasTOFPID
113   Double_t TOFBetaCalc(AliVTrack *track) const;
114   Double_t GetMass(AliHelperParticleSpecies_t id) const;
115   Long64_t Merge(TCollection* list);
116   
117  private:
118   
119   Bool_t fisMC;
120   PIDType_t fPIDType; // PID type
121   Double_t fNSigmaPID; // number of sigma for PID cut
122   Double_t fBayesCut; // Cut on Bayesian probability
123   AliPIDResponse   *fPIDResponse;     // ! PID response object
124   AliPIDCombined   *fPIDCombined;     // PIDCombined
125   TList     *fOutputList;  // List Histo's
126   Double_t fnsigmas[kNSpecies][kNSigmaPIDType+1]; //nsigma values
127   Bool_t fHasDoubleCounting[kNSpecies];//array with compatible identities
128   Bool_t fRequestTOFPID;//if true returns kSpUndefined if the TOF signal is missing
129   Bool_t fRemoveTracksT0Fill;//if true remove tracks for which only StartTime from To-Fill is available (worst resolution)
130   Bool_t fUseExclusiveNSigma;//if true returns the identity only if no double counting
131   Double_t fPtTOFPID; //lower pt bound for the TOF pid
132   Bool_t fHasTOFPID;
133   
134   AliHelperPID(const AliHelperPID&);
135   AliHelperPID& operator=(const AliHelperPID&);
136   
137   ClassDef(AliHelperPID, 4);
138   
139 };
140 #endif
141
142