#!/usr/bin/env bash
# Recreate the C-JEPA expected layout (pendulum_train/pendulum_val) for extract_four.py.
# Usage: bash prepare_cjepa_root.sh [dataset_root]
set -e
ROOT=${1:-/home/user/pendulum3k/dataset}
mkdir -p "$ROOT/cjepa_data_root/pendulum_train/videos" "$ROOT/cjepa_data_root/pendulum_val/videos"
ln -sf "$ROOT"/videos/train/*.mp4 "$ROOT/cjepa_data_root/pendulum_train/videos/"
ln -sf "$ROOT"/videos/val/*.mp4   "$ROOT/cjepa_data_root/pendulum_val/videos/"
echo "linked $(ls "$ROOT/cjepa_data_root/pendulum_train/videos" | wc -l) train + \
$(ls "$ROOT/cjepa_data_root/pendulum_val/videos" | wc -l) val videos into $ROOT/cjepa_data_root"