# Install build dependencies for dora-zed on Windows x64.
# Skips any component that is already installed.
# Run from an elevated PowerShell prompt.
#
# Requirements before running:
#   - CUDA Toolkit 12.x: https://developer.nvidia.com/cuda-downloads
#   - CMake 3.21+:        https://cmake.org/download/
#   - Rust toolchain:     https://rustup.rs

$ErrorActionPreference = "Stop"

$ZED_INSTALL_DIR = "C:\Program Files (x86)\ZED SDK"
$ORT_VERSION     = "1.24.4"
$ORT_NAME        = "onnxruntime-win-x64-gpu_cuda13-$ORT_VERSION"
$ORT_URL         = "https://github.com/microsoft/onnxruntime/releases/download/v$ORT_VERSION/$ORT_NAME.zip"
$ORT_DEST        = "C:\onnxruntime"

# --- ZED SDK ---
if (Test-Path $ZED_INSTALL_DIR) {
    Write-Host ">>> ZED SDK already installed at '$ZED_INSTALL_DIR' — skipping."
} else {
    Write-Host ">>> ZED SDK: download the installer from the release page and run it."
    Write-Host "    https://www.stereolabs.com/developers/release"
    Write-Host "    After install, ZED SDK will be at '$ZED_INSTALL_DIR'."
    Write-Host ""
}

# --- ONNX Runtime ---
if (Test-Path "$ORT_DEST\lib\onnxruntime.dll") {
    Write-Host ">>> ONNX Runtime already installed at '$ORT_DEST' — skipping."
} else {
    Write-Host ">>> Downloading ONNX Runtime $ORT_VERSION GPU (CUDA 13)..."
    $tmp = "$env:TEMP\$ORT_NAME.zip"
    Invoke-WebRequest -Uri $ORT_URL -OutFile $tmp

    Write-Host ">>> Extracting to $ORT_DEST..."
    Expand-Archive -Path $tmp -DestinationPath $env:TEMP -Force
    if (Test-Path $ORT_DEST) { Remove-Item $ORT_DEST -Recurse -Force }
    Move-Item "$env:TEMP\$ORT_NAME" $ORT_DEST
    Remove-Item $tmp

    Write-Host ">>> ONNX Runtime installed to $ORT_DEST"
    Write-Host "    Pass it to CMake: cmake .. -DCMAKE_PREFIX_PATH=C:\onnxruntime"
}

Write-Host ""
Write-Host ">>> Done."