support _orig_spec in _get_continuum

spacetelescope · Dec 7, 2023 · 9b6350f · 9b6350f
1 parent acd7ffe
commit 9b6350f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/jdaviz/configs/cubeviz/plugins/moment_maps/moment_maps.py b/jdaviz/configs/cubeviz/plugins/moment_maps/moment_maps.py
@@ -174,7 +174,6 @@ def calculate_moment(self, add_data=True):
             else:
                 cube = self.dataset.selected_obj
         else:
-            # TODO: need to account for _orig_spec here?
             _, _, cube = self._get_continuum(self.dataset,
                                              'per-pixel',
                                              self.spectral_subset,

diff --git a/jdaviz/configs/specviz/plugins/line_analysis/line_analysis.py b/jdaviz/configs/specviz/plugins/line_analysis/line_analysis.py
@@ -221,7 +221,6 @@ def _is_active_changed(self, msg):
         if self.disabled_msg:
             return
 
-        # TODO: move to mixin?
         for pos, mark in self.continuum_marks.items():
             mark.visible = self.is_active
         self._calculate_statistics(msg)

diff --git a/jdaviz/core/template_mixin.py b/jdaviz/core/template_mixin.py
@@ -2053,7 +2053,10 @@ def _get_continuum(self, dataset, spatial_subset, spectral_subset, update_marks=
         if spatial_subset == 'per-pixel':
             if self.app.config != 'cubeviz':
                 raise ValueError("per-pixel only supported for cubeviz")
-            full_spectrum = dataset.selected_obj
+            if "_orig_spec" in self.dataset.selected_obj.meta:
+                full_spectrum = self.dataset.selected_obj.meta["_orig_spec"]
+            else:
+                full_spectrum = self.dataset.selected_obj
         else:
             full_spectrum = dataset.selected_spectrum_for_spatial_subset(spatial_subset.selected if spatial_subset is not None else None,  # noqa
                                                                          use_display_units=True)