/// <summary>
/// Context Click selection. Here you'll need to register your own using a string identifier
/// </summary>
public static void ContextCallback(object obj)
{
callbackObject cbObj = obj as callbackObject;
curNodeCanvas = cbObj.canvas;
curEditorState = cbObj.editor;
switch (cbObj.message)
{
case "deleteNode":
if (curEditorState.focusedNode != null)
{
curEditorState.focusedNode.Delete();
}
break;
case "duplicateNode":
if (curEditorState.focusedNode != null)
{
ContextCallback(new callbackObject(curEditorState.focusedNode.GetID, curNodeCanvas, curEditorState));
Node duplicatedNode = curNodeCanvas.nodes [curNodeCanvas.nodes.Count - 1];
curEditorState.focusedNode = duplicatedNode;
curEditorState.dragNode = true;
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.panWindow = false;
}
break;
case "startTransition":
if (curEditorState.focusedNode != null)
{
curEditorState.makeTransition = curEditorState.focusedNode;
curEditorState.connectOutput = null;
}
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
default:
Vector2 createPos = ScreenToGUIPos(mousePos);
Node node = NodeTypes.getDefaultNode(cbObj.message);
if (node == null)
{
break;
}
bool acceptTransitions = NodeTypes.nodes [node].transitions;
node = node.Create(createPos);
node.InitBase();
NodeEditorCallbacks.IssueOnAddNode(node);
if (curEditorState.connectOutput != null)
{ // If nodeOutput is defined, link it to the first input of the same type
foreach (NodeInput input in node.Inputs)
{
if (Node.CanApplyConnection(curEditorState.connectOutput, input))
{ // If it can connect (type is equals, it does not cause recursion, ...)
Node.ApplyConnection(curEditorState.connectOutput, input);
break;
}
}
}
else if (acceptTransitions && curEditorState.makeTransition != null)
{
Node.CreateTransition(curEditorState.makeTransition, node);
}
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
}
if (NodeEditor.Repaint != null)
{
NodeEditor.Repaint();
}
}
/// <summary>
/// Evaluates context callbacks previously registered
/// </summary>
public static void ContextCallback(object obj)
{
NodeEditorMenuCallback callback = obj as NodeEditorMenuCallback;
if (callback == null)
{
throw new UnityException("Callback Object passed by context is not of type NodeEditorMenuCallback!");
}
curNodeCanvas = callback.canvas;
curEditorState = callback.editor;
switch (callback.message)
{
case "deleteNode": // Delete request
if (curEditorState.focusedNode != null)
{
curEditorState.focusedNode.Delete();
}
break;
case "duplicateNode": // Duplicate request
if (curEditorState.focusedNode != null)
{
ContextCallback(new NodeEditorMenuCallback(curEditorState.focusedNode.GetID, curNodeCanvas, curEditorState));
Node duplicatedNode = curNodeCanvas.nodes [curNodeCanvas.nodes.Count - 1];
curEditorState.focusedNode = duplicatedNode;
curEditorState.dragNode = true;
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.panWindow = false;
}
break;
case "startTransition": // Starting a new transition
if (curEditorState.focusedNode != null)
{
curEditorState.makeTransition = curEditorState.focusedNode;
curEditorState.connectOutput = null;
}
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
default: // Node creation request
Node node = Node.Create(callback.message, ScreenToGUIPos(callback.contextClickPos));
// Handle auto-connection
if (curEditorState.connectOutput != null)
{ // If nodeOutput is defined, link it to the first input of the same type
foreach (NodeInput input in node.Inputs)
{
if (input.CanApplyConnection(curEditorState.connectOutput))
{ // If it can connect (type is equals, it does not cause recursion, ...)
input.ApplyConnection(curEditorState.connectOutput);
break;
}
}
}
else if (node.AcceptsTranstitions && curEditorState.makeTransition != null)
{
Node.CreateTransition(curEditorState.makeTransition, node);
}
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
}
RepaintClients();
}
/// <summary>
/// Processes input events
/// </summary>
public static void InputEvents(List <Rect> ignoreInput)
{
Event e = Event.current;
mousePos = e.mousePosition;
if (OverlayGUI.HasPopupControl())
{
return;
}
bool insideCanvas = curEditorState.canvasRect.Contains(e.mousePosition);
for (int ignoreCnt = 0; ignoreCnt < ignoreInput.Count; ignoreCnt++)
{
if (ignoreInput [ignoreCnt].Contains(e.mousePosition))
{
insideCanvas = false;
break;
}
}
if (!insideCanvas)
{
return;
}
curEditorState.focusedNode = null;
if (insideCanvas && (e.type == EventType.MouseDown || e.type == EventType.MouseUp))
{
curEditorState.focusedNode = NodeEditor.NodeAtPosition(e.mousePosition);
if (e.button == 0)
{
curEditorState.activeNode = curEditorState.focusedNode;
if (Repaint != null)
{
Repaint();
}
}
}
#if UNITY_EDITOR
if (curEditorState.focusedNode != null)
{
UnityEditor.Selection.activeObject = curEditorState.focusedNode;
}
#endif
switch (e.type)
{
case EventType.MouseDown:
curEditorState.dragNode = false;
curEditorState.panWindow = false;
if (curEditorState.focusedNode != null)
{ // A click on a node
if (e.button == 1)
{ // Right click -> Node Context Click
// TODO: Node Editor: Editor-Independancy - GenericMenu conversion
// #if UNITY_EDITOR
// UnityEditor.GenericMenu menu = new UnityEditor.GenericMenu ();
// #else
GenericMenu menu = new GenericMenu();
// #endif
menu.AddItem(new GUIContent("Delete Node"), false, ContextCallback, new callbackObject("deleteNode", curNodeCanvas, curEditorState));
menu.AddItem(new GUIContent("Duplicate Node"), false, ContextCallback, new callbackObject("duplicateNode", curNodeCanvas, curEditorState));
if (NodeTypes.getNodeData(curEditorState.focusedNode).transitions)
{
menu.AddSeparator("Seperator");
menu.AddItem(new GUIContent("Make Transition"), false, ContextCallback, new callbackObject("startTransition", curNodeCanvas, curEditorState));
}
menu.ShowAsContext();
e.Use();
}
else if (e.button == 0)
{
if (!GUIToScreenRect(curEditorState.focusedNode.rect).Contains(e.mousePosition))
{ // Left click at node edges -> Check for clicked connections to edit
NodeOutput nodeOutput = curEditorState.focusedNode.GetOutputAtPos(e.mousePosition);
if (nodeOutput != null)
{ // Output Node -> New Connection drawn from this
curEditorState.connectOutput = nodeOutput;
e.Use();
}
else
{ // no output clicked, check input
NodeInput nodeInput = curEditorState.focusedNode.GetInputAtPos(e.mousePosition);
if (nodeInput != null && nodeInput.connection != null)
{ // Input node -> Loose and edit Connection
curEditorState.connectOutput = nodeInput.connection;
Node.RemoveConnection(nodeInput);
e.Use();
}
}
}
}
}
else
{ // A click on the empty canvas
if (e.button == 2 || e.button == 0)
{ // Left/Middle Click -> Start scrolling
curEditorState.panWindow = true;
e.delta = Vector2.zero;
}
else if (e.button == 1)
{ // Right click -> Editor Context Click
// TODO: Node Editor: Editor-Independancy - GenericMenu conversion
if (curEditorState.connectOutput != null || curEditorState.makeTransition != null)
{
// #if UNITY_EDITOR
// UnityEditor.GenericMenu menu = new UnityEditor.GenericMenu ();
// #else
GenericMenu menu = new GenericMenu();
// #endif
// Iterate through all compatible nodes
foreach (Node node in NodeTypes.nodes.Keys)
{
if (curEditorState.connectOutput != null)
{
foreach (var input in node.Inputs)
{
if (input.type == curEditorState.connectOutput.type)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes[node].adress), false, ContextCallback, new callbackObject(node.GetID, curNodeCanvas, curEditorState));
break;
}
}
}
else if (curEditorState.makeTransition != null && NodeTypes.nodes [node].transitions)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes[node].adress), false, ContextCallback, new callbackObject(node.GetID, curNodeCanvas, curEditorState));
}
}
menu.ShowAsContext();
}
else
{
// #if UNITY_EDITOR
// UnityEditor.GenericMenu menu = new UnityEditor.GenericMenu ();
// #else
GenericMenu menu = new GenericMenu();
// #endif
foreach (Node node in NodeTypes.nodes.Keys)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes [node].adress), false, ContextCallback, new callbackObject(node.GetID, curNodeCanvas, curEditorState));
}
menu.ShowAsContext();
}
e.Use();
}
}
break;
case EventType.MouseUp:
if (curEditorState.focusedNode != null)
{
if (curEditorState.makeTransition != null)
{
Node.CreateTransition(curEditorState.makeTransition, curEditorState.focusedNode);
}
else if (curEditorState.connectOutput != null)
{ // Apply a connection if theres a clicked input
if (!curEditorState.focusedNode.Outputs.Contains(curEditorState.connectOutput))
{ // If an input was clicked, it'll will now be connected
NodeInput clickedInput = curEditorState.focusedNode.GetInputAtPos(e.mousePosition);
if (Node.CanApplyConnection(curEditorState.connectOutput, clickedInput))
{ // If it can connect (type is equals, it does not cause recursion, ...)
Node.ApplyConnection(curEditorState.connectOutput, clickedInput);
}
}
e.Use();
}
}
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
case EventType.ScrollWheel:
curEditorState.zoom = (float)Math.Round(Math.Min(2.0f, Math.Max(0.6f, curEditorState.zoom + e.delta.y / 15)), 2);
if (Repaint != null)
{
Repaint();
}
break;
case EventType.KeyDown:
// TODO: Node Editor: Shortcuts
if (e.keyCode == KeyCode.N) // Start Navigating (curve to origin / active Node)
{
curEditorState.navigate = true;
}
if (e.keyCode == KeyCode.LeftControl && curEditorState.activeNode != null) // Snap
{
curEditorState.activeNode.rect.position = new Vector2(Mathf.RoundToInt((curEditorState.activeNode.rect.position.x - curEditorState.panOffset.x) / 10) * 10 + curEditorState.panOffset.x,
Mathf.RoundToInt((curEditorState.activeNode.rect.position.y - curEditorState.panOffset.y) / 10) * 10 + curEditorState.panOffset.y);
}
if (Repaint != null)
{
Repaint();
}
break;
case EventType.KeyUp:
if (e.keyCode == KeyCode.N) // Stop Navigating
{
curEditorState.navigate = false;
}
if (Repaint != null)
{
Repaint();
}
break;
case EventType.MouseDrag:
if (curEditorState.panWindow)
{ // Scroll everything with the current mouse delta
curEditorState.panOffset += e.delta * curEditorState.zoom;
for (int nodeCnt = 0; nodeCnt < curNodeCanvas.nodes.Count; nodeCnt++)
{
curNodeCanvas.nodes [nodeCnt].rect.position += e.delta * curEditorState.zoom;
}
e.delta = Vector2.zero;
if (Repaint != null)
{
Repaint();
}
}
else
{
curEditorState.panWindow = false;
}
if (curEditorState.dragNode && curEditorState.activeNode != null && GUIUtility.hotControl == 0)
{ // Drag the active node with the current mouse delta
curEditorState.activeNode.rect.position += e.delta * curEditorState.zoom;
NodeEditorCallbacks.IssueOnMoveNode(curEditorState.activeNode);
e.delta = Vector2.zero;
if (Repaint != null)
{
Repaint();
}
}
else
{
curEditorState.dragNode = false;
}
break;
}
}
/// <summary>
/// Processes input events
/// </summary>
public static void InputEvents()
{
Event e = Event.current;
mousePos = e.mousePosition;
bool leftClick = e.button == 0, rightClick = e.button == 1,
mouseDown = e.type == EventType.MouseDown, mousUp = e.type == EventType.MouseUp;
if (ignoreInput(mousePos))
{
return;
}
#region Change Node selection and focus
// Choose focused and selected Node, accounting for focus changes
curEditorState.focusedNode = null;
if (mouseDown || mousUp)
{
curEditorState.focusedNode = NodeEditor.NodeAtPosition(mousePos);
if (curEditorState.focusedNode != curEditorState.selectedNode)
{
unfocusControls = true;
}
if (mouseDown && leftClick)
{
curEditorState.selectedNode = curEditorState.focusedNode;
RepaintClients();
}
}
// Perform above mentioned focus changes in Repaint, which is the only suitable time to do this
if (unfocusControls && Event.current.type == EventType.Repaint)
{
GUIUtility.hotControl = 0;
GUIUtility.keyboardControl = 0;
unfocusControls = false;
}
#if UNITY_EDITOR
if (curEditorState.focusedNode != null)
{
UnityEditor.Selection.activeObject = curEditorState.focusedNode;
}
#endif
#endregion
switch (e.type)
{
case EventType.MouseDown:
curEditorState.dragNode = false;
curEditorState.panWindow = false;
if (curEditorState.focusedNode != null)
{ // Clicked a Node
if (rightClick)
{ // Node Context Click
GenericMenu menu = new GenericMenu();
menu.AddItem(new GUIContent("Delete Node"), false, ContextCallback, new NodeEditorMenuCallback("deleteNode", curNodeCanvas, curEditorState));
menu.AddItem(new GUIContent("Duplicate Node"), false, ContextCallback, new NodeEditorMenuCallback("duplicateNode", curNodeCanvas, curEditorState));
if (curEditorState.focusedNode.AcceptsTranstitions)
{
menu.AddSeparator("Seperator");
menu.AddItem(new GUIContent("Make Transition"), false, ContextCallback, new NodeEditorMenuCallback("startTransition", curNodeCanvas, curEditorState));
}
menu.ShowAsContext();
e.Use();
}
else if (leftClick)
{ // Detect click on a connection knob
if (!CanvasGUIToScreenRect(curEditorState.focusedNode.rect).Contains(mousePos))
{ // Clicked NodeEdge, check Node Inputs and Outputs
NodeOutput nodeOutput = curEditorState.focusedNode.GetOutputAtPos(e.mousePosition);
if (nodeOutput != null)
{ // Output clicked -> New Connection drawn from this
curEditorState.connectOutput = nodeOutput;
e.Use();
return;
}
NodeInput nodeInput = curEditorState.focusedNode.GetInputAtPos(e.mousePosition);
if (nodeInput != null && nodeInput.connection != null)
{ // Input clicked -> Loose and edit Connection
// TODO: Draw input from NodeInput
curEditorState.connectOutput = nodeInput.connection;
nodeInput.RemoveConnection();
e.Use();
}
}
}
}
else
{ // Clicked on canvas
// NOTE: Panning is not done here but in LateEvents, so buttons on the canvas won't be blocked when clicking
if (rightClick)
{ // Editor Context Click
GenericMenu menu = new GenericMenu();
if (curEditorState.connectOutput != null)
{ // A connection is drawn, so provide a context menu with apropriate nodes to auto-connect
foreach (Node node in NodeTypes.nodes.Keys)
{ // Iterate through all nodes and check for compability
for (int inputCnt = 0; inputCnt < node.Inputs.Count; inputCnt++)
{
if (node.Inputs[inputCnt].type == curEditorState.connectOutput.type)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes[node].adress), false, ContextCallback, new NodeEditorMenuCallback(node.GetID, curNodeCanvas, curEditorState));
break;
}
}
}
}
else if (curEditorState.makeTransition != null && curEditorState.makeTransition.AcceptsTranstitions)
{ // A transition is drawn, so provide a context menu with nodes to auto-connect
foreach (Node node in NodeTypes.nodes.Keys)
{ // Iterate through all nodes and check for compability
if (node.AcceptsTranstitions)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes[node].adress), false, ContextCallback, new NodeEditorMenuCallback(node.GetID, curNodeCanvas, curEditorState));
}
}
}
else
{ // Ordinary context click, add all nodes to add
foreach (Node node in NodeTypes.nodes.Keys)
{
menu.AddItem(new GUIContent("Add " + NodeTypes.nodes [node].adress), false, ContextCallback, new NodeEditorMenuCallback(node.GetID, curNodeCanvas, curEditorState));
}
}
menu.ShowAsContext();
e.Use();
}
}
break;
case EventType.MouseUp:
if (curEditorState.focusedNode != null)
{ // Apply Drawn connections/transition on node
if (curEditorState.makeTransition != null)
{ // Apply a connection if theres a clicked input
Node.CreateTransition(curEditorState.makeTransition, curEditorState.focusedNode);
}
if (curEditorState.connectOutput != null)
{ // Apply a connection if theres a clicked input
if (!curEditorState.focusedNode.Outputs.Contains(curEditorState.connectOutput))
{ // An input was clicked, it'll will now be connected
NodeInput clickedInput = curEditorState.focusedNode.GetInputAtPos(e.mousePosition);
if (clickedInput.CanApplyConnection(curEditorState.connectOutput))
{ // It can connect (type is equals, it does not cause recursion, ...)
clickedInput.ApplyConnection(curEditorState.connectOutput);
}
}
e.Use();
}
}
curEditorState.makeTransition = null;
curEditorState.connectOutput = null;
curEditorState.dragNode = false;
curEditorState.panWindow = false;
break;
case EventType.ScrollWheel:
// Apply Zoom
curEditorState.zoom = (float)Math.Round(Math.Min(2.0f, Math.Max(0.6f, curEditorState.zoom + e.delta.y / 15)), 2);
RepaintClients();
break;
case EventType.KeyDown:
// TODO: Node Editor: Shortcuts
if (e.keyCode == KeyCode.N) // Start Navigating (curve to origin / active Node)
{
curEditorState.navigate = true;
}
if (e.keyCode == KeyCode.LeftControl && curEditorState.selectedNode != null)
{ // Snap selected Node's position to multiples of 10
Vector2 pos = curEditorState.selectedNode.rect.position;
pos = (pos - curEditorState.panOffset) / 10;
pos = new Vector2(Mathf.RoundToInt(pos.x), Mathf.RoundToInt(pos.y));
curEditorState.selectedNode.rect.position = pos * 10 + curEditorState.panOffset;
}
RepaintClients();
break;
case EventType.KeyUp:
if (e.keyCode == KeyCode.N) // Stop Navigating
{
curEditorState.navigate = false;
}
RepaintClients();
break;
case EventType.MouseDrag:
if (curEditorState.panWindow)
{ // Scroll everything with the current mouse delta
curEditorState.panOffset += e.delta * curEditorState.zoom;
for (int nodeCnt = 0; nodeCnt < curNodeCanvas.nodes.Count; nodeCnt++)
{
curNodeCanvas.nodes [nodeCnt].rect.position += e.delta * curEditorState.zoom;
}
e.delta = Vector2.zero;
RepaintClients();
}
if (curEditorState.dragNode && curEditorState.selectedNode != null && GUIUtility.hotControl == 0)
{ // Drag the active node with the current mouse delta
curEditorState.selectedNode.rect.position += e.delta * curEditorState.zoom;
NodeEditorCallbacks.IssueOnMoveNode(curEditorState.selectedNode);
e.delta = Vector2.zero;
RepaintClients();
}
else
{
curEditorState.dragNode = false;
}
break;
}
}
