U3d杂谈 — 使用GraphView做工具 — 02

前面说了如何通过GraphView创建出节点，然后节点如何连接。这一节介绍如何自己创建自定义的节点，这样才能根据需要制作符合功能的节点。下面我们制作2个节点，一个是int的加法节点和输出日志的节点。

展现的效果如下：

我们可以在节点上点击鼠标右键选择“执行”动作来执行当前节点。目前的执行逻辑很简单，是依次向下执行。

using UnityEditor.Experimental.GraphView;
using UnityEngine.UIElements;

public class AddIntNode : MyBaseNode
{
    private TextField Param1;
    private TextField Param2;

    private Port Result;
    public AddIntNode()
    {
        title = "Int加";
        Param1 = new TextField("参数1：");
        Param1.RegisterValueChangedCallback(OnParam1ValueChange);
        Param1.value = "0";

        Param2 = new TextField("参数2：");
        Param2.RegisterValueChangedCallback(OnParam2ValueChange);
        Param2.value = "0";

        Result = Port.Create<Edge>(Orientation.Horizontal, Direction.Output, Port.Capacity.Multi, typeof(int));
        Result.portName = "结果";
        Result.source = 0;

        inputContainer.Add(Param1);
        inputContainer.Add(Param2);

        outputContainer.Add(Result);
    }

    private void OnParam1ValueChange(ChangeEvent<string> e)
    {
        int r = 0;
        if(int.TryParse(e.newValue,out r))
        {
            string s = string.Format("{0}", r);
            Param1.value = s;
        }
        else
        {
            Param1.value = "0";
        }      
    }

    private void OnParam2ValueChange(ChangeEvent<string> e)
    {
        int r = 0;
        if (int.TryParse(e.newValue, out r))
        {
            string s = string.Format("{0}", r);
            Param2.value = s;
        }
        else
        {
            Param2.value = "0";
        }
    }

    public override void OnExecute()
    {
        int r = int.Parse(Param1.value) + int.Parse(Param2.value);
        Result.source = r;
        base.OnExecute();
    }
}

using UnityEditor.Experimental.GraphView;
using UnityEngine;

public class LogNode : MyBaseNode
{
    private Port LogText;
    public LogNode()
    {
        title = "Log打印";

        LogText = Port.Create<Edge>(Orientation.Horizontal, Direction.Input, Port.Capacity.Single, typeof(object));
        LogText.portName = "输出文本";
        LogText.source = 0;

        inputContainer.Add(LogText);
    }

    public override void OnExecute()
    {
        foreach(var v in LogText.connections)
        {
            if(v.output != null)
            {
                LogText.source = v.output.source;
            }
        }
        Debug.Log(LogText.source);
       
        base.OnExecute();
    }
}

他们都来至于一个基类，方便以后扩展相同功能的方法和属性：

using UnityEditor.Experimental.GraphView;
using UnityEngine.UIElements;

public class MyBaseNode : Node
{
    //处理节点计算
    public virtual void OnExecute()
    {
        int count = outputContainer.childCount;
        for(int i = 0; i < count; ++i)
        {
            if(outputContainer.ElementAt(i) is Port)
            {
                Port p = outputContainer.ElementAt(i) as Port;
                foreach(var n in p.connections)
                {
                    if(n.input != null)
                    {
                        MyBaseNode iNode = n.input.node as MyBaseNode;
                        if(iNode != null)
                        {
                            iNode.OnExecute();
                        }
                    }
                }
            }
        }
    }

    public override void BuildContextualMenu(ContextualMenuPopulateEvent evt)
    {
        evt.menu.InsertAction(0, "执行", OnExcMenu);
    }

    private void OnExcMenu(DropdownMenuAction a)
    {
        OnExecute();
    }
}

然后我们在MyGraphView的创建代理函数中，创建的时候默认创建这2个节点。后面我们来看如何扩展创建节点的菜单。

this.nodeCreationRequest += (NodeCreationContext t) =>{
            Node s1 = new AddIntNode();
            Node s2 = new LogNode();

            AddElement(s1);
            AddElement(s2);
        };

我们修改添加上面的几处代码，就能创建这2个节点。我们选中Int加节点鼠标右键，看看是否能够正常计算自己填写的值呢？

目前这种方式存在问题。比如一个节点的输出连接了2个节点，上面的节点的输入又依赖于下面兄弟节点的输出，就会导致问题。因为代码中我们默认通过输出节点向下执行，但是输入的数据可能来不全，这个时候需要通过输入的连接执行父节点。下一节我们探索如何设计节点来规划这个问题。