| 1 |
//--------------------------------------------------------------------------------------------------
|
| 2 |
// $Id: Selector.h,v 1.11 2009/03/12 18:25:10 loizides Exp $
|
| 3 |
//
|
| 4 |
// Our selector class for modular processing of a tree (or chain). In addition to the generic
|
| 5 |
// TAMSelector it determines the begin and end of a run and does the necessary bookkeeping.
|
| 6 |
// Modules can ask the selector to provide the run information.
|
| 7 |
//
|
| 8 |
// Authors: C.Loizides
|
| 9 |
//--------------------------------------------------------------------------------------------------
|
| 10 |
|
| 11 |
#ifndef MITANA_TREEMOD_SELECTOR_H
|
| 12 |
#define MITANA_TREEMOD_SELECTOR_H
|
| 13 |
|
| 14 |
#include "MitAna/TAM/interface/TAModule.h"
|
| 15 |
#include "MitAna/TAM/interface/TAMSelector.h"
|
| 16 |
#include "MitAna/DataTree/interface/Collections.h"
|
| 17 |
#include "MitAna/DataTree/interface/EventHeader.h"
|
| 18 |
#include "MitAna/DataTree/interface/LAHeader.h"
|
| 19 |
#include "MitAna/DataTree/interface/RunInfo.h"
|
| 20 |
|
| 21 |
namespace mithep {
|
| 22 |
class OutputMod;
|
| 23 |
|
| 24 |
class Selector : public TAMSelector
|
| 25 |
{
|
| 26 |
public:
|
| 27 |
Selector();
|
| 28 |
~Selector();
|
| 29 |
|
| 30 |
void AddToTrash(TObject *obj) { fTrash.AddLast(obj); }
|
| 31 |
const char *GetAllEvtTreeName() const { return fAllEvtTreeName; }
|
| 32 |
const char *GetAllEvtHdrBrn() const { return fAllEvtHdrBrn; }
|
| 33 |
const THashTable &GetBranchTable() const { return fBranchTable; }
|
| 34 |
const char *GetEvtHdrName() const { return fEvtHdrName; }
|
| 35 |
const EventHeader *GetEventHeader() const { return fEventHeader; }
|
| 36 |
const char *GetLATreeName() const { return fLATreeName; }
|
| 37 |
const char *GetLAHdrName() const { return fLAHdrName; }
|
| 38 |
const char *GetRunTreeName() const { return fRunTreeName; }
|
| 39 |
const char *GetRunInfoName() const { return fRunInfoName; }
|
| 40 |
const RunInfo *GetRunInfo() const { return fRunInfo; }
|
| 41 |
Bool_t ValidRunInfo() const;
|
| 42 |
Bool_t ValidRunNum() const { return fCurRunNum!=UInt_t(-1); }
|
| 43 |
|
| 44 |
protected:
|
| 45 |
Bool_t BeginRun();
|
| 46 |
Bool_t ConsistentRunNum() const;
|
| 47 |
Bool_t EndRun();
|
| 48 |
Bool_t Notify();
|
| 49 |
Bool_t Process(Long64_t entry);
|
| 50 |
void SlaveBegin(TTree* tree);
|
| 51 |
void UpdateRunInfo();
|
| 52 |
void UpdateRunInfoTree();
|
| 53 |
|
| 54 |
Bool_t fDoRunInfo; //=true then get RunInfo (def=1)
|
| 55 |
TString fEvtHdrName; //name of event header branch
|
| 56 |
TString fRunTreeName; //name of run info tree
|
| 57 |
TString fRunInfoName; //name of run info branch
|
| 58 |
TString fAllEvtHdrBrn; //name of all-event header branch
|
| 59 |
TString fLATreeName; //name of look-ahead tree
|
| 60 |
TString fLAHdrName; //name of look-ahead event header branch
|
| 61 |
TString fAllEvtTreeName; //name of all-event tree
|
| 62 |
TTree *fRunTree; //!run info tree in current file
|
| 63 |
EventHeader *fEventHeader; //!event header for current event
|
| 64 |
RunInfo *fRunInfo; //!run information for current run
|
| 65 |
TTree *fLATree; //!look-ahead tree in current file
|
| 66 |
LAHeader *fLAHeader; //!event header for next event
|
| 67 |
UInt_t fCurRunNum; //!current run number
|
| 68 |
TList fOutputMods; //!pointer(s) to output modules
|
| 69 |
TObjArray fTrash; //!pointers to trashed objects
|
| 70 |
|
| 71 |
private:
|
| 72 |
void SearchOutputMods(const TAModule *mod);
|
| 73 |
|
| 74 |
friend class OutputMod;
|
| 75 |
|
| 76 |
ClassDef(Selector, 1) // Customized selector class
|
| 77 |
};
|
| 78 |
}
|
| 79 |
|
| 80 |
//--------------------------------------------------------------------------------------------------
|
| 81 |
inline Bool_t mithep::Selector::ConsistentRunNum() const
|
| 82 |
{
|
| 83 |
return (ValidRunNum() && fCurRunNum==fEventHeader->RunNum());
|
| 84 |
}
|
| 85 |
|
| 86 |
//--------------------------------------------------------------------------------------------------
|
| 87 |
inline Bool_t mithep::Selector::ValidRunInfo() const
|
| 88 |
{
|
| 89 |
return (fRunInfo && fCurRunNum==fRunInfo->RunNum());
|
| 90 |
}
|
| 91 |
#endif
|
