Logs added
[u/mrichter/AliRoot.git] / TPC / AliTPCPid.h
1 #ifndef ALITPCPID_H
2 #define ALITPCPID_H
3
4 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5  * See cxx source for full Copyright notice                               */
6
7 /* $Id$ */
8
9 #include <TObject.h>
10 #include <TF1.h>
11 #include <TClonesArray.h>
12 #include <TVector.h>
13 #include "../TPC/AliTPCtrack.h"
14 //#include "../ITS/AliITSIOTrack.h"
15 #include "../ITS/AliITStrackV2.h"
16 #include <assert.h>
17 //___________________________________________________________________________
18 class  AliTPCPid :
19   public TObject {
20
21 public:
22                 AliTPCPid(Int_t ntrs=1000);
23                 virtual ~AliTPCPid(){}
24         void    SetEdep(Int_t track,Float_t Edep);
25         void    SetPmom(Int_t track,Float_t Pmom);
26         void    SetPcod(Int_t track,Int_t Pcod);
27         void    Print(Int_t track);
28         void    Tab(void);
29         void    Reset(void);
30         void    SetVec(Int_t track,TVector info);
31         TVector* GetVec(Int_t track);
32         Int_t   GetPcode(TClonesArray*,Float_t);
33         Int_t   GetPcode(Float_t,Float_t);
34         Int_t   GetPcode(AliTPCtrack*);
35 //        Int_t   GetPcode(AliITSIOTrack*); 
36         Int_t   GetPcode(AliITStrackV2*);
37         void    SetCut(Int_t,Float_t,Float_t,Float_t,
38                             Float_t,Float_t,Float_t,Float_t);
39         void    SetAProb(Int_t ivar,Int_t icut,Float_t apro){ aprob[ivar][icut]=apro; } 
40         Float_t GetAProb(Int_t ivar,Int_t icut){ return aprob[ivar][icut]; } 
41         Float_t GetWpi(){return fWpi;}
42         Float_t GetWk(){return fWk;}
43         Float_t GetWp(){return fWp;}
44         Int_t   GetPid(){return fPcode;};
45 protected:
46 public:
47         Float_t cut[13][7],aprob[3][8];
48         Int_t       mxtrs;
49         TClonesArray *trs;
50         Float_t qtot;
51         Float_t fWpi,fWk,fWp;
52         Float_t fRpik,fRppi,fRpka,fRp; 
53         Int_t   fPcode;
54 //private:
55 public:
56         TF1 *fCutKa;
57         TF1 *fCutPr;
58         Float_t fCutKaTune,fCutPrTune;
59         Float_t qcorr(Float_t);
60         int     qcomp(Float_t* qa,Float_t* qb){return qa[0]>qb[0]?1:0;}
61         Float_t qtrm(Int_t track);
62         Float_t qtrm(Float_t qarr[6],Int_t narr);
63         Float_t fSigmin;
64         Int_t   wpik(Int_t,Float_t);
65         Int_t   wpikp(Int_t,Float_t);
66         Int_t   pion(){return /*fWpi=1.,*/fPcode=211;}
67         Int_t   kaon(){return /*fWk=1.,*/fPcode=321;}
68         Int_t   proton(){return /*fWp=1.,*/fPcode=2212;}
69         Int_t   fSilent;
70 public:
71   ClassDef(AliTPCPid,1) // Class for TPC PID
72 };
73
74 #endif  
75
76
77
78