FROM nvidia/cuda:12.4.0-devel-ubuntu22.04

ENV DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/Paris

# Install PDAL and system packages
RUN apt-get update && apt-get install -y --no-install-recommends \
    pdal \
    gdal-bin \
    python3-gdal \
    python3-pip \
    python3-dev \
    build-essential \
    wget \
    && rm -rf /var/lib/apt/lists/*

WORKDIR /data

# Install Python packages via pip
COPY requirements.txt .
RUN pip3 install --no-cache-dir \
    numpy \
    matplotlib \
    whitebox \
    rasterio \
    'laspy[laspy]' \
    scikit-image \
    scikit-learn \
    scipy \
    tqdm

# Install CuPy for GPU acceleration (optional - will fallback to numpy if not available)
RUN pip3 install --no-cache-dir cupy-cuda12x || echo "CuPy not available - GPU acceleration disabled"

# Copy scripts
COPY process_lidar.py /usr/local/bin/
RUN chmod +x /usr/local/bin/process_lidar.py

# Create user with uid/gid 1000:1000 and run as that user
RUN groupadd -g 1000 lidar && \
    useradd -u 1000 -g lidar -m lidar && \
    mkdir -p /data/output /data/input && \
    chown -R lidar:lidar /data /data/output /data/input

USER lidar

VOLUME ["/data"]

CMD ["python3", "/usr/local/bin/process_lidar.py", "/data/input", "-o", "/data/output"]