bugfix
[u/mrichter/AliRoot.git] / ITS / AliITSPid.h
CommitLineData
23efe5f1 1#ifndef ALIITSPID_H
2#define ALIITSPID_H
e75c69cd 3
23efe5f1 4#include <TObject.h>
79b921e1 5#include "AliITSIOTrack.h"
e75c69cd 6
7class TClonesArray;
8class TVector;
9class AliTPCtrack;
10class AliITStrackV2;
11class TF1;
12
23efe5f1 13//___________________________________________________________________________
14class AliITSPid :
15 public TObject {
16
17public:
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;};
41protected:
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