mraspaud · Dec 15, 2023 · May 6, 2022 · May 6, 2022 · May 6, 2022 · May 6, 2022
@@ -46,6 +46,7 @@ The following people have made contributions to this project:
 - [Lu Liu (yukaribbba)](https://github.com/yukaribbba)
 - [Andrea Meraner (ameraner)](https://github.com/ameraner)
 - [Aronne Merrelli (aronnem)](https://github.com/aronnem)
+- [Luca Merucci (lmeru)](https://github.com/lmeru)
 - [Lucas Meyer (LTMeyer)](https://github.com/LTMeyer)
 - [Ondrej Nedelcev (nedelceo)](https://github.com/nedelceo)
 - [Oana Nicola](https://github.com/)

@@ -35,6 +35,7 @@ dependencies:
   - mock
   - libtiff
   - geoviews
+  - hvplot
   - zarr
   - python-eccodes
   # 2.19.1 seems to cause library linking issues

@@ -25,10 +25,11 @@
 
 import numpy as np
 import xarray as xr
+from holoviews import Overlay
 from pyresample.geometry import AreaDefinition, BaseDefinition, SwathDefinition
 from xarray import DataArray
 
-from satpy.composites import IncompatibleAreas
+from satpy.composites import IncompatibleAreas, enhance2dataset
 from satpy.composites.config_loader import load_compositor_configs_for_sensors
 from satpy.dataset import DataID, DataQuery, DatasetDict, combine_metadata, dataset_walker, replace_anc
 from satpy.dependency_tree import DependencyTree
@@ -37,6 +38,12 @@
 from satpy.resample import get_area_def, prepare_resampler, resample_dataset
 from satpy.writers import load_writer
 
+try:
+    import hvplot.xarray  # noqa
+except ImportError:
+    hvplot.xarray = None
+
+
 LOG = logging.getLogger(__name__)
 
 
@@ -1022,6 +1029,70 @@ def to_geoviews(self, gvtype=None, datasets=None, kdims=None, vdims=None, dynami
 
         return gview
 
+    def to_hvplot(self, datasets=None, *args, **kwargs):
+        """Convert satpy Scene to Hvplot.
+
+        Args:
+            datasets (list): Limit included products to these datasets.
+            kwargs: hvplot options list.
+
+        Returns: hvplot object that contains within it the plots of datasets list.
+                 As default it contains all Scene datasets plots and a plot title is shown.
+
+        Example usage:
+           scene_list = ['ash','IR_108']
+           plot = scn.to_hvplot(datasets=scene_list)
+
+           plot.ash+plot.IR_108
+
+        """
+        def _get_crs(xarray_ds):
+            return xarray_ds.area.to_cartopy_crs()
+
+        def _get_timestamp(xarray_ds):
+            time = xarray_ds.attrs['start_time']
+            return time.strftime('%Y %m %d -- %H:%M UTC')
+
+        def _get_units(xarray_ds, variable):
+            return xarray_ds[variable].attrs['units']
+
+        def _plot_rgb(xarray_ds, variable, **defaults):
+            img = enhance2dataset(xarray_ds[variable])
+            return img.hvplot.rgb(bands='bands', title=title,
+                                  clabel='', **defaults)
+
+        def _plot_quadmesh(xarray_ds, variable, **defaults):
+            return xarray_ds[variable].hvplot.quadmesh(
+                clabel=f'[{_get_units(xarray_ds,variable)}]', title=title,
+                **defaults)
+
+        if hvplot.xarray is None:
+            raise ImportError("'hvplot' must be installed to use this feature")
+
+        plot = Overlay()
+        xarray_ds = self.to_xarray_dataset(datasets)
+        ccrs = _get_crs(xarray_ds)
+
+        if datasets is None:
+            datasets = list(xarray_ds.keys())
+
+        defaults = dict(x='x', y='y', data_aspect=1, project=True, geo=True,
+                        crs=ccrs, projection=ccrs, rasterize=True, coastline='110m',
+                        cmap='Plasma', responsive=True, dynamic=False, framewise=True,
+                        colorbar=False, global_extent=False, xlabel='Longitude',
+                        ylabel='Latitude')
+
+        defaults.update(kwargs)
+
+        for element in datasets:
+            title = f'{element} @ {_get_timestamp(xarray_ds)}'
+            if xarray_ds[element].shape[0] == 3:
+                plot[element] = _plot_rgb(xarray_ds, element, **defaults)
+            else:
+                plot[element] = _plot_quadmesh(xarray_ds, element, **defaults)
+
+        return plot
+
     def to_xarray_dataset(self, datasets=None):
         """Merge all xr.DataArrays of a scene to a xr.DataSet.
 

@@ -37,7 +37,9 @@
 
 test_requires = ['behave', 'h5py', 'netCDF4', 'pyhdf', 'imageio', 'pylibtiff',
                  'rasterio', 'geoviews', 'trollimage', 'fsspec', 'bottleneck',
-                 'rioxarray', 'pytest', 'pytest-lazy-fixture', 'defusedxml']
+                 'rioxarray', 'pytest', 'pytest-lazy-fixture', 'defusedxml',
+                 'hvplot']
+
 
 extras_require = {
     # Readers:
@@ -78,6 +80,7 @@
     'doc': ['sphinx', 'sphinx_rtd_theme', 'sphinxcontrib-apidoc'],
     # Other
     'geoviews': ['geoviews'],
+    'hvplot': ['hvplot'],
     'overlays': ['pycoast', 'pydecorate'],
     'tests': test_requires,
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -35,6 +35,7 @@ dependencies: @@
       - mock
       - libtiff
       - geoviews
+      - hvplot
       - zarr
       - python-eccodes
       # 2.19.1 seems to cause library linking issues
@@ Expand Down @@