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 AliVCluster * 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 class TH2F ;
  37
  38 //--- AliRoot system ---
  39 class AliVCluster;
  40 class AliAODPWG4Particle;
  41 class AliEMCALPIDUtils;
  42 class AliCalorimeterUtils;
  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   virtual ~AliCaloPID() ;//virtual dtor
  52
  53 private:
  54   AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
  55   AliCaloPID(const AliCaloPID & g) ; // cpy ctor
  56
  57 public:
  58
  59   enum PidType {
  60     kPhoton = 22,
  61     kPi0 = 111,
  62     kEta = 221,
  63     kElectron = 11,
  64     kEleCon = -11,
  65     kNeutralHadron = 2112,
  66     kChargedHadron = 211,
  67     kNeutralUnknown = 130,
  68     kChargedUnknown=321
  69   };
  70
  71   enum TagType {kPi0Decay, kEtaDecay, kOtherDecay, kConversion, kNoTag = -1};
  72
  73   TList * GetCreateOutputObjects();
  74
  75   void InitParameters();
  76
  77   Int_t GetPdg(const TString calo, const Double_t * pid, const Float_t energy) const ;
  78
  79   Int_t GetPdg(const TString calo,const TLorentzVector mom, const AliVCluster * cluster) const ;
  80
  81   TString       GetPIDParametersList();
  82
  83   void SetPIDBits(const TString calo,  const AliVCluster * cluster, AliAODPWG4Particle *aodph, const AliCalorimeterUtils* cu);
  84
  85   void Print(const Option_t * opt)const;
  86
  87   //Weight getters
  88   Float_t  GetEMCALPhotonWeight() const  { return  fEMCALPhotonWeight  ; }
  89   Float_t  GetEMCALPi0Weight() const     {  return fEMCALPi0Weight  ; }
  90   Float_t  GetEMCALElectronWeight() const  { return  fEMCALElectronWeight  ; }
  91   Float_t  GetEMCALChargeWeight() const     {  return fEMCALChargeWeight  ; }
  92   Float_t  GetEMCALNeutralWeight() const     {  return fEMCALNeutralWeight  ; }
  93   Float_t  GetPHOSPhotonWeight() const   {  return fPHOSPhotonWeight  ; }
  94   Float_t  GetPHOSPi0Weight() const   {  return fPHOSPi0Weight  ; }
  95   Float_t  GetPHOSElectronWeight() const   {  return fPHOSElectronWeight  ; }
  96   Float_t  GetPHOSChargeWeight() const   {  return fPHOSChargeWeight  ; }
  97   Float_t  GetPHOSNeutralWeight() const   {  return fPHOSNeutralWeight  ; }
  98
  99 //  Bool_t  IsPHOSPIDWeightFormulaOn() const   {  return fPHOSWeightFormula  ; }
 100 //  TFormula * GetPHOSPhotonWeightFormula() const     {  return fPHOSPhotonWeightFormula  ; }
 101 //  TFormula * GetPHOSPi0WeightFormula() const    {  return fPHOSPi0WeightFormula  ; }
 102 //
 103   //Weight setters
 104   void SetEMCALPhotonWeight(Float_t  w){  fEMCALPhotonWeight = w ; }
 105   void SetEMCALPi0Weight(Float_t  w){  fEMCALPi0Weight = w ; }
 106   void SetEMCALElectronWeight(Float_t  w){  fEMCALElectronWeight = w ; }
 107   void SetEMCALChargeWeight(Float_t  w){  fEMCALChargeWeight = w ; }
 108   void SetEMCALNeutralWeight(Float_t  w){  fEMCALNeutralWeight = w ; }
 109   void SetPHOSPhotonWeight(Float_t  w){  fPHOSPhotonWeight = w ; }
 110   void SetPHOSPi0Weight(Float_t  w){  fPHOSPi0Weight = w ; }
 111   void SetPHOSElectronWeight(Float_t  w){  fPHOSElectronWeight = w ; }
 112   void SetPHOSChargeWeight(Float_t  w){  fPHOSChargeWeight = w ; }
 113   void SetPHOSNeutralWeight(Float_t  w){  fPHOSNeutralWeight = w ; }
 114
 115 //  void UsePHOSPIDWeightFormula(Bool_t par)  { fPHOSWeightFormula  = par; }
 116 //  void SetPHOSPhotonWeightFormula(TFormula * const photon)    {  fPHOSPhotonWeightFormula  = photon; }
 117 //  void SetPHOSPi0WeightFormula(TFormula * const pi0)   {  fPHOSPi0WeightFormula  = pi0; }
 118
 119   //PID bits setters and getters
 120
 121   Bool_t IsTrackMatched(const AliVCluster * cluster, const AliCalorimeterUtils* cu) const ;
 122
 123   void SetDispersionCut(Float_t dcut ) {fDispCut = dcut; }
 124   Float_t GetDispersionCut() const {return fDispCut ;}
 125
 126   void SetTOFCut(Float_t tcut ) {fTOFCut = tcut; }
 127   Float_t GetTOFCut() const {return fTOFCut ;}
 128
 129   void SetDebug(Int_t deb) {fDebug=deb;}
 130   Int_t GetDebug() const {return fDebug;}
 131
 132   void SwitchOnBayesianRecalculation()  {fRecalculateBayesian = kTRUE ; }
 133   void SwitchOffBayesianRecalculation() {fRecalculateBayesian = kFALSE; }
 134   enum eventType{kLow,kHigh};
 135   void SetLowParticleFlux()  {fParticleFlux = kLow;}
 136   void SetHighParticleFlux() {fParticleFlux = kHigh;}
 137
 138
 139
 140   //Histogrammes setters and getters
 141
 142   virtual void SetHistoERangeAndNBins(Float_t min, Float_t max, Int_t n) {
 143     fHistoNEBins = n ;
 144     fHistoEMax = max ;
 145     fHistoEMin = min ;
 146   }
 147
 148   virtual void SetHistoDEtaRangeAndNBins(Float_t min, Float_t max, Int_t n) {
 149     fHistoNDEtaBins = n ;
 150     fHistoDEtaMax = max ;
 151     fHistoDEtaMin = min ;
 152   }
 153
 154
 155   virtual void SetHistoDPhiRangeAndNBins(Float_t min, Float_t max, Int_t n) {
 156     fHistoNDPhiBins = n ;
 157     fHistoDPhiMax = max ;
 158     fHistoDPhiMin = min ;
 159   }
 160
 161
 162
 163 private:
 164
 165   Float_t      fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL
 166   Float_t      fEMCALPi0Weight;  //Bayesian PID weight for pi0 in EMCAL
 167   Float_t      fEMCALElectronWeight; //Bayesian PID weight for electrons in EMCAL
 168   Float_t      fEMCALChargeWeight;  //Bayesian PID weight for charged hadrons in EMCAL
 169   Float_t      fEMCALNeutralWeight;  //Bayesian PID weight for neutral hadrons in EMCAL
 170   Float_t      fPHOSPhotonWeight; //Bayesian PID weight for photons in PHOS
 171   Float_t      fPHOSPi0Weight; //Bayesian PID weight for pi0 in PHOS
 172   Float_t      fPHOSElectronWeight; //Bayesian PID weight for electrons in PHOS
 173   Float_t      fPHOSChargeWeight; //Bayesian PID weight for charged hadrons in PHOS
 174   Float_t      fPHOSNeutralWeight; //Bayesian PID weight for neutral hadrons in PHOS
 175
 176 //  Bool_t  fPHOSWeightFormula ; //Use parametrized weight threshold, function of energy
 177 //  TFormula * fPHOSPhotonWeightFormula ; //Formula for photon weight
 178 //  TFormula * fPHOSPi0WeightFormula ; //Formula for pi0 weight
 179
 180   Float_t fDispCut;    //Cut on dispersion, used in PID evaluation
 181   Float_t fTOFCut;     //Cut on TOF, used in PID evaluation
 182
 183   Int_t  fDebug; //Debug level
 184
 185   Bool_t fRecalculateBayesian;        // Recalculate PID bayesian or use simple PID?
 186   Int_t  fParticleFlux;               // Particle flux for setting PID parameters
 187   AliEMCALPIDUtils * fEMCALPIDUtils;  // Pointer to EMCALPID to redo the PID Bayesian calculation
 188
 189   //Histograms
 190   Int_t   fHistoNEBins ;    // Number of bins in cluster E axis
 191   Float_t fHistoEMax ;      // Maximum value of cluster E histogram range
 192   Float_t fHistoEMin ;      // Minimum value of cluster E histogram range
 193   Int_t   fHistoNDEtaBins ; // Number of bins in dEta (cluster-track) axis
 194   Float_t fHistoDEtaMax ;   // Maximum value of dEta (cluster-track) histogram range
 195   Float_t fHistoDEtaMin ;   // Minimum value of dEta (cluster-track) histogram range
 196   Int_t   fHistoNDPhiBins ; // Number of bins in dPhi axis
 197   Float_t fHistoDPhiMax ;   // Maximum value of dPhi (cluster-track) histogram range
 198   Float_t fHistoDPhiMin ;   // Minimum value of dPhi (cluster-track) histogram range
 199
 200   TH2F * fhTrackMatchedDEta     ;     //! Eta distance between track and cluster vs cluster E
 201   TH2F * fhTrackMatchedDPhi     ;     //! Phi distance between track and cluster vs cluster E
 202   TH2F * fhTrackMatchedDEtaDPhi ;     //! Eta vs Phi distance between track and cluster, E cluster > 0.5 GeV
 203
 204
 205   ClassDef(AliCaloPID,5)
 206 } ;
 207
 208
 209 #endif //ALICALOPID_H
 210
 211
 212