1)AliCaloPID: Posibility to recalculate PID bayesian in EMCAL
[u/mrichter/AliRoot.git] / PWG4 / PartCorrBase / AliCaloPID.h
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 //_________________________________________________________________________
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.
27 //
28 //*-- Author: Gustavo Conesa (INFN-LNF)
29
30 // --- ROOT system ---
31 #include <TObject.h> 
32 class TString ;
33 class TLorentzVector ;
34 class TFormula ;
35 class TTask;
36
37 //--- AliRoot system ---
38 class AliAODCaloCluster;
39 class AliAODPWG4Particle;
40 #ifdef __EMCALUTIL__
41 class AliEMCALPIDUtils;
42 #endif
43
44 class AliCaloPID : public TObject {
45         
46  public: 
47   
48   AliCaloPID() ; // ctor
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
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; } 
110   void SetPHOSPhotonWeightFormula(TFormula * const photon)    {  fPHOSPhotonWeightFormula  = photon; } 
111   void SetPHOSPi0WeightFormula(TFormula * const pi0)   {  fPHOSPi0WeightFormula  = pi0; }
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   
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
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
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)
159     } ;
160
161
162 #endif //ALICALOPID_H
163
164
165