From 2730c6cdff9cad8b0266c3e105ff437f2d20bb8a Mon Sep 17 00:00:00 2001 From: Marco Giacalone Date: Tue, 7 Jan 2025 12:04:22 +0100 Subject: [PATCH] Added evtpool to template generator --- MC/bin/o2_hybrid_gen.py | 22 +++++++++++++++++----- 1 file changed, 17 insertions(+), 5 deletions(-) diff --git a/MC/bin/o2_hybrid_gen.py b/MC/bin/o2_hybrid_gen.py index 09d042b0c..0ecf2980f 100755 --- a/MC/bin/o2_hybrid_gen.py +++ b/MC/bin/o2_hybrid_gen.py @@ -25,13 +25,13 @@ def main(): args = parser.parse_args() # Check if the mode is valid - mode = "sequential" - if args.mode not in ["sequential", "parallel"]: - print(f"Mode {args.mode} not valid. Please use 'seq' or 'par'") + valid_modes = ["sequential", "parallel"] + mode = args.mode if args.mode in valid_modes else "sequential" + if args.mode and args.mode not in valid_modes: + print(f"Mode {args.mode} not valid. Please use 'sequential' or 'parallel'") print("Setting sequential mode as default") else: - print(f"Running in {args.mode} mode") - mode = args.mode + print(f"Running in {mode} mode") # put in a list all the elementes in the gen flag noConfGen = ["pythia8pp", "pythia8hf", "pythia8hi", "pythia8powheg"] @@ -63,6 +63,18 @@ def main(): "iniFile": "" } }) + elif gen == "evtpool": + gens.append({ + 'name': 'evtpool', + 'config': { + "eventPoolPath": "/path/to/filename.root", + "skipNonTrackable": True, + "roundRobin": False, + "randomize": True, + "rngseed": 0, + "randomphi": False, + } + }) elif gen == "extkinO2": gens.append({ 'name': 'extkinO2',