core-lens 0.1.dev133__tar.gz → 0.1.dev134__tar.gz

{core_lens-0.1.dev133 → core_lens-0.1.dev134}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: core-lens
-Version: 0.1.dev133
+Version: 0.1.dev134
 Summary: Query, analyse, and visualise CoreStack's microwatershed and Earth science data through a clean, composable Python API.
 Project-URL: Homepage, https://github.com/ApoorvaKashyap/core-lens
 Project-URL: Issues, https://github.com/ApoorvaKashyap/core-lens/issues
@@ -34,6 +34,7 @@ Requires-Dist: matplotlib<4,>=3.10.9; extra == 'full'
 Requires-Dist: polars<2,>=1.39.0; extra == 'full'
 Requires-Dist: pyarrow<25,>=23.0.0; extra == 'full'
 Requires-Dist: pydantic<3,>=2.13.3; extra == 'full'
+Requires-Dist: pyogrio<1,>=0.10.0; extra == 'full'
 Requires-Dist: pyproj<4,>=3.7.2; extra == 'full'
 Requires-Dist: scipy<2,>=1.17.1; extra == 'full'
 Requires-Dist: shapely<3,>=2.1.2; extra == 'full'
@@ -52,6 +53,7 @@ Requires-Dist: matplotlib<4,>=3.10.9; extra == 'full-gpu'
 Requires-Dist: polars<2,>=1.39.0; extra == 'full-gpu'
 Requires-Dist: pyarrow<25,>=23.0.0; extra == 'full-gpu'
 Requires-Dist: pydantic<3,>=2.13.3; extra == 'full-gpu'
+Requires-Dist: pyogrio<1,>=0.10.0; extra == 'full-gpu'
 Requires-Dist: pyproj<4,>=3.7.2; extra == 'full-gpu'
 Requires-Dist: scipy<2,>=1.17.1; extra == 'full-gpu'
 Requires-Dist: shapely<3,>=2.1.2; extra == 'full-gpu'
@@ -72,6 +74,7 @@ Requires-Dist: loguru>=0.7.3; extra == 'spatial'
 Requires-Dist: polars<2,>=1.39.0; extra == 'spatial'
 Requires-Dist: pyarrow<25,>=23.0.0; extra == 'spatial'
 Requires-Dist: pydantic<3,>=2.13.3; extra == 'spatial'
+Requires-Dist: pyogrio<1,>=0.10.0; extra == 'spatial'
 Requires-Dist: pyproj<4,>=3.7.2; extra == 'spatial'
 Requires-Dist: shapely<3,>=2.1.2; extra == 'spatial'
 Provides-Extra: stats

{core_lens-0.1.dev133 → core_lens-0.1.dev134}/pyproject.toml

@@ -57,6 +57,7 @@ spatial = [
     "geopandas>=1.1.3,<2",
     "shapely>=2.1.2,<3",
     "pyproj>=3.7.2,<4",
+    "pyogrio>=0.10.0,<1",
 ]
 viz = [
     "core-lens[core]",

{core_lens-0.1.dev133 → core_lens-0.1.dev134}/src/core_lens/_version.py

@@ -18,7 +18,7 @@ version_tuple: tuple[int | str, ...]
 commit_id: str | None
 __commit_id__: str | None
 
-__version__ = version = '0.1.dev133'
-__version_tuple__ = version_tuple = (0, 1, 'dev133')
+__version__ = version = '0.1.dev134'
+__version_tuple__ = version_tuple = (0, 1, 'dev134')
 
 __commit_id__ = commit_id = None

{core_lens-0.1.dev133 → core_lens-0.1.dev134}/src/core_lens/export/formats.py

@@ -187,20 +187,50 @@ def geojson(result: "Result", path: str | pathlib.Path, **kwargs: Any) -> None:
             "Call .with_geometry() first to join the static geometry column before exporting to geojson."
         )
 
-    gdf = result.gdf()
+    geom_col = result.entity.geometry_col
+    data = result.data
+
+    driver = kwargs.pop("driver", "GeoJSON")
+    is_seq = driver == "GeoJSONSeq"
+
+    if kwargs:
+        from loguru import logger
+
+        logger.warning(
+            "geojson() now uses a custom streaming Polars writer. "
+            "Driver-specific kwargs ({}) are ignored.",
+            list(kwargs.keys()),
+        )
 
-    # Resolve write engine: prefer pyogrio (C-backed, no GDAL Python overhead),
-    # fall back to fiona if not installed.
-    engine = kwargs.pop("engine", None)
-    if engine is None:
-        try:
-            import pyogrio as _  # type: ignore[import-untyped]  # noqa: F401
+    import shapely
+    import tempfile
 
-            engine = "pyogrio"
-        except ModuleNotFoundError:
-            engine = "fiona"
+    wkb_arr = data[geom_col].to_numpy()
+    geoms = shapely.from_wkb(wkb_arr)
+    geom_strs = shapely.to_geojson(geoms)
 
-    # Always write as GeoJSON unless the caller overrides driver.
-    kwargs.setdefault("driver", "GeoJSON")
+    with tempfile.NamedTemporaryFile(mode="w+") as tmp:
+        data.drop(geom_col).write_ndjson(tmp.name)
+        tmp.seek(0)
 
-    gdf.to_file(path, engine=engine, **kwargs)
+        with open(path, "w") as out:
+            if not is_seq:
+                out.write('{"type": "FeatureCollection", "features": [\n')
+
+            first = True
+            for i, prop_json_str in enumerate(tmp):
+                if not first:
+                    out.write("\n" if is_seq else ",\n")
+                else:
+                    first = False
+
+                # construct the Feature JSON manually to avoid parsing/re-serialising
+                out.write(
+                    f'{{"type": "Feature", "geometry": {geom_strs[i]}, '
+                    f'"properties": {prop_json_str.strip()}}}'
+                )
+
+            if not is_seq:
+                out.write("\n]}\n")
+            else:
+                out.write("\n")