(refs #7375) Replaced collectFreeVariables method by a visitor
[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / ERealLiteral.java
1 package org.simantics.scl.compiler.elaboration.expressions;
2
3
4 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
5 import org.simantics.scl.compiler.common.names.Names;
6 import org.simantics.scl.compiler.constants.DoubleConstant;
7 import org.simantics.scl.compiler.constants.FloatConstant;
8 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
9 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
10 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
11 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
12 import org.simantics.scl.compiler.errors.ErrorLog;
13 import org.simantics.scl.compiler.errors.Locations;
14 import org.simantics.scl.compiler.types.Type;
15 import org.simantics.scl.compiler.types.Types;
16 import org.simantics.scl.compiler.types.exceptions.MatchException;
17
18 import gnu.trove.map.hash.TObjectIntHashMap;
19 import gnu.trove.set.hash.TIntHashSet;
20
21 public class ERealLiteral extends SimplifiableExpression {
22     public String value;
23     public EVariable constraint;
24
25     public ERealLiteral(String value) {
26         this.value = value;
27     }
28
29     @Override
30     public void collectVars(TObjectIntHashMap<Variable> allVars,
31             TIntHashSet vars) {
32     }
33
34     private Expression tryToConvertToPrimitive(ErrorLog errorLog, Type requiredType) {
35         if(requiredType.equals(Types.DOUBLE))
36             return new ELiteral(new DoubleConstant(Double.parseDouble(value)));
37         else if(requiredType.equals(Types.FLOAT))
38             return new ELiteral(new FloatConstant(Float.parseFloat(value)));
39         else if(requiredType.equals(Types.INTEGER)) {
40             errorLog.log(location, "Cannot convert real literal to Integer.");
41             return new EError();
42         }
43         else if(requiredType.equals(Types.LONG)) {
44             errorLog.log(location, "Cannot convert real literal to Long.");
45             return new EError();
46         }
47         else
48             return null;
49     }
50     
51     @Override
52     public Expression checkBasicType(TypingContext context, Type requiredType) {
53         requiredType = Types.canonical(requiredType);
54         Expression primitive = tryToConvertToPrimitive(context.getErrorLog(), requiredType);
55         if(primitive != null)
56             return primitive;
57         else {
58             setType(requiredType);        
59             constraint = new EVariable(location, null);
60             constraint.setType(Types.pred(Types.REAL, requiredType));
61             context.addConstraintDemand(constraint);
62             return this;
63         }
64     }
65
66     @Override
67     protected void updateType() throws MatchException {
68         throw new InternalCompilerError("TODO");
69     }
70
71     @Override
72     public Expression simplify(SimplificationContext context) {
73         Expression primitive = tryToConvertToPrimitive(context.getErrorLog(), getType());
74         if(primitive != null)
75             return primitive;
76         return context.apply(
77                 context.getConstant(Names.Prelude_fromDouble, getType()),
78                 constraint.simplify(context),
79                 context.literal(new DoubleConstant(Double.parseDouble(value)))
80                 );
81     }
82
83     @Override
84     public Expression resolve(TranslationContext context) {
85         return this;
86     }
87     
88     @Override
89     public Expression resolveAsPattern(TranslationContext context) {
90         return new ELiteral(new DoubleConstant(Double.parseDouble(value)));
91     }
92     
93     @Override
94     public Expression replace(ReplaceContext context) {
95         ERealLiteral copy = new ERealLiteral(value);
96         copy.setType(getType().replace(context.tvarMap));
97         return copy;
98     }
99     
100     @Override
101     public void setLocationDeep(long loc) {
102         if(location == Locations.NO_LOCATION) {
103             location = loc;
104             if(constraint != null)
105                 constraint.setLocationDeep(loc);
106         }
107     }
108     
109     @Override
110     public void accept(ExpressionVisitor visitor) {
111         visitor.visit(this);
112     }
113
114     public String getValue() {
115         return value;
116     }
117     
118     @Override
119     public Expression accept(ExpressionTransformer transformer) {
120         return transformer.transform(this);
121     }
122
123 }