Last fixes

physycom · Feb 13, 2025 · 796261b · 796261b
1 parent 78efdfb
commit 796261b
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/utils/get_osm_data.py b/utils/get_osm_data.py
@@ -18,6 +18,7 @@
 from pathlib import Path
 from matplotlib import pyplot as plt
 import osmnx as ox
+from tqdm import tqdm
 
 __version__ = "2025.2.13"
 
@@ -181,9 +182,21 @@
     )
     if FULL_GRAPH is not None:
         edge_colors = [
-            RGBA_RED if edge not in GRAPH.edges else RGBA_GRAY
-            for edge in FULL_GRAPH.edges
+            (
+                RGBA_GRAY
+                if any(
+                    "geometry" in GRAPH.edges[g_edge]
+                    and "geometry" in FULL_GRAPH.edges[full_edge]
+                    and GRAPH.edges[g_edge]["geometry"].contains(
+                        FULL_GRAPH.edges[full_edge]["geometry"]
+                    )
+                    for g_edge in GRAPH.edges
+                )
+                else RGBA_RED
+            )
+            for full_edge in tqdm(FULL_GRAPH.edges)
         ]
+
         node_colors = [
             RGBA_RED if node not in GRAPH.nodes else RGBA_WHITE
             for node in FULL_GRAPH.nodes