Small fixes

Author: schilling40

flamingo_tools/segmentation/cochlea_mapping.py

@@ -1,4 +1,5 @@
 import math
+import warnings
 from typing import List, Optional, Tuple
 
 import networkx as nx
@@ -58,7 +59,10 @@ def tonotopic_mapping(
     for index, element in zip(labels_subset, centroids_subset):
         coords[index] = element
 
-    _, graph = graph_connected_components(coords, max_edge_distance, min_component_length)
+    components, graph = graph_connected_components(coords, max_edge_distance, min_component_length)
+    if len(components) > 1:
+        warnings.warn(f"There are {len(components)} connected components, expected 1. "
+                      "Check parameters for post-processing (max_edge_distance, min_component_length).")
 
     unfiltered_graph = graph.copy()
 

flamingo_tools/segmentation/postprocessing.py

@@ -352,6 +352,9 @@ def graph_connected_components(coords: dict, max_edge_distance: float, min_compo
                 graph.remove_node(c)
 
     components = [list(s) for s in nx.connected_components(graph)]
+    length_components = [len(c) for c in components]
+    length_components, components = zip(*sorted(zip(length_components, components), reverse=True))
+
     return components, graph
 
 
@@ -414,9 +417,6 @@ def components_sgn(
 
     components, _ = graph_connected_components(coords, max_edge_distance, min_component_length)
 
-    length_components = [len(c) for c in components]
-    length_components, components = zip(*sorted(zip(length_components, components), reverse=True))
-
     # add original coordinates closer to eroded component than threshold
     if postprocess_threshold is not None:
         if postprocess_components is None: