Skip to content

Commit 2730c6c

Browse files
committed
Added evtpool to template generator
1 parent 8cb3cbc commit 2730c6c

File tree

1 file changed

+17
-5
lines changed

1 file changed

+17
-5
lines changed

MC/bin/o2_hybrid_gen.py

Lines changed: 17 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -25,13 +25,13 @@ def main():
2525
args = parser.parse_args()
2626

2727
# Check if the mode is valid
28-
mode = "sequential"
29-
if args.mode not in ["sequential", "parallel"]:
30-
print(f"Mode {args.mode} not valid. Please use 'seq' or 'par'")
28+
valid_modes = ["sequential", "parallel"]
29+
mode = args.mode if args.mode in valid_modes else "sequential"
30+
if args.mode and args.mode not in valid_modes:
31+
print(f"Mode {args.mode} not valid. Please use 'sequential' or 'parallel'")
3132
print("Setting sequential mode as default")
3233
else:
33-
print(f"Running in {args.mode} mode")
34-
mode = args.mode
34+
print(f"Running in {mode} mode")
3535

3636
# put in a list all the elementes in the gen flag
3737
noConfGen = ["pythia8pp", "pythia8hf", "pythia8hi", "pythia8powheg"]
@@ -63,6 +63,18 @@ def main():
6363
"iniFile": ""
6464
}
6565
})
66+
elif gen == "evtpool":
67+
gens.append({
68+
'name': 'evtpool',
69+
'config': {
70+
"eventPoolPath": "/path/to/filename.root",
71+
"skipNonTrackable": True,
72+
"roundRobin": False,
73+
"randomize": True,
74+
"rngseed": 0,
75+
"randomphi": False,
76+
}
77+
})
6678
elif gen == "extkinO2":
6779
gens.append({
6880
'name': 'extkinO2',

0 commit comments

Comments
 (0)