tritrig_beam_slic_to_reco_job.py

Go to the documentation of this file.

"""!
@file tritrig_beam_job.py
 
Merge tritrig and beam events, simulate signal events, and detector readout.
"""
import os
from hpsmc.tools import SLIC, JobManager, LCIOCount, LCIOMerge, ExtractEventsWithHitAtHodoEcal
 
 
inputs = list(job.input_files.values())
 
job.description = 'tritrig beam slic to reco'
 
if 'nevents' in job.params:
    nevents = job.params['nevents']
else:
    nevents = 10000
 
 
tritrig_file_name = 'tritrig_events.stdhep'
 
 
beam_file_names = []
beam_slic_file_names = []
for i in range(len(inputs)):
    if inputs[i] != tritrig_file_name:
        beam_file_names.append(inputs[i])
        filename, file_extension = os.path.splitext(inputs[i])
        beam_slic_file = filename + '.slcio'
        beam_slic_file_names.append(beam_slic_file)
 
 
tritrig_name = 'tritrig'
 
 
beam_name = 'beam'
 
 
tritrig_beam_name = 'tritrig_beam'
 
 
slic = SLIC(inputs=[tritrig_file_name],
            outputs=['%s.slcio' % tritrig_name])
 
 
filter_bunches = ExtractEventsWithHitAtHodoEcal(inputs=slic.output_files(),
                                                outputs=['%s_filt.slcio' % tritrig_name],
                                                event_interval=250, num_hodo_hits=0)
 
 
count_filter = LCIOCount(inputs=filter_bunches.output_files())
 
 
slic_beams = []
for i in range(len(beam_file_names)):
    slic_beams.append(SLIC(inputs=[beam_file_names[i]],
                      outputs=[beam_slic_file_names[i]],
                      nevents=nevents * 250,
                      ignore_job_params=['nevents'])
                      )
 
 
slic_beam_cat = ExtractEventsWithHitAtHodoEcal(inputs=beam_slic_file_names,
                                               outputs=['beam_cat.slcio'],
                                               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'])
 
 
count_merge = LCIOCount(inputs=merge.output_files())
 
 
readout = JobManager(steering='readout',
                     inputs=merge.output_files(),
                     outputs=['%s_readout.slcio' % tritrig_beam_name])
 
 
count_readout = LCIOCount(inputs=readout.output_files())
 
 
recon = JobManager(steering='recon',
                   inputs=readout.output_files(),
                   outputs=['%s_recon.slcio' % tritrig_beam_name])
 
 
count_recon = LCIOCount(inputs=recon.output_files())
 
comps = [slic, filter_bunches, count_filter]
for i in range(len(slic_beams)):
    comps.append(slic_beams[i])
comps.extend([slic_beam_cat, merge, count_merge, readout, count_readout, recon, count_recon])
job.add(comps)