mirror of
https://github.com/pmret/papermario.git
synced 2024-11-08 12:02:30 +01:00
ae66312d8c
* Add Python linter to github actions * wip * Add back splat_ext * Format files * C++ -> C * format 2 files * split workflow into separate file, line length 120, fix excludes * -l 120 in ci * update black locally and apply formatting changes * pyproject.toject --------- Co-authored-by: Ethan Roseman <ethteck@gmail.com>
115 lines
2.9 KiB
Python
Executable File
115 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
import json
|
|
import glob
|
|
import os
|
|
import argparse
|
|
from enum import IntEnum
|
|
|
|
|
|
script_dir = os.path.dirname(os.path.realpath(__file__))
|
|
asm_dir = script_dir + "/../ver/current/asm/nonmatchings"
|
|
|
|
modes = ["min", "max", "avg", "total", "size"]
|
|
|
|
sizes = {}
|
|
|
|
funcs = {}
|
|
|
|
|
|
# Calculate the number of instructions in a .s file
|
|
def calc_insns(f_path):
|
|
ret = 0
|
|
with open(f_path) as f:
|
|
f_lines = f.readlines()
|
|
for line in f_lines:
|
|
if line.startswith("/* "):
|
|
ret += 1
|
|
funcs[f_path.split("/")[-1][:-2]] = ret
|
|
return ret
|
|
|
|
|
|
# Calculate different data points for each .c files and store them as a Tuple
|
|
def do_dir(root, dir):
|
|
max = 0
|
|
min = 0
|
|
total = 0
|
|
|
|
files = glob.glob(os.path.join(root, dir) + "/*.s")
|
|
|
|
for f in files:
|
|
amt = calc_insns(f)
|
|
if amt > max:
|
|
max = amt
|
|
if min == 0 or amt < min:
|
|
min = amt
|
|
total += amt
|
|
|
|
avg = 0 if len(files) == 0 else total / len(files)
|
|
|
|
sizes[root + "/" + dir] = (min, max, total, avg, len(files))
|
|
|
|
|
|
parser = argparse.ArgumentParser(
|
|
description="A tool to receive information about the number of non-matching .s files "
|
|
+ "per .c file, or the size of .s files, measured by their number of instructions. "
|
|
+ "Option -p is used by default if no option is specified."
|
|
)
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument(
|
|
"-f",
|
|
"--files",
|
|
help="Default. Print the number of non-matching .s files per .c file, ordered by size.",
|
|
action="store_true",
|
|
required=False,
|
|
)
|
|
group.add_argument(
|
|
"-a",
|
|
"--alphabetical",
|
|
help="Print the size of .s files, ordered by name.",
|
|
action="store_true",
|
|
required=False,
|
|
)
|
|
group.add_argument(
|
|
"-s",
|
|
"--size",
|
|
help="Print the size of .s files, ordered by size.",
|
|
action="store_true",
|
|
required=False,
|
|
)
|
|
parser.add_argument(
|
|
"-l",
|
|
"--limit",
|
|
help="Only print the .c --files that are greater than or equal to the value.",
|
|
type=int,
|
|
default=0,
|
|
required=False,
|
|
)
|
|
parser.add_argument(
|
|
"-m",
|
|
"--mode",
|
|
help="Switches between output modes for --files. Allowed values are: {min, max, avg, total, size}.",
|
|
choices=modes,
|
|
default="size",
|
|
metavar="",
|
|
required=False,
|
|
)
|
|
|
|
args = parser.parse_args()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
for root, dirs, files in os.walk(asm_dir):
|
|
for asm_dir in dirs:
|
|
do_dir(root, asm_dir)
|
|
|
|
if args.alphabetical:
|
|
print(json.dumps(dict(sorted(funcs.items(), key=lambda f: f[0])), indent=4))
|
|
elif args.size:
|
|
print(json.dumps(dict(sorted(funcs.items(), key=lambda f: f[1])), indent=4))
|
|
else:
|
|
for thing in sorted(sizes.keys(), key=lambda x: sizes[x][modes.index(args.mode)]):
|
|
val = sizes[thing][modes.index(args.mode)]
|
|
if val > args.limit:
|
|
print(thing.split("nonmatchings/")[1].ljust(50) + str(val))
|