New mapping in agreement with the new instructions from Paolo and Giacinto

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

diff --git a/STEER/AliMCEventHandler.h b/STEER/AliMCEventHandler.h
index 59fcfeccdab81cf53a3917f2a2dad22081e4757d..41ce29eb287f5cc921884afc9d750da78ac38ad3 100644 (file)
--- a/STEER/AliMCEventHandler.h
+++ b/STEER/AliMCEventHandler.h
@@ -1,4 +1,4 @@
-// -*- mode: C++ -*- 
+// -*- mode: C++ -*-
 #ifndef ALIMCEVENTHANDLER_H
 #define ALIMCEVENTHANDLER_H
 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
@@ -28,7 +28,7 @@ class AliMCEvent;
 
 
 
-class AliMCEventHandler : public AliVEventHandler 
+class AliMCEventHandler : public AliVEventHandler
 {
 public:
     AliMCEventHandler();
@@ -37,15 +37,20 @@ public:
     virtual void         SetOutputFileName(char* /* fname */) {;}
     virtual char*        GetOutputFileName() {return 0;}
     virtual void         SetInputPath(char* fname); 
+    virtual void         SetInputTree(TTree* /*tree*/) {;}
     virtual TString*     GetInputPath() {return fPathName;}
-    virtual Bool_t       InitIO(Option_t* opt);
-    virtual Bool_t       BeginEvent();
+    virtual Bool_t       Init(Option_t* opt);
+    virtual Bool_t       InitIO(Option_t* opt) {return Init(opt);};
+    virtual Bool_t       Init(TTree* /*tree*/, Option_t* /*opt*/) {return kTRUE;}
+    virtual Bool_t       BeginEvent(Long64_t entry);
+    virtual Bool_t       Notify() { return AliVEventHandler::Notify(); };
     virtual Bool_t       Notify(const char* path);
     virtual Bool_t       FinishEvent();
     virtual Bool_t       Terminate();
     virtual Bool_t       TerminateIO();
     virtual void         ResetIO();
     virtual Bool_t       GetEvent(Int_t iev);
+    virtual void         SetReadTR(Bool_t flag) { fReadTR = flag; }
     //
     AliMCEvent* MCEvent() {return fMCEvent;}
     TTree*      TreeTR() {return fTreeTR;}
@@ -69,7 +74,8 @@ private:
     char             *fExtension;        //! File name extension 
     Int_t             fFileNumber;       //! Input file number
     Int_t             fEventsPerFile;    //! Number of events per file
-    ClassDef(AliMCEventHandler,1)  //MC Truth EventHandler class 
+    Bool_t            fReadTR;           // determines if TR shall be read
+    ClassDef(AliMCEventHandler,1)  //MC Truth EventHandler class
 };
 #endif