Generate top json

Author: nathanaelhuffman

hdl/projects/grapefruit/BUCK

@@ -32,7 +32,8 @@ rdl_file(
         "//hdl/ip/vhd/espi:espi_regs_pkg",
     ],
     outputs = [
-        "gfruit_top_map.html",
+        "gfruit_top_map.html", 
+        "gfruit_top_map.json"
     ]
 )
 

tools/site_cobble/rdl_pkg/json_dump.py

@@ -45,6 +45,13 @@
     MemNode,
 )
 
+def convert_only_map_to_json(rdlc: RDLCompiler, obj: RootNode, path: Union[str, PathLike]):
+    # Convert entire register model to primitive datatypes (a dict/list tree)
+    json_obj = convert_addr_map_only(rdlc, obj.top)
+
+    # Write to a JSON file
+    with open(path, "w") as f:
+        json.dump(json_obj, f, indent=4)
 
 def convert_to_json(rdlc: RDLCompiler, obj: RootNode, path: Union[str, PathLike]):
     # Convert entire register model to primitive datatypes (a dict/list tree)
@@ -114,6 +121,33 @@ def convert_mem(obj: MemNode) -> dict:
     return json_obj
 
 
+def convert_addr_map_only(
+    rdlc: RDLCompiler, obj: Union[AddrmapNode, RegfileNode]
+) -> dict:
+    if obj.is_array:
+        rdlc.msg.fatal("JSON export does not support arrays", obj.inst.inst_src_ref)
+
+    json_obj = dict()
+    if isinstance(obj, AddrmapNode):
+        json_obj["type"] = "addrmap"
+    elif isinstance(obj, RegfileNode):
+        json_obj["type"] = "regfile"
+    else:
+        raise RuntimeError
+
+    json_obj["inst_name"] = obj.inst_name
+    json_obj["addr_offset"] = obj.address_offset
+
+    json_obj["children"] = []
+    for child in obj.children():
+        if isinstance(child, AddrmapNode):
+            json_child = convert_addr_map_only(rdlc, child)
+        else:
+            continue
+        json_obj["children"].append(json_child)
+
+    return json_obj
+
 def convert_addrmap_or_regfile(
     rdlc: RDLCompiler, obj: Union[AddrmapNode, RegfileNode]
 ) -> dict:

tools/site_cobble/rdl_pkg/rdl_cli.py

@@ -21,7 +21,7 @@
 except ModuleNotFoundError:
     from rdl_pkg.exporter import MapExporter, MapofMapsExporter
     from rdl_pkg.listeners import PreExportListener, MyModelPrintingListener
-    from rdl_pkg.json_dump import convert_to_json
+    from rdl_pkg.json_dump import convert_to_json, convert_only_map_to_json
 
 parser = argparse.ArgumentParser()
 parser.add_argument(
@@ -64,6 +64,13 @@ def main():
         # Dump Jinja template-based outputs (filter out .json)
         exporter = MapofMapsExporter()
         exporter.export(pre_export.maps[0], output_filenames_no_json)
+        json_files = [x for x in output_filenames if ".json" in str(x)]
+        if len(json_files) == 1:
+            json_name = Path(json_files[0])
+            convert_only_map_to_json(rdlc, root, json_name)
+        elif len(json_files) > 1:
+            raise Exception(f'Specified too many .json outputs: {json_files.join(",")}')
+
     else:
         # For each standard map, we're going to generate:
         # Standard bsv package from this base address