1 files changed, 157 insertions, 0 deletions
diff --git a/Other/NodeEditorExamples/Assets/UNEB/NodeOutput.cs b/Other/NodeEditorExamples/Assets/UNEB/NodeOutput.cs
new file mode 100644
index 00000000..4a64ec73
--- /dev/null
+++ b/Other/NodeEditorExamples/Assets/UNEB/NodeOutput.cs
@@ -0,0 +1,157 @@
+
+using System.Linq;
+using System.Collections.Generic;
+using UnityEngine;
+
+namespace UNEB
+{
+    public class NodeOutput : NodeConnection
+    {
+        [SerializeField]
+        private bool _bCanHaveMultipleConnections;
+
+        [SerializeField]
+        private List<NodeInput> _inputs = new List<NodeInput>();
+
+        private GUIStyle _style;
+
+        public virtual void Init(Node parent, bool multipleConnections = true)
+        {
+            base.Init(parent);
+            _bCanHaveMultipleConnections = multipleConnections;
+        }
+
+        public List<NodeInput> Inputs
+        {
+            get { return _inputs; }
+        }
+
+        public int InputCount
+        {
+            get { return _inputs.Count; }
+        }
+
+        public NodeInput GetInput(int index)
+        {
+            return _inputs[index];
+        }
+
+        /// <summary>
+        /// Returns true if the connection was added successfully.
+        /// </summary>
+        /// <param name="input"></param>
+        /// <returns></returns>
+        public bool Add(NodeInput input)
+        {
+            if (!CanConnectInput(input)) {
+                return false;
+            }
+
+            // The input cannot be connected to anything.
+            // This test is not inside the CanConnectInput() because
+            // an action can remove the old connections automatically to make it
+            // easier for the user to change connections between nodes.
+            if (input.HasOutputConnected() && !input.bCanHaveMultipleConnections) {
+
+                // Changes to the inputs need to be properly handled by the Action system,
+                // so it works with undo.
+                Debug.LogWarning("Cannot add an input that is already connected");
+                return false;
+            }
+
+            input.Connect(this);
+            _inputs.Add(input);
+
+            parentNode.OnNewInputConnection(input);
+
+            return true;
+        }
+
+        /// <summary>
+        /// Tests to see if the output can be connected to the input.
+        /// </summary>
+        /// <param name="input"></param>
+        /// <returns></returns>
+        public bool CanConnectInput(NodeInput input)
+        {
+            if (input == null) {
+                Debug.LogError("Attempted to add a null input.");
+                return false;
+            }
+
+            // Avoid self-connecting
+            if (parentNode.Inputs.Contains(input)) {
+                Debug.LogWarning("Cannot self connect.");
+                return false;
+            }
+
+            // Avoid connecting when it is already connected.
+            if (input.Outputs.Contains(this)) {
+                Debug.LogWarning("Already Connected");
+                return false;
+            }
+
+            return true;
+        }
+
+        public void Remove(NodeInput input)
+        {
+            if (_inputs.Remove(input)) {
+                parentNode.OnInputConnectionRemoved(input);
+                input.Disconnect(this);
+            }
+        }
+
+        public void RemoveAll()
+        {
+            // Cache the inputs since in order to disconnect them,
+            // they must be removed from _inputs List.
+            var inputs = new List<NodeInput>(_inputs);
+
+            _inputs.Clear();
+
+            foreach (NodeInput input in inputs) {
+                parentNode.OnInputConnectionRemoved(input);
+                input.Disconnect(this);
+            }
+        }
+
+        public override GUIStyle GetStyle()
+        {
+            if (_style == null) {
+                
+                _style = new GUIStyle();
+
+                _style.fixedHeight = kMinSize.y + Node.kKnobOffset;
+                _style.alignment = TextAnchor.MiddleRight;
+                _style.normal.textColor = Color.white * 0.9f;
+                _style.padding.right = (int)kMinHalfSize.x + 5;
+            }
+
+            return _style;
+        }
+
+        /// <summary>
+        /// The output is anchored at the right side of the node.
+        /// </summary>
+        /// <returns></returns>
+        public override float GetNodeAnchor()
+        {
+            return parentNode.bodyRect.xMax;
+        }
+
+        public bool bCanHaveMultipleConnections
+        {
+            get { return _bCanHaveMultipleConnections; }
+        }
+
+        public static NodeOutput Create(Node parent, bool multipleConnections = true)
+        {
+            var output = ScriptableObject.CreateInstance<NodeOutput>();
+            output.Init(parent, multipleConnections);
+
+            NodeConnection.OnConnectionCreated(output);
+            return output;
+        }
+    }
+}
+\ No newline at end of file