【TOOL】提升ROS配置主从机效率的小工具

news2025/1/11 19:41:28

一、简介

因经常需要对不同机器配置主从机(不同机器的主机名一致，IP地址为 “192.168.101.*** ”格式)，故整理此脚本来提高自己频繁切换机器和主从机/单机模式的切换效率，进行一个命令行配置/取消主从机模式。

二、配置步骤

1.编辑脚本文件

脚本文件名为 set_master_slave.sh,内容如下：

#!/bin/bash
# Author:   chatgpt
# Verifier: ZYG
# Date:     2023.06.29

ip_pre="192.168.101."
ip_master_post="143"
ip_slave_post="145"
ip_master=${ip_pre}${ip_master_post}
ip_slave=${ip_pre}${ip_slave_post}

remote_name="robot"
locate_name="gene"

edit_files() {
  # Check if ~/.bashrc contains existing ROS master and slave configuration
  if grep -qE "ROS_MASTER_URI|ROS_HOSTNAME" ~/.bashrc; then
    # Remove existing ROS master and slave configuration
    sed -i '/export ROS_MASTER_URI=/d' ~/.bashrc
    sed -i '/export ROS_HOSTNAME=/d' ~/.bashrc
    #echo "Existing ROS master and slave configuration removed from ~/.bashrc."
  fi

  # Check if /etc/hosts contains existing ROS master and slave configuration
  if grep -qE ".*${remote_name}|.*${locate_name}" /etc/hosts; then
    # Remove existing ROS master and slave configuration
    sudo sed -i -E "/.*(${remote_name}|${locate_name})/d" /etc/hosts
    #echo "Existing ROS master and slave configuration removed from /etc/hosts."
  fi

  # Configuring ROS remote master IP
  echo "export ROS_MASTER_URI=http://$1:11311" >> ~/.bashrc

  # Configuring ROS local slave IP
  echo "export ROS_HOSTNAME=$2" >> ~/.bashrc

  # Update /etc/hosts file
  if grep -qE "${remote_name}|${locate_name}" /etc/hosts; then
    # Remove existing ROS master and slave configuration
    sudo sed -i -E "/(${remote_name}|${locate_name})/d" /etc/hosts
  fi
  echo "$1 ${remote_name}" | sudo tee -a /etc/hosts >/dev/null

  echo "ROS master and slave configuration updated."
  echo "ROS master: ${remote_name} ${ip_master}"
  echo "ROS slave:  ${locate_name} ${ip_slave}"
}
# Check the number of arguments
if [[ $# -eq 2 ]]; then
  # Two arguments provided, configure ROS master and slave IPs
  ip_master=${ip_pre}$1
  ip_slave=${ip_pre}$2

  edit_files ${ip_master} ${ip_slave}
elif [[ $# -eq 1 ]]; then
  if [ "$1" == "n" ]; then
    # Check if ~/.bashrc contains existing ROS master and slave configuration
    if grep -qE "ROS_MASTER_URI|ROS_HOSTNAME" ~/.bashrc; then
      # Remove existing ROS master and slave configuration
      sed -i '/export ROS_MASTER_URI=/d' ~/.bashrc
      sed -i '/export ROS_HOSTNAME=/d' ~/.bashrc
      echo "Removed ROS master and slave configuration from ~/.bashrc."
    fi
    # Check if /etc/hosts contains existing ROS master and slave configuration
    if grep -qE ".*${remote_name}|.*${locate_name}" /etc/hosts; then
      # Remove existing ROS master and slave configuration
      sudo sed -i -E "/.*(${remote_name}|${locate_name})/d" /etc/hosts
      echo "Removed ROS master and slave configuration from /etc/hosts."
    fi
  else
    # One argument provided, configure default ROS master and slave IPs
    ip_master=${ip_pre}$1
    #ip_slave=${ip_pre}$1

    edit_files ${ip_master} ${ip_slave}
  fi
else
  # No arguments provided
  echo "Invalid number of arguments."
  echo "Usage: bash set_master_slave.sh [master_ip] [slave_ip]"
  echo "Example: bash set_master_slave.sh 100 143"
  exit 1
fi

# Source the updated ~/.bashrc file
source ~/.bashrc