show unconnected nodes on map

1 files changed, 8 insertions, 1 deletions

diff --git a/store.py b/store.py
index f98bb25..c74415b 100644
--- a/store.py
+++ b/store.py
@@ -79,14 +79,21 @@ def extract_osm_nodes(f_name):
     # find the largest disjoin tree
     best_size = 0
     best_tree = None
+    best_root = None
     for root in forest:
         tree = forest[root]
         size = len(tree)
         if size > best_size:
             best_size = size
             best_tree = tree
+            best_root = root
 
-    return best_tree
+    unconnected_nodes = []
+    for _, node in nodes.items():
+        if node.find_root().id != best_root:
+            unconnected_nodes.append(node)
+
+    return best_tree, unconnected_nodes
 
 
 def select_nodes_in_rectangle(nodes, min_lat, max_lat, min_long, max_long):