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

   1 package org.simantics.scl.compiler.elaboration.expressions;\r
   2 \r
   3 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;\r
   4 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;\r
   5 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
   6 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
   7 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
   8 import org.simantics.scl.compiler.environment.Environment;\r
   9 import org.simantics.scl.compiler.errors.Locations;\r
  10 import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
  11 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\r
  12 import org.simantics.scl.compiler.internal.elaboration.utils.ExpressionDecorator;\r
  13 import org.simantics.scl.compiler.types.Type;\r
  14 import org.simantics.scl.compiler.types.exceptions.MatchException;\r
  15 \r
  16 import gnu.trove.map.hash.TObjectIntHashMap;\r
  17 import gnu.trove.set.hash.THashSet;\r
  18 import gnu.trove.set.hash.TIntHashSet;\r
  19 \r
  20 public class EAsPattern extends Expression {\r
  21 \r
  22     Variable var;\r
  23     EVar eVar;\r
  24     Expression pattern;\r
  25 \r
  26     public EAsPattern(EVar eVar, Expression pattern) {\r
  27         this.eVar = eVar;\r
  28         this.pattern = pattern;\r
  29     }\r
  30     \r
  31     public EAsPattern(Variable var, Expression pattern) {\r
  32         this.var = var;\r
  33         this.pattern = pattern;\r
  34     }\r
  35     \r
  36     public Variable getVariable() {\r
  37         return var;\r
  38     }\r
  39     \r
  40     public Expression getPattern() {\r
  41         return pattern;\r
  42     }\r
  43 \r
  44     @Override\r
  45     public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
  46             TIntHashSet refs) {\r
  47         pattern.collectRefs(allRefs, refs);\r
  48     }\r
  49 \r
  50     @Override\r
  51     public void collectVars(TObjectIntHashMap<Variable> allVars,\r
  52             TIntHashSet vars) {\r
  53         int id = allVars.get(var);\r
  54         if(id >= 0)\r
  55             vars.add(id);\r
  56         pattern.collectVars(allVars, vars);\r
  57     }\r
  58 \r
  59     @Override\r
  60     public void collectFreeVariables(THashSet<Variable> vars) {\r
  61         vars.add(var);\r
  62         pattern.collectFreeVariables(vars);\r
  63     }\r
  64     \r
  65     @Override\r
  66     public Expression simplify(SimplificationContext context) {\r
  67         pattern = pattern.simplify(context);\r
  68         return this;\r
  69     }\r
  70 \r
  71     @Override\r
  72     public Expression resolve(TranslationContext context) {\r
  73         context.getErrorLog().log(location, "As binding (@) can be used only in patterns.");\r
  74         return this;\r
  75     }\r
  76     \r
  77     @Override\r
  78     public Expression resolveAsPattern(TranslationContext context) {\r
  79         var = context.newVariable(eVar.name);\r
  80         pattern = pattern.resolveAsPattern(context);\r
  81         return this;\r
  82     }\r
  83     \r
  84     @Override\r
  85     protected void updateType() throws MatchException {\r
  86         setType(pattern.getType());\r
  87     }\r
  88 \r
  89     @Override\r
  90     public IVal toVal(Environment env, CodeWriter w) {\r
  91         throw new InternalCompilerError("EAsPattern allowed only in patterns.");\r
  92     }\r
  93     \r
  94     @Override\r
  95     public Expression inferType(TypingContext context) {\r
  96         pattern = pattern.inferType(context);\r
  97         var.setType(pattern.getType());\r
  98         return this;\r
  99     }\r
 100     \r
 101     @Override\r
 102     public Expression checkBasicType(TypingContext context, Type requiredType) {\r
 103         pattern = pattern.checkBasicType(context, requiredType);\r
 104         var.setType(pattern.getType());\r
 105         return this;\r
 106     }\r
 107 \r
 108     @Override\r
 109     public Expression decorate(ExpressionDecorator decorator) {\r
 110         pattern = pattern.decorate(decorator);\r
 111         return decorator.decorate(this);\r
 112     }\r
 113     \r
 114     @Override\r
 115     public Expression replace(ReplaceContext context) {\r
 116         if(!context.inPattern) \r
 117             throw new IllegalStateException();\r
 118         \r
 119         Variable newVariable = new Variable(var.name);\r
 120         newVariable.setType(var.getType().replace(context.tvarMap));\r
 121         context.varMap.put(var, new EVariable(newVariable));\r
 122         \r
 123         EAsPattern result = new EAsPattern(eVar, pattern.replace(context));\r
 124         result.var = newVariable;\r
 125         \r
 126         return result;\r
 127     }\r
 128 \r
 129     @Override\r
 130     public void collectEffects(THashSet<Type> effects) {\r
 131         pattern.collectEffects(effects);\r
 132     }\r
 133     \r
 134     @Override\r
 135     public void setLocationDeep(long loc) {\r
 136         if(location == Locations.NO_LOCATION) {\r
 137             location = loc;\r
 138             pattern.setLocationDeep(loc);\r
 139             if(eVar != null)\r
 140                 eVar.setLocationDeep(loc);\r
 141         }\r
 142     }\r
 143     \r
 144     @Override\r
 145     public void accept(ExpressionVisitor visitor) {\r
 146         visitor.visit(this);\r
 147     }\r
 148 \r
 149     @Override\r
 150     public void forVariables(VariableProcedure procedure) {\r
 151         procedure.execute(eVar.location, var);\r
 152         pattern.forVariables(procedure);\r
 153     }\r
 154     \r
 155     @Override\r
 156     public Expression accept(ExpressionTransformer transformer) {\r
 157         return transformer.transform(this);\r
 158     }\r
 159 \r
 160 }\r