48 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			48 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
import os
 | 
						|
 | 
						|
import bsdiff4
 | 
						|
 | 
						|
from ..Rom import get_base_rom_bytes
 | 
						|
 | 
						|
 | 
						|
def apply_basepatch(base_rom_bytes: bytes) -> bytes:
 | 
						|
    with open(os.path.join(os.path.dirname(__file__), "basepatch.bsdiff4"), "rb") as basepatch:
 | 
						|
        delta: bytes = basepatch.read()
 | 
						|
    return bsdiff4.patch(base_rom_bytes, delta)
 | 
						|
 | 
						|
 | 
						|
def create_basepatch() -> None:
 | 
						|
    from .asar import close as asar_close, geterrors as asar_errors, getprints as asar_prints, \
 | 
						|
        getwarnings as asar_warnings, init as asar_init, patch as asar_patch
 | 
						|
 | 
						|
    os.add_dll_directory(os.path.dirname(__file__))
 | 
						|
    print("Initializing Asar library")
 | 
						|
    asar_init()
 | 
						|
 | 
						|
    print("Opening base ROM")
 | 
						|
    old_rom_data: bytes = get_base_rom_bytes()
 | 
						|
 | 
						|
    print("Patching base ROM")
 | 
						|
    result, new_rom_data = asar_patch(os.path.join(os.path.dirname(__file__), "basepatch.asm"), old_rom_data)
 | 
						|
 | 
						|
    warnings = asar_warnings()
 | 
						|
    print("\nWarnings: " + str(len(warnings)))
 | 
						|
    for w in warnings:
 | 
						|
        print(w)
 | 
						|
 | 
						|
    if result:
 | 
						|
        print("Success")
 | 
						|
        for p in asar_prints():
 | 
						|
            print(p)
 | 
						|
        asar_close()
 | 
						|
        delta: bytes = bsdiff4.diff(old_rom_data, new_rom_data)
 | 
						|
        with open(os.path.join(os.path.dirname(__file__), "basepatch.bsdiff4"), "wb") as f:
 | 
						|
            f.write(delta)
 | 
						|
    else:
 | 
						|
        errors = asar_errors()
 | 
						|
        print("\nErrors: " + str(len(errors)))
 | 
						|
        for error in errors:
 | 
						|
            print(error)
 | 
						|
        asar_close()
 | 
						|
        raise RuntimeError("Asar errors while trying to create basepatch for Lufia II Ancient Cave.")
 |