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