#!/bin/bash
# Quick check for 6-task h=16 generation progress

echo "═══════════════════════════════════════════════════════════"
echo "  6-Task h=16 Collection Generation Status"
echo "═══════════════════════════════════════════════════════════"
echo ""

if [ -z "$1" ]; then
  echo "Usage: $0 <job_id>"
  echo ""
  echo "Or find latest job:"
  sacct -u $USER --format=JobID,JobName,State,Elapsed -S today | grep "dovla_6task_h16" | tail -7
  exit 0
fi

JOB=$1

echo "Job Status:"
sacct -j $JOB --format=JobID,JobName,State,Elapsed,ExitCode | head -10
echo ""

echo "Generated Task Directories:"
ls -d /scratch/$USER/dovla/experiments/six_task_h16_collection/*-v1 2>/dev/null || echo "  (none yet)"
echo ""

echo "Checking each task's progress:"
for task in PickCube-v1 PushCube-v1 PullCube-v1 StackCube-v1 LiftPegUpright-v1 PegInsertionSide-v1; do
  dir="/scratch/$USER/dovla/experiments/six_task_h16_collection/$task"
  if [ -d "$dir" ]; then
    if [ -f "$dir/manifest.json" ]; then
      groups=$(python3 -c "import json; print(json.load(open('$dir/manifest.json'))['group_count'])" 2>/dev/null || echo "?")
      echo "  ✅ $task: $groups groups"
    else
      echo "  🔄 $task: generating..."
    fi
  else
    echo "  ⏳ $task: pending"
  fi
done
echo ""

echo "═══════════════════════════════════════════════════════════"
echo "Expected:"
echo "  PickCube: 1000 groups (~2-3 hours)"
echo "  Others:   500 groups each (~1-2 hours each)"
echo "  Total:    ~4-8 hours (parallel)"
echo "═══════════════════════════════════════════════════════════"