BreakCriticalMachineEdge.h [plain text]
#ifndef LLVM_CODEGEN_BREAKCRITICALMACHINEEDGE_H
#define LLVM_CODEGEN_BREAKCRITICALMACHINEEDGE_H
#include "llvm/CodeGen/MachineJumpTableInfo.h"
#include "llvm/Target/TargetInstrInfo.h"
#include "llvm/Target/TargetMachine.h"
namespace llvm {
MachineBasicBlock* SplitCriticalMachineEdge(MachineBasicBlock* src,
MachineBasicBlock* dst) {
MachineFunction &MF = *src->getParent();
const BasicBlock* srcBB = src->getBasicBlock();
MachineBasicBlock* crit_mbb = MF.CreateMachineBasicBlock(srcBB);
src->removeSuccessor(dst);
src->addSuccessor(crit_mbb);
crit_mbb->addSuccessor(dst);
MF.push_back(crit_mbb);
const TargetMachine& TM = MF.getTarget();
const TargetInstrInfo* TII = TM.getInstrInfo();
std::vector<MachineOperand> emptyConditions;
TII->InsertBranch(*crit_mbb, dst, (MachineBasicBlock*)0,
emptyConditions);
MachineBasicBlock::iterator mii = src->end();
bool found_branch = false;
while (mii != src->begin()) {
mii--;
if (!mii->getDesc().isTerminator()) {
break;
}
for (unsigned i = 0, e = mii->getNumOperands(); i != e; ++i) {
MachineOperand & mo = mii->getOperand(i);
if (mo.isMBB() && mo.getMBB() == dst) {
found_branch = true;
mo.setMBB(crit_mbb);
}
}
}
if(!found_branch) {
TII->InsertBranch(*src, crit_mbb, (MachineBasicBlock*)0,
emptyConditions);
}
for(mii = dst->begin(); mii != dst->end(); mii++) {
if(mii->getOpcode() != TargetInstrInfo::PHI)
break;
std::vector<unsigned> toRemove;
unsigned reg = 0;
for (unsigned u = 0; u != mii->getNumOperands(); ++u)
if (mii->getOperand(u).isMBB() &&
mii->getOperand(u).getMBB() == src) {
reg = mii->getOperand(u-1).getReg();
toRemove.push_back(u-1);
}
for (std::vector<unsigned>::reverse_iterator I = toRemove.rbegin(),
E = toRemove.rend(); I != E; ++I) {
mii->RemoveOperand(*I+1);
mii->RemoveOperand(*I);
}
mii->addOperand(MachineOperand::CreateReg(reg, false));
mii->addOperand(MachineOperand::CreateMBB(crit_mbb));
}
return crit_mbb;
}
}
#endif