import java.util.ArrayList;
+import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
import org.simantics.scl.compiler.common.names.Names;
import org.simantics.scl.compiler.compilation.CompilationContext;
import org.simantics.scl.compiler.constants.NoRepConstant;
import org.simantics.scl.compiler.elaboration.chr.plan.PlanOp;
import org.simantics.scl.compiler.elaboration.chr.plan.PlanRealizer;
import org.simantics.scl.compiler.elaboration.chr.planning.QueryPlanningContext;
+import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
import org.simantics.scl.compiler.errors.Locations;
import org.simantics.scl.compiler.internal.codegen.references.IVal;
import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
-import org.simantics.scl.compiler.types.Type;
import org.simantics.scl.compiler.types.Types;
import org.simantics.scl.compiler.types.exceptions.MatchException;
import org.simantics.scl.compiler.types.kinds.Kinds;
-import gnu.trove.map.hash.TObjectIntHashMap;
-import gnu.trove.set.hash.THashSet;
-import gnu.trove.set.hash.TIntHashSet;
-
public class ECHRSelect extends Expression {
- CHRQuery query;
- Variable[] existentialVariables;
- Expression expression;
+ public CHRQuery query;
+ public Variable[] existentialVariables;
+ public Expression expression;
private ArrayList<PlanOp> planOps;
private CHRRuleset currentRuleset;
this.query = query;
}
- @Override
- public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
- query.collectRefs(allRefs, refs);
- expression.collectRefs(allRefs, refs);
- }
-
- @Override
- public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
- query.collectVars(allVars, vars);
- expression.collectVars(allVars, vars);
- }
-
- @Override
- public void forVariables(VariableProcedure procedure) {
- query.forVariables(procedure);
- expression.forVariables(procedure);
- }
-
@Override
protected void updateType() throws MatchException {
setType(Types.list(expression.getType()));
public Expression simplify(SimplificationContext simplificationContext) {
this.expression = expression.simplify(simplificationContext);
query.simplify(simplificationContext);
-
- CompilationContext compilationContext = simplificationContext.getCompilationContext();
- QueryPlanningContext context = new QueryPlanningContext(compilationContext, existentialVariables);
- if(query.createQueryPlan(context, null, -1, null))
- planOps = context.getPlanOps();
return this;
}
@Override
public IVal toVal(CompilationContext context, CodeWriter w) {
+ QueryPlanningContext queryContext = new QueryPlanningContext(context, existentialVariables);
+ if(query.createQueryPlan(queryContext, null, -1, null))
+ planOps = queryContext.getPlanOps();
+
IVal list = w.apply(location, context.getValue(Names.MList_create).getValue(), NoRepConstant.UNIT);
planOps.add(new PlanOp(location) {
@Override
return w.apply(location, context.getValue(Names.MList_freeze).getValue(), list);
}
- @Override
- public void collectFreeVariables(THashSet<Variable> vars) {
- query.collectFreeVariables(vars);
- expression.collectFreeVariables(vars);
- if(existentialVariables != null)
- for(Variable variable : existentialVariables)
- vars.remove(variable);
- }
-
@Override
public Expression resolve(TranslationContext context) {
currentRuleset = context.currentRuleset;
}
}
- @Override
- public void collectEffects(THashSet<Type> effects) {
- expression.collectEffects(effects);
- query.collectQueryEffects(effects);
- effects.add(Types.PROC);
- }
-
@Override
public void accept(ExpressionVisitor visitor) {
visitor.visit(this);
public Expression accept(ExpressionTransformer transformer) {
return transformer.transform(this);
}
+
+ @Override
+ public Expression replace(ReplaceContext context) {
+ Variable[] newExistentialVariables = new Variable[existentialVariables.length];
+ for(int i=0;i<existentialVariables.length;++i) {
+ Variable newVariable = existentialVariables[i].copy();
+ context.varMap.put(existentialVariables[i], new EVariable(newVariable));
+ newExistentialVariables[i] = newVariable;
+ }
+ ECHRSelect copy = new ECHRSelect(expression.replace(context), query.replace(context));
+ copy.existentialVariables = newExistentialVariables;
+ copy.currentRuleset = currentRuleset;
+ copy.planOps = planOps;
+ if(planOps != null) {
+ copy.planOps = new ArrayList<PlanOp>(planOps.size());
+ throw new InternalCompilerError(location, "Copying of ECHRSelect is not supported.");
+ //for(PlanOp op : planOps)
+ // copy.planOps.add(op.replace(context));
+ }
+ return copy;
+ }
}