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