ap_gen_to_slic_job.py

Go to the documentation of this file.

"""!
@file ap_gen_to_slic_job.py
 
Job script to generate A-prime events, convert to StdHep, apply transformations,
and resulting simulate signal events using SLIC.
 
There are options to generate prompt and displaced events.
You can use them by adjusting the StdHepConverter options as seen below.
"""
 
from hpsmc.generators import MG5, StdHepConverter
from hpsmc.tools import DisplaceUni, Unzip, BeamCoords, AddMotherFullTruth, SLIC
 
job.description = 'ap from generation to slic'
 
 
inputs = list(job.input_files.values())
 
if 'nevents' in job.params:
    nevents = job.params['nevents']
else:
    nevents = 10000
 
 
mg = MG5(name='ap', event_types=['unweighted'])
 
if 'ap_decay_dist' in job.params:
    ap_decay_dist = job.params['ap_decay_dist']
else:
    ap_decay_dist = "lhe_uniform"
 
 
unzip = Unzip(inputs=["ap_unweighted_events.lhe.gz"], outputs=["ap_unweighted_events.lhe"])
 
if ap_decay_dist == "lhe_uniform":
    
    cnv = StdHepConverter(name="lhe_uniform", inputs=["ap_unweighted_events.lhe"])
elif ap_decay_dist == "lhe_prompt":
    
    cnv = StdHepConverter(name="lhe_prompt", inputs=["ap_unweighted_events.lhe"])
elif ap_decay_dist == "displace_time":
    if 'ctau' in job.params:
        
        cnv = DisplaceUni(inputs=["ap_unweighted_events.lhe"])
    else:
        logger.error("Missing parameter ctau")
else:
    logger.error("Invalid ap decay distribution: ap_decay_dist = %s" % ap_decay_dist)
 
 
mom = AddMotherFullTruth(inputs=["ap_unweighted_events.stdhep", unzip.output_files()[0]], outputs=["ap_mom.stdhep"])
 
 
rot = BeamCoords(inputs=mom.output_files(), outputs=["ap_rot.stdhep"])
 
 
slic = SLIC(nevents=nevents + 1, inputs=rot.output_files(), outputs=["ap.slcio"])
 
 
job.add([mg, unzip, cnv, mom, rot, slic])