You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

117 lines
4.4 KiB
Python

#
# Copyright 2012-2013 The libLTE Developers. See the
# COPYRIGHT file at the top-level directory of this distribution.
#
# This file is part of the libLTE library.
#
# libLTE is free software: you can redistribute it and/or modify
# it under the terms of the GNU Lesser General Public License as
# published by the Free Software Foundation, either version 3 of
# the License, or (at your option) any later version.
#
# libLTE is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Lesser General Public License for more details.
#
# A copy of the GNU Lesser General Public License can be found in
# the LICENSE file in the top-level directory of this distribution
# and at http://www.gnu.org/licenses/.
#
import shutil
import os
def MakeModule(m):
dir = os.path.dirname(__file__)
tdir = os.path.join(dir, 'template')
ddir = tdir + m.directory_name
print 'Creating new directory ' + ddir
shutil.copytree(tdir,ddir)
WriteAloeC(m,tdir + '/src/template.c',\
ddir + '/src/' + m.name + '.c')
WriteAloeH(m,tdir + '/src/template.h',\
ddir + '/src/' + m.name + '.h')
def WriteInitParamsGet(m, f):
for p in m.init_params:
f.write('\tif (param_get_' + p['type'] + '_name("' + p['name'] + '", &' + p['variable']+ ')) {\n'
'\t\t' + p['variable'] + ' = ' + str(p['default']) + ';\n'\
'\t}\n')
def WriteInputParamsHandlers(m, f):
for p in m.input_params:
f.write('pmid_t ' + p['name'] + '_id;\n')
def WriteInputParamsId(m, f):
for p in m.input_params:
f.write('\t' + p['name'] + '_id = param_id("' + p['name'] + '");\n')
def WriteInputParamsGet(m, f):
for p in m.input_params:
f.write('\tif (param_get_' + p['type'] + '(' + p['name'] + '_id, &' + p['variable'] + ') != 1) {\n')
if p['default'] == None:
f.write('\t\t' + r'moderror("Error getting parameter ' + p['name'] + '\\n");' + '\n')
f.write('\t\treturn -1;\n\t}\n')
else:
f.write('\t\t' + r'modinfo("Parameter ' + p['name'] + ' not found. Setting to default (' + p['default'] + ')\\n");' + '\n')
f.write('\t\t' + p['variable'] + ' = ' + p['default'] + ';\n\t}\n')
def WriteAloeC(m, input, output):
with open(input,'r') as f:
newlines = []
for line in f.readlines():
newlines.append(line.replace('-name-', m.name))
with open(output, 'w') as f:
for line in newlines:
if '--input_parameters_handlers--' in line:
WriteInputParamsHandlers(m, f)
elif '--input_parameters_getid--' in line:
WriteInputParamsId(m, f)
elif '--init_parameters--' in line:
WriteInitParamsGet(m, f)
elif '--input_parameters--' in line:
WriteInputParamsGet(m, f)
else:
f.write(line)
def WriteAloeH(m, input, output):
with open(input,'r') as f:
newlines = []
for line in f.readlines():
if '-typeinput-' in line:
if m.nof_inputs > 0:
newlines.append(line.replace('-typeinput-', m.input_type))
elif '-numinputs-' in line:
newlines.append(line.replace('-numinputs-', str(m.nof_inputs)))
elif '-sizeinput-' in line:
if m.nof_inputs > 0:
newlines.append(line.replace('-sizeinput-', m.input_size))
else:
newlines.append(line.replace('-sizeinput-', '0'))
elif '-typeoutput-' in line:
if m.nof_outputs > 0:
newlines.append(line.replace('-typeoutput-', m.output_type))
elif '-numoutputs-' in line:
newlines.append(line.replace('-numoutputs-', str(m.nof_outputs)))
elif '-sizeoutput-' in line:
if m.nof_outputs > 0:
newlines.append(line.replace('-sizeoutput-', m.output_size))
else:
newlines.append(line.replace('-sizeoutput-', '0'))
else:
newlines.append(line)
with open(output, 'w') as f:
for line in newlines:
f.write(line)