Initial Commit

Library/PackageCache/com.unity.visualscripting@1.5.2/Editor/VisualScripting.Flow/XFlowGraph.cs

@@ -0,0 +1,34 @@
+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 SuperUnit)?.nest.graph;
+
+                if (nestedGraph != null)
+                {
+                    foreach (var nestedUnit in GetUnitsRecursive(nestedGraph, recursion))
+                    {
+                        yield return nestedUnit;
+                    }
+                }
+            }
+
+            recursion?.Exit(flowGraph);
+        }
+    }
+}