ssb-sgis 1.0.12__tar.gz → 1.0.14__tar.gz

{ssb_sgis-1.0.12 → ssb_sgis-1.0.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ssb-sgis
-Version: 1.0.12
+Version: 1.0.14
 Summary: GIS functions used at Statistics Norway.
 Home-page: https://github.com/statisticsnorway/ssb-sgis
 License: MIT

{ssb_sgis-1.0.12 → ssb_sgis-1.0.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "ssb-sgis"
-version = "1.0.12"
+version = "1.0.14"
 description = "GIS functions used at Statistics Norway."
 authors = ["Morten Letnes <morten.letnes@ssb.no>"]
 license = "MIT"

{ssb_sgis-1.0.12 → ssb_sgis-1.0.14}/src/sgis/__init__.py

@@ -96,6 +96,7 @@ from .maps.maps import explore
 from .maps.maps import explore_locals
 from .maps.maps import qtm
 from .maps.maps import samplemap
+from .maps.norge_i_bilder_wms import NorgeIBilderWms
 from .maps.thematicmap import ThematicMap
 from .maps.tilesources import kartverket as kartverket_tiles
 from .maps.tilesources import xyz as xyztiles

{ssb_sgis-1.0.12 → ssb_sgis-1.0.14}/src/sgis/maps/explore.py

@@ -44,6 +44,7 @@ from ..geopandas_tools.general import clean_geoms
 from ..geopandas_tools.general import make_all_singlepart
 from ..geopandas_tools.geometry_types import get_geom_type
 from ..geopandas_tools.geometry_types import to_single_geom_type
+from .norge_i_bilder_wms import NorgeIBilderWms
 
 try:
     from ..raster.image_collection import Band
@@ -157,7 +158,9 @@ def to_tile(tile: str | xyzservices.TileProvider, max_zoom: int) -> folium.TileL
     except TypeError:
         name = tile
 
-    if not isinstance(tile, str):
+    if isinstance(tile, folium.map.Layer) and not isinstance(tile, folium.GeoJson):
+        return tile
+    elif not isinstance(tile, str):
         try:
             return folium.TileLayer(tile, name=name, max_zoom=max_zoom)
         except TypeError:
@@ -183,7 +186,9 @@ def to_tile(tile: str | xyzservices.TileProvider, max_zoom: int) -> folium.TileL
     return folium.TileLayer(provider, name=name, attr=attr, max_zoom=max_zoom)
 
 
-def _single_band_to_arr(band, mask, name, raster_data_dict):
+def _single_band_to_arr_is_too_much_nodata(
+    band, mask, name, raster_data_dict, max_nodata_percentage
+) -> bool:
     if band.has_array and mask is None:
         arr = band.values
     elif band.has_array:
@@ -191,8 +196,8 @@ def _single_band_to_arr(band, mask, name, raster_data_dict):
     else:
         arr = band.load(indexes=1, bounds=mask).values
 
-    if _is_too_much_nodata([arr], band.nodata):
-        return False
+    if _is_too_much_nodata([arr], band.nodata, max_nodata_percentage):
+        return True
 
     bounds: tuple = (
         _any_to_bbox_crs4326(mask, band.crs)
@@ -216,25 +221,28 @@ def _single_band_to_arr(band, mask, name, raster_data_dict):
     except Exception:
         raster_data_dict["date"] = None
 
-    return True
+    return False
 
 
 def _is_too_much_nodata(
     arrays: list[np.ndarray],
-    nodata: int | None = None,
-    max_nodata_percentage: int = 100,
+    nodata: int | None,
+    max_nodata_percentage: int,
 ) -> bool:
     return (
         any(arr.shape[0] == 0 for arr in arrays)
         or any(
             (
                 isinstance(arr, np.ma.core.MaskedArray)
-                and np.mean((arr.mask) | (arr.data == nodata) | (np.isnan(arr.data)))
+                and np.mean(((arr.mask) | (arr.data == nodata) | (np.isnan(arr.data))))
                 > (max_nodata_percentage / 100)
             )
             for arr in arrays
         )
-        or any(np.mean(arr == nodata) > (max_nodata_percentage / 100) for arr in arrays)
+        or any(
+            np.mean((arr == nodata) | (np.isnan(arr))) > (max_nodata_percentage / 100)
+            for arr in arrays
+        )
     )
 
 
@@ -271,6 +279,8 @@ class Explore(Map):
         decimals: int = 6,
         max_images: int = 10,
         max_nodata_percentage: int = 100,
+        display: bool = True,
+        norge_i_bilder: bool = False,
         **kwargs,
     ) -> None:
         """Initialiser.
@@ -299,6 +309,10 @@ class Explore(Map):
             decimals: Number of decimals in the coordinates.
             max_nodata_percentage: Maximum percentage nodata values (e.g. clouds) ro allow in
                 image arrays.
+            display: Whether to display the map interactively.
+            norge_i_bilder: If True, all Norge i bilder images in bounds will be loaded
+                into the map. Can optionally be set to an instance of NorgeIBilderWms to filter
+                years and names.
             **kwargs: Additional keyword arguments. Can also be geometry-like objects
                 where the key is the label.
         """
@@ -314,6 +328,8 @@ class Explore(Map):
         self.decimals = decimals
         self.max_images = max_images
         self.max_nodata_percentage = max_nodata_percentage
+        self.display = display
+        self.norge_i_bilder = norge_i_bilder
         self.legend = None
 
         self.browser = browser
@@ -457,8 +473,7 @@ class Explore(Map):
 
         if center is None:
             self.to_show = self._gdfs
-            self._explore(**kwargs)
-            return
+            return self._explore(**kwargs)
 
         size = size if size else 1000
 
@@ -622,6 +637,8 @@ class Explore(Map):
                 f.write(self.map._repr_html_())
         elif self.browser:
             run_html_server(self.map._repr_html_())
+        elif not self.display:
+            return
         else:
             display(self.map)
 
@@ -751,6 +768,14 @@ class Explore(Map):
         for tile in tiles:
             to_tile(tile, max_zoom=self.max_zoom).add_to(mapobj)
 
+    def _add_norge_i_bilder(self, map_: folium.Map, bbox: Any) -> None:
+        if not isinstance(self.norge_i_bilder, NorgeIBilderWms):
+            self.norge_i_bilder = NorgeIBilderWms()
+
+        tiles = self.norge_i_bilder.get_tiles(bbox, max_zoom=self.max_zoom)
+        for tile in tiles.values():
+            map_.add_child(tile)
+
     def _create_continous_map(self):
         self._prepare_continous_map()
         if self.scheme:
@@ -957,6 +982,9 @@ class Explore(Map):
             m.get_root().add_child(style)
             # folium.LayerControl(collapsed=False).add_to(m)
 
+        if self.norge_i_bilder:
+            self._add_norge_i_bilder(m, bounds)
+
         return m
 
     def _make_geojson(
@@ -1103,6 +1131,7 @@ class Explore(Map):
             highlight_function=highlight_function,
             smooth_factor=self.smooth_factor,
             show=show,
+            # on_each_feature="function(feature,layer){layer.bindTooltip(feature.properties.NAME,{permanent:true,direction:'center'});return layer;}",
             **kwargs,
         )
 
@@ -1132,6 +1161,8 @@ def _tooltip_popup(
         return folium.GeoJsonTooltip(fields, **kwargs)
     elif type_ == "popup":
         return folium.GeoJsonPopup(fields, **kwargs)
+    else:
+        raise ValueError(type_)
 
 
 def _intersects_if_not_none_or_empty(obj: Any, other: Any) -> bool:
@@ -1345,7 +1376,10 @@ def _add_one_image(
     if len(image) < 3:
         for band in image:
             name = _determine_label(band, band.name or name)
-            _single_band_to_arr(band, mask, name, raster_data_dict)
+            if _single_band_to_arr_is_too_much_nodata(
+                band, mask, name, raster_data_dict, max_nodata_percentage
+            ):
+                return
         return raster_data_dict
 
     def load(band_id: str) -> Band:
@@ -1428,8 +1462,10 @@ def _image_collection_to_background_map(
             return out
 
         raster_data_dict = {}
-        out.append(raster_data_dict)
-        _single_band_to_arr(band, mask, name, raster_data_dict)
+        if not _single_band_to_arr_is_too_much_nodata(
+            band, mask, name, raster_data_dict, max_nodata_percentage
+        ):
+            out.append(raster_data_dict)
         return out
 
     else:
@@ -1462,7 +1498,7 @@ def _image_collection_to_background_map(
                 continue
             i = 1
             while x["label"] in {y["label"] for y in out}:
-                x["label"] = x["label"].rstrip(f"_{i}", "") + f"_{i + 1}"
+                x["label"] = x["label"].rstrip(f"_{i}") + f"_{i + 1}"
                 i += 1
 
             n_added_images += 1

{ssb_sgis-1.0.12 → ssb_sgis-1.0.14}/src/sgis/maps/maps.py

@@ -31,6 +31,7 @@ from ..geopandas_tools.geocoding import address_to_gdf
 from ..geopandas_tools.geometry_types import get_geom_type
 from .explore import Explore
 from .map import Map
+from .norge_i_bilder_wms import NorgeIBilderWms
 from .thematicmap import ThematicMap
 
 try:
@@ -57,6 +58,9 @@ def _get_location_mask(kwargs: dict, gdfs) -> tuple[GeoDataFrame | None, dict]:
         "akersveien": (10.7476367, 59.9222191),
         "kongsvinger": (12.0035242, 60.1875279),
         "stavanger": (5.6960601, 58.8946196),
+        "trondheim": (10.39677054, 63.42687816),
+        "trondhjem": (10.39677054, 63.42687816),
+        "bergen": (5.32574594, 60.39550309),
         "volda": (6.0705987, 62.146643),
     }
 
@@ -87,6 +91,7 @@ def explore(
     size: int | None = None,
     max_images: int = 10,
     max_nodata_percentage: int = 100,
+    norge_i_bilder: bool | NorgeIBilderWms = False,
     **kwargs,
 ) -> Explore:
     """Interactive map of GeoDataFrames with layers that can be toggled on/off.
@@ -118,6 +123,9 @@ def explore(
             map. Defaults to 10.
         max_nodata_percentage: Maximum percentage nodata values (e.g. clouds) ro allow in
             image arrays.
+        norge_i_bilder: If True, all Norge i bilder images in bounds will be loaded
+            into the map. Can optionally be set to an instance of NorgeIBilderWms to filter
+            years and names.
         **kwargs: Keyword arguments to pass to geopandas.GeoDataFrame.explore, for
             instance 'cmap' to change the colors, 'scheme' to change how the data
             is grouped. This defaults to 'fisherjenkssampled' for numeric data.
@@ -167,6 +175,7 @@ def explore(
             max_zoom=max_zoom,
             max_images=max_images,
             max_nodata_percentage=max_nodata_percentage,
+            norge_i_bilder=norge_i_bilder,
             **kwargs,
         )
 
@@ -224,6 +233,7 @@ def explore(
             max_zoom=max_zoom,
             max_images=max_images,
             max_nodata_percentage=max_nodata_percentage,
+            norge_i_bilder=norge_i_bilder,
             **kwargs,
         )
 
@@ -235,6 +245,7 @@ def explore(
         smooth_factor=smooth_factor,
         max_images=max_images,
         max_nodata_percentage=max_nodata_percentage,
+        norge_i_bilder=norge_i_bilder,
         **kwargs,
     )
 
@@ -260,6 +271,7 @@ def samplemap(
     browser: bool = False,
     max_images: int = 10,
     max_nodata_percentage: int = 100,
+    norge_i_bilder: bool | NorgeIBilderWms = False,
     **kwargs,
 ) -> Explore:
     """Shows an interactive map of a random area of GeoDataFrames.
@@ -295,6 +307,9 @@ def samplemap(
             map. Defaults to 10.
         max_nodata_percentage: Maximum percentage nodata values (e.g. clouds) ro allow in
             image arrays.
+        norge_i_bilder: If True, all Norge i bilder images in bounds will be loaded
+            into the map. Can optionally be set to an instance of NorgeIBilderWms to filter
+            years and names.
         **kwargs: Keyword arguments to pass to geopandas.GeoDataFrame.explore, for
             instance 'cmap' to change the colors, 'scheme' to change how the data
             is grouped. This defaults to 'fisherjenkssampled' for numeric data.
@@ -378,6 +393,7 @@ def samplemap(
         smooth_factor=smooth_factor,
         max_images=max_images,
         max_nodata_percentage=max_nodata_percentage,
+        norge_i_bilder=norge_i_bilder,
         **kwargs,
     )
 
@@ -392,6 +408,7 @@ def clipmap(
     browser: bool = False,
     max_images: int = 10,
     max_nodata_percentage: int = 100,
+    norge_i_bilder: bool | NorgeIBilderWms = False,
     **kwargs,
 ) -> Explore | Map:
     """Shows an interactive map of a of GeoDataFrames clipped to the mask extent.
@@ -422,6 +439,9 @@ def clipmap(
             map. Defaults to 10.
         max_nodata_percentage: Maximum percentage nodata values (e.g. clouds) ro allow in
             image arrays.
+        norge_i_bilder: If True, all Norge i bilder images in bounds will be loaded
+            into the map. Can optionally be set to an instance of NorgeIBilderWms to filter
+            years and names.
         **kwargs: Keyword arguments to pass to geopandas.GeoDataFrame.explore, for
             instance 'cmap' to change the colors, 'scheme' to change how the data
             is grouped. This defaults to 'fisherjenkssampled' for numeric data.
@@ -457,6 +477,7 @@ def clipmap(
             smooth_factor=smooth_factor,
             max_images=max_images,
             max_nodata_percentage=max_nodata_percentage,
+            norge_i_bilder=norge_i_bilder,
             **kwargs,
         )
         m.mask = mask