# 322K 全自动: 等生成完 → 构建 → EXP_CUT → TRANS → 训练
# 用法: 生成进程跑完后执行此脚本
$ErrorActionPreference = "Continue"

$PY = "C:/Program Files/Python313/python.exe"
$ICE = "C:\Users\Kirl\Desktop\ICEv2\ICE_Dataset"

Write-Host "===== 322K Pipeline =====" -ForegroundColor Cyan
Write-Host "Waiting for SPE + Random generation to complete..."
Write-Host "SPE dir: $ICE\CCPD_SPE\ccpd_base"
Write-Host "Random dir: $ICE\CCPD_Random_Blue150K_Green100K"

# Wait for SPE (9 types x 8000)
$spe_types = @('police','army','gangao','shiguan_old','shiguan_new','lingguan_old','lingguan_new','coach','new_energy')
do {
    $done = $true
    foreach ($t in $spe_types) {
        $c = @(Get-ChildItem "$ICE\CCPD_SPE\ccpd_base\$t" -Filter *.jpg -ErrorAction SilentlyContinue).Count
        if ($c -lt 8000) { $done = $false; break }
    }
    if (-not $done) { Write-Host "  SPE not ready, waiting 60s..."; Start-Sleep 60 }
} while (-not $done)
Write-Host "SPE: OK" -ForegroundColor Green

# Wait for Random
do {
    $b = @(Get-ChildItem "$ICE\CCPD_Random_Blue150K_Green100K\blue" -Filter *.jpg -ErrorAction SilentlyContinue).Count
    $g = @(Get-ChildItem "$ICE\CCPD_Random_Blue150K_Green100K\green" -Filter *.jpg -ErrorAction SilentlyContinue).Count
    if ($b -lt 150000 -or $g -lt 100000) {
        Write-Host "  Random: blue=$b green=$g, waiting 60s..."
        Start-Sleep 60
    } else { break }
} while ($true)
Write-Host "Random: blue=$b green=$g OK" -ForegroundColor Green

# Now run the pipeline
Write-Host "`nAll data ready. Starting pipeline..." -ForegroundColor Cyan
& "C:/Program Files/Python313/python.exe" "$PSScriptRoot\pipeline_322k.py"