#!/bin/bash

# --- 配置 ---
folder_path="scripts/twitter_simulation/align_with_real_world/big-scale-ablation"
PYTHON_SCRIPT="scripts/twitter_simulation/align_with_real_world/twitter_simulation_large.py"

# --- 执行循环 ---
echo "开始批量运行模拟..."

# 检查文件夹是否存在
if [ ! -d "$folder_path" ]; then
    echo "错误：文件夹 $folder_path 不存在"
    exit 1
fi

# 计算总配置文件数量
total_configs=$(find "$folder_path" -name "*.yaml" | wc -l)
current_run=1

echo "发现 $total_configs 个配置文件"

# 读取文件夹中的所有YAML文件
for file in "$folder_path"/*.yaml; do
    # 检查文件是否存在（防止通配符没有匹配到文件的情况）
    if [ ! -f "$file" ]; then
        echo "警告：没有找到YAML文件在 $folder_path"
        continue
    fi
    
    # 提取文件名（不包括后缀）
    filename=$(basename "$file")
    topicname="${filename%.*}"

    echo "----------------------------------------"
    echo "开始运行第 $current_run / $total_configs 次模拟：$topicname"
    echo "配置文件：$file"
    echo "----------------------------------------"

    # 运行python脚本
    srun -p AI4Good_L1_p --gpus-per-task=1 -c 26 python "$PYTHON_SCRIPT" --config_path "$file"
    
    # 检查是否成功执行
    if [ $? -eq 0 ]; then
        echo "✅ 成功完成：$topicname"
    else
        echo "❌ 执行失败：$topicname"
    fi

    echo "----------------------------------------"
    echo "完成运行第 $current_run / $total_configs 次模拟：$topicname"
    echo "----------------------------------------"
    
    # 增加运行计数器
    current_run=$((current_run + 1))
done

echo "所有模拟运行完成。"
