Laurens-Packages/Packages/com.unity.visualeffectgraph/Editor/Models/VFXUI.cs

using System;
using System.Collections.Generic;
using System.Linq;

using UnityEditor.Experimental.GraphView;
using UnityEngine;
using UnityEngine.Serialization;

namespace UnityEditor.VFX
{
    [Serializable]
    struct VFXNodeID
    {
        public VFXNodeID(VFXModel model, int id)
        {
            this.model = model;
            this.isStickyNote = false;
            this.id = id;
        }

        public VFXNodeID(int id)
        {
            this.model = null;
            this.isStickyNote = true;
            this.id = id;
        }

        public VFXModel model;
        public int id;

        public bool isStickyNote;
    }
    class VFXUI : VFXObject
    {
        [System.Serializable]
        public class UIInfo
        {
            public UIInfo()
            {
            }

            public UIInfo(UIInfo other)
            {
                title = other.title;
                position = other.position;
            }

            public string title;
            public Rect position;
        }

        [System.Serializable]
        public class GroupInfo : UIInfo
        {
            [FormerlySerializedAs("content")]
            public VFXNodeID[] contents;
            public GroupInfo()
            {
            }

            public GroupInfo(GroupInfo other) : base(other)
            {
                contents = other.contents;
            }
        }

        [System.Serializable]
        public class StickyNoteInfo : UIInfo
        {
            public string contents;
            public string theme;
            public string textSize;
            public int colorTheme;

            public StickyNoteInfo()
            {
            }

            public StickyNoteInfo(StickyNoteInfo other) : base(other)
            {
                contents = other.contents;
                theme = other.theme;
                textSize = other.textSize;
                colorTheme = other.colorTheme;
            }
        }

        [System.Serializable]
        public class SystemInfo : UIInfo
        {
            public VFXContext[] contexts;
        }

        public GroupInfo[] groupInfos;
        public StickyNoteInfo[] stickyNoteInfos;

        [Serializable]
        public struct CategoryInfo
        {
            public string name;
            public bool collapsed;
        }

        public List<CategoryInfo> categories =  new ();

        public Rect uiBounds;

        public void Sanitize(VFXGraph graph)
        {
            categories ??= new ();

            if (groupInfos != null)
                foreach (var groupInfo in groupInfos)
                {
                    //Check first, rebuild after because in most case the content will be valid, saving an allocation.
                    if (groupInfo.contents != null && groupInfo.contents.Any(t => (!t.isStickyNote || t.id >= stickyNoteInfos.Length) && !graph.children.Contains(t.model)))
                    {
                        groupInfo.contents = groupInfo.contents.Where(t => (t.isStickyNote && t.id < stickyNoteInfos.Length) || graph.children.Contains(t.model)).ToArray();
                    }
                }

            // Sticky note theme serialization has changed, before it was a string "Light" or "Dark" now it's an integer 1, 2 or 3. (2 being equivalent to "Dark")
            // And groups now have color choice two
            if (graph.version <= 18)
            {
                if (stickyNoteInfos != null)
                {
                    foreach (var stickyNote in stickyNoteInfos)
                    {
                        if (stickyNote.colorTheme == 0)
                        {
                            stickyNote.colorTheme = string.Compare(stickyNote.theme, StickyNoteTheme.Classic.ToString(), StringComparison.OrdinalIgnoreCase) == 0 ? 1 : 2;
                            stickyNote.theme = null;
                        }
                    }
                }
            }
        }
    }
}