commit 7a9aac1ae3f7b836ddb314636d6f50d8ad7ec5d2
author Behdad Esfahbod <behdad@behdad.org> Tue Aug 01 12:05:22 2023 -0600
committer Behdad Esfahbod <behdad@behdad.org> Tue Aug 01 12:17:13 2023 -0600
tree e04a952704e6e28918e145c5223839da33d6eea6
parent 2feac50b40f1dff06655c8efcadcc55088b34dad

[graph] Fixes to parent handling

src/graph/graph.hh

diff --git a/src/graph/graph.hh b/src/graph/graph.hh
index 2b89b99..e464d17 100644
--- a/src/graph/graph.hh
+++ b/src/graph/graph.hh
@@ -199,6 +199,7 @@
 
     void add_parent (unsigned parent_index)
     {
+      assert (parent_index != (unsigned) -1);
       if (incoming_edges_ == 0)
       {
 	single_parent = parent_index;
@@ -207,12 +208,19 @@
       }
       else if (single_parent != (unsigned) -1)
       {
+        assert (incoming_edges_ == 1);
 	if (!parents.set (single_parent, 1))
 	  return;
 	single_parent = (unsigned) -1;
       }
 
-      if (parents.set (parent_index, parents[parent_index] + 1))
+      unsigned *v;
+      if (parents.has (parent_index, &v))
+      {
+        (*v)++;
+	incoming_edges_++;
+      }
+      else if (parents.set (parent_index, 1))
 	incoming_edges_++;
     }
 
@@ -230,7 +238,7 @@
       {
 	incoming_edges_--;
 	if (*v > 1)
-	  *v -= 1;
+	  (*v)--;
 	else
 	  parents.del (parent_index);
 
@@ -284,10 +292,11 @@
         return;
       }
 
-      if (parents.has (old_index))
+      unsigned *v;
+      if (parents.has (old_index, &v))
       {
-	remove_parent (old_index);
-	add_parent (new_index);
+        if (parents.set (new_index, *v))
+	  parents.del (old_index);
       }
     }