parse_game_config.py

from construct import *
import sys

String = PascalString(Byte)
Word = Int16ul
Dword = Int32ul

def inc(ctx, var):
    res = ctx[var]
    ctx[var] += 1
    return res

CFG = Struct(
    "Magic" / Const("CFG\0"),
    "GameName" / String,
    "GameDescription" / String,
    "Version" / String,

    "StartSceneCategoryIndex" / Byte,
    "StartSceneIndex" / Word,

    "Objects" / PrefixedArray(Byte, String),

    "Palettes" / Array(8, Struct(
        "Bitmap" / Word,
        "i" / Computed(0),
        "Columns" / Array(16, Struct(
            "Pixels" / If(lambda ctx: (1 << inc(ctx._, "i")) & ctx._.Bitmap,
                Array(16, Struct(
                    "R" / Byte,
                    "G" / Byte,
                    "B" / Byte
                ))
            )
        ))
    )),

    "WavConfiguration" / PrefixedArray(Byte, Struct(
        "Path" / String,
        "MaxConcurrentPlay" / Byte
    )),

    "TotalScenes" / Word,

    "SceneCategories" / PrefixedArray(Byte, Struct(
        "CategoryName" / String,
        "Scenes" / PrefixedArray(Byte, Struct(
            "Name" / String,
            "Zone" / String,
            "SceneID" / String
        ))
    )),

    "ConfigMemoryEntries" / PrefixedArray(Byte, Struct(
        "Offset" / Dword,
        "Count" / Dword,
        "Data" / Array(this.Count, Dword)
    ))
)

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print "Usage: parse_game_config.py GameConfig.bin"
        sys.exit(-1)

    if not sys.argv[1].endswith(".bin"):
        print "Expected .bin file"
        sys.exit(-1)

    cfg = CFG.parse(open(sys.argv[1], "rb").read())
    print(cfg)