mirror of
https://github.com/smaeul/u-boot.git
synced 2025-10-31 20:18:18 +00:00
d5737b3f6a
Tidy up this tool a little: - define which arguments are needed - split the enum values out into a header file - warn if no enum values are found - display the dtc error if something goes wrong - avoid a Python traceback on error Signed-off-by: Simon Glass <sjg@chromium.org>
139 lines
3.7 KiB
Python
Executable File
139 lines
3.7 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0+
|
|
|
|
"""
|
|
Expo utility - used for testing of expo features
|
|
|
|
Copyright 2023 Google LLC
|
|
Written by Simon Glass <sjg@chromium.org>
|
|
"""
|
|
|
|
import argparse
|
|
import collections
|
|
import io
|
|
import re
|
|
import subprocess
|
|
import sys
|
|
|
|
#from u_boot_pylib import cros_subprocess
|
|
from u_boot_pylib import tools
|
|
|
|
# Parse:
|
|
# SCENE1 = 7,
|
|
# or SCENE2,
|
|
RE_ENUM = re.compile(r'(\S*)(\s*= (\d))?,')
|
|
|
|
# Parse #define <name> "string"
|
|
RE_DEF = re.compile(r'#define (\S*)\s*"(.*)"')
|
|
|
|
def calc_ids(fname):
|
|
"""Figure out the value of the enums in a C file
|
|
|
|
Args:
|
|
fname (str): Filename to parse
|
|
|
|
Returns:
|
|
OrderedDict():
|
|
key (str): enum name
|
|
value (int or str):
|
|
Value of enum, if int
|
|
Value of #define, if string
|
|
"""
|
|
vals = collections.OrderedDict()
|
|
with open(fname, 'r', encoding='utf-8') as inf:
|
|
in_enum = False
|
|
cur_id = 0
|
|
for line in inf.readlines():
|
|
line = line.strip()
|
|
if line == 'enum {':
|
|
in_enum = True
|
|
continue
|
|
if in_enum and line == '};':
|
|
in_enum = False
|
|
|
|
if in_enum:
|
|
if not line or line.startswith('/*'):
|
|
continue
|
|
m_enum = RE_ENUM.match(line)
|
|
if m_enum.group(3):
|
|
cur_id = int(m_enum.group(3))
|
|
vals[m_enum.group(1)] = cur_id
|
|
cur_id += 1
|
|
else:
|
|
m_def = RE_DEF.match(line)
|
|
if m_def:
|
|
vals[m_def.group(1)] = tools.to_bytes(m_def.group(2))
|
|
|
|
return vals
|
|
|
|
|
|
def run_expo(args):
|
|
"""Run the expo program"""
|
|
fname = args.enum_fname or args.layout
|
|
ids = calc_ids(fname)
|
|
if not ids:
|
|
print(f"Warning: No enum ID values found in file '{fname}'")
|
|
|
|
indata = tools.read_file(args.layout)
|
|
|
|
outf = io.BytesIO()
|
|
|
|
for name, val in ids.items():
|
|
if isinstance(val, int):
|
|
outval = b'%d' % val
|
|
else:
|
|
outval = b'"%s"' % val
|
|
find_str = r'\b%s\b' % name
|
|
indata = re.sub(tools.to_bytes(find_str), outval, indata)
|
|
|
|
outf.write(indata)
|
|
data = outf.getvalue()
|
|
|
|
with open('/tmp/asc', 'wb') as outf:
|
|
outf.write(data)
|
|
proc = subprocess.run('dtc', input=data, capture_output=True)
|
|
edtb = proc.stdout
|
|
if proc.stderr:
|
|
print(f"Devicetree compiler error:\n{proc.stderr.decode('utf-8')}")
|
|
return 1
|
|
tools.write_file(args.outfile, edtb)
|
|
return 0
|
|
|
|
|
|
def parse_args(argv):
|
|
"""Parse the command-line arguments
|
|
|
|
Args:
|
|
argv (list of str): List of string arguments
|
|
|
|
Returns:
|
|
tuple: (options, args) with the command-line options and arugments.
|
|
options provides access to the options (e.g. option.debug)
|
|
args is a list of string arguments
|
|
"""
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('-D', '--debug', action='store_true',
|
|
help='Enable full debug traceback')
|
|
parser.add_argument('-e', '--enum-fname', type=str,
|
|
help='.dts or C file containing enum declaration for expo items')
|
|
parser.add_argument('-l', '--layout', type=str, required=True,
|
|
help='Devicetree file source .dts for expo layout (and perhaps enums)')
|
|
parser.add_argument('-o', '--outfile', type=str, required=True,
|
|
help='Filename to write expo layout dtb')
|
|
|
|
return parser.parse_args(argv)
|
|
|
|
def start_expo():
|
|
"""Start the expo program"""
|
|
args = parse_args(sys.argv[1:])
|
|
|
|
if not args.debug:
|
|
sys.tracebacklimit = 0
|
|
|
|
ret_code = run_expo(args)
|
|
sys.exit(ret_code)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
start_expo()
|