def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = ImagesTabWidgetView(parent)
else:
self.view = view
self.image_names_dict = dict()
self.populate_select_image_combo_box_and_update_view()
self.view.select_image_combo_box.currentIndexChanged.connect(
self.update_view)
def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = ImagesTabWidgetView(parent)
else:
self.view = view
# This connection is here so that when the view is updated below the min/max spin boxes have the correct range
# for the scale.
self.view.scale_combo_box.currentTextChanged.connect(self.scale_changed)
self.image_names_dict = dict()
self.populate_select_image_combo_box_and_update_view()
self.view.select_image_combo_box.currentIndexChanged.connect(
self.update_view)
class ImagesTabWidgetPresenter:
def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = ImagesTabWidgetView(parent)
else:
self.view = view
self.image_names_dict = dict()
self.populate_select_image_combo_box_and_update_view()
self.view.select_image_combo_box.currentIndexChanged.connect(
self.update_view)
def apply_properties(self):
props = self.view.get_properties()
# if only one colorbar apply settings to all images
if len(get_colorbars_from_fig(self.fig)) == 1:
# flatten the values into one list
images = sum(self.image_names_dict.values(), [])
else:
images = self.get_selected_image()
for image in images:
if image.colorbar:
image.colorbar.set_label(props.label)
image.set_cmap(props.colormap)
if props.interpolation and not (isinstance(
image, QuadMesh) or isinstance(image, Poly3DCollection)):
image.set_interpolation(props.interpolation)
update_colorbar_scale(self.fig, image, SCALES[props.scale],
props.vmin, props.vmax)
if props.vmin > props.vmax:
self.view.max_min_value_warning.setVisible(True)
self.view.max_min_value_warning.setText(
"<html> <head/> <body> <p> <span style=\"color:#ff0000;\">Max "
"value is less than min value so they have been "
"swapped.</span></p></body></html>")
elif props.vmin == props.vmax:
self.view.max_min_value_warning.setVisible(True)
self.view.max_min_value_warning.setText(
"<html><head/><body><p><span style=\"color:#ff0000;\">Min and max "
"value are the same so they have been "
"adjusted.</span></p></body></html>")
else:
self.view.max_min_value_warning.setVisible(False)
def get_selected_image(self):
return self.image_names_dict[self.view.get_selected_image_name()]
def populate_select_image_combo_box_and_update_view(self):
with block_signals(self.view.select_image_combo_box):
self._populate_select_image_combo_box()
self.update_view()
def update_view(self):
img_props = ImageProperties.from_image(self.get_selected_image())
self.view.label_line_edit.setEnabled(
bool(get_colorbars_from_fig(self.fig)))
self.view.set_label(img_props.label)
self.view.set_colormap(img_props.colormap)
self.view.set_reverse_colormap(img_props.reverse_colormap)
self.view.set_min_value(img_props.vmin)
self.view.set_max_value(img_props.vmax)
if img_props.interpolation:
self.view.enable_interpolation(True)
self.view.set_interpolation(img_props.interpolation)
else:
self.view.set_interpolation('None')
self.view.enable_interpolation(False)
self.view.set_scale(img_props.scale)
@staticmethod
def generate_image_name(image):
"""Generate a name for an image"""
label = image.get_label().lstrip('_')
ax_name = generate_ax_name(image.axes)
if label:
return "{} - {}".format(ax_name, label)
else:
return "{}".format(ax_name)
@staticmethod
def set_name_in_names_dict(name, value, name_dict):
"""Set name of image in image_names_dict"""
idx = 1
base_name = name
while name in name_dict:
name = base_name + "({})".format(idx)
idx += 1
name_dict[name] = value
return name_dict
def _populate_select_image_combo_box(self):
self.view.select_image_combo_box.clear()
for img in get_images_from_fig(self.fig):
self.image_names_dict = self.set_name_in_names_dict(
self.generate_image_name(img[0]), img, self.image_names_dict)
self.view.populate_select_image_combo_box(
sorted(self.image_names_dict.keys()))
class ImagesTabWidgetPresenter:
def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = ImagesTabWidgetView(parent)
else:
self.view = view
self.image_names_dict = dict()
self.populate_select_image_combo_box_and_update_view()
# Signals
self.view.select_image_combo_box.currentIndexChanged.connect(
self.update_view)
def apply_properties(self):
props = self.view.get_properties()
image = self.get_selected_image()
self.set_selected_image_label(props.label)
image.set_cmap(props.colormap)
if props.vmin == 0 and props.scale == "Logarithmic":
props.vmin += 1e-6 # Avoid 0 log scale error
image.set_norm(SCALES[props.scale](vmin=props.vmin, vmax=props.vmax))
if props.interpolation:
image.set_interpolation(props.interpolation)
def get_selected_image(self):
return self.image_names_dict[self.view.get_selected_image_name()]
def populate_select_image_combo_box_and_update_view(self):
with block_signals(self.view.select_image_combo_box):
self._populate_select_image_combo_box()
self.update_view()
def set_selected_image_label(self, label):
image = self.image_names_dict.pop(self.view.get_selected_image_name())
image.set_label(label)
new_name = self.generate_image_name(image)
self.image_names_dict[new_name] = image
self.view.replace_selected_image_name(new_name)
def update_view(self):
# self.view.update(ImageProperties.from_image(self.get_selected_image()))
img_props = ImageProperties.from_image(self.get_selected_image())
self.view.set_label(img_props.label)
self.view.set_colormap(img_props.colormap)
self.view.set_min_value(img_props.vmin)
self.view.set_max_value(img_props.vmax)
if img_props.interpolation:
self.view.enable_interpolation(True)
self.view.set_interpolation(img_props.interpolation)
else:
self.view.set_interpolation('None')
self.view.enable_interpolation(False)
self.view.set_scale(img_props.scale)
@staticmethod
def generate_image_name(image):
"""Generate a name for an image"""
label = image.get_label()
ax_name = generate_ax_name(image.axes)
if label:
return "{} - {}".format(ax_name, label)
else:
return "{}".format(ax_name)
@staticmethod
def set_name_in_names_dict(name, value, name_dict):
"""Set name of image in image_names_dict"""
idx = 1
base_name = name
while name in name_dict:
name = base_name + "({})".format(idx)
idx += 1
name_dict[name] = value
return name_dict
def _populate_select_image_combo_box(self):
self.view.select_image_combo_box.clear()
for img in get_images_from_fig(self.fig):
self.image_names_dict = self.set_name_in_names_dict(
self.generate_image_name(img), img, self.image_names_dict)
self.view.populate_select_image_combo_box(
sorted(self.image_names_dict.keys()))
class ImagesTabWidgetPresenter:
def __init__(self, fig, view=None, parent=None):
self.fig = fig
if not view:
self.view = ImagesTabWidgetView(parent)
else:
self.view = view
# This connection is here so that when the view is updated below the min/max spin boxes have the correct range
# for the scale.
self.view.scale_combo_box.currentTextChanged.connect(
self.scale_changed)
self.image_names_dict = dict()
self.populate_select_image_combo_box_and_update_view()
self.view.select_image_combo_box.currentIndexChanged.connect(
self.update_view)
def apply_properties(self):
props = self.view.get_properties()
image = self.get_selected_image()
self.set_selected_image_label(props.label)
image.set_cmap(props.colormap)
if props.interpolation:
image.set_interpolation(props.interpolation)
update_colorbar_scale(self.fig, image, SCALES[props.scale], props.vmin,
props.vmax)
if props.vmin > props.vmax:
self.view.max_min_value_warning.setVisible(True)
self.view.max_min_value_warning.setText(
"<html> <head/> <body> <p> <span style=\"color:#ff0000;\">Max "
"value is less than min value so they have been "
"swapped.</span></p></body></html>")
elif props.vmin == props.vmax:
self.view.max_min_value_warning.setVisible(True)
self.view.max_min_value_warning.setText(
"<html><head/><body><p><span style=\"color:#ff0000;\">Min and max "
"value are the same so they have been "
"adjusted.</span></p></body></html>")
else:
self.view.max_min_value_warning.setVisible(False)
def get_selected_image(self):
return self.image_names_dict[self.view.get_selected_image_name()]
def populate_select_image_combo_box_and_update_view(self):
with block_signals(self.view.select_image_combo_box):
self._populate_select_image_combo_box()
self.update_view()
def set_selected_image_label(self, label):
image = self.image_names_dict.pop(self.view.get_selected_image_name())
image.set_label(label)
new_name = self.generate_image_name(image)
self.image_names_dict[new_name] = image
self.view.replace_selected_image_name(new_name)
def update_view(self):
img_props = ImageProperties.from_image(self.get_selected_image())
self.view.set_label(img_props.label)
self.view.set_colormap(img_props.colormap)
self.view.set_reverse_colormap(img_props.reverse_colormap)
self.view.set_min_value(img_props.vmin)
self.view.set_max_value(img_props.vmax)
if img_props.interpolation:
self.view.enable_interpolation(True)
self.view.set_interpolation(img_props.interpolation)
else:
self.view.set_interpolation('None')
self.view.enable_interpolation(False)
self.view.set_scale(img_props.scale)
@staticmethod
def generate_image_name(image):
"""Generate a name for an image"""
label = image.get_label()
ax_name = generate_ax_name(image.axes)
if label:
return "{} - {}".format(ax_name, label)
else:
return "{}".format(ax_name)
@staticmethod
def set_name_in_names_dict(name, value, name_dict):
"""Set name of image in image_names_dict"""
idx = 1
base_name = name
while name in name_dict:
name = base_name + "({})".format(idx)
idx += 1
name_dict[name] = value
return name_dict
def _populate_select_image_combo_box(self):
self.view.select_image_combo_box.clear()
for img in get_images_from_fig(self.fig):
self.image_names_dict = self.set_name_in_names_dict(
self.generate_image_name(img), img, self.image_names_dict)
self.view.populate_select_image_combo_box(
sorted(self.image_names_dict.keys()))
def scale_changed(self, scale):
self.view.set_min_max_ranges(scale)