from glue.core import BaseData
from glue_jupyter.bqplot.image import BqplotImageView
from glue_jupyter.bqplot.profile import BqplotProfileView
from glue_jupyter.table import TableViewer
from specutils import Spectrum1D
import astropy
from astropy import units as u
from astropy.utils.introspection import minversion
from jdaviz.core.events import (AddDataToViewerMessage,
RemoveDataFromViewerMessage,
TableClickMessage)
from jdaviz.core.registries import viewer_registry
__all__ = ['MosvizProfileView', 'MosvizImageView', 'MosvizProfile2DView',
'MosvizTableViewer']
[docs]@viewer_registry("mosviz-profile-viewer", label="Profile 1D (Mosviz)")
class MosvizProfileView(BqplotProfileView):
default_class = Spectrum1D
[docs] def data(self, cls=None):
return [layer_state.layer.get_object(cls=cls or self.default_class)
for layer_state in self.state.layers
if hasattr(layer_state, 'layer') and
isinstance(layer_state.layer, BaseData)]
[docs] def set_plot_axes(self):
data = self.data()[0]
# Set axes labels for the spectrum viewer
if not minversion(astropy, '4.3'):
spectral_axis_unit_type = data.spectral_axis.unit.physical_type.title()
else:
# physical_type changed from str to class in astropy 4.3
spectral_axis_unit_type = str(data.spectral_axis.unit.physical_type).title()
# flux_unit_type = data.flux.unit.physical_type.title()
flux_unit_type = "Flux density"
if data.spectral_axis.unit.is_equivalent(u.m):
spectral_axis_unit_type = "Wavelength"
elif data.spectral_axis.unit.is_equivalent(u.pixel):
spectral_axis_unit_type = "pixel"
label_0 = f"{spectral_axis_unit_type} [{data.spectral_axis.unit.to_string()}]"
self.figure.axes[0].label = label_0
self.figure.axes[1].label = f"{flux_unit_type} [{data.flux.unit.to_string()}]"
# Make it so y axis label is not covering tick numbers.
self.figure.axes[1].label_offset = "-50"
# Set Y-axis to scientific notation
self.figure.axes[1].tick_format = '0.1e'
[docs]@viewer_registry("mosviz-image-viewer", label="Image 2D (Mosviz)")
class MosvizImageView(BqplotImageView):
default_class = None
[docs] def data(self, cls=None):
return [layer_state.layer # .get_object(cls=cls or self.default_class)
for layer_state in self.state.layers
if hasattr(layer_state, 'layer') and
isinstance(layer_state.layer, BaseData)]
[docs] def set_plot_axes(self):
self.figure.axes[1].tick_format = None
self.figure.axes[0].tick_format = None
self.figure.axes[1].label = "y: pixels"
self.figure.axes[0].label = "x: pixels"
# Make it so y axis label is not covering tick numbers.
self.figure.axes[1].label_offset = "-50"
[docs]@viewer_registry("mosviz-profile-2d-viewer", label="Spectrum 2D (Mosviz)")
class MosvizProfile2DView(BqplotImageView):
# Due to limitations in CCDData and 2D data that has spectral and spatial
# axes, the default conversion class must handle cubes
default_class = Spectrum1D
tools = ['bqplot:panzoom_x']
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
# Setup viewer option defaults
self.state.aspect = 'auto'
[docs] def data(self, cls=None):
return [layer_state.layer.get_object(cls=cls or self.default_class)
for layer_state in self.state.layers
if hasattr(layer_state, 'layer') and
isinstance(layer_state.layer, BaseData)]
[docs] def set_plot_axes(self):
self.figure.axes[0].visible = False
self.figure.axes[1].label = "y: pixels"
self.figure.axes[1].tick_format = None
self.figure.axes[1].label_location = "start"
# Make it so y axis label is not covering tick numbers.
self.figure.axes[1].label_offset = "-50"
[docs]@viewer_registry("mosviz-table-viewer", label="Table (Mosviz)")
class MosvizTableViewer(TableViewer):
def __init__(self, session, *args, **kwargs):
super().__init__(session, *args, **kwargs)
self.figure_widget.observe(self._on_row_selected, names=['highlighted'])
self._selected_data = {}
self._shared_image = False
self.row_selection_in_progress = False
[docs] def redraw(self):
# Overload to hide components - we do this via overloading instead of
# checking for changes in self.figure_widget.data because some components
# might be added inplace to the dataset.
if self.figure_widget.data is None:
self.figure_widget.hidden_components = []
else:
components_str = [cid.label for cid in self.figure_widget.data.main_components]
hidden = []
for colname in ['Images', '1D Spectra', '2D Spectra']:
if colname in components_str:
hidden.append(self.figure_widget.data.id[colname])
self.figure_widget.hidden_components = hidden
super().redraw()
def _on_row_selected(self, event):
self.row_selection_in_progress = True
# Grab the index of the latest selected row
selected_index = event['new']
mos_data = self.session.data_collection['MOS Table']
for component in mos_data.components:
comp_data = mos_data.get_component(component).data
selected_data = comp_data[selected_index]
if component.label == '1D Spectra':
prev_data = self._selected_data.get('spectrum-viewer')
if prev_data != selected_data:
if prev_data:
# This covers the cases where data is unit converted
# and the name is modified
all_prev_data = [x
for x in self.session.data_collection.labels
if prev_data in x]
for modified_prev_data in all_prev_data:
if modified_prev_data:
remove_data_from_viewer_message = RemoveDataFromViewerMessage(
'spectrum-viewer', modified_prev_data, sender=self)
self.session.hub.broadcast(remove_data_from_viewer_message)
add_data_to_viewer_message = AddDataToViewerMessage(
'spectrum-viewer', selected_data, sender=self)
self.session.hub.broadcast(add_data_to_viewer_message)
self._selected_data['spectrum-viewer'] = selected_data
if component.label == '2D Spectra':
prev_data = self._selected_data.get('spectrum-2d-viewer')
if prev_data != selected_data:
if prev_data:
remove_data_from_viewer_message = RemoveDataFromViewerMessage(
'spectrum-2d-viewer', prev_data, sender=self)
self.session.hub.broadcast(remove_data_from_viewer_message)
add_data_to_viewer_message = AddDataToViewerMessage(
'spectrum-2d-viewer', selected_data, sender=self)
self.session.hub.broadcast(add_data_to_viewer_message)
self._selected_data['spectrum-2d-viewer'] = selected_data
if component.label == 'Images':
prev_data = self._selected_data.get('image-viewer')
if prev_data != selected_data:
if prev_data:
remove_data_from_viewer_message = RemoveDataFromViewerMessage(
'image-viewer', prev_data, sender=self)
self.session.hub.broadcast(remove_data_from_viewer_message)
add_data_to_viewer_message = AddDataToViewerMessage(
'image-viewer', selected_data, sender=self)
self.session.hub.broadcast(add_data_to_viewer_message)
self._selected_data['image-viewer'] = selected_data
message = TableClickMessage(selected_index=selected_index,
shared_image=self._shared_image,
sender=self)
self.session.hub.broadcast(message)
self.row_selection_in_progress = False
[docs] def set_plot_axes(self, *args, **kwargs):
return