bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/query/QMapping.java

   1 package org.simantics.scl.compiler.elaboration.query;
   2
   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;
  14
  15 import gnu.trove.map.hash.TObjectIntHashMap;
  16 import gnu.trove.set.hash.THashSet;
  17 import gnu.trove.set.hash.TIntHashSet;
  18
  19 public class QMapping extends Query {
  20     public final MappingRelation mappingRelation;
  21     public final Expression[] parameters;
  22
  23     public QMapping(MappingRelation mappingRelation, Expression[] parameters) {
  24         this.mappingRelation = mappingRelation;
  25         this.parameters = parameters;
  26     }
  27
  28     @Override
  29     public void collectFreeVariables(THashSet<Variable> vars) {
  30         for(Expression parameter : parameters)
  31             parameter.collectFreeVariables(vars);
  32     }
  33
  34     @Override
  35     public void collectRefs(TObjectIntHashMap<Object> allRefs,
  36             TIntHashSet refs) {
  37         for(Expression parameter : parameters)
  38             parameter.collectRefs(allRefs, refs);
  39     }
  40
  41     @Override
  42     public void collectVars(TObjectIntHashMap<Variable> allVars,
  43             TIntHashSet vars) {
  44         for(Expression parameter : parameters)
  45             parameter.collectVars(allVars, vars);
  46     }
  47
  48     @Override
  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.");
  54         else
  55             for(int i=0;i<parameters.length;++i)
  56                 parameters[i] = parameters[i].checkType(context, parameterTypes[i]);
  57     }
  58
  59     @Override
  60     public void collectConstraints(ConstraintCollectionContext context)
  61             throws UnsolvableQueryException {
  62         throw new InternalCompilerError();
  63     }
  64
  65     @Override
  66     public Query replace(ReplaceContext context) {
  67         return new QMapping(mappingRelation,
  68                 Expression.replace(context, parameters));
  69     }
  70
  71     @Override
  72     public void setLocationDeep(long loc) {
  73         if(location == Locations.NO_LOCATION) {
  74             location = loc;
  75             for(Expression parameter : parameters)
  76                 parameter.setLocationDeep(loc);
  77         }
  78     }
  79
  80     @Override
  81     public void accept(QueryVisitor visitor) {
  82         visitor.visit(this);
  83     }
  84
  85     @Override
  86     public Query accept(QueryTransformer transformer) {
  87         return transformer.transform(this);
  88     }
  89 }