3.2 资源守恒与不变量检验

#!/usr/bin/env python3
# file: sim/gd_mech_03_conservation.py
# tier: S
# verify: /opt/homebrew/bin/python3 sim/gd_mech_03_conservation.py
"""3.2 资源守恒与不变量检验（S 层）。

快照给终态，轨迹给步序；两者都要能被**检验**。本脚本对同一份 play_spec.json
在**多组种子**下跑同一策略，逐局检查四条不变量，输出误差表——**全部应为 0**：

  ① 牌数守恒   hand + draw + discard == deckSize(12)      → cardErr
  ② 伤害守恒   dealt == params.enemyHp - enemyHp          → dmgErr
  ③ 能量范围   0 <= energy <= params.energyMax            → energyErr
  ④ 格挡非负   block >= 0                                 → blockErr

误差是**整数**，判据是「恰好为 0」，不是「约等于」。种子是**显式列举的整数**，
不使用 random 模块——同一份种子表在任何机器上给出同一张误差表。

最后给一个**人为破坏守恒的负例**（出牌时忘了把牌放进弃牌堆），证明检验器真的会报错，
而不是恒返回 0。
"""
from __future__ import annotations

import os
import sys

import ashlib

HERE = os.path.dirname(os.path.abspath(__file__))
ROOT = os.path.dirname(HERE)

SEEDS = [20261014, 1, 7, 42, 100, 999, 123456, 2147483646]
HEADER = ["seed", "cardErr", "dmgErr", "energyErr", "blockErr", "turn", "won"]


def check(spec: dict, snap: dict) -> dict:
    """四条不变量的误差（整数，合格即全 0）。"""
    deck_n = ashlib.deck_size(spec)
    p = spec["params"]
    return {
        "cardErr": snap["cardCount"] - deck_n,
        "dmgErr": snap["dealt"] - (p["enemyHp"] - snap["enemyHp"]),
        "energyErr": max(0, -snap["energy"]) + max(0, snap["energy"] - p["energyMax"]),
        "blockErr": max(0, -snap["block"]),
    }


def run_broken(spec: dict) -> dict:
    """负例：出牌时**忘了** discard.append(c)，牌数立刻不再守恒。其余规则照旧。"""
    p = spec["params"]
    deck: list = []
    for c in spec["cards"]:
        deck.extend([c] * c["count"])
    deck = ashlib.shuffle_deck(deck, spec["seed"])
    hand = deck[: p["handSize"]]
    draw = deck[p["handSize"]:]
    discard: list = []
    energy, player_hp, enemy_hp, block, dealt = p["energyMax"], p["playerHp"], p["enemyHp"], 0, 0
    for cmd in spec["strategy"]:
        if enemy_hp <= 0:
            break
        if cmd.startswith("p"):
            idx = int(cmd[1:])
            if idx < 0 or idx >= len(hand):
                continue
            c = hand[idx]
            if c["cost"] > energy:
                continue
            hand.pop(idx)
            # BUG 就在这里：漏了 discard.append(c)
            energy -= c["cost"]
            block += c["block"]
            d = min(c["damage"], enemy_hp)
            enemy_hp -= d
            dealt += d
        else:
            absorbed = min(block, p["enemyAttack"])
            block -= absorbed
            player_hp = max(0, player_hp - (p["enemyAttack"] - absorbed))
            block = 0
            energy = p["energyMax"]
            while len(hand) < p["handSize"] and draw:
                hand.append(draw.pop(0))
    return {
        "cardCount": len(hand) + len(draw) + len(discard),
        "dealt": dealt, "enemyHp": enemy_hp, "energy": energy, "block": block,
    }


def main() -> int:
    spec = ashlib.load_spec()
    rows: list = []
    total = 0
    for seed in SEEDS:
        s = dict(spec)
        s["seed"] = seed
        snap = ashlib.run_spec(s)
        err = check(s, snap)
        total += sum(err.values())
        rows.append([seed, err["cardErr"], err["dmgErr"], err["energyErr"],
                     err["blockErr"], snap["turn"], str(snap["won"]).lower()])

    out = os.path.join(ROOT, "out", "mech03_conservation.csv")
    n = ashlib.write_csv(out, HEADER, rows)

    print(f"seeds={len(SEEDS)} deckSize={ashlib.deck_size(spec)} energyMax={spec['params']['energyMax']}")
    print(" ".join(HEADER))
    for r in rows:
        print(" ".join(str(x) for x in r))
    print(f"totalError={total} allZero={total == 0}")

    broken = run_broken(spec)
    berr = check(spec, broken)
    print(f"broken: cardCount={broken['cardCount']} cardErr={berr['cardErr']} "
          f"dmgErr={berr['dmgErr']} energyErr={berr['energyErr']} blockErr={berr['blockErr']}")
    print(f"detected={any(v != 0 for v in berr.values())}")
    print(f"wrote out/mech03_conservation.csv rows={n}")
    return 0 if total == 0 and any(v != 0 for v in berr.values()) else 1


if __name__ == "__main__":
    sys.exit(main())

seeds=8 deckSize=12 energyMax=3
seed cardErr dmgErr energyErr blockErr turn won
20261014 0 0 0 0 4 true
1 0 0 0 0 4 true
7 0 0 0 0 4 true
42 0 0 0 0 4 true
100 0 0 0 0 3 true
999 0 0 0 0 4 true
123456 0 0 0 0 3 true
2147483646 0 0 0 0 4 true
totalError=0 allZero=True
broken: cardCount=3 cardErr=-9 dmgErr=0 energyErr=0 blockErr=0
detected=True
wrote out/mech03_conservation.csv rows=8

seed,cardErr,dmgErr,energyErr,blockErr,turn,won
20261014,0,0,0,0,4,true
1,0,0,0,0,4,true
7,0,0,0,0,4,true