(refs #7375) Replaced collectFreeVariables method by a visitor
[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / EEnforce.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.elaboration.contexts.ReplaceContext;
5 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
6 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
7 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
8 import org.simantics.scl.compiler.elaboration.query.QExists;
9 import org.simantics.scl.compiler.elaboration.query.Query;
10 import org.simantics.scl.compiler.elaboration.query.compilation.EnforcingContext;
11 import org.simantics.scl.compiler.errors.Locations;
12 import org.simantics.scl.compiler.types.Types;
13 import org.simantics.scl.compiler.types.exceptions.MatchException;
14
15 import gnu.trove.map.hash.TObjectIntHashMap;
16 import gnu.trove.set.hash.TIntHashSet;
17
18 public class EEnforce extends SimplifiableExpression {
19     
20     public Query query;
21     
22     public EEnforce(Query query) {
23         this.query = query;
24     }
25     
26     public Query getQuery() {
27         return query;
28     }
29
30     @Override
31     public void collectVars(TObjectIntHashMap<Variable> allVars,
32             TIntHashSet vars) {
33         query.collectVars(allVars, vars);
34     }
35
36     @Override
37     protected void updateType() throws MatchException {
38         setType(Types.tupleConstructor(0));
39     }
40     
41     @Override
42     public Expression inferType(TypingContext context) {
43         query.checkType(context);
44         //context.declareEffect(location, query.getEffect(Query.W));
45         return compile(context);
46     }
47     
48     public Expression compile(TypingContext context) {
49         return query.generateEnforce(new EnforcingContext(context));
50     }
51
52     @Override
53     public Expression simplify(SimplificationContext context) {
54         /*query = query.simplify(context);
55         return query.generateEnforce(context);*/
56         throw new InternalCompilerError();
57     }
58
59     @Override
60     public Expression resolve(TranslationContext context) {
61         context.pushExistentialFrame();
62         query = query.resolve(context);
63         Variable[] variables = context.popExistentialFrame();
64         if(variables.length > 0)
65             query = new QExists(variables, query);
66         return this;
67     }
68     
69     @Override
70     public void setLocationDeep(long loc) {
71         if(location == Locations.NO_LOCATION) {
72             location = loc;
73             query.setLocationDeep(loc);
74         }
75     }
76     
77     @Override
78     public void accept(ExpressionVisitor visitor) {
79         visitor.visit(this);
80     }
81     
82     @Override
83     public Expression replace(ReplaceContext context) {
84         return new EEnforce(query.replace(context));
85     }
86     
87     @Override
88     public Expression accept(ExpressionTransformer transformer) {
89         return transformer.transform(this);
90     }
91
92 }