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