launchtree_plugin.py

Go to the documentation of this file.

#!/usr/bin/env python
import rospy
from rqt_gui_py.plugin import Plugin

from python_qt_binding.QtCore import Qt
from python_qt_binding.QtGui import QInputDialog

from rqt_launchtree.launchtree_widget import LaunchtreeWidget


class LaunchtreePlugin(Plugin):

    _SETTING_LASTPKG = 'last_pkg'
    _SETTING_LASTLAUNCHFILE = 'last_launch'
    _SETTING_LASTLAUNCHARGS = 'last_args'

    def __init__(self, context):
        super(LaunchtreePlugin, self).__init__(context)

        self._widget = LaunchtreeWidget(context)
        if context.serial_number() > 1:
            self._widget.setWindowTitle(self._widget.windowTitle() +
                                        (' (%d)' % context.serial_number()))
        context.add_widget(self._widget)

    def shutdown_plugin(self):
        self._widget.shutdown()

    def save_settings(self, plugin_settings, instance_settings):
        instance_settings.set_value('editor', self._widget.editor)
        _curr_pkg = self._widget.package_select.currentText()
        rospy.logdebug('save_settings) currentIndex={}'.format(_curr_pkg))
        instance_settings.set_value(self._SETTING_LASTPKG, _curr_pkg)
        instance_settings.set_value(self._SETTING_LASTLAUNCHFILE, self._widget.launchfile_select.currentText())
        instance_settings.set_value(self._SETTING_LASTLAUNCHARGS, self._widget.args_input.text())

    def restore_settings(self, plugin_settings, instance_settings):
        self._widget.editor = instance_settings.value('editor', 'gedit')
        self._widget.args_input.setText(instance_settings.value(self._SETTING_LASTLAUNCHARGS, ''))
        pkg_idx = self._widget.package_select.findText(instance_settings.value(self._SETTING_LASTPKG))
        if pkg_idx >= 0:
            self._widget.package_select.blockSignals(True)
            self._widget.package_select.setCurrentIndex(pkg_idx)
            self._widget.package_select.blockSignals(False)
            self._widget.update_launchfiles(pkg_idx)
            # only set launch file if pkg was restored
            launch_idx = self._widget.launchfile_select.findText(instance_settings.value(self._SETTING_LASTLAUNCHFILE))
            if launch_idx >= 0:
                self._widget.launchfile_select.blockSignals(True)
                self._widget.launchfile_select.setCurrentIndex(launch_idx)
                self._widget.launchfile_select.blockSignals(False)
        self._widget.block_load(False)
        self._widget.load_launchfile()

    def trigger_configuration(self):
        (text, ok) = QInputDialog.getText(self._widget,
            'Settings for %s' % self._widget.windowTitle(),
            'Command to edit launch files (vim, gedit, ...), can accept args:',
            text = self._widget.editor
        )
        if ok:
            self._widget.editor = text