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.TIntHashSet;
  15
  16 public class QIf extends Query {
  17     public Expression condition;
  18     public Query thenQuery;
  19     public Query elseQuery;
  20
  21     public QIf(Expression condition, Query thenQuery, Query elseQuery) {
  22         this.condition = condition;
  23         this.thenQuery = thenQuery;
  24         this.elseQuery = elseQuery;
  25     }
  26
  27     @Override
  28     public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  29         condition.collectVars(allVars, vars);
  30         thenQuery.collectVars(allVars, vars);
  31         elseQuery.collectVars(allVars, vars);
  32     }
  33
  34     @Override
  35     public void checkType(TypingContext context) {
  36         condition.checkType(context, Types.BOOLEAN);
  37         thenQuery.checkType(context);
  38         elseQuery.checkType(context);
  39     }
  40
  41     @Override
  42     public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
  43         // TODO Auto-generated method stub
  44
  45     }
  46
  47     @Override
  48     public Query replace(ReplaceContext context) {
  49         return new QIf(
  50                 condition.replace(context),
  51                 thenQuery.replace(context),
  52                 elseQuery.replace(context));
  53     }
  54
  55     @Override
  56     public void setLocationDeep(long loc) {
  57         if(location == Locations.NO_LOCATION) {
  58             this.location = loc;
  59             condition.setLocationDeep(loc);
  60             elseQuery.setLocationDeep(loc);
  61             thenQuery.setLocationDeep(loc);
  62         }
  63     }
  64
  65     @Override
  66     public void accept(QueryVisitor visitor) {
  67         visitor.visit(this);
  68     }
  69
  70     @Override
  71     public Query accept(QueryTransformer transformer) {
  72         return transformer.transform(this);
  73     }
  74
  75 }