Fix an issue with wire route callbacks being cleared on reload

Algoryx · Jan 21, 2025 · b4fd0f4 · b4fd0f4
1 parent 76b86c0
commit b4fd0f4
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/AGXUnity/WireRoute.cs b/AGXUnity/WireRoute.cs
@@ -18,7 +18,7 @@ namespace AGXUnity
   /// Wire.RouteNode myNode = route.FirstOrDefault( node => node.Frame == thisFrame );
   /// </example>
   [Serializable]
-  public class WireRoute : Route<WireRouteNode>
+  public class WireRoute : Route<WireRouteNode>, ISerializationCallbackReceiver
   {
     /// <summary>
     /// Checks validity of current route.
@@ -177,5 +177,16 @@ private void OnRemovedFromList( WireRouteNode node, int index )
     {
       node.Wire = null;
     }
+
+    public void OnBeforeSerialize() { }
+
+    public void OnAfterDeserialize()
+    {
+      // Remove callback if it exists to avoid any possible duplicates
+      OnNodeAdded   -= this.OnAddedToList;
+      OnNodeRemoved -= this.OnRemovedFromList;
+      OnNodeAdded   += this.OnAddedToList;
+      OnNodeRemoved += this.OnRemovedFromList;
+    }
   }
 }