File size: 912 Bytes
18a519f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | using System.Collections.Generic;
namespace Unity.VisualScripting
{
public static class XFlowGraph
{
public static IEnumerable<IUnit> GetUnitsRecursive(this FlowGraph flowGraph, Recursion recursion)
{
Ensure.That(nameof(flowGraph)).IsNotNull(flowGraph);
if (!recursion?.TryEnter(flowGraph) ?? false)
{
yield break;
}
foreach (var unit in flowGraph.units)
{
yield return unit;
var nestedGraph = (unit as SubgraphUnit)?.nest.graph;
if (nestedGraph != null)
{
foreach (var nestedUnit in GetUnitsRecursive(nestedGraph, recursion))
{
yield return nestedUnit;
}
}
}
recursion?.Exit(flowGraph);
}
}
}
|