geo-explorer 0.9.8__py3-none-any.whl → 0.9.9__py3-none-any.whl

geo_explorer/nc.py

@@ -0,0 +1,199 @@
+import abc
+from typing import ClassVar
+
+import numpy as np
+import pandas as pd
+import pyproj
+import rasterio
+import shapely
+from geopandas import GeoDataFrame
+from geopandas import GeoSeries
+from shapely.geometry import Polygon
+
+try:
+    from xarray import DataArray
+    from xarray import Dataset
+except ImportError:
+
+    class DataArray:
+        """Placeholder."""
+
+    class Dataset:
+        """Placeholder."""
+
+
+from .utils import _PROFILE_DICT
+from .utils import get_xarray_bounds
+from .utils import time_method_call
+
+
+class AbstractImageConfig(abc.ABC):
+    rgb_bands: ClassVar[list[str] | None] = None
+    reducer: ClassVar[str | None] = None
+
+    def __init__(self, code_block: str | None = None) -> None:
+        self._code_block = code_block
+        self.code_block = code_block  # trigger setter
+
+    @property
+    def code_block(self) -> str | None:
+        return self._code_block
+
+    @code_block.setter
+    def code_block(self, value: str | None):
+        if value and (
+            "xarr=" not in value.replace(" ", "")
+            or not any(txt in value.replace(" ", "") for txt in ("=ds", "(ds"))
+        ):
+            raise ValueError(
+                "'code_block' must be a piece of code that takes the xarray object 'ds' and defines the object 'xarr'. "
+                f"Got '{value}'"
+            )
+        self._code_block = value
+
+    @abc.abstractmethod
+    def get_crs(self, ds: Dataset, path: str) -> pyproj.CRS:
+        pass
+
+    @abc.abstractmethod
+    def get_bounds(self, ds: Dataset, path: str) -> tuple[float, float, float, float]:
+        pass
+
+    @time_method_call(_PROFILE_DICT)
+    def filter_ds(
+        self,
+        ds: Dataset,
+        bounds: tuple[float, float, float, float],
+        code_block: str | None,
+    ) -> GeoDataFrame | None:
+        crs = self.get_crs(ds, None)
+        ds_bounds = get_xarray_bounds(ds)
+
+        bbox_correct_crs = (
+            GeoSeries([shapely.box(*bounds)], crs=4326).to_crs(crs).union_all()
+        )
+        clipped_bbox = bbox_correct_crs.intersection(shapely.box(*ds_bounds))
+        minx, miny, maxx, maxy = clipped_bbox.bounds
+
+        ds = ds.sel(
+            x=slice(minx, maxx),
+            y=slice(maxy, miny),
+        )
+
+        return _run_code_block(ds, code_block)
+
+    @time_method_call(_PROFILE_DICT)
+    def to_numpy(self, xarr: Dataset | DataArray) -> GeoDataFrame | None:
+        if isinstance(xarr, Dataset) and len(xarr.data_vars) == 1:
+            xarr = xarr[next(iter(xarr.data_vars))]
+        elif isinstance(xarr, Dataset):
+            try:
+                xarr = xarr[self.rgb_bands]
+            except Exception:
+                pass
+
+        if "time" in set(xarr.dims) and (
+            not hasattr(xarr["time"].values, "__len__") or len(xarr["time"].values) > 1
+        ):
+            if self.reducer is None:
+                xarr = xarr.isel(time=0)
+            else:
+                xarr = getattr(xarr, self.reducer)(dim="time")
+
+        if isinstance(xarr, Dataset) and self.rgb_bands:
+            return np.array([xarr[band].values for band in self.rgb_bands])
+        elif isinstance(xarr, Dataset):
+            return np.array([xarr[var].values for var in xarr.data_vars])
+
+        if isinstance(xarr, np.ndarray):
+            return xarr
+
+        return xarr.values
+
+    def __str__(self) -> str:
+        code_block = f"'{self.code_block}'" if self.code_block else None
+        return f"{self.__class__.__name__}({code_block})"
+
+    def __repr__(self) -> str:
+        return str(self)
+
+
+def _run_code_block(
+    ds: DataArray | Dataset, code_block: str | None
+) -> Dataset | DataArray:
+    if not code_block:
+        return ds
+
+    try:
+        xarr = eval(code_block)
+        if callable(xarr):
+            xarr = xarr(ds)
+    except SyntaxError:
+        loc = {}
+        exec(code_block, globals=globals() | {"ds": ds}, locals=loc)
+        xarr = loc["xarr"]
+
+    if isinstance(xarr, np.ndarray) and isinstance(ds, DataArray):
+        ds.values = xarr
+        return ds
+    elif isinstance(xarr, np.ndarray) and isinstance(ds, Dataset):
+        raise ValueError(
+            "Cannot return np.ndarray from 'code_block' if ds is xarray.Dataset."
+        )
+    return xarr
+
+
+class GeoTIFFConfig(AbstractImageConfig):
+    def get_crs(self, ds, path):
+        with rasterio.open(path) as src:
+            return src.crs
+
+    def get_bounds(self, ds, path) -> tuple[float, float, float, float]:
+        with rasterio.open(path) as src:
+            return tuple(
+                GeoSeries([shapely.box(*src.bounds)], crs=src.crs)
+                .to_crs(4326)
+                .total_bounds
+            )
+
+
+class NetCDFConfig(AbstractImageConfig):
+    """Sets the configuration for reading NetCDF files and getting crs and bounds.
+
+    Args:
+        code_block: String of Python code that takes an xarray.Dataset (ds) and returns an xarray.DataArray (xarr).
+            Note that the input Dataset must be references as 'ds' and the output must be assigned to 'xarr'.
+    """
+
+    rgb_bands: ClassVar[list[str]] = ["B4", "B3", "B2"]
+
+    def get_bounds(self, ds, path) -> tuple[float, float, float, float]:
+        return get_xarray_bounds(ds)
+
+    def get_crs(self, ds: Dataset, path: str) -> pyproj.CRS:
+        attrs = [x for x in ds.attrs if "projection" in x.lower() or "crs" in x.lower()]
+        if not attrs:
+            raise ValueError(f"Could not find CRS attribute in dataset: {ds}")
+        for i, attr in enumerate(attrs):
+            try:
+                return pyproj.CRS(ds.attrs[attr])
+            except Exception as e:
+                if i == len(attrs) - 1:
+                    attrs_dict = {attr: ds.attrs[attr] for attr in attrs}
+                    raise ValueError(
+                        f"No valid CRS attribute found among {attrs_dict}"
+                    ) from e
+
+
+class NBSNetCDFConfig(NetCDFConfig):
+    def get_crs(self, ds: Dataset, path: str) -> pyproj.CRS:
+        return pyproj.CRS(ds.UTM_projection.epsg_code)
+
+
+class Sentinel2NBSNetCDFConfig(NBSNetCDFConfig):
+    rgb_bands: ClassVar[list[str]] = ["B4", "B3", "B2"]
+
+
+def _pd():
+    """Function that makes sure 'pd' is not removed by 'ruff' fixes. Because pd is useful in code_block."""
+    pd

geo_explorer/utils.py

@@ -1,47 +1,24 @@
 import time
+from collections.abc import Callable
 from functools import wraps
 from pathlib import Path
-from collections.abc import Callable
 
 import dash_bootstrap_components as dbc
 from dash import html
 
+_PROFILE_DICT = {}
 
-def _standardize_path(path: str | Path) -> str:
-    """Make sure delimiter is '/' and path ends without '/'."""
-    return str(path).replace("\\", "/").replace(r"\"", "/")
 
+DEBUG: bool = 1
 
-def _clicked_button_style():
-    return {
-        "color": "#e4e4e4",
-        "background": "#2F3034",
-    }
 
-
-def _unclicked_button_style():
-    return {
-        "background": "#e4e4e4",
-        "color": "black",
-    }
-
-
-def get_button_with_tooltip(
-    button_text, id, tooltip_text: str, n_clicks=0, **button_kwargs
-) -> list[html.Button, dbc.Tooltip]:
-    return [
-        html.Button(
-            button_text,
-            id=id,
-            n_clicks=n_clicks,
-            **button_kwargs,
-        ),
-        dbc.Tooltip(
-            tooltip_text,
-            target=id,
-            delay={"show": 500, "hide": 100},
-        ),
-    ]
+def debug_print(*args):
+    print(
+        *(
+            f"{type(arg).__name__}: {arg}" if isinstance(arg, Exception) else arg
+            for arg in args
+        )
+    )
 
 
 def time_method_call(method_dict) -> Callable:
@@ -82,3 +59,75 @@ def time_function_call(method_dict):
         return wrapper
 
     return decorator
+
+
+if not DEBUG:
+
+    def debug_print(*args):
+        pass
+
+    def time_method_call(_) -> Callable:
+        def decorator(method):
+            @wraps(method)
+            def wrapper(self, *args, **kwargs):
+                return method(self, *args, **kwargs)
+
+            return wrapper
+
+        return decorator
+
+    def time_function_call(_):
+        def decorator(func):
+            @wraps(func)
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+
+            return wrapper
+
+        return decorator
+
+
+def get_xarray_bounds(ds) -> tuple[float, float, float, float]:
+    return (
+        float(ds["x"].min().values),
+        float(ds["y"].min().values),
+        float(ds["x"].max().values),
+        float(ds["y"].max().values),
+    )
+
+
+def _standardize_path(path: str | Path) -> str:
+    """Make sure delimiter is '/' and path ends without '/'."""
+    return str(path).replace("\\", "/").replace(r"\"", "/")
+
+
+def _clicked_button_style():
+    return {
+        "color": "#e4e4e4",
+        "background": "#2F3034",
+    }
+
+
+def _unclicked_button_style():
+    return {
+        "background": "#e4e4e4",
+        "color": "black",
+    }
+
+
+def get_button_with_tooltip(
+    button_text, id, tooltip_text: str, n_clicks=0, **button_kwargs
+) -> list[html.Button, dbc.Tooltip]:
+    return [
+        html.Button(
+            button_text,
+            id=id,
+            n_clicks=n_clicks,
+            **button_kwargs,
+        ),
+        dbc.Tooltip(
+            tooltip_text,
+            target=id,
+            delay={"show": 500, "hide": 100},
+        ),
+    ]

{geo_explorer-0.9.8.dist-info → geo_explorer-0.9.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: geo-explorer
-Version: 0.9.8
+Version: 0.9.9
 Summary: Explore geodata interactively.
 License: MIT
 Author: Morten Letnes
@@ -20,7 +20,7 @@ Requires-Dist: fsspec (>=2024.10.1)
 Requires-Dist: geopandas (>=0.14.0)
 Requires-Dist: jenkspy (>=0.3.2)
 Requires-Dist: matplotlib (>=3.7.0)
-Requires-Dist: msgspec (>=0.19.0,<0.20.0)
+Requires-Dist: msgspec (>=0.19.0)
 Requires-Dist: numpy (>=1.26.4)
 Requires-Dist: pandas (>=2.2.1)
 Requires-Dist: polars (>=1.32.0)

geo_explorer-0.9.9.dist-info/RECORD

@@ -0,0 +1,14 @@
+geo_explorer/__init__.py,sha256=LAnELq1tD-bpPHVM_bPa8DqAD4kY6dE9ZalZSp3R0Ug,108
+geo_explorer/assets/chroma.min.js,sha256=cNBWTSvPBSwnJYaqi4SM3IFXRjpWCLsa4VVfJ8yYWNw,46173
+geo_explorer/assets/on_each_feature.js,sha256=pjpr3MGlaTZf8pa191hRB5Iui8un7jBG5mFw_GNGqxE,2630
+geo_explorer/assets/stylesheet.css,sha256=1Dow3oh532wDZMEFr-VlkCdfekuf_20K97AVKatJun0,1957
+geo_explorer/file_browser.py,sha256=NmMAKHznJ1TOJ8TH-axwfvKaZ7r0gH9QSCmVCXHnfZw,23270
+geo_explorer/fs.py,sha256=N_WtpAjohO6OKAzoWzNxko0Rv_Y4oH0I8ackcoPu4f8,1693
+geo_explorer/geo_explorer.py,sha256=PzOfYFtiMwiO6YadXIsNwpZFi5fAQoRW9q43VQ_dD4s,192229
+geo_explorer/nc.py,sha256=zjW0X_prH1M-KHgJ2qspTlI5pgyBntWRD3GxGB0eO4A,6203
+geo_explorer/utils.py,sha256=qRLhthS881rKOUdItJCyo__3a4I2Xv3l50eeumO-8Ek,3234
+geo_explorer-0.9.9.dist-info/LICENSE,sha256=lL2h0dNKGTKAE0CjTy62SDbRennVD1xPgM5LzGqhKeo,1074
+geo_explorer-0.9.9.dist-info/LICENSE.md,sha256=hxspefYgWP3U6OZFhCifqWMI5ksnKzgFxNKgQnG7Ozc,1074
+geo_explorer-0.9.9.dist-info/METADATA,sha256=9vaB4hjPmzQF94l57v09FnCeAuu3rbhuIszV2wAavpc,8487
+geo_explorer-0.9.9.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+geo_explorer-0.9.9.dist-info/RECORD,,

geo_explorer-0.9.8.dist-info/RECORD

@@ -1,13 +0,0 @@
-geo_explorer/__init__.py,sha256=LAnELq1tD-bpPHVM_bPa8DqAD4kY6dE9ZalZSp3R0Ug,108
-geo_explorer/assets/chroma.min.js,sha256=cNBWTSvPBSwnJYaqi4SM3IFXRjpWCLsa4VVfJ8yYWNw,46173
-geo_explorer/assets/on_each_feature.js,sha256=pjpr3MGlaTZf8pa191hRB5Iui8un7jBG5mFw_GNGqxE,2630
-geo_explorer/assets/stylesheet.css,sha256=1Dow3oh532wDZMEFr-VlkCdfekuf_20K97AVKatJun0,1957
-geo_explorer/file_browser.py,sha256=A7BZyioiq1lJS5Eo1d292G4al9EVrUhzxjh_5shhQP8,22386
-geo_explorer/fs.py,sha256=N_WtpAjohO6OKAzoWzNxko0Rv_Y4oH0I8ackcoPu4f8,1693
-geo_explorer/geo_explorer.py,sha256=Dj3T64Jg7RW4LxIFOHGSm4CAZPMQXLDNgqdivlatjcs,179810
-geo_explorer/utils.py,sha256=Mg1NqUm8hq17z681YQExCve49eoJZjhoG8xWKwYfExE,2249
-geo_explorer-0.9.8.dist-info/LICENSE,sha256=lL2h0dNKGTKAE0CjTy62SDbRennVD1xPgM5LzGqhKeo,1074
-geo_explorer-0.9.8.dist-info/LICENSE.md,sha256=hxspefYgWP3U6OZFhCifqWMI5ksnKzgFxNKgQnG7Ozc,1074
-geo_explorer-0.9.8.dist-info/METADATA,sha256=dGNYMcISz5MOmN4DqpOBuvaPNf81efUYbhiyEBEzP7w,8495
-geo_explorer-0.9.8.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-geo_explorer-0.9.8.dist-info/RECORD,,