HepMCCopy.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ServiceRegistry/interface/Service.h"

#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include "RecoParticleFlow/Configuration/plugins/HepMCCopy.h"
#include "HepMC/GenEvent.h"

HepMCCopy::HepMCCopy(edm::ParameterSet const & p)  
{    
  // This producer produces a HepMCProduct, a copy of the original one
  produces<edm::HepMCProduct>();
}

void HepMCCopy::produce(edm::Event & iEvent, const edm::EventSetup & es)
{
  
  edm::Handle<edm::HepMCProduct> theHepMCProduct;
  bool source = iEvent.getByLabel("generatorSmeared",theHepMCProduct);
  if ( !source ) { 
    auto pu_product = std::make_unique<edm::HepMCProduct>();  
    iEvent.put(std::move(pu_product));
  } else { 
    auto pu_product = std::make_unique<edm::HepMCProduct>(*theHepMCProduct);  
    iEvent.put(std::move(pu_product));
  }

}

DEFINE_FWK_MODULE(HepMCCopy);