signal_pulser_overlay_to_recon_job.py

Go to the documentation of this file.

"""!
@file signal_pulser_overlay_to_recon_job.py
 
Overlay MC with pulser data (evio or slcio) and run readout + reconstruction.
Automatically detects pulser input format from file name.
"""
 
from hpsmc.tools import ExtractEventsWithHitAtHodoEcal, EvioToLcio, JobManager, FilterBunches, LCIOCount, HPSTR
 
job.description = 'signal-pulse from overlay to recon'
 
 
inputs = list(job.input_files.values())
 
 
signal_file_name = []
 
 
pulser_file_name = []
 
for input in inputs:
    if "signal" in input:
        signal_file_name.append(input)
    if "pulser" in input:
        pulser_file_name.append(input)
 
 
if len(signal_file_name) == 0:
    raise Exception("Missing required input file(s) for signal")
if len(pulser_file_name) == 0:
    raise Exception("Missing required input file(s) for pulser data")
 
 
pulser_is_evio = 'evio' in pulser_file_name[0]
 
 
signal_name = 'signal'
 
 
pulser_name = 'pulser'
 
 
signal_pulser_name = 'signal_pulser'
 
 
filter_events = ExtractEventsWithHitAtHodoEcal(inputs=signal_file_name,
                                               outputs=['%s_filt.slcio' % signal_name],
                                               event_interval=0, num_hodo_hits=1)
 
 
count_filter = LCIOCount(inputs=filter_events.output_files())
 
 
evio_to_lcio = EvioToLcio(steering='evio_to_lcio', inputs=pulser_file_name, output=['%s.slcio' % pulser_name])
 
 
count_pulser = LCIOCount(inputs=evio_to_lcio.output_files())
 
 
overlay = JobManager(steering='overlay',
                     inputs=filter_events.output_files(),
                     overlay_file=pulser_file_name[0],
                     outputs=['%s.slcio' % signal_pulser_name])
 
 
space_overlay = FilterBunches(inputs=overlay.output_files(),
                              filter_no_cuts=True,
                              outputs=['%s_spaced.slcio' % signal_pulser_name],
                              filter_event_interval=250)
 
 
count_space_overlay = LCIOCount(inputs=space_overlay.output_files())
 
 
readout = JobManager(steering='readout',
                     inputs=space_overlay.output_files(),
                     outputs=['%s_readout.slcio' % signal_pulser_name])
 
 
count_readout = LCIOCount(inputs=readout.output_files())
 
 
recon = JobManager(steering='recon',
                   inputs=readout.output_files(),
                   outputs=['%s_recon.slcio' % signal_pulser_name])
 
 
count_recon = LCIOCount(inputs=recon.output_files())
 
 
cnv = HPSTR(inputs=recon.output_files(), cfg='cnv')
 
 
if pulser_is_evio:
    job.add([filter_events, count_filter, evio_to_lcio, overlay, space_overlay,
             count_space_overlay, readout, count_readout, recon, count_recon, cnv])
else:
    job.add([filter_events, count_filter, overlay, space_overlay,
             count_space_overlay, readout, count_readout, recon, count_recon, cnv])