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

#ifndef ALIITSPID_H
#define ALIITSPID_H

#include <TObject.h>
#include "AliITSIOTrack.h"

class TClonesArray;
class TVector;
class AliTPCtrack;
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);
	void	Tab(void);
	void    Reset(void);
	void	SetVec(Int_t track,TVector info);
	TVector* GetVec(Int_t track);
	Int_t	GetPcode(TClonesArray* rps,Float_t pm);
	Int_t	GetPcode(Float_t p,Float_t pm);
	Int_t   GetPcode(AliTPCtrack* 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){ return fAprob[ivar][icut]; } 
	Float_t GetWpi(){return fWpi;}
	Float_t GetWk(){return fWk;}
	Float_t GetWp(){return fWp;}
	Int_t	GetPid(){return fPcode;};
protected:
	Float_t fCut[13][7],fAprob[3][8];
	Int_t       fMxtrs;
	TClonesArray *fTrs;
	Float_t fWpi,fWk,fWp;
	Float_t fRpik,fRppi,fRpka,fRp; 
	Int_t 	fPcode;

	int	Qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
	Float_t Qtrm(Int_t track);
	Float_t Qtrm(Float_t qarr[6],Int_t narr);
	Float_t fSigmin;
	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;}
        Int_t   fSilent;
	TF1*    fCutKa;
	TF1*    fCutPr;
	TF1*    fGGpi[6];
	TF1*    fGGka[3];
        TF1*    fGGpr[3];
	TF1*    fggpi;
	TF1*    fggka;
	TF1*    fggpr;

  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>
79b921e1	5	#include "AliITSIOTrack.h"
e75c69cd	6
	7	class TClonesArray;
	8	class TVector;
	9	class AliTPCtrack;
	10	class AliITStrackV2;
	11	class TF1;
	12
23efe5f1	13	//___________________________________________________________________________
	14	class AliITSPid :
	15	public TObject {
	16
	17	public:
	18	AliITSPid(Int_t ntrs=1000);
	19	virtual ~AliITSPid(){}
	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);
	24	void Tab(void);
	25	void Reset(void);
	26	void SetVec(Int_t track,TVector info);
	27	TVector* GetVec(Int_t track);
e75c69cd	28	Int_t GetPcode(TClonesArray* rps,Float_t pm);
	29	Int_t GetPcode(Float_t p,Float_t pm);
	30	Int_t GetPcode(AliTPCtrack* track);
	31	Int_t GetPcode(AliITSIOTrack* track);
	32	Int_t GetPcode(AliITStrackV2* track);
	33	void SetCut(Int_t n,Float_t pm,Float_t pilo,Float_t pihi,
	34	Float_t klo,Float_t khi,Float_t plo,Float_t phi);
	35	void SetAProb(Int_t ivar,Int_t icut,Float_t apro){ fAprob[ivar][icut]=apro; }
	36	Float_t GetAProb(Int_t ivar,Int_t icut){ return fAprob[ivar][icut]; }
23efe5f1	37	Float_t GetWpi(){return fWpi;}
	38	Float_t GetWk(){return fWk;}
	39	Float_t GetWp(){return fWp;}
	40	Int_t GetPid(){return fPcode;};
	41	protected:
e75c69cd	42	Float_t fCut[13][7],fAprob[3][8];
	43	Int_t fMxtrs;
	44	TClonesArray *fTrs;
23efe5f1	45	Float_t fWpi,fWk,fWp;
	46	Float_t fRpik,fRppi,fRpka,fRp;
	47	Int_t fPcode;
e75c69cd	48
	49	int Qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
	50	Float_t Qtrm(Int_t track);
	51	Float_t Qtrm(Float_t qarr[6],Int_t narr);
38f7c306	52	Float_t fSigmin;
e75c69cd	53	Int_t Wpik(Float_t pm,Float_t q);
	54	Int_t Wpikp(Float_t pm,Float_t q);
	55	Int_t Pion(){return fWpi=1.,fPcode=211;}
	56	Int_t Kaon(){return fWk=1.,fPcode=321;}
	57	Int_t Proton(){return fWp=1.,fPcode=2212;}
11b3ae13	58	Int_t fSilent;
	59	TF1* fCutKa;
	60	TF1* fCutPr;
	61	TF1* fGGpi[6];
	62	TF1* fGGka[3];
	63	TF1* fGGpr[3];
	64	TF1* fggpi;
	65	TF1* fggka;
	66	TF1* fggpr;
e75c69cd	67
e6727e13	68	ClassDef(AliITSPid,2) // Class for ITS PID
23efe5f1	69	};
	70
	71	#endif
	72
	73
	74
	75