#!/bin/bash

echo "Setting up NVIDIA GPU for Gazebo..."

if ! command -v nvidia-smi &> /dev/null; then
    echo "NVIDIA driver not found. Skipping GPU setup."
    exit 0
fi

echo "NVIDIA GPU detected:"
nvidia-smi --query-gpu=name,driver_version --format=csv,noheader

export __NV_PRIME_RENDER_OFFLOAD=1
export __GLX_VENDOR_LIBRARY_NAME=nvidia
export VK_ICD_FILENAMES=/usr/share/vulkan/icd.d/nvidia_icd.json

if [ -f /usr/lib/x86_64-linux-gnu/libEGL_nvidia.so.0 ]; then
    export __EGL_VENDOR_LIBRARY_FILENAMES=/usr/share/glvnd/egl_vendor.d/10_nvidia.json
fi

echo "NVIDIA GPU environment configured."
echo ""
echo "Add these to your ~/.bashrc for permanent setup:"
echo "  export __NV_PRIME_RENDER_OFFLOAD=1"
echo "  export __GLX_VENDOR_LIBRARY_NAME=nvidia"