signal_beam_merge_to_recon_job.py

Go to the documentation of this file.

"""!
@file signal_beam_merge_to_recon_job.py
 
Merging of signal and beam events, simulation of readout,
and reconstruction of events.
"""
from hpsmc.tools import ExtractEventsWithHitAtHodoEcal, JobManager, LCIOCount, LCIOMerge
 
job.description = 'signal-beam from merge to recon'
 
 
inputs = list(job.input_files.values())
 
 
signal_file_name = []
 
 
beam_file_name = []
 
for input in inputs:
    if "signal" in input:
        signal_file_name.append(input)
    if "beam" in input:
        beam_file_name.append(input)
 
 
if len(signal_file_name) == 0:
    raise Exception("Missing required input file(s) for signal")
if len(beam_file_name) == 0:
    raise Exception("Missing required input file(s) for beam")
 
 
signal_name = 'signal'
 
 
beam_name = 'beam'
 
 
signal_beam_name = 'signal-beam'
 
 
filter_events = ExtractEventsWithHitAtHodoEcal(inputs=signal_file_name,
                                               outputs=['%s_filt.slcio' % signal_name],
                                               event_interval=250, num_hodo_hits=1)
 
 
count_filter = LCIOCount(inputs=filter_events.output_files())
 
 
catenate_beam = ExtractEventsWithHitAtHodoEcal(inputs=beam_file_name,
                                               outputs=['%s_filt.slcio' % beam_name],
                                               event_interval=0, num_hodo_hits=0)
 
 
count_beam = LCIOCount(inputs=catenate_beam.output_files())
 
 
merge = LCIOMerge(inputs=[filter_events.output_files()[0],
                          catenate_beam.output_files()[0]],
                  outputs=['%s.slcio' % signal_beam_name],
                  ignore_job_params=['nevents'])
 
 
count_merge = LCIOCount(inputs=merge.output_files())
 
 
readout = JobManager(steering='readout',
                     inputs=merge.output_files(),
                     outputs=['%s_readout.slcio' % signal_beam_name])
 
 
count_readout = LCIOCount(inputs=readout.output_files())
 
 
recon = JobManager(steering='recon',
                   inputs=readout.output_files(),
                   outputs=['%s_recon.slcio' % signal_beam_name])
 
 
count_recon = LCIOCount(inputs=recon.output_files())
 
 
job.add([filter_events, count_filter, catenate_beam, count_beam, merge,
         count_merge, readout, count_readout, recon, count_recon])