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.query.compilation.ConstraintCollectionContext;
  12 import org.simantics.scl.compiler.elaboration.query.compilation.DerivateException;
  13 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
  14 import org.simantics.scl.compiler.elaboration.relations.LocalRelation;
  15 import org.simantics.scl.compiler.elaboration.relations.SCLRelation;
  16 import org.simantics.scl.compiler.types.Types;
  17 import org.simantics.scl.compiler.types.kinds.Kinds;
  18
  19 import gnu.trove.map.hash.THashMap;
  20
  21 public class QExists extends QAbstractModifier {
  22     public Variable[] variables;
  23
  24     public QExists(Variable[] variables, Query query) {
  25         super(query);
  26         this.variables = variables;
  27     }
  28
  29     public QExists(Collection<Variable> variables, Query query) {
  30         this(variables.toArray(new Variable[variables.size()]), query);
  31     }
  32
  33     @Override
  34     public void checkType(TypingContext context) {
  35         for(Variable var : variables)
  36             var.setType(Types.metaVar(Kinds.STAR));
  37         super.checkType(context);
  38     }
  39
  40     @Override
  41     public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
  42         for(Variable variable : variables)
  43             context.addVariable(variable);
  44         query.collectConstraints(context);
  45     }
  46
  47     @Override
  48     public Diff[] derivate(THashMap<LocalRelation, Diffable> diffables) throws DerivateException {
  49         Diff[] result = query.derivate(diffables);
  50         for(int i=0;i<result.length;++i)
  51             result[i].query = new QExists(variables, result[i].query);
  52         return result;
  53     }
  54
  55     @Override
  56     public Query replace(ReplaceContext context) {
  57         Variable[] newVariables = new Variable[variables.length];
  58         for(int i=0;i<variables.length;++i) {
  59             Variable variable = variables[i];
  60             Variable newVariable = new Variable(variable.getName(), variable.getType().replace(context.tvarMap));
  61             newVariables[i] = newVariable;
  62             context.varMap.put(variable, new EVariable(newVariable));
  63         }
  64         Query result = new QExists(newVariables, query.replace(context));
  65         for(Variable v : variables)
  66             context.varMap.remove(v);
  67         return result;
  68     }
  69
  70     @Override
  71     public Query removeRelations(Set<SCLRelation> relations) {
  72         Query newQuery = query.removeRelations(relations);
  73         if(newQuery == query)
  74             return this;
  75         else if(newQuery == EMPTY_QUERY)
  76             return EMPTY_QUERY;
  77         else
  78             return new QExists(variables, newQuery);
  79     }
  80
  81     @Override
  82     public void accept(QueryVisitor visitor) {
  83         visitor.visit(this);
  84     }
  85
  86     @Override
  87     public Query accept(QueryTransformer transformer) {
  88         return transformer.transform(this);
  89     }
  90
  91 }