def test_curve_has_all_errorbars_on_replot_after_error_every_increase(self): fig = figure() ax = fig.add_subplot(111) curve = ax.errorbar([0, 1, 2, 4], [0, 1, 2, 4], yerr=[0.1, 0.2, 0.3, 0.4]) new_curve = CurvesTabWidgetPresenter._replot_mpl_curve(ax, curve, {'errorevery': 2}) self.assertEqual(2, len(new_curve[2][0].get_segments())) new_curve = CurvesTabWidgetPresenter._replot_mpl_curve(ax, new_curve, {'errorevery': 1}) self.assertTrue(hasattr(new_curve, 'errorbar_data')) self.assertEqual(4, len(new_curve[2][0].get_segments()))
def _generate_presenter(self, mock_view=None, fig=None): if not mock_view: mock_view = Mock(get_selected_ax_name=lambda: "Axes 0: (0, 0)", get_selected_curve_name=lambda: "Workspace") if not fig: fig = self.fig return CurvesTabWidgetPresenter(fig=fig, view=mock_view)
def __init__(self, fig, view=None, parent=None): self.fig = fig if view: self.view = view else: self.view = PlotConfigDialogView(parent) self.view.show() self.tab_widget_presenters = [None, None, None] self.tab_widget_views = [None, None, None] # Axes tab if len(self.fig.get_axes()) > 0: axes_tab = AxesTabWidgetPresenter(self.fig, parent=self.view) self.tab_widget_presenters[0] = axes_tab self.tab_widget_views[0] = (axes_tab.view, "Axes") # Curves tab if curve_in_figure(self.fig): curves_tab = CurvesTabWidgetPresenter(self.fig, parent=self.view) self.tab_widget_presenters[1] = curves_tab self.tab_widget_views[1] = (curves_tab.view, "Curves") # Images tab if image_in_figure(self.fig): images_tab = ImagesTabWidgetPresenter(self.fig, parent=self.view) self.tab_widget_presenters[2] = images_tab self.tab_widget_views[2] = (images_tab.view, "Images") self._add_tab_widget_views() # Signals self.view.ok_button.clicked.connect(self.apply_properties_and_exit) self.view.apply_button.clicked.connect(self.apply_properties) self.view.cancel_button.clicked.connect(self.exit)
def toggle_error_bars_for(ax, curve, make_visible=None): # get all curve properties curve_props = CurveProperties.from_curve(curve) # and remove the ones that matplotlib doesn't recognise plot_kwargs = curve_props.get_plot_kwargs() new_curve = CurvesTabWidgetPresenter.replot_curve( ax, curve, plot_kwargs) # Inverts either the current state of hide_errors # or the make_visible kwarg that forces a state: # If make visible is True, then hide_errors must be False # for the intended effect curve_props.hide_errors = not curve_props.hide_errors if make_visible is None else not make_visible CurvesTabWidgetPresenter.toggle_errors(new_curve, curve_props) CurvesTabWidgetPresenter.update_limits_and_legend(ax)
def test_curves_and_legend_tabs_close_when_all_curves_removed(self): fig, ax = subplots() ax.plot([[0], [0]], label='line1') ax.legend() mock_view = Mock(get_selected_ax_name=lambda: "(0, 0)", get_current_curve_name=lambda: "line1", get_selected_curves_names=lambda: ["line1"]) mock_view.select_curve_list=Mock(selectedItems=lambda: [], count=lambda: 0) mock_view.select_axes_combo_box.count = Mock(return_value=0) parent_presenter_mock = Mock() legend_presenter = LegendTabWidgetPresenter(fig=fig, view=mock_view, parent_presenter=parent_presenter_mock) curves_presenter =CurvesTabWidgetPresenter( fig=fig, view=mock_view, parent_presenter=parent_presenter_mock, legend_tab=legend_presenter) curves_presenter.remove_selected_curves() curves_presenter.parent_presenter.forget_tab_from_presenter.assert_has_calls( [call(curves_presenter), call(legend_presenter)]) mock_view.close.assert_called()
def __init__(self, fig, view=None, parent=None): self.fig = fig if view: self.view = view else: self.view = PlotConfigDialogView(parent) self.view.show() self.tab_widget_presenters = [None, None, None, None] self.tab_widget_views = [None, None, None, None] legend_tab = None # Axes tab if len(self.fig.get_axes()) > 0: axes_tab = AxesTabWidgetPresenter( self.fig, parent=self.view, success_callback=self.success_callback, error_callback=self.error_callback) self.tab_widget_presenters[1] = axes_tab self.tab_widget_views[0] = (axes_tab.view, "Axes") # Legend tab if legend_in_figure(self.fig): legend_tab = LegendTabWidgetPresenter(self.fig, parent_view=self.view, parent_presenter=self) self.tab_widget_presenters[0] = legend_tab self.tab_widget_views[3] = (legend_tab.view, "Legend") # Curves tab if curve_in_figure(self.fig): curves_tab = CurvesTabWidgetPresenter(self.fig, parent_view=self.view, parent_presenter=self, legend_tab=legend_tab) self.tab_widget_presenters[2] = curves_tab self.tab_widget_views[1] = (curves_tab.view, "Curves") # Images tab if image_in_figure(self.fig): images_tab = ImagesTabWidgetPresenter(self.fig, parent=self.view) self.tab_widget_presenters[3] = images_tab self.tab_widget_views[2] = (images_tab.view, "Images") self._add_tab_widget_views() # Signals self.view.ok_button.clicked.connect(self.apply_properties_and_exit) self.view.apply_button.clicked.connect(self.apply_properties) self.view.cancel_button.clicked.connect(self.exit) self.view.help_button.clicked.connect(self.open_help_window) self.error_state = False
def add_error_bars_menu(self, parent_menu, ax): """ Add menu actions to toggle the errors for all lines in the plot. Lines without errors are added in the context menu first, then lines containing errors are appended afterwards. This is done so that the context menu always has the same order of curves as the legend is currently showing - and the legend always appends curves with errors after the lines without errors. Relevant source, as of 10 July 2019: https://github.com/matplotlib/matplotlib/blob/154922992722db37a9d9c8680682ccc4acf37f8c/lib/matplotlib/legend.py#L1201 :param parent_menu: The menu to which the actions will be added :type parent_menu: QMenu :param ax: The Axes containing lines to toggle errors on """ # if the ax is not a MantidAxes, and there are no errors plotted, # then do not add any options for the menu if not isinstance(ax, MantidAxes) and len(ax.containers) == 0: return error_bars_menu = QMenu(self.ERROR_BARS_MENU_TEXT, parent_menu) error_bars_menu.addAction( self.SHOW_ERROR_BARS_BUTTON_TEXT, partial(self._update_plot_after, self._toggle_all_errors, ax, make_visible=True)) error_bars_menu.addAction( self.HIDE_ERROR_BARS_BUTTON_TEXT, partial(self._update_plot_after, self._toggle_all_errors, ax, make_visible=False)) parent_menu.addMenu(error_bars_menu) self.active_lines = CurvesTabWidgetPresenter.get_curves_from_ax(ax) # if there's more than one line plotted, then # add a sub menu, containing an action to hide the # error bar for each line error_bars_menu.addSeparator() add_later = [] for index, line in enumerate(self.active_lines): if curve_has_errors(line): curve_props = CurveProperties.from_curve(line) # Add lines without errors first, lines with errors are appended later. Read docstring for more info if not isinstance(line, ErrorbarContainer): action = error_bars_menu.addAction( line.get_label(), partial(self._update_plot_after, self.toggle_error_bars_for, ax, line)) action.setCheckable(True) action.setChecked(not curve_props.hide_errors) else: add_later.append( (line.get_label(), partial(self._update_plot_after, self.toggle_error_bars_for, ax, line), not curve_props.hide_errors)) for label, function, visible in add_later: action = error_bars_menu.addAction(label, function) action.setCheckable(True) action.setChecked(visible)