]>
Commit | Line | Data |
---|---|---|
1 | // -*- mode: C++ -*- | |
2 | #ifndef ALIESD_H | |
3 | #define ALIESD_H | |
4 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
5 | * See cxx source for full Copyright notice */ | |
6 | ||
7 | ||
8 | /* $Id$ */ | |
9 | ||
10 | //------------------------------------------------------------------------- | |
11 | // Class AliESD | |
12 | // This is the class to deal with during the physical analysis of data | |
13 | // | |
14 | // Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch | |
15 | //------------------------------------------------------------------------- | |
16 | ||
17 | #include <TClonesArray.h> | |
18 | #include <TObject.h> | |
19 | #include <TArrayF.h> | |
20 | ||
21 | #include "AliESDMuonTrack.h" | |
22 | #include "AliESDPmdTrack.h" | |
23 | #include "AliESDTrdTrack.h" | |
24 | #include "AliESDVertex.h" | |
25 | #include "AliESDcascade.h" | |
26 | #include "AliESDkink.h" | |
27 | #include "AliESDtrack.h" | |
28 | #include "AliESDHLTtrack.h" | |
29 | #include "AliESDCaloCluster.h" | |
30 | #include "AliESDv0.h" | |
31 | #include "AliESDFMD.h" | |
32 | #include "AliESDVZERO.h" | |
33 | #include "AliMultiplicity.h" | |
34 | #include "AliRawDataErrorLog.h" | |
35 | ||
36 | class AliESDfriend; | |
37 | ||
38 | class AliESD : public TObject { | |
39 | public: | |
40 | AliESD(); | |
41 | virtual ~AliESD(); | |
42 | ||
43 | void SetESDfriend(const AliESDfriend *f); | |
44 | void GetESDfriend(AliESDfriend *f) const; | |
45 | ||
46 | void SetEventNumberInFile(Int_t n) {fEventNumberInFile=n;} | |
47 | void SetBunchCrossNumber(UShort_t n) {fBunchCrossNumber=n;} | |
48 | void SetOrbitNumber(UInt_t n) {fOrbitNumber=n;} | |
49 | void SetPeriodNumber(UInt_t n) {fPeriodNumber=n;} | |
50 | void SetRunNumber(Int_t n) {fRunNumber=n;} | |
51 | void SetTimeStamp(UInt_t timeStamp){fTimeStamp = timeStamp;} | |
52 | void SetEventType(UInt_t eventType){fEventType = eventType;} | |
53 | void SetTriggerMask(ULong64_t n) {fTriggerMask=n;} | |
54 | void SetTriggerCluster(UChar_t n) {fTriggerCluster = n;} | |
55 | void SetMagneticField(Float_t mf){fMagneticField = mf;} | |
56 | Float_t GetMagneticField() const {return fMagneticField;} | |
57 | ||
58 | AliESDtrack *GetTrack(Int_t i) const { | |
59 | return (AliESDtrack *)fTracks.UncheckedAt(i); | |
60 | } | |
61 | AliESDHLTtrack *GetHLTConfMapTrack(Int_t i) const { | |
62 | return (AliESDHLTtrack *)fHLTConfMapTracks.UncheckedAt(i); | |
63 | } | |
64 | AliESDHLTtrack *GetHLTHoughTrack(Int_t i) const { | |
65 | return (AliESDHLTtrack *)fHLTHoughTracks.UncheckedAt(i); | |
66 | } | |
67 | AliESDMuonTrack *GetMuonTrack(Int_t i) const { | |
68 | return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i); | |
69 | } | |
70 | AliESDPmdTrack *GetPmdTrack(Int_t i) const { | |
71 | return (AliESDPmdTrack *)fPmdTracks.UncheckedAt(i); | |
72 | } | |
73 | AliESDTrdTrack *GetTrdTrack(Int_t i) const { | |
74 | return (AliESDTrdTrack *)fTrdTracks.UncheckedAt(i); | |
75 | } | |
76 | ||
77 | Int_t AddTrack(const AliESDtrack *t) { | |
78 | AliESDtrack * track = new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t); | |
79 | track->SetID(fTracks.GetEntriesFast()-1); | |
80 | return track->GetID(); | |
81 | ||
82 | } | |
83 | void AddHLTConfMapTrack(const AliESDHLTtrack *t) { | |
84 | new(fHLTConfMapTracks[fHLTConfMapTracks.GetEntriesFast()]) AliESDHLTtrack(*t); | |
85 | } | |
86 | void AddHLTHoughTrack(const AliESDHLTtrack *t) { | |
87 | new(fHLTHoughTracks[fHLTHoughTracks.GetEntriesFast()]) AliESDHLTtrack(*t); | |
88 | } | |
89 | void AddMuonTrack(const AliESDMuonTrack *t) { | |
90 | new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t); | |
91 | } | |
92 | void AddPmdTrack(const AliESDPmdTrack *t) { | |
93 | new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t); | |
94 | } | |
95 | void AddTrdTrack(const AliESDTrdTrack *t) { | |
96 | new(fTrdTracks[fTrdTracks.GetEntriesFast()]) AliESDTrdTrack(*t); | |
97 | } | |
98 | ||
99 | AliESDv0 *GetV0(Int_t i) const { | |
100 | return (AliESDv0 *)fV0s.UncheckedAt(i); | |
101 | } | |
102 | Int_t AddV0(const AliESDv0 *v); | |
103 | ||
104 | AliESDcascade *GetCascade(Int_t i) const { | |
105 | return (AliESDcascade *)fCascades.UncheckedAt(i); | |
106 | } | |
107 | void AddCascade(const AliESDcascade *c) { | |
108 | new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c); | |
109 | } | |
110 | ||
111 | AliESDkink *GetKink(Int_t i) const { | |
112 | return (AliESDkink *)fKinks.UncheckedAt(i); | |
113 | } | |
114 | Int_t AddKink(const AliESDkink *c) { | |
115 | AliESDkink * kink = new(fKinks[fKinks.GetEntriesFast()]) AliESDkink(*c); | |
116 | kink->SetID(fKinks.GetEntriesFast()); | |
117 | return fKinks.GetEntriesFast()-1; | |
118 | } | |
119 | ||
120 | AliESDCaloCluster *GetCaloCluster(Int_t i) const { | |
121 | return (AliESDCaloCluster *)fCaloClusters.UncheckedAt(i); | |
122 | } | |
123 | Int_t AddCaloCluster(const AliESDCaloCluster *c) { | |
124 | AliESDCaloCluster *clus = new(fCaloClusters[fCaloClusters.GetEntriesFast()]) AliESDCaloCluster(*c); | |
125 | clus->SetID(fCaloClusters.GetEntriesFast()-1); | |
126 | return fCaloClusters.GetEntriesFast()-1; | |
127 | } | |
128 | ||
129 | void AddPHOSTriggerPosition(TArrayF array) { fPHOSTriggerPosition = new TArrayF(array) ; } | |
130 | void AddPHOSTriggerAmplitudes(TArrayF array) { fPHOSTriggerAmplitudes = new TArrayF(array) ; } | |
131 | void AddEMCALTriggerPosition(TArrayF array) { fEMCALTriggerPosition = new TArrayF(array) ; } | |
132 | void AddEMCALTriggerAmplitudes(TArrayF array){ fEMCALTriggerAmplitudes = new TArrayF(array) ; } | |
133 | ||
134 | void SetVertex(const AliESDVertex *vertex) { | |
135 | new (&fSPDVertex) AliESDVertex(*vertex); | |
136 | } | |
137 | const AliESDVertex *GetVertex() const {return &fSPDVertex;} | |
138 | ||
139 | void SetMultiplicity(const AliMultiplicity *mul) { | |
140 | new (&fSPDMult) AliMultiplicity(*mul); | |
141 | } | |
142 | const AliMultiplicity *GetMultiplicity() const {return &fSPDMult;} | |
143 | ||
144 | void SetPrimaryVertex(const AliESDVertex *vertex) { | |
145 | new (&fPrimaryVertex) AliESDVertex(*vertex); | |
146 | } | |
147 | const AliESDVertex *GetPrimaryVertex() const {return &fPrimaryVertex;} | |
148 | ||
149 | void SetDiamond(const AliESDVertex *vertex) { | |
150 | fDiamondXY[0]=vertex->GetXv(); | |
151 | fDiamondXY[1]=vertex->GetYv(); | |
152 | Double_t cov[6]; | |
153 | vertex->GetCovMatrix(cov); | |
154 | fDiamondCovXY[0]=cov[0]; | |
155 | fDiamondCovXY[1]=cov[1]; | |
156 | fDiamondCovXY[2]=cov[2]; | |
157 | } | |
158 | Float_t GetDiamondX() const {return fDiamondXY[0];} | |
159 | Float_t GetDiamondY() const {return fDiamondXY[1];} | |
160 | Float_t GetSigma2DiamondX() const {return fDiamondCovXY[0];} | |
161 | Float_t GetSigma2DiamondY() const {return fDiamondCovXY[2];} | |
162 | void GetDiamondCovXY(Float_t cov[3]) const { | |
163 | for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return; | |
164 | } | |
165 | ||
166 | Int_t GetEventNumberInFile() const {return fEventNumberInFile;} | |
167 | UShort_t GetBunchCrossNumber() const {return fBunchCrossNumber;} | |
168 | UInt_t GetOrbitNumber() const {return fOrbitNumber;} | |
169 | UInt_t GetPeriodNumber() const {return fPeriodNumber;} | |
170 | Int_t GetRunNumber() const {return fRunNumber;} | |
171 | UInt_t GetTimeStamp() const { return fTimeStamp;} | |
172 | UInt_t GetEventType() const { return fEventType;} | |
173 | ULong64_t GetTriggerMask() const {return fTriggerMask;} | |
174 | UChar_t GetTriggerCluster() const {return fTriggerCluster;} | |
175 | ||
176 | Int_t GetNumberOfTracks() const {return fTracks.GetEntriesFast();} | |
177 | Int_t GetNumberOfHLTConfMapTracks() const {return fHLTConfMapTracks.GetEntriesFast();} | |
178 | Int_t GetNumberOfHLTHoughTracks() const {return fHLTHoughTracks.GetEntriesFast();} | |
179 | Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();} | |
180 | Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();} | |
181 | Int_t GetNumberOfTrdTracks() const {return fTrdTracks.GetEntriesFast();} | |
182 | Int_t GetNumberOfV0s() const {return fV0s.GetEntriesFast();} | |
183 | Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();} | |
184 | Int_t GetNumberOfKinks() const {return fKinks.GetEntriesFast();} | |
185 | Int_t GetNumberOfCaloClusters() const {return fCaloClusters.GetEntriesFast();} | |
186 | ||
187 | Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;} | |
188 | void SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;} | |
189 | Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;} | |
190 | void SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;} | |
191 | TArrayF *GetEMCALTriggerPosition() const {return fEMCALTriggerPosition;} | |
192 | TArrayF *GetEMCALTriggerAmplitudes() const {return fEMCALTriggerAmplitudes;} | |
193 | ||
194 | Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;} | |
195 | void SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; } | |
196 | void SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; } | |
197 | Int_t GetFirstPHOSCluster() const { return fFirstPHOSCluster ; } | |
198 | TArrayF *GetPHOSTriggerPosition() const {return fPHOSTriggerPosition;} | |
199 | TArrayF *GetPHOSTriggerAmplitudes() const {return fPHOSTriggerAmplitudes;} | |
200 | ||
201 | ||
202 | Float_t GetT0zVertex() const {return fT0zVertex;} | |
203 | void SetT0zVertex(Float_t z) {fT0zVertex=z;} | |
204 | Float_t GetT0() const {return fT0timeStart;} | |
205 | void SetT0(Float_t timeStart) {fT0timeStart = timeStart;} | |
206 | const Float_t * GetT0time() const {return fT0time;} | |
207 | void SetT0time(Float_t time[24]) { | |
208 | for (Int_t i=0; i<24; i++) fT0time[i] = time[i]; | |
209 | } | |
210 | const Float_t * GetT0amplitude() const {return fT0amplitude;} | |
211 | void SetT0amplitude(Float_t amp[24]) { | |
212 | for (Int_t i=0; i<24; i++) fT0amplitude[i] = amp[i]; | |
213 | } | |
214 | ||
215 | Float_t GetZDCN1Energy() const {return fZDCN1Energy;} | |
216 | Float_t GetZDCP1Energy() const {return fZDCP1Energy;} | |
217 | Float_t GetZDCN2Energy() const {return fZDCN2Energy;} | |
218 | Float_t GetZDCP2Energy() const {return fZDCP2Energy;} | |
219 | Float_t GetZDCEMEnergy() const {return fZDCEMEnergy;} | |
220 | Int_t GetZDCParticipants() const {return fZDCParticipants;} | |
221 | void SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t emEnergy, | |
222 | Float_t n2Energy, Float_t p2Energy, Int_t participants) | |
223 | {fZDCN1Energy=n1Energy; fZDCP1Energy=p1Energy; fZDCEMEnergy=emEnergy; | |
224 | fZDCN2Energy=n2Energy; fZDCP2Energy=p2Energy; fZDCParticipants=participants;} | |
225 | ||
226 | void ResetV0s() { fV0s.Clear(); } | |
227 | void ResetCascades() { fCascades.Clear(); } | |
228 | void Reset(); | |
229 | ||
230 | void Print(Option_t *option="") const; | |
231 | ||
232 | void SetFMDData(AliESDFMD * obj) { fESDFMD = new AliESDFMD(*obj); } | |
233 | AliESDFMD *GetFMDData(){ return fESDFMD; } | |
234 | ||
235 | void SetVZEROData(AliESDVZERO * obj) { fESDVZERO = new AliESDVZERO(*obj); } | |
236 | AliESDVZERO *GetVZEROData(){ return fESDVZERO; } | |
237 | ||
238 | AliRawDataErrorLog *GetErrorLog(Int_t i) const { | |
239 | return (AliRawDataErrorLog *)fErrorLogs.UncheckedAt(i); | |
240 | } | |
241 | void AddRawDataErrorLog(const AliRawDataErrorLog *log) { | |
242 | new(fErrorLogs[fErrorLogs.GetEntriesFast()]) AliRawDataErrorLog(*log); | |
243 | } | |
244 | Int_t GetNumberOfErrorLogs() const {return fErrorLogs.GetEntriesFast();} | |
245 | ||
246 | protected: | |
247 | AliESD(const AliESD&); | |
248 | AliESD &operator=(const AliESD& source); | |
249 | ||
250 | // Event Identification | |
251 | Int_t fEventNumberInFile;// running Event count in the file | |
252 | UShort_t fBunchCrossNumber;// Bunch Crossing Number | |
253 | UInt_t fOrbitNumber; // Orbit Number | |
254 | UInt_t fPeriodNumber; // Period Number | |
255 | Int_t fRunNumber; // Run Number | |
256 | UInt_t fTimeStamp; // Time stamp | |
257 | UInt_t fEventType; // Type of Event | |
258 | ULong64_t fTriggerMask; // Trigger Type (mask) | |
259 | UChar_t fTriggerCluster; // Trigger cluster (mask) | |
260 | Int_t fRecoVersion; // Version of reconstruction | |
261 | Float_t fMagneticField; // Solenoid Magnetic Field in kG : for compatibility with AliMagF | |
262 | ||
263 | Float_t fZDCN1Energy; // reconstructed energy in the neutron ZDC | |
264 | Float_t fZDCP1Energy; // reconstructed energy in the proton ZDC | |
265 | Float_t fZDCN2Energy; // reconstructed energy in the neutron ZDC | |
266 | Float_t fZDCP2Energy; // reconstructed energy in the proton ZDC | |
267 | Float_t fZDCEMEnergy; // reconstructed energy in the electromagnetic ZDC | |
268 | Int_t fZDCParticipants; // number of participants estimated by the ZDC | |
269 | ||
270 | Float_t fT0zVertex; // vertex z position estimated by the T0 | |
271 | AliESDVertex fSPDVertex; // Primary vertex estimated by the SPD | |
272 | AliESDVertex fPrimaryVertex; // Primary vertex estimated using ESD tracks | |
273 | Float_t fDiamondXY[2]; // Interaction diamond (x,y) in RUN | |
274 | Float_t fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN | |
275 | AliMultiplicity fSPDMult; // SPD tracklet multiplicity | |
276 | ||
277 | Float_t fT0timeStart; // interaction time estimated by the T0 | |
278 | Float_t fT0time[24]; // best TOF on each T0 PMT | |
279 | Float_t fT0amplitude[24]; // number of particles(MIPs) on each T0 PMT | |
280 | ||
281 | TClonesArray fTracks; // ESD tracks | |
282 | TClonesArray fHLTConfMapTracks;// HLT ESD tracks from Conformal Mapper method | |
283 | TClonesArray fHLTHoughTracks; // HLT ESD tracks from Hough Transform method | |
284 | TClonesArray fMuonTracks; // MUON ESD tracks | |
285 | TClonesArray fPmdTracks; // PMD ESD tracks | |
286 | TClonesArray fTrdTracks; // TRD ESD tracks (triggered) | |
287 | TClonesArray fV0s; // V0 vertices | |
288 | TClonesArray fCascades; // Cascade vertices | |
289 | TClonesArray fKinks; // Kinks | |
290 | TClonesArray fCaloClusters; // Calorimeter clusters for PHOS/EMCAL | |
291 | Int_t fEMCALClusters; // Number of EMCAL clusters (subset of caloclusters) | |
292 | Int_t fFirstEMCALCluster; // First EMCAL cluster in the fCaloClusters list | |
293 | TArrayF *fEMCALTriggerPosition; ///(x,y,z of 2x2 and x,y,z of nxn) not position of centroid but of patch corner | |
294 | TArrayF *fEMCALTriggerAmplitudes; //(2x2 max ampl, 2x2 amp out of patch, nxn max ampl, nxn amp out of patch) | |
295 | ||
296 | Int_t fPHOSClusters; // Number of PHOS clusters (subset of caloclusters) | |
297 | Int_t fFirstPHOSCluster; // First PHOS cluster in the fCaloClusters list | |
298 | TArrayF *fPHOSTriggerPosition; //(x,y,z of 2x2 and x,y,z of nxn), not position of centroid but of patch corner | |
299 | TArrayF *fPHOSTriggerAmplitudes; //(2x2 max ampl, 2x2 amp out of patch, nxn max ampl, nxn amp out of patch) | |
300 | ||
301 | AliESDFMD *fESDFMD; // FMD object containing rough multiplicity | |
302 | AliESDVZERO *fESDVZERO; // VZERO object containing rough multiplicity | |
303 | ||
304 | TClonesArray fErrorLogs; // Raw-data reading error messages | |
305 | ||
306 | ClassDef(AliESD,21) //ESD class | |
307 | }; | |
308 | #endif | |
309 |