refactor unit conversion messaging

jdaviz/app.py

@@ -1345,8 +1345,7 @@ def _get_display_unit(self, axis):
                 raise ValueError(f"could not find units for axis='{axis}'")
         uc = self._jdaviz_helper.plugins.get('Unit Conversion')._obj
         if axis == 'spectral_y':
-            # translate options from uc.spectral_y_type to the prefix used in uc.??_unit_selected
-            axis = {'Surface Brightness': 'sb', 'Flux': 'flux'}[uc.spectral_y_type_selected]
+            return uc.spectral_y_unit
         try:
             return getattr(uc, f'{axis}_unit_selected')
         except AttributeError:

jdaviz/configs/cubeviz/plugins/mixins.py

@@ -89,6 +89,12 @@ def slice_values(self):
         take_inds = [2, 1, 0]
         take_inds.remove(self.slice_index)
         converted_axis = np.array([])
+
+        # Retrieve display units
+        slice_display_units = self.jdaviz_app._get_display_unit(
+            self.slice_display_unit_name
+        )
+
         for layer in self.layers:
             world_comp_ids = layer.layer.data.world_component_ids
 
@@ -100,11 +106,6 @@ def slice_values(self):
                 # Case where 2D image is loaded in image viewer
                 continue
 
-            # Retrieve display units
-            slice_display_units = self.jdaviz_app._get_display_unit(
-                self.slice_display_unit_name
-            )
-
             try:
                 # Retrieve layer data and units using the slice index of the world components ids
                 data_comp = layer.layer.data.get_component(world_comp_ids[self.slice_index])

jdaviz/configs/cubeviz/plugins/slice/slice.py

@@ -139,12 +139,13 @@ def _initialize_location(self, *args):
             if str(viewer.state.x_att) not in self.valid_slice_att_names:
                 # avoid setting value to degs, before x_att is changed to wavelength, for example
                 continue
-            # ensure the cache is reset (if previous attempts to initialize failed resulting in an
-            # empty list as the cache)
-            viewer._clear_cache('slice_values')
+            if self.app._get_display_unit(viewer.slice_display_unit_name) == '':
+                # viewer is not ready to retrieve slice_values in display units
+                continue
             slice_values = viewer.slice_values
             if len(slice_values):
-                self.value = slice_values[int(len(slice_values)/2)]
+                new_value = slice_values[int(len(slice_values)/2)]
+                self.value = new_value
                 self._indicator_initialized = True
                 return
 
@@ -229,22 +230,30 @@ def _on_select_slice_message(self, msg):
             self.value = msg.value
 
     def _on_global_display_unit_changed(self, msg):
-        if not self.app.config == 'cubeviz':
-            return
-
         if msg.axis != self.slice_display_unit_name:
             return
+        self._clear_cache()
         if not self.value_unit:
             self.value_unit = str(msg.unit)
             return
+        if not self._indicator_initialized:
+            self._initialize_location()
+            return
         prev_unit = u.Unit(self.value_unit)
         self.value_unit = str(msg.unit)
-        self._clear_cache()
-        self.value = (self.value * prev_unit).to_value(msg.unit, equivalencies=u.spectral())
+        self.value = self._convert_value_to_unit(self.value, prev_unit, msg.unit)
+
+    def _convert_value_to_unit(self, value, prev_unit, new_unit):
+        return (value * prev_unit).to_value(new_unit, equivalencies=u.spectral())
 
     def _clear_cache(self, *attrs):
         if not len(attrs):
             attrs = self._cached_properties
+        if len(attrs):
+            # most internally cached properties rely on
+            # viewer slice_values, so let's also clear those caches
+            for viewer in self.slice_selection_viewers:
+                viewer._clear_cache('slice_values')
         for attr in attrs:
             if attr in self.__dict__:
                 del self.__dict__[attr]

jdaviz/configs/cubeviz/plugins/spectral_extraction/tests/test_spectral_extraction.py

@@ -584,6 +584,7 @@ def test_spectral_extraction_unit_conv_one_spec(
     uc = cubeviz_helper.plugins["Unit Conversion"]
     assert uc.flux_unit == "Jy"
     uc.flux_unit.selected = "MJy"
+    assert spectrum_viewer.state.y_display_unit == "MJy"
     spec_extr_plugin = cubeviz_helper.plugins['Spectral Extraction']
     # Overwrite the one and only default extraction.
     collapsed = spec_extr_plugin.extract()

jdaviz/configs/cubeviz/plugins/tests/test_parsers.py

@@ -155,7 +155,6 @@ def test_spectrum3d_no_wcs_parse(cubeviz_helper):
     assert flux.units == 'nJy / pix2'
 
 
-@pytest.mark.skip(reason="unskip after 3192 merged")
 def test_spectrum1d_parse(spectrum1d, cubeviz_helper):
     cubeviz_helper.load_data(spectrum1d)
 

jdaviz/configs/default/plugins/viewers.py

@@ -770,8 +770,11 @@ def set_plot_axes(self):
 
         if solid_angle_unit is not None:
 
-            for un in locally_defined_flux_units:
-                locally_defined_sb_unit = un / solid_angle_unit
+            # create an equivalency for each flux unit for flux <> flux/pix2.
+            # for similar reasons to the 'untranslatable units' issue, custom
+            # equivs. can't be combined, so a workaround is creating an equiv
+            # for each flux that may need an additional equiv.
+            angle_to_pixel_equivs = [_eqv_sb_per_pixel_to_per_angle(un) for un in locally_defined_flux_units]  # noqa
 
                 # create an equivalency for each flux unit for flux <> flux/pix2.
                 # for similar reasons to the 'untranslatable units' issue, custom

jdaviz/configs/specviz/plugins/parsers.py

@@ -11,7 +11,6 @@
 from jdaviz.core.events import SnackbarMessage
 from jdaviz.core.registries import data_parser_registry
 from jdaviz.utils import standardize_metadata, download_uri_to_path
-from jdaviz.core.validunits import check_if_unit_is_per_solid_angle
 
 
 __all__ = ["specviz_spectrum1d_parser"]
@@ -160,15 +159,6 @@ def specviz_spectrum1d_parser(app, data, data_label=None, format=None, show_in_v
             # Make metadata layout conform with other viz.
             spec.meta = standardize_metadata(spec.meta)
 
-            # If this is the first loaded data, we want to set spectral y unit type to Flux or
-            # Surface Brightness as appropriate
-            if len(app.data_collection) == 0 and "Unit Conversion" in app._jdaviz_helper.plugins:
-                uc = app._jdaviz_helper.plugins["Unit Conversion"]
-                if check_if_unit_is_per_solid_angle(flux_units):
-                    uc._obj.spectral_y_type = "Surface Brightness"
-                else:
-                    uc._obj.spectral_y_type = "Flux"
-
             app.add_data(spec, data_label[i])
 
             # handle display, with the SpectrumList special case in mind.

jdaviz/configs/specviz/plugins/unit_conversion/tests/test_unit_conversion.py

@@ -100,7 +100,7 @@ def test_conv_no_data(specviz_helper, spectrum1d):
     # spectrum not load is in Flux units, sb_unit and flux_unit
     # should be enabled, spectral_y_type should not be
     plg = specviz_helper.plugins["Unit Conversion"]
-    with pytest.raises(ValueError, match="no valid unit choices"):
+    with pytest.raises(ValueError, match="could not find match in valid x display units"):
         plg.spectral_unit = "micron"
     assert len(specviz_helper.app.data_collection) == 0
 
@@ -290,11 +290,17 @@ def test_contour_unit_conversion(cubeviz_helper, spectrum1d_cube_fluxunit_jy_per
     # Make sure that the contour values get updated
     po_plg.contour_visible = True
 
+    assert uc_plg.spectral_y_type == 'Flux'
+    assert uc_plg.flux_unit == 'Jy'
+    assert uc_plg.sb_unit == "Jy / sr"
+    assert cubeviz_helper.viewers['flux-viewer']._obj.layers[0].state.attribute_display_unit == "Jy / sr"  # noqa
     assert np.allclose(po_plg.contour_max.value, 199)
 
-    uc_plg._obj.spectral_y_type_selected = 'Surface Brightness'
+    uc_plg.spectral_y_type = 'Surface Brightness'
     uc_plg.flux_unit = 'MJy'
 
+    assert uc_plg.sb_unit == "MJy / sr"
+    assert cubeviz_helper.viewers['flux-viewer']._obj.layers[0].state.attribute_display_unit == "MJy / sr"  # noqa
     assert np.allclose(po_plg.contour_max.value, 1.99e-4)