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

bool	hasAbilityToProduceInBeginLumis () const final

bool	hasAbilityToProduceInBeginProcessBlocks () const final

bool	hasAbilityToProduceInBeginRuns () const final

bool	hasAbilityToProduceInEndLumis () const final

bool	hasAbilityToProduceInEndProcessBlocks () const final

bool	hasAbilityToProduceInEndRuns () const final

Private Member Functions
HepMC::FourVector	FourVector (const reco::Candidate::LorentzVector &lvec)

HepMC::FourVector	FourVector (const reco::Candidate::Point &point)

Private Attributes
const double	cmEnergy_

edm::EDGetTokenT< GenEventInfoProduct >	genEventInfoToken_

edm::EDGetTokenT< reco::CandidateView >	genParticlesToken_

edm::EDGetTokenT< GenRunInfoProduct >	genRunInfoToken_

edm::ESHandle< ParticleDataTable >	pTable_

std::vector< int >	signalParticlePdgIds_

HepMC::GenCrossSection	xsec_

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

typedef CacheTypes::GlobalCache	GlobalCache

typedef AbilityChecker< T... >	HasAbility

typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache

typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext

typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache

typedef CacheTypes::RunCache	RunCache

typedef RunContextT< RunCache, GlobalCache >	RunContext

typedef CacheTypes::RunSummaryCache	RunSummaryCache

Detailed Description

Definition at line 27 of file GenParticles2HepMCConverter.cc.

Constructor & Destructor Documentation

◆ GenParticles2HepMCConverter()

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

explicit

Definition at line 56 of file GenParticles2HepMCConverter.cc.

     : 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"));
   signalParticlePdgIds_ = pset.getParameter<std::vector<int>>("signalParticlePdgIds");
  
   produces<edm::HepMCProduct>("unsmeared");
 }

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

◆ ~GenParticles2HepMCConverter()

GenParticles2HepMCConverter::~GenParticles2HepMCConverter ( )

inlineoverride

Definition at line 30 of file GenParticles2HepMCConverter.cc.

30 {};

Member Function Documentation

◆ beginRun()

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

override

Definition at line 67 of file GenParticles2HepMCConverter.cc.

                                                                                    {
   edm::Handle<GenRunInfoProduct> genRunInfoHandle;
   iRun.getByToken(genRunInfoToken_, genRunInfoHandle);
  
   xsec_.set_cross_section(genRunInfoHandle->internalXSec().value(), genRunInfoHandle->internalXSec().error());
 }

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

◆ FourVector() [1/2]

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

inlineprivate

Definition at line 50 of file GenParticles2HepMCConverter.cc.

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

References SiStripPI::max.

◆ FourVector() [2/2]

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

inlineprivate

Definition at line 46 of file GenParticles2HepMCConverter.cc.

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

References point.

Referenced by produce().

◆ produce()

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

override

Definition at line 74 of file GenParticles2HepMCConverter.cc.

                                                                                           {
   edm::Handle<reco::CandidateView> genParticlesHandle;
   event.getByToken(genParticlesToken_, genParticlesHandle);
  
   edm::Handle<GenEventInfoProduct> genEventInfoHandle;
   event.getByToken(genEventInfoToken_, genEventInfoHandle);
  
   eventSetup.getData(pTable_);
  
   HepMC::GenEvent* hepmc_event = new HepMC::GenEvent();
   hepmc_event->set_event_number(event.id().event());
   hepmc_event->set_signal_process_id(genEventInfoHandle->signalProcessID());
   hepmc_event->set_event_scale(genEventInfoHandle->qScale());
   hepmc_event->set_alphaQED(genEventInfoHandle->alphaQED());
   hepmc_event->set_alphaQCD(genEventInfoHandle->alphaQCD());
  
   hepmc_event->weights() = genEventInfoHandle->weights();
  
   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;
     HepMC::PdfInfo hepmc_pdfInfo(pdf_id1, pdf_id2, pdf_x1, pdf_x2, pdf_scalePDF, pdf_xPDF1, pdf_xPDF2);
     hepmc_event->set_pdf_info(hepmc_pdfInfo);
   }
  
   // Prepare list of HepMC::GenParticles
   std::map<const reco::Candidate*, HepMC::GenParticle*> genCandToHepMCMap;
   HepMC::GenParticle *hepmc_parton1 = nullptr, *hepmc_parton2 = nullptr;
   std::vector<HepMC::GenParticle*> 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);
     HepMC::GenParticle* hepmc_particle = new HepMC::GenParticle(FourVector(p->p4()), p->pdgId(), p->status());
     hepmc_particle->suggest_barcode(i + 1);
  
     // Assign particle's generated mass from the standard particle data table
     double particleMass;
     if (pTable_->particle(p->pdgId()))
       particleMass = pTable_->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;
       }
     }
   }
  
   HepMC::GenVertex* vertex1 = nullptr;
   HepMC::GenVertex* vertex2 = nullptr;
   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 = new HepMC::GenVertex(FourVector(nullP4));
     vertex2 = new HepMC::GenVertex(FourVector(nullP4));
     hepmc_parton1 = new HepMC::GenParticle(FourVector(+beamP4), 2212, 4);
     hepmc_parton2 = new HepMC::GenParticle(FourVector(-beamP4), 2212, 4);
   } else {
     // Put incident beam particles : proton -> parton vertex
     vertex1 = new HepMC::GenVertex(FourVector(parton1->vertex()));
     vertex2 = new HepMC::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*, HepMC::GenVertex*> 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);
       HepMC::GenVertex* vertex;
       if (particleToVertexMap.find(elder) == particleToVertexMap.end()) {
         vertex = new HepMC::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
   bool hasSignalVertex = false;
   if (!signalParticlePdgIds_.empty()) {
     // Loop over all vertices to assign the signal vertex, decaying to a signal particle
     for (auto v = hepmc_event->vertices_begin(); v != hepmc_event->vertices_end(); ++v) {
       for (auto p = (*v)->particles_begin(HepMC::children); p != (*v)->particles_end(HepMC::children); ++p) {
         const int pdgId = (*p)->pdg_id();
         if (std::find(signalParticlePdgIds_.begin(), signalParticlePdgIds_.end(), pdgId) !=
             signalParticlePdgIds_.end()) {
           hepmc_event->set_signal_process_vertex(*v);
           hasSignalVertex = true;
           break;
         }
       }
       if (hasSignalVertex)
         break;
     }
   }
   // Set the default signal vertex if still not set
   if (!hasSignalVertex)
     hepmc_event->set_signal_process_vertex(*(vertex1->vertices_begin()));
  
   std::unique_ptr<edm::HepMCProduct> hepmc_product(new edm::HepMCProduct());
   hepmc_product->addHepMCData(hepmc_event);
   event.put(std::move(hepmc_product), "unsmeared");
 }