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.compilation.CompilationContext;
   5 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
   6 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
   7 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   8 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
   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.interpreted.IExpression;
  13 import org.simantics.scl.compiler.top.ExpressionInterpretationContext;
  14 import org.simantics.scl.compiler.types.TVar;
  15 import org.simantics.scl.compiler.types.Types;
  16 import org.simantics.scl.compiler.types.exceptions.MatchException;
  17
  18 import gnu.trove.map.hash.TObjectIntHashMap;
  19 import gnu.trove.set.hash.TIntHashSet;
  20
  21 public class ELambdaType extends Expression {
  22     public TVar[] parameters;
  23     public Expression value;
  24
  25     public ELambdaType(TVar[] parameters, Expression value) {
  26         super(value.getLocation());
  27         this.parameters = parameters;
  28         this.value = value;
  29     }
  30
  31     @Override
  32     public void collectVars(TObjectIntHashMap<Variable> allVars,
  33             TIntHashSet vars) {
  34         value.collectVars(allVars, vars);
  35     }
  36
  37     @Override
  38     protected void updateType() throws MatchException {
  39         setType(Types.forAll(parameters, value.getType()));
  40     }
  41
  42     @Override
  43     public IVal toVal(CompilationContext context, CodeWriter w) {
  44         return lambdaToVal(context, w);
  45     }
  46
  47     @Override
  48     public Expression simplify(SimplificationContext context) {
  49         value = value.simplify(context);
  50         return this;
  51     }
  52
  53     @Override
  54     public Expression resolve(TranslationContext context) {
  55         value = value.resolve(context);
  56         return this;
  57     }
  58
  59     @Override
  60     public Expression replace(ReplaceContext context) {
  61         TVar[] newParameters = new TVar[parameters.length];
  62         for(int i=0;i<parameters.length;++i) {
  63             TVar var = Types.var(parameters[i].getKind());
  64             newParameters[i] = var;
  65             context.tvarMap.put(parameters[i], var);
  66         }
  67
  68         ELambdaType result = new ELambdaType(newParameters, value.replace(context));
  69         for(int i=0;i<parameters.length;++i)
  70             context.tvarMap.remove(parameters[i]);
  71         return result;
  72     }
  73
  74     @Override
  75     public IExpression toIExpression(ExpressionInterpretationContext target) {
  76         return value.toIExpression(target);
  77     }
  78
  79     @Override
  80     public Expression inferType(TypingContext context) {
  81         throw new InternalCompilerError("Should not type check " + getClass().getSimpleName() + ".");
  82     }
  83
  84     @Override
  85     public void setLocationDeep(long loc) {
  86         if(location == Locations.NO_LOCATION) {
  87             location = loc;
  88             value.setLocationDeep(loc);
  89         }
  90     }
  91
  92     @Override
  93     public void accept(ExpressionVisitor visitor) {
  94         visitor.visit(this);
  95     }
  96
  97     @Override
  98     public Expression accept(ExpressionTransformer transformer) {
  99         return transformer.transform(this);
 100     }
 101
 102     @Override
 103     public int getSyntacticFunctionArity() {
 104         return value.getSyntacticFunctionArity();
 105     }
 106
 107 }