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

#ifndef ALIESDINPUTHANDLER_H
#define ALIESDINPUTHANDLER_H
/* Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

//-------------------------------------------------------------------------
//     ESD Input Handler realisation of the AliVEventHandler interface
//     Author: Andreas Morsch, CERN
//-------------------------------------------------------------------------

#include "AliInputEventHandler.h"
#include "AliESDEvent.h"
class TChain;
class TTree;
class AliRunTag;
class AliEventTag;
class TMap;
class AliESDfriend;
class AliESDpid;


class AliESDInputHandler : public AliInputEventHandler {

 public:
    AliESDInputHandler();
    AliESDInputHandler(const char* name, const char* title);
    virtual ~AliESDInputHandler();
    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 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();
    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;}
    // Friends&Co
    AliESDfriend        *GetESDfriend()    const {return fFriend;}
    AliESDpid           *GetESDpid()       const {return fESDpid;}
    void                 SetESDpid(AliESDpid* pid)     {fESDpid = pid;}
    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;}
    const AliEventTag   *GetEventTag() const     {return fEventTag;}
    // Get the statistics object (currently TH2). Option can be BIN0.
    virtual TObject     *GetStatistics(Option_t *option="") const;
 private:
    AliESDInputHandler(const AliESDInputHandler& handler);             
    AliESDInputHandler& operator=(const AliESDInputHandler& handler);  
 protected:
    // ESD event
    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
    // ESD Tag Cut Summary
    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
    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
Commit	Line	Data
	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
	23
	24	class AliESDInputHandler : public AliInputEventHandler {
	25
	26	public:
	27	AliESDInputHandler();
	28	AliESDInputHandler(const char* name, const char* title);
	29	virtual ~AliESDInputHandler();
	30	virtual Bool_t Init(Option_t* opt) {return AliInputEventHandler::Init(opt);}
	31	virtual Bool_t Init(TTree* tree, Option_t* opt);
	32	virtual Bool_t BeginEvent(Long64_t entry);
	33	virtual Bool_t Notify() { return AliInputEventHandler::Notify(); };
	34	virtual Bool_t Notify(const char* path);
	35	virtual Bool_t FinishEvent();
	36	void CheckSelectionMask();
	37	AliESDEvent *GetEvent() const {return fEvent;}
	38	Option_t *GetAnalysisType() const {return fAnalysisType;}
	39	Option_t *GetDataType() const;
	40	// Tag cut summary analysis
	41	Int_t GetNEventAcceptedInFile();
	42	Int_t GetNEventRejectedInFile();
	43	Bool_t GetCutSummaryForChain(Int_t aTotal, Int_t aAccepted, Int_t *aRejected);
	44	Int_t GetNFilesEmpty();
	45	// HLT analysis
	46	AliESDEvent *GetHLTEvent() const {return fHLTEvent;}
	47	TTree *GetHLTTree() const {return fHLTTree;}
	48	void SetReadHLT() {fUseHLT = kTRUE;}
	49	// Friends&Co
	50	AliESDfriend *GetESDfriend() const {return fFriend;}
	51	AliESDpid *GetESDpid() const {return fESDpid;}
	52	void SetESDpid(AliESDpid* pid) {fESDpid = pid;}
	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	private:
	62	AliESDInputHandler(const AliESDInputHandler& handler);
	63	AliESDInputHandler& operator=(const AliESDInputHandler& handler);
	64	protected:
	65	// ESD event
	66	AliESDEvent *fEvent; //! Pointer to the event
	67	AliESDfriend *fFriend; //! Pointer to the esd friend
	68	AliESDpid *fESDpid; //! Pointer to PID information
	69	Option_t *fAnalysisType; //! local, proof, grid
	70	Int_t fNEvents; //! Number of events in the current tree
	71	// HLT event
	72	AliESDEvent *fHLTEvent; //! Pointer to the HLT Event (if present)
	73	TTree *fHLTTree; //! Pointer to the HLT Event (if present)
	74	Bool_t fUseHLT; // Flag to access HLT Events
	75	// ESD Tag Cut Summary
	76	TMap *fTagCutSumm; //! Tag cut summary map
	77	// ESD Tags (optional)
	78	Bool_t fUseTags; // Flag to use tags
	79	TChain *fChainT; //! File with event tags
	80	TTree *fTreeT; //! Tree of tags
	81	AliRunTag *fRunTag; //! Pointer to the run tag
	82	const AliEventTag *fEventTag; //! Current event tag
	83	// Friends
	84	Bool_t fReadFriends; // Flag for friends reading
	85	TString fFriendFileName;// Name of the file containing the frien tree (branch)
	86	ClassDef(AliESDInputHandler, 6);
	87	};
	88
	89	#endif