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