#!/bin/bash
# ============================================================
# Helper script to submit SLURM jobs
# ============================================================
#
# Usage:
#   ./scripts/submit.sh test          # Quick interactive test
#   ./scripts/submit.sh synthetic     # Full synthetic experiment
#   ./scripts/submit.sh shd           # Full SHD experiment
#   ./scripts/submit.sh custom        # Show custom submission example
# ============================================================

set -e
cd "$(dirname "$0")/.."

case "$1" in
    test)
        echo "Submitting quick test job (interactive partition)..."
        sbatch scripts/test_interactive.sbatch
        ;;
    synthetic)
        echo "Submitting synthetic data experiment..."
        sbatch scripts/run_depth_experiment.sbatch
        ;;
    shd)
        echo "Submitting SHD experiment..."
        sbatch scripts/run_depth_experiment_shd.sbatch
        ;;
    custom)
        echo "Custom submission example:"
        echo ""
        echo "  # Override parameters:"
        echo "  sbatch --export=EPOCHS=100,DEPTHS=\"1 2 4 8\",LAMBDA_REG=0.2 scripts/run_depth_experiment.sbatch"
        echo ""
        echo "  # Use different partition:"
        echo "  sbatch --partition=gpuA100x4 scripts/run_depth_experiment.sbatch"
        echo ""
        echo "  # Use H200 GPUs for faster training:"
        echo "  sbatch --partition=gpuH200x8 scripts/run_depth_experiment_shd.sbatch"
        ;;
    status)
        echo "Your current jobs:"
        squeue -u $USER
        ;;
    *)
        echo "Usage: $0 {test|synthetic|shd|custom|status}"
        echo ""
        echo "  test      - Quick test on interactive partition (~5 min)"
        echo "  synthetic - Full experiment with synthetic data (~2 hours)"
        echo "  shd       - Full experiment with SHD dataset (~4 hours)"
        echo "  custom    - Show custom submission examples"
        echo "  status    - Show your current jobs"
        exit 1
        ;;
esac