14 from beamparameters
import add_beamparameters
20 mg_beamenergy =
'10.355/2.'
21 mg_generate =
'e+ e- > u u~'
25 mg_steeringtemplate = \
26 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/examples/run_sm.steeringtemplate'
31 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/output/sm_qqbar')
37 mg_externals =
'mg5_aMC'
41 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/examples/run_sm_uubar.steering'
46 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/cards/run_card.dat'
51 mydict[
'MGOUTPUT'] = mg_outputdir
52 mydict[
'MGRUNDCARD'] = mg_runcard
53 mydict[
'MGNEVENTS'] = mg_nevents
54 mydict[
'MGBEAMENERGY'] = mg_beamenergy
55 mydict[
'MGGENERATE'] = mg_generate
56 mydict[
'MGSEED'] = mg_seed
58 fp1 = open(mg_steeringfile,
'w')
59 fp2 = open(mg_steeringtemplate,
'r')
62 for (key, value)
in mydict.items():
63 data = data.replace(key, value)
68 subprocess.check_call([mg_externals, mg_steeringfile])
71 subprocess.check_call([
'gunzip', mg_outputdir +
'/Events/run_01/unweighted_events.lhe.gz'])
74 set_log_level(LogLevel.ERROR)
80 beamparameters = add_beamparameters(main,
"Y3S")
82 lhereader = register_module(
'LHEInput')
83 lhereader.param(
'makeMaster',
True)
84 lhereader.param(
'runNum', 1)
85 lhereader.param(
'expNum', 1)
86 lhereader.param(
'inputFileList', [mg_outputdir +
'/Events/run_01/unweighted_events.lhe'])
87 lhereader.param(
'useWeights',
False)
88 lhereader.param(
'nInitialParticles', 2)
89 lhereader.param(
'nVirtualParticles', 0)
90 lhereader.param(
'boost2Lab',
False)
91 lhereader.param(
'wrongSignPz',
True)
94 main.add_module(lhereader)
95 print_params(lhereader)
98 progress = register_module(
'Progress')
99 progress.set_log_level(LogLevel.INFO)
100 main.add_module(progress)
103 rootoutput = register_module(
'RootOutput')
104 rootoutput.param(
'outputFileName',
'LHEReaderMasterOutputSM.root')
105 main.add_module(rootoutput)