New classes for MUON and PHOS ESD (Yu.Belikov
[u/mrichter/AliRoot.git] / STEER / AliESD.h
index eed56ba097614ba1e66690dd21c6a357ab68198a..19df050e99a2c9c2ee8fd169d22a3d3383cee155 100644 (file)
@@ -16,6 +16,8 @@
 #include "TObject.h"
 #include "TClonesArray.h"
 #include  "AliESDtrack.h"
+#include  "AliESDMuonTrack.h"
+#include  "AliESDCaloTrack.h"
 #include  "AliESDv0.h"
 #include  "AliESDcascade.h"
 
@@ -24,6 +26,8 @@ public:
   AliESD();
   virtual ~AliESD() {
     fTracks.Delete();
+    fCaloTracks.Delete();
+    fMuonTracks.Delete();
     fV0s.Delete();
     fCascades.Delete();
   }
@@ -33,9 +37,22 @@ public:
   AliESDtrack *GetTrack(Int_t i) {
     return (AliESDtrack *)fTracks.UncheckedAt(i);
   }
+  AliESDCaloTrack *GetCaloTrack(Int_t i) {
+    return (AliESDCaloTrack *)fCaloTracks.UncheckedAt(i);
+  }
+  AliESDMuonTrack *GetMuonTrack(Int_t i) {
+    return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
+  }
+
   void AddTrack(const AliESDtrack *t) {
     new(fTracks[fTracks.GetEntriesFast()]) AliESDtrack(*t);
   }
+  void AddCaloTrack(const AliESDCaloTrack *t) {
+    new(fCaloTracks[fCaloTracks.GetEntriesFast()]) AliESDCaloTrack(*t);
+  }
+  void AddMuonTrack(const AliESDMuonTrack *t) {
+    new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
+  }
 
   AliESDv0 *GetV0(Int_t i) {
     return (AliESDv0 *)fV0s.UncheckedAt(i);
@@ -55,7 +72,9 @@ public:
   Int_t  GetRunNumber() const {return fRunNumber;}
   Long_t GetTrigger() const {return fTrigger;}
   
-  Int_t GetNumberOfTracks()   const {return fTracks.GetEntriesFast();}
+  Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
+  Int_t GetNumberOfCaloTracks() const {return fCaloTracks.GetEntriesFast();}
+  Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
   
@@ -68,6 +87,8 @@ protected:
   Int_t        fRecoVersion;     // Version of reconstruction 
 
   TClonesArray  fTracks;         // ESD tracks
+  TClonesArray  fCaloTracks;     // Calorimeters' ESD tracks
+  TClonesArray  fMuonTracks;     // MUON ESD tracks
   TClonesArray  fV0s;            // V0 vertices
   TClonesArray  fCascades;       // Cascade vertices