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 "AliVirtualEventHandler.h"
  14
  15 class AliAODEvent;
  16 class TFile;
  17 class TTree;
  18
  19
  20
  21 class AliAODHandler : public AliVirtualEventHandler {
  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       InitIO(Option_t* option);
  30     virtual Bool_t       Fill();
  31     virtual Bool_t       Terminate();
  32     virtual Bool_t       TerminateIO();
  33     virtual Bool_t       Notify() {return kTRUE;}
  34     //
  35     AliAODEvent*         GetAOD()  {return fAODEvent;}
  36     TTree*               GetTree() {return fTreeA;}
  37     void                 CreateTree();
  38     void                 FillTree();
  39     void                 AddAODtoTreeUserInfo();
  40  private:
  41     AliAODHandler(const AliAODHandler&);             // Not implemented
  42     AliAODHandler& operator=(const AliAODHandler&);  // Not implemented
  43  private:
  44     AliAODEvent             *fAODEvent;  //! Pointer to the AOD event
  45     TTree                   *fTreeA;     //! tree for AOD persistency
  46     TFile                   *fFileA;     //! Output file
  47     char                    *fName;      //! Output file name
  48     ClassDef(AliAODHandler, 1);
  49 };
  50
  51 #endif