Split TypedSelector into Selector and TypeMask.

pkg/compiler/lib/src/resolution/tree_elements.dart

 // Copyright (c) 2015, 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.
 
 part of resolution;
 
 abstract class TreeElements {
   AnalyzableElement get analyzedElement;
   Iterable<Node> get superUses;
 
   /// Iterables of the dependencies that this [TreeElement] records of
   /// [analyzedElement].
   Iterable<Element> get allElements;
   void forEachConstantNode(f(Node n, ConstantExpression c));
 
   /// A set of additional dependencies.  See [registerDependency] below.
   Iterable<Element> get otherDependencies;
 
   Element operator[](Node node);
 
   SendStructure getSendStructure(Send send);
 
   // TODO(johnniwinther): Investigate whether [Node] could be a [Send].
   Selector getSelector(Node node);
   Selector getGetterSelectorInComplexSendSet(SendSet node);
   Selector getOperatorSelectorInComplexSendSet(SendSet node);
   DartType getType(Node node);
-  void setSelector(Node node, Selector selector);
-  void setGetterSelectorInComplexSendSet(SendSet node, Selector selector);
-  void setOperatorSelectorInComplexSendSet(SendSet node, Selector selector);
+  TypeMask getTypeMask(Node node);
+  TypeMask getGetterTypeMaskInComplexSendSet(SendSet node);
+  TypeMask getOperatorTypeMaskInComplexSendSet(SendSet node);
+  void setTypeMask(Node node, TypeMask mask);
+  void setGetterTypeMaskInComplexSendSet(SendSet node, TypeMask mask);
+  void setOperatorTypeMaskInComplexSendSet(SendSet node, TypeMask mask);
 
   /// Returns the for-in loop variable for [node].
   Element getForInVariable(ForIn node);
   Selector getIteratorSelector(ForIn node);
   Selector getMoveNextSelector(ForIn node);
   Selector getCurrentSelector(ForIn node);
-  void setIteratorSelector(ForIn node, Selector selector);
-  void setMoveNextSelector(ForIn node, Selector selector);
-  void setCurrentSelector(ForIn node, Selector selector);
+  TypeMask getIteratorTypeMask(ForIn node);
+  TypeMask getMoveNextTypeMask(ForIn node);
+  TypeMask getCurrentTypeMask(ForIn node);
+  void setIteratorTypeMask(ForIn node, TypeMask mask);
+  void setMoveNextTypeMask(ForIn node, TypeMask mask);
+  void setCurrentTypeMask(ForIn node, TypeMask mask);
   void setConstant(Node node, ConstantExpression constant);
   ConstantExpression getConstant(Node node);
   bool isAssert(Send send);
 
   /// Returns the [FunctionElement] defined by [node].
   FunctionElement getFunctionDefinition(FunctionExpression node);
 
   /// Returns target constructor for the redirecting factory body [node].
   ConstructorElement getRedirectingTargetConstructor(
       RedirectingFactoryBody node);
@@ skipping 32 matching lines @@
   /// Returns the jump target of the [node].
   JumpTarget getTargetOf(GotoStatement node);
 
   /// Returns the label defined by [node].
   LabelDefinition getLabelDefinition(Label node);
 
   /// Returns the label that [node] targets.
   LabelDefinition getTargetLabel(GotoStatement node);
 }
 
-class TreeElementMapping implements TreeElements {
+class TreeElementMapping extends TreeElements {
   final AnalyzableElement analyzedElement;
   Map<Spannable, Selector> _selectors;
+  Map<Spannable, TypeMask> _typeMasks;
   Map<Node, DartType> _types;
   Setlet<Node> _superUses;
   Setlet<Element> _otherDependencies;
   Map<Node, ConstantExpression> _constants;
   Map<VariableElement, List<Node>> _potentiallyMutated;
   Map<Node, Map<VariableElement, List<Node>>> _potentiallyMutatedIn;
   Map<VariableElement, List<Node>> _potentiallyMutatedInClosure;
   Map<Node, Map<VariableElement, List<Node>>> _accessedByClosureIn;
   Setlet<Element> _elements;
   Setlet<Send> _asserts;
@@ skipping 329 matching lines @@
     if (_targetLabels == null) {
       _targetLabels = new Maplet<GotoStatement, LabelDefinition>();
     }
     _targetLabels[node] = label;
   }
 
   LabelDefinition getTargetLabel(GotoStatement node) {
     assert(node.target != null);
     return _targetLabels != null ? _targetLabels[node] : null;
   }
+
+  TypeMask _getTypeMask(Spannable node) {
+    return _typeMasks != null ? _typeMasks[node] : null;
+  }
+
+  void _setTypeMask(Spannable node, TypeMask mask) {
+    if (_typeMasks == null) {
+      _typeMasks = new Maplet<Spannable, TypeMask>();
+    }
+    _typeMasks[node] = mask;
+  }
+
+  void setTypeMask(Node node, TypeMask mask) {
+    _setTypeMask(node, mask);
+  }
+
+  TypeMask getTypeMask(Node node) => _getTypeMask(node);
+
+  void setGetterTypeMaskInComplexSendSet(SendSet node, TypeMask mask) {
+    _setTypeMask(node.selector, mask);
+  }
+
+  TypeMask getGetterTypeMaskInComplexSendSet(SendSet node) {
+    return _getTypeMask(node.selector);
+  }
+
+  void setOperatorTypeMaskInComplexSendSet(SendSet node, TypeMask mask) {
+    _setTypeMask(node.assignmentOperator, mask);
+  }
+
+  TypeMask getOperatorTypeMaskInComplexSendSet(SendSet node) {
+    return _getTypeMask(node.assignmentOperator);
+  }
+
+  // The following methods set selectors on the "for in" node. Since
+  // we're using three selectors, we need to use children of the node,
+  // and we arbitrarily choose which ones.
+
+  void setIteratorTypeMask(ForIn node, TypeMask mask) {
+    _setTypeMask(node, mask);
+  }
+
+  TypeMask getIteratorTypeMask(ForIn node) {
+    return _getTypeMask(node);
+  }
+
+  void setMoveNextTypeMask(ForIn node, TypeMask mask) {
+    _setTypeMask(node.forToken, mask);
+  }
+
+  TypeMask getMoveNextTypeMask(ForIn node) {
+    return _getTypeMask(node.forToken);
+  }
+
+  void setCurrentTypeMask(ForIn node, TypeMask mask) {
+    _setTypeMask(node.inToken, mask);
+  }
+
+  TypeMask getCurrentTypeMask(ForIn node) {
+    return _getTypeMask(node.inToken);
+  }
 }
 
 TreeElements _ensureTreeElements(AnalyzableElementX element) {
   if (element._treeElements == null) {
     element._treeElements = new TreeElementMapping(element);
   }
   return element._treeElements;
 }
 
 abstract class AnalyzableElementX implements AnalyzableElement {
   TreeElements _treeElements;
 
   bool get hasTreeElements => _treeElements != null;
 
   TreeElements get treeElements {
     assert(invariant(this, _treeElements !=null,
         message: "TreeElements have not been computed for $this."));
     return _treeElements;
   }
 
   void reuseElement() {
     _treeElements = null;
   }
 }