1 package org.simantics.scl.compiler.elaboration.query;
3 import java.util.Collection;
6 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
7 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
8 import org.simantics.scl.compiler.elaboration.expressions.EVariable;
9 import org.simantics.scl.compiler.elaboration.expressions.QueryTransformer;
10 import org.simantics.scl.compiler.elaboration.expressions.Variable;
11 import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;
12 import org.simantics.scl.compiler.elaboration.query.compilation.ConstraintCollectionContext;
13 import org.simantics.scl.compiler.elaboration.query.compilation.DerivateException;
14 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
15 import org.simantics.scl.compiler.elaboration.relations.LocalRelation;
16 import org.simantics.scl.compiler.elaboration.relations.SCLRelation;
17 import org.simantics.scl.compiler.types.Types;
18 import org.simantics.scl.compiler.types.kinds.Kinds;
20 import gnu.trove.map.hash.THashMap;
21 import gnu.trove.set.hash.THashSet;
23 public class QExists extends QAbstractModifier {
26 public QExists(Variable[] variables, Query query) {
28 this.variables = variables;
31 public QExists(Collection<Variable> variables, Query query) {
32 this(variables.toArray(new Variable[variables.size()]), query);
36 public void collectFreeVariables(THashSet<Variable> vars) {
37 super.collectFreeVariables(vars);
38 for(Variable variable : variables)
39 vars.remove(variable);
43 public void checkType(TypingContext context) {
44 for(Variable var : variables)
45 var.setType(Types.metaVar(Kinds.STAR));
46 super.checkType(context);
50 public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
51 for(Variable variable : variables)
52 context.addVariable(variable);
53 query.collectConstraints(context);
57 public Diff[] derivate(THashMap<LocalRelation, Diffable> diffables) throws DerivateException {
58 Diff[] result = query.derivate(diffables);
59 for(int i=0;i<result.length;++i)
60 result[i].query = new QExists(variables, result[i].query);
65 public Query replace(ReplaceContext context) {
66 Variable[] newVariables = new Variable[variables.length];
67 for(int i=0;i<variables.length;++i) {
68 Variable variable = variables[i];
69 Variable newVariable = new Variable(variable.getName(), variable.getType().replace(context.tvarMap));
70 newVariables[i] = newVariable;
71 context.varMap.put(variable, new EVariable(newVariable));
73 Query result = new QExists(newVariables, query.replace(context));
74 for(Variable v : variables)
75 context.varMap.remove(v);
80 public Query removeRelations(Set<SCLRelation> relations) {
81 Query newQuery = query.removeRelations(relations);
84 else if(newQuery == EMPTY_QUERY)
87 return new QExists(variables, newQuery);
91 public void accept(QueryVisitor visitor) {
96 public void forVariables(VariableProcedure procedure) {
97 for(Variable variable : variables)
98 procedure.execute(location, variable);
99 super.forVariables(procedure);
103 public Query accept(QueryTransformer transformer) {
104 return transformer.transform(this);