#!/usr/bin/env bash
# step-5000 效果演示：同一首帧，一组自造指令 + replay，8 卡并行各跑一条（50 步）。
set -euo pipefail
ROOT=/opt/dlami/nvme/zhiyangdeng/ActionRoPE
export DIFFSYNTH_SKIP_DOWNLOAD=True PYTHONPATH=$ROOT
CK=${CK:-$ROOT/outputs/arope_5k/step-5000.safetensors}
OUT=${OUT:-$ROOT/outputs/samples/arope_5k}; mkdir -p "$OUT"
FF=clip_Eybx_200000958_000273; SCENE=tidal_flats
run() { local gpu=$1 name=$2; shift 2
  CUDA_VISIBLE_DEVICES=$gpu "$ROOT/.venv/bin/python" -m actionrope.infer --ckpt "$CK" --steps 50 --out "$OUT/$name.mp4" "$@" \
    > "$OUT/$name.log" 2>&1 && grep -E "实测位移|PSNR" "$OUT/$name.log" | sed "s/^/[$name] /" || echo "[$name] FAILED"; }
run 0 replay      --replay $FF &
run 1 right_x1    --first_frame $FF --scene $SCENE --actions "right:1.0:21" &
run 2 left_x1     --first_frame $FF --scene $SCENE --actions "left:1.0:21" &
run 3 up_x1       --first_frame $FF --scene $SCENE --actions "up:1.0:21" &
run 4 down_x1     --first_frame $FF --scene $SCENE --actions "down:1.0:21" &
run 5 right_x0.5  --first_frame $FF --scene $SCENE --actions "right:0.5:21" &
run 6 right_x1.5  --first_frame $FF --scene $SCENE --actions "right:1.5:21" &
run 7 there_back  --first_frame $FF --scene $SCENE --actions "right:1.0:10,left:1.0:11" &
wait
echo DONE