]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESD.h
Additional protection in case of zero magnetic field (Yu.Belikov)
[u/mrichter/AliRoot.git] / STEER / AliESD.h
index 2786872fa83847e7ca1a627fc33d30be96272c78..d6352bab8579d4dc2b82cedb88fdbc745b70674a 100644 (file)
@@ -1,5 +1,5 @@
-#ifndef ALIESDEVENT_H
-#define ALIESDEVENT_H
+#ifndef ALIESD_H
+#define ALIESD_H
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
  * See cxx source for full Copyright notice                               */
 
 //         Origin: Iouri Belikov, CERN, Jouri.Belikov@cern.ch 
 //-------------------------------------------------------------------------
 
-#include "TObject.h"
-#include "TClonesArray.h"
-#include  "AliESDtrack.h"
-#include  "AliESDMuonTrack.h"
-#include  "AliESDCaloTrack.h"
-#include  "AliESDv0.h"
-#include  "AliESDcascade.h"
+#include <TClonesArray.h>
+#include <TObject.h>
+
+#include "AliESDMuonTrack.h"
+#include "AliESDPmdTrack.h"
+#include "AliESDVertex.h"
+#include "AliESDcascade.h"
+#include "AliESDtrack.h"
+#include "AliESDHLTtrack.h"
+#include "AliESDv0.h"
 
 class AliESD : public TObject {
 public:
   AliESD();
-  virtual ~AliESD() {
-    fTracks.Delete();
-    fCaloTracks.Delete();
-    fMuonTracks.Delete();
-    fV0s.Delete();
-    fCascades.Delete();
-  }
+  virtual ~AliESD(); 
 
   void SetEventNumber(Int_t n) {fEventNumber=n;}
-
+  void SetRunNumber(Int_t n) {fRunNumber=n;}
+  void SetTrigger(Long_t n) {fTrigger=n;}
+  void SetMagneticField(Float_t mf){fMagneticField = mf;}
+  Float_t GetMagneticField() const {return fMagneticField;}
+  
   AliESDtrack *GetTrack(Int_t i) const {
     return (AliESDtrack *)fTracks.UncheckedAt(i);
   }
-  AliESDCaloTrack *GetCaloTrack(Int_t i) const {
-    return (AliESDCaloTrack *)fCaloTracks.UncheckedAt(i);
+  AliESDHLTtrack *GetHLTConfMapTrack(Int_t i) const {
+    return (AliESDHLTtrack *)fHLTConfMapTracks.UncheckedAt(i);
+  }
+  AliESDHLTtrack *GetHLTHoughTrack(Int_t i) const {
+    return (AliESDHLTtrack *)fHLTHoughTracks.UncheckedAt(i);
   }
   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
     return (AliESDMuonTrack *)fMuonTracks.UncheckedAt(i);
   }
+  AliESDPmdTrack *GetPmdTrack(Int_t i) const {
+    return (AliESDPmdTrack *)fPmdTracks.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 AddHLTConfMapTrack(const AliESDHLTtrack *t) {
+    new(fHLTConfMapTracks[fHLTConfMapTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
+  }
+  void AddHLTHoughTrack(const AliESDHLTtrack *t) {
+    new(fHLTHoughTracks[fHLTHoughTracks.GetEntriesFast()]) AliESDHLTtrack(*t);
   }
   void AddMuonTrack(const AliESDMuonTrack *t) {
     new(fMuonTracks[fMuonTracks.GetEntriesFast()]) AliESDMuonTrack(*t);
   }
+  void AddPmdTrack(const AliESDPmdTrack *t) {
+    new(fPmdTracks[fPmdTracks.GetEntriesFast()]) AliESDPmdTrack(*t);
+  }
 
   AliESDv0 *GetV0(Int_t i) const {
     return (AliESDv0 *)fV0s.UncheckedAt(i);
@@ -68,19 +81,46 @@ public:
     new(fCascades[fCascades.GetEntriesFast()]) AliESDcascade(*c);
   }
 
-  void SetVertex(const Double_t *vtx, const Double_t *cvtx=0);
-  void GetVertex(Double_t *vtx, Double_t *cvtx) const;
+  void SetVertex(AliESDVertex* vertex) {
+    new(&fPrimaryVertex) AliESDVertex(*vertex);
+  }
+  const AliESDVertex* GetVertex() const {return &fPrimaryVertex;};
 
   Int_t  GetEventNumber() const {return fEventNumber;}
   Int_t  GetRunNumber() const {return fRunNumber;}
   Long_t GetTrigger() const {return fTrigger;}
   
   Int_t GetNumberOfTracks()     const {return fTracks.GetEntriesFast();}
-  Int_t GetNumberOfCaloTracks() const {return fCaloTracks.GetEntriesFast();}
+  Int_t GetNumberOfHLTConfMapTracks()     const {return fHLTConfMapTracks.GetEntriesFast();}
+  Int_t GetNumberOfHLTHoughTracks()     const {return fHLTHoughTracks.GetEntriesFast();}
   Int_t GetNumberOfMuonTracks() const {return fMuonTracks.GetEntriesFast();}
+  Int_t GetNumberOfPmdTracks() const {return fPmdTracks.GetEntriesFast();}
   Int_t GetNumberOfV0s()      const {return fV0s.GetEntriesFast();}
   Int_t GetNumberOfCascades() const {return fCascades.GetEntriesFast();}
-  
+  Int_t GetNumberOfPHOSParticles() const {return fPHOSParticles;}
+  void  SetNumberOfPHOSParticles(Int_t part) { fPHOSParticles = part ; }
+  void  SetFirstPHOSParticle(Int_t index) { fFirstPHOSParticle = index ; } 
+  Int_t GetNumberOfEMCALParticles() const {return fEMCALParticles;}
+  void  SetNumberOfEMCALParticles(Int_t part) { fEMCALParticles = part ; }
+  void  SetFirstEMCALParticle(Int_t index) { fFirstEMCALParticle = index ; } 
+
+
+  Float_t GetT0zVertex() const {return fT0zVertex;}
+  void SetT0zVertex(Float_t z) {fT0zVertex=z;}
+
+  Float_t GetZDCNEnergy() const {return fZDCNEnergy;}
+  Float_t GetZDCPEnergy() const {return fZDCPEnergy;}
+  Float_t GetZDCEMEnergy() const {return fZDCEMEnergy;}
+  Int_t   GetZDCParticipants() const {return fZDCParticipants;}
+  void    SetZDC(Float_t nEnergy, Float_t pEnergy, Float_t emEnergy,
+                 Int_t participants) 
+   {fZDCNEnergy=nEnergy; fZDCPEnergy=pEnergy; fZDCEMEnergy=emEnergy;
+    fZDCParticipants=participants;}
+
+  void  Reset();
+
+  void  Print(Option_t *option="") const;
+   
 protected:
 
   // Event Identification
@@ -88,18 +128,29 @@ protected:
   Int_t        fRunNumber;       // Run Number
   Long_t       fTrigger;         // Trigger Type
   Int_t        fRecoVersion;     // Version of reconstruction 
-
-  Double_t fVtx[3];              // Primary vertex position
-  Double_t fCovVtx[6];           // Cov. matrix of the primary vertex position
-
-  TClonesArray  fTracks;         // ESD tracks
-  TClonesArray  fCaloTracks;     // Calorimeters' ESD tracks
-  TClonesArray  fMuonTracks;     // MUON ESD tracks
-  TClonesArray  fV0s;            // V0 vertices
-  TClonesArray  fCascades;       // Cascade vertices
-  
-  ClassDef(AliESD,1)  //ESD class 
+  Float_t      fMagneticField;   // Solenoid Magnetic Field in kG : for compatibility with AliMagF
+
+  Float_t      fZDCNEnergy;      // reconstructed energy in the neutron ZDC
+  Float_t      fZDCPEnergy;      // reconstructed energy in the proton ZDC
+  Float_t      fZDCEMEnergy;     // reconstructed energy in the electromagnetic ZDC
+  Int_t        fZDCParticipants; // number of participants estimated by the ZDC
+
+  Float_t      fT0zVertex;       // vertex z position estimated by the START
+  AliESDVertex fPrimaryVertex;   // Primary vertex estimated by the ITS
+
+  TClonesArray fTracks;          // ESD tracks
+  TClonesArray fHLTConfMapTracks;// HLT ESD tracks from Conformal Mapper method
+  TClonesArray fHLTHoughTracks;  // HLT ESD tracks from Hough Transform method
+  TClonesArray fMuonTracks;      // MUON ESD tracks
+  TClonesArray fPmdTracks;       // PMD ESD tracks
+  TClonesArray fV0s;             // V0 vertices
+  TClonesArray fCascades;        // Cascade vertices
+  Int_t        fPHOSParticles;   // Number of PHOS particles (stored as fTracks)
+  Int_t        fEMCALParticles;  // Number of EMCAL particles (stored as fTracks)
+  Int_t        fFirstPHOSParticle; // First PHOS particle in the fTracks list 
+  Int_t        fFirstEMCALParticle;// First EMCAL particle in the fTracks list 
+  ClassDef(AliESD,8)  //ESD class 
 };
-
 #endif