rospy节点一边接收topic,一边将topic数据可视化

如果想要matplotlib动态更新画图,只能将matplotlib放在主线程中,如果放进子线程,就会报这个错

ValueError: set_wakeup_fd only works in main thread

但是如果要订阅ros的topic,主线程会被阻塞,matplotlib无法动态更新。

因此,通过将matplotlib嵌入tkinter中,实现一边订阅topic,一边动态更新画图

python 复制代码
import math
import rospy
import message_filters
import tkinter as tk
from matplotlib.figure import Figure
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg
from my_pkg.msg import MyArrayMessage


class ROS_Tkinter_App:
    def __init__(self, master):
        self.master = master
        master.title("ROS Tkinter App")

        # Matplotlib setup for scatter plot
        self.figure = Figure(figsize=(10, 8), dpi=100)
        self.scatter_plot = self.figure.add_subplot(111)
        self.scatter_plot.set_title('Scatter Plot')
        self.canvas = FigureCanvasTkAgg(self.figure, master=master)
        self.canvas.get_tk_widget().pack()

        # ROS Initialization
        rospy.init_node('ros_tkinter_app', anonymous=True)

        # Define subscribers for the two topics
        sub_topic1 = message_filters.Subscriber("/global_path", MyArrayMessage)
        sub_topic2 = message_filters.Subscriber("/vehicle_position", MyArrayMessage)

        # Synchronize the subscribers using ApproximateTimeSynchronizer
        ts = message_filters.ApproximateTimeSynchronizer([sub_topic1, sub_topic2], queue_size=10, slop=0.1)
        ts.registerCallback(self.callback)

        # Bind the update_plot_size function to the <Configure> event
        # master.bind("<Configure>", self.update_plot_size)

    def callback(self, data1, data2):
        # This method will be called when a new message is received on the /ros_topic
        rospy.loginfo("ROS Topic Data")

        # 这里处理topic内的数据

        # Update scatter plot with new data
        self.update_scatter_plot(Target_Angle, Center_y, Center_x, waypoints_y, waypoints_x)  # Assuming the data is a float, update accordingly

    def update_scatter_plot(self, angle_radians, Center_x, Center_y, waypoints_x, waypoints_y):
        # Update the scatter plot with new data
        self.scatter_plot.clear()
        self.scatter_plot.set_title('Scatter Plot')

        self.scatter_plot.scatter(Center_x, Center_y, color='red', marker='o', label='Center Point')
        
        self.scatter_plot.plot([Center_x, end_x], [Center_y, end_y], color='blue', linestyle='--', label='Line Segment')

        self.scatter_plot.scatter(waypoints_x, waypoints_y, color='green', marker='o', label='Waypoint')

        # 设置x轴和y轴的显示范围
        self.scatter_plot.set_xlim(-100, 100)  # 设置x轴的范围
        self.scatter_plot.set_ylim(-100, 100)  # 设置y轴的范围
        # 设置 x 轴和 y 轴的相同比例尺
        self.scatter_plot.set_aspect('equal')

        # 设置坐标轴标签
        self.scatter_plot.set_xlabel('CARLA Y Axis')
        self.scatter_plot.set_ylabel('CARLA X Axis')

        self.canvas.draw()

    def update_plot_size(self, event):
        # Update the plot size based on the new window size
        new_width = event.width / self.canvas.figure.get_dpi()
        new_height = event.height / self.canvas.figure.get_dpi()
        self.figure.set_size_inches(new_width, new_height)
        self.canvas.draw()


def main():
    root = tk.Tk()
    app = ROS_Tkinter_App(root)
    root.mainloop()


if __name__ == '__main__':
    main()
相关推荐
进击的六角龙40 分钟前
Python中处理Excel的基本概念(如工作簿、工作表等)
开发语言·python·excel
一只爱好编程的程序猿1 小时前
Java后台生成指定路径下创建指定名称的文件
java·python·数据下载
千汇数据的老司机1 小时前
webGIS在可视化大屏中的应用场景有哪些,有哪些类库
信息可视化·webgis·#web
Aniay_ivy1 小时前
深入探索 Java 8 Stream 流:高效操作与应用场景
java·开发语言·python
gonghw4031 小时前
DearPyGui学习
python·gui
向阳12181 小时前
Bert快速入门
人工智能·python·自然语言处理·bert
engchina1 小时前
Neo4j 和 Python 初学者指南:如何使用可选关系匹配优化 Cypher 查询
数据库·python·neo4j
兆。1 小时前
掌握 PyQt5:从零开始的桌面应用开发
开发语言·爬虫·python·qt
小华同学ai2 小时前
AJ-Report:一款开源且非常强大的数据可视化大屏和报表工具
数据库·信息可视化·开源
南宫理的日知录2 小时前
99、Python并发编程:多线程的问题、临界资源以及同步机制
开发语言·python·学习·编程学习