コード例 #1
0
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(
            self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(
            self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(
            self._do_show_algorithm_history)

        self.workspacewidget.workspaceDoubleClicked.connect(
            self._action_double_click_workspace)
コード例 #2
0
ファイル: workspacewidget.py プロジェクト: gemmaguest/mantid
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=False))

        self.workspacewidget.plotMDHistoClicked.connect(
            partial(self._do_plot_1d_md, errors=False, overplot=False))
        self.workspacewidget.overplotMDHistoClicked.connect(
            partial(self._do_plot_1d_md, errors=False, overplot=True))
        self.workspacewidget.plotMDHistoWithErrorsClicked.connect(
            partial(self._do_plot_1d_md, errors=True, overplot=False))
        self.workspacewidget.overplotMDHistoWithErrorsClicked.connect(
            partial(self._do_plot_1d_md, errors=True, overplot=True))

        self.workspacewidget.plotBinClicked.connect(
            partial(self._do_plot_bin, errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(self._do_show_algorithm_history)
        self.workspacewidget.showDetectorsClicked.connect(self._do_show_detectors)
        self.workspacewidget.plotAdvancedClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=False, advanced=True))
        self.workspacewidget.plotSurfaceClicked.connect(
            partial(self._do_plot_3D, plot_type='surface'))
        self.workspacewidget.plotWireframeClicked.connect(
            partial(self._do_plot_3D, plot_type='wireframe'))
        self.workspacewidget.plotContourClicked.connect(
            partial(self._do_plot_3D, plot_type='contour'))
        self.workspacewidget.sampleMaterialClicked.connect(self._do_sample_material)
        self.workspacewidget.superplotClicked.connect(self._do_superplot)
        self.workspacewidget.superplotWithErrsClicked.connect(
                self._do_superplot_with_errors)
        self.workspacewidget.superplotBinsClicked.connect(self._do_superplot_bins)
        self.workspacewidget.superplotBinsWithErrsClicked.connect(
                self._do_superplot_bins_with_errors)
        self.workspacewidget.contextMenuAboutToShow.connect(
            self._on_context_menu)

        self.workspacewidget.workspaceDoubleClicked.connect(self._action_double_click_workspace)
コード例 #3
0
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(partial(self._do_plot_spectrum,
                                                                 errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(partial(self._do_plot_spectrum,
                                                                     errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(partial(self._do_plot_spectrum,
                                                                           errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(partial(self._do_plot_spectrum,
                                                                               errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(self._do_show_algorithm_history)

        self.workspacewidget.workspaceDoubleClicked.connect(self._action_double_click_workspace)
コード例 #4
0
ファイル: workspacewidget.py プロジェクト: PeterParker/mantid
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(
            functools.partial(self._do_plot_spectrum,
                              errors=False,
                              overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(
            functools.partial(self._do_plot_spectrum,
                              errors=False,
                              overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(
            functools.partial(self._do_plot_spectrum,
                              errors=True,
                              overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(
            functools.partial(self._do_plot_spectrum,
                              errors=True,
                              overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(
            self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
コード例 #5
0
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        # layout
        workspacewidget = WorkspaceTreeWidget(MantidTreeModel())
        self.workspacewidget = workspacewidget
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        workspacewidget.plotSpectrumClicked.connect(
            functools.partial(self._do_plot_spectrum, errors=False))
        workspacewidget.plotSpectrumWithErrorsClicked.connect(
            functools.partial(self._do_plot_spectrum, errors=True))
        workspacewidget.plotColorfillClicked.connect(self._do_plot_colorfill)
コード例 #6
0
class WorkspaceWidget(PluginWidget):
    """Provides a Workspace Widget for workspace manipulation"""

    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(partial(self._do_plot_spectrum,
                                                                 errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(partial(self._do_plot_spectrum,
                                                                     errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(partial(self._do_plot_spectrum,
                                                                           errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(partial(self._do_plot_spectrum,
                                                                               errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(self._do_show_algorithm_history)

        self.workspacewidget.workspaceDoubleClicked.connect(self._action_double_click_workspace)

    # ----------------- Plugin API --------------------

    def register_plugin(self):
        self.main.add_dockwidget(self)

    def get_plugin_title(self):
        return "Workspaces"

    def readSettings(self, _):
        pass

    def writeSettings(self, _):
        pass

    # ----------------- Behaviour --------------------

    def _do_plot_spectrum(self, names, errors, overplot):
        """
        Plot spectra from the selected workspaces

        :param names: A list of workspace names
        :param errors: If true then error bars will be plotted on the points
        :param overplot: If true then the add to the current figure if one
                         exists and it is a compatible figure
        """
        if overplot:
            compatible, error_msg = can_overplot()
            if not compatible:
                QMessageBox.warning(self, "", error_msg)
                return

        plot_from_names(names, errors, overplot)

    def _do_plot_colorfill(self, names):
        """
        Plot a colorfill from the selected workspaces

        :param names: A list of workspace names
        """
        try:
            pcolormesh(self._ads.retrieveWorkspaces(names, unrollGroups=True))
        except BaseException:
            import traceback
            traceback.print_exc()

    def _do_sample_logs(self, names):
        """
        Show the sample log window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                SampleLogs(ws=ws, parent=self)
            except Exception as exception:
                logger.warning("Could not open sample logs for workspace '{}'."
                               "".format(ws.name()))
                logger.debug("{}: {}".format(type(exception).__name__,
                                             exception))

    def _do_slice_viewer(self, names):
        """
        Show the sliceviewer window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                SliceViewer(ws=ws, parent=self)
            except Exception as exception:
                logger.warning("Could not open slice viewer for workspace '{}'."
                               "".format(ws.name()))
                logger.debug("{}: {}".format(type(exception).__name__,
                                             exception))

    def _do_show_instrument(self, names):
        """
        Show an instrument widget for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            if ws.getInstrument().getName():
                try:
                    presenter = InstrumentViewPresenter(ws, parent=self)
                    presenter.show_view()
                except Exception as exception:
                    logger.warning("Could not show instrument for workspace "
                                   "'{}':\n{}.\n".format(ws.name(), exception))
            else:
                logger.warning("Could not show instrument for workspace '{}':"
                               "\nNo instrument available.\n"
                               "".format(ws.name()))

    def _do_show_data(self, names):
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                MatrixWorkspaceDisplay.supports(ws)
                # the plot function is being injected in the presenter
                # this is done so that the plotting library is mockable in testing
                presenter = MatrixWorkspaceDisplay(ws, plot=plot, parent=self)
                presenter.show_view()
            except ValueError:
                try:
                    TableWorkspaceDisplay.supports(ws)
                    presenter = TableWorkspaceDisplay(ws, plot=matplotlib.pyplot, parent=self)
                    presenter.show_view()
                except ValueError:
                    logger.error(
                        "Could not open workspace: {0} with neither "
                        "MatrixWorkspaceDisplay nor TableWorkspaceDisplay."
                        "".format(ws.name()))

    def _do_show_algorithm_history(self, names):
        for name in names:
            if not isinstance(self._ads.retrieve(name), WorkspaceGroup):
                try:
                    AlgorithmHistoryWindow(self, name).show()
                except Exception as exception:
                    logger.warning("Could not open history of '{}'. "
                                   "".format(name))
                    logger.warning("{}: {}".format(type(exception).__name__, exception))

    def _action_double_click_workspace(self, name):
        self._do_show_data([name])

    def refresh_workspaces(self):
        self.workspacewidget.refreshWorkspaces()
コード例 #7
0
ファイル: workspacewidget.py プロジェクト: ethoeng/mantid
class WorkspaceWidget(PluginWidget):
    """Provides a Workspace Widget for workspace manipulation"""
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=False))
        self.workspacewidget.plotBinClicked.connect(
            partial(self._do_plot_bin, errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(
            self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(
            self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(
            self._do_show_algorithm_history)
        self.workspacewidget.showDetectorsClicked.connect(
            self._do_show_detectors)
        self.workspacewidget.plotAdvancedClicked.connect(
            partial(self._do_plot_spectrum,
                    errors=False,
                    overplot=False,
                    advanced=True))
        self.workspacewidget.plotSurfaceClicked.connect(
            partial(self._do_plot_3D, plot_type='surface'))
        self.workspacewidget.plotWireframeClicked.connect(
            partial(self._do_plot_3D, plot_type='wireframe'))
        self.workspacewidget.plotContourClicked.connect(
            partial(self._do_plot_3D, plot_type='contour'))

        self.workspacewidget.workspaceDoubleClicked.connect(
            self._action_double_click_workspace)

    # ----------------- Plugin API --------------------

    def register_plugin(self):
        self.main.add_dockwidget(self)

    def get_plugin_title(self):
        return "Workspaces"

    def readSettings(self, _):
        pass

    def writeSettings(self, _):
        pass

    # ----------------- Behaviour --------------------

    def _do_plot_spectrum(self, names, errors, overplot, advanced=False):
        """
        Plot spectra from the selected workspaces

        :param names: A list of workspace names
        :param errors: If true then error bars will be plotted on the points
        :param overplot: If true then the add to the current figure if one
                         exists and it is a compatible figure
        :param advanced: If true then the advanced options will be shown in
                         the spectra selector dialog.
        """
        if overplot:
            compatible, error_msg = can_overplot()
            if not compatible:
                QMessageBox.warning(self, "", error_msg)
                return
        try:
            plot_from_names(names, errors, overplot, advanced=advanced)
        except RuntimeError as re:
            logger.error(str(re))

    def _do_plot_bin(self, names, errors, overplot):
        """
        Plot a single bin from the selected workspaces

        :param names: A list of workspace names
        :param errors: If true then error bars will be plotted on the points
        :param overplot: If true then the add to the current figure if one
                         exists and it is a compatible figure
        """
        if overplot:
            compatible, error_msg = can_overplot()
            if not compatible:
                QMessageBox.warning(self, "", error_msg)
                return
        plot_kwargs = {"axis": MantidAxType.BIN}
        plot(self._ads.retrieveWorkspaces(names, unrollGroups=True),
             errors=errors,
             overplot=overplot,
             wksp_indices=[0],
             plot_kwargs=plot_kwargs)

    def _do_plot_colorfill(self, names):
        """
        Plot a colorfill from the selected workspaces

        :param names: A list of workspace names
        :param contour: An optional bool for whether to draw contour lines.
        """
        try:
            pcolormesh(names)
        except BaseException:
            import traceback
            traceback.print_exc()

    def _do_plot_3D(self, workspaces, plot_type):
        """
        Make a 3D plot from the selected workspace.

        :param workspaces: A list of workspace names.
        :param plot_type: The type of 3D plot, either 'surface', 'wireframe', or 'contour'.
        """
        plot_function = getattr(functions, f'plot_{plot_type}', None)

        if plot_function is None:
            return

        try:
            plot_function(workspaces)
        except RuntimeError as re:
            logger.error(str(re))

    def _do_sample_logs(self, names):
        """
        Show the sample log window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                SampleLogs(ws=ws, parent=self)
            except Exception as exception:
                logger.warning("Could not open sample logs for workspace '{}'."
                               "".format(ws.name()))
                logger.warning("{}: {}".format(
                    type(exception).__name__, exception))

    def _do_slice_viewer(self, names):
        """
        Show the sliceviewer window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                presenter = SliceViewer(ws=ws, parent=self)
                presenter.view.show()
            except Exception as exception:
                logger.warning(
                    "Could not open slice viewer for workspace '{}'."
                    "".format(ws.name()))
                logger.warning("{}: {}".format(
                    type(exception).__name__, exception))

    def _do_show_instrument(self, names):
        """
        Show an instrument widget for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            if ws.getInstrument().getName():
                try:
                    presenter = InstrumentViewPresenter(ws, parent=self)
                    presenter.show_view()
                except Exception as exception:
                    logger.warning("Could not show instrument for workspace "
                                   "'{}':\n{}\n".format(ws.name(), exception))
            else:
                logger.warning("Could not show instrument for workspace '{}':"
                               "\nNo instrument available.\n"
                               "".format(ws.name()))

    def _do_show_data(self, names):
        # local import to allow this module to be imported without pyplot being imported
        import matplotlib.pyplot
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                MatrixWorkspaceDisplay.supports(ws)
                # the plot function is being injected in the presenter
                # this is done so that the plotting library is mockable in testing
                presenter = MatrixWorkspaceDisplay(ws, plot=plot, parent=self)
                presenter.show_view()
            except ValueError:
                try:
                    TableWorkspaceDisplay.supports(ws)
                    presenter = TableWorkspaceDisplay(ws,
                                                      plot=matplotlib.pyplot,
                                                      parent=self)
                    presenter.show_view()
                except ValueError:
                    logger.error(
                        "Could not open workspace: {0} with neither "
                        "MatrixWorkspaceDisplay nor TableWorkspaceDisplay."
                        "".format(ws.name()))

    def _do_show_algorithm_history(self, names):
        for name in names:
            if not isinstance(self._ads.retrieve(name), WorkspaceGroup):
                try:
                    AlgorithmHistoryWindow(self, name).show()
                except Exception as exception:
                    logger.warning("Could not open history of '{}'. "
                                   "".format(name))
                    logger.warning("{}: {}".format(
                        type(exception).__name__, exception))

    def _do_show_detectors(self, names):
        successful_workspaces = []
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                task = BlockingAsyncTaskWithCallback(
                    self._run_create_detector_table, [ws],
                    blocking_cb=QApplication.processEvents)
                task.start()
            except RuntimeError:
                continue
            else:
                successful_workspaces.append(ws.name())

        self._do_show_data(
            list(map(lambda x: x + "-Detectors", successful_workspaces)))

    def _run_create_detector_table(self, ws):
        CreateDetectorTable(InputWorkspace=ws)

    def _action_double_click_workspace(self, name):
        ws = self._ads.retrieve(name)
        try:
            # if this is a table workspace (or peaks workspace),
            # then it can't be plotted automatically, so the data is shown instead
            TableWorkspaceDisplay.supports(ws)
            self._do_show_data([name])
        except ValueError:
            if hasattr(ws, 'blocksize') and ws.blocksize() == 1:
                #If this is ws is just a single value show the data, else plot the bin
                if hasattr(ws, 'getNumberHistograms'
                           ) and ws.getNumberHistograms() == 1:
                    self._do_show_data([name])
                else:
                    plot_kwargs = {"axis": MantidAxType.BIN}
                    plot([ws],
                         errors=False,
                         overplot=False,
                         wksp_indices=[0],
                         plot_kwargs=plot_kwargs)
            else:
                plot_from_names([name],
                                errors=False,
                                overplot=False,
                                show_colorfill_btn=True)

    def refresh_workspaces(self):
        self.workspacewidget.refreshWorkspaces()
コード例 #8
0
 def test_widget_creation(self):
     param = MantidTreeModel()
     widget = WorkspaceTreeWidget(param)
     self.assertTrue(widget is not None)
コード例 #9
0
class WorkspaceWidget(PluginWidget):
    """Provides a Workspace Widget for workspace manipulation"""
    def __init__(self, parent):
        super(WorkspaceWidget, self).__init__(parent)

        self._ads = AnalysisDataService.Instance()

        # layout
        self.workspacewidget = WorkspaceTreeWidget()
        layout = QVBoxLayout()
        layout.addWidget(self.workspacewidget)
        self.setLayout(layout)

        # behaviour
        self.workspacewidget.plotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=False))
        self.workspacewidget.overplotSpectrumClicked.connect(
            partial(self._do_plot_spectrum, errors=False, overplot=True))
        self.workspacewidget.plotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=False))
        self.workspacewidget.overplotSpectrumWithErrorsClicked.connect(
            partial(self._do_plot_spectrum, errors=True, overplot=True))
        self.workspacewidget.plotColorfillClicked.connect(
            self._do_plot_colorfill)
        self.workspacewidget.sampleLogsClicked.connect(self._do_sample_logs)
        self.workspacewidget.sliceViewerClicked.connect(self._do_slice_viewer)
        self.workspacewidget.showDataClicked.connect(self._do_show_data)
        self.workspacewidget.showInstrumentClicked.connect(
            self._do_show_instrument)
        self.workspacewidget.showAlgorithmHistoryClicked.connect(
            self._do_show_algorithm_history)

        self.workspacewidget.workspaceDoubleClicked.connect(
            self._action_double_click_workspace)

    # ----------------- Plugin API --------------------

    def register_plugin(self):
        self.main.add_dockwidget(self)

    def get_plugin_title(self):
        return "Workspaces"

    def readSettings(self, _):
        pass

    def writeSettings(self, _):
        pass

    # ----------------- Behaviour --------------------

    def _do_plot_spectrum(self, names, errors, overplot):
        """
        Plot spectra from the selected workspaces

        :param names: A list of workspace names
        :param errors: If true then error bars will be plotted on the points
        :param overplot: If true then the add to the current figure if one
                         exists and it is a compatible figure
        """
        if overplot:
            compatible, error_msg = can_overplot()
            if not compatible:
                QMessageBox.warning(self, "", error_msg)
                return

        plot_from_names(names, errors, overplot)

    def _do_plot_colorfill(self, names):
        """
        Plot a colorfill from the selected workspaces

        :param names: A list of workspace names
        """
        try:
            pcolormesh(self._ads.retrieveWorkspaces(names, unrollGroups=True))
        except BaseException:
            import traceback
            traceback.print_exc()

    def _do_sample_logs(self, names):
        """
        Show the sample log window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                SampleLogs(ws=ws, parent=self)
            except Exception as exception:
                logger.warning("Could not open sample logs for workspace '{}'."
                               "".format(ws.name()))
                logger.debug("{}: {}".format(
                    type(exception).__name__, exception))

    def _do_slice_viewer(self, names):
        """
        Show the sliceviewer window for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                SliceViewer(ws=ws, parent=self)
            except Exception as exception:
                logger.warning(
                    "Could not open slice viewer for workspace '{}'."
                    "".format(ws.name()))
                logger.debug("{}: {}".format(
                    type(exception).__name__, exception))

    def _do_show_instrument(self, names):
        """
        Show an instrument widget for the given workspaces

        :param names: A list of workspace names
        """
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            if ws.getInstrument().getName():
                try:
                    presenter = InstrumentViewPresenter(ws, parent=self)
                    presenter.show_view()
                except Exception as exception:
                    logger.warning("Could not show instrument for workspace "
                                   "'{}':\n{}.\n".format(ws.name(), exception))
            else:
                logger.warning("Could not show instrument for workspace '{}':"
                               "\nNo instrument available.\n"
                               "".format(ws.name()))

    def _do_show_data(self, names):
        for ws in self._ads.retrieveWorkspaces(names, unrollGroups=True):
            try:
                MatrixWorkspaceDisplay.supports(ws)
                # the plot function is being injected in the presenter
                # this is done so that the plotting library is mockable in testing
                presenter = MatrixWorkspaceDisplay(ws, plot=plot, parent=self)
                presenter.show_view()
            except ValueError:
                try:
                    TableWorkspaceDisplay.supports(ws)
                    presenter = TableWorkspaceDisplay(ws,
                                                      plot=matplotlib.pyplot,
                                                      parent=self)
                    presenter.show_view()
                except ValueError:
                    logger.error(
                        "Could not open workspace: {0} with neither "
                        "MatrixWorkspaceDisplay nor TableWorkspaceDisplay."
                        "".format(ws.name()))

    def _do_show_algorithm_history(self, names):
        for name in names:
            if not isinstance(self._ads.retrieve(name), WorkspaceGroup):
                try:
                    AlgorithmHistoryWindow(self, name).show()
                except Exception as exception:
                    logger.warning("Could not open history of '{}'. "
                                   "".format(name))
                    logger.warning("{}: {}".format(
                        type(exception).__name__, exception))

    def _action_double_click_workspace(self, name):
        self._do_show_data([name])

    def refresh_workspaces(self):
        self.workspacewidget.refreshWorkspaces()
コード例 #10
0
 def test_widget_creation(self):
     widget = WorkspaceTreeWidget()
     self.assertTrue(widget is not None)
コード例 #11
0
 def test_widget_creation(self):
     widget = WorkspaceTreeWidget()
     self.assertNotEqual(widget, None)
     sip.delete(widget)