author | lagergren |
Fri, 19 Apr 2013 16:11:16 +0200 | |
changeset 17233 | 72ccf78a8216 |
parent 16530 | 201d682e75f4 |
child 17523 | cb4a7c901e0d |
permissions | -rw-r--r-- |
16147 | 1 |
/* |
16151 | 2 |
* Copyright (c) 2010, 2013, Oracle and/or its affiliates. All rights reserved. |
16147 | 3 |
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
4 |
* |
|
5 |
* This code is free software; you can redistribute it and/or modify it |
|
6 |
* under the terms of the GNU General Public License version 2 only, as |
|
7 |
* published by the Free Software Foundation. Oracle designates this |
|
8 |
* particular file as subject to the "Classpath" exception as provided |
|
9 |
* by Oracle in the LICENSE file that accompanied this code. |
|
10 |
* |
|
11 |
* This code is distributed in the hope that it will be useful, but WITHOUT |
|
12 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
|
13 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
|
14 |
* version 2 for more details (a copy is included in the LICENSE file that |
|
15 |
* accompanied this code). |
|
16 |
* |
|
17 |
* You should have received a copy of the GNU General Public License version |
|
18 |
* 2 along with this work; if not, write to the Free Software Foundation, |
|
19 |
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
|
20 |
* |
|
21 |
* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
|
22 |
* or visit www.oracle.com if you need additional information or have any |
|
23 |
* questions. |
|
24 |
*/ |
|
25 |
||
26 |
package jdk.nashorn.internal.ir; |
|
27 |
||
28 |
import java.util.ArrayList; |
|
29 |
import java.util.Collections; |
|
30 |
import java.util.List; |
|
17233 | 31 |
|
16240
e1468b33e201
8008239: Unpublicized parts of the code generator package that were only package internal.
lagergren
parents:
16160
diff
changeset
|
32 |
import jdk.nashorn.internal.codegen.Label; |
17233 | 33 |
import jdk.nashorn.internal.ir.annotations.Immutable; |
16147 | 34 |
import jdk.nashorn.internal.ir.visitor.NodeVisitor; |
35 |
import jdk.nashorn.internal.runtime.Source; |
|
36 |
||
37 |
/** |
|
38 |
* IR representation of a SWITCH statement. |
|
39 |
*/ |
|
17233 | 40 |
@Immutable |
41 |
public final class SwitchNode extends BreakableNode { |
|
16147 | 42 |
/** Switch expression. */ |
17233 | 43 |
private final Node expression; |
44 |
||
45 |
/** Switch cases. */ |
|
46 |
private final List<CaseNode> cases; |
|
47 |
||
48 |
/** Switch default index. */ |
|
49 |
private final int defaultCaseIndex; |
|
16147 | 50 |
|
51 |
/** Tag symbol. */ |
|
52 |
private Symbol tag; |
|
53 |
||
54 |
/** |
|
55 |
* Constructor |
|
56 |
* |
|
17233 | 57 |
* @param source the source |
58 |
* @param token token |
|
59 |
* @param finish finish |
|
60 |
* @param expression switch expression |
|
61 |
* @param cases cases |
|
62 |
* @param defaultCase the default case node - null if none, otherwise has to be present in cases list |
|
16147 | 63 |
*/ |
17233 | 64 |
public SwitchNode(final Source source, final long token, final int finish, final Node expression, final List<CaseNode> cases, final CaseNode defaultCase) { |
65 |
super(source, token, finish, new Label("switch_break")); |
|
66 |
this.expression = expression; |
|
67 |
this.cases = cases; |
|
68 |
this.defaultCaseIndex = defaultCase == null ? -1 : cases.indexOf(defaultCase); |
|
16147 | 69 |
} |
70 |
||
17233 | 71 |
private SwitchNode(final SwitchNode switchNode, final Node expression, final List<CaseNode> cases, final int defaultCase) { |
16147 | 72 |
super(switchNode); |
17233 | 73 |
this.expression = expression; |
74 |
this.cases = cases; |
|
75 |
this.defaultCaseIndex = defaultCase; |
|
76 |
this.tag = switchNode.getTag(); //TODO are symbols inhereted as references? |
|
16147 | 77 |
} |
78 |
||
79 |
@Override |
|
17233 | 80 |
public Node ensureUniqueLabels(final LexicalContext lc) { |
81 |
final List<CaseNode> newCases = new ArrayList<>(); |
|
82 |
for (final CaseNode caseNode : cases) { |
|
83 |
newCases.add(new CaseNode(caseNode, caseNode.getTest(), caseNode.getBody())); |
|
84 |
} |
|
85 |
return Node.replaceInLexicalContext(lc, this, new SwitchNode(this, expression, newCases, defaultCaseIndex)); |
|
16147 | 86 |
} |
87 |
||
88 |
@Override |
|
17233 | 89 |
public boolean isTerminal() { |
90 |
//there must be a default case, and that including all other cases must terminate |
|
91 |
if (!cases.isEmpty() && defaultCaseIndex != -1) { |
|
92 |
for (final CaseNode caseNode : cases) { |
|
93 |
if (!caseNode.isTerminal()) { |
|
94 |
return false; |
|
95 |
} |
|
96 |
} |
|
97 |
return true; |
|
98 |
} |
|
99 |
return false; |
|
16147 | 100 |
|
17233 | 101 |
} |
16147 | 102 |
|
17233 | 103 |
@Override |
104 |
public Node accept(final LexicalContext lc, final NodeVisitor visitor) { |
|
105 |
if (visitor.enterSwitchNode(this)) { |
|
106 |
return visitor.leaveSwitchNode( |
|
107 |
setExpression(visitor.getLexicalContext(), expression.accept(visitor)). |
|
108 |
setCases(visitor.getLexicalContext(), Node.accept(visitor, CaseNode.class, cases), defaultCaseIndex)); |
|
16147 | 109 |
} |
110 |
||
111 |
return this; |
|
112 |
} |
|
113 |
||
114 |
@Override |
|
115 |
public void toString(final StringBuilder sb) { |
|
116 |
sb.append("switch ("); |
|
117 |
expression.toString(sb); |
|
118 |
sb.append(')'); |
|
119 |
} |
|
120 |
||
121 |
/** |
|
17233 | 122 |
* Return the case node that is default case |
123 |
* @return default case or null if none |
|
124 |
*/ |
|
125 |
public CaseNode getDefaultCase() { |
|
126 |
return defaultCaseIndex == -1 ? null : cases.get(defaultCaseIndex); |
|
127 |
} |
|
128 |
||
129 |
/** |
|
16147 | 130 |
* Get the cases in this switch |
131 |
* @return a list of case nodes |
|
132 |
*/ |
|
133 |
public List<CaseNode> getCases() { |
|
134 |
return Collections.unmodifiableList(cases); |
|
135 |
} |
|
136 |
||
137 |
/** |
|
17233 | 138 |
* Replace case nodes with new list. the cases have to be the same |
139 |
* and the default case index the same. This is typically used |
|
140 |
* by NodeVisitors who perform operations on every case node |
|
141 |
* @param lc lexical context |
|
142 |
* @param cases list of cases |
|
143 |
* @return new switcy node or same if no state was changed |
|
16147 | 144 |
*/ |
17233 | 145 |
public SwitchNode setCases(final LexicalContext lc, final List<CaseNode> cases) { |
146 |
return setCases(lc, cases, defaultCaseIndex); |
|
147 |
} |
|
148 |
||
149 |
private SwitchNode setCases(final LexicalContext lc, final List<CaseNode> cases, final int defaultCaseIndex) { |
|
150 |
if (this.cases == cases) { |
|
151 |
return this; |
|
152 |
} |
|
153 |
return Node.replaceInLexicalContext(lc, this, new SwitchNode(this, expression, cases, defaultCaseIndex)); |
|
16147 | 154 |
} |
155 |
||
156 |
/** |
|
17233 | 157 |
* Set or reset the list of cases in this switch |
158 |
* @param lc lexical context |
|
159 |
* @param cases a list of cases, case nodes |
|
160 |
* @param defaultCase a case in the list that is the default - must be in the list or class will assert |
|
161 |
* @return new switch node or same if no state was changed |
|
16147 | 162 |
*/ |
17233 | 163 |
public SwitchNode setCases(final LexicalContext lc, final List<CaseNode> cases, final CaseNode defaultCase) { |
164 |
return setCases(lc, cases, defaultCase == null ? -1 : cases.indexOf(defaultCase)); |
|
16147 | 165 |
} |
166 |
||
167 |
/** |
|
168 |
* Return the expression to switch on |
|
169 |
* @return switch expression |
|
170 |
*/ |
|
171 |
public Node getExpression() { |
|
172 |
return expression; |
|
173 |
} |
|
174 |
||
175 |
/** |
|
176 |
* Set or reset the expression to switch on |
|
17233 | 177 |
* @param lc lexical context |
16147 | 178 |
* @param expression switch expression |
17233 | 179 |
* @return new switch node or same if no state was changed |
16147 | 180 |
*/ |
17233 | 181 |
public SwitchNode setExpression(final LexicalContext lc, final Node expression) { |
182 |
if (this.expression == expression) { |
|
183 |
return this; |
|
184 |
} |
|
185 |
return Node.replaceInLexicalContext(lc, this, new SwitchNode(this, expression, cases, defaultCaseIndex)); |
|
16147 | 186 |
} |
187 |
||
188 |
/** |
|
189 |
* Get the tag symbol for this switch. The tag symbol is where |
|
190 |
* the switch expression result is stored |
|
191 |
* @return tag symbol |
|
192 |
*/ |
|
193 |
public Symbol getTag() { |
|
194 |
return tag; |
|
195 |
} |
|
196 |
||
197 |
/** |
|
198 |
* Set the tag symbol for this switch. The tag symbol is where |
|
199 |
* the switch expression result is stored |
|
200 |
* @param tag a symbol |
|
201 |
*/ |
|
202 |
public void setTag(final Symbol tag) { |
|
203 |
this.tag = tag; |
|
204 |
} |
|
205 |
} |
|
206 |