Skip to content

Geometry

Utility geometry operations functions.

Functions

srai.geometry.remove_interiors 

remove_interiors(geometry: Union[Polygon, MultiPolygon]) -> Polygon

Close polygon holes by limitation to the exterior ring.

PARAMETER DESCRIPTION
geometry

Polygon to close.

TYPE: Union[Polygon, MultiPolygon]

RETURNS DESCRIPTION
Polygon

Closed polygon.

TYPE: Polygon

Source code in srai/geometry.py 
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
def remove_interiors(geometry: Union[Polygon, MultiPolygon]) -> Polygon:
    """
    Close polygon holes by limitation to the exterior ring.

    Args:
        geometry (Union[Polygon, MultiPolygon])): Polygon to close.

    Returns:
        Polygon: Closed polygon.
    """
    if isinstance(geometry, MultiPolygon):
        return unary_union([remove_interiors(sub_polygon) for sub_polygon in geometry.geoms])
    if geometry.interiors:
        return Polygon(list(geometry.exterior.coords))
    return geometry

srai.geometry.flatten_geometry_series 

flatten_geometry_series(geometry_series: gpd.GeoSeries) -> list[BaseGeometry]

Flatten all geometries from a series into a list of BaseGeometries.

Source code in srai/geometry.py 
35
36
37
38
def flatten_geometry_series(geometry_series: gpd.GeoSeries) -> list[BaseGeometry]:
    """Flatten all geometries from a series into a list of BaseGeometries."""
    geometries: list[BaseGeometry] = seq(geometry_series).flat_map(flatten_geometry).to_list()
    return geometries

srai.geometry.merge_disjointed_gdf_geometries 

merge_disjointed_gdf_geometries(gdf: gpd.GeoDataFrame) -> MultiPolygon

Merges geometries from a GeoDataFrame into a single MultiPolygon.

Input geometries are expected to be disjointed.

PARAMETER DESCRIPTION
gdf

GeoDataFrame with geometries to merge.

TYPE: GeoDataFrame

RETURNS DESCRIPTION
MultiPolygon

Merged polygon

TYPE: MultiPolygon

Source code in srai/geometry.py 
118
119
120
121
122
123
124
125
126
127
128
129
130
def merge_disjointed_gdf_geometries(gdf: gpd.GeoDataFrame) -> MultiPolygon:
    """
    Merges geometries from a GeoDataFrame into a single MultiPolygon.

    Input geometries are expected to be disjointed.

    Args:
        gdf (gpd.GeoDataFrame): GeoDataFrame with geometries to merge.

    Returns:
        MultiPolygon: Merged polygon
    """
    return merge_disjointed_polygons(list(gdf.geometry))

srai.geometry.buffer_geometry 

buffer_geometry(geometry: BaseGeometry, meters: float) -> BaseGeometry

Buffer geometry by a given radius in meters.

Projects geometry into azimuthal projection before applying buffer and then changes values back to WGS84 coordinates.

Doesn't work with polygons covering the whole earth (from -180 to 180 longitude).

PARAMETER DESCRIPTION
geometry

Geometry to buffer.

TYPE: BaseGeometry

meters

Radius distance in meters.

TYPE: float

RETURNS DESCRIPTION
BaseGeometry

Buffered geometry.

TYPE: BaseGeometry

Source code in srai/geometry.py 
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
def buffer_geometry(geometry: BaseGeometry, meters: float) -> BaseGeometry:
    """
    Buffer geometry by a given radius in meters.

    Projects geometry into azimuthal projection before applying buffer and then changes values
    back to WGS84 coordinates.

    Doesn't work with polygons covering the whole earth (from -180 to 180 longitude).

    Args:
        geometry (BaseGeometry): Geometry to buffer.
        meters (float): Radius distance in meters.

    Returns:
        BaseGeometry: Buffered geometry.
    """
    _lon, _lat = geometry.centroid.coords[0]

    aeqd_proj = pyproj.Proj(proj="aeqd", ellps="WGS84", datum="WGS84", lat_0=_lat, lon_0=_lon)
    wgs84_proj = pyproj.Proj(proj="latlong", ellps="WGS84")

    wgs84_to_aeqd = pyproj.Transformer.from_proj(wgs84_proj, aeqd_proj, always_xy=True).transform
    aeqd_to_wgs84 = pyproj.Transformer.from_proj(aeqd_proj, wgs84_proj, always_xy=True).transform

    projected_geometry = shapely_transform(wgs84_to_aeqd, geometry)
    bufferred_projected_geometry = projected_geometry.buffer(meters)

    return shapely_transform(aeqd_to_wgs84, bufferred_projected_geometry)

srai.geometry.flatten_geometry 

flatten_geometry(geometry: BaseGeometry) -> list[BaseGeometry]

Flatten all geometries into a list of BaseGeometries.

Source code in srai/geometry.py 
41
42
43
44
45
46
def flatten_geometry(geometry: BaseGeometry) -> list[BaseGeometry]:
    """Flatten all geometries into a list of BaseGeometries."""
    if isinstance(geometry, BaseMultipartGeometry):
        geometries: list[BaseGeometry] = seq(geometry.geoms).flat_map(flatten_geometry).to_list()
        return geometries
    return [geometry]

srai.geometry.merge_disjointed_polygons 

merge_disjointed_polygons(
    polygons: list[Union[Polygon, MultiPolygon]],
) -> MultiPolygon

Merges all polygons into a single MultiPolygon.

Input polygons are expected to be disjointed.

PARAMETER DESCRIPTION
polygons

List of polygons to merge

TYPE: List[Union[Polygon, MultiPolygon]]

RETURNS DESCRIPTION
MultiPolygon

Merged polygon

TYPE: MultiPolygon

Source code in srai/geometry.py 
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
def merge_disjointed_polygons(polygons: list[Union[Polygon, MultiPolygon]]) -> MultiPolygon:
    """
    Merges all polygons into a single MultiPolygon.

    Input polygons are expected to be disjointed.

    Args:
        polygons (List[Union[Polygon, MultiPolygon]]): List of polygons to merge

    Returns:
        MultiPolygon: Merged polygon
    """
    single_polygons = []
    for geom in polygons:
        if isinstance(geom, Polygon):
            single_polygons.append(geom)
        else:
            single_polygons.extend(geom.geoms)
    return MultiPolygon(single_polygons)