rad_gen_job.py

Go to the documentation of this file.

"""!
@file rad_gen_job.py
 
Simulation of radiative events.
"""
from hpsmc.generators import MG5, StdHepConverter
from hpsmc.tools import Unzip, BeamCoords, AddMotherFullTruth
 
job.description = 'rad generation'
 
 
inputs = list(job.input_files.values())
 
if 'nevents' in job.params:
    nevents = job.params['nevents']
else:
    nevents = 10000
 
 
mg = MG5(name='RAD', event_types=['unweighted'])
 
 
cnv = StdHepConverter(name="lhe_rad", inputs=mg.output_files())
 
 
unzip = Unzip(inputs=mg.output_files(), outputs=["rad.lhe"])
 
 
mom = AddMotherFullTruth(inputs=[cnv.output_files()[0], unzip.output_files()[0]], outputs=["rad_mom.stdhep"])
 
 
rot = BeamCoords(inputs=mom.output_files(), outputs=["rad_rot.stdhep"])
 
 
job.add([mg, cnv, unzip, mom, rot])