Number of sigma pedestal cut increased to 4
[u/mrichter/AliRoot.git] / STEER / AOD / AliAODPid.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
7  * Permission to use, copy, modify and distribute this software and its   *
8  * documentation strictly for non-commercial purposes is hereby granted   *
9  * without fee, provided that the above copyright notice appears in all   *
10  * copies and that both the copyright notice and this permission notice   *
11  * appear in the supporting documentation. The authors make no claims     *
12  * about the suitability of this software for any purpose. It is          *
13  * provided "as is" without express or implied warranty.                  *
14  **************************************************************************/
15
16 /* $Id$ */
17
18 //-------------------------------------------------------------------------
19 //     AOD Pid class to store additional pid information
20 //     Author: Annalisa Mastroserio
21 //-------------------------------------------------------------------------
22
23 #include "AliAODPid.h"
24 #include "AliESDtrack.h"
25 #include "AliLog.h"
26 #include "AliPID.h"
27
28 ClassImp(AliAODPid)
29
30
31 //______________________________________________________________________________
32 AliAODPid::AliAODPid():
33     fITSsignal(0), 
34     fTPCsignal(0),
35     fTPCsignalN(0),
36     fTPCmomentum(0),
37     fTRDnSlices(0),
38     fTRDslices(0x0),
39     fTOFesdsignal(0),
40     fHMPIDsignal(0)
41 {
42   // default constructor
43     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]   = 0; 
44     for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = 0.;
45     for(Int_t i=0; i<3; i++) fEMCALPosition[i]    = 0.;
46     for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
47     for(Int_t i=0; i<6; i++) fTRDmomentum[i]      = 0.;
48     for(Int_t i=0; i<3; i++) fEMCALMomentum[i]    = 0.;
49   
50 }
51
52 //______________________________________________________________________________
53 AliAODPid::~AliAODPid() 
54 {
55   delete [] fTRDslices;
56   fTRDslices = 0;
57   // destructor
58 }
59
60
61 //______________________________________________________________________________
62 AliAODPid::AliAODPid(const AliAODPid& pid) : 
63   TObject(pid),
64   fITSsignal(pid.fITSsignal), 
65   fTPCsignal(pid.fTPCsignal),
66   fTPCsignalN(pid.fTPCsignalN),
67   fTPCmomentum(pid.fTPCmomentum),
68   fTRDnSlices(pid.fTRDnSlices),
69   fTRDslices(0x0),
70   fTOFesdsignal(pid.fTOFesdsignal),
71   fHMPIDsignal(pid.fHMPIDsignal)
72 {
73   // Copy constructor
74     fTRDslices = new Double32_t[fTRDnSlices];
75     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
76     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
77     for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[i];
78     for(Int_t i=0; i<3; i++) {
79       fEMCALPosition[i]=pid.fEMCALPosition[i];
80       fEMCALMomentum[i]=pid.fEMCALMomentum[i];
81     }
82     for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
83
84     for(Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
85 }
86
87 //______________________________________________________________________________
88 AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
89 {
90   // Assignment operator
91   if(this!=&pid) {
92     // copy stuff
93     fITSsignal=pid.fITSsignal; 
94     fTPCsignal=pid.fTPCsignal;
95     
96     if(pid.fTRDnSlices<=0||(fTRDnSlices!=pid.fTRDnSlices)){
97       // only delete if number changed or is 0
98       delete [] fTRDslices;
99       fTRDslices = 0;
100       if(pid.fTRDnSlices>0) fTRDslices = new Double32_t[fTRDnSlices];
101     }
102     fTRDnSlices=pid.fTRDnSlices;
103     
104     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
105     fTOFesdsignal=pid.fTOFesdsignal;
106     fHMPIDsignal=pid.fHMPIDsignal;
107     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
108     for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[i];
109     for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
110     for(Int_t i=0; i<3; i++) {
111       fEMCALPosition[i]=pid.fEMCALPosition[i];
112       fEMCALMomentum[i]=pid.fEMCALMomentum[i];
113     }
114     for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
115   }
116
117   return *this;
118 }
119 //_______________________________________________________________________________
120 void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES]) const
121 {
122  // Returns the array with integrated times for each particle hypothesis
123 for(Int_t i=0; i<kSPECIES; i++) timeint[i]=fIntTime[i];
124 }
125 //_______________________________________________________________________________
126 void AliAODPid::SetIntegratedTimes(Double_t timeint[kSPECIES])
127 {
128  // Returns the array with integrated times for each particle hypothesis
129 for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=timeint[i];
130 }
131 //_______________________________________________________________________________
132 void AliAODPid::GetEMCALPosition(Double_t emcalpos[3]) const
133 {
134  // Returns the array with extrapolated track position at the EMCAL surface
135   for(Int_t i=0; i<3; i++) emcalpos[i]=fEMCALPosition[i];
136 }
137 //_______________________________________________________________________________
138 void AliAODPid::SetEMCALPosition(Double_t emcpos[3])
139 {
140  // Sets the array with extrapolated track position at the EMCAL surface
141   for(Int_t i=0; i<3; i++) fEMCALPosition[i]=emcpos[i];
142 }
143 //_______________________________________________________________________________
144 void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3]) const
145 {
146  // Returns the array with extrapolated track momentum at the EMCAL surface
147   for(Int_t i=0; i<3; i++) emcalmom[i]=fEMCALMomentum[i];
148 }
149 //_______________________________________________________________________________
150 void AliAODPid::SetEMCALMomentum(Double_t emcmom[3])
151 {
152  // Sets the array with extrapolated track momentum at the EMCAL surface
153   for(Int_t i=0; i<3; i++) fEMCALMomentum[i]=emcmom[i];
154 }
155 //______________________________________________________________________________
156 void AliAODPid::SetTOFpidResolution(Double_t tofPIDres[5])
157 {
158   for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=tofPIDres[i];
159
160 }
161 //______________________________________________________________________________
162 void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
163 {
164   for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
165 }
166
167 //______________________________________________________________________________
168 void AliAODPid::SetHMPIDprobs(Double_t hmpPid[5]) 
169 {
170   //
171   // Set the HMPID PID probablities that are read from ESD
172   //  
173   for(Int_t i = 0; i < 5; i++ ) fHMPIDprobs[i] =  hmpPid[i];
174 }
175 //______________________________________________________________________________
176 void AliAODPid::GetHMPIDprobs(Double_t *p) const
177 {
178   //
179   // Set the HMPID PID probablities that are read from ESD
180   //  
181   for(Int_t i = 0; i < AliPID::kSPECIES; i++ ) p[i] =  fHMPIDprobs[i];
182 }