#!/bin/bash

# 初始化 Conda 环境路径
CONDA_BASE=$(conda info --base)
source "$CONDA_BASE/etc/profile.d/conda.sh"

# 确保主路径在 PYTHONPATH 中
export PYTHONPATH=$PYTHONPATH:/root/autodl-tmp/SplatAtlas

# 定义任务队列：(环境绝对路径 脚本相对路径)
declare -a tasks=(

    "/root/autodl-tmp/envs/gaussianpro scripts/run_gaussianpro_benchmark.py"

    "/root/autodl-tmp/envs/conegs_env scripts/run_conegs_benchmark.py"
    "/root/autodl-tmp/envs/steepgs_env scripts/run_steepgs.py"
    
    "/root/autodl-tmp/envs/absgs_env scripts/run_absgs_benchmark.py"
    "/root/autodl-tmp/envs/reactgs_env scripts/run_reactgs_benchmark.py"
    "/root/autodl-tmp/envs/lapis_gs scripts/run_lapisgs_benchmark.py"
    "/root/autodl-tmp/envs/co_adaptation_3dgs scripts/run_coadaptgs_benchmark.py"
    "/root/autodl-tmp/envs/mini_splatting scripts/run_minisplatting_benchmark.py"
    "/root/autodl-tmp/envs/gs_lpm scripts/run_gslpm.py"
    "/root/autodl-tmp/envs/opti3dgs scripts/run_opti3dgs_benchmark.py"
    "/root/autodl-tmp/envs/ghap_env scripts/run_ghap_benchmark.py"
    "/root/autodl-tmp/envs/3dgs_mcmc scripts/run_3dgsmcmc_benchmark.py"
         # "/root/autodl-tmp/envs/lod_gs scripts/run_lod_gs_benchmark.py"
         
#conegs可能需要重跑，test可能没有被划分          
)

echo "🚀 SplatAtlas: Starting Robust Optimization Pipeline..."

for task in "${tasks[@]}"; do
    # 解析环境路径和脚本
    read -r env_path script_path <<< "$task"
    
    echo "----------------------------------------------------------------"
    echo "🔄 Method: $(basename "$script_path")"
    
    # 1. 激活环境
    conda activate "$env_path"
    
    if [ $? -eq 0 ]; then
        # 2. 环境自动体检：添加了 scikit-image 的安装
        echo "🛠️ Checking requirements in $env_path..."
        # 增加 scikit-image 以解决 ModuleNotFoundError: No module named 'skimage'
        pip install lpips imageio scikit-image -q
        
        # 3. 执行 Benchmark
        echo "✅ Environment ready. Executing..."
        "$CONDA_PREFIX/bin/python" "/root/autodl-tmp/SplatAtlas/$script_path"
        
        if [ $? -eq 0 ]; then
            echo "✨ Finished: $(basename "$script_path")"
        else
            echo "❌ Error: $(basename "$script_path") failed."
        fi
    else
        echo "❌ Failed to activate: $env_path"
    fi
    
    # 释放当前环境
    conda deactivate
done

echo "================================================================"
echo "🏁 Optimization Pipeline completed."