Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / EIntegerLiteral.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EIntegerLiteral.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EIntegerLiteral.java
new file mode 100755 (executable)
index 0000000..7ccd66d
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EIntegerLiteral.java
@@ -0,0 +1,165 @@
+package org.simantics.scl.compiler.elaboration.expressions;\r
+\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+import gnu.trove.set.hash.THashSet;\r
+import gnu.trove.set.hash.TIntHashSet;\r
+\r
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;\r
+import org.simantics.scl.compiler.constants.DoubleConstant;\r
+import org.simantics.scl.compiler.constants.FloatConstant;\r
+import org.simantics.scl.compiler.constants.IntegerConstant;\r
+import org.simantics.scl.compiler.constants.LongConstant;\r
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
+import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
+import org.simantics.scl.compiler.types.Type;\r
+import org.simantics.scl.compiler.types.Types;\r
+import org.simantics.scl.compiler.types.exceptions.MatchException;\r
+\r
+public class EIntegerLiteral extends SimplifiableExpression {\r
+    public String value;\r
+    EVariable constraint;\r
+\r
+    public EIntegerLiteral(String value) {\r
+        this.value = value;\r
+    }\r
+\r
+    @Override\r
+    public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
+            TIntHashSet refs) {\r
+    }\r
+\r
+    @Override\r
+    public void collectVars(TObjectIntHashMap<Variable> allVars,\r
+            TIntHashSet vars) {\r
+    }\r
+    \r
+    public String getValue() {\r
+        return value;\r
+    }\r
+    \r
+    private Expression tryToConvertToPrimitive(Type requiredType) {\r
+        if(requiredType.equals(Types.INTEGER))\r
+            return new ELiteral(new IntegerConstant(Integer.parseInt(value)));\r
+        else if(requiredType.equals(Types.DOUBLE))\r
+            return new ELiteral(new DoubleConstant(Double.parseDouble(value)));\r
+        else if(requiredType.equals(Types.FLOAT))\r
+            return new ELiteral(new FloatConstant(Float.parseFloat(value)));\r
+        else if(requiredType.equals(Types.LONG))\r
+            return new ELiteral(new LongConstant(Long.parseLong(value)));\r
+        else\r
+            return null;\r
+    }\r
+    \r
+    @Override\r
+    public Expression checkBasicType(TypingContext context, Type requiredType) {\r
+        requiredType = Types.canonical(requiredType);\r
+        \r
+        try {\r
+            Expression primitive = tryToConvertToPrimitive(requiredType);\r
+            if(primitive != null)\r
+                return primitive;\r
+        } catch(NumberFormatException e) {\r
+            context.getErrorLog().log(getLocation(), "Invalid number format.");\r
+        }\r
+        \r
+        setType(requiredType);\r
+        constraint = new EVariable(location, null);\r
+        constraint.setType(Types.pred(Types.RING, requiredType));\r
+        context.addConstraintDemand(constraint);\r
+        return this;        \r
+    }\r
+    \r
+    @Override\r
+    protected void updateType() throws MatchException {\r
+        throw new InternalCompilerError();\r
+    }\r
+\r
+    @Override\r
+    public void collectFreeVariables(THashSet<Variable> vars) {\r
+    }\r
+\r
+    @Override\r
+    public Expression simplify(SimplificationContext context) {\r
+        try {\r
+            Expression primitive = tryToConvertToPrimitive(Types.canonical(getType()));\r
+            if(primitive != null)\r
+                return primitive;\r
+            return context.apply(\r
+                    context.getConstant(SimplificationContext.FROM_INTEGER, getType()),\r
+                    constraint.simplify(context),\r
+                    context.literal(new IntegerConstant(Integer.parseInt(value)))\r
+                    );\r
+        } catch(NumberFormatException e) {\r
+            context.getErrorLog().log(getLocation(), "Invalid number format (maybe too long for the expected number type).");\r
+            return this;\r
+        }\r
+    }\r
+\r
+    @Override\r
+    public Expression resolve(TranslationContext context) {\r
+        return this;\r
+    }\r
+    \r
+    @Override\r
+    public Expression resolveAsPattern(TranslationContext context) {\r
+        return this;\r
+    }\r
+    \r
+    @Override\r
+    public Expression replace(ReplaceContext context) {\r
+        EIntegerLiteral copy = new EIntegerLiteral(value);\r
+        copy.setType(getType().replace(context.tvarMap));\r
+        copy.constraint = (EVariable)constraint.replace(context);\r
+        return copy;\r
+    }\r
+\r
+    @Override\r
+    public Expression decorate(ExpressionDecorator decorator) {     \r
+        return decorator.decorate(this);\r
+    }\r
+    \r
+    @Override\r
+    public boolean isEffectful() {\r
+       return false;\r
+    }\r
+\r
+    @Override\r
+    public void collectEffects(THashSet<Type> effects) {\r
+    }\r
+    \r
+    @Override\r
+    public void setLocationDeep(long loc) {\r
+        if(location == Locations.NO_LOCATION) {\r
+            location = loc;\r
+            if(constraint != null)\r
+                constraint.setLocationDeep(loc);\r
+        }\r
+    }\r
+    \r
+    @Override\r
+    public void accept(ExpressionVisitor visitor) {\r
+        visitor.visit(this);\r
+    }\r
+\r
+    @Override\r
+    public void forVariables(VariableProcedure procedure) {\r
+        if(constraint != null)\r
+            constraint.forVariables(procedure);\r
+    }\r
+    \r
+    @Override\r
+    public boolean isPattern(int arity) {\r
+        return arity == 0;\r
+    }\r
+    \r
+    @Override\r
+    public Expression accept(ExpressionTransformer transformer) {\r
+        return transformer.transform(this);\r
+    }\r
+\r
+}\r