ros2 robot interface command is tested.

Go2Py/robot/interface/ros2.py

@@ -13,6 +13,8 @@ from geometry_msgs.msg import TransformStamped
 from Go2Py.joy import xKeySwitch, xRockerBtn
 from geometry_msgs.msg import TwistStamped
 from unitree_go.msg import LowState
+from nav_msgs.msg import Odometry   
+
 
 
 def ros2_init(args=None):
@@ -56,25 +58,31 @@ class ROS2ExecutorManager:
             self.executor_thread.join()
 
 
-class GO2Robot(Node):
+class GO2Real(Node):
     def __init__(
         self,
-        test=False,
+        mode = 'highlevel', # 'highlevel' or 'lowlevel'
         vx_max=0.5,
         vy_max=0.4,
         ωz_max=0.5,
-        node_name="go2py_highlevel_subscriber",
     ):
-        self.node_name = node_name
+        assert mode in ['highlevel', 'lowlevel'], "mode should be either 'highlevel' or 'lowlevel'"
+        self.mode = mode
+        self.node_name = "go2py_highlevel_subscriber"
         self.highcmd_topic = "/go2/twist_cmd"
         self.joint_state_topic = "/go2/joint_states"
         self.lowstate_topic = "/lowstate"
         super().__init__(self.node_name)
         
-        self.subscription = self.create_subscription(
+        self.lowstate_subscriber = self.create_subscription(
             LowState, self.lowstate_topic, self.lowstate_callback, 1
         )
-        self.cmd_publisher = self.create_publisher(TwistStamped, self.highcmd_topic, 1)
+
+        self.odometry_subscriber = self.create_subscription(
+            Odometry, "/utlidar/robot_odom", self.odom_callback, 1
+        )
+
+        self.highcmd_publisher = self.create_publisher(TwistStamped, self.highcmd_topic, 1)
         self.highcmd = TwistStamped()
         # create pinocchio robot
         # self.pin_robot = PinRobot()
@@ -93,16 +101,31 @@ class GO2Robot(Node):
         """
         self.state = msg
 
+    def odom_callback(self, msg):
+        """
+        Retrieve the odometry of the robot
+        """
+        self.odom = msg
+
+    def getOdometry(self):
+        """Returns the odometry of the robot"""
+        stamp = self.odom.header.stamp
+        position = np.array([self.odom.pose.pose.position.x, self.odom.pose.pose.position.y, self.odom.pose.pose.position.z])
+        orientation = np.array([self.odom.pose.pose.orientation.x, self.odom.pose.pose.orientation.y, self.odom.pose.pose.orientation.z, self.odom.pose.pose.orientation.w])
+        stamp_nanosec = stamp.sec + stamp.nanosec * 1e-9
+        return {'stamp_nanosec':stamp_nanosec, 'position':position, 'orientation':orientation}
+
     def getIMU(self):
-        accel = self.state.imu.accelerometer
+        accel = self.state.imu_state.accelerometer
-        gyro = self.state.imu.gyroscope
+        gyro = self.state.imu_state.gyroscope
-        quat = self.state.imu.quaternion
+        quat = self.state.imu_state.quaternion
-        rpy = self.state.imu.rpy
+        rpy = self.state.imu_state.rpy
-        return accel, gyro, quat, rpy
+        temp = self.state.imu_state.temperature
+        return {'accel':accel, 'gyro':gyro, 'quat':quat, "rpy":rpy, 'temp':temp}
 
     def getFootContacts(self):
-        """Returns the foot contact states"""
+        """Returns the raw foot contact forces"""
-        footContacts = self.state.foot_force_est
+        footContacts = self.state.foot_force
         return np.array(footContacts)
 
     def getJointStates(self):
@@ -113,10 +136,19 @@ class GO2Robot(Node):
         )
         q, dq = np.array(_q), np.array(_dq)
 
-        return q, dq
+        return {'q':q, 'dq':dq}
 
     def getRemoteState(self):
-        """Returns the state of the remote control"""
+        """A method to get the state of the wireless remote control. 
+        Returns a xRockerBtn object: 
+        - head: [head1, head2]
+        - keySwitch: xKeySwitch object
+        - lx: float
+        - rx: float
+        - ry: float
+        - L2: float
+        - ly: float
+        """
         wirelessRemote = self.state.wireless_remote[:24]
 
         binary_data = bytes(wirelessRemote)
@@ -151,7 +183,7 @@ class GO2Robot(Node):
         v_x = ly * self.vx_max
         v_y = lx * self.vy_max
         ω = rx * self.ωz_max
-
+        
         return v_x, v_y, ω
 
     def getBatteryState(self):
@@ -160,16 +192,17 @@ class GO2Robot(Node):
         return batteryState.SOC
 
     def setCommands(self, v_x, v_y, ω_z, bodyHeight=0.0, footRaiseHeight=0.0, mode=2):
-        assert mode in [0, 2]  # Only mode 2: walking and mode 0: idea is allowed
+        if self.mode == 'highlevel':
-        self.cmd_watchdog_timer = time.time()
+            self.cmd_watchdog_timer = time.time()
-        self.cmd.mode = mode
+            _v_x, _v_y, _ω_z = self.clip_velocity(v_x, v_y, ω_z)
-        self.cmd.body_height = np.clip(bodyHeight, -0.15, 0.1)
+            self.highcmd.header.stamp = self.get_clock().now().to_msg()
-        self.cmd.foot_raise_height = np.clip(footRaiseHeight, -0.1, 0.1)
+            self.highcmd.header.frame_id = "base_link"
-        _v_x, _v_y, _ω_z = self.clip_velocity(v_x, v_y, ω_z)
+            self.highcmd.twist.linear.x = _v_x
-        self.cmd.velocity = [_v_x, _v_y]
+            self.highcmd.twist.linear.y = _v_y
-        self.cmd.yaw_speed = _ω_z
+            self.highcmd.twist.angular.z = _ω_z
-        # Publish the command as a ROS2 message
+            self.highcmd_publisher.publish(self.highcmd)
-        self.cmd_publisher.publish(self.cmd)
+        else:
+            raise NotImplementedError("Low level control command is not implemented yet")
 
     def close(self):
         self.running = False

examples/highlevel_ros2_interface.ipynb

@@ -12,9 +12,18 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 1,
    "metadata": {},
-   "outputs": [],
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "pygame 2.5.2 (SDL 2.28.2, Python 3.8.18)\n",
+      "Hello from the pygame community. https://www.pygame.org/contribute.html\n"
+     ]
+    }
+   ],
    "source": [
     "from Go2Py.joy import Logitech3DPro\n",
     "# joy = Logitech3DPro(joy_id=0)"
@@ -31,19 +40,59 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
-    "from Go2Py.robot.interface.ros2 import GO2Robot, ros2_init, ROS2ExecutorManager\n",
+    "from Go2Py.robot.interface.ros2 import GO2Real, ros2_init, ROS2ExecutorManager\n",
     "import time\n",
     "ros2_init()\n",
-    "robot = GO2Robot()\n",
+    "robot = GO2Real()\n",
     "ros2_exec_manager = ROS2ExecutorManager()\n",
     "ros2_exec_manager.add_node(robot)\n",
     "ros2_exec_manager.start()"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [
+    {
+     "ename": "AttributeError",
+     "evalue": "'float' object has no attribute 'nanosec'",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mAttributeError\u001b[0m                            Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[4], line 2\u001b[0m\n\u001b[1;32m      1\u001b[0m \u001b[38;5;66;03m# robot.getIMU()\u001b[39;00m\n\u001b[0;32m----> 2\u001b[0m \u001b[43mrobot\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mgetOdometry\u001b[49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m[\u001b[49m\u001b[38;5;124;43m'\u001b[39;49m\u001b[38;5;124;43mstamp_nanosec\u001b[39;49m\u001b[38;5;124;43m'\u001b[39;49m\u001b[43m]\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mnanosec\u001b[49m\n",
+      "\u001b[0;31mAttributeError\u001b[0m: 'float' object has no attribute 'nanosec'"
+     ]
+    }
+   ],
+   "source": [
+    "# robot.getIMU()\n",
+    "robot.getOdometry()['stamp_nanosec'].nanosec"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 48,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from nav_msgs.msg import Odometry "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 21,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "robot.setCommands(0, 0, 0.1)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},
@@ -95,7 +144,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.8.10"
+   "version": "3.8.18"
   }
  },
  "nbformat": 4,