1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
"""This module defines the common interface for convention implementations
and provides helpers for selecting conventions.
"""
from typing import Optional, Protocol, Union
import rasterio.crs
import xarray
from affine import Affine
class ConventionProtocol(Protocol):
"""Protocol defining the interface for convention modules."""
@classmethod
def read_crs(
cls, obj: Union[xarray.Dataset, xarray.DataArray]
) -> Optional[rasterio.crs.CRS]:
"""Read CRS from the object using this convention."""
@classmethod
def read_transform(
cls, obj: Union[xarray.Dataset, xarray.DataArray]
) -> Optional[Affine]:
"""Read transform from the object using this convention."""
@classmethod
def read_spatial_dimensions(
cls,
obj: Union[xarray.Dataset, xarray.DataArray],
) -> Optional[tuple[str, str]]:
"""Read spatial dimensions (y_dim, x_dim) from the object using this convention."""
@classmethod
def write_crs(
cls,
obj: Union[xarray.Dataset, xarray.DataArray],
*,
crs: rasterio.crs.CRS,
**kwargs,
) -> Union[xarray.Dataset, xarray.DataArray]:
"""Write CRS to the object using this convention."""
@classmethod
def write_transform(
cls,
obj: Union[xarray.Dataset, xarray.DataArray],
*,
transform: Affine,
**kwargs,
) -> Union[xarray.Dataset, xarray.DataArray]:
"""Write transform to the object using this convention."""
|
