beam_gen_sample_job.py

Go to the documentation of this file.

"""!
@file beam_gen_sample_job.py
 
Generate beam events, transform them to beam coodinates and randomly sample them.
"""
from hpsmc.generators import EGS5
from hpsmc.tools import BeamCoords, RandomSample
 
 
inputs = list(job.input_files.values())
 
if 'nevents' in job.params:
    nevents = job.params['nevents']
else:
    nevents = 250000
 
 
egs5 = EGS5(name="beam_v7_%s" % job.params['run_params'])
 
 
rot = BeamCoords()
 
 
sample = RandomSample()
 
 
job.add([egs5, rot, sample])