[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 "AliAODPid.h"
#include "AliAODTrack.h"
#include "AliPIDResponse.h"
#include "AliPIDCombined.h"

class AliAODPidHF : public AliAODPid{

 public:

 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 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 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 SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
 
 //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;
 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;}
 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;
 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 BayesianProbability(AliAODTrack *track,Double_t *pid) 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);

 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);

 protected:

 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;
 void BayesianProbabilityITS(AliAODTrack *track,Double_t *prob) const;
 void BayesianProbabilityTPC(AliAODTrack *track,Double_t *prob) const;
 void BayesianProbabilityTOF(AliAODTrack *track,Double_t *prob) const;
 void BayesianProbabilityTRD(AliAODTrack *track,Double_t *prob) 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 fTOFSigma; // TOF precision 
 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  
 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 

 AliTPCPIDResponse* fTPCResponse; // TPC response 

 ClassDef(AliAODPidHF,16) // 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	////***********************************************************
	14
	15	#include "AliAODPid.h"
	16	#include "AliAODTrack.h"
b79bfc3e	17	#include "AliPIDResponse.h"
11690a06	18	#include "AliPIDCombined.h"
7ce8802c	19
228f3aba	20	class AliAODPidHF : public AliAODPid{
7ce8802c	21
	22	public:
	23
	24	AliAODPidHF();
	25	AliAODPidHF(const AliAODPidHF& pid);
	26	AliAODPidHF& operator=(const AliAODPidHF& pid);
	27	virtual ~AliAODPidHF();
	28
	29	//Setters
79f3c225	30	void SetSigma(Double_t *sigma){
	31	for(Int_t i=0; i<fnNSigma; i++) fnSigma[i]=sigma[i];
	32	}
228f3aba	33	void SetSigma(Int_t idet,Double_t sigma){fnSigma[idet]=sigma;return;}
79f3c225	34	void SetSigmaForTPC(Double_t *sigma){for(Int_t i=0;i<3;i++) fnSigma[i]=sigma[i];return;}
ad42e35b	35	void SetSigmaForTPCCompat(Double_t sigma){fnSigmaCompat[0]=sigma;return;}
ad42e35b	36	void SetSigmaForTOFCompat(Double_t sigma){fnSigmaCompat[1]=sigma;return;}
79f3c225	37	void SetSigmaForTOF(Double_t sigma){fnSigma[3]=sigma;return;}
79f3c225	38	void SetSigmaForITS(Double_t sigma){fnSigma[4]=sigma;return;}
228f3aba	39	void SetTofSigma(Double_t sigma){fTOFSigma=sigma;return;}
7ce8802c	40	void SetPriors(Double_t *priors){fPriors=priors;return;}
79f3c225	41	void SetPLimit(Double_t *plim){for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
79f3c225	42	void SetPLimit(Double_t *plim,Int_t npLim){fnPLimit=npLim;for(Int_t i=0;i<fnPLimit;i++) fPLimit[i]=plim[i];return;}
228f3aba	43	void SetAsym(Bool_t asym){fAsym=asym;return;}
cdac45d3	44	void SetTPC(Bool_t tpc){fTPC=tpc;return;}
	45	void SetTOF(Bool_t tof){fTOF=tof;return;}
	46	void SetITS(Bool_t its){fITS=its;return;}
	47	void SetTRD(Bool_t trd){fTRD=trd;return;}
79f3c225	48	void SetMatch(Int_t match){fMatch=match;return;}
cdac45d3	49	void SetCompat(Bool_t comp){fCompat=comp;return;}
913d1957	50	void SetMC(Bool_t mc){fMC=mc;return;}
5821ea5b	51	void SetMClowenpp2011(Bool_t mc){fMCLowEn2011=mc;return;}
12141139	52	void SetOnePad(Bool_t onepad){fOnePad=onepad;return;}
67a82ef2	53	void SetppLowEn2011(Bool_t opt){fppLowEn2011=opt;return;}
17c4a8d5	54	void SetPbPb(Bool_t pbpb){fPbPb=pbpb;return;}
ad42e35b	55	void SetPCompatTOF(Double_t pTOF){fPCompatTOF=pTOF;return;}
439d61b7	56	void SetTOFdecide(Bool_t tOFdecide){fTOFdecide=tOFdecide;return;}
b79bfc3e	57	void SetOldPid(Bool_t oldPid){fOldPid=oldPid;return;}
2c9eefd4	58	void SetPtThresholdTPC(Double_t ptThresholdTPC){fPtThresholdTPC=ptThresholdTPC;return;}
b79bfc3e	59	void SetPidResponse(AliPIDResponse *pidResp) {fPidResponse=pidResp;return;}
7ce8802c	60
79f3c225	61	//Getters
451c9a4e	62
	63	Int_t GetnSigmaTPC(AliAODTrack *track, Int_t species, Double_t &sigma) const;
	64	Int_t GetnSigmaTOF(AliAODTrack *track, Int_t species, Double_t &sigma) const;
1cb0f658	65	Double_t GetSigma(Int_t idet) const{return fnSigma[idet];}
1cb0f658	66	Double_t GetTofSigma() const{return fTOFSigma;}
4fc75076	67	//void GetPriors(Double_t *priors) const{priors=fPriors;return;}
	68	//void GetPLimit(Double_t *plim) const{plim=fPLimit;}
	69	void GetPriors(Double_t *priors) const{for(Int_t i=0;i<fnPriors;i++){priors[i]=fPriors[i];}return;}
	70	void GetPLimit(Double_t *plim) const{for(Int_t i=0;i<fnPLimit;i++){plim[i]=fPLimit[i];}return;}
1cb0f658	71	Bool_t GetAsym() const{return fAsym;}
	72	Bool_t GetTPC() const{return fTPC;}
	73	Bool_t GetTOF() const{return fTOF;}
	74	Bool_t GetITS() const{return fITS;}
	75	Bool_t GetTRD() const{return fTRD;}
	76	Int_t GetMatch() const{return fMatch;}
	77	Bool_t GetCompat() const{return fCompat;}
913d1957	78	Bool_t GetMC() const{return fMC;}
17c4a8d5	79	Bool_t GetOnePad() const{return fOnePad;}
67a82ef2	80	Bool_t GetppLowEn2011() const {return fppLowEn2011;}
5821ea5b	81	Bool_t GetMCLowEn2011() const {return fMCLowEn2011;}
17c4a8d5	82	Bool_t GetPbPb() const{return fPbPb;}
439d61b7	83	Bool_t GetTOFdecide() const{return fTOFdecide;}
ad42e35b	84	Double_t GetPCompatTOF() const{return fPCompatTOF;}
	85	Double_t GetnSigmaCompatTPC() const{return fnSigmaCompat[0];}
	86	Double_t GetnSigmaCompatTOF() const{return fnSigmaCompat[1];}
b79bfc3e	87	Bool_t GetOldPid(){return fOldPid;}
2c9eefd4	88	Double_t GetPtThresholdTPC(){return fPtThresholdTPC;}
11690a06	89	AliPIDResponse *GetPidResponse() const {return fPidResponse;}
11690a06	90	AliPIDCombined *GetPidCombined() const {return fPidCombined;}
7ce8802c	91
228f3aba	92	Int_t RawSignalPID (AliAODTrack *track, TString detector) const;
	93	Bool_t IsKaonRaw (AliAODTrack *track, TString detector) const;
	94	Bool_t IsPionRaw (AliAODTrack *track, TString detector) const;
	95	Bool_t IsProtonRaw (AliAODTrack *track, TString detector) const;
	96	Bool_t IsElectronRaw (AliAODTrack *track, TString detector) const;
cdac45d3	97	void BayesianProbability(AliAODTrack track,Double_t pid) const;
228f3aba	98	void CombinedProbability(AliAODTrack track,Bool_t type) const; //0 = pion, 1 = kaon, 2 = proton
	99	Bool_t CheckStatus(AliAODTrack *track,TString detectors) const;
	100
70e398d3	101	Bool_t CheckITSPIDStatus(AliAODTrack *track) const;
	102	Bool_t CheckTPCPIDStatus(AliAODTrack *track) const;
	103	Bool_t CheckTOFPIDStatus(AliAODTrack *track) const;
	104	Bool_t CheckTRDPIDStatus(AliAODTrack *track) const;
	105
228f3aba	106	Bool_t TPCRawAsym(AliAODTrack* track,Int_t specie) const;
70e398d3	107	Int_t MatchTPCTOF(AliAODTrack *track,Int_t specie);
228f3aba	108
cdac45d3	109	Int_t MakeRawPid(AliAODTrack *track,Int_t specie); //general method to perform PID using raw signals
cdac45d3	110
d50b25df	111	Bool_t IsTOFPiKexcluded(AliAODTrack *track,Double_t nsigmaK);
d50b25df	112
70e398d3	113	void GetTPCBetheBlochParams(Double_t alephParameters[5]) const;
70e398d3	114	void SetBetheBloch();
64e39959	115	// method for AliPIDCombined object
70e398d3	116	void SetSelectedSpecies(Int_t ispecies = AliPID::kSPECIES){GetPidCombined()->SetSelectedSpecies(ispecies);};
	117	void SetPriorDistribution(AliPID::EParticleType type,TH1F *prior);
	118	void DrawPrior(AliPID::EParticleType type);
64e39959	119
7ce8802c	120	protected:
7ce8802c	121
228f3aba	122	Int_t ApplyPidTPCRaw(AliAODTrack *track,Int_t specie) const;
	123	Int_t ApplyPidTOFRaw(AliAODTrack *track,Int_t specie) const;
	124	Int_t ApplyPidITSRaw(AliAODTrack *track,Int_t specie) const;
	125	void BayesianProbabilityITS(AliAODTrack track,Double_t prob) const;
	126	void BayesianProbabilityTPC(AliAODTrack track,Double_t prob) const;
	127	void BayesianProbabilityTOF(AliAODTrack track,Double_t prob) const;
	128	void BayesianProbabilityTRD(AliAODTrack track,Double_t prob) const;
7ce8802c	129
7ce8802c	130	private:
cdac45d3	131	Int_t fnNSigma; // number of sigmas
ad42e35b	132	Double_t *fnSigma; // [fnNSigma], sigma for the raw signal PID: 0-2 for TPC, 3 for TOF, 4 for ITS
228f3aba	133	Double_t fTOFSigma; // TOF precision
cdac45d3	134	Int_t fnPriors; //number of priors
	135	Double_t *fPriors; // [fnPriors], set of priors
	136	Int_t fnPLimit; //number of Plimit
	137	Double_t *fPLimit; // [fnPLimit], limit of p intervals for asimmetric PID: fPLimit<p[0], fPLimit[0]<p<fPLimit[1], p>fPLimit[1]
228f3aba	138	Bool_t fAsym; // asimmetric PID required
cdac45d3	139	Bool_t fTPC; //switch to include or exclude TPC
	140	Bool_t fTOF; // switch to include or exclude TOF
	141	Bool_t fITS; //switch to include or exclude ITS
	142	Bool_t fTRD; // switch to include or exclude TRD
79f3c225	143	Int_t fMatch; //switch to combine the info from more detectors: 1 = \|\| , 2 = &, 3 = p region
cdac45d3	144	Bool_t fCompat; // compatibility region : useful only if fMatch=1
ad42e35b	145	Double_t fPCompatTOF; // compatibility p limit for TOF
02baac36	146	Int_t fnNSigmaCompat; // number of sigmas
02baac36	147	Double_t *fnSigmaCompat; //[fnNSigmaCompat] 0: n sigma for TPC compatibility band, 1: for TOF
913d1957	148	Bool_t fMC; // MC(kTRUE) or real data (kFALSE, default option)
12141139	149	Bool_t fOnePad; // real data with one pad clusters
5821ea5b	150	Bool_t fMCLowEn2011; // MC for low energy MC
67a82ef2	151	Bool_t fppLowEn2011; // Data for low energy pp 2011
17c4a8d5	152	Bool_t fPbPb; // real data PbPb
439d61b7	153	Bool_t fTOFdecide; // real data PbPb
b79bfc3e	154	Bool_t fOldPid; // old PID method implemented
2c9eefd4	155	Double_t fPtThresholdTPC; // pT threshold to use TPC PID
b79bfc3e	156	AliPIDResponse *fPidResponse; //pid response
11690a06	157	AliPIDCombined* fPidCombined; //combined PID object
7ce8802c	158
70e398d3	159	AliTPCPIDResponse* fTPCResponse; // TPC response
7ce8802c	160
70e398d3	161	ClassDef(AliAODPidHF,16) // AliAODPid for heavy flavor PID
7ce8802c	162
	163	};
	164
	165	#endif
d50b25df	166