1 //===-- lib/MC/MCFunction.cpp -----------------------------------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
10 #include "llvm/MC/MCFunction.h"
11 #include "llvm/MC/MCAtom.h"
12 #include "llvm/MC/MCModule.h"
19 MCFunction::MCFunction(StringRef Name, MCModule *Parent)
20 : Name(Name), ParentModule(Parent)
23 MCFunction::~MCFunction() {
24 for (iterator I = begin(), E = end(); I != E; ++I)
28 MCBasicBlock &MCFunction::createBlock(const MCTextAtom &TA) {
29 MCBasicBlock *MCBB = new MCBasicBlock(TA, this);
30 Blocks.push_back(MCBB);
34 const MCBasicBlock *MCFunction::find(uint64_t StartAddr) const {
35 for (const_iterator I = begin(), E = end(); I != E; ++I)
36 if ((*I)->getInsts()->getBeginAddr() == StartAddr)
41 MCBasicBlock *MCFunction::find(uint64_t StartAddr) {
42 return const_cast<MCBasicBlock *>(
43 const_cast<const MCFunction *>(this)->find(StartAddr));
48 MCBasicBlock::MCBasicBlock(const MCTextAtom &Insts, MCFunction *Parent)
49 : Insts(&Insts), Parent(Parent) {
50 getParent()->getParent()->trackBBForAtom(&Insts, this);
53 void MCBasicBlock::addSuccessor(const MCBasicBlock *MCBB) {
54 if (!isSuccessor(MCBB))
55 Successors.push_back(MCBB);
58 bool MCBasicBlock::isSuccessor(const MCBasicBlock *MCBB) const {
59 return std::find(Successors.begin(), Successors.end(),
60 MCBB) != Successors.end();
63 void MCBasicBlock::addPredecessor(const MCBasicBlock *MCBB) {
64 if (!isPredecessor(MCBB))
65 Predecessors.push_back(MCBB);
68 bool MCBasicBlock::isPredecessor(const MCBasicBlock *MCBB) const {
69 return std::find(Predecessors.begin(), Predecessors.end(),
70 MCBB) != Predecessors.end();
73 void MCBasicBlock::splitBasicBlock(MCBasicBlock *SplitBB) {
74 assert(Insts->getEndAddr() + 1 == SplitBB->Insts->getBeginAddr() &&
75 "Splitting unrelated basic blocks!");
76 SplitBB->addPredecessor(this);
77 assert(SplitBB->Successors.empty() &&
78 "Split basic block shouldn't already have successors!");
79 SplitBB->Successors = Successors;
81 addSuccessor(SplitBB);