]> git.uio.no Git - u/mrichter/AliRoot.git/blame - STEER/AliESD.h
Coding conventions and minor fixes
[u/mrichter/AliRoot.git] / STEER / AliESD.h
CommitLineData
9da38871 1// -*- mode: C++ -*-
af7ba10c 2#ifndef ALIESD_H
3#define ALIESD_H
8a8d023f 4/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
6
88cb7938 7
8/* $Id$ */
9
ae982df3 10//-------------------------------------------------------------------------
11// Class AliESD
12// This is the class to deal with during the physical analysis of data
13//
14// Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
15//-------------------------------------------------------------------------
8a8d023f 16
af7ba10c 17#include <TClonesArray.h>
18#include <TObject.h>
64df000d 19#include <TArrayF.h>
b545009a 20
af7ba10c 21#include "AliESDMuonTrack.h"
22#include "AliESDPmdTrack.h"
0ee00e25 23#include "AliESDTrdTrack.h"
af7ba10c 24#include "AliESDVertex.h"
25#include "AliESDcascade.h"
51ad6848 26#include "AliESDkink.h"
af7ba10c 27#include "AliESDtrack.h"
af885e0f 28#include "AliESDHLTtrack.h"
85c60a8e 29#include "AliESDCaloCluster.h"
af7ba10c 30#include "AliESDv0.h"
9da38871 31#include "AliESDFMD.h"
dd735cf0 32#include "AliESDVZERO.h"
32e449be 33#include "AliMultiplicity.h"
899ca440 34#include "AliRawDataErrorLog.h"
8a8d023f 35
d75007f6 36class AliESDfriend;
53203d32 37
ae982df3 38class AliESD : public TObject {
8a8d023f 39public:
ae982df3 40 AliESD();
e149da78 41 AliESD(const AliESD&);
af7ba10c 42 virtual ~AliESD();
8a8d023f 43
d75007f6 44 void SetESDfriend(const AliESDfriend *f);
45 void GetESDfriend(AliESDfriend *f) const;
46
af885e0f 47 void SetEventNumberInFile(Int_t n) {fEventNumberInFile=n;}
48 void SetBunchCrossNumber(UShort_t n) {fBunchCrossNumber=n;}
49 void SetOrbitNumber(UInt_t n) {fOrbitNumber=n;}
50 void SetPeriodNumber(UInt_t n) {fPeriodNumber=n;}
51 void SetRunNumber(Int_t n) {fRunNumber=n;}
52 void SetTimeStamp(UInt_t timeStamp){fTimeStamp = timeStamp;}
53 void SetEventType(UInt_t eventType){fEventType = eventType;}
54 void SetTriggerMask(ULong64_t n) {fTriggerMask=n;}
55 void SetTriggerCluster(UChar_t n) {fTriggerCluster = n;}
56 void SetMagneticField(Float_t mf){fMagneticField = mf;}
57 Float_t GetMagneticField() const {return fMagneticField;}
58
59 AliESDtrack *GetTrack(Int_t i) const {
60 return (AliESDtrack *)fTracks.UncheckedAt(i);
482070f2 61 }
af885e0f 62 AliESDHLTtrack *GetHLTConfMapTrack(Int_t i) const {
63 return (AliESDHLTtrack *)fHLTConfMapTracks.UncheckedAt(i);
482070f2 64 }
af885e0f 65 AliESDHLTtrack *GetHLTHoughTrack(Int_t i) const {
66 return (AliESDHLTtrack *)fHLTHoughTracks.UncheckedAt(i);
672b5f43 67 }
af885e0f 68 AliESDMuonTrack *GetMuonTrack(Int_t i) const {
69 return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
561b7b31 70 }
af885e0f 71 AliESDPmdTrack *GetPmdTrack(Int_t i) const {
72 return (AliESDPmdTrack *)fPmdTracks.UncheckedAt(i);
ae982df3 73 }
af885e0f 74 AliESDTrdTrack *GetTrdTrack(Int_t i) const {
75 return (AliESDTrdTrack *)fTrdTracks.UncheckedAt(i);
53203d32 76 }
53203d32 77
7f68891d 78 Bool_t Clean(Float_t *cleanPars);
a023d8d8 79 Bool_t RemoveKink(Int_t i);
80 Bool_t RemoveV0(Int_t i);
7f68891d 81 Bool_t RemoveTrack(Int_t i);
d64bd07d 82
af885e0f 83 Int_t AddTrack(const AliESDtrack *t) {
84 AliESDtrack * track = new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
85 track->SetID(fTracks.GetEntriesFast()-1);
86 return track->GetID();
87
482070f2 88 }
af885e0f 89 void AddHLTConfMapTrack(const AliESDHLTtrack *t) {
90 new(fHLTConfMapTracks[fHLTConfMapTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
53203d32 91 }
af885e0f 92 void AddHLTHoughTrack(const AliESDHLTtrack *t) {
93 new(fHLTHoughTracks[fHLTHoughTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
482070f2 94 }
672b5f43 95 void AddMuonTrack(const AliESDMuonTrack *t) {
af885e0f 96 new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
672b5f43 97 }
561b7b31 98 void AddPmdTrack(const AliESDPmdTrack *t) {
af885e0f 99 new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t);
561b7b31 100 }
0ee00e25 101 void AddTrdTrack(const AliESDTrdTrack *t) {
af885e0f 102 new(fTrdTracks[fTrdTracks.GetEntriesFast()]) AliESDTrdTrack(*t);
0ee00e25 103 }
8a8d023f 104
873f1f73 105 AliESDv0 *GetV0(Int_t i) const {
af885e0f 106 return (AliESDv0 *)fV0s.UncheckedAt(i);
e23730c7 107 }
d6a49f20 108 Int_t AddV0(const AliESDv0 *v);
e23730c7 109
873f1f73 110 AliESDcascade *GetCascade(Int_t i) const {
af885e0f 111 return (AliESDcascade *)fCascades.UncheckedAt(i);
e23730c7 112 }
113 void AddCascade(const AliESDcascade *c) {
af885e0f 114 new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
e23730c7 115 }
116
51ad6848 117 AliESDkink *GetKink(Int_t i) const {
af885e0f 118 return (AliESDkink *)fKinks.UncheckedAt(i);
119 }
120 Int_t AddKink(const AliESDkink *c) {
121 AliESDkink * kink = new(fKinks[fKinks.GetEntriesFast()]) AliESDkink(*c);
122 kink->SetID(fKinks.GetEntriesFast());
123 return fKinks.GetEntriesFast()-1;
51ad6848 124 }
51ad6848 125
85c60a8e 126 AliESDCaloCluster *GetCaloCluster(Int_t i) const {
af885e0f 127 return (AliESDCaloCluster *)fCaloClusters.UncheckedAt(i);
128 }
129 Int_t AddCaloCluster(const AliESDCaloCluster *c) {
130 AliESDCaloCluster *clus = new(fCaloClusters[fCaloClusters.GetEntriesFast()]) AliESDCaloCluster(*c);
131 clus->SetID(fCaloClusters.GetEntriesFast()-1);
132 return fCaloClusters.GetEntriesFast()-1;
85c60a8e 133 }
af885e0f 134
135 void AddPHOSTriggerPosition(TArrayF array) { fPHOSTriggerPosition = new TArrayF(array) ; }
136 void AddPHOSTriggerAmplitudes(TArrayF array) { fPHOSTriggerAmplitudes = new TArrayF(array) ; }
137 void AddEMCALTriggerPosition(TArrayF array) { fEMCALTriggerPosition = new TArrayF(array) ; }
138 void AddEMCALTriggerAmplitudes(TArrayF array){ fEMCALTriggerAmplitudes = new TArrayF(array) ; }
873f1f73 139
af885e0f 140 void SetVertex(const AliESDVertex *vertex) {
141 new (&fSPDVertex) AliESDVertex(*vertex);
9f57c1aa 142 }
af885e0f 143 const AliESDVertex *GetVertex() const {return &fSPDVertex;}
144
145 void SetMultiplicity(const AliMultiplicity *mul) {
146 new (&fSPDMult) AliMultiplicity(*mul);
9f57c1aa 147 }
af885e0f 148 const AliMultiplicity *GetMultiplicity() const {return &fSPDMult;}
9f57c1aa 149
af885e0f 150 void SetPrimaryVertex(const AliESDVertex *vertex) {
151 new (&fPrimaryVertex) AliESDVertex(*vertex);
152 }
153 const AliESDVertex *GetPrimaryVertex() const {return &fPrimaryVertex;}
154
155 void SetDiamond(const AliESDVertex *vertex) {
156 fDiamondXY[0]=vertex->GetXv();
157 fDiamondXY[1]=vertex->GetYv();
158 Double_t cov[6];
159 vertex->GetCovMatrix(cov);
160 fDiamondCovXY[0]=cov[0];
161 fDiamondCovXY[1]=cov[1];
162 fDiamondCovXY[2]=cov[2];
163 }
164 Float_t GetDiamondX() const {return fDiamondXY[0];}
165 Float_t GetDiamondY() const {return fDiamondXY[1];}
166 Float_t GetSigma2DiamondX() const {return fDiamondCovXY[0];}
167 Float_t GetSigma2DiamondY() const {return fDiamondCovXY[2];}
168 void GetDiamondCovXY(Float_t cov[3]) const {
169 for(Int_t i=0;i<3;i++) cov[i]=fDiamondCovXY[i]; return;
170 }
171
172 Int_t GetEventNumberInFile() const {return fEventNumberInFile;}
173 UShort_t GetBunchCrossNumber() const {return fBunchCrossNumber;}
174 UInt_t GetOrbitNumber() const {return fOrbitNumber;}
175 UInt_t GetPeriodNumber() const {return fPeriodNumber;}
176 Int_t GetRunNumber() const {return fRunNumber;}
177 UInt_t GetTimeStamp() const { return fTimeStamp;}
178 UInt_t GetEventType() const { return fEventType;}
179 ULong64_t GetTriggerMask() const {return fTriggerMask;}
180 UChar_t GetTriggerCluster() const {return fTriggerCluster;}
181
182 Int_t GetNumberOfTracks() const {return fTracks.GetEntriesFast();}
183 Int_t GetNumberOfHLTConfMapTracks() const {return fHLTConfMapTracks.GetEntriesFast();}
184 Int_t GetNumberOfHLTHoughTracks() const {return fHLTHoughTracks.GetEntriesFast();}
185 Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
186 Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();}
187 Int_t GetNumberOfTrdTracks() const {return fTrdTracks.GetEntriesFast();}
188 Int_t GetNumberOfV0s() const {return fV0s.GetEntriesFast();}
189 Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
190 Int_t GetNumberOfKinks() const {return fKinks.GetEntriesFast();}
191 Int_t GetNumberOfCaloClusters() const {return fCaloClusters.GetEntriesFast();}
85c60a8e 192
193 Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;}
194 void SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;}
195 Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;}
196 void SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;}
af885e0f 197 TArrayF *GetEMCALTriggerPosition() const {return fEMCALTriggerPosition;}
198 TArrayF *GetEMCALTriggerAmplitudes() const {return fEMCALTriggerAmplitudes;}
85c60a8e 199
200 Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;}
201 void SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; }
202 void SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; }
203 Int_t GetFirstPHOSCluster() const { return fFirstPHOSCluster ; }
af885e0f 204 TArrayF *GetPHOSTriggerPosition() const {return fPHOSTriggerPosition;}
205 TArrayF *GetPHOSTriggerAmplitudes() const {return fPHOSTriggerAmplitudes;}
206
207
208 Float_t GetT0zVertex() const {return fT0zVertex;}
209 void SetT0zVertex(Float_t z) {fT0zVertex=z;}
210 Float_t GetT0() const {return fT0timeStart;}
211 void SetT0(Float_t timeStart) {fT0timeStart = timeStart;}
212 const Float_t * GetT0time() const {return fT0time;}
213 void SetT0time(Float_t time[24]) {
214 for (Int_t i=0; i<24; i++) fT0time[i] = time[i];
215 }
216 const Float_t * GetT0amplitude() const {return fT0amplitude;}
217 void SetT0amplitude(Float_t amp[24]) {
218 for (Int_t i=0; i<24; i++) fT0amplitude[i] = amp[i];
219 }
220
221 Float_t GetZDCN1Energy() const {return fZDCN1Energy;}
222 Float_t GetZDCP1Energy() const {return fZDCP1Energy;}
223 Float_t GetZDCN2Energy() const {return fZDCN2Energy;}
224 Float_t GetZDCP2Energy() const {return fZDCP2Energy;}
225 Float_t GetZDCEMEnergy() const {return fZDCEMEnergy;}
226 Int_t GetZDCParticipants() const {return fZDCParticipants;}
227 void SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t emEnergy,
228 Float_t n2Energy, Float_t p2Energy, Int_t participants)
229 {fZDCN1Energy=n1Energy; fZDCP1Energy=p1Energy; fZDCEMEnergy=emEnergy;
230 fZDCN2Energy=n2Energy; fZDCP2Energy=p2Energy; fZDCParticipants=participants;}
ef278eae 231
af885e0f 232 void ResetV0s() { fV0s.Clear(); }
233 void ResetCascades() { fCascades.Clear(); }
50ca5f39 234 void Reset();
bf25155c 235
ef278eae 236 void Print(Option_t *option="") const;
9da38871 237
af885e0f 238 void SetFMDData(AliESDFMD * obj) { fESDFMD = new AliESDFMD(*obj); }
239 AliESDFMD *GetFMDData(){ return fESDFMD; }
240
241 void SetVZEROData(AliESDVZERO * obj) { fESDVZERO = new AliESDVZERO(*obj); }
242 AliESDVZERO *GetVZEROData(){ return fESDVZERO; }
899ca440 243
af885e0f 244 AliRawDataErrorLog *GetErrorLog(Int_t i) const {
245 return (AliRawDataErrorLog *)fErrorLogs.UncheckedAt(i);
246 }
247 void AddRawDataErrorLog(const AliRawDataErrorLog *log) {
248 new(fErrorLogs[fErrorLogs.GetEntriesFast()]) AliRawDataErrorLog(*log);
249 }
250 Int_t GetNumberOfErrorLogs() const {return fErrorLogs.GetEntriesFast();}
8a8d023f 251
af885e0f 252protected:
e149da78 253 AliESD &operator=(const AliESD& source);
8a8d023f 254
af885e0f 255 // Event Identification
256
257 Int_t fEventNumberInFile;// running Event count in the file
258 UShort_t fBunchCrossNumber;// Bunch Crossing Number
259 UInt_t fOrbitNumber; // Orbit Number
260 UInt_t fPeriodNumber; // Period Number
261 Int_t fRunNumber; // Run Number
262 UInt_t fTimeStamp; // Time stamp
263 UInt_t fEventType; // Type of Event
264 ULong64_t fTriggerMask; // Trigger Type (mask)
265 UChar_t fTriggerCluster; // Trigger cluster (mask)
266 Int_t fRecoVersion; // Version of reconstruction
267 Float_t fMagneticField; // Solenoid Magnetic Field in kG : for compatibility with AliMagF
268
269 Float_t fZDCN1Energy; // reconstructed energy in the neutron ZDC
270 Float_t fZDCP1Energy; // reconstructed energy in the proton ZDC
271 Float_t fZDCN2Energy; // reconstructed energy in the neutron ZDC
272 Float_t fZDCP2Energy; // reconstructed energy in the proton ZDC
273 Float_t fZDCEMEnergy; // reconstructed energy in the electromagnetic ZDC
274 Int_t fZDCParticipants; // number of participants estimated by the ZDC
275
276 Float_t fT0zVertex; // vertex z position estimated by the T0
277 AliESDVertex fSPDVertex; // Primary vertex estimated by the SPD
278 AliESDVertex fPrimaryVertex; // Primary vertex estimated using ESD tracks
279 Float_t fDiamondXY[2]; // Interaction diamond (x,y) in RUN
280 Float_t fDiamondCovXY[3]; // Interaction diamond covariance (x,y) in RUN
281 AliMultiplicity fSPDMult; // SPD tracklet multiplicity
282
283 Float_t fT0timeStart; // interaction time estimated by the T0
284 Float_t fT0time[24]; // best TOF on each T0 PMT
285 Float_t fT0amplitude[24]; // number of particles(MIPs) on each T0 PMT
286
287 TClonesArray fTracks; // ESD tracks
288 TClonesArray fHLTConfMapTracks;// HLT ESD tracks from Conformal Mapper method
289 TClonesArray fHLTHoughTracks; // HLT ESD tracks from Hough Transform method
290 TClonesArray fMuonTracks; // MUON ESD tracks
291 TClonesArray fPmdTracks; // PMD ESD tracks
292 TClonesArray fTrdTracks; // TRD ESD tracks (triggered)
293 TClonesArray fV0s; // V0 vertices
294 TClonesArray fCascades; // Cascade vertices
295 TClonesArray fKinks; // Kinks
296 TClonesArray fCaloClusters; // Calorimeter clusters for PHOS/EMCAL
85c60a8e 297 Int_t fEMCALClusters; // Number of EMCAL clusters (subset of caloclusters)
298 Int_t fFirstEMCALCluster; // First EMCAL cluster in the fCaloClusters list
af885e0f 299 TArrayF *fEMCALTriggerPosition; ///(x,y,z of 2x2 and x,y,z of nxn) not position of centroid but of patch corner
300 TArrayF *fEMCALTriggerAmplitudes; //(2x2 max ampl, 2x2 amp out of patch, nxn max ampl, nxn amp out of patch)
85c60a8e 301
302 Int_t fPHOSClusters; // Number of PHOS clusters (subset of caloclusters)
303 Int_t fFirstPHOSCluster; // First PHOS cluster in the fCaloClusters list
af885e0f 304 TArrayF *fPHOSTriggerPosition; //(x,y,z of 2x2 and x,y,z of nxn), not position of centroid but of patch corner
305 TArrayF *fPHOSTriggerAmplitudes; //(2x2 max ampl, 2x2 amp out of patch, nxn max ampl, nxn amp out of patch)
306
307 AliESDFMD *fESDFMD; // FMD object containing rough multiplicity
308 AliESDVZERO *fESDVZERO; // VZERO object containing rough multiplicity
309
310 TClonesArray fErrorLogs; // Raw-data reading error messages
899ca440 311
af885e0f 312 ClassDef(AliESD,21) //ESD class
8a8d023f 313};
8a8d023f 314#endif
315