///Generate and return task menu
GenericMenu GetMenu(Action <Task> callback)
{
var menu = new GenericMenu();
menu.AddItem(new GUIContent("Open Script"), false, () => { EditorUtils.OpenScriptOfType(task.GetType()); });
menu.AddItem(new GUIContent("Copy"), false, () => { CopyBuffer.SetCache <Task>(task); });
foreach (var _m in task.GetType().RTGetMethods())
{
var m = _m;
var att = m.RTGetAttribute <ContextMenu>(true);
if (att != null)
{
menu.AddItem(new GUIContent(att.menuItem), false, () => { m.Invoke(task, null); });
}
}
menu.AddSeparator("/");
menu.AddItem(new GUIContent("Delete"), false, () =>
{
if (callback != null)
{
UndoUtility.RecordObject(task.ownerSystem.contextObject, "Delete Task");
callback(null);
}
});
return(menu);
}
///<summary>Generate and return task menu</summary>
GenericMenu GetMenu(Action <Task> callback)
{
var menu = new GenericMenu();
var taskType = task.GetType();
menu.AddItem(new GUIContent("Open Script"), false, () => { EditorUtils.OpenScriptOfType(taskType); });
menu.AddItem(new GUIContent("Copy"), false, () => { CopyBuffer.SetCache <Task>(task); });
foreach (var _m in taskType.RTGetMethods())
{
var m = _m;
var att = m.RTGetAttribute <ContextMenu>(true);
if (att != null)
{
menu.AddItem(new GUIContent(att.menuItem), false, () => { m.Invoke(task, null); });
}
}
if (taskType.IsGenericType)
{
menu = EditorUtils.GetPreferedTypesSelectionMenu(taskType.GetGenericTypeDefinition(), (t) => { callback(Task.Create(t, task.ownerSystem)); }, menu, "Change Generic Type");
}
menu.AddSeparator("/");
menu.AddItem(new GUIContent("Delete"), false, () =>
{
if (callback != null)
{
UndoUtility.RecordObject(task.ownerSystem.contextObject, "Delete Task");
callback(null);
}
});
return(menu);
}
///----------------------------------------------------------------------------------------------
///<summary>Graph events AFTER nodes</summary>
static void HandlePostNodesGraphEvents(Graph graph, Vector2 canvasMousePos)
{
//Shortcuts
if (GUIUtility.keyboardControl == 0)
{
if (e.type == EventType.ValidateCommand)
{
if (e.commandName == "Copy" || e.commandName == "Cut" || e.commandName == "Paste" || e.commandName == "SoftDelete" || e.commandName == "Delete" || e.commandName == "Duplicate")
{
e.Use();
}
}
if (e.type == EventType.ExecuteCommand)
{
//COPY/CUT
if (e.commandName == "Copy" || e.commandName == "Cut")
{
List <Node> selection = null;
if (GraphEditorUtility.activeNode != null)
{
selection = new List <Node> {
GraphEditorUtility.activeNode
};
}
if (GraphEditorUtility.activeElements != null && GraphEditorUtility.activeElements.Count > 0)
{
selection = GraphEditorUtility.activeElements.Cast <Node>().ToList();
}
if (selection != null)
{
CopyBuffer.SetCache <Node[]>(Graph.CloneNodes(selection).ToArray());
if (e.commandName == "Cut")
{
foreach (Node node in selection)
{
graph.RemoveNode(node);
}
}
}
e.Use();
}
//PASTE
if (e.commandName == "Paste")
{
if (CopyBuffer.HasCache <Node[]>())
{
TryPasteNodesInGraph(graph, CopyBuffer.GetCache <Node[]>(), canvasMousePos + new Vector2(500, 500) / graph.zoomFactor);
}
e.Use();
}
//DUPLICATE
if (e.commandName == "Duplicate")
{
if (GraphEditorUtility.activeElements != null && GraphEditorUtility.activeElements.Count > 0)
{
TryPasteNodesInGraph(graph, GraphEditorUtility.activeElements.OfType <Node>().ToArray(), default(Vector2));
}
if (GraphEditorUtility.activeNode != null)
{
GraphEditorUtility.activeElement = GraphEditorUtility.activeNode.Duplicate(graph);
}
//Connections can't be duplicated by themselves. They do so as part of multiple node duplication (at least 2).
e.Use();
}
//DELETE
if (e.commandName == "SoftDelete" || e.commandName == "Delete")
{
if (GraphEditorUtility.activeElements != null && GraphEditorUtility.activeElements.Count > 0)
{
foreach (var obj in GraphEditorUtility.activeElements.ToArray())
{
if (obj is Node)
{
graph.RemoveNode(obj as Node);
}
if (obj is Connection)
{
graph.RemoveConnection(obj as Connection);
}
}
GraphEditorUtility.activeElements = null;
}
if (GraphEditorUtility.activeNode != null)
{
graph.RemoveNode(GraphEditorUtility.activeNode);
GraphEditorUtility.activeElement = null;
}
if (GraphEditorUtility.activeConnection != null)
{
graph.RemoveConnection(GraphEditorUtility.activeConnection);
GraphEditorUtility.activeElement = null;
}
e.Use();
}
}
}
//No panel is obscuring
if (GraphEditorUtility.allowClick)
{
if (e.type == EventType.MouseDown && e.clickCount == 2 && e.button == 0)
{
current.maximized = !current.maximized;
e.Use();
}
//Right click or shortcut canvas context menu. Opens browser for adding new nodes.
var isContext = e.type == EventType.ContextClick && !e.alt;
var isShortcut = e.type == EventType.KeyDown && e.keyCode == KeyCode.Space && GUIUtility.keyboardControl == 0 && !e.shift;
if (isContext || isShortcut)
{
GenericMenuBrowser.ShowAsync(e.mousePosition, "Add Node", graph.baseNodeType, () =>
{
var menu = GetAddNodeMenu(graph, canvasMousePos);
if (CopyBuffer.TryGetCache <Node[]>(out Node[] copiedNodes) && copiedNodes.Length > 0)
{
if (copiedNodes[0].GetType().IsSubclassOf(graph.baseNodeType))
{
menu.AddSeparator("/");
var suffix = copiedNodes.Length == 1 ? copiedNodes[0].GetType().FriendlyName() : copiedNodes.Length.ToString();
if (copiedNodes.Length > 0)
{
menu.AddItem(new GUIContent(string.Format("Paste Node(s) ({0})", suffix)), false, () => { TryPasteNodesInGraph(graph, copiedNodes, canvasMousePos); });
}
}
}
return(menu);
});