bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/elaboration/expressions/ELambdaType.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.environment.Environment;
   9 import org.simantics.scl.compiler.errors.Locations;
  10 import org.simantics.scl.compiler.internal.codegen.references.IVal;
  11 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
  12 import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;
  13 import org.simantics.scl.compiler.internal.interpreted.IExpression;
  14 import org.simantics.scl.compiler.top.ExpressionInterpretationContext;
  15 import org.simantics.scl.compiler.types.TVar;
  16 import org.simantics.scl.compiler.types.Type;
  17 import org.simantics.scl.compiler.types.Types;
  18 import org.simantics.scl.compiler.types.exceptions.MatchException;
  19
  20 import gnu.trove.map.hash.TObjectIntHashMap;
  21 import gnu.trove.set.hash.THashSet;
  22 import gnu.trove.set.hash.TIntHashSet;
  23
  24 public class ELambdaType extends Expression {
  25     public TVar[] parameters;
  26     public Expression value;
  27
  28     public ELambdaType(TVar[] parameters, Expression value) {
  29         super(value.getLocation());
  30         this.parameters = parameters;
  31         this.value = value;
  32     }
  33
  34         public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
  35         value.collectRefs(allRefs, refs);
  36     }
  37
  38         @Override
  39         public void collectVars(TObjectIntHashMap<Variable> allVars,
  40                 TIntHashSet vars) {
  41             value.collectVars(allVars, vars);
  42         }
  43
  44         @Override
  45         protected void updateType() throws MatchException {
  46             setType(Types.forAll(parameters, value.getType()));
  47         }
  48
  49         @Override
  50     public IVal toVal(Environment env, CodeWriter w) {
  51         return lambdaToVal(env, w);
  52     }
  53
  54     @Override
  55     public void collectFreeVariables(THashSet<Variable> vars) {
  56         value.collectFreeVariables(vars);
  57     }
  58
  59     @Override
  60     public Expression simplify(SimplificationContext context) {
  61         value = value.simplify(context);
  62         return this;
  63     }
  64
  65     @Override
  66     public Expression resolve(TranslationContext context) {
  67         value = value.resolve(context);
  68         return this;
  69     }
  70
  71     @Override
  72     public Expression replace(ReplaceContext context) {
  73         TVar[] newParameters = new TVar[parameters.length];
  74         for(int i=0;i<parameters.length;++i) {
  75             TVar var = Types.var(parameters[i].getKind());
  76             newParameters[i] = var;
  77             context.tvarMap.put(parameters[i], var);
  78         }
  79
  80         ELambdaType result = new ELambdaType(newParameters, value.replace(context));
  81         for(int i=0;i<parameters.length;++i)
  82             context.tvarMap.remove(parameters[i]);
  83         return result;
  84     }
  85
  86     @Override
  87     public IExpression toIExpression(ExpressionInterpretationContext target) {
  88         return value.toIExpression(target);
  89     }
  90
  91     @Override
  92     public Expression inferType(TypingContext context) {
  93         throw new InternalCompilerError("Should not type check " + getClass().getSimpleName() + ".");
  94     }
  95
  96     @Override
  97     public Expression decorate(ExpressionDecorator decorator) {
  98         value = value.decorate(decorator);
  99         return decorator.decorate(this);
 100     }
 101
 102     @Override
 103     public void collectEffects(THashSet<Type> effects) {
 104     }
 105
 106     @Override
 107     public void setLocationDeep(long loc) {
 108         if(location == Locations.NO_LOCATION) {
 109             location = loc;
 110             value.setLocationDeep(loc);
 111         }
 112     }
 113
 114     @Override
 115     public void accept(ExpressionVisitor visitor) {
 116         visitor.visit(this);
 117     }
 118
 119     @Override
 120     public void forVariables(VariableProcedure procedure) {
 121         value.forVariables(procedure);
 122     }
 123
 124     @Override
 125     public Expression accept(ExpressionTransformer transformer) {
 126         return transformer.transform(this);
 127     }
 128
 129     @Override
 130     public int getSyntacticFunctionArity() {
 131         return value.getSyntacticFunctionArity();
 132     }
 133
 134 }