Using pseudo-rapidity getter.

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

diff --git a/STEER/AliAODHandler.h b/STEER/AliAODHandler.h
index 99b3cb1eb6e3a86f97e73cad52772ad803b7442a..86c965d1b37b9f9f75be4860a5a4b23d803d2ad5 100644 (file)
--- a/STEER/AliAODHandler.h
+++ b/STEER/AliAODHandler.h
@@ -10,7 +10,7 @@
 //     Author: Andreas Morsch, CERN
 //-------------------------------------------------------------------------
 
-#include "AliVirtualEventHandler.h"
+#include "AliVEventHandler.h"
 
 class AliAODEvent;
 class TFile;
@@ -18,33 +18,47 @@ class TTree;
 
 
 
-class AliAODHandler : public AliVirtualEventHandler {
+class AliAODHandler : public AliVEventHandler {
     
  public:
     AliAODHandler();
     AliAODHandler(const char* name, const char* title);
     virtual ~AliAODHandler();
-    virtual Bool_t       InitIO();
-    virtual Bool_t       Fill();
+    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 Bool_t       Notify() {return kTRUE;}
+    //
+    virtual void         SetCreateNonStandardAOD()   {fIsStandard = kFALSE;}
+    virtual void         SetNeedsHeaderReplication() {fNeedsHeaderReplication = 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               NeedsHeaderReplication() {return  fNeedsHeaderReplication;}
+    //
+    void                 SetInputTree(TTree* /*tree*/) {;}
  private:
     AliAODHandler(const AliAODHandler&);             // Not implemented
     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  private:
-    AliAODEvent             *fAODEvent;  //! Pointer to the AOD event
-    TFile                   *fAODFile;   //! Pointer to the AOD file
-    TTree                   *fTreeA;     //! tree for AOD persistency
-
-    
-    ClassDef(AliAODHandler, 1);
+    Bool_t                   fIsStandard;             // Flag for standard aod creation
+    Bool_t                   fNeedsHeaderReplication; // Flag for header replication
+    AliAODEvent             *fAODEvent;               //! Pointer to the AOD event
+    TTree                   *fTreeA;                  //! tree for AOD persistency
+    TFile                   *fFileA;                  //! Output file
+    TString                  fFileName;               //  Output file name
+    ClassDef(AliAODHandler, 2);
 };
 
 #endif