+node example

1 files changed, 106 insertions, 0 deletions

diff --git a/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/CreateConnection.cs b/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/CreateConnection.cs
new file mode 100644
index 00000000..09d3e521
--- /dev/null
+++ b/Other/NodeEditorExamples/Assets/UNEB/Editor/Actions/CreateConnection.cs
@@ -0,0 +1,106 @@
+
+using System.Linq;
+using System.Collections.Generic;
+
+using UnityEngine;
+using UnityEditor;
+
+namespace UNEB
+{
+    public class CreateConnection : MultiStageAction
+    {
+        private NodeInput _input;
+        private NodeOutput _output;
+
+        // The output of the old node it was connected to.
+        private NodeOutput _oldConnectedOutput;
+
+        // Old inputs of the node.
+        private List<NodeInput> _oldConnectedInputs;
+
+        public override void Do()
+        {
+            manager.window.state.selectedOutput = _output;
+        }
+
+        public override void Undo()
+        {
+            _output.Remove(_input);
+            reconnectOldConnections();
+        }
+
+        public override void Redo()
+        {
+            disconnectOldConnections();
+            _output.Add(_input);
+        }
+
+        private void reconnectOldConnections()
+        {
+            //  Re-connect old connections
+            if (_oldConnectedOutput != null) {
+                _oldConnectedOutput.Add(_input);
+            }
+
+            if (_oldConnectedInputs != null) {
+                foreach (var input in _oldConnectedInputs) {
+                    _output.Add(input);
+                }
+            }
+        }
+
+        private void disconnectOldConnections()
+        {
+            // Remove old connections
+            if (_oldConnectedOutput != null) {
+                _oldConnectedOutput.Remove(_input);
+            }
+
+            if (_oldConnectedInputs != null) {
+                _output.RemoveAll();
+            }
+        }
+
+        public override void OnActionStart()
+        {
+            _output = manager.window.state.selectedOutput;
+        }
+
+        public override bool OnActionEnd()
+        {
+            manager.window.state.selectedOutput = null;
+            manager.window.editor.OnMouseOverInput((input) => { _input = input; });
+
+            // Make the connection.
+            if (_input != null && _output.CanConnectInput(_input)) {
+
+                if (!_output.bCanHaveMultipleConnections)
+                {
+                    _output.RemoveAll();
+                }
+
+                if (!_input.bCanHaveMultipleConnections) {
+                    cacheOldConnections();
+                    disconnectOldConnections();
+                }
+
+                return _output.Add(_input);
+            }
+
+            return false;
+        }
+
+        private void cacheOldConnections()
+        {
+            // Check if the receiving node was already connected.
+            if (_input != null && _input.HasOutputConnected()) {
+                _oldConnectedOutput = _input.Outputs[0];
+            }
+
+            // Check if the origin node already had inputs
+            if (!_output.bCanHaveMultipleConnections && _output.InputCount > 0) {
+                _oldConnectedInputs = _output.Inputs.ToList();
+            }
+        }
+    }
+}
\ No newline at end of file