vidavis 0.0.5__py3-none-any.whl → 0.0.7__py3-none-any.whl

vidavis/__version__.py

@@ -1 +1 @@
-__version__ = '0.0.5'
+__version__ = '0.0.7'

vidavis/apps/_ms_raster.py

@@ -7,15 +7,15 @@ import time
 from bokeh.models.formatters import NumeralTickFormatter
 import holoviews as hv
 import numpy as np
-import panel as pn
 from pandas import to_datetime
+import panel as pn
 
 from vidavis.bokeh._palette import available_palettes
+from vidavis.data.measurement_set.processing_set._ps_coords import set_index_coordinates
 from vidavis.plot.ms_plot._time_ticks import get_time_formatter
 from vidavis.plot.ms_plot._ms_plot import MsPlot
-from vidavis.plot.ms_plot._ms_plot_constants import VIS_AXIS_OPTIONS, SPECTRUM_AXIS_OPTIONS, PS_SELECTION_OPTIONS, MS_SELECTION_OPTIONS
-from vidavis.plot.ms_plot._ms_plot_selectors import (file_selector, title_selector, style_selector, axis_selector,
-aggregation_selector, iteration_selector, selection_selector, plot_starter)
+from vidavis.plot.ms_plot._ms_plot_constants import VIS_AXIS_OPTIONS, SPECTRUM_AXIS_OPTIONS, PS_SELECTION_OPTIONS, MS_SELECTION_OPTIONS, TIME_FORMAT
+from vidavis.plot.ms_plot._raster_plot_gui import create_raster_gui
 from vidavis.plot.ms_plot._raster_plot_inputs import check_inputs
 from vidavis.plot.ms_plot._raster_plot import RasterPlot
 
@@ -43,23 +43,21 @@ class MsRaster(MsPlot):
     def __init__(self, ms=None, log_level="info", log_to_file=True, show_gui=False):
         super().__init__(ms, log_level, log_to_file, show_gui, "MsRaster")
         self._raster_plot = RasterPlot()
+        self._plot_data = None
 
         # Calculations for color limits
         self._spw_stats = {}
         self._spw_color_limits = {}
 
         if show_gui:
-            # GUI based on panel widgets
-            self._gui_layout = None
-
-            # Check if plot inputs changed and new plot is needed.
-            # GUI can change subparams which do not change plot
+            # For checking if plot inputs or cursor position changed and new plot or cursor location info is needed.
             self._last_plot_inputs = None
             self._last_style_inputs = None
-            # Last plot when no new plot created (plot inputs same) or is iter Layout plot (opened in tab)
-            self._last_gui_plot = None
+            self._last_cursor = None
 
             # Return plot for gui DynamicMap:
+            # Last plot when no new plot created (plot inputs same) or is iter Layout plot (opened in tab)
+            self._last_gui_plot = None
             # Empty plot when ms not set or plot fails
             self._empty_plot = self._create_empty_plot()
 
@@ -165,25 +163,14 @@ class MsRaster(MsPlot):
             x_axis (str): Plot x-axis. Default 'baseline' ('antenna_name' for spectrum data).
             y_axis (str): Plot y-axis. Default 'time'.
             vis_axis (str): Complex visibility component to plot (amp, phase, real, imag). Default 'amp'.
-                Call data_groups() to see options.
-                MeasurementSet selection:
-                    'data_group': name for correlated data, flags, weights, and uvw. Default value 'base'.
-                        Use data_groups() to get data group names.
-                    Dimensions:
-                        Visibility dimensions: 'baseline' 'time', 'frequency', 'polarization'
-                        Spectrum dimensions: 'antenna_name', 'time', 'frequency', 'polarization'
-                        Default value is index 0 (after user selection) for non-axis dimensions unless aggregated.
-                        Use antennas() to get antenna names. Select 'baseline' as "<name1> & <name2>".
-                        Use summary() to list frequencies and polarizations.
-                        TODO: how to select time?
-            aggregator (str): reduction for rasterization. Default None.
+            aggregator (None, str): reduction for rasterization. Default None.
                 Options include 'max', 'mean', 'min', 'std', 'sum', 'var'.
-            agg_axis (str, list): which dimension to apply aggregator across. Default None.
+            agg_axis (None, str, list): which dimension to apply aggregator across. Default None.
                 Options include one or more dimensions.
                 If agg_axis is None and aggregator is set, aggregates over all non-axis dimensions.
                 If one agg_axis is selected, the non-agg dimension will be selected.
-            iter_axis (str): dimension over which to iterate values (using iter_range).
-            iter_range (tuple): (start, end) inclusive index values for iteration plots.
+            iter_axis (None, str): dimension over which to iterate values (using iter_range).
+            iter_range (None, tuple): (start, end) inclusive index values for iteration plots.
                 Default (0, 0) (first iteration only). Use (0, -1) for all iterations.
                 If subplots is a grid, the range is limited by the grid size.
                 If subplots is a single plot, all iteration plots in the range can be saved using export_range in save().
@@ -194,12 +181,12 @@ class MsRaster(MsPlot):
                 Options include None (use data limits), 'auto' (calculate limits for amplitude), and 'manual' (use range in color_range).
                 'auto' is equivalent to None if vis_axis is not 'amp'.
                 When subplots is set, the 'auto' or 'manual' range will be used for all plots.
-            color_range (tuple): (min, max) of colorbar to use if color_mode is 'manual'.
-            title (str): Plot title, default None (no title)
+            color_range (None, tuple): (min, max) of colorbar to use if color_mode is 'manual'.
+            title (None, str): Plot title, default None (no title)
                 Set title='ms' to generate title from ms name and iter_axis value, if any.
             clear_plots (bool): whether to clear list of plots. Default True.
 
-        If not show_gui and plotting is successful, use show() or save() to view/save the plot only.
+        If plot is successful, use show() or save() to view/save the plot.
         '''
         inputs = locals() # collect arguments into dict (not unused as pylint complains!)
         if self._plot_inputs['selection']:
@@ -244,10 +231,13 @@ class MsRaster(MsPlot):
         Args:
             filename (str): Name of file to save. Default '': the plot will be saved as {ms}_raster.{ext}.
                 If fmt is not set for extension, plot will be saved as .png.
-            fmt (str): Format of file to save ('png', 'html', or 'gif').
+            fmt (str): Format of file to save ('png', 'svg', or 'html').
                 Default 'auto': inferred from filename extension.
-            width (int): width of exported plot.
-            height (int): height of exported plot.
+            width (int): width of exported plot in pixels.
+            height (int): height of exported plot in pixels.
+
+        If subplots defines a grid layout, width and height describe the size of each plot in the layout.
+            The layout plot size will be (width * columns, height * rows) pixels.
 
         If iteration plots were created:
             If subplots is a grid, the layout plot will be saved to a single file.
@@ -270,6 +260,7 @@ class MsRaster(MsPlot):
 
         # Select vis_axis data to plot and update selection; returns xarray Dataset
         raster_data = self._data.get_raster_data(plot_inputs)
+        self._plot_data = raster_data
 
         # Add params needed for plot: auto color range and ms name
         self._set_auto_color_range(plot_inputs) # set calculated limits if auto mode
@@ -423,85 +414,34 @@ class MsRaster(MsPlot):
     ### -----------------------------------------------------------------------
     def _launch_gui(self):
         ''' Use Holoviz Panel to create a dashboard for plot inputs. '''
-        # Select MS
-        file_selectors = file_selector('Path to MeasurementSet (ms or zarr) for plot', '~' , self._set_filename)
-
-        # Select style - colormaps, colorbar, color limits
-        style_selectors = style_selector(self._set_style_params, self._set_color_range)
-
-        # Set title
-        title_input = title_selector(self._set_title)
-
-        # Select x, y, and vis axis
+        callbacks = {
+            'filename': self._set_filename,
+            'style': self._set_style_params,
+            'color': self._set_color_range,
+            'axes': self._set_axes,
+            'select_ps': self._set_ps_selection,
+            'select_ms': self._set_ms_selection,
+            'aggregation': self._set_aggregation,
+            'iter_values': self._set_iter_values,
+            'iteration': self._set_iteration,
+            'title': self._set_title,
+            'plot_updating': self._update_plot_spinner,
+            'update_plot': self._update_plot,
+        }
+        data_dims = self._ms_info['data_dims'] if 'data_dims' in self._ms_info else None
         x_axis = self._plot_inputs['x_axis']
         y_axis = self._plot_inputs['y_axis']
-        data_dims = self._ms_info['data_dims'] if 'data_dims' in self._ms_info else None
-        axis_selectors = axis_selector(x_axis, y_axis, data_dims, True, self._set_axes)
-
-        # Select from ProcessingSet and MeasurementSet
-        selection_selectors = selection_selector(self._set_ps_selection, self._set_ms_selection)
-
-        # Generic axis options, updated when ms is set
-        axis_options = data_dims if data_dims else []
-
-        # Select aggregator and axes to aggregate
-        agg_selectors = aggregation_selector(axis_options, self._set_aggregation)
-
-        # Select iter_axis and iter value or range
-        iter_selectors = iteration_selector(axis_options, self._set_iter_values, self._set_iteration)
-
-        # Put user input widgets in accordion with only one card active at a time
-        selectors = pn.Accordion(
-            ("Select file", file_selectors),         # [0]
-            ("Plot style", style_selectors),         # [1]
-            ("Data Selection", selection_selectors), # [2]
-            ("Plot axes", axis_selectors),           # [3]
-            ("Aggregation", agg_selectors),          # [4]
-            ("Iteration", iter_selectors),           # [5]
-            ("Plot title", title_input),             # [6]
-        )
-        selectors.toggle = True
-
-        # Plot button and spinner while plotting
-        init_plot = plot_starter(self._update_plot_spinner)
-
-        # Connect plot to filename and selector widgets
-        dmap = hv.DynamicMap(
-            pn.bind(
-                self._update_plot,
-                ms=file_selectors[0][0],
-                do_plot=init_plot[0],
-            ),
-        )
-
-        # Layout plot and input widgets in a row
-        self._gui_layout = pn.Row(
-            pn.Tabs(             # [0]
-                ('Plot', dmap),               # [0]
-                ('Plot Inputs', pn.Column()), # [1]
-                sizing_mode='stretch_width',
-            ),
-            pn.Spacer(width=10), # [1]
-            pn.Column(           # [2]
-                pn.Spacer(height=25), # [0]
-                selectors,            # [1]
-                init_plot,            # [2]
-                width_policy='min',
-                width=400,
-                sizing_mode='stretch_height',
-            ),
-            sizing_mode='stretch_height',
-        )
-
+        self._gui_layout = create_raster_gui(callbacks, data_dims, x_axis, y_axis)
         # Show gui
-        # print("gui layout:", self._gui_layout) # for debugging
         self._gui_layout.show(title=self._app_name, threaded=True)
 
     ###
     ### Main callback to create plot if inputs changed
     ###
-    def _update_plot(self, ms, do_plot):
-        ''' Create plot with inputs from GUI.  Must return plot, even if empty plot, for DynamicMap. '''
+    def _update_plot(self, ms, do_plot, x, y):
+        ''' Create plot with inputs from GUI, or update cursor (x, y) information.
+            Must return plot, even if empty plot or same plot as before, for DynamicMap.
+        '''
         if self._toast:
             self._toast.destroy()
 
@@ -510,12 +450,18 @@ class MsRaster(MsPlot):
             # Launched GUI with no MS
             return self._empty_plot
 
-        # If not first plot, user has to click Plot button (do_plot=True).
+        # If not first plot, user has to click Plot button (do_plot=True) to update plot.
+        # Respond to pointer callback only.
         first_plot = not self._last_gui_plot
         if not do_plot and not first_plot:
+            if not self._last_cursor or (self._last_cursor[0] != x or self._last_cursor[1] != y):
+                # Callback is for new cursor location
+                self._update_cursor_location(x, y)
+                self._last_cursor = (x, y)
             # Not ready to update plot yet, return last plot.
             return self._last_gui_plot
 
+        # Callback is for initial plot for input ms, or user clicked Plot button
         if (self._set_ms(ms) or first_plot) and self._data and self._data.is_valid():
             # New MS set and is valid
             self._update_gui_axis_options()
@@ -811,7 +757,6 @@ class MsRaster(MsPlot):
             #filename_input.width = len(filename[-1])
             filename_input.value = filename[-1]
 
-
     def _set_iter_values(self, iter_axis):
         ''' Set up player with values when iter_axis is selected '''
         iter_axis = None if iter_axis == 'None' else iter_axis
@@ -876,6 +821,98 @@ class MsRaster(MsPlot):
             for param in self._plot_params:
                 inputs_column.append(pn.pane.Str(param))
 
+    def _update_cursor_location(self, x, y):
+        ''' Show metadata for cursor x,y position '''
+        # Convert plot values to selection values to select plot data
+        x_axis = self._plot_inputs['x_axis']
+        y_axis = self._plot_inputs['y_axis']
+        x = round(x) if x_axis == 'baseline' else x
+        y = round(y) if y_axis == 'baseline' else y
+        position = {x_axis: x, y_axis: y}
+        location_values, units = self._get_cursor_location_values(position)
+        self._update_cursor_box(location_values, units)
+
+    def _get_cursor_location_values(self, cursor_position):
+        values = cursor_position.copy()
+        units = {}
+
+        if self._plot_data:
+            try:
+                xds = set_index_coordinates(self._plot_data, tuple(cursor_position.keys()))
+                sel_xds = xds.sel(indexers=None, method='nearest', tolerance=None, drop=False, **cursor_position)
+                for coord in sel_xds.coords:
+                    if coord != 'uvw_label':
+                        val, unit = self._get_xda_val_unit(sel_xds[coord])
+                        values[coord] = val
+                        units[coord] = unit
+                for data_var in sel_xds.data_vars:
+                    val, unit = self._get_xda_val_unit(sel_xds[data_var])
+                    if data_var == 'UVW':
+                        names = ['U', 'V', 'W']
+                        for i, name in enumerate(names):
+                            values[name] = val[i]
+                            units[name] = unit[i]
+                    else:
+                        values[data_var] = val
+                        units[data_var] = unit
+            except KeyError:
+                pass
+
+        # Set complex component name for visibilities
+        if 'VISIBILITY' in values:
+            values[self._plot_inputs['vis_axis'].capitalize()] = values.pop('VISIBILITY')
+        return values, units
+
+    def _update_cursor_box(self, location_values, units):
+        ''' Update cursor location widget box with info in dict '''
+        cursor_location_box = self._gui_layout[0][0][1]
+        cursor_location_box.clear() # pn.WidgetBox
+        location_layout = pn.Column(pn.widgets.StaticText(name="Cursor Location"))
+        info_row = pn.Row()
+        info_col = pn.Column()
+
+        for name, value in location_values.items():
+            # 4 entries per column; append to row and start new column
+            if len(info_col.objects) == 4:
+                info_row.append(info_col)
+                info_col = pn.Column()
+
+            if not isinstance(value, str):
+                if name == "FLAG":
+                    value = "nan" if np.isnan(value) else int(value)
+                elif isinstance(value, float):
+                    if np.isnan(value):
+                        value = "nan"
+                    elif value < 1e6:
+                        value = f"{value:.4f}"
+                    else:
+                        value = f"{value:.4e}"
+                elif isinstance(value, np.datetime64):
+                    value = to_datetime(np.datetime_as_string(value)).strftime(TIME_FORMAT)
+                    units.pop(name) # no unit for datetime string
+            unit = units[name] if name in units else ""
+            info = pn.widgets.StaticText(name=name, value=f"{value} {unit}")
+            info.margin = (0, 10) # default (5, 10)
+            info_col.append(info)
+        info_row.append(info_col)
+        location_layout.append(info_row)
+        cursor_location_box.append(location_layout)
+
+    def _get_xda_val_unit(self, xda):
+        ''' Get value and unit as str not list '''
+        # Value
+        val = xda.values
+        if isinstance(val, np.ndarray) and val.size == 1:
+            val = val.item()
+        # Unit
+        try:
+            unit = xda.attrs['units']
+            unit = unit[0] if (isinstance(unit, list) and len(unit) == 1) else unit
+            unit = '' if unit == 'unkown' else unit
+        except KeyError:
+            unit = ''
+        return val, unit
+
     ###
     ### Callbacks for widgets which update plot inputs
     ###

vidavis/plot/ms_plot/_ms_plot.py

@@ -5,11 +5,13 @@ Base class for ms plots
 import os
 import time
 
-from bokeh.plotting import show
+from bokeh.io import export_png, export_svg
+from bokeh.plotting import save, show
 import hvplot
 import holoviews as hv
 import numpy as np
 import panel as pn
+from selenium import webdriver
 
 try:
     from toolviper.utils.logger import get_logger, setup_logger
@@ -57,6 +59,9 @@ class MsPlot:
         self._plots_locked = False
         self._plots = []
 
+        # Initialize gui
+        self._gui_layout = None
+
         # Set data (if ms)
         self._data = None
         self._ms_info = {}
@@ -140,27 +145,21 @@ class MsPlot:
         self._plots_locked = True
 
         # Single plot or combine plots into layout using subplots (rows, columns)
-        # Not layout if subplots is single plot (default if None) or if only one plot saved
-        subplots = self._plot_inputs['subplots']
-        layout_plot, is_layout = self._layout_plots(subplots)
-
-        # Render to bokeh figure
-        if is_layout:
-            # Show plots in columns
-            plot = hv.render(layout_plot.cols(subplots[1]))
-        else:
-            # Show single plot
-            plot = hv.render(layout_plot)
+        layout_plot = self._layout_plots(self._plot_inputs['subplots'])
+
+        # Render plot as Bokeh Figure or GridPlot so can show() in script without tying up thread
+        bokeh_fig = hv.render(layout_plot)
 
         self._plots_locked = False
         if self._plot_params:
+            # Show plot and plot inputs in tabs
             column = pn.Column()
             for param in self._plot_params:
                 column.append(pn.pane.Str(param))
-            tabs = pn.Tabs(('Plot', plot), ('Plot Inputs', column))
+            tabs = pn.Tabs(('Plot', bokeh_fig), ('Plot Inputs', column))
             tabs.show(title=self._app_name, threaded=True)
         else:
-            show(plot)
+            show(bokeh_fig)
 
     def save(self, filename='ms_plot.png', fmt='auto', width=900, height=600):
         '''
@@ -175,78 +174,103 @@ class MsPlot:
 
         start_time = time.time()
 
-        # Save single plot or combine plots into layout using subplots (rows, columns).
-        # Not layout if subplots is single plot or if only one plot saved.
-        subplots = self._plot_inputs['subplots']
-        layout_plot, is_layout = self._layout_plots(subplots)
+        name, ext = os.path.splitext(filename)
+        fmt = ext[1:] if fmt=='auto' else fmt
 
-        if is_layout:
-            # Save plots combined into one layout
-            hvplot.save(layout_plot.cols(subplots[1]), filename=filename, fmt=fmt)
-            self._logger.info("Saved plot to %s.", filename)
-        else:
-            # Save plots individually, with index appended if exprange='all' and multiple plots.
-            if self._plot_inputs['iter_axis'] is None:
-                hvplot.save(layout_plot.opts(width=width, height=height), filename=filename, fmt=fmt)
-                self._logger.info("Saved plot to %s.", filename)
-            else:
-                name, ext = os.path.splitext(filename)
-                iter_range = self._plot_inputs['iter_range'] # None or (start, end)
-                plot_idx = 0 if iter_range is None else iter_range[0]
-
-                for plot in self._plots:
-                    exportname = f"{name}_{plot_idx}{ext}"
-                    hvplot.save(plot.opts(width=width, height=height), filename=exportname, fmt=fmt)
-                    self._logger.info("Saved plot to %s.", exportname)
-                    plot_idx += 1
+        # Combine plots into layout using subplots (rows, columns) if not single plot.
+        # Set fixed size for export.
+        layout_plot = self._layout_plots(self._plot_inputs['subplots'], (width, height))
 
+        if not isinstance(layout_plot, hv.Layout) and self._plot_inputs['iter_axis']:
+            # Save iterated plots individually, with index appended to filename
+            plot_idx = 0 if self._plot_inputs['iter_range'] is None else self._plot_inputs['iter_range'][0]
+            for plot in self._plots:
+                exportname = f"{name}_{plot_idx}{ext}"
+                self._save_plot(plot, exportname, fmt)
+                plot_idx += 1
+        else:
+            self._save_plot(layout_plot, filename, fmt)
         self._logger.debug("Save elapsed time: %.2fs.", time.time() - start_time)
 
-    def _layout_plots(self, subplots):
+    def _layout_plots(self, subplots, fixed_size=None):
+        ''' Combine plots in a layout, using fixed size for the layout if given '''
         subplots = (1, 1) if subplots is None else subplots
-        num_plots = len(self._plots)
-        num_layout_plots = min(num_plots, np.prod(subplots))
+        num_plots = min(len(self._plots), np.prod(subplots))
+        plot_width = fixed_size[0] if fixed_size else None
+        plot_height = fixed_size[1] if fixed_size else None
 
-        if num_layout_plots == 1:
-            return self._plots[0], False
+        if num_plots == 1:
+            # Single plot, not layout
+            plot = self._plots[0]
+            if fixed_size:
+                plot = plot.opts(responsive=False, width=plot_width, height=plot_height, clone=True)
+            return plot
 
         # Set plots in layout
-        plot_count = 0
-        layout = None
-        for i in range(num_layout_plots):
+        layout_plot = None
+        for i in range(num_plots):
             plot = self._plots[i]
-            layout = plot if layout is None else layout + plot
-            plot_count += 1
-
-        is_layout = plot_count > 1
-        return layout, is_layout
+            if fixed_size:
+                plot = plot.opts(responsive=False, width=plot_width, height=plot_height, clone=True)
+            layout_plot = plot if layout_plot is None else layout_plot + plot
+
+        # Layout in columns
+        return layout_plot.cols(subplots[1])
+
+    def _save_plot(self, plot, filename, fmt):
+        ''' Save plot using hvplot, else bokeh '''
+        # Remove toolbar unless html
+        toolbar = 'right' if fmt=='html' else None
+        plot = plot.opts(toolbar=toolbar, clone=True)
+
+        try:
+            hvplot.save(plot, filename=filename, fmt=fmt)
+        except (Exception, RuntimeError) as exc:
+            # Fails if hvplot cannot find web driver or fmt is svg.
+            # Render a Bokeh Figure or GridPlot, create webdriver, then use Bokeh to export.
+            fig = hv.render(plot)
+            if fmt=='html':
+                save(fig, filename)
+            elif fmt in ['png', 'svg']:
+                # Use Chrome web driver
+                service = webdriver.ChromeService()
+                options = webdriver.ChromeOptions()
+                options.add_argument('--headless')
+                options.add_argument('--no-sandbox')
+
+                with webdriver.Chrome(service=service, options=options) as driver:
+                    if fmt=='png':
+                        export_png(fig, filename=filename, webdriver=driver)
+                    elif fmt=='svg':
+                        export_svg(fig, filename=filename, webdriver=driver)
+            else:
+                raise ValueError(f"Invalid fmt or filename extension {fmt} for save()") from exc
+        self._logger.info("Saved plot to %s.", filename)
 
     def _set_ms(self, ms_path):
         ''' Set MsData and update ms info for input ms filepath (MSv2 or zarr), if set.
-            Return whether ms changed (false if ms is None). '''
+            Return whether ms changed (false if ms_path is None, not set yet), even if error. '''
         self._ms_info['ms'] = ms_path
         ms_error = ""
-        ms_changed = ms_path and (not self._data or not self._data.is_ms_path(ms_path))
-
-        if ms_changed:
-            try:
-                # Set new MS data
-                self._data = MsData(ms_path, self._logger)
-                data_path = self._data.get_path()
-                self._ms_info['ms'] = data_path
-                root, ext = os.path.splitext(os.path.basename(data_path))
-                while ext != '':
-                    root, ext = os.path.splitext(root)
-                self._ms_info['basename'] = root
-                self._ms_info['data_dims'] = self._data.get_data_dimensions()
-            except RuntimeError as e:
-                ms_error = str(e)
-                self._data = None
-
+        if not ms_path or (self._data and self._data.is_ms_path(ms_path)):
+            return False
+
+        try:
+            # Set new MS data
+            self._data = MsData(ms_path, self._logger)
+            data_path = self._data.get_path()
+            self._ms_info['ms'] = data_path
+            root, ext = os.path.splitext(os.path.basename(data_path))
+            while ext != '':
+                root, ext = os.path.splitext(root)
+            self._ms_info['basename'] = root
+            self._ms_info['data_dims'] = self._data.get_data_dimensions()
+        except RuntimeError as e:
+            ms_error = str(e)
+            self._data = None
         if ms_error:
             self._notify(ms_error, 'error', 0)
-
-        return ms_changed
+        return True
 
     def _notify(self, message, level, duration=3000):
         ''' Log message. If show_gui, notify user with toast for duration in ms.

vidavis/plot/ms_plot/_raster_plot_gui.py

@@ -0,0 +1,91 @@
+'''
+    Create interactive GUI for ms raster plotting
+'''
+
+import holoviews as hv
+import panel as pn
+from vidavis.plot.ms_plot._ms_plot_selectors import (file_selector, title_selector, style_selector,
+    axis_selector, aggregation_selector, iteration_selector, selection_selector, plot_starter)
+
+def create_raster_gui(callbacks, data_dims, x_axis, y_axis):
+    ''' Use Holoviz Panel to create a dashboard for plot inputs and raster plot display. '''
+    # ------------------
+    # PLOT INPUTS COLUMN
+    # ------------------
+    # Select MS
+    file_selectors = file_selector('Path to MeasurementSet (ms or zarr) for plot', '~' , callbacks['filename'])
+
+    # Select style - colormaps, colorbar, color limits
+    style_selectors = style_selector(callbacks['style'], callbacks['color'])
+
+    # Select x, y, and vis axis
+    axis_selectors = axis_selector(x_axis, y_axis, data_dims, True, callbacks['axes'])
+
+    # Select from ProcessingSet and MeasurementSet
+    selection_selectors = selection_selector(callbacks['select_ps'], callbacks['select_ms'])
+
+    # Generic axis options, updated when ms is set
+    axis_options = data_dims if data_dims else []
+
+    # Select aggregator and axes to aggregate
+    agg_selectors = aggregation_selector(axis_options, callbacks['aggregation'])
+
+    # Select iter_axis and iter value or range
+    iter_selectors = iteration_selector(axis_options, callbacks['iter_values'], callbacks['iteration'])
+
+    # Set title
+    title_input = title_selector(callbacks['title'])
+
+    # Put user input widgets in accordion with only one card active at a time (toggle)
+    selectors = pn.Accordion(
+        ("Select file", file_selectors),         # [0]
+        ("Plot style", style_selectors),         # [1]
+        ("Data Selection", selection_selectors), # [2]
+        ("Plot axes", axis_selectors),           # [3]
+        ("Aggregation", agg_selectors),          # [4]
+        ("Iteration", iter_selectors),           # [5]
+        ("Plot title", title_input),             # [6]
+    )
+    selectors.toggle = True
+
+    # Plot button and spinner while plotting
+    init_plot = plot_starter(callbacks['plot_updating'])
+
+    # -------------------------
+    # PLOT WITH CURSOR POSITION
+    # -------------------------
+    # Connect plot to filename and plot button; add pointer stream for cursor info
+    dmap = hv.DynamicMap(
+        pn.bind(
+            callbacks['update_plot'],
+            ms=file_selectors[0][0],
+            do_plot=init_plot[0],
+        ),
+        streams=[hv.streams.PointerXY()] # for cursor location
+    )
+
+    # ----------------------------------------------
+    # GUI LAYOUT OF PLOT TABS AND PLOT INPUTS COLUMN
+    # ----------------------------------------------
+    return pn.Row(
+        pn.Tabs(             # Row [0]
+            ('Plot',
+                pn.Column(                # Tabs [0]
+                    dmap,           # [0]
+                    pn.WidgetBox(), # [1] cursor location
+                )
+            ),
+            ('Plot Inputs', pn.Column()), # Tabs [1]
+            sizing_mode='stretch_width',
+        ),
+        pn.Spacer(width=10), # Row [1]
+        pn.Column(  # Row [2]
+            pn.Spacer(height=25), # Column [0]
+            selectors,            # Column [1]
+            init_plot,            # Column [2]
+            width_policy='min',
+            width=400,
+            sizing_mode='stretch_height',
+        ),
+        sizing_mode='stretch_height',
+    )

{vidavis-0.0.5.dist-info → vidavis-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vidavis
-Version: 0.0.5
+Version: 0.0.7
 Summary: Radio astronomy visibility data visualization
 License: LGPL
 Author-email: Darrell Schiebel <darrell@schiebel.us>,Pam Harris <pharris@nrao.edu>
@@ -8,6 +8,8 @@ Requires-Python: >=3.11, <3.14
 Requires-Dist: bokeh==3.6.1
 Requires-Dist: graphviper
 Requires-Dist: hvplot
+Requires-Dist: matplotlib
+Requires-Dist: selenium
 Description-Content-Type: text/x-rst
 
 vidavis - VIsibility DAta VISualization
@@ -46,9 +48,6 @@ Requirements
 - Optionally `python-casacore <https://pypi.org/project/python-casacore/>`_ or
   `casatools <https://pypi.org/project/casatools/>`_ for MSv2 conversion
 
-- Optionally `Selenium <https://www.selenium.dev/documentation/en/>`_ along with
-  a web driver to export to file using ``save()``
-
 Install
 ```````
 
@@ -57,33 +56,14 @@ Install
 MSv2 Conversion
 ^^^^^^^^^^^^^^^
 
-To enable conversion from MSv2 to MSv4 with **python-casacore** use (this only works for Linux):
+To enable conversion from MSv2 to MSv4 with **python-casacore** for Linux only:
 
 - :code:`pip install "xradio[python-casacore]"`
 
-On macOS it is required to pre-install `python-casacore` using:
+On macOS it is required to pre-install **python-casacore**:
 
 - :code:`conda install -c conda-forge python-casacore`
 
-Exporting Plots
-^^^^^^^^^^^^^^^
-
-To enable exporting plots to file without showing the plot, using preferred web
-driver:
-
-**Selenium** with **geckodriver** and **Firefox** (to ensure compatible versions):
-
-- :code:`conda install -c conda-forge selenium firefox geckodriver`
-
-**Selenium** with **ChromeDriver** (Chrome), with the executable
-**chromedriver** in your PATH:
-
-- :code:`conda install -c conda-forge selenium python-chromedriver-binary`
-
-or:
-
-- :code:`pip install selenium chromedriver-binary`
-
 Simple MsRaster Usage Example
 `````````````````````````````
 

{vidavis-0.0.5.dist-info → vidavis-0.0.7.dist-info}/RECORD

@@ -1,7 +1,7 @@
 vidavis/LICENSE.rst,sha256=qzGpkvhDzf_MgF1PIn6rCmYPrcEhkfrBUchosLJj-U4,26371
 vidavis/__init__.py,sha256=nhvlqlYhl4NzFO2WPTYfhyZkNdpfkBUTUOgJWKCK4Tc,1681
 vidavis/apps/__init__.py,sha256=ZQ5v1VFtjn3ztmuOHLOk5WbC1uLNIgL9rbHQ4v0zJwY,87
-vidavis/apps/_ms_raster.py,sha256=A7KCJWW2-f6cONMTHjwFfHIqFhAJt00ophe1Zm2XX1E,44996
+vidavis/apps/_ms_raster.py,sha256=z6abcuznCSudJVcBnp0KpF5mwuRzbON8IFZtKhWnWKk,46831
 vidavis/bokeh/__init__.py,sha256=gdPPxBCe0enCSjvPFxkgMlhpVnAMFXKcw9GN28tVdXM,95
 vidavis/bokeh/_palette.py,sha256=gzfJHuUgqxd8hJpZe-gQPFTCPq9f5I8uLEkHAK5FNDM,2480
 vidavis/data/__init__.py,sha256=-RDRe0PYK6vPlhdRV2Dy1vGbnDGoXWDATmfxaR-gXcE,48
@@ -18,10 +18,11 @@ vidavis/data/measurement_set/processing_set/_ps_stats.py,sha256=4uLImKCANXLUM8jO
 vidavis/data/measurement_set/processing_set/_xds_data.py,sha256=qLO2VkLINkSAQ7CGRFmpWQYpHrP4XoJJkwA4pp9DO8M,5253
 vidavis/plot/__init__.py,sha256=thxe5vAGdpEiqoKPHLJoWUqKMVrUVx0ajpsGf5pVP98,95
 vidavis/plot/ms_plot/__init__.py,sha256=wY0_7gY9M6K1D6tKQsr89L_uSs3seJlD-uicx7dx5Mo,74
-vidavis/plot/ms_plot/_ms_plot.py,sha256=pS-l6-cxmnsIa4Dm-yyPkgimh7sFSl6qBIpXxK9HV_Y,11414
+vidavis/plot/ms_plot/_ms_plot.py,sha256=FlCeoeqLbvjK5ARae-S2QivJZ1Yig3IhKOvi1FRhXbk,12660
 vidavis/plot/ms_plot/_ms_plot_constants.py,sha256=cX_TQhKJ3hJzPuRYmuRJxue1sjq82yl_ZN2_w6TshmI,930
 vidavis/plot/ms_plot/_ms_plot_selectors.py,sha256=BZQwARvMPdk78n6Rh2tOaSc8GenZBrxHZb14oFD9gJM,10785
 vidavis/plot/ms_plot/_raster_plot.py,sha256=lNa9i_eJ8F8Fc2zcHLRcaxKKOELk3x_QmXT__T76pg8,10999
+vidavis/plot/ms_plot/_raster_plot_gui.py,sha256=ttUC_HW-INm82ToOEZzlNjAAsQGCLpxcZH5-te-W4jk,3448
 vidavis/plot/ms_plot/_raster_plot_inputs.py,sha256=vPjZPDuB26ENxwv4z7dUa_c5TqByXejScY6hqEnLFLU,4768
 vidavis/plot/ms_plot/_time_ticks.py,sha256=j-DcPh7RfGE8iX2bPjLQDQPIbiAbmjiEWQnKmdMWA3I,1773
 vidavis/plot/ms_plot/_xds_plot_axes.py,sha256=EeWvAbiKV33nEWdI8V3M0uwLTnycq4bFYBOyVWkxCu0,4429
@@ -29,8 +30,8 @@ vidavis/toolbox/__init__.py,sha256=jqFa-eziVz_frNnXxwjJFK36qNpz1H38s-VlpBcq-R8,1
 vidavis/toolbox/_app_context.py,sha256=H7gtF8RrAH46FqDcMobv3KM1Osbnapgu6aTG-m3VCWA,3049
 vidavis/toolbox/_logging.py,sha256=OEisrd8FM8VTNBMc7neLh9ekelf29ZILYB5pScebly0,2739
 vidavis/toolbox/_static.py,sha256=HJLMtClppgOJXWAtV6Umn5EqN80u0oZiIouQ1JsB9PM,2346
-vidavis/__version__.py,sha256=0Uc3zDF9XadX9j08cOIZbGoZsI0JPHTxemWum3gkos4,21
-vidavis-0.0.5.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
-vidavis-0.0.5.dist-info/METADATA,sha256=sxHI7zSrrwDcnRaaeZLSmqyDRRsHM-5hSBrHzB-g5TQ,2850
-vidavis-0.0.5.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
-vidavis-0.0.5.dist-info/RECORD,,
+vidavis/__version__.py,sha256=7sNbee72r3qCAVloFm2RRWCTa9gVjMS7nCJcJ-URzl4,21
+vidavis-0.0.7.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
+vidavis-0.0.7.dist-info/METADATA,sha256=OxbpEuebhkcUmr7Ka5Tx4rK2GMfmw15Ls8yegy3XwRA,2242
+vidavis-0.0.7.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+vidavis-0.0.7.dist-info/RECORD,,