#!/bin/bash
# Pipeline LiDAR Archéologique
# Utilisation: ./run.sh [options]
#
# Options:
#   -r RESOLUTION   Résolution en m/px (défaut: 0.5)
#   -w WORKERS      Nombre de workers parallèles (défaut: 1)
#   -g              Activer l'accélération GPU
#   -v              Mode verbeux (timestamps + niveaux)
#   --debug         Mode debug (détails internes fichier:ligne)
#   -f / --force    Régénérer tous les fichiers même si existants
#   --keep-tif       Conserver les fichiers TIFF intermédiaires
#   --no-viewer     Ne pas générer le viewer web (COGs + HTML)
#   --force-classification
#                   Reclassifier le sol même si le fichier .las existe déjà
#   --ground-classification {auto,smrf,pmf,csf}
#                   Méthode de classification du sol (défaut: auto)
#   --file NOM...   Traiter un ou plusieurs fichiers LAZ spécifiques
#   --test          Exécuter les tests unitaires
#   serve           Démarrer le serveur cartographique
#   -h              Afficher l'aide complète

set -e

SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
INPUT_DIR="${SCRIPT_DIR}/input"
OUTPUT_DIR="${SCRIPT_DIR}/output"
IMAGE_NAME="lidar-lidar"

# Serve command — start the web map server
if [ "$1" = "serve" ]; then
    # Build l'image si elle n'existe pas
    if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then
        echo "Build de l'image Docker..."
        docker build -t "$IMAGE_NAME" "$SCRIPT_DIR"
    fi
    mkdir -p "$OUTPUT_DIR"
    echo "============================================"
    echo " Serveur cartographique LiDAR"
    echo "============================================"
    echo "  Viewer:  http://localhost:8000/viewer"
    echo "  TiTiler: http://localhost:8000/cog/"
    echo "============================================"
    docker run --rm -p 8000:8000 \
        -v "${OUTPUT_DIR}:/data/output" \
        "$IMAGE_NAME" \
        python3 -m lidar_pipeline.server /data/output
    exit 0
fi

# Afficher l'aide si aucun argument
if [ $# -eq 0 ]; then
    echo "Pipeline LiDAR Archéologique"
    echo ""
    echo "Usage: $0 [options]"
    echo "  $0 serve                                  # Démarrer le serveur cartographique"
    echo ""
    echo "Options:"
    echo "  -r RESOLUTION   Résolution en m/px (défaut: 0.5)"
    echo "  -w WORKERS      Nombre de workers CPU parallèles (défaut: 1)"
    echo "  -g              Activer l'accélération GPU NVIDIA"
    echo "  -v              Mode verbeux (timestamps + niveaux)"
    echo "  --debug         Mode debug (détails internes fichier:ligne)"
    echo "  -f / --force    Régénérer tous les fichiers même si les WebP existent"
    echo "  --force-classification"
    echo "                  Reclassifier le sol même si le fichier .las existe"
    echo "  --keep-tif      Conserver les fichiers TIFF intermédiaires"
    echo "  --no-viewer     Ne pas générer le viewer web"
    echo "  --ground-classification {auto,smrf,pmf,csf}"
    echo "                  Méthode de classification du sol (défaut: auto)"
    echo "  --file NOM...   Traiter un ou plusieurs fichiers LAZ (nom complet sans .laz/.las)"
    echo "  --test          Exécuter les tests unitaires"
    echo "  serve           Démarrer le serveur cartographique"
    echo "  -h              Afficher cette aide"
    echo ""
    echo "Exemples:"
    echo "  $0 -g                                    # GPU, auto"
    echo "  $0 -g -w 4                               # GPU + 4 workers"
    echo "  $0 -g -v                                 # GPU + verbeux"
    echo "  $0 -g -r 0.2                             # Haute résolution"
    echo "  $0 -g --force                            # Tout régénérer (WebP + classification)"
    echo "  $0 -g --force-classification             # Reclassifier le sol seulement"
    echo "  $0 -g --ground-classification pmf        # Forcer PMF"
    echo "  $0 -g --file LHD_...IGN69.copc           # Un fichier"
    echo "  $0 serve                                  # Démarrer le serveur web"
    exit 0
fi

RESOLUTION=0.5
WORKERS=1
GPU_FLAG=""
VERBOSE_FLAG=""
FORCE_FLAG=""
FILE_ARGS=""
GROUND_METHOD=""
FORCE_CLASSIFY_FLAG=""
KEEP_TIF_FLAG=""
NO_VIEWER_FLAG=""

# Parse arguments manually (more robust than getopts for mixed short/long options)
while [ $# -gt 0 ]; do
    case "$1" in
        -r) RESOLUTION="$2"; shift 2 ;;
        -w) WORKERS="$2"; shift 2 ;;
        -g) GPU_FLAG="--gpus all"; shift ;;
        -v) VERBOSE_FLAG="-v"; shift ;;
        -f) FORCE_FLAG="--force"; shift ;;
        --debug) VERBOSE_FLAG="--debug"; shift ;;
        --force) FORCE_FLAG="--force"; shift ;;
        --force-classification) FORCE_CLASSIFY_FLAG="--force-classification"; shift ;;
        --keep-tif) KEEP_TIF_FLAG="--keep-tif"; shift ;;
        --no-viewer) NO_VIEWER_FLAG="--no-viewer"; shift ;;
        --ground-classification) GROUND_METHOD="$2"; shift 2 ;;
        --ground-classification=*) GROUND_METHOD="${1#--ground-classification=}"; shift ;;
        --file) shift; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do FILE_ARGS="$FILE_ARGS $1"; shift; done ;;
        --test) ;;  # Handled below
        -h|--help|-help)
            echo "Pipeline LiDAR Archéologique"
            echo ""
            echo "Usage: $0 [options]"
            echo "  $0 serve                                  # Démarrer le serveur cartographique"
            echo ""
            echo "Options:"
            echo "  -r RESOLUTION   Résolution en m/px (défaut: 0.5)"
            echo "  -w WORKERS      Nombre de workers CPU parallèles (défaut: 1)"
            echo "  -g              Activer l'accélération GPU NVIDIA"
            echo "  -v              Mode verbeux (timestamps + niveaux)"
            echo "  --debug         Mode debug (détails internes fichier:ligne)"
            echo "  -f / --force    Régénérer tous les fichiers même si les WebP existent"
            echo "  --force-classification"
            echo "                  Reclassifier le sol même si le fichier .las existe"
            echo "  --keep-tif      Conserver les fichiers TIFF intermédiaires"
            echo "  --no-viewer     Ne pas générer le viewer web"
            echo "  --ground-classification {auto,smrf,pmf,csf}"
            echo "                  Méthode de classification du sol (défaut: auto)"
            echo "  --file NOM...   Traiter un ou plusieurs fichiers LAZ (nom complet sans .laz/.las)"
            echo "  --test          Exécuter les tests unitaires"
            echo "  serve           Démarrer le serveur cartographique"
            echo "  -h              Afficher cette aide"
            echo ""
            echo "Exemples:"
            echo "  $0 -g                                    # GPU, auto"
            echo "  $0 -g -w 4                               # GPU + 4 workers"
            echo "  $0 -g -v                                 # GPU + verbeux"
            echo "  $0 -g -r 0.2                             # Haute résolution"
            echo "  $0 -g --force                            # Tout régénérer (WebP + classification)"
            echo "  $0 -g --force-classification             # Reclassifier le sol seulement"
            echo "  $0 -g --ground-classification pmf        # Forcer PMF"
            echo "  $0 -g --file LHD_...IGN69.copc           # Un fichier"
            echo "  $0 serve                                  # Démarrer le serveur web"
            exit 0
            ;;
        *) echo "Option invalide: $1" >&2; exit 1 ;;
    esac
done

# Trim FILE_ARGS whitespace
FILE_ARGS=$(echo "$FILE_ARGS" | xargs)

# Check for --test flag first
if [[ " $* " == *" --test "* ]]; then
    # Build l'image si elle n'existe pas
    if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then
        echo "Build de l'image Docker..."
        docker build -t "$IMAGE_NAME" "$SCRIPT_DIR"
    fi
    echo "============================================"
    echo " Tests unitaires LiDAR Pipeline"
    echo "============================================"
    docker run --rm $GPU_FLAG \
        "$IMAGE_NAME" \
        python3 -m pytest -v --pyargs lidar_pipeline.tests
    exit $?
fi

# Build l'image si elle n'existe pas
if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then
    echo "Build de l'image Docker..."
    docker build -t "$IMAGE_NAME" "$SCRIPT_DIR"
fi

# Créer les répertoires s'ils n'existent pas
mkdir -p "$INPUT_DIR" "$OUTPUT_DIR"

# Lancer le pipeline
echo "============================================"
echo " Pipeline LiDAR Archéologique"
echo "============================================"
echo "  Résolution : ${RESOLUTION}m/px"
echo "  Workers    : ${WORKERS}"
echo "  GPU        : $([ -n "$GPU_FLAG" ] && echo 'OUI' || echo 'non')"
echo "  Verbeux    : $([ -n "$VERBOSE_FLAG" ] && echo 'OUI' || echo 'non')"
echo "  Force      : $([ -n "$FORCE_FLAG" ] && echo 'OUI' || echo 'non')"
echo "  Force classif.: $([ -n "$FORCE_CLASSIFY_FLAG" ] && echo 'OUI' || echo 'non')"
echo "  Keep TIFF    : $([ -n "$KEEP_TIF_FLAG" ] && echo 'OUI' || echo 'non')"
echo "  Viewer web   : $([ -n "$NO_VIEWER_FLAG" ] && echo 'non' || echo 'OUI')"
echo "  Classification sol : $([ -n "$GROUND_METHOD" ] && echo "$GROUND_METHOD" || echo 'auto')"
if [ -n "$FILE_ARGS" ]; then
    echo "  Fichiers   :${FILE_ARGS}"
fi
echo "============================================"

CMD_ARGS="-o /data/output -r $RESOLUTION -w $WORKERS $VERBOSE_FLAG $FORCE_FLAG $FORCE_CLASSIFY_FLAG $KEEP_TIF_FLAG $NO_VIEWER_FLAG"
if [ -n "$GROUND_METHOD" ]; then
    CMD_ARGS="$CMD_ARGS --ground-classification $GROUND_METHOD"
fi
if [ -n "$FILE_ARGS" ]; then
    CMD_ARGS="$CMD_ARGS --file $FILE_ARGS"
fi

docker run --rm $GPU_FLAG \
    --user 1000:1000 \
    -v "${INPUT_DIR}:/data/input:ro" \
    -v "${OUTPUT_DIR}:/data/output" \
    "$IMAGE_NAME" \
    python3 -m lidar_pipeline /data/input \
    $CMD_ARGS