Skip to content
Snippets Groups Projects

[RTADPA BW Tests] When evaluating nevts in input DIGIs should use 'ROOT' not 'MDF'

Merged Luke Grazette requested to merge lugrazet-BW-hlt1-inputnevts-isdigi into master
All threads resolved!
2 files
+ 10
6
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -18,7 +18,7 @@ from GaudiConf import IOHelper
from PRConfig.bandwidth_helpers import FileNameHelper, parse_yaml
def _extract_evt_numbers(ifiles, evtmax):
def _extract_evt_numbers(ifiles, evtmax, isdigi):
"""
Extract all event numbers saved to a single MDF.
"""
@@ -36,7 +36,8 @@ def _extract_evt_numbers(ifiles, evtmax):
unpack_rawevent(bank_types=['ODIN'], configurables=True),
createODIN(ODIN='myODIN')
])
IOHelper("MDF").inputFiles(ifiles)
IOHelper("ROOT").inputFiles(ifiles) if isdigi else IOHelper(
"MDF").inputFiles(ifiles)
appMgr = GP.AppMgr()
evt = appMgr.evtsvc()
@@ -57,11 +58,12 @@ def _extract_evt_numbers(ifiles, evtmax):
return event_numbers
def input_nevts(input_files, evtmax):
def input_nevts(input_files, evtmax, isdigi):
"""
For an arbitrary set of input_files, returns: min(total number of events, evtmax).
"""
event_numbers = _extract_evt_numbers(ifiles=input_files, evtmax=evtmax)
event_numbers = _extract_evt_numbers(
ifiles=input_files, evtmax=evtmax, isdigi=isdigi)
n_evts = len(event_numbers)
print(f"Found {n_evts} event numbers in input_files")
@@ -97,7 +99,8 @@ def main():
ifile = fname_helper.mdf_fname_for_reading(args.stream_config, args.stream)
evtmax = int(parse_yaml(fname_helper.input_nevts_json())['n_evts'])
event_numbers = _extract_evt_numbers(ifiles=[ifile], evtmax=evtmax)
event_numbers = _extract_evt_numbers(
ifiles=[ifile], evtmax=evtmax, isdigi=False)
ofile = fname_helper.event_no_fname(args.stream_config, args.stream)
with open(ofile, 'w') as f:
Loading