Additional flags for AOD replication (Roberta Arnaldi, AM)

diff --git a/STEER/AliAODHandler.cxx b/STEER/AliAODHandler.cxx
index cb7a5126ecf9d7bec834bb2c683d1fb21ae28119..90f264d1e6f7cc1e936a83e0881faf23b7fd307c 100644 (file)
--- a/STEER/AliAODHandler.cxx
+++ b/STEER/AliAODHandler.cxx
@@ -35,6 +35,15 @@ AliAODHandler::AliAODHandler() :
     AliVEventHandler(),
     fIsStandard(kTRUE),
     fNeedsHeaderReplication(kFALSE),
+    fNeedsTracksBranchReplication(kFALSE),
+    fNeedsVerticesBranchReplication(kFALSE),
+    fNeedsV0sBranchReplication(kFALSE),
+    fNeedsTrackletsBranchReplication(kFALSE),
+    fNeedsPMDClustersBranchReplication(kFALSE),
+    fNeedsJetsBranchReplication(kFALSE),
+    fNeedsFMDClustersBranchReplication(kFALSE),
+    fNeedsCaloClustersBranchReplication(kFALSE),
+    fAODIsReplicated(kFALSE),
     fAODEvent(NULL),
     fTreeA(NULL),
     fFileA(NULL),
@@ -48,6 +57,15 @@ AliAODHandler::AliAODHandler(const char* name, const char* title):
     AliVEventHandler(name, title),
     fIsStandard(kTRUE),
     fNeedsHeaderReplication(kFALSE),
+    fNeedsTracksBranchReplication(kFALSE),
+    fNeedsVerticesBranchReplication(kFALSE),
+    fNeedsV0sBranchReplication(kFALSE),
+    fNeedsTrackletsBranchReplication(kFALSE),
+    fNeedsPMDClustersBranchReplication(kFALSE),
+    fNeedsJetsBranchReplication(kFALSE),
+    fNeedsFMDClustersBranchReplication(kFALSE),
+    fNeedsCaloClustersBranchReplication(kFALSE),
+    fAODIsReplicated(kFALSE),
     fAODEvent(NULL),
     fTreeA(NULL),
     fFileA(NULL),
@@ -108,6 +126,9 @@ Bool_t AliAODHandler::FinishEvent()
     fAODEvent->MakeEntriesReferencable();
     FillTree();
     if (fIsStandard) fAODEvent->ResetStd();
+    // Reset AOD replication flag
+    fAODIsReplicated = kFALSE;
+    
     return kTRUE;
 }
 

diff --git a/STEER/AliAODHandler.h b/STEER/AliAODHandler.h
index 86c965d1b37b9f9f75be4860a5a4b23d803d2ad5..7f154ad822775d82aacd8163fd636ce7861109ba 100644 (file)
--- a/STEER/AliAODHandler.h
+++ b/STEER/AliAODHandler.h
@@ -37,6 +37,15 @@ class AliAODHandler : public AliVEventHandler {
     //
     virtual void         SetCreateNonStandardAOD()   {fIsStandard = kFALSE;}
     virtual void         SetNeedsHeaderReplication() {fNeedsHeaderReplication = kTRUE;}
+    virtual void         SetNeedsTracksBranchReplication() {fNeedsTracksBranchReplication = kTRUE;}
+    virtual void         SetNeedsVerticesBranchReplication() {fNeedsVerticesBranchReplication = kTRUE;}
+    virtual void         SetNeedsV0sBranchReplication() {fNeedsV0sBranchReplication = kTRUE;}
+    virtual void         SetNeedsTrackletsBranchReplication() {fNeedsTrackletsBranchReplication = kTRUE;}
+    virtual void         SetNeedsPMDClustersBranchReplication() {fNeedsPMDClustersBranchReplication = kTRUE;}
+    virtual void         SetNeedsJetsBranchReplication() {fNeedsJetsBranchReplication = kTRUE;}
+    virtual void         SetNeedsFMDClustersBranchReplication() {fNeedsFMDClustersBranchReplication = kTRUE;}
+    virtual void         SetNeedsCaloClustersBranchReplication() {fNeedsCaloClustersBranchReplication = kTRUE;}
+    virtual void         SetAODIsReplicated() {fAODIsReplicated = kTRUE;}
     //
     AliAODEvent*         GetAOD()  {return fAODEvent;}
     virtual TTree*       GetTree() const {return fTreeA;}
@@ -46,14 +55,33 @@ class AliAODHandler : public AliVEventHandler {
     void                 AddBranch(const char* cname, void* addobj);
     Bool_t               IsStandard() {return fIsStandard;}
     Bool_t               NeedsHeaderReplication() {return  fNeedsHeaderReplication;}
+    Bool_t               NeedsTracksBranchReplication() {return  fNeedsTracksBranchReplication;}
+    Bool_t               NeedsVerticesBranchReplication() {return  fNeedsVerticesBranchReplication;}
+    Bool_t               NeedsV0sBranchReplication() {return  fNeedsV0sBranchReplication;}
+    Bool_t               NeedsTrackletsBranchReplication() {return  fNeedsTrackletsBranchReplication;}
+    Bool_t               NeedsPMDClustersBranchReplication() {return  fNeedsPMDClustersBranchReplication;}
+    Bool_t               NeedsJetsBranchReplication() {return  fNeedsJetsBranchReplication;}
+    Bool_t               NeedsFMDClustersBranchReplication() {return  fNeedsFMDClustersBranchReplication;}
+    Bool_t               NeedsCaloClustersBranchReplication() {return  fNeedsCaloClustersBranchReplication;}
+    Bool_t               AODIsReplicated() {return fAODIsReplicated;}
     //
     void                 SetInputTree(TTree* /*tree*/) {;}
  private:
     AliAODHandler(const AliAODHandler&);             // Not implemented
     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  private:
-    Bool_t                   fIsStandard;             // Flag for standard aod creation
-    Bool_t                   fNeedsHeaderReplication; // Flag for header replication
+    Bool_t                   fIsStandard;                         // Flag for standard aod creation
+    Bool_t                   fNeedsHeaderReplication;             // Flag for header replication
+    Bool_t                   fNeedsTracksBranchReplication;       // Flag for tracks replication
+    Bool_t                   fNeedsVerticesBranchReplication;     // Flag for vertices replication
+    Bool_t                   fNeedsV0sBranchReplication;          // Flag for V0s replication
+    Bool_t                   fNeedsTrackletsBranchReplication;    // Flag for Tracklets replication
+    Bool_t                   fNeedsPMDClustersBranchReplication;  // Flag for PMDClusters replication
+    Bool_t                   fNeedsJetsBranchReplication;         // Flag for Jets replication
+    Bool_t                   fNeedsFMDClustersBranchReplication;  // Flag for FMDClusters replication
+    Bool_t                   fNeedsCaloClustersBranchReplication; // Flag for CaloClusters replication
+    Bool_t                   fAODIsReplicated;                    // Flag true if replication as been executed
+    
     AliAODEvent             *fAODEvent;               //! Pointer to the AOD event
     TTree                   *fTreeA;                  //! tree for AOD persistency
     TFile                   *fFileA;                  //! Output file