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