#!/usr/bin/env bash
# Download script for UZH FPV DAVIS rosbags with ground truth
# Skips files that already exist in the current directory.
# Uses the final download.ifi.uzh.ch URLs directly (avoids 301 redirect chain).
set -euo pipefail

BASE="https://download.ifi.uzh.ch/rpg/web/datasets/uzh-fpv-newer-versions/v3"

URLS=(
    # Indoor forward facing
    "${BASE}/indoor_forward_3_davis_with_gt.bag"
    "${BASE}/indoor_forward_5_davis_with_gt.bag"
    "${BASE}/indoor_forward_6_davis_with_gt.bag"
    "${BASE}/indoor_forward_9_davis_with_gt.bag"
    "${BASE}/indoor_forward_10_davis_with_gt.bag"

    # Indoor 45 degree downward
    "${BASE}/indoor_45_2_davis_with_gt.bag"
    "${BASE}/indoor_45_4_davis_with_gt.bag"
    "${BASE}/indoor_45_9_davis_with_gt.bag"
    "${BASE}/indoor_45_12_davis_with_gt.bag"
    "${BASE}/indoor_45_13_davis_with_gt.bag"
    "${BASE}/indoor_45_14_davis_with_gt.bag"

    # Outdoor forward facing
    "${BASE}/outdoor_forward_1_davis_with_gt.bag"
    "${BASE}/outdoor_forward_5_davis_with_gt.bag"

    # Outdoor 45 degree downward
    "${BASE}/outdoor_45_1_davis_with_gt.bag"
)

for url in "${URLS[@]}"; do
    filename=$(basename "$url")
    if [ -f "$filename" ]; then
        echo "SKIP: $filename already exists"
    else
        echo "DOWNLOAD: $filename"
        wget --continue --show-progress "$url" -O "$filename"
    fi
done

echo ""
echo "Done. All DAVIS with-GT rosbags downloaded."