|
| 1 | +#!/bin/bash |
| 2 | +# Assuming this script is invoked from the root of the repository... |
| 3 | +help() { |
| 4 | + echo "usage: $0 [COMMAND] [ROS]" |
| 5 | + echo "COMMAND:" |
| 6 | + echo " - stop" |
| 7 | + echo " - build_pick_and_place" |
| 8 | + echo " - start_pick_and_place" |
| 9 | + echo " - build_ros" |
| 10 | + echo " - start_ros" |
| 11 | + echo " - run_ros_color_publisher" |
| 12 | + echo " - run_ros_pose_service_client" |
| 13 | + echo " - run_ros_position_service" |
| 14 | + echo "ROS" |
| 15 | + echo " - ros1" |
| 16 | + echo " - ros2" |
| 17 | +} |
| 18 | + |
| 19 | +COMMAND=$1 |
| 20 | +ROS=$2 |
| 21 | + |
| 22 | +if [ "$COMMAND" == "stop" ]; then |
| 23 | + echo "Terminating process $3" |
| 24 | + pkill -15 -P $3 |
| 25 | + sleep 10 |
| 26 | + |
| 27 | +elif [ "$COMMAND" == "build_pick_and_place" ]; then |
| 28 | + source /opt/ros/noetic/setup.bash |
| 29 | + pushd $PWD |
| 30 | + cd tutorials/pick_and_place/ROS |
| 31 | + catkin_make |
| 32 | + source devel/setup.bash |
| 33 | + popd |
| 34 | + |
| 35 | +elif [ "$COMMAND" == "start_pick_and_place" ]; then |
| 36 | + echo "Starting ROS for Pick and Place" |
| 37 | + source tutorials/pick_and_place/ROS/devel/setup.bash |
| 38 | + roslaunch niryo_moveit part_3.launch |
| 39 | + |
| 40 | +elif [ "$COMMAND" == "build_ros" ]; then |
| 41 | + if [ "$ROS" == "ros1" ]; then |
| 42 | + export ROS_WORKSPACE=$(pwd)/ros1_ws |
| 43 | + mkdir -p $ROS_WORKSPACE/src |
| 44 | + cp -r tutorials/ros_unity_integration/ros_packages/ $ROS_WORKSPACE/src/ |
| 45 | + git clone https://github.com/Unity-Technologies/ROS-TCP-Endpoint $ROS_WORKSPACE/src/ros_tcp_endpoint -b main |
| 46 | + /bin/bash tutorials/ros_unity_integration/ros_docker/set-up-workspace |
| 47 | + chmod +x $ROS_WORKSPACE/src/ros_tcp_endpoint/src/ros_tcp_endpoint/*.py |
| 48 | + elif [ "$ROS" == "ros2" ]; then |
| 49 | + export ROS_WORKSPACE=$(pwd)/ros2_ws |
| 50 | + mkdir -p $ROS_WORKSPACE/src |
| 51 | + cp -r tutorials/ros_unity_integration/ros2_packages/ $ROS_WORKSPACE/src/ |
| 52 | + git clone https://github.com/Unity-Technologies/ROS-TCP-Endpoint $ROS_WORKSPACE/src/ros_tcp_endpoint -b main-ros2 |
| 53 | + source /opt/ros/$ROS_DISTRO/setup.sh |
| 54 | + pushd $(pwd) |
| 55 | + cd $ROS_WORKSPACE |
| 56 | + colcon build |
| 57 | + popd |
| 58 | + else |
| 59 | + help |
| 60 | + fi |
| 61 | + |
| 62 | +elif [ "$COMMAND" == "start_ros" ]; then |
| 63 | + if [ "$ROS" == "ros1" ]; then |
| 64 | + source ros1_ws/devel/setup.bash |
| 65 | + echo "Starting ROS1 master" |
| 66 | + roscore & |
| 67 | + sleep 5 # Wait ROS master to stand up |
| 68 | + rosparam set ROS_IP 127.0.0.1 |
| 69 | + echo "Starting ROS1 default server endpoint" |
| 70 | + rosrun ros_tcp_endpoint default_server_endpoint.py |
| 71 | + elif [ "$ROS" == "ros2" ]; then |
| 72 | + source ros2_ws/install/setup.bash |
| 73 | + echo "Starting ROS2 default server endpoint" |
| 74 | + ros2 run ros_tcp_endpoint default_server_endpoint --ros-args -p ROS_IP:=127.0.0.1 |
| 75 | + else |
| 76 | + help |
| 77 | + fi |
| 78 | + |
| 79 | +elif [ "$COMMAND" == "run_ros_color_publisher" ]; then |
| 80 | + if [ "$ROS" == "ros1" ]; then |
| 81 | + source ros1_ws/devel/setup.bash |
| 82 | + elif [ "$ROS" == "ros2" ]; then |
| 83 | + source ros2_ws/install/setup.bash |
| 84 | + else |
| 85 | + help |
| 86 | + fi |
| 87 | + echo "Starting to run $ROS color publisher every 30 seconds" |
| 88 | + count=0 |
| 89 | + while [[ $count -le 6 ]] |
| 90 | + do |
| 91 | + sleep 5 |
| 92 | + if [ "$ROS" == "ros1" ]; then |
| 93 | + rosrun unity_robotics_demo color_publisher.py |
| 94 | + elif [ "$ROS" == "ros2" ]; then |
| 95 | + ros2 run unity_robotics_demo color_publisher |
| 96 | + else |
| 97 | + help |
| 98 | + fi |
| 99 | + count=$(( $count + 1 )) |
| 100 | + done |
| 101 | + echo "Completed run: $ROS color publisher" |
| 102 | + |
| 103 | +elif [ "$COMMAND" == "run_ros_pose_service_client" ]; then |
| 104 | + if [ "$ROS" == "ros1" ]; then |
| 105 | + source ros1_ws/devel/setup.bash |
| 106 | + elif [ "$ROS" == "ros2" ]; then |
| 107 | + source ros2_ws/install/setup.bash |
| 108 | + else |
| 109 | + help |
| 110 | + fi |
| 111 | + echo "Starting to run $ROS pose service client and send requests every 30 seconds" |
| 112 | + count=0 |
| 113 | + while [[ $count -le 6 ]] |
| 114 | + do |
| 115 | + sleep 5 |
| 116 | + if [ "$ROS" == "ros1" ]; then |
| 117 | + rosservice call /obj_pose_srv Cube |
| 118 | + elif [ "$ROS" == "ros2" ]; then |
| 119 | + ros2 service call obj_pose_srv unity_robotics_demo_msgs/ObjectPoseService "{object_name: Cube}" |
| 120 | + else |
| 121 | + help |
| 122 | + fi |
| 123 | + count=$(( $count + 1 )) |
| 124 | + done |
| 125 | + echo "Completed run: $ROS pose service client" |
| 126 | + |
| 127 | +elif [ "$COMMAND" == "run_ros_position_service" ]; then |
| 128 | + if [ "$ROS" == "ros1" ]; then |
| 129 | + source ros1_ws/devel/setup.bash |
| 130 | + elif [ "$ROS" == "ros2" ]; then |
| 131 | + source ros2_ws/install/setup.bash |
| 132 | + else |
| 133 | + help |
| 134 | + fi |
| 135 | + echo "Starting $ROS position service" |
| 136 | + if [ "$ROS" == "ros1" ]; then |
| 137 | + rosrun unity_robotics_demo position_service.py |
| 138 | + elif [ "$ROS" == "ros2" ]; then |
| 139 | + ros2 run unity_robotics_demo position_service |
| 140 | + else |
| 141 | + help |
| 142 | + fi |
| 143 | + |
| 144 | +else |
| 145 | + help |
| 146 | +fi |
0 commit comments