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/MCAnalysis/MCFunction.h"
11 #include "llvm/MC/MCAnalysis/MCAtom.h"
12 #include "llvm/MC/MCAnalysis/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
createBlock(const MCTextAtom & TA)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();
27 }
28
find(uint64_t StartAddr)29 MCBasicBlock *MCFunction::find(uint64_t StartAddr) {
30 for (const_iterator I = begin(), E = end(); I != E; ++I)
31 if ((*I)->getInsts()->getBeginAddr() == StartAddr)
32 return I->get();
33 return nullptr;
34 }
35
find(uint64_t StartAddr) const36 const MCBasicBlock *MCFunction::find(uint64_t StartAddr) const {
37 return const_cast<MCFunction *>(this)->find(StartAddr);
38 }
39
40 // MCBasicBlock
41
MCBasicBlock(const MCTextAtom & Insts,MCFunction * Parent)42 MCBasicBlock::MCBasicBlock(const MCTextAtom &Insts, MCFunction *Parent)
43 : Insts(&Insts), Parent(Parent) {
44 getParent()->getParent()->trackBBForAtom(&Insts, this);
45 }
46
addSuccessor(const MCBasicBlock * MCBB)47 void MCBasicBlock::addSuccessor(const MCBasicBlock *MCBB) {
48 if (!isSuccessor(MCBB))
49 Successors.push_back(MCBB);
50 }
51
isSuccessor(const MCBasicBlock * MCBB) const52 bool MCBasicBlock::isSuccessor(const MCBasicBlock *MCBB) const {
53 return std::find(Successors.begin(), Successors.end(),
54 MCBB) != Successors.end();
55 }
56
addPredecessor(const MCBasicBlock * MCBB)57 void MCBasicBlock::addPredecessor(const MCBasicBlock *MCBB) {
58 if (!isPredecessor(MCBB))
59 Predecessors.push_back(MCBB);
60 }
61
isPredecessor(const MCBasicBlock * MCBB) const62 bool MCBasicBlock::isPredecessor(const MCBasicBlock *MCBB) const {
63 return std::find(Predecessors.begin(), Predecessors.end(),
64 MCBB) != Predecessors.end();
65 }
66
splitBasicBlock(MCBasicBlock * SplitBB)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;
74 Successors.clear();
75 addSuccessor(SplitBB);
76 }
77