전공/ROS
ROS2 여러 토픽 구독, 발행하기
흔한 학생
2024. 11. 14. 20:48
항상 그랬듯 새로운 파이썬 파일 만들고서는 setup.py 수정하기
setup.py에서 엔트리 포인트를 추가해주어야 함
두 가지 토픽을 구독하는 노드 만들기
import rclpy as rp
from rclpy.node import Node
from turtlesim.msg import Pose
from geometry_msgs.msg import Twist
from my_first_package_msgs.msg import CmdAndPoseVel
class CmdAndPose(Node):
def __init__(self):
super().__init__('turtle_cmd_pose')
self.sub_pose = self.create_subscription(Pose, '/turtle1/pose', self.callback_pose, 10)
self.sub_cmdvel = self.create_subscription(Twist, 'turtle1/cmd_vel', self.callback_cmd, 10)
self.cmd_pose = CmdAndPoseVel()
def callback_pose(self, msg):
self.cmd_pose.pose_x = msg.x
self.cmd_pose.pose_y = msg.y
self.cmd_pose.linear_vel = msg.linear_velocity
self.cmd_pose.angular_vel = msg.angular_velocity
def callback_cmd(self, msg):
self.cmd_pose.cmd_vel_linear = msg.linear.x
self.cmd_pose.cmd_vel_angular = msg.angular.z
def main(args=None):
rp.init(args=args)
turtle_cmd_pose_node = CmdAndPose()
rp.spin(turtle_cmd_pose_node)
turtle_cmd_pose_node.destroy_node()
rp.shutdown()
if __name__ == '__main__':
main()
pose와 cmd_vel 토픽을 받아오고 만들었던 메시지 타입으로 넣어준다.
이전에 했던 구독 노드를 융합한 느낌이다.
callback 함수에서 msg를 받아온다.
__init__ 아래에서 볼 수 있듯이
각 함수에서는 msg 인자로 각 Pose 타입의 /turtle1/pose와 Twist 타입의 /turtle1/cmd_vel을 받는다.
두 가지 토픽 구독 + 발행하는 노드 만들기
import rclpy as rp
from rclpy.node import Node
from turtlesim.msg import Pose
from geometry_msgs.msg import Twist
from my_first_package_msgs.msg import CmdAndPoseVel
class CmdAndPose(Node):
def __init__(self):
super().__init__('turtle_cmd_pose')
self.sub_pose = self.create_subscription(Pose, '/turtle1/pose', self.callback_pose, 10)
self.sub_cmdvel = self.create_subscription(Twist, 'turtle1/cmd_vel', self.callback_cmd, 10)
self.timer_period = 1.0
self.publisher = self.create_publisher(CmdAndPoseVel, '/cmd_and_pose', 10)
self.timer = self.create_timer(self.timer_period, self.timer_callback)
self.cmd_pose = CmdAndPoseVel()
def callback_pose(self, msg):
self.cmd_pose.pose_x = msg.x
self.cmd_pose.pose_y = msg.y
self.cmd_pose.linear_vel = msg.linear_velocity
self.cmd_pose.angular_vel = msg.angular_velocity
def callback_cmd(self, msg):
self.cmd_pose.cmd_vel_linear = msg.linear.x
self.cmd_pose.cmd_vel_angular = msg.angular.z
def timer_callback(self):
self.publisher.publish(self.cmd_pose)
def main(args=None):
rp.init(args=args)
turtle_cmd_pose_node = CmdAndPose()
rp.spin(turtle_cmd_pose_node)
turtle_cmd_pose_node.destroy_node()
rp.shutdown()
if __name__ == '__main__':
main()
앞의 구독 노드와 이전의 토픽 발행 노드를 합친 느낌이다.
이 글은 민형기 강사님의 강의를 보고 복습을 위해 정리한 글입니다.
때문에 오류가 있을 수 있으며 자세한 사항은 유튜브를 참고해 공부하시면 좋을 것 같습니다.