384 lines
14 KiB
C#
384 lines
14 KiB
C#
|
using System;
|
|||
|
|
using System.Collections.Generic;
|
|||
|
|
using System.IO;
|
|||
|
|
using System.Linq;
|
|||
|
|
using Convention.WindowsUI.Variant;
|
|||
|
|
using UnityEngine;
|
|||
|
|
using UnityEngine.EventSystems;
|
|||
|
|
using UnityEngine.InputSystem;
|
|||
|
|
|
|||
|
|
namespace Convention.Workflow
|
|||
|
|
{
|
|||
|
|
[Serializable, ArgPackage]
|
|||
|
|
public class NodeResult
|
|||
|
|
{
|
|||
|
|
public int nodeID;
|
|||
|
|
public string nodeTitle;
|
|||
|
|
public Dictionary<string, string> result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable, ArgPackage]
|
|||
|
|
public class ContextResult
|
|||
|
|
{
|
|||
|
|
public string hashID;
|
|||
|
|
public List<NodeResult> results;
|
|||
|
|
public float progress;
|
|||
|
|
public int task_count;
|
|||
|
|
public List<int> current_running_nodes;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable, ArgPackage]
|
|||
|
|
public class FunctionModel
|
|||
|
|
{
|
|||
|
|
public string name = "unknown";
|
|||
|
|
public string description = "unknown";
|
|||
|
|
public Dictionary<string, string> parameters = new();
|
|||
|
|
public Dictionary<string, string> returns = new();
|
|||
|
|
public string module = "global";
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Serializable, ArgPackage]
|
|||
|
|
public class Workflow
|
|||
|
|
{
|
|||
|
|
public List<NodeInfo> Datas = new();
|
|||
|
|
[NonSerialized] public List<Node> Nodes = new();
|
|||
|
|
public List<FunctionModel> Functions = new();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public class WorkflowManager : MonoSingleton<WorkflowManager>
|
|||
|
|
{
|
|||
|
|
public static float CameraZ = 0;
|
|||
|
|
|
|||
|
|
[Content] public Workflow m_workflow;
|
|||
|
|
public Workflow workflow
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
return m_workflow;
|
|||
|
|
}
|
|||
|
|
//set
|
|||
|
|
//{
|
|||
|
|
// if (m_workflow == value)
|
|||
|
|
// return;
|
|||
|
|
// ClearWorkflowGraph();
|
|||
|
|
// m_workflow = value;
|
|||
|
|
// BuildWorkflowGraph();
|
|||
|
|
//}
|
|||
|
|
}
|
|||
|
|
[Resources, OnlyNotNullMode, SerializeField] private Transform m_CameraTransform;
|
|||
|
|
[Setting] public float ScrollSpeed = 1;
|
|||
|
|
[Setting, OnlyNotNullMode] public ScriptableObject GraphNodePrefabs;
|
|||
|
|
[Setting, HopeNotNull] public ScriptableObject TextLabels;
|
|||
|
|
|
|||
|
|
//[Resources, SerializeField, OnlyNotNullMode, Header("Prefabs")]
|
|||
|
|
//private GameObject GraphNodePrefab;
|
|||
|
|
[Resources, SerializeField, OnlyNotNullMode, Header("Content")]
|
|||
|
|
public RectTransform ContentPlane;
|
|||
|
|
[Resources, SerializeField, OnlyNotNullMode, Header("Mouse Click")]
|
|||
|
|
public RectTransform focusObject;
|
|||
|
|
[SerializeField, OnlyNotNullMode]
|
|||
|
|
public RectTransform UIFocusObject;
|
|||
|
|
private List<SharedModule.CallbackData> callbackDatas = new();
|
|||
|
|
private HashSet<Type> registeredCallbackNodeType = new();
|
|||
|
|
|
|||
|
|
public Dictionary<string, List<FunctionModel>> CallableFunctionModels = new();
|
|||
|
|
|
|||
|
|
|
|||
|
|
public void RegisterFunctionModel([In] FunctionModel func)
|
|||
|
|
{
|
|||
|
|
if (!CallableFunctionModels.ContainsKey(func.module))
|
|||
|
|
CallableFunctionModels[func.module] = new();
|
|||
|
|
CallableFunctionModels[func.module].Add(func);
|
|||
|
|
Debug.Log($"[{String.Join(", ", func.returns.ToList().ConvertAll(x => $"{x.Key}: {x.Value}"))}] {func.name}" +
|
|||
|
|
$"({String.Join(", ", func.parameters.ToList().ConvertAll(x => $"{x.Key}: {x.Value}"))}): {func.description}");
|
|||
|
|
}
|
|||
|
|
public void UnregisterFunctionModel([In] FunctionModel func)
|
|||
|
|
{
|
|||
|
|
CallableFunctionModels[func.module].Remove(func);
|
|||
|
|
}
|
|||
|
|
public List<string> GetAllModuleName()
|
|||
|
|
{
|
|||
|
|
return CallableFunctionModels.Keys.ToList();
|
|||
|
|
}
|
|||
|
|
public List<string> GetAllFunctionName(string module)
|
|||
|
|
{
|
|||
|
|
return CallableFunctionModels[module].ConvertAll(x => x.name);
|
|||
|
|
}
|
|||
|
|
public bool ContainsModule(string module)
|
|||
|
|
{
|
|||
|
|
return CallableFunctionModels.ContainsKey(module);
|
|||
|
|
}
|
|||
|
|
public bool ContainsFunctionModel(string module, string functionName)
|
|||
|
|
{
|
|||
|
|
return ContainsModule(module) && CallableFunctionModels[module].Any(y => y.name == functionName);
|
|||
|
|
}
|
|||
|
|
public List<FunctionModel> GetAllFunctionModel(string module)
|
|||
|
|
{
|
|||
|
|
return CallableFunctionModels[module];
|
|||
|
|
}
|
|||
|
|
[return: ReturnMayNull]
|
|||
|
|
public FunctionModel GetFunctionModel(string module,string functionName)
|
|||
|
|
{
|
|||
|
|
if (ContainsModule(module))
|
|||
|
|
return CallableFunctionModels[module].FirstOrDefault(x => x.name == functionName);
|
|||
|
|
return null;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public delegate bool CustomTransformer([In] string word, out string late);
|
|||
|
|
public List<CustomTransformer> customTransformers = new();
|
|||
|
|
public static string Transformer([In] string str)
|
|||
|
|
{
|
|||
|
|
if (string.IsNullOrEmpty(str))
|
|||
|
|
return str;
|
|||
|
|
if (instance != null)
|
|||
|
|
{
|
|||
|
|
foreach (var customTransformer in instance.customTransformers)
|
|||
|
|
{
|
|||
|
|
if (customTransformer(str, out var result))
|
|||
|
|
{
|
|||
|
|
var menuButton = instance.callbackDatas.Find(x => x.name == str);
|
|||
|
|
if (menuButton != null)
|
|||
|
|
{
|
|||
|
|
menuButton.name = result;
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
if (instance.TextLabels == null)
|
|||
|
|
return str;
|
|||
|
|
if (instance.TextLabels.symbols.ContainsKey(str))
|
|||
|
|
return instance.TextLabels.symbols[str];
|
|||
|
|
}
|
|||
|
|
return str;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void SetupWorkflowGraphNodeType([In] SharedModule.CallbackData callback)
|
|||
|
|
{
|
|||
|
|
callbackDatas.Add(callback);
|
|||
|
|
}
|
|||
|
|
public void SetupWorkflowGraphNodeType(params NodeInfo[] templates)
|
|||
|
|
{
|
|||
|
|
foreach (NodeInfo nodeInfo in templates)
|
|||
|
|
{
|
|||
|
|
if (registeredCallbackNodeType.Contains(nodeInfo.GetType()))
|
|||
|
|
continue;
|
|||
|
|
registeredCallbackNodeType.Add(nodeInfo.GetType());
|
|||
|
|
string label = nodeInfo.GetType().Name;
|
|||
|
|
if (label.EndsWith("Info"))
|
|||
|
|
label = label[..^4];
|
|||
|
|
SetupWorkflowGraphNodeType(new SharedModule.CallbackData(Transformer(label), x =>
|
|||
|
|
{
|
|||
|
|
var info = nodeInfo.TemplateClone();
|
|||
|
|
var node = this.CreateGraphNode(info);
|
|||
|
|
var pos = focusObject.position;
|
|||
|
|
node.transform.position = new Vector2(pos.x, pos.y);
|
|||
|
|
}));
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Start()
|
|||
|
|
{
|
|||
|
|
callbackDatas = new();
|
|||
|
|
Architecture.RegisterWithDuplicateAllow(typeof(WorkflowManager), this, () =>
|
|||
|
|
{
|
|||
|
|
Debug.Log($"{nameof(WorkflowManager)} registered");
|
|||
|
|
if (GraphNodePrefabs == null)
|
|||
|
|
GraphNodePrefabs = Resources.Load<ScriptableObject>("Workflow/Nodes");
|
|||
|
|
SetupWorkflowGraphNodeType(
|
|||
|
|
new TextNodeInfo(),
|
|||
|
|
new ValueNodeInfo(),
|
|||
|
|
new ResourceNodeInfo(),
|
|||
|
|
new StepNodeInfo(),
|
|||
|
|
new EndNodeInfo());
|
|||
|
|
}, typeof(GraphInputWindow), typeof(GraphInspector));
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void Update()
|
|||
|
|
{
|
|||
|
|
if (Keyboard.current[Key.LeftCtrl].isPressed)
|
|||
|
|
{
|
|||
|
|
var t = -Mouse.current.scroll.y.ReadValue() * ScrollSpeed * 0.001f;
|
|||
|
|
var z = m_CameraTransform.transform.localPosition.z;
|
|||
|
|
if (z - t > -100 && z - t < -5)
|
|||
|
|
m_CameraTransform.transform.Translate(new Vector3(0, 0, -t), Space.Self);
|
|||
|
|
}
|
|||
|
|
UIFocusObject.position = Mouse.current.position.ReadValue();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void LateUpdate()
|
|||
|
|
{
|
|||
|
|
CameraZ = Camera.main.transform.position.z;
|
|||
|
|
}
|
|||
|
|
public void RefreshImmediate()
|
|||
|
|
{
|
|||
|
|
foreach (var node in workflow.Nodes)
|
|||
|
|
{
|
|||
|
|
node.RefreshImmediate();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
public void ClearWorkflowGraph()
|
|||
|
|
{
|
|||
|
|
foreach (var node in workflow.Nodes)
|
|||
|
|
{
|
|||
|
|
GameObject.Destroy(node.gameObject);
|
|||
|
|
}
|
|||
|
|
workflow.Nodes.Clear();
|
|||
|
|
workflow.Datas.Clear();
|
|||
|
|
}
|
|||
|
|
public void BuildWorkflowGraph()
|
|||
|
|
{
|
|||
|
|
foreach (var info in workflow.Datas)
|
|||
|
|
{
|
|||
|
|
CreateGraphNode(info);
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public Node CreateGraphNode([In] NodeInfo info, bool isRefresh = true)
|
|||
|
|
{
|
|||
|
|
var node = info.Instantiate();
|
|||
|
|
node.gameObject.SetActive(true);
|
|||
|
|
node.transform.SetParent(ContentPlane);
|
|||
|
|
node.transform.localScale = Vector3.one;
|
|||
|
|
node.transform.eulerAngles = Vector3.zero;
|
|||
|
|
node.SetupFromInfo(info.TemplateClone(), isRefresh);
|
|||
|
|
workflow.Nodes.Add(node);
|
|||
|
|
node.MyNodeTab = GraphInputWindow.instance.RegisterOnHierarchyWindow(node.info);
|
|||
|
|
return node;
|
|||
|
|
}
|
|||
|
|
public void DestroyNode(Node node)
|
|||
|
|
{
|
|||
|
|
if (!workflow.Nodes.Remove(node))
|
|||
|
|
{
|
|||
|
|
throw new InvalidOperationException("node is not in current workflow");
|
|||
|
|
}
|
|||
|
|
GameObject.Destroy(node.gameObject);
|
|||
|
|
}
|
|||
|
|
public bool ContainsNode(int id)
|
|||
|
|
{
|
|||
|
|
if (id < 0)
|
|||
|
|
return false;
|
|||
|
|
return workflow.Nodes.Count < id;
|
|||
|
|
}
|
|||
|
|
public Node GetGraphNode(int id)
|
|||
|
|
{
|
|||
|
|
if (id < 0)
|
|||
|
|
return null;
|
|||
|
|
return workflow.Nodes[id];
|
|||
|
|
}
|
|||
|
|
public int GetGraphNodeID(Node node)
|
|||
|
|
{
|
|||
|
|
if (node == null)
|
|||
|
|
return -1;
|
|||
|
|
return workflow.Nodes.IndexOf(node);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public void SaveWorkflowWithSystemPlugin()
|
|||
|
|
{
|
|||
|
|
var str = PluginExtenion.SaveFile("工作流|*.workflow;*.json", "保存工作流");
|
|||
|
|
if (string.IsNullOrEmpty(str) == false)
|
|||
|
|
SaveWorkflow(str);
|
|||
|
|
}
|
|||
|
|
public void LoadWorkflowWithSystemPlugin()
|
|||
|
|
{
|
|||
|
|
var str = PluginExtenion.SelectFile("工作流|*.workflow;*.json", "加载工作流");
|
|||
|
|
if (string.IsNullOrEmpty(str) == false)
|
|||
|
|
LoadWorkflow(str);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
[Content, OnlyPlayMode] public string LastSavePath = null;
|
|||
|
|
|
|||
|
|
public static string WorkflowFileKey = "workflow";
|
|||
|
|
|
|||
|
|
public ToolFile SaveWorkflow(string workflowPath)
|
|||
|
|
{
|
|||
|
|
LastSavePath = workflowPath;
|
|||
|
|
ToolFile local = new(workflowPath);
|
|||
|
|
ToolFile parent = local.GetParentDir();
|
|||
|
|
if (parent.Exists() == false)
|
|||
|
|
throw new FileNotFoundException($"{parent} is not exist");
|
|||
|
|
workflow.Datas.Clear();
|
|||
|
|
Debug.Log(workflow.Nodes.Count);
|
|||
|
|
foreach (var node in workflow.Nodes)
|
|||
|
|
{
|
|||
|
|
node.info.CopyFromNode(node);
|
|||
|
|
workflow.Datas.Add(node.info);
|
|||
|
|
}
|
|||
|
|
local.SaveAsJson(workflow, WorkflowFileKey);
|
|||
|
|
return local;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
public Workflow LoadWorkflow(Workflow workflow)
|
|||
|
|
{
|
|||
|
|
ClearWorkflowGraph();
|
|||
|
|
workflow.Datas.Sort((x, y) => x.nodeID.CompareTo(y.nodeID));
|
|||
|
|
for (int i = 0; i < workflow.Datas.Count; i++)
|
|||
|
|
{
|
|||
|
|
if (workflow.Datas[i].nodeID != i)
|
|||
|
|
throw new InvalidOperationException("Bad workflow: nodeID != node index");
|
|||
|
|
}
|
|||
|
|
this.m_workflow = new();
|
|||
|
|
foreach (var info in workflow.Datas)
|
|||
|
|
{
|
|||
|
|
var node = CreateGraphNode(info, false);
|
|||
|
|
this.workflow.Datas.Add(node.info);
|
|||
|
|
node.ClearSlots();
|
|||
|
|
node.BuildSlots();
|
|||
|
|
}
|
|||
|
|
ConventionUtility.CreateSteps().Next(() =>
|
|||
|
|
{
|
|||
|
|
for (int i = 0; i < workflow.Datas.Count; i++)
|
|||
|
|
{
|
|||
|
|
var info = workflow.Datas[i];
|
|||
|
|
var node = GetGraphNode(i);
|
|||
|
|
foreach (var (key, slot) in info.inmapping)
|
|||
|
|
{
|
|||
|
|
if (slot.targetNodeID != -1)
|
|||
|
|
this.workflow.Nodes[i].LinkInslotToOtherNodeOutslot(GetGraphNode(slot.targetNodeID), slot.slotName, slot.targetSlotName);
|
|||
|
|
}
|
|||
|
|
node.RefreshRectTransform();
|
|||
|
|
node.RefreshPosition();
|
|||
|
|
}
|
|||
|
|
}).Wait(0.1f, () =>
|
|||
|
|
{
|
|||
|
|
foreach (var node in this.workflow.Nodes)
|
|||
|
|
{
|
|||
|
|
node.RefreshImmediate();
|
|||
|
|
}
|
|||
|
|
}).Wait(1f, () =>
|
|||
|
|
{
|
|||
|
|
foreach (var node in this.workflow.Nodes)
|
|||
|
|
{
|
|||
|
|
node.RefreshImmediate();
|
|||
|
|
}
|
|||
|
|
}).Invoke();
|
|||
|
|
Debug.Log($"Current Workflow Nodes: count={this.workflow.Nodes.Count}");
|
|||
|
|
return this.workflow;
|
|||
|
|
}
|
|||
|
|
public Workflow LoadWorkflow(string workflowPath)
|
|||
|
|
{
|
|||
|
|
ToolFile local = new(workflowPath);
|
|||
|
|
if (local.Exists() == false)
|
|||
|
|
throw new FileNotFoundException($"{local} is not exist");
|
|||
|
|
LastSavePath = workflowPath;
|
|||
|
|
var loadedWorkflow = local.LoadAsJson<Workflow>(WorkflowFileKey);
|
|||
|
|
return LoadWorkflow(loadedWorkflow);
|
|||
|
|
}
|
|||
|
|
public void OpenMenu(PointerEventData data)
|
|||
|
|
{
|
|||
|
|
focusObject.position = data.pointerCurrentRaycast.worldPosition;
|
|||
|
|
#if UNITY_EDITOR
|
|||
|
|
if (callbackDatas.Count == 0)
|
|||
|
|
SharedModule.instance.OpenCustomMenu(UIFocusObject, new SharedModule.CallbackData("Empty", x => Debug.Log(x)));
|
|||
|
|
else
|
|||
|
|
#endif
|
|||
|
|
{
|
|||
|
|
foreach (var callbackData in callbackDatas)
|
|||
|
|
{
|
|||
|
|
Transformer(callbackData.name);
|
|||
|
|
}
|
|||
|
|
SharedModule.instance.OpenCustomMenu(UIFocusObject, callbackDatas.ToArray());
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|