Update script with the Type -> PDG rename

key4hep · Feb 23, 2024 · 2945663 · 2945663
1 parent 3e090d2
commit 2945663
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/scripts/createEDM4hepFile.py b/scripts/createEDM4hepFile.py
@@ -307,7 +307,10 @@
 
     particles = edm4hep.ReconstructedParticleCollection()
     particle = particles.create()
-    particle.setType(next(counter))
+    if args.use_pre1:
+        particle.setType(next(counter))
+    else:
+        particle.setPDG(next(counter))
     particle.setEnergy(next(counter))
     particle.setMomentum(edm4hep.Vector3f(next(counter), next(counter), next(counter)))
     particle.setReferencePoint(