papermario/tools/star_rod_enum_to_decomp.py

136 lines
3.8 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
import argparse
import sys
import re
import os
from glob import glob
def create_enum(file_content, prefix, ordering):
ret = ""
max_size = 0
if ordering:
for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
if len(key) > max_size:
max_size = len(key)
else:
for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
if len(value) > max_size:
max_size = len(value)
if prefix is not None:
max_size += len(prefix)
else:
prefix = ""
for key, value in re.findall(r"(\S+)\s+=\s+(\S+)", file_content):
if ordering:
key = (
"_".join(re.sub(r"([A-Z]{1,2})", r" \1", key).split())
.replace("N_P_C", "NPC")
.replace("__", "_")
.replace("-", "")
)
key = prefix.upper() + "_{:<{width}}".format(key, width=max_size + 2).upper()
ret += " " + key + " = 0x" + "{:>{fill}{width}}".format(value, fill=0, width=8) + ",\n"
else:
value = (
"_".join(re.sub(r"([A-Z]{1,2})", r" \1", value).split())
.replace("N_P_C", "NPC")
.replace("__", "_")
.replace("-", "")
)
value = prefix.upper() + "_{:<{width}}".format(value, width=max_size + 2).upper()
ret += " " + value + " = 0x" + "{:>{fill}{width}}".format(key, fill=0, width=8) + ",\n"
ret += "};\n"
return ret
def single_translation(file_path):
ret = ""
try:
file = open(file_path)
file_content = file.read()
file.close()
except:
print("File not found at the given path.")
return "t", "y"
enum_namespace = re.search(r"(\w*)\s+%\s+namespace", file_content).group(1)
enum_name = re.search(r"(\w*)\s+%\s+library name", file_content).group(1)
reversed_order = re.search(r"(\w*)\s+%\s+reversed", file_content).group(1)
ret += "enum " + enum_name + " {\n"
if reversed_order == "true":
ret += create_enum(file_content, enum_namespace, True)
else:
ret += create_enum(file_content, enum_namespace, False)
return enum_name, ret
def recursive_translation(database_path):
ret = ""
stuff = [f for f in glob(database_path + "/**/*.*", recursive=True) if f.endswith(".enum") or f.endswith(".flags")]
for element in stuff:
try:
file = open(element)
file_content = file.read()
file.close()
except:
continue
enum_namespace = re.search(r"(\w*)\s+%\s+namespace", file_content).group(1)
enum_name = re.search(r"(\w*)\s+%\s+library name", file_content).group(1)
reversed_order = re.search(r"(\w*)\s+%\s+reversed", file_content).group(1)
ret += "enum " + enum_name + " {\n"
if reversed_order == "true":
ret += create_enum(file_content, enum_namespace, True)
else:
ret += create_enum(file_content, enum_namespace, False)
ret += "\n"
return "enums_and_flags", ret
def main(args):
ret = ""
if args.recursive:
enum_name, result = recursive_translation(args.query)
else:
enum_name, result = single_translation(args.query)
ret += result
file = open("" + enum_name + ".txt", "w+")
file.write(ret)
file.close()
parser = argparse.ArgumentParser(
description="Convert a StarRod enum into an enum that is in a decomp compatible format"
)
parser.add_argument("query", help="StarRod enum file or folder")
parser.add_argument(
"-r",
"--recursive",
help="recursively convert all files to enums",
type=bool,
required=False,
)
args = parser.parse_args()
if __name__ == "__main__":
main(parser.parse_args())