| b8def77a |
1 | #ifndef ALITPCPID_H |
| 2 | #define ALITPCPID_H |
| 733304f0 |
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 | |
| b8def77a |
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 | |
git://git.uio.no / u / mrichter / AliRoot.git / blame