STEER/AliESD.h

   1 #ifndef ALIESDEVENT_H
   2 #define ALIESDEVENT_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 "TObject.h"
  17 #include "TClonesArray.h"
  18 #include  "AliESDVertex.h"
  19 #include  "AliESDtrack.h"
  20 #include  "AliESDMuonTrack.h"
  21 #include  "AliESDCaloTrack.h"
  22 #include  "AliESDv0.h"
  23 #include  "AliESDcascade.h"
  24
  25 class AliESD : public TObject {
  26 public:
  27   AliESD();
  28   virtual ~AliESD() {
  29     fTracks.Delete();
  30     fCaloTracks.Delete();
  31     fMuonTracks.Delete();
  32     fV0s.Delete();
  33     fCascades.Delete();
  34   }
  35
  36   void SetEventNumber(Int_t n) {fEventNumber=n;}
  37   void SetRunNumber(Int_t n) {fRunNumber=n;}
  38   void SetTrigger(Long_t n) {fTrigger=n;}
  39   void SetMagneticField(Float_t mf){fMagneticField = mf;}
  40   Float_t GetMagneticField() const {return fMagneticField;}
  41
  42   AliESDtrack *GetTrack(Int_t i) const {
  43     return (AliESDtrack *)fTracks.UncheckedAt(i);
  44   }
  45   AliESDCaloTrack *GetCaloTrack(Int_t i) const {
  46     return (AliESDCaloTrack *)fCaloTracks.UncheckedAt(i);
  47   }
  48   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
  49     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
  50   }
  51
  52   void AddTrack(const AliESDtrack *t) {
  53     new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
  54   }
  55   void AddCaloTrack(const AliESDCaloTrack *t) {
  56     new(fCaloTracks[fCaloTracks.GetEntriesFast()]) AliESDCaloTrack(*t);
  57   }
  58   void AddMuonTrack(const AliESDMuonTrack *t) {
  59     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
  60   }
  61
  62   AliESDv0 *GetV0(Int_t i) const {
  63     return (AliESDv0 *)fV0s.UncheckedAt(i);
  64   }
  65   void AddV0(const AliESDv0 *v) {
  66     new(fV0s[fV0s.GetEntriesFast()]) AliESDv0(*v);
  67   }
  68
  69   AliESDcascade *GetCascade(Int_t i) const {
  70     return (AliESDcascade *)fCascades.UncheckedAt(i);
  71   }
  72   void AddCascade(const AliESDcascade *c) {
  73     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
  74   }
  75
  76   void SetVertex(AliESDVertex* vertex) {
  77     new(&fPrimaryVertex) AliESDVertex(*vertex);
  78   }
  79   const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
  80
  81   Int_t  GetEventNumber() const {return fEventNumber;}
  82   Int_t  GetRunNumber() const {return fRunNumber;}
  83   Long_t GetTrigger() const {return fTrigger;}
  84
  85   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
  86   Int_t GetNumberOfCaloTracks() const {return fCaloTracks.GetEntriesFast();}
  87   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
  88   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
  89   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
  90
  91   void  Print(Option_t *option="") const;
  92
  93 protected:
  94
  95   // Event Identification
  96   Int_t        fEventNumber;     // Event Number
  97   Int_t        fRunNumber;       // Run Number
  98   Long_t       fTrigger;         // Trigger Type
  99   Int_t        fRecoVersion;     // Version of reconstruction
 100   Float_t      fMagneticField;   // Solenoid Magnetic Field in kG : for compatibility with AliMagF
 101
 102   AliESDVertex  fPrimaryVertex;  // Primary vertex
 103
 104   TClonesArray  fTracks;         // ESD tracks
 105   TClonesArray  fCaloTracks;     // Calorimeters' ESD tracks
 106   TClonesArray  fMuonTracks;     // MUON ESD tracks
 107   TClonesArray  fV0s;            // V0 vertices
 108   TClonesArray  fCascades;       // Cascade vertices
 109
 110   ClassDef(AliESD,3)  //ESD class
 111                       //ver. 2: Magnetic Field Added; skowron
 112 };
 113
 114 #endif
 115