contrib/llvm/lib/Analysis/IPA/CallPrinter.cpp

   1 //===- CallPrinter.cpp - DOT printer for call graph -----------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9 //
  10 // This file defines '-dot-callgraph', which emit a callgraph.<fnname>.dot
  11 // containing the call graph of a module.
  12 //
  13 // There is also a pass available to directly call dotty ('-view-callgraph').
  14 //
  15 //===----------------------------------------------------------------------===//
  16
  17 #include "llvm/Analysis/CallGraph.h"
  18 #include "llvm/Analysis/CallPrinter.h"
  19 #include "llvm/Analysis/DOTGraphTraitsPass.h"
  20
  21 using namespace llvm;
  22
  23 namespace llvm {
  24
  25 template<>
  26 struct DOTGraphTraits<CallGraph*> : public DefaultDOTGraphTraits {
  27   DOTGraphTraits (bool isSimple=false) : DefaultDOTGraphTraits(isSimple) {}
  28
  29   static std::string getGraphName(CallGraph *Graph) {
  30     return "Call graph";
  31   }
  32
  33   std::string getNodeLabel(CallGraphNode *Node, CallGraph *Graph) {
  34     if (Function *Func = Node->getFunction())
  35       return Func->getName();
  36
  37     return "external node";
  38   }
  39 };
  40
  41 } // end llvm namespace
  42
  43 namespace {
  44
  45 struct CallGraphViewer
  46   : public DOTGraphTraitsModuleViewer<CallGraph, true> {
  47   static char ID;
  48
  49   CallGraphViewer()
  50     : DOTGraphTraitsModuleViewer<CallGraph, true>("callgraph", ID) {
  51     initializeCallGraphViewerPass(*PassRegistry::getPassRegistry());
  52   }
  53 };
  54
  55 struct CallGraphPrinter
  56   : public DOTGraphTraitsModulePrinter<CallGraph, true> {
  57   static char ID;
  58
  59   CallGraphPrinter()
  60     : DOTGraphTraitsModulePrinter<CallGraph, true>("callgraph", ID) {
  61       initializeCallGraphPrinterPass(*PassRegistry::getPassRegistry());
  62   }
  63 };
  64
  65 } // end anonymous namespace
  66
  67 char CallGraphViewer::ID = 0;
  68 INITIALIZE_PASS(CallGraphViewer, "view-callgraph",
  69                 "View call graph",
  70                 false, false)
  71
  72 char CallGraphPrinter::ID = 0;
  73 INITIALIZE_PASS(CallGraphPrinter, "dot-callgraph",
  74                 "Print call graph to 'dot' file",
  75                 false, false)
  76
  77 // Create methods available outside of this file, to use them
  78 // "include/llvm/LinkAllPasses.h". Otherwise the pass would be deleted by
  79 // the link time optimization.
  80
  81 ModulePass *llvm::createCallGraphViewerPass() {
  82   return new CallGraphViewer();
  83 }
  84
  85 ModulePass *llvm::createCallGraphPrinterPass() {
  86   return new CallGraphPrinter();
  87 }