bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/EEnforce.java

   1 package org.simantics.scl.compiler.elaboration.expressions;
   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.SimplificationContext;
   6 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   7 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   8 import org.simantics.scl.compiler.elaboration.query.QExists;
   9 import org.simantics.scl.compiler.elaboration.query.Query;
  10 import org.simantics.scl.compiler.elaboration.query.compilation.EnforcingContext;
  11 import org.simantics.scl.compiler.errors.Locations;
  12 import org.simantics.scl.compiler.types.Type;
  13 import org.simantics.scl.compiler.types.Types;
  14 import org.simantics.scl.compiler.types.exceptions.MatchException;
  15
  16 import gnu.trove.map.hash.TObjectIntHashMap;
  17 import gnu.trove.set.hash.THashSet;
  18 import gnu.trove.set.hash.TIntHashSet;
  19
  20 public class EEnforce extends SimplifiableExpression {
  21
  22     Query query;
  23
  24     public EEnforce(Query query) {
  25         this.query = query;
  26     }
  27
  28     public Query getQuery() {
  29         return query;
  30     }
  31
  32     @Override
  33     public void collectVars(TObjectIntHashMap<Variable> allVars,
  34             TIntHashSet vars) {
  35         query.collectVars(allVars, vars);
  36     }
  37
  38     @Override
  39     public void collectEffects(THashSet<Type> effects) {
  40         throw new InternalCompilerError(location, getClass().getSimpleName() + " does not support collectEffects.");
  41     }
  42
  43     @Override
  44     protected void updateType() throws MatchException {
  45         setType(Types.tupleConstructor(0));
  46     }
  47
  48     @Override
  49     public Expression inferType(TypingContext context) {
  50         query.checkType(context);
  51         //context.declareEffect(location, query.getEffect(Query.W));
  52         return compile(context);
  53     }
  54
  55     public Expression compile(TypingContext context) {
  56         return query.generateEnforce(new EnforcingContext(context));
  57     }
  58
  59     @Override
  60     public void collectFreeVariables(THashSet<Variable> vars) {
  61         query.collectFreeVariables(vars);
  62     }
  63
  64     @Override
  65     public Expression simplify(SimplificationContext context) {
  66         /*query = query.simplify(context);
  67         return query.generateEnforce(context);*/
  68         throw new InternalCompilerError();
  69     }
  70
  71     @Override
  72     public Expression resolve(TranslationContext context) {
  73         context.pushExistentialFrame();
  74         query = query.resolve(context);
  75         Variable[] variables = context.popExistentialFrame();
  76         if(variables.length > 0)
  77             query = new QExists(variables, query);
  78         return this;
  79     }
  80
  81     @Override
  82     public void setLocationDeep(long loc) {
  83         if(location == Locations.NO_LOCATION) {
  84             location = loc;
  85             query.setLocationDeep(loc);
  86         }
  87     }
  88
  89     @Override
  90     public void accept(ExpressionVisitor visitor) {
  91         visitor.visit(this);
  92     }
  93
  94     @Override
  95     public Expression replace(ReplaceContext context) {
  96         return new EEnforce(query.replace(context));
  97     }
  98
  99     @Override
 100     public Expression accept(ExpressionTransformer transformer) {
 101         return transformer.transform(this);
 102     }
 103
 104 }