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 MCBasicBlock &MCFunction::createBlock(const MCTextAtom &TA) {
24 std::unique_ptr<MCBasicBlock> MCBB(new MCBasicBlock(TA, this));
25 Blocks.push_back(std::move(MCBB));
26 return *Blocks.back();
29 MCBasicBlock *MCFunction::find(uint64_t StartAddr) {
30 for (const_iterator I = begin(), E = end(); I != E; ++I)
31 if ((*I)->getInsts()->getBeginAddr() == StartAddr)
36 const MCBasicBlock *MCFunction::find(uint64_t StartAddr) const {
37 return const_cast<MCFunction *>(this)->find(StartAddr);
42 MCBasicBlock::MCBasicBlock(const MCTextAtom &Insts, MCFunction *Parent)
43 : Insts(&Insts), Parent(Parent) {
44 getParent()->getParent()->trackBBForAtom(&Insts, this);
47 void MCBasicBlock::addSuccessor(const MCBasicBlock *MCBB) {
48 if (!isSuccessor(MCBB))
49 Successors.push_back(MCBB);
52 bool MCBasicBlock::isSuccessor(const MCBasicBlock *MCBB) const {
53 return std::find(Successors.begin(), Successors.end(),
54 MCBB) != Successors.end();
57 void MCBasicBlock::addPredecessor(const MCBasicBlock *MCBB) {
58 if (!isPredecessor(MCBB))
59 Predecessors.push_back(MCBB);
62 bool MCBasicBlock::isPredecessor(const MCBasicBlock *MCBB) const {
63 return std::find(Predecessors.begin(), Predecessors.end(),
64 MCBB) != Predecessors.end();
67 void MCBasicBlock::splitBasicBlock(MCBasicBlock *SplitBB) {
68 assert(Insts->getEndAddr() + 1 == SplitBB->Insts->getBeginAddr() &&
69 "Splitting unrelated basic blocks!");
70 SplitBB->addPredecessor(this);
71 assert(SplitBB->Successors.empty() &&
72 "Split basic block shouldn't already have successors!");
73 SplitBB->Successors = Successors;
75 addSuccessor(SplitBB);