#!/usr/bin/env bash
# ============================================================
# 种子步进链执行模板 (seed_nc [→ nl]) — 由 new_test.sh 拷为 scripts/run.sh
# 规范: docs/testing_workflow_2026_08_11.md §四；references/tlusty-stages.md
# 输入: seed_nc.5 seed_nc.nst fort.8(邻居收敛模型)；若跑到 nl 还需 nl.5 nl.nst
# 用法: bash scripts/run.sh [timeout_per_stage]
# ============================================================
set -u
TEST_DIR="$(cd "$(dirname "$0")/.." && pwd)"
TIMEOUT="${1:-1800}"   # 种子链 seed_nc 给宽点
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 seed_nc.5 seed_nc.nst fort.8; do
  [ -f "inputs/$f" ] || { echo "❌ 缺输入: inputs/$f"; exit 1; }
done
echo "  seed_nc.5 L2:   $(sed -n '2p' inputs/seed_nc.5)    (应含 F F)"
echo "  seed_nc.nst L1: $(head -1 inputs/seed_nc.nst)       (应含 NITER=, ICHANG=0, ORELAX=0.3)"
echo "  fort.8 NaN行:   $(grep -c -iE 'nan|inf|\*{3,}' inputs/fort.8 || true)   (应为 0)"
mkdir -p run outputs
ln -sfn "$RUNTIME_DATA" run/data
echo "  data -> $(readlink run/data)"

# ---------- 准备 run/ ----------
cp inputs/seed_nc.5 inputs/seed_nc.nst inputs/fort.8 run/ 2>/dev/null
[ -f inputs/nl.5 ] && cp inputs/nl.5 inputs/nl.nst run/ 2>/dev/null
rm -f run/fort.6 run/fort.7 run/fort.8.run run/fort.9 run/fort.14
# seed_nc 阶段种子 = 邻居模型 (inputs/fort.8 已经拷进 run/)

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; }
    # seed_nc: run/fort.8 已是邻居种子；nl: 上阶段 fort.7→fort.8
    if [ "$stage" = "nl" ]; then
        [ -f run/seed_nc.7 ] && cp run/seed_nc.7 run/fort.8 || { echo "⚠ 无 seed_nc.7 作 nl 种子"; }
    fi
    cd run
    timeout "$TIMEOUT" "$BIN" < "$stage.5" > fort.6 2> fort.14
    local rc=$?
    cd ..
    # rc=124 = timeout 杀死。种子链 seed_nc 本就慢，更要注意截断。
    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
    echo "  $stage rc=$rc"
}

echo "[$(date +%H:%M:%S)] $TEST_DIR 开始"
run_stage seed_nc
# nl 仅在 nl.5 存在时执行
if [ -f inputs/nl.5 ]; then
    [ -f run/seed_nc.7 ] && cp run/seed_nc.7 run/fort.8
    run_stage nl
fi
echo "[$(date +%H:%M:%S)] $TEST_DIR 完成"

# ---------- 归档 ----------
echo "=== [归档] -> outputs/ ==="
for s in seed_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/seed_nc.9"
echo "    python3 ../../../../../../.agents/skills/tlusty-synspec-test/scripts/check_temperature.py outputs/seed_nc.7 --teff <Teff>"