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