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 }