bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/query/QExists.java

   1 package org.simantics.scl.compiler.elaboration.query;
   2
   3 import java.util.Collection;
   4 import java.util.Set;
   5
   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;
  19
  20 import gnu.trove.map.hash.THashMap;
  21 import gnu.trove.set.hash.THashSet;
  22
  23 public class QExists extends QAbstractModifier {
  24     Variable[] variables;
  25
  26     public QExists(Variable[] variables, Query query) {
  27         super(query);
  28         this.variables = variables;
  29     }
  30
  31     public QExists(Collection<Variable> variables, Query query) {
  32         this(variables.toArray(new Variable[variables.size()]), query);
  33     }
  34
  35     @Override
  36     public void collectFreeVariables(THashSet<Variable> vars) {
  37         super.collectFreeVariables(vars);
  38         for(Variable variable : variables)
  39             vars.remove(variable);
  40     }
  41
  42     @Override
  43     public void checkType(TypingContext context) {
  44         for(Variable var : variables)
  45             var.setType(Types.metaVar(Kinds.STAR));
  46         super.checkType(context);
  47     }
  48
  49     @Override
  50     public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
  51         for(Variable variable : variables)
  52             context.addVariable(variable);
  53         query.collectConstraints(context);
  54     }
  55
  56     @Override
  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);
  61         return result;
  62     }
  63
  64     @Override
  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));
  72         }
  73         Query result = new QExists(newVariables, query.replace(context));
  74         for(Variable v : variables)
  75             context.varMap.remove(v);
  76         return result;
  77     }
  78
  79     @Override
  80     public Query removeRelations(Set<SCLRelation> relations) {
  81         Query newQuery = query.removeRelations(relations);
  82         if(newQuery == query)
  83             return this;
  84         else if(newQuery == EMPTY_QUERY)
  85             return EMPTY_QUERY;
  86         else
  87             return new QExists(variables, newQuery);
  88     }
  89
  90     @Override
  91     public void accept(QueryVisitor visitor) {
  92         visitor.visit(this);
  93     }
  94
  95     @Override
  96     public void forVariables(VariableProcedure procedure) {
  97         for(Variable variable : variables)
  98             procedure.execute(location, variable);
  99         super.forVariables(procedure);
 100     }
 101
 102     @Override
 103     public Query accept(QueryTransformer transformer) {
 104         return transformer.transform(this);
 105     }
 106
 107 }