Primary vertex reconstruction and standalone ITS tracking in the reconstruction chain
[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  "AliESDVertex.h"
19 #include  "AliESDtrack.h"
20 #include  "AliESDMuonTrack.h"
21 #include  "AliESDCaloTrack.h"
22 #include  "AliESDv0.h"
23 #include  "AliESDcascade.h"
24
25 class AliESD : public TObject {
26 public:
27   AliESD();
28   virtual ~AliESD() {
29     fTracks.Delete();
30     fCaloTracks.Delete();
31     fMuonTracks.Delete();
32     fV0s.Delete();
33     fCascades.Delete();
34   }
35
36   void SetEventNumber(Int_t n) {fEventNumber=n;}
37   void SetRunNumber(Int_t n) {fRunNumber=n;}
38   void SetTrigger(Long_t n) {fTrigger=n;}
39   void SetMagneticField(Float_t mf){fMagneticField = mf;}
40   Float_t GetMagneticField() const {return fMagneticField;}
41   
42   AliESDtrack *GetTrack(Int_t i) const {
43     return (AliESDtrack *)fTracks.UncheckedAt(i);
44   }
45   AliESDCaloTrack *GetCaloTrack(Int_t i) const {
46     return (AliESDCaloTrack *)fCaloTracks.UncheckedAt(i);
47   }
48   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
49     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
50   }
51
52   void AddTrack(const AliESDtrack *t) {
53     new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
54   }
55   void AddCaloTrack(const AliESDCaloTrack *t) {
56     new(fCaloTracks[fCaloTracks.GetEntriesFast()]) AliESDCaloTrack(*t);
57   }
58   void AddMuonTrack(const AliESDMuonTrack *t) {
59     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
60   }
61
62   AliESDv0 *GetV0(Int_t i) const {
63     return (AliESDv0 *)fV0s.UncheckedAt(i);
64   }
65   void AddV0(const AliESDv0 *v) {
66     new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
67   }
68
69   AliESDcascade *GetCascade(Int_t i) const {
70     return (AliESDcascade *)fCascades.UncheckedAt(i);
71   }
72   void AddCascade(const AliESDcascade *c) {
73     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
74   }
75
76   void SetVertex(AliESDVertex* vertex) {
77     new(&fPrimaryVertex) AliESDVertex(*vertex);
78   }
79   const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
80
81   Int_t  GetEventNumber() const {return fEventNumber;}
82   Int_t  GetRunNumber() const {return fRunNumber;}
83   Long_t GetTrigger() const {return fTrigger;}
84   
85   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
86   Int_t GetNumberOfCaloTracks() const {return fCaloTracks.GetEntriesFast();}
87   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
88   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
89   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
90
91   void  Print(Option_t *option="") const;
92    
93 protected:
94
95   // Event Identification
96   Int_t        fEventNumber;     // Event Number
97   Int_t        fRunNumber;       // Run Number
98   Long_t       fTrigger;         // Trigger Type
99   Int_t        fRecoVersion;     // Version of reconstruction 
100   Float_t      fMagneticField;   // Solenoid Magnetic Field in kG : for compatibility with AliMagF
101
102   AliESDVertex  fPrimaryVertex;  // Primary vertex
103
104   TClonesArray  fTracks;         // ESD tracks
105   TClonesArray  fCaloTracks;     // Calorimeters' ESD tracks
106   TClonesArray  fMuonTracks;     // MUON ESD tracks
107   TClonesArray  fV0s;            // V0 vertices
108   TClonesArray  fCascades;       // Cascade vertices
109   
110   ClassDef(AliESD,3)  //ESD class 
111                       //ver. 2: Magnetic Field Added; skowron
112 };
113
114 #endif 
115