doppy 0.2.4__tar.gz → 0.3.1__tar.gz

{doppy-0.2.4 → doppy-0.3.1}/Cargo.lock

@@ -106,7 +106,7 @@ checksum = "248e3bacc7dc6baa3b21e405ee045c3047101a49145e7e9eca583ab4c2ca5345"
 
 [[package]]
 name = "doppy_rs"
-version = "0.2.4"
+version = "0.3.1"
 dependencies = [
  "doprs",
  "numpy",
@@ -115,7 +115,7 @@ dependencies = [
 
 [[package]]
 name = "doprs"
-version = "0.2.4"
+version = "0.3.1"
 dependencies = [
  "chrono",
  "rayon",

{doppy-0.2.4 → doppy-0.3.1}/Cargo.toml

@@ -4,6 +4,6 @@ resolver = "2"
 
 [workspace.package]
 edition = "2021"
-version = "0.2.4"
+version = "0.3.1"
 authors = ["Niko Leskinen <niko.leskinen@fmi.fi>"]
 license-file = "LICENSE"

{doppy-0.2.4 → doppy-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: doppy
-Version: 0.2.4
+Version: 0.3.1
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10

{doppy-0.2.4 → doppy-0.3.1}/src/doppy/netcdf.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import warnings
+from types import TracebackType
 from typing import Literal, TypeAlias
 
 import netCDF4
@@ -14,6 +15,17 @@ class Dataset:
     def __init__(self, filename: str) -> None:
         self.nc = netCDF4.Dataset(filename, mode="w")
 
+    def __enter__(self) -> Dataset:
+        return self
+
+    def __exit__(
+        self,
+        exc_type: type[BaseException] | None,
+        exc_val: BaseException | None,
+        exc_tb: TracebackType | None,
+    ) -> None:
+        self.close()
+
     def add_dimension(self, dim: str) -> Dataset:
         self.nc.createDimension(dim, None)
         return self
@@ -23,7 +35,7 @@ class Dataset:
         return self
 
     def add_atribute(self, key: str, val: str) -> Dataset:
-        warnings.warn("Use add_attribute", DeprecationWarning)
+        warnings.warn("Use add_attribute", DeprecationWarning, stacklevel=2)
         return self.add_attribute(key, val)
 
     def add_time(

{doppy-0.2.4 → doppy-0.3.1}/src/doppy/product/__init__.py

@@ -1,5 +1,6 @@
 from doppy.product.stare import Stare
+from doppy.product.stare_depol import StareDepol
 from doppy.product.wind import Options as WindOptions
 from doppy.product.wind import Wind
 
-__all__ = ["Stare", "Wind", "WindOptions"]
+__all__ = ["Stare", "StareDepol", "Wind", "WindOptions"]

doppy-0.3.1/src/doppy/product/stare_depol.py

@@ -0,0 +1,83 @@
+import numpy as np
+import numpy.typing as npt
+
+from doppy.product.stare import Stare
+
+
+class StareDepol:
+    """
+    Stare product with depolarisation ratio derived from co-polarised and
+    cross-polarised stare data.
+
+    Attributes:
+    -----------
+    time
+        An array of datetime64 objects representing the observation times.
+    radial_distance
+        An array of radial distances from the observation point, in meters.
+    elevation
+        An array of elevation angles corresponding to the observation points, in
+        degrees.
+    beta
+        An array of backscatter coefficients for the co-polarised signal, in
+        sr-1 m-1.
+    radial_velocity
+        An array of radial velocities of the co-polarised signal, in m s-1.
+    mask
+        A boolean array indicating signal (True) or noise (False) data points.
+    depolarisation
+        An array of depolarisation ratios calculated as the ratio of
+        co-polarised to cross-polarised backscatter coefficients.
+    wavelength
+        The wavelength of the lidar, in meters.
+    system_id
+        A string identifier for the lidar.
+
+    Raises
+    ------
+    ValueError
+        If the input `co` and `cross` products have mismatched wavelengths,
+        system IDs, radial distances, or elevation angles, this exception is
+        raised.
+    """
+
+    time: npt.NDArray[np.datetime64]
+    radial_distance: npt.NDArray[np.float64]
+    elevation: npt.NDArray[np.float64]
+    beta: npt.NDArray[np.float64]
+    radial_velocity: npt.NDArray[np.float64]
+    mask: npt.NDArray[np.bool_]
+    depolarisation: npt.NDArray[np.float64]
+    wavelength: float
+    system_id: str
+
+    def __init__(self, co: Stare, cross: Stare):
+        if co.wavelength != cross.wavelength:
+            raise ValueError(
+                "Different wavelength in co and cross: "
+                f"{co.wavelength} vs {cross.wavelength}"
+            )
+        if co.system_id != cross.system_id:
+            raise ValueError(
+                "Different system ID in co and cross: "
+                f"{co.system_id} vs {cross.system_id}"
+            )
+        if not np.allclose(co.radial_distance, cross.radial_distance, atol=1):
+            raise ValueError("Different radial distance in co and cross")
+
+        time_ind = np.argmin(np.abs(co.time - cross.time[:, np.newaxis]), axis=0)
+        cross_elevation = cross.elevation[time_ind]
+        cross_beta = cross.beta[time_ind, :]
+
+        if not np.allclose(co.elevation, cross_elevation, atol=1):
+            raise ValueError("Different elevation in co and cross")
+
+        self.time = co.time
+        self.radial_distance = co.radial_distance
+        self.elevation = co.elevation
+        self.beta = co.beta
+        self.radial_velocity = co.radial_velocity
+        self.mask = co.mask
+        self.depolarisation = cross_beta / co.beta
+        self.wavelength = co.wavelength
+        self.system_id = co.system_id