averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-21 12:14:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

silgen support for breaking out of a switch. rdar://16736255

Browse Source 
Swift SVN r16903
This commit is contained in:
Chris Lattner
2014-04-27 01:48:14 +00:00
parent 60044a9c54
commit a9a1e9fc94
2 changed files with 22 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
lib/SILGen/SILGenPattern.cpp
View File

@@ -1592,13 +1592,23 @@ static void emitCaseBody(SILGenFunction &gen,
/// Emit cases for a scope, identified by its continuation BB.
static void emitCasesForScope(SILGenFunction &gen,
SILBasicBlock *contBB,
const CaseMap &caseMap) {
const CaseMap &caseMap,
SwitchStmt *S) {
for (auto &cases : caseMap) {
CaseStmt *caseStmt = cases.first;
const CaseBlock &caseBlock = cases.second;
if (caseBlock.cont == contBB)
emitCaseBody(gen, caseStmt, caseBlock);
if (caseBlock.cont != contBB) continue;
gen.BreakContinueDestStack.push_back({
S,
JumpDest(contBB, gen.getCleanupsDepth(), CleanupLocation(caseStmt)),
JumpDest(caseStmt)
});
emitCaseBody(gen, caseStmt, caseBlock);
gen.BreakContinueDestStack.pop_back();
}
}
@@ -1702,6 +1712,8 @@ recur:
contLoc);
Scope dispatchScope(gen.Cleanups, contLoc);
gen.BreakContinueDestStack.push_back({ stmt, contDest, JumpDest(contLoc) });
defaultBB
= emitDispatchAndDestructure(gen, specializers,
clauses.getOccurrences()[0],
@@ -1741,7 +1753,7 @@ recur:
&& "recursive emitDecisionTree did not terminate all its BBs");
// Emit cases in this scope.
emitCasesForScope(gen, innerContBB, caseMap);
emitCasesForScope(gen, innerContBB, caseMap, stmt);
if (innerContBB->pred_empty()) {
// If the continuation wasn't used, kill it.
@@ -1756,6 +1768,8 @@ recur:
if (gen.B.hasValidInsertionPoint())
gen.Cleanups.emitBranchAndCleanups(contDest, CleanupLocation(stmt));
}
gen.BreakContinueDestStack.pop_back();
}
// If the dispatch was exhaustive, then emitDispatchAndDestructure returns
@@ -1808,7 +1822,7 @@ void SILGenFunction::emitSwitchStmt(SwitchStmt *S) {
// Create a continuation bb for life after the switch.
SILBasicBlock *contBB = createBasicBlock();
// Set up an initial clause matrix.
// Set up an initial clause matrix.
ClauseMatrix clauses(subject, S->getCases().size(),
S->getCases()[0]);
@@ -1840,7 +1854,7 @@ void SILGenFunction::emitSwitchStmt(SwitchStmt *S) {
"emitDecisionTree did not terminate all its BBs");
// Emit cases for the outermost scope.
emitCasesForScope(*this, contBB, caseMap);
emitCasesForScope(*this, contBB, caseMap, S);
SwitchStack.pop_back();