Test the CapturedScope for local functions

tests/compiler/dart2js/closure/closure_test.dart

 // Copyright (c) 2017, 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.
 
 import 'dart:io' hide Link;
 import 'package:async_helper/async_helper.dart';
 import 'package:compiler/src/closure.dart';
 import 'package:compiler/src/common.dart';
 import 'package:compiler/src/compiler.dart';
 import 'package:compiler/src/diagnostics/diagnostic_listener.dart';
@@ skipping 77 matching lines @@
   ClosureAstComputer(DiagnosticReporter reporter, Map<Id, ActualData> actualMap,
       ResolvedAst resolvedAst, this.closureDataLookup,
       {this.verbose: false})
       : super(reporter, actualMap, resolvedAst) {
     pushMember(resolvedAst.element as MemberElement);
   }
 
   visitFunctionExpression(ast.FunctionExpression node) {
     Entity localFunction = resolvedAst.elements.getFunctionDefinition(node);
     if (localFunction is LocalFunctionElement) {
+      pushMember(localFunction.callMethod);
       pushLocalFunction(node);
       super.visitFunctionExpression(node);
       popLocalFunction();
+      popMember();
     } else {
       super.visitFunctionExpression(node);
     }
   }
 
   @override
   String computeNodeValue(ast.Node node, [AstElement element]) {
     if (element != null && element.isLocal) {
       if (element.isFunction) {
-        return computeObjectValue(element);
+        LocalFunctionElement localFunction = element;
+        return computeObjectValue(localFunction.callMethod);
       } else {
         LocalElement local = element;
         return computeLocalValue(local);
       }
     }
     // TODO(johnniwinther,efortuna): Collect data for other nodes?
     return null;
   }
 
   @override
-  String computeElementValue(AstElement element) {
+  String computeElementValue(covariant MemberElement element) {
     // TODO(johnniwinther,efortuna): Collect data for the member
     // (has thisLocal, has box, etc.).
     return computeObjectValue(element);
   }
 }
 
 /// Kernel IR visitor for computing closure data.
 class ClosureIrChecker extends IrDataExtractor with ComputeValueMixin<ir.Node> {
   final MemberEntity member;
   final ClosureDataLookup<ir.Node> closureDataLookup;
   final KernelToLocalsMap _localsMap;
   final bool verbose;
 
   ClosureIrChecker(
       Map<Id, ActualData> actualMap,
       KernelToElementMapForBuilding elementMap,
       this.member,
       this._localsMap,
       this.closureDataLookup,
       {this.verbose: false})
       : super(actualMap) {
     pushMember(member);
   }
 
   visitFunctionExpression(ir.FunctionExpression node) {
+    ClosureRepresentationInfo info = closureDataLookup.getClosureInfo(node);
+    pushMember(info.callMethod);
     pushLocalFunction(node);
     super.visitFunctionExpression(node);
     popLocalFunction();
+    popMember();
   }
 
   visitFunctionDeclaration(ir.FunctionDeclaration node) {
+    ClosureRepresentationInfo info = closureDataLookup.getClosureInfo(node);
+    pushMember(info.callMethod);
     pushLocalFunction(node);
     super.visitFunctionDeclaration(node);
     popLocalFunction();
+    popMember();
   }
 
   @override
   String computeNodeValue(ir.Node node) {
     if (node is ir.VariableDeclaration) {
       if (node.parent is ir.FunctionDeclaration) {
-        return computeObjectValue(node.parent);
+        ClosureRepresentationInfo info =
+            closureDataLookup.getClosureInfo(node.parent);
+        return computeObjectValue(info.callMethod);
       }
       Local local = _localsMap.getLocalVariable(node);
       return computeLocalValue(local);
     } else if (node is ir.FunctionExpression) {
-      return computeObjectValue(node);
+      ClosureRepresentationInfo info = closureDataLookup.getClosureInfo(node);
+      return computeObjectValue(info.callMethod);
     }
     return null;
   }
 
   @override
   String computeMemberValue(ir.Member node) {
     return computeObjectValue(member);
   }
 }
 
 abstract class ComputeValueMixin<T> {
   bool get verbose;
+  Map<BoxLocal, String> boxNames = <BoxLocal, String>{};
   ClosureDataLookup<T> get closureDataLookup;
   Link<ScopeInfo> scopeInfoStack = const Link<ScopeInfo>();
   ScopeInfo get scopeInfo => scopeInfoStack.head;
-  CapturedScope capturedScope;
+  CapturedScope get capturedScope => capturedScopeStack.head;
+  Link<CapturedScope> capturedScopeStack = const Link<CapturedScope>();
   Link<ClosureRepresentationInfo> closureRepresentationInfoStack =
       const Link<ClosureRepresentationInfo>();
   ClosureRepresentationInfo get closureRepresentationInfo =>
       closureRepresentationInfoStack.isNotEmpty
           ? closureRepresentationInfoStack.head
           : null;
 
   void pushMember(MemberEntity member) {
     scopeInfoStack =
         scopeInfoStack.prepend(closureDataLookup.getScopeInfo(member));
-    capturedScope = closureDataLookup.getCapturedScope(member);
+    capturedScopeStack =
+        capturedScopeStack.prepend(closureDataLookup.getCapturedScope(member));
+    if (capturedScope.hasBox) {
+      boxNames[capturedScope.box] = 'box${boxNames.length}';
+    }
     dump(member);
   }
 
   void popMember() {
     scopeInfoStack = scopeInfoStack.tail;
+    capturedScopeStack = capturedScopeStack.tail;
   }
 
   void pushLocalFunction(T node) {
     closureRepresentationInfoStack = closureRepresentationInfoStack
         .prepend(closureDataLookup.getClosureInfo(node));
     dump(node);
   }
 
   void popLocalFunction() {
     closureRepresentationInfoStack = closureRepresentationInfoStack.tail;
@@ skipping 23 matching lines @@
     if (scopeInfo.localIsUsedInTryOrSync(local)) {
       features.add('inTry');
       // TODO(johnniwinther,efortuna): Should this be enabled and checked?
       //Expect.isTrue(capturedScope.localIsUsedInTryOrSync(local));
     } else {
       //Expect.isFalse(capturedScope.localIsUsedInTryOrSync(local));
     }
     if (capturedScope.isBoxed(local)) {
       features.add('boxed');
     }
-    if (capturedScope.context == local) {
-      features.add('local');
+    if (capturedScope.box == local) {
+      // TODO(johnniwinther): This can't happen!

[Emily Fortuna, 2017/08/22 23:47:37]

but this is happening? 
if not... throw?

[Johnni Winther, 2017/08/23 07:47:18]

I'm still in the process of making this accurate. I left it in to remember that
we cannot test the box value this way.

+      features.add('box');
     }
     if (capturedScope is CapturedLoopScope) {
       CapturedLoopScope loopScope = capturedScope;
       if (loopScope.boxedLoopVariables.contains(local)) {
         features.add('loop');
       }
     }
     if (closureRepresentationInfo != null) {
       if (closureRepresentationInfo.createdFieldEntities.contains(local)) {
         features.add('field');
       }
-      if (closureRepresentationInfo.isVariableBoxed(local)) {
-        features.add('variable-boxed');
-      }
     }
     // TODO(johnniwinther,efortuna): Add more info?
     return (features.toList()..sort()).join(',');
   }
 
-  String computeObjectValue(Object object) {
+  String computeObjectValue(MemberEntity member) {
     Map<String, String> features = <String, String>{};
 
     void addLocals(String name, forEach(f(Local local, _))) {
       List<String> names = <String>[];
       forEach((Local local, _) {
         if (local is BoxLocal) {
-          names.add('box');
+          names.add(boxNames[local]);
         } else {
           names.add(local.name);
         }
       });
       String value = names.isEmpty ? null : '[${(names..sort()).join(',')}]';
       if (features.containsKey(name)) {
         Expect.equals(
-            features[name], value, "Inconsistent values for $name on $object.");
+            features[name], value, "Inconsistent values for $name on $member.");
       }
       features[name] = value;
     }
 
-    if (object is MemberEntity) {
-      if (scopeInfo.thisLocal != null) {
-        features['hasThis'] = '';
-      }
-
-      if (capturedScope.requiresContextBox) {
-        features['requiresBox'] = '';
-      }
-      addLocals('boxed', capturedScope.forEachBoxedVariable);
+    if (scopeInfo.thisLocal != null) {
+      features['hasThis'] = '';
     }
+    if (capturedScope.hasBox) {
+      //print('------------- $object from $capturedScope');

[Emily Fortuna, 2017/08/22 23:47:37]

remove commented out line?

+      features['box'] = boxNames[capturedScope.box];
+    }
+    addLocals('boxed', capturedScope.forEachBoxedVariable);
 
     if (closureRepresentationInfo != null) {
-      addLocals('boxed', closureRepresentationInfo.forEachBoxedVariable);
-      addLocals('captured', closureRepresentationInfo.forEachCapturedVariable);
       addLocals('free', closureRepresentationInfo.forEachFreeVariable);
     }
 
     StringBuffer sb = new StringBuffer();
     bool needsComma = false;
     for (String name in features.keys.toList()..sort()) {
       String value = features[name];
       if (value != null) {
         if (needsComma) {
           sb.write(',');
         }
         sb.write(name);
         if (value != '') {
           sb.write('=');
           sb.write(value);
         }
         needsComma = true;
       }
     }
     return sb.toString();
   }
 }