UnityGame/Library/PackageCache/com.unity.shadergraph/Editor/Data/Nodes/NodeClassCache.cs

using System;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Profiling;

namespace UnityEditor.ShaderGraph
{
    [GenerationAPI]
    [InitializeOnLoad]
    internal static class NodeClassCache
    {
        private class PostProcessor : AssetPostprocessor
        {
            static void OnPostprocessAllAssets(string[] importedAssets, string[] deletedAssets, string[] movedAssets, string[] movedFromAssetPaths)
            {
                if (m_KnownSubGraphLookupTable != null)
                {
                    foreach (string str in deletedAssets)
                    {
                        var guid = AssetDatabase.AssetPathToGUID(str);
                        if (m_KnownSubGraphLookupTable.ContainsKey(guid))
                        {
                            m_KnownSubGraphLookupTable.Remove(guid);
                        }
                    }
                    foreach (string str in movedFromAssetPaths)
                    {
                        var guid = AssetDatabase.AssetPathToGUID(str);
                        if (m_KnownSubGraphLookupTable.ContainsKey(guid))
                        {
                            m_KnownSubGraphLookupTable.Remove(guid);
                        }
                    }
                }

                foreach (string str in importedAssets)
                {
                    if (str.EndsWith(ShaderSubGraphImporter.Extension))
                    {
                        UpdateSubGraphEntry(str);
                    }
                }
                foreach (string str in movedAssets)
                {
                    if (str.EndsWith(ShaderSubGraphImporter.Extension))
                    {
                        UpdateSubGraphEntry(str);
                    }
                }
            }
        }


        private static Dictionary<Type, List<ContextFilterableAttribute>> m_KnownTypeLookupTable;
        private static Dictionary<Type, List<ContextFilterableAttribute>> KnownTypeLookupTable
        {
            get
            {
                EnsureKnownTypeLookupTable();
                return m_KnownTypeLookupTable;
            }
        }

        public static IEnumerable<Type> knownNodeTypes
        {
            get => KnownTypeLookupTable.Keys;
        }


        private static Dictionary<string, SubGraphAsset> m_KnownSubGraphLookupTable;
        private static Dictionary<string, SubGraphAsset> KnownSubGraphLookupTable
        {
            get
            {
                EnsureKnownSubGraphLookupTable();
                return m_KnownSubGraphLookupTable;
            }
        }

        public static IEnumerable<SubGraphAsset> knownSubGraphAssets
        {
            get => KnownSubGraphLookupTable.Values;
        }

        public static void UpdateSubGraphEntry(string path)
        {
            string guid = AssetDatabase.AssetPathToGUID(path);
            if (guid.Length == 0)
            {
                return;
            }
            var asset = AssetDatabase.LoadAssetAtPath<SubGraphAsset>(path);

            bool valid = asset != null && asset.isValid;
            if (KnownSubGraphLookupTable.TryGetValue(guid, out SubGraphAsset known))
            {
                if (!valid)
                {
                    KnownSubGraphLookupTable.Remove(guid);
                }
                else if (asset != known)
                {
                    KnownSubGraphLookupTable[guid] = asset;
                }
            }
            else if (valid)
            {
                KnownSubGraphLookupTable.Add(guid, asset);
            }
        }

        public static IEnumerable<ContextFilterableAttribute> GetFilterableAttributesOnNodeType(Type nodeType)
        {
            if (nodeType == null)
            {
                throw new ArgumentNullException("Cannot get attributes on a null Type");
            }

            if (KnownTypeLookupTable.TryGetValue(nodeType, out List<ContextFilterableAttribute> filterableAttributes))
            {
                return filterableAttributes;
            }
            else
            {
                throw new ArgumentException($"The passed in Type {nodeType.FullName} was not found in the loaded assemblies as a child class of AbstractMaterialNode");
            }
        }

        public static T GetAttributeOnNodeType<T>(Type nodeType) where T : ContextFilterableAttribute
        {
            var filterableAttributes = GetFilterableAttributesOnNodeType(nodeType);
            foreach (var attr in filterableAttributes)
            {
                if (attr is T searchTypeAttr)
                {
                    return searchTypeAttr;
                }
            }
            return null;
        }

        private static void EnsureKnownTypeLookupTable()
        {
            if (m_KnownTypeLookupTable == null)
            {
                Profiler.BeginSample("EnsureKnownTypeLookupTable");
                m_KnownTypeLookupTable = new Dictionary<Type, List<ContextFilterableAttribute>>();
                foreach (Type nodeType in TypeCache.GetTypesDerivedFrom<AbstractMaterialNode>())
                {
                    if (!nodeType.IsAbstract)
                    {
                        List<ContextFilterableAttribute> filterableAttributes = new List<ContextFilterableAttribute>();
                        foreach (Attribute attribute in Attribute.GetCustomAttributes(nodeType))
                        {
                            Type attributeType = attribute.GetType();
                            if (!attributeType.IsAbstract && attribute is ContextFilterableAttribute contextFilterableAttribute)
                            {
                                filterableAttributes.Add(contextFilterableAttribute);
                            }
                        }
                        m_KnownTypeLookupTable.Add(nodeType, filterableAttributes);
                    }
                }
                Profiler.EndSample();
            }
        }

        private static void EnsureKnownSubGraphLookupTable()
        {
            if (m_KnownSubGraphLookupTable == null)
            {
                Profiler.BeginSample("EnsureKnownSubGraphLookupTable");
                m_KnownSubGraphLookupTable = new Dictionary<string, SubGraphAsset>();
                foreach (var guid in AssetDatabase.FindAssets(string.Format("t:{0}", typeof(SubGraphAsset))))
                {
                    var asset = AssetDatabase.LoadAssetAtPath<SubGraphAsset>(AssetDatabase.GUIDToAssetPath(guid));
                    if (asset != null && asset.isValid)
                    {
                        m_KnownSubGraphLookupTable.Add(guid, asset);
                    }
                }
                Profiler.EndSample();
            }
        }

        private static void DebugPrintKnownNodes()
        {
            foreach (var entry in KnownTypeLookupTable)
            {
                var nodeType = entry.Key;
                var filterableAttributes = entry.Value;
                String attrs = "";
                foreach (var filterable in filterableAttributes)
                {
                    attrs += filterable.ToString() + ", ";
                }
                Debug.Log(nodeType.ToString() + $": [{attrs}]");
            }
        }
    }
}