Laurens-Packages/Packages/com.unity.visualeffectgraph/Editor/GraphView/Elements/Controllers/VFXNodeController.cs

using System;
using UnityEditor.Experimental.GraphView;
using UnityEngine;
using System.Collections.Generic;
using System.Reflection;
using System.Linq;

using Object = UnityEngine.Object;
using System.Collections.ObjectModel;
using UnityEngine.VFX;

namespace UnityEditor.VFX.UI
{
    abstract class VFXNodeController : VFXController<VFXModel>, IGizmoController
    {
        protected List<VFXDataAnchorController> m_InputPorts = new List<VFXDataAnchorController>();

        protected List<VFXDataAnchorController> m_OutputPorts = new List<VFXDataAnchorController>();

        public ReadOnlyCollection<VFXDataAnchorController> inputPorts
        {
            get { return m_InputPorts.AsReadOnly(); }
        }

        public ReadOnlyCollection<VFXDataAnchorController> outputPorts
        {
            get { return m_OutputPorts.AsReadOnly(); }
        }

        public VFXNodeController(VFXModel model, VFXViewController viewController) : base(viewController, model)
        {
            var settings = model.GetSettings(true);
            m_Settings = new VFXSettingController[settings.Count()];
            int cpt = 0;
            foreach (var setting in settings)
            {
                var settingController = new VFXSettingController();
                settingController.Init(viewController, this.slotContainer, setting.field.Name, setting.field.FieldType);
                m_Settings[cpt++] = settingController;
            }
        }

        public virtual void ForceUpdate()
        {
            ModelChanged(model);
        }

        protected virtual void NewInputSet(List<VFXDataAnchorController> newInputs)
        {
        }

        public bool CouldLink(VFXDataAnchorController myAnchor, VFXDataAnchorController otherAnchor, VFXDataAnchorController.CanLinkCache cache)
        {
            if (myAnchor.direction == Direction.Input)
                return CouldLinkMyInputTo(myAnchor, otherAnchor, cache);
            else
                return otherAnchor.sourceNode.CouldLinkMyInputTo(otherAnchor, myAnchor, cache);
        }

        protected virtual bool CouldLinkMyInputTo(VFXDataAnchorController myInput, VFXDataAnchorController otherOutput, VFXDataAnchorController.CanLinkCache cache)
        {
            return false;
        }

        public void UpdateAllEditable()
        {
            foreach (var port in inputPorts)
            {
                port.UpdateEditable();
            }
        }

        protected override void ModelChanged(UnityEngine.Object obj)
        {
            var inputs = inputPorts;
            var newAnchors = new List<VFXDataAnchorController>();

            m_SyncingSlots = true;
            m_HasActivationAnchor = slotContainer.activationSlot != null;
            bool changed = false;
            if (m_HasActivationAnchor)
                changed = UpdateSlots(newAnchors, new[] { slotContainer.activationSlot }, true, true);
            changed |= UpdateSlots(newAnchors, slotContainer.inputSlots, true, true);
            NewInputSet(newAnchors);

            foreach (var anchorController in m_InputPorts.Except(newAnchors))
            {
                anchorController.OnDisable();
                changed = true;
            }
            m_InputPorts = newAnchors;


            newAnchors = new List<VFXDataAnchorController>();
            changed |= UpdateSlots(newAnchors, slotContainer.outputSlots, true, false);

            foreach (var anchorController in m_OutputPorts.Except(newAnchors))
            {
                anchorController.OnDisable();
                changed = true;
            }
            m_OutputPorts = newAnchors;
            m_SyncingSlots = false;

            // Call after base.ModelChanged which ensure the ui has been refreshed before we try to create potiential new edges.
            if (changed)
            {
                viewController.DataEdgesMightHaveChanged();
                VFXSlotContainerEditor.SceneViewVFXSlotContainerOverlay.ClearGizmos();
            }

            NotifyChange(AnyThing);
        }

        public virtual Vector2 position
        {
            get
            {
                return model.position;
            }
            set
            {
                value = new Vector2(Mathf.Round(value.x), Mathf.Round(value.y));
                if (model.position != value)
                {
                    Undo.RecordObject(model, "Move VFXModel: " + value);
                    model.position = value;
                }
            }
        }
        public virtual bool superCollapsed
        {
            get
            {
                return model.superCollapsed;
            }
            set
            {
                if (value != model.superCollapsed)
                {
                    Undo.RecordObject(model, "Collapse VFXModel: " + value);
                    model.superCollapsed = value;
                    if (model.superCollapsed)
                    {
                        model.collapsed = false;
                    }
                }
            }
        }

        public virtual string title
        {
            get
            {
                var name = model.name;
                var eof = name.IndexOf('\n');
                if (eof == -1)
                    return name;
                return name.Substring(0, eof);
            }
        }

        public string subtitle
        {
            get
            {
                var name = model.name;
                var eof = name.IndexOf('\n');
                if (eof == -1)
                    return string.Empty;
                return name.Substring(eof + 1);
            }
        }

        public override IEnumerable<Controller> allChildren
        {
            get { return inputPorts.Cast<Controller>().Concat(outputPorts.Cast<Controller>()).Concat(m_Settings.Cast<Controller>()); }
        }

        public virtual int id
        {
            get { return 0; }
        }

        private bool m_HasActivationAnchor = false;
        public bool HasActivationAnchor => m_HasActivationAnchor;

        bool m_SyncingSlots;
        public void DataEdgesMightHaveChanged()
        {
            if (viewController != null && !m_SyncingSlots)
            {
                viewController.DataEdgesMightHaveChanged();
            }
        }

        public virtual void NodeGoingToBeRemoved()
        {
            var outputEdges = outputPorts.SelectMany(t => t.connections);

            foreach (var edge in outputEdges)
            {
                edge.input.sourceNode.OnEdgeFromInputGoingToBeRemoved(edge.input);
            }

            var inputEdges = inputPorts.SelectMany(t => t.connections);

            foreach (var edge in inputEdges)
            {
                edge.output.sourceNode.OnEdgeFromOutputGoingToBeRemoved(edge.output, edge.input);
            }
        }

        public virtual void OnEdgeFromInputGoingToBeRemoved(VFXDataAnchorController myInput)
        {
        }

        public virtual void OnEdgeFromOutputGoingToBeRemoved(VFXDataAnchorController myOutput, VFXDataAnchorController otherInput)
        {
        }

        public virtual void WillCreateLink(ref VFXSlot myInput, ref VFXSlot otherOutput, bool revertTypeConstraint = false)
        {
        }

        protected virtual bool UpdateSlots(List<VFXDataAnchorController> newAnchors, IEnumerable<VFXSlot> slotList, bool expanded, bool input)
        {
            VFXSlot[] slots = slotList.ToArray();
            bool changed = false;
            foreach (VFXSlot slot in slots)
            {
                VFXDataAnchorController propController = GetPropertyController(slot, input);

                if (propController == null)
                {
                    propController = AddDataAnchor(slot, input, !expanded);
                    changed = true;
                }
                newAnchors.Add(propController);

                if (!VFXDataAnchorController.SlotShouldSkipFirstLevel(slot))
                {
                    changed |= UpdateSlots(newAnchors, slot.children, expanded && propController.expandedSelf, input);
                }
                else
                {
                    VFXSlot firstSlot = slot.children.First();
                    changed |= UpdateSlots(newAnchors, firstSlot.children, expanded && propController.expandedSelf, input);
                }
            }

            return changed;
        }

        public VFXDataAnchorController GetPropertyController(VFXSlot slot, bool input)
        {
            VFXDataAnchorController result = null;

            if (input)
                result = inputPorts.Cast<VFXDataAnchorController>().Where(t => t.model == slot).FirstOrDefault();
            else
                result = outputPorts.Cast<VFXDataAnchorController>().Where(t => t.model == slot).FirstOrDefault();

            return result;
        }

        protected abstract VFXDataAnchorController AddDataAnchor(VFXSlot slot, bool input, bool hidden);

        public IVFXSlotContainer slotContainer { get { return model as IVFXSlotContainer; } }

        public VFXSettingController[] settings
        {
            get { return m_Settings; }
        }

        public virtual bool expanded
        {
            get
            {
                return !slotContainer.collapsed;
            }

            set
            {
                if (value != !slotContainer.collapsed)
                {
                    Undo.RecordObject(model, "Collapse VFXModel: " + !value);
                    slotContainer.collapsed = !value;
                }
            }
        }

        public virtual void CollectGizmos()
        {
            m_GizmoableAnchors.Clear();
            foreach (VFXDataAnchorController controller in inputPorts)
            {
                if (controller.model != null && controller.model.IsMasterSlot() && VFXGizmoUtility.HasGizmo(controller.portType))
                {
                    m_GizmoableAnchors.Add(controller);
                }
            }
        }

        public virtual void DrawGizmos(VisualEffect component)
        {
            if (currentGizmoable is VFXDataAnchorController gizmoable)
            {
                gizmoable.DrawGizmo(component);
            }
        }

        public virtual Bounds GetGizmoBounds(VisualEffect component)
        {
            if (currentGizmoable != null)
                return ((VFXDataAnchorController)currentGizmoable).GetGizmoBounds(component);

            return new Bounds();
        }

        public virtual GizmoError GetGizmoError(VisualEffect component)
        {
            return currentGizmoable is IGizmoError gizmoController
                ? gizmoController.GetGizmoError(component)
                : GizmoError.NotAvailable;
        }

        protected List<IGizmoable> m_GizmoableAnchors = new();

        public ReadOnlyCollection<IGizmoable> gizmoables => m_GizmoableAnchors.AsReadOnly();

        public IGizmoable currentGizmoable { get; set; }

        private VFXSettingController[] m_Settings;
    }
}