mirror of
https://github.com/RPCSX/llvm.git
synced 2024-12-11 05:35:11 +00:00
562873f9be
There is another pass by the generic name 'CallGraphPrinter' which is actually just a call graph printer tucked away inside the opt tool. I'd like to bring it out and make it follow the same patterns as the rest of the CallGraph code, but doing so would end up conflicting with the name of the DOT printing pass. So this makes the DOT printing pass name be more precise. No functionality changed here. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@263100 91177308-0d34-0410-b5e6-96231b3b80d8
93 lines
3.0 KiB
C++
93 lines
3.0 KiB
C++
//===- CallPrinter.cpp - DOT printer for call graph -----------------------===//
|
|
//
|
|
// The LLVM Compiler Infrastructure
|
|
//
|
|
// This file is distributed under the University of Illinois Open Source
|
|
// License. See LICENSE.TXT for details.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file defines '-dot-callgraph', which emit a callgraph.<fnname>.dot
|
|
// containing the call graph of a module.
|
|
//
|
|
// There is also a pass available to directly call dotty ('-view-callgraph').
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Analysis/CallGraph.h"
|
|
#include "llvm/Analysis/CallPrinter.h"
|
|
#include "llvm/Analysis/DOTGraphTraitsPass.h"
|
|
|
|
using namespace llvm;
|
|
|
|
namespace llvm {
|
|
|
|
template <> struct DOTGraphTraits<CallGraph *> : public DefaultDOTGraphTraits {
|
|
DOTGraphTraits(bool isSimple = false) : DefaultDOTGraphTraits(isSimple) {}
|
|
|
|
static std::string getGraphName(CallGraph *Graph) { return "Call graph"; }
|
|
|
|
std::string getNodeLabel(CallGraphNode *Node, CallGraph *Graph) {
|
|
if (Function *Func = Node->getFunction())
|
|
return Func->getName();
|
|
|
|
return "external node";
|
|
}
|
|
};
|
|
|
|
struct AnalysisCallGraphWrapperPassTraits {
|
|
static CallGraph *getGraph(CallGraphWrapperPass *P) {
|
|
return &P->getCallGraph();
|
|
}
|
|
};
|
|
|
|
} // end llvm namespace
|
|
|
|
namespace {
|
|
|
|
struct CallGraphViewer
|
|
: public DOTGraphTraitsModuleViewer<CallGraphWrapperPass, true, CallGraph *,
|
|
AnalysisCallGraphWrapperPassTraits> {
|
|
static char ID;
|
|
|
|
CallGraphViewer()
|
|
: DOTGraphTraitsModuleViewer<CallGraphWrapperPass, true, CallGraph *,
|
|
AnalysisCallGraphWrapperPassTraits>(
|
|
"callgraph", ID) {
|
|
initializeCallGraphViewerPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
};
|
|
|
|
struct CallGraphDOTPrinter : public DOTGraphTraitsModulePrinter<
|
|
CallGraphWrapperPass, true, CallGraph *,
|
|
AnalysisCallGraphWrapperPassTraits> {
|
|
static char ID;
|
|
|
|
CallGraphDOTPrinter()
|
|
: DOTGraphTraitsModulePrinter<CallGraphWrapperPass, true, CallGraph *,
|
|
AnalysisCallGraphWrapperPassTraits>(
|
|
"callgraph", ID) {
|
|
initializeCallGraphDOTPrinterPass(*PassRegistry::getPassRegistry());
|
|
}
|
|
};
|
|
|
|
} // end anonymous namespace
|
|
|
|
char CallGraphViewer::ID = 0;
|
|
INITIALIZE_PASS(CallGraphViewer, "view-callgraph", "View call graph", false,
|
|
false)
|
|
|
|
char CallGraphDOTPrinter::ID = 0;
|
|
INITIALIZE_PASS(CallGraphDOTPrinter, "dot-callgraph",
|
|
"Print call graph to 'dot' file", false, false)
|
|
|
|
// Create methods available outside of this file, to use them
|
|
// "include/llvm/LinkAllPasses.h". Otherwise the pass would be deleted by
|
|
// the link time optimization.
|
|
|
|
ModulePass *llvm::createCallGraphViewerPass() { return new CallGraphViewer(); }
|
|
|
|
ModulePass *llvm::createCallGraphDOTPrinterPass() {
|
|
return new CallGraphDOTPrinter();
|
|
}
|