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

   1 package org.simantics.scl.compiler.elaboration.query;
   2
   3 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
   4 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   5 import org.simantics.scl.compiler.elaboration.expressions.Expression;
   6 import org.simantics.scl.compiler.elaboration.expressions.QueryTransformer;
   7 import org.simantics.scl.compiler.elaboration.expressions.Variable;
   8 import org.simantics.scl.compiler.elaboration.query.compilation.ConstraintCollectionContext;
   9 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
  10 import org.simantics.scl.compiler.errors.Locations;
  11 import org.simantics.scl.compiler.types.Types;
  12
  13 import gnu.trove.map.hash.TObjectIntHashMap;
  14 import gnu.trove.set.hash.THashSet;
  15 import gnu.trove.set.hash.TIntHashSet;
  16
  17 public class QIf extends Query {
  18     public Expression condition;
  19     public Query thenQuery;
  20     public Query elseQuery;
  21
  22     public QIf(Expression condition, Query thenQuery, Query elseQuery) {
  23         this.condition = condition;
  24         this.thenQuery = thenQuery;
  25         this.elseQuery = elseQuery;
  26     }
  27
  28     @Override
  29     public void collectFreeVariables(THashSet<Variable> vars) {
  30         condition.collectFreeVariables(vars);
  31         thenQuery.collectFreeVariables(vars);
  32         elseQuery.collectFreeVariables(vars);
  33     }
  34
  35     @Override
  36     public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
  37         condition.collectRefs(allRefs, refs);
  38         thenQuery.collectRefs(allRefs, refs);
  39         elseQuery.collectRefs(allRefs, refs);
  40     }
  41
  42     @Override
  43     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  44         condition.collectVars(allVars, vars);
  45         thenQuery.collectVars(allVars, vars);
  46         elseQuery.collectVars(allVars, vars);
  47     }
  48
  49     @Override
  50     public void checkType(TypingContext context) {
  51         condition.checkType(context, Types.BOOLEAN);
  52         thenQuery.checkType(context);
  53         elseQuery.checkType(context);
  54     }
  55
  56     @Override
  57     public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
  58         // TODO Auto-generated method stub
  59
  60     }
  61
  62     @Override
  63     public Query replace(ReplaceContext context) {
  64         return new QIf(
  65                 condition.replace(context),
  66                 thenQuery.replace(context),
  67                 elseQuery.replace(context));
  68     }
  69
  70     @Override
  71     public void setLocationDeep(long loc) {
  72         if(location == Locations.NO_LOCATION) {
  73             this.location = loc;
  74             condition.setLocationDeep(loc);
  75             elseQuery.setLocationDeep(loc);
  76             thenQuery.setLocationDeep(loc);
  77         }
  78     }
  79
  80     @Override
  81     public void accept(QueryVisitor visitor) {
  82         visitor.visit(this);
  83     }
  84
  85     @Override
  86     public Query accept(QueryTransformer transformer) {
  87         return transformer.transform(this);
  88     }
  89
  90 }