testss

Library/PackageCache/com.unity.shadergraph@11.0.0/Editor/Data/Graphs/GraphDataUtils.cs

@@ -0,0 +1,66 @@
+using System;
+using UnityEngine;
+using UnityEditor.Graphing;
+using UnityEngine.Pool;
+
+namespace UnityEditor.ShaderGraph
+{
+    sealed partial class GraphData : ISerializationCallbackReceiver
+    {
+        public static class GraphDataUtils
+        {
+            public static void ApplyActionLeafFirst(GraphData graph, Action<AbstractMaterialNode> action)
+            {
+                var temporaryMarks = PooledHashSet<string>.Get();
+                var permanentMarks = PooledHashSet<string>.Get();
+                var slots = ListPool<MaterialSlot>.Get();
+
+                // Make sure we process a node's children before the node itself.
+                var stack = StackPool<AbstractMaterialNode>.Get();
+                foreach (var node in graph.GetNodes<AbstractMaterialNode>())
+                {
+                    stack.Push(node);
+                }
+                while (stack.Count > 0)
+                {
+                    var node = stack.Pop();
+                    if (permanentMarks.Contains(node.objectId))
+                    {
+                        continue;
+                    }
+
+                    if (temporaryMarks.Contains(node.objectId))
+                    {
+                        action.Invoke(node);
+                        permanentMarks.Add(node.objectId);
+                    }
+                    else
+                    {
+                        temporaryMarks.Add(node.objectId);
+                        stack.Push(node);
+                        node.GetInputSlots(slots);
+                        foreach (var inputSlot in slots)
+                        {
+                            var nodeEdges = graph.GetEdges(inputSlot.slotReference);
+                            foreach (var edge in nodeEdges)
+                            {
+                                var fromSocketRef = edge.outputSlot;
+                                var childNode = fromSocketRef.node;
+                                if (childNode != null)
+                                {
+                                    stack.Push(childNode);
+                                }
+                            }
+                        }
+                        slots.Clear();
+                    }
+                }
+
+                StackPool<AbstractMaterialNode>.Release(stack);
+                ListPool<MaterialSlot>.Release(slots);
+                temporaryMarks.Dispose();
+                permanentMarks.Dispose();
+            }
+        }
+    }
+}