Add function to replace a destination MBB in a single jump table

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@88804 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Jim Grosbach
2009-11-14 20:09:13 +00:00
parent 1b226abcd9
commit 68bb60f6a4
2 changed files with 24 additions and 8 deletions

View File

@ -69,6 +69,11 @@ public:
/// the jump tables to branch to New instead. /// the jump tables to branch to New instead.
bool ReplaceMBBInJumpTables(MachineBasicBlock *Old, MachineBasicBlock *New); bool ReplaceMBBInJumpTables(MachineBasicBlock *Old, MachineBasicBlock *New);
/// ReplaceMBBInJumpTable - If Old is a target of the jump tables, update
/// the jump table to branch to New instead.
bool ReplaceMBBInJumpTable(unsigned Idx, MachineBasicBlock *Old,
MachineBasicBlock *New);
/// getEntrySize - Returns the size of an individual field in a jump table. /// getEntrySize - Returns the size of an individual field in a jump table.
/// ///
unsigned getEntrySize() const { return EntrySize; } unsigned getEntrySize() const { return EntrySize; }

View File

@ -545,14 +545,25 @@ MachineJumpTableInfo::ReplaceMBBInJumpTables(MachineBasicBlock *Old,
MachineBasicBlock *New) { MachineBasicBlock *New) {
assert(Old != New && "Not making a change?"); assert(Old != New && "Not making a change?");
bool MadeChange = false; bool MadeChange = false;
for (size_t i = 0, e = JumpTables.size(); i != e; ++i) { for (size_t i = 0, e = JumpTables.size(); i != e; ++i)
MachineJumpTableEntry &JTE = JumpTables[i]; ReplaceMBBInJumpTable(i, Old, New);
for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j) return MadeChange;
if (JTE.MBBs[j] == Old) { }
JTE.MBBs[j] = New;
MadeChange = true; /// ReplaceMBBInJumpTable - If Old is a target of the jump tables, update
} /// the jump table to branch to New instead.
} bool
MachineJumpTableInfo::ReplaceMBBInJumpTable(unsigned Idx,
MachineBasicBlock *Old,
MachineBasicBlock *New) {
assert(Old != New && "Not making a change?");
bool MadeChange = false;
MachineJumpTableEntry &JTE = JumpTables[Idx];
for (size_t j = 0, e = JTE.MBBs.size(); j != e; ++j)
if (JTE.MBBs[j] == Old) {
JTE.MBBs[j] = New;
MadeChange = true;
}
return MadeChange; return MadeChange;
} }