vibespatial.raster.rasterize

Vector-to-raster conversion: rasterize geometries onto a grid.

GPU path uses an NVRTC per-pixel point-in-polygon kernel adapted from kernels/predicates/point_in_polygon.py. CPU fallback uses rasterio.features.

Bead o17.8.8 (GridSpec + CPU baseline) and o17.8.9 (GPU rasterize).

Functions

rasterize_cpu(...)

Rasterize geometries onto a grid using rasterio (CPU).

rasterize_gpu(...)

Rasterize polygons onto a grid using GPU per-pixel PIP kernel.

rasterize_owned(...)

Rasterize geometries onto a grid with automatic GPU/CPU dispatch.

Module Contents

vibespatial.raster.rasterize.rasterize_cpu(geometries, values: numpy.ndarray, grid_spec: vibespatial.raster.buffers.GridSpec) vibespatial.raster.buffers.OwnedRasterArray

Rasterize geometries onto a grid using rasterio (CPU).

Parameters

geometriessequence of Shapely geometries

Geometries to rasterize.

valuesnp.ndarray

Value for each geometry (same length as geometries).

grid_specGridSpec

Target grid specification.

Returns

OwnedRasterArray

vibespatial.raster.rasterize.rasterize_gpu(geometries, values: numpy.ndarray, grid_spec: vibespatial.raster.buffers.GridSpec) vibespatial.raster.buffers.OwnedRasterArray

Rasterize polygons onto a grid using GPU per-pixel PIP kernel.

Parameters

geometriessequence of Shapely geometries

Polygon/MultiPolygon geometries to rasterize.

valuesnp.ndarray

Value for each geometry.

grid_specGridSpec

Target grid specification.

Returns

OwnedRasterArray

vibespatial.raster.rasterize.rasterize_owned(geometries, values: numpy.ndarray, grid_spec: vibespatial.raster.buffers.GridSpec, *, use_gpu: bool | None = None) vibespatial.raster.buffers.OwnedRasterArray

Rasterize geometries onto a grid with automatic GPU/CPU dispatch.

Parameters

geometriessequence of Shapely geometries

Geometries to rasterize.

valuesnp.ndarray

Value per geometry.

grid_specGridSpec

Target grid specification.

use_gpubool or None

Force GPU (True), force CPU (False), or auto-dispatch (None). Auto uses GPU when available and pixel count exceeds threshold.

Returns

OwnedRasterArray