vegetation_water_change.py

from geopandas import GeoDataFrame, GeoSeries

from city_metrix.layers import VegetationWaterMap

# TODO: layer generation and zonal stats use different spatial resolutions

def vegetation_water_change_gain_area(zones: GeoDataFrame, spatial_resolution=10) -> GeoSeries:
    gain_counts = VegetationWaterMap(greenwater_layer='gaingreenwaterSlope').groupby(zones).count()
    gain_area = gain_counts * spatial_resolution ** 2

    return gain_area


def vegetation_water_change_loss_area(zones: GeoDataFrame, spatial_resolution=10) -> GeoSeries:
    loss_counts = VegetationWaterMap(greenwater_layer='lossgreenwaterSlope').groupby(zones).count()
    loss_area = loss_counts * spatial_resolution ** 2

    return loss_area


def vegetation_water_change_gain_loss_ratio(zones: GeoDataFrame) -> GeoSeries:
    start_counts = VegetationWaterMap(greenwater_layer='startgreenwaterIndex').groupby(zones).count()
    loss_counts = VegetationWaterMap(greenwater_layer='lossgreenwaterSlope').groupby(zones).count()
    gain_counts = VegetationWaterMap(greenwater_layer='gaingreenwaterSlope').groupby(zones).count()

    return (gain_counts - loss_counts) / start_counts