DDSim: allow EDM4hep File reader

AIDASoft · Dec 13, 2024 · 0969e1f · 0969e1f
1 parent cb5cf31
commit 0969e1f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/DDG4/python/DDSim/DD4hepSimulation.py b/DDG4/python/DDSim/DD4hepSimulation.py
@@ -49,6 +49,7 @@
     ".stdhep", ".slcio", ".HEPEvt", ".hepevt",
     ".pairs",
     ".hepmc",
+    ".root",
     ] + HEPMC3_SUPPORTED_EXTENSIONS
 
 
@@ -445,6 +446,9 @@ def run(self):
         gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/GuineaPig%d" % index)
         gen.Input = "Geant4EventReaderGuineaPig|" + inputFile
         gen.Parameters = self.guineapig.getParameters()
+      elif inputFile.endswith(".root"):
+        gen = DDG4.GeneratorAction(kernel, "Geant4InputAction/EDM4hep%d" % index)
+        gen.Input = "EDM4hepFileReader|" + inputFile
       else:
         # this should never happen because we already check at the top, but in case of some LogicError...
         raise RuntimeError("Unknown input file type: %s" % inputFile)