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){fnSigma=sigma;return;}
  27  void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
  28  void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
  29  void SetPriors(Double_t *priors){fPriors=priors;return;}
  30  void SetPLimit(Double_t *plim){fPLimit=plim;return;}
  31  void SetAsym(Bool_t asym){fAsym=asym;return;}
  32
  33
  34  Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
  35  Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
  36  Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
  37  Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
  38  Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
  39  void BayesianProbability(AliAODTrack *track,TString detectors,Double_t *pid) const;
  40  void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
  41  Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
  42
  43  Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
  44  Int_t MatchTPCTOF(AliAODTrack *track,Int_t mode,Int_t specie,Bool_t compat);
  45
  46
  47
  48  protected:
  49
  50  Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
  51  Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
  52  Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
  53  void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const;
  54  void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
  55  void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
  56  void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const;
  57
  58  private:
  59
  60  Int_t fnNSigma; // size of the nsigma array
  61  Double_t *fnSigma; //[fnNSigma] sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
  62  Double_t fTOFSigma; // TOF precision
  63  Int_t fnPriors;    // size of the priors array
  64  Double_t *fPriors; //[fnPriors] set of priors
  65  Int_t fnPLimit;  // size of the plimit array
  66  Double_t *fPLimit; //[fnPLimit] limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
  67  Bool_t fAsym; // asimmetric PID required
  68
  69
  70
  71  ClassDef(AliAODPidHF,4) // AliAODPid for heavy flavor PID
  72
  73 };
  74
  75 #endif