#define LLVM_ANALYSIS_PROFILEINFO_H
#include "llvm/BasicBlock.h"
-#include "llvm/Support/raw_ostream.h"
#include <cassert>
#include <string>
#include <map>
namespace llvm {
class Function;
class Pass;
+ class raw_ostream;
/// ProfileInfo Class - This class holds and maintains profiling
/// information for some unit of code.
// getFunction() - Returns the Function for an Edge, checking for validity.
static const Function* getFunction(Edge e) {
- assert(e.second && "Invalid ProfileInfo::Edge");
- return e.second->getParent();
+ if (e.first) {
+ return e.first->getParent();
+ } else if (e.second) {
+ return e.second->getParent();
+ }
+ assert(0 && "Invalid ProfileInfo::Edge");
+ return (const Function*)0;
}
// getEdge() - Creates an Edge from two BasicBlocks.
/// it available to the optimizers.
Pass *createProfileLoaderPass(const std::string &Filename);
- static raw_ostream& operator<<(raw_ostream &O,
- ProfileInfo::Edge E) ATTRIBUTE_USED;
- static raw_ostream& operator<<(raw_ostream &O,
- ProfileInfo::Edge E) {
- O<<"(";
- O<<(E.first?E.first->getNameStr():"0");
- O<<",";
- O<<(E.second?E.second->getNameStr():"0");
- return O<<")";
- }
+ raw_ostream& operator<<(raw_ostream &O, ProfileInfo::Edge E);
} // End llvm namespace