Files
af/tools/progress.py
T
Anghelo Carvajal 817a05c1ab Frogress script (#10)
* initial progress script

* uploadfrogress script

* Add Frogress step to Jenkinsfile

* rename to animalforest

* badge

* ZeldaRET
2023-04-03 12:22:52 -04:00

81 lines
2.7 KiB
Python
Executable File

#!/usr/bin/env python3
# SPDX-FileCopyrightText: © 2023 ZeldaRET
# SPDX-License-Identifier: MIT
from __future__ import annotations
import argparse
import mapfile_parser
from pathlib import Path
ASMPATH = Path("asm")
NONMATCHINGS = "nonmatchings"
def getProgressFromMapFile(mapFile: mapfile_parser.MapFile, asmPath: Path, nonmatchings: Path, aliases: dict[str, str]=dict(), pathIndex: int=2) -> tuple[mapfile_parser.ProgressStats, dict[str, mapfile_parser.ProgressStats]]:
totalStats = mapfile_parser.ProgressStats()
progressPerFolder: dict[str, mapfile_parser.ProgressStats] = dict()
for file in mapFile:
if len(file.symbols) == 0:
continue
folder = file.filepath.parts[pathIndex]
if folder in aliases:
folder = aliases[folder]
if folder not in progressPerFolder:
progressPerFolder[folder] = mapfile_parser.ProgressStats()
originalFilePath = Path(*file.filepath.parts[pathIndex:])
fullAsmFile = asmPath / originalFilePath.with_suffix(".s")
wholeFileIsUndecomped = fullAsmFile.exists()
for func in file.symbols:
if wholeFileIsUndecomped:
totalStats.undecompedSize += func.size
progressPerFolder[folder].undecompedSize += func.size
elif mapFile.findSymbolByName(func.name) is not None:
totalStats.undecompedSize += func.size
progressPerFolder[folder].undecompedSize += func.size
else:
totalStats.decompedSize += func.size
progressPerFolder[folder].decompedSize += func.size
return totalStats, progressPerFolder
def getProgress(mapPath: Path, version: str) -> tuple[mapfile_parser.ProgressStats, dict[str, mapfile_parser.ProgressStats]]:
mapFile = mapfile_parser.MapFile()
mapFile.readMapFile(mapPath)
for file in mapFile:
if len(file.symbols) == 0:
continue
filepathParts = list(file.filepath.parts)
if version in filepathParts:
filepathParts.remove(version)
file.filepath = Path(*filepathParts)
nonMatchingsPath = ASMPATH / version / NONMATCHINGS
return mapFile.filterBySegmentType(".text").getProgress(ASMPATH / version, nonMatchingsPath, aliases={"ultralib": "libultra"})
def progressMain():
parser = argparse.ArgumentParser()
parser.add_argument("-v", "--version", help="version to process", default="jp")
args = parser.parse_args()
mapPath = Path("build") / f"animalforest.{args.version}.map"
totalStats, progressPerFolder = getProgress(mapPath, args.version)
mapfile_parser.progress_stats.printStats(totalStats, progressPerFolder)
if __name__ == "__main__":
progressMain()