Adding endpoint vertex of MCParticle

HEP-FCC · Feb 20, 2024 · 0854134 · 0854134
1 parent ba03ab5
commit 0854134
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/k4Gen/src/components/HepMCToEDMConverter.cpp b/k4Gen/src/components/HepMCToEDMConverter.cpp
@@ -33,6 +33,12 @@ edm4hep::MutableMCParticle HepMCToEDMConverter::convert(std::shared_ptr<const He
     edm_particle.setVertex( {float(pos.x()), float(pos.y()), float(pos.z())} );
   }
 
+  auto endVtx = hepmcParticle->end_vertex();
+  if ( endVtx != nullptr ) {
+    auto& pos = endVtx->position();
+    edm_particle.setEndpoint( {pos.x(), pos.y(), pos.z()} );
+  }
+
   return edm_particle;
 }