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