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

#ifndef ALIITSPID_H
#define ALIITSPID_H
#include <TObject.h>
#include <TClonesArray.h>
#include <TVector.h>
#include "../TPC/AliTPCtrack.h"
#include "AliITSIOTrack.h"
#include "AliITStrackV2.h"
#include <assert.h>
//___________________________________________________________________________
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*,Float_t);
	Int_t	GetPcode(Float_t,Float_t);
	Int_t   GetPcode(AliTPCtrack*);
        Int_t   GetPcode(AliITSIOTrack*); 
        Int_t   GetPcode(AliITStrackV2*);
	void	SetCut(Int_t,Float_t,Float_t,Float_t,
			    Float_t,Float_t,Float_t,Float_t);
	void    SetAProb(Int_t ivar,Int_t icut,Float_t apro){ aprob[ivar][icut]=apro; } 
	Float_t GetAProb(Int_t ivar,Int_t icut){ return aprob[ivar][icut]; } 
	Float_t GetWpi(){return fWpi;}
	Float_t GetWk(){return fWk;}
	Float_t GetWp(){return fWp;}
	Int_t	GetPid(){return fPcode;};
protected:
public:
	Float_t cut[13][7],aprob[3][8];
	Int_t       mxtrs;
	TClonesArray *trs;
	Float_t qtot;
	Float_t fWpi,fWk,fWp;
	Float_t fRpik,fRppi,fRpka,fRp; 
	Int_t 	fPcode;
//private:
public:
	Float_t qcorr(Float_t);
	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(Int_t,Float_t);
	Int_t	wpikp(Int_t,Float_t);
	Int_t	pion(){return fWpi=1.,fPcode=211;}
	Int_t	kaon(){return fWk=1.,fPcode=321;}
	Int_t	proton(){return fWp=1.,fPcode=2212;}
public:
  ClassDef(AliITSPid,1) // Class for ITS PID
};

#endif
Commit	Line	Data
23efe5f1	1	#ifndef ALIITSPID_H
	2	#define ALIITSPID_H
	3	#include <TObject.h>
	4	#include <TClonesArray.h>
	5	#include <TVector.h>
79b921e1	6	#include "../TPC/AliTPCtrack.h"
	7	#include "AliITSIOTrack.h"
	8	#include "AliITStrackV2.h"
23efe5f1	9	#include <assert.h>
23efe5f1	10	//___________________________________________________________________________
	11	class AliITSPid :
	12	public TObject {
	13
	14	public:
	15	AliITSPid(Int_t ntrs=1000);
	16	virtual ~AliITSPid(){}
	17	void SetEdep(Int_t track,Float_t Edep);
	18	void SetPmom(Int_t track,Float_t Pmom);
	19	void SetPcod(Int_t track,Int_t Pcod);
	20	void Print(Int_t track);
	21	void Tab(void);
	22	void Reset(void);
	23	void SetVec(Int_t track,TVector info);
	24	TVector* GetVec(Int_t track);
	25	Int_t GetPcode(TClonesArray*,Float_t);
	26	Int_t GetPcode(Float_t,Float_t);
38f7c306	27	Int_t GetPcode(AliTPCtrack*);
	28	Int_t GetPcode(AliITSIOTrack*);
	29	Int_t GetPcode(AliITStrackV2*);
23efe5f1	30	void SetCut(Int_t,Float_t,Float_t,Float_t,
23efe5f1	31	Float_t,Float_t,Float_t,Float_t);
38f7c306	32	void SetAProb(Int_t ivar,Int_t icut,Float_t apro){ aprob[ivar][icut]=apro; }
38f7c306	33	Float_t GetAProb(Int_t ivar,Int_t icut){ return aprob[ivar][icut]; }
23efe5f1	34	Float_t GetWpi(){return fWpi;}
	35	Float_t GetWk(){return fWk;}
	36	Float_t GetWp(){return fWp;}
	37	Int_t GetPid(){return fPcode;};
	38	protected:
	39	public:
38f7c306	40	Float_t cut[13][7],aprob[3][8];
23efe5f1	41	Int_t mxtrs;
	42	TClonesArray *trs;
	43	Float_t qtot;
	44	Float_t fWpi,fWk,fWp;
	45	Float_t fRpik,fRppi,fRpka,fRp;
	46	Int_t fPcode;
	47	//private:
	48	public:
	49	Float_t qcorr(Float_t);
	50	int qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
	51	Float_t qtrm(Int_t track);
79b921e1	52	Float_t qtrm(Float_t qarr[6],Int_t narr);
38f7c306	53	Float_t fSigmin;
23efe5f1	54	Int_t wpik(Int_t,Float_t);
	55	Int_t wpikp(Int_t,Float_t);
	56	Int_t pion(){return fWpi=1.,fPcode=211;}
	57	Int_t kaon(){return fWk=1.,fPcode=321;}
	58	Int_t proton(){return fWp=1.,fPcode=2212;}
	59	public:
	60	ClassDef(AliITSPid,1) // Class for ITS PID
	61	};
	62
	63	#endif
	64
	65
	66
	67