ssb-sgis 1.0.6__py3-none-any.whl → 1.0.7__py3-none-any.whl

sgis/helpers.py

@@ -130,6 +130,12 @@ def in_jupyter() -> bool:
         return False
 
 
+def _fix_path(path: str) -> str:
+    return (
+        str(path).replace("\\", "/").replace(r"\"", "/").replace("//", "/").rstrip("/")
+    )
+
+
 def get_all_files(root: str, recursive: bool = True) -> list[str]:
     """Fetch all files in a directory.
 
@@ -141,11 +147,11 @@ def get_all_files(root: str, recursive: bool = True) -> list[str]:
         A list of file paths.
     """
     if not recursive:
-        return [path for path in glob.glob(str(Path(root)) + "/**")]
+        return [_fix_path(path) for path in glob.glob(str(Path(root)) + "/**")]
     paths = []
     for root_dir, _, files in os.walk(root):
         for file in files:
-            path = os.path.join(root_dir, file)
+            path = _fix_path(os.path.join(root_dir, file))
             paths.append(path)
     return paths
 

sgis/maps/explore.py

@@ -194,8 +194,6 @@ def _single_band_to_arr(band, mask, name, raster_data_dict):
             gpd.GeoSeries(box(*band.bounds), crs=band.crs).to_crs(4326).geometry.values
         ).bounds
     )
-    # if np.max(arr) > 0:
-    #     arr = arr / 255
     try:
         raster_data_dict["cmap"] = band.get_cmap(arr)
     except Exception:
@@ -533,8 +531,6 @@ class Explore(Map):
                 arr = arr.data
             if "bool" in str(arr.dtype):
                 arr = np.where(arr, 1, 0)
-            # if np.max(arr[~np.isnan(arr)]) > 255:
-            #     arr = (arr - np.min(arr)) / (np.max(arr) - np.min(arr))
             try:
                 arr = (arr - np.min(arr)) / (np.max(arr) - np.min(arr))
             except Exception:
@@ -1182,28 +1178,17 @@ class Explore(Map):
                 break
 
             crs = red_band.crs
-
-            bounds: tuple = (
-                _any_to_bbox_crs4326(mask, crs)
-                if mask is not None
-                else (
-                    union_all(
-                        gpd.GeoSeries(box(*red_band.bounds), crs=crs)
-                        .to_crs(4326)
-                        .geometry.values
-                    ).bounds
-                )
-            )
+            bounds = to_bbox(to_gdf(red_band.bounds, crs).to_crs(4326))
 
             red_band = red_band.values
             blue_band = blue_band.values
             green_band = green_band.values
 
-            if red_band.shape[0] == 0:
-                continue
-            if blue_band.shape[0] == 0:
-                continue
-            if green_band.shape[0] == 0:
+            if (
+                red_band.shape[0] == 0
+                or blue_band.shape[0] == 0
+                or green_band.shape[0] == 0
+            ):
                 continue
 
             # to 3d array in shape (x, y, 3)
@@ -1268,16 +1253,10 @@ def _determine_label(
         )
         if does_not_have_generic_name:
             return obj_name
-    # try:
-    #     if obj.tile and obj.date:
-    #         name = f"{obj.tile}_{obj.date[:8]}"
-    # except (ValueError, AttributeError):
-    #     name = None
-
     try:
         # Images/Bands/Collections constructed from arrays have no path stems
-        if obj.stem:
-            name = obj.stem
+        if obj.name:
+            name = obj.name
         else:
             name = str(obj)[:23]
     except (AttributeError, ValueError):

sgis/maps/map.py

@@ -14,7 +14,11 @@ import numpy as np
 import pandas as pd
 from geopandas import GeoDataFrame
 from geopandas import GeoSeries
-from jenkspy import jenks_breaks
+
+try:
+    from jenkspy import jenks_breaks
+except ImportError:
+    pass
 from mapclassify import classify
 from pandas.errors import PerformanceWarning
 from shapely import Geometry

sgis/raster/base.py

@@ -189,30 +189,6 @@ def _gdf_to_geojson_with_col(gdf: GeoDataFrame, values: np.ndarray) -> list[dict
         ]
 
 
-def _shapely_to_raster(
-    geometry: Geometry,
-    res: int | float,
-    fill: int = 0,
-    all_touched: bool = False,
-    merge_alg: Callable = MergeAlg.replace,
-    default_value: int = 1,
-    dtype: Any | None = None,
-) -> np.array:
-    shape = _get_shape_from_bounds(geometry.bounds, res=res, indexes=1)
-    transform = _get_transform_from_bounds(geometry.bounds, shape)
-
-    return features.rasterize(
-        [(geometry, default_value)],
-        out_shape=shape,
-        transform=transform,
-        fill=fill,
-        all_touched=all_touched,
-        merge_alg=merge_alg,
-        default_value=default_value,
-        dtype=dtype,
-    )
-
-
 @contextmanager
 def memfile_from_array(array: np.ndarray, **profile) -> rasterio.MemoryFile:
     """Yield a memory file from a numpy array."""
@@ -228,33 +204,3 @@ def get_index_mapper(df: pd.DataFrame) -> tuple[dict[int, int], str]:
     idx_mapper = dict(enumerate(df.index))
     idx_name = df.index.name
     return idx_mapper, idx_name
-
-
-NESSECARY_META = [
-    "path",
-    "type",
-    "bounds",
-    "crs",
-]
-
-PROFILE_ATTRS = [
-    "driver",
-    "dtype",
-    "nodata",
-    "crs",
-    "height",
-    "width",
-    "blockysize",
-    "blockxsize",
-    "tiled",
-    "compress",
-    "interleave",
-    "count",  # TODO: this should be based on band_index / array depth, so will have no effect
-    "indexes",  # TODO
-]
-
-ALLOWED_KEYS = (
-    NESSECARY_META
-    + PROFILE_ATTRS
-    + ["array", "res", "transform", "name", "date", "regex"]
-)