doppy 0.2.1__cp310-abi3-macosx_10_12_x86_64.whl → 0.2.3__cp310-abi3-macosx_10_12_x86_64.whl

doppy/product/__init__.py

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

doppy/product/wind.py

@@ -18,6 +18,11 @@ import doppy
 SelectionGroupKeyType: TypeAlias = tuple[int, int, tuple[int, ...]]
 
 
+@dataclass
+class Options:
+    azimuth_offset_deg: float | None
+
+
 @dataclass
 class Wind:
     time: npt.NDArray[np.datetime64]
@@ -45,6 +50,7 @@ class Wind:
         | Sequence[Path]
         | Sequence[bytes]
         | Sequence[BufferedIOBase],
+        options: Options | None = None,
     ) -> Wind:
         raws = doppy.raw.HaloHpl.from_srcs(data)
 
@@ -60,6 +66,9 @@ class Wind:
         if len(raw.time) == 0:
             raise doppy.exceptions.NoDataError("No suitable data for the wind product")
 
+        if options and options.azimuth_offset_deg:
+            raw.azimuth += options.azimuth_offset_deg
+
         groups = _group_scans_by_azimuth_rotation(raw)
         time_list = []
         elevation_list = []
@@ -104,6 +113,7 @@ class Wind:
         | Sequence[Path]
         | Sequence[bytes]
         | Sequence[BufferedIOBase],
+        options: Options | None = None,
     ) -> Wind:
         raws = doppy.raw.WindCube.from_vad_srcs(data)
 
@@ -119,6 +129,9 @@ class Wind:
         if len(raw.time) == 0:
             raise doppy.exceptions.NoDataError("No suitable data for the wind product")
 
+        if options and options.azimuth_offset_deg:
+            raw.azimuth += options.azimuth_offset_deg
+
         time_list = []
         elevation_list = []
         wind_list = []
@@ -159,6 +172,7 @@ class Wind:
         | Sequence[Path]
         | Sequence[bytes]
         | Sequence[BufferedIOBase],
+        options: Options | None = None,
     ) -> Wind:
         raws = doppy.raw.Wls70.from_srcs(data)
 
@@ -170,6 +184,20 @@ class Wind:
             .sorted_by_time()
             .non_strictly_increasing_timesteps_removed()
         )
+
+        if options and options.azimuth_offset_deg:
+            theta = np.deg2rad(options.azimuth_offset_deg)
+            cos_theta = np.cos(theta)
+            sin_theta = np.sin(theta)
+
+            meridional_wind = (
+                sin_theta * raw.zonal_wind + cos_theta * raw.meridional_wind
+            )
+            zonal_wind = cos_theta * raw.zonal_wind - sin_theta * raw.meridional_wind
+        else:
+            meridional_wind = raw.meridional_wind
+            zonal_wind = raw.zonal_wind
+
         mask = (
             np.isnan(raw.meridional_wind)
             | np.isnan(raw.zonal_wind)
@@ -178,8 +206,8 @@ class Wind:
         return Wind(
             time=raw.time,
             height=raw.altitude,
-            zonal_wind=raw.zonal_wind,
-            meridional_wind=raw.meridional_wind,
+            zonal_wind=zonal_wind,
+            meridional_wind=meridional_wind,
             vertical_wind=raw.vertical_wind,
             mask=mask,
             system_id=raw.system_id,

doppy/raw/wls70.py

@@ -33,6 +33,7 @@ class Wls70:
     ]  # v := meridional wind?, dim: (time, altitude)
     vertical_wind: npt.NDArray[np.float64]  # w := vertical wind?, dim: (time, altitude)
     system_id: str
+    cnr_threshold: float
 
     @classmethod
     def from_srcs(
@@ -111,6 +112,7 @@ class Wls70:
                 meridional_wind=self.meridional_wind[index],
                 vertical_wind=self.vertical_wind[index],
                 system_id=self.system_id,
+                cnr_threshold=self.cnr_threshold,
             )
         raise TypeError
 
@@ -137,6 +139,9 @@ class Wls70:
             meridional_wind=np.concatenate(tuple(r.meridional_wind for r in raws)),
             vertical_wind=np.concatenate(tuple(r.vertical_wind for r in raws)),
             system_id=merge_all_equal("system_id", [r.system_id for r in raws]),
+            cnr_threshold=merge_all_equal(
+                "cnr_threshold", [r.cnr_threshold for r in raws]
+            ),
         )
 
     def non_strictly_increasing_timesteps_removed(self) -> Wls70:
@@ -158,6 +163,7 @@ def _raw_rs_to_wls70(
     info, cols, data = raw_rs
     altitude = info["altitude"]
     system_id = info["system_id"]
+    cnr_threshold = float(info["cnr_threshold"])
     data = data.reshape(-1, len(cols))
     time_ts = data[:, 0]
     time = np.array([datetime64(datetime.utcfromtimestamp(ts)) for ts in time_ts])
@@ -173,7 +179,10 @@ def _raw_rs_to_wls70(
     u = data[:, 9::8]
     v = data[:, 10::8]
     w = data[:, 11::8]
-
+    mask = (np.abs(u) > 90) | (np.abs(v) > 90) | (np.abs(w) > 90)
+    u[mask] = np.nan
+    v[mask] = np.nan
+    w[mask] = np.nan
     return Wls70(
         time=time,
         altitude=altitude,
@@ -189,4 +198,5 @@ def _raw_rs_to_wls70(
         meridional_wind=v,
         vertical_wind=w,
         system_id=system_id,
+        cnr_threshold=cnr_threshold,
     )

doppy-0.2.3.dist-info/METADATA

@@ -0,0 +1,137 @@
+Metadata-Version: 2.3
+Name: doppy
+Version: 0.2.3
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
+Requires-Dist: requests
+Requires-Dist: urllib3
+Requires-Dist: numpy
+Requires-Dist: netcdf4
+Requires-Dist: typer
+Requires-Dist: matplotlib
+Requires-Dist: scikit-learn
+Requires-Dist: scipy
+Requires-Dist: mypy ; extra == 'dev'
+Requires-Dist: ruff ; extra == 'dev'
+Requires-Dist: pytest ; extra == 'dev'
+Requires-Dist: types-requests ; extra == 'dev'
+Requires-Dist: py-spy ; extra == 'dev'
+Requires-Dist: maturin ==1.4 ; extra == 'dev'
+Requires-Dist: release-version ; extra == 'dev'
+Requires-Dist: pre-commit ; extra == 'dev'
+Provides-Extra: dev
+License-File: LICENSE
+License-File: LICENSE
+Author: Niko Leskinen <niko.leskinen@fmi.fi>
+Author-email: Niko Leskinen <niko.leskinen@fmi.fi>
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
+Project-URL: Homepage, https://github.com/actris-cloudnet/doppy
+Project-URL: Repository, https://github.com/actris-cloudnet/doppy
+Project-URL: Changelog, https://github.com/actris-cloudnet/doppy/blob/main/CHANGELOG.md
+Project-URL: Bug Tracker, https://github.com/actris-cloudnet/doppy/issues
+
+# Doppy - Wind doppler lidar processing
+
+[![CI](https://github.com/actris-cloudnet/doppy/actions/workflows/ci.yml/badge.svg)](https://github.com/actris-cloudnet/doppy/actions/workflows/ci.yml)
+[![PyPI version](https://badge.fury.io/py/doppy.svg)](https://badge.fury.io/py/doppy)
+
+## Products
+
+- [Stare](https://github.com/actris-cloudnet/doppy/blob/main/src/doppy/product/stare.py): [Examples](https://cloudnet.fmi.fi/search/visualizations?experimental=true&product=doppler-lidar&dateFrom=2024-06-05&dateTo=2024-06-05)
+- [Wind](https://github.com/actris-cloudnet/doppy/blob/main/src/doppy/product/wind.py): [Examples](https://cloudnet.fmi.fi/search/visualizations?experimental=true&product=doppler-lidar-wind&dateFrom=2024-06-05&dateTo=2024-06-05)
+
+## Instruments
+
+- HALO Photonics Streamline lidars (stare, wind)
+- Leosphere WindCube WLS200S (wind)
+- Leosphere WindCube WLS70 (wind)
+
+## Install
+
+```sh
+pip install doppy
+```
+
+## Usage
+
+```python
+import doppy
+
+stare = doppy.product.Stare.from_halo_data(
+    data=LIST_OF_STARE_FILE_PATHS,
+    data_bg=LIST_OF_BACKGROUND_FILE_PATHS,
+    bg_correction_method=doppy.options.BgCorrectionMethod.FIT,
+)
+
+
+(
+    doppy.netcdf.Dataset(FILENAME)
+    .add_dimension("time")
+    .add_dimension("range")
+    .add_time(
+        name="time",
+        dimensions=("time",),
+        standard_name="time",
+        long_name="Time UTC",
+        data=stare.time,
+        dtype="f8",
+    )
+    .add_variable(
+        name="range",
+        dimensions=("range",),
+        units="m",
+        data=stare.radial_distance,
+        dtype="f4",
+    )
+    .add_variable(
+        name="elevation",
+        dimensions=("time",),
+        units="degrees",
+        data=stare.elevation,
+        dtype="f4",
+        long_name="elevation from horizontal",
+    )
+    .add_variable(
+        name="beta_raw",
+        dimensions=("time", "range"),
+        units="sr-1 m-1",
+        data=stare.beta,
+        dtype="f4",
+    )
+    .add_variable(
+        name="beta",
+        dimensions=("time", "range"),
+        units="sr-1 m-1",
+        data=stare.beta,
+        dtype="f4",
+        mask=stare.mask,
+    )
+    .add_variable(
+        name="v",
+        dimensions=("time", "range"),
+        units="m s-1",
+        long_name="Doppler velocity",
+        data=stare.radial_velocity,
+        dtype="f4",
+        mask=stare.mask,
+    )
+    .add_scalar_variable(
+        name="wavelength",
+        units="m",
+        standard_name="radiation_wavelength",
+        data=stare.wavelength,
+        dtype="f4",
+    )
+    .add_atribute("serial_number", stare.system_id)
+    .add_atribute("doppy_version", doppy.__version__)
+).close()
+
+```
+

{doppy-0.2.1.dist-info → doppy-0.2.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
-doppy-0.2.1.dist-info/METADATA,sha256=0T7yh-b9uQUGEeBymLzElKfO6tw2QAlDqffDPUNda7k,1789
-doppy-0.2.1.dist-info/WHEEL,sha256=HYZ_jUi0IGqtX7pjBFQby7ex6mkgDnUsBYfR-onY1Fw,105
-doppy-0.2.1.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
-doppy-0.2.1.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
-doppy-0.2.1.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.2.3.dist-info/METADATA,sha256=oQQ7h-pTzAIlKSN7C0vAqEw70QLwmzVjeuNwmvxMYGY,4142
+doppy-0.2.3.dist-info/WHEEL,sha256=yeRC21W_lRgy51Xt7_Plp8suG9tGryQjlozuN-L_Ook,105
+doppy-0.2.3.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.2.3.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.2.3.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
 doppy/options.py,sha256=73BDODO4OYHn2qOshhwz6u6G3J1kNd3uj6P0a3V4HBE,205
 doppy/__init__.py,sha256=Z9aEUlbPRWRUAoB8_-djkgrJuS4-6pjem4-mVSB6Z9I,191
-doppy/product/wind.py,sha256=MAK-ZTx37gL7y-x3RbqN_QrZR9kjAiLxdIMn_c6M5UA,12151
-doppy/product/__init__.py,sha256=lyp88zs7GBk8uUzkj8lxaXPuYwPa52xXeTwf5qwFddU,103
+doppy/product/wind.py,sha256=mL8Ru7Hefq8YBYFIklBkySO3OEoRQ32ZPgB3deIhO7c,13042
+doppy/product/__init__.py,sha256=DwTrvhMY_Eq-RG3H6Qyg_R_E9ie1wv0ofKr_1rSJ4Sk,172
 doppy/product/stare.py,sha256=aMhOA5yXjcnwgzAp4GvLF1HT1cD-Zuqg9sEQIPlxWa0,19907
 doppy/bench.py,sha256=iVNYveMVGGRES2oe3Orsn31jQFCKTXOmxRFuFiJ8_OA,248
 doppy/netcdf.py,sha256=uEiJfqxXxHwe5ghT6OpIu8xeY1HgWDW1ENTBlFLOGtE,3416
@@ -20,10 +20,10 @@ doppy/data/api.py,sha256=QaVKj304OPcu8OF5xgtduQzDis8Srn-I6UgR9qb2u9E,1863
 doppy/data/exceptions.py,sha256=6CS6OHIWq8CqlxiceEvC1j0EfWUYoIfM7dW88apQVn4,89
 doppy/raw/halo_sys_params.py,sha256=IXH40xBHyXCGX0ZE79KnSeXRj1wbqoqL0RYUQyBJqdE,3937
 doppy/raw/__init__.py,sha256=AMHyONuH0aUJUQz20EhlANaq9UjWJtSZf7kWUVx3ZjA,228
-doppy/raw/wls70.py,sha256=paFT8PddRemtiDAXxkZVcMljCq_CFw1DHmxLHXY0ZhE,7008
+doppy/raw/wls70.py,sha256=PMyr_JYTiKuYv1JpWHgVX6OA5zU83DOnr8vBCxFyFAw,7420
 doppy/raw/halo_bg.py,sha256=kO03yGlKS-DpMMGHYuy_BuidyeUL38TxT5vMn8H_8lE,4809
 doppy/raw/halo_hpl.py,sha256=VkWemJfGFu-tF-fkhjYwF6dCZPrNQanqL0NBFJVS-TQ,18485
 doppy/raw/windcube.py,sha256=ZaOswQJbVDPBYj5oU1pTYmsX8F-mKIbjJRZmLYEMXP0,9906
 doppy/__main__.py,sha256=zrKQJVj0k0ypBQCGK65Czt9G9FZ_qx3ussw6Q9VJ14g,346
-doppy/rs.abi3.so,sha256=xJ24N0vZB2LmuWNoCt2wUI4f1gFnUVOh5NxgvYPjlSE,2876040
-doppy-0.2.1.dist-info/RECORD,,
+doppy/rs.abi3.so,sha256=uS-gr5NekphI-rQ6plXqpuEDCiEJXGUZfyvl5atqSKE,2844096
+doppy-0.2.3.dist-info/RECORD,,

{doppy-0.2.1.dist-info → doppy-0.2.3.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.5.1)
+Generator: maturin (1.6.0)
 Root-Is-Purelib: false
 Tag: cp310-abi3-macosx_10_12_x86_64

doppy-0.2.1.dist-info/METADATA

@@ -1,44 +0,0 @@
-Metadata-Version: 2.3
-Name: doppy
-Version: 0.2.1
-Classifier: Development Status :: 4 - Beta
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: License :: OSI Approved :: MIT License
-Classifier: Intended Audience :: Science/Research
-Classifier: Operating System :: OS Independent
-Requires-Dist: requests
-Requires-Dist: urllib3
-Requires-Dist: numpy
-Requires-Dist: netcdf4
-Requires-Dist: typer
-Requires-Dist: matplotlib
-Requires-Dist: scikit-learn
-Requires-Dist: scipy
-Requires-Dist: mypy ; extra == 'dev'
-Requires-Dist: ruff ; extra == 'dev'
-Requires-Dist: pytest ; extra == 'dev'
-Requires-Dist: types-requests ; extra == 'dev'
-Requires-Dist: py-spy ; extra == 'dev'
-Requires-Dist: maturin ==1.4 ; extra == 'dev'
-Requires-Dist: release-version ; extra == 'dev'
-Requires-Dist: pre-commit ; extra == 'dev'
-Provides-Extra: dev
-License-File: LICENSE
-License-File: LICENSE
-Author: Niko Leskinen <niko.leskinen@fmi.fi>
-Author-email: Niko Leskinen <niko.leskinen@fmi.fi>
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
-Project-URL: Homepage, https://github.com/actris-cloudnet/doppy
-Project-URL: Repository, https://github.com/actris-cloudnet/doppy
-Project-URL: Changelog, https://github.com/actris-cloudnet/doppy/blob/main/CHANGELOG.md
-Project-URL: Bug Tracker, https://github.com/actris-cloudnet/doppy/issues
-
-# Doppy - Wind doppler lidar processing
-
-[![CI](https://github.com/actris-cloudnet/doppy/actions/workflows/ci.yml/badge.svg)](https://github.com/actris-cloudnet/doppy/actions/workflows/ci.yml)
-[![PyPI version](https://badge.fury.io/py/doppy.svg)](https://badge.fury.io/py/doppy)
-