1 package org.simantics.scl.compiler.elaboration.query;
\r
3 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
\r
4 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
\r
5 import org.simantics.scl.compiler.elaboration.expressions.Expression;
\r
6 import org.simantics.scl.compiler.elaboration.expressions.QueryTransformer;
\r
7 import org.simantics.scl.compiler.elaboration.expressions.Variable;
\r
8 import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;
\r
9 import org.simantics.scl.compiler.elaboration.query.compilation.ConstraintCollectionContext;
\r
10 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
\r
11 import org.simantics.scl.compiler.errors.Locations;
\r
12 import org.simantics.scl.compiler.types.Types;
\r
14 import gnu.trove.map.hash.TObjectIntHashMap;
\r
15 import gnu.trove.set.hash.THashSet;
\r
16 import gnu.trove.set.hash.TIntHashSet;
\r
18 public class QIf extends Query {
\r
19 public Expression condition;
\r
20 public Query thenQuery;
\r
21 public Query elseQuery;
\r
23 public QIf(Expression condition, Query thenQuery, Query elseQuery) {
\r
24 this.condition = condition;
\r
25 this.thenQuery = thenQuery;
\r
26 this.elseQuery = elseQuery;
\r
30 public void collectFreeVariables(THashSet<Variable> vars) {
\r
31 condition.collectFreeVariables(vars);
\r
32 thenQuery.collectFreeVariables(vars);
\r
33 elseQuery.collectFreeVariables(vars);
\r
37 public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
\r
38 condition.collectRefs(allRefs, refs);
\r
39 thenQuery.collectRefs(allRefs, refs);
\r
40 elseQuery.collectRefs(allRefs, refs);
\r
44 public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
\r
45 condition.collectVars(allVars, vars);
\r
46 thenQuery.collectVars(allVars, vars);
\r
47 elseQuery.collectVars(allVars, vars);
\r
51 public void checkType(TypingContext context) {
\r
52 condition.checkType(context, Types.BOOLEAN);
\r
53 thenQuery.checkType(context);
\r
54 elseQuery.checkType(context);
\r
58 public void collectConstraints(ConstraintCollectionContext context) throws UnsolvableQueryException {
\r
59 // TODO Auto-generated method stub
\r
64 public Query replace(ReplaceContext context) {
\r
66 condition.replace(context),
\r
67 thenQuery.replace(context),
\r
68 elseQuery.replace(context));
\r
72 public void setLocationDeep(long loc) {
\r
73 if(location == Locations.NO_LOCATION) {
\r
74 this.location = loc;
\r
75 condition.setLocationDeep(loc);
\r
76 elseQuery.setLocationDeep(loc);
\r
77 thenQuery.setLocationDeep(loc);
\r
82 public void accept(QueryVisitor visitor) {
\r
83 visitor.visit(this);
\r
87 public void forVariables(VariableProcedure procedure) {
\r
88 condition.forVariables(procedure);
\r
89 elseQuery.forVariables(procedure);
\r
90 thenQuery.forVariables(procedure);
\r
94 public Query accept(QueryTransformer transformer) {
\r
95 return transformer.transform(this);
\r