STEER/AliESD.h

   1 #ifndef ALIESD_H
   2 #define ALIESD_H
   3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   4  * See cxx source for full Copyright notice                               */
   5
   6
   7 /* $Id$ */
   8
   9 //-------------------------------------------------------------------------
  10 //                          Class AliESD
  11 //   This is the class to deal with during the physical analysis of data
  12 //
  13 //         Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch
  14 //-------------------------------------------------------------------------
  15
  16 #include <TClonesArray.h>
  17 #include <TObject.h>
  18
  19 #include "AliESDMuonTrack.h"
  20 #include "AliESDPmdTrack.h"
  21 #include "AliESDVertex.h"
  22 #include "AliESDcascade.h"
  23 #include "AliESDtrack.h"
  24 #include "AliESDv0.h"
  25
  26 class AliESD : public TObject {
  27 public:
  28   AliESD();
  29   virtual ~AliESD();
  30
  31   void SetEventNumber(Int_t n) {fEventNumber=n;}
  32   void SetRunNumber(Int_t n) {fRunNumber=n;}
  33   void SetTrigger(Long_t n) {fTrigger=n;}
  34   void SetMagneticField(Float_t mf){fMagneticField = mf;}
  35   Float_t GetMagneticField() const {return fMagneticField;}
  36
  37   AliESDtrack *GetTrack(Int_t i) const {
  38     return (AliESDtrack *)fTracks.UncheckedAt(i);
  39   }
  40   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
  41     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
  42   }
  43   AliESDPmdTrack *GetPmdTrack(Int_t i) const {
  44     return (AliESDPmdTrack *)fPmdTracks.UncheckedAt(i);
  45   }
  46
  47   void AddTrack(const AliESDtrack *t) {
  48     new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
  49   }
  50
  51   void AddMuonTrack(const AliESDMuonTrack *t) {
  52     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
  53   }
  54   void AddPmdTrack(const AliESDPmdTrack *t) {
  55     new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t);
  56   }
  57
  58   AliESDv0 *GetV0(Int_t i) const {
  59     return (AliESDv0 *)fV0s.UncheckedAt(i);
  60   }
  61   void AddV0(const AliESDv0 *v) {
  62     new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
  63   }
  64
  65   AliESDcascade *GetCascade(Int_t i) const {
  66     return (AliESDcascade *)fCascades.UncheckedAt(i);
  67   }
  68   void AddCascade(const AliESDcascade *c) {
  69     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
  70   }
  71
  72   void SetVertex(AliESDVertex* vertex) {
  73     new(&fPrimaryVertex) AliESDVertex(*vertex);
  74   }
  75   const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
  76
  77   Int_t  GetEventNumber() const {return fEventNumber;}
  78   Int_t  GetRunNumber() const {return fRunNumber;}
  79   Long_t GetTrigger() const {return fTrigger;}
  80
  81   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
  82   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
  83   Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();}
  84   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
  85   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
  86   Int_t GetNumberOfPHOSParticles() const {return fPHOSParticles;}
  87   void  SetNumberOfPHOSParticles(Int_t part) { fPHOSParticles = part ; }
  88   void  SetFirstPHOSParticle(Int_t index) { fFirstPHOSParticle = index ; }
  89   Int_t GetNumberOfEMCALParticles() const {return fEMCALParticles;}
  90   void  SetNumberOfEMCALParticles(Int_t part) { fEMCALParticles = part ; }
  91   void  SetFirstEMCALParticle(Int_t index) { fFirstEMCALParticle = index ; }
  92
  93
  94   Float_t GetT0zVertex() const {return fT0zVertex;}
  95   void SetT0zVertex(Float_t z) {fT0zVertex=z;}
  96
  97   void  Print(Option_t *option="") const;
  98
  99 protected:
 100
 101   // Event Identification
 102   Int_t        fEventNumber;     // Event Number
 103   Int_t        fRunNumber;       // Run Number
 104   Long_t       fTrigger;         // Trigger Type
 105   Int_t        fRecoVersion;     // Version of reconstruction
 106   Float_t      fMagneticField;   // Solenoid Magnetic Field in kG : for compatibility with AliMagF
 107
 108   Float_t      fT0zVertex;       // vertex z position estimated by the START
 109   AliESDVertex  fPrimaryVertex;  // Primary vertex estimated by the ITS
 110
 111   TClonesArray  fTracks;         // ESD tracks
 112   TClonesArray  fMuonTracks;     // MUON ESD tracks
 113   TClonesArray  fPmdTracks;      // PMD ESD tracks
 114   TClonesArray  fV0s;            // V0 vertices
 115   TClonesArray  fCascades;       // Cascade vertices
 116   Int_t         fPHOSParticles;  // Number of PHOS particles (stored as fTracks)
 117   Int_t         fEMCALParticles; // Number of EMCAL particles (stored as fTracks)
 118   Int_t         fFirstPHOSParticle; // First PHOS particle in the fTracks list
 119   Int_t         fFirstEMCALParticle;// First EMCAL particle in the fTracks list
 120
 121   ClassDef(AliESD,6)  //ESD class
 122                       //ver. 2: Magnetic Field Added; skowron
 123 };
 124
 125 #endif
 126