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