import hashlib import os import Utils from Patch import read_rom, APDeltaPatch JAP10HASH = '21f3e98df4780ee1c667b84e57d88675' ROM_PLAYER_LIMIT = 65535 class SMDeltaPatch(APDeltaPatch): hash = JAP10HASH game = "Super Metroid" patch_file_ending = ".apsm" @classmethod def get_source_data(cls) -> bytes: return get_base_rom_bytes() def get_base_rom_bytes(file_name: str = "") -> bytes: base_rom_bytes = getattr(get_base_rom_bytes, "base_rom_bytes", None) if not base_rom_bytes: file_name = get_base_rom_path(file_name) base_rom_bytes = bytes(read_rom(open(file_name, "rb"))) basemd5 = hashlib.md5() basemd5.update(base_rom_bytes) if JAP10HASH != basemd5.hexdigest(): raise Exception('Supplied Base Rom does not match known MD5 for JAP(1.0) release. ' 'Get the correct game and version, then dump it') get_base_rom_bytes.base_rom_bytes = base_rom_bytes return base_rom_bytes def get_base_rom_path(file_name: str = "") -> str: options = Utils.get_options() if not file_name: file_name = options["sm_options"]["rom_file"] if not os.path.exists(file_name): file_name = Utils.user_path(file_name) return file_name