#!/bin/bash

export CUDA_HOME=/usr/local/cuda-11.8
export PATH=${CUDA_HOME}/bin:${PATH}
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}

# Exit on error
# set -e # Commented out to allow checking specific error codes if needed

# Set CUDA environment (this part will likely be overridden or become redundant)
if [ -d "/usr/local/cuda" ]; then # This might point to a different version
    export CUDA_HOME_DETECTED=/usr/local/cuda 
    echo "System default CUDA found at $CUDA_HOME_DETECTED (might be different from enforced $CUDA_HOME)"
elif [ -d "/usr/lib/cuda" ]; then
    export CUDA_HOME=/usr/lib/cuda
    echo "CUDA environment set to $CUDA_HOME"
else
    echo "CUDA not found. The project requires CUDA to run."
    echo "Please install CUDA and try again."
    exit 1
fi

# Set LibTorch library path
if [ -d "$HOME/libtorch" ]; then
    export LD_LIBRARY_PATH=$HOME/libtorch/lib:$LD_LIBRARY_PATH
    echo "LibTorch library path set to $HOME/libtorch/lib"
elif [ -d "/usr/local/libtorch" ]; then
    export LD_LIBRARY_PATH=/usr/local/libtorch/lib:$LD_LIBRARY_PATH
    echo "LibTorch library path set to /usr/local/libtorch/lib"
fi

# Run the demo
echo "Starting tracking demo..."
./bin/tracking_demo