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