doppy 0.0.2__cp310-abi3-macosx_11_0_arm64.whl → 0.0.3__cp310-abi3-macosx_11_0_arm64.whl

doppy/product/stare.py

@@ -507,8 +507,38 @@ def _select_raws_for_stare(
     if len(raws) == 0:
         raise doppy.exceptions.NoDataError("No data to select from")
 
-    # Select files that stare vertically
-    raws_stare = [raw for raw in raws if raw.elevation_angles == {90}]
+    # Select files that stare
+    raws_stare = [
+        raw
+        for raw in raws
+        if len(raw.azimuth_angles) == 1 or raw.azimuth_angles == {0, 360}
+    ]
+    if len(raws_stare) == 0:
+        raise doppy.exceptions.NoDataError(
+            "No data suitable for stare product. Data is probably from scans"
+        )
+    raws_stare = [raw for raw in raws if len(raw.elevation_angles) == 1]
+    if len(raws_stare) == 0:
+        raise doppy.exceptions.NoDataError(
+            "No data suitable for stare product. "
+            "Elevation angle does not remain constant"
+        )
+    elevation_angles = []
+    for raw in raws_stare:
+        elevation_angles += list(raw.elevation_angles)
+    max_elevation_angle = max(elevation_angles)
+
+    ELEVATION_ANGLE_FLUCTUATION_THRESHOLD = 2
+    ELEVATION_ANGLE_VERTICAL_OFFSET_THRESHOLD = 15
+
+    raws_stare = [
+        raw
+        for raw in raws
+        if abs(next(iter(raw.elevation_angles)) - max_elevation_angle)
+        < ELEVATION_ANGLE_FLUCTUATION_THRESHOLD
+        and abs(next(iter(raw.elevation_angles)) - 90)
+        < ELEVATION_ANGLE_VERTICAL_OFFSET_THRESHOLD
+    ]
 
     if len(raws_stare) == 0:
         raise doppy.exceptions.NoDataError("No data suitable for stare product")
@@ -554,6 +584,8 @@ def _cluster_background_profiles(
     background_signal: npt.NDArray[np.float64], radial_distance: npt.NDArray[np.float64]
 ) -> npt.NDArray[np.int64]:
     default_labels = np.zeros(len(background_signal), dtype=int)
+    if len(background_signal) < 2:
+        return default_labels
     radial_distance_mask = (90 < radial_distance) & (radial_distance < 1500)
 
     normalised_background_signal = background_signal / np.median(

{doppy-0.0.2.dist-info → doppy-0.0.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: doppy
-Version: 0.0.2
+Version: 0.0.3
 Classifier: Development Status :: 4 - Beta
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10

{doppy-0.0.2.dist-info → doppy-0.0.3.dist-info}/RECORD

@@ -1,12 +1,12 @@
-doppy-0.0.2.dist-info/METADATA,sha256=4aQ594EyPD3__rvNn-ptVFg2C8Mii5c5mnUYzMyEktE,1583
-doppy-0.0.2.dist-info/WHEEL,sha256=lD_nV8vaA2uMWYS4vgMNGHE0f6_P8sTCC7Nx6jThods,103
-doppy-0.0.2.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
-doppy-0.0.2.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
-doppy-0.0.2.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.0.3.dist-info/METADATA,sha256=1ohcoPWtZ3frQhXFr64TY_jXkFZx7BwUmSK1pZ-Y3go,1583
+doppy-0.0.3.dist-info/WHEEL,sha256=lD_nV8vaA2uMWYS4vgMNGHE0f6_P8sTCC7Nx6jThods,103
+doppy-0.0.3.dist-info/entry_points.txt,sha256=9b_Ca7vJoh6AwL3W8qAPh_UmJ_1Pa6hi-TDfCTDjvSk,43
+doppy-0.0.3.dist-info/license_files/LICENSE,sha256=V-0iroMNMI8ctnLgUau1kdFvwhkYhr9vi-5kWKxw2wc,1089
+doppy-0.0.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/__init__.py,sha256=QtS5Eq0KGUYkaylDGAW8Oz6OIlab4WQYd5fTTUshTbc,59
-doppy/product/stare.py,sha256=04rAnG6bSsdyFSe43jONtthvtQn9S1RBsQ3DY8iqzjA,18581
+doppy/product/stare.py,sha256=b-7P9NjP-KJKrDWcaXxM-fBNiBnHUB09ceosWXv7jPU,19664
 doppy/bench.py,sha256=iVNYveMVGGRES2oe3Orsn31jQFCKTXOmxRFuFiJ8_OA,248
 doppy/netcdf.py,sha256=NW5zTW2OY-M4ApMzgxwN7ysmL0LRRWOfyKFPinCw6gg,3403
 doppy/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -21,5 +21,5 @@ doppy/raw/__init__.py,sha256=GrBbsTkoD2DH_Xwt6apZGwpT1cCIbfOWHzdQVjAPVh0,151
 doppy/raw/halo_bg.py,sha256=kO03yGlKS-DpMMGHYuy_BuidyeUL38TxT5vMn8H_8lE,4809
 doppy/raw/halo_hpl.py,sha256=GNAA3strYeukGKulZ432R8MOXw_JoH3KlRZ9SLDTnoA,17894
 doppy/__main__.py,sha256=zrKQJVj0k0ypBQCGK65Czt9G9FZ_qx3ussw6Q9VJ14g,346
-doppy/rs.abi3.so,sha256=9K58AkzRZsXJHqMSYj6JG2dDjE0uKf4gMAAvNKPye5Q,2606093
-doppy-0.0.2.dist-info/RECORD,,
+doppy/rs.abi3.so,sha256=-NFDTAQTwbRoFELYEYy2-5uiTmiY_yH_P5VjoHbsxvE,2661421
+doppy-0.0.3.dist-info/RECORD,,