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         throw new InternalCompilerError(location, "Cannot collect free variables for a pattern.");\r
  62     }\r
  63     \r
  64     @Override\r
  65     public void removeFreeVariables(THashSet<Variable> vars) {\r
  66         vars.remove(var);\r
  67         pattern.removeFreeVariables(vars);\r
  68     }\r
  69     \r
  70     @Override\r
  71     public Expression simplify(SimplificationContext context) {\r
  72         pattern = pattern.simplify(context);\r
  73         return this;\r
  74     }\r
  75 \r
  76     @Override\r
  77     public Expression resolve(TranslationContext context) {\r
  78         context.getErrorLog().log(location, "As binding (@) can be used only in patterns.");\r
  79         return this;\r
  80     }\r
  81     \r
  82     @Override\r
  83     public Expression resolveAsPattern(TranslationContext context) {\r
  84         var = context.newVariable(eVar.name);\r
  85         pattern = pattern.resolveAsPattern(context);\r
  86         return this;\r
  87     }\r
  88     \r
  89     @Override\r
  90     protected void updateType() throws MatchException {\r
  91         setType(pattern.getType());\r
  92     }\r
  93 \r
  94     @Override\r
  95     public IVal toVal(Environment env, CodeWriter w) {\r
  96         throw new InternalCompilerError("EAsPattern allowed only in patterns.");\r
  97     }\r
  98     \r
  99     @Override\r
 100     public Expression inferType(TypingContext context) {\r
 101         pattern = pattern.inferType(context);\r
 102         var.setType(pattern.getType());\r
 103         return this;\r
 104     }\r
 105     \r
 106     @Override\r
 107     public Expression checkBasicType(TypingContext context, Type requiredType) {\r
 108         pattern = pattern.checkBasicType(context, requiredType);\r
 109         var.setType(pattern.getType());\r
 110         return this;\r
 111     }\r
 112 \r
 113     @Override\r
 114     public Expression decorate(ExpressionDecorator decorator) {\r
 115         pattern = pattern.decorate(decorator);\r
 116         return decorator.decorate(this);\r
 117     }\r
 118     \r
 119     @Override\r
 120     public Expression replace(ReplaceContext context) {\r
 121         if(!context.inPattern) \r
 122             throw new IllegalStateException();\r
 123         \r
 124         Variable newVariable = new Variable(var.name);\r
 125         newVariable.setType(var.getType().replace(context.tvarMap));\r
 126         context.varMap.put(var, new EVariable(newVariable));\r
 127         \r
 128         EAsPattern result = new EAsPattern(eVar, pattern.replace(context));\r
 129         result.var = newVariable;\r
 130         \r
 131         return result;\r
 132     }\r
 133 \r
 134     @Override\r
 135     public void collectEffects(THashSet<Type> effects) {\r
 136         pattern.collectEffects(effects);\r
 137     }\r
 138     \r
 139     @Override\r
 140     public void setLocationDeep(long loc) {\r
 141         if(location == Locations.NO_LOCATION) {\r
 142             location = loc;\r
 143             pattern.setLocationDeep(loc);\r
 144             if(eVar != null)\r
 145                 eVar.setLocationDeep(loc);\r
 146         }\r
 147     }\r
 148     \r
 149     @Override\r
 150     public void accept(ExpressionVisitor visitor) {\r
 151         visitor.visit(this);\r
 152     }\r
 153 \r
 154     @Override\r
 155     public void forVariables(VariableProcedure procedure) {\r
 156         procedure.execute(eVar.location, var);\r
 157         pattern.forVariables(procedure);\r
 158     }\r
 159     \r
 160     @Override\r
 161     public Expression accept(ExpressionTransformer transformer) {\r
 162         return transformer.transform(this);\r
 163     }\r
 164 \r
 165 }\r