mirror of
https://github.com/pmret/papermario.git
synced 2024-11-30 09:00:47 +00:00
ae66312d8c
* Add Python linter to github actions * wip * Add back splat_ext * Format files * C++ -> C * format 2 files * split workflow into separate file, line length 120, fix excludes * -l 120 in ci * update black locally and apply formatting changes * pyproject.toject --------- Co-authored-by: Ethan Roseman <ethteck@gmail.com>
136 lines
3.8 KiB
Python
Executable File
136 lines
3.8 KiB
Python
Executable File
#!/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())
|