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