]> gerrit.simantics Code Review - simantics/platform.git/blob
0a11d9b99b169e4787bc4526babd8384cbc01a6a
[simantics/platform.git] /
1 package org.simantics.scl.compiler.elaboration.expressions;\r
2 \r
3 import gnu.trove.map.hash.TObjectIntHashMap;\r
4 import gnu.trove.set.hash.THashSet;\r
5 import gnu.trove.set.hash.TIntHashSet;\r
6 \r
7 import org.simantics.scl.compiler.common.exceptions.InternalCompilerError;\r
8 import org.simantics.scl.compiler.elaboration.contexts.ReplaceContext;\r
9 import org.simantics.scl.compiler.elaboration.contexts.SimplificationContext;\r
10 import org.simantics.scl.compiler.elaboration.contexts.TranslationContext;\r
11 import org.simantics.scl.compiler.elaboration.contexts.TypingContext;\r
12 import org.simantics.scl.compiler.environment.Environment;\r
13 import org.simantics.scl.compiler.errors.Locations;\r
14 import org.simantics.scl.compiler.internal.codegen.references.IVal;\r
15 import org.simantics.scl.compiler.internal.codegen.writer.CodeWriter;\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.exceptions.MatchException;\r
19 \r
20 public class EAsPattern extends Expression {\r
21 \r
22     Variable var;\r
23     EVar eVar;\r
24     Expression pattern;\r
25 \r
26     public EAsPattern(EVar eVar, Expression pattern) {\r
27         this.eVar = eVar;\r
28         this.pattern = pattern;\r
29     }\r
30     \r
31     public EAsPattern(Variable var, Expression pattern) {\r
32         this.var = var;\r
33         this.pattern = pattern;\r
34     }\r
35     \r
36     public Variable getVariable() {\r
37         return var;\r
38     }\r
39     \r
40     public Expression getPattern() {\r
41         return pattern;\r
42     }\r
43 \r
44     @Override\r
45     public void collectRefs(TObjectIntHashMap<Object> allRefs,\r
46             TIntHashSet refs) {\r
47         pattern.collectRefs(allRefs, refs);\r
48     }\r
49 \r
50     @Override\r
51     public void collectVars(TObjectIntHashMap<Variable> allVars,\r
52             TIntHashSet vars) {\r
53         int id = allVars.get(var);\r
54         if(id >= 0)\r
55             vars.add(id);\r
56         pattern.collectVars(allVars, vars);\r
57     }\r
58 \r
59     @Override\r
60     public void collectFreeVariables(THashSet<Variable> vars) {\r
61         vars.add(var);\r
62         pattern.collectFreeVariables(vars);\r
63     }\r
64     \r
65     @Override\r
66     public Expression simplify(SimplificationContext context) {\r
67         pattern = pattern.simplify(context);\r
68         return this;\r
69     }\r
70 \r
71     @Override\r
72     public Expression resolve(TranslationContext context) {\r
73         context.getErrorLog().log(location, "As binding (@) can be used only in patterns.");\r
74         return this;\r
75     }\r
76     \r
77     @Override\r
78     public Expression resolveAsPattern(TranslationContext context) {\r
79         var = context.newVariable(eVar.name);\r
80         pattern = pattern.resolveAsPattern(context);\r
81         return this;\r
82     }\r
83     \r
84     @Override\r
85     protected void updateType() throws MatchException {\r
86         setType(pattern.getType());\r
87     }\r
88 \r
89     @Override\r
90     public IVal toVal(Environment env, CodeWriter w) {\r
91         throw new InternalCompilerError("EAsPattern allowed only in patterns.");\r
92     }\r
93     \r
94     @Override\r
95     public Expression inferType(TypingContext context) {\r
96         pattern = pattern.inferType(context);\r
97         var.setType(pattern.getType());\r
98         return this;\r
99     }\r
100     \r
101     @Override\r
102     public Expression checkBasicType(TypingContext context, Type requiredType) {\r
103         pattern = pattern.checkBasicType(context, requiredType);\r
104         var.setType(pattern.getType());\r
105         return this;\r
106     }\r
107 \r
108     @Override\r
109     public Expression decorate(ExpressionDecorator decorator) {\r
110         pattern = pattern.decorate(decorator);\r
111         return decorator.decorate(this);\r
112     }\r
113     \r
114     @Override\r
115     public Expression replace(ReplaceContext context) {\r
116         if(!context.inPattern) \r
117             throw new IllegalStateException();\r
118         \r
119         Variable newVariable = new Variable(var.name);\r
120         newVariable.setType(var.getType().replace(context.tvarMap));\r
121         context.varMap.put(var, new EVariable(newVariable));\r
122         \r
123         EAsPattern result = new EAsPattern(eVar, pattern.replace(context));\r
124         result.var = newVariable;\r
125         \r
126         return result;\r
127     }\r
128 \r
129     @Override\r
130     public void collectEffects(THashSet<Type> effects) {\r
131         pattern.collectEffects(effects);\r
132     }\r
133     \r
134     @Override\r
135     public void setLocationDeep(long loc) {\r
136         if(location == Locations.NO_LOCATION) {\r
137             location = loc;\r
138             pattern.setLocationDeep(loc);\r
139             if(eVar != null)\r
140                 eVar.setLocationDeep(loc);\r
141         }\r
142     }\r
143     \r
144     @Override\r
145     public void accept(ExpressionVisitor visitor) {\r
146         visitor.visit(this);\r
147     }\r
148 \r
149     @Override\r
150     public void forVariables(VariableProcedure procedure) {\r
151         procedure.execute(eVar.location, var);\r
152         pattern.forVariables(procedure);\r
153     }\r
154     \r
155     @Override\r
156     public Expression accept(ExpressionTransformer transformer) {\r
157         return transformer.transform(this);\r
158     }\r
159 \r
160 }\r