(refs #7375) Replaced collectFreeVariables method by a visitor
[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / ELambdaType.java
1 package org.simantics.scl.compiler.elaboration.expressions;
2
3 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
4 import org.simantics.scl.compiler.compilation.CompilationContext;
5 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
6 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
7 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
8 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
9 import org.simantics.scl.compiler.errors.Locations;
10 import org.simantics.scl.compiler.internal.codegen.references.IVal;
11 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
12 import org.simantics.scl.compiler.internal.interpreted.IExpression;
13 import org.simantics.scl.compiler.top.ExpressionInterpretationContext;
14 import org.simantics.scl.compiler.types.TVar;
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 ELambdaType extends Expression {
22     public TVar[] parameters;
23     public Expression value;
24     
25     public ELambdaType(TVar[] parameters, Expression value) {
26         super(value.getLocation());
27         this.parameters = parameters;
28         this.value = value;
29     }
30
31     @Override
32     public void collectVars(TObjectIntHashMap<Variable> allVars,
33             TIntHashSet vars) {
34         value.collectVars(allVars, vars);
35     }
36
37     @Override
38     protected void updateType() throws MatchException {
39         setType(Types.forAll(parameters, value.getType()));
40     }
41
42     @Override
43     public IVal toVal(CompilationContext context, CodeWriter w) {
44         return lambdaToVal(context, w);
45     }
46
47     @Override
48     public Expression simplify(SimplificationContext context) {
49         value = value.simplify(context);
50         return this;
51     }
52
53     @Override
54     public Expression resolve(TranslationContext context) {
55         value = value.resolve(context);
56         return this;
57     }
58
59     @Override
60     public Expression replace(ReplaceContext context) {
61         TVar[] newParameters = new TVar[parameters.length];
62         for(int i=0;i<parameters.length;++i) {
63             TVar var = Types.var(parameters[i].getKind());
64             newParameters[i] = var;
65             context.tvarMap.put(parameters[i], var);
66         }
67             
68         ELambdaType result = new ELambdaType(newParameters, value.replace(context));
69         for(int i=0;i<parameters.length;++i)
70             context.tvarMap.remove(parameters[i]);
71         return result;
72     }
73     
74     @Override
75     public IExpression toIExpression(ExpressionInterpretationContext target) {
76         return value.toIExpression(target);
77     }
78     
79     @Override
80     public Expression inferType(TypingContext context) {
81         throw new InternalCompilerError("Should not type check " + getClass().getSimpleName() + ".");
82     }
83     
84     @Override
85     public void setLocationDeep(long loc) {
86         if(location == Locations.NO_LOCATION) {
87             location = loc;
88             value.setLocationDeep(loc);
89         }
90     }
91     
92     @Override
93     public void accept(ExpressionVisitor visitor) {
94         visitor.visit(this);
95     }
96     
97     @Override
98     public Expression accept(ExpressionTransformer transformer) {
99         return transformer.transform(this);
100     }
101     
102     @Override
103     public int getSyntacticFunctionArity() {
104         return value.getSyntacticFunctionArity();
105     }
106
107 }