b9fd53fbef
Convert some scripts to use `argparse.FileType` where the change is relatively easily doable. This allows better error messages as e.g. missing input files will be detected during argument parsing. And it also makes writing to stdout in absence of an explicit argument simpler. Signed-off-by: Barnabás Pőcze <barnabas.pocze@ideasonboard.com> Reviewed-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
105 lines
3.1 KiB
Python
Executable File
105 lines
3.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
# Copyright (C) 2019, Google Inc.
|
|
#
|
|
# Author: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
|
|
#
|
|
# Generate control definitions from YAML
|
|
|
|
import argparse
|
|
import jinja2
|
|
import os
|
|
import sys
|
|
import yaml
|
|
|
|
from controls import Control
|
|
|
|
|
|
def snake_case(s):
|
|
return ''.join([c.isupper() and ('_' + c) or c for c in s]).strip('_')
|
|
|
|
|
|
def format_description(description):
|
|
description = description.strip('\n').split('\n')
|
|
for i in range(1, len(description)):
|
|
line = description[i]
|
|
description[i] = (line and ' * ' or ' *') + line
|
|
return '\n'.join(description)
|
|
|
|
|
|
def extend_control(ctrl, id, ranges):
|
|
ctrl.id = ranges[ctrl.vendor] + id + 1
|
|
|
|
if ctrl.vendor != 'libcamera':
|
|
ctrl.namespace = f'{ctrl.vendor}::'
|
|
else:
|
|
ctrl.namespace = ''
|
|
|
|
return ctrl
|
|
|
|
|
|
def main(argv):
|
|
|
|
# Parse command line arguments
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--mode', '-m', type=str, required=True, choices=['controls', 'properties'],
|
|
help='Mode of operation')
|
|
parser.add_argument('--output', '-o', metavar='file', default=sys.stdout,
|
|
type=argparse.FileType('w', encoding='utf-8'),
|
|
help='Output file name. Defaults to standard output if not specified.')
|
|
parser.add_argument('--ranges', '-r', required=True, type=argparse.FileType('rb'),
|
|
help='Control id range reservation file.')
|
|
parser.add_argument('--template', '-t', dest='template', required=True,
|
|
type=argparse.FileType('r', encoding='utf-8'),
|
|
help='Template file name.')
|
|
parser.add_argument('input', nargs='+',
|
|
type=argparse.FileType('rb'),
|
|
help='Input file name.')
|
|
|
|
args = parser.parse_args(argv[1:])
|
|
|
|
ranges = yaml.safe_load(args.ranges)['ranges']
|
|
|
|
controls = {}
|
|
for input in args.input:
|
|
data = yaml.safe_load(input)
|
|
|
|
vendor = data['vendor']
|
|
if vendor not in ranges.keys():
|
|
raise RuntimeError(f'Control id range is not defined for vendor {vendor}')
|
|
|
|
ctrls = controls.setdefault(vendor, [])
|
|
|
|
for i, ctrl in enumerate(data['controls']):
|
|
ctrl = Control(*ctrl.popitem(), vendor, args.mode)
|
|
ctrls.append(extend_control(ctrl, i, ranges))
|
|
|
|
# Sort the vendors by range numerical value
|
|
controls = [[vendor, ctrls] for vendor, ctrls in controls.items()]
|
|
controls.sort(key=lambda item: ranges[item[0]])
|
|
|
|
filename = {
|
|
'controls': 'control_ids',
|
|
'properties': 'property_ids',
|
|
}[args.mode]
|
|
|
|
data = {
|
|
'filename': filename,
|
|
'mode': args.mode,
|
|
'controls': controls,
|
|
}
|
|
|
|
env = jinja2.Environment()
|
|
env.filters['format_description'] = format_description
|
|
env.filters['snake_case'] = snake_case
|
|
template = env.from_string(args.template.read())
|
|
string = template.render(data)
|
|
|
|
args.output.write(string)
|
|
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main(sys.argv))
|