circuitpython/tools/gen_ld_files.py

61 lines
1.8 KiB
Python
Raw Normal View History

2019-10-20 23:50:12 -04:00
#! /usr/bin/env python3
import argparse
import os
import os.path
import sys
import re
from string import Template
parser = argparse.ArgumentParser(description='Apply #define values to .template.ld file.')
parser.add_argument('template_files', metavar='TEMPLATE_FILE', type=argparse.FileType('r'),
nargs='+', help="template filename: <something>.template.ld")
parser.add_argument('--defines', type=argparse.FileType('r'), required=True)
parser.add_argument('--out_dir', required=True)
args = parser.parse_args()
defines = {}
2019-12-05 22:45:53 -05:00
#
REMOVE_UL_RE = re.compile('([0-9]+)UL')
def remove_UL(s):
return REMOVE_UL_RE.sub(r'\1', s)
# We skip all lines before
# // START_LD_DEFINES
# Then we look for lines like this:
# /*NAME_OF_VALUE=*/ NAME_OF_VALUE;
VALUE_LINE_RE = re.compile(r'^/\*\s*(\w+)\s*=\*/\s*(.*);\s*$')
start_processing = False
2019-10-20 23:50:12 -04:00
for line in args.defines:
2019-12-05 22:45:53 -05:00
line = line.strip()
if line == '// START_LD_DEFINES':
start_processing = True
continue
if start_processing:
match = VALUE_LINE_RE.match(line)
if match:
name = match.group(1)
value = match.group(2).strip()
defines[name] = remove_UL(value)
2019-10-20 23:50:12 -04:00
fail = False
for template_file in args.template_files:
ld_template_basename = os.path.basename(template_file.name)
ld_pathname = os.path.join(args.out_dir, ld_template_basename.replace('.template.ld', '.ld'))
with open(ld_pathname, 'w') as output:
2019-12-05 22:45:53 -05:00
for k,v in defines.items():
print('/*', k, '=', v, '*/', file=output)
print(file=output)
2019-10-20 23:50:12 -04:00
try:
output.write(Template(template_file.read()).substitute(defines))
except KeyError as e:
print("ERROR: {}: No #define for '{}'".format(ld_pathname, e.args[0]), file=sys.stderr)
fail = True
if fail:
sys.exit(1)