#!/bin/bash
# Setup script for BA pipeline dependencies

set -e

echo "Setting up BA pipeline dependencies..."

# Check if COLMAP is installed
if ! command -v colmap &> /dev/null; then
    echo "ERROR: COLMAP is not installed."
    echo "Please install COLMAP first:"
    echo "  macOS: brew install colmap"
    echo "  Ubuntu: sudo apt-get install colmap"
    echo "  Or build from source: https://colmap.github.io/install.html"
    exit 1
fi

echo "✓ COLMAP found"

# Install Python dependencies
echo "Installing Python dependencies..."

pip install pycolmap

# Install hloc
if [ ! -d "hloc" ]; then
    echo "Cloning hloc..."
    git clone https://github.com/cvg/Hierarchical-Localization.git hloc
    cd hloc
    pip install -e .
    cd ..
else
    echo "✓ hloc directory exists, skipping clone"
fi

# Install LightGlue
echo "Installing LightGlue..."
pip install git+https://github.com/cvg/LightGlue.git

echo ""
echo "✓ BA pipeline setup complete!"
echo ""
echo "To verify installation, run:"
echo "  python -c 'import pycolmap; from hloc import extract_features; print(\"OK\")'"