fix: color in source location in feature map

seung-lab · Apr 11, 2024 · 435051b · 435051b
1 parent c3e1215
commit 435051b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/dijkstra3d.hpp b/dijkstra3d.hpp
@@ -1920,7 +1920,9 @@ OUT* edf_with_feature_map(
   uint64_t src = 1;
   for (uint64_t source : sources) {
     dist[source] = -0;
-    queue.emplace(0.0, source, src++);
+    feature_map[source] = src;
+    queue.emplace(0.0, source, src);
+    src++;
   }
 
   uint64_t loc, next_loc;