Source code for disdrodb.utils.manipulations

#!/usr/bin/env python3

# -----------------------------------------------------------------------------.
# Copyright (c) 2021-2023 DISDRODB developers
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
# -----------------------------------------------------------------------------.
"""Include functions helping for DISDRODB product manipulations."""

import numpy as np

from disdrodb.utils.xarray import unstack_datarray_dimension




def get_diameter_bin_edges(ds):
    """Retrieve diameter bin edges."""
    bin_edges = np.append(ds["diameter_bin_lower"].compute().data, ds["diameter_bin_upper"].compute().data[-1])
    return bin_edges





def convert_from_decibel(x):
    """Convert dB to unit."""
    return np.power(10.0, 0.1 * x)  # x/10





def convert_to_decibel(x):
    """Convert unit to dB."""
    return 10 * np.log10(x)





def unstack_radar_variables(ds):
    """Unstack radar variables."""
    from disdrodb.scattering import RADAR_VARIABLES

    for var in RADAR_VARIABLES:
        if var in ds:
            ds_unstack = unstack_datarray_dimension(ds[var], dim="frequency", prefix="", suffix="_")
            ds.update(ds_unstack)
            ds = ds.drop_vars(var)
    if "frequency" in ds.dims:
        ds = ds.drop_dims("frequency")
    return ds





def resample_drop_number_concentration(da, diameter_bin_edges, method="linear"):
    """Resample drop number concentration N(D) DataArray to high resolution diameter bins."""
    diameters_bin_center = diameter_bin_edges[:-1] + np.diff(diameter_bin_edges) / 2

    da = da.interp(coords={"diameter_bin_center": diameters_bin_center}, method=method)
    diameter_bin_width = np.diff(diameter_bin_edges)
    diameter_bin_lower = diameter_bin_edges[:-1]
    diameter_bin_upper = diameter_bin_edges[1:]
    da = da.assign_coords(
        {
            "diameter_bin_width": ("diameter_bin_center", diameter_bin_width),
            "diameter_bin_lower": ("diameter_bin_center", diameter_bin_lower),
            "diameter_bin_upper": ("diameter_bin_center", diameter_bin_upper),
        },
    )
    return da