feat: calculate layout on node changed

Author: SolarianZ

Editor/Scripts/GraphView/PlayableGraphView.cs

@@ -56,7 +56,7 @@ private void ClearView()
         {
             foreach (var playableOutputNode in _rootOutputNodes)
             {
-                playableOutputNode.RemoveFromContainer();
+                playableOutputNode.RemoveFromView();
             }
 
             _rootOutputNodes.Clear();
@@ -88,7 +88,7 @@ private void UpdateView()
 
                 // create new node
                 var playableOutputNode = PlayableOutputNodeFactory.CreateNode(playableOutput);
-                playableOutputNode.AddToContainer(this);
+                playableOutputNode.AddToView(this, null);
                 playableOutputNode.AddFlag(NodeFlag.Active);
 
                 _rootOutputNodes.Add(playableOutputNode);
@@ -100,7 +100,7 @@ private void UpdateView()
                 var rootOutputNode = _rootOutputNodes[i];
                 if (!rootOutputNode.CheckFlag(NodeFlag.Active))
                 {
-                    rootOutputNode.RemoveFromContainer();
+                    rootOutputNode.RemoveFromView();
 
                     _rootOutputNodes.RemoveAt(i);
                     continue;

Editor/Scripts/GraphView/PlayableGraphView.cs.meta

@@ -12,8 +12,10 @@ namespace GBG.PlayableGraphMonitor.Editor.Node
     public enum NodeFlag : uint
     {
         None = 0,
+
         Active = 1 << 0,
-        //Dirty = 1 << 1,
+
+        HierarchyDirty = 1 << 1,
     }
 
     public readonly struct NodeInput
@@ -52,7 +54,7 @@ public abstract class GraphViewNode : UNode
 
         protected List<NodeInput> InternalInputs { get; } = new List<NodeInput>();
 
-        //protected GraphViewNode Parent { get; private set; }
+        protected GraphViewNode Parent { get; private set; }
 
 
         public virtual void Update() { }
@@ -90,13 +92,14 @@ public override void BuildContextualMenu(ContextualMenuPopulateEvent evt)
 
         #region Hierarchy
 
-        public void AddToContainer(UGraphView container)
+        public void AddToView(UGraphView container, GraphViewNode parentNode)
         {
             Container = container;
             Container.AddElement(this);
+            Parent = parentNode;
         }
 
-        public void RemoveFromContainer()
+        public void RemoveFromView()
         {
             // self
             Container.RemoveElement(this);
@@ -106,12 +109,13 @@ public void RemoveFromContainer()
             {
                 var input = InternalInputs[i];
                 Container.RemoveElement(input.Edge);
-                input.Node.RemoveFromContainer();
+                input.Node.RemoveFromView();
             }
 
             InternalInputs.Clear();
 
             Container = null;
+            Parent = null;
         }
 
 
@@ -209,11 +213,25 @@ public bool CheckFlag(NodeFlag flag)
         public void AddFlag(NodeFlag flag)
         {
             _flags |= (uint)flag;
+            OnFlagsChanged();
         }
 
         public void RemoveFlag(NodeFlag flag)
         {
             _flags &= ~(uint)flag;
+            OnFlagsChanged();
+        }
+
+
+        protected virtual void OnFlagsChanged()
+        {
+            if (CheckFlag(NodeFlag.HierarchyDirty))
+            {
+                _hierarchySize = null;
+                RemoveFlag(NodeFlag.HierarchyDirty);
+
+                Parent?.AddFlag(NodeFlag.HierarchyDirty);
+            }
         }
 
         #endregion

Editor/Scripts/Node/GraphViewNode.cs

@@ -62,14 +62,16 @@ public override void Update()
 
                 // create new node
                 var inputPlayableNode = PlayableNodeFactory.CreateNode(inputPlayable);
-                inputPlayableNode.AddToContainer(Container);
+                inputPlayableNode.AddToView(Container, this);
                 inputPlayableNode.AddFlag(NodeFlag.Active);
 
                 var inputPlayableNodeOutputPort = inputPlayableNode.OutputPorts[0];
                 var selfInputPort = InternalInputPorts[i];
                 var edge = selfInputPort.ConnectTo(inputPlayableNodeOutputPort);
                 Container.AddElement(edge);
                 InternalInputs.Add(new NodeInput(edge, inputPlayableNode, i));
+
+                AddFlag(NodeFlag.HierarchyDirty);
             }
 
             // check and update children
@@ -81,9 +83,12 @@ public override void Update()
                 if (!input.Node.CheckFlag(NodeFlag.Active))
                 {
                     Container.RemoveElement(input.Edge);
-                    input.Node.RemoveFromContainer();
+                    input.Node.RemoveFromView();
 
                     InternalInputs.RemoveAt(i);
+
+                    AddFlag(NodeFlag.HierarchyDirty);
+
                     continue;
                 }
 

Editor/Scripts/Node/PlayableNode.cs

@@ -47,14 +47,16 @@ public override void Update()
                 {
                     // create new node
                     var sourcePlayableNode = PlayableNodeFactory.CreateNode(sourcePlayable);
-                    sourcePlayableNode.AddToContainer(Container);
+                    sourcePlayableNode.AddToView(Container, this);
                     sourcePlayableNode.AddFlag(NodeFlag.Active);
 
                     var sourcePlayableOutputPort = sourcePlayableNode.OutputPorts[0];
                     var selfSourcePort = InternalInputPorts[0];
                     var edge = selfSourcePort.ConnectTo(sourcePlayableOutputPort);
                     Container.AddElement(edge);
                     InternalInputs.Add(new NodeInput(edge, sourcePlayableNode, 0));
+
+                    AddFlag(NodeFlag.HierarchyDirty);
                 }
             }
 
@@ -65,9 +67,12 @@ public override void Update()
                 if (!input.Node.CheckFlag(NodeFlag.Active))
                 {
                     Container.RemoveElement(input.Edge);
-                    input.Node.RemoveFromContainer();
+                    input.Node.RemoveFromView();
 
                     InternalInputs.RemoveAt(i);
+
+                    AddFlag(NodeFlag.HierarchyDirty);
+
                     continue;
                 }