Adding AODMCHeader, new class and in the Filter and Handler

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

diff --git a/STEER/AliAODHandler.h b/STEER/AliAODHandler.h
index 73cc666a03b5f60d00eacece14bff8c012e88115..b2fcc3633a7675e5e4853673e2b87c3513d67966 100644 (file)
--- a/STEER/AliAODHandler.h
+++ b/STEER/AliAODHandler.h
@@ -15,6 +15,9 @@
 class AliAODEvent;
 class TFile;
 class TTree;
+class AliMCEventHandler;
+class AliAODMCHeader;
+class AliGenEventHeader;
 
 
 
@@ -24,31 +27,75 @@ class AliAODHandler : public AliVEventHandler {
     AliAODHandler();
     AliAODHandler(const char* name, const char* title);
     virtual ~AliAODHandler();
-    virtual void         SetOutputFileName(char* fname) {fName = fname;}
-    virtual char*        GetOutputFileName() {return fName;}
-    virtual Bool_t       InitIO(Option_t* option);
-    virtual Bool_t       BeginEvent(){ return kTRUE;}
+    virtual void         SetOutputFileName(const char* fname);
+    virtual const char*  GetOutputFileName();
+    virtual Bool_t       Init(Option_t* option);
+    virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*option*/)  {return kTRUE;}
+    virtual Bool_t       BeginEvent(Long64_t /*entry*/){return kTRUE;}
+    virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
     virtual Bool_t       Notify(const char * /* path */) {return kTRUE;}
     virtual Bool_t       FinishEvent();
     virtual Bool_t       Terminate();
     virtual Bool_t       TerminateIO();
     //
+    virtual void         SetCreateNonStandardAOD()   {fIsStandard = kFALSE;}
+    virtual void         SetFillAOD(Bool_t b)      {fFillAOD = b;}
+    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;}
-    TTree*               GetTree() {return fTreeA;}
-    void                 CreateTree();
+    virtual TTree*       GetTree() const {return fTreeA;}
+    void                 CreateTree(Int_t flag);
     void                 FillTree();
     void                 AddAODtoTreeUserInfo();
+    void                 AddBranch(const char* cname, void* addobj);
+    Bool_t               IsStandard() {return fIsStandard;}
+    Bool_t               GetFillAOD(){return fFillAOD;} 
+    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*/) {;}
+    void                 SetMCEventHandler(AliMCEventHandler* mcH) {fMCEventH = mcH;} // For internal use
  private:
+    void StoreMCParticles();
+    void SetMCHeaderInfo(AliAODMCHeader *mcHeader,AliGenEventHeader *genHeader); // Utility function t catch different types of eventheaders
     AliAODHandler(const AliAODHandler&);             // Not implemented
     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  private:
-    AliAODEvent             *fAODEvent;  //! Pointer to the AOD event
-    TTree                   *fTreeA;     //! tree for AOD persistency
-    TFile                   *fFileA;     //! Output file
-    char                    *fName;      //! Output file name
-    ClassDef(AliAODHandler, 1);
+    Bool_t                   fIsStandard;                         // Flag for standard aod creation
+    Bool_t                   fFillAOD;                          // Flag for filling of the AOD tree at the end (all or nothing)
+    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
+    AliMCEventHandler       *fMCEventH;               //! Pointer to mc event handler needed not to depend on the manager
+    TTree                   *fTreeA;                  //! tree for AOD persistency
+    TFile                   *fFileA;                  //! Output file
+    TString                  fFileName;               //  Output file name
+    ClassDef(AliAODHandler, 3)
 };
 
 #endif