21037270339

Committed 15 Jan 2026 03:47PM UTC coverage: 78.884% (-0.7%) from 79.615%

Build # 21037270339

Build Type

Pull #221

github

Committed by

web-flow

Commit Message

Merge d5dad0f2b into eedc483f5

Pull Request Pull Request #221: Pixel_based_queries_rewrite

Run Details

3097 of 3926 relevant lines covered (78.88%)

0.79 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

81.2

geoengine/types.py

# pylint: disable=too-many-lines

"""
Different type mappings of geo engine types
"""

from __future__ import annotations

from abc import abstractmethod
from datetime import datetime, timezone
from enum import Enum
from typing import Any, Literal, cast
from uuid import UUID

import geoengine_openapi_client
import geoengine_openapi_client.models
import geoengine_openapi_client.models.raster_to_dataset_query_rectangle
import numpy as np
from attr import dataclass

from geoengine.colorizer import Colorizer
from geoengine.error import GeoEngineException, InputException, TypeException

DEFAULT_ISO_TIME_FORMAT = "%Y-%m-%dT%H:%M:%S.%f%z"


class SpatialBounds:
    """A spatial bounds object"""

    xmin: float
    ymin: float
    xmax: float
    ymax: float

    def __init__(self, xmin: float, ymin: float, xmax: float, ymax: float) -> None:
        """Initialize a new `SpatialBounds` object"""
        if (xmin > xmax) or (ymin > ymax):
            raise InputException("Bbox: Malformed since min must be <= max")

        self.xmin = xmin
        self.ymin = ymin
        self.xmax = xmax
        self.ymax = ymax

    def as_bbox_str(self, y_axis_first=False) -> str:
        """
        A comma-separated string representation of the spatial bounds with OGC axis ordering
        """
        bbox_tuple = self.as_bbox_tuple(y_axis_first=y_axis_first)
        return f"{bbox_tuple[0]},{bbox_tuple[1]},{bbox_tuple[2]},{bbox_tuple[3]}"

    def as_bbox_tuple(self, y_axis_first=False) -> tuple[float, float, float, float]:
        """
        Return the bbox with OGC axis ordering of the srs
        """

        if y_axis_first:
            return (self.ymin, self.xmin, self.ymax, self.xmax)

        return (self.xmin, self.ymin, self.xmax, self.ymax)

    def x_axis_size(self) -> float:
        """The size of the x axis"""
        return self.xmax - self.xmin

    def y_axis_size(self) -> float:
        """The size of the y axis"""
        return self.ymax - self.ymin


class BoundingBox2D(SpatialBounds):
    """'A 2D bounding box."""

    def to_api_dict(self) -> geoengine_openapi_client.BoundingBox2D:
        return geoengine_openapi_client.BoundingBox2D(
            lower_left_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.xmin,
                y=self.ymin,
            ),
            upper_right_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.xmax,
                y=self.ymax,
            ),
        )

    def intersection(self, other: BoundingBox2D) -> BoundingBox2D:
        return BoundingBox2D(
            min(self.xmin, other.xmin),
            min(self.ymin, other.ymin),
            max(self.xmax, other.xmax),
            max(self.ymax, other.ymax),
        )

    @staticmethod
    def from_response(response: geoengine_openapi_client.BoundingBox2D) -> BoundingBox2D:
        """create a `BoundingBox2D` from an API response"""
        lower_left = response.lower_left_coordinate
        upper_right = response.upper_right_coordinate

        return BoundingBox2D(
            lower_left.x,
            lower_left.y,
            upper_right.x,
            upper_right.y,
        )

    def __repr__(self) -> str:
        return f"BoundingBox2D(xmin={self.xmin}, ymin={self.ymin}, xmax={self.xmax}, ymax={self.ymax})"


class SpatialPartition2D(SpatialBounds):
    """A 2D spatial partition."""

    @staticmethod
    def from_response(response: geoengine_openapi_client.SpatialPartition2D) -> SpatialPartition2D:
        """create a `SpatialPartition2D` from an API response"""
        upper_left = response.upper_left_coordinate
        lower_right = response.lower_right_coordinate

        return SpatialPartition2D(
            upper_left.x,
            lower_right.y,
            lower_right.x,
            upper_left.y,
        )

    def to_api_dict(self) -> geoengine_openapi_client.SpatialPartition2D:
        return geoengine_openapi_client.SpatialPartition2D(
            upper_left_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.xmin,
                y=self.ymax,
            ),
            lower_right_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.xmax,
                y=self.ymin,
            ),
        )

    def to_bounding_box(self) -> BoundingBox2D:
        """convert to a `BoundingBox2D`"""
        return BoundingBox2D(self.xmin, self.ymin, self.xmax, self.ymax)

    @staticmethod
    def from_bounding_box(bbox: BoundingBox2D) -> SpatialPartition2D:
        """Creates a  `SpatialPartition2D` from a `BoundingBox2D`"""
        return SpatialPartition2D(bbox.xmin, bbox.ymin, bbox.xmax, bbox.ymax)

    def __repr__(self) -> str:
        return f"SpatialPartition2D(xmin={self.xmin}, ymin={self.ymin}, xmax={self.xmax}, ymax={self.ymax})"


class TimeInterval:
    """'A time interval."""

    start: np.datetime64
    end: np.datetime64 | None

    def __init__(self, start: datetime | np.datetime64, end: datetime | np.datetime64 | None = None) -> None:
        """Initialize a new `TimeInterval` object"""

        if isinstance(start, np.datetime64):
            self.start = start
        elif isinstance(start, datetime):
            # We assume that a datetime without a timezone means UTC
            if start.tzinfo is not None:
                start = start.astimezone(tz=timezone.utc).replace(tzinfo=None)
            self.start = np.datetime64(start)
        else:
            raise InputException("`start` must be of type `datetime.datetime` or `numpy.datetime64`")

        if end is None:
            self.end = self.start
        elif isinstance(end, np.datetime64):
            self.end = end
        elif isinstance(end, datetime):
            # We assume that a datetime without a timezone means UTC
            if end.tzinfo is not None:
                end = end.astimezone(tz=timezone.utc).replace(tzinfo=None)
            self.end = np.datetime64(end)
        else:
            raise InputException("`end` must be of type `datetime.datetime` or `numpy.datetime64`")

        # Check validity of time interval if an `end` exists
        if end is not None and start > end:
            raise InputException("Time inverval: Start must be <= End")

    def is_instant(self) -> bool:
        return self.end is None or self.start == self.end

    @property
    def time_str(self) -> str:
        """
        Return the time instance or interval as a string representation
        """

        start_iso = TimeInterval.__datetime_to_iso_str(self.start)

        if self.end is None or self.start == self.end:
            return start_iso

        end_iso = TimeInterval.__datetime_to_iso_str(self.end)

        return start_iso + "/" + end_iso

    @staticmethod
    def from_response(response: geoengine_openapi_client.models.TimeInterval) -> TimeInterval:
        """create a `TimeInterval` from an API response"""

        if response.start is None:
            raise TypeException("TimeInterval must have a start")

        start = cast(int, response.start)
        end = None
        if response.end is not None:
            end = cast(int, response.end)

        if start == end:
            end = None

        return TimeInterval(
            np.datetime64(start, "ms"),
            np.datetime64(end, "ms") if end is not None else None,
        )

    def __repr__(self) -> str:
        return f"TimeInterval(start={self.start}, end={self.end})"

    def to_api_dict(self) -> geoengine_openapi_client.TimeInterval:
        """create a openapi `TimeInterval` from self"""
        start = self.start.astype("datetime64[ms]").astype(int)
        end = self.end.astype("datetime64[ms]").astype(int) if self.end is not None else None

        # The openapi Timeinterval does not accept end: None. So we set it to start IF self is an instant.
        end = end if end is not None else start

        print(self, start, end)

        return geoengine_openapi_client.TimeInterval(start=int(start), end=int(end))

    @staticmethod
    def __datetime_to_iso_str(timestamp: np.datetime64) -> str:
        return str(np.datetime_as_string(timestamp, unit="ms", timezone="UTC")).replace("Z", "+00:00")

    def __eq__(self, other: Any) -> bool:
        """Check if two `TimeInterval` objects are equal."""
        if not isinstance(other, TimeInterval):
            return False
        return self.start == other.start and self.end == other.end


class SpatialResolution:
    """'A spatial resolution."""

    x_resolution: float
    y_resolution: float

    def __init__(self, x_resolution: float, y_resolution: float) -> None:
        """Initialize a new `SpatialResolution` object"""
        if x_resolution <= 0 or y_resolution <= 0:
            raise InputException("Resolution: Must be positive")

        self.x_resolution = x_resolution
        self.y_resolution = y_resolution

    def to_api_dict(self) -> geoengine_openapi_client.SpatialResolution:
        return geoengine_openapi_client.SpatialResolution(
            x=self.x_resolution,
            y=self.y_resolution,
        )

    @staticmethod
    def from_response(response: geoengine_openapi_client.SpatialResolution) -> SpatialResolution:
        """create a `SpatialResolution` from an API response"""
        return SpatialResolution(x_resolution=response.x, y_resolution=response.y)

    def as_tuple(self) -> tuple[float, float]:
        return (self.x_resolution, self.y_resolution)

    def resolution_ogc(self, srs_code: str) -> tuple[float, float]:
        """
        Return the resolution in OGC style
        """

        # TODO: why is the y resolution in this case negative but not in all other cases?
        if srs_code == "EPSG:4326":
            return (-self.y_resolution, self.x_resolution)

        return self.as_tuple()

    def __str__(self) -> str:
        return str(f"{self.x_resolution},{self.y_resolution}")

    def __repr__(self) -> str:
        return str(f"SpatialResolution(x={self.x_resolution}, y={self.y_resolution})")


class QueryRectangle:
    """
    A multi-dimensional query rectangle, consisting of spatial and temporal information.
    """

    __spatial_bounds: BoundingBox2D
    __time_interval: TimeInterval
    __srs: str

    def __init__(
        self,
        spatial_bounds: BoundingBox2D | SpatialPartition2D | tuple[float, float, float, float],
        time_interval: TimeInterval | tuple[datetime, datetime | None],
        srs="EPSG:4326",
    ) -> None:
        """
        Initialize a new `QueryRectangle` object

        Parameters
        ----------
        spatial_bounds
            The spatial bounds of the query rectangle.
            Either a `BoundingBox2D` or a tuple of floats (xmin, ymin, xmax, ymax)
        time_interval
            The time interval of the query rectangle.
            Either a `TimeInterval` or a tuple of `datetime.datetime` objects (start, end)
        """

        if not isinstance(spatial_bounds, BoundingBox2D):
            if isinstance(spatial_bounds, SpatialPartition2D):
                spatial_bounds = spatial_bounds.to_bounding_box()
            else:
                spatial_bounds = BoundingBox2D(*spatial_bounds)
        if not isinstance(time_interval, TimeInterval):
            time_interval = TimeInterval(*time_interval)

        self.__spatial_bounds = spatial_bounds
        self.__time_interval = time_interval
        self.__srs = srs

    @property
    def bbox_str(self) -> str:
        """
        A comma-separated string representation of the spatial bounds
        """
        return self.__spatial_bounds.as_bbox_str()

    @property
    def bbox_ogc_str(self) -> str:
        """
        A comma-separated string representation of the spatial bounds with OGC axis ordering
        """
        y_axis_first = self.__srs == "EPSG:4326"
        return self.__spatial_bounds.as_bbox_str(y_axis_first=y_axis_first)

    @property
    def bbox_ogc(self) -> tuple[float, float, float, float]:
        """
        Return the bbox with OGC axis ordering of the srs
        """

        # TODO: properly handle axis order
        y_axis_first = self.__srs == "EPSG:4326"
        return self.__spatial_bounds.as_bbox_tuple(y_axis_first=y_axis_first)

    @property
    def time(self) -> TimeInterval:
        """
        Return the time instance or interval
        """
        return self.__time_interval

    @property
    def spatial_bounds(self) -> BoundingBox2D:
        """
        Return the spatial bounds
        """
        return self.__spatial_bounds

    @property
    def time_str(self) -> str:
        """
        Return the time instance or interval as a string representation
        """
        return self.time.time_str

    @property
    def srs(self) -> str:
        """
        Return the SRS string
        """
        return self.__srs

    def __repr__(self) -> str:
        """Return a string representation of the query rectangle."""
        r = "QueryRectangle( \n"
        r += "    " + repr(self.__spatial_bounds) + "\n"
        r += "    " + repr(self.__time_interval) + "\n"
        r += f"    srs={self.__srs} \n"
        r += ")"
        return r

    def with_raster_bands(self, raster_bands: list[int]) -> RasterQueryRectangle:
        """Converts a `QueryRectangle` into a `RasterQueryRectangle`"""
        return RasterQueryRectangle(self.spatial_bounds, self.time, raster_bands, self.srs)


class RasterQueryRectangle(QueryRectangle):
    """
    A multi-dimensional query rectangle, consisting of spatial and temporal information and raster bands.
    """

    __bands: list[int] = []

    def __init__(
        self,
        spatial_bounds: BoundingBox2D | SpatialPartition2D | tuple[float, float, float, float],
        time_interval: TimeInterval | tuple[datetime, datetime | None],
        raster_bands: list[int] | None | int,
        srs="EPSG:4326",
    ) -> None:
        """
        Initialize a new `QueryRectangle` object

        Parameters
        ----------
        spatial_bounds
            The spatial bounds of the query rectangle.
            Either a `BoundingBox2D` or a tuple of floats (xmin, ymin, xmax, ymax)
        time_interval
            The time interval of the query rectangle.
            Either a `TimeInterval` or a tuple of `datetime.datetime` objects (start, end)
        bands
            The raster bands of the query rectangle.
            A List of ints representing the band numbers.
        """

        super().__init__(spatial_bounds, time_interval, srs)
        if raster_bands is None:
            self.__bands = [0]
        elif isinstance(raster_bands, int):
            self.__bands = [raster_bands]
        else:
            self.__bands = raster_bands

    @property
    def raster_bands(self) -> list[int]:
        """
        Return the query bands
        """
        return self.__bands

    def __repr__(self) -> str:
        """Return a string representation of the query rectangle."""
        r = "RasterQueryRectangle( \n"
        r += "    " + repr(self.spatial_bounds) + "\n"
        r += "    " + repr(self.time) + "\n"
        r += "    " + repr(self.__bands) + "\n"
        r += f"    srs={self.srs} \n"
        r += ")"
        return r


class ResultDescriptor:  # pylint: disable=too-few-public-methods
    """
    Base class for result descriptors
    """

    __spatial_reference: str
    __time_bounds: TimeInterval | None

    def __init__(
        self,
        spatial_reference: str,
        time_bounds: TimeInterval | None = None,
    ) -> None:
        """Initialize a new `ResultDescriptor` object"""

        self.__spatial_reference = spatial_reference
        self.__time_bounds = time_bounds

    @staticmethod
    def from_response(response: geoengine_openapi_client.TypedResultDescriptor) -> ResultDescriptor:
        """
        Parse a result descriptor from an http response
        """

        inner = response.actual_instance

        if isinstance(inner, geoengine_openapi_client.TypedRasterResultDescriptor):
            return RasterResultDescriptor.from_response_raster(inner)
        if isinstance(inner, geoengine_openapi_client.TypedVectorResultDescriptor):
            return VectorResultDescriptor.from_response_vector(inner)
        if isinstance(inner, geoengine_openapi_client.TypedPlotResultDescriptor):
            return PlotResultDescriptor.from_response_plot(inner)

        raise TypeException("Unknown `ResultDescriptor` type")

    @classmethod
    def is_raster_result(cls) -> bool:
        """
        Return true if the result is of type raster
        """
        return False

    @classmethod
    def is_vector_result(cls) -> bool:
        """
        Return true if the result is of type vector
        """
        return False

    @classmethod
    def is_plot_result(cls) -> bool:
        """
        Return true if the result is of type plot
        """

        return False

    @property
    def spatial_reference(self) -> str:
        """Return the spatial reference"""

        return self.__spatial_reference

    @property
    def time_bounds(self) -> TimeInterval | None:
        """Return the time bounds"""

        return self.__time_bounds

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.TypedResultDescriptor:
        pass


class VectorResultDescriptor(ResultDescriptor):
    """
    A vector result descriptor
    """

    __spatial_bounds: BoundingBox2D | None
    __data_type: VectorDataType
    __columns: dict[str, VectorColumnInfo]

    def __init__(  # pylint: disable=too-many-arguments,too-many-positional-arguments
        self,
        spatial_reference: str,
        data_type: VectorDataType,
        columns: dict[str, VectorColumnInfo],
        time_bounds: TimeInterval | None = None,
        spatial_bounds: BoundingBox2D | None = None,
    ) -> None:
        """Initialize a vector result descriptor"""
        super().__init__(spatial_reference, time_bounds)
        self.__data_type = data_type
        self.__columns = columns
        self.__spatial_bounds = spatial_bounds

    @staticmethod
    def from_response_vector(response: geoengine_openapi_client.TypedVectorResultDescriptor) -> VectorResultDescriptor:
        """Parse a vector result descriptor from an http response"""
        sref = response.spatial_reference
        data_type = VectorDataType.from_string(response.data_type)
        columns = {name: VectorColumnInfo.from_response(info) for name, info in response.columns.items()}

        time_bounds = None
        if response.time is not None:
            time_bounds = TimeInterval.from_response(response.time)
        spatial_bounds = None
        if response.bbox is not None:
            spatial_bounds = BoundingBox2D.from_response(response.bbox)

        return VectorResultDescriptor(sref, data_type, columns, time_bounds, spatial_bounds)

    @classmethod
    def is_vector_result(cls) -> bool:
        return True

    @property
    def data_type(self) -> VectorDataType:
        """Return the data type"""
        return self.__data_type

    @property
    def spatial_reference(self) -> str:
        """Return the spatial reference"""
        return super().spatial_reference

    @property
    def columns(self) -> dict[str, VectorColumnInfo]:
        """Return the columns"""

        return self.__columns

    @property
    def spatial_bounds(self) -> BoundingBox2D | None:
        """Return the spatial bounds"""
        return self.__spatial_bounds

    def __repr__(self) -> str:
        """Display representation of the vector result descriptor"""
        r = ""
        r += f"Data type:         {self.data_type.value}\n"
        r += f"Spatial Reference: {self.spatial_reference}\n"

        r += "Columns:\n"
        for column_name in self.columns:
            column_info = self.columns[column_name]
            r += f"  {column_name}:\n"
            r += f"    Column Type: {column_info.data_type.value}\n"
            r += f"    Measurement: {column_info.measurement}\n"

        return r

    def to_api_dict(self) -> geoengine_openapi_client.TypedResultDescriptor:
        """Convert the vector result descriptor to a dictionary"""

        return geoengine_openapi_client.TypedResultDescriptor(
            geoengine_openapi_client.TypedVectorResultDescriptor(
                type="vector",
                data_type=self.data_type.to_api_enum(),
                spatial_reference=self.spatial_reference,
                columns={name: column_info.to_api_dict() for name, column_info in self.columns.items()},
                time=self.time_bounds.to_api_dict() if self.time_bounds is not None else None,
                bbox=self.spatial_bounds.to_api_dict() if self.spatial_bounds is not None else None,
            )
        )


class FeatureDataType(str, Enum):
    """Vector column data type"""

    CATEGORY = "category"
    INT = "int"
    FLOAT = "float"
    TEXT = "text"
    BOOL = "bool"
    DATETIME = "dateTime"

    @staticmethod
    def from_string(data_type: str) -> FeatureDataType:
        """Create a new `VectorColumnDataType` from a string"""

        return FeatureDataType(data_type)

    def to_api_enum(self) -> geoengine_openapi_client.FeatureDataType:
        """Convert to an API enum"""

        return geoengine_openapi_client.FeatureDataType(self.value)


@dataclass
class VectorColumnInfo:
    """Vector column information"""

    data_type: FeatureDataType
    measurement: Measurement

    @staticmethod
    def from_response(response: geoengine_openapi_client.VectorColumnInfo) -> VectorColumnInfo:
        """Create a new `VectorColumnInfo` from a JSON response"""

        return VectorColumnInfo(
            FeatureDataType.from_string(data_type=response.data_type), Measurement.from_response(response.measurement)
        )

    def to_api_dict(self) -> geoengine_openapi_client.VectorColumnInfo:
        """Convert to a dictionary"""

        return geoengine_openapi_client.VectorColumnInfo(
            data_type=self.data_type.to_api_enum(),
            measurement=self.measurement.to_api_dict(),
        )


@dataclass(repr=False)
class RasterBandDescriptor:
    """A raster band descriptor"""

    name: str
    measurement: Measurement

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.RasterBandDescriptor) -> RasterBandDescriptor:
        """Parse an http response to a `RasterBandDescriptor` object"""
        return RasterBandDescriptor(response.name, Measurement.from_response(response.measurement))

    def to_api_dict(self) -> geoengine_openapi_client.RasterBandDescriptor:
        return geoengine_openapi_client.RasterBandDescriptor(
            name=self.name,
            measurement=self.measurement.to_api_dict(),
        )

    def __repr__(self) -> str:
        """Display representation of a raster band descriptor"""
        return f"{self.name}: {self.measurement}"


@dataclass
class GridIdx2D:
    """A grid index"""

    x_idx: int
    y_idx: int

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.GridIdx2D) -> GridIdx2D:
        """Parse an http response to a `GridIdx2D` object"""
        return GridIdx2D(x_idx=response.x_idx, y_idx=response.y_idx)

    def to_api_dict(self) -> geoengine_openapi_client.GridIdx2D:
        return geoengine_openapi_client.GridIdx2D(y_idx=self.y_idx, x_idx=self.x_idx)


@dataclass
class GridBoundingBox2D:
    """A grid boundingbox where lower right is inclusive index"""

    top_left_idx: GridIdx2D
    bottom_right_idx: GridIdx2D

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.GridBoundingBox2D) -> GridBoundingBox2D:
        """Parse an http response to a `GridBoundingBox2D` object"""
        ul_idx = GridIdx2D.from_response(response.top_left_idx)
        lr_idx = GridIdx2D.from_response(response.bottom_right_idx)
        return GridBoundingBox2D(top_left_idx=ul_idx, bottom_right_idx=lr_idx)

    def to_api_dict(self) -> geoengine_openapi_client.GridBoundingBox2D:
        return geoengine_openapi_client.GridBoundingBox2D(
            top_left_idx=self.top_left_idx.to_api_dict(),
            bottom_right_idx=self.bottom_right_idx.to_api_dict(),
        )

    @property
    def width(self) -> int:
        return abs(self.bottom_right_idx.x_idx - self.top_left_idx.x_idx)

    @property
    def height(self) -> int:
        return abs(self.top_left_idx.y_idx - self.bottom_right_idx.y_idx)

    def contains_idx(self, idx: GridIdx2D) -> bool:
        """Test if a `GridIdx2D` is contained by this"""
        contains_x = self.top_left_idx.x_idx <= idx.x_idx <= self.bottom_right_idx.x_idx
        contains_y = self.top_left_idx.y_idx <= idx.y_idx <= self.bottom_right_idx.y_idx
        return contains_x and contains_y


@dataclass
class SpatialGridDefinition:
    """A grid boundingbox where lower right is inclusive index"""

    geo_transform: GeoTransform
    grid_bounds: GridBoundingBox2D

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.SpatialGridDefinition) -> SpatialGridDefinition:
        """Parse an http response to a `SpatialGridDefinition` object"""
        geo_transform = GeoTransform.from_response(response.geo_transform)
        grid_bounds = GridBoundingBox2D.from_response(response.grid_bounds)
        return SpatialGridDefinition(geo_transform=geo_transform, grid_bounds=grid_bounds)

    def to_api_dict(self) -> geoengine_openapi_client.SpatialGridDefinition:
        return geoengine_openapi_client.SpatialGridDefinition(
            geo_transform=self.geo_transform.to_api_dict(),
            grid_bounds=self.grid_bounds.to_api_dict(),
        )

    def contains_idx(self, idx: GridIdx2D) -> bool:
        return self.grid_bounds.contains_idx(idx)

    def spatial_bounds(self) -> SpatialPartition2D:
        return self.geo_transform.grid_bounds_to_spatial_bounds(self.grid_bounds)

    def spatial_resolution(self) -> SpatialResolution:
        return self.geo_transform.spatial_resolution()

    def __repr__(self) -> str:
        """Display representation of the SpatialGridDefinition"""
        r = "SpatialGridDefinition: \n"
        r += f"    GeoTransform: {self.geo_transform}\n"
        r += f"    GridBounds: {self.grid_bounds}\n"
        return r


@dataclass
class SpatialGridDescriptor:
    """A grid boundingbox where lower right is inclusive index"""

    spatial_grid: SpatialGridDefinition
    descriptor: geoengine_openapi_client.SpatialGridDescriptorState

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.SpatialGridDescriptor) -> SpatialGridDescriptor:
        """Parse an http response to a `SpatialGridDefinition` object"""
        spatial_grid = SpatialGridDefinition.from_response(response.spatial_grid)
        return SpatialGridDescriptor(spatial_grid=spatial_grid, descriptor=response.descriptor)

    def to_api_dict(self) -> geoengine_openapi_client.SpatialGridDescriptor:
        return geoengine_openapi_client.SpatialGridDescriptor(
            spatial_grid=self.spatial_grid.to_api_dict(),
            descriptor=self.descriptor,
        )

    def contains_idx(self, idx: GridIdx2D) -> bool:
        return self.spatial_grid.contains_idx(idx)

    def spatial_resolution(self) -> SpatialResolution:
        return self.spatial_grid.spatial_resolution()

    def spatial_bounds(self) -> SpatialPartition2D:
        return self.spatial_grid.spatial_bounds()

    def is_source(self) -> bool:
        return self.descriptor == "source"

    def is_derived(self) -> bool:
        return self.descriptor == "derived"

    def __repr__(self) -> str:
        """Display representation of the SpatialGridDescriptor"""
        r = "SpatialGridDescriptor: \n"
        r += f"    Definition: {self.spatial_grid}\n"
        r += f"    Is a {self.descriptor} grid.\n"
        return r


class RasterDataType(str, Enum):
    """Raster data type enum"""

    U8 = "U8"
    U16 = "U16"
    U32 = "U32"
    U64 = "U64"
    I8 = "I8"
    I16 = "I16"
    I32 = "I32"
    I64 = "I64"
    F32 = "F32"
    F64 = "F64"

    @staticmethod
    def from_string(data_type: str) -> RasterDataType:
        """Create a new `RasterDataType` from a string"""

        if data_type not in RasterDataType._value2member_map_:
            raise ValueError(f"Unknown RasterDataType: {data_type}")

        return RasterDataType(data_type)

    @staticmethod
    def from_literal(
        data_type: Literal["U8", "U16", "U32", "U64", "I8", "I16", "I32", "I64", "F32", "F64"],
    ) -> RasterDataType:
        """Create a new `RasterDataType` from a literal"""

        return RasterDataType(data_type)

    def to_literal(
        self,
    ) -> Literal["U8", "U16", "U32", "U64", "I8", "I16", "I32", "I64", "F32", "F64"]:
        """Convert to a literal"""

        return cast(
            Literal["U8", "U16", "U32", "U64", "I8", "I16", "I32", "I64", "F32", "F64"],
            self.value,
        )

    @staticmethod
    def from_api_enum(data_type: geoengine_openapi_client.RasterDataType) -> RasterDataType:
        """Create a new `RasterDataType` from an API enum"""

        return RasterDataType(data_type.value)

    def to_api_enum(self) -> geoengine_openapi_client.RasterDataType:
        """Convert to an API enum"""

        return geoengine_openapi_client.RasterDataType(self.value)

    def to_np_dtype(self) -> np.dtype:
        """Convert to a numpy dtype"""
        mapping = {
            RasterDataType.U8: np.uint8,
            RasterDataType.U16: np.uint16,
            RasterDataType.U32: np.uint32,
            RasterDataType.U64: np.uint64,
            RasterDataType.I8: np.int8,
            RasterDataType.I16: np.int16,
            RasterDataType.I32: np.int32,
            RasterDataType.I64: np.int64,
            RasterDataType.F32: np.float32,
            RasterDataType.F64: np.float64,
        }
        return np.dtype(mapping[self])


class TimeDimension:
    """A time dimension"""

    @classmethod
    def from_response(
        cls, response: geoengine_openapi_client.TimeDimension
    ) -> RegularTimeDimension | IrregularTimeDimension:
        """Parse a time dimension from an http response"""
        actual = response.actual_instance

        if actual is None:
            raise ValueError("input is None")

        if actual.type == "regular":
            if not isinstance(actual, geoengine_openapi_client.RegularTimeDimension):
                raise ValueError("Type should be regular!")
            return RegularTimeDimension.from_response(response)

        if actual.type == "irregular":
            return IrregularTimeDimension.from_response(response)

        raise ValueError("unknown input type")

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.TimeDimension:
        pass


class RegularTimeDimension(TimeDimension):
    """
    A regular time dimension
    """

    origin: np.datetime64
    step: TimeStep

    def __init__(self, step: TimeStep, origin: np.datetime64 | None = None) -> None:
        """Initialize a new `RegularTimeDimension`"""

        self.origin = origin if origin is not None else np.datetime64("1970-01-01T00:00:00Z")
        self.step = step

    def to_api_dict(self) -> geoengine_openapi_client.TimeDimension:
        """Convert the regular time dimension to a dictionary"""
        time_origin = self.origin.astype("datetime64[ms]").astype(int)
        return geoengine_openapi_client.TimeDimension(
            {"type": "regular", "origin": int(time_origin), "step": self.step.to_api_dict()}
        )

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.TimeDimension) -> RegularTimeDimension:
        """Parse a regular time dimension from an http response"""

        actual = response.actual_instance

        if actual is None or actual.type != "regular":
            raise ValueError("type must be regular")

        if not isinstance(actual, geoengine_openapi_client.RegularTimeDimension):
            raise ValueError("Not a valid RegularTimeDimension")

        origin = np.datetime64(actual.origin, "ms")
        step = TimeStep.from_response(actual.step)
        return RegularTimeDimension(step=step, origin=origin)


class IrregularTimeDimension(TimeDimension):
    """The irregular time dimension"""

    def to_api_dict(self) -> geoengine_openapi_client.TimeDimension:
        """Convert the irregular time dimension to a dictionary"""

        return geoengine_openapi_client.TimeDimension({"type": "irregular"})

    @classmethod
    def from_response(cls, response: Any) -> IrregularTimeDimension:
        """Parse an irregular time dimension from an http response"""
        return IrregularTimeDimension()


class TimeDescriptor:
    """A time descriptor"""

    bounds: TimeInterval | None
    dimension: TimeDimension

    def __init__(self, dimension: TimeDimension, bounds: TimeInterval | None = None) -> None:
        """Initialize a new `TimeDescriptor`"""
        self.dimension = dimension
        self.bounds = bounds

    def to_api_dict(self) -> geoengine_openapi_client.TimeDescriptor:
        """Convert the time descriptor to a dictionary"""
        return geoengine_openapi_client.TimeDescriptor(
            dimension=self.dimension.to_api_dict(),
            bounds=self.bounds.to_api_dict() if self.bounds is not None else None,
        )

    @staticmethod
    def from_response(response: geoengine_openapi_client.TimeDescriptor) -> TimeDescriptor:
        """Parse a time descriptor from an http response"""
        bounds = None
        dimension = None

        if response.bounds is not None:
            bounds = TimeInterval.from_response(response.bounds)

        dimension = TimeDimension.from_response(response.dimension)

        return TimeDescriptor(bounds=bounds, dimension=dimension)


class RasterResultDescriptor(ResultDescriptor):
    """
    A raster result descriptor
    """

    __data_type: RasterDataType
    __bands: list[RasterBandDescriptor]
    __spatial_grid: SpatialGridDescriptor
    __time: TimeDescriptor

    def __init__(  # pylint: disable=too-many-arguments,too-many-positional-arguments
        self,
        data_type: RasterDataType | Literal["U8", "U16", "U32", "U64", "I8", "I16", "I32", "I64", "F32", "F64"],
        bands: list[RasterBandDescriptor],
        spatial_reference: str,
        spatial_grid: SpatialGridDescriptor,
        time: TimeDescriptor | TimeInterval | TimeDimension | None = None,
    ) -> None:
        """Initialize a new `RasterResultDescriptor`"""

        time_descriptor = time
        if isinstance(time, TimeInterval):
            time_descriptor = TimeDescriptor(IrregularTimeDimension(), time)
        elif isinstance(time, TimeDimension):
            time_descriptor = TimeDescriptor(time, None)
        elif time is None:
            time_descriptor = TimeDescriptor(IrregularTimeDimension(), None)

        if not isinstance(time_descriptor, TimeDescriptor):
            raise ValueError(f"no valid TimeDimension, got {type(time)}")

        if isinstance(data_type, str):
            data_type = RasterDataType.from_string(data_type)
        elif not isinstance(data_type, RasterDataType):
            raise ValueError(f"no valid RasterDataType, got {type(data_type)}")

        super().__init__(spatial_reference, time_descriptor.bounds)
        self.__data_type = data_type
        self.__bands = bands
        self.__spatial_grid = spatial_grid
        self.__time = time_descriptor

    def to_api_dict(self) -> geoengine_openapi_client.TypedResultDescriptor:
        """Convert the raster result descriptor to a dictionary"""

        return geoengine_openapi_client.TypedResultDescriptor(
            geoengine_openapi_client.TypedRasterResultDescriptor(
                type="raster",
                data_type=self.data_type.to_api_enum(),
                bands=[band.to_api_dict() for band in self.__bands],
                spatial_reference=self.spatial_reference,
                time=self.__time.to_api_dict(),
                spatial_grid=self.__spatial_grid.to_api_dict(),
            )
        )

    @staticmethod
    def from_response_raster(response: geoengine_openapi_client.TypedRasterResultDescriptor) -> RasterResultDescriptor:
        """Parse a raster result descriptor from an http response"""
        spatial_ref = response.spatial_reference
        data_type = RasterDataType.from_api_enum(response.data_type)
        bands = [RasterBandDescriptor.from_response(band) for band in response.bands]

        spatial_grid = SpatialGridDescriptor.from_response(response.spatial_grid)

        time_bounds = TimeDescriptor.from_response(response.time)

        return RasterResultDescriptor(
            data_type=data_type,
            bands=bands,
            spatial_reference=spatial_ref,
            time=time_bounds,
            spatial_grid=spatial_grid,
        )

    @classmethod
    def is_raster_result(cls) -> bool:
        return True

    @property
    def data_type(self) -> RasterDataType:
        return self.__data_type

    @property
    def bands(self) -> list[RasterBandDescriptor]:
        return self.__bands

    @property
    def spatial_grid(self) -> SpatialGridDescriptor:
        return self.__spatial_grid

    @property
    def spatial_bounds(self) -> SpatialPartition2D:
        return self.spatial_grid.spatial_bounds()

    @property
    def geo_transform(self) -> GeoTransform:
        return self.spatial_grid.spatial_grid.geo_transform

    @property
    def spatial_reference(self) -> str:
        """Return the spatial reference"""

        return super().spatial_reference

    def __repr__(self) -> str:
        """Display representation of the raster result descriptor"""
        r = ""
        r += f"Data type:         {self.data_type}\n"
        r += f"Spatial Reference: {self.spatial_reference}\n"
        r += f"Spatial Grid: {self.spatial_grid} \n"
        r += f"Time Bounds: {self.time_bounds}\n"
        r += "Bands:\n"

        for band in self.__bands:
            r += f"    {band}\n"

        return r


class PlotResultDescriptor(ResultDescriptor):
    """
    A plot result descriptor
    """

    __spatial_bounds: BoundingBox2D | None

    def __init__(  # pylint: disable=too-many-arguments]
        self,
        spatial_reference: str,
        time_bounds: TimeInterval | None = None,
        spatial_bounds: BoundingBox2D | None = None,
    ) -> None:
        """Initialize a new `PlotResultDescriptor`"""
        super().__init__(spatial_reference, time_bounds)
        self.__spatial_bounds = spatial_bounds

    def __repr__(self) -> str:
        """Display representation of the plot result descriptor"""
        r = "Plot Result"

        return r

    @staticmethod
    def from_response_plot(response: geoengine_openapi_client.TypedPlotResultDescriptor) -> PlotResultDescriptor:
        """Create a new `PlotResultDescriptor` from a JSON response"""
        spatial_ref = response.spatial_reference

        time_bounds = None
        if response.time is not None:
            time_bounds = TimeInterval.from_response(response.time)
        spatial_bounds = None
        if response.bbox is not None:
            spatial_bounds = BoundingBox2D.from_response(response.bbox)

        return PlotResultDescriptor(
            spatial_reference=spatial_ref, time_bounds=time_bounds, spatial_bounds=spatial_bounds
        )

    @classmethod
    def is_plot_result(cls) -> bool:
        return True

    @property
    def spatial_reference(self) -> str:
        """Return the spatial reference"""
        return super().spatial_reference

    @property
    def spatial_bounds(self) -> BoundingBox2D | None:
        return self.__spatial_bounds

    def to_api_dict(self) -> geoengine_openapi_client.TypedResultDescriptor:
        """Convert the plot result descriptor to a dictionary"""

        return geoengine_openapi_client.TypedResultDescriptor(
            geoengine_openapi_client.TypedPlotResultDescriptor(
                type="plot",
                spatial_reference=self.spatial_reference,
                time=self.time_bounds.to_api_dict() if self.time_bounds is not None else None,
                bbox=self.spatial_bounds.to_api_dict() if self.spatial_bounds is not None else None,
            )
        )


class VectorDataType(str, Enum):
    """An enum of vector data types"""

    DATA = "Data"
    MULTI_POINT = "MultiPoint"
    MULTI_LINE_STRING = "MultiLineString"
    MULTI_POLYGON = "MultiPolygon"

    @classmethod
    def from_geopandas_type_name(cls, name: str) -> VectorDataType:
        """Resolve vector data type from geopandas geometry type"""

        name_map = {
            "Point": VectorDataType.MULTI_POINT,
            "MultiPoint": VectorDataType.MULTI_POINT,
            "Line": VectorDataType.MULTI_LINE_STRING,
            "MultiLine": VectorDataType.MULTI_LINE_STRING,
            "Polygon": VectorDataType.MULTI_POLYGON,
            "MultiPolygon": VectorDataType.MULTI_POLYGON,
        }

        if name in name_map:
            return name_map[name]

        raise InputException("Invalid vector data type")

    def to_api_enum(self) -> geoengine_openapi_client.VectorDataType:
        return geoengine_openapi_client.VectorDataType(self.value)

    @staticmethod
    def from_literal(literal: Literal["Data", "MultiPoint", "MultiLineString", "MultiPolygon"]) -> VectorDataType:
        """Resolve vector data type from literal"""
        return VectorDataType(literal)

    @staticmethod
    def from_api_enum(data_type: geoengine_openapi_client.VectorDataType) -> VectorDataType:
        """Resolve vector data type from API enum"""
        return VectorDataType(data_type.value)

    @staticmethod
    def from_string(string: str) -> VectorDataType:
        """Resolve vector data type from string"""
        if string not in VectorDataType.__members__.values():
            raise InputException("Invalid vector data type: " + string)
        return VectorDataType(string)


class TimeStepGranularity(Enum):
    """An enum of time step granularities"""

    MILLIS = "millis"
    SECONDS = "seconds"
    MINUTES = "minutes"
    HOURS = "hours"
    DAYS = "days"
    MONTHS = "months"
    YEARS = "years"

    def to_api_enum(self) -> geoengine_openapi_client.TimeGranularity:
        return geoengine_openapi_client.TimeGranularity(self.value)


@dataclass
class TimeStep:
    """A time step that consists of a granularity and a step size"""

    step: int
    granularity: TimeStepGranularity

    def __init__(self, step: int, granularity: TimeStepGranularity | str) -> None:
        """Initialize a new `TimeStep` object"""
        self.step = step
        if isinstance(granularity, str):
            self.granularity = TimeStepGranularity(granularity)
        elif isinstance(granularity, TimeStepGranularity):
            self.granularity = granularity
        else:
            raise InputException("Invalid granularity type. Got: " + str(type(granularity)))

    def to_api_dict(self) -> geoengine_openapi_client.TimeStep:
        return geoengine_openapi_client.TimeStep(
            step=self.step,
            granularity=self.granularity.to_api_enum(),
        )

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.TimeStep) -> TimeStep:
        """Parse an http response to a `TimeStep` object"""
        granularity = TimeStepGranularity(response.granularity.value)
        return TimeStep(step=response.step, granularity=granularity)


@dataclass
class Provenance:
    """Provenance information as triplet of citation, license and uri"""

    citation: str
    license: str
    uri: str

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.Provenance) -> Provenance:
        """Parse an http response to a `Provenance` object"""
        return Provenance(response.citation, response.license, response.uri)

    def to_api_dict(self) -> geoengine_openapi_client.Provenance:
        return geoengine_openapi_client.Provenance(
            citation=self.citation,
            license=self.license,
            uri=self.uri,
        )


@dataclass
class ProvenanceEntry:
    """Provenance of a dataset"""

    data: list[DataId]
    provenance: Provenance

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.ProvenanceEntry) -> ProvenanceEntry:
        """Parse an http response to a `ProvenanceEntry` object"""

        dataset = [DataId.from_response(data) for data in response.data]
        provenance = Provenance.from_response(response.provenance)

        return ProvenanceEntry(dataset, provenance)


class Symbology:
    """Base class for symbology"""

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.Symbology:
        pass

    @staticmethod
    def from_response(response: geoengine_openapi_client.Symbology) -> Symbology:
        """Parse an http response to a `Symbology` object"""
        inner = response.actual_instance

        if isinstance(
            inner,
            geoengine_openapi_client.PointSymbology
            | geoengine_openapi_client.LineSymbology
            | geoengine_openapi_client.PolygonSymbology,
        ):
            # return VectorSymbology.from_response_vector(response)
            return VectorSymbology()  # TODO: implement
        if isinstance(inner, geoengine_openapi_client.RasterSymbology):
            return RasterSymbology.from_response_raster(inner)

        raise InputException("Invalid symbology type")

    def __repr__(self) -> str:
        return "Symbology"


class VectorSymbology(Symbology):
    """A vector symbology"""

    # TODO: implement

    def to_api_dict(self) -> geoengine_openapi_client.Symbology:
        return None  # type: ignore


class RasterColorizer:
    """Base class for raster colorizer"""

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.RasterColorizer) -> RasterColorizer:
        """Parse an http response to a `RasterColorizer` object"""
        inner = response.actual_instance

        if isinstance(inner, geoengine_openapi_client.SingleBandRasterColorizer):
            return SingleBandRasterColorizer.from_single_band_response(inner)
        if isinstance(inner, geoengine_openapi_client.MultiBandRasterColorizer):
            return MultiBandRasterColorizer.from_multi_band_response(inner)

        raise GeoEngineException({"message": "Unknown RasterColorizer type"})

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.RasterColorizer:
        pass


@dataclass
class SingleBandRasterColorizer(RasterColorizer):
    """A raster colorizer for a specified band"""

    band: int
    band_colorizer: Colorizer

    @staticmethod
    def from_single_band_response(response: geoengine_openapi_client.SingleBandRasterColorizer) -> RasterColorizer:
        return SingleBandRasterColorizer(response.band, Colorizer.from_response(response.band_colorizer))

    def to_api_dict(self) -> geoengine_openapi_client.RasterColorizer:
        return geoengine_openapi_client.RasterColorizer(
            geoengine_openapi_client.SingleBandRasterColorizer(
                type="singleBand",
                band=self.band,
                band_colorizer=self.band_colorizer.to_api_dict(),
            )
        )


@dataclass
class MultiBandRasterColorizer(RasterColorizer):
    """A raster colorizer for multiple bands"""

    blue_band: int
    blue_max: float
    blue_min: float
    blue_scale: float | None
    green_band: int
    green_max: float
    green_min: float
    green_scale: float | None
    red_band: int
    red_max: float
    red_min: float
    red_scale: float | None

    @staticmethod
    def from_multi_band_response(response: geoengine_openapi_client.MultiBandRasterColorizer) -> RasterColorizer:
        return MultiBandRasterColorizer(
            blue_band=response.blue_band,
            blue_max=response.blue_max,
            blue_min=response.blue_min,
            blue_scale=response.blue_scale if response.blue_scale is not None else None,
            green_band=response.green_band,
            green_max=response.green_max,
            green_min=response.green_min,
            green_scale=response.green_scale if response.green_scale is not None else None,
            red_band=response.red_band,
            red_max=response.red_max,
            red_min=response.red_min,
            red_scale=response.red_scale if response.red_scale is not None else None,
        )

    def to_api_dict(self) -> geoengine_openapi_client.RasterColorizer:
        return geoengine_openapi_client.RasterColorizer(
            geoengine_openapi_client.MultiBandRasterColorizer(
                type="multiBand",
                blue_band=self.blue_band,
                blue_max=self.blue_max,
                blue_min=self.blue_min,
                blue_scale=self.blue_scale,
                green_band=self.green_band,
                green_max=self.green_max,
                green_min=self.green_min,
                green_scale=self.green_scale,
                red_band=self.red_band,
                red_max=self.red_max,
                red_min=self.red_min,
                red_scale=self.red_scale,
            )
        )


class RasterSymbology(Symbology):
    """A raster symbology"""

    opacity: float
    raster_colorizer: RasterColorizer

    def __init__(self, raster_colorizer: RasterColorizer, opacity: float = 1.0) -> None:
        """Initialize a new `RasterSymbology`"""

        self.raster_colorizer = raster_colorizer
        self.opacity = opacity

    def to_api_dict(self) -> geoengine_openapi_client.Symbology:
        """Convert the raster symbology to a dictionary"""

        return geoengine_openapi_client.Symbology(
            geoengine_openapi_client.RasterSymbology(
                type="raster",
                raster_colorizer=self.raster_colorizer.to_api_dict(),
                opacity=self.opacity,
            )
        )

    @staticmethod
    def from_response_raster(response: geoengine_openapi_client.RasterSymbology) -> RasterSymbology:
        """Parse an http response to a `RasterSymbology` object"""

        raster_colorizer = RasterColorizer.from_response(response.raster_colorizer)

        return RasterSymbology(raster_colorizer, response.opacity)

    def __repr__(self) -> str:
        return str(self.__class__) + f"({self.raster_colorizer}, {self.opacity})"

    def __eq__(self, value):
        """Check if two RasterSymbologies are equal"""

        if not isinstance(value, self.__class__):
            return False
        return self.opacity == value.opacity and self.raster_colorizer == value.raster_colorizer


class DataId:  # pylint: disable=too-few-public-methods
    """Base class for data ids"""

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.DataId) -> DataId:
        """Parse an http response to a `DataId` object"""
        inner = response.actual_instance

        if isinstance(inner, geoengine_openapi_client.InternalDataId):
            return InternalDataId.from_response_internal(inner)
        if isinstance(inner, geoengine_openapi_client.ExternalDataId):
            return ExternalDataId.from_response_external(inner)

        raise GeoEngineException({"message": "Unknown DataId type"})

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.DataId:
        pass


class InternalDataId(DataId):
    """An internal data id"""

    __dataset_id: UUID

    def __init__(self, dataset_id: UUID):
        self.__dataset_id = dataset_id

    @classmethod
    def from_response_internal(cls, response: geoengine_openapi_client.InternalDataId) -> InternalDataId:
        """Parse an http response to a `InternalDataId` object"""
        return InternalDataId(response.dataset_id)

    def to_api_dict(self) -> geoengine_openapi_client.DataId:
        return geoengine_openapi_client.DataId(
            geoengine_openapi_client.InternalDataId(type="internal", dataset_id=str(self.__dataset_id))
        )

    def __str__(self) -> str:
        return str(self.__dataset_id)

    def __repr__(self) -> str:
        """Display representation of an internal data id"""
        return str(self)

    def __eq__(self, other) -> bool:
        """Check if two internal data ids are equal"""
        if not isinstance(other, self.__class__):
            return False

        return self.__dataset_id == other.__dataset_id  # pylint: disable=protected-access


class ExternalDataId(DataId):
    """An external data id"""

    __provider_id: UUID
    __layer_id: str

    def __init__(self, provider_id: UUID, layer_id: str):
        self.__provider_id = provider_id
        self.__layer_id = layer_id

    @classmethod
    def from_response_external(cls, response: geoengine_openapi_client.ExternalDataId) -> ExternalDataId:
        """Parse an http response to a `ExternalDataId` object"""

        return ExternalDataId(response.provider_id, response.layer_id)

    def to_api_dict(self) -> geoengine_openapi_client.DataId:
        return geoengine_openapi_client.DataId(
            geoengine_openapi_client.ExternalDataId(
                type="external",
                provider_id=str(self.__provider_id),
                layer_id=self.__layer_id,
            )
        )

    def __str__(self) -> str:
        return f"{self.__provider_id}:{self.__layer_id}"

    def __repr__(self) -> str:
        """Display representation of an external data id"""
        return str(self)

    def __eq__(self, other) -> bool:
        """Check if two external data ids are equal"""
        if not isinstance(other, self.__class__):
            return False

        return self.__provider_id == other.__provider_id and self.__layer_id == other.__layer_id  # pylint: disable=protected-access


class Measurement:  # pylint: disable=too-few-public-methods
    """
    Base class for measurements
    """

    @staticmethod
    def from_response(response: geoengine_openapi_client.Measurement) -> Measurement:
        """
        Parse a result descriptor from an http response
        """
        inner = response.actual_instance

        if isinstance(inner, geoengine_openapi_client.UnitlessMeasurement):
            return UnitlessMeasurement()
        if isinstance(inner, geoengine_openapi_client.ContinuousMeasurement):
            return ContinuousMeasurement.from_response_continuous(inner)
        if isinstance(inner, geoengine_openapi_client.ClassificationMeasurement):
            return ClassificationMeasurement.from_response_classification(inner)

        raise TypeException("Unknown `Measurement` type")

    @abstractmethod
    def to_api_dict(self) -> geoengine_openapi_client.Measurement:
        pass


class UnitlessMeasurement(Measurement):
    """A measurement that is unitless"""

    def __str__(self) -> str:
        """String representation of a unitless measurement"""
        return "unitless"

    def __repr__(self) -> str:
        """Display representation of a unitless measurement"""
        return str(self)

    def to_api_dict(self) -> geoengine_openapi_client.Measurement:
        return geoengine_openapi_client.Measurement(geoengine_openapi_client.UnitlessMeasurement(type="unitless"))


class ContinuousMeasurement(Measurement):
    """A measurement that is continuous"""

    __measurement: str
    __unit: str | None

    def __init__(self, measurement: str, unit: str | None) -> None:
        """Initialize a new `ContiuousMeasurement`"""

        super().__init__()

        self.__measurement = measurement
        self.__unit = unit

    @staticmethod
    def from_response_continuous(response: geoengine_openapi_client.ContinuousMeasurement) -> ContinuousMeasurement:
        """Initialize a new `ContiuousMeasurement from a JSON response"""

        return ContinuousMeasurement(response.measurement, response.unit)

    def __str__(self) -> str:
        """String representation of a continuous measurement"""

        if self.__unit is None:
            return self.__measurement

        return f"{self.__measurement} ({self.__unit})"

    def __repr__(self) -> str:
        """Display representation of a continuous measurement"""
        return str(self)

    def to_api_dict(self) -> geoengine_openapi_client.Measurement:
        return geoengine_openapi_client.Measurement(
            geoengine_openapi_client.ContinuousMeasurement(
                type="continuous", measurement=self.__measurement, unit=self.__unit
            )
        )

    @property
    def measurement(self) -> str:
        return self.__measurement

    @property
    def unit(self) -> str | None:
        return self.__unit


class ClassificationMeasurement(Measurement):
    """A measurement that is a classification"""

    __measurement: str
    __classes: dict[int, str]

    def __init__(self, measurement: str, classes: dict[int, str]) -> None:
        """Initialize a new `ClassificationMeasurement`"""

        super().__init__()

        self.__measurement = measurement
        self.__classes = classes

    @staticmethod
    def from_response_classification(
        response: geoengine_openapi_client.ClassificationMeasurement,
    ) -> ClassificationMeasurement:
        """Initialize a new `ClassificationMeasurement from a JSON response"""

        measurement = response.measurement

        str_classes: dict[str, str] = response.classes
        classes = {int(k): v for k, v in str_classes.items()}

        return ClassificationMeasurement(measurement, classes)

    def to_api_dict(self) -> geoengine_openapi_client.Measurement:
        str_classes: dict[str, str] = {str(k): v for k, v in self.__classes.items()}

        return geoengine_openapi_client.Measurement(
            geoengine_openapi_client.ClassificationMeasurement(
                type="classification", measurement=self.__measurement, classes=str_classes
            )
        )

    def __str__(self) -> str:
        """String representation of a classification measurement"""
        classes_str = ", ".join(f"{k}: {v}" for k, v in self.__classes.items())
        return f"{self.__measurement} ({classes_str})"

    def __repr__(self) -> str:
        """Display representation of a classification measurement"""
        return str(self)

    @property
    def measurement(self) -> str:
        return self.__measurement

    @property
    def classes(self) -> dict[int, str]:
        return self.__classes


class GeoTransform:
    """The `GeoTransform` specifies the relationship between pixel coordinates and geographic coordinates."""

    x_min: float
    y_max: float
    """In Geo Engine, x_pixel_size is always positive."""
    x_pixel_size: float
    """In Geo Engine, y_pixel_size is always negative."""
    y_pixel_size: float

    def __init__(self, x_min: float, y_max: float, x_pixel_size: float, y_pixel_size: float):
        """Initialize a new `GeoTransform`"""

        assert x_pixel_size > 0, "In Geo Engine, x_pixel_size is always positive."
        assert y_pixel_size < 0, "In Geo Engine, y_pixel_size is always negative."

        self.x_min = x_min
        self.y_max = y_max
        self.x_pixel_size = x_pixel_size
        self.y_pixel_size = y_pixel_size

    @classmethod
    def from_response_gdal_geo_transform(
        cls, response: geoengine_openapi_client.GdalDatasetGeoTransform
    ) -> GeoTransform:
        """Parse a geotransform from an HTTP JSON response"""
        return GeoTransform(
            x_min=response.origin_coordinate.x,
            y_max=response.origin_coordinate.y,
            x_pixel_size=response.x_pixel_size,
            y_pixel_size=response.y_pixel_size,
        )

    @classmethod
    def from_response(cls, response: geoengine_openapi_client.GeoTransform) -> GeoTransform:
        """Parse a geotransform from an HTTP JSON response"""

        return GeoTransform(
            x_min=response.origin_coordinate.x,
            y_max=response.origin_coordinate.y,
            x_pixel_size=response.x_pixel_size,
            y_pixel_size=response.y_pixel_size,
        )

    def to_api_dict(self) -> geoengine_openapi_client.GeoTransform:
        return geoengine_openapi_client.GeoTransform(
            origin_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.x_min,
                y=self.y_max,
            ),
            x_pixel_size=self.x_pixel_size,
            y_pixel_size=self.y_pixel_size,
        )

    def to_api_dict_gdal_geo_transform(self) -> geoengine_openapi_client.GdalDatasetGeoTransform:
        return geoengine_openapi_client.GdalDatasetGeoTransform(
            origin_coordinate=geoengine_openapi_client.Coordinate2D(
                x=self.x_min,
                y=self.y_max,
            ),
            x_pixel_size=self.x_pixel_size,
            y_pixel_size=self.y_pixel_size,
        )

    def to_gdal(self) -> tuple[float, float, float, float, float, float]:
        """Convert to a GDAL geotransform"""
        return (self.x_min, self.x_pixel_size, 0, self.y_max, 0, self.y_pixel_size)

    def __str__(self) -> str:
        return (
            f"Origin: ({self.x_min}, {self.y_max}), "
            f"X Pixel Size: {self.x_pixel_size}, "
            f"Y Pixel Size: {self.y_pixel_size}"
        )

    def __repr__(self) -> str:
        return str(self)

    @property
    def x_half_pixel_size(self) -> float:
        return self.x_pixel_size / 2.0

    @property
    def y_half_pixel_size(self) -> float:
        return self.y_pixel_size / 2.0

    def pixel_x_to_coord_x(self, pixel: int) -> float:
        return self.x_min + pixel * self.x_pixel_size

    def pixel_y_to_coord_y(self, pixel: int) -> float:
        return self.y_max + pixel * self.y_pixel_size

    def coord_to_pixel_ul(self, x_cord: float, y_coord: float) -> GridIdx2D:
        """Convert a coordinate to a pixel index rould towards top left"""
        return GridIdx2D(
            x_idx=int(np.floor((x_cord - self.x_min) / self.x_pixel_size)),
            y_idx=int(np.ceil((y_coord - self.y_max) / self.y_pixel_size)),
        )

    def coord_to_pixel_lr(self, x_cord: float, y_coord: float) -> GridIdx2D:
        """Convert a coordinate to a pixel index ound towards lower right"""
        return GridIdx2D(
            x_idx=int(np.ceil((x_cord - self.x_min) / self.x_pixel_size)),
            y_idx=int(np.floor((y_coord - self.y_max) / self.y_pixel_size)),
        )

    def pixel_ul_to_coord(self, x_pixel: int, y_pixel: int) -> tuple[float, float]:
        """Convert a pixel position into a coordinate"""
        x = self.pixel_x_to_coord_x(x_pixel)
        y = self.pixel_y_to_coord_y(y_pixel)
        return (x, y)

    def pixel_lr_to_coord(self, x_pixel: int, y_pixel: int) -> tuple[float, float]:
        (x, y) = self.pixel_ul_to_coord(x_pixel, y_pixel)
        return (x + self.x_pixel_size, y + self.y_pixel_size)

    def pixel_center_to_coord(self, x_pixel, y_pixel) -> tuple[float, float]:
        (x, y) = self.pixel_ul_to_coord(x_pixel, y_pixel)
        return (x + self.x_half_pixel_size, y + self.y_half_pixel_size)

    def spatial_resolution(self) -> SpatialResolution:
        return SpatialResolution(x_resolution=abs(self.x_pixel_size), y_resolution=abs(self.y_pixel_size))

    def spatial_to_grid_bounds(self, bounds: SpatialPartition2D | BoundingBox2D) -> GridBoundingBox2D:
        """Converts a BoundingBox2D or a SpatialPartition2D into a GridBoundingBox2D"""
        ul = self.coord_to_pixel_ul(bounds.xmin, bounds.ymax)
        rl = self.coord_to_pixel_lr(bounds.xmax, bounds.ymin)
        return GridBoundingBox2D(top_left_idx=ul, bottom_right_idx=rl)

    def grid_bounds_to_spatial_bounds(self, bounds: GridBoundingBox2D) -> SpatialPartition2D:
        """Converts a GridBoundingBox2D into a SpatialPartition2D"""
        xmin, ymax = self.pixel_ul_to_coord(bounds.top_left_idx.x_idx, bounds.top_left_idx.y_idx)
        xmax, ymin = self.pixel_lr_to_coord(bounds.bottom_right_idx.x_idx, bounds.bottom_right_idx.y_idx)
        return SpatialPartition2D(xmin, ymin, xmax, ymax)

    def __eq__(self, other) -> bool:
        """Check if two geotransforms are equal"""
        if not isinstance(other, GeoTransform):
            return False

        return (
            self.x_min == other.x_min
            and self.y_max == other.y_max
            and self.x_pixel_size == other.x_pixel_size
            and self.y_pixel_size == other.y_pixel_size
        )

geo-engine / geoengine-python / 21037270339

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous