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.exceptions.InternalCompilerError;
   6 import org.simantics.scl.compiler.common.names.Names;
   7 import org.simantics.scl.compiler.compilation.CompilationContext;
   8 import org.simantics.scl.compiler.constants.NoRepConstant;
   9 import org.simantics.scl.compiler.elaboration.chr.CHRQuery;
  10 import org.simantics.scl.compiler.elaboration.chr.CHRRuleset;
  11 import org.simantics.scl.compiler.elaboration.chr.plan.PlanContext;
  12 import org.simantics.scl.compiler.elaboration.chr.plan.PlanOp;
  13 import org.simantics.scl.compiler.elaboration.chr.plan.PlanRealizer;
  14 import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;
  15 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
  16 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
  17 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
  18 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
  19 import org.simantics.scl.compiler.errors.Locations;
  20 import org.simantics.scl.compiler.internal.codegen.references.IVal;
  21 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
  22 import org.simantics.scl.compiler.types.Types;
  23 import org.simantics.scl.compiler.types.exceptions.MatchException;
  24 import org.simantics.scl.compiler.types.kinds.Kinds;
  25
  26 public class ECHRSelect extends Expression {
  27     public CHRQuery query;
  28     public Variable[] existentialVariables;
  29     public Expression expression;
  30     private ArrayList<PlanOp> planOps;
  31     private CHRRuleset currentRuleset;
  32
  33     public ECHRSelect(Expression expression, CHRQuery query) {
  34         this.expression = expression;
  35         this.query = query;
  36     }
  37
  38     @Override
  39     protected void updateType() throws MatchException {
  40         setType(Types.list(expression.getType()));
  41     }
  42
  43     @Override
  44     public Expression inferType(TypingContext context) {
  45         for(Variable variable : existentialVariables)
  46             variable.setType(Types.metaVar(Kinds.STAR));
  47         query.checkType(context);
  48         expression = expression.inferType(context);
  49         return this;
  50     }
  51
  52     @Override
  53     public Expression simplify(SimplificationContext simplificationContext) {
  54         this.expression = expression.simplify(simplificationContext);
  55         query.simplify(simplificationContext);
  56
  57         return this;
  58     }
  59
  60     @Override
  61     public IVal toVal(CompilationContext context, CodeWriter w) {
  62         QueryPlanningContext queryContext = new QueryPlanningContext(context, existentialVariables);
  63         if(query.createQueryPlan(queryContext, null, -1, null))
  64             planOps = queryContext.getPlanOps();
  65
  66         IVal list = w.apply(location, context.getValue(Names.MList_create).getValue(), NoRepConstant.UNIT);
  67         planOps.add(new PlanOp(location) {
  68             @Override
  69             public void generateCode(CompilationContext context, PlanContext planContext, CodeWriter w) {
  70                 w.apply(location, context.getValue(Names.MList_add).getValue(), list, expression.toVal(context, w));
  71             }
  72         });
  73         PlanRealizer realizer = new PlanRealizer(context, currentRuleset, currentRuleset != null ? currentRuleset.runtimeRulesetVariable : null, null, planOps);
  74         realizer.nextOp(w);
  75         return w.apply(location, context.getValue(Names.MList_freeze).getValue(), list);
  76     }
  77
  78     @Override
  79     public Expression resolve(TranslationContext context) {
  80         currentRuleset = context.currentRuleset;
  81
  82         context.pushExistentialFrame();
  83         query.resolve(context);
  84         context.disallowNewExistentials();
  85         expression = expression.resolve(context);
  86         existentialVariables = context.popExistentialFrame();
  87         return this;
  88     }
  89
  90     @Override
  91     public void setLocationDeep(long loc) {
  92         if(location == Locations.NO_LOCATION) {
  93             query.setLocationDeep(loc);
  94             expression.setLocationDeep(loc);
  95         }
  96     }
  97
  98     @Override
  99     public void accept(ExpressionVisitor visitor) {
 100         visitor.visit(this);
 101     }
 102
 103     @Override
 104     public Expression accept(ExpressionTransformer transformer) {
 105         return transformer.transform(this);
 106     }
 107
 108     @Override
 109     public Expression replace(ReplaceContext context) {
 110         Variable[] newExistentialVariables = new Variable[existentialVariables.length];
 111         for(int i=0;i<existentialVariables.length;++i) {
 112             Variable newVariable = existentialVariables[i].copy();
 113             context.varMap.put(existentialVariables[i], new EVariable(newVariable));
 114             newExistentialVariables[i] = newVariable;
 115         }
 116         ECHRSelect copy = new ECHRSelect(expression.replace(context), query.replace(context));
 117         copy.existentialVariables = newExistentialVariables;
 118         copy.currentRuleset = currentRuleset;
 119         copy.planOps = planOps;
 120         if(planOps != null) {
 121             copy.planOps = new ArrayList<PlanOp>(planOps.size());
 122             throw new InternalCompilerError(location, "Copying of ECHRSelect is not supported.");
 123             //for(PlanOp op : planOps)
 124             //    copy.planOps.add(op.replace(context));
 125         }
 126         return copy;
 127     }
 128 }