[u/mrichter/AliRoot.git] / ITS / AliITSPid.h

#ifndef ALIITSPID_H
#define ALIITSPID_H

#include <TObject.h>

class TClonesArray;
class TVector;
class AliITSIOTrack;
class AliKalmanTrack;
class AliITStrackV2;
class TF1;


class  AliITSPid : public TObject {

public:
		AliITSPid(Int_t ntrs=1000);
		virtual ~AliITSPid(){}

	void	SetEdep(Int_t track,Float_t Edep);
	void	SetPmom(Int_t track,Float_t Pmom);
	void	SetPcod(Int_t track,Int_t Pcod);
	void	Print(Int_t track);
        virtual void Print(Option_t *option="") const {TObject::Print(option);}
	void	Tab(void);
	void    Reset(void);
	void	SetVec(Int_t track,TVector info);
	TVector* GetVec(Int_t track) const;
	Int_t	GetPcode(TClonesArray* rps,Float_t pm);
	Int_t	GetPcode(Float_t p,Float_t pm);
	Int_t   GetPcode(AliKalmanTrack* track);
        Int_t   GetPcode(AliITSIOTrack* track); 
        Int_t   GetPcode(AliITStrackV2* track);
	void	SetCut(Int_t n,Float_t pm,Float_t pilo,Float_t pihi,
		       Float_t klo,Float_t khi,Float_t plo,Float_t phi);
	void    SetAProb(Int_t ivar,Int_t icut,Float_t apro){ fAprob[ivar][icut]=apro; } 
	Float_t GetAProb(Int_t ivar,Int_t icut) const { return fAprob[ivar][icut]; } 
	Float_t GetWpi() const {return fWpi;}
	Float_t GetWk() const {return fWk;}
	Float_t GetWp() const {return fWp;}
	Int_t	GetPid() const {return fPcode;};
protected:
	// copy constructor and assignment operator are protected
	// since they are not allowed
        AliITSPid(const AliITSPid &source); // copy constructor. 
        AliITSPid& operator=(const AliITSPid& /* source */); // = operator.

	int	Qcomp(Float_t* qa,Float_t* qb) const {return qa[0]>qb[0]?1:0;}
	Float_t Qtrm(Int_t track);
	Float_t Qtrm(Float_t qarr[6],Int_t narr) const;
	Int_t	Wpik(Float_t pm,Float_t q);
        Int_t	Wpikp(Float_t pm,Float_t q);
	Int_t	Pion(){return fWpi=1.,fPcode=211;}
	Int_t	Kaon(){return fWk=1.,fPcode=321;}
	Int_t	Proton(){return fWp=1.,fPcode=2212;}
	//================ Data members ========================
	Float_t fCut[13][7],fAprob[3][8]; //Cuts and prior probs tables
	Int_t       fMxtrs; //Maximum tracks limit
	TClonesArray *fTrs; //Tracks set under investigation
	Float_t fWpi,fWk,fWp; //Probabilities for pions,kaons,protons        
	Float_t fRpik,fRppi,fRpka,fRp; //Signal ratios
	Int_t 	fPcode;  //Particle code
	Float_t fSigmin; // Tuning parameter
        Int_t   fSilent; // Output suppresion flag
	TF1*    fCutKa;  // Pions-kaons cut function
	TF1*    fCutPr;  // Kaons-protons cut function
	TF1*    fGGpi[6];// Pions signal parametrization for Hijing
	TF1*    fGGka[3];// Kaons          --//--
        TF1*    fGGpr[3];// Protons        --//--
	TF1*    fggpi;   // Pions signal for given momentum
	TF1*    fggka;   // Kaons          --//--
	TF1*    fggpr;   // Protons        --//--
  ClassDef(AliITSPid,2) // Class for ITS PID
};

#endif
Commit	Line	Data
23efe5f1	1	#ifndef ALIITSPID_H
23efe5f1	2	#define ALIITSPID_H
e75c69cd	3
23efe5f1	4	#include <TObject.h>
e75c69cd	5
	6	class TClonesArray;
	7	class TVector;
8057c644	8	class AliITSIOTrack;
5aa865de	9	class AliKalmanTrack;
e75c69cd	10	class AliITStrackV2;
	11	class TF1;
	12
8057c644	13
8057c644	14	class AliITSPid : public TObject {
23efe5f1	15
	16	public:
	17	AliITSPid(Int_t ntrs=1000);
	18	virtual ~AliITSPid(){}
8057c644	19
23efe5f1	20	void SetEdep(Int_t track,Float_t Edep);
	21	void SetPmom(Int_t track,Float_t Pmom);
	22	void SetPcod(Int_t track,Int_t Pcod);
	23	void Print(Int_t track);
d2f55a22	24	virtual void Print(Option_t *option="") const {TObject::Print(option);}
23efe5f1	25	void Tab(void);
	26	void Reset(void);
	27	void SetVec(Int_t track,TVector info);
8057c644	28	TVector* GetVec(Int_t track) const;
e75c69cd	29	Int_t GetPcode(TClonesArray* rps,Float_t pm);
e75c69cd	30	Int_t GetPcode(Float_t p,Float_t pm);
5aa865de	31	Int_t GetPcode(AliKalmanTrack* track);
e75c69cd	32	Int_t GetPcode(AliITSIOTrack* track);
	33	Int_t GetPcode(AliITStrackV2* track);
	34	void SetCut(Int_t n,Float_t pm,Float_t pilo,Float_t pihi,
	35	Float_t klo,Float_t khi,Float_t plo,Float_t phi);
	36	void SetAProb(Int_t ivar,Int_t icut,Float_t apro){ fAprob[ivar][icut]=apro; }
8057c644	37	Float_t GetAProb(Int_t ivar,Int_t icut) const { return fAprob[ivar][icut]; }
	38	Float_t GetWpi() const {return fWpi;}
	39	Float_t GetWk() const {return fWk;}
	40	Float_t GetWp() const {return fWp;}
	41	Int_t GetPid() const {return fPcode;};
23efe5f1	42	protected:
8057c644	43	// copy constructor and assignment operator are protected
	44	// since they are not allowed
	45	AliITSPid(const AliITSPid &source); // copy constructor.
	46	AliITSPid& operator=(const AliITSPid& /* source */); // = operator.
e75c69cd	47
8057c644	48	int Qcomp(Float_t* qa,Float_t* qb) const {return qa[0]>qb[0]?1:0;}
e75c69cd	49	Float_t Qtrm(Int_t track);
8057c644	50	Float_t Qtrm(Float_t qarr[6],Int_t narr) const;
e75c69cd	51	Int_t Wpik(Float_t pm,Float_t q);
	52	Int_t Wpikp(Float_t pm,Float_t q);
	53	Int_t Pion(){return fWpi=1.,fPcode=211;}
	54	Int_t Kaon(){return fWk=1.,fPcode=321;}
	55	Int_t Proton(){return fWp=1.,fPcode=2212;}
8057c644	56	//================ Data members ========================
	57	Float_t fCut[13][7],fAprob[3][8]; //Cuts and prior probs tables
	58	Int_t fMxtrs; //Maximum tracks limit
	59	TClonesArray *fTrs; //Tracks set under investigation
	60	Float_t fWpi,fWk,fWp; //Probabilities for pions,kaons,protons
	61	Float_t fRpik,fRppi,fRpka,fRp; //Signal ratios
	62	Int_t fPcode; //Particle code
	63	Float_t fSigmin; // Tuning parameter
	64	Int_t fSilent; // Output suppresion flag
	65	TF1* fCutKa; // Pions-kaons cut function
	66	TF1* fCutPr; // Kaons-protons cut function
	67	TF1* fGGpi[6];// Pions signal parametrization for Hijing
	68	TF1* fGGka[3];// Kaons --//--
	69	TF1* fGGpr[3];// Protons --//--
	70	TF1* fggpi; // Pions signal for given momentum
	71	TF1* fggka; // Kaons --//--
	72	TF1* fggpr; // Protons --//--
e6727e13	73	ClassDef(AliITSPid,2) // Class for ITS PID
23efe5f1	74	};
	75
	76	#endif
	77
	78
	79
	80