Refactoring CHR handling code
[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.Type;
14 import org.simantics.scl.compiler.types.exceptions.MatchException;
15
16 public class ECHRRuleset extends Expression {
17     public CHRRuleset ruleset;
18     public Expression in;
19     
20     public ECHRRuleset(CHRRuleset ruleset, Expression in) {
21         this.ruleset = ruleset;
22         this.in = in;
23     }
24
25     @Override
26     protected void updateType() throws MatchException {
27         setType(in.getType());
28     }
29
30     @Override
31     public IVal toVal(CompilationContext context, CodeWriter w) {
32         ruleset.generateCode(w);
33         return in.toVal(context, w);
34     }
35
36     @Override
37     public Expression resolve(TranslationContext context) {
38         if(context.currentRuleset != null) {
39             context.getErrorLog().log(location, "Current version of SCL compiler does not support nested rulesets.");
40             return this;
41         }
42         context.currentRuleset = ruleset;
43         
44         context.pushFrame();
45         context.pushCHRConstraintFrame();
46         ruleset.resolve(context);
47         in = in.resolve(context);
48         context.popCHRConstraintFrame(ruleset);
49         context.popFrame();
50         
51         context.currentRuleset = null;
52         
53         return this;
54     }
55     @Override
56     public void setLocationDeep(long loc) {
57         if(location == Locations.NO_LOCATION) {
58             this.location = loc;
59             ruleset.setLocationDeep(loc);
60             in.setLocationDeep(loc);
61         }
62     }
63     
64     @Override
65     public void accept(ExpressionVisitor visitor) {
66         visitor.visit(this);
67     }
68     
69     @Override
70     public Expression inferType(TypingContext context) {
71         ruleset.checkType(context);
72         in = in.inferType(context);
73         return this;
74     }
75     
76     @Override
77     public Expression checkBasicType(TypingContext context, Type requiredType) {
78         ruleset.checkType(context);
79         in = in.checkType(context, requiredType);
80         return this;
81     }
82     
83     @Override
84     public Expression simplify(SimplificationContext context) {
85         ruleset.simplifyAndCompileIfNeeded(context);
86         in = in.simplify(context);
87         return this;
88     }
89     
90     @Override
91     public Expression accept(ExpressionTransformer transformer) {
92         return transformer.transform(this);
93     }
94     
95     @Override
96     public IExpression toIExpression(ExpressionInterpretationContext context) {
97         throw new UnsupportedOperationException();
98     }
99
100 }