#include <FWCore/FWLite/interface/TFWLiteSelectorBasic.h>
Public Member Functions | |
virtual void | begin (TList *&in)=0 |
virtual void | postProcessing (TList &out)=0 |
virtual void | preProcessing (const TList *in, TList &out)=0 |
virtual void | process (const edm::Event &event)=0 |
virtual void | terminate (TList &out)=0 |
TFWLiteSelectorBasic () | |
virtual | ~TFWLiteSelectorBasic () |
Private Member Functions | |
virtual void | Begin (TTree *) |
virtual void | Init (TTree *) |
virtual Bool_t | Notify () |
const TFWLiteSelectorBasic & | operator= (const TFWLiteSelectorBasic &) |
virtual Bool_t | Process (Long64_t) |
void | setupNewFile (TFile &) |
virtual void | SlaveBegin (TTree *) |
virtual void | SlaveTerminate () |
virtual void | Terminate () |
TFWLiteSelectorBasic (const TFWLiteSelectorBasic &) | |
virtual Int_t | Version () const |
Private Attributes | |
bool | everythingOK_ |
boost::shared_ptr < edm::root::TFWLiteSelectorMembers > | m_ |
Description: A ROOT TSelector which accesses data using an edm::Event
Usage: By inheriting from this class one can make a TSelector for ROOT which works with PROOF and which allows you to access data using an edm::Event.
Definition at line 41 of file TFWLiteSelectorBasic.h.
TFWLiteSelectorBasic::TFWLiteSelectorBasic | ( | ) |
Definition at line 155 of file TFWLiteSelectorBasic.cc.
: m_(new edm::root::TFWLiteSelectorMembers), everythingOK_(false) { }
TFWLiteSelectorBasic::~TFWLiteSelectorBasic | ( | ) | [virtual] |
Definition at line 164 of file TFWLiteSelectorBasic.cc.
{ }
TFWLiteSelectorBasic::TFWLiteSelectorBasic | ( | const TFWLiteSelectorBasic & | ) | [private] |
virtual void TFWLiteSelectorBasic::begin | ( | TList *& | in | ) | [pure virtual] |
Called each time the 'client' begins processing (remote 'slaves' do not see this message)
in | an assignable pointer to a list of objects you want passed to 'preProcessing'. This list is used to communicate with remote slaves. NOTE: you are responsible for deleting this TList and its content once you are done with it. |
Implemented in tselname, tfwliteselectortest::ThingsTSelector, tfwliteselectortest::ThingsTSelector2, and examples::TrackTSelector.
Referenced by Begin().
void TFWLiteSelectorBasic::Begin | ( | TTree * | iTree | ) | [private, virtual] |
void TFWLiteSelectorBasic::Init | ( | TTree * | iTree | ) | [private, virtual] |
Definition at line 195 of file TFWLiteSelectorBasic.cc.
References m_.
Referenced by Begin(), and SlaveBegin().
{ if(iTree == 0) return; m_->setTree(iTree); }
Bool_t TFWLiteSelectorBasic::Notify | ( | ) | [private, virtual] |
Definition at line 202 of file TFWLiteSelectorBasic.cc.
References gather_cfg::cout, everythingOK_, mergeVDriftHistosByStation::file, m_, and setupNewFile().
{ //std::cout << "Notify start" << std::endl; //we have switched to a new file //get new file from Tree if(0 == m_->tree_) { std::cout << "No tree" << std::endl; return kFALSE; } TFile* file = m_->tree_->GetCurrentFile(); if(0 == file) { //When in Rome, do as the Romans TChain* chain = dynamic_cast<TChain*>(m_->tree_); if(0 == chain) { std::cout << "No file" << std::endl; return kFALSE; } file = chain->GetFile(); if(0 == file) { std::cout << "No file" << std::endl; return kFALSE; } } setupNewFile(*file); return everythingOK_ ? kTRUE: kFALSE; }
const TFWLiteSelectorBasic& TFWLiteSelectorBasic::operator= | ( | const TFWLiteSelectorBasic & | ) | [private] |
virtual void TFWLiteSelectorBasic::postProcessing | ( | TList & | out | ) | [pure virtual] |
Called each time the 'slave' has seen all the events
out | the list of objects that will be sent to 'terminate'. You can Add() additional objects to 'out' at this point as well. |
Implemented in TFWLiteSelector< TWorker >, tfwliteselectortest::ThingsTSelector, TFWLiteSelector< tselnameWorker >, TFWLiteSelector< ThingsWorker >, and TFWLiteSelector< TrackAnalysisAlgorithm >.
Referenced by SlaveTerminate().
virtual void TFWLiteSelectorBasic::preProcessing | ( | const TList * | in, |
TList & | out | ||
) | [pure virtual] |
Called each time the 'slave' is about to start processing
in | a pointer to the list of objects created in 'begin()'. The pointer can be 0 |
out | a list of objects that are the result of processing (e.g. histograms). You should call 'Add()' for each object you want sent to the 'terminate' method. |
Implemented in TFWLiteSelector< TWorker >, tfwliteselectortest::ThingsTSelector, TFWLiteSelector< tselnameWorker >, TFWLiteSelector< ThingsWorker >, and TFWLiteSelector< TrackAnalysisAlgorithm >.
Referenced by SlaveBegin().
Bool_t TFWLiteSelectorBasic::Process | ( | Long64_t | iEntry | ) | [private, virtual] |
Definition at line 240 of file TFWLiteSelectorBasic.cc.
References printConversionInfo::aux, edm::poolNames::branchListIndexesBranchName(), edm::BranchTypeToAuxiliaryBranchName(), gather_cfg::cout, event(), edm::poolNames::eventSelectionsBranchName(), everythingOK_, exception, Exception, edm::errors::FatalRootError, edm::InEvent, m_, process(), edm::EventAuxiliary::run(), and edm::EventAuxiliary::time().
{ //std::cout << "Process start" << std::endl; if(everythingOK_) { std::auto_ptr<edm::EventAuxiliary> eaux(new edm::EventAuxiliary()); edm::EventAuxiliary& aux = *eaux; edm::EventAuxiliary* pAux= eaux.get(); TBranch* branch = m_->tree_->GetBranch(edm::BranchTypeToAuxiliaryBranchName(edm::InEvent).c_str()); branch->SetAddress(&pAux); branch->GetEntry(iEntry); //NEW m_->processNames_ = aux.processHistory(); // std::cout << "ProcessNames\n"; // for(edm::ProcessNameList::const_iterator itName = m_->processNames_.begin(), // itNameEnd = m_->processNames_.end(); // itName != itNameEnd; // ++itName) { // std::cout << " " << *itName << std::endl; // } boost::shared_ptr<edm::EventSelectionIDVector> eventSelectionIDs_(new edm::EventSelectionIDVector); edm::EventSelectionIDVector* pEventSelectionIDVector = eventSelectionIDs_.get(); TBranch* eventSelectionsBranch = m_->tree_->GetBranch(edm::poolNames::eventSelectionsBranchName().c_str()); if(!eventSelectionsBranch) { throw edm::Exception(edm::errors::FatalRootError) << "Failed to find event Selections branch in event tree"; } eventSelectionsBranch->SetAddress(&pEventSelectionIDVector); eventSelectionsBranch->GetEntry(iEntry); boost::shared_ptr<edm::BranchListIndexes> branchListIndexes_(new edm::BranchListIndexes); edm::BranchListIndexes* pBranchListIndexes = branchListIndexes_.get(); TBranch* branchListIndexBranch = m_->tree_->GetBranch(edm::poolNames::branchListIndexesBranchName().c_str()); if(!branchListIndexBranch) { throw edm::Exception(edm::errors::FatalRootError) << "Failed to find branch list index branch in event tree"; } branchListIndexBranch->SetAddress(&pBranchListIndexes); branchListIndexBranch->GetEntry(iEntry); m_->branchIDListHelper_->fixBranchListIndexes(*branchListIndexes_); try { m_->reader_->setEntry(iEntry); boost::shared_ptr<edm::RunAuxiliary> runAux(new edm::RunAuxiliary(aux.run(), aux.time(), aux.time())); boost::shared_ptr<edm::RunPrincipal> rp(new edm::RunPrincipal(runAux, m_->reg_, m_->pc_)); boost::shared_ptr<edm::LuminosityBlockAuxiliary> lumiAux( new edm::LuminosityBlockAuxiliary(rp->run(), 1, aux.time(), aux.time())); boost::shared_ptr<edm::LuminosityBlockPrincipal>lbp( new edm::LuminosityBlockPrincipal(lumiAux, m_->reg_, m_->pc_)); m_->ep_->fillEventPrincipal(*eaux, eventSelectionIDs_, branchListIndexes_, m_->mapper_, m_->reader_.get()); lbp->setRunPrincipal(rp); m_->ep_->setLuminosityBlockPrincipal(lbp); m_->processNames_ = m_->ep_->processHistory(); edm::Event event(*m_->ep_, m_->md_); //Make the event principal accessible to edm::Ref's Operate sentry(m_->ep_->prodGetter()); process(event); } catch(std::exception const& iEx) { std::cout << "While processing entry " << iEntry << " the following exception was caught \n" << iEx.what() << std::endl; } catch(...) { std::cout << "While processing entry " << iEntry << " an unknown exception was caught" << std::endl; } } return everythingOK_ ? kTRUE: kFALSE; }
virtual void TFWLiteSelectorBasic::process | ( | const edm::Event & | event | ) | [pure virtual] |
Call each time the 'slave' gets a new Event
event | a standard edm::Event which works just like it does in cmsRun |
Implemented in TFWLiteSelector< TWorker >, tfwliteselectortest::ThingsTSelector, TFWLiteSelector< tselnameWorker >, TFWLiteSelector< ThingsWorker >, and TFWLiteSelector< TrackAnalysisAlgorithm >.
Referenced by Process().
void TFWLiteSelectorBasic::setupNewFile | ( | TFile & | iFile | ) | [private] |
Definition at line 321 of file TFWLiteSelectorBasic.cc.
References edm::poolNames::branchIDListBranchName(), edm::BranchDescription::branchName(), edm::BranchDescription::branchType(), edm::BranchDescription::className(), gather_cfg::cout, everythingOK_, Exception, edm::poolNames::fileFormatVersionBranchName(), edm::errors::FileReadError, edm::fillProductRegistryTransients(), edm::BranchDescription::friendlyClassName(), edm::friendlyname::friendlyName(), i, edm::poolNames::idToParameterSetBlobsBranchName(), edm::InEvent, edm::BranchDescription::init(), edm::detail::ThreadSafeRegistry< KEY, T, E >::insertMapped(), instance, m_, edm::poolNames::metaDataTreeName(), edm::poolNames::parameterSetMapBranchName(), edm::poolNames::parameterSetsTreeName(), edm::poolNames::processConfigurationBranchName(), edm::poolNames::processHistoryBranchName(), parseEventContent::prod, edm::poolNames::productDescriptionBranchName(), edm::BranchDescription::setDropped(), edm::ParameterSet::setID(), AlCaHLTBitMon_QueryRunRegistry::string, groupFilesInBlocks::temp, edm::errors::UnimplementedFeature, and edm::BranchDescription::updateFriendlyClassName().
Referenced by Notify().
{ //look up meta-data //get product registry //std::vector<edm::EventProcessHistoryID> eventProcessHistoryIDs_; TTree* metaDataTree = dynamic_cast<TTree*>(iFile.Get(edm::poolNames::metaDataTreeName().c_str())); if(!metaDataTree) { std::cout << "could not find TTree " << edm::poolNames::metaDataTreeName() << std::endl; everythingOK_ = false; return; } edm::FileFormatVersion* fftPtr = &(m_->fileFormatVersion_); if(metaDataTree->FindBranch(edm::poolNames::fileFormatVersionBranchName().c_str()) != 0) { metaDataTree->SetBranchAddress(edm::poolNames::fileFormatVersionBranchName().c_str(), &fftPtr); } edm::ProductRegistry* pReg = &(*m_->reg_); metaDataTree->SetBranchAddress(edm::poolNames::productDescriptionBranchName().c_str(), &(pReg)); m_->reg_->setFrozen(); typedef std::map<edm::ParameterSetID, edm::ParameterSetBlob> PsetMap; PsetMap psetMap; PsetMap *psetMapPtr = &psetMap; if(metaDataTree->FindBranch(edm::poolNames::parameterSetMapBranchName().c_str()) != 0) { metaDataTree->SetBranchAddress(edm::poolNames::parameterSetMapBranchName().c_str(), &psetMapPtr); } else { TTree* psetTree = dynamic_cast<TTree *>(iFile.Get(edm::poolNames::parameterSetsTreeName().c_str())); if(0 == psetTree) { throw edm::Exception(edm::errors::FileReadError) << "Could not find tree " << edm::poolNames::parameterSetsTreeName() << " in the input file.\n"; } typedef std::pair<edm::ParameterSetID, edm::ParameterSetBlob> IdToBlobs; IdToBlobs idToBlob; IdToBlobs* pIdToBlob = &idToBlob; psetTree->SetBranchAddress(edm::poolNames::idToParameterSetBlobsBranchName().c_str(), &pIdToBlob); for(long long i = 0; i != psetTree->GetEntries(); ++i) { psetTree->GetEntry(i); psetMap.insert(idToBlob); } } edm::ProcessHistoryRegistry::vector_type pHistVector; edm::ProcessHistoryRegistry::vector_type *pHistVectorPtr = &pHistVector; if(metaDataTree->FindBranch(edm::poolNames::processHistoryBranchName().c_str()) != 0) { metaDataTree->SetBranchAddress(edm::poolNames::processHistoryBranchName().c_str(), &pHistVectorPtr); } edm::ProcessConfigurationVector procConfigVector; edm::ProcessConfigurationVector* procConfigVectorPtr = &procConfigVector; if(metaDataTree->FindBranch(edm::poolNames::processConfigurationBranchName().c_str()) != 0) { metaDataTree->SetBranchAddress(edm::poolNames::processConfigurationBranchName().c_str(), &procConfigVectorPtr); } boost::shared_ptr<edm::BranchIDListHelper> branchIDListsHelper(new edm::BranchIDListHelper); edm::BranchIDLists const* branchIDListsPtr = &branchIDListsHelper->branchIDLists(); if(metaDataTree->FindBranch(edm::poolNames::branchIDListBranchName().c_str()) != 0) { metaDataTree->SetBranchAddress(edm::poolNames::branchIDListBranchName().c_str(), &branchIDListsPtr); } metaDataTree->GetEntry(0); // Merge into the registries. For now, we do NOT merge the product registry. edm::pset::Registry& psetRegistry = *edm::pset::Registry::instance(); for(PsetMap::const_iterator i = psetMap.begin(), iEnd = psetMap.end(); i != iEnd; ++i) { edm::ParameterSet pset(i->second.pset()); pset.setID(i->first); psetRegistry.insertMapped(pset); } edm::ProcessHistoryRegistry::instance()->insertCollection(pHistVector); edm::ProcessConfigurationRegistry::instance()->insertCollection(procConfigVector); m_->pointerToBranchBuffer_.erase(m_->pointerToBranchBuffer_.begin(), m_->pointerToBranchBuffer_.end()); fillProductRegistryTransients(procConfigVector, *m_->reg_); std::auto_ptr<edm::ProductRegistry> newReg(new edm::ProductRegistry()); edm::ProductRegistry::ProductList const& prodList = m_->reg_->productList(); { for(edm::ProductRegistry::ProductList::const_iterator it = prodList.begin(), itEnd = prodList.end(); it != itEnd; ++it) { edm::BranchDescription const& prod = it->second; //std::cout << "productname = " << it->second << " end " << std::endl; std::string newFriendlyName = edm::friendlyname::friendlyName(prod.className()); if(newFriendlyName == prod.friendlyClassName()) { newReg->copyProduct(prod); } else { if(m_->fileFormatVersion_.splitProductIDs()) { throw edm::Exception(edm::errors::UnimplementedFeature) << "Cannot change friendly class name algorithm without more development work\n" << "to update BranchIDLists. Contact the framework group.\n"; } edm::BranchDescription newBD(prod); newBD.updateFriendlyClassName(); newReg->copyProduct(newBD); // Need to call init to get old branch name. prod.init(); } } newReg->setFrozen(); m_->reg_.reset(newReg.release()); } edm::ProductRegistry::ProductList const& prodList2 = m_->reg_->productList(); std::vector<edm::EventEntryDescription> temp(prodList2.size(), edm::EventEntryDescription()); m_->prov_.swap(temp); std::vector<edm::EventEntryDescription>::iterator itB = m_->prov_.begin(); m_->pointerToBranchBuffer_.reserve(prodList2.size()); for(edm::ProductRegistry::ProductList::const_iterator it = prodList2.begin(), itEnd = prodList2.end(); it != itEnd; ++it, ++itB) { edm::BranchDescription const& prod = it->second; if(prod.branchType() == edm::InEvent) { prod.init(); //NEED to do this and check to see if branch exists if(m_->tree_->GetBranch(prod.branchName().c_str()) == 0) { prod.setDropped(); } //std::cout << "id " << it->first << " branch " << it->second << std::endl; //m_->pointerToBranchBuffer_.push_back(&(*itB)); //void* tmp = &(m_->pointerToBranchBuffer_.back()); //edm::EventEntryDescription* tmp = &(*itB); //CDJ need to fix provenance and be backwards compatible, for now just don't read the branch //m_->metaTree_->SetBranchAddress(prod.branchName().c_str(), tmp); } } m_->branchIDListHelper_->updateFromInput(*branchIDListsPtr); m_->ep_.reset(new edm::EventPrincipal(m_->reg_, m_->branchIDListHelper_, m_->pc_)); everythingOK_ = true; }
void TFWLiteSelectorBasic::SlaveBegin | ( | TTree * | iTree | ) | [private, virtual] |
Definition at line 189 of file TFWLiteSelectorBasic.cc.
References Init(), and preProcessing().
{ Init(iTree); preProcessing(fInput, *fOutput); }
void TFWLiteSelectorBasic::SlaveTerminate | ( | ) | [private, virtual] |
Definition at line 311 of file TFWLiteSelectorBasic.cc.
References postProcessing().
{ postProcessing(*fOutput); }
void TFWLiteSelectorBasic::Terminate | ( | ) | [private, virtual] |
Definition at line 316 of file TFWLiteSelectorBasic.cc.
References terminate().
{ terminate(*fOutput); }
virtual void TFWLiteSelectorBasic::terminate | ( | TList & | out | ) | [pure virtual] |
Called each time the 'client' has finished processing.
out | contains the accumulated output of all slaves. |
Implemented in tselname, tfwliteselectortest::ThingsTSelector, tfwliteselectortest::ThingsTSelector2, and examples::TrackTSelector.
Referenced by Terminate().
virtual Int_t TFWLiteSelectorBasic::Version | ( | ) | const [inline, private, virtual] |
Definition at line 95 of file TFWLiteSelectorBasic.h.
{ return 1; }
bool TFWLiteSelectorBasic::everythingOK_ [private] |
Definition at line 100 of file TFWLiteSelectorBasic.h.
Referenced by Notify(), Process(), and setupNewFile().
boost::shared_ptr<edm::root::TFWLiteSelectorMembers> TFWLiteSelectorBasic::m_ [private] |
Definition at line 99 of file TFWLiteSelectorBasic.h.
Referenced by Init(), Notify(), Process(), and setupNewFile().