sm64pc/tools/demo_data_converter.py

#!/usr/bin/env python3
import sys
import re
import json

def main():
    need_help = False
    defines = []
    skip_next = 0
    prog_args = []
    for i, a in enumerate(sys.argv[1:], 1):
        if skip_next > 0:
            skip_next -= 1
            continue
        if a == "--help" or a == "-h":
            need_help = True
        if a == "-D":
            defines.append(sys.argv[i + 1])
            skip_next = 1
        elif a.startswith("-D"):
            defines.append(a[2:])
        else:
            prog_args.append(a)

    defines = [d.split("=")[0] for d in defines]
    
    if len(prog_args) < 1 or need_help:
        print("Usage: {} <demo_data.json> [-D <symbol>] > <demo_data.c>".format(sys.argv[0]))
        sys.exit(0 if need_help else 1)

    with open(prog_args[0], "r") as file:
        descr = json.loads(re.sub(r"/\*[\w\W]*?\*/", "", file.read()))

    table = []
    for item in descr["table"]:
        if not "ifdef" in item or any(d in defines for d in item["ifdef"]):
            table.append(item)

    demofiles = []
    for item in descr["demofiles"]:
        if not "ifdef" in item or any(d in defines for d in item["ifdef"]):
            demofiles.append(item)

    structdef = ["u32 numEntries;",
                 "const void *addrPlaceholder;",
                 "struct OffsetSizePair entries[" + str(len(table)) + "];"]
    structobj = [str(len(table)) + ",",
                 "NULL,"]

    structobj.append("{")
    for item in table:
        offset_to_data = "offsetof(struct DemoInputsObj, " + item["demofile"] + ")"
        size = "sizeof(gDemoInputs." + item["demofile"] + ")"
        if "extraSize" in item:
            size += " + " + str(item["extraSize"])
        structobj.append("{" + offset_to_data + ", " + size + "},")
    structobj.append("},")

    for item in demofiles:
        with open("assets/demos/" + item["name"] + ".bin", "rb") as file:
            demobytes = file.read()
        structdef.append("u8 " + item["name"] + "[" + str(len(demobytes)) + "];")
        structobj.append("{" + ",".join(hex(x) for x in demobytes) + "},")

    print("#include \"types.h\"")
    print("#include <stddef.h>")
    print("")

    print("struct DemoInputsObj {")
    for s in structdef:
        print(s)
    print("} gDemoInputs = {")
    for s in structobj:
        print(s)
    print("};")

if __name__ == "__main__":
    main()