Fixed GetTriggerAmplittudes

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

diff --git a/STEER/AliMCEventHandler.h b/STEER/AliMCEventHandler.h
index 59fcfeccdab81cf53a3917f2a2dad22081e4757d..ee9dee4df17bee9157cf252a8a40bdb31fb69224 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,31 +28,39 @@ class AliMCEvent;
 
 
 
-class AliMCEventHandler : public AliVEventHandler 
+class AliMCEventHandler : public AliVEventHandler
 {
 public:
     AliMCEventHandler();
     AliMCEventHandler(const char* name, const char* title);
     virtual ~AliMCEventHandler();
-    virtual void         SetOutputFileName(char* /* fname */) {;}
-    virtual char*        GetOutputFileName() {return 0;}
-    virtual void         SetInputPath(char* fname); 
-    virtual TString*     GetInputPath() {return fPathName;}
-    virtual Bool_t       InitIO(Option_t* opt);
-    virtual Bool_t       BeginEvent();
+    virtual void         SetOutputFileName(const char* /* fname */) {;}
+    virtual const char*  GetOutputFileName() {return 0;}
+    virtual void         SetInputPath(const char* fname); 
+    virtual void         SetInputTree(TTree* /*tree*/) {;}
+    virtual TString*     GetInputPath() const {return fPathName;}
+    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;}
+    AliMCEvent* MCEvent() const {return fMCEvent;} 
+    TTree*      TreeTR()  const {return fTreeTR;}
+    TTree*      TreeK()   const {return fTreeK;}
     Int_t       GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
     void        DrawCheck(Int_t i, Int_t search=0);
 private:
     Bool_t      OpenFile(Int_t i);
+    AliMCEventHandler(const AliMCEventHandler& handler);             
+    AliMCEventHandler& operator=(const AliMCEventHandler& handler);  
 private:
     AliMCEvent       *fMCEvent;          //! MC Event
     TFile            *fFileE;            //! File with TreeE
@@ -69,7 +77,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