class TChain;
class TTree;
class AliRunTag;
+class AliEventTag;
class TMap;
+class AliESDfriend;
+class AliESDpid;
+class AliESDEvent;
+class AliPIDResponse;
+
class AliESDInputHandler : public AliInputEventHandler {
AliESDInputHandler();
AliESDInputHandler(const char* name, const char* title);
virtual ~AliESDInputHandler();
- virtual Bool_t Init(Option_t* /*opt*/) {return kTRUE;}
+ virtual Bool_t Init(Option_t* opt) {return AliInputEventHandler::Init(opt);}
virtual Bool_t Init(TTree* tree, Option_t* opt);
virtual Bool_t BeginEvent(Long64_t entry);
- virtual Bool_t Notify() { return AliVEventHandler::Notify(); };
+ virtual Bool_t Notify() { return AliInputEventHandler::Notify(); };
virtual Bool_t Notify(const char* path);
virtual Bool_t FinishEvent();
+ void CheckSelectionMask();
AliESDEvent *GetEvent() const {return fEvent;}
Option_t *GetAnalysisType() const {return fAnalysisType;}
Option_t *GetDataType() const;
// Tag cut summary analysis
Int_t GetNEventAcceptedInFile();
Int_t GetNEventRejectedInFile();
- // HLT analysis
+ Bool_t GetCutSummaryForChain(Int_t *aTotal, Int_t *aAccepted, Int_t *aRejected);
+ Int_t GetNFilesEmpty();
+ // HLT analysis
AliESDEvent *GetHLTEvent() const {return fHLTEvent;}
TTree *GetHLTTree() const {return fHLTTree;}
- void SetReadHLT() {fUseHLT = kTRUE;}
+ void SetReadHLT() {fUseHLT = kTRUE;}
+ // Friends&Co
+ AliESDfriend *GetESDfriend() const {return fFriend;}
+ void SetReadFriends(Bool_t flag) {fReadFriends = flag;}
+ void SetFriendFileName(const char *fname) {fFriendFileName = fname;}
// Tag analysis
- void SetReadTags() {fUseTags = kTRUE;}
- AliRunTag *GetRunTag() const {return fRunTag;}
-
+ void SetReadTags() {fUseTags = kTRUE;}
+ AliRunTag *GetRunTag() const {return fRunTag;}
+ const AliEventTag *GetEventTag() const {return fEventTag;}
+ // Get the statistics object (currently TH2). Option can be BIN0.
+ virtual TObject *GetStatistics(Option_t *option="") const;
+
+ //PID response
+ virtual AliPIDResponse* GetPIDResponse() {return (AliPIDResponse*)fESDpid;}
+ virtual void CreatePIDResponse(Bool_t isMC=kFALSE);
+ AliESDpid *GetESDpid() const {return fESDpid;}
+ void SetESDpid(AliESDpid* pid) {fESDpid = pid;}
+
private:
AliESDInputHandler(const AliESDInputHandler& handler);
AliESDInputHandler& operator=(const AliESDInputHandler& handler);
protected:
// ESD event
- AliESDEvent *fEvent; //! Pointer to the event
- Option_t *fAnalysisType; //! local, proof, grid
- Int_t fNEvents; //! Number of events in the current tree
+ AliESDEvent *fEvent; //! Pointer to the event
+ AliESDfriend *fFriend; //! Pointer to the esd friend
+ AliESDpid *fESDpid; //! Pointer to PID information
+ Option_t *fAnalysisType; //! local, proof, grid
+ Int_t fNEvents; //! Number of events in the current tree
// HLT event
- AliESDEvent *fHLTEvent; //! Pointer to the HLT Event (if present)
- TTree *fHLTTree; //! Pointer to the HLT Event (if present)
- Bool_t fUseHLT; // Flag to access HLT Events
+ AliESDEvent *fHLTEvent; //! Pointer to the HLT Event (if present)
+ TTree *fHLTTree; //! Pointer to the HLT Event (if present)
+ Bool_t fUseHLT; // Flag to access HLT Events
// ESD Tag Cut Summary
- TMap *fTagCutSumm; //! Tag cut summary map
+ TMap *fTagCutSumm; //! Tag cut summary map
// ESD Tags (optional)
- Bool_t fUseTags; // Flag to use tags
- TChain *fChainT; //! File with event tags
- TTree *fTreeT; //! Tree of tags
- AliRunTag *fRunTag; //! Pointer to the run tag
- ClassDef(AliESDInputHandler, 4);
+ Bool_t fUseTags; // Flag to use tags
+ TChain *fChainT; //! File with event tags
+ TTree *fTreeT; //! Tree of tags
+ AliRunTag *fRunTag; //! Pointer to the run tag
+ const AliEventTag *fEventTag; //! Current event tag
+ // Friends
+ Bool_t fReadFriends; // Flag for friends reading
+ TString fFriendFileName;// Name of the file containing the frien tree (branch)
+ ClassDef(AliESDInputHandler, 6);
};
#endif