Changeset 2447 for trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization/ProbabilisticTreeCreator.cs

Timestamp:

10/27/09 09:47:03 (14 years ago)

Author:

gkronber

Message:

Added until test project for HeuristicLab.GP plugins. #791 (Tests for GP operators and structure identification evaluators)

File:

• trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization/ProbabilisticTreeCreator.cs (modified) (2 diffs)

trunk/sources/HeuristicLab.GP.Operators/3.3/Initialization/ProbabilisticTreeCreator.cs

@@ -27 +27 @@
 namespace HeuristicLab.GP.Operators {
   public class ProbabilisticTreeCreator : OperatorBase {
-    private int MAX_TRIES { get { return 100; } }
+    private static int MAX_TRIES { get { return 100; } }
 
     public override string Description {
@@ -50 +50 @@
       int maxTreeHeight = GetVariableValue<IntData>("MaxTreeHeight", scope, true).Data;
 
-      int treeSize = random.Next(minTreeSize, maxTreeSize + 1);
-      IFunctionTree root;
-      int tries = 0;
-      TreeGardener gardener = new TreeGardener(random, funLibrary);
-      do {
-        root = gardener.PTC2(treeSize, maxTreeHeight);
-        if (tries++ >= MAX_TRIES) {
-          // try a different size
-          treeSize = random.Next(minTreeSize, maxTreeSize + 1);
-          tries = 0;
-        }
-      } while (root.GetSize() > maxTreeSize || root.GetHeight() > maxTreeHeight);
-
+      IFunctionTree root = Create(random, funLibrary, minTreeSize, maxTreeSize, maxTreeHeight);
       scope.AddVariable(new HeuristicLab.Core.Variable(scope.TranslateName("FunctionTree"), new GeneticProgrammingModel(root)));
       return Util.CreateInitializationOperation(TreeGardener.GetAllSubTrees(root), scope);
     }
 
- }
+
+    public static IFunctionTree Create(IRandom random, FunctionLibrary funLib, int minSize, int maxSize, int maxHeight) {
+      int treeSize = random.Next(minSize, maxSize + 1);
+      IFunctionTree root;
+      int tries = 0;
+      TreeGardener gardener = new TreeGardener(random, funLib);
+      do {
+        root = gardener.PTC2(treeSize, maxHeight);
+        if (tries++ >= MAX_TRIES) {
+          // try a different size
+          treeSize = random.Next(minSize, maxSize + 1);
+          tries = 0;
+        }
+      } while (root.GetSize() > maxHeight || root.GetHeight() > maxHeight);
+      return root;
+    }
+  }
 }