1 //===-- lib/MC/MCFunction.cpp -----------------------------------*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "llvm/MC/MCFunction.h"
11 #include "llvm/MC/MCAtom.h"
12 #include "llvm/MC/MCModule.h"
13 #include <algorithm>
14
15 using namespace llvm;
16
17 // MCFunction
18
MCFunction(StringRef Name,MCModule * Parent)19 MCFunction::MCFunction(StringRef Name, MCModule *Parent)
20 : Name(Name), ParentModule(Parent)
21 {}
22
~MCFunction()23 MCFunction::~MCFunction() {
24 for (iterator I = begin(), E = end(); I != E; ++I)
25 delete *I;
26 }
27
createBlock(const MCTextAtom & TA)28 MCBasicBlock &MCFunction::createBlock(const MCTextAtom &TA) {
29 MCBasicBlock *MCBB = new MCBasicBlock(TA, this);
30 Blocks.push_back(MCBB);
31 return *MCBB;
32 }
33
find(uint64_t StartAddr)34 MCBasicBlock *MCFunction::find(uint64_t StartAddr) {
35 for (const_iterator I = begin(), E = end(); I != E; ++I)
36 if ((*I)->getInsts()->getBeginAddr() == StartAddr)
37 return *I;
38 return 0;
39 }
40
find(uint64_t StartAddr) const41 const MCBasicBlock *MCFunction::find(uint64_t StartAddr) const {
42 return const_cast<MCFunction *>(this)->find(StartAddr);
43 }
44
45 // MCBasicBlock
46
MCBasicBlock(const MCTextAtom & Insts,MCFunction * Parent)47 MCBasicBlock::MCBasicBlock(const MCTextAtom &Insts, MCFunction *Parent)
48 : Insts(&Insts), Parent(Parent) {
49 getParent()->getParent()->trackBBForAtom(&Insts, this);
50 }
51
addSuccessor(const MCBasicBlock * MCBB)52 void MCBasicBlock::addSuccessor(const MCBasicBlock *MCBB) {
53 if (!isSuccessor(MCBB))
54 Successors.push_back(MCBB);
55 }
56
isSuccessor(const MCBasicBlock * MCBB) const57 bool MCBasicBlock::isSuccessor(const MCBasicBlock *MCBB) const {
58 return std::find(Successors.begin(), Successors.end(),
59 MCBB) != Successors.end();
60 }
61
addPredecessor(const MCBasicBlock * MCBB)62 void MCBasicBlock::addPredecessor(const MCBasicBlock *MCBB) {
63 if (!isPredecessor(MCBB))
64 Predecessors.push_back(MCBB);
65 }
66
isPredecessor(const MCBasicBlock * MCBB) const67 bool MCBasicBlock::isPredecessor(const MCBasicBlock *MCBB) const {
68 return std::find(Predecessors.begin(), Predecessors.end(),
69 MCBB) != Predecessors.end();
70 }
71
splitBasicBlock(MCBasicBlock * SplitBB)72 void MCBasicBlock::splitBasicBlock(MCBasicBlock *SplitBB) {
73 assert(Insts->getEndAddr() + 1 == SplitBB->Insts->getBeginAddr() &&
74 "Splitting unrelated basic blocks!");
75 SplitBB->addPredecessor(this);
76 assert(SplitBB->Successors.empty() &&
77 "Split basic block shouldn't already have successors!");
78 SplitBB->Successors = Successors;
79 Successors.clear();
80 addSuccessor(SplitBB);
81 }
82