+node example

1 files changed, 128 insertions, 0 deletions

diff --git a/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/DeleteNodeAction.cs b/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/DeleteNodeAction.cs
new file mode 100644
index 00000000..2e2e6672
--- /dev/null
+++ b/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/DeleteNodeAction.cs
@@ -0,0 +1,128 @@
+
+using System;
+using System.Linq;
+using System.Collections.Generic;
+using UnityEngine;
+using UNEB.Utility;
+
+namespace UNEB
+{
+    // Each input can have many outputs
+    using InputToOutputPair = Pair<NodeInput, List<NodeOutput>>;
+
+    // Each output can have many inputs
+    using OutputToInputsPair = Pair<NodeOutput, List<NodeInput>>;
+
+    public class DeleteNodeAction : UndoableAction
+    {
+        private NodeGraph _graph;
+        private Node _nodeRemoved = null;
+
+        private List<InputToOutputPair> _oldConnectedOutputs;
+        private List<OutputToInputsPair> _oldConnectedInputs;
+
+        // The node referenced can only be destroyed if the 
+        // delete action has been done or redone.
+        private bool _bCanDeleteNode = false;
+
+        public DeleteNodeAction()
+        {
+
+            _oldConnectedOutputs = new List<InputToOutputPair>();
+            _oldConnectedInputs = new List<OutputToInputsPair>();
+        }
+
+        public override bool Init()
+        {
+            return manager.window.state.selectedNode != null;
+        }
+
+        public override void Do()
+        {
+            _graph = manager.window.graph;
+            _nodeRemoved = manager.window.state.selectedNode;
+            _graph.Remove(_nodeRemoved);
+
+            // Remember all the old outputs the inputs were connected to.
+            foreach (var input in _nodeRemoved.Inputs) {
+
+                if (input.HasOutputConnected()) {
+                    _oldConnectedOutputs.Add(new InputToOutputPair(input, input.Outputs.ToList()));
+                }
+            }
+
+            // Remember all the old input connections that the outputs were connected to.
+            foreach (var output in _nodeRemoved.Outputs) {
+
+                if (output.InputCount != 0) {
+                    _oldConnectedInputs.Add(new OutputToInputsPair(output, output.Inputs.ToList()));
+                }
+            }
+
+            disconnectOldConnections();
+
+            _bCanDeleteNode = true;
+        }
+
+        public override void Undo()
+        {
+            _graph.Add(_nodeRemoved);
+            reconnectOldConnections();
+
+            _bCanDeleteNode = false;
+        }
+
+        public override void Redo()
+        {
+            _graph.Remove(_nodeRemoved);
+            disconnectOldConnections();
+
+            _bCanDeleteNode = true;
+        }
+
+        private void disconnectOldConnections()
+        {
+            // For all the outputs for this node, remove all the connected inputs.
+            foreach (var output in _nodeRemoved.Outputs) {
+                output.RemoveAll();
+            }
+
+            // For all the inputs for this node, remove all the connected outputs.
+            foreach (var input in _nodeRemoved.Inputs) {
+                input.RemoveAll();
+            }
+        }
+
+        private void reconnectOldConnections()
+        {
+            // For all the remembered inputs (of this node) to output pairs, reconnect.
+            foreach (InputToOutputPair inOutPair in _oldConnectedOutputs) {
+
+                NodeInput input = inOutPair.item1;
+                List<NodeOutput> outputs = inOutPair.item2;
+
+                foreach (var output in outputs) {
+                    output.Add(input);
+                }
+            }
+
+            // For all the remembered outputs (of this node) to inputs, reconnect.
+            foreach (OutputToInputsPair outInsPair in _oldConnectedInputs) {
+
+                NodeOutput output = outInsPair.item1;
+                List<NodeInput> inputs = outInsPair.item2;
+
+                foreach (var input in inputs) {
+                    output.Add(input);
+                }
+            }
+        }
+
+        public override void OnDestroy()
+        {
+            if (_bCanDeleteNode && _nodeRemoved) {
+                ScriptableObject.DestroyImmediate(_nodeRemoved, true);
+            }
+        }
+    }
+}
\ No newline at end of file