xNode/Editor/GroupNodeEditor.cs

using System;
using System.Collections.Generic;
using UnityEngine;
using XNode;
using System.Linq;
using UnityEditor;
using XNodeEditor.Internal;

namespace XNodeEditor
{
    [CustomNodeEditor(typeof(XNode.GroupNode))]
    public class GroupNodeEditor : NodeEditor
    {
        private static Vector4 padding = new Vector4(32, 48, 32, 32);
        private bool _selected = false;
        private UnityEngine.Object[] lastSelection;
        private List<RerouteReference> lastSelectionReroutes;

        public override void OnHeaderGUI()
        {
            if (target is XNode.GroupNode node)
            {
                bool selectChildren = NodeEditorWindow.currentActivity is NodeEditorWindow.NodeActivity.HoldNode or NodeEditorWindow.NodeActivity.DragNode;
                if (!_selected && selectChildren && Selection.objects.Contains(target))
                {
                    _selected = true;
                    var selection = Selection.objects.ToList();
                    lastSelection = selection.ToArray();
                    lastSelectionReroutes = new List<RerouteReference>(window.selectedReroutes);
                    SelectChildren(node, ref selection);
                    //selection.AddRange(GetChildren(node));
                    Selection.objects = selection.Distinct().ToArray();
                    NodeEditorWindow.current.Repaint();
                }
                else if (_selected && !selectChildren)
                {
                    _selected = false;
                    Selection.objects = lastSelection;
                    window.selectedReroutes = lastSelectionReroutes;
                }
            }
            base.OnHeaderGUI();
        }

        private void SelectChildren(XNode.GroupNode group, ref List<UnityEngine.Object> selectionList)
        {
            var groupRect = new Rect(group.position, new Vector2(GetWidth(), GetHeight()));
            foreach (var child in group.children)
            {
                if (selectionList.Contains(child)) continue;
                selectionList.Add(child);
                var pc = child.Ports.Count();
                for (int p = 0; p < pc; p++)
                {
                    var port = child.Ports.ElementAt(p);
                    for (int i = 0; i < port.ConnectionCount; i++)
                    {
                        var connectionIndex = i;
                        if (port.direction == NodePort.IO.Input)
                        {
                            var newPort = port.GetConnection(i);
                            connectionIndex = newPort.GetConnectionIndex(port);
                            port = newPort;
                        }
                        var reroutes = port.GetReroutePoints(connectionIndex);
                        for (int j = 0; j < reroutes.Count; j++)
                        {
                            if (groupRect.Contains(reroutes[j])) window.selectedReroutes.Add(new RerouteReference(port, connectionIndex, j));
                        }
                    }
                }
                if (child is XNode.GroupNode _group) SelectChildren(_group, ref selectionList);
            }
        }

        public override void OnBodyGUI()
        {
            var e = Event.current;
            var node = target as XNode.GroupNode;

            for (int i = node.children.Count - 1; i >= 0; i--)
            {
                if (node.children[i] == null) node.children.RemoveAt(i);
            }

            var nodes = node.graph.nodes;
            if (e.type == EventType.Repaint)
            {
                if (nodes.FirstOrDefault() != node)
                {
                    nodes.Remove(node);
                    int targetIndex = 0;
                    for (int i = 0; i < nodes.Count; i++)
                    {
                        if (nodes[i] is XNode.GroupNode _group)
                        {
                            if (_group.children.Contains(node))
                            {
                                targetIndex = i + 1;
                                break;
                            }
                        }
                        else break;
                    }
                    nodes.Insert(targetIndex, node);
                }
            }

            if (node == null || node.children == null || node.children.Count == 0) return;
            node.position = new Vector2(
                node.children.Min(x => x.position.x) - padding.x,
                node.children.Min(x => x.position.y) - padding.y
                );
            GUILayout.Label(GUIContent.none);
        }

        public override Color GetTint()
        {
            Type type = target.GetType();
            if (type.TryGetAttributeTint(out Color color)) return color;

            if (!(target is XNode.GroupNode node)) return NodeEditorPreferences.GetSettings().tintColor;
            return node.color;
        }

        public override int GetWidth()
        {
            int min = base.GetWidth();
            var group = target as XNode.GroupNode;
            if (group == null || group.children == null || group.children.Count == 0) return min + (int)padding.x + (int)padding.z;
            return Mathf.Max(min,
                             group.children.Max(x =>
                                                   GetSize(x).x + (int)x.position.x) - (int)target.position.x)
                 + (int)padding.z;
        }

        public override int GetHeight()
        {
            int min = base.GetHeight();
            var group = target as XNode.GroupNode;
            if (group == null || group.children == null || group.children.Count == 0) return min + (int)padding.y + (int)padding.w;
            return Mathf.Max(min,
                             group.children.Max(x =>
                                                   GetSize(x).y + (int) x.position.y) - (int)target.position.y)
                 + (int) padding.w;
        }

        public Vector2Int GetSize(Node node) => Vector2Int.RoundToInt(window.nodeSizes.TryGetValue(node, out var size) ? size : new Vector2(0,0));

        private static Texture2D nodeGroupBody => _nodeGroupBody != null ? _nodeGroupBody : _nodeGroupBody = Resources.Load<Texture2D>("xnode_group");
        private static Texture2D _nodeGroupBody;
        private static GUIStyle guiStyle;

        public override GUIStyle GetBodyStyle() => guiStyle ??= new GUIStyle(base.GetBodyStyle())
        {
            normal =
            {
                background = nodeGroupBody
            }
        };
    }
}