unleashed-firmware/scripts/fwsize.py
hedger c3ececcf96
[FL-3174] Dolphin builder in ufbt; minor ufbt/fbt improvements (#2601)
* ufbt: added "dolphin_ext" target (expects "external" subfolder in cwd with dolphin assets); cleaned up unused code
* ufbt: codestyle fixes
* scripts: fixed style according to ruff linter
* scripts: additional cleanup & codestyle fixes
* github: pass target hw code when installing local SDK with ufbt
* ufbt: added error message for missing folder in dolphin builder
* scripts: more linter fixes
* sdk: added flipper_format_stream; ufbt: support for --extra-define
* fbt: reduced amount of global defines
* scripts, fbt: rearranged imports

Co-authored-by: Aleksandr Kutuzov <alleteam@gmail.com>
2023-05-03 14:48:49 +09:00

56 lines
1.7 KiB
Python

#!/usr/bin/env python3
import math
import os
import subprocess
from ansi.color import fg
from flipper.app import App
class Main(App):
def init(self):
self.subparsers = self.parser.add_subparsers(help="sub-command help")
self.parser_elfsize = self.subparsers.add_parser("elf", help="Dump elf stats")
self.parser_elfsize.add_argument("elfname", action="store")
self.parser_elfsize.set_defaults(func=self.process_elf)
self.parser_binsize = self.subparsers.add_parser("bin", help="Dump bin stats")
self.parser_binsize.add_argument("binname", action="store")
self.parser_binsize.set_defaults(func=self.process_bin)
def process_elf(self):
all_sizes = subprocess.check_output(
["arm-none-eabi-size", "-A", self.args.elfname], shell=False
)
all_sizes = all_sizes.splitlines()
sections_to_keep = (".text", ".rodata", ".data", ".bss", ".free_flash")
for line in all_sizes:
line = line.decode("utf-8")
parts = line.split()
if len(parts) != 3:
continue
section, size, _ = parts
if section not in sections_to_keep:
continue
print(f"{section:<11} {size:>8} ({(int(size)/1024):6.2f} K)")
return 0
def process_bin(self):
PAGE_SIZE = 4096
binsize = os.path.getsize(self.args.binname)
pages = math.ceil(binsize / PAGE_SIZE)
last_page_state = (binsize % PAGE_SIZE) * 100 / PAGE_SIZE
print(
fg.yellow(
f"{os.path.basename(self.args.binname):<11}: {pages:>4} flash pages (last page {last_page_state:.02f}% full)"
)
)
return 0
if __name__ == "__main__":
Main()()