Update Install Scripts 2
This commit is contained in:
@@ -1,6 +1,14 @@
|
||||
#!/bin/bash
|
||||
# Full simulation launch script with ArduPilot SITL
|
||||
# This script launches Gazebo, ArduPilot SITL, and MAVROS
|
||||
# =============================================================================
|
||||
# UAV-UGV Simulation - Run Script
|
||||
# =============================================================================
|
||||
# Launches Gazebo with ArduPilot SITL for GPS-denied navigation testing
|
||||
#
|
||||
# Usage:
|
||||
# ./scripts/run_simulation.sh # Default world
|
||||
# ./scripts/run_simulation.sh --world runway # Runway world
|
||||
# ./scripts/run_simulation.sh --software-render # For WSL/no GPU
|
||||
# =============================================================================
|
||||
|
||||
set -e
|
||||
|
||||
@@ -28,51 +36,32 @@ fi
|
||||
|
||||
# Detect ROS distro
|
||||
ROS_DISTRO=""
|
||||
for distro in humble jazzy iron galactic; do
|
||||
for distro in jazzy humble iron galactic; do
|
||||
if [ -d "/opt/ros/$distro" ]; then
|
||||
ROS_DISTRO="$distro"
|
||||
break
|
||||
fi
|
||||
done
|
||||
|
||||
if [ -z "$ROS_DISTRO" ]; then
|
||||
echo -e "${RED}ERROR: No ROS 2 installation found!${NC}"
|
||||
echo "Please install ROS 2 first. See docs/setup_guide.md"
|
||||
exit 1
|
||||
if [ -n "$ROS_DISTRO" ]; then
|
||||
source /opt/ros/${ROS_DISTRO}/setup.bash
|
||||
echo -e "${GREEN}ROS 2: $ROS_DISTRO${NC}"
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}Using ROS 2: $ROS_DISTRO${NC}"
|
||||
|
||||
# Check ArduPilot installation
|
||||
# ArduPilot paths
|
||||
ARDUPILOT_HOME="${ARDUPILOT_HOME:-$HOME/ardupilot}"
|
||||
if [ ! -d "$ARDUPILOT_HOME" ]; then
|
||||
echo -e "${RED}ERROR: ArduPilot not found at $ARDUPILOT_HOME${NC}"
|
||||
echo ""
|
||||
echo "Please install ArduPilot first:"
|
||||
echo " bash scripts/install_ardupilot.sh"
|
||||
echo ""
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Source environments
|
||||
source /opt/ros/${ROS_DISTRO}/setup.bash
|
||||
|
||||
if [ -f /usr/share/gazebo/setup.bash ]; then
|
||||
source /usr/share/gazebo/setup.bash
|
||||
elif [ -f /usr/share/gazebo-11/setup.bash ]; then
|
||||
source /usr/share/gazebo-11/setup.bash
|
||||
fi
|
||||
ARDUPILOT_GZ="${ARDUPILOT_GZ:-$HOME/ardupilot_gazebo}"
|
||||
|
||||
# Parse arguments
|
||||
WORLD="$PROJECT_DIR/worlds/iris_runway.world"
|
||||
VEHICLE="copter"
|
||||
WORLD_NAME="iris_runway"
|
||||
VEHICLE="ArduCopter"
|
||||
USE_SOFTWARE_RENDER=false
|
||||
INSTANCE=0
|
||||
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case $1 in
|
||||
--world)
|
||||
WORLD="$2"
|
||||
WORLD_NAME="$2"
|
||||
shift 2
|
||||
;;
|
||||
--vehicle)
|
||||
@@ -91,40 +80,58 @@ while [[ $# -gt 0 ]]; do
|
||||
echo "Usage: $0 [OPTIONS]"
|
||||
echo ""
|
||||
echo "Options:"
|
||||
echo " --world FILE World file (default: iris_runway.world)"
|
||||
echo " --vehicle TYPE Vehicle type: copter, rover (default: copter)"
|
||||
echo " --world NAME World name (default: iris_runway)"
|
||||
echo " --vehicle TYPE ArduCopter or Rover (default: ArduCopter)"
|
||||
echo " --software-render Force software rendering (for WSL)"
|
||||
echo " --instance N Vehicle instance number (default: 0)"
|
||||
echo " --instance N SITL instance number (default: 0)"
|
||||
echo " --help Show this help"
|
||||
echo ""
|
||||
echo "Available worlds in ardupilot_gazebo:"
|
||||
ls "$ARDUPILOT_GZ/worlds/"*.sdf 2>/dev/null | xargs -I {} basename {} .sdf || echo " (none found)"
|
||||
exit 0
|
||||
;;
|
||||
*)
|
||||
if [ -f "$1" ]; then
|
||||
WORLD="$1"
|
||||
elif [ -f "$PROJECT_DIR/worlds/$1" ]; then
|
||||
WORLD="$PROJECT_DIR/worlds/$1"
|
||||
fi
|
||||
WORLD_NAME="$1"
|
||||
shift
|
||||
;;
|
||||
esac
|
||||
done
|
||||
|
||||
# Setup Gazebo paths
|
||||
export GAZEBO_MODEL_PATH="$PROJECT_DIR/models:${GAZEBO_MODEL_PATH:-}"
|
||||
export GAZEBO_RESOURCE_PATH="$PROJECT_DIR/worlds:${GAZEBO_RESOURCE_PATH:-}"
|
||||
export GZ_SIM_RESOURCE_PATH="$PROJECT_DIR/models:$PROJECT_DIR/worlds:${GZ_SIM_RESOURCE_PATH:-}"
|
||||
|
||||
if [ -d "$HOME/ardupilot_gazebo" ]; then
|
||||
export GAZEBO_MODEL_PATH="$HOME/ardupilot_gazebo/models:$GAZEBO_MODEL_PATH"
|
||||
export GAZEBO_RESOURCE_PATH="$HOME/ardupilot_gazebo/worlds:$GAZEBO_RESOURCE_PATH"
|
||||
if [ -d "$ARDUPILOT_GZ" ]; then
|
||||
export GZ_SIM_SYSTEM_PLUGIN_PATH="$ARDUPILOT_GZ/build:${GZ_SIM_SYSTEM_PLUGIN_PATH:-}"
|
||||
export GZ_SIM_RESOURCE_PATH="$ARDUPILOT_GZ/models:$ARDUPILOT_GZ/worlds:$GZ_SIM_RESOURCE_PATH"
|
||||
fi
|
||||
|
||||
# Find world file
|
||||
WORLD_FILE=""
|
||||
if [ -f "$PROJECT_DIR/worlds/${WORLD_NAME}.sdf" ]; then
|
||||
WORLD_FILE="$PROJECT_DIR/worlds/${WORLD_NAME}.sdf"
|
||||
elif [ -f "$ARDUPILOT_GZ/worlds/${WORLD_NAME}.sdf" ]; then
|
||||
WORLD_FILE="$ARDUPILOT_GZ/worlds/${WORLD_NAME}.sdf"
|
||||
elif [ -f "$WORLD_NAME" ]; then
|
||||
WORLD_FILE="$WORLD_NAME"
|
||||
else
|
||||
echo -e "${RED}ERROR: World file not found: $WORLD_NAME${NC}"
|
||||
echo ""
|
||||
echo "Available worlds:"
|
||||
echo " Project worlds:"
|
||||
ls "$PROJECT_DIR/worlds/"*.sdf 2>/dev/null | xargs -I {} basename {} .sdf || echo " (none)"
|
||||
echo " ArduPilot Gazebo worlds:"
|
||||
ls "$ARDUPILOT_GZ/worlds/"*.sdf 2>/dev/null | xargs -I {} basename {} .sdf || echo " (none)"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# WSL/Display setup
|
||||
if $IS_WSL; then
|
||||
if [ -f "$PROJECT_DIR/wsl_env.sh" ]; then
|
||||
source "$PROJECT_DIR/wsl_env.sh"
|
||||
fi
|
||||
if [ -z "$DISPLAY" ]; then
|
||||
export DISPLAY=:0
|
||||
if [ -d "/mnt/wslg" ]; then
|
||||
export DISPLAY=:0
|
||||
else
|
||||
export DISPLAY=$(cat /etc/resolv.conf | grep nameserver | awk '{print $2}'):0
|
||||
fi
|
||||
fi
|
||||
echo -e "${BLUE}DISPLAY: $DISPLAY${NC}"
|
||||
fi
|
||||
@@ -140,15 +147,12 @@ cleanup() {
|
||||
echo ""
|
||||
echo -e "${YELLOW}Shutting down simulation...${NC}"
|
||||
|
||||
# Kill processes
|
||||
pkill -f "sim_vehicle.py" 2>/dev/null || true
|
||||
pkill -f "mavproxy" 2>/dev/null || true
|
||||
pkill -f "ArduCopter" 2>/dev/null || true
|
||||
pkill -f "ArduRover" 2>/dev/null || true
|
||||
pkill -f "gzserver" 2>/dev/null || true
|
||||
pkill -f "gzclient" 2>/dev/null || true
|
||||
pkill -f "gazebo" 2>/dev/null || true
|
||||
pkill -f "mavros" 2>/dev/null || true
|
||||
pkill -f "arducopter" 2>/dev/null || true
|
||||
pkill -f "ardurover" 2>/dev/null || true
|
||||
pkill -f "ruby" 2>/dev/null || true # gz sim uses ruby
|
||||
pkill -f "gz sim" 2>/dev/null || true
|
||||
|
||||
sleep 2
|
||||
echo -e "${GREEN}Cleanup complete.${NC}"
|
||||
@@ -156,125 +160,78 @@ cleanup() {
|
||||
|
||||
trap cleanup EXIT INT TERM
|
||||
|
||||
# Activate virtual environment
|
||||
if [ -f "$PROJECT_DIR/venv/bin/activate" ]; then
|
||||
source "$PROJECT_DIR/venv/bin/activate"
|
||||
# Check ArduPilot installation
|
||||
if [ ! -d "$ARDUPILOT_HOME" ]; then
|
||||
echo -e "${RED}ERROR: ArduPilot not found at $ARDUPILOT_HOME${NC}"
|
||||
echo "Run setup.sh first to install ArduPilot"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Use ardupilot_gazebo iris world if no custom world specified
|
||||
if [ ! -f "$WORLD" ]; then
|
||||
if [ -f "$HOME/ardupilot_gazebo/worlds/iris_arducopter_runway.world" ]; then
|
||||
WORLD="$HOME/ardupilot_gazebo/worlds/iris_arducopter_runway.world"
|
||||
else
|
||||
echo -e "${RED}ERROR: World file not found: $WORLD${NC}"
|
||||
exit 1
|
||||
fi
|
||||
# Check Gazebo plugin
|
||||
if [ ! -f "$ARDUPILOT_GZ/build/libArduPilotPlugin.so" ]; then
|
||||
echo -e "${RED}ERROR: ArduPilot Gazebo plugin not found${NC}"
|
||||
echo "Run setup.sh first to build the plugin"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo -e "${GREEN}Configuration:${NC}"
|
||||
echo " Vehicle: $VEHICLE"
|
||||
echo " World: $(basename $WORLD)"
|
||||
echo " World: $(basename $WORLD_FILE .sdf)"
|
||||
echo " Instance: $INSTANCE"
|
||||
echo ""
|
||||
|
||||
# Calculate ports
|
||||
MAVLINK_PORT=$((14550 + INSTANCE * 10))
|
||||
SITL_PORT=$((5760 + INSTANCE * 10))
|
||||
|
||||
# Start Gazebo in background
|
||||
echo -e "${GREEN}Starting Gazebo...${NC}"
|
||||
gazebo --verbose "$WORLD" &
|
||||
GAZEBO_PID=$!
|
||||
gz sim -v4 -r "$WORLD_FILE" &
|
||||
GZ_PID=$!
|
||||
|
||||
sleep 5
|
||||
|
||||
# Check if Gazebo started
|
||||
if ! kill -0 $GAZEBO_PID 2>/dev/null; then
|
||||
if ! kill -0 $GZ_PID 2>/dev/null; then
|
||||
echo -e "${RED}ERROR: Gazebo failed to start${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}Gazebo running (PID: $GAZEBO_PID)${NC}"
|
||||
echo -e "${GREEN}Gazebo running (PID: $GZ_PID)${NC}"
|
||||
|
||||
# Start ArduPilot SITL
|
||||
echo -e "${GREEN}Starting ArduPilot SITL...${NC}"
|
||||
cd "$ARDUPILOT_HOME"
|
||||
|
||||
if [ "$VEHICLE" = "copter" ]; then
|
||||
VEHICLE_DIR="ArduCopter"
|
||||
FRAME="gazebo-iris"
|
||||
elif [ "$VEHICLE" = "rover" ]; then
|
||||
VEHICLE_DIR="Rover"
|
||||
FRAME="gazebo-rover"
|
||||
else
|
||||
VEHICLE_DIR="ArduCopter"
|
||||
FRAME="gazebo-iris"
|
||||
fi
|
||||
export PATH=$PATH:$ARDUPILOT_HOME/Tools/autotest:$HOME/.local/bin
|
||||
|
||||
cd "$ARDUPILOT_HOME/$VEHICLE_DIR"
|
||||
|
||||
# Start sim_vehicle.py in background
|
||||
python3 "$ARDUPILOT_HOME/Tools/autotest/sim_vehicle.py" \
|
||||
-v "$VEHICLE_DIR" \
|
||||
-f "$FRAME" \
|
||||
--no-mavproxy \
|
||||
-I "$INSTANCE" \
|
||||
--no-rebuild \
|
||||
sim_vehicle.py \
|
||||
-v $VEHICLE \
|
||||
-f gazebo-iris \
|
||||
--model JSON \
|
||||
-I $INSTANCE \
|
||||
--console \
|
||||
&
|
||||
SITL_PID=$!
|
||||
|
||||
sleep 10
|
||||
|
||||
# Check if SITL started
|
||||
if ! kill -0 $SITL_PID 2>/dev/null; then
|
||||
echo -e "${RED}ERROR: ArduPilot SITL failed to start${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${GREEN}ArduPilot SITL running (PID: $SITL_PID)${NC}"
|
||||
|
||||
# Start MAVROS
|
||||
echo -e "${GREEN}Starting MAVROS...${NC}"
|
||||
ros2 run mavros mavros_node --ros-args \
|
||||
-p fcu_url:="udp://127.0.0.1:${MAVLINK_PORT}@" \
|
||||
-p gcs_url:="" \
|
||||
-p target_system_id:=1 \
|
||||
-p target_component_id:=1 \
|
||||
&>/dev/null &
|
||||
MAVROS_PID=$!
|
||||
|
||||
sleep 5
|
||||
|
||||
echo ""
|
||||
echo -e "${GREEN}==========================================${NC}"
|
||||
echo -e "${GREEN} Simulation Running${NC}"
|
||||
echo -e "${GREEN}==========================================${NC}"
|
||||
echo ""
|
||||
echo -e "${BLUE}Components:${NC}"
|
||||
echo " - Gazebo: PID $GAZEBO_PID"
|
||||
echo " - Gazebo: PID $GZ_PID"
|
||||
echo " - ArduPilot SITL: PID $SITL_PID"
|
||||
echo " - MAVROS: PID $MAVROS_PID"
|
||||
echo ""
|
||||
echo -e "${BLUE}MAVROS Topics:${NC}"
|
||||
echo " ros2 topic list | grep mavros"
|
||||
echo ""
|
||||
echo -e "${BLUE}Control UAV:${NC}"
|
||||
echo " # Arm"
|
||||
echo " ros2 service call /mavros/cmd/arming mavros_msgs/srv/CommandBool \"{value: true}\""
|
||||
echo ""
|
||||
echo " # Set mode to GUIDED"
|
||||
echo " ros2 service call /mavros/set_mode mavros_msgs/srv/SetMode \"{base_mode: 0, custom_mode: 'GUIDED'}\""
|
||||
echo ""
|
||||
echo " # Takeoff to 5 meters"
|
||||
echo " ros2 service call /mavros/cmd/takeoff mavros_msgs/srv/CommandTOL \"{altitude: 5}\""
|
||||
echo ""
|
||||
echo " # Send position setpoint (x, y, z in meters)"
|
||||
echo " ros2 topic pub /mavros/setpoint_position/local geometry_msgs/PoseStamped \\"
|
||||
echo " \"{header: {frame_id: 'map'}, pose: {position: {x: 10, y: 5, z: 5}}}\""
|
||||
echo -e "${BLUE}MAVProxy Commands (in console):${NC}"
|
||||
echo " mode guided # Switch to GUIDED mode"
|
||||
echo " arm throttle # Arm the drone"
|
||||
echo " takeoff 5 # Takeoff to 5 meters"
|
||||
echo " guided 10 5 -10 # Fly to position (N, E, D)"
|
||||
echo " rtl # Return to launch"
|
||||
echo " land # Land"
|
||||
echo ""
|
||||
echo -e "${YELLOW}Press Ctrl+C to stop simulation${NC}"
|
||||
echo ""
|
||||
|
||||
# Wait for all processes
|
||||
# Wait for processes
|
||||
wait
|
||||
|
||||
Reference in New Issue
Block a user