STEER/AliESDHandler.h

   1 #ifndef ALIESDHANDLER_H
   2 #define ALIESDHANDLER_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 //
  10 //     Implementation of the Event Handler Interface for ESD
  11 //
  12 //-------------------------------------------------------------------------
  13
  14 #include "AliVEventHandler.h"
  15
  16 class AliESDEvent;
  17 class AliESDfriend;
  18 class TFile;
  19 class TTree;
  20
  21 class AliESDHandler : public AliVEventHandler {
  22
  23  public:
  24         AliESDHandler();
  25         AliESDHandler(const char* name, const char* title);
  26         virtual ~AliESDHandler();
  27         virtual void SetOutputFileName(const char* fname){fFileName = fname;}
  28         virtual const char* GetOutputFileName(){return fFileName.Data();}
  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*/){fIsEventSelectedForFriends = kFALSE; 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         AliESDfriend* GetESDfriend()  {return fesdf;}
  39         virtual TTree* GetTree() const {return fTreeEF;}
  40         void FillTree();
  41         void SetInputTree(TTree* /*tree*/) {;}
  42         void SelectEventForFriends() {fIsEventSelectedForFriends = kTRUE;}
  43
  44  private:
  45
  46         AliESDHandler(const AliESDHandler&);             // Not implemented
  47         AliESDHandler& operator=(const AliESDHandler&);  // Not implemented
  48
  49         AliESDfriend* fesdf;    //! Pointer to the ESD friend
  50         TTree* fTreeEF;         //! Output tree for friends
  51         TFile* fFileEF;         //! Output file for friends
  52         TString fFileName;      //! Output file name for friends
  53         Bool_t fIsEventSelectedForFriends; //! flag to indicate if the event was selected to have the friends kept
  54
  55     ClassDef(AliESDHandler, 3)
  56 };
  57 #endif