67 lines
2.4 KiB
C#
67 lines
2.4 KiB
C#
|
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();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|