23 from beamparameters
import add_beamparameters
28 print(
'Please provide ISR parameter')
30 if sys.argv[1] ==
"1":
32 if sys.argv[1] ==
"0":
36 isr = int(sys.argv[1])
46 mg_beamenergy =
'10.58/2.'
47 mg_generate =
'e+ e- > a ap, ap > mu+ mu-'
48 mg_parameter_wap =
'0.03102254'
49 mg_parameter_map =
'1.0e0'
61 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/models/darkphoton'
66 mg_steeringtemplate = \
67 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/examples/run_darkphoton_isr.steeringtemplate'
70 mg_steeringtemplate = \
71 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/examples/run_darkphoton.steeringtemplate'
76 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/output/darkphoton_mumu'
83 mg_externals =
'mg5_aMC'
87 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/examples/run_darkphoton.steering'
92 os.path.expandvars(
'$BELLE2_LOCAL_DIR/generators/madgraph/cards/run_card.dat'
97 mydict[
'MGMODEL'] = mg_model
98 mydict[
'MGOUTPUT'] = mg_outputdir
99 mydict[
'MGRUNDCARD'] = mg_runcard
100 mydict[
'MGNEVENTS'] = mg_nevents
101 mydict[
'MGBEAMENERGY'] = mg_beamenergy
102 mydict[
'MGGENERATE'] = mg_generate
103 mydict[
'MGPARAMETERWAP'] = mg_parameter_wap
104 mydict[
'MGPARAMETERMAP'] = mg_parameter_map
105 mydict[
'MGSEED'] = mg_seed
106 mydict[
'MGlpp1'] = mg_lpp1
107 mydict[
'MGlpp2'] = mg_lpp2
108 mydict[
'MGge'] = mg_ge
109 mydict[
'MGgchi'] = mg_gchi
110 mydict[
'MGel'] = mg_el
111 mydict[
'MGea'] = mg_ea
112 mydict[
'MGetaa'] = mg_etaa
113 mydict[
'MGetal'] = mg_etal
114 mydict[
'MGmll'] = mg_mll
115 mydict[
'MGbwcutoff'] = mg_bwcutoff
117 fp1 = open(mg_steeringfile,
'w')
118 fp2 = open(mg_steeringtemplate,
'r')
121 for (key, value)
in mydict.items():
122 data = data.replace(key, value)
127 subprocess.check_call([mg_externals, mg_steeringfile])
130 subprocess.check_call([
'gunzip', mg_outputdir +
'/Events/run_01/unweighted_events.lhe.gz'])
134 set_log_level(LogLevel.ERROR)
140 beamparameters = add_beamparameters(main,
"Y4S")
142 lhereader = register_module(
'LHEInput')
143 lhereader.param(
'makeMaster',
True)
144 lhereader.param(
'inputFileList', [mg_outputdir +
'/Events/run_01/unweighted_events.lhe'])
145 lhereader.param(
'useWeights',
False)
146 lhereader.param(
'nInitialParticles', 2)
147 lhereader.param(
'nVirtualParticles', 0)
148 lhereader.param(
'boost2Lab',
True)
149 lhereader.param(
'wrongSignPz',
True)
153 main.add_module(lhereader)
154 print_params(lhereader)
157 progress = register_module(
'Progress')
158 progress.set_log_level(LogLevel.INFO)
159 main.add_module(progress)
162 rootoutput = register_module(
'RootOutput')
163 rootoutput.param(
'outputFileName',
'LHEReaderMasterOutputDarkMuMu.root')
164 main.add_module(rootoutput)
167 main.add_module(
'PrintMCParticles', logLevel=LogLevel.DEBUG,