lib/MC/MCFunction.cpp

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