romconv.py

#!/usr/bin/env python3
import sys

PATCHES = {
    "galaga_rom_cpu1":
    [
        # jump over tile ram test
        ( 0x3382, 0x06, 0xc3 ),     # 06, jp
        ( 0x3383, 0x0a, 0x35 ),     # xx35
	( 0x3384, 0xd9, 0x34 ),     # 34xx
	# only one ramtest round, instead of 30
	( 0x348a, 0x1e, 0x01 ),
        # skip rom test
	( 0x352b, 0xe5, 0xc9 )      # ret
    ]    
}

def parse_rom(id, infiles, outfile, apply_patches = False):
    offset = 0
    of = open(outfile, "w")

    print("const unsigned char "+id+"[] = {\n  ", end="", file=of)
    
    for name_idx in range(len(infiles)):
        f = open(infiles[name_idx], "rb")
        rom_data = f.read()
        f.close()

        # the first frogger audio cpu rom has bits
        # d0 and d1 swapped. Fix this
        if rom_data[:8] == bytes([0x05,0x00,0x22,0x00,0x40,0xc3,0x0b,0x02]):
            rom_data = list(rom_data)
            for i in range(len(rom_data)):
                rom_data[i] = (rom_data[i] & 0xfc) | ((rom_data[i] & 2)>>1) | ((rom_data[i] & 1)<<1)
            rom_data = bytes(rom_data)
            
        # apply patches
        if apply_patches:
            rom_data = list(rom_data)
            for i in PATCHES:
                if i == id:
                    for p in PATCHES[i]:
                        if p[0] - offset < len(rom_data):
                            if rom_data[p[0] - offset] == p[1]:
                                print("Patching", hex(p[0]), ":", p[1], "->", p[2])
                                rom_data[p[0] - offset] = p[2]
                            else:
                                raise ValueError("Unexpected patchdata")
        
        offset += len(rom_data)
        for i in range(len(rom_data)):
            print("0x{:02X}".format(rom_data[i]), end="", file=of)
            if i != len(rom_data)-1 or name_idx != len(infiles)-1:
                print(",", end="", file=of)
                if i&15 == 15:
                    print("\n  ", end="", file=of)
            else:
                print("", file=of)
        
    print("};", file=of)
    of.close()

if len(sys.argv) < 3:
    print("Usage:",sys.argv[0], "name <infiles> <outfile>")
    print("  -p  - apply patches (e.g. fast boot)")
    print("  Galaga CPU1:      ", sys.argv[0], "-p galaga_rom_cpu1 ../roms/gg1_1b.3p ../roms/gg1_2b.3m ../roms/gg1_3.2m ../roms/gg1_4b.2l ../galagino/galaga_rom1.h")
    print("  Galaga CPU2:      ", sys.argv[0], "galaga_rom_cpu2 ../roms/gg1_5b.3f ../galagino/galaga_rom2.h")
    print("  Galaga CPU3:      ", sys.argv[0], "galaga_rom_cpu3 ../roms/gg1_7b.2c ../galagino/galaga_rom3.h")
    print("  Pacman:           ", sys.argv[0], "pacman_rom ../roms/pacman.6e ../roms/pacman.6f ../roms/pacman.6h ../roms/pacman.6j ../galagino/pacman_rom.h")
    print("  Donkey Kong CPU1: ", sys.argv[0], "dkong_rom_cpu1 ../roms/c_5et_g.bin ../roms/c_5ct_g.bin ../roms/c_5bt_g.bin ../roms/c_5at_g.bin ../galagino/dkong_rom1.h")
    print("  Donkey Kong CPU2: ", sys.argv[0], "dkong_rom_cpu2 ../roms/s_3i_b.bin ../roms/s_3j_b.bin ../galagino/dkong_rom2.h")
    print("  Frogger CPU1:     ", sys.argv[0], "frogger_rom_cpu1 ../roms/frogger.26 ../roms/frogger.27 ../roms/frsm3.7 ../galagino/frogger_rom1.h")
    print("  Frogger CPU2:     ", sys.argv[0], "frogger_rom_cpu2 ../roms/frogger.608 ../roms/frogger.609 ../roms/frogger.610 ../galagino/frogger_rom2.h")
    print("  Digdug CPU1:      ", sys.argv[0], "digdug_rom_cpu1 ../roms/dd1a.1 ../roms/dd1a.2 ../roms/dd1a.3 ../roms/dd1a.4 ../galagino/digdug_rom1.h")
    print("  Digdug CPU2:      ", sys.argv[0], "digdug_rom_cpu2 ../roms/dd1a.5 ../roms/dd1a.6 ../galagino/digdug_rom2.h")
    print("  Digdug CPU3:      ", sys.argv[0], "digdug_rom_cpu3 ../roms/dd1.7 ../galagino/digdug_rom3.h")
    print("  Digdug Playfield: ", sys.argv[0], "digdug_playfield ../roms/dd1.10b ../galagino/digdug_playfield.h")
    print("  1942 CPU1:        ", sys.argv[0], "_1942_rom_cpu1 ../roms/srb-03.m3 ../roms/srb-04.m4 ../galagino/1942_rom1.h")
    print("  1942 CPU1 bank 0: ", sys.argv[0], "_1942_rom_cpu1_b0 ../roms/srb-05.m5 ../galagino/1942_rom1_b0.h")
    print("  1942 CPU1 bank 1: ", sys.argv[0], "_1942_rom_cpu1_b1 ../roms/srb-06.m6 ../galagino/1942_rom1_b1.h")
    print("  1942 CPU1 bank 2: ", sys.argv[0], "_1942_rom_cpu1_b2 ../roms/srb-07.m7 ../galagino/1942_rom1_b2.h")
    print("  1942 CPU2:        ", sys.argv[0], "_1942_rom_cpu2 ../roms/sr-01.c11 ../galagino/1942_rom2.h")
    
    exit(-1)

if sys.argv[1] == "-p":       
    parse_rom(sys.argv[2], sys.argv[3:-1], sys.argv[-1], True)
else:
    parse_rom(sys.argv[1], sys.argv[2:-1], sys.argv[-1])