]> git.uio.no Git - u/mrichter/AliRoot.git/blame_incremental - STEER/AliESDEvent.h
Fixing part of the Coding violation
[u/mrichter/AliRoot.git] / STEER / AliESDEvent.h
... / ...
CommitLineData
1// -*- mode: C++ -*-
2#ifndef ALIESDEVENT_H
3#define ALIESDEVENT_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 AliESDEvent
12// This is the class to deal with during the physics analysis of data.
13// It also ensures the backward compatibility with the old ESD format.
14//
15// Origin: Christian Klein-Boesing, CERN, Christian.Klein-Boesing@cern.ch
16//-------------------------------------------------------------------------
17
18#include <TClonesArray.h>
19#include <TObject.h>
20#include <TTree.h>
21#include <TArrayF.h>
22
23class TList;
24
25#include "AliVEvent.h"
26// some includes for delegated methods
27#include "AliESDCaloTrigger.h"
28#include "AliESDRun.h"
29#include "AliESDHeader.h"
30#include "AliESDTZERO.h"
31#include "AliESDZDC.h"
32#include "AliESDACORDE.h"
33
34// AliESDtrack has to be included so that the compiler
35// knows its inheritance tree (= that it is a AliVParticle).
36#include "AliESDtrack.h"
37
38class AliESDfriend;
39class AliESDVZERO;
40class AliESDHLTtrack;
41class AliESDVertex;
42class AliESDPmdTrack;
43class AliESDFMD;
44class AliESDkink;
45class AliESDCaloCluster;
46class AliESDCaloCells;
47class AliESDv0;
48class AliMultiplicity;
49class AliRawDataErrorLog;
50class AliESDRun;
51class AliESDTrdTrack;
52class AliESDMuonTrack;
53class AliESD;
54class AliESDcascade;
55class TRefArray;
56class AliESDACORDE;
57
58class AliESDEvent : public AliVEvent {
59public:
60
61
62 enum ESDListIndex {kESDRun,
63 kHeader,
64 kESDZDC,
65 kESDFMD,
66 kESDVZERO,
67 kESDTZERO,
68 kTPCVertex,
69 kSPDVertex,
70 kPrimaryVertex,
71 kSPDMult,
72 kPHOSTrigger,
73 kEMCALTrigger,
74 kTracks,
75 kMuonTracks,
76 kPmdTracks,
77 kTrdTracks,
78 kV0s,
79 kCascades,
80 kKinks,
81 kCaloClusters,
82 kEMCALCells,
83 kPHOSCells,
84 kErrorLogs,
85 kESDACORDE,
86 kESDListN
87 };
88
89 AliESDEvent();
90 virtual ~AliESDEvent();
91 AliESDEvent &operator=(const AliESDEvent& source); // or make private and use only copy?
92 virtual void Copy(TObject& obj) const;
93
94 // RUN
95 // move this to the UserData!!!
96 const AliESDRun* GetESDRun() const {return fESDRun;}
97
98 // Delegated methods for fESDRun
99 void SetRunNumber(Int_t n) {fESDRun->SetRunNumber(n);}
100 Int_t GetRunNumber() const {return fESDRun->GetRunNumber();}
101 void SetPeriodNumber(UInt_t n){fESDRun->SetPeriodNumber(n);}
102 UInt_t GetPeriodNumber() const {return fESDRun->GetPeriodNumber();}
103 void SetMagneticField(Double_t mf){fESDRun->SetMagneticField(mf);}
104 Double_t GetMagneticField() const {return fESDRun->GetMagneticField();}
105 void SetDiamond(const AliESDVertex *vertex) { fESDRun->SetDiamond(vertex);}
106 Double_t GetDiamondX() const {return fESDRun->GetDiamondX();}
107 Double_t GetDiamondY() const {return fESDRun->GetDiamondY();}
108 Double_t GetSigma2DiamondX() const {return fESDRun->GetSigma2DiamondX();}
109 Double_t GetSigma2DiamondY() const {return fESDRun->GetSigma2DiamondY();}
110 void GetDiamondCovXY(Float_t cov[3]) const {fESDRun->GetDiamondCovXY(cov);}
111 void SetTriggerClass(const char*name, Int_t index) {fESDRun->SetTriggerClass(name,index);}
112
113
114 // HEADER
115 AliESDHeader* GetHeader() const {return fHeader;}
116
117 // Delegated methods for fHeader
118 void SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
119 void SetOrbitNumber(UInt_t n) {fHeader->SetOrbitNumber(n);}
120 void SetTimeStamp(UInt_t timeStamp){fHeader->SetTimeStamp(timeStamp);}
121 void SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
122 void SetEventNumberInFile(Int_t n) {fHeader->SetEventNumberInFile(n);}
123 // void SetRunNumber(Int_t n) {fHeader->SetRunNumber(n);}
124 void SetBunchCrossNumber(UShort_t n) {fHeader->SetBunchCrossNumber(n);}
125 void SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
126
127 ULong64_t GetTriggerMask() const {return fHeader->GetTriggerMask();}
128 TString GetFiredTriggerClasses() const {return fESDRun->GetFiredTriggerClasses(fHeader->GetTriggerMask());}
129 Bool_t IsTriggerClassFired(const char *name) const {return fESDRun->IsTriggerClassFired(fHeader->GetTriggerMask(),name);}
130 UInt_t GetOrbitNumber() const {return fHeader->GetOrbitNumber();}
131 UInt_t GetTimeStamp() const { return fHeader->GetTimeStamp();}
132 UInt_t GetEventType() const { return fHeader->GetEventType();}
133 Int_t GetEventNumberInFile() const {return fHeader->GetEventNumberInFile();}
134 UShort_t GetBunchCrossNumber() const {return fHeader->GetBunchCrossNumber();}
135 UChar_t GetTriggerCluster() const {return fHeader->GetTriggerCluster();}
136
137 // ZDC CKB: put this in the header?
138 AliESDZDC* GetESDZDC() const {return fESDZDC;}
139
140 // Delegated methods for fESDZDC
141 Double_t GetZDCN1Energy() const {return fESDZDC->GetZDCN1Energy();}
142 Double_t GetZDCP1Energy() const {return fESDZDC->GetZDCP1Energy();}
143 Double_t GetZDCN2Energy() const {return fESDZDC->GetZDCN2Energy();}
144 Double_t GetZDCP2Energy() const {return fESDZDC->GetZDCP2Energy();}
145 Double_t GetZDCEMEnergy(Int_t i=0) const {return fESDZDC->GetZDCEMEnergy(i);}
146 Int_t GetZDCParticipants() const {return fESDZDC->GetZDCParticipants();}
147 void SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t em1Energy, Float_t em2Energy,
148 Float_t n2Energy, Float_t p2Energy, Int_t participants)
149 {fESDZDC->SetZDC(n1Energy, p1Energy, em1Energy, em2Energy, n2Energy, p2Energy, participants);}
150
151
152 // FMD
153 void SetFMDData(AliESDFMD * obj);
154 AliESDFMD *GetFMDData(){ return fESDFMD; }
155
156
157 // TZERO CKB: put this in the header?
158 const AliESDTZERO* GetESDTZERO() const {return fESDTZERO;}
159 // delegetated methods for fESDTZERO
160
161 Double_t GetT0zVertex() const {return fESDTZERO->GetT0zVertex();}
162 void SetT0zVertex(Float_t z) {fESDTZERO->SetT0zVertex(z);}
163 Double_t GetT0() const {return fESDTZERO->GetT0();}
164 void SetT0(Float_t timeStart) {fESDTZERO->SetT0(timeStart);}
165 const Double_t * GetT0time() const {return fESDTZERO->GetT0time();}
166 void SetT0time(Float_t time[24]) {fESDTZERO->SetT0time(time);}
167 const Double_t * GetT0amplitude() const {return fESDTZERO->GetT0amplitude();}
168 void SetT0amplitude(Float_t amp[24]){fESDTZERO->SetT0amplitude(amp);}
169
170 // VZERO
171 AliESDVZERO *GetVZEROData() const { return fESDVZERO; }
172 void SetVZEROData(AliESDVZERO * obj);
173
174 // ACORDE
175 AliESDACORDE *GetACORDEData() const { return fESDACORDE;}
176 void SetACORDEData(AliESDACORDE * obj);
177
178 void SetESDfriend(const AliESDfriend *f) const;
179 void GetESDfriend(AliESDfriend *f) const;
180
181
182
183 void SetPrimaryVertexTPC(const AliESDVertex *vertex);
184 const AliESDVertex *GetPrimaryVertexTPC() const {return fTPCVertex;}
185
186 void SetPrimaryVertexSPD(const AliESDVertex *vertex);
187 const AliESDVertex *GetPrimaryVertexSPD() const {return fSPDVertex;}
188 const AliESDVertex *GetVertex() const {
189 //For the backward compatibily only
190 return GetPrimaryVertexSPD();
191 }
192
193 void SetPrimaryVertex(const AliESDVertex *vertex);
194 const AliESDVertex *GetPrimaryVertex() const {return fPrimaryVertex;}
195
196 void SetMultiplicity(const AliMultiplicity *mul);
197
198 const AliMultiplicity *GetMultiplicity() const {return fSPDMult;}
199
200
201 Bool_t Clean(Float_t *cleanPars);
202 Bool_t RemoveKink(Int_t i) const;
203 Bool_t RemoveV0(Int_t i) const;
204 Bool_t RemoveTrack(Int_t i) const;
205
206 AliESDtrack *GetTrack(Int_t i) const {
207 return (AliESDtrack *)fTracks->UncheckedAt(i);
208 }
209 Int_t AddTrack(const AliESDtrack *t);
210
211
212 AliESDHLTtrack *GetHLTConfMapTrack(Int_t /*i*/) const {
213 // return (AliESDHLTtrack *)fHLTConfMapTracks->UncheckedAt(i);
214 return 0;
215 }
216 void AddHLTConfMapTrack(const AliESDHLTtrack */*t*/) {
217 printf("ESD:: AddHLTConfMapTrack do nothing \n");
218 // TClonesArray &fhlt = *fHLTConfMapTracks;
219 // new(fhlt[fHLTConfMapTracks->GetEntriesFast()]) AliESDHLTtrack(*t);
220 }
221
222
223 AliESDHLTtrack *GetHLTHoughTrack(Int_t /*i*/) const {
224 // return (AliESDHLTtrack *)fHLTHoughTracks->UncheckedAt(i);
225 return 0;
226 }
227 void AddHLTHoughTrack(const AliESDHLTtrack */*t*/) {
228 printf("ESD:: AddHLTHoughTrack do nothing \n");
229 // TClonesArray &fhlt = *fHLTHoughTracks;
230 // new(fhlt[fHLTHoughTracks->GetEntriesFast()]) AliESDHLTtrack(*t);
231 }
232
233 AliESDMuonTrack *GetMuonTrack(Int_t i) const {
234 return (AliESDMuonTrack *)fMuonTracks->UncheckedAt(i);
235 }
236
237 void AddMuonTrack(const AliESDMuonTrack *t);
238
239 AliESDPmdTrack *GetPmdTrack(Int_t i) const {
240 return (AliESDPmdTrack *)fPmdTracks->UncheckedAt(i);
241 }
242
243 void AddPmdTrack(const AliESDPmdTrack *t);
244
245
246 AliESDTrdTrack *GetTrdTrack(Int_t i) const {
247 return (AliESDTrdTrack *)fTrdTracks->UncheckedAt(i);
248 }
249
250
251 void AddTrdTrack(const AliESDTrdTrack *t);
252
253 AliESDv0 *GetV0(Int_t i) const {
254 return (AliESDv0*)fV0s->UncheckedAt(i);
255 }
256 Int_t AddV0(const AliESDv0 *v);
257
258 AliESDcascade *GetCascade(Int_t i) const {
259 return (AliESDcascade *)fCascades->UncheckedAt(i);
260 }
261
262 void AddCascade(const AliESDcascade *c);
263
264 AliESDkink *GetKink(Int_t i) const {
265 return (AliESDkink *)fKinks->UncheckedAt(i);
266 }
267 Int_t AddKink(const AliESDkink *c);
268
269 AliESDCaloCluster *GetCaloCluster(Int_t i) const {
270 return (AliESDCaloCluster *)fCaloClusters->UncheckedAt(i);
271 }
272
273 Int_t AddCaloCluster(const AliESDCaloCluster *c);
274
275 AliESDCaloCells *GetEMCALCells() const {return fEMCALCells; }
276 AliESDCaloCells *GetPHOSCells() const {return fPHOSCells; }
277
278 AliRawDataErrorLog *GetErrorLog(Int_t i) const {
279 return (AliRawDataErrorLog *)fErrorLogs->UncheckedAt(i);
280 }
281 void AddRawDataErrorLog(const AliRawDataErrorLog *log) const;
282
283 Int_t GetNumberOfErrorLogs() const {return fErrorLogs->GetEntriesFast();}
284
285
286 void AddPHOSTriggerPosition(TArrayF array) { fPHOSTrigger->AddTriggerPosition(array); }
287 void AddPHOSTriggerAmplitudes(TArrayF array) { fPHOSTrigger->AddTriggerAmplitudes(array);}
288 void AddEMCALTriggerPosition(TArrayF array) { fEMCALTrigger->AddTriggerPosition(array); }
289 void AddEMCALTriggerAmplitudes(TArrayF array){ fEMCALTrigger->AddTriggerAmplitudes(array); }
290
291 Int_t GetNumberOfTracks() const {return fTracks->GetEntriesFast();}
292 Int_t GetNumberOfHLTConfMapTracks() const {return 0;}
293 // fHLTConfMapTracks->GetEntriesFast();}
294 Int_t GetNumberOfHLTHoughTracks() const {return 0; }
295 // fHLTHoughTracks->GetEntriesFast(); }
296
297 Int_t GetNumberOfMuonTracks() const {return fMuonTracks->GetEntriesFast();}
298 Int_t GetNumberOfPmdTracks() const {return fPmdTracks->GetEntriesFast();}
299 Int_t GetNumberOfTrdTracks() const {return fTrdTracks->GetEntriesFast();}
300 Int_t GetNumberOfV0s() const {return fV0s->GetEntriesFast();}
301 Int_t GetNumberOfCascades() const {return fCascades->GetEntriesFast();}
302 Int_t GetNumberOfKinks() const {return fKinks->GetEntriesFast();}
303
304 Int_t GetEMCALClusters(TRefArray *clusters) const;
305 Int_t GetPHOSClusters(TRefArray *clusters) const;
306 Int_t GetNumberOfCaloClusters() const {return fCaloClusters->GetEntriesFast();}
307
308
309 // Remove this stuff CKB?
310 //---------------------------------------------------
311 Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;}
312 void SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;}
313 Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;}
314 void SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;}
315
316 Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;}
317 void SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; }
318 void SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; }
319 Int_t GetFirstPHOSCluster() const { return fFirstPHOSCluster ; }
320 //-------------------------------------------------------
321
322 TArrayF *GetEMCALTriggerPosition() const {return fEMCALTrigger->GetTriggerPosition();}
323 TArrayF *GetEMCALTriggerAmplitudes() const {return fEMCALTrigger->GetTriggerAmplitudes();}
324 TArrayF *GetPHOSTriggerPosition() const {return fPHOSTrigger->GetTriggerPosition();}
325 TArrayF *GetPHOSTriggerAmplitudes() const {return fPHOSTrigger->GetTriggerAmplitudes();}
326
327 void ResetV0s() { fV0s->Clear(); }
328 void ResetCascades() { fCascades->Clear(); }
329 void Reset();
330
331 void Print(Option_t *option="") const;
332
333 void AddObject(TObject* obj);
334 void ReadFromTree(TTree *tree);
335 TObject* FindListObject(const char *name);
336 AliESD *GetAliESDOld(){return fESDOld;}
337 const void WriteToTree(TTree* tree) const;
338 void GetStdContent();
339 void ResetStdContent();
340 void CreateStdContent();
341 void SetStdNames();
342 void CopyFromOldESD();
343 TList* GetList() const {return fESDObjects;}
344
345protected:
346 AliESDEvent(const AliESDEvent&);
347
348
349 TList *fESDObjects; // List of esd Objects
350
351 AliESDRun *fESDRun; //! Run information tmp put in the Userdata
352 AliESDHeader *fHeader; //! ESD Event Header
353 AliESDZDC *fESDZDC; //! ZDC information
354 AliESDFMD *fESDFMD; //! FMD object containing rough multiplicity
355 AliESDVZERO *fESDVZERO; //! VZERO object containing rough multiplicity
356 AliESDTZERO *fESDTZERO; //! TZEROObject
357 AliESDVertex *fTPCVertex; //! Primary vertex estimated by the TPC
358 AliESDVertex *fSPDVertex; //! Primary vertex estimated by the SPD
359 AliESDVertex *fPrimaryVertex; //! Primary vertex estimated using ESD tracks
360 AliMultiplicity *fSPDMult; //! SPD tracklet multiplicity
361 AliESDCaloTrigger* fPHOSTrigger; //! PHOS Trigger information
362 AliESDCaloTrigger* fEMCALTrigger; //! PHOS Trigger information
363 AliESDACORDE *fESDACORDE; //! ACORDE ESD object caontaining bit pattern
364
365 TClonesArray *fTracks; //! ESD tracks
366 TClonesArray *fMuonTracks; //! MUON ESD tracks
367 TClonesArray *fPmdTracks; //! PMD ESD tracks
368 TClonesArray *fTrdTracks; //! TRD ESD tracks (triggered)
369 TClonesArray *fV0s; //! V0 vertices
370 TClonesArray *fCascades; //! Cascade vertices
371 TClonesArray *fKinks; //! Kinks
372 TClonesArray *fCaloClusters; //! Calorimeter clusters for PHOS/EMCAL
373 AliESDCaloCells *fEMCALCells; //! EMCAL cell info
374 AliESDCaloCells *fPHOSCells; //! PHOS cell info
375 TClonesArray *fErrorLogs; //! Raw-data reading error messages
376
377
378
379 AliESD *fESDOld; //! Old esd Structure
380 AliESDfriend *fESDFriendOld; //! Old friend esd Structure
381 Bool_t fConnected; //! flag if leaves are alreday connected
382
383 static const char* fgkESDListName[kESDListN]; //!
384
385 // Remove this stuff CKB
386 Int_t fEMCALClusters; // Number of EMCAL clusters (subset of caloclusters)
387 Int_t fFirstEMCALCluster; // First EMCAL cluster in the fCaloClusters list
388
389 Int_t fPHOSClusters; // Number of PHOS clusters (subset of caloclusters)
390 Int_t fFirstPHOSCluster; // First PHOS cluster in the fCaloClusters list
391
392 ClassDef(AliESDEvent,8) //ESDEvent class
393};
394#endif
395