1 #ifndef ALIAODINPUTHANDLER_H
2 #define ALIAODINPUTHANDLER_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
8 //-------------------------------------------------------------------------
9 // AOD Input Handler realisation of the AliVEventHandler interface
10 // Author: Andreas Morsch, CERN
11 //-------------------------------------------------------------------------
13 #include "AliInputEventHandler.h"
14 #include "AliAODEvent.h"
15 #include "AliMCEvent.h"
24 class AliAODInputHandler : public AliInputEventHandler {
28 AliAODInputHandler(const char* name, const char* title);
29 virtual ~AliAODInputHandler();
30 virtual Bool_t Init(Option_t* /*opt*/) {return kTRUE;}
31 virtual Bool_t Init(TTree* tree, Option_t* opt);
32 AliAODEvent *GetEvent() const {return fEvent;}
33 AliMCEvent *MCEvent() const {return fMCEvent;}
34 virtual void AddFriend(char* filename);
35 virtual Bool_t BeginEvent(Long64_t entry);
36 virtual Bool_t Notify() { return AliVEventHandler::Notify();};
37 virtual Bool_t Notify(const char* path);
38 virtual Bool_t FinishEvent();
39 Option_t *GetDataType() const;
40 // Get the statistics object (currently TH2). Option can be BIN0.
41 virtual TObject *GetStatistics(Option_t *option="") const;
42 // Provisions for event merging
43 void SetMergeEvents(Bool_t flag) {fMergeEvents = flag;}
44 Bool_t GetMergeEvents() const {return fMergeEvents;}
45 AliAODEvent* GetEventToMerge() {return fAODEventToMerge;}
46 TTree* GetTreeToMerge() const {return fTreeToMerge;}
47 void SetMergeOffset(Int_t ioff) {fMergeOffset = ioff;}
48 Int_t GetMergeOffset() const {return fMergeOffset;}
49 void SetMergeTracks(Bool_t flag) {fMergeTracks = flag;}
50 Bool_t GetMergeTracks() const {return fMergeTracks;}
51 void SetMergeEMCALClusters(Bool_t flag) {fMergeEMCALClusters = flag;}
52 Bool_t GetMergeEMCALClusters() const {return fMergeEMCALClusters;}
53 void SetMergeEMCALCells(Bool_t flag) {fMergeEMCALCells = flag;}
54 Bool_t GetMergeEMCALCells() const {return fMergeEMCALCells ;}
55 void SetMergePHOSClusters(Bool_t flag) {fMergePHOSClusters = flag;}
56 Bool_t GetMergePHOSClusters() const {return fMergePHOSClusters ;}
57 void SetMergePHOSCells(Bool_t flag) {fMergePHOSCells = flag;}
58 Bool_t GetMergePHOSCells() const {return fMergePHOSCells ;}
60 virtual AliPIDResponse* GetPIDResponse() {return (AliPIDResponse*)fAODpidUtil;}
61 virtual void CreatePIDResponse(Bool_t isMC=kFALSE);
62 AliAODpidUtil *GetAODpidUtil() const { return fAODpidUtil; }
65 void ConnectFriends();
66 AliAODInputHandler(const AliAODInputHandler& handler);
67 AliAODInputHandler& operator=(const AliAODInputHandler& handler);
69 AliAODEvent *fEvent; //! Pointer to the event
70 AliMCEvent *fMCEvent; //! Pointer to the MCEvent
71 TList *fFriends; // List of friend trees
72 AliAODpidUtil *fAODpidUtil; //! Pointer to PID information
74 // Support for merged events
75 Bool_t fMergeEvents; // Flag for event merging
76 Bool_t fMergeTracks; // Merge tracks
77 Bool_t fMergeEMCALClusters; // Merge PHOS cluster
78 Bool_t fMergePHOSClusters; // Merge EMCAL cluster
79 Bool_t fMergeEMCALCells; // Merge PHOS cluster
80 Bool_t fMergePHOSCells; // Merge EMCAL cluster
81 Bool_t fFriendsConnected;// Friends are connected
82 TFile *fFileToMerge; //! File for merging
83 TTree *fTreeToMerge; //! Tree for merging
84 AliAODEvent *fAODEventToMerge; //! Event for merging
85 Int_t fMergeOffset; //! Event offset for merging
86 TH2F* fHistStatistics[2]; //! how many events are cut away why {all,bin 0}
87 ClassDef(AliAODInputHandler, 3);