diff options
| author | Ryan Myers <ryan.p.myers@gmail.com> | 2023-12-21 10:52:51 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-12-21 12:52:51 -0300 |
| commit | 015ce283b47d5cd15b349bde600cb5b97e7247b7 (patch) | |
| tree | e0eb824c337a89dca4f82509c37078bacebda8b6 /tools/progress.py | |
| parent | 11197f6c28f3b96abd3be448811ab31474c646c3 (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.py | 114 |
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) |
