Newer
Older
# Mantid Repository : https://github.com/mantidproject/mantid
#
# Copyright © 2019 ISIS Rutherford Appleton Laboratory UKRI,
# NScD Oak Ridge National Laboratory, European Spallation Source
# & Institut Laue - Langevin
# SPDX - License - Identifier: GPL - 3.0 +
# This file is part of the mantid workbench.
from __future__ import (absolute_import, unicode_literals)
from mantidqt.widgets.plotconfigdialog import generate_ax_name, get_axes_names_dict
from mantidqt.widgets.plotconfigdialog.axestabwidget import AxProperties
from mantidqt.widgets.plotconfigdialog.axestabwidget.view import AxesTabWidgetView
class AxesTabWidgetPresenter:
def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = AxesTabWidgetView(parent)
else:
self.view = view
# Dictionary mapping ax name to Axes object
self.axes_names_dict = get_axes_names_dict(self.fig)
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# Add axes names to "select axes" combo box
self.populate_select_axes_combo_box()
# Display top axes' properties in input fields
self.set_selected_ax_view_properties()
# Signals
self.view.select_axes_combo_box.currentIndexChanged.connect(
self.set_selected_ax_view_properties
)
def apply_properties(self):
"""Update the axes with the user inputted properties"""
ax = self.get_current_ax()
new_props = self.view.get_properties()
self.set_ax_title(ax, new_props.title)
ax.set_xlim(new_props.xlim)
ax.set_xlabel(new_props.xlabel)
ax.set_xscale(new_props.xscale)
ax.set_ylim(new_props.ylim)
ax.set_ylabel(new_props.ylabel)
ax.set_yscale(new_props.yscale)
def get_current_ax(self):
"""Get Axes object of currently selected axes"""
return self.axes_names_dict[self.view.get_selected_ax_name()]
def get_current_ax_name(self):
"""Get name of currently selected axes"""
return self.view.get_selected_ax_name()
def get_current_ax_properties(self):
"""Get axes properties from currently selected axes"""
return AxProperties.from_ax_object(self.get_current_ax())
def populate_select_axes_combo_box(self):
"""Add axes' names to the 'select axes' combo box"""
# Sort names by axes position
names = sorted(self.axes_names_dict.keys(),
key=lambda x: x[x.rfind("("):])
self.view.populate_select_axes_combo_box(names)
def rename_current_axes(self, new_name):
"""
Rename the current axes, updating the axes_names_dict and
the select_axes_combo_box
"""
old_name = self.get_current_ax_name()
self.view.set_current_axes_selector_text(new_name)
self.axes_names_dict[new_name] = self.axes_names_dict.pop(old_name)
def set_ax_title(self, ax, new_title):
"""Set axes' title and update its entry in the axes selector"""
ax.set_title(new_title)
self.rename_current_axes(generate_ax_name(ax))
def set_selected_ax_view_properties(self):
"""Update the properties in the view from the selected axes"""
ax_props = self.get_current_ax_properties()
self.view.set_properties(ax_props)