rio-tiler 7.7.3.post1__tar.gz → 7.8.0__tar.gz

{rio_tiler-7.7.3.post1 → rio_tiler-7.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rio-tiler
-Version: 7.7.3.post1
+Version: 7.8.0
 Summary: User friendly Rasterio plugin to read raster datasets.
 Project-URL: Homepage, https://cogeotiff.github.io/rio-tiler/
 Project-URL: Documentation, https://cogeotiff.github.io/rio-tiler/

{rio_tiler-7.7.3.post1 → rio_tiler-7.8.0}/pyproject.toml

@@ -161,7 +161,7 @@ exclude = [
 max-complexity = 14
 
 [tool.bumpversion]
-current_version = "7.7.3.post1"
+current_version = "7.8.0"
 parse = """(?x)
     (?P<major>\\d+)\\.
     (?P<minor>\\d+)\\.

{rio_tiler-7.7.3.post1 → rio_tiler-7.8.0}/rio_tiler/__init__.py

@@ -1,6 +1,6 @@
 """rio-tiler."""
 
-__version__ = "7.7.3.post1"
+__version__ = "7.8.0"
 
 from . import (  # noqa
     colormap,

{rio_tiler-7.7.3.post1 → rio_tiler-7.8.0}/rio_tiler/io/xarray.py

@@ -108,7 +108,14 @@ class XarrayReader(BaseReader):
         self._dims = [
             d
             for d in self.input.dims
-            if d not in [self.input.rio.x_dim, self.input.rio.y_dim]
+            if d
+            not in [
+                self.input.rio.x_dim,
+                self.input.rio.y_dim,
+                "spatial_ref",
+                "crs_wkt",
+                "grid_mapping",
+            ]
         ]
         assert len(self._dims) in [0, 1], "Can't handle >=4D DataArray"
 
@@ -130,9 +137,20 @@ class XarrayReader(BaseReader):
         `Bands` are all dimensions not defined as spatial dims by rioxarray.
         """
         if not self._dims:
-            coords_name = list(self.input.coords)
-            if len(coords_name) > 3 and (coord := coords_name[2]):
-                return [str(self.input.coords[coord].data)]
+            coords_name = [
+                d
+                for d in self.input.coords
+                if d
+                not in [
+                    self.input.rio.x_dim,
+                    self.input.rio.y_dim,
+                    "spatial_ref",
+                    "crs_wkt",
+                    "grid_mapping",
+                ]
+            ]
+            if coords_name:
+                return [str(self.input.coords[coords_name[0]].data)]
 
             return [self.input.name or "array"]
 

{rio_tiler-7.7.3.post1 → rio_tiler-7.8.0}/rio_tiler/models.py

@@ -6,6 +6,7 @@ from typing import Any, Dict, List, Literal, Optional, Sequence, Tuple, Union
 
 import attr
 import numpy
+import rasterio
 from affine import Affine
 from color_operations import parse_operations, scale_dtype, to_math_type
 from numpy.typing import NDArray
@@ -719,6 +720,42 @@ class ImageData:
 
         return render(array.data, img_format=img_format, colormap=colormap, **kwargs)
 
+    def to_raster(self, dst_path: str, *, driver: str = "GTIFF", **kwargs: Any) -> None:
+        """Save ImageData array to file."""
+        if driver.upper() == "GTIFF":
+            if "transform" not in kwargs:
+                kwargs.update({"transform": self.transform})
+            if "crs" not in kwargs and self.crs:
+                kwargs.update({"crs": self.crs})
+
+        write_nodata = "nodata" in kwargs
+        count, height, width = self.array.shape
+
+        output_profile = {
+            "dtype": self.array.dtype,
+            "count": count if write_nodata else count + 1,
+            "height": height,
+            "width": width,
+        }
+        output_profile.update(kwargs)
+
+        with warnings.catch_warnings():
+            warnings.filterwarnings(
+                "ignore",
+                category=NotGeoreferencedWarning,
+                module="rasterio",
+            )
+            with rasterio.open(dst_path, "w", driver=driver, **output_profile) as dst:
+                dst.write(self.data, indexes=list(range(1, count + 1)))
+
+                # Use Mask as an alpha band
+                if not write_nodata:
+                    if ColorInterp.alpha not in dst.colorinterp:
+                        dst.colorinterp = *dst.colorinterp[:-1], ColorInterp.alpha
+                    dst.write(self.mask.astype(self.array.dtype), indexes=count + 1)
+
+        return
+
     def statistics(
         self,
         categorical: bool = False,
@@ -750,7 +787,7 @@ class ImageData:
         shape_crs: CRS = WGS84_CRS,
         cover_scale: int = 10,
     ) -> NDArray[numpy.floating]:
-        """Post-process image data.
+        """Get Coverage array for a Geometry.
 
         Args:
             shape (Dict): GeoJSON geometry or Feature.