Propagate tags for edges, not all nodes that have an edge

Change-Id: I42288adf9332ba17f7a43bf94541fa88eebdfac9
project-oak · Mar 4, 2022 · 4a53fdb · 4a53fdb
1 parent 1b9714a
commit 4a53fdb
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ibis/src/recipes.rs b/ibis/src/recipes.rs
@@ -104,11 +104,11 @@ ibis! {
         KnownType(apply!(y_generic, y_arg));
 
     HasTag(s, n, n, tag) <- Solution(s), Claim(n, tag);
-    HasTag(s, source, down, tag) <- // Propagate tags 'downstream'
-        Node(_down_particle, down, _, _),
+    HasTag(s, source, *down, tag) <- // Propagate tags 'downstream'
         HasTag(s, source, curr, tag),
-        !TrustedToRemoveTag(down, tag),
-        (s.has_edge(curr, down));
+        for (up, down) in &s.solution().edges,
+        (*up == curr),
+        !TrustedToRemoveTag(*down, tag);
 
     HasTag(s, source, down, tag) <- // Propagate tags 'across stream' (i.e. inside a particle)
         HasTag(s, source, curr, tag),