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