Source code for glue.viewers.scatter.viewer

from glue.core.subset import roi_to_subset_state
from glue.core.util import update_ticks
from glue.core.roi_pretransforms import ProjectionMplTransform, RadianTransform

from glue.utils import mpl_to_datetime64
from glue.viewers.scatter.compat import update_scatter_viewer_state
from glue.viewers.matplotlib.mpl_axes import init_mpl


__all__ = ['MatplotlibScatterMixin']


[docs]class MatplotlibScatterMixin(object):
[docs] def setup_callbacks(self): self.state.add_callback('x_att', self._update_axes) self.state.add_callback('y_att', self._update_axes) self.state.add_callback('x_log', self._update_axes) self.state.add_callback('y_log', self._update_axes) self.state.add_callback('plot_mode', self._update_projection) self.state.add_callback('angle_unit', self._update_angle_unit) self.state.add_callback('x_min', self._x_limits_to_mpl) self.state.add_callback('x_max', self._x_limits_to_mpl) self.state.add_callback('y_min', self.limits_to_mpl) self.state.add_callback('y_max', self.limits_to_mpl) self.state.add_callback('x_axislabel', self._update_polar_ticks) self.state.add_callback('y_axislabel', self._update_polar_ticks) self._update_axes()
def _update_ticks(self, *args): radians = hasattr(self.state, 'angle_unit') and self.state.angle_unit == 'radians' if self.state.x_att is not None: # Update ticks, which sets the labels to categories if components are categorical update_ticks(self.axes, 'x', self.state.x_kinds, self.state.x_log, self.state.x_categories, projection=self.state.plot_mode, radians=radians, label=self.state.x_axislabel) if self.state.y_att is not None: # Update ticks, which sets the labels to categories if components are categorical update_ticks(self.axes, 'y', self.state.y_kinds, self.state.y_log, self.state.y_categories, projection=self.state.plot_mode, radians=radians, label=self.state.y_axislabel) def _update_axes(self, *args): self._update_ticks(args) if self.state.x_att is not None: self.state.x_axislabel = self.state.x_att.label if self.state.y_att is not None: self.state.y_axislabel = self.state.y_att.label self.axes.figure.canvas.draw_idle() def _update_polar_ticks(self, *args): if self.using_polar(): self._update_ticks(args) self.axes.figure.canvas.draw_idle() def _update_projection(self, *args): self.figure.delaxes(self.axes) _, self.axes = init_mpl(self.figure, projection=self.state.plot_mode) self.remove_all_toolbars() self.initialize_toolbar() for layer in self.layers: layer._set_axes(self.axes) layer.state.vector_mode = 'Cartesian' layer.state._update_points_mode() layer.update() self.axes.callbacks.connect('xlim_changed', self.limits_from_mpl) self.axes.callbacks.connect('ylim_changed', self.limits_from_mpl) self.update_x_axislabel() self.update_y_axislabel() self.update_x_ticklabel() self.update_y_ticklabel() # Reset and roundtrip the limits to have reasonable and synced limits when changing self.state.x_log = self.state.y_log = False self.state.reset_limits() if self.using_polar(): self.state.full_circle() self.limits_to_mpl() self.limits_from_mpl() # We need to update the tick marks # to account for the radians/degrees switch in polar mode # Also need to add/remove axis labels as necessary self._update_axes() self.figure.canvas.draw_idle()
[docs] def using_rectilinear(self): return self.state.plot_mode == 'rectilinear'
[docs] def using_polar(self): return self.state.plot_mode == 'polar'
def _update_angle_unit(self, *args): self._update_axes() for layer in self.layers: layer.update() def _x_limits_to_mpl(self, *args, **kwargs): if self.using_polar(): self.state.full_circle() self.limits_to_mpl()
[docs] def update_x_axislabel(self, *event): if self.using_polar(): self.axes.set_xlabel("") else: super().update_x_axislabel()
# Because of how the polar plot is drawn, we need to give the y-axis label more padding # Since this mixin is first in the MRO, we can 'override' the MatplotlibDataViewer method
[docs] def update_y_axislabel(self, *event): if self.using_polar(): self.axes.set_ylabel("") else: self.axes.set_ylabel(self.state.y_axislabel, weight=self.state.y_axislabel_weight, size=self.state.y_axislabel_size, labelpad=None) self.redraw()
[docs] def apply_roi(self, roi, override_mode=None): # Force redraw to get rid of ROI. We do this because applying the # subset state below might end up not having an effect on the viewer, # for example there may not be any layers, or the active subset may not # be one of the layers. So we just explicitly redraw here to make sure # a redraw will happen after this method is called. self.redraw() if len(self.layers) == 0: return x_date = 'datetime' in self.state.x_kinds y_date = 'datetime' in self.state.y_kinds if x_date or y_date: roi = roi.transformed(xfunc=mpl_to_datetime64 if x_date else None, yfunc=mpl_to_datetime64 if y_date else None) use_transform = not self.using_rectilinear() subset_state = roi_to_subset_state(roi, x_att=self.state.x_att, x_categories=self.state.x_categories, y_att=self.state.y_att, y_categories=self.state.y_categories, use_pretransform=use_transform) if use_transform: transform = ProjectionMplTransform(self.state.plot_mode, self.axes.get_xlim(), self.axes.get_ylim(), self.axes.get_xscale(), self.axes.get_yscale()) # If we're using degrees, we need to staple on the degrees -> radians conversion beforehand if self.state.using_degrees: coords = ['x'] if self.using_polar() else ['x', 'y'] transform = RadianTransform(coords=coords, next_transform=transform) subset_state.pretransform = transform self.apply_subset_state(subset_state, override_mode=override_mode)
[docs] @staticmethod def update_viewer_state(rec, context): return update_scatter_viewer_state(rec, context)