]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDEvent.h
A getter for the number of tracks composing the vertex
[u/mrichter/AliRoot.git] / STEER / AliESDEvent.h
index 35c39f06332a1f78bc316ce2a3cfa37a299e5a2c..6c0f52c6cdf7f987576c10fe458c2e86f3dadf92 100644 (file)
@@ -9,7 +9,7 @@
 
 //-------------------------------------------------------------------------
 //                          Class AliESDEvent
-//   This is the class to deal with during the physical analysis of data.
+//   This is the class to deal with during the physics analysis of data.
 //   It also ensures the backward compatibility with the old ESD format.
 //      
 // Origin: Christian Klein-Boesing, CERN, Christian.Klein-Boesing@cern.ch 
 
 class TList;
 
-
-#include "AliESDMuonTrack.h"
-#include "AliESDPmdTrack.h"
-#include "AliESDTrdTrack.h"
-#include "AliESDVertex.h"
-#include "AliESDcascade.h"
-#include "AliESDkink.h"
-#include "AliESDtrack.h"
-#include "AliESDCaloCluster.h"
-#include "AliESDv0.h"
-#include "AliESDFMD.h"
-#include "AliESDVZERO.h"
-#include "AliMultiplicity.h"
-#include "AliRawDataErrorLog.h"
+#include "AliVEvent.h"
+// some includes for delegated methods
+#include "AliESDCaloTrigger.h"
 #include "AliESDRun.h"
 #include "AliESDHeader.h"
-#include "AliESDZDC.h"
 #include "AliESDTZERO.h"
-#include "AliESDCaloTrigger.h"
+#include "AliESDZDC.h"
+
+// AliESDtrack has to be included so that the compiler 
+// knows its inheritance tree (= that it is a AliVParticle).
+#include "AliESDtrack.h"
 
 class AliESDfriend;
 class AliESDVZERO;
 class AliESDHLTtrack;
+class AliESDVertex;
+class AliESDPmdTrack;
 class AliESDFMD;
+class AliESDkink;
+class AliESDCaloCluster;
+class AliESDCaloCells;
+class AliESDv0;
+class AliMultiplicity;
+class AliRawDataErrorLog;
+class AliESDRun;
+class AliESDTrdTrack;
+class AliESDMuonTrack;
 class AliESD;
+class AliESDcascade;
+class TRefArray;
 
-class AliESDEvent : public TObject {
+class AliESDEvent : public AliVEvent {
 public:
 
 
-  enum ESDListIndex_t   {kESDRun,
+  enum ESDListIndex   {kESDRun,
                       kHeader,
                       kESDZDC,
                       kESDFMD,
@@ -71,6 +76,8 @@ public:
                       kCascades,
                       kKinks,
                       kCaloClusters,
+                      kEMCALCells,
+                      kPHOSCells,
                       kErrorLogs,
                       kESDListN
   };
@@ -84,22 +91,22 @@ public:
   const AliESDRun*    GetESDRun() const {return fESDRun;}
 
   // Delegated methods for fESDRun
-  void    SetRunNumber(Int_t n) {fESDRun->SetRunNumber(n);}
-  Int_t   GetRunNumber() const {return fESDRun->GetRunNumber();}
-  void    SetPeriodNumber(Int_t n){fESDRun->SetPeriodNumber(n);}
-  Int_t   GetPeriodNumber() const {return fESDRun->GetPeriodNumber();}
-  void    SetMagneticField(Float_t mf){fESDRun->SetMagneticField(mf);}
-  Float_t GetMagneticField() const {return fESDRun->GetMagneticField();}
-  void SetDiamond(const AliESDVertex *vertex) { fESDRun->SetDiamond(vertex);}
-  Float_t GetDiamondX() const {return fESDRun->GetDiamondX();}
-  Float_t GetDiamondY() const {return fESDRun->GetDiamondY();}
-  Float_t GetSigma2DiamondX() const {return  fESDRun->GetSigma2DiamondX();}
-  Float_t GetSigma2DiamondY() const {return  fESDRun->GetSigma2DiamondY();}
-  void GetDiamondCovXY(Float_t cov[3]) const {fESDRun->GetDiamondCovXY(cov);}   
+  void     SetRunNumber(Int_t n) {fESDRun->SetRunNumber(n);}
+  Int_t    GetRunNumber() const {return fESDRun->GetRunNumber();}
+  void     SetPeriodNumber(UInt_t n){fESDRun->SetPeriodNumber(n);}
+  UInt_t   GetPeriodNumber() const {return fESDRun->GetPeriodNumber();}
+  void     SetMagneticField(Double_t mf){fESDRun->SetMagneticField(mf);}
+  Double_t GetMagneticField() const {return fESDRun->GetMagneticField();}
+  void     SetDiamond(const AliESDVertex *vertex) { fESDRun->SetDiamond(vertex);}
+  Double_t  GetDiamondX() const {return fESDRun->GetDiamondX();}
+  Double_t  GetDiamondY() const {return fESDRun->GetDiamondY();}
+  Double_t  GetSigma2DiamondX() const {return  fESDRun->GetSigma2DiamondX();}
+  Double_t  GetSigma2DiamondY() const {return  fESDRun->GetSigma2DiamondY();}
+  void      GetDiamondCovXY(Float_t cov[3]) const {fESDRun->GetDiamondCovXY(cov);}   
   
 
   // HEADER
-  const AliESDHeader* GetHeader() const {return fHeader;}
+  AliESDHeader* GetHeader() const {return fHeader;}
 
   // Delegated methods for fHeader
   void      SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
@@ -110,6 +117,7 @@ public:
   //  void      SetRunNumber(Int_t n) {fHeader->SetRunNumber(n);}
   void      SetBunchCrossNumber(UShort_t n) {fHeader->SetBunchCrossNumber(n);}
   void      SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
+  
   ULong64_t GetTriggerMask() const {return fHeader->GetTriggerMask();}
   UInt_t    GetOrbitNumber() const {return fHeader->GetOrbitNumber();}
   UInt_t    GetTimeStamp()  const { return fHeader->GetTimeStamp();}
@@ -119,18 +127,18 @@ public:
   UChar_t   GetTriggerCluster() const {return fHeader->GetTriggerCluster();}
 
   // ZDC CKB: put this in the header?
-  const AliESDZDC*    GetESDZDC() const {return fESDZDC;}
+  AliESDZDC*    GetESDZDC() const {return fESDZDC;}
 
   // Delegated methods for fESDZDC
-  Float_t GetZDCN1Energy() const {return fESDZDC->GetZDCN1Energy();}
-  Float_t GetZDCP1Energy() const {return fESDZDC->GetZDCP1Energy();}
-  Float_t GetZDCN2Energy() const {return fESDZDC->GetZDCN2Energy();}
-  Float_t GetZDCP2Energy() const {return fESDZDC->GetZDCP2Energy();}
-  Float_t GetZDCEMEnergy() const {return fESDZDC->GetZDCEMEnergy();}
+  Double_t GetZDCN1Energy() const {return fESDZDC->GetZDCN1Energy();}
+  Double_t GetZDCP1Energy() const {return fESDZDC->GetZDCP1Energy();}
+  Double_t GetZDCN2Energy() const {return fESDZDC->GetZDCN2Energy();}
+  Double_t GetZDCP2Energy() const {return fESDZDC->GetZDCP2Energy();}
+  Double_t GetZDCEMEnergy(Int_t i=0) const {return fESDZDC->GetZDCEMEnergy(i);}
   Int_t   GetZDCParticipants() const {return fESDZDC->GetZDCParticipants();}
-  void    SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t emEnergy,
+  void    SetZDC(Float_t n1Energy, Float_t p1Energy, Float_t em1Energy, Float_t em2Energy,
                  Float_t n2Energy, Float_t p2Energy, Int_t participants)
-  {fESDZDC->SetZDC(n1Energy, p1Energy, emEnergy, n2Energy, p2Energy, participants);}
+  {fESDZDC->SetZDC(n1Energy, p1Energy, em1Energy, em2Energy, n2Energy, p2Energy, participants);}
 
 
   // FMD
@@ -142,13 +150,13 @@ public:
   const AliESDTZERO*    GetESDTZERO() const {return fESDTZERO;}
   // delegetated methods for fESDTZERO
 
-  Float_t GetT0zVertex() const {return fESDTZERO->GetT0zVertex();}
+  Double_t GetT0zVertex() const {return fESDTZERO->GetT0zVertex();}
   void SetT0zVertex(Float_t z) {fESDTZERO->SetT0zVertex(z);}
-  Float_t GetT0() const {return fESDTZERO->GetT0();}
+  Double_t GetT0() const {return fESDTZERO->GetT0();}
   void SetT0(Float_t timeStart) {fESDTZERO->SetT0(timeStart);}
-  const Float_t * GetT0time() const {return fESDTZERO->GetT0time();}
+  const Double_t * GetT0time() const {return fESDTZERO->GetT0time();}
   void SetT0time(Float_t time[24]) {fESDTZERO->SetT0time(time);}
-  const Float_t * GetT0amplitude() const {return fESDTZERO->GetT0amplitude();}
+  const Double_t * GetT0amplitude() const {return fESDTZERO->GetT0amplitude();}
   void SetT0amplitude(Float_t amp[24]){fESDTZERO->SetT0amplitude(amp);}
 
   // VZERO 
@@ -161,26 +169,22 @@ public:
 
 
 
-  void SetVertex(const AliESDVertex *vertex) {
-    *fSPDVertex  = *vertex;
-    fSPDVertex->SetName("SPDVertex");// error prone use class wide names?
-    //CKB or new with placement
-  }
+  void SetVertex(const AliESDVertex *vertex); 
   const AliESDVertex *GetVertex() const {return fSPDVertex;}
 
-  void SetPrimaryVertex(const AliESDVertex *vertex) {
-    *fPrimaryVertex = *vertex;
-    fPrimaryVertex->SetName("PrimaryVertex");// error prone use class wide names?
-  }
+  void SetPrimaryVertex(const AliESDVertex *vertex);
   const AliESDVertex *GetPrimaryVertex() const {return fPrimaryVertex;}
 
-  void SetMultiplicity(const AliMultiplicity *mul) {
-    *fSPDMult = *mul;
-    // CKB 
-    //     new (&fSPDMult) AliMultiplicity(*mul);
-  }
+  void SetMultiplicity(const AliMultiplicity *mul);
+
   const AliMultiplicity *GetMultiplicity() const {return fSPDMult;}
-  
+
+
+  Bool_t Clean(Float_t *cleanPars);
+  Bool_t RemoveKink(Int_t i);
+  Bool_t RemoveV0(Int_t i);
+  Bool_t RemoveTrack(Int_t i);
+
   AliESDtrack *GetTrack(Int_t i) const {
     return (AliESDtrack *)fTracks->UncheckedAt(i);
   }
@@ -211,26 +215,22 @@ public:
   AliESDMuonTrack *GetMuonTrack(Int_t i) const {
     return (AliESDMuonTrack *)fMuonTracks->UncheckedAt(i);
   }
-  void AddMuonTrack(const AliESDMuonTrack *t) {
-    TClonesArray &fmu = *fMuonTracks;
-    new(fmu[fMuonTracks->GetEntriesFast()]) AliESDMuonTrack(*t);
-  }
+
+  void AddMuonTrack(const AliESDMuonTrack *t);
 
   AliESDPmdTrack *GetPmdTrack(Int_t i) const {
     return (AliESDPmdTrack *)fPmdTracks->UncheckedAt(i);
   }
-  void AddPmdTrack(const AliESDPmdTrack *t) {
-    TClonesArray &fpmd = *fPmdTracks;
-    new(fpmd[fPmdTracks->GetEntriesFast()]) AliESDPmdTrack(*t);
-  }
+
+  void AddPmdTrack(const AliESDPmdTrack *t);
+
 
   AliESDTrdTrack *GetTrdTrack(Int_t i) const {
     return (AliESDTrdTrack *)fTrdTracks->UncheckedAt(i);
   }
-  void AddTrdTrack(const AliESDTrdTrack *t) {
-    TClonesArray &ftrd = *fTrdTracks;
-    new(ftrd[fTrdTracks->GetEntriesFast()]) AliESDTrdTrack(*t);
-  }
+
+  
+  void AddTrdTrack(const AliESDTrdTrack *t);
 
   AliESDv0 *GetV0(Int_t i) const {
     return (AliESDv0*)fV0s->UncheckedAt(i);
@@ -240,10 +240,8 @@ public:
   AliESDcascade *GetCascade(Int_t i) const {
     return (AliESDcascade *)fCascades->UncheckedAt(i);
   }
-  void AddCascade(const AliESDcascade *c) {
-    TClonesArray &fc = *fCascades;
-    new(fc[fCascades->GetEntriesFast()]) AliESDcascade(*c);
-  }
+
+  void AddCascade(const AliESDcascade *c);
 
   AliESDkink *GetKink(Int_t i) const {
     return (AliESDkink *)fKinks->UncheckedAt(i);
@@ -253,16 +251,17 @@ public:
   AliESDCaloCluster *GetCaloCluster(Int_t i) const {
     return (AliESDCaloCluster *)fCaloClusters->UncheckedAt(i);
   }
+
   Int_t AddCaloCluster(const AliESDCaloCluster *c);
 
+  AliESDCaloCells *GetEMCALCells() const {return fEMCALCells; }  
+  AliESDCaloCells *GetPHOSCells() const {return fPHOSCells; }  
+
   AliRawDataErrorLog *GetErrorLog(Int_t i) const {
     return (AliRawDataErrorLog *)fErrorLogs->UncheckedAt(i);
   }
-  void  AddRawDataErrorLog(const AliRawDataErrorLog *log) {
-    // CKB inline this??
-    TClonesArray &errlogs = *fErrorLogs;
-    new(errlogs[errlogs.GetEntriesFast()])  AliRawDataErrorLog(*log);
-  }
+  void  AddRawDataErrorLog(const AliRawDataErrorLog *log);
+
   Int_t GetNumberOfErrorLogs()   const {return fErrorLogs->GetEntriesFast();}
 
     
@@ -271,7 +270,6 @@ public:
   void AddEMCALTriggerPosition(TArrayF array)  { fEMCALTrigger->AddTriggerPosition(array); }
   void AddEMCALTriggerAmplitudes(TArrayF array){ fEMCALTrigger->AddTriggerAmplitudes(array); }
 
-
   Int_t GetNumberOfTracks()     const {return fTracks->GetEntriesFast();}
   Int_t GetNumberOfHLTConfMapTracks()     const {return 0;} 
   // fHLTConfMapTracks->GetEntriesFast();}
@@ -284,19 +282,27 @@ public:
   Int_t GetNumberOfV0s()      const {return fV0s->GetEntriesFast();}
   Int_t GetNumberOfCascades() const {return fCascades->GetEntriesFast();}
   Int_t GetNumberOfKinks() const {return fKinks->GetEntriesFast();}
+  
+  Int_t GetEMCALClusters(TRefArray *clusters) const;
+  Int_t GetPHOSClusters(TRefArray *clusters) const;
   Int_t GetNumberOfCaloClusters() const {return fCaloClusters->GetEntriesFast();}
 
+  
+  // Remove this stuff CKB?
+  //---------------------------------------------------
   Int_t GetNumberOfEMCALClusters() const {return fEMCALClusters;}
   void  SetNumberOfEMCALClusters(Int_t clus) {fEMCALClusters = clus;}
   Int_t GetFirstEMCALCluster() const {return fFirstEMCALCluster;}
   void  SetFirstEMCALCluster(Int_t index) {fFirstEMCALCluster = index;}
-  TArrayF *GetEMCALTriggerPosition() const {return  fEMCALTrigger->GetTriggerPosition();}
-  TArrayF *GetEMCALTriggerAmplitudes() const {return  fEMCALTrigger->GetTriggerAmplitudes();}
-
   Int_t GetNumberOfPHOSClusters() const {return fPHOSClusters;}
   void  SetNumberOfPHOSClusters(Int_t part) { fPHOSClusters = part ; }
   void  SetFirstPHOSCluster(Int_t index) { fFirstPHOSCluster = index ; } 
   Int_t GetFirstPHOSCluster() const  { return fFirstPHOSCluster ; }
+  //-------------------------------------------------------
+
+  TArrayF *GetEMCALTriggerPosition() const {return  fEMCALTrigger->GetTriggerPosition();}
+  TArrayF *GetEMCALTriggerAmplitudes() const {return  fEMCALTrigger->GetTriggerAmplitudes();}
   TArrayF *GetPHOSTriggerPosition() const {return  fPHOSTrigger->GetTriggerPosition();}
   TArrayF *GetPHOSTriggerAmplitudes() const {return  fPHOSTrigger->GetTriggerAmplitudes();}
 
@@ -310,13 +316,13 @@ public:
   void ReadFromTree(TTree *tree);
   TObject* FindListObject(const char *name);
   AliESD *GetAliESDOld(){return fESDOld;}
-  const void WriteToTree(TTree* tree) const {tree->Branch(fESDObjects);}
+  const void WriteToTree(TTree* tree) const;
   void GetStdContent();
   void ResetStdContent();
   void CreateStdContent();
   void SetStdNames();
   void CopyFromOldESD();
-  TList* GetList(){return fESDObjects;}
+  TList* GetList() const {return fESDObjects;}
 
 protected:
   AliESDEvent(const AliESDEvent&);
@@ -345,11 +351,17 @@ protected:
   TClonesArray *fCascades;         //! Cascade vertices
   TClonesArray *fKinks;            //! Kinks
   TClonesArray *fCaloClusters;     //! Calorimeter clusters for PHOS/EMCAL
+  AliESDCaloCells *fEMCALCells;     //! EMCAL cell info
+  AliESDCaloCells *fPHOSCells;     //! PHOS cell info
   TClonesArray *fErrorLogs;        //! Raw-data reading error messages
  
 
 
-  AliESD    *fESDOld;              //! Old esd Structure
+  AliESD       *fESDOld;           //! Old esd Structure
+  AliESDfriend *fESDFriendOld;     //! Old friend esd Structure
+  Bool_t    fConnected;            //! flag if leaves are alreday connected
+
+  static const char* fgkESDListName[kESDListN]; //!
 
   // Remove this stuff CKB
   Int_t        fEMCALClusters;   // Number of EMCAL clusters (subset of caloclusters)
@@ -358,7 +370,7 @@ protected:
   Int_t        fPHOSClusters;     // Number of PHOS clusters (subset of caloclusters)
   Int_t        fFirstPHOSCluster; // First PHOS cluster in the fCaloClusters list 
 
-  ClassDef(AliESDEvent,1)  //ESDEvent class 
+  ClassDef(AliESDEvent,6)  //ESDEvent class 
 };
 #endif