1 package org.simantics.scl.compiler.elaboration.query;
3 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;
4 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
5 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
6 import org.simantics.scl.compiler.elaboration.expressions.Expression;
7 import org.simantics.scl.compiler.elaboration.expressions.QueryTransformer;
8 import org.simantics.scl.compiler.elaboration.expressions.Variable;
9 import org.simantics.scl.compiler.elaboration.expressions.VariableProcedure;
10 import org.simantics.scl.compiler.elaboration.query.compilation.ConstraintCollectionContext;
11 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
12 import org.simantics.scl.compiler.elaboration.rules.MappingRelation;
13 import org.simantics.scl.compiler.errors.Locations;
14 import org.simantics.scl.compiler.types.Type;
16 import gnu.trove.map.hash.TObjectIntHashMap;
17 import gnu.trove.set.hash.THashSet;
18 import gnu.trove.set.hash.TIntHashSet;
20 public class QMapping extends Query {
21 public final MappingRelation mappingRelation;
22 public final Expression[] parameters;
24 public QMapping(MappingRelation mappingRelation, Expression[] parameters) {
25 this.mappingRelation = mappingRelation;
26 this.parameters = parameters;
30 public void collectFreeVariables(THashSet<Variable> vars) {
31 for(Expression parameter : parameters)
32 parameter.collectFreeVariables(vars);
36 public void collectRefs(TObjectIntHashMap<Object> allRefs,
38 for(Expression parameter : parameters)
39 parameter.collectRefs(allRefs, refs);
43 public void collectVars(TObjectIntHashMap<Variable> allVars,
45 for(Expression parameter : parameters)
46 parameter.collectVars(allVars, vars);
50 public void checkType(TypingContext context) {
51 // Check parameter types
52 Type[] parameterTypes = mappingRelation.parameterTypes;
53 if(parameterTypes.length != parameters.length)
54 context.getErrorLog().log(location, "Relation is applied with wrong number of parameters.");
56 for(int i=0;i<parameters.length;++i)
57 parameters[i] = parameters[i].checkType(context, parameterTypes[i]);
61 public void collectConstraints(ConstraintCollectionContext context)
62 throws UnsolvableQueryException {
63 throw new InternalCompilerError();
67 public Query replace(ReplaceContext context) {
68 return new QMapping(mappingRelation,
69 Expression.replace(context, parameters));
73 public void setLocationDeep(long loc) {
74 if(location == Locations.NO_LOCATION) {
76 for(Expression parameter : parameters)
77 parameter.setLocationDeep(loc);
82 public void accept(QueryVisitor visitor) {
87 public void forVariables(VariableProcedure procedure) {
88 for(Expression parameter : parameters)
89 parameter.forVariables(procedure);
93 public Query accept(QueryTransformer transformer) {
94 return transformer.transform(this);