上一篇我们通过C++实现了发布者和订阅者,本篇我们试试用Python来实现同样的功能。
1.创建功能包
首先我们在之前创建的dev_ws工作空间中来创建一个放置节点代码的功能包。 打开一个新的终端,cd到dev_ws/src目录下,然后运行创建功能包的指令:
ros2 pkg create --build-type ament_python py_pubsub
很快就创建好了一个叫做py_pubsub的功能包。
2.创建发布者节点
在py_pubsub/py_pubsub文件夹下,创建一个发布者节点的代码文件publisher_member_function.py,然后拷贝以下代码放进去:
import rclpy
from rclpy.node import Node
from std_msgs.msg import String
class MinimalPublisher(Node):
def __init__(self):
super().__init__('minimal_publisher')
self.publisher_ = self.create_publisher(String, 'topic', 10)
timer_period = 0.5 # seconds
self.timer = self.create_timer(timer_period, self.timer_callback)
self.i = 0
def timer_callback(self):
msg = String()
msg.data = 'Hello World: %d' % self.i
self.publisher_.publish(msg)
self.get_logger().info('Publishing: "%s"' % msg.data)
self.i += 1
def main(args=None):
rclpy.init(args=args)
minimal_publisher = MinimalPublisher()
rclpy.spin(minimal_publisher)
# Destroy the node explicitly
# (optional - otherwise it will be done automatically
# when the garbage collector destroys the node object)
minimal_publisher.destroy_node()
rclpy.shutdown()
if __name__ == '__main__':
main()
这就是一个简单的发布者代码,我们来对代码做下解析。
import rclpy
from rclpy.node import Node
首先是引入需要的模块,包括ROS2的python接口和节点类。
from std_msgs.msg import String
紧接着是导入String字符串消息。
class MinimalPublisher(Node):
然后创建一个继承于Node基类的MinimalPublisher节点子类。
def __init__(self):
super().__init__('minimal_publisher')
self.publisher_ = self.create_publisher(String, 'topic', 10)
timer_period = 0.5 # seconds
self.timer = self.create_timer(timer_period, self.timer_callback)
self.i = 0
这里是MinimalPublisherd的构造函数,super().__init__ 中会调用父类Node的构造函数,节点名初始化为“minimal_publisher”。然后创建了一个发布者,发布的话题名是topic,话题消息是String,保存消息的队列长度是10,然后创建了一个定时器timer_,做了一个0.5秒的定时,每次触发定时器后,都会运行回调函数timer_callback。
def timer_callback(self):
msg = String()
msg.data = 'Hello World: %d' % self.i
self.publisher_.publish(msg)
self.get_logger().info('Publishing: "%s"' % msg.data)
self.i += 1
timer_callback是这里的关键,每次触发都会发布一次话题消息。message中保存的字符串是Hello world加一个计数值,然后通过info函数打印一次日志信息,再通过发布者的publish方法将消息发布出去。
def main(args=None):
rclpy.init(args=args)
minimal_publisher = MinimalPublisher()
rclpy.spin(minimal_publisher)
# Destroy the node explicitly
# (optional - otherwise it will be done automatically
# when the garbage collector destroys the node object)
minimal_publisher.destroy_node()
rclpy.shutdown()
最后就是main函数啦,先初始化rclpy,再创建MinimalPublisher,进入自旋锁,当退出锁时,就会关闭节点结束啦。 完成以上发布者的代码后,功能包里还有一些内容需要设置。
- 设置依赖项
打开功能包的package.xml文件,先把这些基础信息填写好:
<description>Examples of minimal publisher/subscriber using rclpy</description>
<maintainer email="you@email.com">Your Name</maintainer>
<license>Apache License 2.0</license>
然后还需要添加依赖项,放到ament_python下边:
<exec_depend>rclpy</exec_depend>
<exec_depend>std_msgs</exec_depend>
- 设置程序入口
接下来打开setup.py文件,同样需要补充以下内容:
maintainer='YourName',
maintainer_email='you@email.com',
description='Examples of minimal publisher/subscriber using rclpy',
license='Apache License 2.0',
然后在 entry_points 下添加如下内容:
entry_points={
'console_scripts': [
'talker = py_pubsub.publisher_member_function:main',
],},
- 检查setup.cfg文件
setup.cfg文件中内容是自动添加的,可以打开看下,内容如下:
[develop]script-dir=$base/lib/py_pubsub
[install]install-scripts=$base/lib/py_pubsub
主要含义就是把可执行的python文件在编译时放到install的lib下边,这样ros2 run命令才能找的到。
3.创建订阅者
还是回到功能包py_pubsub/py_pubsub文件夹下,接下来创建订阅者的代码subscriber_member_function.py:
import rclpyfrom rclpy.node import Node
from std_msgs.msg import String
class MinimalSubscriber(Node):
def __init__(self):
super().__init__('minimal_subscriber')
self.subscription = self.create_subscription(
String,
'topic',
self.listener_callback,
10)
self.subscription # prevent unused variable warning
def listener_callback(self, msg):
self.get_logger().info('I heard: "%s"' % msg.data)
def main(args=None):
rclpy.init(args=args)
minimal_subscriber = MinimalSubscriber()
rclpy.spin(minimal_subscriber)
# Destroy the node explicitly
# (optional - otherwise it will be done automatically
# when the garbage collector destroys the node object)
minimal_subscriber.destroy_node()
rclpy.shutdown()
if __name__ == '__main__':
main()
还是来解析下代码。
self.subscription = self.create_subscription(
String,
'topic',
self.listener_callback,
10)
订阅者的代码整体流程和发布者类似,现在的节点名叫minimal_subscriber,构造函数中创建了订阅者,订阅String消息,订阅的话题名叫做“topic”,保存消息的队列长度是10,当订阅到数据时,会进入回调函数topic_callback。
def listener_callback(self, msg):
self.get_logger().info('I heard: "%s"' % msg.data)
回调函数中会收到String消息,然后并没有做太多处理,只是通过info打印出来。 main函数中的内容和发布者几乎是一致的,就不再赘述。 由于该节点的依赖项和发布者一样,我们就不需要修改package.xml和 setup.cfg文件了,不过程序入口 setup.py还是得加一些内容:
entry_points={
'console_scripts': [
'talker = py_pubsub.publisher_member_function:main',
'listener = py_pubsub.subscriber_member_function:main',
],},
4.编译并运行
编译前先确认下功能包的依赖项有没有都安装好,在dev_ws路径下运行如下命令:
rosdep install -i --from-path src --rosdistro foxy -y
安装完毕后还是在该路径下编译py_pubsub功能包:
colcon build --packages-select py_pubsub
编译完成后,打开一个新的终端,设置工作空间的环境变量后,运行发布者:
. install/setup.bash
ros2 run py_pubsub talker
运行成功后可以看到终端每隔0.5s打印一次日志信息: 再打开一个新的终端,类似的操作,运行订阅者:
. install/setup.bash
ros2 run py_pubsub listener