lib/Analysis/ProfileInfo.cpp

   1 //===- ProfileInfo.cpp - Profile Info Interface ---------------------------===//
   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 // This file implements the abstract ProfileInfo interface, and the default
  11 // "no profile" implementation.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "llvm/Analysis/Passes.h"
  16 #include "llvm/Analysis/ProfileInfo.h"
  17 #include "llvm/Pass.h"
  18 #include "llvm/Support/CFG.h"
  19 #include "llvm/Support/Compiler.h"
  20 #include <set>
  21 using namespace llvm;
  22
  23 // Register the ProfileInfo interface, providing a nice name to refer to.
  24 static RegisterAnalysisGroup<ProfileInfo> Z("Profile Information");
  25 char ProfileInfo::ID = 0;
  26
  27 ProfileInfo::~ProfileInfo() {}
  28
  29 double ProfileInfo::getExecutionCount(const BasicBlock *BB) {
  30   std::map<const Function*, BlockCounts>::iterator J =
  31     BlockInformation.find(BB->getParent());
  32   if (J != BlockInformation.end()) {
  33     BlockCounts::iterator I = J->second.find(BB);
  34     if (I != J->second.end())
  35       return I->second;
  36   }
  37
  38   pred_const_iterator PI = pred_begin(BB), PE = pred_end(BB);
  39
  40   // Are there zero predecessors of this block?
  41   if (PI == PE) {
  42     // If this is the entry block, look for the Null -> Entry edge.
  43     if (BB == &BB->getParent()->getEntryBlock())
  44       return getEdgeWeight(getEdge(0, BB));
  45     else
  46       return 0;   // Otherwise, this is a dead block.
  47   }
  48
  49   // Otherwise, if there are predecessors, the execution count of this block is
  50   // the sum of the edge frequencies from the incoming edges.
  51   std::set<const BasicBlock*> ProcessedPreds;
  52   double Count = 0;
  53   for (; PI != PE; ++PI)
  54     if (ProcessedPreds.insert(*PI).second) {
  55       double w = getEdgeWeight(getEdge(*PI, BB));
  56       if (w == MissingValue) {
  57         Count = MissingValue; break;
  58       }
  59       Count += w;
  60     }
  61
  62   BlockInformation[BB->getParent()][BB] = Count;
  63   return Count;
  64 }
  65
  66 double ProfileInfo::getExecutionCount(const Function *F) {
  67   if (F->isDeclaration()) return MissingValue;
  68   std::map<const Function*, double>::iterator J =
  69     FunctionInformation.find(F);
  70   if (J != FunctionInformation.end())
  71     return J->second;
  72
  73   double Count = getExecutionCount(&F->getEntryBlock());
  74   FunctionInformation[F] = Count;
  75   return Count;
  76 }
  77
  78
  79 //===----------------------------------------------------------------------===//
  80 //  NoProfile ProfileInfo implementation
  81 //
  82
  83 namespace {
  84   struct VISIBILITY_HIDDEN NoProfileInfo
  85     : public ImmutablePass, public ProfileInfo {
  86     static char ID; // Class identification, replacement for typeinfo
  87     NoProfileInfo() : ImmutablePass(&ID) {}
  88   };
  89 }  // End of anonymous namespace
  90
  91 char NoProfileInfo::ID = 0;
  92 // Register this pass...
  93 static RegisterPass<NoProfileInfo>
  94 X("no-profile", "No Profile Information", false, true);
  95
  96 // Declare that we implement the ProfileInfo interface
  97 static RegisterAnalysisGroup<ProfileInfo, true> Y(X);
  98
  99 ImmutablePass *llvm::createNoProfileInfoPass() { return new NoProfileInfo(); }