Allow binary operators on compile-time-constants.

frog/leg/compile_time_constants.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.
 
 /**
  * The [CompileTimeConstantHandler] keeps track of compile-time constants,
  * initializations of global and static fields, and default values of
  * optional parameters.
  */
 class CompileTimeConstantHandler extends CompilerTask {
@@ skipping 78 matching lines @@
           && element.modifiers.isFinal();
     });
   }
 
   StringBuffer writeJsCodeForVariable(StringBuffer buffer,
                                       VariableElement element) {
     var value = initialVariableValues[element];
     if (value === null) {
       buffer.add("(void 0)");
     } else if (value is num) {
-      buffer.add("($value)");
+      if (value.isNaN()) {
+        buffer.add("(0/0)");
+      } else if (value == double.INFINITY) {
+        buffer.add("(1/0)");
+      } else if (value == -double.INFINITY) {
+        buffer.add("(-1/0)");
+      } else {
+        buffer.add("($value)");        
+      }
     } else if (value === true) {
       buffer.add("true");
     } else if (value === false) {
       buffer.add("false");
     } else if (value is DartString) {
       buffer.add("'");
       writeEscapedString(value, buffer, (reason) {
         compiler.cancel("failed to write escaped string: $value");
       });
       buffer.add("'");
     } else {
       // TODO(floitsch): support more values.
       compiler.unimplemented("CompileTimeConstantHandler" +
                              "writeJsCodeForVariable",
-                             node: element.parseNode(compiler));
+                             element: element);
     }
     return buffer;
   }
 
   /**
    * Write the contents of the quoted string to a [StringBuffer] in
    * a form that is valid as JavaScript string literal content.
    * The string is assumed quoted by single quote characters.
    */
   static void writeEscapedString(DartString string,
@@ skipping 64 matching lines @@
   }
 
   visitLiteral(Literal literal) {
     if (literal is LiteralString) {
       assert(literal.asLiteralString().isValidated());
       return literal.asLiteralString().dartString;
     }
     return literal.value;
   }
 
+  // TODO(floitsch): provide better error-messages.
   visitSend(Send send) {
     Element element = definitions[send];
-    if (element !== null && element.kind == ElementKind.FIELD) {
-      if (element.isInstanceMember() ||
-          element.modifiers === null ||
+    if (Elements.isStaticOrTopLevelField(element)) {
+      if (element.modifiers === null ||
           !element.modifiers.isFinal()) {
-        error(element);
+        error(send);
       }
       return constantHandler.compileVariable(element);
+    } else if (send.isPrefix) {
+      assert(send.isOperator);
+      var receiverValue = evaluate(send.receiver);
+      Operator op = send.selector;
+      switch (op.source.stringValue) {
+        case "-":
+          if (receiverValue is !num) error(send);
+          return -receiverValue;
+        case "~": 
+          if (receiverValue is !int) error(send); 
+          return ~receiverValue;
+        case "!":
+          if (receiverValue is !bool) error(send);
+          return !receiverValue;
+        default: error(send);

[ngeoffray, 2012/02/21 11:22:09]

new line after default?

[floitsch, 2012/02/22 10:17:51]

Done.

+      }
+    } else if (send.isOperator && !send.isPostfix) {
+      assert(send.argumentCount() == 1);
+      var left = evaluate(send.receiver);
+      var right = evaluate(send.argumentsNode.nodes.head);
+      String op = send.selector.asOperator().source.stringValue;
+
+      if (op == "==" || op == "===") {
+        // We use == instead of === so that non-canonicalized DartStrings can
+        // use their equality operator.
+        return left == right;
+      } else if (op == "!=" || op == "!==") {
+        // We use == instead of === so that non-canonicalized DartStrings can

[ngeoffray, 2012/02/21 11:22:09]

Remove comment, or adjust it to !=.

[floitsch, 2012/02/22 10:17:51]

Done.

+        // use their equality operator.
+        return left != right;
+      }
+      if (left is num && right is num) {
+        switch (op) {
+          case "+": return left + right;
+          case "-": return left - right;
+          case "*": return left * right;
+          case "/": return left / right;
+          case "~/":
+          case "%":
+            if (left is int && right is int && right == 0) {
+              error(send);
+            }
+            return op == "~/" ? left ~/ right : left % right;
+          case "<": return left < right;
+          case "<=": return left <= right;
+          case ">": return left > right;
+          case ">=": return left >= right;
+        }
+      }
+      if (left is int && right is int) {
+        switch (op) {
+          case "|": return left | right;
+          case "&": return left & right;
+          case "<<":
+            // TODO(floitsch): find a better way to guard against shifts to the
+            // left.
+            if (right > 100) error(send);
+            if (right < 0) error(send);
+            return left << right;
+          case ">>":
+            if (right < 0) error(send);
+            return left >> right;        
+          case "^": return left ^ right;        
+        }
+      }
+      if (left is DartString) {
+        DartString dartString = left;
+        if (op == "+") {
+          if (right is DartString) {
+            return new ConsDartString(dartString, right);

[ngeoffray, 2012/02/21 11:22:09]

Could we have operator+ defined in DartString? (likely to get removed, I know
:))

[floitsch, 2012/02/22 10:17:51]

let's keep it without +.

+          } else if (right is num ||
+                     right is bool ||
+                     right === null) {
+            return new ConsDartString(left,

[ngeoffray, 2012/02/21 11:22:09]

Please move ConsDartString from ssa nodes to ast nodes. Even better if you move
it to util.

[floitsch, 2012/02/22 10:17:51]

Done.

+                                      new DartString.literal(right.toString()));
+          } else {
+            error(send);
+          }
+        } else if (op == "[]" && right is int) {

[ngeoffray, 2012/02/21 11:22:09]

Is "op == '[]'" equivalent to send.isIndex?

[floitsch, 2012/02/22 10:17:51]

Done.

+          if (0 <= right && right < left.length) {
+            Iterator iterator = dartString.iterator();
+            for (int i = 0; i < right; i++) iterator.next();
+            String oneCharString = new String.fromCharCodes([iterator.next()]);
+            return new DartString.literal(oneCharString);
+          } else {
+            error(send);
+          }
+        }
+      }
+    } else if (send.isPropertyAccess) {
+      Identifier selector = send.selector.asIdentifier();
+      if (selector == null) error(send);
+      if (selector.source != const SourceString("length")) error(send);
+      var receiver = evaluate(send.receiver);
+      if (receiver is DartString) {
+        DartString dartString = receiver;
+        return dartString.length;
+      }
+      compiler.unimplemented(
+          "CompileTimeConstantEvaluator 'length' property-lookup on non-string",
+          node: send);
     }
     return super.visitSend(send);
   }
 
-  error(Element element) {
+  visitSendSet(SendSet node) {
+    error(node);
+  }
+
+  error(Node node) {
     // TODO(floitsch): get the list of constants that are currently compiled
     // and present some kind of stack-trace.
     MessageKind kind = MessageKind.NOT_A_COMPILE_TIME_CONSTANT;
-    List arguments = [element.name];
-    Node node = element.parseNode(compiler);
-    compiler.reportError(node, new CompileTimeConstantError(kind, arguments));
+    compiler.reportError(node, new CompileTimeConstantError(kind, const []));
   }
 }