18f89ccdf7dcb885eb943176f6cdbf136d3e0a4e
[simantics/platform.git] / 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 }