"""Assert the reconstructed lamp tables byte-for-byte against BTL4OPT.EXE.

#98: the per-condenser lamp table was "corrected" on a consistency argument and
was wrong.  This pins every entry to the image so the next person does not have
to trust a comment -- run it and it either matches or it does not.
"""
import re, struct, sys

EXE = r"C:\git\bt411\content\BTL4OPT.EXE"
SRC = r"C:\git\bt411\game\reconstructed\btl4galm.cpp"

d = open(EXE, "rb").read()
e = struct.unpack_from("<I", d, 0x3C)[0]; coff = e + 4
ns = struct.unpack_from("<H", d, coff + 2)[0]
osz = struct.unpack_from("<H", d, coff + 16)[0]
opt = coff + 20; ib = struct.unpack_from("<I", d, opt + 28)[0]; st = opt + osz
# section header +8: VirtualSize, VirtualAddress, SizeOfRawData, PointerToRawData
_raw = [struct.unpack_from("<IIII", d, st + i * 40 + 8) for i in range(ns)]
secs = [(va, vsize, rawptr, rawsize) for (vsize, va, rawsize, rawptr) in _raw]


def off(va):
    r = va - ib
    for sva, vsize, rawptr, rawsize in secs:
        if sva <= r < sva + max(vsize, rawsize):
            return rawptr + (r - sva)
    raise SystemExit("VA %08x not mapped" % va)


def image_ints(va, n):
    o = off(va)
    return [struct.unpack_from("<i", d, o + 4 * i)[0] for i in range(n)]


src = open(SRC, encoding="utf-8", errors="replace").read()


def source_array(name):
    m = re.search(r"%s\s*\[\s*\d*\s*\]\s*=\s*\{([^}]*)\}" % re.escape(name), src)
    if not m:
        raise SystemExit("array %s not found in %s" % (name, SRC))
    return [int(x.strip(), 0) for x in m.group(1).split(",") if x.strip()]


CASES = [
    # (source array,          image VA,   how many, what it is)
    ("kBTCondenserLamp", 0x51D058, 7, "per-condenser lamps (index = condenserNumber, 1-based)"),
    ("kBTPlacementLamp", 0x51D070, 5, "per-placement lamps (index = auxScreenPlacement)"),
]

bad = 0
for name, va, n, what in CASES:
    got = source_array(name)[:n]
    want = image_ints(va, n)
    ok = got == want
    bad += (0 if ok else 1)
    print("%-20s @%08x  %-6s" % (name, va, "MATCH" if ok else "DIFFER"))
    print("   image : %s" % ["0x%X" % v for v in want])
    print("   source: %s" % ["0x%X" % v for v in got])
    print("   (%s)" % what)

# The coolingLoop switch is code, not a table: FUN_004cc148 maps 0..5.
print("\nkFixed (FUN_004cc148 switch) expected 0x2F 0x2E 0x2D 0x2C 0x2B 0x2A")
m = re.search(r"kFixed\s*\[\s*\d*\s*\]\s*=\s*\{([^}]*)\}", src)
fixed = [int(x.strip(), 0) for x in m.group(1).split(",") if x.strip()]
want = [0x2F, 0x2E, 0x2D, 0x2C, 0x2B, 0x2A]
ok = fixed == want
bad += (0 if ok else 1)
print("   source: %s  %s" % (["0x%X" % v for v in fixed], "MATCH" if ok else "DIFFER"))
print("\nNOTE the condenser set (2F 2E 2D 2B 2A 29) and the coolingLoop set")
print("(2F 2E 2D 2C 2B 2A) are DIFFERENT -- overlapping but not interchangeable.")

sys.exit(1 if bad else 0)