Possibility to add new branch to AOD event.
[u/mrichter/AliRoot.git] / STEER / AliAODHandler.h
1 #ifndef ALIAODHANDLER_H
2 #define ALIAODHANDLER_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //-------------------------------------------------------------------------
9 //     Implementation of the Event Handler Interface for AOD
10 //     Author: Andreas Morsch, CERN
11 //-------------------------------------------------------------------------
12
13 #include "AliVEventHandler.h"
14
15 class AliAODEvent;
16 class TFile;
17 class TTree;
18
19
20
21 class AliAODHandler : public AliVEventHandler {
22     
23  public:
24     AliAODHandler();
25     AliAODHandler(const char* name, const char* title);
26     virtual ~AliAODHandler();
27     virtual void         SetOutputFileName(char* fname) {fName = fname;}
28     virtual char*        GetOutputFileName() {return fName;}
29     virtual Bool_t       Init(Option_t* option);
30     virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*option*/)  {return kTRUE;}
31     virtual Bool_t       BeginEvent(Long64_t /*entry*/)  {return kTRUE;}
32     virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
33     virtual Bool_t       Notify(const char * /* path */) {return kTRUE;}
34     virtual Bool_t       FinishEvent();
35     virtual Bool_t       Terminate();
36     virtual Bool_t       TerminateIO();
37     //
38     AliAODEvent*         GetAOD()  {return fAODEvent;}
39     TTree*               GetTree() {return fTreeA;}
40     void                 CreateTree(Int_t flag);
41     void                 FillTree();
42     void                 AddAODtoTreeUserInfo();
43     void                 AddBranch(const char* bname, const char* cname, TObject* addobj);
44     //
45     void                 SetInputTree(TTree* /*tree*/) {;}
46  private:
47     AliAODHandler(const AliAODHandler&);             // Not implemented
48     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
49  private:
50     AliAODEvent             *fAODEvent;  //! Pointer to the AOD event
51     TTree                   *fTreeA;     //! tree for AOD persistency
52     TFile                   *fFileA;     //! Output file
53     char                    *fName;      //! Output file name
54     ClassDef(AliAODHandler, 1);
55 };
56
57 #endif