rio-tiler 7.7.4__tar.gz → 7.8.1__tar.gz

{rio_tiler-7.7.4 → rio_tiler-7.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rio-tiler
-Version: 7.7.4
+Version: 7.8.1
 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.4 → rio_tiler-7.8.1}/pyproject.toml

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

{rio_tiler-7.7.4 → rio_tiler-7.8.1}/rio_tiler/__init__.py

@@ -1,6 +1,6 @@
 """rio-tiler."""
 
-__version__ = "7.7.4"
+__version__ = "7.8.1"
 
 from . import (  # noqa
     colormap,

{rio_tiler-7.7.4 → rio_tiler-7.8.1}/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.

{rio_tiler-7.7.4 → rio_tiler-7.8.1}/rio_tiler/reader.py

@@ -297,6 +297,19 @@ def read(
             numpy.multiply(data, scales, out=data, casting="unsafe")
             numpy.add(data, offsets, out=data, casting="unsafe")
 
+            # apply scale/offsets to stats
+            if dataset_statistics:
+                scales = numpy.array(dataset.scales)[numpy.array(indexes) - 1].reshape(
+                    (-1, 1)
+                )
+                offsets = numpy.array(dataset.offsets)[numpy.array(indexes) - 1].reshape(
+                    (-1, 1)
+                )
+                stats_array = numpy.array(dataset_statistics)
+                numpy.multiply(stats_array, scales, out=stats_array, casting="unsafe")
+                numpy.add(stats_array, offsets, out=stats_array, casting="unsafe")
+                dataset_statistics = [tuple(s) for s in stats_array.tolist()]
+
         if post_process:
             data = post_process(data)