Support break and labeled statements.

frog/leg/ssa/codegen.dart

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 class SsaCodeGeneratorTask extends CompilerTask {
   SsaCodeGeneratorTask(Compiler compiler) : super(compiler);
   String get name() => 'SSA code generator';
 
   String generate(WorkItem work, HGraph graph) {
     return measure(() {
@@ skipping 208 matching lines @@
     }
   }
 
   visit(HInstruction node, int expectedPrecedence) {
     int oldPrecedence = this.expectedPrecedence;
     this.expectedPrecedence = expectedPrecedence;
     node.accept(this);
     this.expectedPrecedence = oldPrecedence;
   }
 
+  void handleLabeledBlock(HBasicBlock node) {
+    HLabeledBlockInformation labeledBlockInfo = node.labeledBlockInformation;
+    if (labeledBlockInfo.start === node) {
+      addIndentation();
+      for (SourceString label in labeledBlockInfo.labels) {
+        addLabel(label);
+        buffer.add(":");
+      }
+      buffer.add("{\n");
+      indent++;
+    } else {
+      assert(labeledBlockInfo.end === node);
+      assert((){
+        // Check that this block is (transitively) dominated by the start block.
+        HBasicBlock block = node;
+        while (block.dominator !== null) {
+          block = block.dominator;
+          if (block === labeledBlockInfo.start) return true;
+        }
+        return false;
+      });
+      indent--;
+      addIndentation();
+      buffer.add("}\n");
+    }
+  }
+
   visitBasicBlock(HBasicBlock node) {
     currentBlock = node;
 
-    // While loop will be closed by the conditional loop-branch.
-    // TODO(floitsch): HACK HACK HACK.
-    if (currentBlock.isLoopHeader()) beginLoop(node);
+    if (node.hasLabeledBlockInformation()) {
+      handleLabeledBlock(node);
+    } else if (currentBlock.isLoopHeader()) {
+      // While loop will be closed by the conditional loop-branch.
+      // TODO(floitsch): HACK HACK HACK.
+      beginLoop(node);
+    }
 
     HInstruction instruction = node.first;
     while (instruction != null) {
       if (instruction is HGoto || instruction is HExit || instruction is HTry) {
         visit(instruction, JSPrecedence.STATEMENT_PRECEDENCE);
         return;
       } else if (!instruction.generateAtUseSite()) {
         if (instruction is !HIf && instruction is !HBailoutTarget) {
           addIndentation();
         }
@@ skipping 120 matching lines @@
     // is responsible for visiting the successor.
     if (dominated.isEmpty()) return;
     if (dominated.length > 2) unreachable();
     if (dominated.length == 2 && currentBlock !== currentGraph.entry) {
       unreachable();
     }
     assert(dominated[0] == currentBlock.successors[0]);
     visitBasicBlock(dominated[0]);
   }
 
+  // Used to write the name of labels.
+  // The default implementation uses the unmodified Dart label name.
+  // Specializations might change this.
+  void addLabel(SourceString label) {
+    buffer.add(label.toString());
+  }
+
+  visitBreak(HBreak node) {
+    assert(currentBlock.successors.length == 1);
+    // No block finishing with a 'break' can have more than
+    // one dominated block (since it has only one successor).
+    // If the successor is dominated by another block, then the other block
+    // is responsible for visiting the successor.
+    List<HBasicBlock> dominated = currentBlock.dominatedBlocks;
+    assert(dominated.isEmpty());
+    // Otherwise we would have bailed out in the builder.
+    addIndentation();
+    buffer.add("break");
+    if (node.label !== null) {
+      buffer.add(" ");
+      addLabel(node.label);
+    }
+    buffer.add(";\n");
+  }
+
   visitTry(HTry node) {
     addIndentation();
     buffer.add('try {\n');
     indent++;
     List<HBasicBlock> successors = node.block.successors;
     visitBasicBlock(successors[0]);
     indent--;
 
     if (node.finallyBlock != successors[1]) {
       // Printing the catch part.
@@ skipping 14 matching lines @@
       visitBasicBlock(node.finallyBlock);
       indent--;
     }
     addIndentation();
     buffer.add('}\n');
 
     visitBasicBlock(node.joinBlock);
   }
 
   visitIf(HIf node) {
+    List<HBasicBlock> dominated = node.block.dominatedBlocks;
     startIf(node);
     assert(!node.generateAtUseSite());
     startThen(node);
+    assert(node.thenBlock === dominated[0]);
     visitBasicBlock(node.thenBlock);
+    int preVisitedBlocks = 1;
     endThen(node);
     if (node.hasElse) {
       startElse(node);
+      assert(node.elseBlock === dominated[1]);
       visitBasicBlock(node.elseBlock);
+      preVisitedBlocks = 2;
       endElse(node);
     }
     endIf(node);
 
-    // Normally the HIf dominates the join-block. In this case there is one
-    // dominated block that we need to visit:
-    // If both the then and else blocks return/throw, then the join-block is
-    // either the exit-block, or there is none.
-    // We can also have the case where the HIf has no else, but the then-branch
-    // terminates. If the code after the 'if' terminates, then the
-    // if could become the dominator of the exit-block, thus having
-    // three dominated blocks: the then, the code after the if, and the exit
-    // block.
-
-    List<HBasicBlock> dominated = node.block.dominatedBlocks;
+    // Visit all the dominated blocks that are not part of the then or else
+    // branches. Depending on how the then/else branches terminate
+    // (e.g., return/throw/break) there can be any number of these.
     int dominatedCount = dominated.length;
-    if (node.hasElse && (dominatedCount == 3 || dominatedCount == 4)) {
-      // Normal case. The third dominated block is either the join-block or
-      // the exit-block (if both branches terminate).
-      // If the if dominates 4 blocks, then at least one branch does a
-      // conditional return, and the 4th block is the exit-block.
-      assert(dominatedCount != 4 || dominated.last().isExitBlock());
-      visitBasicBlock(dominated[2]);
-    } else if (node.hasElse) {
-      // Both branches terminate, but this HIf is not the dominator of the exit
-      // block.
-      assert(dominatedCount == 2);
-    } else if (!node.hasElse && dominatedCount == 2) {
-      // Normal case. Even if the then-branch terminated there is still
-      // a join-block.
-      assert(!dominated.last().isExitBlock());
-      visitBasicBlock(dominated[1]);
-    } else {
-      // The then-branch terminates, and the code following the if terminates
-      // too. The if happens to dominate the exit-block.
-      assert(!node.hasElse);
-      assert(dominatedCount == 3);
-      assert(dominated.last().isExitBlock());
-      visitBasicBlock(dominated[1]);
-      visitBasicBlock(dominated[2]);
+    for (int i = preVisitedBlocks; i < dominatedCount; i++) {
+      HBasicBlock dominatedBlock = dominated[i];
+      assert(dominatedBlock.dominator === node.block);
+      visitBasicBlock(dominatedBlock);
     }
   }
 
   visitInvokeDynamicMethod(HInvokeDynamicMethod node) {
     beginExpression(JSPrecedence.CALL_PRECEDENCE);
     use(node.receiver, JSPrecedence.MEMBER_PRECEDENCE);
     buffer.add('.');
     // Avoid adding the generative constructor name to the list of
     // seen selectors.
     if (node.inputs[0] is HForeignNew) {
@@ skipping 167 matching lines @@
   visitLoopBranch(HLoopBranch node) {
     HBasicBlock branchBlock = currentBlock;
     handleLoopCondition(node);
     List<HBasicBlock> dominated = currentBlock.dominatedBlocks;
     // For a do while loop, the body has already been visited.
     if (!node.isDoWhile()) {
       visitBasicBlock(dominated[0]);
     }
     endLoop(node.block);
     visitBasicBlock(branchBlock.successors[1]);
-    // TODO(floitsch): with labeled breaks we can have more dominated blocks.
-    assert(dominated.length <= 3);
-    if (dominated.length == 3) {
-      // This happens when the body contains a 'return', and the exit-block is
-      // not dominated by a dominator of the while loop.
-      assert(dominated[2].isExitBlock());
-      visitBasicBlock(dominated[2]);
+    // With labeled breaks we can have more dominated blocks.
+    if (dominated.length >= 3) {
+      for (int i = 2; i < dominated.length; i++) {
+        visitBasicBlock(dominated[i]);
+      }
     }
   }
 
   visitNot(HNot node) {
     assert(node.inputs.length == 1);
     beginExpression(JSPrecedence.PREFIX_PRECEDENCE);
     buffer.add('!');
     use(node.inputs[0], JSPrecedence.PREFIX_PRECEDENCE);
     endExpression(JSPrecedence.PREFIX_PRECEDENCE);
   }
@@ skipping 345 matching lines @@
       checkArray(input, '!==');
       buffer.add(')) ');
       bailout(node, 'Not a string or array');
     } else {
       unreachable();
     }
   }
 
   void beginLoop(HBasicBlock block) {
     addIndentation();
+    for (SourceString label in block.loopInformation.labels) {
+      buffer.add("${label.stringValue}:");
+    }
     buffer.add('while (true) {\n');
     indent++;
   }
 
   void endLoop(HBasicBlock block) {
     indent--;
     addIndentation();
     buffer.add('}\n');  // Close 'while' loop.
   }
 
@@ skipping 141 matching lines @@
     indent++;
   }
 
   void endBailoutSwitch() {
     indent--; // Close 'case'.
     indent--;
     addIndentation();
     buffer.add('}\n');  // Close 'switch'.
   }
 
+  // Adds a "$" in front of names of labels from the original source.
+  // This avoids conflicts with labels introduced by bailouts, which
+  // starts with a non-"$" character.
+  void addLabel(SourceString label) {
+    buffer.add("\$$label");
+  }
+
   void beginLoop(HBasicBlock block) {
     // TODO(ngeoffray): Don't put labels on loops that don't bailout.
     String newLabel = pushLabel();
     if (block.hasBailouts()) {
       startBailoutCase(block.bailouts, const <HBailoutTarget>[]);
     }
 
     addIndentation();
+    for (SourceString label in block.loopInformation.labels) {
+      addLabel(label);
+      buffer.add(":");
+    }
     buffer.add('$newLabel: while (true) {\n');
     indent++;
 
     if (block.hasBailouts()) {
       startBailoutSwitch();
     }
   }
 
   void endLoop(HBasicBlock block) {
     popLabel();
@@ skipping 69 matching lines @@
       startBailoutSwitch();
     }
   }
 
   void endElse(HIf node) {
     if (node.elseBlock.hasBailouts()) {
       endBailoutSwitch();
     }
   }
 }