gerrit.simantics Code Review - simantics/platform.git/blob

   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.Type;
  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.THashSet;
  20 import gnu.trove.set.hash.TIntHashSet;
  21
  22 public class EApplyType extends Expression {
  23     Expression expression;
  24     Type parameter;
  25
  26     public EApplyType(Expression expression, Type parameter) {
  27         this.expression = expression;
  28         this.parameter = parameter;
  29     }
  30
  31     public static Expression create(Expression expression, Type ... parameters) {
  32         for(Type parameter : parameters)
  33             expression = new EApplyType(expression, parameter);
  34         return expression;
  35     }
  36
  37     public EApplyType(long loc, Expression expression, Type parameter) {
  38         super(loc);
  39         if(parameter == null)
  40             throw new NullPointerException();
  41         this.expression = expression;
  42         this.parameter = parameter;
  43     }
  44
  45     public Expression getExpression() {
  46         return expression;
  47     }
  48
  49     public Type getParameter() {
  50         return parameter;
  51     }
  52
  53         public void collectRefs(TObjectIntHashMap<Object> allRefs, TIntHashSet refs) {
  54         expression.collectRefs(allRefs, refs);
  55     }
  56
  57         public void collectVars(TObjectIntHashMap<Variable> allVars, TIntHashSet vars) {
  58             expression.collectVars(allVars, vars);
  59         }
  60
  61         @Override
  62         protected void updateType() throws MatchException {
  63             setType(Types.instantiate(expression.getType(), parameter));
  64         }
  65
  66         @Override
  67         public IVal toVal(CompilationContext context, CodeWriter w) {
  68         IVal val = expression.toVal(context, w);
  69         return val.createSpecialization(parameter);
  70     }
  71
  72     @Override
  73     public void collectFreeVariables(THashSet<Variable> vars) {
  74         expression.collectFreeVariables(vars);
  75     }
  76
  77     @Override
  78     public Expression simplify(SimplificationContext context) {
  79         expression = expression.simplify(context);
  80         if(expression instanceof EConstant) {
  81             ((EConstant)expression).addTypeParameters(parameter);
  82             return expression;
  83         }
  84         return this;
  85     }
  86
  87     @Override
  88     public Expression resolve(TranslationContext context) {
  89         expression = expression.resolve(context);
  90         return this;
  91     }
  92
  93     @Override
  94     public void removeFreeVariables(THashSet<Variable> vars) {
  95         expression.removeFreeVariables(vars);
  96     }
  97
  98     @Override
  99     public Expression replace(ReplaceContext context) {
 100         return new EApplyType(expression.replace(context),
 101                 parameter.replace(context.tvarMap));
 102     }
 103
 104     @Override
 105     public IExpression toIExpression(ExpressionInterpretationContext target) {
 106         return expression.toIExpression(target);
 107     }
 108
 109     @Override
 110     public Expression inferType(TypingContext context) {
 111         throw new InternalCompilerError("Should not type check " + getClass().getSimpleName() + ".");
 112     }
 113
 114     @Override
 115     public boolean isEffectful() {
 116         return expression.isEffectful();
 117     }
 118
 119     @Override
 120     public void collectEffects(THashSet<Type> effects) {
 121         expression.collectEffects(effects);
 122     }
 123
 124     @Override
 125     public void setLocationDeep(long loc) {
 126         if(location == Locations.NO_LOCATION) {
 127             location = loc;
 128             expression.setLocationDeep(loc);
 129         }
 130     }
 131
 132     @Override
 133     public void accept(ExpressionVisitor visitor) {
 134         visitor.visit(this);
 135     }
 136
 137     @Override
 138     public void forVariables(VariableProcedure procedure) {
 139         expression.forVariables(procedure);
 140     }
 141
 142     @Override
 143     public Expression accept(ExpressionTransformer transformer) {
 144         return transformer.transform(this);
 145     }
 146
 147 }