1 package org.simantics.scl.compiler.elaboration.expressions;
3 import java.util.ArrayList;
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;
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;
33 public ECHRSelect(Expression expression, CHRQuery query) {
34 this.expression = expression;
39 protected void updateType() throws MatchException {
40 setType(Types.list(expression.getType()));
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);
53 public Expression simplify(SimplificationContext simplificationContext) {
54 this.expression = expression.simplify(simplificationContext);
55 query.simplify(simplificationContext);
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();
66 IVal list = w.apply(location, context.getValue(Names.MList_create).getValue(), NoRepConstant.UNIT);
67 planOps.add(new PlanOp(location) {
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));
73 PlanRealizer realizer = new PlanRealizer(context, currentRuleset, currentRuleset != null ? currentRuleset.runtimeRulesetVariable : null, null, planOps);
75 return w.apply(location, context.getValue(Names.MList_freeze).getValue(), list);
79 public Expression resolve(TranslationContext context) {
80 currentRuleset = context.currentRuleset;
82 context.pushExistentialFrame();
83 query.resolve(context);
84 context.disallowNewExistentials();
85 expression = expression.resolve(context);
86 existentialVariables = context.popExistentialFrame();
91 public void setLocationDeep(long loc) {
92 if(location == Locations.NO_LOCATION) {
93 query.setLocationDeep(loc);
94 expression.setLocationDeep(loc);
99 public void accept(ExpressionVisitor visitor) {
104 public Expression accept(ExpressionTransformer transformer) {
105 return transformer.transform(this);
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;
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));