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

   1 package org.simantics.scl.compiler.elaboration.expressions;
   2
   3 import java.util.ArrayList;
   4
   5 import org.simantics.scl.compiler.compilation.CompilationContext;
   6 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;
   7 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;
   8 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;
   9 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;
  10 import org.simantics.scl.compiler.errors.Locations;
  11 import org.simantics.scl.compiler.internal.codegen.references.IVal;
  12 import org.simantics.scl.compiler.internal.codegen.ssa.exits.Throw;
  13 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;
  14 import org.simantics.scl.compiler.internal.elaboration.matching.PatternMatchingCompiler;
  15 import org.simantics.scl.compiler.internal.elaboration.matching.Row;
  16 import org.simantics.scl.compiler.types.Type;
  17 import org.simantics.scl.compiler.types.Types;
  18 import org.simantics.scl.compiler.types.kinds.Kinds;
  19
  20 import gnu.trove.map.hash.TObjectIntHashMap;
  21 import gnu.trove.set.hash.TIntHashSet;
  22
  23 public class EMatch extends Expression {
  24
  25     public Expression[] scrutinee;
  26     public Case[] cases;
  27
  28     public EMatch(Expression[] scrutinee, Case ... cases) {
  29         this.scrutinee = scrutinee;
  30         this.cases = cases;
  31     }
  32
  33     public EMatch(Expression scrutinee, Case ... cases) {
  34         this(new Expression[] {scrutinee}, cases);
  35     }
  36
  37     public EMatch(long loc, Expression[] scrutinee, Case ... cases) {
  38         super(loc);
  39         this.scrutinee = scrutinee;
  40         this.cases = cases;
  41     }
  42
  43     @Override
  44     public void collectVars(TObjectIntHashMap<Variable> allVars,
  45             TIntHashSet vars) {
  46         for(Expression s : scrutinee)
  47             s.collectVars(allVars, vars);
  48         for(Case case_ : cases)
  49             case_.collectVars(allVars, vars);
  50     }
  51
  52     @Override
  53     protected void updateType() {
  54         setType(cases[0].value.getType());
  55     }
  56
  57     @Override
  58     public IVal toVal(CompilationContext context, CodeWriter w) {
  59         ArrayList<Row> rows = new ArrayList<Row>(cases.length);
  60         for(Case case_ : cases)
  61             rows.add(new Row(case_.patterns, case_.value));
  62
  63         IVal[] scrutineeVals = new IVal[scrutinee.length];
  64         for(int i=0;i<scrutinee.length;++i)
  65             scrutineeVals[i] = scrutinee[i].toVal(context, w);
  66
  67         CodeWriter joinPoint = w.createBlock(getType());
  68         CodeWriter failurePoint = w.createBlock(); // TODO generate only one failurePoint per function
  69         PatternMatchingCompiler.split(w, context, scrutineeVals, joinPoint.getContinuation(), failurePoint.getContinuation(), rows);
  70         failurePoint.throw_(location, Throw.MatchingException, "Matching failure at: " + toString());
  71         w.continueAs(joinPoint);
  72         return w.getParameters()[0];
  73     }
  74
  75     @Override
  76     public Expression simplify(SimplificationContext context) {
  77         for(int i=0;i<scrutinee.length;++i)
  78             scrutinee[i] = scrutinee[i].simplify(context);
  79         for(Case case_ : cases)
  80             case_.simplify(context);
  81         if(cases.length == 1 && scrutinee.length == 1) {
  82             Case case_ = cases[0];
  83             Expression pattern = case_.patterns[0];
  84             if(case_.patterns[0] instanceof EVariable
  85                     && !(case_.value instanceof GuardedExpressionGroup)) {
  86                 Variable var = ((EVariable)pattern).variable;
  87                 return new ESimpleLet(var, scrutinee[0], case_.value);
  88             }
  89         }
  90         return this;
  91     }
  92
  93     @Override
  94     public Expression resolve(TranslationContext context) {
  95         for(int i=0;i<scrutinee.length;++i)
  96             scrutinee[i] = scrutinee[i].resolve(context);
  97         for(Case case_ : cases)
  98             case_.resolve(context);
  99         return this;
 100     }
 101
 102     @Override
 103     public void setLocationDeep(long loc) {
 104         if(location == Locations.NO_LOCATION) {
 105             location = loc;
 106             for(Case case_ : cases)
 107                 case_.setLocationDeep(loc);
 108             for(Expression e : scrutinee)
 109                 e.setLocationDeep(loc);
 110         }
 111     }
 112
 113     @Override
 114     public Expression replace(ReplaceContext context) {
 115         Expression[] newScrutinee = new Expression[scrutinee.length];
 116         for(int i=0;i<scrutinee.length;++i)
 117             newScrutinee[i] = scrutinee[i].replace(context);
 118         Case[] newCases = new Case[cases.length];
 119         for(int i=0;i<cases.length;++i)
 120             newCases[i] = cases[i].replace(context);
 121         return new EMatch(getLocation(), newScrutinee, newCases);
 122     }
 123
 124     @Override
 125     public Expression checkBasicType(TypingContext context, Type requiredType) {
 126         Type[] scrutineeTypes = new Type[scrutinee.length];
 127         for(int i=0;i<scrutinee.length;++i) {
 128             scrutinee[i] = scrutinee[i].checkType(context, Types.metaVar(Kinds.STAR));
 129             scrutineeTypes[i] = scrutinee[i].getType();
 130         }
 131         for(Case case_ : cases)
 132             case_.checkType(context, scrutineeTypes, requiredType);
 133         setType(requiredType);
 134         return this;
 135     }
 136
 137     @Override
 138     public Expression checkIgnoredType(TypingContext context) {
 139         Type[] scrutineeTypes = new Type[scrutinee.length];
 140         for(int i=0;i<scrutinee.length;++i) {
 141             scrutinee[i] = scrutinee[i].checkType(context, Types.metaVar(Kinds.STAR));
 142             scrutineeTypes[i] = scrutinee[i].getType();
 143         }
 144         for(Case case_ : cases)
 145             case_.checkIgnoredType(context, scrutineeTypes);
 146         setType(Types.UNIT);
 147         return this;
 148     }
 149
 150     @Override
 151     public void accept(ExpressionVisitor visitor) {
 152         visitor.visit(this);
 153     }
 154
 155     public Expression[] getScrutinee() {
 156         return scrutinee;
 157     }
 158
 159     public Case[] getCases() {
 160         return cases;
 161     }
 162
 163     @Override
 164     public Expression accept(ExpressionTransformer transformer) {
 165         return transformer.transform(this);
 166     }
 167
 168     @Override
 169     public int getSyntacticFunctionArity() {
 170         int result = 0;
 171         for(Case case_ : cases)
 172             result = Math.max(result, case_.value.getSyntacticFunctionArity());
 173         return result;
 174     }
 175 }