[u/mrichter/AliRoot.git] / PWGHF / vertexingHF / AliAODPidHF.h

#ifndef ALIAODPIDHF_H
#define ALIAODPIDHF_H

/* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */ 

//***********************************************************
//// Class AliAODPidHF
//// class for PID with AliAODRecoDecayHF 
//// 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
////***********************************************************

#include <TString.h>
#include <TH1F.h>
#include <TObject.h>
#include "AliAODTrack.h"
#include "AliPIDResponse.h"
#include "AliPIDCombined.h"
#include "AliPID.h"

class AliAODPidHF : public TObject{

 public:

  enum ECombDetectors {
    kTPC,
    kTOF,
    kTPCTOF,
    kTPCITS,
    kTPCAndTOF
  };

  AliAODPidHF();
  AliAODPidHF(const AliAODPidHF& pid);
  AliAODPidHF& operator=(const AliAODPidHF& pid);
  virtual ~AliAODPidHF();

  //Setters
  void SetSigma(Double_t *sigma){
    for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
  }
  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 SetCutOnTOFmismatchProb(Double_t cut=0.01){fCutTOFmismatch=cut;}
  void DisableCutOnTOFmismatchProb(){fCutTOFmismatch=999.;}

  void SetMinNClustersTPCPID(Int_t minc) {fMinNClustersTPCPID=minc;}

  void SetCombinednSigmaCutForPiKP(Float_t sigpi, Float_t sigk, Float_t sigp){
    fMaxnSigmaCombined[0]=sigpi; 
    fMaxnSigmaCombined[1]=sigk;
    fMaxnSigmaCombined[2]=sigp;
  }
  void SetTPCnSigmaRangeForPions(Float_t smin, Float_t smax){
    fMinnSigmaTPC[0]=smin;
    fMaxnSigmaTPC[0]=smax;
  }
  void SetTOFnSigmaRangeForPions(Float_t smin, Float_t smax){
    fMinnSigmaTOF[0]=smin;
    fMaxnSigmaTOF[0]=smax;
  }
  void SetTPCnSigmaRangeForKaons(Float_t smin, Float_t smax){
    fMinnSigmaTPC[1]=smin;
    fMaxnSigmaTPC[1]=smax;
  }
  void SetTOFnSigmaRangeForKaons(Float_t smin, Float_t smax){
    fMinnSigmaTOF[1]=smin;
    fMaxnSigmaTOF[1]=smax;
  }
  void SetTPCnSigmaRangeForProtons(Float_t smin, Float_t smax){
    fMinnSigmaTPC[2]=smin;
    fMaxnSigmaTPC[2]=smax;
  }
  void SetTOFnSigmaRangeForProtons(Float_t smin, Float_t smax){
    fMinnSigmaTOF[2]=smin;
    fMaxnSigmaTOF[2]=smax;
  }

  void SetPriors(Double_t *priors){fPriors=priors;return;}
  void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
  void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
  void SetAsym(Bool_t asym){fAsym=asym;return;}
  void SetUseAsymmnSigmaTOF(Double_t nsmin, Double_t nsmax, Double_t nscompmin, Double_t nscompmax){
    fUseAsymTOF=kTRUE; 
    fLownSigmaTOF=nsmin; fUpnSigmaTOF=nsmax;
    fLownSigmaCompatTOF=nscompmin; fUpnSigmaCompatTOF=nscompmax;
  }
  void SetTPC(Bool_t tpc){fTPC=tpc;return;}
  void SetTOF(Bool_t tof){fTOF=tof;return;}
  void SetITS(Bool_t its){fITS=its;return;}
  void SetTRD(Bool_t trd){fTRD=trd;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 SetMaxTrackMomForCombinedPID(Double_t mom){fMaxTrackMomForCombinedPID=mom;}
  void SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
  void SetCombDetectors(ECombDetectors pidComb) {
    fCombDetectors=pidComb;
  }
  void SetPionPriorHisto(TH1F* histo){
    if(fPriorsH[AliPID::kPion]) delete fPriorsH[AliPID::kPion];
    fPriorsH[AliPID::kPion] = new TH1F(*histo);
  }
  void SetKaonPriorHisto(TH1F* histo){
    if(fPriorsH[AliPID::kKaon]) delete fPriorsH[AliPID::kKaon];
    fPriorsH[AliPID::kKaon] = new TH1F(*histo);
  }
  void SetProtonPriorHisto(TH1F* histo){
    if(fPriorsH[AliPID::kProton]) delete fPriorsH[AliPID::kProton];
    fPriorsH[AliPID::kProton] = new TH1F(*histo);
  }

 
  //Getters
 
  Int_t GetnSigmaTPC(AliAODTrack *track, Int_t species, Double_t &sigma) const;
  Int_t GetnSigmaTOF(AliAODTrack *track, Int_t species, Double_t &sigma) const;
  Int_t GetnSigmaITS(AliAODTrack *track, Int_t species, Double_t &sigma) const;
  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{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;}
  Bool_t GetITS() const{return fITS;}
  Bool_t GetTRD() const{return fTRD;}
  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;}
  Double_t GetMaxTrackMomForCombinedPID(){return fMaxTrackMomForCombinedPID;}
  AliPIDResponse *GetPidResponse() const {return fPidResponse;}
  AliPIDCombined *GetPidCombined() const {return fPidCombined;}
  ECombDetectors GetCombDetectors() const {
    return fCombDetectors;
  }
  Bool_t GetUseCombined() {return fUseCombined;}
  Bool_t GetDefaultPriors() {return fDefaultPriors;}
 
  Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
  Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
  Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
  void CombinedProbability(AliAODTrack *track,Bool_t *type) const; //0 = pion, 1 = kaon, 2 = proton
  Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;

  Bool_t CheckITSPIDStatus(AliAODTrack *track) const;
  Bool_t CheckTPCPIDStatus(AliAODTrack *track) const;
  Bool_t CheckTOFPIDStatus(AliAODTrack *track) const;
  Bool_t CheckTRDPIDStatus(AliAODTrack *track) const;

  Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
  Int_t MatchTPCTOF(AliAODTrack *track,Int_t specie);

  Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals

  Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);

  Bool_t IsExcluded(AliAODTrack *track, Int_t labelTrack, Double_t nsigmaCut, TString detector);

  void GetTPCBetheBlochParams(Double_t alephParameters[5]) const;
  void SetBetheBloch();
  // 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);
  void SetPriorsHistos(TString priorFileName);
  void SetUpCombinedPID();
  void SetUseCombined(Bool_t useCombined=kTRUE) {fUseCombined=useCombined;}
  void SetUseDefaultPriors(Bool_t defaultP)	    {fDefaultPriors=defaultP;}
  Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
  Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
  Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
  Int_t ApplyTOFCompatibilityBand(AliAODTrack *track,Int_t specie) const;

  void PrintAll() const;

 protected:


 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 fTOFSigma; // TOF precision 
  Double_t fCutTOFmismatch; // Cut of TOF mismatch probability
  UInt_t fMinNClustersTPCPID;       // Minimum TPC PID clusters cut
  Int_t fnPriors; //number of priors
  Double_t *fPriors; // [fnPriors], set of priors
  Int_t fnPLimit; //number of Plimit
  Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
  Bool_t fAsym; // asimmetric PID required
  Bool_t fTPC; //switch to include or exclude TPC 
  Bool_t fTOF; // switch to include or exclude TOF
  Bool_t fITS; //switch to include or exclude ITS
  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  
  Bool_t fUseAsymTOF; // flag for using asymmetrig nSigmaCut in TOF for fMatch==1
  Double_t fLownSigmaTOF;  // lower nsigma TOF (for fUseAsymTOF)
  Double_t fUpnSigmaTOF;  // upper nsigma TOF (for fUseAsymTOF)
  Double_t fLownSigmaCompatTOF;  // lower nsigma TOF (for fUseAsymTOF)
  Double_t fUpnSigmaCompatTOF;  // upper nsigma TOF (for fUseAsymTOF)
  Int_t fnNSigmaCompat; // number of sigmas
  Double_t *fnSigmaCompat; //[fnNSigmaCompat]  0: n sigma for TPC compatibility band, 1: for TOF  
  Double_t fMaxnSigmaCombined[3];  // nSigma cut for pi,K,p (TPC^2+TOF^2)
  Double_t fMinnSigmaTPC[3]; //min. of nSigma range for pi,K,p in TPC (match==5)
  Double_t fMaxnSigmaTPC[3]; //max. of nSigma range for pi,K,p in TPC (match==5)
  Double_t fMinnSigmaTOF[3]; //min. of nSigma range for pi,K,p in TOF (match==5)
  Double_t fMaxnSigmaTOF[3]; //max. of nSigma range for pi,K,p in TOF (match==5)
  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
  Double_t fMaxTrackMomForCombinedPID; // momentum threshold to use PID
  AliPIDResponse *fPidResponse; //! pid response
  AliPIDCombined* fPidCombined; //! combined PID object 

  AliTPCPIDResponse* fTPCResponse; //! TPC response 

  TH1F* fPriorsH[AliPID::kSPECIES]; // priors histos
  ECombDetectors fCombDetectors; // detectors to be involved for combined PID
  Bool_t fUseCombined; // detectors to be involved for combined PID
  Bool_t fDefaultPriors; // use default priors for combined PID

  ClassDef(AliAODPidHF,22) // AliAODPid for heavy flavor PID

    };

#endif
Commit	Line	Data
7ce8802c	1	#ifndef ALIAODPIDHF_H
	2	#define ALIAODPIDHF_H
	3
	4	/* Copyright(c) 1998-2006, ALICE Experiment at CERN, All rights reserved. *
27de2dfb	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id$ */
7ce8802c	8
	9	//***********************************************************
	10	//// Class AliAODPidHF
	11	//// class for PID with AliAODRecoDecayHF
228f3aba	12	//// 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
7ce8802c	13	////***********************************************************
7ce8802c	14
4c177f98	15	#include <TString.h>
4c177f98	16	#include <TH1F.h>
fcff8ce7	17	#include <TObject.h>
7ce8802c	18	#include "AliAODTrack.h"
b79bfc3e	19	#include "AliPIDResponse.h"
11690a06	20	#include "AliPIDCombined.h"
4c177f98	21	#include "AliPID.h"
7ce8802c	22
fcff8ce7	23	class AliAODPidHF : public TObject{
7ce8802c	24
	25	public:
	26
fcff8ce7	27	enum ECombDetectors {
	28	kTPC,
	29	kTOF,
	30	kTPCTOF,
	31	kTPCITS,
	32	kTPCAndTOF
	33	};
	34
	35	AliAODPidHF();
	36	AliAODPidHF(const AliAODPidHF& pid);
	37	AliAODPidHF& operator=(const AliAODPidHF& pid);
	38	virtual ~AliAODPidHF();
	39
	40	//Setters
	41	void SetSigma(Double_t *sigma){
79f3c225	42	for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
4c177f98	43	}
fcff8ce7	44	void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
	45	void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
	46	void SetSigmaForTPCCompat(Double_t sigma){fnSigmaCompat[0]=sigma;return;}
	47	void SetSigmaForTOFCompat(Double_t sigma){fnSigmaCompat[1]=sigma;return;}
	48	void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
	49	void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
	50	void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
	51
	52	void SetCutOnTOFmismatchProb(Double_t cut=0.01){fCutTOFmismatch=cut;}
	53	void DisableCutOnTOFmismatchProb(){fCutTOFmismatch=999.;}
	54
	55	void SetMinNClustersTPCPID(Int_t minc) {fMinNClustersTPCPID=minc;}
	56
	57	void SetCombinednSigmaCutForPiKP(Float_t sigpi, Float_t sigk, Float_t sigp){
	58	fMaxnSigmaCombined[0]=sigpi;
	59	fMaxnSigmaCombined[1]=sigk;
	60	fMaxnSigmaCombined[2]=sigp;
	61	}
	62	void SetTPCnSigmaRangeForPions(Float_t smin, Float_t smax){
	63	fMinnSigmaTPC[0]=smin;
	64	fMaxnSigmaTPC[0]=smax;
	65	}
	66	void SetTOFnSigmaRangeForPions(Float_t smin, Float_t smax){
	67	fMinnSigmaTOF[0]=smin;
	68	fMaxnSigmaTOF[0]=smax;
	69	}
	70	void SetTPCnSigmaRangeForKaons(Float_t smin, Float_t smax){
	71	fMinnSigmaTPC[1]=smin;
	72	fMaxnSigmaTPC[1]=smax;
	73	}
	74	void SetTOFnSigmaRangeForKaons(Float_t smin, Float_t smax){
	75	fMinnSigmaTOF[1]=smin;
	76	fMaxnSigmaTOF[1]=smax;
	77	}
	78	void SetTPCnSigmaRangeForProtons(Float_t smin, Float_t smax){
	79	fMinnSigmaTPC[2]=smin;
	80	fMaxnSigmaTPC[2]=smax;
	81	}
	82	void SetTOFnSigmaRangeForProtons(Float_t smin, Float_t smax){
	83	fMinnSigmaTOF[2]=smin;
	84	fMaxnSigmaTOF[2]=smax;
	85	}
	86
	87	void SetPriors(Double_t *priors){fPriors=priors;return;}
	88	void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
	89	void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
	90	void SetAsym(Bool_t asym){fAsym=asym;return;}
23e088a9	91	void SetUseAsymmnSigmaTOF(Double_t nsmin, Double_t nsmax, Double_t nscompmin, Double_t nscompmax){
	92	fUseAsymTOF=kTRUE;
	93	fLownSigmaTOF=nsmin; fUpnSigmaTOF=nsmax;
	94	fLownSigmaCompatTOF=nscompmin; fUpnSigmaCompatTOF=nscompmax;
	95	}
fcff8ce7	96	void SetTPC(Bool_t tpc){fTPC=tpc;return;}
	97	void SetTOF(Bool_t tof){fTOF=tof;return;}
	98	void SetITS(Bool_t its){fITS=its;return;}
	99	void SetTRD(Bool_t trd){fTRD=trd;return;}
	100	void SetMatch(Int_t match){fMatch=match;return;}
	101	void SetCompat(Bool_t comp){fCompat=comp;return;}
	102	void SetMC(Bool_t mc){fMC=mc;return;}
	103	void SetMClowenpp2011(Bool_t mc){fMCLowEn2011=mc;return;}
	104	void SetOnePad(Bool_t onepad){fOnePad=onepad;return;}
	105	void SetppLowEn2011(Bool_t opt){fppLowEn2011=opt;return;}
	106	void SetPbPb(Bool_t pbpb){fPbPb=pbpb;return;}
	107	void SetPCompatTOF(Double_t pTOF){fPCompatTOF=pTOF;return;}
	108	void SetTOFdecide(Bool_t tOFdecide){fTOFdecide=tOFdecide;return;}
	109	void SetOldPid(Bool_t oldPid){fOldPid=oldPid;return;}
	110	void SetPtThresholdTPC(Double_t ptThresholdTPC){fPtThresholdTPC=ptThresholdTPC;return;}
	111	void SetMaxTrackMomForCombinedPID(Double_t mom){fMaxTrackMomForCombinedPID=mom;}
	112	void SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
	113	void SetCombDetectors(ECombDetectors pidComb) {
	114	fCombDetectors=pidComb;
	115	}
	116	void SetPionPriorHisto(TH1F* histo){
	117	if(fPriorsH[AliPID::kPion]) delete fPriorsH[AliPID::kPion];
	118	fPriorsH[AliPID::kPion] = new TH1F(*histo);
	119	}
	120	void SetKaonPriorHisto(TH1F* histo){
	121	if(fPriorsH[AliPID::kKaon]) delete fPriorsH[AliPID::kKaon];
	122	fPriorsH[AliPID::kKaon] = new TH1F(*histo);
	123	}
	124	void SetProtonPriorHisto(TH1F* histo){
	125	if(fPriorsH[AliPID::kProton]) delete fPriorsH[AliPID::kProton];
	126	fPriorsH[AliPID::kProton] = new TH1F(*histo);
	127	}
03a82f54	128
7ce8802c	129
fcff8ce7	130	//Getters
451c9a4e	131
fcff8ce7	132	Int_t GetnSigmaTPC(AliAODTrack *track, Int_t species, Double_t &sigma) const;
	133	Int_t GetnSigmaTOF(AliAODTrack *track, Int_t species, Double_t &sigma) const;
	134	Int_t GetnSigmaITS(AliAODTrack *track, Int_t species, Double_t &sigma) const;
	135	Double_t GetSigma(Int_t idet) const{return fnSigma[idet];}
	136	Double_t GetTofSigma() const{return fTOFSigma;}
	137	//void GetPriors(Double_t *priors) const{priors=fPriors;return;}
	138	//void GetPLimit(Double_t *plim) const{plim=fPLimit;}
	139	void GetPriors(Double_t *priors) const{for(Int_t i=0;i<fnPriors;i++){priors[i]=fPriors[i];}return;}
	140	void GetPLimit(Double_t *plim) const{for(Int_t i=0;i<fnPLimit;i++){plim[i]=fPLimit[i];}return;}
	141	Bool_t GetAsym() const{return fAsym;}
	142	Bool_t GetTPC() const{return fTPC;}
	143	Bool_t GetTOF() const{return fTOF;}
	144	Bool_t GetITS() const{return fITS;}
	145	Bool_t GetTRD() const{return fTRD;}
	146	Int_t GetMatch() const{return fMatch;}
	147	Bool_t GetCompat() const{return fCompat;}
	148	Bool_t GetMC() const{return fMC;}
	149	Bool_t GetOnePad() const{return fOnePad;}
	150	Bool_t GetppLowEn2011() const {return fppLowEn2011;}
	151	Bool_t GetMCLowEn2011() const {return fMCLowEn2011;}
	152	Bool_t GetPbPb() const{return fPbPb;}
	153	Bool_t GetTOFdecide() const{return fTOFdecide;}
	154	Double_t GetPCompatTOF() const{return fPCompatTOF;}
	155	Double_t GetnSigmaCompatTPC() const{return fnSigmaCompat[0];}
	156	Double_t GetnSigmaCompatTOF() const{return fnSigmaCompat[1];}
	157	Bool_t GetOldPid(){return fOldPid;}
	158	Double_t GetPtThresholdTPC(){return fPtThresholdTPC;}
	159	Double_t GetMaxTrackMomForCombinedPID(){return fMaxTrackMomForCombinedPID;}
	160	AliPIDResponse *GetPidResponse() const {return fPidResponse;}
	161	AliPIDCombined *GetPidCombined() const {return fPidCombined;}
	162	ECombDetectors GetCombDetectors() const {
	163	return fCombDetectors;
4c177f98	164	}
fcff8ce7	165	Bool_t GetUseCombined() {return fUseCombined;}
e51f1625	166	Bool_t GetDefaultPriors() {return fDefaultPriors;}
e51f1625	167
fcff8ce7	168	Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
	169	Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
	170	Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
	171	Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
	172	Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
	173	void CombinedProbability(AliAODTrack track,Bool_t type) const; //0 = pion, 1 = kaon, 2 = proton
	174	Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
228f3aba	175
fcff8ce7	176	Bool_t CheckITSPIDStatus(AliAODTrack *track) const;
	177	Bool_t CheckTPCPIDStatus(AliAODTrack *track) const;
	178	Bool_t CheckTOFPIDStatus(AliAODTrack *track) const;
	179	Bool_t CheckTRDPIDStatus(AliAODTrack *track) const;
70e398d3	180
fcff8ce7	181	Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
fcff8ce7	182	Int_t MatchTPCTOF(AliAODTrack *track,Int_t specie);
228f3aba	183
fcff8ce7	184	Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
cdac45d3	185
fcff8ce7	186	Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);
d50b25df	187
fcff8ce7	188	Bool_t IsExcluded(AliAODTrack *track, Int_t labelTrack, Double_t nsigmaCut, TString detector);
e7af8919	189
fcff8ce7	190	void GetTPCBetheBlochParams(Double_t alephParameters[5]) const;
fcff8ce7	191	void SetBetheBloch();
64e39959	192	// method for AliPIDCombined object
fcff8ce7	193	void SetSelectedSpecies(Int_t ispecies = AliPID::kSPECIES){GetPidCombined()->SetSelectedSpecies(ispecies);};
	194	void SetPriorDistribution(AliPID::EParticleType type,TH1F *prior);
	195	void DrawPrior(AliPID::EParticleType type);
	196	void SetPriorsHistos(TString priorFileName);
	197	void SetUpCombinedPID();
	198	void SetUseCombined(Bool_t useCombined=kTRUE) {fUseCombined=useCombined;}
e51f1625	199	void SetUseDefaultPriors(Bool_t defaultP) {fDefaultPriors=defaultP;}
fcff8ce7	200	Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
	201	Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
	202	Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
23e088a9	203	Int_t ApplyTOFCompatibilityBand(AliAODTrack *track,Int_t specie) const;
e7af8919	204
f35149be	205	void PrintAll() const;
f35149be	206
e7af8919	207	protected:
e7af8919	208
7ce8802c	209
7ce8802c	210	private:
fcff8ce7	211	Int_t fnNSigma; // number of sigmas
	212	Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
	213	Double_t fTOFSigma; // TOF precision
	214	Double_t fCutTOFmismatch; // Cut of TOF mismatch probability
	215	UInt_t fMinNClustersTPCPID; // Minimum TPC PID clusters cut
	216	Int_t fnPriors; //number of priors
	217	Double_t *fPriors; // [fnPriors], set of priors
	218	Int_t fnPLimit; //number of Plimit
	219	Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
	220	Bool_t fAsym; // asimmetric PID required
	221	Bool_t fTPC; //switch to include or exclude TPC
	222	Bool_t fTOF; // switch to include or exclude TOF
	223	Bool_t fITS; //switch to include or exclude ITS
	224	Bool_t fTRD; // switch to include or exclude TRD
	225	Int_t fMatch; //switch to combine the info from more detectors: 1 = \|\| , 2 = &, 3 = p region
	226	Bool_t fCompat; // compatibility region : useful only if fMatch=1
	227	Double_t fPCompatTOF; // compatibility p limit for TOF
23e088a9	228	Bool_t fUseAsymTOF; // flag for using asymmetrig nSigmaCut in TOF for fMatch==1
	229	Double_t fLownSigmaTOF; // lower nsigma TOF (for fUseAsymTOF)
	230	Double_t fUpnSigmaTOF; // upper nsigma TOF (for fUseAsymTOF)
	231	Double_t fLownSigmaCompatTOF; // lower nsigma TOF (for fUseAsymTOF)
	232	Double_t fUpnSigmaCompatTOF; // upper nsigma TOF (for fUseAsymTOF)
fcff8ce7	233	Int_t fnNSigmaCompat; // number of sigmas
	234	Double_t *fnSigmaCompat; //[fnNSigmaCompat] 0: n sigma for TPC compatibility band, 1: for TOF
	235	Double_t fMaxnSigmaCombined[3]; // nSigma cut for pi,K,p (TPC^2+TOF^2)
	236	Double_t fMinnSigmaTPC[3]; //min. of nSigma range for pi,K,p in TPC (match==5)
	237	Double_t fMaxnSigmaTPC[3]; //max. of nSigma range for pi,K,p in TPC (match==5)
	238	Double_t fMinnSigmaTOF[3]; //min. of nSigma range for pi,K,p in TOF (match==5)
	239	Double_t fMaxnSigmaTOF[3]; //max. of nSigma range for pi,K,p in TOF (match==5)
	240	Bool_t fMC; // MC(kTRUE) or real data (kFALSE, default option)
	241	Bool_t fOnePad; // real data with one pad clusters
	242	Bool_t fMCLowEn2011; // MC for low energy MC
	243	Bool_t fppLowEn2011; // Data for low energy pp 2011
	244	Bool_t fPbPb; // real data PbPb
	245	Bool_t fTOFdecide; // real data PbPb
	246	Bool_t fOldPid; // old PID method implemented
	247	Double_t fPtThresholdTPC; // pT threshold to use TPC PID
	248	Double_t fMaxTrackMomForCombinedPID; // momentum threshold to use PID
	249	AliPIDResponse *fPidResponse; //! pid response
	250	AliPIDCombined* fPidCombined; //! combined PID object
	251
	252	AliTPCPIDResponse* fTPCResponse; //! TPC response
	253
	254	TH1F* fPriorsH[AliPID::kSPECIES]; // priors histos
	255	ECombDetectors fCombDetectors; // detectors to be involved for combined PID
	256	Bool_t fUseCombined; // detectors to be involved for combined PID
e51f1625	257	Bool_t fDefaultPriors; // use default priors for combined PID
fcff8ce7	258
23e088a9	259	ClassDef(AliAODPidHF,22) // AliAODPid for heavy flavor PID
fcff8ce7	260
fcff8ce7	261	};
7ce8802c	262
7ce8802c	263	#endif
d50b25df	264