include/llvm/Analysis/ProfileInfoLoader.h

   1 //===- ProfileInfoLoader.h - Load & convert profile information -*- C++ -*-===//
   2 //
   3 //                      The LLVM Compiler Infrastructure
   4 //
   5 // This file was developed by the LLVM research group and is distributed under
   6 // the University of Illinois Open Source License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // The ProfileInfoLoader class is used to load and represent profiling
  11 // information read in from the dump file.  If conversions between formats are
  12 // needed, it can also do this.
  13 //
  14 //===----------------------------------------------------------------------===//
  15
  16 #ifndef LLVM_ANALYSIS_PROFILEINFOLOADER_H
  17 #define LLVM_ANALYSIS_PROFILEINFOLOADER_H
  18
  19 #include <vector>
  20 #include <string>
  21 #include <utility>
  22
  23 namespace llvm {
  24
  25 class Module;
  26 class Function;
  27 class BasicBlock;
  28
  29 class ProfileInfoLoader {
  30   Module &M;
  31   std::vector<std::string> CommandLines;
  32   std::vector<unsigned>    FunctionCounts;
  33   std::vector<unsigned>    BlockCounts;
  34   std::vector<unsigned>    EdgeCounts;
  35 public:
  36   // ProfileInfoLoader ctor - Read the specified profiling data file, exiting
  37   // the program if the file is invalid or broken.
  38   ProfileInfoLoader(const char *ToolName, const std::string &Filename,
  39                     Module &M);
  40
  41   unsigned getNumExecutions() const { return CommandLines.size(); }
  42   const std::string &getExecution(unsigned i) const { return CommandLines[i]; }
  43
  44   // getFunctionCounts - This method is used by consumers of function counting
  45   // information.  If we do not directly have function count information, we
  46   // compute it from other, more refined, types of profile information.
  47   //
  48   void getFunctionCounts(std::vector<std::pair<Function*, unsigned> > &Counts);
  49
  50   // hasAccurateBlockCounts - Return true if we can synthesize accurate block
  51   // frequency information from whatever we have.
  52   //
  53   bool hasAccurateBlockCounts() const {
  54     return !BlockCounts.empty() || !EdgeCounts.empty();
  55   }
  56
  57   // hasAccurateEdgeCounts - Return true if we can synthesize accurate edge
  58   // frequency information from whatever we have.
  59   //
  60   bool hasAccurateEdgeCounts() const {
  61     return !EdgeCounts.empty();
  62   }
  63
  64   // getBlockCounts - This method is used by consumers of block counting
  65   // information.  If we do not directly have block count information, we
  66   // compute it from other, more refined, types of profile information.
  67   //
  68   void getBlockCounts(std::vector<std::pair<BasicBlock*, unsigned> > &Counts);
  69
  70   // getEdgeCounts - This method is used by consumers of edge counting
  71   // information.  If we do not directly have edge count information, we compute
  72   // it from other, more refined, types of profile information.
  73   //
  74   // Edges are represented as a pair, where the first element is the basic block
  75   // and the second element is the successor number.
  76   //
  77   typedef std::pair<BasicBlock*, unsigned> Edge;
  78   void getEdgeCounts(std::vector<std::pair<Edge, unsigned> > &Counts);
  79 };
  80
  81 } // End llvm namespace
  82
  83 #endif