mathe/Library/PackageCache/com.unity.shadergraph@14.0.8/Editor/Data/Nodes/NodeClassCache.cs

202 lines
7.2 KiB
C#
Raw Normal View History

2024-09-20 20:30:10 +02:00
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}]");
}
}
}
}