-
Notifications
You must be signed in to change notification settings - Fork 0
/
maddump.py
210 lines (162 loc) · 8.7 KB
/
maddump.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
from distutils import dir_util
import os
from madgraph.iolibs.files import cp, ln, mv
import madgraph.iolibs.export_v4 as export_v4
import madgraph.iolibs.file_writers as writers
import madgraph.various.misc as misc
import madgraph.various.banner as banner_mod
import madgraph.iolibs.group_subprocs as group_subprocs
import fit2D_card as fit2D
import bremsstrahlung_card as bremsstrahlung
import ebeampdffit_card as ebeampdffit
pjoin = os.path.join
class MadDump(export_v4.ProcessExporterFortranMEGroup):
check = True
# check status of the directory. Remove it if already exists
# Language type: 'v4' for f77/ 'cpp' for C++ output
exporter = 'v4'
# Output type:
#[Template/dir/None] copy the Template, just create dir or do nothing
output = 'Template'
# Decide which type of merging to used [madevent/madweight]
grouped_mode = 'madevent'
# if no grouping on can decide to merge uu~ and u~u anyway:
sa_symmetry = False
def __init__(self, *args, **opts):
misc.sprint("Initialise MadDump")
super(MadDump, self).__init__(*args, **opts)
def copy_template(self, *args, **opts):
maddump_dir = os.path.dirname(os.path.realpath( __file__ ))
# Copy the madevent templates
misc.sprint("copy the associate template")
super(MadDump, self).copy_template(*args, **opts)
# Add the PLUGIN templates
dir_util.copy_tree(pjoin(maddump_dir, 'FITPACK'),
pjoin(self.dir_path, 'Source/FITPACK'))
temp_dir= pjoin(maddump_dir, 'Templates/')
# cp(temp_dir + 'Cards/fit2D_card.dat',
# self.dir_path + '/Cards/fit2D_card.dat')
cp(temp_dir + 'Inc/fit2D.inc',
self.dir_path + '/Source/fit2D.inc')
cp(temp_dir + 'Inc/bremsstrahlung.inc',
self.dir_path + '/Source/bremsstrahlung.inc')
cp(temp_dir + 'Inc/ebeampdf_fit.inc',
self.dir_path + '/Source/ebeampdf_fit.inc')
#cp(temp_dir + 'meshfitter2D.py',
# self.dir_path + '/Cards/meshfitter2D.py')
#cp(temp_dir + 'lhe-meshfitter.py',
# self.dir_path + '/Cards/lhe-meshfitter.py')
#cp(temp_dir + 'README_MADDUMP',
#self.dir_path + '/Cards/README_MADDUMP')
# Update the makefile to compile the FITPACK routine
with open(self.dir_path+'/SubProcesses/makefile') as f:
text = []
for line in f:
if 'LINKLIBS' in line:
text.append(line)
text.append('LINKLIBS += -lfitpack\n')
else:
text.append(line)
# Add/Overwrite some functions in the Fortran code
with open(self.dir_path+'/SubProcesses/makefile', "w") as f:
for line in text:
f.write(line)
#modiy existing fortran code in SubProcess
files = ["dummy_fct.f","unwgt.f","reweight.f"]
remove_list = [['get_dummy_x1','get_dummy_x1_x2'],["store_events","write_leshouche"],["setclscales"]]
for name, to_rm in zip(files, remove_list):
template = open(pjoin(self.dir_path, "SubProcesses", name),"r").read()
plugin = open(pjoin(maddump_dir, "Templates",name),"r").read()
ff = writers.FortranWriter(pjoin(self.dir_path, "SubProcesses", name))
ff.remove_routine(template, to_rm, formatting=False)
ff.writelines(plugin, formatting=False)
ff.close()
#modiy existing fortran code in Source
files = ["rw_events.f","setrun.f","PDF/pdg2pdf.f"]
remove_list = [["read_event","write_event_to_stream","write_event"],["setrun"],["pdg2pdf"]]
for name, to_rm in zip(files, remove_list):
template = open(pjoin(self.dir_path, "Source", name),"r").read()
plugin = open(pjoin(maddump_dir, "Templates",name),"r").read()
ff = writers.FortranWriter(pjoin(self.dir_path, "Source", name))
ff.remove_routine(template, to_rm, formatting=False)
ff.writelines(plugin, formatting=False)
ff.close()
def write_source_makefile(self, writer):
"""Write the nexternal.inc file for MG4"""
replace_dict = super(MadDump, self).write_source_makefile(None)
libfitpack_line='''$(LIBDIR)libfitpack.$(libext):\n\tcd FITPACK; make\n\nlifitpack: $(LIBDIR)libfitpack.$(libext)\n'''
replace_dict['libraries'] += ' $(LIBDIR)libfitpack.$(libext)'
replace_dict['additional_dependencies'] += libfitpack_line
if writer:
path = pjoin(self.mgme_dir, 'madgraph', 'iolibs','template_files','madevent_makefile_source')
text = open(path).read() % replace_dict
writer.write(text)
writer.write('\tcd FITPACK; make clean; cd ..\n')
return replace_dict
def make_source_links(self):
""" Create the links from the files in sources """
ln(self.dir_path + '/Source/fit2D.inc', self.dir_path + '/SubProcesses', log=False)
ln(self.dir_path + '/Source/bremsstrahlung.inc', self.dir_path + '/SubProcesses', log=False)
ln(self.dir_path + '/Source/ebeampdf_fit.inc', self.dir_path + '/SubProcesses', log=False)
return super(MadDump, self).make_source_links()
def link_files_in_SubProcess(self, Ppath):
""" Create the necessary links in the P* directory path Ppath"""
#import ehist.dat and cell_frotran.dat into Subprocesses/P
ln(self.dir_path + '/Cards/ehist.dat', cwd=Ppath)
ln(self.dir_path + '/Cards/cell_fortran.dat', cwd=Ppath)
# import fit2D.in into Subprocesses/P
ln('../fit2D.inc', cwd=Ppath)
ln('../bremsstrahlung.inc', cwd=Ppath)
ln('../ebeampdf_fit.inc', cwd=Ppath)
super(MadDump, self).link_files_in_SubProcess(Ppath)
def pass_information_from_cmd(self, cmd):
"""pass information from the command interface to the exporter.
Please do not modify any object of the interface from the exporter.
"""
self.cmd = cmd
self.proc_characteristic['DM'] = cmd._dm_candidate[0]['pdg_code']
self.proc_characteristic['BSM_model'] = cmd._curr_model.get('modelpath')
return super(MadDump, self).pass_information_from_cmd(cmd)
def finalize(self,*args, **opts):
self.create_fit2D_card()
self.create_bremsstrahlung_card()
self.create_ebeampdffit_card()
from madgraph import MG5DIR
filename = os.path.join(self.cmd._export_dir, 'Cards', 'me5_configuration.txt')
self.cmd.do_save('options %s' % filename.replace(' ', '\ '), check=False,
to_keep={'mg5_path':MG5DIR})
return super(MadDump, self).finalize(*args, **opts)
#===========================================================================
# create the run_card
#===========================================================================
def create_run_card(self, matrix_elements, history):
""" """
run_card = banner_mod.RunCard()
run_card.remove_all_cut()
run_card['lpp1'] = 9
run_card.write(pjoin(self.dir_path, 'Cards', 'run_card_default.dat'))
run_card.write(pjoin(self.dir_path, 'Cards', 'run_card.dat'))
#===========================================================================
# create the fit2D_card
#===========================================================================
def create_fit2D_card(self):
""" """
fit2D_card = fit2D.Fit2DCard()
fit2D_card.write(pjoin(self.dir_path, 'Cards', 'fit2D_card_default.dat'))
fit2D_card.write(pjoin(self.dir_path, 'Cards', 'fit2D_card.dat'))
#===========================================================================
# create the bremsstrahlung_card
#===========================================================================
def create_bremsstrahlung_card(self):
""" """
bremsstrahlung_card = bremsstrahlung.BremsstrahlungCard()
bremsstrahlung_card.write(pjoin(self.dir_path, 'Cards', 'bremsstrahlung_card_default.dat'))
bremsstrahlung_card.write(pjoin(self.dir_path, 'Cards', 'bremsstrahlung_card.dat'))
#===========================================================================
# create the bremsstrahlung_card
#===========================================================================
def create_ebeampdffit_card(self):
""" """
ebeampdf_card = ebeampdffit.EbeamPdfFitCard()
ebeampdf_card.write(pjoin(self.dir_path, 'Cards', 'ebeampdf_fit_card_default.dat'))
ebeampdf_card.write(pjoin(self.dir_path, 'Cards', 'ebeampdf_fit_card.dat'))