Add the ability to print out llvm code when the -annotated-llvm option is given
[oota-llvm.git] / tools / llvm-prof / ProfileInfo.h
1 //===- ProfileInfo.h - Represents 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 ProfileInfo class is used to represent profiling information read in from
11 // the dump file.
12 //
13 //===----------------------------------------------------------------------===//
14
15 #ifndef PROFILEINFO_H
16 #define PROFILEINFO_H
17
18 #include <vector>
19 #include <string>
20 #include <utility>
21 class Module;
22 class Function;
23 class BasicBlock;
24
25 class ProfileInfo {
26   Module &M;
27   std::vector<std::string> CommandLines;
28   std::vector<unsigned>    FunctionCounts;
29   std::vector<unsigned>    BlockCounts;
30 public:
31   // ProfileInfo ctor - Read the specified profiling data file, exiting the
32   // program if the file is invalid or broken.
33   ProfileInfo(const char *ToolName, const std::string &Filename, Module &M);
34
35   unsigned getNumExecutions() const { return CommandLines.size(); }
36   const std::string &getExecution(unsigned i) const { return CommandLines[i]; }
37
38   // getFunctionCounts - This method is used by consumers of function counting
39   // information.  If we do not directly have function count information, we
40   // compute it from other, more refined, types of profile information.
41   //
42   void getFunctionCounts(std::vector<std::pair<Function*, unsigned> > &Counts);
43
44   // hasAccurateBlockCounts - Return true if we can synthesize accurate block
45   // frequency information from whatever we have.
46   //
47   bool hasAccurateBlockCounts() const {
48     return !BlockCounts.empty();
49   }
50
51   // getBlockCounts - This method is used by consumers of block counting
52   // information.  If we do not directly have block count information, we
53   // compute it from other, more refined, types of profile information.
54   //
55   void getBlockCounts(std::vector<std::pair<BasicBlock*, unsigned> > &Counts);
56 };
57
58 #endif