43 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Python
		
	
	
	
			
		
		
	
	
			43 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			Python
		
	
	
	
import hashlib
 | 
						|
import os
 | 
						|
 | 
						|
import Utils
 | 
						|
from Patch import read_rom, APDeltaPatch
 | 
						|
 | 
						|
SMJUHASH = '21f3e98df4780ee1c667b84e57d88675'
 | 
						|
ROM_PLAYER_LIMIT = 65535
 | 
						|
 | 
						|
 | 
						|
class SMDeltaPatch(APDeltaPatch):
 | 
						|
    hash = SMJUHASH
 | 
						|
    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 SMJUHASH != basemd5.hexdigest():
 | 
						|
            raise Exception('Supplied Base Rom does not match known MD5 for Japan+US 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
 |