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

   1 package org.simantics.scl.compiler.elaboration.expressions;
   2
   3 import java.util.ArrayList;
   4
   5 import org.simantics.scl.compiler.common.names.Names;
   6 import org.simantics.scl.compiler.compilation.CompilationContext;
   7 import org.simantics.scl.compiler.constants.NoRepConstant;
   8 import org.simantics.scl.compiler.elaboration.chr.CHRQuery;
   9 import org.simantics.scl.compiler.elaboration.chr.CHRRuleset;
  10 import org.simantics.scl.compiler.elaboration.chr.plan.PlanContext;
  11 import org.simantics.scl.compiler.elaboration.chr.plan.PlanOp;
  12 import org.simantics.scl.compiler.elaboration.chr.plan.PlanRealizer;
  13 import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;
  14 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
  15 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
  16 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
  17 import org.simantics.scl.compiler.errors.Locations;
  18 import org.simantics.scl.compiler.internal.codegen.references.IVal;
  19 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
  20 import org.simantics.scl.compiler.types.Types;
  21 import org.simantics.scl.compiler.types.exceptions.MatchException;
  22 import org.simantics.scl.compiler.types.kinds.Kinds;
  23
  24 import gnu.trove.map.hash.TObjectIntHashMap;
  25 import gnu.trove.set.hash.TIntHashSet;
  26
  27 public class ECHRSelect extends Expression {
  28     public CHRQuery query;
  29     public Variable[] existentialVariables;
  30     public Expression expression;
  31     private ArrayList<PlanOp> planOps;
  32     private CHRRuleset currentRuleset;
  33
  34     public ECHRSelect(Expression expression, CHRQuery query) {
  35         this.expression = expression;
  36         this.query = query;
  37     }
  38
  39     @Override
  40     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  41         query.collectVars(allVars, vars);
  42         expression.collectVars(allVars, vars);
  43     }
  44
  45     @Override
  46     protected void updateType() throws MatchException {
  47         setType(Types.list(expression.getType()));
  48     }
  49
  50     @Override
  51     public Expression inferType(TypingContext context) {
  52         for(Variable variable : existentialVariables)
  53             variable.setType(Types.metaVar(Kinds.STAR));
  54         query.checkType(context);
  55         expression = expression.inferType(context);
  56         return this;
  57     }
  58
  59     @Override
  60     public Expression simplify(SimplificationContext simplificationContext) {
  61         this.expression = expression.simplify(simplificationContext);
  62         query.simplify(simplificationContext);
  63
  64         CompilationContext compilationContext = simplificationContext.getCompilationContext();
  65         QueryPlanningContext context = new QueryPlanningContext(compilationContext, existentialVariables);
  66         if(query.createQueryPlan(context, null, -1, null))
  67             planOps = context.getPlanOps();
  68
  69         return this;
  70     }
  71
  72     @Override
  73     public IVal toVal(CompilationContext context, CodeWriter w) {
  74         IVal list = w.apply(location, context.getValue(Names.MList_create).getValue(), NoRepConstant.UNIT);
  75         planOps.add(new PlanOp(location) {
  76             @Override
  77             public void generateCode(CompilationContext context, PlanContext planContext, CodeWriter w) {
  78                 w.apply(location, context.getValue(Names.MList_add).getValue(), list, expression.toVal(context, w));
  79             }
  80         });
  81         PlanRealizer realizer = new PlanRealizer(context, currentRuleset, currentRuleset != null ? currentRuleset.runtimeRulesetVariable : null, null, planOps);
  82         realizer.nextOp(w);
  83         return w.apply(location, context.getValue(Names.MList_freeze).getValue(), list);
  84     }
  85
  86     @Override
  87     public Expression resolve(TranslationContext context) {
  88         currentRuleset = context.currentRuleset;
  89
  90         context.pushExistentialFrame();
  91         query.resolve(context);
  92         context.disallowNewExistentials();
  93         expression = expression.resolve(context);
  94         existentialVariables = context.popExistentialFrame();
  95         return this;
  96     }
  97
  98     @Override
  99     public void setLocationDeep(long loc) {
 100         if(location == Locations.NO_LOCATION) {
 101             query.setLocationDeep(loc);
 102             expression.setLocationDeep(loc);
 103         }
 104     }
 105
 106     @Override
 107     public void accept(ExpressionVisitor visitor) {
 108         visitor.visit(this);
 109     }
 110
 111     @Override
 112     public Expression accept(ExpressionTransformer transformer) {
 113         return transformer.transform(this);
 114     }
 115 }