Further integration of the HLT reconstruction (C.Cheshkov)
[u/mrichter/AliRoot.git] / STEER / AliESD.h
CommitLineData
af7ba10c 1#ifndef ALIESD_H
2#define ALIESD_H
8a8d023f 3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
88cb7938 6
7/* $Id$ */
8
ae982df3 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//-------------------------------------------------------------------------
8a8d023f 15
af7ba10c 16#include <TClonesArray.h>
17#include <TObject.h>
18
af7ba10c 19#include "AliESDMuonTrack.h"
20#include "AliESDPmdTrack.h"
21#include "AliESDVertex.h"
22#include "AliESDcascade.h"
51ad6848 23#include "AliESDkink.h"
af7ba10c 24#include "AliESDtrack.h"
482070f2 25#include "AliESDHLTtrack.h"
af7ba10c 26#include "AliESDv0.h"
51ad6848 27#include "AliESDV0MI.h"
8a8d023f 28
ae982df3 29class AliESD : public TObject {
8a8d023f 30public:
ae982df3 31 AliESD();
af7ba10c 32 virtual ~AliESD();
8a8d023f 33
ae982df3 34 void SetEventNumber(Int_t n) {fEventNumber=n;}
5121f0b3 35 void SetRunNumber(Int_t n) {fRunNumber=n;}
2257f27e 36 void SetTrigger(Long_t n) {fTrigger=n;}
a866ac60 37 void SetMagneticField(Float_t mf){fMagneticField = mf;}
38 Float_t GetMagneticField() const {return fMagneticField;}
39
873f1f73 40 AliESDtrack *GetTrack(Int_t i) const {
ae982df3 41 return (AliESDtrack *)fTracks.UncheckedAt(i);
42 }
482070f2 43 AliESDHLTtrack *GetHLTConfMapTrack(Int_t i) const {
44 return (AliESDHLTtrack *)fHLTConfMapTracks.UncheckedAt(i);
45 }
46 AliESDHLTtrack *GetHLTHoughTrack(Int_t i) const {
47 return (AliESDHLTtrack *)fHLTHoughTracks.UncheckedAt(i);
48 }
873f1f73 49 AliESDMuonTrack *GetMuonTrack(Int_t i) const {
672b5f43 50 return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
51 }
561b7b31 52 AliESDPmdTrack *GetPmdTrack(Int_t i) const {
53 return (AliESDPmdTrack *)fPmdTracks.UncheckedAt(i);
54 }
672b5f43 55
51ad6848 56 Int_t AddTrack(const AliESDtrack *t) {
57 AliESDtrack * track = new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
58 track->SetID(fTracks.GetEntriesFast()-1);
59 return track->GetID();
60
ae982df3 61 }
482070f2 62 void AddHLTConfMapTrack(const AliESDHLTtrack *t) {
63 new(fHLTConfMapTracks[fHLTConfMapTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
64 }
65 void AddHLTHoughTrack(const AliESDHLTtrack *t) {
66 new(fHLTHoughTracks[fHLTHoughTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
67 }
672b5f43 68 void AddMuonTrack(const AliESDMuonTrack *t) {
69 new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
70 }
561b7b31 71 void AddPmdTrack(const AliESDPmdTrack *t) {
72 new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t);
73 }
8a8d023f 74
873f1f73 75 AliESDv0 *GetV0(Int_t i) const {
e23730c7 76 return (AliESDv0 *)fV0s.UncheckedAt(i);
77 }
78 void AddV0(const AliESDv0 *v) {
79 new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
80 }
51ad6848 81 void UpdateV0PIDs();
e23730c7 82
873f1f73 83 AliESDcascade *GetCascade(Int_t i) const {
e23730c7 84 return (AliESDcascade *)fCascades.UncheckedAt(i);
85 }
86 void AddCascade(const AliESDcascade *c) {
87 new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
88 }
89
51ad6848 90 AliESDkink *GetKink(Int_t i) const {
91 return (AliESDkink *)fKinks.UncheckedAt(i);
92 }
93 Int_t AddKink(const AliESDkink *c) {
94 AliESDkink * kink = new(fKinks[fKinks.GetEntriesFast()]) AliESDkink(*c);
95 kink->SetID(fKinks.GetEntriesFast());
96 return fKinks.GetEntriesFast()-1;
97 }
98
99 AliESDV0MI *GetV0MI(Int_t i) const {
100 return (AliESDV0MI *)fV0MIs.UncheckedAt(i);
101 }
102 Int_t AddV0MI(const AliESDV0MI *c) {
103 AliESDV0MI * v0 = new(fV0MIs[fV0MIs.GetEntriesFast()]) AliESDV0MI(*c);
104 v0->SetID(fV0MIs.GetEntriesFast()-1);
105 return fV0MIs.GetEntriesFast()-1;
106 }
107
108
1f46a9ae 109 void SetVertex(const AliESDVertex* vertex) {
2257f27e 110 new(&fPrimaryVertex) AliESDVertex(*vertex);
111 }
112 const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
873f1f73 113
ae982df3 114 Int_t GetEventNumber() const {return fEventNumber;}
115 Int_t GetRunNumber() const {return fRunNumber;}
116 Long_t GetTrigger() const {return fTrigger;}
8a8d023f 117
672b5f43 118 Int_t GetNumberOfTracks() const {return fTracks.GetEntriesFast();}
482070f2 119 Int_t GetNumberOfHLTConfMapTracks() const {return fHLTConfMapTracks.GetEntriesFast();}
120 Int_t GetNumberOfHLTHoughTracks() const {return fHLTHoughTracks.GetEntriesFast();}
672b5f43 121 Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
561b7b31 122 Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();}
e23730c7 123 Int_t GetNumberOfV0s() const {return fV0s.GetEntriesFast();}
124 Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
51ad6848 125 Int_t GetNumberOfKinks() const {return fKinks.GetEntriesFast();}
126 Int_t GetNumberOfV0MIs() const {return fV0MIs.GetEntriesFast();}
a2882fb4 127 Int_t GetNumberOfPHOSParticles() const {return fPHOSParticles;}
128 void SetNumberOfPHOSParticles(Int_t part) { fPHOSParticles = part ; }
129 void SetFirstPHOSParticle(Int_t index) { fFirstPHOSParticle = index ; }
704be597 130 Int_t GetNumberOfEMCALParticles() const {return fEMCALParticles;}
131 void SetNumberOfEMCALParticles(Int_t part) { fEMCALParticles = part ; }
132 void SetFirstEMCALParticle(Int_t index) { fFirstEMCALParticle = index ; }
133
a2882fb4 134
4a78b8c5 135 Float_t GetT0zVertex() const {return fT0zVertex;}
136 void SetT0zVertex(Float_t z) {fT0zVertex=z;}
ef278eae 137
878bc0c2 138 Float_t GetZDCNEnergy() const {return fZDCNEnergy;}
139 Float_t GetZDCPEnergy() const {return fZDCPEnergy;}
140 Float_t GetZDCEMEnergy() const {return fZDCEMEnergy;}
141 Int_t GetZDCParticipants() const {return fZDCParticipants;}
142 void SetZDC(Float_t nEnergy, Float_t pEnergy, Float_t emEnergy,
143 Int_t participants)
144 {fZDCNEnergy=nEnergy; fZDCPEnergy=pEnergy; fZDCEMEnergy=emEnergy;
145 fZDCParticipants=participants;}
146
bf25155c 147 void Reset();
148
ef278eae 149 void Print(Option_t *option="") const;
150
8a8d023f 151protected:
152
153 // Event Identification
ae982df3 154 Int_t fEventNumber; // Event Number
155 Int_t fRunNumber; // Run Number
156 Long_t fTrigger; // Trigger Type
157 Int_t fRecoVersion; // Version of reconstruction
a866ac60 158 Float_t fMagneticField; // Solenoid Magnetic Field in kG : for compatibility with AliMagF
8a8d023f 159
878bc0c2 160 Float_t fZDCNEnergy; // reconstructed energy in the neutron ZDC
161 Float_t fZDCPEnergy; // reconstructed energy in the proton ZDC
162 Float_t fZDCEMEnergy; // reconstructed energy in the electromagnetic ZDC
163 Int_t fZDCParticipants; // number of participants estimated by the ZDC
164
4a78b8c5 165 Float_t fT0zVertex; // vertex z position estimated by the START
8bbc564d 166 AliESDVertex fPrimaryVertex; // Primary vertex estimated by the ITS
167
168 TClonesArray fTracks; // ESD tracks
169 TClonesArray fHLTConfMapTracks;// HLT ESD tracks from Conformal Mapper method
170 TClonesArray fHLTHoughTracks; // HLT ESD tracks from Hough Transform method
171 TClonesArray fMuonTracks; // MUON ESD tracks
172 TClonesArray fPmdTracks; // PMD ESD tracks
173 TClonesArray fV0s; // V0 vertices
174 TClonesArray fCascades; // Cascade vertices
51ad6848 175 TClonesArray fKinks; // Kinks
176 TClonesArray fV0MIs; // V0MI
8bbc564d 177 Int_t fPHOSParticles; // Number of PHOS particles (stored as fTracks)
178 Int_t fEMCALParticles; // Number of EMCAL particles (stored as fTracks)
179 Int_t fFirstPHOSParticle; // First PHOS particle in the fTracks list
180 Int_t fFirstEMCALParticle;// First EMCAL particle in the fTracks list
704be597 181
878bc0c2 182 ClassDef(AliESD,8) //ESD class
8a8d023f 183};
8a8d023f 184#endif
185