Make friends available after the AliAnalysisManager::Init() phase
[u/mrichter/AliRoot.git] / STEER / ESD / AliESDInputHandler.h
1 #ifndef ALIESDINPUTHANDLER_H
2 #define ALIESDINPUTHANDLER_H
3 /* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8 //-------------------------------------------------------------------------
9 //     ESD Input Handler realisation of the AliVEventHandler interface
10 //     Author: Andreas Morsch, CERN
11 //-------------------------------------------------------------------------
12
13 #include "AliInputEventHandler.h"
14 #include "AliESDEvent.h"
15 class TChain;
16 class TTree;
17 class AliRunTag;
18 class AliEventTag;
19 class TMap;
20 class AliESDfriend;
21 class AliESDpid;
22 class AliESDEvent;
23 class AliPIDResponse;
24
25
26 class AliESDInputHandler : public AliInputEventHandler {
27
28  public:
29     AliESDInputHandler();
30     AliESDInputHandler(const char* name, const char* title);
31     virtual ~AliESDInputHandler();
32     virtual Bool_t       Init(Option_t* opt) {return AliInputEventHandler::Init(opt);}
33     virtual Bool_t       Init(TTree* tree, Option_t* opt);
34     virtual Bool_t       BeginEvent(Long64_t entry);
35     virtual Bool_t       Notify() { return AliInputEventHandler::Notify(); };
36     virtual Bool_t       Notify(const char* path);
37     virtual Bool_t       FinishEvent();
38     void                 CheckSelectionMask();
39     AliESDEvent         *GetEvent()        const {return fEvent;}
40     Option_t            *GetAnalysisType() const {return fAnalysisType;}
41     Option_t            *GetDataType() const;
42     // Tag cut summary analysis
43     Int_t                GetNEventAcceptedInFile();
44     Int_t                GetNEventRejectedInFile();
45     Bool_t               GetCutSummaryForChain(Int_t *aTotal, Int_t *aAccepted, Int_t *aRejected);
46     Int_t                GetNFilesEmpty();
47     // HLT  analysis
48     AliESDEvent         *GetHLTEvent()     const {return fHLTEvent;}
49     TTree               *GetHLTTree()      const {return fHLTTree;}    
50     void                 SetReadHLT()            {fUseHLT = kTRUE;}
51     // Friends&Co
52     AliESDfriend        *GetESDfriend()    const {return fFriend;}
53     void                 SetReadFriends(Bool_t flag)   {fReadFriends = flag;}
54     void                 SetFriendFileName(const char *fname)  {fFriendFileName = fname;}
55     // Tag analysis
56     void                 SetReadTags()           {fUseTags = kTRUE;}
57     AliRunTag           *GetRunTag() const       {return fRunTag;}
58     const AliEventTag   *GetEventTag() const     {return fEventTag;}
59     // Get the statistics object (currently TH2). Option can be BIN0.
60     virtual TObject     *GetStatistics(Option_t *option="") const;
61
62     //PID response
63     virtual AliPIDResponse* GetPIDResponse() {return (AliPIDResponse*)fESDpid;}
64     virtual void CreatePIDResponse(Bool_t isMC=kFALSE);
65     AliESDpid           *GetESDpid()       const {return fESDpid;}
66     void                 SetESDpid(AliESDpid* pid)     {fESDpid = pid;}
67   
68  private:
69     AliESDInputHandler(const AliESDInputHandler& handler);             
70     AliESDInputHandler& operator=(const AliESDInputHandler& handler);  
71     void                 ConnectFriends();
72  protected:
73     // ESD event
74     AliESDEvent    *fEvent;         //! Pointer to the event
75     AliESDfriend   *fFriend;        //! Pointer to the esd friend
76     AliESDpid      *fESDpid;        //! Pointer to PID information
77     Option_t       *fAnalysisType;  //! local, proof, grid
78     Int_t           fNEvents;       //! Number of events in the current tree
79     // HLT event
80     AliESDEvent    *fHLTEvent;      //! Pointer to the HLT Event (if present)
81     TTree          *fHLTTree;       //! Pointer to the HLT Event (if present)
82     Bool_t          fUseHLT;        //  Flag to access HLT Events
83     // ESD Tag Cut Summary
84     TMap           *fTagCutSumm;    //! Tag cut summary map
85     // ESD Tags (optional)
86     Bool_t          fUseTags;       //  Flag to use tags
87     TChain         *fChainT;        //! File with event tags
88     TTree          *fTreeT;         //! Tree of tags
89     AliRunTag      *fRunTag;        //! Pointer to the run tag
90     const AliEventTag *fEventTag;      //! Current event tag
91     // Friends
92     Bool_t          fReadFriends;   //  Flag for friends reading 
93     TString         fFriendFileName;//  Name of the file containing the frien tree (branch)
94     ClassDef(AliESDInputHandler, 6);
95 };
96
97 #endif