1)AliCaloPID: Posibility to recalculate PID bayesian in EMCAL
[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliCaloPID.h
CommitLineData
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 */
5/* $Id: $ */
6
7//_________________________________________________________________________
bdd2a262 8// Class for PID selection with calorimeters
9// The Output of the 2 main methods GetPdg is a PDG number identifying the cluster,
10// being kPhoton, kElectron, kPi0 ... as defined in the header file
11// - GetPdg(const TString calo, const Double_t * pid, const Float_t energy)
12// Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle
13// - GetPdg(const TString calo,const TLorentzVector mom, const AliAODCaloCluster * cluster)
14// Recalcultes PID, the bayesian or any new one to be implemented in the future
15// Right now only the possibility to recalculate EMCAL with bayesian and simple PID.
16// In order to recalculate Bayesian, it is necessary to load the EMCALUtils library
17// and do SwitchOnBayesianRecalculation().
18// To change the PID parameters from Low to High like the ones by default, use the constructor
19// AliCaloPID(flux)
20// where flux is AliCaloPID::kLow or AliCaloPID::kHigh
21// If it is necessary to change the parameters use the constructor
22// AliCaloPID(AliEMCALPIDUtils *utils) and set the parameters before.
23// - SetPIDBits: Simple PID, depending on the thresholds fDispCut fTOFCut and even the
24// result of the PID bayesian a different PID bit is set.
25//
26// All these methods can be called in the analysis you are interested.
1c5acb87 27//
28//*-- Author: Gustavo Conesa (INFN-LNF)
29
30// --- ROOT system ---
31#include <TObject.h>
32class TString ;
33class TLorentzVector ;
34class TFormula ;
bdd2a262 35class TTask;
1c5acb87 36
37//--- AliRoot system ---
1c5acb87 38class AliAODCaloCluster;
39class AliAODPWG4Particle;
bdd2a262 40#ifdef __EMCALUTIL__
41class AliEMCALPIDUtils;
42#endif
1c5acb87 43
44class AliCaloPID : public TObject {
45
477d6cee 46 public:
47
48 AliCaloPID() ; // ctor
bdd2a262 49 AliCaloPID(const Int_t particleFlux) ; // ctor, to be used when recalculating bayesian PID
50 AliCaloPID(const TTask * emcalpid) ; // ctor, to be used when recalculating bayesian PID and need different parameters
477d6cee 51 AliCaloPID(const AliCaloPID & g) ; // cpy ctor
52 AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
53 virtual ~AliCaloPID() ;//virtual dtor
54
55 enum PidType {
56 kPhoton = 22,
57 kPi0 = 111,
58 kEta = 221,
59 kElectron = 11,
60 kEleCon = -11,
61 kNeutralHadron = 2112,
62 kChargedHadron = 211,
63 kNeutralUnknown = 130,
64 kChargedUnknown=321
65 };
66
67 enum TagType {kPi0Decay, kEtaDecay, kOtherDecay, kConversion, kNoTag = -1};
68
69 void InitParameters();
70
71 Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
72
73 Int_t GetPdg(const TString calo,const TLorentzVector mom, const AliAODCaloCluster * cluster) const ;
74
75 TString GetPIDParametersList();
76
77 void SetPIDBits(const TString calo, const AliAODCaloCluster * cluster, AliAODPWG4Particle *aodph);
78
79 void Print(const Option_t * opt)const;
80
81 //Weight getters
82 Float_t GetEMCALPhotonWeight() const { return fEMCALPhotonWeight ; }
83 Float_t GetEMCALPi0Weight() const { return fEMCALPi0Weight ; }
84 Float_t GetEMCALElectronWeight() const { return fEMCALElectronWeight ; }
85 Float_t GetEMCALChargeWeight() const { return fEMCALChargeWeight ; }
86 Float_t GetEMCALNeutralWeight() const { return fEMCALNeutralWeight ; }
87 Float_t GetPHOSPhotonWeight() const { return fPHOSPhotonWeight ; }
88 Float_t GetPHOSPi0Weight() const { return fPHOSPi0Weight ; }
89 Float_t GetPHOSElectronWeight() const { return fPHOSElectronWeight ; }
90 Float_t GetPHOSChargeWeight() const { return fPHOSChargeWeight ; }
91 Float_t GetPHOSNeutralWeight() const { return fPHOSNeutralWeight ; }
92
93 Bool_t IsPHOSPIDWeightFormulaOn() const { return fPHOSWeightFormula ; }
94 TFormula * GetPHOSPhotonWeightFormula() const { return fPHOSPhotonWeightFormula ; }
95 TFormula * GetPHOSPi0WeightFormula() const { return fPHOSPi0WeightFormula ; }
96
97 //Weight setters
98 void SetEMCALPhotonWeight(Float_t w){ fEMCALPhotonWeight = w ; }
99 void SetEMCALPi0Weight(Float_t w){ fEMCALPi0Weight = w ; }
100 void SetEMCALElectronWeight(Float_t w){ fEMCALElectronWeight = w ; }
101 void SetEMCALChargeWeight(Float_t w){ fEMCALChargeWeight = w ; }
102 void SetEMCALNeutralWeight(Float_t w){ fEMCALNeutralWeight = w ; }
103 void SetPHOSPhotonWeight(Float_t w){ fPHOSPhotonWeight = w ; }
104 void SetPHOSPi0Weight(Float_t w){ fPHOSPi0Weight = w ; }
105 void SetPHOSElectronWeight(Float_t w){ fPHOSElectronWeight = w ; }
106 void SetPHOSChargeWeight(Float_t w){ fPHOSChargeWeight = w ; }
107 void SetPHOSNeutralWeight(Float_t w){ fPHOSNeutralWeight = w ; }
108
109 void UsePHOSPIDWeightFormula(Bool_t par) { fPHOSWeightFormula = par; }
bdd2a262 110 void SetPHOSPhotonWeightFormula(TFormula * const photon) { fPHOSPhotonWeightFormula = photon; }
111 void SetPHOSPi0WeightFormula(TFormula * const pi0) { fPHOSPi0WeightFormula = pi0; }
477d6cee 112
113 //PID bits setters and getters
114 void SetDispersionCut(Float_t dcut ) {fDispCut = dcut; }
115 Float_t GetDispersionCut() const {return fDispCut ;}
116
117 void SetTOFCut(Float_t tcut ) {fTOFCut = tcut; }
118 Float_t GetTOFCut() const {return fTOFCut ;}
119
120 void SetDebug(Int_t deb) {fDebug=deb;}
121 Int_t GetDebug() const {return fDebug;}
122
bdd2a262 123 void SwitchOnBayesianRecalculation() {fRecalculateBayesian = kTRUE ; }
124 void SwitchOffBayesianRecalculation() {fRecalculateBayesian = kFALSE; }
125 enum eventType{kLow,kHigh};
126 void SetLowParticleFlux() {fParticleFlux = kLow;}
127 void SetHighParticleFlux() {fParticleFlux = kHigh;}
128
477d6cee 129 private:
130
131 Float_t fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL
132 Float_t fEMCALPi0Weight; //Bayesian PID weight for pi0 in EMCAL
133 Float_t fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL
134 Float_t fEMCALChargeWeight; //Bayesian PID weight for charged hadrons in EMCAL
135 Float_t fEMCALNeutralWeight; //Bayesian PID weight for neutral hadrons in EMCAL
136 Float_t fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS
137 Float_t fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS
138 Float_t fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS
139 Float_t fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS
140 Float_t fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS
141
142 Bool_t fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
143 TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
144 TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight
145
146 Float_t fDispCut; //Cut on dispersion, used in PID evaluation
147 Float_t fTOFCut; //Cut on TOF, used in PID evaluation
148
149 Int_t fDebug; //Debug level
bdd2a262 150
151 Bool_t fRecalculateBayesian; // Recalculate PID bayesian or use simple PID?
152 Int_t fParticleFlux; // Particle flux for setting PID parameters
153#ifdef __EMCALUTIL__
154 AliEMCALPIDUtils * fEMCALPIDUtils; //Pointer to EMCALPID to redo the PID Bayesian calculation
155#endif
156
157
158 ClassDef(AliCaloPID,4)
477d6cee 159 } ;
1c5acb87 160
161
162#endif //ALICALOPID_H
163
164
165