Getting rid of obsolete objects.

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

diff --git a/STEER/AliAODEvent.h b/STEER/AliAODEvent.h
index 268ccc08a94309b7b1d6a6dbf616501a76c1e5bd..56ae6a27134b67fb40ecdd52068317e0f20f101b 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"
@@ -30,6 +31,7 @@
 #include "AliAODFmdCluster.h"
 
 class TTree;
+class TFolder;
 
 class AliAODEvent : public AliVEvent {
 
@@ -38,6 +40,7 @@ class AliAODEvent : public AliVEvent {
                       kAODTracks,
                       kAODVertices,
                       kAODv0,
+                      kAODcascade,
                       kAODTracklets,
                       kAODJets,
                       kAODEmcalCells,
@@ -51,8 +54,8 @@ class AliAODEvent : public AliVEvent {
   AliAODEvent();
   virtual ~AliAODEvent();
 
-  //AliAODEvent(const AliAODEvent& aodevent);  // not implemented
-  //AliAODEvent& operator=(const AliAODEvent& aodevent);  // not implemented
+  AliAODEvent(const AliAODEvent& aodevent);
+  AliAODEvent& operator=(const AliAODEvent& aodevent);
 
   void          AddObject(TObject *obj);
   void          RemoveObject(TObject *obj);
@@ -73,12 +76,16 @@ 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     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 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);}
@@ -105,7 +112,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;}
@@ -115,11 +122,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(); }
@@ -127,6 +141,10 @@ class AliAODEvent : public AliVEvent {
   Int_t         AddCaloCluster(const AliAODCaloCluster* clus)
   {new((*fCaloClusters)[fCaloClusters->GetEntriesFast()]) AliAODCaloCluster(*clus); return fCaloClusters->GetEntriesFast()-1;}
 
+  Int_t GetEMCALClusters(TRefArray *clusters) const;
+  Int_t GetPHOSClusters(TRefArray *clusters) const;
+
+
   // -- FMD Cluster
   TClonesArray *GetFmdClusters()        const { return fFmdClusters; }
   Int_t         GetNFmdClusters()       const { return fFmdClusters->GetEntriesFast(); }
@@ -159,28 +177,34 @@ class AliAODEvent : public AliVEvent {
   void    CreateStdContent();
   void    SetStdNames();
   void    GetStdContent();
+  void    CreateStdFolders();
   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    ReadFromTree(TTree *tree, Option_t* opt = "");
+  void    WriteToTree(TTree* tree) const {tree->Branch(fAODObjects);}
 
   void  Print(Option_t *option="") const;
-
+  void  MakeEntriesReferencable();
+  static void AssignIDtoCollection(TCollection* col);
  private :
 
-  TList *fAODObjects; //  list of AODObjects
-  
+  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
@@ -191,7 +215,7 @@ class AliAODEvent : public AliVEvent {
 
   static const char* fAODListName[kAODListN]; //!
 
-  ClassDef(AliAODEvent,3);
+  ClassDef(AliAODEvent, 5);
 };
 
 #endif