bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/ECHRRuleset.java

   1 package org.simantics.scl.compiler.elaboration.expressions;\r
   2 \r
   3 import org.simantics.scl.compiler.elaboration.chr.CHRRuleset;\r
   4 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
   5 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
   6 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
   7 import org.simantics.scl.compiler.environment.Environment;\r
   8 import org.simantics.scl.compiler.errors.Locations;\r
   9 import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
  10 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\r
  11 import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
  12 import org.simantics.scl.compiler.internal.interpreted.IExpression;\r
  13 import org.simantics.scl.compiler.top.ExpressionInterpretationContext;\r
  14 import org.simantics.scl.compiler.types.Type;\r
  15 import org.simantics.scl.compiler.types.exceptions.MatchException;\r
  16 \r
  17 import gnu.trove.map.hash.TObjectIntHashMap;\r
  18 import gnu.trove.set.hash.THashSet;\r
  19 import gnu.trove.set.hash.TIntHashSet;\r
  20 \r
  21 public class ECHRRuleset extends Expression {\r
  22     CHRRuleset ruleset;\r
  23     Expression in;\r
  24     \r
  25     public ECHRRuleset(CHRRuleset ruleset, Expression in) {\r
  26         this.ruleset = ruleset;\r
  27         this.in = in;\r
  28     }\r
  29     \r
  30     @Override\r
  31     public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {\r
  32         ruleset.collectRefs(allRefs, refs);\r
  33         in.collectRefs(allRefs, refs);\r
  34     }\r
  35     @Override\r
  36     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {\r
  37         ruleset.collectVars(allVars, vars);\r
  38         in.collectVars(allVars, vars);\r
  39     }\r
  40     @Override\r
  41     public void forVariables(VariableProcedure procedure) {\r
  42         ruleset.forVariables(procedure);\r
  43         in.forVariables(procedure);\r
  44     }\r
  45     @Override\r
  46     protected void updateType() throws MatchException {\r
  47         setType(in.getType());\r
  48     }\r
  49     @Override\r
  50     public IVal toVal(Environment env, CodeWriter w) {\r
  51         ruleset.generateCode(w);\r
  52         return in.toVal(env, w);\r
  53     }\r
  54     @Override\r
  55     public void collectFreeVariables(THashSet<Variable> vars) {\r
  56         ruleset.collectFreeVariables(vars);\r
  57         in.collectFreeVariables(vars);\r
  58     }\r
  59     @Override\r
  60     public Expression resolve(TranslationContext context) {\r
  61         context.pushFrame();\r
  62         context.pushCHRConstraintFrame();\r
  63         ruleset.resolve(context);\r
  64         in = in.resolve(context);\r
  65         context.popCHRConstraintFrame(ruleset.constraints);\r
  66         context.popFrame();\r
  67         return this;\r
  68     }\r
  69     @Override\r
  70     public void setLocationDeep(long loc) {\r
  71         if(location == Locations.NO_LOCATION) {\r
  72             this.location = loc;\r
  73             ruleset.setLocationDeep(loc);\r
  74             in.setLocationDeep(loc);\r
  75         }\r
  76     }\r
  77     @Override\r
  78     public Expression decorate(ExpressionDecorator decorator) {\r
  79         in = in.decorate(decorator);\r
  80         return this;\r
  81     }\r
  82     @Override\r
  83     public void collectEffects(THashSet<Type> effects) {\r
  84         ruleset.collectEffects(effects);\r
  85         in.collectEffects(effects);\r
  86     }\r
  87     @Override\r
  88     public void accept(ExpressionVisitor visitor) {\r
  89         visitor.visit(this);\r
  90     }\r
  91     \r
  92     @Override\r
  93     public Expression inferType(TypingContext context) {\r
  94         ruleset.checkType(context);\r
  95         in = in.inferType(context);\r
  96         return this;\r
  97     }\r
  98     \r
  99     @Override\r
 100     public Expression simplify(SimplificationContext context) {\r
 101         ruleset.simplify(context);\r
 102         ruleset.compile(context);\r
 103         in = in.simplify(context);\r
 104         return this;\r
 105     }\r
 106     \r
 107     @Override\r
 108     public Expression accept(ExpressionTransformer transformer) {\r
 109         return transformer.transform(this);\r
 110     }\r
 111     \r
 112     @Override\r
 113     public IExpression toIExpression(ExpressionInterpretationContext context) {\r
 114         throw new UnsupportedOperationException();\r
 115     }\r
 116 \r
 117 }\r