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 15using namespace llvm; 16 17// MCFunction 18 19MCFunction::MCFunction(StringRef Name, MCModule *Parent) 20 : Name(Name), ParentModule(Parent) 21{} 22 23MCFunction::~MCFunction() { 24 for (iterator I = begin(), E = end(); I != E; ++I) 25 delete *I; 26} 27 28MCBasicBlock &MCFunction::createBlock(const MCTextAtom &TA) { 29 MCBasicBlock *MCBB = new MCBasicBlock(TA, this); 30 Blocks.push_back(MCBB); 31 return *MCBB; 32} 33 34MCBasicBlock *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 41const MCBasicBlock *MCFunction::find(uint64_t StartAddr) const { 42 return const_cast<MCFunction *>(this)->find(StartAddr); 43} 44 45// MCBasicBlock 46 47MCBasicBlock::MCBasicBlock(const MCTextAtom &Insts, MCFunction *Parent) 48 : Insts(&Insts), Parent(Parent) { 49 getParent()->getParent()->trackBBForAtom(&Insts, this); 50} 51 52void MCBasicBlock::addSuccessor(const MCBasicBlock *MCBB) { 53 if (!isSuccessor(MCBB)) 54 Successors.push_back(MCBB); 55} 56 57bool MCBasicBlock::isSuccessor(const MCBasicBlock *MCBB) const { 58 return std::find(Successors.begin(), Successors.end(), 59 MCBB) != Successors.end(); 60} 61 62void MCBasicBlock::addPredecessor(const MCBasicBlock *MCBB) { 63 if (!isPredecessor(MCBB)) 64 Predecessors.push_back(MCBB); 65} 66 67bool MCBasicBlock::isPredecessor(const MCBasicBlock *MCBB) const { 68 return std::find(Predecessors.begin(), Predecessors.end(), 69 MCBB) != Predecessors.end(); 70} 71 72void 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