Merge pull request #47604 from mkirsano/hepmc3togenparticles

Add reco::GenParticle collection production from HepMC3 event record

Author: cmsbuild

PhysicsTools/HepMCCandAlgos/BuildFile.xml

@@ -3,6 +3,7 @@
 <use name="DataFormats/HepMCCandidate"/>
 <use name="CommonTools/UtilAlgos"/>
 <use name="hepmc"/>
+<use name="hepmc3"/>
 <use name="eigen"/>
 <use name="CLHEP"/>
 <export>

PhysicsTools/HepMCCandAlgos/interface/MCTruthHelper.h

@@ -6,6 +6,10 @@
 #include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"
 #include "DataFormats/HepMCCandidate/interface/GenStatusFlags.h"
 
+#include "HepMC3/GenEvent.h"
+#include "HepMC3/GenParticle.h"
+#include "HepMC3/GenVertex.h"
+
 #include <iostream>
 #include <unordered_set>
 
@@ -143,36 +147,54 @@ class MCTruthHelper {
   //pdgid
   int pdgId(const HepMC::GenParticle &p) const;
 
+  //pdgid
+  int pdgId(const HepMC3::GenParticle &p) const;
+
   //abs(pdgid)
   int absPdgId(const reco::GenParticle &p) const;
 
   //abs(pdgid)
   int absPdgId(const HepMC::GenParticle &p) const;
 
+  //abs(pdgid)
+  int absPdgId(const HepMC3::GenParticle &p) const;
+
   //number of mothers
   unsigned int numberOfMothers(const reco::GenParticle &p) const;
 
   //number of mothers
   unsigned int numberOfMothers(const HepMC::GenParticle &p) const;
 
+  //number of mothers
+  unsigned int numberOfMothers(const HepMC3::GenParticle &p) const;
+
   //mother
   const reco::GenParticle *mother(const reco::GenParticle &p, unsigned int imoth = 0) const;
 
   //mother
   const HepMC::GenParticle *mother(const HepMC::GenParticle &p, unsigned int imoth = 0) const;
 
+  //mother
+  const HepMC3::GenParticle *mother(const HepMC3::GenParticle &p, unsigned int imoth = 0) const;
+
   //number of daughters
   unsigned int numberOfDaughters(const reco::GenParticle &p) const;
 
   //number of daughters
   unsigned int numberOfDaughters(const HepMC::GenParticle &p) const;
 
+  //number of daughters
+  unsigned int numberOfDaughters(const HepMC3::GenParticle &p) const;
+
   //ith daughter
   const reco::GenParticle *daughter(const reco::GenParticle &p, unsigned int idau) const;
 
   //ith daughter
   const HepMC::GenParticle *daughter(const HepMC::GenParticle &p, unsigned int idau) const;
 
+  //ith daughter
+  const HepMC3::GenParticle *daughter(const HepMC3::GenParticle &p, unsigned int idau) const;
+
   /////////////////////////////////////////////////////////////////////////////
   //Helper function to fill status flags
   void fillGenStatusFlags(const P &p, reco::GenStatusFlags &statusFlags) const;
@@ -576,6 +598,12 @@ int MCTruthHelper<P>::pdgId(const HepMC::GenParticle &p) const {
   return p.pdg_id();
 }
 
+//////////////////////////////////////////////////////////////
+template <typename P>
+int MCTruthHelper<P>::pdgId(const HepMC3::GenParticle &p) const {
+  return p.pid();
+}
+
 //////////////////////////////////////////////////////////////
 template <typename P>
 int MCTruthHelper<P>::absPdgId(const reco::GenParticle &p) const {
@@ -588,6 +616,12 @@ int MCTruthHelper<P>::absPdgId(const HepMC::GenParticle &p) const {
   return std::abs(p.pdg_id());
 }
 
+//////////////////////////////////////////////////////////////
+template <typename P>
+int MCTruthHelper<P>::absPdgId(const HepMC3::GenParticle &p) const {
+  return std::abs(p.pid());
+}
+
 /////////////////////////////////////////////////////////////////////////////
 template <typename P>
 unsigned int MCTruthHelper<P>::numberOfMothers(const reco::GenParticle &p) const {
@@ -600,6 +634,12 @@ unsigned int MCTruthHelper<P>::numberOfMothers(const HepMC::GenParticle &p) cons
   return p.production_vertex() ? p.production_vertex()->particles_in_size() : 0;
 }
 
+/////////////////////////////////////////////////////////////////////////////
+template <typename P>
+unsigned int MCTruthHelper<P>::numberOfMothers(const HepMC3::GenParticle &p) const {
+  return p.production_vertex() ? p.production_vertex()->particles_in_size() : 0;
+}
+
 /////////////////////////////////////////////////////////////////////////////
 template <typename P>
 const reco::GenParticle *MCTruthHelper<P>::mother(const reco::GenParticle &p, unsigned int imoth) const {
@@ -614,6 +654,20 @@ const HepMC::GenParticle *MCTruthHelper<P>::mother(const HepMC::GenParticle &p,
              : nullptr;
 }
 
+/////////////////////////////////////////////////////////////////////////////
+template <typename P>
+const HepMC3::GenParticle *MCTruthHelper<P>::mother(const HepMC3::GenParticle &p, unsigned int imoth) const {
+  if (numberOfMothers(p) > 0) {
+    unsigned int imothi = 0;
+    for (const HepMC3::ConstGenParticlePtr &mother : (p.production_vertex())->particles_in()) {
+      if (imothi == imoth)
+        return mother.get();
+      imothi++;
+    }
+  }
+  return nullptr;
+}
+
 /////////////////////////////////////////////////////////////////////////////
 template <typename P>
 unsigned int MCTruthHelper<P>::numberOfDaughters(const reco::GenParticle &p) const {
@@ -626,6 +680,12 @@ unsigned int MCTruthHelper<P>::numberOfDaughters(const HepMC::GenParticle &p) co
   return p.end_vertex() ? p.end_vertex()->particles_out_size() : 0;
 }
 
+/////////////////////////////////////////////////////////////////////////////
+template <typename P>
+unsigned int MCTruthHelper<P>::numberOfDaughters(const HepMC3::GenParticle &p) const {
+  return p.end_vertex() ? p.end_vertex()->particles_out_size() : 0;
+}
+
 /////////////////////////////////////////////////////////////////////////////
 template <typename P>
 const reco::GenParticle *MCTruthHelper<P>::daughter(const reco::GenParticle &p, unsigned int idau) const {
@@ -638,6 +698,20 @@ const HepMC::GenParticle *MCTruthHelper<P>::daughter(const HepMC::GenParticle &p
   return *(p.end_vertex()->particles_out_const_begin() + idau);
 }
 
+/////////////////////////////////////////////////////////////////////////////
+template <typename P>
+const HepMC3::GenParticle *MCTruthHelper<P>::daughter(const HepMC3::GenParticle &p, unsigned int idau) const {
+  if (numberOfDaughters(p) > 0) {
+    unsigned int idaui = 0;
+    for (const HepMC3::ConstGenParticlePtr &daughter : (p.end_vertex())->particles_out()) {
+      if (idaui == idau)
+        return daughter.get();
+      idaui++;
+    }
+  }
+  return nullptr;
+}
+
 /////////////////////////////////////////////////////////////////////////////
 template <typename P>
 void MCTruthHelper<P>::fillGenStatusFlags(const P &p, reco::GenStatusFlags &statusFlags) const {