Changeset 9038

Timestamp:

12/12/12 13:09:21 (11 years ago)

Author:

mkommend

Message:

#1996: Added OperatorGraph.Iterate method which returns all reachable operators.

File:

• trunk/sources/HeuristicLab.Core/3.3/OperatorGraph.cs (modified) (2 diffs)

trunk/sources/HeuristicLab.Core/3.3/OperatorGraph.cs

@@ -21 +21 @@
 
 using System;
+using System.Collections.Generic;
 using System.Linq;
 using HeuristicLab.Collections;
@@ -244 +245 @@
     }
     #endregion
+
+    // <summary>
+    /// Iterates an operator graph so that it jumps from the intial operator to all other operators and yields each operator it touches.
+    /// Cycles are detected and not iterated twice.
+    /// </summary>
+    /// <returns>An enumeration of all the operators that could be found.</returns>
+    public IEnumerable<IOperator> Iterate() {
+      if (InitialOperator == null) yield break;
+
+      var open = new Stack<IOperator>();
+      var visited = new HashSet<IOperator>();
+      open.Push(InitialOperator);
+
+      while (open.Any()) {
+        IOperator current = open.Pop();
+        if (visited.Contains(current)) continue;
+        visited.Add(current);
+
+        foreach (var parameter in current.Parameters.OfType<IValueParameter>()) {
+          if (!typeof(IOperator).IsAssignableFrom(parameter.DataType)) continue;
+          if (parameter.Value == null) continue;
+
+          open.Push((IOperator)parameter.Value);
+        }
+
+        yield return current;
+      }
+    }
   }
 }