#!/bin/bash

# Submit DPPO jobs to SLURM
# Usage: ./submit_job.sh [mode] [task] [experiment_type]

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
cd "$SCRIPT_DIR"

# Default values
MODE=${1:-dev}  # dev, gym, robomimic, d3il, furniture
TASK=${2:-hopper}  # Task specific to each mode
EXPERIMENT_TYPE=${3:-pretrain}  # pretrain or finetune

echo "Submitting DPPO job..."
echo "Mode: $MODE"
echo "Task: $TASK"
echo "Experiment type: $EXPERIMENT_TYPE"

case $MODE in
    dev)
        echo "Submitting development test job..."
        sbatch slurm/run_dppo_dev.sh
        ;;
    gym)
        echo "Submitting Gym job..."
        sbatch slurm/run_dppo_gym.sh "$TASK" "$EXPERIMENT_TYPE"
        ;;
    *)
        echo "Unknown mode: $MODE"
        echo "Supported modes: dev, gym"
        echo ""
        echo "Usage examples:"
        echo "  ./submit_job.sh dev                    # Run quick dev test"
        echo "  ./submit_job.sh gym hopper pretrain    # Pre-train on Gym Hopper"
        echo "  ./submit_job.sh gym walker2d finetune  # Fine-tune on Gym Walker2D"
        exit 1
        ;;
esac

echo ""
echo "Job submitted! Check status with:"
echo "  squeue -u $USER"
echo ""
echo "Check logs in: logs/ directory"