IOMC/RandomEngine/plugins/RandomEngineStateProducer.cc - master - githubmirror/cmssw

IOMC/RandomEngine/plugins/RandomEngineStateProducer.cc

40 строк2 KB

Matti Kortelainen

Move RandomNumberGenerator service base class to FWCore/AbstractServices

28 мар 2025, 00:56

28 мар 2025, 00:56c4a96e9


#include "RandomEngineStateProducer.h"

#include "FWCore/AbstractServices/interface/RandomNumberGenerator.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/LuminosityBlock.h"
#include "FWCore/ParameterSet/interface/ConfigurationDescriptions.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "SimDataFormats/RandomEngine/interface/RandomEngineStates.h"

#include <memory>

RandomEngineStateProducer::RandomEngineStateProducer(edm::ParameterSet const&) {
  produces<edm::RandomEngineStates, edm::Transition::BeginLuminosityBlock>("beginLumi");
  produces<edm::RandomEngineStates>();
}

RandomEngineStateProducer::~RandomEngineStateProducer() {}

void RandomEngineStateProducer::produce(edm::StreamID iID, edm::Event& ev, edm::EventSetup const&) const {
  edm::Service<edm::RandomNumberGenerator>().and_then([&ev](auto const& randomService) {
    auto states = std::make_unique<edm::RandomEngineStates>();
    states->setRandomEngineStates(randomService.getEventCache(ev.streamID()));
    ev.put(std::move(states));
  });
}

void RandomEngineStateProducer::globalBeginLuminosityBlockProduce(edm::LuminosityBlock& lb,
                                                                  edm::EventSetup const&) const {
  edm::Service<edm::RandomNumberGenerator>().and_then([&lb](auto const& randomService) {
    auto states = std::make_unique<edm::RandomEngineStates>();
    states->setRandomEngineStates(randomService.getLumiCache(lb.index()));
    lb.put(std::move(states), "beginLumi");
  });
}

void RandomEngineStateProducer::fillDescriptions(edm::ConfigurationDescriptions& descriptions) {
  edm::ParameterSetDescription desc;
  descriptions.add("randomEngineStateProducer", desc);
}