#define ALIAODPIDHF_H
/* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
- * * See cxx source for full Copyright notice */
+ * See cxx source for full Copyright notice */
+
+/* $Id$ */
//***********************************************************
//// Class AliAODPidHF
#include "AliAODPid.h"
#include "AliAODTrack.h"
-#include "AliTPCPIDResponse.h"
+#include "AliPIDResponse.h"
+#include "AliPIDCombined.h"
class AliAODPidHF : public AliAODPid{
}
void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
+ void SetSigmaForTPCCompat(Double_t sigma){fnSigmaCompat[0]=sigma;return;}
+ void SetSigmaForTOFCompat(Double_t sigma){fnSigmaCompat[1]=sigma;return;}
void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
void SetMatch(Int_t match){fMatch=match;return;}
void SetCompat(Bool_t comp){fCompat=comp;return;}
void SetMC(Bool_t mc){fMC=mc;return;}
+ void SetMClowenpp2011(Bool_t mc){fMCLowEn2011=mc;return;}
void SetOnePad(Bool_t onepad){fOnePad=onepad;return;}
+ void SetppLowEn2011(Bool_t opt){fppLowEn2011=opt;return;}
+ void SetPbPb(Bool_t pbpb){fPbPb=pbpb;return;}
+ void SetPCompatTOF(Double_t pTOF){fPCompatTOF=pTOF;return;}
+ void SetTOFdecide(Bool_t tOFdecide){fTOFdecide=tOFdecide;return;}
+ void SetOldPid(Bool_t oldPid){fOldPid=oldPid;return;}
+ void SetPtThresholdTPC(Double_t ptThresholdTPC){fPtThresholdTPC=ptThresholdTPC;return;}
+ void SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
//Getters
Double_t GetSigma(Int_t idet) const{return fnSigma[idet];}
Double_t GetTofSigma() const{return fTOFSigma;}
- void GetPriors(Double_t *priors) const{priors=fPriors;return;}
- void GetPLimit(Double_t *plim) const{plim=fPLimit;}
+ //void GetPriors(Double_t *priors) const{priors=fPriors;return;}
+ //void GetPLimit(Double_t *plim) const{plim=fPLimit;}
+ void GetPriors(Double_t *priors) const{for(Int_t i=0;i<fnPriors;i++){priors[i]=fPriors[i];}return;}
+ void GetPLimit(Double_t *plim) const{for(Int_t i=0;i<fnPLimit;i++){plim[i]=fPLimit[i];}return;}
Bool_t GetAsym() const{return fAsym;}
Bool_t GetTPC() const{return fTPC;}
Bool_t GetTOF() const{return fTOF;}
Int_t GetMatch() const{return fMatch;}
Bool_t GetCompat() const{return fCompat;}
Bool_t GetMC() const{return fMC;}
+ Bool_t GetOnePad() const{return fOnePad;}
+ Bool_t GetppLowEn2011() const {return fppLowEn2011;}
+ Bool_t GetMCLowEn2011() const {return fMCLowEn2011;}
+ Bool_t GetPbPb() const{return fPbPb;}
+ Bool_t GetTOFdecide() const{return fTOFdecide;}
+ Double_t GetPCompatTOF() const{return fPCompatTOF;}
+ Double_t GetnSigmaCompatTPC() const{return fnSigmaCompat[0];}
+ Double_t GetnSigmaCompatTOF() const{return fnSigmaCompat[1];}
+ Bool_t GetOldPid(){return fOldPid;}
+ Double_t GetPtThresholdTPC(){return fPtThresholdTPC;}
+ AliPIDResponse *GetPidResponse() const {return fPidResponse;}
+ AliPIDCombined *GetPidCombined() const {return fPidCombined;}
Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
+ Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);
+
+ void SetBetheBloch(AliTPCPIDResponse &tpcResp) const;
+ // method for AliPIDCombined object
+ void SetSelectedSpecies(Int_t ispecies = AliPID::kSPECIES){GetPidCombined()->SetSelectedSpecies(ispecies);};
+ void SetPriorDistribution(AliPID::EParticleType type,TH1F *prior);
+ void DrawPrior(AliPID::EParticleType type);
protected:
void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) const;
- void SetBetheBloch(AliTPCPIDResponse tpcResp) const;
private:
Int_t fnNSigma; // number of sigmas
- Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
+ Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
Double_t fTOFSigma; // TOF precision
Int_t fnPriors; //number of priors
Double_t *fPriors; // [fnPriors], set of priors
Bool_t fTRD; // switch to include or exclude TRD
Int_t fMatch; //switch to combine the info from more detectors: 1 = || , 2 = &, 3 = p region
Bool_t fCompat; // compatibility region : useful only if fMatch=1
+ Double_t fPCompatTOF; // compatibility p limit for TOF
+ Int_t fnNSigmaCompat; // number of sigmas
+ Double_t *fnSigmaCompat; //[fnNSigmaCompat] 0: n sigma for TPC compatibility band, 1: for TOF
Bool_t fMC; // MC(kTRUE) or real data (kFALSE, default option)
Bool_t fOnePad; // real data with one pad clusters
-
+ Bool_t fMCLowEn2011; // MC for low energy MC
+ Bool_t fppLowEn2011; // Data for low energy pp 2011
+ Bool_t fPbPb; // real data PbPb
+ Bool_t fTOFdecide; // real data PbPb
+ Bool_t fOldPid; // old PID method implemented
+ Double_t fPtThresholdTPC; // pT threshold to use TPC PID
+ AliPIDResponse *fPidResponse; //pid response
+ AliPIDCombined* fPidCombined; //combined PID object
- ClassDef(AliAODPidHF,5) // AliAODPid for heavy flavor PID
+ ClassDef(AliAODPidHF,15) // AliAODPid for heavy flavor PID
};
#endif
+