1 package org.simantics.scl.compiler.elaboration.query;
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;
13 import gnu.trove.map.hash.TObjectIntHashMap;
14 import gnu.trove.set.hash.THashSet;
15 import gnu.trove.set.hash.TIntHashSet;
17 public class QIf extends Query {
18 public Expression condition;
19 public Query thenQuery;
20 public Query elseQuery;
22 public QIf(Expression condition, Query thenQuery, Query elseQuery) {
23 this.condition = condition;
24 this.thenQuery = thenQuery;
25 this.elseQuery = elseQuery;
29 public void collectFreeVariables(THashSet<Variable> vars) {
30 condition.collectFreeVariables(vars);
31 thenQuery.collectFreeVariables(vars);
32 elseQuery.collectFreeVariables(vars);
36 public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
37 condition.collectVars(allVars, vars);
38 thenQuery.collectVars(allVars, vars);
39 elseQuery.collectVars(allVars, vars);
43 public void checkType(TypingContext context) {
44 condition.checkType(context, Types.BOOLEAN);
45 thenQuery.checkType(context);
46 elseQuery.checkType(context);
50 public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
51 // TODO Auto-generated method stub
56 public Query replace(ReplaceContext context) {
58 condition.replace(context),
59 thenQuery.replace(context),
60 elseQuery.replace(context));
64 public void setLocationDeep(long loc) {
65 if(location == Locations.NO_LOCATION) {
67 condition.setLocationDeep(loc);
68 elseQuery.setLocationDeep(loc);
69 thenQuery.setLocationDeep(loc);
74 public void accept(QueryVisitor visitor) {
79 public Query accept(QueryTransformer transformer) {
80 return transformer.transform(this);