Changeset 10028 for branches/Robocode.TrunkInt/HeuristicLab.Problems.Robocode/3.3/Symbols/Statements/IfThenElseStat.cs

Timestamp:

10/07/13 08:49:34 (11 years ago)

Author:

jkarder

Message:

#2069:

• merged IfStat and ElseStat into IfThenElseStat
• changed the font of the suffixCode text box in the CodeNodeView

File:

• branches/Robocode.TrunkInt/HeuristicLab.Problems.Robocode/3.3/Symbols/Statements/IfThenElseStat.cs (moved) (moved from branches/Robocode.TrunkInt/HeuristicLab.Problems.Robocode/3.3/Symbols/Statements/IfStat.cs) (2 diffs)

branches/Robocode.TrunkInt/HeuristicLab.Problems.Robocode/3.3/Symbols/Statements/IfThenElseStat.cs

@@ -28 +28 @@
 namespace HeuristicLab.Problems.Robocode {
   [StorableClass]
-  public class IfStat : CodeNode {
+  public class IfThenElseStat : CodeNode {
     public override int MinimumArity { get { return 2; } }
     public override int MaximumArity { get { return 3; } }
@@ -39 +39 @@
 
     [StorableConstructor]
-    protected IfStat(bool deserializing) : base(deserializing) { }
-    protected IfStat(IfStat original, Cloner cloner) : base(original, cloner) { }
+    protected IfThenElseStat(bool deserializing) : base(deserializing) { }
+    protected IfThenElseStat(IfThenElseStat original, Cloner cloner) : base(original, cloner) { }
 
-    public IfStat()
-      : base("IfStat", "An if statement.") {
-      Prefix = "if (";
-      Suffix = ") {";
-    }
+    public IfThenElseStat() : base("IfThenElseStat", "An if statement.") { }
 
     public override IDeepCloneable Clone(Cloner cloner) {
-      return new IfStat(this, cloner);
+      return new IfThenElseStat(this, cloner);
     }
 
     public override string Interpret(ISymbolicExpressionTreeNode node, IEnumerable<ISymbolicExpressionTreeNode> children) {
-      string condition = string.Empty, ifTrue = string.Empty, ifElse = string.Empty;
-      foreach (var c in children) {
-        if (c.Symbol is LogicalExpression)
-          condition = ((CodeNode)c.Symbol).Interpret(c, c.Subtrees);
-        else if (c.Symbol is Stat)
-          ifTrue = ((CodeNode)c.Symbol).Interpret(c, c.Subtrees);
-        else if (c.Symbol is ElseStat || c.Symbol is EmptyEvent || c.Symbol is DoNothing)
-          ifElse = ((CodeNode)c.Symbol).Interpret(c, c.Subtrees);
-        else throw new ArgumentException("Unexpected children.", "children");
+      ISymbolicExpressionTreeNode condition = null, truePart = null, falsePart = null;
+      var enumerator = children.GetEnumerator();
+      int childCount = 0;
+      while (enumerator.MoveNext()) {
+        childCount++;
+        switch (childCount) {
+          case 1: condition = enumerator.Current; break;
+          case 2: truePart = enumerator.Current; break;
+          case 3: falsePart = enumerator.Current; break;
+          default: throw new ArgumentException("Unexpected number of children. Expected a maximum of 3 children.");
+        }
       }
-      string result = Prefix + condition + Suffix + Environment.NewLine + ifTrue + Environment.NewLine + "}";
-      result += string.IsNullOrEmpty(ifElse) ? Environment.NewLine : ifElse;
+      if (childCount < 2) throw new ArgumentException("Unexpected number of children. Expected at least 2 children.");
+
+      var parts = new[] {
+        ((CodeNode)condition.Symbol).Interpret(condition, condition.Subtrees),
+        ((CodeNode)truePart.Symbol).Interpret(truePart, truePart.Subtrees),
+        ((CodeNode)falsePart.Symbol).Interpret(falsePart, falsePart.Subtrees)
+      };
+
+      string result = "if (" + parts[0] + ") {" + parts[1] + "}";
+      if (!string.IsNullOrEmpty(parts[2]))
+        result += " else {" + result[2] + "}";
       return result;
     }