mirror of
https://github.com/open-goal/jak-project
synced 2026-05-23 06:54:31 -04:00
water111
60 lines
1.6 KiB
Python
60 lines
1.6 KiB
Python
import os
|
|
import glob
|
|
import argparse
|
|
|
|
### Script to track decompilation progress.
|
|
### Example usage: python3 scripts/decomp_progress.py ~/jak-project/goal_src
|
|
|
|
def get_goal_files(root_dir):
|
|
"""Get all GOAL source files under root_dir"""
|
|
return [goal_file for file in os.walk(root_dir) for goal_file in glob.glob(os.path.join(file[0], '*.gc'))]
|
|
|
|
def lines_in_file(file_path):
|
|
with open(file_path) as f:
|
|
lines = 0
|
|
for _ in f:
|
|
lines += 1
|
|
return lines
|
|
|
|
|
|
def print_table(stats):
|
|
total_lines = 0
|
|
print("| {: <24} | {: <6} |".format("file name", "lines"))
|
|
print("-------------------------------------")
|
|
for x in stats:
|
|
print(" {: <24} | {: >6} |".format(x[0], x[1]))
|
|
total_lines += x[1]
|
|
print("-------------------------------------")
|
|
print("| {: <24} | {: >6} |".format("TOTAL", total_lines))
|
|
print("-------------------------------------")
|
|
estimated_lines = 500000
|
|
print("Progress: {}/{} lines ({:.2f}%)".format(total_lines, estimated_lines, 100. * total_lines / estimated_lines))
|
|
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(dest='goal_src', help='the goal_src folder')
|
|
args = parser.parse_args()
|
|
all_files = get_goal_files(args.goal_src)
|
|
|
|
file_stats = []
|
|
|
|
excluded_files = {"all_files.gc", "goal-lib.gc"}
|
|
|
|
for fn in all_files:
|
|
short_name = os.path.basename(fn)
|
|
line_count = lines_in_file(fn)
|
|
if line_count == 7 or short_name in excluded_files:
|
|
# the template has 7 lines, just skip it.
|
|
continue
|
|
|
|
file_stats.append((short_name, line_count))
|
|
file_stats.sort(key=lambda x: x[1])
|
|
|
|
print_table(file_stats)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|