shithub: pokecrystal

ref: 97a11c623ebd2c3e8d40a9c01b42669b2094da2c
dir: /tools/free_space.py/

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
Usage: python3 free_space.py [BANK=none] [pokecrystal.map]

Calculate the free space in the ROM or its individual banks.

The BANK argument allows printing free space in one, all, or none of the ROM's banks.
Valid arguments are numbers (in decimal "42" or hexadecimal "0x2A"), "all" or "none".
If not specified, defaults to "none".
"""

import sys

from mapreader import MapReader

def main():
	print_bank = 'none'
	mapfile = 'pokecrystal.map'

	for arg in sys.argv[1:]:
		if arg.startswith('BANK='):
			print_bank = arg.split('=', 1)[-1]
		else:
			mapfile = arg

	if print_bank not in {'all', 'none'}:
		try:
			if print_bank.startswith('0x') or print_bank.startswith('0X'):
				print_bank = int(print_bank[2:], 16)
			else:
				print_bank = int(print_bank)
		except ValueError:
			error = 'Error: invalid BANK: %s' % print_bank
			if print_bank.isalnum():
				error += ' (did you mean: 0x%s?)' % print_bank
			print(error, file=sys.stderr)
			sys.exit(1)

	num_banks = 0x80
	bank_size = 0x4000 # bytes
	total_size = num_banks * bank_size

	r = MapReader()
	with open(mapfile, 'r', encoding='utf-8') as f:
		l = f.readlines()
	r.read_map_data(l)

	free_space = 0
	per_bank = []
	default_bank_data = {'sections': [], 'used': 0, 'slack': bank_size}
	for bank in range(num_banks):
		bank_data = r.bank_data['ROM0 bank'] if bank == 0 else r.bank_data['ROMX bank']
		data = bank_data.get(bank, default_bank_data)
		used = data['used']
		slack = data['slack']
		per_bank.append((used, slack))
		free_space += slack

	print('Free space: %d/%d (%.2f%%)' % (free_space, total_size, free_space * 100.0 / total_size))
	if print_bank != 'none':
		print()
		print('bank, used, free')
	for bank in range(num_banks):
		used, slack = per_bank[bank]
		if print_bank in {'all', bank}:
				print('$%02X, %d, %d' % (bank, used, slack))

if __name__ == '__main__':
	main()