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

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