doppy 0.5.2__cp310-abi3-win_amd64.whl → 0.5.4__cp310-abi3-win_amd64.whl

doppy/raw/__init__.py

@@ -3,5 +3,14 @@ from .halo_hpl import HaloHpl
 from .halo_sys_params import HaloSysParams
 from .windcube import WindCube, WindCubeFixed
 from .wls70 import Wls70
+from .wls77 import Wls77
 
-__all__ = ["HaloHpl", "HaloBg", "HaloSysParams", "WindCube", "WindCubeFixed", "Wls70"]
+__all__ = [
+    "HaloHpl",
+    "HaloBg",
+    "HaloSysParams",
+    "WindCube",
+    "WindCubeFixed",
+    "Wls70",
+    "Wls77",
+]

doppy/raw/wls77.py

@@ -0,0 +1,163 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from datetime import datetime, timezone
+from io import BufferedIOBase
+from pathlib import Path
+from typing import Any, Sequence
+
+import numpy as np
+import numpy.typing as npt
+from numpy import datetime64
+
+import doppy
+from doppy import exceptions
+from doppy.raw.utils import bytes_from_src
+from doppy.utils import merge_all_equal
+
+
+@dataclass
+class Wls77:
+    time: npt.NDArray[datetime64]  # dim: (time, )
+    altitude: npt.NDArray[np.float64]  # dim: (altitude, )
+    position: npt.NDArray[np.float64]  # dim: (time, )
+    temperature: npt.NDArray[np.float64]  # dim: (time, )
+    wiper_count: npt.NDArray[np.float64]  # dim: (time, )
+    cnr: npt.NDArray[np.float64]  # dim: (time, altitude)
+    radial_velocity: npt.NDArray[np.float64]  # dim: (time, altitude)
+    radial_velocity_deviation: npt.NDArray[np.float64]  # dim: (time, altitude)
+    wind_speed: npt.NDArray[np.float64]  # dim: (time, altitude)
+    wind_direction: npt.NDArray[np.float64]  # dim: (time, altitude)
+    zonal_wind: npt.NDArray[np.float64]  # u := zonal wind?, dim: (time, altitude)
+    meridional_wind: npt.NDArray[
+        np.float64
+    ]  # v := meridional wind?, dim: (time, altitude)
+    vertical_wind: npt.NDArray[np.float64]  # w := vertical wind?, dim: (time, altitude)
+    cnr_threshold: float
+    system_id: str
+
+    @classmethod
+    def from_srcs(
+        cls, data: Sequence[str | bytes | Path | BufferedIOBase]
+    ) -> list[Wls77]:
+        data_bytes = [bytes_from_src(src) for src in data]
+        raws = doppy.rs.raw.wls77.from_bytes_srcs(data_bytes)
+        try:
+            return [_raw_rs_to_wls77(r) for r in raws]
+        except RuntimeError as err:
+            raise exceptions.RawParsingError(err) from err
+
+    @classmethod
+    def from_src(cls, data: str | Path | bytes | BufferedIOBase) -> Wls77:
+        data_bytes = bytes_from_src(data)
+        try:
+            return _raw_rs_to_wls77(doppy.rs.raw.wls77.from_bytes_src(data_bytes))
+        except RuntimeError as err:
+            raise exceptions.RawParsingError(err) from err
+
+    def __getitem__(
+        self,
+        index: int
+        | slice
+        | list[int]
+        | npt.NDArray[np.int64]
+        | npt.NDArray[np.bool_]
+        | tuple[slice, slice],
+    ) -> Wls77:
+        if isinstance(index, (int, slice, list, np.ndarray)):
+            return Wls77(
+                time=self.time[index],
+                altitude=self.altitude,
+                position=self.position[index],
+                temperature=self.temperature[index],
+                wiper_count=self.wiper_count[index],
+                cnr=self.cnr[index],
+                radial_velocity=self.radial_velocity[index],
+                radial_velocity_deviation=self.radial_velocity_deviation[index],
+                wind_speed=self.wind_speed[index],
+                wind_direction=self.wind_direction[index],
+                zonal_wind=self.zonal_wind[index],
+                meridional_wind=self.meridional_wind[index],
+                vertical_wind=self.vertical_wind[index],
+                system_id=self.system_id,
+                cnr_threshold=self.cnr_threshold,
+            )
+        raise TypeError
+
+    def sorted_by_time(self) -> Wls77:
+        sort_indices = np.argsort(self.time)
+        return self[sort_indices]
+
+    @classmethod
+    def merge(cls, raws: Sequence[Wls77]) -> Wls77:
+        return cls(
+            time=np.concatenate(tuple(r.time for r in raws)),
+            altitude=raws[0].altitude,
+            position=np.concatenate(tuple(r.position for r in raws)),
+            temperature=np.concatenate(tuple(r.temperature for r in raws)),
+            wiper_count=np.concatenate(tuple(r.wiper_count for r in raws)),
+            cnr=np.concatenate(tuple(r.cnr for r in raws)),
+            radial_velocity=np.concatenate(tuple(r.radial_velocity for r in raws)),
+            radial_velocity_deviation=np.concatenate(
+                tuple(r.radial_velocity_deviation for r in raws)
+            ),
+            wind_speed=np.concatenate(tuple(r.wind_speed for r in raws)),
+            wind_direction=np.concatenate(tuple(r.wind_direction for r in raws)),
+            zonal_wind=np.concatenate(tuple(r.zonal_wind for r in raws)),
+            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) -> Wls77:
+        if len(self.time) == 0:
+            return self
+        mask = np.ones_like(self.time, dtype=np.bool_)
+        latest_time = self.time[0]
+        for i, t in enumerate(self.time[1:], start=1):
+            if t <= latest_time:
+                mask[i] = False
+            else:
+                latest_time = t
+        return self[mask]
+
+
+def _raw_rs_to_wls77(
+    raw: dict[str, Any],
+) -> Wls77:
+    time_ts = raw["time"]
+    time = np.array(
+        [
+            datetime64(datetime.fromtimestamp(ts, timezone.utc).replace(tzinfo=None))
+            for ts in time_ts
+        ]
+    )
+
+    n = time.size
+
+    return Wls77(
+        time=time,
+        altitude=np.array(raw["altitude"], dtype=np.float64),
+        position=np.array(raw["position"], dtype=np.float64),
+        temperature=np.array(raw["temperature"], dtype=np.float64),
+        wiper_count=np.array(raw["wiper_count"], dtype=np.float64),
+        cnr=np.array(raw["cnr"], dtype=np.float64).reshape(n, -1),
+        radial_velocity=np.array(raw["radial_velocity"], dtype=np.float64).reshape(
+            n, -1
+        ),
+        radial_velocity_deviation=np.array(
+            raw["radial_velocity_deviation"], dtype=np.float64
+        ).reshape(n, -1),
+        wind_speed=np.array(raw["wind_speed"], dtype=np.float64).reshape(n, -1),
+        wind_direction=np.array(raw["wind_direction"], dtype=np.float64).reshape(n, -1),
+        zonal_wind=np.array(raw["zonal_wind"], dtype=np.float64).reshape(n, -1),
+        meridional_wind=np.array(raw["meridional_wind"], dtype=np.float64).reshape(
+            n, -1
+        ),
+        vertical_wind=np.array(raw["vertical_wind"], dtype=np.float64).reshape(n, -1),
+        cnr_threshold=raw["cnr_threshold"],
+        system_id=raw["system_id"],
+    )

{doppy-0.5.2.dist-info → doppy-0.5.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: doppy
-Version: 0.5.2
+Version: 0.5.4
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
@@ -28,6 +28,7 @@ Requires-Dist: maturin==1.8 ; extra == 'dev'
 Requires-Dist: release-version ; extra == 'dev'
 Requires-Dist: pre-commit ; extra == 'dev'
 Requires-Dist: xarray[io] ; extra == 'dev'
+Requires-Dist: seaborn ; extra == 'dev'
 Provides-Extra: dev
 License-File: LICENSE
 License-File: LICENSE

{doppy-0.5.2.dist-info → doppy-0.5.4.dist-info}/RECORD

@@ -1,7 +1,7 @@
-doppy-0.5.2.dist-info/METADATA,sha256=mHDuliE31HGflXwmDeo0uCEscZHGj2e8iQ0r20nYKHI,4414
-doppy-0.5.2.dist-info/WHEEL,sha256=md9qofgGs0CN4-5nDhVd0IzxxzM8cFGCnkeRThUNGbI,95
-doppy-0.5.2.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
-doppy-0.5.2.dist-info/licenses/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
+doppy-0.5.4.dist-info/METADATA,sha256=L0tX0_qaQLLmArtnqhCJW4EJBvLgasgkPyqOjI0v_34,4454
+doppy-0.5.4.dist-info/WHEEL,sha256=GbZa854Nd4wq8FJ_cHwge55xWx0Hgf1aqY1IVa8I2To,95
+doppy-0.5.4.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.5.4.dist-info/licenses/LICENSE,sha256=RIAxFjJLTw0wQ3_SM73JoTeppoD99DJJ72cjvVuRrW4,1110
 doppy/bench.py,sha256=X4yPXFPAM708ptzXKDYxVBOC0E3cMj_KxXzmD_Q9_RM,321
 doppy/data/api.py,sha256=IoLRodfITNHs4Pb2IvVvOjSirSeK-xNoSO4tsQYjsKc,1933
 doppy/data/cache.py,sha256=ERFzH-KjLLXV8fCePsewTepKZXmLwtXxeazmHiFdA80,1203
@@ -24,9 +24,10 @@ doppy/raw/halo_sys_params.py,sha256=9aZpAgaLFniADjYv7EO8wKtmBD6McJP2v7YUz94hrTM,
 doppy/raw/utils.py,sha256=pBeBk87_LMxop14_RK5Xj64iYBMu8AhH-IqBRoGgUzE,436
 doppy/raw/windcube.py,sha256=FBoiLHXNluN2HISe6OO5TE32BtZYTuvSwyvcYIafLtE,19626
 doppy/raw/wls70.py,sha256=lWSnuZlmdCGqrlYt6SUDFvPRBOOdtt6LiDiu_nhnPiU,6494
-doppy/raw/__init__.py,sha256=ycT-_mCiI19b2TH33L74-Eq_MYUTPe3B8NhwNodSx1E,267
+doppy/raw/wls77.py,sha256=SOIFGC1GhoAPGePH6B-fAuyWqfW5jevi1zbku9JOBng,6623
+doppy/raw/__init__.py,sha256=QqAPpV53Xbz3tHo4zCYv9_pgLYw_F2YIRH2-Mg9l2gI,341
 doppy/utils.py,sha256=2PwDiO8GyMjsHC-EYcTGyN2mEfPaCi3hhEn2twETOl0,814
 doppy/__init__.py,sha256=Af7_8p3oN1nTqS9fo0mVKVuiKf5CAEK69uQa32CSFBA,197
 doppy/__main__.py,sha256=38hIWWfanILuBBGorQiAaleSC4qYJoIxuzVBkxf7Dng,371
-doppy/rs.pyd,sha256=DDQzPRR6UTBJqDpmZfP9FL-1TJcg-26nSktIeVnVVww,2164736
-doppy-0.5.2.dist-info/RECORD,,
+doppy/rs.pyd,sha256=8H1KXP5TZTafe-bXfbVS55yAJE6mmYRvE4OFJOYMAKA,2198528
+doppy-0.5.4.dist-info/RECORD,,

{doppy-0.5.2.dist-info → doppy-0.5.4.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: maturin (1.8.2)
+Generator: maturin (1.8.3)
 Root-Is-Purelib: false
 Tag: cp310-abi3-win_amd64