From f74825436c2a62ae16874465479a02b2859f216e Mon Sep 17 00:00:00 2001
From: Chris Lattner <sabre@nondot.org>
Date: Mon, 4 Nov 2002 02:55:30 +0000
Subject: [PATCH] Implement a new -print-callgraph analysis that turns a
 callgraph into a dot graph

git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@4524 91177308-0d34-0410-b5e6-96231b3b80d8
---
 tools/analyze/GraphPrinters.cpp | 46 ++++++++++++++++++++++++++++++++-
 tools/opt/GraphPrinters.cpp     | 46 ++++++++++++++++++++++++++++++++-
 2 files changed, 90 insertions(+), 2 deletions(-)

diff --git a/tools/analyze/GraphPrinters.cpp b/tools/analyze/GraphPrinters.cpp
index 35bff565a1f..28b18d1480e 100644
--- a/tools/analyze/GraphPrinters.cpp
+++ b/tools/analyze/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");
+};
diff --git a/tools/opt/GraphPrinters.cpp b/tools/opt/GraphPrinters.cpp
index 35bff565a1f..28b18d1480e 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");
+};
-- 
2.34.1