Switch-implementation.

frog/leg/resolver.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.
 
 interface TreeElements {
   Element operator[](Node node);
   Selector getSelector(Send send);
 }
 
 class TreeElementMapping implements TreeElements {
@@ skipping 469 matching lines @@
 
 class LabeledStatementLabelScope implements LabelScope {
   final LabelScope outer;
   final LabelElement label;
   LabeledStatementLabelScope(this.outer, this.label);
   LabelElement lookup(String labelName) {
     if (this.label.labelName == labelName) return label;
     return outer.lookup(labelName);
   }
 }
 

[ahe, 2012/03/09 13:58:32]

Remove extra line

[Lasse Reichstein Nielsen, 2012/03/12 13:05:23]

Done.

+
+class SwitchLabelScope implements LabelScope {
+  final LabelScope outer;
+  final Map<String, LabelElement> caseLabels;
+  SwitchLabelScope(this.outer, this.caseLabels);
+  LabelElement lookup(String labelName) {
+    LabelElement result = caseLabels[labelName];
+    if (result !== null) return result;
+    return outer.lookup(labelName);
+  }
+}
+

[ahe, 2012/03/09 13:58:32]

Extra line.

[Lasse Reichstein Nielsen, 2012/03/12 13:05:23]

Done.

+
 class EmptyLabelScope implements LabelScope {
   const EmptyLabelScope();
   LabelElement lookup(String label) => null;
   LabelScope get outer() {
     throw 'internal error: empty label scope has no outer';
   }
 }
 
 class StatementScope {
   LabelScope labels;
   Link<StatementElement> breakTargetStack;
   Link<StatementElement> continueTargetStack;
 
   StatementScope()
       : labels = const EmptyLabelScope(),
         breakTargetStack = const EmptyLink<StatementElement>(),
         continueTargetStack = const EmptyLink<StatementElement>();
 
-  LabelElement lookupLabel(String label) => labels.lookup(label);
+  LabelElement lookupLabel(String label) =>
+      labels.lookup(label);
+
   StatementElement currentBreakTarget() =>
     breakTargetStack.isEmpty() ? null : breakTargetStack.head;
+
   StatementElement currentContinueTarget() =>
     continueTargetStack.isEmpty() ? null : continueTargetStack.head;
 
   void enterLabelScope(LabelElement element) {
     labels = new LabeledStatementLabelScope(labels, element);
   }
+
   void exitLabelScope() {
     labels = labels.outer;
   }
+
   void enterLoop(StatementElement element) {
     breakTargetStack = breakTargetStack.prepend(element);
     continueTargetStack = continueTargetStack.prepend(element);
   }
+
   void exitLoop() {
     breakTargetStack = breakTargetStack.tail;
     continueTargetStack = continueTargetStack.tail;
   }
+
+  void enterSwitch(StatementElement breakElement,
+                   Map<String, LabelElement> continueElements) {
+    breakTargetStack = breakTargetStack.prepend(breakElement);
+    labels = new SwitchLabelScope(labels, continueElements);
+  }
+
+  void exitSwitch() {
+    breakTargetStack = breakTargetStack.tail;
+    labels = labels.outer;
+  }
 }
 
 
 class ResolverVisitor extends CommonResolverVisitor<Element> {
   final TreeElementMapping mapping;
   final Element enclosingElement;
   bool inInstanceContext;
   Scope context;
   ClassElement currentClass;
   bool typeRequired = false;
   StatementScope statementScope;
   int allowedCategory = ElementCategory.VARIABLE | ElementCategory.FUNCTION;
+  // For generating locally unique ids.
+  int sequence = 0;
 
   ResolverVisitor(Compiler compiler, Element element)
     : this.mapping  = new TreeElementMapping(),
       this.enclosingElement = element,
       inInstanceContext = element.isInstanceMember()
           || element.isGenerativeConstructor(),
       this.context  = element.isMember()
         ? new ClassScope(element.enclosingElement, element.getLibrary())
         : new TopScope(element.getLibrary()),
       this.currentClass = element.isMember() ? element.enclosingElement : null,
       this.statementScope = new StatementScope(),
       super(compiler);
 
   Element lookup(Node node, SourceString name) {
     Element result = context.lookup(name);
     if (!inInstanceContext && result != null && result.isInstanceMember()) {
       error(node, MessageKind.NO_INSTANCE_AVAILABLE, [node]);
     }
     return result;
   }
 
   // Create, or reuse an already created, statement element for a statement.
   StatementElement getOrCreateStatementElement(Node statement) {
     StatementElement element = mapping[statement];
     if (element !== null) return element;
-    element = new StatementElement(statement, enclosingElement);
+    if (statement is SwitchStatement) {

[ahe, 2012/03/09 13:58:32]

statement.asSwitchStatement() !== null

[Lasse Reichstein Nielsen, 2012/03/12 13:05:23]

Why use something that's harder to read and is possibly less efficient?
I.e., BUT WHY?!?!?

[Lasse Reichstein Nielsen, 2012/03/12 13:21:28]

But I rewrote it anyway :)

+      element = new SwitchStatementElement(statement,
+                                           enclosingElement,
+                                           sequence++);
+    } else {
+      element = new StatementElement(statement, enclosingElement);
+    }
     mapping[statement] = element;
     return element;
   }
 
   inStaticContext(action()) {
     bool wasInstanceContext = inInstanceContext;
     inInstanceContext = false;
     action();
     inInstanceContext = wasInstanceContext;
   }
@@ skipping 453 matching lines @@
       }
       target.isBreakTarget = true;
     } else {
       String labelName = node.target.source.slowToString();
       LabelElement label = statementScope.lookupLabel(labelName);
       if (label === null) {
         error(node.target, MessageKind.UNBOUND_LABEL, [labelName]);
         return;
       }
       target = label.target;
+      if (!target.statement.isValidBreakTarget()) {
+        error(node.target, MessageKind.INVALID_BREAK, [labelName]);
+        return;
+      }
       label.setBreakTarget();
       mapping[node.target] = label;
     }
     mapping[node] = target;
   }
 
   visitContinueStatement(ContinueStatement node) {
     StatementElement target;
     if (node.target === null) {
       target = statementScope.currentContinueTarget();
@@ skipping 68 matching lines @@
   visitLiteralMapEntry(LiteralMapEntry node) {
     node.visitChildren(this);
   }
 
   visitNamedArgument(NamedArgument node) {
     visit(node.expression);
   }
 
   visitSwitchStatement(SwitchStatement node) {
     node.expression.accept(this);
-    StatementElement element = getOrCreateStatementElement(node);
-    statementScope.enterLoop(element);
+
+    StatementElement breakElement = getOrCreateStatementElement(node);
+    Map<String, LabelElement> continueLabels = <LabelElement>{};
+    for (SwitchCase switchCase in node.cases) {
+      if (switchCase.label !== null) {
+        Identifier labelIdentifier = switchCase.label;
+        String labelName = labelIdentifier.source.slowToString();
+
+        LabelElement existingElement = continueLabels[labelName];
+        if (existingElement !== null) {
+          // It's an error if the same label occurs twice in the same switch.
+          warning(labelIdentifier, MessageKind.DUPLICATE_LABEL, [labelName]);
+          error(existingElement.label, MessageKind.EXISTING_LABEL, [labelName]);
+        } else {
+          // It's only a warning if it shadows another label.
+          existingElement = statementScope.lookupLabel(labelName);
+          if (existingElement !== null) {
+            warning(labelIdentifier, MessageKind.DUPLICATE_LABEL, [labelName]);
+            warning(existingElement.label,
+                    MessageKind.EXISTING_LABEL, [labelName]);
+          }
+        }
+
+        StatementElement statementElement =
+            new StatementElement(switchCase, enclosingElement);
+        mapping[switchCase] = statementElement;
+
+        LabelElement label =
+            new LabelElement(labelIdentifier, labelName,
+                             statementElement, enclosingElement);
+        mapping[labelIdentifier] = label;
+        continueLabels[labelName] = label;
+      }
+    }
+    statementScope.enterSwitch(breakElement, continueLabels);
     node.cases.accept(this);
-    statementScope.exitLoop();
+    statementScope.exitSwitch();
+
+    // Clean-up unused labels
+    continueLabels.forEach((String key, LabelElement label) {
+      StatementElement statementElement = label.target;
+      SwitchCase switchCase = statementElement.statement;
+      if (!label.isContinueTarget) {
+        mapping.remove(switchCase);
+        mapping.remove(label.label);
+      }
+    });
   }
 
   visitSwitchCase(SwitchCase node) {
-    // TODO(ahe): What about the label?
-    node.expression.accept(this);
+    // The label was handled in [visitSwitchStatement(SwitchStatement)].
+    node.expressions.accept(this);
     visitIn(node.statements, new BlockScope(context));
   }
 
-  visitDefaultCase(DefaultCase node) {
-    // TODO(ahe): What about the label?
-    visitIn(node.statements, new BlockScope(context));
-  }
-
   visitTryStatement(TryStatement node) {
     visit(node.tryBlock);
     if (node.catchBlocks.isEmpty() && node.finallyBlock == null) {
       // TODO(ngeoffray): The precise location is
       // node.getEndtoken.next. Adjust when issue #1581 is fixed.
       error(node, MessageKind.NO_CATCH_NOR_FINALLY);
     }
     visit(node.catchBlocks);
     visit(node.finallyBlock);
   }
@@ skipping 409 matching lines @@
 class TopScope extends Scope {
   LibraryElement get library() => element;
 
   TopScope(LibraryElement library) : super(null, library);
   Element lookup(SourceString name) => library.find(name);
 
   Element add(Element element) {
     throw "Cannot add an element in the top scope";
   }
 }