rasterix 0.1a3__tar.gz → 0.1.1__tar.gz

{rasterix-0.1a3 → rasterix-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rasterix
-Version: 0.1a3
+Version: 0.1.1
 Summary: Raster extensions for Xarray
 License-Expression: Apache-2.0
 License-File: LICENSE
@@ -17,6 +17,7 @@ Requires-Dist: affine
 Requires-Dist: numpy>=2
 Requires-Dist: pandas>=2
 Requires-Dist: xarray>=2025
+Requires-Dist: xproj>=0.2.0
 Provides-Extra: dask
 Requires-Dist: dask-geopandas; extra == 'dask'
 Provides-Extra: docs
@@ -44,65 +45,29 @@ Requires-Dist: exactextract; extra == 'exactextract'
 Requires-Dist: sparse; extra == 'exactextract'
 Provides-Extra: rasterize
 Requires-Dist: rasterio; extra == 'rasterize'
-Provides-Extra: test
-Requires-Dist: dask-geopandas; extra == 'test'
-Requires-Dist: exactextract; extra == 'test'
-Requires-Dist: geodatasets; extra == 'test'
-Requires-Dist: hypothesis; extra == 'test'
-Requires-Dist: netcdf4; extra == 'test'
-Requires-Dist: pooch; extra == 'test'
-Requires-Dist: rasterio; extra == 'test'
-Requires-Dist: rioxarray; extra == 'test'
-Requires-Dist: sparse; extra == 'test'
 Description-Content-Type: text/markdown
 
 # rasterix: Raster tricks for Xarray
 
-[![GitHub Workflow CI Status](https://img.shields.io/github/actions/workflow/status/dcherian/rasterix/test.yml?branch=main&logo=github&style=flat)](https://github.com/dcherian/rasterix/actions)
+[![GitHub Workflow CI Status](https://img.shields.io/github/actions/workflow/status/xarray-contrib/rasterix/test.yml?branch=main&logo=github&style=flat)](https://github.com/xarray-contrib/rasterix/actions)
 [![Documentation Status](https://readthedocs.org/projects/rasterix/badge/?version=latest)](https://rasterix.readthedocs.io/en/latest/?badge=latest)
 [![PyPI](https://img.shields.io/pypi/v/rasterix.svg?style=flat)](https://pypi.org/project/rasterix/)
 [![Conda-forge](https://img.shields.io/conda/vn/conda-forge/rasterix.svg?style=flat)](https://anaconda.org/conda-forge/rasterix)
 
-<img src="rasterix.png" width="300">
+<img src="_static/rasterix.png" width="300">
 
 This WIP project contains tools to make it easier to analyze raster data with Xarray.
-
-The intent is to provide reusable building blocks for the many sub-ecosystems around: e.g. rioxarray, odc-geo, etc.
-
-## Contents
-
 It currently has two pieces.
 
-### 1. RasterIndex
-
-See `src/ rasterix/raster_index.py` and `notebooks/raster_index.ipynb` for a brief demo.
-
-### 2. Dask-aware rasterization wrappers
+1. `RasterIndex` for indexing using the affine transform recorded in GeoTIFFs.
+1. Dask-aware rasterization wrappers around `exactextract`, `rasterio.features.rasterize`, and `rasterio.features.geometry_mask`.
 
-See `src/rasterix/rasterize.py` for dask-aware wrappers around [`exactextract`](https://github.com/dcherian/rasterix/blob/ec3f51e60e25aa312e6f48c4b22f91bec70413ed/rasterize.py#L165), [`rasterio.features.rasterize`](https://github.com/dcherian/rasterix/blob/ec3f51e60e25aa312e6f48c4b22f91bec70413ed/rasterize.py#L307), and [`rasterio.features.geometry_mask`](https://github.com/dcherian/rasterix/blob/ec3f51e60e25aa312e6f48c4b22f91bec70413ed/rasterize.py#L472).
-
-This code is likely to move elsewhere!
+Our intent is to provide reusable building blocks for the many sub-ecosystems around: e.g. `rioxarray`, `odc.geo`, etc.
 
 ## Installing
 
-### PyPI
-
 `rasterix` alpha releases are available on pypi
 
 ```
 pip install rasterix
 ```
-
-## Developing
-
-1. Clone the repo
-   ```
-   git remote add upstream git@github.com:dcherian/rasterix.git
-   cd rasterix
-   ```
-1. [Install hatch](https://hatch.pypa.io/1.12/install/)
-1. Run the tests
-   ```
-   hatch env run --env test.py3.13 run-pytest  # Run the tests without coverage reports
-   hatch env run --env test.py3.13 run-coverage-html   # Run the tests with an html coverage report
-   ```

rasterix-0.1.1/README.md

@@ -0,0 +1,24 @@
+# rasterix: Raster tricks for Xarray
+
+[![GitHub Workflow CI Status](https://img.shields.io/github/actions/workflow/status/xarray-contrib/rasterix/test.yml?branch=main&logo=github&style=flat)](https://github.com/xarray-contrib/rasterix/actions)
+[![Documentation Status](https://readthedocs.org/projects/rasterix/badge/?version=latest)](https://rasterix.readthedocs.io/en/latest/?badge=latest)
+[![PyPI](https://img.shields.io/pypi/v/rasterix.svg?style=flat)](https://pypi.org/project/rasterix/)
+[![Conda-forge](https://img.shields.io/conda/vn/conda-forge/rasterix.svg?style=flat)](https://anaconda.org/conda-forge/rasterix)
+
+<img src="_static/rasterix.png" width="300">
+
+This WIP project contains tools to make it easier to analyze raster data with Xarray.
+It currently has two pieces.
+
+1. `RasterIndex` for indexing using the affine transform recorded in GeoTIFFs.
+1. Dask-aware rasterization wrappers around `exactextract`, `rasterio.features.rasterize`, and `rasterio.features.geometry_mask`.
+
+Our intent is to provide reusable building blocks for the many sub-ecosystems around: e.g. `rioxarray`, `odc.geo`, etc.
+
+## Installing
+
+`rasterix` alpha releases are available on pypi
+
+```
+pip install rasterix
+```

{rasterix-0.1a3 → rasterix-0.1.1}/pyproject.toml

@@ -29,6 +29,7 @@ dependencies = [
     "affine",
     "pandas>=2",
     "numpy>=2",
+    "xproj>=0.2.0",
     "xarray>=2025",
 ]
 dynamic=["version"]
@@ -37,17 +38,6 @@ dynamic=["version"]
 dask = ["dask-geopandas"]
 rasterize = ["rasterio"]
 exactextract = ["exactextract", "sparse"]
-test = [
-  "geodatasets",
-  "pooch",
-  "dask-geopandas",
-  "rasterio",
-  "rioxarray",
-  "exactextract",
-  "sparse",
-  "netCDF4",
-  "hypothesis",
-]
 docs = [
   "geodatasets",
   "pooch",
@@ -75,6 +65,9 @@ docs = [
 [tool.hatch]
 version.source = "vcs"
 
+[tool.hatch.metadata]
+allow-direct-references = true
+
 [tool.hatch.build]
 hooks.vcs.version-file = "src/rasterix/_version.py"
 
@@ -87,14 +80,26 @@ exclude = [
     "doc",
 ]
 
-[tool.hatch.envs.test]
-dependencies = [
+[dependency-groups]
+test = [
     "coverage",
+    "dask-geopandas",
+    "exactextract",
+    "geodatasets",
+    "hypothesis",
+    "netCDF4",
+    "pooch",
+    "pytest",
     "pytest",
     "pytest-cov",
-    "pytest-xdist"
+    "pytest-xdist",
+    "rasterio",
+    "sparse",
+    "xarray @ git+https://github.com/pydata/xarray.git@26ccc7f8f014f29c551fd566a04d6e9f878c0b0b", # https://github.com/pydata/xarray/pull/10980
 ]
-features = ["test"]
+
+[tool.hatch.envs.test]
+dependency-groups = ["test"]
 
 [[tool.hatch.envs.test.matrix]]
 python = ["3.11", "3.13"]

{rasterix-0.1a3 → rasterix-0.1.1}/src/rasterix/_version.py

@@ -1,7 +1,14 @@
 # file generated by setuptools-scm
 # don't change, don't track in version control
 
-__all__ = ["__version__", "__version_tuple__", "version", "version_tuple"]
+__all__ = [
+    "__version__",
+    "__version_tuple__",
+    "version",
+    "version_tuple",
+    "__commit_id__",
+    "commit_id",
+]
 
 TYPE_CHECKING = False
 if TYPE_CHECKING:
@@ -9,13 +16,19 @@ if TYPE_CHECKING:
     from typing import Union
 
     VERSION_TUPLE = Tuple[Union[int, str], ...]
+    COMMIT_ID = Union[str, None]
 else:
     VERSION_TUPLE = object
+    COMMIT_ID = object
 
 version: str
 __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
+commit_id: COMMIT_ID
+__commit_id__: COMMIT_ID
 
-__version__ = version = '0.1a3'
-__version_tuple__ = version_tuple = (0, 1, 'a3')
+__version__ = version = '0.1.1'
+__version_tuple__ = version_tuple = (0, 1, 1)
+
+__commit_id__ = commit_id = None

rasterix-0.1.1/src/rasterix/lib.py

@@ -0,0 +1,106 @@
+"""Shared library utilities for rasterix."""
+
+import logging
+
+from affine import Affine
+
+# Define TRACE level (lower than DEBUG)
+TRACE = 5
+logging.addLevelName(TRACE, "TRACE")
+
+
+class TraceLogger(logging.Logger):
+    """Logger with trace level support."""
+
+    def trace(self, message, *args, **kwargs):
+        """Log a message with severity 'TRACE'."""
+        if self.isEnabledFor(TRACE):
+            self._log(TRACE, message, args, **kwargs)
+
+
+# Set the custom logger class
+logging.setLoggerClass(TraceLogger)
+
+# Create logger for the rasterix package
+logger = logging.getLogger("rasterix")
+
+
+def affine_from_tiepoint_and_scale(
+    tiepoint: list[float] | tuple[float, ...],
+    scale: list[float] | tuple[float, ...],
+) -> Affine:
+    """Create an Affine transform from GeoTIFF tiepoint and pixel scale.
+
+    Parameters
+    ----------
+    tiepoint : list or tuple
+        GeoTIFF model tiepoint in format [I, J, K, X, Y, Z]
+        where (I, J, K) are pixel coords and (X, Y, Z) are world coords.
+    scale : list or tuple
+        GeoTIFF model pixel scale in format [ScaleX, ScaleY, ScaleZ].
+
+    Returns
+    -------
+    Affine
+        Affine transformation matrix.
+
+    Raises
+    ------
+    AssertionError
+        If ScaleZ is not 0 (only 2D rasters are supported).
+
+    Examples
+    --------
+    >>> tiepoint = [0.0, 0.0, 0.0, 323400.0, 4265400.0, 0.0]
+    >>> scale = [30.0, 30.0, 0.0]
+    >>> affine = affine_from_tiepoint_and_scale(tiepoint, scale)
+    """
+    if len(tiepoint) < 6:
+        raise ValueError(f"tiepoint must have at least 6 elements, got {len(tiepoint)}")
+    if len(scale) < 3:
+        raise ValueError(f"scale must have at least 3 elements, got {len(scale)}")
+
+    i, j, k, x, y, z = tiepoint[:6]
+    scale_x, scale_y, scale_z = scale[:3]
+
+    # We only support 2D rasters
+    assert scale_z == 0, f"Z pixel scale must be 0 for 2D rasters, got {scale_z}"
+
+    # The tiepoint gives us the world coordinates at pixel (I, J)
+    # Affine transform: x_world = c + i * a, y_world = f + j * e
+    # So: c = x - i * scale_x, f = y - j * scale_y
+    c = x - i * scale_x
+    f = y - j * scale_y
+
+    return Affine.translation(c, f) * Affine.scale(scale_x, scale_y)
+
+
+def affine_from_stac_proj_metadata(metadata: dict) -> Affine | None:
+    """Extract Affine transform from STAC projection metadata.
+
+    Parameters
+    ----------
+    metadata : dict
+        Dictionary containing STAC metadata. Should contain a 'proj:transform' key.
+
+    Returns
+    -------
+    Affine or None
+        Affine transformation matrix if 'proj:transform' is found, None otherwise.
+
+    Examples
+    --------
+    >>> metadata = {"proj:transform": [30.0, 0.0, 323400.0, 0.0, 30.0, 4268400.0]}
+    >>> affine = affine_from_stac_proj_metadata(metadata)
+    """
+    if "proj:transform" not in metadata:
+        return None
+
+    transform = metadata["proj:transform"]
+    # proj:transform is a 3x3 matrix in row-major order, but typically only 6 elements
+    # [a, b, c, d, e, f, 0, 0, 1] where the affine is constructed from first 6 elements
+    if len(transform) < 6:
+        raise ValueError(f"proj:transform must have at least 6 elements, got {len(transform)}")
+
+    a, b, c, d, e, f = transform[:6]
+    return Affine(a, b, c, d, e, f)