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].reset(workerPtr.release());
      return m_workerMap[moduleLabel].get(); 
    } 
    return (workerIt->second.get());
  }
}