summaryrefslogtreecommitdiff
path: root/tools/progress.py
diff options
context:
space:
mode:
authorRyan Myers <ryan.p.myers@gmail.com>2023-12-21 10:52:51 -0500
committerGitHub <noreply@github.com>2023-12-21 12:52:51 -0300
commit015ce283b47d5cd15b349bde600cb5b97e7247b7 (patch)
treee0eb824c337a89dca4f82509c37078bacebda8b6 /tools/progress.py
parent11197f6c28f3b96abd3be448811ab31474c646c3 (diff)
Match over 20 new funcs in sf 77e40 (#55)
* 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
Diffstat (limited to 'tools/progress.py')
-rw-r--r--tools/progress.py114
1 files changed, 114 insertions, 0 deletions
diff --git a/tools/progress.py b/tools/progress.py
new file mode 100644
index 00000000..d840954c
--- /dev/null
+++ b/tools/progress.py
@@ -0,0 +1,114 @@
+#!/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)