Merge pull request #330 from eteq/specviz-helper-helper

add specviz property to cubeviz/mosviz helpers

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

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