dart2js cps: Share interceptors by default and propagate to use later.

pkg/compiler/lib/src/js_backend/codegen/unsugar.dart

 library dart2js.unsugar_cps;
 
 import '../../cps_ir/cps_ir_nodes.dart';
 
 import '../../cps_ir/optimizers.dart' show ParentVisitor;
 import '../../constants/expressions.dart';
 import '../../constants/values.dart';
-import '../../elements/elements.dart' show
-    ClassElement,
-    FieldElement,
-    FunctionElement,
-    Local,
-    ExecutableElement;
+import '../../elements/elements.dart';
 import '../../js_backend/codegen/glue.dart';
 import '../../dart2jslib.dart' show Selector, World;
 import '../../cps_ir/cps_ir_builder.dart' show ThisParameterLocal;
 
 class ExplicitReceiverParameterEntity implements Local {
   String get name => 'receiver';
   final ExecutableElement executableContext;
   ExplicitReceiverParameterEntity(this.executableContext);
   toString() => 'ExplicitReceiverParameterEntity($executableContext)';
 }
 
+/// Suggested name for an interceptor.
+class InterceptorEntity extends Entity {
+  Entity interceptedVariable;
+
+  InterceptorEntity(this.interceptedVariable);
+
+  String get name => interceptedVariable.name + '_';
+}
+
+
 /// Rewrites the initial CPS IR to make Dart semantics explicit and inserts
 /// special nodes that respect JavaScript behavior.
 ///
 /// Performs the following rewrites:
 ///  - Rewrite [IsTrue] in a [Branch] to do boolean conversion.
 ///  - Add interceptors at call sites that use interceptor calling convention.
 ///  - Add explicit receiver argument for methods that are called in interceptor
 ///    calling convention.
 ///  - Convert two-parameter exception handlers to one-parameter ones.
 class UnsugarVisitor extends RecursiveVisitor {
   Glue _glue;
   ParentVisitor _parentVisitor = new ParentVisitor();
 
   Parameter thisParameter;
   Parameter explicitReceiverParameter;
 
+  Map<Primitive, Interceptor> interceptors = <Primitive, Interceptor>{};
+
   // In a catch block, rethrow implicitly throws the block's exception
   // parameter.  This is the exception parameter when nested in a catch
   // block and null otherwise.
   Parameter _exceptionParameter = null;
 
   UnsugarVisitor(this._glue);
 
   void rewrite(FunctionDefinition function) {
     bool inInterceptedMethod = _glue.isInterceptedMethod(function.element);
 
@@ skipping 169 matching lines @@
     // Rethrow can only appear in a catch block.  It throws that block's
     // (wrapped) caught exception.
     Throw replacement = new Throw(_exceptionParameter);
     InteriorNode parent = node.parent;
     parent.body = replacement;
     replacement.parent = parent;
     // The original rethrow does not have any references that we need to
     // worry about unlinking.
   }
 
+  /// Returns an interceptor for the given value, capable of responding to
+  /// [selector].
+  /// 
+  /// A single getInterceptor call will be created per primitive, bound
+  /// immediately after the primitive is bound.
+  /// 
+  /// The type propagation pass will later narrow the set of interceptors
+  /// based on the input type, and the let sinking pass will propagate the
+  /// getInterceptor call closer to its use when this is profitable.
+  Interceptor getInterceptorFor(Primitive prim, Selector selector) {
+    Interceptor interceptor = interceptors[prim];
+    if (interceptor == null) {
+      interceptor = new Interceptor(prim);
+      interceptors[prim] = interceptor;
+      InteriorNode parent = prim.parent;
+      insertLetPrim(interceptor, parent.body);
+      if (prim.hint != null) {
+        interceptor.hint = new InterceptorEntity(prim.hint);
+      }
+    }
+    // Add the interceptor classes that can respond to the given selector.
+    interceptor.interceptedClasses.addAll(
+        _glue.getInterceptedClassesOn(selector));
+    return interceptor;
+  }
+
   processInvokeMethod(InvokeMethod node) {
     Selector selector = node.selector;
     if (!_glue.isInterceptedSelector(selector)) return;
 
     Primitive receiver = node.receiver.definition;
     Primitive newReceiver;
 
     if (receiver == explicitReceiverParameter) {
       // If the receiver is the explicit receiver, we are calling a method in
       // the same interceptor:
       //  Change 'receiver.foo()'  to  'this.foo(receiver)'.
       newReceiver = thisParameter;
     } else {
-      // TODO(sra): Move the computation of interceptedClasses to a much later
-      // phase and take into account the remaining uses of the interceptor.
-      Set<ClassElement> interceptedClasses =
-        _glue.getInterceptedClassesOn(selector);
-      _glue.registerSpecializedGetInterceptor(interceptedClasses);
-      newReceiver = new Interceptor(receiver, interceptedClasses);
-      insertLetPrim(newReceiver, node);
+      newReceiver = getInterceptorFor(receiver, node.selector);
     }
 
     node.arguments.insert(0, node.receiver);
     node.receiver = new Reference<Primitive>(newReceiver);
   }
 
   processInvokeMethodDirectly(InvokeMethodDirectly node) {
     if (_glue.isInterceptedMethod(node.target)) {
       Primitive nullPrim = nullConstant;
       insertLetPrim(nullPrim, node);
@@ skipping 23 matching lines @@
     condition.value.unlink();
     node.trueContinuation.unlink();
     node.falseContinuation.unlink();
     parent.body = newNode;
   }
 
   processInterceptor(Interceptor node) {
     _glue.registerSpecializedGetInterceptor(node.interceptedClasses);
   }
 }