cubevis 0.5.2__py3-none-any.whl

cubevis/apps/_plotants.py

@@ -0,0 +1,286 @@
+########################################################################
+#
+# Copyright (C) 2022
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+"""
+plotants module
+"""
+
+import os
+
+import numpy as np
+from bokeh.io import export_png, export_svgs
+from bokeh.models import ColumnDataSource, LabelSet
+from bokeh.plotting import figure, show
+
+try:
+    from cairosvg import svg2pdf
+    _have_svg2pdf = True
+except ImportError:
+    _have_svg2pdf = False
+
+try:
+    import casatools as ct
+    from casatools import table, msmetadata, quanta, ms, measures
+except:
+    ct = None
+    from cubevis.utils import warn_import
+    warn_import('casatools')
+
+_FIGURE_PLOT_WIDTH = 450
+_FIGURE_PLOT_HEIGHT = 450
+
+
+def __get_observatory_info(msname):
+    """Extract the observatory information from `msname`.
+
+    Parameters
+    ----------
+    msname: string
+        Path to a CASA measurement set.
+
+    Returns
+    -------
+    ( string, list )
+        string is the telescope name and the list is the array position
+    """
+    metadata = ct.msmetadata()
+    metadata.open(msname)
+    telescope = metadata.observatorynames()[0]
+    positions = metadata.observatoryposition()
+    metadata.close()
+    return telescope, positions
+
+
+def __get_antenna_info(msname, log, exclude, checkbaselines):
+    """Return the antenna position info.
+
+    Parameters
+    ----------
+    msname: string
+        Path to the CASA measurement set.
+    log: boolean
+        whether to plot logarithmic positions
+    exclude: [ int ]
+        list antenna name/id selection to exclude from plot
+    checkbaselines: boolean
+        whether to check baselines in the main table
+    """
+    if ct is None:
+        raise RuntimeError('casatools is not available')
+
+    me = ct.measures()
+    qa = ct.quanta()
+    tb = ct.table()
+
+    telescope, positions = __get_observatory_info(msname)
+    positions_wgs84 = me.measure(positions, "WGS84")
+    array_lon, array_lat, = [positions_wgs84[i]["value"] for i in ["m0", "m1", "m2"]]
+
+    # Open the ANTENNA subtable to get the names of the antennas in this MS and
+    # their positions.  Note that the entries in the ANTENNA subtable are pretty
+    # much in random order, so antenna_names translates between their index and name
+    # (e.g., index 11 = STD155).  We'll need these indices for later, since the
+    # main data table refers to the antennas by their indices, not names.
+
+    anttabname = msname + "/ANTENNA"
+    tb.open(anttabname)
+    # Get antenna names from antenna table
+    antenna_names = np.array(tb.getcol("NAME")).tolist()
+    station_names = np.array(tb.getcol("STATION")).tolist()
+    if telescope == "VLBA":  # names = ant@station
+        antenna_names = ["@".join(antsta) for antsta in zip(antenna_names, station_names)]
+    # Get antenna positions from antenna table
+    antenna_positions = np.array(
+        [
+            me.position("ITRF", qa.quantity(x, "m"), qa.quantity(y, "m"), qa.quantity(z, "m"))
+            for (x, y, z) in tb.getcol("POSITION").transpose()
+        ]
+    )
+    tb.close()
+
+    all_ant_ids = range(len(antenna_names))
+    if checkbaselines:
+        # Get antenna ids from main table; this will add to runtime
+        tb.open(msname)
+        ants1 = tb.getcol("ANTENNA1")
+        ants2 = tb.getcol("ANTENNA2")
+        tb.close()
+        ant_ids_used = list(set(np.append(ants1, ants2)))
+    else:
+        # use them all!
+        ant_ids_used = all_ant_ids
+
+    # handle exclude -- remove from ant_ids_used
+    for ant_id in exclude:
+        try:
+            ant_name_id = antenna_names[ant_id] + " (id " + str(ant_id) + ")"
+            ant_ids_used.remove(ant_id)
+            print( f'''Exclude antenna {ant_name_id}''' )
+        except ValueError:
+            print( f'''Cannot exclude antenna {ant_name_id}: not in main table''' )
+
+    # apply ant_ids_used mask
+    antenna_names = [antenna_names[i] for i in ant_ids_used]
+    antenna_positions = [antenna_positions[i] for i in ant_ids_used]
+    station_names = [station_names[i] for i in ant_ids_used]
+
+    n_ants = len(ant_ids_used)
+    # casalog.post("Number of points being plotted: " + str(n_ants))
+    if n_ants == 0:  # excluded all antennas
+        return telescope, antenna_names, [], [], [], []
+
+    # Get the names, indices, and lat/lon/alt coords of "good" antennas.
+    ant_wgs84s = np.array([me.measure(pos, "WGS84") for pos in antenna_positions])
+
+    # Convert from lat, lon, alt to X, Y, Z (unless VLBA)
+    # where X is east, Y is north, Z is up,
+    # and 0, 0, 0 is the center
+    # Note: this conversion is NOT exact, since it doesn't take into account
+    # Earth's ellipticity!  But it's close enough.
+    if telescope == "VLBA" and not log:
+        ant_lons, ant_lats = [[pos[i] for pos in ant_wgs84s] for i in ["m0", "m1"]]
+        ant_xs = [qa.convert(lon, "deg")["value"] for lon in ant_lons]
+        ant_ys = [qa.convert(lat, "deg")["value"] for lat in ant_lats]
+    else:
+        ant_lons, ant_lats = [np.array([pos[i]["value"] for pos in ant_wgs84s]) for i in ["m0", "m1"]]
+        rade = 6370000.0 # radE
+        ant_xs = (ant_lons - array_lon) * rade * np.cos(array_lat)
+        ant_ys = (ant_lats - array_lat) * rade
+    return telescope, antenna_names, ant_ids_used, ant_xs, ant_ys, station_names
+
+
+def __plot_antennas_log(telescope, names, ids, xpos, ypos, antindex, stations, title):
+    raise NotImplementedError("This is a placeholder for another type of plot. It is not implemented yet.")
+
+
+def __plot_antennas(telescope, names, ids, xpos, ypos, antindex, stations, title):
+    if telescope == "VLBA":
+        labelx = "Longitude (deg)"
+        labely = "Latitude (deg)"
+    else:
+        # use m or km units
+        units = " (m)"
+        if np.median(xpos) > 1e6 or np.median(ypos) > 1e6:
+            xpos /= 1e3
+            ypos /= 1e3
+            units = " (km)"
+        labelx = "X" + units
+        labely = "Y" + units
+    if antindex:
+        names = [f"{name} ({idx})" for name, idx in zip(names, ids)]
+    source = ColumnDataSource(data=dict(x=[], y=[], labels=[]))
+    source.data = dict(x=xpos, y=ypos, labels=names)
+    labels = LabelSet(x="x", y="y", text="labels", x_offset=5, y_offset=5, source=source, text_font_size="10pt")
+    plot = figure(plot_height=_FIGURE_PLOT_HEIGHT, plot_width=_FIGURE_PLOT_WIDTH)
+    plot.scatter("x", "y", source=source, size=5, line_color="red", fill_color="red", fill_alpha=0.5)
+    plot.title.text = title
+    plot.xaxis[0].axis_label = labelx
+    plot.yaxis[0].axis_label = labely
+    plot.add_layout(labels)
+    return plot
+
+
+def plotants(vis, figfile="", antindex=False, logpos=False, exclude=[], checkbaselines=False, title="", showgui=True):
+    """Plot the antenna distribution in the local reference frame:
+    The location of the antennas in the MS will be plotted with
+    X-toward local east; Y-toward local north.  The name of each
+    antenna is shown next to its respective location.
+
+    Parameters
+    ----------
+    vis: string
+    Path to the input visibility file
+
+    antindex: boolean, default: False
+        Label antennas with name and antenna ID
+
+    logpos: boolean, default: False
+        Produce a logarithmic position plot.
+
+    exclude: list, default: []
+        antenna IDs or names to exclude from plotting, for example:
+        exclude=[2,3,4], exclude=['DV15']
+
+    checkbaselines: boolean, default: False
+        Only plot antennas in the MAIN table. This can be useful after a split.
+        WARNING:  Setting checkbaselines to True will add to runtime in proportion
+        to the number of rows in the dataset.
+
+    title: string, default: ''
+        Title written along top of plot
+    """
+    if os.path.exists(vis) is False:
+        raise Exception(f"Visibility file {vis} does not exist")  # could be a print + return
+    # remove trailing / for title basename
+    if vis.endswith("/"):
+        vis = vis[:-1]
+
+    myms = ct.ms()
+    try:
+        exclude = myms.msseltoindex(vis, baseline=exclude)["antenna1"].tolist()
+    except RuntimeError as rterr:  # MSSelection failed
+        errmsg = str(rterr)
+        errmsg = errmsg.replace("specificion", "specification")
+        errmsg = errmsg.replace("Antenna Expression: ", "")
+        raise RuntimeError("Exclude selection error: " + errmsg) from rterr
+
+    # Get the antenna positions
+    telescope, names, ids, xpos, ypos, stations = __get_antenna_info(vis, logpos, exclude, checkbaselines)
+    if not names:
+        raise ValueError("No antennas selected. Exiting plotants.")
+
+    if title == "":
+        msname = os.path.basename(vis)
+        title = "Antenna Positions for "
+        if len(msname) > 55:
+            title += "\n"
+        title += msname
+
+    if logpos:
+        fig = __plot_antennas_log(telescope, names, ids, xpos, ypos, antindex, stations, title)
+    else:
+        fig = __plot_antennas(telescope, names, ids, xpos, ypos, antindex, stations, title)
+
+    if showgui is not False:
+        show(fig)
+
+    if figfile != "":
+        if figfile.endswith(".png"):
+            export_png(fig, filename=figfile)
+        elif figfile.endswith(".svg"):
+            fig.output_backend = "svg"
+            export_svgs(fig, filename=figfile)
+        elif figfile.endswith(".pdf"):
+            if _have_svg2pdf == True:
+                fig.output_backend = "svg"
+                export_svgs(fig, filename=figfile.replace(".pdf", ".svg"))
+                svg2pdf(url=figfile.replace(".pdf", ".svg"), write_to=figfile)
+                os.system("rm " + figfile.replace(".pdf", ".svg"))
+            else:
+                raise RuntimeError("cairosvg is required for generating PDF output, but it is not available")
+        else:
+            raise ValueError("Invalid output file type.  Must be .png or .svg or .pdf")

cubevis/apps/_plotbandpass.py

@@ -0,0 +1,7 @@
+"""Plotbandpass module
+"""
+
+def plotbandpass():
+    """Plotbandpass function
+    """
+    raise NotImplementedError("This is a placeholder for plotbandpass. It is not implemented yet.")

cubevis/bokeh/__init__.py

@@ -0,0 +1,29 @@
+########################################################################
+#
+# Copyright (C) 2021, 2022
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+'''This module contains the extensions and additions to the functionality
+provided by Bokeh'''

cubevis/bokeh/annotations/__init__.py

@@ -0,0 +1 @@
+from ._ev_poly_annotation import EvPolyAnnotation

cubevis/bokeh/annotations/_ev_poly_annotation.py

@@ -0,0 +1,6 @@
+from bokeh.models import PolyAnnotation
+
+class EvPolyAnnotation(PolyAnnotation):
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)

cubevis/bokeh/components/__init__.py

@@ -0,0 +1,28 @@
+########################################################################
+#
+# Copyright (C) 2021,2022,2023
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+'''Custom Bokeh GUI Components'''

cubevis/bokeh/format/__init__.py

@@ -0,0 +1,31 @@
+########################################################################
+#
+# Copyright (C) 2023
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+'''Implementation of Bokeh formatting extensions provided by ``cubevis``'''
+
+from ._time_ticks import get_time_formatter
+from ._wcs_ticks import WcsTicks

cubevis/bokeh/format/_time_ticks.py

@@ -0,0 +1,44 @@
+########################################################################
+#
+# Copyright (C) 2025
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+
+''' Format time ticks for consistency at various zoom levels '''
+
+from bokeh.models.formatters import DatetimeTickFormatter
+
+def get_time_formatter():
+    ''' Tick formatter which includes H:M:S in most ticks '''
+    return DatetimeTickFormatter(
+        context_which = 'all',
+        microseconds='%fus',
+        milliseconds='%H:%M:%S %3Nms',
+        seconds='%H:%M:%S',
+        minsec='%H:%M:%S',
+        minutes='%H:%M:%S',
+        hourmin='%H:%M:%S',
+        hours='%H:%M:%S',
+        days='%F')

cubevis/bokeh/format/_wcs_ticks.py

@@ -0,0 +1,45 @@
+########################################################################
+#
+# Copyright (C) 2023
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+from bokeh.models import TickFormatter
+from bokeh.util.compiler import TypeScript
+from bokeh.core.properties import Instance, String
+from cubevis.bokeh.sources import ImageDataSource
+from ..state import casalib_url, cubevisjs_url
+
+class WcsTicks(TickFormatter):
+
+    ## which axis are we labeling
+    axis = String( )
+
+    ## source containing the WCS information
+    image_source = Instance(ImageDataSource)
+
+    __javascript__ = [ casalib_url( ), cubevisjs_url( ) ]
+
+    def __init__( self, *args, **kwargs ):
+        super( ).__init__( *args, **kwargs )

cubevis/bokeh/models/__init__.py

@@ -0,0 +1,4 @@
+from ._tip_button import TipButton
+from ._tip import Tip
+from ._edit_span import EditSpan
+from ._ev_text_input import EvTextInput

cubevis/bokeh/models/_edit_span.py

@@ -0,0 +1,7 @@
+from bokeh.models import Span
+
+class EditSpan(Span):
+
+    # explicit __init__ to support Init signatures
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)

cubevis/bokeh/models/_ev_text_input.py

@@ -0,0 +1,6 @@
+from bokeh.models import TextInput
+
+class EvTextInput(TextInput):
+
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)

cubevis/bokeh/models/_tip.py

@@ -0,0 +1,37 @@
+from bokeh.models import Tooltip
+from bokeh.models.layouts import LayoutDOM, UIElement
+from bokeh.core.properties import Instance, Required, Float, Int, Either
+
+class Tip(LayoutDOM):
+    '''Display a tooltip for the child element
+    '''
+
+    def __init__(self, *args, **kwargs) -> None:
+        if len(args) != 1 and "child" not in kwargs:
+            raise ValueError("a 'child' argument must be supplied")
+        elif len(args) == 1 and "child" in kwargs:
+            raise ValueError("'child' supplied as both a positional argument and a keyword")
+        elif len(args) > 1:
+            raise ValueError("only one 'child' can be supplied as a positional argument")
+        elif len(args) > 0:
+            kwargs["child"] = args[0]
+
+        super().__init__(**kwargs)
+
+    child = Required(Instance(UIElement), help="""
+    A child, which can be other components including plots, rows, columns, and widgets.
+    """)
+
+    tooltip = Required(Instance(Tooltip), help="""
+    A tooltip with plain text or rich HTML contents, providing general help or
+    description of a widget's or component's function.
+    """)
+
+    hover_wait = Either( Float, Int, default=1.5, help='''
+    amount of time (in seconds) to wait before displaying
+    ''' )
+
+    def _sphinx_height_hint(self):
+        if child._sphinx_height_hint() is None:
+            return None
+        return child._sphinx_height_hint()

cubevis/bokeh/models/_tip_button.py

@@ -0,0 +1,50 @@
+
+from bokeh.models.callbacks import Callback
+from bokeh.util.callback_manager import EventCallback
+from bokeh.events import ButtonClick
+from bokeh.models.widgets.buttons import AbstractButton
+from bokeh.models.ui.icons import BuiltinIcon
+from bokeh.core.properties import Instance, Required, Override, Nullable, Float, Int, Either
+from bokeh.models import Tooltip
+
+
+
+class TipButton(AbstractButton):
+    """ A button with a help symbol that displays additional text when hovered
+    over or clicked.
+    """
+
+    # explicit __init__ to support Init signatures
+    def __init__(self, *args, **kwargs) -> None:
+        super().__init__(*args, **kwargs)
+
+    tooltip = Required(Instance(Tooltip), help="""
+    A tooltip with plain text or rich HTML contents, providing general help or
+    description of a widget's or component's function.
+    """)
+
+    hover_wait = Either( Float, Int, default=1.5, help='''
+    amount of time (in seconds) to wait before displaying
+    ''' )
+
+    label = Override(default="")
+
+    icon = Override(default=lambda: BuiltinIcon("help", size=18))
+
+    button_type = Override(default="default")
+
+    def on_click(self, handler: EventCallback) -> None:
+        ''' Set up a handler for button clicks.
+
+        Args:
+            handler (func) : handler function to call when button is clicked.
+
+        Returns:
+            None
+
+        '''
+        self.on_event(ButtonClick, handler)
+
+    def js_on_click(self, handler: Callback) -> None:
+        ''' Set up a JavaScript handler for button clicks. '''
+        self.js_on_event(ButtonClick, handler)

cubevis/bokeh/sources/__init__.py

@@ -0,0 +1,35 @@
+########################################################################
+#
+# Copyright (C) 2021,2022
+# Associated Universities, Inc. Washington DC, USA.
+#
+# This script is free software; you can redistribute it and/or modify it
+# under the terms of the GNU Library General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or (at your
+# option) any later version.
+#
+# This library is distributed in the hope that it will be useful, but WITHOUT
+# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
+# FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Library General Public
+# License for more details.
+#
+# You should have received a copy of the GNU Library General Public License
+# along with this library; if not, write to the Free Software Foundation,
+# Inc., 675 Massachusetts Ave, Cambridge, MA 02139, USA.
+#
+# Correspondence concerning AIPS++ should be adressed as follows:
+#        Internet email: casa-feedback@nrao.edu.
+#        Postal address: AIPS++ Project Office
+#                        National Radio Astronomy Observatory
+#                        520 Edgemont Road
+#                        Charlottesville, VA 22903-2475 USA
+#
+########################################################################
+'''Implementation of the Bokeh data source extensions provided by
+``cubevis``'''
+
+from ._data_pipe import DataPipe
+from ._image_pipe import ImagePipe
+from ._image_data_source import ImageDataSource
+from ._spectra_data_source import SpectraDataSource
+from ._updatable_data_source import UpdatableDataSource