ProfileInfo interface tweaks.

- Add getExecutionCount(const Function).

 - Add helper Edge type.

 - constify.

 - No functionality change.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@75623 91177308-0d34-0410-b5e6-96231b3b80d8
This commit is contained in:
Daniel Dunbar
2009-07-14 06:58:59 +00:00
parent e11c4db848
commit 858cb8a5e0
2 changed files with 22 additions and 10 deletions

View File

@@ -26,17 +26,22 @@
namespace llvm {
class BasicBlock;
class Function;
class Pass;
/// ProfileInfo Class - This class holds and maintains edge profiling
/// information for some unit of code.
class ProfileInfo {
public:
// Types for handling profiling information.
typedef std::pair<const BasicBlock*, const BasicBlock*> Edge;
protected:
// EdgeCounts - Count the number of times a transition between two blocks is
// executed. As a special case, we also hold an edge from the null
// BasicBlock to the entry block to indicate how many times the function was
// entered.
std::map<std::pair<BasicBlock*, BasicBlock*>, unsigned> EdgeCounts;
std::map<Edge, unsigned> EdgeCounts;
public:
static char ID; // Class identification, replacement for typeinfo
virtual ~ProfileInfo(); // We want to be subclassed
@@ -44,10 +49,13 @@ namespace llvm {
//===------------------------------------------------------------------===//
/// Profile Information Queries
///
unsigned getExecutionCount(BasicBlock *BB) const;
unsigned getExecutionCount(const Function *F) const;
unsigned getEdgeWeight(BasicBlock *Src, BasicBlock *Dest) const {
std::map<std::pair<BasicBlock*, BasicBlock*>, unsigned>::const_iterator I=
unsigned getExecutionCount(const BasicBlock *BB) const;
unsigned getEdgeWeight(const BasicBlock *Src,
const BasicBlock *Dest) const {
std::map<Edge, unsigned>::const_iterator I =
EdgeCounts.find(std::make_pair(Src, Dest));
return I != EdgeCounts.end() ? I->second : 0;
}