]>
Commit | Line | Data |
---|---|---|
1c5acb87 | 1 | #ifndef ALICALOPID_H |
2 | #define ALICALOPID_H | |
3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
1c5acb87 | 5 | |
6 | //_________________________________________________________________________ | |
bdd2a262 | 7 | // Class for PID selection with calorimeters |
49b5c49b | 8 | // The Output of the main method GetIdentifiedParticleType is a PDG number identifying the cluster, |
bdd2a262 | 9 | // being kPhoton, kElectron, kPi0 ... as defined in the header file |
3c1d9afb | 10 | // - GetIdentifiedParticleType(const AliVCluster * cluster) |
49b5c49b | 11 | // Assignes a PID tag to the cluster, right now there is the possibility to : use bayesian weights from reco, |
12 | // recalculate them (EMCAL) or use other procedures not used in reco. | |
bdd2a262 | 13 | // In order to recalculate Bayesian, it is necessary to load the EMCALUtils library |
14 | // and do SwitchOnBayesianRecalculation(). | |
15 | // To change the PID parameters from Low to High like the ones by default, use the constructor | |
16 | // AliCaloPID(flux) | |
17 | // where flux is AliCaloPID::kLow or AliCaloPID::kHigh | |
18 | // If it is necessary to change the parameters use the constructor | |
19 | // AliCaloPID(AliEMCALPIDUtils *utils) and set the parameters before. | |
49b5c49b | 20 | |
3c1d9afb | 21 | // - GetGetIdentifiedParticleTypeFromBayesian(const Double_t * pid, const Float_t energy) |
49b5c49b | 22 | // Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle, |
3c1d9afb | 23 | // executed when bayesian is ON by GetIdentifiedParticleType(const AliVCluster * cluster) |
9a6fa057 | 24 | // - SetPIDBits: Simple PID, depending on the thresholds fLOCut fTOFCut and even the |
bdd2a262 | 25 | // result of the PID bayesian a different PID bit is set. |
26 | // | |
1c5acb87 | 27 | // |
28 | //*-- Author: Gustavo Conesa (INFN-LNF) | |
29 | ||
30 | // --- ROOT system --- | |
31 | #include <TObject.h> | |
32 | class TString ; | |
33 | class TLorentzVector ; | |
a5fb4114 | 34 | #include <TFormula.h> |
f21fc003 | 35 | class TList; |
d39cba7e | 36 | class TH2F ; |
1c5acb87 | 37 | |
38 | //--- AliRoot system --- | |
0ae57829 | 39 | class AliVCluster; |
3c1d9afb | 40 | class AliVCaloCells; |
1c5acb87 | 41 | class AliAODPWG4Particle; |
c5693f62 | 42 | class AliEMCALPIDUtils; |
f2ccb5b8 | 43 | class AliCalorimeterUtils; |
49b5c49b | 44 | class AliVEvent; |
1c5acb87 | 45 | |
46 | class AliCaloPID : public TObject { | |
47 | ||
477d6cee | 48 | public: |
49 | ||
50 | AliCaloPID() ; // ctor | |
bdd2a262 | 51 | AliCaloPID(const Int_t particleFlux) ; // ctor, to be used when recalculating bayesian PID |
f21fc003 | 52 | AliCaloPID(const TNamed * emcalpid) ; // ctor, to be used when recalculating bayesian PID and need different parameters |
477d6cee | 53 | virtual ~AliCaloPID() ;//virtual dtor |
c5693f62 | 54 | |
3c1d9afb | 55 | enum PidType |
56 | { | |
a5fb4114 | 57 | kPhoton = 22, |
58 | kPi0 = 111, | |
59 | kEta = 221, | |
60 | kElectron = 11, | |
61 | kEleCon =-11, | |
62 | kNeutralHadron = 2112, | |
63 | kChargedHadron = 211, | |
477d6cee | 64 | kNeutralUnknown = 130, |
a5fb4114 | 65 | kChargedUnknown = 321 |
477d6cee | 66 | }; |
67 | ||
68 | enum TagType {kPi0Decay, kEtaDecay, kOtherDecay, kConversion, kNoTag = -1}; | |
69 | ||
49b5c49b | 70 | // Main methods |
71 | ||
a5fb4114 | 72 | TList * GetCreateOutputObjects(); |
d39cba7e | 73 | |
a5fb4114 | 74 | void InitParameters(); |
9a6fa057 | 75 | |
995c6150 | 76 | Bool_t IsInPi0SplitAsymmetryRange(const Float_t energy, const Float_t asy, const Int_t nlm); |
77 | ||
78 | Bool_t IsInPi0SplitMassRange (const Float_t energy, const Float_t mass, const Int_t nlm); | |
79 | ||
5b4c2f5b | 80 | Bool_t IsInPi0M02Range (const Float_t energy, const Float_t m02, const Int_t nlm); |
81 | Bool_t IsInEtaM02Range (const Float_t energy, const Float_t m02, const Int_t nlm); | |
82 | Bool_t IsInConM02Range (const Float_t energy, const Float_t m02, const Int_t nlm); | |
5a72d9af | 83 | |
5a72d9af | 84 | |
3c1d9afb | 85 | Int_t GetIdentifiedParticleTypeFromBayesWeights(const Bool_t isEMCAL, const Double_t * pid, const Float_t energy) ; |
86 | ||
87 | Int_t GetIdentifiedParticleTypeFromClusterSplitting(AliVCluster * cluster, AliVCaloCells* cells, | |
88 | AliCalorimeterUtils * caloutils, | |
89 | Double_t vertex[3], | |
bfdcf7fb | 90 | Int_t & nLocMax, Double_t & mass, Double_t & angle, |
19391b8c | 91 | Double_t & e1 , Double_t & e2, |
92 | Int_t & absId1, Int_t & absId2) ; | |
477d6cee | 93 | |
3c1d9afb | 94 | Int_t GetIdentifiedParticleType(const AliVCluster * cluster) ; |
477d6cee | 95 | |
9a6fa057 | 96 | TString GetPIDParametersList(); |
477d6cee | 97 | |
49b5c49b | 98 | Bool_t IsTrackMatched(AliVCluster * cluster, AliCalorimeterUtils* cu, AliVEvent* event) const ; |
99 | ||
3c1d9afb | 100 | void SetPIDBits(AliVCluster * cluster, AliAODPWG4Particle *aodph, |
49b5c49b | 101 | AliCalorimeterUtils* cu, AliVEvent* event); |
477d6cee | 102 | |
a5fb4114 | 103 | void Print(const Option_t * opt)const; |
104 | ||
3c1d9afb | 105 | void PrintClusterPIDWeights(const Double_t * pid) const; |
106 | ||
49b5c49b | 107 | //Check if cluster photon-like. Uses photon cluster parameterization in real pp data |
108 | //Returns distance in sigmas. Recommended cut 2.5 | |
109 | Float_t TestPHOSDispersion(const Double_t pt, const Double_t m20, const Double_t m02) const ; | |
110 | //Checks distance to the closest track. Takes into account | |
111 | //non-perpendicular incidence of tracks. | |
112 | Float_t TestPHOSChargedVeto(const Double_t dx, const Double_t dz, const Double_t ptTrack, | |
113 | const Int_t chargeTrack, const Double_t mf) const ; | |
114 | ||
115 | // Setters, getters | |
116 | ||
117 | void SetDebug(Int_t deb) { fDebug = deb ; } | |
118 | Int_t GetDebug() const { return fDebug ; } | |
119 | ||
120 | enum eventType{kLow,kHigh}; | |
121 | void SetLowParticleFlux() { fParticleFlux = kLow ; } | |
122 | void SetHighParticleFlux() { fParticleFlux = kHigh ; } | |
123 | // not really used, only for bayesian recalculation in EMCAL, but could be useful in future | |
124 | ||
125 | // Bayesian | |
126 | ||
127 | void SwitchOnBayesian() { fUseBayesianWeights = kTRUE ; } | |
128 | void SwitchOffBayesian() { fUseBayesianWeights = kFALSE; } | |
129 | void SwitchOnBayesianRecalculation() { fRecalculateBayesian = kTRUE ; fUseBayesianWeights = kTRUE ;} // EMCAL | |
130 | void SwitchOffBayesianRecalculation() { fRecalculateBayesian = kFALSE; } // EMCAL | |
131 | ||
c5693f62 | 132 | AliEMCALPIDUtils * GetEMCALPIDUtils() ; |
477d6cee | 133 | |
134 | //Weight getters | |
49b5c49b | 135 | Float_t GetEMCALPhotonWeight() const { return fEMCALPhotonWeight ; } |
136 | Float_t GetEMCALPi0Weight() const { return fEMCALPi0Weight ; } | |
137 | Float_t GetEMCALElectronWeight() const { return fEMCALElectronWeight ; } | |
138 | Float_t GetEMCALChargeWeight() const { return fEMCALChargeWeight ; } | |
139 | Float_t GetEMCALNeutralWeight() const { return fEMCALNeutralWeight ; } | |
140 | Float_t GetPHOSPhotonWeight() const { return fPHOSPhotonWeight ; } | |
141 | Float_t GetPHOSPi0Weight() const { return fPHOSPi0Weight ; } | |
142 | Float_t GetPHOSElectronWeight() const { return fPHOSElectronWeight ; } | |
143 | Float_t GetPHOSChargeWeight() const { return fPHOSChargeWeight ; } | |
144 | Float_t GetPHOSNeutralWeight() const { return fPHOSNeutralWeight ; } | |
145 | ||
146 | Bool_t IsPHOSPIDWeightFormulaOn() const { return fPHOSWeightFormula ; } | |
147 | ||
148 | TFormula * GetPHOSPhotonWeightFormula() { | |
a5fb4114 | 149 | if(!fPHOSPhotonWeightFormula) |
150 | fPHOSPhotonWeightFormula = new TFormula("phos_photon_weight", | |
151 | fPHOSPhotonWeightFormulaExpression); | |
49b5c49b | 152 | return fPHOSPhotonWeightFormula ; } |
477d6cee | 153 | |
49b5c49b | 154 | TFormula * GetPHOSPi0WeightFormula() { |
a5fb4114 | 155 | if(!fPHOSPi0WeightFormula) |
156 | fPHOSPi0WeightFormula = new TFormula("phos_pi0_weight", | |
157 | fPHOSPi0WeightFormulaExpression); | |
49b5c49b | 158 | return fPHOSPi0WeightFormula ; } |
5ae09196 | 159 | |
49b5c49b | 160 | TString GetPHOSPhotonWeightFormulaExpression() const { return fPHOSPhotonWeightFormulaExpression ; } |
161 | TString GetPHOSPi0WeightFormulaExpression() const { return fPHOSPi0WeightFormulaExpression ; } | |
5ae09196 | 162 | |
a5fb4114 | 163 | //Weight setters |
49b5c49b | 164 | void SetEMCALPhotonWeight (Float_t w) { fEMCALPhotonWeight = w ; } |
165 | void SetEMCALPi0Weight (Float_t w) { fEMCALPi0Weight = w ; } | |
166 | void SetEMCALElectronWeight(Float_t w) { fEMCALElectronWeight = w ; } | |
167 | void SetEMCALChargeWeight (Float_t w) { fEMCALChargeWeight = w ; } | |
168 | void SetEMCALNeutralWeight (Float_t w) { fEMCALNeutralWeight = w ; } | |
169 | void SetPHOSPhotonWeight (Float_t w) { fPHOSPhotonWeight = w ; } | |
170 | void SetPHOSPi0Weight (Float_t w) { fPHOSPi0Weight = w ; } | |
171 | void SetPHOSElectronWeight (Float_t w) { fPHOSElectronWeight = w ; } | |
172 | void SetPHOSChargeWeight (Float_t w) { fPHOSChargeWeight = w ; } | |
173 | void SetPHOSNeutralWeight (Float_t w) { fPHOSNeutralWeight = w ; } | |
174 | ||
175 | void UsePHOSPIDWeightFormula (Bool_t ok ) { fPHOSWeightFormula = ok ; } | |
176 | void SetPHOSPhotonWeightFormulaExpression(TString ph) { fPHOSPhotonWeightFormulaExpression = ph ; } | |
177 | void SetPHOSPi0WeightFormulaExpression (TString pi) { fPHOSPi0WeightFormulaExpression = pi ; } | |
d39cba7e | 178 | |
49b5c49b | 179 | //PID cuts |
d39cba7e | 180 | |
49b5c49b | 181 | void SetEMCALLambda0CutMax(Float_t lcut ) { fEMCALL0CutMax = lcut ; } |
182 | Float_t GetEMCALLambda0CutMax() const { return fEMCALL0CutMax ; } | |
183 | ||
184 | void SetEMCALLambda0CutMin(Float_t lcut ) { fEMCALL0CutMin = lcut ; } | |
185 | Float_t GetEMCALLambda0CutMin() const { return fEMCALL0CutMin ; } | |
186 | ||
187 | void SetEMCALDEtaCut(Float_t dcut ) { fEMCALDEtaCut = dcut ; } | |
188 | Float_t GetEMCALDEtaCut() const { return fEMCALDEtaCut ; } | |
189 | ||
190 | void SetEMCALDPhiCut(Float_t dcut ) { fEMCALDPhiCut = dcut ; } | |
191 | Float_t GetEMCALDPhiCut() const { return fEMCALDPhiCut ; } | |
192 | ||
193 | void SetTOFCut(Float_t tcut ) { fTOFCut = tcut ; } | |
194 | Float_t GetTOFCut() const { return fTOFCut ; } | |
195 | ||
196 | void SetPHOSRCut(Float_t rcut ) { fPHOSRCut = rcut ; } | |
197 | Float_t GetPHOSRCut() const { return fPHOSRCut ; } | |
a5fb4114 | 198 | |
49b5c49b | 199 | void SetPHOSDispersionCut(Float_t dcut ) { fPHOSDispersionCut = dcut ; } |
200 | Float_t GetPHOSDispersionCut() const { return fPHOSDispersionCut ; } | |
201 | ||
3c1d9afb | 202 | // Cluster splitting analysis |
203 | ||
204 | void SwitchOnClusterSplittingPID() { fDoClusterSplitting = kTRUE ; } | |
205 | void SwitchOffClusterplittingPID() { fDoClusterSplitting = kFALSE ; } | |
5a72d9af | 206 | |
207 | void SwitchOnSimpleSplitMassCut() { fUseSimpleMassCut = kTRUE ; } | |
208 | void SwitchOffSimpleSplitMassCut() { fUseSimpleMassCut = kFALSE ; } | |
3c1d9afb | 209 | |
5a72d9af | 210 | void SwitchOnSimpleSplitM02Cut() { fUseSimpleM02Cut = kTRUE ; } |
211 | void SwitchOffSimpleSplitM02Cut() { fUseSimpleM02Cut = kFALSE ; } | |
212 | ||
667432ef | 213 | void SwitchOnSplitAsymmetryCut() { fUseSplitAsyCut = kTRUE ; } |
214 | void SwitchOffSplitAsymmetryCut() { fUseSplitAsyCut = kFALSE ; } | |
215 | ||
3c1d9afb | 216 | void SetClusterSplittingM02Cut(Float_t min=0, Float_t max=100) |
217 | { fSplitM02MinCut = min ; fSplitM02MaxCut = max ; } | |
218 | ||
667432ef | 219 | void SetClusterSplittingMinNCells(Int_t cut) { fSplitMinNCells = cut ; } |
220 | ||
4d97a954 | 221 | void SetSplitEnergyFractionMinimum(Int_t i, Float_t min){ if (i < 3 && i >=0 ) fSplitEFracMin[i] = min ; } |
222 | Float_t GetSplitEnergyFractionMinimum(Int_t i) const { if( i < 3 && i >=0 ) return fSplitEFracMin[i] ; else return 0 ; } | |
3c1d9afb | 223 | |
5a72d9af | 224 | Float_t GetPi0MinMass() const { return fMassPi0Min ; } // Simple cut case |
225 | Float_t GetEtaMinMass() const { return fMassEtaMin ; } // Simple cut case | |
3c1d9afb | 226 | Float_t GetPhotonMinMass() const { return fMassPhoMin ; } |
227 | Float_t GetPi0MaxMass() const { return fMassPi0Max ; } | |
228 | Float_t GetEtaMaxMass() const { return fMassEtaMax ; } | |
229 | Float_t GetPhotonMaxMass() const { return fMassPhoMax ; } | |
230 | ||
5a72d9af | 231 | void SetSplitWidthSigma(Float_t s) { fSplitWidthSigma = s ; } |
995c6150 | 232 | void SetPi0MassWidthSelectionParameters (Int_t iparam, Float_t param) { if(iparam < 7 ) fMassWidthPi0Param[iparam] = param ; } |
5b4c2f5b | 233 | void SetM02MaximumSelectionParameters (Int_t inlm, Int_t iparam, Float_t param) |
234 | { if(iparam < 6 && inlm < 2) fM02MaxParam[inlm][iparam] = param ; } | |
a5a3f703 | 235 | void SetM02MinimumSelectionParameters (Int_t inlm, Int_t iparam, Float_t param) |
667432ef | 236 | { if(iparam < 6 && inlm < 2) fM02MinParam[inlm][iparam] = param ; } |
afc83530 | 237 | void SetAsymmetryMinimumSelectionParameters(Int_t inlm, Int_t iparam, Float_t param) |
667432ef | 238 | { if(iparam < 6 && inlm < 2) fAsyMinParam[inlm][iparam] = param ; } |
5a72d9af | 239 | |
240 | void SetPi0MassRange(Float_t min, Float_t max) { fMassPi0Min = min ; fMassPi0Max = max ; } // Simple case | |
241 | void SetEtaMassRange(Float_t min, Float_t max) { fMassEtaMin = min ; fMassEtaMax = max ; } | |
242 | void SetPhotonMassRange(Float_t min, Float_t max) { fMassPhoMin = min ; fMassPhoMax = max ; } | |
667432ef | 243 | |
f2ccb5b8 | 244 | private: |
477d6cee | 245 | |
49b5c49b | 246 | Int_t fDebug; // Debug level |
247 | Int_t fParticleFlux; // Particle flux for setting PID parameters | |
248 | ||
249 | // Bayesian | |
250 | AliEMCALPIDUtils * fEMCALPIDUtils; // Pointer to EMCALPID to redo the PID Bayesian calculation | |
251 | Bool_t fUseBayesianWeights; // Select clusters based on weights calculated in reconstruction | |
252 | Bool_t fRecalculateBayesian; // Recalculate PID bayesian or use simple PID? | |
253 | ||
254 | Float_t fEMCALPhotonWeight; // Bayesian PID weight for photons in EMCAL | |
255 | Float_t fEMCALPi0Weight; // Bayesian PID weight for pi0 in EMCAL | |
256 | Float_t fEMCALElectronWeight; // Bayesian PID weight for electrons in EMCAL | |
257 | Float_t fEMCALChargeWeight; // Bayesian PID weight for charged hadrons in EMCAL | |
258 | Float_t fEMCALNeutralWeight; // Bayesian PID weight for neutral hadrons in EMCAL | |
259 | Float_t fPHOSPhotonWeight; // Bayesian PID weight for photons in PHOS | |
260 | Float_t fPHOSPi0Weight; // Bayesian PID weight for pi0 in PHOS | |
261 | Float_t fPHOSElectronWeight; // Bayesian PID weight for electrons in PHOS | |
262 | Float_t fPHOSChargeWeight; // Bayesian PID weight for charged hadrons in PHOS | |
263 | Float_t fPHOSNeutralWeight; // Bayesian PID weight for neutral hadrons in PHOS | |
a5fb4114 | 264 | |
9a6fa057 | 265 | Bool_t fPHOSWeightFormula ; // Use parametrized weight threshold, function of energy |
266 | TFormula *fPHOSPhotonWeightFormula ; // Formula for photon weight | |
267 | TFormula *fPHOSPi0WeightFormula ; // Formula for pi0 weight | |
a5fb4114 | 268 | TString fPHOSPhotonWeightFormulaExpression; // Photon weight formula in string |
269 | TString fPHOSPi0WeightFormulaExpression; // Pi0 weight formula in string | |
270 | ||
49b5c49b | 271 | // PID calculation |
272 | Float_t fEMCALL0CutMax; // Max Cut on shower shape lambda0, used in PID evaluation, only EMCAL | |
273 | Float_t fEMCALL0CutMin; // Min Cut on shower shape lambda0, used in PID evaluation, only EMCAL | |
274 | Float_t fEMCALDEtaCut; // Track matching cut on Dz | |
275 | Float_t fEMCALDPhiCut; // Track matching cut on Dx | |
ae182e60 | 276 | |
49b5c49b | 277 | Float_t fTOFCut; // Cut on TOF, used in PID evaluation |
278 | ||
279 | Float_t fPHOSDispersionCut; // Shower shape elipse radious cut | |
280 | Float_t fPHOSRCut; // Track-Cluster distance cut for track matching in PHOS | |
49b5c49b | 281 | |
3c1d9afb | 282 | // Cluster splitting mass ranges |
283 | Bool_t fDoClusterSplitting; // Cluster splitting analysis | |
5a72d9af | 284 | Bool_t fUseSimpleMassCut; // Use simple min-max pi0 mass cut |
285 | Bool_t fUseSimpleM02Cut; // Use simple min-max M02 cut | |
667432ef | 286 | Bool_t fUseSplitAsyCut ; // Remove splitted clusters with too large asymmetry, range defined in AliCaloPID |
287 | Float_t fSplitM02MaxCut ; // Study clusters with l0 smaller than cut | |
5a72d9af | 288 | Float_t fSplitM02MinCut ; // Study clusters with l0 larger than cut // simple case |
3c1d9afb | 289 | Int_t fSplitMinNCells ; // Study clusters with ncells larger than cut |
290 | Float_t fMassEtaMin ; // Min Eta mass | |
291 | Float_t fMassEtaMax ; // Max Eta mass | |
5a72d9af | 292 | Float_t fMassPi0Min ; // Min Pi0 mass // simple cut case |
293 | Float_t fMassPi0Max ; // Min Pi0 mass // simple cut case | |
3c1d9afb | 294 | Float_t fMassPhoMin ; // Min Photon mass |
295 | Float_t fMassPhoMax ; // Min Photon mass | |
5a72d9af | 296 | Float_t fMassWidthPi0Param[7] ; // 3 param for pol2 fit on width, 2 param for mass position NLM=1 and NLM>1 for pi0 selection |
5b4c2f5b | 297 | Float_t fM02MinParam[2][6] ; // 4 param for pol3 fit on M02 minimum for pi0 selection (maximum for conversions) |
298 | Float_t fM02MaxParam[2][6] ; // 4 param for pol3 fit on M02 maximum for pi0 selection | |
667432ef | 299 | Float_t fAsyMinParam[2][6] ; // 4 param for pol3 fit on asymmetry minimum, for 2 cases, NLM=1 and NLM>=2 |
4d97a954 | 300 | Float_t fSplitEFracMin[3] ; // Do not use clusters with too large energy in cluster compared |
301 | // to energy in splitted clusters, depeding on NLM | |
5a72d9af | 302 | Float_t fSplitWidthSigma; // Cut on mass+-width*fSplitWidthSigma |
303 | ||
304 | ||
3c1d9afb | 305 | |
5a72d9af | 306 | AliCaloPID & operator = (const AliCaloPID & cpid) ; // cpy assignment |
307 | AliCaloPID( const AliCaloPID & cpid) ; // cpy ctor | |
c5693f62 | 308 | |
5b4c2f5b | 309 | ClassDef(AliCaloPID,17) |
3c1d9afb | 310 | |
e5dbdaf0 | 311 | } ; |
1c5acb87 | 312 | |
313 | ||
314 | #endif //ALICALOPID_H | |
315 | ||
316 | ||
317 |