def set_timeline_data(self,
len_timeline=None,
color_callback=None,
pause_callback=None):
if len_timeline:
rp = rospkg.RosPack()
ui_file = os.path.join(rp.get_path('rqt_robot_monitor'),
'resource', 'timelinepane.ui')
loadUi(ui_file, self, {'TimelineView': TimelineView})
self._pause_callback = pause_callback
self._timeline_view.set_init_data(1, len_timeline, 5,
color_callback)
self._scene = QGraphicsScene(self._timeline_view)
self._timeline_view.setScene(self._scene)
self._timeline_view.show()
self._queue_diagnostic = deque()
self._len_timeline = len_timeline
self._paused = False
self._tracking_latest = True
self._last_sec_marker_at = 2
self._last_msg = None
self._pause_button.clicked[bool].connect(self._pause)
self.sig_update.connect(self._timeline_view.slot_redraw)
def __init__(self, ros_package):
super(GraphWidget, self).__init__()
self._current_graph = None
self._lock = Lock()
self._load_ui(ros_package)
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
factory = DmgItemFactory()
factory.set_color(QColor(50, 50, 50))
factory.set_highlighted_color(QColor(0, 150, 0))
self._dot_to_qt = DotToQtGenerator(factory)
self.initialized = False
self.setObjectName('GraphWidget')
self.graphics_view.setScene(self._scene)
self.open_button.setIcon(QIcon.fromTheme('document-open'))
self.open_button.pressed.connect(self._import)
self.export_button.setIcon(QIcon.fromTheme('document-export'))
self.export_button.pressed.connect(self._export)
self.fit_to_view_button.setIcon(QIcon.fromTheme('zoom-fit-best'))
self.fit_to_view_button.pressed.connect(self._fit_to_view)
self.decision_graphs_combo_box.setSizeAdjustPolicy(QComboBox.AdjustToMinimumContentsLength)
self.decision_graphs_combo_box.currentIndexChanged['QString'].connect(self._graph_item_changed)
self._dot_processor = DotProcessor(self._dot_to_qt)
self.decision_graphs = dict()
self.states = dict()
def __init__(self, parent=None):
super(QCameraView, self).__init__(parent)
self.scene = QGraphicsScene(self)
self.setScene(self.scene)
self._load_default_image()
self.connect(self, SIGNAL("load_default_image"),
self._load_default_image)
class EsterelPlanViewerWidget(QWidget):
# plan view
_scene = QGraphicsScene()
_webview = QGraphicsWebView()
_svg = QGraphicsSvgItem()
_renderer = QSvgRenderer()
def __init__(self, plugin=None):
super(EsterelPlanViewerWidget, self).__init__()
# Create QWidget
ui_file = os.path.join(rospkg.RosPack().get_path('rosplan_rqt'),
'resource', 'esterel_plan_viewer.ui')
loadUi(ui_file, self)
self.setObjectName('ROSPlanEsterelPlanViewer')
self.graphicsView.setScene(self._scene)
self._scene.addItem(self._svg)
self.refreshButton.clicked[bool].connect(self._handle_refresh_clicked)
self._sub = rospy.Subscriber("/plan_graph", String, self.plan_received)
self._plugin = plugin
"""
updating plan view
"""
def plan_received(self, msg):
graph = pydot.graph_from_dot_data(msg.data)
svg_string = graph.create_svg()
self._renderer.load(QByteArray(svg_string))
self._svg.setSharedRenderer(self._renderer)
"""
called when the refresh button is clicked
"""
def _handle_refresh_clicked(self, checked):
self._sub.unregister()
self._sub = rospy.Subscriber(self.topicText.text(), String,
self.plan_received)
"""
Qt methods
"""
def shutdown_plugin(self):
pass
def save_settings(self, plugin_settings, instance_settings):
pass
def restore_settings(self, plugin_settings, instance_settings):
pass
def __init__(self, context):
super(RosTfTree, self).__init__(context)
self.initialized = False
self.setObjectName('RosTfTree')
self._current_dotcode = None
self._widget = QWidget()
# factory builds generic dotcode items
self.dotcode_factory = PydotFactory()
# self.dotcode_factory = PygraphvizFactory()
# generator builds rosgraph
self.dotcode_generator = RosTfTreeDotcodeGenerator()
self.tf2_buffer_ = tf2_ros.Buffer()
self.tf2_listener_ = tf2_ros.TransformListener(self.tf2_buffer_)
# dot_to_qt transforms into Qt elements using dot layout
self.dot_to_qt = DotToQtGenerator()
rp = rospkg.RosPack()
ui_file = os.path.join(rp.get_path('rqt_tf_tree'), 'resource', 'RosTfTree.ui')
loadUi(ui_file, self._widget, {'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('RosTfTreeUi')
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() + (' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.refresh_graph_push_button.setIcon(QIcon.fromTheme('view-refresh'))
self._widget.refresh_graph_push_button.pressed.connect(self._update_tf_graph)
self._widget.highlight_connections_check_box.toggled.connect(self._redraw_graph_view)
self._widget.auto_fit_graph_check_box.toggled.connect(self._redraw_graph_view)
self._widget.fit_in_view_push_button.setIcon(QIcon.fromTheme('zoom-original'))
self._widget.fit_in_view_push_button.pressed.connect(self._fit_in_view)
self._widget.load_dot_push_button.setIcon(QIcon.fromTheme('document-open'))
self._widget.load_dot_push_button.pressed.connect(self._load_dot)
self._widget.save_dot_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_dot_push_button.pressed.connect(self._save_dot)
self._widget.save_as_svg_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_as_svg_push_button.pressed.connect(self._save_svg)
self._widget.save_as_image_push_button.setIcon(QIcon.fromTheme('image-x-generic'))
self._widget.save_as_image_push_button.pressed.connect(self._save_image)
self._deferred_fit_in_view.connect(self._fit_in_view,
Qt.QueuedConnection)
self._deferred_fit_in_view.emit()
context.add_widget(self._widget)
self._force_refresh = False
def __init__(self, camera_topic='/image_raw'):
super(CameraView, self).__init__()
self._scene = QGraphicsScene()
self.bridge = CvBridge()
self._map_item = 0
self.image_changed.connect(self._update)
self._sub = rospy.Subscriber(camera_topic, Image, self.callback)
self.setScene(self._scene)
def __init__(self, context):
self._context = context
super(ConductorGraph, self).__init__(context)
self.initialised = False
self.setObjectName('Conductor Graph')
self._node_items = None
self._edge_items = None
self._node_item_events = {}
self._edge_item_events = {}
self._client_info_list = {}
self._widget = QWidget()
self.cur_selected_client_name = ""
self.pre_selected_client_name = ""
# factory builds generic dotcode items
self.dotcode_factory = PydotFactory()
# self.dotcode_factory=PygraphvizFactory()
self.dotcode_generator = ConductorGraphDotcodeGenerator()
self.dot_to_qt = DotToQtGenerator()
self._graph = ConductorGraphInfo(self._update_conductor_graph_relay,
self._set_network_statisics)
rospack = rospkg.RosPack()
ui_file = os.path.join(rospack.get_path('concert_conductor_graph'),
'ui', 'conductor_graph.ui')
loadUi(ui_file, self._widget,
{'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('ConductorGraphUi')
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() +
(' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.highlight_connections_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.auto_fit_graph_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.clusters_check_box.toggled.connect(
self._redraw_graph_view)
self.signal_deferred_fit_in_view.connect(self._fit_in_view,
Qt.QueuedConnection)
self.signal_deferred_fit_in_view.emit()
self._widget.tabWidget.currentChanged.connect(self._change_client_tab)
self.signal_update_conductor_graph.connect(
self._update_conductor_graph)
context.add_widget(self._widget)
def __init__(self, parent, len_timeline,
msg_callback,
color_callback,
pause_callback=None):
"""
@param msg_callback: This can be a function that takes either
{ DiagnosticArray, DiagnosticsStatus } as an argument. If your timeline
is to show the status of all devices, choose the former.
Else (eg. your timeline only handles the status of an arbitrary
node) use the latter.
@param color_callback: Not directly used within this class. Instead,
this will be passed and used in TimelineView class.
@todo: pause_callback is MUST since it's used in both by
RobotMonitorWidget & InspectorWindow.
"""
super(TimelinePane, self).__init__()
self._parent = parent
ui_file = os.path.join(os.path.dirname(os.path.realpath(__file__)),
'rqt_robot_monitor_timelinepane.ui')
loadUi(ui_file, self, {'TimelineView': TimelineView})
# Initialization process for TimeLineView obj continues.
self._pause_callback = pause_callback
self._msg_callback = msg_callback
self._timeline_view.set_init_data(1, len_timeline, 5,
# self._get_color_for_value)
color_callback)
self._scene = QGraphicsScene(self._timeline_view)
# self._colors = [QColor('green'), QColor('yellow'), QColor('red')]
self._timeline_view.setScene(self._scene)
self._timeline_view.show()
# self._messages = [None for x in range(len_timeline)] # DiagnosticStatus
# self._queue_diagnostic = [1 for x in range(20)]
self._queue_diagnostic = deque()
self._len_timeline = len_timeline
self._paused = False
self._tracking_latest = True
self._last_sec_marker_at = 2
self._last_msg = None
self._pause_button.clicked[bool].connect(self._pause)
self._sig_update.connect(self._timeline_view.slot_redraw)
def __init__(self,
map_topic='/map',
paths=[
'/move_base/SBPLLatticePlanner/plan',
'/move_base/TrajectoryPlannerROS/local_plan'
],
polygons=['/move_base/local_costmap/robot_footprint'],
tf=None,
parent=None):
super(NavView, self).__init__()
self._parent = parent
self.last_path = None
self.map_changed.connect(self._update)
self.destroyed.connect(self.close)
#ScrollHandDrag
self.setDragMode(QGraphicsView.ScrollHandDrag)
self._map = None
self._map_item = None
self.w = 0
self.h = 0
self._paths = {}
self._polygons = {}
self.path_changed.connect(self._update_path)
self.polygon_changed.connect(self._update_polygon)
self._colors = [(238, 34, 116), (68, 134, 252), (236, 228, 46),
(102, 224, 18), (242, 156, 6), (240, 64, 10),
(196, 30, 250)]
self._scene = QGraphicsScene()
if tf is None:
self._tf = tf.TransformListener()
else:
self._tf = tf
self.map_sub = rospy.Subscriber('/map', OccupancyGrid, self.map_cb)
for path in paths:
self.add_path(path)
for poly in polygons:
self.add_polygon(poly)
self.setScene(self._scene)
def __init__(self, context):
super(CapabilityGraph, self).__init__(context)
self.setObjectName('CapabilityGraph')
self.__current_dotcode = None
self.__running_providers = []
self.__spec_index = None
self.__widget = QWidget()
self.__dot_to_qt = DotToQtGenerator()
rp = rospkg.RosPack()
ui_file = os.path.join(rp.get_path('rqt_capabilities'), 'resources',
'CapabilityGraph.ui')
loadUi(
ui_file, self.__widget, {
'CapabilitiesInteractiveGraphicsView':
CapabilitiesInteractiveGraphicsView
})
self.__widget.setObjectName('CapabilityGraphUI')
if context.serial_number() > 1:
self.__widget.setWindowTitle(self.__widget.windowTitle() +
(' (%d)' % context.serial_number()))
self.__scene = QGraphicsScene()
self.__scene.setBackgroundBrush(Qt.white)
self.__widget.graphics_view.setScene(self.__scene)
self.__widget.refresh_graph_push_button.setIcon(
QIcon.fromTheme('view-refresh'))
self.__widget.refresh_graph_push_button.pressed.connect(
self.__refresh_view)
self.__refresh_view()
self.__deferred_fit_in_view.connect(self.__fit_in_view,
Qt.QueuedConnection)
self.__deferred_fit_in_view.emit()
self.__redraw_graph.connect(self.__update_capabilities_graph)
# TODO: use user provided server node name
rospy.Subscriber('/capability_server/events', CapabilityEvent,
self.__handle_event)
context.add_widget(self.__widget)
def __init__(self, timeline, parent):
super(ImageView, self).__init__(timeline, parent)
self._image = None
self._image_topic = None
self._image_stamp = None
self.quality = Image.NEAREST # quality hint for scaling
# TODO put the image_topic and image_stamp on the picture or display them in some fashion
self._overlay_font_size = 14.0
self._overlay_indent = (4, 4)
self._overlay_color = (0.2, 0.2, 1.0)
self._image_view = QGraphicsView(parent)
self._image_view.resizeEvent = self._resizeEvent
self._scene = QGraphicsScene()
self._image_view.setScene(self._scene)
parent.layout().addWidget(self._image_view)
def _redraw_graph_view(self):
key = str(self.get_current_message().header.stamp)
if key in self._scene_cache:
new_scene = self._scene_cache[key]
else: # cache miss
new_scene = QGraphicsScene()
new_scene.setBackgroundBrush(Qt.white)
if self._widget.highlight_connections_check_box.isChecked():
highlight_level = 3
else:
highlight_level = 1
# (nodes, edges) = self.dot_to_qt.graph_to_qt_items(self.dotcode_generator.graph,
# highlight_level)
# this function is very expensive
(nodes, edges) = self.dot_to_qt.dotcode_to_qt_items(
self._current_dotcode, highlight_level)
for node_item in iter(nodes.values()):
new_scene.addItem(node_item)
for edge_items in iter(edges.values()):
for edge_item in edge_items:
edge_item.add_to_scene(new_scene)
new_scene.setSceneRect(new_scene.itemsBoundingRect())
# put the scene in the cache
self._scene_cache[key] = new_scene
self._scene_cache_keys.append(key)
if len(self._scene_cache) > self._scene_cache_capacity:
oldest = self._scene_cache_keys[0]
del self._scene_cache[oldest]
self._scene_cache_keys.remove(oldest)
# after construction, set the scene and fit to the view
self._scene = new_scene
self._widget.graphics_view.setScene(self._scene)
self._widget.message_label.setText(self._tip_message)
if self._widget.auto_fit_graph_check_box.isChecked():
self._fit_in_view()
def __init__(self, parent=None):
super(QMapView, self).__init__()
self._parent = parent
self._scene = QGraphicsScene()
self.map_changed.connect(self._update_map)
self.destroyed.connect(self.close)
# ScrollHandDrag
self.setDragMode(QGraphicsView.ScrollHandDrag)
self._map = None
self.w = 0
self.h = 0
self._map_item = {}
self._polygons = {}
self._colors = [(238, 34, 116), (68, 134, 252), (236, 228, 46), (102, 224, 18), (242, 156, 6), (240, 64, 10), (196, 30, 250)]
self.setScene(self._scene)
def __init__(self, parent):
"""Cannot take args other than parent due to loadUi limitation."""
super(TimelineView, self).__init__()
self._parent = parent
self._timeline_marker = QIcon.fromTheme('system-search')
self._min = 0
self._max = 0
self._xpos_marker = 5
self._timeline_marker_width = 15
self._timeline_marker_height = 15
self._last_marker_at = 2
self.redraw.connect(self._slot_redraw)
self._timeline = None
self.setUpdatesEnabled(True)
self._scene = QGraphicsScene(self)
self.setScene(self._scene)
def __init__(self, context):
super(RosGraph, self).__init__(context)
self.initialized = False
self.setObjectName('RosGraph')
self._graph = None
self._current_dotcode = None
self._widget = QWidget()
# factory builds generic dotcode items
self.dotcode_factory = PydotFactory()
# self.dotcode_factory = PygraphvizFactory()
# generator builds rosgraph
self.dotcode_generator = RosGraphDotcodeGenerator()
# dot_to_qt transforms into Qt elements using dot layout
self.dot_to_qt = DotToQtGenerator()
ui_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'RosGraph.ui')
loadUi(ui_file, self._widget, {'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('RosGraphUi')
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() + (' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.graph_type_combo_box.insertItem(0, self.tr('Nodes only'), NODE_NODE_GRAPH)
self._widget.graph_type_combo_box.insertItem(1, self.tr('Nodes/Topics (active)'), NODE_TOPIC_GRAPH)
self._widget.graph_type_combo_box.insertItem(2, self.tr('Nodes/Topics (all)'), NODE_TOPIC_ALL_GRAPH)
self._widget.graph_type_combo_box.setCurrentIndex(0)
self._widget.graph_type_combo_box.currentIndexChanged.connect(self._refresh_rosgraph)
self.node_completionmodel = NamespaceCompletionModel(self._widget.filter_line_edit, False)
completer = RepeatedWordCompleter(self.node_completionmodel, self)
completer.setCompletionMode(QCompleter.PopupCompletion)
completer.setWrapAround(True)
completer.setCaseSensitivity(Qt.CaseInsensitive)
self._widget.filter_line_edit.editingFinished.connect(self._refresh_rosgraph)
self._widget.filter_line_edit.setCompleter(completer)
self.topic_completionmodel = NamespaceCompletionModel(self._widget.topic_filter_line_edit, False)
topic_completer = RepeatedWordCompleter(self.topic_completionmodel, self)
topic_completer.setCompletionMode(QCompleter.PopupCompletion)
topic_completer.setWrapAround(True)
topic_completer.setCaseSensitivity(Qt.CaseInsensitive)
self._widget.topic_filter_line_edit.editingFinished.connect(self._refresh_rosgraph)
self._widget.topic_filter_line_edit.setCompleter(topic_completer)
self._widget.namespace_cluster_check_box.clicked.connect(self._refresh_rosgraph)
self._widget.actionlib_check_box.clicked.connect(self._refresh_rosgraph)
self._widget.dead_sinks_check_box.clicked.connect(self._refresh_rosgraph)
self._widget.leaf_topics_check_box.clicked.connect(self._refresh_rosgraph)
self._widget.quiet_check_box.clicked.connect(self._refresh_rosgraph)
self._widget.refresh_graph_push_button.setIcon(QIcon.fromTheme('view-refresh'))
self._widget.refresh_graph_push_button.pressed.connect(self._update_rosgraph)
self._widget.highlight_connections_check_box.toggled.connect(self._redraw_graph_view)
self._widget.auto_fit_graph_check_box.toggled.connect(self._redraw_graph_view)
self._widget.fit_in_view_push_button.setIcon(QIcon.fromTheme('zoom-original'))
self._widget.fit_in_view_push_button.pressed.connect(self._fit_in_view)
self._widget.load_dot_push_button.setIcon(QIcon.fromTheme('document-open'))
self._widget.load_dot_push_button.pressed.connect(self._load_dot)
self._widget.save_dot_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_dot_push_button.pressed.connect(self._save_dot)
self._widget.save_as_svg_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_as_svg_push_button.pressed.connect(self._save_svg)
self._widget.save_as_image_push_button.setIcon(QIcon.fromTheme('image'))
self._widget.save_as_image_push_button.pressed.connect(self._save_image)
self._update_rosgraph()
self._deferred_fit_in_view.connect(self._fit_in_view, Qt.QueuedConnection)
self._deferred_fit_in_view.emit()
context.add_widget(self._widget)
class EsterelPlanViewerWidget(QWidget):
# plan view
_scene = QGraphicsScene()
_webview = QGraphicsWebView()
_svg = QGraphicsSvgItem()
_renderer = QSvgRenderer()
def __init__(self, plugin=None):
super(EsterelPlanViewerWidget, self).__init__()
# Create QWidget
ui_file = os.path.join(rospkg.RosPack().get_path('rosplan_rqt'),
'resource', 'esterel_plan_viewer.ui')
loadUi(ui_file, self,
{'InteractiveGraphicsView': InteractiveGraphicsView})
self.setObjectName('ROSPlanEsterelPlanViewer')
self.graphicsView.setScene(self._scene)
self._scene.addItem(self._svg)
self.refreshButton.clicked[bool].connect(self._handle_refresh_clicked)
self.saveButton.clicked[bool].connect(self._handle_save_button_clicked)
self._sub = rospy.Subscriber('/rosplan_plan_dispatcher/plan_graph',
String, self.planReceivedCallback)
self._plugin = plugin
# to store the graph msg received in callback, later on is used to save the graph if needed
self.graph = None
def planReceivedCallback(self, msg):
'''
updating plan view
'''
# save graph in member variable in case user clicks save button later
self.graph = msg.data
# render graph
temp_buffer_graph = pydot.graph_from_dot_data(self.graph)
svg_string = temp_buffer_graph.create_svg()
self._renderer.load(QByteArray(svg_string))
self._svg.setSharedRenderer(self._renderer)
def _handle_refresh_clicked(self, checked):
'''
called when the refresh button is clicked
'''
self._sub.unregister()
self._sub = rospy.Subscriber(self.topicText.text(), String,
self.planReceivedCallback)
def save_graph(self, full_path):
'''
check if last graph msg received is valid (non empty), then save in file.dot
'''
if self.graph:
dot_file = open(full_path, 'w')
dot_file.write(self.graph)
dot_file.close()
rospy.loginfo('graph saved succesfully in %s', full_path)
else:
# if self.graph is None it will fall in this case
rospy.logerr('Could not save Graph: is empty, currently subscribing to: %s, try' +\
' clicking "Update subscriber" button and make sure graph is published at least one time', self.topicText.text())
def _handle_save_button_clicked(self, checked):
'''
called when the save button is clicked
'''
rospy.loginfo('Saving esterel plan to dot file')
fileName = QFileDialog.getSaveFileName(
self, 'Save Esterel plan to dot file', '',
'Graph xdot Files (*.dot)')
if fileName[0] == '':
rospy.loginfo("User has cancelled saving process")
else:
# add .dot at the end of the filename
full_dot_path = fileName[0]
if not '.dot' in full_dot_path:
full_dot_path += '.dot'
rospy.loginfo("path to save dot file: %s", full_dot_path)
self.save_graph(full_dot_path)
# Qt methods
def shutdown_plugin(self):
pass
def save_settings(self, plugin_settings, instance_settings):
pass
def restore_settings(self, plugin_settings, instance_settings):
pass
def __init__(self, context):
super(GatewayGraph, self).__init__(context)
self.initialised = False
self.setObjectName('Gateway Graph')
self._current_dotcode = None
self._widget = QWidget()
# factory builds generic dotcode items
self.dotcode_factory = PydotFactory()
# self.dotcode_factory = PygraphvizFactory()
self.dotcode_generator = RosGraphDotcodeGenerator()
self.dot_to_qt = DotToQtGenerator()
self._graph = Graph()
rospack = rospkg.RosPack()
ui_file = os.path.join(rospack.get_path('rocon_gateway_graph'), 'ui',
'gateway_graph.ui')
#ui_file = os.path.join(os.path.dirname(os.path.realpath(__file__)), 'ui', 'gateway_graph.ui')
loadUi(ui_file, self._widget,
{'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('GatewayGraphUi')
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() +
(' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.refresh_graph_push_button.setIcon(
QIcon.fromTheme('view-refresh'))
self._widget.refresh_graph_push_button.pressed.connect(
self._update_gateway_graph)
self._widget.graph_type_combo_box.insertItem(0, self.tr('Gateways'),
GATEWAY_GATEWAY_GRAPH)
self._widget.graph_type_combo_box.insertItem(
1, self.tr('Pulled Connections'), GATEWAY_PULLED_GRAPH)
self._widget.graph_type_combo_box.insertItem(
2, self.tr('Flipped Connections'), GATEWAY_FLIPPED_GRAPH)
self._widget.graph_type_combo_box.setCurrentIndex(0)
self._widget.graph_type_combo_box.currentIndexChanged.connect(
self._refresh_rosgraph)
self.node_completionmodel = NamespaceCompletionModel(
self._widget.filter_line_edit, False)
completer = RepeatedWordCompleter(self.node_completionmodel, self)
completer.setCompletionMode(QCompleter.PopupCompletion)
completer.setWrapAround(True)
completer.setCaseSensitivity(Qt.CaseInsensitive)
self._widget.filter_line_edit.editingFinished.connect(
self._refresh_rosgraph)
self._widget.filter_line_edit.setCompleter(completer)
self.topic_completionmodel = NamespaceCompletionModel(
self._widget.topic_filter_line_edit, False)
topic_completer = RepeatedWordCompleter(self.topic_completionmodel,
self)
topic_completer.setCompletionMode(QCompleter.PopupCompletion)
topic_completer.setWrapAround(True)
topic_completer.setCaseSensitivity(Qt.CaseInsensitive)
self._widget.topic_filter_line_edit.editingFinished.connect(
self._refresh_rosgraph)
self._widget.topic_filter_line_edit.setCompleter(topic_completer)
self._widget.namespace_cluster_check_box.clicked.connect(
self._refresh_rosgraph)
self._widget.watchlist_check_box.clicked.connect(
self._refresh_rosgraph)
self._widget.all_advertisements_check_box.clicked.connect(
self._refresh_rosgraph)
self._widget.highlight_connections_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.auto_fit_graph_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.fit_in_view_push_button.setIcon(
QIcon.fromTheme('zoom-original'))
self._widget.fit_in_view_push_button.pressed.connect(self._fit_in_view)
self._widget.load_dot_push_button.setIcon(
QIcon.fromTheme('document-open'))
self._widget.load_dot_push_button.pressed.connect(self._load_dot)
self._widget.save_dot_push_button.setIcon(
QIcon.fromTheme('document-save-as'))
self._widget.save_dot_push_button.pressed.connect(self._save_dot)
self._widget.save_as_svg_push_button.setIcon(
QIcon.fromTheme('document-save-as'))
self._widget.save_as_svg_push_button.pressed.connect(self._save_svg)
self._widget.save_as_image_push_button.setIcon(
QIcon.fromTheme('image'))
self._widget.save_as_image_push_button.pressed.connect(
self._save_image)
self._update_gateway_graph()
self._deferred_fit_in_view.connect(self._fit_in_view,
Qt.QueuedConnection)
self._deferred_fit_in_view.emit()
context.add_widget(self._widget)
def __init__(self, context):
super(RosPackGraph, self).__init__(context)
self.initialized = False
self._current_dotcode = None
self._update_thread = WorkerThread(self._update_thread_run, self._update_finished)
self._nodes = {}
self._edges = {}
self._options = {}
self._options_serialized = ''
self.setObjectName('RosPackGraph')
rospack = rospkg.RosPack()
rosstack = rospkg.RosStack()
# factory builds generic dotcode items
self.dotcode_factory = PydotFactory()
# self.dotcode_factory = PygraphvizFactory()
# generator builds rosgraph
self.dotcode_generator = RosPackageGraphDotcodeGenerator(rospack, rosstack)
# dot_to_qt transforms into Qt elements using dot layout
self.dot_to_qt = DotToQtGenerator()
self._widget = QWidget()
rp = rospkg.RosPack()
ui_file = os.path.join(rp.get_path('rqt_dep'), 'resource', 'RosPackGraph.ui')
loadUi(ui_file, self._widget, {'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('RosPackGraphUi')
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() + (' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.depth_combo_box.insertItem(0, self.tr('infinite'), -1)
self._widget.depth_combo_box.insertItem(1, self.tr('1'), 2)
self._widget.depth_combo_box.insertItem(2, self.tr('2'), 3)
self._widget.depth_combo_box.insertItem(3, self.tr('3'), 4)
self._widget.depth_combo_box.insertItem(4, self.tr('4'), 5)
self._widget.depth_combo_box.currentIndexChanged.connect(self._refresh_rospackgraph)
self._widget.directions_combo_box.insertItem(0, self.tr('depends'), 0)
self._widget.directions_combo_box.insertItem(1, self.tr('depends_on'), 1)
self._widget.directions_combo_box.insertItem(2, self.tr('both'), 2)
self._widget.directions_combo_box.currentIndexChanged.connect(self._refresh_rospackgraph)
self._widget.package_type_combo_box.insertItem(0, self.tr('wet & dry'), 3)
self._widget.package_type_combo_box.insertItem(1, self.tr('wet only'), 2)
self._widget.package_type_combo_box.insertItem(2, self.tr('dry only'), 1)
self._widget.package_type_combo_box.currentIndexChanged.connect(self._refresh_rospackgraph)
completionmodel = StackageCompletionModel(self._widget.filter_line_edit, rospack, rosstack)
completer = RepeatedWordCompleter(completionmodel, self)
completer.setCompletionMode(QCompleter.PopupCompletion)
completer.setWrapAround(True)
completer.setCaseSensitivity(Qt.CaseInsensitive)
self._widget.filter_line_edit.editingFinished.connect(self._refresh_rospackgraph)
self._widget.filter_line_edit.setCompleter(completer)
self._widget.filter_line_edit.selectionChanged.connect(self._clear_filter)
self._widget.with_stacks_check_box.clicked.connect(self._refresh_rospackgraph)
self._widget.mark_check_box.clicked.connect(self._refresh_rospackgraph)
self._widget.colorize_check_box.clicked.connect(self._refresh_rospackgraph)
self._widget.hide_transitives_check_box.clicked.connect(self._refresh_rospackgraph)
self._widget.refresh_graph_push_button.setIcon(QIcon.fromTheme('view-refresh'))
self._widget.refresh_graph_push_button.pressed.connect(self._update_rospackgraph)
self._widget.highlight_connections_check_box.toggled.connect(self._refresh_rospackgraph)
self._widget.auto_fit_graph_check_box.toggled.connect(self._refresh_rospackgraph)
self._widget.fit_in_view_push_button.setIcon(QIcon.fromTheme('zoom-original'))
self._widget.fit_in_view_push_button.pressed.connect(self._fit_in_view)
self._widget.load_dot_push_button.setIcon(QIcon.fromTheme('document-open'))
self._widget.load_dot_push_button.pressed.connect(self._load_dot)
self._widget.save_dot_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_dot_push_button.pressed.connect(self._save_dot)
self._widget.save_as_svg_push_button.setIcon(QIcon.fromTheme('document-save-as'))
self._widget.save_as_svg_push_button.pressed.connect(self._save_svg)
self._widget.save_as_image_push_button.setIcon(QIcon.fromTheme('image'))
self._widget.save_as_image_push_button.pressed.connect(self._save_image)
self._deferred_fit_in_view.connect(self._fit_in_view, Qt.QueuedConnection)
self._deferred_fit_in_view.emit()
context.add_widget(self._widget)
# If in either of following case, this turnes True
# - 1st filtering key is already input by user
# - filtering key is restored
self._filtering_started = False
def __init__(self, context):
self._context = context
super(TeleopApp, self).__init__(context)
self.initialised = False
self.setObjectName('Teleop App')
self.is_setting_dlg_live = False
self._widget = QWidget()
rospack = rospkg.RosPack()
ui_file = os.path.join(rospack.get_path('concert_teleop_app'), 'ui',
'teleop_app.ui')
self._widget.setObjectName('TeleopAppUi')
loadUi(ui_file, self._widget, {})
if context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() +
(' (%d)' % context.serial_number()))
#list item click event
self._widget.robot_list_tree_widget.itemClicked.connect(
self._select_robot_list_tree_item)
self._widget.robot_list_tree_widget.itemDoubleClicked.connect(
self._dbclick_robot_list_item)
#button event connection
self._widget.backward_btn.setAutoRepeat(True)
self._widget.forward_btn.setAutoRepeat(True)
self._widget.left_turn_btn.setAutoRepeat(True)
self._widget.right_turn_btn.setAutoRepeat(True)
self._widget.backward_btn.clicked.connect(self._backward)
self._widget.forward_btn.clicked.connect(self._forward)
self._widget.left_turn_btn.clicked.connect(self._left_turn)
self._widget.right_turn_btn.clicked.connect(self._right_turn)
self._widget.backward_btn.released.connect(self._stop)
self._widget.forward_btn.released.connect(self._stop)
self._widget.left_turn_btn.released.connect(self._stop)
self._widget.right_turn_btn.released.connect(self._stop)
self._widget.capture_teleop_btn.pressed.connect(self._capture_teleop)
self._widget.release_teleop_btn.pressed.connect(self._release_teleop)
#signal event connection
self._widget.destroyed.connect(self._exit)
self._update_robot_list_signal.connect(self._update_robot_list)
self.connect(self, SIGNAL("capture"),
self._show_capture_teleop_message)
self.connect(self, SIGNAL("release"),
self._show_release_teleop_message)
self.connect(self, SIGNAL("error"), self._show_error_teleop_message)
context.add_widget(self._widget)
#init
self.scene = QGraphicsScene()
self._widget.camera_view.setScene(self.scene)
self.timer = QTimer(self._widget)
self.timer.timeout.connect(self._display_image)
self.timer.start(self.CAMERA_FPS)
self._widget.release_teleop_btn.setEnabled(False)
self.teleop_app_info = TeleopAppInfo()
self.teleop_app_info._reg_event_callback(self._refresh_robot_list)
self.teleop_app_info._reg_capture_event_callback(
self._capture_event_callback)
self.teleop_app_info._reg_release_event_callback(
self._release_event_callback)
self.teleop_app_info._reg_error_event_callback(
self._error_event_callback)
self.robot_item_list = {}
self.current_robot = None
self.current_captured_robot = None
def __init__(self, context):
super(RosBehaviourTree, self).__init__(context)
self.setObjectName('RosBehaviourTree')
parser = argparse.ArgumentParser()
RosBehaviourTree.add_arguments(parser, False)
# if the context doesn't have an argv attribute then assume we're running with --no-roscore
if not hasattr(context, 'argv'):
args = sys.argv[1:]
# Can run the viewer with or without live updating. Running without is
# intended for viewing of bags only
self.live_update = False
else:
args = context.argv()
self.live_update = True
parsed_args = parser.parse_args(args)
self.context = context
self.initialized = False
self._current_dotcode = None # dotcode for the tree that is currently displayed
self._viewing_bag = False # true if a bag file is loaded
# True if next or previous buttons are pressed. Reset if the tree being
# viewed is the last one in the list.
self._browsing_timeline = False
self._widget = QWidget()
# factory builds generic dotcode items
self.dotcode_factory = PygraphvizFactory() # PydotFactory()
# self.dotcode_factory = PygraphvizFactory()
# generator builds rosgraph
self.dotcode_generator = RosBehaviourTreeDotcodeGenerator()
self.current_topic = None
self.behaviour_sub = None
self._tip_message = None # message of the tip of the tree
self._saved_settings_topic = None # topic subscribed to by previous instance
self.visibility_level = py_trees.common.VisibilityLevel.DETAIL
# dot_to_qt transforms into Qt elements using dot layout
self.dot_to_qt = DotToQtGenerator()
rp = rospkg.RosPack()
ui_file = os.path.join(rp.get_path('rqt_py_trees'), 'resource',
'RosBehaviourTree.ui')
loadUi(ui_file, self._widget,
{'InteractiveGraphicsView': InteractiveGraphicsView})
self._widget.setObjectName('RosBehaviourTreeUi')
if hasattr(context, 'serial_number') and context.serial_number() > 1:
self._widget.setWindowTitle(self._widget.windowTitle() +
(' (%d)' % context.serial_number()))
self._scene = QGraphicsScene()
self._scene.setBackgroundBrush(Qt.white)
self._widget.graphics_view.setScene(self._scene)
self._widget.highlight_connections_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.auto_fit_graph_check_box.toggled.connect(
self._redraw_graph_view)
self._widget.fit_in_view_push_button.setIcon(
QIcon.fromTheme('zoom-original'))
self._widget.fit_in_view_push_button.pressed.connect(self._fit_in_view)
self._widget.load_bag_push_button.setIcon(
QIcon.fromTheme('document-open'))
self._widget.load_bag_push_button.pressed.connect(self._load_bag)
self._widget.load_dot_push_button.setIcon(
QIcon.fromTheme('document-open'))
self._widget.load_dot_push_button.pressed.connect(self._load_dot)
self._widget.save_dot_push_button.setIcon(
QIcon.fromTheme('document-save-as'))
self._widget.save_dot_push_button.pressed.connect(self._save_dot)
self._widget.save_as_svg_push_button.setIcon(
QIcon.fromTheme('document-save-as'))
self._widget.save_as_svg_push_button.pressed.connect(self._save_svg)
self._widget.save_as_image_push_button.setIcon(
QIcon.fromTheme('image'))
self._widget.save_as_image_push_button.pressed.connect(
self._save_image)
for text in visibility.combo_to_py_trees:
self._widget.visibility_level_combo_box.addItem(text)
self._widget.visibility_level_combo_box.setCurrentIndex(
self.visibility_level)
self._widget.visibility_level_combo_box.currentIndexChanged[
'QString'].connect(self._update_visibility_level)
# set up the function that is called whenever the box is resized -
# ensures that the timeline is correctly drawn.
self._widget.resizeEvent = self._resize_event
self._timeline = None
self._timeline_listener = None
# Connect the message changed function of this object to a corresponding
# signal. This signal will be activated whenever the message being
# viewed changes.
self._message_changed.connect(self.message_changed)
self._message_cleared.connect(self.message_cleared)
# Set up combo box for topic selection
# when the refresh_combo signal happens, update the combo topics available
self._refresh_combo.connect(self._update_combo_topics)
# filter events to catch the event which opens the combo box
self._combo_event_filter = RosBehaviourTree.ComboBoxEventFilter(
self._refresh_combo)
self._widget.topic_combo_box.installEventFilter(
self._combo_event_filter)
self._widget.topic_combo_box.activated.connect(self._choose_topic)
self._update_combo_topics()
# Set up navigation buttons
self._widget.previous_tool_button.pressed.connect(self._previous)
self._widget.previous_tool_button.setIcon(
QIcon.fromTheme('go-previous'))
self._widget.next_tool_button.pressed.connect(self._next)
self._widget.next_tool_button.setIcon(QIcon.fromTheme('go-next'))
self._widget.first_tool_button.pressed.connect(self._first)
self._widget.first_tool_button.setIcon(QIcon.fromTheme('go-first'))
self._widget.last_tool_button.pressed.connect(self._last)
self._widget.last_tool_button.setIcon(QIcon.fromTheme('go-last'))
# play, pause and stop buttons
self._widget.play_tool_button.pressed.connect(self._play)
self._widget.play_tool_button.setIcon(
QIcon.fromTheme('media-playback-start'))
self._widget.stop_tool_button.pressed.connect(self._stop)
self._widget.stop_tool_button.setIcon(
QIcon.fromTheme('media-playback-stop'))
# also connect the navigation buttons so that they stop the timer when
# pressed while the tree is playing.
self._widget.first_tool_button.pressed.connect(self._stop)
self._widget.previous_tool_button.pressed.connect(self._stop)
self._widget.last_tool_button.pressed.connect(self._stop)
self._widget.next_tool_button.pressed.connect(self._stop)
# set up shortcuts for navigation (vim)
next_shortcut_vi = QShortcut(QKeySequence("l"), self._widget)
next_shortcut_vi.activated.connect(
self._widget.next_tool_button.pressed)
previous_shortcut_vi = QShortcut(QKeySequence("h"), self._widget)
previous_shortcut_vi.activated.connect(
self._widget.previous_tool_button.pressed)
first_shortcut_vi = QShortcut(QKeySequence("^"), self._widget)
first_shortcut_vi.activated.connect(
self._widget.first_tool_button.pressed)
last_shortcut_vi = QShortcut(QKeySequence("$"), self._widget)
last_shortcut_vi.activated.connect(
self._widget.last_tool_button.pressed)
# shortcuts for emacs
next_shortcut_emacs = QShortcut(QKeySequence("Ctrl+f"), self._widget)
next_shortcut_emacs.activated.connect(
self._widget.next_tool_button.pressed)
previous_shortcut_emacs = QShortcut(QKeySequence("Ctrl+b"),
self._widget)
previous_shortcut_emacs.activated.connect(
self._widget.previous_tool_button.pressed)
first_shortcut_emacs = QShortcut(QKeySequence("Ctrl+a"), self._widget)
first_shortcut_emacs.activated.connect(
self._widget.first_tool_button.pressed)
last_shortcut_emacs = QShortcut(QKeySequence("Ctrl+e"), self._widget)
last_shortcut_emacs.activated.connect(
self._widget.last_tool_button.pressed)
# set up stuff for dotcode cache
self._dotcode_cache_capacity = 50
self._dotcode_cache = {}
# cache is ordered on timestamps from messages, but earliest timestamp
# isn't necessarily the message that was viewed the longest time ago, so
# need to store keys
self._dotcode_cache_keys = []
# set up stuff for scene cache (dotcode cache doesn't seem to make much difference)
self._scene_cache_capacity = 50
self._scene_cache = {}
self._scene_cache_keys = []
# Update the timeline buttons to correspond with a completely
# uninitialised state.
self._set_timeline_buttons(first_snapshot=False,
previous_snapshot=False,
next_snapshot=False,
last_snapshot=False)
self._deferred_fit_in_view.connect(self._fit_in_view,
Qt.QueuedConnection)
self._deferred_fit_in_view.emit()
# This is used to store a timer which controls how fast updates happen when the play button is pressed.
self._play_timer = None
# updates the view
self._refresh_view.connect(self._refresh_tree_graph)
self._force_refresh = False
if self.live_update:
context.add_widget(self._widget)
else:
self.initialized = True # this needs to be set for trees to be displayed
context.setCentralWidget(self._widget)
if parsed_args.bag:
self._load_bag(parsed_args.bag)
elif parsed_args.latest_bag:
# if the latest bag is requested, load it from the default directory, or
# the one specified in the args
bag_dir = parsed_args.bag_dir or os.getenv(
'ROS_HOME', os.path.expanduser('~/.ros')) + '/behaviour_trees'
self.open_latest_bag(bag_dir, parsed_args.by_time)
