Removing obsolete macros
[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"
23efe5f1 9#include <assert.h>
23efe5f1 10//___________________________________________________________________________
11class AliITSPid :
12 public TObject {
13
14public:
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,
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; }
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;};
38protected:
39public:
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:
48public:
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;}
59public:
60 ClassDef(AliITSPid,1) // Class for ITS PID
61};
62
63#endif
64
65
66
67