- Reshuffling of the particle codes in AliPID. Now the light nuclei are between the
[u/mrichter/AliRoot.git] / STEER / AOD / AliAODPid.h
1 #ifndef AliAODPID_H
2 #define AliAODPID_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //-------------------------------------------------------------------------
9 //     AOD Pid object for additional pid information
10 //     Author: Annalisa Mastroserio, CERN
11 //-------------------------------------------------------------------------
12
13 #include <TObject.h>
14
15 class AliTPCdEdxInfo;
16
17 class AliAODPid : public TObject {
18
19  public:
20   AliAODPid();
21   virtual ~AliAODPid();
22   AliAODPid(const AliAODPid& pid); 
23   AliAODPid& operator=(const AliAODPid& pid);
24   
25   enum{kTRDnPlanes=6};
26
27  //setters
28   void      SetITSsignal(Double_t its)                         {fITSsignal=its;}
29   void      SetITSdEdxSamples(const Double_t s[4]);
30   void      SetTPCsignal(Double_t tpc)                         {fTPCsignal=tpc;}
31   void      SetTPCsignalN(UShort_t tpcN)                       {fTPCsignalN=(UChar_t)((tpcN<160)?tpcN:160);}
32   void      SetTPCmomentum(Double_t tpcMom)                    {fTPCmomentum=tpcMom;}
33   void      SetTPCdEdxInfo(AliTPCdEdxInfo * dEdxInfo);
34   inline void  SetTRDsignal(Int_t nslices, const Double_t * const trdslices);  
35   void      SetTRDmomentum(Int_t nplane, Float_t trdMom)       {fTRDmomentum[nplane]=trdMom;}
36   inline void  SetTRDncls(UChar_t ncls, Int_t layer = -1);
37   void      SetTRDntrackletsPID(UChar_t ntls) {fTRDntls = ntls;}
38   void      SetTRDChi2(Double_t chi2)          {fTRDChi2 = chi2;}
39   void      SetTOFsignal(Double_t tof)                         {fTOFesdsignal=tof;}
40   void      SetTOFpidResolution(Double_t tofPIDres[5]);
41   void      SetIntegratedTimes(Double_t timeint[5]);
42
43   Double_t  GetITSsignal()       const {return  fITSsignal;}
44   void      GetITSdEdxSamples(Double_t *s) const;
45   Double_t  GetITSdEdxSample(Int_t i) const {
46     if(i>=0 && i<4) return fITSdEdxSamples[i];
47     else return 0.;
48   }
49   Double_t  GetTPCsignal()       const {return  fTPCsignal;}
50   UShort_t  GetTPCsignalN()      const {return  (UShort_t)fTPCsignalN;}
51   AliTPCdEdxInfo * GetTPCdEdxInfo()const{return fTPCdEdxInfo;}
52
53   Double_t  GetTPCmomentum()     const {return  fTPCmomentum;}
54   Int_t     GetTRDnSlices()      const {return  fTRDnSlices;}
55   Double_t* GetTRDsignal()       const {return  fTRDslices;}
56   Double_t  GetTRDChi2()         const {return fTRDChi2;}
57   const Double_t*  GetTRDmomentum() const {return  fTRDmomentum;}
58   UChar_t   GetTRDncls(UChar_t layer) const { if(layer > 5) return 0; return fTRDncls[layer];}
59   inline UChar_t GetTRDncls() const;
60   UChar_t   GetTRDntrackletsPID() const {return fTRDntls;}
61   Double_t  GetTOFsignal()       const {return  fTOFesdsignal;}
62
63   void      GetIntegratedTimes(Double_t timeint[5])  const; 
64   void      GetTOFpidResolution (Double_t tofRes[5]) const;
65
66  private :
67   Double32_t  fITSsignal;        //[0.,0.,10] detector raw signal
68   Double32_t  fITSdEdxSamples[4];//[0.,0.,10] ITS dE/dx samples
69
70   Double32_t  fTPCsignal;        //[0.,0.,10] detector raw signal
71   UChar_t     fTPCsignalN;       // number of points used for TPC dE/dx
72   Double32_t  fTPCmomentum;      //[0.,0.,20] momentum at the inner wall of TPC;
73
74   Int_t       fTRDnSlices;       // N slices used for PID in the TRD
75   UChar_t     fTRDntls;          // number of tracklets used for PID calculation
76   UChar_t     fTRDncls[6];       // number of clusters used for dE/dx calculation
77   Double32_t* fTRDslices;        //[fTRDnSlices][0.,0.,10]
78   Double32_t  fTRDmomentum[6];   //[0.,0.,10]  momentum at the TRD layers
79   Double32_t  fTRDChi2;          //[0.,0.,10]  TRD chi2
80
81   Double32_t  fTOFesdsignal;     //[0.,0.,20] TOF signal - t0 (T0 interaction time)
82   Double32_t  fTOFpidResolution[5]; //[0.,0.,20] TOF pid resolution for each mass hypotesys 
83   Double32_t  fIntTime[5];       //[0.,0.,20] track time hypothesis
84  
85   AliTPCdEdxInfo * fTPCdEdxInfo; // object containing dE/dx information for different pad regions
86
87   ClassDef(AliAODPid, 13);
88 };
89
90 //_____________________________________________________________
91 void AliAODPid::SetTRDsignal(Int_t nslices, const Double_t * const trdslices) {
92   //
93   // Set TRD dE/dx slices and the number of dE/dx slices per track
94   //
95   if(fTRDslices && fTRDnSlices != nslices) {
96     delete [] fTRDslices; fTRDslices = NULL;
97   };
98   if(!fTRDslices) fTRDslices = new Double32_t[nslices];
99   fTRDnSlices = nslices; 
100   for(Int_t is = 0; is < fTRDnSlices; is++) fTRDslices[is] = trdslices[is];
101 }
102
103 //_____________________________________________________________
104 void AliAODPid::SetTRDncls(UChar_t ncls, Int_t layer) { 
105   //
106   // Set the number of clusters / tracklet
107   // If no layer is specified the full number of clusters will be put in layer 0
108   //
109   if(layer > 5) return; 
110   if(layer < 0) fTRDncls[0] = ncls;
111   else fTRDncls[layer] = ncls;
112 }
113
114 //_____________________________________________________________
115 UChar_t AliAODPid::GetTRDncls() const{
116   //
117   // Get number of clusters per track
118   // Calculated as sum of the number of clusters per tracklet
119   //
120   UChar_t ncls = 0;
121   for(Int_t ily = 0; ily < 6; ily++) ncls += fTRDncls[ily];
122   return ncls;
123 }
124 #endif