mirror of
https://github.com/HarbourMasters/Starship.git
synced 2025-01-23 21:45:00 +03:00
015ce283b4
* Match func_8007B9DC and func_8007BB14 * Remove macro and add debug notices * Fix merge diff * Match func_8007BC7C * Match func_8007BCE8 * Match func_8007BDE0 * Match func_8007BE54 and an audio func func_8001A500 * Match func_8007BEE8 * Match func_8007BF64 * Match func_8007BFFC * Match func_8007C088 * Match func_8007C120 * Match func_8007C1AC * Match func_8007C250 * Fix merge from master * Match func_8007C3B4 * Match func_8007C484 and func_8007C50C * Match func_8007C608 and func_8007C688 * Match func_8007C6FC and func_8007C85C * Match func_8007C8C4 * Add a progress script WIP
115 lines
3.4 KiB
Python
115 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import argparse
|
|
import git
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from colour import Color
|
|
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
root_dir = os.path.join(script_dir, "..")
|
|
asm_dir = os.path.join(root_dir, "asm", "us", "nonmatchings")
|
|
build_dir = os.path.join(root_dir, "build")
|
|
elf_path = os.path.join(build_dir, "starfox64.us.elf")
|
|
|
|
def get_func_sizes():
|
|
try:
|
|
result = subprocess.run(['objdump', '-x', elf_path], stdout=subprocess.PIPE)
|
|
nm_lines = result.stdout.decode().split("\n")
|
|
except:
|
|
print(f"Error: Could not run objdump on {elf_path} - make sure that the project is built")
|
|
sys.exit(1)
|
|
|
|
sizes = {}
|
|
total = 0
|
|
|
|
for line in nm_lines:
|
|
if " F " in line:
|
|
components = line.split()
|
|
size = int(components[4], 16)
|
|
name = components[5]
|
|
total += size
|
|
sizes[name] = size
|
|
|
|
return sizes, total
|
|
|
|
def get_nonmatching_funcs():
|
|
funcs = set()
|
|
for root, dirs, files in os.walk(asm_dir):
|
|
for f in files:
|
|
if f.endswith(".s"):
|
|
funcs.add(f[:-2])
|
|
|
|
return funcs
|
|
|
|
def get_funcs_sizes(sizes, matchings, nonmatchings):
|
|
msize = 0
|
|
nmsize = 0
|
|
|
|
for func in matchings:
|
|
msize += sizes[func]
|
|
|
|
for func in nonmatchings:
|
|
if func not in sizes:
|
|
pass
|
|
# print(func)
|
|
else:
|
|
nmsize += sizes[func]
|
|
#print("% s,%i" % (func, sizes[func]))
|
|
|
|
return msize, nmsize
|
|
|
|
def lerp(a, b, alpha):
|
|
return a + (b - a) * alpha
|
|
|
|
def main(args):
|
|
func_sizes, total_size = get_func_sizes()
|
|
all_funcs = set(func_sizes.keys())
|
|
|
|
nonmatching_funcs = get_nonmatching_funcs()
|
|
matching_funcs = all_funcs - nonmatching_funcs
|
|
|
|
matching_size, nonmatching_size = get_funcs_sizes(func_sizes, matching_funcs, nonmatching_funcs)
|
|
|
|
if len(all_funcs) == 0:
|
|
funcs_matching_ratio = 0.0
|
|
matching_ratio = 0.0
|
|
else:
|
|
funcs_matching_ratio = (len(matching_funcs) / len(all_funcs)) * 100
|
|
matching_ratio = (matching_size / total_size) * 100
|
|
|
|
if args.csv:
|
|
version = 1
|
|
git_object = git.Repo().head.object
|
|
timestamp = str(git_object.committed_date)
|
|
git_hash = git_object.hexsha
|
|
csv_list = [str(version), timestamp, git_hash, str(len(all_funcs)), str(len(nonmatching_funcs)),
|
|
str(len(matching_funcs)), str(total_size), str(nonmatching_size), str(matching_size)]
|
|
print(",".join(csv_list))
|
|
elif args.shield_json:
|
|
import json
|
|
|
|
# https://shields.io/endpoint
|
|
color = Color("#50ca22", hue=lerp(0, 105/255, matching_ratio / 100))
|
|
print(json.dumps({
|
|
"schemaVersion": 1,
|
|
"label": f"progress",
|
|
"message": f"{matching_ratio:.2f}%",
|
|
"color": color.hex,
|
|
}))
|
|
else:
|
|
if matching_size + nonmatching_size != total_size:
|
|
print("Warning: category/total size mismatch!\n")
|
|
print(f"{len(matching_funcs)} matched functions / {len(all_funcs)} total ({funcs_matching_ratio:.2f}%)")
|
|
print(f"{matching_size} matching bytes / {total_size} total ({matching_ratio:.2f}%)")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Reports progress for the project")
|
|
parser.add_argument("--csv", action="store_true")
|
|
parser.add_argument("--shield-json", action="store_true")
|
|
args = parser.parse_args()
|
|
|
|
main(args)
|