#!/usr/bin/env bash
# ============================================================
# 冷启动链执行模板 (lte → nc → nl) — 由 new_test.sh 拷为 scripts/run.sh
# 规范: docs/testing_workflow_2026_08_11.md §四；references/tlusty-stages.md
# 用法: bash scripts/run.sh [timeout_per_stage]
# ============================================================
set -u
TEST_DIR="$(cd "$(dirname "$0")/.." && pwd)"
TIMEOUT="${1:-1200}"   # 每阶段超时（秒）
BIN="/home/fmq/program/tlusty/tl208-s54/dcts/assets/tlusty_static"
RUNTIME_DATA="/home/fmq/program/tlusty/tl208-s54/dcts/assets/data"

cd "$TEST_DIR" || { echo "无法进入 $TEST_DIR"; exit 1; }

# ---------- 执行前检查 ----------
echo "=== [检查] 输入文件 ==="
for f in lte.5 nc.5 nl.5 lte.nst nc.nst nl.nst; do
  [ -f "inputs/$f" ] || { echo "❌ 缺输入: inputs/$f"; exit 1; }
done
echo "  lte.5 L2: $(sed -n '2p' inputs/lte.5)   (应含 T T)"
echo "  nc.5  L2: $(sed -n '2p' inputs/nc.5)    (应含 F F)"
echo "  nl.5  L2: $(sed -n '2p' inputs/nl.5)    (应含 F F)"
for s in lte nc nl; do echo "  $s.nst L1: $(head -1 inputs/$s.nst)"; done
# 冷启动链不应有 fort.8
[ -f "inputs/fort.8" ] && { echo "❌ 冷启动链不应有 inputs/fort.8（lte 阶段 ltgray=T 会删它）"; exit 1; }
mkdir -p run outputs
ln -sfn "$RUNTIME_DATA" run/data
echo "  data -> $(readlink run/data)"

# ---------- 准备 run/ ----------
echo "=== [准备] run/ 工作目录 ==="
cp inputs/*.5 inputs/*.nst run/ 2>/dev/null
rm -f run/fort.6 run/fort.7 run/fort.8 run/fort.9 run/fort.14

# ---------- 阶段执行（每阶段产物按阶段重命名） ----------
run_stage() {
    local stage="$1"
    echo "=== [执行] $stage (timeout ${TIMEOUT}s) ==="
    cp "run/$stage.nst" run/nst 2>/dev/null || { echo "❌ 无 run/$stage.nst"; exit 1; }
    # 冷启动 lte: 删 fort.8 (ltgray=T 强制灰大气重建)；nc/nl: 保留上阶段 fort.7→fort.8
    if [ "$stage" = "lte" ]; then rm -f run/fort.8; fi
    cd run
    timeout "$TIMEOUT" "$BIN" < "$stage.5" > fort.6 2> fort.14
    local rc=$?
    cd ..
    # rc=124 = timeout 杀死（GNU timeout）。结果可能未完成/被截断，写标记供事后审查注意。
    if [ "$rc" -eq 124 ]; then
      echo "  ⚠⚠ $stage TIMEOUT (rc=124, 超过 ${TIMEOUT}s 被 kill —— 迭代可能未完成，产物被截断)"
      touch "$TEST_DIR/outputs/${stage}.TIMEOUT"
    fi
    cp run/fort.6 "$stage.6" 2>/dev/null
    [ -f run/fort.7 ] && cp run/fort.7 "$stage.7"
    [ -f run/fort.9 ] && cp run/fort.9 "$stage.9"
    cp run/fort.14 "$stage.14" 2>/dev/null
    [ -f run/fort.7 ] && cp run/fort.7 run/fort.8   # 下阶段种子
    echo "  $stage rc=$rc"
}

echo "[$(date +%H:%M:%S)] $TEST_DIR 开始"
run_stage lte
run_stage nc
run_stage nl
echo "[$(date +%H:%M:%S)] $TEST_DIR 完成"

# ---------- 归档 ----------
echo "=== [归档] -> outputs/ ==="
for s in lte nc nl; do
    [ -f "$s.6" ] && cp "$s.6" "$s.7" "$s.9" "$s.14" outputs/ 2>/dev/null
done
echo "outputs/: $(ls outputs/ | wc -l) 个文件"
echo "=== 完成。请做事后审查（references/physics-checks.md）："
echo "    python3 ../../../../../../.agents/skills/tlusty-synspec-test/scripts/check_fort9.py outputs/nl.9"
echo "    python3 ../../../../../../.agents/skills/tlusty-synspec-test/scripts/check_temperature.py outputs/nl.7 --teff <Teff>"