event display add flag to draw secondaries

key4hep · Sep 4, 2024 · 1e04ea3 · 1e04ea3
1 parent 84082e6
commit 1e04ea3
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/CLDConfig/event_display.py b/CLDConfig/event_display.py
@@ -40,6 +40,13 @@
     default=os.environ["K4GEO"] + "/FCCee/CLD/compact/CLD_o2_v07/CLD_o2_v07.xml"
 )
 
+parser.add_argument(
+    "--drawSecondaries",
+    action="store_true",
+    help="Enable drawing of secondary MCParticles",
+    default=False
+)
+
 reco_args = parser.parse_known_args()[0]
 
 algList = []
@@ -79,6 +86,7 @@
                           "DrawDetectorID": ["0"],
                           "DrawHelixForPFOs": ["0"],
                           "DrawHelixForTrack": ["0"],
+                          "DrawMCParticlesCreatedInSimulation": ["true" if reco_args.drawSecondaries else "false"],
                           "DrawInLayer": [
                               "VXDCollection", "0", "5", "1",
                               "SITCollection", "0", "5", "1",
@@ -197,7 +205,7 @@
                           "UseColorForHelixTracks": ["1"],
                           "UseTrackerExtentForLimitsOfHelix": ["true"],
                           "UsingParticleGun": ["false"],
-                          "WaitForKeyboard": ["1"]
+                          "WaitForKeyboard": ["1"],
                           }
 
 algList.append(MyCEDViewer)