Clean-up of include files
[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 <TVector.h>
11 #include "AliTPCtrack.h"
12 #include <assert.h>
13
14 class TF1;
15 class TClonesArray;
16 //___________________________________________________________________________
17 class  AliTPCPid :
18   public TObject {
19
20 public:
21                 AliTPCPid(Int_t ntrs=1000);
22                 virtual ~AliTPCPid(){}
23                 AliTPCPid( const AliTPCPid& r);
24                 AliTPCPid &operator = (const AliTPCPid & param); //assignment
25         void    SetEdep(Int_t track,Float_t Edep);
26         void    SetPmom(Int_t track,Float_t Pmom);
27         void    SetPcod(Int_t track,Int_t Pcod);
28         void    PrintPID(Int_t track);
29         void    Tab(void);
30         void    Reset(void);
31         void    SetVec(Int_t track,TVector info) const;
32         TVector* GetVec(Int_t track) const;
33         Int_t   GetPcode(TClonesArray* ,Float_t) const;
34         Int_t   GetPcode(Float_t q,Float_t pm);
35         Int_t   GetPcode(AliTPCtrack*track);
36         void    SetCut(Int_t n, Float_t pm, Float_t pilo, Float_t pihi,
37                             Float_t klo, Float_t khi, Float_t plo, 
38                             Float_t phi);
39         void    SetAProb(Int_t ivar,Int_t icut,Float_t apro){ faprob[ivar][icut]=apro; } 
40         Float_t GetAProb(Int_t ivar,Int_t icut) const
41                         { return faprob[ivar][icut]; } 
42         Float_t GetWpi() const {return fWpi;}
43         Float_t GetWk() const {return fWk;}
44         Float_t GetWp() const {return fWp;}
45         Int_t   GetPid() const {return fPcode;};
46         Float_t Qcorr(Float_t xc);
47         Int_t   Qcomp(Float_t* qa,Float_t* qb) const {return qa[0]>qb[0]?1:0;}
48         Float_t Qtrm(Int_t track) const;
49         Float_t Qtrm(Float_t qarr[6],Int_t narr);
50         Int_t   Wpik(Int_t nc, Float_t q);
51         Int_t   Wpikp(Int_t nc, Float_t q);
52         Int_t   Pion(){return /*fWpi=1.,*/fPcode=211;}
53         Int_t   Kaon(){return /*fWk=1.,*/fPcode=321;}
54         Int_t   Proton(){return /*fWp=1.,*/fPcode=2212;}
55 private:
56         TF1 *fCutKa; // function
57         TF1 *fCutPr; // function
58         Float_t fCutKaTune,fCutPrTune; // tune cuts
59         Float_t fSigmin; // sigma min
60         Int_t   fSilent; // flag
61         Float_t fcut[13][7],faprob[3][8]; //cuts
62         Int_t       fmxtrs; // fmxtrs
63         TClonesArray *trs; //pointer
64         Float_t fqtot; // tot q
65         Float_t fWpi,fWk,fWp; // weights
66         Float_t fRpik,fRppi,fRpka,fRp; // ratios
67         Int_t   fPcode; //p-code
68
69   ClassDef(AliTPCPid,1) // Class for TPC PID
70
71 };
72
73 #endif  
74
75
76
77