12165399321a2496a6148ff77b8d48422a73c164
[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / elaboration / expressions / ECHRRuleset.java
1 package org.simantics.scl.compiler.elaboration.expressions;
2
3 import org.simantics.scl.compiler.compilation.CompilationContext;
4 import org.simantics.scl.compiler.elaboration.chr.CHRRuleset;
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.errors.Locations;
9 import org.simantics.scl.compiler.internal.codegen.references.IVal;
10 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
11 import org.simantics.scl.compiler.internal.interpreted.IExpression;
12 import org.simantics.scl.compiler.top.ExpressionInterpretationContext;
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 ECHRRuleset extends Expression {
20     CHRRuleset ruleset;
21     Expression in;
22     
23     public ECHRRuleset(CHRRuleset ruleset, Expression in) {
24         this.ruleset = ruleset;
25         this.in = in;
26     }
27     
28     @Override
29     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
30         ruleset.collectVars(allVars, vars);
31         in.collectVars(allVars, vars);
32     }
33
34     @Override
35     protected void updateType() throws MatchException {
36         setType(in.getType());
37     }
38
39     @Override
40     public IVal toVal(CompilationContext context, CodeWriter w) {
41         ruleset.generateCode(w);
42         return in.toVal(context, w);
43     }
44
45     @Override
46     public void collectFreeVariables(THashSet<Variable> vars) {
47         ruleset.collectFreeVariables(vars);
48         in.collectFreeVariables(vars);
49     }
50
51     @Override
52     public Expression resolve(TranslationContext context) {
53         if(context.currentRuleset != null) {
54             context.getErrorLog().log(location, "Current version of SCL compiler does not support nested rulesets.");
55             return this;
56         }
57         context.currentRuleset = ruleset;
58         
59         context.pushFrame();
60         context.pushCHRConstraintFrame();
61         ruleset.resolve(context);
62         in = in.resolve(context);
63         context.popCHRConstraintFrame(ruleset.constraints);
64         context.popFrame();
65         
66         context.currentRuleset = null;
67         
68         return this;
69     }
70     @Override
71     public void setLocationDeep(long loc) {
72         if(location == Locations.NO_LOCATION) {
73             this.location = loc;
74             ruleset.setLocationDeep(loc);
75             in.setLocationDeep(loc);
76         }
77     }
78     
79     @Override
80     public void accept(ExpressionVisitor visitor) {
81         visitor.visit(this);
82     }
83     
84     @Override
85     public Expression inferType(TypingContext context) {
86         ruleset.checkType(context);
87         in = in.inferType(context);
88         return this;
89     }
90     
91     @Override
92     public Expression simplify(SimplificationContext context) {
93         ruleset.simplify(context);
94         ruleset.compile(context);
95         in = in.simplify(context);
96         return this;
97     }
98     
99     @Override
100     public Expression accept(ExpressionTransformer transformer) {
101         return transformer.transform(this);
102     }
103     
104     @Override
105     public IExpression toIExpression(ExpressionInterpretationContext context) {
106         throw new UnsupportedOperationException();
107     }
108
109 }