tritrig_beam_job.py

Go to the documentation of this file.

"""!
@file tritrig_beam_job.py
 
Merge tritrig and beam events.
"""
from hpsmc.tools import LCIOMerge, ExtractEventsWithHitAtHodoEcal
 
 
inputs = list(job.input_files.values())
 
job.description = 'tritrig beam'
 
if 'event_interval' in job.params:
    event_interval = job.params['event_interval']
else:
    event_interval = 1
 
 
tritrig_file_name = 'tritrig_events.slcio'
 
 
beam_file_names = []
for i in range(len(inputs)):
    if inputs[i] != tritrig_file_name:
        beam_file_names.append(inputs[i])
 
 
tritrig_name = 'tritrig'
 
 
tritrig_beam_name = 'tritrig_beam'
 
 
filter_bunches = ExtractEventsWithHitAtHodoEcal(inputs=[tritrig_file_name],
                                                outputs=['%s_filt.slcio' % tritrig_name],
                                                event_interval=event_interval, num_hodo_hits=0)
 
 
slic_beam_cat = ExtractEventsWithHitAtHodoEcal(inputs=beam_file_names,
                                               outputs=['beam_cat.slcio'],
                                               ignore_job_params=['event_interval'],
                                               event_interval=0, num_hodo_hits=0)
 
 
merge = LCIOMerge(inputs=[filter_bunches.output_files()[0],
                          slic_beam_cat.outputs[0]],
                  outputs=['%s.slcio' % tritrig_beam_name],
                  ignore_job_params=['nevents'])
 
comps = [filter_bunches, slic_beam_cat, merge]
job.add(comps)