env.sh

#!/bin/bash

# ----------------------------------------------------------------------
#
#   This script configures a terminal environment 
#   to run properly in interaction with this repository
#
# ----------------------------------------------------------------------

WORKSPACE_ROOT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"

source ~/.bashrc

# ROS2 specific environment
if [ -d /opt/ros ]; then
    source /opt/ros/humble/setup.bash
fi
if [ -d /usr/share/colcon_cd ]; then
    source /usr/share/colcon_cd/function/colcon_cd.sh
    export _colcon_cd_root=/opt/ros/humble/
fi
if [ -d /usr/share/colcon_argcomplete ]; then
    source /usr/share/colcon_argcomplete/hook/colcon-argcomplete.bash
fi
if [ -f "${WORKSPACE_ROOT_DIR}/install/local_setup.bash" ]; then 
    source "${WORKSPACE_ROOT_DIR}/install/local_setup.bash"
fi

#
# Add local python packages to PYTHONPATH
#

PYTHON_VERSION="python3.10"  # Replace with your Python version

# Check if the workspace is using merge-install (look for 'install/local/lib/<python_version>/dist-packages')
MERGE_INSTALL_PATH="$WORKSPACE_ROOT_DIR/install/local/lib/$PYTHON_VERSION/dist-packages"

if [ -d "$MERGE_INSTALL_PATH" ]; then
    # Merge-install setup detected
    # echo "Merge-install setup detected."
    export PYTHONPATH="$MERGE_INSTALL_PATH:$PYTHONPATH"
else
    # Non-merge-install setup detected
    # echo "Non-merge-install setup detected."

    # Loop through all packages in the workspace src directory
    for PACKAGE in "$WORKSPACE_ROOT_DIR"/src/*; do
        SITE_PACKAGES_PATH="$WORKSPACE_ROOT_DIR/install/$(basename "$PACKAGE")/local/lib/$PYTHON_VERSION/dist-packages"
        if [ -d "$SITE_PACKAGES_PATH" ]; then
            # Append the dist-packages path to PYTHONPATH if it exists
            export PYTHONPATH="$SITE_PACKAGES_PATH:$PYTHONPATH"
        fi
    done
fi

#
# Related programs
#

export PX4_ROS2_ENV_DEFAULT_BASE_DIR="$HOME/Desktop"

# Change these variables if necessary
export PX4_ROS2_ENV_PX4_AUTOPILOT_DIR="${PX4_ROS2_ENV_DEFAULT_BASE_DIR}/PX4-Autopilot"
export PX4_ROS2_ENV_QGROUNDCONTROL_APP="${PX4_ROS2_ENV_DEFAULT_BASE_DIR}/QGroundControl.AppImage"
export PX4_ROS2_ENV_GROOT2_APP="${PX4_ROS2_ENV_DEFAULT_BASE_DIR}/Groot2-v1.6.1-x86_64.AppImage"

if [ -f "$PX4_ROS2_ENV_QGROUNDCONTROL_APP" ] && [ $(stat -c "%a" "$PX4_ROS2_ENV_QGROUNDCONTROL_APP") -ne 755 ]; then
    chmod +x "$PX4_ROS2_ENV_QGROUNDCONTROL_APP"
fi
if [ -f "$PX4_ROS2_ENV_GROOT2_APP" ] && [ $(stat -c "%a" "$PX4_ROS2_ENV_GROOT2_APP") -ne 755 ]; then
    chmod +x "$PX4_ROS2_ENV_GROOT2_APP"
fi