setup_spot.bash

# Run this script by source setup_movo.bash
if [[ ! $PWD = *robotdev ]]; then
    echo "You must be in the root directory of the robotdev repository."
    return 1
else
    . "./tools.sh"
fi
repo_root=$PWD
export REPO_ROOT=${repo_root}

# Path to Spot workspace, relative to repository root;
# No begin or trailing slash.
SPOT_PATH="spot"
SPOT_ROS_PATH="spot/ros_ws"  # path to spot ros workspace

# Set the ID of the Spot you are working on
# Required input from user.
if [ "$#" -ne 1 ]; then
    echo -e "Usage: setup_spot.bash <spot_id>"
    return 1
fi

SPOT_ID=$1
# Load the spot config (for this SPOT_ID)
source $repo_root/.spot_config


#------------- FUNCTIONS  ----------------
# Always assume at the start of a function,
# or any if clause, the working directory is
# the root directory of the repository.
# Detect your Spot connection.
function detect_spot_connection
{
    # Detects the spot connection by pinging.
    # Sets two variables, 'spot_conn' and 'spot_ip'
    echo -e "Pinging Spot WiFi IP $SPOT_WIFI_IP..."
    if ping_success $SPOT_WIFI_IP; then
        echo -e "OK"
        spot_conn="spot wifi"
        spot_ip=$SPOT_WIFI_IP
        true && return
    fi

    echo -e "Pinging Spot Ethernet IP $SPOT_ETH_IP..."
    if ping_success $SPOT_ETH_IP; then
        echo -e "OK"
        spot_conn="ethernet"
        spot_ip=$SPOT_ETH_IP
        true && return
    fi

    echo -e "Pinging Spot IP as a client on WiFi $SPOT_WIFI_CLIENT_IP..."
    if ping_success $SPOT_WIFI_CLIENT_IP; then
        echo -e "OK"
        spot_conn="client"
        spot_ip=$SPOT_WIFI_CLIENT_IP
        true && return
    fi

    echo "Cannot connect to Spot"
    spot_conn=""
    spot_ip=""
    false
}

SPOT_ADDITIONAL_BUILD_OPTIONS=""
SPOT_CMAKE_BUILD_TYPE="Release"
function build_spot
{
    cd $repo_root/${SPOT_ROS_PATH}

    if catkin_make\
        --cmake-args\
        -DCMAKE_BUILD_TYPE=${SPOT_CMAKE_BUILD_TYPE}\
        -DPYTHON_EXECUTABLE=/usr/bin/python3\
        -DPYTHON_INCLUDE_DIR=/usr/include/python3.8\
        -DPYTHON_LIBRARY=/usr/lib/x86_64-linux-gnu/libpython3.8.so\
        ${SPOT_ADDITIONAL_BUILD_OPTIONS}\
        $1; then
        echo "SPOT SETUP DONE." >> src/.DONE_SETUP
    else
        rm src/.DONE_SETUP
    fi
}

function ping_spot
{
    if [ -z $SPOT_IP ]; then
        echo -e "It appears that Spot is not connected"
    else
        ping $SPOT_IP
    fi
}

function install_rtabmap_ros_from_source
{
    cd $repo_root/${SPOT_ROS_PATH}/src
    if [ ! -d "rtabmap_ros" ]; then
        git clone -b noetic-devel https://github.com/introlab/rtabmap_ros.git
    fi
    SPOT_ADDITIONAL_BUILD_OPTIONS="-DRTABMAP_SYNC_MULTI_RGBD=ON"
    cd $repo_root
}

# Add a few alias for pinging spot.
#------------- Main Logic  ----------------

# We have only tested Spot stack with Ubuntu 20.04.
if ! ubuntu_version_equal 20.04; then
    echo "SPOT development requires Ubuntu 20.04 and ROS noetic. Abort."
    return 1
fi

# need to use ros for the following commands
if ! useros; then
    echo "Cannot use ROS. Abort."
    exit 1
fi  # so that catkin_make is available for build_ros_ws

# Creates spot workspace.
# create the spot workspace directory
if [ ! -d "${SPOT_ROS_PATH}/src" ]; then
    mkdir -p ${SPOT_ROS_PATH}/src
fi

# create a dedicated virtualenv for spot workspace
if [ ! -d "${SPOT_PATH}/venv/spot" ]; then
    cd ${SPOT_PATH}/
    virtualenv -p python3 venv/spot
    cd ..
fi

# activate virtualenv; Note that this is the only
# functionality of this script if spot has been setup
# before.
source ${SPOT_PATH}/venv/spot/bin/activate

if first_time_build $SPOT_ROS_PATH; then
    pip uninstall -y em
    pip install empy catkin-pkg rospkg defusedxml
    pip install pyqt5
    pip install PySide2
    pip install bosdyn-client==3.2.1
    pip install bosdyn-mission==3.2.1
    pip install bosdyn-api==3.2.1
    pip install bosdyn-core==3.2.1
    pip install bosdyn-choreography-client==3.2.1
    pip install rosdep
    # other necessary packages
    pip install numpy
    pip install pydot
    pip install graphviz
    pip install opencv-python
    pip install pandas
    pip install open3d

    # rosdep install dependencies
    rosdep update
    rosdep install --from-paths src --ignore-src -y

    # install pykdl, needed by tf2_geometry_msgs
    sudo apt-get install -y python3-pykdl

    # other ROS utlities/packages
    sudo apt-get install -y ros-noetic-rqt-graph
    sudo apt-get install -y ros-noetic-rqt-tf-tree
    sudo apt-get install -y ros-noetic-navigation
    sudo apt-get install -y ros-noetic-gmapping
    sudo apt-get install -y ros-noetic-kdl-parser-py
    sudo apt-get install -y ros-noetic-fiducials
    sudo apt-get install -y ros-noetic-find-object-2d
    sudo apt-get install -y ros-noetic-apriltag-ros
    sudo apt-get install -y ros-noetic-rtabmap
    sudo apt-get install -y ros-noetic-ros-numpy

    # Mapping library
    install_rtabmap_ros_from_source
    sudo apt-get install -y ros-noetic-octomap-rviz-plugins

    # Uninstall PyQt5 and PyQt5-sip in pip, so that
    # rqt_* GUI software work.
    # reference: https://github.com/ros-visualization/rqt_graph/issues/51#issuecomment-782062642
    pip uninstall -y PyQt5
    pip uninstall -y PyQt5-sip

    # Downgrade yaml so that rqt_* stuff runs
    # reference: https://stackoverflow.com/a/69565230/2893053
    !pip install pyyaml==5.4.1

    # PCL
    sudo apt-get install python3-pcl pcl-tools

    # Octomap
    sudo apt-get install ros-noetic-octomap
    sudo apt-get install ros-noetic-octomap-mapping
    sudo apt-get install ros-noetic-octomap-server
    sudo apt-get install ros-noetic-octomap-rviz-plugins

    # scikit-iamge
    pip install -U scikit-image

    # vision-msgs
    sudo apt-get install ros-noetic-vision-msgs
fi

# catkin make and end.
if first_time_build $SPOT_ROS_PATH; then
    build_spot
    echo -e "alias dospot='cd ~/repo/robotdev/; source setup_spot.bash'" >> ~/.bashrc
else
    echo -e "If you want to build the spot project, run 'build_spot'"
fi

export ROS_PACKAGE_PATH=$repo_root/${SPOT_ROS_PATH}/src/:${ROS_PACKAGE_PATH}
export PYTHONPATH=""
source $repo_root/${SPOT_ROS_PATH}/devel/setup.bash
# We'd like to use packages in the virtualenv, what's already on /usr/lib,
# and in the workspace (done by above step). NOTE: Using /usr/lib is
# necessary so that PyKDL can be imported (it could only be installed
# via sudo apt-get install python3-pykdl, for some unknown reason).
export PYTHONPATH="$repo_root/${SPOT_PATH}/venv/spot/lib/python3.8/site-packages:${PYTHONPATH}:/usr/lib/python3/dist-packages"
if confirm "Are you working on the real robot ?"; then
    # Check if the environment variable SPOT_IP is set.
    # If not, then try to detect spot connection and set it.
    if [ -z $SPOT_IP ]; then
       if detect_spot_connection; then
           export SPOT_IP=${spot_ip}
           export SPOT_CONN=${spot_conn}
       fi
    fi

    # If Spot is connected, then SPOT_IP should be set.
    if [ -z $SPOT_IP ]; then
        echo -e "Unable to connect to spot."
    else
        if ping_success $SPOT_IP; then
            echo -e "Spot connected! IP: ${SPOT_IP}; Method: ${SPOT_CONN}"
        else
            echo -e "Spot connection lost."
            export SPOT_IP=""
            export SPOT_CONN=""
        fi
    fi

    # get spot password variables
    source $repo_root/.spot_config
fi
cd $repo_root