ap_gen_to_slic_mg4_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 MG4, 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
 
# if 'ap_decay_dist' in job.params:
#     ap_decay_dist = job.params['ap_decay_dist']
# else:
#     ap_decay_dist = "lhe_uniform"
 
 
mg = MG4(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])