From 6dd1e346d330e662e42dd8bc75d17317223908fd Mon Sep 17 00:00:00 2001 From: Jan Tuomi Date: Sat, 24 Feb 2024 20:30:17 +0200 Subject: Add setup.py, fix bugs --- atk16_asm/pad_bin.py | 42 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 atk16_asm/pad_bin.py (limited to 'atk16_asm/pad_bin.py') diff --git a/atk16_asm/pad_bin.py b/atk16_asm/pad_bin.py new file mode 100644 index 0000000..a9fde17 --- /dev/null +++ b/atk16_asm/pad_bin.py @@ -0,0 +1,42 @@ +import os + +def pad_binary(file_path: str, target_size: int) -> int: + """ + Pads a binary file with zeros until it reaches a specified size in KiB. + + :param file_path: Path to the binary file to be padded. + :param target_size: Target size in bytes. + """ + + # Check current file size + current_size = os.path.getsize(file_path) + + # Calculate needed padding + padding_size = target_size - current_size + + if padding_size < 0: + raise ValueError(f"File is larger than the target size ({current_size} > {target_size}).") + + # Append zeros if needed + if padding_size > 0: + with open(file_path, 'ab') as file: + file.write(b'\x00' * padding_size) + + return padding_size + +def convert_size_to_bytes(size_str: str) -> int: + """ + Converts a size string with K, M, or G suffix to bytes. + + :param size_str: Size string (e.g., "64K", "1M", "2G"). + :return: Size in bytes. + """ + size_str = size_str.upper() + if size_str.endswith('K'): + return int(size_str[:-1]) * 1024 + elif size_str.endswith('M'): + return int(size_str[:-1]) * 1024 ** 2 + elif size_str.endswith('G'): + return int(size_str[:-1]) * 1024 ** 3 + else: + return int(size_str) -- cgit v1.3