56d0558e7a2972ee49754b7b6694e2480f652181
[u/mrichter/AliRoot.git] / STEER / AliESDEvent.h
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
23 class 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
38 class AliESDfriend;
39 class AliESDVZERO;
40 class AliESDHLTtrack;
41 class AliESDVertex;
42 class AliESDPmdTrack;
43 class AliESDFMD;
44 class AliESDkink;
45 class AliESDCaloCluster;
46 class AliESDCaloCells;
47 class AliESDv0;
48 class AliMultiplicity;
49 class AliRawDataErrorLog;
50 class AliESDRun;
51 class AliESDTrdTrack;
52 class AliESDMuonTrack;
53 class AliESD;
54 class AliESDcascade;
55 class TRefArray;
56 class AliESDACORDE;
57
58 class AliESDEvent : public AliVEvent {
59 public:
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
92
93   // RUN
94   // move this to the UserData!!!
95   const AliESDRun*    GetESDRun() const {return fESDRun;}
96
97   // Delegated methods for fESDRun
98   void     SetRunNumber(Int_t n) {fESDRun->SetRunNumber(n);}
99   Int_t    GetRunNumber() const {return fESDRun->GetRunNumber();}
100   void     SetPeriodNumber(UInt_t n){fESDRun->SetPeriodNumber(n);}
101   UInt_t   GetPeriodNumber() const {return fESDRun->GetPeriodNumber();}
102   void     SetMagneticField(Double_t mf){fESDRun->SetMagneticField(mf);}
103   Double_t GetMagneticField() const {return fESDRun->GetMagneticField();}
104   void     SetDiamond(const AliESDVertex *vertex) { fESDRun->SetDiamond(vertex);}
105   Double_t  GetDiamondX() const {return fESDRun->GetDiamondX();}
106   Double_t  GetDiamondY() const {return fESDRun->GetDiamondY();}
107   Double_t  GetSigma2DiamondX() const {return  fESDRun->GetSigma2DiamondX();}
108   Double_t  GetSigma2DiamondY() const {return  fESDRun->GetSigma2DiamondY();}
109   void      GetDiamondCovXY(Float_t cov[3]) const {fESDRun->GetDiamondCovXY(cov);}   
110   void     SetTriggerClass(const char*name, Int_t index) {fESDRun->SetTriggerClass(name,index);}
111   
112
113   // HEADER
114   AliESDHeader* GetHeader() const {return fHeader;}
115
116   // Delegated methods for fHeader
117   void      SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
118   void      SetOrbitNumber(UInt_t n) {fHeader->SetOrbitNumber(n);}
119   void      SetTimeStamp(UInt_t timeStamp){fHeader->SetTimeStamp(timeStamp);}
120   void      SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
121   void      SetEventNumberInFile(Int_t n) {fHeader->SetEventNumberInFile(n);}
122   //  void      SetRunNumber(Int_t n) {fHeader->SetRunNumber(n);}
123   void      SetBunchCrossNumber(UShort_t n) {fHeader->SetBunchCrossNumber(n);}
124   void      SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
125   
126   ULong64_t GetTriggerMask() const {return fHeader->GetTriggerMask();}
127   TString   GetFiredTriggerClasses() const {return fESDRun->GetFiredTriggerClasses(fHeader->GetTriggerMask());}
128   Bool_t    IsTriggerClassFired(const char *name) const {return fESDRun->IsTriggerClassFired(fHeader->GetTriggerMask(),name);}
129   UInt_t    GetOrbitNumber() const {return fHeader->GetOrbitNumber();}
130   UInt_t    GetTimeStamp()  const { return fHeader->GetTimeStamp();}
131   UInt_t    GetEventType()  const { return fHeader->GetEventType();}
132   Int_t     GetEventNumberInFile() const {return fHeader->GetEventNumberInFile();}
133   UShort_t  GetBunchCrossNumber() const {return fHeader->GetBunchCrossNumber();}
134   UChar_t   GetTriggerCluster() const {return fHeader->GetTriggerCluster();}
135
136   // ZDC CKB: put this in the header?
137   AliESDZDC*    GetESDZDC() const {return fESDZDC;}
138
139   // Delegated methods for fESDZDC
140   Double_t GetZDCN1Energy() const {return fESDZDC->GetZDCN1Energy();}
141   Double_t GetZDCP1Energy() const {return fESDZDC->GetZDCP1Energy();}
142   Double_t GetZDCN2Energy() const {return fESDZDC->GetZDCN2Energy();}
143   Double_t GetZDCP2Energy() const {return fESDZDC->GetZDCP2Energy();}
144   Double_t GetZDCEMEnergy(Int_t i=0) const {return fESDZDC->GetZDCEMEnergy(i);}
145   Int_t   GetZDCParticipants() const {return fESDZDC->GetZDCParticipants();}
146   void    SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t em1Energy, Float_t em2Energy,
147                  Float_t n2Energy, Float_t p2Energy, Int_t participants)
148   {fESDZDC->SetZDC(n1Energy, p1Energy, em1Energy, em2Energy, n2Energy, p2Energy, participants);}
149
150
151   // FMD
152   void SetFMDData(AliESDFMD * obj);
153   AliESDFMD *GetFMDData(){ return fESDFMD; }
154
155
156   // TZERO CKB: put this in the header?
157   const AliESDTZERO*    GetESDTZERO() const {return fESDTZERO;}
158   // delegetated methods for fESDTZERO
159
160   Double_t GetT0zVertex() const {return fESDTZERO->GetT0zVertex();}
161   void SetT0zVertex(Float_t z) {fESDTZERO->SetT0zVertex(z);}
162   Double_t GetT0() const {return fESDTZERO->GetT0();}
163   void SetT0(Float_t timeStart) {fESDTZERO->SetT0(timeStart);}
164   const Double_t * GetT0time() const {return fESDTZERO->GetT0time();}
165   void SetT0time(Float_t time[24]) {fESDTZERO->SetT0time(time);}
166   const Double_t * GetT0amplitude() const {return fESDTZERO->GetT0amplitude();}
167   void SetT0amplitude(Float_t amp[24]){fESDTZERO->SetT0amplitude(amp);}
168
169   // VZERO 
170   AliESDVZERO *GetVZEROData() const { return fESDVZERO; }
171   void SetVZEROData(AliESDVZERO * obj);
172
173  // ACORDE
174   AliESDACORDE *GetACORDEData() const { return fESDACORDE;}
175   void SetACORDEData(AliESDACORDE * obj);
176
177   void SetESDfriend(const AliESDfriend *f) const;
178   void GetESDfriend(AliESDfriend *f) const;
179
180
181
182   void SetPrimaryVertexTPC(const AliESDVertex *vertex); 
183   const AliESDVertex *GetPrimaryVertexTPC() const {return fTPCVertex;}
184
185   void SetPrimaryVertexSPD(const AliESDVertex *vertex); 
186   const AliESDVertex *GetPrimaryVertexSPD() const {return fSPDVertex;}
187   const AliESDVertex *GetVertex() const {
188     //For the backward compatibily only
189      return GetPrimaryVertexSPD();
190   }
191
192   void SetPrimaryVertex(const AliESDVertex *vertex);
193   const AliESDVertex *GetPrimaryVertex() const {return fPrimaryVertex;}
194
195   void SetMultiplicity(const AliMultiplicity *mul);
196
197   const AliMultiplicity *GetMultiplicity() const {return fSPDMult;}
198
199
200   Bool_t Clean(Float_t *cleanPars);
201   Bool_t RemoveKink(Int_t i)   const;
202   Bool_t RemoveV0(Int_t i)     const;
203   Bool_t RemoveTrack(Int_t i)  const;
204
205   AliESDtrack *GetTrack(Int_t i) const {
206     return (AliESDtrack *)fTracks->UncheckedAt(i);
207   }
208   Int_t  AddTrack(const AliESDtrack *t);
209
210   
211   AliESDHLTtrack *GetHLTConfMapTrack(Int_t /*i*/) const {
212     //    return (AliESDHLTtrack *)fHLTConfMapTracks->UncheckedAt(i);
213     return 0;
214   }
215   void AddHLTConfMapTrack(const AliESDHLTtrack */*t*/) {
216     printf("ESD:: AddHLTConfMapTrack do nothing \n");
217     //    TClonesArray &fhlt = *fHLTConfMapTracks;
218     //  new(fhlt[fHLTConfMapTracks->GetEntriesFast()]) AliESDHLTtrack(*t);
219   }
220   
221
222   AliESDHLTtrack *GetHLTHoughTrack(Int_t /*i*/) const {
223     //    return (AliESDHLTtrack *)fHLTHoughTracks->UncheckedAt(i);
224     return 0;
225   }
226   void AddHLTHoughTrack(const AliESDHLTtrack */*t*/) {
227     printf("ESD:: AddHLTHoughTrack do nothing \n");
228     //    TClonesArray &fhlt = *fHLTHoughTracks;
229     //     new(fhlt[fHLTHoughTracks->GetEntriesFast()]) AliESDHLTtrack(*t);
230   }
231   
232   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
233     return (AliESDMuonTrack *)fMuonTracks->UncheckedAt(i);
234   }
235
236   void AddMuonTrack(const AliESDMuonTrack *t);
237
238   AliESDPmdTrack *GetPmdTrack(Int_t i) const {
239     return (AliESDPmdTrack *)fPmdTracks->UncheckedAt(i);
240   }
241
242   void AddPmdTrack(const AliESDPmdTrack *t);
243
244
245   AliESDTrdTrack *GetTrdTrack(Int_t i) const {
246     return (AliESDTrdTrack *)fTrdTracks->UncheckedAt(i);
247   }
248
249   
250   void AddTrdTrack(const AliESDTrdTrack *t);
251
252   AliESDv0 *GetV0(Int_t i) const {
253     return (AliESDv0*)fV0s->UncheckedAt(i);
254   }
255   Int_t AddV0(const AliESDv0 *v);
256
257   AliESDcascade *GetCascade(Int_t i) const {
258     return (AliESDcascade *)fCascades->UncheckedAt(i);
259   }
260
261   void AddCascade(const AliESDcascade *c);
262
263   AliESDkink *GetKink(Int_t i) const {
264     return (AliESDkink *)fKinks->UncheckedAt(i);
265   }
266   Int_t AddKink(const AliESDkink *c);
267
268   AliESDCaloCluster *GetCaloCluster(Int_t i) const {
269     return (AliESDCaloCluster *)fCaloClusters->UncheckedAt(i);
270   }
271
272   Int_t AddCaloCluster(const AliESDCaloCluster *c);
273
274   AliESDCaloCells *GetEMCALCells() const {return fEMCALCells; }  
275   AliESDCaloCells *GetPHOSCells() const {return fPHOSCells; }  
276
277   AliRawDataErrorLog *GetErrorLog(Int_t i) const {
278     return (AliRawDataErrorLog *)fErrorLogs->UncheckedAt(i);
279   }
280   void  AddRawDataErrorLog(const AliRawDataErrorLog *log) const;
281
282   Int_t GetNumberOfErrorLogs()   const {return fErrorLogs->GetEntriesFast();}
283
284     
285   void AddPHOSTriggerPosition(TArrayF array)   { fPHOSTrigger->AddTriggerPosition(array); }
286   void AddPHOSTriggerAmplitudes(TArrayF array) { fPHOSTrigger->AddTriggerAmplitudes(array);}
287   void AddEMCALTriggerPosition(TArrayF array)  { fEMCALTrigger->AddTriggerPosition(array); }
288   void AddEMCALTriggerAmplitudes(TArrayF array){ fEMCALTrigger->AddTriggerAmplitudes(array); }
289
290   Int_t GetNumberOfTracks()     const {return fTracks->GetEntriesFast();}
291   Int_t GetNumberOfHLTConfMapTracks()     const {return 0;} 
292   // fHLTConfMapTracks->GetEntriesFast();}
293   Int_t GetNumberOfHLTHoughTracks()     const {return  0;  }
294   //  fHLTHoughTracks->GetEntriesFast();  }
295
296   Int_t GetNumberOfMuonTracks() const {return fMuonTracks->GetEntriesFast();}
297   Int_t GetNumberOfPmdTracks() const {return fPmdTracks->GetEntriesFast();}
298   Int_t GetNumberOfTrdTracks() const {return fTrdTracks->GetEntriesFast();}
299   Int_t GetNumberOfV0s()      const {return fV0s->GetEntriesFast();}
300   Int_t GetNumberOfCascades() const {return fCascades->GetEntriesFast();}
301   Int_t GetNumberOfKinks() const {return fKinks->GetEntriesFast();}
302   
303   Int_t GetEMCALClusters(TRefArray *clusters) const;
304   Int_t GetPHOSClusters(TRefArray *clusters) const;
305   Int_t GetNumberOfCaloClusters() const {return fCaloClusters->GetEntriesFast();}
306
307   
308   // Remove this stuff CKB?
309   //---------------------------------------------------
310   Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;}
311   void  SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;}
312   Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;}
313   void  SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;}
314  
315   Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;}
316   void  SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; }
317   void  SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; } 
318   Int_t GetFirstPHOSCluster() const  { return fFirstPHOSCluster ; }
319   //-------------------------------------------------------
320
321   TArrayF *GetEMCALTriggerPosition() const {return  fEMCALTrigger->GetTriggerPosition();}
322   TArrayF *GetEMCALTriggerAmplitudes() const {return  fEMCALTrigger->GetTriggerAmplitudes();}
323   TArrayF *GetPHOSTriggerPosition() const {return  fPHOSTrigger->GetTriggerPosition();}
324   TArrayF *GetPHOSTriggerAmplitudes() const {return  fPHOSTrigger->GetTriggerAmplitudes();}
325
326   void ResetV0s() { fV0s->Clear(); }
327   void ResetCascades() { fCascades->Clear(); }
328   void Reset();
329
330   void  Print(Option_t *option="") const;
331
332   void AddObject(TObject* obj);
333   void ReadFromTree(TTree *tree);
334   TObject* FindListObject(const char *name);
335   AliESD *GetAliESDOld(){return fESDOld;}
336   const void WriteToTree(TTree* tree) const;
337   void GetStdContent();
338   void ResetStdContent();
339   void CreateStdContent();
340   void SetStdNames();
341   void CopyFromOldESD();
342   TList* GetList() const {return fESDObjects;}
343
344 protected:
345   AliESDEvent(const AliESDEvent&);
346   AliESDEvent &operator=(const AliESDEvent& source);
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