Laurens-Packages/Packages/com.unity.shadergraph/Editor/Importers/ShaderGraphMetadata.cs

using System;
using System.Collections.Generic;
using System.Linq;
using UnityEditor.ShaderGraph.Internal;
using UnityEngine;
using static UnityEditor.ShaderGraph.CategoryDataCollection;
using Object = UnityEngine.Object;

namespace UnityEditor.ShaderGraph
{
    [Serializable]
    struct GraphInputData
    {
        public string referenceName;
        public bool isKeyword;
        public PropertyType propertyType;
        public KeywordType keywordType;

        public bool isCompoundProperty;
        public List<SubPropertyData> subProperties;

        public string tooltip;

        public override bool Equals(object obj)
        {
            if (obj is null || obj is not GraphInputData o)
                return false;

            if (subProperties is null != o.subProperties is null
                || referenceName is null != o.referenceName is null
                || tooltip is null != o.tooltip is null
                || referenceName is not null && !referenceName.Equals(o.referenceName)
                || !isKeyword.Equals(o.isKeyword)
                || !propertyType.Equals(o.propertyType)
                || !keywordType.Equals(o.keywordType)
                || !isCompoundProperty.Equals(o.isCompoundProperty)
                // || tooltip is not null && !tooltip.Equals(o.tooltip)
                || subProperties is not null && !subProperties.Count.Equals(o.subProperties.Count))
                return false;

            if (subProperties is not null)
            for (int i = 0; i < subProperties.Count; ++i)
                if (!subProperties[i].Equals(o.subProperties[i]))
                    return false;

            return true;
        }

        public override int GetHashCode()
        {
            var hash = new HashCode();
            hash.Add(referenceName);
            hash.Add(isKeyword);
            hash.Add(propertyType);
            hash.Add(keywordType);
            hash.Add(isCompoundProperty);
            // hash.Add(tooltip); This is not part of the graph inputs identity.

            if (subProperties is not null)
            foreach (var subproperty in subProperties)
            {
                hash.Add(subproperty.referenceName);
                hash.Add(subproperty.propertyType);
            }
            return hash.ToHashCode();
        }
    }

    [Serializable]
    struct SubPropertyData
    {
        public string referenceName;
        public PropertyType propertyType;
    }

    [Serializable]
    class MinimalCategoryData
    {
        // ShaderLab doesn't understand virtual texture inputs, they need to be processed to replace the virtual texture input with the layers that compose it instead
        public static GraphInputData ProcessVirtualTextureProperty(VirtualTextureShaderProperty virtualTextureShaderProperty)
        {
            var layerReferenceNames = new List<string>();
            virtualTextureShaderProperty.GetPropertyReferenceNames(layerReferenceNames);
            var virtualTextureLayerDataList = new List<SubPropertyData>();

            // Skip the first entry in this list as that's the Virtual Texture reference name itself, which won't exist in ShaderLab
            foreach (var referenceName in layerReferenceNames.Skip(1))
            {
                var layerPropertyData = new SubPropertyData() { referenceName = referenceName, propertyType = PropertyType.Texture2D };
                virtualTextureLayerDataList.Add(layerPropertyData);
            }

            var virtualTexturePropertyData = new GraphInputData() { referenceName = virtualTextureShaderProperty.displayName, propertyType = PropertyType.VirtualTexture, isKeyword = false };
            virtualTexturePropertyData.isCompoundProperty = true;
            virtualTexturePropertyData.subProperties = virtualTextureLayerDataList;
            return virtualTexturePropertyData;
        }

        public static bool TryProcessInput(ShaderInput input, out GraphInputData data)
        {
            data = default;
            if (!input.isExposed)
                return false;

            string tooltip = "";
            if (input.promoteToFinalShader)
                tooltip = $"from {input.PromotedAssetPath}";

            switch(input)
            {
                case VirtualTextureShaderProperty vt:
                    data = ProcessVirtualTextureProperty(vt);
                    return true;
                case AbstractShaderProperty prop:
                    data = new GraphInputData() { referenceName = prop.referenceName, propertyType = prop.propertyType, isKeyword = false, tooltip = tooltip };
                    return true;
                case ShaderKeyword keyword:
                    var sanitizedReferenceName = keyword.referenceName;
                    if (keyword.keywordType == KeywordType.Boolean && keyword.referenceName.Contains("_ON"))
                        sanitizedReferenceName = sanitizedReferenceName.Replace("_ON", String.Empty);
                    data = new GraphInputData() { referenceName = sanitizedReferenceName, keywordType = keyword.keywordType, isKeyword = true, tooltip = tooltip };
                    return true;

                default:
                    return false;
            }
        }

        public string categoryName;
        public List<GraphInputData> propertyDatas;
        [NonSerialized]
        public bool expanded = true;
    }

    class ShaderGraphMetadata : ScriptableObject
    {
        public string outputNodeTypeName;

        // these asset dependencies are stored here as a way for "Export Package..." to discover them
        // and automatically pull them in to the .unitypackage
        public List<Object> assetDependencies;

        public List<MinimalCategoryData> categoryDatas;
    }

    internal class CategoryDataCollection
    {
        HashSet<(GraphInputData input, string category)> lookup = new();
        Dictionary<string, int> categoryScores = new();
        Dictionary<GraphInputData, int> propertyScores = new();

        internal void Set(string categoryName, GraphInputData data, int propertyScore = 0, int categoryScore = 0)
        {
            if (!categoryScores.TryAdd(categoryName, categoryScore) && categoryScore < categoryScores[categoryName])
                categoryScores[categoryName] = categoryScore;

            if (!propertyScores.TryAdd(data, propertyScore) && propertyScore < propertyScores[data])
                propertyScores[data] = propertyScore;

            // append the tooltip if it doesn't already exist.
            if (lookup.TryGetValue((data, categoryName), out var currentEntry) && !currentEntry.input.tooltip.Contains(data.tooltip))
            {
                lookup.Remove((data, categoryName));
                currentEntry.input.tooltip += $"\n{data.tooltip}";
                lookup.Add(currentEntry);
            }
            else lookup.Add((data, categoryName));
        }

        internal List<MinimalCategoryData> GenerateMCD()
        {
            var result = new List<MinimalCategoryData>();
            var buckets = new Dictionary<string, List<GraphInputData>>();

            var order = new List<string>(categoryScores.Keys);
            order.Sort((a, b) => {
                    var score = categoryScores[a].CompareTo(categoryScores[b]);
                    if (score == 0) score = a.CompareTo(b);
                    return score;
                });

            foreach(var data in lookup)
            {
                buckets.TryAdd(data.category, new());
                buckets[data.category].Add(data.input);
            }

            foreach(var name in order)
            {
                if (!buckets.ContainsKey(name))
                    continue;

                buckets[name].Sort((a, b) => {
                    var score = propertyScores[a].CompareTo(propertyScores[b]);
                    if (score == 0) score = a.referenceName.CompareTo(b.referenceName);
                    return score;
                });

                result.Add(new() { categoryName = name, propertyDatas = buckets[name] });
            }
            return result;
        }
    }
}