diff --git a/jdaviz/configs/cubeviz/helper.py b/jdaviz/configs/cubeviz/helper.py index 058389d1ce..e4bf813fc9 100644 --- a/jdaviz/configs/cubeviz/helper.py +++ b/jdaviz/configs/cubeviz/helper.py @@ -1,7 +1,18 @@ from jdaviz.core.helpers import ConfigHelper -from ..default.plugins.line_lists.line_list_mixin import LineListMixin +from jdaviz.configs.default.plugins.line_lists.line_list_mixin import LineListMixin +from jdaviz.configs.specviz import SpecViz class CubeViz(ConfigHelper, LineListMixin): """CubeViz Helper class""" _default_configuration = 'cubeviz' + + @property + def specviz(self): + """ + A specviz helper (`~jdaviz.configs.specviz.SpecViz`) for the Jdaviz + application that is wrapped by cubeviz + """ + if not hasattr(self, '_specviz'): + self._specviz = SpecViz(app=self.app) + return self._specviz diff --git a/jdaviz/configs/mosviz/helper.py b/jdaviz/configs/mosviz/helper.py index 8f3b44f920..a9d5d928c7 100644 --- a/jdaviz/configs/mosviz/helper.py +++ b/jdaviz/configs/mosviz/helper.py @@ -1,11 +1,13 @@ import numpy as np from pathlib import Path -from jdaviz.core.helpers import ConfigHelper -from jdaviz.core.events import SnackbarMessage from astropy.table import QTable import astropy.units as u +from jdaviz.core.helpers import ConfigHelper +from jdaviz.core.events import SnackbarMessage +from jdaviz.configs.specviz import SpecViz + class MosViz(ConfigHelper): """MosViz Helper class""" @@ -282,3 +284,13 @@ def to_csv(self, filename="MOS_data.csv", selected=False, overwrite=False): table_df = table_df.drop(labels="Pixel Axis 0 [x]", axis=1) table_df.to_csv(filename, index_label="Table Index") + + @property + def specviz(self): + """ + A specviz helper (`~jdaviz.configs.specviz.SpecViz`) for the Jdaviz + application that is wrapped by mosviz + """ + if not hasattr(self, '_specviz'): + self._specviz = SpecViz(app=self.app) + return self._specviz