Possibility to use nSigma in ITS for PID (Annalisa)
[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAODPidHF.h
1 #ifndef ALIAODPIDHF_H
2 #define ALIAODPIDHF_H
3
4 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */ 
8
9 //***********************************************************
10 //// Class AliAODPidHF
11 //// class for PID with AliAODRecoDecayHF 
12 //// Authors: D. Caffarri caffarri@pd.infn.it, A.Dainese andrea.dainese@pd.infn.it, S. Dash dash@to.infn.it, F. Prino prino@to.infn.it, R. Romita r.romita@gsi.de, Y. Wang yifei@pi0.physi.uni-heidelberg.de
13 ////***********************************************************
14
15 #include <TString.h>
16 #include <TH1F.h>
17 #include "AliAODPid.h"
18 #include "AliAODTrack.h"
19 #include "AliPIDResponse.h"
20 #include "AliPIDCombined.h"
21 #include "AliPID.h"
22
23 class AliAODPidHF : public AliAODPid{
24
25  public:
26
27  enum ECombDetectors {
28   kTPC,
29   kTOF,
30   kTPCTOF,
31   kTPCITS
32  };
33
34  AliAODPidHF();
35  AliAODPidHF(const AliAODPidHF& pid);
36  AliAODPidHF& operator=(const AliAODPidHF& pid);
37  virtual ~AliAODPidHF();
38
39  //Setters
40  void SetSigma(Double_t *sigma){
41     for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
42      }
43  void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
44  void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
45  void SetSigmaForTPCCompat(Double_t sigma){fnSigmaCompat[0]=sigma;return;}
46  void SetSigmaForTOFCompat(Double_t sigma){fnSigmaCompat[1]=sigma;return;}
47  void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
48  void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
49  void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
50  void SetPriors(Double_t *priors){fPriors=priors;return;}
51  void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
52  void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
53  void SetAsym(Bool_t asym){fAsym=asym;return;}
54  void SetTPC(Bool_t tpc){fTPC=tpc;return;}
55  void SetTOF(Bool_t tof){fTOF=tof;return;}
56  void SetITS(Bool_t its){fITS=its;return;}
57  void SetTRD(Bool_t trd){fTRD=trd;return;}
58  void SetMatch(Int_t match){fMatch=match;return;}
59  void SetCompat(Bool_t comp){fCompat=comp;return;}
60  void SetMC(Bool_t mc){fMC=mc;return;}
61  void SetMClowenpp2011(Bool_t mc){fMCLowEn2011=mc;return;}
62  void SetOnePad(Bool_t onepad){fOnePad=onepad;return;}
63  void SetppLowEn2011(Bool_t opt){fppLowEn2011=opt;return;}
64  void SetPbPb(Bool_t pbpb){fPbPb=pbpb;return;}
65  void SetPCompatTOF(Double_t pTOF){fPCompatTOF=pTOF;return;}
66  void SetTOFdecide(Bool_t tOFdecide){fTOFdecide=tOFdecide;return;}
67  void SetOldPid(Bool_t oldPid){fOldPid=oldPid;return;}
68  void SetPtThresholdTPC(Double_t ptThresholdTPC){fPtThresholdTPC=ptThresholdTPC;return;}
69  void SetMaxTrackMomForCombinedPID(Double_t mom){fMaxTrackMomForCombinedPID=mom;}
70  void SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
71  void SetCombDetectors(ECombDetectors pidComb) {
72    fCombDetectors=pidComb;
73   }
74  void SetPionPriorHisto(TH1F* histo){
75    if(fPriorsH[AliPID::kPion]) delete fPriorsH[AliPID::kPion];
76    fPriorsH[AliPID::kPion] = new TH1F(*histo);
77  }
78  void SetKaonPriorHisto(TH1F* histo){
79    if(fPriorsH[AliPID::kKaon]) delete fPriorsH[AliPID::kKaon];
80    fPriorsH[AliPID::kKaon] = new TH1F(*histo);
81  }
82  void SetProtonPriorHisto(TH1F* histo){
83    if(fPriorsH[AliPID::kProton]) delete fPriorsH[AliPID::kProton];
84    fPriorsH[AliPID::kProton] = new TH1F(*histo);
85  }
86
87  
88  //Getters
89  
90  Int_t GetnSigmaTPC(AliAODTrack *track, Int_t species, Double_t &sigma) const;
91  Int_t GetnSigmaTOF(AliAODTrack *track, Int_t species, Double_t &sigma) const;
92  Int_t GetnSigmaITS(AliAODTrack *track, Int_t species, Double_t &sigma) const;
93  Double_t GetSigma(Int_t idet) const{return fnSigma[idet];}
94  Double_t GetTofSigma() const{return fTOFSigma;}
95  //void GetPriors(Double_t *priors) const{priors=fPriors;return;}
96  //void GetPLimit(Double_t *plim) const{plim=fPLimit;}
97  void GetPriors(Double_t *priors) const{for(Int_t i=0;i<fnPriors;i++){priors[i]=fPriors[i];}return;}
98  void GetPLimit(Double_t *plim) const{for(Int_t i=0;i<fnPLimit;i++){plim[i]=fPLimit[i];}return;}
99  Bool_t GetAsym() const{return fAsym;}
100  Bool_t GetTPC() const{return fTPC;}
101  Bool_t GetTOF() const{return fTOF;}
102  Bool_t GetITS() const{return fITS;}
103  Bool_t GetTRD() const{return fTRD;}
104  Int_t GetMatch() const{return fMatch;}
105  Bool_t GetCompat() const{return fCompat;}
106  Bool_t GetMC() const{return fMC;}
107  Bool_t GetOnePad() const{return fOnePad;}
108  Bool_t GetppLowEn2011() const {return fppLowEn2011;}
109  Bool_t GetMCLowEn2011() const {return fMCLowEn2011;}
110  Bool_t GetPbPb() const{return fPbPb;}
111  Bool_t GetTOFdecide() const{return fTOFdecide;}
112  Double_t GetPCompatTOF() const{return fPCompatTOF;}
113  Double_t GetnSigmaCompatTPC() const{return fnSigmaCompat[0];}
114  Double_t GetnSigmaCompatTOF() const{return fnSigmaCompat[1];}
115  Bool_t GetOldPid(){return fOldPid;}
116  Double_t GetPtThresholdTPC(){return fPtThresholdTPC;}
117  Double_t GetMaxTrackMomForCombinedPID(){return fMaxTrackMomForCombinedPID;}
118  AliPIDResponse *GetPidResponse() const {return fPidResponse;}
119  AliPIDCombined *GetPidCombined() const {return fPidCombined;}
120  ECombDetectors GetCombDetectors() const {
121    return fCombDetectors;
122   }
123  Bool_t GetUseCombined() {return fUseCombined;}
124
125  Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
126  Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
127  Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
128  Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
129  Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
130  void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
131  Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
132
133  Bool_t CheckITSPIDStatus(AliAODTrack *track) const;
134  Bool_t CheckTPCPIDStatus(AliAODTrack *track) const;
135  Bool_t CheckTOFPIDStatus(AliAODTrack *track) const;
136  Bool_t CheckTRDPIDStatus(AliAODTrack *track) const;
137
138  Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
139  Int_t MatchTPCTOF(AliAODTrack *track,Int_t specie);
140
141  Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
142
143  Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);
144
145  Bool_t IsExcluded(AliAODTrack *track, Int_t labelTrack, Double_t nsigmaCut, TString detector);
146
147  void GetTPCBetheBlochParams(Double_t alephParameters[5]) const;
148  void SetBetheBloch();
149   // method for AliPIDCombined object
150  void SetSelectedSpecies(Int_t ispecies = AliPID::kSPECIES){GetPidCombined()->SetSelectedSpecies(ispecies);};
151  void SetPriorDistribution(AliPID::EParticleType type,TH1F *prior);
152  void DrawPrior(AliPID::EParticleType type);
153  void SetPriorsHistos(TString priorFileName);
154  void SetUpCombinedPID();
155  void SetUseCombined(Bool_t useCombined=kTRUE) {fUseCombined=useCombined;}
156
157  Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
158  Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
159  Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
160
161  protected:
162
163
164  private:
165  Int_t fnNSigma; // number of sigmas
166  Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
167  Double_t fTOFSigma; // TOF precision 
168  Int_t fnPriors; //number of priors
169  Double_t *fPriors; // [fnPriors], set of priors
170  Int_t fnPLimit; //number of Plimit
171  Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
172  Bool_t fAsym; // asimmetric PID required
173  Bool_t fTPC; //switch to include or exclude TPC 
174  Bool_t fTOF; // switch to include or exclude TOF
175  Bool_t fITS; //switch to include or exclude ITS
176  Bool_t fTRD; // switch to include or exclude TRD
177  Int_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region
178  Bool_t fCompat; // compatibility region : useful only if fMatch=1
179  Double_t fPCompatTOF; //  compatibility p limit for TOF  
180  Int_t fnNSigmaCompat; // number of sigmas
181  Double_t *fnSigmaCompat; //[fnNSigmaCompat]  0: n sigma for TPC compatibility band, 1: for TOF  
182  Bool_t fMC; // MC(kTRUE) or real data (kFALSE, default option)
183  Bool_t fOnePad; //  real data with one pad clusters 
184  Bool_t fMCLowEn2011; //  MC for low energy MC
185  Bool_t fppLowEn2011; //  Data for low energy pp 2011
186  Bool_t fPbPb; //  real data PbPb 
187  Bool_t fTOFdecide; //  real data PbPb 
188  Bool_t fOldPid; //  old PID method implemented
189  Double_t fPtThresholdTPC; //  pT threshold to use TPC PID
190  Double_t fMaxTrackMomForCombinedPID; // momentum threshold to use PID
191  AliPIDResponse *fPidResponse; //! pid response
192  AliPIDCombined* fPidCombined; //! combined PID object 
193
194  AliTPCPIDResponse* fTPCResponse; //! TPC response 
195
196  TH1F* fPriorsH[AliPID::kSPECIES]; // priors histos
197  ECombDetectors fCombDetectors; // detectors to be involved for combined PID
198  Bool_t fUseCombined; // detectors to be involved for combined PID
199
200  ClassDef(AliAODPidHF,19) // AliAODPid for heavy flavor PID
201
202 };
203
204 #endif
205