Also adding zeta search in _search_indices_vertical_s

OceanParcels · Jan 29, 2025 · 762d911 · 762d911
1 parent c0122d2
commit 762d911
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/parcels/field.py b/parcels/field.py
@@ -1091,6 +1091,12 @@ def _search_indices_vertical_s(
             else:
                 zi = depth_indices.argmin() - 1 if z < depth_vector[0] else 0
         zeta = (z - depth_vector[zi]) / (depth_vector[zi + 1] - depth_vector[zi])
+        while zeta > 1:
+            zi += 1
+            zeta = (z - depth_vector[zi]) / (depth_vector[zi + 1] - depth_vector[zi])
+        while zeta < 0:
+            zi -= 1
+            zeta = (z - depth_vector[zi]) / (depth_vector[zi + 1] - depth_vector[zi])
         return (zi, zeta)
 
     @deprecated_made_private  # TODO: Remove 6 months after v3.1.0