EMD DA updated

[u/mrichter/AliRoot.git] / STEER / AliAODEvent.h

diff --git a/STEER/AliAODEvent.h b/STEER/AliAODEvent.h
index 2639342112bc5f42b309261b075489f25fbb82ef..24b386e89bad169310451f6040244437ff629017 100644 (file)
--- a/STEER/AliAODEvent.h
+++ b/STEER/AliAODEvent.h
@@ -22,6 +22,7 @@
 #include "AliAODTrack.h"
 #include "AliAODVertex.h"
 #include "AliAODv0.h"
+#include "AliAODcascade.h"
 #include "AliAODTracklets.h"
 #include "AliAODJet.h"
 #include "AliAODCaloCells.h"
@@ -39,6 +40,7 @@ class AliAODEvent : public AliVEvent {
                       kAODTracks,
                       kAODVertices,
                       kAODv0,
+                      kAODcascade,
                       kAODTracklets,
                       kAODJets,
                       kAODEmcalCells,
@@ -52,8 +54,8 @@ class AliAODEvent : public AliVEvent {
   AliAODEvent();
   virtual ~AliAODEvent();
 
-  AliAODEvent(const AliAODEvent& aodevent);             
-  AliAODEvent& operator=(const AliAODEvent& aodevent);  
+  AliAODEvent(const AliAODEvent& aodevent);
+  AliAODEvent& operator=(const AliAODEvent& aodevent);
 
   void          AddObject(TObject *obj);
   void          RemoveObject(TObject *obj);
@@ -74,21 +76,27 @@ class AliAODEvent : public AliVEvent {
   void     SetOrbitNumber(UInt_t n) {if (fHeader) fHeader->SetOrbitNumber(n);}
   void     SetBunchCrossNumber(UShort_t n) {if (fHeader) fHeader->SetBunchCrossNumber(n);}
   void     SetMagneticField(Double_t mf){if (fHeader) fHeader->SetMagneticField(mf);}
+  void     SetMuonMagFieldScale(Double_t mf){if (fHeader) fHeader->SetMuonMagFieldScale(mf);}
+  void     SetDiamond(Float_t xy[2],Float_t cov[3]){if (fHeader) fHeader->SetDiamond(xy,cov);}
 
   Int_t    GetRunNumber() const {return fHeader ? fHeader->GetRunNumber() : -999;}
   UInt_t   GetPeriodNumber() const {return fHeader ? fHeader->GetPeriodNumber() : 0;}
   UInt_t   GetOrbitNumber() const {return fHeader ? fHeader->GetOrbitNumber() : 0;}
   UShort_t GetBunchCrossNumber() const {return fHeader ? fHeader->GetBunchCrossNumber() : 0;}
   Double_t GetMagneticField() const {return fHeader ? fHeader->GetMagneticField() : -999.;}
+  Double_t GetMuonMagFieldScale() const {return fHeader ? fHeader->GetMuonMagFieldScale() : -999.;}
+  Double_t GetDiamondX() const {return fHeader ? fHeader->GetDiamondX() : -999.;}
+  Double_t GetDiamondY() const {return fHeader ? fHeader->GetDiamondY() : -999.;}
+  void     GetDiamondCovXY(Float_t cov[3]) const {cov[0]=-999.; if(fHeader) fHeader->GetDiamondCovXY(cov);}
   
   void      SetEventType(UInt_t eventType){fHeader->SetEventType(eventType);}
   void      SetTriggerMask(ULong64_t n) {fHeader->SetTriggerMask(n);}
   void      SetTriggerCluster(UChar_t n) {fHeader->SetTriggerCluster(n);}
 
-  UInt_t    GetEventType()  const { return fHeader ? fHeader->GetEventType() : 0;}
-  ULong64_t GetTriggerMask() const {return fHeader ? fHeader->GetTriggerMask() : 0;}
-  UChar_t   GetTriggerCluster() const {return fHeader ? fHeader->GetTriggerCluster() : 0;}
-
+  UInt_t    GetEventType()          const { return fHeader ? fHeader->GetEventType() : 0;}
+  ULong64_t GetTriggerMask()        const { return fHeader ? fHeader->GetTriggerMask() : 0;}
+  UChar_t   GetTriggerCluster()     const { return fHeader ? fHeader->GetTriggerCluster() : 0;}
+  TString   GetFiredTriggerClasses()const { return fHeader->GetFiredTriggerClasses();};
   Double_t  GetZDCN1Energy()        const { return fHeader ? fHeader->GetZDCN1Energy() : -999.; }
   Double_t  GetZDCP1Energy()        const { return fHeader ? fHeader->GetZDCP1Energy() : -999.; }
   Double_t  GetZDCN2Energy()        const { return fHeader ? fHeader->GetZDCN2Energy() : -999.; }
@@ -106,7 +114,7 @@ class AliAODEvent : public AliVEvent {
 
   // -- Vertex
   TClonesArray *GetVertices()            const { return fVertices; }
-  Int_t         GetNVertices()           const { return fVertices->GetEntriesFast(); }
+  Int_t         GetNumberOfVertices()    const { return fVertices->GetEntriesFast(); }
   AliAODVertex *GetVertex(Int_t nVertex) const { return (AliAODVertex*)fVertices->UncheckedAt(nVertex); }
   Int_t         AddVertex(const AliAODVertex* vtx)
   {new((*fVertices)[fVertices->GetEntriesFast()]) AliAODVertex(*vtx); return fVertices->GetEntriesFast()-1;}
@@ -116,11 +124,18 @@ class AliAODEvent : public AliVEvent {
 
   // V0
   TClonesArray *GetV0s()                 const { return fV0s; }
-  Int_t         GetNV0s()                const { return fV0s->GetEntriesFast(); }
+  Int_t         GetNumberOfV0s()         const { return fV0s->GetEntriesFast(); }
   AliAODv0     *GetV0(Int_t nV0)         const { return (AliAODv0*)fV0s->UncheckedAt(nV0); }
   Int_t         AddV0(const AliAODv0* v0)
   {new((*fV0s)[fV0s->GetEntriesFast()]) AliAODv0(*v0); return fV0s->GetEntriesFast()-1;}
 
+  // Cascades
+  TClonesArray  *GetCascades()            const { return fCascades; }
+  Int_t          GetNumberOfCascades()    const { return fCascades->GetEntriesFast(); }
+  AliAODcascade *GetCascade(Int_t nCasc)  const { return (AliAODcascade*)fCascades->UncheckedAt(nCasc); }
+  Int_t          AddCascade(const AliAODcascade* cascade)
+  {new((*fCascades)[fCascades->GetEntriesFast()]) AliAODcascade(*cascade); return fCascades->GetEntriesFast()-1;}
+
   // -- EMCAL and PHOS Cluster
   TClonesArray *GetCaloClusters()        const { return fCaloClusters; }
   Int_t         GetNCaloClusters()       const { return fCaloClusters->GetEntriesFast(); }
@@ -168,13 +183,16 @@ class AliAODEvent : public AliVEvent {
   void    ResetStd(Int_t trkArrSize = 0, 
                   Int_t vtxArrSize = 0, 
                   Int_t v0ArrSize = 0, 
+                  Int_t cascadeArrSize = 0,
                   Int_t jetSize = 0, 
                   Int_t caloClusSize = 0, 
                   Int_t fmdClusSize = 0, 
-                  Int_t pmdClusSize = 0);
+                  Int_t pmdClusSize = 0
+                  );
   void    ClearStd();
-  void    ReadFromTree(TTree *tree);
-  const void WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}
+  void    Reset() {ClearStd();} 
+  void    ReadFromTree(TTree *tree, Option_t* opt = "");
+  void    WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}
 
   void  Print(Option_t *option="") const;
   void  MakeEntriesReferencable();
@@ -183,12 +201,13 @@ class AliAODEvent : public AliVEvent {
 
   TList   *fAODObjects; //  list of AODObjects
   TFolder *fAODFolder;  //  folder structure of branches
- 
+  Bool_t   fConnected;  //! flag if leaves are alreday connected 
   // standard content
   AliAODHeader    *fHeader;       //! event information
   TClonesArray    *fTracks;       //! charged tracks
   TClonesArray    *fVertices;     //! vertices
   TClonesArray    *fV0s;          //! V0s
+  TClonesArray    *fCascades;     //! Cascades
   AliAODTracklets *fTracklets;    //! SPD tracklets
   TClonesArray    *fJets;         //! jets
   AliAODCaloCells *fEmcalCells;   //! EMCAL calorimenter cells
@@ -199,7 +218,7 @@ class AliAODEvent : public AliVEvent {
 
   static const char* fAODListName[kAODListN]; //!
 
-  ClassDef(AliAODEvent, 4);
+  ClassDef(AliAODEvent, 5);
 };
 
 #endif