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