Changes in QA to be able to process separately different triggers (Ruben)

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

diff --git a/STEER/AliAODInputHandler.h b/STEER/AliAODInputHandler.h
index 4d4c3a1a6bc58377908443ce27db337b7bb5fc62..a29f3764fb745a7ce2b0c85309cb29568e042ea8 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,26 +33,58 @@ 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;}
+    TTree*               GetTreeToMerge()  const {return fTreeToMerge;}
     void                 SetMergeOffset(Int_t ioff) {fMergeOffset = ioff;}
+    Int_t                GetMergeOffset()     const {return fMergeOffset;}
+    void                 SetMergeTracks(Bool_t flag) {fMergeTracks = flag;}
+    Bool_t               GetMergeTracks()      const {return fMergeTracks;}
+    void                 SetMergeEMCALClusters(Bool_t flag) {fMergeEMCALClusters = flag;}
+    Bool_t               GetMergeEMCALClusters()      const {return fMergeEMCALClusters;}
+    void                 SetMergeEMCALCells(Bool_t flag)    {fMergeEMCALCells    = flag;}
+    Bool_t               GetMergeEMCALCells()         const {return fMergeEMCALCells   ;} 
+    void                 SetMergePHOSClusters(Bool_t flag) {fMergePHOSClusters   = flag;}
+    Bool_t               GetMergePHOSClusters()      const {return fMergePHOSClusters  ;}
+    void                 SetMergePHOSCells(Bool_t flag)    {fMergePHOSCells      = flag;}
+    Bool_t               GetMergePHOSCells()         const {return fMergePHOSCells     ;}  
+    //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 
+    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          fMergeTracks;        // Merge tracks
+    Bool_t          fMergeEMCALClusters; // Merge PHOS  cluster
+    Bool_t          fMergePHOSClusters;  // Merge EMCAL cluster
+    Bool_t          fMergeEMCALCells;    // Merge PHOS  cluster
+    Bool_t          fMergePHOSCells;     // Merge EMCAL cluster
+    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
-    ClassDef(AliAODInputHandler, 2);
+    TH2F*           fHistStatistics[2]; //! how many events are cut away why {all,bin 0}
+    ClassDef(AliAODInputHandler, 3);
 };
 
 #endif