22a61c29565820b2f508088021a52f79bb5fd24e
[u/mrichter/AliRoot.git] / PWG4 / 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 /* History of cvs commits:
8  *
9  * $Log$
10  *
11  *
12  */
13
14 //_________________________________________________________________________
15 // Class for track/cluster acceptance selection
16 // Selection in Central barrel, EMCAL and PHOS
17 //
18 //*-- Author: Gustavo Conesa (INFN-LNF)
19
20 // --- ROOT system ---
21 #include <TObject.h> 
22 class TString ;
23 class Riostream ;
24 class TLorentzVector ;
25 class TFormula ;
26
27 //--- AliRoot system ---
28 class AliLog ;
29
30 class AliCaloPID : public TObject {
31
32 public: 
33
34   AliCaloPID() ; // ctor
35   AliCaloPID(const AliCaloPID & g) ; // cpy ctor
36   AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
37   virtual ~AliCaloPID() ;//virtual dtor
38
39   enum PidType {
40     kPhoton = 22,
41     kPi0 = 111,
42     kEta = 221, 
43     kElectron = 11, 
44     kEleCon = -11, 
45     kNeutralHadron = 2112, 
46     kChargedHadron = 211, 
47     kNeutralUnknown = 130, 
48     kChargedUnknown=321
49   };
50
51   enum PhotonStatusType {
52     kPromptPhoton=2,
53     kFragmentPhoton=3,
54     kPi0DecayPhoton=4, 
55     kEtaDecayPhoton=5, 
56     kOtherDecayPhoton=6,
57     kUnknown=7
58   };
59
60   void InitParameters();
61   
62   void Print(const Option_t * opt)const;
63   
64   Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
65   Int_t GetPdg(const TString calo,  const TLorentzVector mom, const Double_t l0, const Double_t l1, const Double_t disp, const Double_t tof, const Double_t distCPV) const ;
66
67   //Weight getters
68   Float_t  GetEMCALPhotonWeight() const  { return  fEMCALPhotonWeight  ; }
69   Float_t  GetEMCALPi0Weight() const     {  return fEMCALPi0Weight  ; }
70   Float_t  GetEMCALElectronWeight() const  { return  fEMCALElectronWeight  ; }
71   Float_t  GetEMCALChargeWeight() const     {  return fEMCALChargeWeight  ; }
72   Float_t  GetEMCALNeutralWeight() const     {  return fEMCALNeutralWeight  ; }
73   Float_t  GetPHOSPhotonWeight() const   {  return fPHOSPhotonWeight  ; }
74   Float_t  GetPHOSPi0Weight() const   {  return fPHOSPi0Weight  ; }
75   Float_t  GetPHOSElectronWeight() const   {  return fPHOSElectronWeight  ; }
76   Float_t  GetPHOSChargeWeight() const   {  return fPHOSChargeWeight  ; }
77   Float_t  GetPHOSNeutralWeight() const   {  return fPHOSNeutralWeight  ; }
78   
79   Bool_t  IsPHOSPIDWeightFormulaOn() const   {  return fPHOSWeightFormula  ; } 
80   TFormula * GetPHOSPhotonWeightFormula() const     {  return fPHOSPhotonWeightFormula  ; } 
81   TFormula * GetPHOSPi0WeightFormula() const    {  return fPHOSPi0WeightFormula  ; }
82   
83   //Weight setters
84   void SetEMCALPhotonWeight(Float_t  w){  fEMCALPhotonWeight = w ; }
85   void SetEMCALPi0Weight(Float_t  w){  fEMCALPi0Weight = w ; }
86   void SetEMCALElectronWeight(Float_t  w){  fEMCALElectronWeight = w ; }
87   void SetEMCALChargeWeight(Float_t  w){  fEMCALChargeWeight = w ; }
88   void SetEMCALNeutralWeight(Float_t  w){  fEMCALNeutralWeight = w ; }
89   void SetPHOSPhotonWeight(Float_t  w){  fPHOSPhotonWeight = w ; }
90   void SetPHOSPi0Weight(Float_t  w){  fPHOSPi0Weight = w ; }
91   void SetPHOSElectronWeight(Float_t  w){  fPHOSElectronWeight = w ; }
92   void SetPHOSChargeWeight(Float_t  w){  fPHOSChargeWeight = w ; }
93   void SetPHOSNeutralWeight(Float_t  w){  fPHOSNeutralWeight = w ; }
94   
95   void UsePHOSPIDWeightFormula(Bool_t par)  { fPHOSWeightFormula  = par; } 
96   void SetPHOSPhotonWeightFormula(TFormula * photon)    {  fPHOSPhotonWeightFormula  = photon; } 
97   void SetPHOSPi0WeightFormula(TFormula * pi0)   {  fPHOSPi0WeightFormula  = pi0; }
98   
99  private:
100   
101   Float_t      fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL 
102   Float_t      fEMCALPi0Weight;  //Bayesian PID weight for pi0 in EMCAL 
103   Float_t      fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL 
104   Float_t      fEMCALChargeWeight;  //Bayesian PID weight for charged hadrons in EMCAL 
105   Float_t      fEMCALNeutralWeight;  //Bayesian PID weight for neutral hadrons in EMCAL 
106   Float_t      fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS 
107   Float_t      fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS 
108   Float_t      fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS 
109   Float_t      fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS 
110   Float_t      fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS 
111
112   Bool_t  fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
113   TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
114   TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight
115
116   
117   ClassDef(AliCaloPID,1)
118 } ;
119
120
121 #endif //ALICALOPID_H
122
123
124