tritrig_gen_to_slic_job.py

Go to the documentation of this file.

"""!
@file tritrig_gen_to_beam_coords_job.py
 
Simulate tritrig events, add mother particle information and rotate events into beam coordinates.
"""
from hpsmc.generators import MG5, StdHepConverter
from hpsmc.tools import Unzip, AddMotherFullTruth, BeamCoords, SLIC
 
job.description = 'Generate tritrig events and simulate passage through detector'
 
if 'nevents' in job.params:
    nevents = job.params['nevents']
else:
    nevents = 10000
 
 
mg = MG5(name='tritrig')
 
 
unzip = Unzip(inputs=mg.output_files())
 
 
cnv = StdHepConverter(inputs=mg.output_files(),
                      outputs=['tritrig.stdhep'])
 
 
mom = AddMotherFullTruth(inputs=[cnv.output_files()[0], unzip.output_files()[0]],
                         outputs=['tritrig_mom.stdhep'])
 
 
rot = BeamCoords(inputs=['tritrig_mom.stdhep'],
                 outputs=['tritrig_mom_rot.stdhep'])
 
 
job.ptag('gen', 'tritrig_mom_rot.stdhep')
 
 
slic = SLIC(inputs=['tritrig_mom_rot.stdhep'],
            outputs=['tritrig_mom_rot.slcio'],
            nevents=nevents + 1)
 
 
job.add([mg, unzip, cnv, mom, rot, slic])