#!/bin/sh
# nt-overcall-gladiator-ab.sh — A/B for Gladiator advances of our 1NT overcall of
# their MAJOR (`set_nt_overcall_gladiator`, docs/bba-gap-campaign.md def-r1 lever):
# a `2♣` weak relay + a cue-of-their-major Stayman for the one unbid major +
# natural INV + splinter / Leaping-Michaels, replacing the opening-1NT graft over
# major openings only.  OFF = shipped default (systems-on opening-1NT graft over
# majors); ON = --ns-nt-overcall-gladiator.  Both arms keep systems-on for the
# MINOR openings (untouched), so the minor split is a ~0 sanity check and the
# major split is the real test.  Both vulnerabilities, three scorers (plain+pd via
# ab-dump-diff, sd via ab-dump-sd), arms sequential, one shared SEED_BASE.  Do NOT
# touch the codebase while it runs (bba-gen-parallel re-invokes cargo build; must
# stay a no-op).
#
#   PER_SHARD=6400 SPLIT=<split_by_opening.py> setsid nohup scripts/idle-run.sh \
#       scripts/nt-overcall-gladiator-ab.sh ab-results/nt-overcall-gladiator \
#       >ab-results/nt-overcall-gladiator.log 2>&1 &
set -eu
cd "$(dirname "$0")/.."

R=${1:?usage: nt-overcall-gladiator-ab.sh RESULTS_DIR}
SPLIT=${SPLIT:?set SPLIT to the split_by_opening.py path}
mkdir -p "$R"
SHA=$(git rev-parse --short HEAD)
DIFF=target/release/examples/ab-dump-diff
SD=target/release/examples/ab-dump-sd
PER_SHARD=${PER_SHARD:-6400}
SHARDS=$(nproc)

cargo build --release --features serde --example bba-gen --example ab-dump-diff --example ab-dump-sd

log() { echo "$(date -u +%FT%TZ) $*" | tee -a "$R/log" >&2; }
if [ ! -s "$R/seed" ]; then date +%s >"$R/seed"; fi
SEED_BASE=$(cat "$R/seed")

arm() {
    name=$1; vul=$2; shift 2
    dir="$R/$name-$vul"
    [ -d "$dir" ] && { log "skip $dir (exists)"; return 0; }
    log "generate $dir (SEED_BASE=$SEED_BASE, flags: $*)"
    SEED_BASE=$SEED_BASE scripts/bba-gen-parallel.sh "$dir" "$PER_SHARD" -v "$vul" "$@" \
        >>"$R/log" 2>&1
}

# diffpair ON_DIR OFF_DIR TAG — plain + pd, 8 solvers wide
diffpair() {
    on=$1; off=$2; tag=$3; vul=$4
    for score in plain pd; do
        out="$R/diff.$tag.$score.txt"
        [ -s "$out" ] && { log "skip $out"; continue; }
        log "diff $tag ($score)"
        i=0
        while [ "$i" -lt "$SHARDS" ]; do
            [ -f "$on/shard-$i.json" ] && "$DIFF" "$on/shard-$i.json" "$off/shard-$i.json" \
                --score "$score" --show 0 >"$out.shard-$i" 2>&1 &
            [ $(((i + 1) % 8)) -eq 0 ] && wait
            i=$((i + 1))
        done
        wait
        cat "$out".shard-* >"$out" 2>/dev/null; rm -f "$out".shard-*
    done
}

# sddiff ON_DIR OFF_DIR TAG VUL — sd-lead, 16 worlds
sddiff() {
    on=$1; off=$2; tag=$3; vul=$4
    out="$R/sd.$tag.txt"
    [ -s "$out" ] && { log "skip $out"; return 0; }
    log "sd-diff $tag"
    i=0
    while [ "$i" -lt "$SHARDS" ]; do
        [ -f "$on/shard-$i.json" ] && "$SD" "$on/shard-$i.json" "$off/shard-$i.json" \
            -v "$vul" --sd-worlds 16 --show 0 >"$out.shard-$i" 2>&1 &
        [ $(((i + 1) % 8)) -eq 0 ] && wait
        i=$((i + 1))
    done
    wait
    cat "$out".shard-* >"$out" 2>/dev/null; rm -f "$out".shard-*
}

log "=== nt-overcall-gladiator A/B start, sha=$SHA, SEED_BASE=$SEED_BASE, ${SHARDS}x${PER_SHARD} bd/arm/vul"
for vul in none both; do
    arm off "$vul"
    arm on  "$vul" --ns-nt-overcall-gladiator
    python3 "$SPLIT" "$R/on-$vul"
    python3 "$SPLIT" "$R/off-$vul"
    for kind in minor major; do
        diffpair "$R/on-$vul-$kind" "$R/off-$vul-$kind" "$vul.$kind" "$vul"
        sddiff   "$R/on-$vul-$kind" "$R/off-$vul-$kind" "$vul.$kind" "$vul"
    done
done
log "=== nt-overcall-gladiator A/B done"