LaserAlignmentProducer.cc

Go to the documentation of this file.

// system include files
#include "FWCore/Framework/interface/Event.h"

// user include files
#include "Alignment/LaserAlignmentSimulation/plugins/LaserAlignmentProducer.h"
#include "FWCore/Framework/interface/MakerMacros.h"

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

//
// constructors and destructor
//
LaserAlignmentProducer::LaserAlignmentProducer(const edm::ParameterSet &) : EDProducer(), theEvent(nullptr) {
  // register your products
  produces<edm::HepMCProduct>("unsmeared");

  // now do what ever other initialization is needed
}

LaserAlignmentProducer::~LaserAlignmentProducer() {
  // no need to cleanup theEvent since it's done in HepMCProduct
}

// ------------ method called to produce the event  ------------
void LaserAlignmentProducer::produce(edm::Event &iEvent, const edm::EventSetup &) {
  // create the event
  theEvent = new HepMC::GenEvent();

  // create a primary vertex
  HepMC::GenVertex *theVtx = new HepMC::GenVertex(HepMC::FourVector(0., 0., 0.));

  // add a particle to the vertex; this is needed to avoid crashes in
  // OscarProducer. Use a electron neutrino, with zero energy and mass
  HepMC::GenParticle *theParticle = new HepMC::GenParticle(HepMC::FourVector(0., 0., 0., 0.), 12, 1);

  theVtx->add_particle_out(theParticle);

  // add the vertex to the event
  theEvent->add_vertex(theVtx);

  // set the event number
  theEvent->set_event_number(iEvent.id().event());
  // set the signal process id
  theEvent->set_signal_process_id(20);

  // create an empty output collection
  auto theOutput = std::make_unique<edm::HepMCProduct>();
  theOutput->addHepMCData(theEvent);

  // put the output to the event
  iEvent.put(std::move(theOutput));
}

// define this as a plug-in

DEFINE_FWK_MODULE(LaserAlignmentProducer);