08d1b86297
Co-authored-by: philmoz <philmoz@users.noreply.github.com> Co-authored-by: Peter Feerick <peter.feerick@gmail.com>
94 lines
2.3 KiB
Python
94 lines
2.3 KiB
Python
|
|
import re
|
|
import sys
|
|
import json
|
|
|
|
from hal_switches import Switch, parse_switches
|
|
from hal_adc import ADCInput, SPI_ADCInput, ADC, ADCInputParser
|
|
from hal_keys import Key, Trim, parse_trims, parse_keys
|
|
|
|
import legacy_names
|
|
|
|
#
|
|
# Return a file handle or STDIN
|
|
#
|
|
def open_file(filename):
|
|
|
|
if filename and not filename == '-':
|
|
return open(filename)
|
|
else:
|
|
return sys.stdin
|
|
|
|
#
|
|
# Read lines of defines into a dictionary
|
|
#
|
|
def parse_hw_defs(filename):
|
|
|
|
hw_defs = {}
|
|
|
|
with open_file(filename) as file:
|
|
for line in file.readlines():
|
|
m = re.match(r'#define ([^\s]*)\s*(.*)', line.rstrip())
|
|
name = m.group(1)
|
|
value = m.group(2)
|
|
if value.isnumeric():
|
|
value = int(value)
|
|
elif not value:
|
|
value = None
|
|
hw_defs[name] = value
|
|
|
|
return hw_defs
|
|
|
|
def prune_dict(d):
|
|
# ret = {}
|
|
# for k, v in d.items():
|
|
# if v is not None:
|
|
# ret[k] = v
|
|
# return ret
|
|
return d
|
|
|
|
class DictEncoder(json.JSONEncoder):
|
|
|
|
def default(self, obj):
|
|
if isinstance(obj, Switch):
|
|
return prune_dict(obj.__dict__)
|
|
if isinstance(obj, ADCInput):
|
|
return prune_dict(obj.__dict__)
|
|
if isinstance(obj, SPI_ADCInput):
|
|
return prune_dict(obj.__dict__)
|
|
if isinstance(obj, ADC):
|
|
return prune_dict(obj.__dict__)
|
|
if isinstance(obj, Trim):
|
|
return prune_dict(obj.__dict__)
|
|
if isinstance(obj, Key):
|
|
return prune_dict(obj.__dict__)
|
|
|
|
# Let the base class default method raise the TypeError
|
|
return json.JSONEncoder.default(self, obj)
|
|
|
|
#
|
|
# Parse HAL defines into JSON
|
|
#
|
|
def parse_defines(filename, target):
|
|
|
|
hw_defs = parse_hw_defs(filename)
|
|
out_defs = {}
|
|
|
|
# parse ADC first, we might have switches using ADC
|
|
legacy_inputs = legacy_names.inputs_by_target(target)
|
|
adc_parser = ADCInputParser(target,hw_defs,legacy_inputs)
|
|
adc_inputs = adc_parser.parse_inputs()
|
|
out_defs["adc_inputs"] = adc_inputs
|
|
|
|
switches = parse_switches(target, hw_defs, adc_parser)
|
|
out_defs["switches"] = switches
|
|
|
|
keys = parse_keys(target, hw_defs)
|
|
out_defs["keys"] = keys
|
|
|
|
trims = parse_trims(hw_defs)
|
|
out_defs["trims"] = trims
|
|
|
|
print(json.dumps(out_defs, cls=DictEncoder, indent=2))
|
|
|