- Another couple of tries for the channels

Author: Leguark

gempy_plugins/orientations_generator/_orientations_generator.py

@@ -12,8 +12,8 @@ class NearestSurfacePointsSearcher(enum.Enum):
 
 
 def select_nearest_surfaces_points(surface_points_xyz: np.ndarray, searchcrit: Optional[int|float] = 3,
-                                   search_type: NearestSurfacePointsSearcher = NearestSurfacePointsSearcher.KNN
-                                   ) -> np.ndarray:
+                                   search_type: NearestSurfacePointsSearcher = NearestSurfacePointsSearcher.KNN,
+                                   filter_less_than: Optional[int] = None) -> np.ndarray:
     match search_type:
         case NearestSurfacePointsSearcher.KNN:
             Tree = NearestNeighbors(n_neighbors=searchcrit)
@@ -26,4 +26,7 @@ def select_nearest_surfaces_points(surface_points_xyz: np.ndarray, searchcrit: O
         case _:
             raise ValueError(f"Invalid search type: {search_type}")
 
+    if filter_less_than is not None:
+        neighbours_surfaces = [np.array(neigh) for neigh in neighbours_surfaces if len(neigh) >= filter_less_than]
+        
     return neighbours_surfaces