[CFE] handle set literals in constant evaluator.

askeksa · commit-bot@chromium.org · commit bc2b78886d0a · 2019-03-04T10:20:56.000Z
Change-Id: I9dc5f7a3d2e7bc0f53e0370042bc59d688a8c4b3 Reviewed-on: https://dart-review.googlesource.com/c/95021 Commit-Queue: Aske Simon Christensen <askesc@google.com> Reviewed-by: Kevin Millikin <kmillikin@google.com>
diff --git a/pkg/kernel/lib/transformations/constants.dart b/pkg/kernel/lib/transformations/constants.dart
@@ -562,6 +562,29 @@ class ConstantEvaluator extends RecursiveVisitor {
     return canonicalize(new ListConstant(typeArgument, entries));
   }
 
+  visitSetLiteral(SetLiteral node) {
+    if (!node.isConst) {
+      throw new _AbortCurrentEvaluation(
+          errorReporter.nonConstLiteral(contextChain, node, 'Set'));
+    }
+    final List<Constant> entries = new List<Constant>(node.expressions.length);
+    for (int i = 0; i < node.expressions.length; ++i) {
+      entries[i] = _evaluateSubexpression(node.expressions[i]);
+    }
+    if (hasUnevaluatedChild(node)) {
+      final expressions = new List<Expression>(node.expressions.length);
+      for (int i = 0; i < node.expressions.length; ++i) {
+        expressions[i] = unique(entries[i].asExpression());
+      }
+      return unevaluated(
+          node,
+          new SetLiteral(expressions,
+              typeArgument: node.typeArgument, isConst: true));
+    }
+    final DartType typeArgument = evaluateDartType(node, node.typeArgument);
+    return canonicalize(new SetConstant(typeArgument, entries));
+  }
+
   visitMapLiteral(MapLiteral node) {
     if (!node.isConst) {
       throw new _AbortCurrentEvaluation(
