From 4b8a920e5521358deb0e330456db2c695456f510 Mon Sep 17 00:00:00 2001
From: yjinzero <yjinzero@gmail.com>
Date: Thu, 13 Feb 2025 21:53:25 +0900
Subject: [PATCH] state pub

---
 deploy/deploy_real/state_pub.py | 65 +++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 deploy/deploy_real/state_pub.py

diff --git a/deploy/deploy_real/state_pub.py b/deploy/deploy_real/state_pub.py
new file mode 100644
index 0000000..69160ce
--- /dev/null
+++ b/deploy/deploy_real/state_pub.py
@@ -0,0 +1,65 @@
+from math import sin, cos, pi
+import rclpy
+from rclpy.node import Node
+from rclpy.qos import QoSProfile
+from geometry_msgs.msg import Quaternion
+from sensor_msgs.msg import JointState
+# from tf2_ros import TransformBroadcaster, TransformStamped
+from unitree_hg.msg import LowCmd as LowCmdHG, LowState as LowStateHG
+import yaml
+
+class StatePublisher(Node):
+
+    def __init__(self):
+        rclpy.init()
+        
+        with open('./configs/ik.yaml', 'r') as fp:
+            self.joint_names = yaml.safe_load(fp)['motor_joint']
+
+        super().__init__('state_publisher')
+        qos_profile = QoSProfile(depth=10)
+
+        self.low_state = LowStateHG()
+        self.low_state_subscriber = self._node.create_subscription(LowStateHG,
+                    'lowstate', self.on_low_state, 10)
+        self.joint_pub = self.create_publisher(JointState,
+                                               'joint_states', qos_profile)
+        self.nodeName = self.get_name()
+        self.get_logger().info("{0} started".format(self.nodeName))
+        self.joint_state = JointState()
+
+    # def wait_for_low_state(self):
+    #     while self.low_state.crc == 0:
+    #         print(self.low_state)
+    #         time.sleep(self.config.control_dt)
+    #     print("Successfully connected to the robot.")
+
+    def on_low_state(self, msg: LowStateHG):
+        self.low_state = msg
+        joint_state = self.joint_state
+        now = self.get_clock().now()
+        joint_state.header.stamp = now.to_msg()
+        joint_state.name = self.joint_names
+        joint_state.position = [0.0 for _ in self.joint_names]
+        for i in range(len(self.low_state.motor_state)):
+            joint_state.position[i] = self.low_state.motor_state[i].q
+        self.joint_pub.publish(joint_state)
+    
+    def run(self):
+        loop_rate = self.create_rate(30)
+        try:
+            # rclpy.spin()
+            while rclpy.ok():
+                rclpy.spin_once(self)
+                loop_rate.sleep()
+        except KeyboardInterrupt:
+            pass
+
+
+
+def main():
+    node = StatePublisher()
+    node.run()
+
+if __name__ == '__main__':
+    main()
\ No newline at end of file