UnityGame/Library/PackageCache/com.unity.visualscripting/Editor/VisualScripting.Shared/EmptyGraphWindow.cs

550 lines
16 KiB
C#
Raw Permalink Normal View History

2024-10-27 10:53:47 +03:00
using System;
using System.Collections.Generic;
using System.IO;
using UnityEditor;
using UnityEditor.SceneManagement;
using UnityEngine;
using UnityEngine.SceneManagement;
using UnityObject = UnityEngine.Object;
namespace Unity.VisualScripting
{
internal class DropdownOptions
{
internal bool interactable = true;
internal string label;
internal string tooltip;
internal Action callback;
}
internal class SplitDropdown
{
private bool reset;
private List<DropdownOptions> listOfOptions;
internal SplitDropdown(List<DropdownOptions> listOfOptions)
{
this.listOfOptions = listOfOptions;
}
internal DropdownOptions GetOption(int index)
{
return listOfOptions[index];
}
internal void Reset()
{
reset = true;
}
internal bool Draw(Rect area, string label, ref bool toggleState)
{
GUI.Box(area, string.Empty, EmptyGraphWindow.buttonStyle);
if (GUI.Button(new Rect(area.x, area.y, area.width - area.height, area.height), label,
EmptyGraphWindow.labelStyleButton))
{
toggleState = false;
return true;
}
Rect toggleRect = new Rect(area.x + area.width - area.height - 2, area.y - 1, area.height + 3,
area.height + 3);
toggleState = GUI.Toggle(toggleRect, toggleState, "", EmptyGraphWindow.labelStyleButton);
GUI.Label(toggleRect, EmptyGraphWindow.dropdownIcon, EmptyGraphWindow.titleStyle);
if (toggleState)
{
GUI.BeginGroup(new Rect(area.x, area.y + area.height - 1, area.width, area.height * 2),
EmptyGraphWindow.buttonStyle);
Rect areaChild = new Rect(0, 0, EmptyGraphWindow.buttonWidth,
EmptyGraphWindow.buttonHeight);
GUIContent contentOption;
foreach (DropdownOptions dropdownOption in listOfOptions)
{
EditorGUI.BeginDisabledGroup(!dropdownOption.interactable);
if (dropdownOption.interactable)
{
contentOption = new GUIContent(dropdownOption.label);
}
else
{
contentOption = new GUIContent(dropdownOption.label, dropdownOption.tooltip);
}
if (GUI.Button(areaChild, contentOption, EmptyGraphWindow.labelStyleDropdownOptions))
{
toggleState = false;
dropdownOption.callback();
}
EditorGUI.EndDisabledGroup();
areaChild.y += areaChild.height;
}
GUI.EndGroup();
}
if (reset)
{
toggleState = false;
reset = false;
}
return false;
}
}
internal class EmptyGraphWindow : EditorWindow
{
internal static GUIContent dropdownIcon;
private bool toggleOnScript = false;
private bool toggleOnState = false;
internal static GUIStyle buttonStyle;
internal static GUIStyle labelStyleButton;
internal static GUIStyle labelStyleDropdownOptions;
internal static GUIStyle titleStyle;
internal static int titleHeight = 120;
internal static int buttonWidth = 200;
internal static int buttonHeight = 22;
private bool shouldCloseWindow;
private Vector2 scrollPosition;
private Rect scrollArea;
private SplitDropdown splitDropdownScriptGraph;
private SplitDropdown splitDropdownStateGraph;
const string k_OnSelectedGameObject = "...on selected GameObject";
const string k_OnNewGameObject = "...on new GameObject";
const string k_SelectedGameObject = "Please, select a GameObject";
[MenuItem("Window/Visual Scripting/Visual Scripting Graph", false, 3010)]
private static void ShowWindow()
{
EmptyGraphWindow window = GetWindow<EmptyGraphWindow>();
window.titleContent = new GUIContent("Visual Scripting");
}
private void OnEnable()
{
string pathRoot = PathUtility.GetPackageRootPath();
UnityObject icon = EditorGUIUtility.Load(Path.Combine(pathRoot,
"Editor/VisualScripting.Shared/EditorAssetResources/SplitButtonArrow.png"));
dropdownIcon = new GUIContent(icon as Texture2D);
scrollArea = new Rect(0, 0, 630, 300);
toggleOnScript = false;
toggleOnState = false;
shouldCloseWindow = false;
var listOfOptions = new List<DropdownOptions>
{
new DropdownOptions
{
label = k_OnSelectedGameObject,
tooltip = k_SelectedGameObject,
callback = CreateScriptGraphOnSelectedGameObject
},
new DropdownOptions
{
label = k_OnNewGameObject,
callback = CreateScriptGraphOnNewGameObject
}
};
splitDropdownScriptGraph = new SplitDropdown(listOfOptions);
listOfOptions = new List<DropdownOptions>
{
new DropdownOptions
{
label = k_OnSelectedGameObject,
tooltip = k_SelectedGameObject,
callback = CreateStateGraphOnSelectedGameObject
},
new DropdownOptions
{
label = k_OnNewGameObject,
callback = CreateStateGraphOnNewGameObject
}
};
splitDropdownStateGraph = new SplitDropdown(listOfOptions);
}
private void OpenGraphAsset(UnityObject unityObject, bool shouldSetSceneAsDirty)
{
shouldCloseWindow = true;
ScriptGraphAsset scriptGraphAsset = unityObject as ScriptGraphAsset;
GraphReference graphReference = null;
if (scriptGraphAsset != null)
{
graphReference = GraphReference.New(scriptGraphAsset, true);
}
else
{
StateGraphAsset stateGraphAsset = unityObject as StateGraphAsset;
if (stateGraphAsset != null)
{
graphReference = GraphReference.New(stateGraphAsset, true);
}
}
if (shouldSetSceneAsDirty)
{
EditorSceneManager.MarkSceneDirty(SceneManager.GetActiveScene());
}
GraphWindow.OpenActive(graphReference);
}
private void OpenGraphFromPath(string path, bool shouldSetSceneAsDirty)
{
path = path.Replace(Application.dataPath, "Assets");
UnityObject unityObject = AssetDatabase.LoadAssetAtPath(path, typeof(UnityObject));
OpenGraphAsset(unityObject, shouldSetSceneAsDirty);
}
private void OpenGraph()
{
EditorGUIUtility.ShowObjectPicker<MacroScriptableObject>(null, false, String.Empty,
EditorGUIUtility.GetControlID(FocusType.Passive));
}
private bool CreateScriptGraphAsset(GameObject gameObject = null, bool updateName = false)
{
var path = EditorUtility.SaveFilePanelInProject("Save Graph", "New Script Graph", "asset", null);
if (string.IsNullOrEmpty(path))
{
return false;
}
VSUsageUtility.isVisualScriptingUsed = true;
var macro = (IMacro)CreateInstance(typeof(ScriptGraphAsset));
var macroObject = (UnityObject)macro;
macro.graph = FlowGraph.WithStartUpdate();
if (gameObject != null)
{
ScriptMachine flowMachine = gameObject.AddComponent<ScriptMachine>();
flowMachine.nest.macro = (ScriptGraphAsset)macro;
}
string filename = Path.GetFileNameWithoutExtension(path);
if (updateName)
{
gameObject.name = filename;
}
macroObject.name = filename;
AssetDatabase.CreateAsset(macroObject, path);
bool shouldSetSceneAsDirty = gameObject != null;
OpenGraphFromPath(path, shouldSetSceneAsDirty);
return true;
}
private void CreateScriptGraph()
{
Selection.activeGameObject = null;
if (CreateScriptGraphAsset())
{
shouldCloseWindow = true;
}
}
private void CreateScriptGraphOnNewGameObject()
{
Selection.activeGameObject = null;
GameObject newGameObject = new GameObject();
if (!CreateScriptGraphAsset(newGameObject, true))
{
DestroyImmediate(newGameObject);
}
}
private void CreateScriptGraphOnSelectedGameObject()
{
if (Selection.activeGameObject != null)
{
if (CreateScriptGraphAsset(Selection.activeGameObject))
{
shouldCloseWindow = true;
}
}
}
private bool CreateStateGraphAsset(GameObject gameObject = null, bool updateName = false)
{
var path = EditorUtility.SaveFilePanelInProject("Save Graph", "New State Graph", "asset", null);
if (string.IsNullOrEmpty(path))
{
return false;
}
VSUsageUtility.isVisualScriptingUsed = true;
var macro = (IMacro)CreateInstance(typeof(StateGraphAsset));
var macroObject = (UnityObject)macro;
macro.graph = StateGraph.WithStart();
if (gameObject != null)
{
StateMachine stateMachine = gameObject.AddComponent<StateMachine>();
stateMachine.nest.macro = (StateGraphAsset)macro;
}
string filename = Path.GetFileNameWithoutExtension(path);
if (updateName)
{
gameObject.name = filename;
}
macroObject.name = filename;
AssetDatabase.CreateAsset(macroObject, path);
bool shouldSetSceneAsDirty = gameObject != null;
OpenGraphFromPath(path, shouldSetSceneAsDirty);
return true;
}
private void CreateStateGraph()
{
Selection.activeGameObject = null;
if (CreateStateGraphAsset())
{
shouldCloseWindow = true;
}
}
private void CreateStateGraphOnNewGameObject()
{
Selection.activeGameObject = null;
GameObject newGameObject = new GameObject();
if (!CreateStateGraphAsset(newGameObject, true))
{
DestroyImmediate(newGameObject);
}
}
private void CreateStateGraphOnSelectedGameObject()
{
if (Selection.activeGameObject != null)
{
if (CreateStateGraphAsset(Selection.activeGameObject))
{
shouldCloseWindow = true;
}
}
}
private void CreateStyles()
{
if (buttonStyle == null)
{
buttonStyle = new GUIStyle("Button");
}
if (labelStyleButton == null)
{
labelStyleButton = new GUIStyle("Label") { alignment = TextAnchor.MiddleCenter };
}
if (labelStyleDropdownOptions == null)
{
labelStyleDropdownOptions = new GUIStyle("ToolbarButton")
{
alignment = TextAnchor.MiddleLeft,
padding = new RectOffset(20, 0, 0, 0)
};
}
if (titleStyle == null)
{
titleStyle = new GUIStyle("Label") { alignment = TextAnchor.MiddleCenter, fontSize = 20 };
}
}
private void ResetToggle()
{
if (Event.current.rawType == EventType.MouseUp)
{
if (toggleOnScript)
{
splitDropdownScriptGraph.Reset();
Repaint();
}
if (toggleOnState)
{
splitDropdownStateGraph.Reset();
Repaint();
}
}
}
private void OpenGraphFromPicker()
{
if (Event.current.commandName == "ObjectSelectorUpdated")
{
UnityObject selectedObject = EditorGUIUtility.GetObjectPickerObject();
OpenGraphAsset(selectedObject, false);
}
}
private void OnGUI()
{
CreateStyles();
ResetToggle();
DropAreaGUI();
OpenGraphFromPicker();
scrollPosition = GUI.BeginScrollView(new Rect(0, 0, position.width, position.height), scrollPosition,
scrollArea);
GUILayout.BeginVertical(GUILayout.ExpandHeight(true));
Vector2 groupSize = new Vector2(scrollArea.width, 280);
GUI.BeginGroup(new Rect((position.width / 2) - (groupSize.x / 2), (position.height / 2) - (groupSize.y / 2),
groupSize.x, groupSize.y));
GUI.Label(new Rect(0, 0, groupSize.x, titleHeight), "Drag and drop a Visual Scripting Graph asset here\nor",
titleStyle);
int buttonX = 10;
if (GUI.Button(new Rect(buttonX, titleHeight, buttonWidth, buttonHeight), "Browse to open a Graph"))
{
OpenGraph();
}
buttonX += (buttonWidth + 10);
Rect area = new Rect(buttonX, titleHeight, buttonWidth, buttonHeight);
splitDropdownScriptGraph.GetOption(0).interactable = Selection.activeGameObject != null;
if (splitDropdownScriptGraph.Draw(area, "Create new Script Graph", ref toggleOnScript))
{
CreateScriptGraph();
}
if (toggleOnScript)
{
toggleOnState = false;
}
buttonX += (buttonWidth + 10);
area = new Rect(buttonX, titleHeight, buttonWidth, buttonHeight);
splitDropdownStateGraph.GetOption(0).interactable = Selection.activeGameObject != null;
if (splitDropdownStateGraph.Draw(area, "Create new State Graph", ref toggleOnState))
{
CreateStateGraph();
}
if (toggleOnState)
{
toggleOnScript = false;
}
GUI.EndGroup();
GUILayout.EndVertical();
GUI.EndScrollView();
if (shouldCloseWindow)
{
Close();
}
}
private void DropAreaGUI()
{
Event currentEvent = Event.current;
Rect activeArea = GUILayoutUtility.GetRect(0, 0, GUILayout.ExpandWidth(true), GUILayout.ExpandHeight(true));
switch (currentEvent.type)
{
case EventType.DragUpdated:
case EventType.DragPerform:
if (!activeArea.Contains(currentEvent.mousePosition))
{
return;
}
DragAndDrop.visualMode = DragAndDropVisualMode.Copy;
if (currentEvent.type == EventType.DragPerform)
{
DragAndDrop.AcceptDrag();
foreach (UnityObject draggedObject in DragAndDrop.objectReferences)
{
ScriptGraphAsset scriptGraphAsset = draggedObject as ScriptGraphAsset;
if (scriptGraphAsset != null)
{
shouldCloseWindow = true;
GraphWindow.OpenActive(GraphReference.New(scriptGraphAsset, true));
break;
}
}
}
break;
}
}
}
}