79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
#!/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()
|