papermario/tools/sortsymz.py
Ethan Roseman 021592d219
Misc decomp 70 (#716)
* aura_appendGfx

* 3 more action commands

* fire_shell

* hammer, hurricane

* spiny_surge

* data stuff, cleanup, whirlwind

* warnings

* water_block data and func

* water_block

* effect stuff cleanup

* big_snowflakes effect

* data cleanup, fpr abi regs

* 2 battle item funcs + data

* 2 mo

* shooting_star func

* 3 16F740 funcs

* btl_state_update_end_battle

* map funcs

* more dedupe

* cleanin

* draw_entity_model_C/D

* effect butterflies, btl state funcs

* it is done

* it is done

* PR comments, some cleanup
2022-05-27 22:03:19 +09:00

68 lines
2.1 KiB
Python
Executable File

#!/usr/bin/env python3
import os
import re
syms = []
groups = {}
with open("tools/symz.txt") as f:
for line in f.readlines():
if line.strip() and not line.startswith("//"):
name, addr = line.strip().strip(";").split(" = ")
syms.append({"name": name, "address": int(addr, 0), "found_in": set(), "dead": False})
# Search src for syms
for root, dirs, files in os.walk("src"):
for file in files:
if file.endswith(".c") and "bss" not in file:
with open(os.path.join(root, file)) as f:
text = f.read()
for sym in syms:
if sym["name"] in text:
sym["found_in"].add(os.path.join(root, file).replace("src/", ""))
# Search asm for syms
for root, dirs, files in os.walk("ver/us/asm"):
for file in files:
if file.endswith(".s"):
with open(os.path.join(root, file)) as f:
text = f.read()
for sym in syms:
if sym["name"] in text:
sym["found_in"].add(os.path.join(root, file).replace("ver/us/asm/nonmatchings/", ""))
if re.match(r"E[A-F][0-9A-F]{4}", root.split("/")[-1]):
sym["dead"] = True
else:
sym["dead"] = False
for sym in syms:
if len(sym["found_in"]) == 0:
group = ""
elif sym["dead"]:
group = "dead"
elif len(sym["found_in"]) > 1:
group = "multig"
else:
group = next(iter(sym["found_in"]))
if group not in groups:
groups[group] = []
groups[group].append(sym)
for group in groups:
if group == "":
print("// The following symbols can be simply removed from undefined_syms:")
elif group == "multig":
print("// The following symbols are found in multiple locations:")
elif group == "dead":
print("// The following symbols are dead:")
else:
print("// " + group + ":")
sorted_syms = sorted(groups[group], key=lambda tup: tup["address"])
for sym in sorted_syms:
print(f"{sym['name']} = 0x{sym['address']:X}")
print()