summaryrefslogtreecommitdiff
path: root/tools/opt/GraphPrinters.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tools/opt/GraphPrinters.cpp')
-rw-r--r--tools/opt/GraphPrinters.cpp46
1 files changed, 45 insertions, 1 deletions
diff --git a/tools/opt/GraphPrinters.cpp b/tools/opt/GraphPrinters.cpp
index 35bff565a1..28b18d1480 100644
--- a/tools/opt/GraphPrinters.cpp
+++ b/tools/opt/GraphPrinters.cpp
@@ -10,10 +10,15 @@
#include "Support/GraphWriter.h"
#include "llvm/Pass.h"
#include "llvm/iTerminators.h"
+#include "llvm/Analysis/CallGraph.h"
#include "llvm/Support/CFG.h"
#include <sstream>
#include <fstream>
+//===----------------------------------------------------------------------===//
+// Control Flow Graph Printer
+//===----------------------------------------------------------------------===//
+
template<>
struct DOTGraphTraits<Function*> : public DefaultDOTGraphTraits {
static std::string getGraphName(Function *F) {
@@ -70,7 +75,6 @@ static void WriteGraphToFile(std::ostream &O, const std::string &GraphName,
namespace {
struct CFGPrinter : public FunctionPass {
- Function *F;
virtual bool runOnFunction(Function &Func) {
WriteGraphToFile(std::cerr, "cfg."+Func.getName(), &Func);
return false;
@@ -86,3 +90,43 @@ namespace {
RegisterAnalysis<CFGPrinter> P1("print-cfg",
"Print CFG of function to 'dot' file");
};
+
+
+
+//===----------------------------------------------------------------------===//
+// Call Graph Printer
+//===----------------------------------------------------------------------===//
+
+template<>
+struct DOTGraphTraits<CallGraph*> : public DefaultDOTGraphTraits {
+ static std::string getGraphName(CallGraph *F) {
+ return "Call Graph";
+ }
+
+ static std::string getNodeLabel(CallGraphNode *Node, CallGraph *Graph) {
+ if (Node->getFunction())
+ return Node->getFunction()->getName();
+ else
+ return "Indirect call node";
+ }
+};
+
+
+namespace {
+ struct CallGraphPrinter : public Pass {
+ virtual bool run(Module &M) {
+ WriteGraphToFile(std::cerr, "callgraph", &getAnalysis<CallGraph>());
+ return false;
+ }
+
+ void print(std::ostream &OS) const {}
+
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.addRequired<CallGraph>();
+ AU.setPreservesAll();
+ }
+ };
+
+ RegisterAnalysis<CallGraphPrinter> P2("print-callgraph",
+ "Print Call Graph to 'dot' file");
+};