PassThroughEvtVtxGenerator.cc

Go to the documentation of this file.

/*
*/

#include "IOMC/EventVertexGenerators/interface/PassThroughEvtVtxGenerator.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

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

#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/Utilities/interface/RandomNumberGenerator.h"

#include "FWCore/Utilities/interface/Exception.h"

#include "DataFormats/Provenance/interface/Provenance.h"
#include "FWCore/Utilities/interface/EDMException.h"

//#include "HepMC/GenEvent.h"
// #include "CLHEP/Vector/ThreeVector.h"
// #include "HepMC/SimpleVector.h"

using namespace edm;
using namespace CLHEP;
//using namespace HepMC;


PassThroughEvtVtxGenerator::PassThroughEvtVtxGenerator( const ParameterSet& pset )
  : BaseEvtVtxGenerator(pset)
{
   Service<RandomNumberGenerator> rng;
   if ( ! rng.isAvailable()) {
     throw cms::Exception("Configuration")
       << "The PassThroughEvtVtxGenerator requires the RandomNumberGeneratorService\n"
          "which is not present in the configuration file. \n"
          "You must add the service\n"
          "in the configuration file or remove the modules that require it.";
   }
   sourceToken=consumes<edm::HepMCProduct>(pset.getParameter<edm::InputTag>("src"));
}

PassThroughEvtVtxGenerator::~PassThroughEvtVtxGenerator()
{
}

HepMC::FourVector PassThroughEvtVtxGenerator::newVertex(CLHEP::HepRandomEngine*) const  {
  return HepMC::FourVector(0.,0.,0.,0);
}

void PassThroughEvtVtxGenerator::produce( Event& evt, const EventSetup& )
{
   edm::Service<edm::RandomNumberGenerator> rng;

   Handle<HepMCProduct> HepUnsmearedMCEvt ;

   evt.getByToken( sourceToken, HepUnsmearedMCEvt ) ;

   // Copy the HepMC::GenEvent
   HepMC::GenEvent* genevt = new HepMC::GenEvent(*HepUnsmearedMCEvt->GetEvent());
   std::unique_ptr<edm::HepMCProduct> HepMCEvt(new edm::HepMCProduct(genevt));

   evt.put(std::move(HepMCEvt)) ;

   return ;
}