mirror of
https://github.com/pmret/papermario.git
synced 2024-11-09 12:32:38 +01:00
118 lines
3.6 KiB
Python
Executable File
118 lines
3.6 KiB
Python
Executable File
#!/usr/bin/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())
|