CMS 3D CMS Logo

List of all members | Public Types | Public Member Functions | Private Attributes
edm::WorkerManager Class Reference

#include <WorkerManager.h>

Public Types

typedef std::vector< Worker * > AllWorkers
 

Public Member Functions

ExceptionToActionTable const & actionTable () const
 
void addToAllWorkers (Worker *w)
 
void addToUnscheduledWorkers (ParameterSet &pset, ProductRegistry &preg, PreallocationConfiguration const *prealloc, std::shared_ptr< ProcessConfiguration > processConfiguration, std::string label, std::set< std::string > &unscheduledLabels, std::vector< std::string > &shouldBeUsedLabels)
 
AllWorkers const & allWorkers () const
 
void beginJob (ProductRegistry const &iRegistry, eventsetup::ESRecordsToProxyIndices const &)
 
void beginStream (StreamID iID, StreamContext &streamContext)
 
void deleteModuleIfExists (std::string const &moduleLabel)
 
void endJob ()
 
void endJob (ExceptionCollector &collector)
 
void endStream (StreamID iID, StreamContext &streamContext)
 
WorkergetWorker (ParameterSet &pset, ProductRegistry &preg, PreallocationConfiguration const *prealloc, std::shared_ptr< ProcessConfiguration const > processConfiguration, std::string const &label)
 
template<typename T >
void processAccumulatorsAsync (WaitingTaskHolder, typename T::TransitionInfoType const &, ServiceToken const &, StreamID, ParentContext const &, typename T::Context const *)
 
template<typename T , typename U >
void processOneOccurrenceAsync (WaitingTaskHolder, typename T::TransitionInfoType &, ServiceToken const &, StreamID, typename T::Context const *topContext, U const *context)
 
void resetAll ()
 
void setupOnDemandSystem (EventTransitionInfo const &)
 
void setupResolvers (Principal &principal)
 
 WorkerManager (std::shared_ptr< ActivityRegistry > actReg, ExceptionToActionTable const &actions)
 
 WorkerManager (std::shared_ptr< ModuleRegistry > modReg, std::shared_ptr< ActivityRegistry > actReg, ExceptionToActionTable const &actions)
 
 WorkerManager (WorkerManager &&)=default
 

Private Attributes

ExceptionToActionTable const * actionTable_
 
AllWorkers allWorkers_
 
void const * lastSetupEventPrincipal_
 
UnscheduledCallProducer unscheduled_
 
WorkerRegistry workerReg_
 

Detailed Description

Definition at line 35 of file WorkerManager.h.

Member Typedef Documentation

◆ AllWorkers

typedef std::vector<Worker*> edm::WorkerManager::AllWorkers

Definition at line 37 of file WorkerManager.h.

Constructor & Destructor Documentation

◆ WorkerManager() [1/3]

edm::WorkerManager::WorkerManager ( std::shared_ptr< ActivityRegistry actReg,
ExceptionToActionTable const &  actions 
)

Definition at line 17 of file WorkerManager.cc.

18  : workerReg_(areg),
20  allWorkers_(),
21  unscheduled_(*areg),
22  lastSetupEventPrincipal_(nullptr) {} // WorkerManager::WorkerManager

◆ WorkerManager() [2/3]

edm::WorkerManager::WorkerManager ( WorkerManager &&  )
default

◆ WorkerManager() [3/3]

edm::WorkerManager::WorkerManager ( std::shared_ptr< ModuleRegistry modReg,
std::shared_ptr< ActivityRegistry actReg,
ExceptionToActionTable const &  actions 
)

Definition at line 24 of file WorkerManager.cc.

27  : workerReg_(areg, modReg),
29  allWorkers_(),
30  unscheduled_(*areg),
31  lastSetupEventPrincipal_(nullptr) {} // WorkerManager::WorkerManager

Member Function Documentation

◆ actionTable()

ExceptionToActionTable const& edm::WorkerManager::actionTable ( ) const
inline

Definition at line 86 of file WorkerManager.h.

86 { return *actionTable_; }

References actionTable_.

Referenced by edm::StreamSchedule::actionTable().

◆ addToAllWorkers()

void edm::WorkerManager::addToAllWorkers ( Worker w)

Definition at line 133 of file WorkerManager.cc.

133  {
134  if (!search_all(allWorkers_, w)) {
135  allWorkers_.push_back(w);
136  }
137  }

References allWorkers_, edm::search_all(), and w.

Referenced by edm::StreamSchedule::addToAllWorkers(), and addToUnscheduledWorkers().

◆ addToUnscheduledWorkers()

void edm::WorkerManager::addToUnscheduledWorkers ( ParameterSet pset,
ProductRegistry preg,
PreallocationConfiguration const *  prealloc,
std::shared_ptr< ProcessConfiguration processConfiguration,
std::string  label,
std::set< std::string > &  unscheduledLabels,
std::vector< std::string > &  shouldBeUsedLabels 
)

Definition at line 52 of file WorkerManager.cc.

58  {
59  //Need to
60  // 1) create worker
61  // 2) if it is a WorkerT<EDProducer>, add it to our list
62  auto modType = pset.getParameter<std::string>("@module_edm_type");
63  if (modType == kProducerType || modType == kFilterType) {
64  Worker* newWorker = getWorker(pset, preg, prealloc, processConfiguration, label);
65  assert(newWorker->moduleType() == Worker::kProducer || newWorker->moduleType() == Worker::kFilter);
66  unscheduledLabels.insert(label);
67  unscheduled_.addWorker(newWorker);
68  //add to list so it gets reset each new event
69  addToAllWorkers(newWorker);
70  } else {
71  shouldBeUsedLabels.push_back(label);
72  }
73  }

References addToAllWorkers(), edm::UnscheduledCallProducer::addWorker(), cms::cuda::assert(), getWorker(), edm::Worker::kFilter, kFilterType(), edm::Worker::kProducer, kProducerType(), label, edm::Worker::moduleType(), muonDTDigis_cfi::pset, AlCaHLTBitMon_QueryRunRegistry::string, and unscheduled_.

Referenced by edm::SecondaryEventProvider::SecondaryEventProvider(), and edm::StreamSchedule::StreamSchedule().

◆ allWorkers()

AllWorkers const& edm::WorkerManager::allWorkers ( ) const
inline

Definition at line 82 of file WorkerManager.h.

82 { return allWorkers_; }

References allWorkers_.

Referenced by edm::StreamSchedule::allWorkers().

◆ beginJob()

void edm::WorkerManager::beginJob ( ProductRegistry const &  iRegistry,
eventsetup::ESRecordsToProxyIndices const &  iESIndices 
)

Definition at line 91 of file WorkerManager.cc.

92  {
93  auto const processBlockLookup = iRegistry.productLookup(InProcess);
94  auto const runLookup = iRegistry.productLookup(InRun);
95  auto const lumiLookup = iRegistry.productLookup(InLumi);
96  auto const eventLookup = iRegistry.productLookup(InEvent);
97  if (!allWorkers_.empty()) {
98  auto const& processName = allWorkers_[0]->description()->processName();
99  auto processBlockModuleToIndicies = processBlockLookup->indiciesForModulesInProcess(processName);
100  auto runModuleToIndicies = runLookup->indiciesForModulesInProcess(processName);
101  auto lumiModuleToIndicies = lumiLookup->indiciesForModulesInProcess(processName);
102  auto eventModuleToIndicies = eventLookup->indiciesForModulesInProcess(processName);
103  for (auto& worker : allWorkers_) {
104  worker->updateLookup(InProcess, *processBlockLookup);
105  worker->updateLookup(InRun, *runLookup);
106  worker->updateLookup(InLumi, *lumiLookup);
107  worker->updateLookup(InEvent, *eventLookup);
108  worker->updateLookup(iESIndices);
109  worker->resolvePutIndicies(InProcess, processBlockModuleToIndicies);
110  worker->resolvePutIndicies(InRun, runModuleToIndicies);
111  worker->resolvePutIndicies(InLumi, lumiModuleToIndicies);
112  worker->resolvePutIndicies(InEvent, eventModuleToIndicies);
113  }
114 
115  for_all(allWorkers_, std::bind(&Worker::beginJob, std::placeholders::_1));
116  }
117  }

References allWorkers_, edm::Worker::beginJob(), edm::for_all(), edm::InEvent, edm::InLumi, edm::InProcess, edm::InRun, SimL1EmulatorRepack_CalouGT_cff::processName, and edm::ProductRegistry::productLookup().

Referenced by edm::SecondaryEventProvider::beginJob().

◆ beginStream()

void edm::WorkerManager::beginStream ( StreamID  iID,
StreamContext streamContext 
)

Definition at line 119 of file WorkerManager.cc.

119  {
120  for (auto& worker : allWorkers_) {
121  worker->beginStream(iID, streamContext);
122  }
123  }

References allWorkers_.

Referenced by edm::SecondaryEventProvider::beginStream(), and edm::StreamSchedule::beginStream().

◆ deleteModuleIfExists()

void edm::WorkerManager::deleteModuleIfExists ( std::string const &  moduleLabel)

Definition at line 33 of file WorkerManager.cc.

33  {
34  auto worker = workerReg_.get(moduleLabel);
35  if (worker != nullptr) {
36  auto eraseBeg = std::remove(allWorkers_.begin(), allWorkers_.end(), worker);
37  allWorkers_.erase(eraseBeg, allWorkers_.end());
38  unscheduled_.removeWorker(worker);
40  }
41  }

References allWorkers_, edm::WorkerRegistry::deleteModule(), edm::WorkerRegistry::get(), HerwigMaxPtPartonFilter_cfi::moduleLabel, MatrixUtil::remove(), edm::UnscheduledCallProducer::removeWorker(), unscheduled_, and workerReg_.

Referenced by edm::StreamSchedule::deleteModule().

◆ endJob() [1/2]

void edm::WorkerManager::endJob ( void  )

Definition at line 75 of file WorkerManager.cc.

75  {
76  for (auto& worker : allWorkers_) {
77  worker->endJob();
78  }
79  }

References allWorkers_.

Referenced by edm::SecondaryEventProvider::endJob().

◆ endJob() [2/2]

void edm::WorkerManager::endJob ( ExceptionCollector collector)

Definition at line 81 of file WorkerManager.cc.

81  {
82  for (auto& worker : allWorkers_) {
83  try {
84  convertException::wrap([&]() { worker->endJob(); });
85  } catch (cms::Exception const& ex) {
86  collector.addException(ex);
87  }
88  }
89  }

References edm::ExceptionCollector::addException(), allWorkers_, and edm::convertException::wrap().

◆ endStream()

void edm::WorkerManager::endStream ( StreamID  iID,
StreamContext streamContext 
)

Definition at line 125 of file WorkerManager.cc.

125  {
126  for (auto& worker : allWorkers_) {
127  worker->endStream(iID, streamContext);
128  }
129  }

References allWorkers_.

Referenced by edm::SecondaryEventProvider::endStream(), and edm::StreamSchedule::endStream().

◆ getWorker()

Worker * edm::WorkerManager::getWorker ( ParameterSet pset,
ProductRegistry preg,
PreallocationConfiguration const *  prealloc,
std::shared_ptr< ProcessConfiguration const >  processConfiguration,
std::string const &  label 
)

Definition at line 43 of file WorkerManager.cc.

47  {
48  WorkerParams params(&pset, preg, prealloc, processConfiguration, *actionTable_);
50  }

References actionTable_, edm::WorkerRegistry::getWorker(), label, CalibrationSummaryClient_cfi::params, muonDTDigis_cfi::pset, and workerReg_.

Referenced by addToUnscheduledWorkers(), and edm::StreamSchedule::fillWorkers().

◆ processAccumulatorsAsync()

template<typename T >
void edm::WorkerManager::processAccumulatorsAsync ( WaitingTaskHolder  task,
typename T::TransitionInfoType const &  info,
ServiceToken const &  token,
StreamID  streamID,
ParentContext const &  parentContext,
typename T::Context const *  context 
)

◆ processOneOccurrenceAsync()

template<typename T , typename U >
void edm::WorkerManager::processOneOccurrenceAsync ( WaitingTaskHolder  task,
typename T::TransitionInfoType &  info,
ServiceToken const &  token,
StreamID  streamID,
typename T::Context const *  topContext,
U const *  context 
)

Definition at line 105 of file WorkerManager.h.

110  {
111  //make sure the unscheduled items see this run or lumi transition
112  unscheduled_.runNowAsync<T, U>(std::move(task), info, token, streamID, topContext, context);
113  }

References info(), eostools::move(), edm::UnscheduledCallProducer::runNowAsync(), TrackValidation_cff::task, unpackBuffers-CaloStage2::token, mitigatedMETSequence_cff::U, and unscheduled_.

Referenced by edm::StreamSchedule::processOneStreamAsync().

◆ resetAll()

void edm::WorkerManager::resetAll ( )

Definition at line 131 of file WorkerManager.cc.

131 { for_all(allWorkers_, std::bind(&Worker::reset, std::placeholders::_1)); }

References allWorkers_, edm::for_all(), and edm::Worker::reset().

Referenced by edm::StreamSchedule::processOneStreamAsync(), and setupResolvers().

◆ setupOnDemandSystem()

void edm::WorkerManager::setupOnDemandSystem ( EventTransitionInfo const &  info)

◆ setupResolvers()

void edm::WorkerManager::setupResolvers ( Principal principal)

Member Data Documentation

◆ actionTable_

ExceptionToActionTable const* edm::WorkerManager::actionTable_
private

Definition at line 98 of file WorkerManager.h.

Referenced by actionTable(), and getWorker().

◆ allWorkers_

AllWorkers edm::WorkerManager::allWorkers_
private

◆ lastSetupEventPrincipal_

void const* edm::WorkerManager::lastSetupEventPrincipal_
private

Definition at line 101 of file WorkerManager.h.

Referenced by setupResolvers().

◆ unscheduled_

UnscheduledCallProducer edm::WorkerManager::unscheduled_
private

◆ workerReg_

WorkerRegistry edm::WorkerManager::workerReg_
private

Definition at line 97 of file WorkerManager.h.

Referenced by deleteModuleIfExists(), and getWorker().

kProducerType
static const std::string kProducerType("EDProducer")
edm::WorkerManager::addToAllWorkers
void addToAllWorkers(Worker *w)
Definition: WorkerManager.cc:133
CalibrationSummaryClient_cfi.params
params
Definition: CalibrationSummaryClient_cfi.py:14
edm::WorkerManager::resetAll
void resetAll()
Definition: WorkerManager.cc:131
edm::Worker::beginJob
void beginJob()
Definition: Worker.cc:347
kFilterType
static const std::string kFilterType("EDFilter")
edm::UnscheduledCallProducer::runAccumulatorsAsync
void runAccumulatorsAsync(WaitingTaskHolder task, typename T::TransitionInfoType const &info, ServiceToken const &token, StreamID streamID, ParentContext const &parentContext, typename T::Context const *context)
Definition: UnscheduledCallProducer.h:92
cms::cuda::assert
assert(be >=bs)
info
static const TGPicture * info(bool iBackgroundIsBlack)
Definition: FWCollectionSummaryWidget.cc:153
edm::Worker::kProducer
Definition: Worker.h:93
edm::InRun
Definition: BranchType.h:11
edm::Worker::reset
void reset()
Definition: Worker.h:176
edm::for_all
Func for_all(ForwardSequence &s, Func f)
wrapper for std::for_each
Definition: Algorithms.h:14
config
Definition: config.py:1
edm::InProcess
Definition: BranchType.h:11
edm::WorkerRegistry::getWorker
Worker * getWorker(WorkerParams const &p, std::string const &moduleLabel)
Retrieve the particular instance of the worker.
Definition: WorkerRegistry.cc:28
edm::Worker::kFilter
Definition: Worker.h:93
w
const double w
Definition: UKUtility.cc:23
submitPVResolutionJobs.config
config
parse the configuration file
Definition: submitPVResolutionJobs.py:281
edm::convertException::wrap
auto wrap(F iFunc) -> decltype(iFunc())
Definition: ConvertException.h:19
edm::WorkerManager::unscheduled_
UnscheduledCallProducer unscheduled_
Definition: WorkerManager.h:100
TrackValidation_cff.task
task
Definition: TrackValidation_cff.py:252
edm::WorkerRegistry::get
Worker const * get(std::string const &moduleLabel) const
Definition: WorkerRegistry.cc:48
edm::InEvent
Definition: BranchType.h:11
edm::UnscheduledCallProducer::setEventTransitionInfo
void setEventTransitionInfo(EventTransitionInfo const &info)
Definition: UnscheduledCallProducer.h:62
edm::WorkerManager::actionTable_
ExceptionToActionTable const * actionTable_
Definition: WorkerManager.h:98
edm::WorkerManager::workerReg_
WorkerRegistry workerReg_
Definition: WorkerManager.h:97
mitigatedMETSequence_cff.U
U
Definition: mitigatedMETSequence_cff.py:36
AlCaHLTBitMon_QueryRunRegistry.string
string
Definition: AlCaHLTBitMon_QueryRunRegistry.py:256
edm::InLumi
Definition: BranchType.h:11
edm::UnscheduledCallProducer::addWorker
void addWorker(Worker *aWorker)
Definition: UnscheduledCallProducer.h:47
edm::WorkerManager::allWorkers_
AllWorkers allWorkers_
Definition: WorkerManager.h:99
edm::UnscheduledCallProducer::removeWorker
void removeWorker(Worker const *worker)
Definition: UnscheduledCallProducer.h:55
edm::UnscheduledCallProducer::runNowAsync
void runNowAsync(WaitingTaskHolder task, typename T::TransitionInfoType const &info, ServiceToken const &token, StreamID streamID, typename T::Context const *topContext, U const *context) const
Definition: UnscheduledCallProducer.h:70
SimL1EmulatorRepack_CalouGT_cff.processName
processName
Definition: SimL1EmulatorRepack_CalouGT_cff.py:17
eostools.move
def move(src, dest)
Definition: eostools.py:511
edm::UnscheduledCallProducer::auxiliary
UnscheduledAuxiliary const & auxiliary() const
Definition: UnscheduledCallProducer.h:64
T
long double T
Definition: Basic3DVectorLD.h:48
edm::WorkerRegistry::deleteModule
void deleteModule(std::string const &moduleLabel)
Deletes the module of the Worker, but the Worker continues to exist.
Definition: WorkerRegistry.cc:56
UnscheduledConfigurator
MatrixUtil.remove
def remove(d, key, TELL=False)
Definition: MatrixUtil.py:219
edm::WorkerManager::getWorker
Worker * getWorker(ParameterSet &pset, ProductRegistry &preg, PreallocationConfiguration const *prealloc, std::shared_ptr< ProcessConfiguration const > processConfiguration, std::string const &label)
Definition: WorkerManager.cc:43
actions
roAction_t actions[nactions]
Definition: GenABIO.cc:181
edm::search_all
bool search_all(ForwardSequence const &s, Datum const &d)
Definition: Algorithms.h:36
cms::Exception
Definition: Exception.h:70
HerwigMaxPtPartonFilter_cfi.moduleLabel
moduleLabel
Definition: HerwigMaxPtPartonFilter_cfi.py:4
SiStripBadComponentsDQMServiceTemplate_cfg.ep
ep
Definition: SiStripBadComponentsDQMServiceTemplate_cfg.py:86
label
const char * label
Definition: PFTauDecayModeTools.cc:11
muonDTDigis_cfi.pset
pset
Definition: muonDTDigis_cfi.py:27
unpackBuffers-CaloStage2.token
token
Definition: unpackBuffers-CaloStage2.py:316
edm::WorkerManager::lastSetupEventPrincipal_
void const * lastSetupEventPrincipal_
Definition: WorkerManager.h:101