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.query.compilation.ConstraintCollectionContext;
10 import org.simantics.scl.compiler.elaboration.query.compilation.UnsolvableQueryException;
11 import org.simantics.scl.compiler.elaboration.rules.MappingRelation;
12 import org.simantics.scl.compiler.errors.Locations;
13 import org.simantics.scl.compiler.types.Type;
15 import gnu.trove.map.hash.TObjectIntHashMap;
16 import gnu.trove.set.hash.THashSet;
17 import gnu.trove.set.hash.TIntHashSet;
19 public class QMapping extends Query {
20 public final MappingRelation mappingRelation;
21 public final Expression[] parameters;
23 public QMapping(MappingRelation mappingRelation, Expression[] parameters) {
24 this.mappingRelation = mappingRelation;
25 this.parameters = parameters;
29 public void collectFreeVariables(THashSet<Variable> vars) {
30 for(Expression parameter : parameters)
31 parameter.collectFreeVariables(vars);
35 public void collectRefs(TObjectIntHashMap<Object> allRefs,
37 for(Expression parameter : parameters)
38 parameter.collectRefs(allRefs, refs);
42 public void collectVars(TObjectIntHashMap<Variable> allVars,
44 for(Expression parameter : parameters)
45 parameter.collectVars(allVars, vars);
49 public void checkType(TypingContext context) {
50 // Check parameter types
51 Type[] parameterTypes = mappingRelation.parameterTypes;
52 if(parameterTypes.length != parameters.length)
53 context.getErrorLog().log(location, "Relation is applied with wrong number of parameters.");
55 for(int i=0;i<parameters.length;++i)
56 parameters[i] = parameters[i].checkType(context, parameterTypes[i]);
60 public void collectConstraints(ConstraintCollectionContext context)
61 throws UnsolvableQueryException {
62 throw new InternalCompilerError();
66 public Query replace(ReplaceContext context) {
67 return new QMapping(mappingRelation,
68 Expression.replace(context, parameters));
72 public void setLocationDeep(long loc) {
73 if(location == Locations.NO_LOCATION) {
75 for(Expression parameter : parameters)
76 parameter.setLocationDeep(loc);
81 public void accept(QueryVisitor visitor) {
86 public Query accept(QueryTransformer transformer) {
87 return transformer.transform(this);