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