from hpsmc.generators import MG4 from hpsmc.tools import LHECount job.description = 'WAB generation using MadGraph4' # generate tritrig in MG5 mg = MG4(name="wab") # check that at least 80% of the requested events were generated or fail the job #check = LHECount(minevents=params.nevents*0.8, inputs=["wab_unweighted_events.lhe.gz"]) # run the job #job.components=[mg, check] job.add([mg])
# setup job including default parameters
job = Job(name="AP event gen job")
job.set_default_params(def_params)
job.initialize()
# get job params
params = job.params
# base file name
filename = "aprime"
# generate A-prime events using Madgraph 4
ap = MG4(name="ap",
run_card="run_card_" + params['run_params'] + ".dat",
params={"APMASS": params['apmass']},
outputs=[filename],
nevents=params['nevents'])
# unzip the LHE events to local file
unzip = Unzip(inputs=[filename + "_events.lhe.gz"])
# displace the time of decay using ctau param
displ = StdHepTool(name="lhe_tridents_displacetime",
inputs=[filename + "_events.lhe"],
outputs=[filename + ".stdhep"],
args=["-l", str(params['ctau'])])
# rotate events into beam coordinates and move vertex by 5 mm
rot = StdHepTool(name="beam_coords",
inputs=[filename + ".stdhep"],
import sys, os, argparse
from hpsmc.job import Job
from hpsmc.generators import MG4, StdHepConverter
from hpsmc.tools import Unzip, StdHepTool, SLIC, FilterMCBunches, JobManager, LCIOTool
job = Job(name="AP job")
job.initialize()
params = job.params
filename = "aprime"
# generate A-prime events using Madgraph 4
ap = MG4(name="ap",
run_card="run_card_" + params.run_params + ".dat",
params={"APMASS": params.apmass},
outputs=[filename],
nevents=params.nevents)
# unzip the LHE events to local file
unzip = Unzip(inputs=[filename + "_events.lhe.gz"])
# displace the time of decay using ctau param
displ = StdHepTool(name="lhe_tridents_displacetime",
inputs=[filename + "_events.lhe"],
outputs=[filename + ".stdhep"],
args=["-l", str(params.ctau)])
# rotate events into beam coordinates and move vertex by 5 mm
rot = StdHepTool(name="beam_coords",
inputs=[filename + ".stdhep"],
from hpsmc.generators import MG4, StdHepConverter
from hpsmc.tools import Unzip, BeamCoords, AddMotherFullTruth, SLIC
job.description = 'ap-displaced from generation to slic'
# Get job input file targets
inputs = list(job.input_files.values())
if 'nevents' in job.params:
nevents = job.params['nevents']
else:
nevents = 10000
# Generate rad in MG4
mg = MG4(name='ap', event_types=['unweighted'])
# Convert LHE output to stdhep
cnv = StdHepConverter(name="lhe_prompt", inputs=mg.output_files())
# Unzip the LHE events to a local file
unzip = Unzip(inputs=mg.output_files(), outputs=["ap.lhe"])
# Add mother particle to tag trident particles
mom = AddMotherFullTruth(
inputs=[cnv.output_files()[0],
unzip.output_files()[0]],
outputs=["ap_mom.stdhep"])
# Rotate events into beam coords
rot = BeamCoords(inputs=mom.output_files(), outputs=["ap_rot.stdhep"])
from hpsmc.generators import MG4, StdHepConverter
from hpsmc.job import Job
from hpsmc.run_params import RunParameters
from hpsmc.tools import Unzip, StdHepTool
ap = MG4(description="Generate A-prime events with APMASS param",
name="ap",
run_card="run_card_1pt05.dat",
params={"APMASS": 40.0},
outputs=["ap"],
rand_seed=1,
nevents=500)
unzip = Unzip(inputs=["ap_events.lhe.gz"])
rand_seed = 1
z = 1.0
displ = StdHepTool(name="lhe_tridents_displacetime",
inputs=["ap_events.lhe"],
outputs=["ap.stdhep"],
args=["-s", str(rand_seed), "-l",
str(z)])
dump = StdHepTool(name="print_stdhep", inputs=["ap.stdhep"])
job = Job(name="AP test", components=[ap, unzip, displ, dump])
job.params = {}
job.run()
""" Job script to generate A-prime events, convert to StdHep, and apply transformations. """ from hpsmc.generators import MG4 from hpsmc.tools import Unzip, StdHepTool, DisplaceTime, BeamCoords job.description = 'A-prime generation using MadGraph' # Generate A-prime events using MadGraph4 ap = MG4(name="ap", event_types=['unweighted']) # Unzip the LHE events to a local file unzip = Unzip() # Create a stdhep file, displacing the time of decay using the ctau param displace = DisplaceTime() # Rotate events into beam coordinates and move the vertices rotate = BeamCoords() # Add components to the job job.add([ap, unzip, displace, rotate])