New classes for MUON and PHOS ESD (Yu.Belikov
[u/mrichter/AliRoot.git] / STEER / AliESD.h
1 #ifndef ALIESDEVENT_H
2 #define ALIESDEVENT_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6
7 /* $Id$ */
8
9 //-------------------------------------------------------------------------
10 //                          Class AliESD
11 //   This is the class to deal with during the physical analysis of data
12 //      
13 //         Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
14 //-------------------------------------------------------------------------
15
16 #include "TObject.h"
17 #include "TClonesArray.h"
18 #include  "AliESDtrack.h"
19 #include  "AliESDMuonTrack.h"
20 #include  "AliESDCaloTrack.h"
21 #include  "AliESDv0.h"
22 #include  "AliESDcascade.h"
23
24 class AliESD : public TObject {
25 public:
26   AliESD();
27   virtual ~AliESD() {
28     fTracks.Delete();
29     fCaloTracks.Delete();
30     fMuonTracks.Delete();
31     fV0s.Delete();
32     fCascades.Delete();
33   }
34
35   void SetEventNumber(Int_t n) {fEventNumber=n;}
36
37   AliESDtrack *GetTrack(Int_t i) {
38     return (AliESDtrack *)fTracks.UncheckedAt(i);
39   }
40   AliESDCaloTrack *GetCaloTrack(Int_t i) {
41     return (AliESDCaloTrack *)fCaloTracks.UncheckedAt(i);
42   }
43   AliESDMuonTrack *GetMuonTrack(Int_t i) {
44     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
45   }
46
47   void AddTrack(const AliESDtrack *t) {
48     new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
49   }
50   void AddCaloTrack(const AliESDCaloTrack *t) {
51     new(fCaloTracks[fCaloTracks.GetEntriesFast()]) AliESDCaloTrack(*t);
52   }
53   void AddMuonTrack(const AliESDMuonTrack *t) {
54     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
55   }
56
57   AliESDv0 *GetV0(Int_t i) {
58     return (AliESDv0 *)fV0s.UncheckedAt(i);
59   }
60   void AddV0(const AliESDv0 *v) {
61     new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
62   }
63
64   AliESDcascade *GetCascade(Int_t i) {
65     return (AliESDcascade *)fCascades.UncheckedAt(i);
66   }
67   void AddCascade(const AliESDcascade *c) {
68     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
69   }
70
71   Int_t  GetEventNumber() const {return fEventNumber;}
72   Int_t  GetRunNumber() const {return fRunNumber;}
73   Long_t GetTrigger() const {return fTrigger;}
74   
75   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
76   Int_t GetNumberOfCaloTracks() const {return fCaloTracks.GetEntriesFast();}
77   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
78   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
79   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
80   
81 protected:
82
83   // Event Identification
84   Int_t        fEventNumber;     // Event Number
85   Int_t        fRunNumber;       // Run Number
86   Long_t       fTrigger;         // Trigger Type
87   Int_t        fRecoVersion;     // Version of reconstruction 
88
89   TClonesArray  fTracks;         // ESD tracks
90   TClonesArray  fCaloTracks;     // Calorimeters' ESD tracks
91   TClonesArray  fMuonTracks;     // MUON ESD tracks
92   TClonesArray  fV0s;            // V0 vertices
93   TClonesArray  fCascades;       // Cascade vertices
94   
95   ClassDef(AliESD,1)  //ESD class 
96 };
97
98 #endif 
99