(refs #7375) Replaced collectFreeVariables method by a visitor
[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.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 }