Coverity fix

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

diff --git a/STEER/AliAODInputHandler.h b/STEER/AliAODInputHandler.h
index a69e207f3640863231eb3a6ba62042f0ca09380f..2eaecc6d75f298303ad1d30e7bba1eb253b411ab 100644 (file)
--- a/STEER/AliAODInputHandler.h
+++ b/STEER/AliAODInputHandler.h
@@ -13,9 +13,13 @@
 #include "AliInputEventHandler.h"
 #include "AliAODEvent.h"
 #include "AliMCEvent.h"
-
 class TList;
 class AliMCEvent;
+class TH2F;
+class AliMCEvent;
+class AliAODpidUtil;
+class AliPIDResponse;
+
 
 class AliAODInputHandler : public AliInputEventHandler {
 
@@ -29,15 +33,42 @@ class AliAODInputHandler : public AliInputEventHandler {
     AliMCEvent          *MCEvent()  const {return fMCEvent;}
     virtual void         AddFriend(char* filename);
     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();
     Option_t            *GetDataType() const;
+    // Get the statistics object (currently TH2). Option can be BIN0.
+    virtual TObject     *GetStatistics(Option_t *option="") const;
+    // Provisions for event merging
+    void                 SetMergeEvents(Bool_t flag) {fMergeEvents = flag;}
+    Bool_t               GetMergeEvents() const {return fMergeEvents;}
+    AliAODEvent*         GetEventToMerge() {return fAODEventToMerge;}
+    void                 SetMergeOffset(Int_t ioff) {fMergeOffset = ioff;}
+  
+    //PID response
+    virtual AliPIDResponse* GetPIDResponse() {return (AliPIDResponse*)fAODpidUtil;}
+    virtual void CreatePIDResponse(Bool_t isMC=kFALSE);
+    AliAODpidUtil *GetAODpidUtil() const { return fAODpidUtil; }
+  
  private:
+    void ConnectFriends();
     AliAODInputHandler(const AliAODInputHandler& handler);             
     AliAODInputHandler& operator=(const AliAODInputHandler& handler);  
  private:
     AliAODEvent    *fEvent;   //! Pointer to the event
     AliMCEvent     *fMCEvent; //! Pointer to the MCEvent
-    TList          *fFriends; //  List of friend trees 
-    ClassDef(AliAODInputHandler, 1);
+    TList          *fFriends; //  List of friend trees
+    AliAODpidUtil  *fAODpidUtil; //! Pointer to PID information
+  
+// Support for merged events
+    Bool_t          fMergeEvents;     // Flag for event merging
+    Bool_t          fFriendsConnected;// Friends are connected
+    TFile          *fFileToMerge;     //! File for merging
+    TTree          *fTreeToMerge;     //! Tree for merging
+    AliAODEvent    *fAODEventToMerge; //! Event for merging
+    Int_t           fMergeOffset;     //! Event offset for merging
+    TH2F*           fHistStatistics[2]; //! how many events are cut away why {all,bin 0}
+    ClassDef(AliAODInputHandler, 3);
 };
 
 #endif