diff --git a/Scripts/Editor/NodeEditorGUILayout.cs b/Scripts/Editor/NodeEditorGUILayout.cs
index 19e9ca5..1e17f4a 100644
--- a/Scripts/Editor/NodeEditorGUILayout.cs
+++ b/Scripts/Editor/NodeEditorGUILayout.cs
@@ -165,8 +165,7 @@ namespace XNodeEditor {
rect.size = new Vector2(16, 16);
- NodeEditor editor = NodeEditor.GetEditor(port.node, NodeEditorWindow.current);
- Color backgroundColor = editor.GetTint();
+ Color backgroundColor = NodeEditorWindow.current.graphEditor.GetPortBackgroundColor(port);
Color col = NodeEditorWindow.current.graphEditor.GetPortColor(port);
DrawPortHandle(rect, backgroundColor, col);
@@ -219,8 +218,7 @@ namespace XNodeEditor {
Rect rect = new Rect(position, new Vector2(16, 16));
- NodeEditor editor = NodeEditor.GetEditor(port.node, NodeEditorWindow.current);
- Color backgroundColor = editor.GetTint();
+ Color backgroundColor = NodeEditorWindow.current.graphEditor.GetPortBackgroundColor(port);
Color col = NodeEditorWindow.current.graphEditor.GetPortColor(port);
DrawPortHandle(rect, backgroundColor, col);
@@ -246,8 +244,7 @@ namespace XNodeEditor {
rect.size = new Vector2(16, 16);
- NodeEditor editor = NodeEditor.GetEditor(port.node, NodeEditorWindow.current);
- Color backgroundColor = editor.GetTint();
+ Color backgroundColor = NodeEditorWindow.current.graphEditor.GetPortBackgroundColor(port);
Color col = NodeEditorWindow.current.graphEditor.GetPortColor(port);
DrawPortHandle(rect, backgroundColor, col);
diff --git a/Scripts/Editor/NodeGraphEditor.cs b/Scripts/Editor/NodeGraphEditor.cs
index a8bde28..8b671e2 100644
--- a/Scripts/Editor/NodeGraphEditor.cs
+++ b/Scripts/Editor/NodeGraphEditor.cs
@@ -4,10 +4,12 @@ using System.Linq;
using UnityEditor;
using UnityEngine;
-namespace XNodeEditor {
+namespace XNodeEditor
+{
/// Base class to derive custom Node Graph editors from. Use this to override how graphs are drawn in the editor.
[CustomNodeGraphEditor(typeof(XNode.NodeGraph))]
- public class NodeGraphEditor : XNodeEditor.Internal.NodeEditorBase {
+ public class NodeGraphEditor : XNodeEditor.Internal.NodeEditorBase
+ {
[Obsolete("Use window.position instead")]
public Rect position { get { return window.position; } set { window.position = value; } }
/// Are we currently renaming a node?
@@ -17,28 +19,32 @@ namespace XNodeEditor {
/// Called when opened by NodeEditorWindow
public virtual void OnOpen() { }
-
+
/// Called when NodeEditorWindow gains focus
public virtual void OnWindowFocus() { }
/// Called when NodeEditorWindow loses focus
public virtual void OnWindowFocusLost() { }
- public virtual Texture2D GetGridTexture() {
+ public virtual Texture2D GetGridTexture()
+ {
return NodeEditorPreferences.GetSettings().gridTexture;
}
- public virtual Texture2D GetSecondaryGridTexture() {
+ public virtual Texture2D GetSecondaryGridTexture()
+ {
return NodeEditorPreferences.GetSettings().crossTexture;
}
/// Return default settings for this graph type. This is the settings the user will load if no previous settings have been saved.
- public virtual NodeEditorPreferences.Settings GetDefaultPreferences() {
+ public virtual NodeEditorPreferences.Settings GetDefaultPreferences()
+ {
return new NodeEditorPreferences.Settings();
}
/// Returns context node menu path. Null or empty strings for hidden nodes.
- public virtual string GetNodeMenuName(Type type) {
+ public virtual string GetNodeMenuName(Type type)
+ {
//Check if type has the CreateNodeMenuAttribute
XNode.Node.CreateNodeMenuAttribute attrib;
if (NodeEditorUtilities.GetAttrib(type, out attrib)) // Return custom path
@@ -48,7 +54,8 @@ namespace XNodeEditor {
}
/// The order by which the menu items are displayed.
- public virtual int GetNodeMenuOrder(Type type) {
+ public virtual int GetNodeMenuOrder(Type type)
+ {
//Check if type has the CreateNodeMenuAttribute
XNode.Node.CreateNodeMenuAttribute attrib;
if (NodeEditorUtilities.GetAttrib(type, out attrib)) // Return custom path
@@ -58,10 +65,12 @@ namespace XNodeEditor {
}
/// Add items for the context menu when right-clicking this node. Override to add custom menu items.
- public virtual void AddContextMenuItems(GenericMenu menu) {
+ public virtual void AddContextMenuItems(GenericMenu menu)
+ {
Vector2 pos = NodeEditorWindow.current.WindowToGridPosition(Event.current.mousePosition);
var nodeTypes = NodeEditorReflection.nodeTypes.OrderBy(type => GetNodeMenuOrder(type)).ToArray();
- for (int i = 0; i < nodeTypes.Length; i++) {
+ for (int i = 0; i < nodeTypes.Length; i++)
+ {
Type type = nodeTypes[i];
//Get node context menu path
@@ -71,14 +80,16 @@ namespace XNodeEditor {
// Check if user is allowed to add more of given node type
XNode.Node.DisallowMultipleNodesAttribute disallowAttrib;
bool disallowed = false;
- if (NodeEditorUtilities.GetAttrib(type, out disallowAttrib)) {
+ if (NodeEditorUtilities.GetAttrib(type, out disallowAttrib))
+ {
int typeCount = target.nodes.Count(x => x.GetType() == type);
if (typeCount >= disallowAttrib.max) disallowed = true;
}
// Add node entry to context menu
if (disallowed) menu.AddItem(new GUIContent(path), false, null);
- else menu.AddItem(new GUIContent(path), false, () => {
+ else menu.AddItem(new GUIContent(path), false, () =>
+ {
XNode.Node node = CreateNode(type, pos);
NodeEditorWindow.current.AutoConnect(node);
});
@@ -93,11 +104,13 @@ namespace XNodeEditor {
/// Returned gradient is used to color noodles
/// The output this noodle comes from. Never null.
/// The output this noodle comes from. Can be null if we are dragging the noodle.
- public virtual Gradient GetNoodleGradient(XNode.NodePort output, XNode.NodePort input) {
+ public virtual Gradient GetNoodleGradient(XNode.NodePort output, XNode.NodePort input)
+ {
Gradient grad = new Gradient();
// If dragging the noodle, draw solid, slightly transparent
- if (input == null) {
+ if (input == null)
+ {
Color a = GetTypeColor(output.ValueType);
grad.SetKeys(
new GradientColorKey[] { new GradientColorKey(a, 0f) },
@@ -105,11 +118,13 @@ namespace XNodeEditor {
);
}
// If normal, draw gradient fading from one input color to the other
- else {
+ else
+ {
Color a = GetTypeColor(output.ValueType);
Color b = GetTypeColor(input.ValueType);
// If any port is hovered, tint white
- if (window.hoveredPort == output || window.hoveredPort == input) {
+ if (window.hoveredPort == output || window.hoveredPort == input)
+ {
a = Color.Lerp(a, Color.white, 0.8f);
b = Color.Lerp(b, Color.white, 0.8f);
}
@@ -124,34 +139,48 @@ namespace XNodeEditor {
/// Returned float is used for noodle thickness
/// The output this noodle comes from. Never null.
/// The output this noodle comes from. Can be null if we are dragging the noodle.
- public virtual float GetNoodleThickness(XNode.NodePort output, XNode.NodePort input) {
+ public virtual float GetNoodleThickness(XNode.NodePort output, XNode.NodePort input)
+ {
return NodeEditorPreferences.GetSettings().noodleThickness;
}
- public virtual NoodlePath GetNoodlePath(XNode.NodePort output, XNode.NodePort input) {
+ public virtual NoodlePath GetNoodlePath(XNode.NodePort output, XNode.NodePort input)
+ {
return NodeEditorPreferences.GetSettings().noodlePath;
}
- public virtual NoodleStroke GetNoodleStroke(XNode.NodePort output, XNode.NodePort input) {
+ public virtual NoodleStroke GetNoodleStroke(XNode.NodePort output, XNode.NodePort input)
+ {
return NodeEditorPreferences.GetSettings().noodleStroke;
}
/// Returned color is used to color ports
- public virtual Color GetPortColor(XNode.NodePort port) {
+ public virtual Color GetPortColor(XNode.NodePort port)
+ {
return GetTypeColor(port.ValueType);
}
+ /// The returned color is used to color the background of the door.
+ /// Usually used for outer edge effect
+ public virtual Color GetPortBackgroundColor(XNode.NodePort port)
+ {
+ return Color.gray;
+ }
+
/// Returns generated color for a type. This color is editable in preferences
- public virtual Color GetTypeColor(Type type) {
+ public virtual Color GetTypeColor(Type type)
+ {
return NodeEditorPreferences.GetTypeColor(type);
}
/// Override to display custom tooltips
- public virtual string GetPortTooltip(XNode.NodePort port) {
+ public virtual string GetPortTooltip(XNode.NodePort port)
+ {
Type portType = port.ValueType;
string tooltip = "";
tooltip = portType.PrettyName();
- if (port.IsOutput) {
+ if (port.IsOutput)
+ {
object obj = port.node.GetValue(port);
tooltip += " = " + (obj != null ? obj.ToString() : "null");
}
@@ -159,12 +188,14 @@ namespace XNodeEditor {
}
/// Deal with objects dropped into the graph through DragAndDrop
- public virtual void OnDropObjects(UnityEngine.Object[] objects) {
+ public virtual void OnDropObjects(UnityEngine.Object[] objects)
+ {
if (GetType() != typeof(NodeGraphEditor)) Debug.Log("No OnDropObjects override defined for " + GetType());
}
/// Create a node and save it in the graph asset
- public virtual XNode.Node CreateNode(Type type, Vector2 position) {
+ public virtual XNode.Node CreateNode(Type type, Vector2 position)
+ {
Undo.RecordObject(target, "Create Node");
XNode.Node node = target.AddNode(type);
Undo.RegisterCreatedObjectUndo(node, "Create Node");
@@ -177,7 +208,8 @@ namespace XNodeEditor {
}
/// Creates a copy of the original node in the graph
- public virtual XNode.Node CopyNode(XNode.Node original) {
+ public virtual XNode.Node CopyNode(XNode.Node original)
+ {
Undo.RecordObject(target, "Duplicate Node");
XNode.Node node = target.CopyNode(original);
Undo.RegisterCreatedObjectUndo(node, "Duplicate Node");
@@ -188,13 +220,16 @@ namespace XNodeEditor {
}
/// Return false for nodes that can't be removed
- public virtual bool CanRemove(XNode.Node node) {
+ public virtual bool CanRemove(XNode.Node node)
+ {
// Check graph attributes to see if this node is required
Type graphType = target.GetType();
XNode.NodeGraph.RequireNodeAttribute[] attribs = Array.ConvertAll(
graphType.GetCustomAttributes(typeof(XNode.NodeGraph.RequireNodeAttribute), true), x => x as XNode.NodeGraph.RequireNodeAttribute);
- if (attribs.Any(x => x.Requires(node.GetType()))) {
- if (target.nodes.Count(x => x.GetType() == node.GetType()) <= 1) {
+ if (attribs.Any(x => x.Requires(node.GetType())))
+ {
+ if (target.nodes.Count(x => x.GetType() == node.GetType()) <= 1)
+ {
return false;
}
}
@@ -202,7 +237,8 @@ namespace XNodeEditor {
}
/// Safely remove a node and all its connections.
- public virtual void RemoveNode(XNode.Node node) {
+ public virtual void RemoveNode(XNode.Node node)
+ {
if (!CanRemove(node)) return;
// Remove the node
@@ -218,18 +254,21 @@ namespace XNodeEditor {
[AttributeUsage(AttributeTargets.Class)]
public class CustomNodeGraphEditorAttribute : Attribute,
- XNodeEditor.Internal.NodeEditorBase.INodeEditorAttrib {
+ XNodeEditor.Internal.NodeEditorBase.INodeEditorAttrib
+ {
private Type inspectedType;
public string editorPrefsKey;
/// Tells a NodeGraphEditor which Graph type it is an editor for
/// Type that this editor can edit
/// Define unique key for unique layout settings instance
- public CustomNodeGraphEditorAttribute(Type inspectedType, string editorPrefsKey = "xNode.Settings") {
+ public CustomNodeGraphEditorAttribute(Type inspectedType, string editorPrefsKey = "xNode.Settings")
+ {
this.inspectedType = inspectedType;
this.editorPrefsKey = editorPrefsKey;
}
- public Type GetInspectedType() {
+ public Type GetInspectedType()
+ {
return inspectedType;
}
}