disdrodb 0.0.21__py3-none-any.whl → 0.1.0__py3-none-any.whl

disdrodb/scattering/routines.py

@@ -0,0 +1,456 @@
+# -----------------------------------------------------------------------------.
+# 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/>.
+# -----------------------------------------------------------------------------.
+"""Implement PSD scattering routines."""
+
+import itertools
+
+import dask
+import numpy as np
+import xarray as xr
+from pytmatrix import orientation, radar, refractive, tmatrix_aux
+from pytmatrix.psd import BinnedPSD, PSDIntegrator
+from pytmatrix.tmatrix import Scatterer
+
+from disdrodb.psd.models import create_psd, get_required_parameters
+from disdrodb.scattering.axis_ratio import check_axis_ratio, get_axis_ratio_method
+from disdrodb.utils.warnings import suppress_warnings
+
+# Wavelengths for which the refractive index is defined in pytmatrix (in mm)
+wavelength_dict = {
+    "S": tmatrix_aux.wl_S,
+    "C": tmatrix_aux.wl_C,
+    "X": tmatrix_aux.wl_X,
+    "Ku": tmatrix_aux.wl_Ku,
+    "Ka": tmatrix_aux.wl_Ka,
+    "W": tmatrix_aux.wl_W,
+}
+
+
+def available_radar_bands():
+    """Return a list of the available radar bands."""
+    return list(wavelength_dict)
+
+
+def check_radar_band(radar_band):
+    """Check the validity of the specified radar band."""
+    available_bands = available_radar_bands()
+    if radar_band not in available_bands:
+        raise ValueError(f"{radar_band} is an invalid radar band. Valid radar bands: {available_bands}.")
+    return radar_band
+
+
+def get_radar_wavelength(radar_band):
+    """Get the wavelength of a radar band."""
+    wavelength = wavelength_dict[radar_band]
+    return wavelength
+
+
+def initialize_scatterer(wavelength, canting_angle_std=7, D_max=8, axis_ratio="Thurai2007"):
+    """Initialize T-matrix scatterer object for a given wavelength."""
+    # Retrieve custom axis ratio function
+    axis_ratio_func = get_axis_ratio_method(axis_ratio)
+
+    # Retrieve water complex refractive index
+    # - Here we currently assume 10 °C
+    # - m_w_0C and m_w_20C are also available
+    # TODO: should be another dimension ? Or use scatterer.psd_integrator.m_func?
+    water_refractive_index = refractive.m_w_10C[wavelength]
+
+    # ---------------------------------------------------------------.
+    # Initialize Scatterer class
+    scatterer = Scatterer(wavelength=wavelength, m=water_refractive_index)
+    # - Define particle orientation PDF for orientational averaging
+    # --> The standard deviation of the angle with respect to vertical orientation (the canting angle).
+    scatterer.or_pdf = orientation.gaussian_pdf(std=canting_angle_std)
+    # - Define orientation methods
+    # --> Alternatives: orient_averaged_fixed, orient_single
+    scatterer.orient = orientation.orient_averaged_fixed
+
+    # ---------------------------------------------------------------.
+    # Initialize PSDIntegrator
+    scatterer.psd_integrator = PSDIntegrator()
+    # - Define axis_ratio_func
+    # --> The Scatterer class expects horizontal to vertical
+    scatterer.psd_integrator.axis_ratio_func = lambda D: 1.0 / axis_ratio_func(D)
+    # - Define function to compute refrative index (as function of D)
+    # scatterer.psd_integrator.m_func = None    # Use constant value of scatterer.m
+    # - Define number of points over which to integrate
+    scatterer.psd_integrator.num_points = 1024
+    # - Define maximum drop diameter
+    scatterer.psd_integrator.D_max = D_max
+    # - Define geometries
+    scatterer.psd_integrator.geometries = (tmatrix_aux.geom_horiz_back, tmatrix_aux.geom_horiz_forw)
+    # ---------------------------------------------------------------.
+    # Initialize scattering table
+    scatterer.psd_integrator.init_scatter_table(scatterer)
+    return scatterer
+
+
+def compute_radar_variables(scatterer):
+    """Compute radar variables for a given scatter object with a specified PSD.
+
+    To speed up computations, this function should input a scatterer object with
+    a preinitialized scattering table.
+    """
+    # Compute radar parameters
+    radar_vars = {}
+    scatterer.set_geometry(tmatrix_aux.geom_horiz_back)
+    radar_vars["Zh"] = 10 * np.log10(radar.refl(scatterer, h_pol=True))  # dBZ
+    radar_vars["Zdr"] = 10 * np.log10(radar.Zdr(scatterer))  # dB
+    radar_vars["rho_hv"] = radar.rho_hv(scatterer)
+    radar_vars["ldr"] = radar.ldr(scatterer)
+    scatterer.set_geometry(tmatrix_aux.geom_horiz_forw)
+    radar_vars["Kdp"] = radar.Kdp(scatterer)
+    radar_vars["Ai"] = radar.Ai(scatterer)
+    return radar_vars
+
+
+def _estimate_empirical_radar_parameters(
+    drop_number_concentration,
+    bin_edges,
+    scatterer,
+    output_dictionary,
+):
+    # Initialize bad results
+    if output_dictionary:
+        null_output = {"Zh": np.nan, "Zdr": np.nan, "rho_hv": np.nan, "ldr": np.nan, "Kdp": np.nan, "Ai": np.nan}
+    else:
+        null_output = np.array([np.nan, np.nan, np.nan, np.nan, np.nan, np.nan])
+
+    # Assign PSD model to the scatterer object
+    scatterer.psd = BinnedPSD(bin_edges, drop_number_concentration)
+
+    # Get radar variables
+    with suppress_warnings():
+        try:
+            radar_vars = compute_radar_variables(scatterer)
+            output = radar_vars if output_dictionary else np.array(list(radar_vars.values()))
+        except Exception:
+            output = null_output
+    return output
+
+
+def _estimate_model_radar_parameters(
+    parameters,
+    psd_model,
+    psd_parameters_names,
+    scatterer,
+    output_dictionary,
+):
+    # Initialize bad results
+    if output_dictionary:
+        null_output = {"Zh": np.nan, "Zdr": np.nan, "rho_hv": np.nan, "ldr": np.nan, "Kdp": np.nan, "Ai": np.nan}
+    else:
+        null_output = np.array([np.nan, np.nan, np.nan, np.nan, np.nan, np.nan])
+
+    # Assign PSD model to the scatterer object
+    parameters = dict(zip(psd_parameters_names, parameters))
+    scatterer.psd = create_psd(psd_model, parameters)
+
+    # Get radar variables
+    with suppress_warnings():
+        radar_vars = compute_radar_variables(scatterer)
+        try:
+            radar_vars = compute_radar_variables(scatterer)
+            output = radar_vars if output_dictionary else np.array(list(radar_vars.values()))
+        except Exception:
+            output = null_output
+    return output
+
+
+def get_psd_parameters(ds):
+    """Return a xr.Dataset with only the PSD parameters as variable."""
+    psd_model = ds.attrs["disdrodb_psd_model"]
+    required_parameters = get_required_parameters(psd_model)
+    missing_parameters = [param for param in required_parameters if param not in ds]
+    if len(missing_parameters) > 0:
+        raise ValueError(f"The {psd_model} parameters {missing_parameters} are not present in the dataset.")
+    return ds[required_parameters]
+
+
+def get_model_radar_parameters(
+    ds,
+    radar_band,
+    canting_angle_std=7,
+    diameter_max=8,
+    axis_ratio="Thurai2007",
+):
+    """Compute radar parameters from a PSD model.
+
+    Parameters
+    ----------
+    ds : xarray.Dataset
+        Dataset containing the parameters of the PSD model.
+        The dataset attribute disdrodb_psd_model specifies the PSD model to use.
+    radar_band : str
+        Radar band to be used.
+    canting_angle_std : float, optional
+        Standard deviation of the canting angle.  The default value is 7.
+    diameter_max : float, optional
+        Maximum diameter. The default value is 8 mm.
+    axis_ratio : str, optional
+        Method to compute the axis ratio. The default method is ``Thurai2007``.
+
+    Returns
+    -------
+    xarray.Dataset
+        Dataset containing the computed radar parameters.
+    """
+    # Retrieve psd model and parameters.
+    psd_model = ds.attrs["disdrodb_psd_model"]
+    required_parameters = get_required_parameters(psd_model)
+    ds_parameters = get_psd_parameters(ds)
+
+    # Check argument validity
+    axis_ratio = check_axis_ratio(axis_ratio)
+    radar_band = check_radar_band(radar_band)
+
+    # Retrieve wavelengths in mm
+    wavelength = get_radar_wavelength(radar_band)
+
+    # Create DataArray with PSD parameters
+    da_parameters = ds_parameters.to_array(dim="psd_parameters").compute()
+
+    # Initialize scattering table
+    scatterer = initialize_scatterer(
+        wavelength=wavelength,
+        canting_angle_std=canting_angle_std,
+        D_max=diameter_max,
+        axis_ratio=axis_ratio,
+    )
+
+    # Define kwargs
+    kwargs = {
+        "output_dictionary": False,
+        "psd_model": psd_model,
+        "psd_parameters_names": required_parameters,
+        "scatterer": scatterer,
+    }
+
+    # Loop over each PSD (not in parallel --> dask="forbidden")
+    # - It costs much more to initiate the scatterer rather than looping over timesteps !
+    da_radar = xr.apply_ufunc(
+        _estimate_model_radar_parameters,
+        da_parameters,
+        kwargs=kwargs,
+        input_core_dims=[["psd_parameters"]],
+        output_core_dims=[["radar_variables"]],
+        vectorize=True,
+        dask="forbidden",
+        dask_gufunc_kwargs={"output_sizes": {"radar_variables": 5}},  # lengths of the new output_core_dims dimensions.
+        output_dtypes=["float64"],
+    )
+
+    # Add parameters coordinates
+    da_radar = da_radar.assign_coords({"radar_variables": ["Zh", "Zdr", "rho_hv", "ldr", "Kdp", "Ai"]})
+
+    # Create parameters dataset
+    ds_radar = da_radar.to_dataset(dim="radar_variables")
+
+    # Expand dimensions for later merging
+    dims_dict = {
+        "radar_band": [radar_band],
+        "axis_ratio": [axis_ratio],
+        "canting_angle_std": [canting_angle_std],
+        "diameter_max": [diameter_max],
+    }
+    ds_radar = ds_radar.expand_dims(dim=dims_dict)
+    return ds_radar
+
+
+def get_empirical_radar_parameters(
+    ds,
+    radar_band=None,
+    canting_angle_std=7,
+    diameter_max=8,
+    axis_ratio="Thurai2007",
+):
+    """Compute radar parameters from empirical drop number concentration.
+
+    Parameters
+    ----------
+    ds : xarray.Dataset
+        Dataset containing the drop number concentration variable.
+    radar_band : str
+        Radar band to be used.
+    canting_angle_std : float, optional
+        Standard deviation of the canting angle.  The default value is 7.
+    diameter_max : float, optional
+        Maximum diameter. The default value is 8 mm.
+    axis_ratio : str, optional
+        Method to compute the axis ratio. The default method is ``Thurai2007``.
+
+    Returns
+    -------
+    xarray.Dataset
+        Dataset containing the computed radar parameters.
+    """
+    # Define inputs
+    da_drop_number_concentration = ds["drop_number_concentration"].compute()
+
+    # Define bin edges
+    bin_edges = np.append(ds["diameter_bin_lower"].compute().data, ds["diameter_bin_upper"].compute().data[-1])
+
+    # Check argument validity
+    axis_ratio = check_axis_ratio(axis_ratio)
+    radar_band = check_radar_band(radar_band)
+
+    # Retrieve wavelengths in mm
+    wavelength = get_radar_wavelength(radar_band)
+
+    # Initialize scattering table
+    scatterer = initialize_scatterer(
+        wavelength=wavelength,
+        canting_angle_std=canting_angle_std,
+        D_max=diameter_max,
+        axis_ratio=axis_ratio,
+    )
+
+    # Define kwargs
+    kwargs = {
+        "output_dictionary": False,
+        "bin_edges": bin_edges,
+        "scatterer": scatterer,
+    }
+
+    # Loop over each PSD (not in parallel --> dask="forbidden")
+    # - It costs much more to initiate the scatterer rather than looping over timesteps !
+    da_radar = xr.apply_ufunc(
+        _estimate_empirical_radar_parameters,
+        da_drop_number_concentration,
+        kwargs=kwargs,
+        input_core_dims=[["diameter_bin_center"]],
+        output_core_dims=[["radar_variables"]],
+        vectorize=True,
+        dask="forbidden",
+        dask_gufunc_kwargs={"output_sizes": {"radar_variables": 5}},  # lengths of the new output_core_dims dimensions.
+        output_dtypes=["float64"],
+    )
+
+    # Add parameters coordinates
+    da_radar = da_radar.assign_coords({"radar_variables": ["Zh", "Zdr", "rho_hv", "ldr", "Kdp", "Ai"]})
+
+    # Create parameters dataset
+    ds_radar = da_radar.to_dataset(dim="radar_variables")
+
+    # Expand dimensions for later merging
+    dims_dict = {
+        "radar_band": [radar_band],
+        "axis_ratio": [axis_ratio],
+        "canting_angle_std": [canting_angle_std],
+        "diameter_max": [diameter_max],
+    }
+    ds_radar = ds_radar.expand_dims(dim=dims_dict)
+    return ds_radar
+
+
+def get_radar_parameters(
+    ds,
+    radar_band=None,
+    canting_angle_std=7,
+    diameter_max=8,
+    axis_ratio="Thurai2007",
+    parallel=True,
+):
+    """Compute radar parameters from empirical drop number concentration or PSD model.
+
+    Parameters
+    ----------
+    ds : xarray.Dataset
+        Dataset containing the drop number concentration variable.
+    radar_band : str or list of str, optional
+        Radar band(s) to be used.
+        If ``None`` (the default), all available radar bands are used.
+    canting_angle_std : float or list of float, optional
+        Standard deviation of the canting angle.  The default value is 7.
+    diameter_max : float or list of float, optional
+        Maximum diameter. The default value is 8 mm.
+    axis_ratio : str or list of str, optional
+        Method to compute the axis ratio. The default method is ``Thurai2007``.
+    parallel : bool, optional
+        Whether to compute radar variables in parallel.
+        The default value is ``True``.
+
+    Returns
+    -------
+    xarray.Dataset
+        Dataset containing the computed radar parameters.
+    """
+    # Decide whether to simulate radar parameters based on empirical PSD or model PSD
+    if "disdrodb_psd_model" not in ds.attrs and "drop_number_concentration" not in ds:
+        raise ValueError("The input dataset is not a DISDRODB L2E or L2M product.")
+    # Model-based simulation
+    if "disdrodb_psd_model" in ds.attrs:
+        func = get_model_radar_parameters
+        ds_subset = get_psd_parameters(ds).compute()
+    # Empirical PSD simulation
+    else:
+        func = get_empirical_radar_parameters
+        ds_subset = ds[["drop_number_concentration"]].compute()
+
+    # Initialize radar band if not provided
+    if radar_band is None:
+        radar_band = available_radar_bands()
+
+    # Ensure parameters are list
+    diameter_max = np.atleast_1d(diameter_max)
+    canting_angle_std = np.atleast_1d(canting_angle_std)
+    axis_ratio = np.atleast_1d(axis_ratio)
+    radar_band = np.atleast_1d(radar_band)
+
+    # Check parameters validity
+    axis_ratio = [check_axis_ratio(method) for method in axis_ratio]
+    radar_band = [check_radar_band(band) for band in radar_band]
+
+    # Order radar band from longest to shortest wavelength
+    # - ["S", "C", "X", "Ku", "Ka", "W"]
+    radar_band = sorted(radar_band, key=lambda x: wavelength_dict[x])[::-1]
+
+    # Retrieve combination of parameters
+    list_params = [
+        {
+            "radar_band": rb.item(),
+            "canting_angle_std": cas.item(),
+            "axis_ratio": ar.item(),
+            "diameter_max": d_max.item(),
+        }
+        for rb, cas, ar, d_max in itertools.product(radar_band, canting_angle_std, axis_ratio, diameter_max)
+    ]
+
+    # Compute radar variables for each configuration in parallel
+    # - The function expects the data into memory (no dask arrays !)
+    if parallel:
+        list_ds = [dask.delayed(func)(ds_subset, **params) for params in list_params]
+        list_ds = dask.compute(*list_ds)
+    else:
+        list_ds = [func(ds_subset, **params) for params in list_params]
+
+    # Merge into a single dataset
+    # - Order radar bands from longest to shortest wavelength
+    ds_radar = xr.merge(list_ds)
+    ds_radar = ds_radar.sel(radar_band=radar_band)
+
+    # Copy global attributes from input dataset
+    ds_radar.attrs = ds.attrs.copy()
+
+    # Remove single dimensions (add info to attributes)
+    parameters = ["radar_band", "canting_angle_std", "axis_ratio", "diameter_max"]
+    for param in parameters:
+        if ds_radar.sizes[param] == 1:
+            ds_radar.attrs[f"disdrodb_scattering_{param}"] = ds_radar[param].item()
+    ds_radar = ds_radar.squeeze()
+    return ds_radar

disdrodb/utils/__init__.py

@@ -0,0 +1,17 @@
+# -----------------------------------------------------------------------------.
+# 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/>.
+# -----------------------------------------------------------------------------.
+"""DISDRODB Utils Module."""

disdrodb/utils/attrs.py

@@ -0,0 +1,208 @@
+#!/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/>.
+# -----------------------------------------------------------------------------.
+"""DISDRODB netCDF4 attributes utilities."""
+import datetime
+
+from disdrodb import ARCHIVE_VERSION, CONVENTIONS, SOFTWARE_VERSION
+
+####---------------------------------------------------------------------.
+#### Variable attributes
+
+
+def set_attrs(ds, attrs_dict):
+    """Set attributes to the variables of the xr.Dataset."""
+    for var in attrs_dict:
+        if var in ds:
+            ds[var].attrs.update(attrs_dict[var])
+    return ds
+
+
+####---------------------------------------------------------------------.
+#### Coordinates attributes
+
+
+def get_coords_attrs_dict():
+    """Return dictionary with DISDRODB coordinates attributes."""
+    attrs_dict = {}
+    # Define diameter attributes
+    attrs_dict["diameter_bin_center"] = {
+        "name": "diameter_bin_center",
+        "standard_name": "diameter_bin_center",
+        "long_name": "diameter_bin_center",
+        "units": "mm",
+        "description": "Bin center drop diameter value",
+    }
+    attrs_dict["diameter_bin_width"] = {
+        "name": "diameter_bin_width",
+        "standard_name": "diameter_bin_width",
+        "long_name": "diameter_bin_width",
+        "units": "mm",
+        "description": "Drop diameter bin width",
+    }
+    attrs_dict["diameter_bin_upper"] = {
+        "name": "diameter_bin_upper",
+        "standard_name": "diameter_bin_upper",
+        "long_name": "diameter_bin_upper",
+        "units": "mm",
+        "description": "Bin upper bound drop diameter value",
+    }
+    attrs_dict["velocity_bin_lower"] = {
+        "name": "velocity_bin_lower",
+        "standard_name": "velocity_bin_lower",
+        "long_name": "velocity_bin_lower",
+        "units": "mm",
+        "description": "Bin lower bound drop diameter value",
+    }
+    # Define velocity attributes
+    attrs_dict["velocity_bin_center"] = {
+        "name": "velocity_bin_center",
+        "standard_name": "velocity_bin_center",
+        "long_name": "velocity_bin_center",
+        "units": "m/s",
+        "description": "Bin center drop fall velocity value",
+    }
+    attrs_dict["velocity_bin_width"] = {
+        "name": "velocity_bin_width",
+        "standard_name": "velocity_bin_width",
+        "long_name": "velocity_bin_width",
+        "units": "m/s",
+        "description": "Drop fall velocity bin width",
+    }
+    attrs_dict["velocity_bin_upper"] = {
+        "name": "velocity_bin_upper",
+        "standard_name": "velocity_bin_upper",
+        "long_name": "velocity_bin_upper",
+        "units": "m/s",
+        "description": "Bin upper bound drop fall velocity value",
+    }
+    attrs_dict["velocity_bin_lower"] = {
+        "name": "velocity_bin_lower",
+        "standard_name": "velocity_bin_lower",
+        "long_name": "velocity_bin_lower",
+        "units": "m/s",
+        "description": "Bin lower bound drop fall velocity value",
+    }
+    # Define geolocation attributes
+    attrs_dict["latitude"] = {
+        "name": "latitude",
+        "standard_name": "latitude",
+        "long_name": "Latitude",
+        "units": "degrees_north",
+    }
+    attrs_dict["longitude"] = {
+        "name": "longitude",
+        "standard_name": "longitude",
+        "long_name": "Longitude",
+        "units": "degrees_east",
+    }
+    attrs_dict["altitude"] = {
+        "name": "altitude",
+        "standard_name": "altitude",
+        "long_name": "Altitude",
+        "units": "m",
+        "description": "Elevation above sea level",
+    }
+    # Define time attributes
+    attrs_dict["time"] = {
+        "name": "time",
+        "standard_name": "time",
+        "long_name": "time",
+        "description": "UTC Time",
+    }
+
+    return attrs_dict
+
+
+def set_coordinate_attributes(ds):
+    """Set coordinates attributes."""
+    # Get attributes dictionary
+    attrs_dict = get_coords_attrs_dict()
+    # Set attributes
+    ds = set_attrs(ds, attrs_dict)
+    return ds
+
+
+####-------------------------------------------------------------------------.
+#### DISDRODB Global Attributes
+
+
+def set_disdrodb_attrs(ds, product: str):
+    """Add DISDRODB processing information to the netCDF global attributes.
+
+    It assumes stations metadata are already added the dataset.
+
+    Parameters
+    ----------
+    ds : xarray.Dataset
+        Dataset
+    product: str
+        DISDRODB product.
+
+    Returns
+    -------
+    xarray dataset
+        Dataset.
+    """
+    # Add dataset conventions
+    ds.attrs["Conventions"] = CONVENTIONS
+
+    # Add featureType
+    if "platform_type" in ds.attrs:
+        platform_type = ds.attrs["platform_type"]
+        if platform_type == "fixed":
+            ds.attrs["featureType"] = "timeSeries"
+        else:
+            ds.attrs["featureType"] = "trajectory"
+
+    # Update DISDRODDB attributes
+    ds = update_disdrodb_attrs(ds=ds, product=product)
+    return ds
+
+
+def update_disdrodb_attrs(ds, product: str):
+    """Add DISDRODB processing information to the netCDF global attributes.
+
+    It assumes stations metadata are already added the dataset.
+
+    Parameters
+    ----------
+    ds : xarray dataset.
+        Dataset
+    product: str
+        DISDRODB product.
+
+    Returns
+    -------
+    xarray dataset
+        Dataset.
+    """
+    # Add time_coverage_start and time_coverage_end
+    ds.attrs["time_coverage_start"] = str(ds["time"].data[0])
+    ds.attrs["time_coverage_end"] = str(ds["time"].data[-1])
+
+    # DISDRODDB attributes
+    # - Add DISDRODB processing info
+    now = datetime.datetime.utcnow()
+    current_time = now.strftime("%Y-%m-%d %H:%M:%S")
+    ds.attrs["disdrodb_processing_date"] = current_time
+    # - Add DISDRODB product and version
+    ds.attrs["disdrodb_product_version"] = ARCHIVE_VERSION
+    ds.attrs["disdrodb_software_version"] = SOFTWARE_VERSION
+    ds.attrs["disdrodb_product"] = product
+    return ds