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

vidavis/__version__.py

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

vidavis/apps/_ms_raster.py

@@ -244,10 +244,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.

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
@@ -140,27 +142,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
+        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,51 +171,78 @@ 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)
-
-        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]
+        name, ext = os.path.splitext(filename)
+        fmt = ext[1:] if fmt=='auto' else fmt
 
-                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.

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: vidavis
-Version: 0.0.5
+Version: 0.0.6
 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
 ```````
 
@@ -65,25 +64,6 @@ On macOS it is required to pre-install `python-casacore` using:
 
 - :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.6.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=laZNFItWW5X2bvKlBI8WXfCegFQBVvxdcmDiR1mrqV4,45205
 vidavis/bokeh/__init__.py,sha256=gdPPxBCe0enCSjvPFxkgMlhpVnAMFXKcw9GN28tVdXM,95
 vidavis/bokeh/_palette.py,sha256=gzfJHuUgqxd8hJpZe-gQPFTCPq9f5I8uLEkHAK5FNDM,2480
 vidavis/data/__init__.py,sha256=-RDRe0PYK6vPlhdRV2Dy1vGbnDGoXWDATmfxaR-gXcE,48
@@ -18,7 +18,7 @@ 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=lf_9vSal2sMmSdp9BxjMat38T1SJ3ptF_8-UUZ5tGqw,12592
 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
@@ -29,8 +29,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=UwR0idQSHgdEemLAk-o2bPOXYWCj-lyyQzAPFRLbziA,21
+vidavis-0.0.6.dist-info/WHEEL,sha256=B19PGBCYhWaz2p_UjAoRVh767nYQfk14Sn4TpIZ-nfU,87
+vidavis-0.0.6.dist-info/METADATA,sha256=u-RsO4qX6UALqttKxzQIatAw2CmEx8Elb2gY-NrakcI,2263
+vidavis-0.0.6.dist-info/licenses/LICENSE,sha256=IMF9i4xIpgCADf0U-V1cuf9HBmqWQd3qtI3FSuyW4zE,26526
+vidavis-0.0.6.dist-info/RECORD,,