WorkerRegistry.cc

Go to the documentation of this file.

 
#include <memory>
#include "FWCore/Framework/src/WorkerRegistry.h"
#include "FWCore/Framework/src/Worker.h"
#include "FWCore/Framework/src/ModuleHolder.h"
#include "FWCore/Framework/src/MakeModuleParams.h"
#include "FWCore/Framework/src/ModuleRegistry.h"
#include "FWCore/ServiceRegistry/interface/ActivityRegistry.h"
 
namespace edm {
 
  WorkerRegistry::WorkerRegistry(std::shared_ptr<ActivityRegistry> areg)
      : modRegistry_(new ModuleRegistry), m_workerMap(), actReg_(areg) {}
 
  WorkerRegistry::WorkerRegistry(std::shared_ptr<ActivityRegistry> areg, std::shared_ptr<ModuleRegistry> modReg)
      : modRegistry_(modReg), m_workerMap(), actReg_(areg) {}
 
  WorkerRegistry::~WorkerRegistry() {}
 
  void WorkerRegistry::clear() { m_workerMap.clear(); }
 
  Worker* WorkerRegistry::getWorker(WorkerParams const& p, std::string const& moduleLabel) {
    WorkerMap::iterator workerIt = m_workerMap.find(moduleLabel);
 
    // if the worker is not there, make it
    if (workerIt == m_workerMap.end()) {
      MakeModuleParams mmp(p.pset_, *p.reg_, p.preallocate_, p.processConfiguration_);
      auto modulePtr = modRegistry_->getModule(
          mmp, moduleLabel, actReg_->preModuleConstructionSignal_, actReg_->postModuleConstructionSignal_);
      auto workerPtr = modulePtr->makeWorker(p.actions_);
 
      workerPtr->setActivityRegistry(actReg_);
 
      // Transfer ownership of worker to the registry
      m_workerMap[moduleLabel] =
          std::shared_ptr<Worker>(workerPtr.release());  // propagate_const<T> has no reset() function
      return m_workerMap[moduleLabel].get();
    }
    return (workerIt->second.get());
  }
}  // namespace edm