Inheritance diagram for GenParticles2HepMCConverter:

Public Member Functions
void	beginRun (edm::Run const &iRun, edm::EventSetup const &) override

	GenParticles2HepMCConverter (const edm::ParameterSet &pset)

void	produce (edm::Event &event, const edm::EventSetup &eventSetup) override

	~GenParticles2HepMCConverter () override

Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default

	EDProducer (const EDProducer &)=delete

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

const EDProducer &	operator= (const EDProducer &)=delete

Private Member Functions
HepMC3::FourVector	FourVector (const reco::Candidate::Point &point)

HepMC3::FourVector	FourVector (const reco::Candidate::LorentzVector &lvec)

Private Attributes
const double	cmEnergy_

edm::EDGetTokenT< GenEventInfoProduct >	genEventInfoToken_

edm::EDGetTokenT< reco::CandidateView >	genParticlesToken_

edm::EDGetTokenT< GenRunInfoProduct >	genRunInfoToken_

edm::ESGetToken< HepPDT::ParticleDataTable, PDTRecord >	pTable_

HepMC3::GenCrossSectionPtr	xsec_

Additional Inherited Members
Public Types inherited from edm::stream::EDProducer<>
using	CacheTypes = CacheContexts< T... >

using	GlobalCache = typename CacheTypes::GlobalCache

using	HasAbility = AbilityChecker< T... >

using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache

using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache

using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >

using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache

using	RunCache = typename CacheTypes::RunCache

using	RunContext = RunContextT< RunCache, GlobalCache >

using	RunSummaryCache = typename CacheTypes::RunSummaryCache

Detailed Description

Definition at line 31 of file GenParticles2HepMCConverter.cc.

Constructor & Destructor Documentation

◆ GenParticles2HepMCConverter()

GenParticles2HepMCConverter::GenParticles2HepMCConverter ( const edm::ParameterSet & pset )

explicit

Definition at line 59 of file GenParticles2HepMCConverter.cc.

References genEventInfoToken_, genParticlesToken_, genRunInfoToken_, muonDTDigis_cfi::pset, and pTable_.

     : cmEnergy_(pset.getUntrackedParameter<double>("cmEnergy", 13000)) {
   genParticlesToken_ = consumes<reco::CandidateView>(pset.getParameter<edm::InputTag>("genParticles"));
   genEventInfoToken_ = consumes<GenEventInfoProduct>(pset.getParameter<edm::InputTag>("genEventInfo"));
   genRunInfoToken_ = consumes<GenRunInfoProduct, edm::InRun>(pset.getParameter<edm::InputTag>("genEventInfo"));
   pTable_ = esConsumes<HepPDT::ParticleDataTable, PDTRecord>();
 
   produces<edm::HepMC3Product>("unsmeared");
 }

◆ ~GenParticles2HepMCConverter()

GenParticles2HepMCConverter::~GenParticles2HepMCConverter ( )

inlineoverride

Definition at line 34 of file GenParticles2HepMCConverter.cc.

34 {}

Member Function Documentation

◆ beginRun()

void GenParticles2HepMCConverter::beginRun	(	edm::Run const &	iRun,
		edm::EventSetup const &
	)

override

Definition at line 70 of file GenParticles2HepMCConverter.cc.

References GenRunInfoProduct::XSec::error(), genRunInfoToken_, edm::Run::getByToken(), GenRunInfoProduct::internalXSec(), edm::HandleBase::isValid(), GenRunInfoProduct::XSec::value(), and xsec_.

                                                                                    {
   edm::Handle<GenRunInfoProduct> genRunInfoHandle;
   iRun.getByToken(genRunInfoToken_, genRunInfoHandle);
 
   xsec_ = make_shared<HepMC3::GenCrossSection>();
   if (genRunInfoHandle.isValid()) {
     xsec_->set_cross_section(genRunInfoHandle->internalXSec().value(), genRunInfoHandle->internalXSec().error());
   } else {
     // dummy cross section
     xsec_->set_cross_section(1., 0.);
   }
 }

◆ FourVector() [1/2]

HepMC3::FourVector GenParticles2HepMCConverter::FourVector ( const reco::Candidate::Point & point )

inlineprivate

Definition at line 49 of file GenParticles2HepMCConverter.cc.

References point.

Referenced by produce().

                                                                       {
     return HepMC3::FourVector(10 * point.x(), 10 * point.y(), 10 * point.z(), 0);
   };

◆ FourVector() [2/2]

HepMC3::FourVector GenParticles2HepMCConverter::FourVector ( const reco::Candidate::LorentzVector & lvec )

inlineprivate

Definition at line 53 of file GenParticles2HepMCConverter.cc.

References Matriplex::hypot(), and WZElectronSkims53X_cff::max.

                                                                              {
     // Avoid negative mass, set minimum m^2 = 0
     return HepMC3::FourVector(lvec.px(), lvec.py(), lvec.pz(), std::hypot(lvec.P(), std::max(0., lvec.mass())));
   };

◆ produce()

void GenParticles2HepMCConverter::produce	(	edm::Event &	event,
		const edm::EventSetup &	eventSetup
	)

override

Definition at line 83 of file GenParticles2HepMCConverter.cc.

References funct::abs(), GenEventInfoProduct::alphaQCD(), GenEventInfoProduct::alphaQED(), edm::View< T >::at(), cmEnergy_, options_cfi::eventSetup, FourVector(), genEventInfoToken_, genParticlesToken_, mps_fire::i, gen::PdfInfo::id, dqmiolumiharvest::j, EgHLTOffHistBins_cfi::mass, eostools::move(), create_idmaps::n, or, AlCaHLTBitMon_ParallelJobs::p, GenEventInfoProduct::pdf(), pTable_, GenEventInfoProduct::qScale(), gen::PdfInfo::scalePDF, GenEventInfoProduct::signalProcessID(), edm::View< T >::size(), bphysicsOniaDQM_cfi::vertex, reco::Candidate::vertex(), GenEventInfoProduct::weights(), gen::PdfInfo::x, gen::PdfInfo::xPDF, and xsec_.

                                                                                           {
   edm::Handle<reco::CandidateView> genParticlesHandle;
   event.getByToken(genParticlesToken_, genParticlesHandle);
 
   edm::Handle<GenEventInfoProduct> genEventInfoHandle;
   event.getByToken(genEventInfoToken_, genEventInfoHandle);
 
   auto const& pTableData = eventSetup.getData(pTable_);
 
   HepMC3::GenEvent hepmc_event;
   hepmc_event.set_event_number(event.id().event());
   hepmc_event.add_attribute("signal_process_id",
                             std::make_shared<HepMC3::IntAttribute>(genEventInfoHandle->signalProcessID()));
   hepmc_event.add_attribute("event_scale", std::make_shared<HepMC3::DoubleAttribute>(genEventInfoHandle->qScale()));
   hepmc_event.add_attribute("alphaQCD", std::make_shared<HepMC3::DoubleAttribute>(genEventInfoHandle->alphaQCD()));
   hepmc_event.add_attribute("alphaQED", std::make_shared<HepMC3::DoubleAttribute>(genEventInfoHandle->alphaQED()));
 
   hepmc_event.weights() = genEventInfoHandle->weights();
   // add dummy weight if necessary
   if (hepmc_event.weights().size() == 0) {
     hepmc_event.weights().push_back(1.);
   }
 
   // resize cross section to number of weights
   if (xsec_->xsecs().size() < hepmc_event.weights().size()) {
     xsec_->set_cross_section(std::vector<double>(hepmc_event.weights().size(), xsec_->xsec(0)),
                              std::vector<double>(hepmc_event.weights().size(), xsec_->xsec_err(0)));
   }
   hepmc_event.set_cross_section(xsec_);
 
   // Set PDF
   const gen::PdfInfo* pdf = genEventInfoHandle->pdf();
   if (pdf != nullptr) {
     const int pdf_id1 = pdf->id.first, pdf_id2 = pdf->id.second;
     const double pdf_x1 = pdf->x.first, pdf_x2 = pdf->x.second;
     const double pdf_scalePDF = pdf->scalePDF;
     const double pdf_xPDF1 = pdf->xPDF.first, pdf_xPDF2 = pdf->xPDF.second;
     HepMC3::GenPdfInfoPtr hepmc_pdfInfo = make_shared<HepMC3::GenPdfInfo>();
     hepmc_pdfInfo->set(pdf_id1, pdf_id2, pdf_x1, pdf_x2, pdf_scalePDF, pdf_xPDF1, pdf_xPDF2);
     hepmc_event.set_pdf_info(hepmc_pdfInfo);
   }
 
   // Prepare list of HepMC3::GenParticles
   std::map<const reco::Candidate*, HepMC3::GenParticlePtr> genCandToHepMCMap;
   HepMC3::GenParticlePtr hepmc_parton1, hepmc_parton2;
   std::vector<HepMC3::GenParticlePtr> hepmc_particles;
   const reco::Candidate *parton1 = nullptr, *parton2 = nullptr;
   for (unsigned int i = 0, n = genParticlesHandle->size(); i < n; ++i) {
     const reco::Candidate* p = &genParticlesHandle->at(i);
     HepMC3::GenParticlePtr hepmc_particle =
         std::make_shared<HepMC3::GenParticle>(FourVector(p->p4()), p->pdgId(), p->status());
 
     // Assign particle's generated mass from the standard particle data table
     double particleMass;
     if (pTableData.particle(p->pdgId()))
       particleMass = pTableData.particle(p->pdgId())->mass();
     else
       particleMass = p->mass();
 
     hepmc_particle->set_generated_mass(particleMass);
 
     hepmc_particles.push_back(hepmc_particle);
     genCandToHepMCMap[p] = hepmc_particle;
 
     // Find incident proton pair
     if (p->mother() == nullptr and std::abs(p->eta()) > 5 and std::abs(p->pz()) > 1000) {
       if (!parton1 and p->pz() > 0) {
         parton1 = p;
         hepmc_parton1 = hepmc_particle;
       } else if (!parton2 and p->pz() < 0) {
         parton2 = p;
         hepmc_parton2 = hepmc_particle;
       }
     }
   }
 
   HepMC3::GenVertexPtr vertex1;
   HepMC3::GenVertexPtr vertex2;
   if (parton1 == nullptr || parton2 == nullptr) {
     // Particle gun samples do not have incident partons. Put dummy incident particle and prod vertex
     // Note: leave parton1 and parton2 as nullptr since it is not used anymore after creating hepmc_parton1 and 2
     const reco::Candidate::LorentzVector nullP4(0, 0, 0, 0);
     const reco::Candidate::LorentzVector beamP4(0, 0, cmEnergy_ / 2, cmEnergy_ / 2);
     vertex1 = make_shared<HepMC3::GenVertex>(FourVector(nullP4));
     vertex2 = make_shared<HepMC3::GenVertex>(FourVector(nullP4));
     hepmc_parton1 = make_shared<HepMC3::GenParticle>(FourVector(+beamP4), 2212, 4);
     hepmc_parton2 = make_shared<HepMC3::GenParticle>(FourVector(-beamP4), 2212, 4);
   } else {
     // Put incident beam particles : proton -> parton vertex
     vertex1 = make_shared<HepMC3::GenVertex>(FourVector(parton1->vertex()));
     vertex2 = make_shared<HepMC3::GenVertex>(FourVector(parton2->vertex()));
   }
   hepmc_event.add_vertex(vertex1);
   hepmc_event.add_vertex(vertex2);
   vertex1->add_particle_in(hepmc_parton1);
   vertex2->add_particle_in(hepmc_parton2);
   //hepmc_event.set_beam_particles(hepmc_parton1, hepmc_parton2);
 
   // Prepare vertex list
   typedef std::map<const reco::Candidate*, HepMC3::GenVertexPtr> ParticleToVertexMap;
   ParticleToVertexMap particleToVertexMap;
   particleToVertexMap[parton1] = vertex1;
   particleToVertexMap[parton2] = vertex2;
   for (unsigned int i = 0, n = genParticlesHandle->size(); i < n; ++i) {
     const reco::Candidate* p = &genParticlesHandle->at(i);
     if (p == parton1 or p == parton2)
       continue;
 
     // Connect mother-daughters for the other cases
     for (unsigned int j = 0, nMothers = p->numberOfMothers(); j < nMothers; ++j) {
       // Mother-daughter hierarchy defines vertex
       const reco::Candidate* elder = p->mother(j)->daughter(0);
       HepMC3::GenVertexPtr vertex;
       if (particleToVertexMap.find(elder) == particleToVertexMap.end()) {
         vertex = make_shared<HepMC3::GenVertex>(FourVector(elder->vertex()));
         hepmc_event.add_vertex(vertex);
         particleToVertexMap[elder] = vertex;
       } else {
         vertex = particleToVertexMap[elder];
       }
 
       // Vertex is found. Now connect each other
       const reco::Candidate* mother = p->mother(j);
       vertex->add_particle_in(genCandToHepMCMap[mother]);
       vertex->add_particle_out(hepmc_particles[i]);
     }
   }
 
   // Finalize HepMC event record
   auto hepmc_product = std::make_unique<edm::HepMC3Product>(&hepmc_event);
   event.put(std::move(hepmc_product), "unsmeared");
 }