mirror of https://github.com/PentHertz/srsLTE.git
117 lines
4.4 KiB
Python
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)
|
|
|
|
|