diff --git a/tools/regconvert.py b/tools/regconvert.py index fdc1090b67..287cc13720 100755 --- a/tools/regconvert.py +++ b/tools/regconvert.py @@ -1,6 +1,7 @@ #!/usr/bin/env python3 import argparse +import re GROUP_SIZE = 0x60 DATA_OFFSET = 0x14 @@ -11,13 +12,96 @@ def get_reg_macro(offset): reg = (offset - DATA_OFFSET) // 2 group = reg // GROUP_SIZE reg_in_group = reg % GROUP_SIZE - return "%cREG(%d)\n" % (REGISTER_NAMES[group], reg_in_group) + return "%cREG(%d)" % (REGISTER_NAMES[group], reg_in_group) + +def parse_number(string, force_hex): + if string.lower().startswith("0x") or force_hex: + return int(string, 16) + return int(string, 0) + +def index_to_offset(index): + return index * 2 + DATA_OFFSET + +def read_file(filename): + parsed_contents = "" + + src_file = open(filename) + file_contents = src_file.read() + src_file.close() + + pattern = re.compile("gGameInfo->data\[((0[xX])?[0-9a-fA-F]+)\]") + + match = pattern.search(file_contents) + while match: + index = parse_number(match.group(1), False) + offset = index_to_offset(index) + + start, end = match.span() + parsed_contents += file_contents[:start] + parsed_contents += get_reg_macro(offset) + + file_contents = file_contents[end:] + match = pattern.search(file_contents) + + parsed_contents += file_contents + return parsed_contents + +def write_file(filename, contents): + with open(filename, "w") as f: + f.write(contents) + +def check_valid_index(index): + return 0x0 <= index <= 0xADF + +def check_valid_offset(offset): + return 0x14 <= index <= 0x15D2 def main(): - parser = argparse.ArgumentParser(description="Converts a GameInfo offset to a REG macro.") - parser.add_argument("offset", help="offset to GameInfo in hexadecimal") + parser = argparse.ArgumentParser(description="Converts a gGameInfo->data index to a REG macro.") + + # TODO: Add a description and a better title + # index_group = parser.add_argument_group(title="index", description="") + index_group = parser.add_argument_group() + index_group.add_argument("index", help="index of gGameInfo->data in decimal (or hexadecimal if starts with 0x)") + index_group.add_argument("--offset", help="treat index argument as an offset to gGameInfo instead", action="store_true") + index_group.add_argument("--hex", help="treat index argument as hexadecimal, even without the 0x", action="store_true") + + # TODO: Add a description and a better title + # file_group = parser.add_argument_group(title="file", description="") + file_group = parser.add_argument_group() + file_group.add_argument("--file", help="use the parameter as a filepath, then replace every occurrence of `gGameInfo->data[number]` in that file a REG macro", action="store_true") + file_group.add_argument("--stdout", help="print the processed file to stdout instead of overwriting the original", action="store_true") args = parser.parse_args() - print(get_reg_macro(int(args.offset, 16))) + + if args.file: + filename = args.index + contents = read_file(filename) + if args.stdout: + print(contents) + else: + write_file(filename, contents) + else: + index = parse_number(args.index, args.hex) + if args.offset: + offset = index + if not check_valid_offset(offset): + print("Invalid offset: " + args.index) + if args.hex: + print("Offset should be in the range from 0x14 to 0x15D2") + else: + print("Offset should be in the range from 20 to 5586") + exit(-1) + else: + if not check_valid_index(index): + print("Invalid index: " + args.index) + if args.hex: + print("Index should be in the range from 0x0 to 0xADF") + else: + print("Index should be in the range from 0 to 2783") + exit(-1) + offset = index_to_offset(index) + + print(get_reg_macro(offset)+"\n") if __name__ == "__main__": main()