PWG3/vertexingHF/AliAODPidHF.h

   1 #ifndef ALIAODPIDHF_H
   2 #define ALIAODPIDHF_H
   3
   4 /* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
   5  *  * See cxx source for full Copyright notice                               */
   6
   7 //***********************************************************
   8 //// Class AliAODPidHF
   9 //// class for PID with AliAODRecoDecayHF
  10 //// Authors: D. Caffarri caffarri@pd.infn.it, A.Dainese andrea.dainese@pd.infn.it, S. Dash dash@to.infn.it, F. Prino prino@to.infn.it, R. Romita r.romita@gsi.de, Y. Wang yifei@pi0.physi.uni-heidelberg.de
  11 ////***********************************************************
  12
  13 #include "AliAODPid.h"
  14 #include "AliAODTrack.h"
  15
  16 class AliAODPidHF : public AliAODPid{
  17
  18  public:
  19
  20  AliAODPidHF();
  21  AliAODPidHF(const AliAODPidHF& pid);
  22  AliAODPidHF& operator=(const AliAODPidHF& pid);
  23  virtual ~AliAODPidHF();
  24
  25  //Setters
  26  void SetSigma(Double_t *sigma){
  27     for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
  28      }
  29  void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
  30  void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
  31  void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
  32  void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
  33  void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
  34  void SetPriors(Double_t *priors){fPriors=priors;return;}
  35  void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
  36  void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
  37  void SetAsym(Bool_t asym){fAsym=asym;return;}
  38  void SetTPC(Bool_t tpc){fTPC=tpc;return;}
  39  void SetTOF(Bool_t tof){fTOF=tof;return;}
  40  void SetITS(Bool_t its){fITS=its;return;}
  41  void SetTRD(Bool_t trd){fTRD=trd;return;}
  42  void SetMatch(Int_t match){fMatch=match;return;}
  43  void SetCompat(Bool_t comp){fCompat=comp;return;}
  44
  45  //Getters
  46  Double_t GetSigma(Int_t idet){return fnSigma[idet];}
  47  Double_t GetTofSigma(){return fTOFSigma;}
  48  void GetPriors(Double_t *priors){priors=fPriors;return;}
  49  void GetPLimit(Double_t *plim){plim=fPLimit;}
  50  Bool_t GetAsym(){return fAsym;}
  51  Bool_t GetTPC(){return fTPC;}
  52  Bool_t GetTOF(){return fTOF;}
  53  Bool_t GetITS(){return fITS;}
  54  Bool_t GetTRD(){return fTRD;}
  55  Int_t GetMatch(){return fMatch;}
  56  Bool_t GetCompat(){return fCompat;}
  57
  58  Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
  59  Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
  60  Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
  61  Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
  62  Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
  63  void BayesianProbability(AliAODTrack *track,Double_t *pid) const;
  64  void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
  65  Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
  66
  67  Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
  68  Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat);
  69
  70  Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
  71
  72
  73
  74  protected:
  75
  76  Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
  77  Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
  78  Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
  79  void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const;
  80  void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
  81  void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
  82  void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const;
  83
  84  private:
  85  Int_t fnNSigma; // number of sigmas
  86  Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
  87  Double_t fTOFSigma; // TOF precision
  88  Int_t fnPriors; //number of priors
  89  Double_t *fPriors; // [fnPriors], set of priors
  90  Int_t fnPLimit; //number of Plimit
  91  Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
  92  Bool_t fAsym; // asimmetric PID required
  93  Bool_t fTPC; //switch to include or exclude TPC
  94  Bool_t fTOF; // switch to include or exclude TOF
  95  Bool_t fITS; //switch to include or exclude ITS
  96  Bool_t fTRD; // switch to include or exclude TRD
  97  Int_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region
  98  Bool_t fCompat; // compatibility region : useful only if fMatch=1
  99
 100
 101
 102  ClassDef(AliAODPidHF,5) // AliAODPid for heavy flavor PID
 103
 104 };
 105
 106 #endif