from glue.core.subset import roi_to_subset_state
from glue.core.util import update_ticks
from glue.core.roi_pretransforms import FullSphereLongitudeTransform, 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
from glue.viewers.matplotlib.viewer import SimpleMatplotlibViewer
from glue.viewers.scatter.state import ScatterViewerState
from glue.viewers.scatter.layer_artist import ScatterLayerArtist
__all__ = ['MatplotlibScatterMixin', 'SimpleScatterViewer']
[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._y_limits_to_mpl)
self.state.add_callback('y_max', self._y_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'
[docs] def using_fullsphere(self):
return self.state.plot_mode in ['aitoff', 'hammer', 'lambert', 'mollweide']
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()
elif self.using_fullsphere():
return
self.limits_to_mpl()
def _y_limits_to_mpl(self, *args, **kwargs):
if self.using_fullsphere():
return
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_full_sphere:
transform = FullSphereLongitudeTransform(next_transform=transform)
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)
[docs]class SimpleScatterViewer(MatplotlibScatterMixin, SimpleMatplotlibViewer):
_state_cls = ScatterViewerState
_data_artist_cls = ScatterLayerArtist
_subset_artist_cls = ScatterLayerArtist
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
MatplotlibScatterMixin.setup_callbacks(self)
