From f19f6f2d544783caff02c87a1b5ca945660f71b2 Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wdconinc@gmail.com>
Date: Mon, 17 Oct 2022 17:04:26 -0500
Subject: [PATCH] feat: automatically interpret hepmc3 input files
---
DDG4/python/DDSim/DD4hepSimulation.py | 9 ++++++++-
1 file changed, 8 insertions(+), 1 deletion(-)
diff --git a/DDG4/python/DDSim/DD4hepSimulation.py b/DDG4/python/DDSim/DD4hepSimulation.py
index c8397eae2..ac543d7e6 100644
--- a/DDG4/python/DDSim/DD4hepSimulation.py
+++ b/DDG4/python/DDSim/DD4hepSimulation.py
@@ -39,7 +39,7 @@ try:
except ImportError:
ARGCOMPLETEENABLED = False
-POSSIBLEINPUTFILES = (".stdhep", ".slcio", ".HEPEvt", ".hepevt", ".hepmc", ".pairs")
+POSSIBLEINPUTFILES = (".stdhep", ".slcio", ".HEPEvt", ".hepevt", ".hepmc", ".hepmc3", ".pairs")
class DD4hepSimulation(object):
@@ -411,6 +411,13 @@ class DD4hepSimulation(object):
else:
gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/hepmc%d" % index)
gen.Input = "Geant4EventReaderHepMC|" + inputFile
+ elif inputFile.endswith(".hepmc3"):
+ if self.hepmc3.useHepMC3:
+ gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/hepmc%d" % index)
+ gen.Parameters = self.hepmc3.getParameters()
+ gen.Input = "HEPMC3FileReader|" + inputFile
+ else:
+ raise RuntimeError("HepMC3 input file support not enabled")
elif inputFile.endswith(".pairs"):
gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/GuineaPig%d" % index)
gen.Input = "Geant4EventReaderGuineaPig|" + inputFile
--
GitLab