mirror of https://gitlab.freedesktop.org/mesa/mesa
75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
#
|
|
# Copyright 2019 Advanced Micro Devices, Inc.
|
|
#
|
|
# SPDX-License-Identifier: MIT
|
|
#
|
|
"""
|
|
Helper script that was used during the generation of the JSON data.
|
|
|
|
usage: python3 canonicalize.py FILE
|
|
|
|
Reads the register database from FILE, performs canonicalization
|
|
(de-duplication of enums and register types, implicitly sorting JSON by name)
|
|
and attempts to deduce missing register types.
|
|
|
|
Notes about deduced register types as well as the output JSON are printed on
|
|
stdout.
|
|
"""
|
|
|
|
from collections import defaultdict
|
|
import json
|
|
import re
|
|
import sys
|
|
|
|
from regdb import RegisterDatabase, deduplicate_enums, deduplicate_register_types
|
|
|
|
RE_number = re.compile('[0-9]+')
|
|
|
|
def deduce_missing_register_types(regdb):
|
|
"""
|
|
This is a heuristic for filling in missing register types based on
|
|
sequentially named registers.
|
|
"""
|
|
buckets = defaultdict(list)
|
|
for regmap in regdb.register_mappings():
|
|
buckets[RE_number.sub('0', regmap.name)].append(regmap)
|
|
|
|
for bucket in buckets.values():
|
|
if len(bucket) <= 1:
|
|
continue
|
|
|
|
regtypenames = set(
|
|
regmap.type_ref for regmap in bucket if hasattr(regmap, 'type_ref')
|
|
)
|
|
if len(regtypenames) == 1:
|
|
regtypename = regtypenames.pop()
|
|
for regmap in bucket:
|
|
if not hasattr(regmap, 'type_ref'):
|
|
print('Deducing {0} -> {1}'.format(regmap.name, regtypename), file=sys.stderr)
|
|
regmap.type_ref = regtypename
|
|
|
|
|
|
def json_canonicalize(filp, chips = None):
|
|
regdb = RegisterDatabase.from_json(json.load(filp))
|
|
|
|
if chips is not None:
|
|
for regmap in regdb.register_mappings():
|
|
assert not hasattr(regmap, 'chips')
|
|
regmap.chips = [chips]
|
|
|
|
deduplicate_enums(regdb)
|
|
deduplicate_register_types(regdb)
|
|
deduce_missing_register_types(regdb)
|
|
regdb.garbage_collect()
|
|
|
|
return regdb.encode_json_pretty()
|
|
|
|
|
|
def main():
|
|
print(json_canonicalize(open(sys.argv[1], 'r'), sys.argv[2]))
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|
|
# kate: space-indent on; indent-width 4; replace-tabs on;
|