Corrections needed on HP
[u/mrichter/AliRoot.git] / ITS / AliITSPid.h
CommitLineData
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"
11b3ae13 9#include <TF1.h>
23efe5f1 10#include <assert.h>
23efe5f1 11//___________________________________________________________________________
12class AliITSPid :
13 public TObject {
14
15public:
16 AliITSPid(Int_t ntrs=1000);
17 virtual ~AliITSPid(){}
18 void SetEdep(Int_t track,Float_t Edep);
19 void SetPmom(Int_t track,Float_t Pmom);
20 void SetPcod(Int_t track,Int_t Pcod);
21 void Print(Int_t track);
22 void Tab(void);
23 void Reset(void);
24 void SetVec(Int_t track,TVector info);
25 TVector* GetVec(Int_t track);
26 Int_t GetPcode(TClonesArray*,Float_t);
27 Int_t GetPcode(Float_t,Float_t);
38f7c306 28 Int_t GetPcode(AliTPCtrack*);
29 Int_t GetPcode(AliITSIOTrack*);
30 Int_t GetPcode(AliITStrackV2*);
23efe5f1 31 void SetCut(Int_t,Float_t,Float_t,Float_t,
32 Float_t,Float_t,Float_t,Float_t);
38f7c306 33 void SetAProb(Int_t ivar,Int_t icut,Float_t apro){ aprob[ivar][icut]=apro; }
34 Float_t GetAProb(Int_t ivar,Int_t icut){ return aprob[ivar][icut]; }
23efe5f1 35 Float_t GetWpi(){return fWpi;}
36 Float_t GetWk(){return fWk;}
37 Float_t GetWp(){return fWp;}
38 Int_t GetPid(){return fPcode;};
39protected:
40public:
38f7c306 41 Float_t cut[13][7],aprob[3][8];
23efe5f1 42 Int_t mxtrs;
43 TClonesArray *trs;
44 Float_t qtot;
45 Float_t fWpi,fWk,fWp;
46 Float_t fRpik,fRppi,fRpka,fRp;
47 Int_t fPcode;
48//private:
49public:
23efe5f1 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;
11b3ae13 54 Int_t wpik(Float_t,Float_t);
55 Int_t wpikp(Float_t,Float_t);
23efe5f1 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;}
11b3ae13 59 Int_t fSilent;
60 TF1* fCutKa;
61 TF1* fCutPr;
62 TF1* fGGpi[6];
63 TF1* fGGka[3];
64 TF1* fGGpr[3];
65 TF1* fggpi;
66 TF1* fggka;
67 TF1* fggpr;
23efe5f1 68public:
e6727e13 69 ClassDef(AliITSPid,2) // Class for ITS PID
23efe5f1 70};
71
72#endif
73
74
75
76