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

   1 package org.simantics.scl.compiler.elaboration.expressions;
   2
   3 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
   4 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   5 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   6 import org.simantics.scl.compiler.errors.Locations;
   7 import org.simantics.scl.compiler.types.TPred;
   8 import org.simantics.scl.compiler.types.exceptions.MatchException;
   9
  10 import gnu.trove.map.hash.TObjectIntHashMap;
  11 import gnu.trove.set.hash.TIntHashSet;
  12
  13 public class EGetConstraint extends SimplifiableExpression {
  14     TPred constraint;
  15     public EVariable evidence;
  16
  17     public EGetConstraint(long loc, TPred constraint) {
  18         super(loc);
  19         this.constraint = constraint;
  20     }
  21
  22     @Override
  23     public void collectVars(TObjectIntHashMap<Variable> allVars,
  24             TIntHashSet vars) {
  25         if(evidence != null)
  26             evidence.collectVars(allVars, vars);
  27     }
  28
  29     @Override
  30     public Expression inferType(TypingContext context) {
  31         Variable variable = new Variable("evidence");
  32         variable.setType(constraint);
  33         evidence = new EVariable(getLocation(), variable);
  34         evidence.setType(variable.getType());
  35         context.addConstraintDemand(evidence);
  36         return this;
  37     }
  38
  39     @Override
  40     protected void updateType() throws MatchException {
  41         setType(constraint);
  42     }
  43
  44     @Override
  45     public Expression simplify(SimplificationContext context) {
  46         return evidence.simplify(context);
  47     }
  48
  49     @Override
  50     public Expression resolve(TranslationContext context) {
  51         return this;
  52     }
  53
  54     @Override
  55     public void setLocationDeep(long loc) {
  56         if(location == Locations.NO_LOCATION) {
  57             location = loc;
  58             if(evidence != null)
  59                 evidence.setLocationDeep(loc);
  60         }
  61     }
  62
  63     @Override
  64     public void accept(ExpressionVisitor visitor) {
  65         visitor.visit(this);
  66     }
  67
  68     @Override
  69     public Expression accept(ExpressionTransformer transformer) {
  70         return transformer.transform(this);
  71     }
  72
  73 }