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
  20 #include "AliESDMuonTrack.h"
  21 #include "AliESDPmdTrack.h"
  22 #include "AliESDVertex.h"
  23 #include "AliESDcascade.h"
  24 #include "AliESDkink.h"
  25 #include "AliESDtrack.h"
  26 #include "AliESDHLTtrack.h"
  27 #include "AliESDCaloCluster.h"
  28 #include "AliESDv0.h"
  29 #include "AliESDV0MI.h"
  30 #include "AliESDFMD.h"
  31
  32 class AliESD : public TObject {
  33 public:
  34   AliESD();
  35   virtual ~AliESD();
  36
  37   void SetEventNumber(Int_t n) {fEventNumber=n;}
  38   void SetRunNumber(Int_t n) {fRunNumber=n;}
  39   void SetTrigger(Long_t n) {fTrigger=n;}
  40   void SetMagneticField(Float_t mf){fMagneticField = mf;}
  41   Float_t GetMagneticField() const {return fMagneticField;}
  42
  43   AliESDtrack *GetTrack(Int_t i) const {
  44     return (AliESDtrack *)fTracks.UncheckedAt(i);
  45   }
  46   AliESDHLTtrack *GetHLTConfMapTrack(Int_t i) const {
  47     return (AliESDHLTtrack *)fHLTConfMapTracks.UncheckedAt(i);
  48   }
  49   AliESDHLTtrack *GetHLTHoughTrack(Int_t i) const {
  50     return (AliESDHLTtrack *)fHLTHoughTracks.UncheckedAt(i);
  51   }
  52   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
  53     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
  54   }
  55   AliESDPmdTrack *GetPmdTrack(Int_t i) const {
  56     return (AliESDPmdTrack *)fPmdTracks.UncheckedAt(i);
  57   }
  58
  59   Int_t  AddTrack(const AliESDtrack *t) {
  60     AliESDtrack * track = new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
  61     track->SetID(fTracks.GetEntriesFast()-1);
  62     return  track->GetID();
  63
  64   }
  65   void AddHLTConfMapTrack(const AliESDHLTtrack *t) {
  66     new(fHLTConfMapTracks[fHLTConfMapTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
  67   }
  68   void AddHLTHoughTrack(const AliESDHLTtrack *t) {
  69     new(fHLTHoughTracks[fHLTHoughTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
  70   }
  71   void AddMuonTrack(const AliESDMuonTrack *t) {
  72     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
  73   }
  74   void AddPmdTrack(const AliESDPmdTrack *t) {
  75     new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t);
  76   }
  77
  78   AliESDv0 *GetV0(Int_t i) const {
  79     return (AliESDv0 *)fV0s.UncheckedAt(i);
  80   }
  81   void AddV0(const AliESDv0 *v) {
  82     new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
  83   }
  84   void UpdateV0PIDs();
  85
  86   AliESDcascade *GetCascade(Int_t i) const {
  87     return (AliESDcascade *)fCascades.UncheckedAt(i);
  88   }
  89   void AddCascade(const AliESDcascade *c) {
  90     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
  91   }
  92
  93   AliESDkink *GetKink(Int_t i) const {
  94     return (AliESDkink *)fKinks.UncheckedAt(i);
  95   }
  96   Int_t AddKink(const AliESDkink *c) {
  97     AliESDkink * kink = new(fKinks[fKinks.GetEntriesFast()]) AliESDkink(*c);
  98     kink->SetID(fKinks.GetEntriesFast());
  99     return fKinks.GetEntriesFast()-1;
 100   }
 101
 102   AliESDV0MI *GetV0MI(Int_t i) const {
 103     return (AliESDV0MI *)fV0MIs.UncheckedAt(i);
 104   }
 105   Int_t AddV0MI(const AliESDV0MI *c) {
 106     AliESDV0MI * v0 = new(fV0MIs[fV0MIs.GetEntriesFast()]) AliESDV0MI(*c);
 107     v0->SetID(fV0MIs.GetEntriesFast()-1);
 108     return fV0MIs.GetEntriesFast()-1;
 109   }
 110
 111   AliESDCaloCluster *GetCaloCluster(Int_t i) const {
 112     return (AliESDCaloCluster *)fCaloClusters.UncheckedAt(i);
 113   }
 114   Int_t AddCaloCluster(const AliESDCaloCluster *c) {
 115     AliESDCaloCluster *clus = new(fCaloClusters[fCaloClusters.GetEntriesFast()]) AliESDCaloCluster(*c);
 116     clus->SetID(fCaloClusters.GetEntriesFast()-1);
 117     return fCaloClusters.GetEntriesFast()-1;
 118   }
 119
 120   void SetVertex(const AliESDVertex* vertex) {
 121     new(&fPrimaryVertex) AliESDVertex(*vertex);
 122   }
 123   const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
 124
 125   Int_t  GetEventNumber() const {return fEventNumber;}
 126   Int_t  GetRunNumber() const {return fRunNumber;}
 127   Long_t GetTrigger() const {return fTrigger;}
 128
 129   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
 130   Int_t GetNumberOfHLTConfMapTracks()     const {return fHLTConfMapTracks.GetEntriesFast();}
 131   Int_t GetNumberOfHLTHoughTracks()     const {return fHLTHoughTracks.GetEntriesFast();}
 132   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
 133   Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();}
 134   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
 135   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
 136   Int_t GetNumberOfKinks() const {return fKinks.GetEntriesFast();}
 137   Int_t GetNumberOfV0MIs() const {return fV0MIs.GetEntriesFast();}
 138   Int_t GetNumberOfCaloClusters() const {return fCaloClusters.GetEntriesFast();}
 139
 140   Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;}
 141   void  SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;}
 142   Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;}
 143   void  SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;}
 144
 145   Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;}
 146   void  SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; }
 147   void  SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; }
 148   Int_t GetFirstPHOSCluster() const  { return fFirstPHOSCluster ; }
 149
 150   Float_t GetT0zVertex() const {return fT0zVertex;}
 151   void SetT0zVertex(Float_t z) {fT0zVertex=z;}
 152
 153   Float_t GetZDCNEnergy() const {return fZDCNEnergy;}
 154   Float_t GetZDCPEnergy() const {return fZDCPEnergy;}
 155   Float_t GetZDCEMEnergy() const {return fZDCEMEnergy;}
 156   Int_t   GetZDCParticipants() const {return fZDCParticipants;}
 157   void    SetZDC(Float_t nEnergy, Float_t pEnergy, Float_t emEnergy,
 158                  Int_t participants)
 159    {fZDCNEnergy=nEnergy; fZDCPEnergy=pEnergy; fZDCEMEnergy=emEnergy;
 160     fZDCParticipants=participants;}
 161
 162   void ResetV0s() { fV0s.Clear(); }
 163   void ResetCascades() { fCascades.Clear(); }
 164   void Reset();
 165
 166   void  Print(Option_t *option="") const;
 167
 168   void SetFMDData(AliESDFMD * obj) {
 169     fESDFMD = new AliESDFMD(*obj);
 170   }
 171
 172   AliESDFMD * GetFMDData(){ return fESDFMD;}
 173
 174 protected:
 175
 176   // Event Identification
 177   Int_t        fEventNumber;     // Event Number
 178   Int_t        fRunNumber;       // Run Number
 179   Long_t       fTrigger;         // Trigger Type
 180   Int_t        fRecoVersion;     // Version of reconstruction
 181   Float_t      fMagneticField;   // Solenoid Magnetic Field in kG : for compatibility with AliMagF
 182
 183   Float_t      fZDCNEnergy;      // reconstructed energy in the neutron ZDC
 184   Float_t      fZDCPEnergy;      // reconstructed energy in the proton ZDC
 185   Float_t      fZDCEMEnergy;     // reconstructed energy in the electromagnetic ZDC
 186   Int_t        fZDCParticipants; // number of participants estimated by the ZDC
 187
 188   Float_t      fT0zVertex;       // vertex z position estimated by the START
 189   AliESDVertex fPrimaryVertex;   // Primary vertex estimated by the ITS
 190
 191   TClonesArray fTracks;          // ESD tracks
 192   TClonesArray fHLTConfMapTracks;// HLT ESD tracks from Conformal Mapper method
 193   TClonesArray fHLTHoughTracks;  // HLT ESD tracks from Hough Transform method
 194   TClonesArray fMuonTracks;      // MUON ESD tracks
 195   TClonesArray fPmdTracks;       // PMD ESD tracks
 196   TClonesArray fV0s;             // V0 vertices
 197   TClonesArray fCascades;        // Cascade vertices
 198   TClonesArray fKinks;           // Kinks
 199   TClonesArray fV0MIs;           // V0MI
 200   TClonesArray fCaloClusters;    // Calorimeter clusters for PHOS/EMCAL
 201   Int_t        fEMCALClusters;   // Number of EMCAL clusters (subset of caloclusters)
 202   Int_t        fFirstEMCALCluster; // First EMCAL cluster in the fCaloClusters list
 203
 204   Int_t        fPHOSClusters;     // Number of PHOS clusters (subset of caloclusters)
 205   Int_t        fFirstPHOSCluster; // First PHOS cluster in the fCaloClusters list
 206
 207   AliESDFMD *  fESDFMD; // FMD object containing rough multiplicity
 208
 209   ClassDef(AliESD,9)  //ESD class
 210 };
 211 #endif
 212