Skip to content

Commit

Permalink
Move more self-contained functionality away from tools/opt/opt.cpp
Browse files Browse the repository at this point in the history
BreakpointPrinter moves to its own module.


git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@201242 91177308-0d34-0410-b5e6-96231b3b80d8
  • Loading branch information
eliben committed Feb 12, 2014
1 parent 7580df3 commit a0ea8fa
Show file tree
Hide file tree
Showing 4 changed files with 110 additions and 61 deletions.
82 changes: 82 additions & 0 deletions tools/opt/BreakpointPrinter.cpp
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
//===- BreakpointPrinter.cpp - Breakpoint location printer ----------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
///
/// \file
/// \brief Breakpoint location printer.
///
//===----------------------------------------------------------------------===//
#include "BreakpointPrinter.h"
#include "llvm/ADT/StringSet.h"
#include "llvm/DebugInfo.h"
#include "llvm/IR/Module.h"
#include "llvm/Pass.h"
#include "llvm/Support/raw_ostream.h"

using namespace llvm;

namespace {

struct BreakpointPrinter : public ModulePass {
raw_ostream &Out;
static char ID;
DITypeIdentifierMap TypeIdentifierMap;

BreakpointPrinter(raw_ostream &out) : ModulePass(ID), Out(out) {}

void getContextName(DIDescriptor Context, std::string &N) {
if (Context.isNameSpace()) {
DINameSpace NS(Context);
if (!NS.getName().empty()) {
getContextName(NS.getContext(), N);
N = N + NS.getName().str() + "::";
}
} else if (Context.isType()) {
DIType TY(Context);
if (!TY.getName().empty()) {
getContextName(TY.getContext().resolve(TypeIdentifierMap), N);
N = N + TY.getName().str() + "::";
}
}
}

virtual bool runOnModule(Module &M) {
TypeIdentifierMap.clear();
NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu");
if (CU_Nodes)
TypeIdentifierMap = generateDITypeIdentifierMap(CU_Nodes);

StringSet<> Processed;
if (NamedMDNode *NMD = M.getNamedMetadata("llvm.dbg.sp"))
for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) {
std::string Name;
DISubprogram SP(NMD->getOperand(i));
assert((!SP || SP.isSubprogram()) &&
"A MDNode in llvm.dbg.sp should be null or a DISubprogram.");
if (!SP)
continue;
getContextName(SP.getContext().resolve(TypeIdentifierMap), Name);
Name = Name + SP.getDisplayName().str();
if (!Name.empty() && Processed.insert(Name)) {
Out << Name << "\n";
}
}
return false;
}

virtual void getAnalysisUsage(AnalysisUsage &AU) const {
AU.setPreservesAll();
}
};

char BreakpointPrinter::ID = 0;
}

ModulePass *llvm::createBreakpointPrinter(raw_ostream &out) {
return new BreakpointPrinter(out);
}
25 changes: 25 additions & 0 deletions tools/opt/BreakpointPrinter.h
Original file line number Diff line number Diff line change
@@ -0,0 +1,25 @@
//===- BreakpointPrinter.h - Breakpoint location printer ------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
///
/// \file
/// \brief Breakpoint location printer.
///
//===----------------------------------------------------------------------===//
#ifndef LLVM_TOOLS_OPT_BREAKPOINTPRINTER_H
#define LLVM_TOOLS_OPT_BREAKPOINTPRINTER_H

namespace llvm {

class ModulePass;
class raw_ostream;

ModulePass *createBreakpointPrinter(raw_ostream &out);
}

#endif // LLVM_TOOLS_OPT_BREAKPOINTPRINTER_H
1 change: 1 addition & 0 deletions tools/opt/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ set(LLVM_NO_DEAD_STRIP 1)

add_llvm_tool(opt
AnalysisWrappers.cpp
BreakpointPrinter.cpp
GraphPrinters.cpp
NewPMDriver.cpp
Passes.cpp
Expand Down
63 changes: 2 additions & 61 deletions tools/opt/opt.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -12,17 +12,16 @@
//
//===----------------------------------------------------------------------===//

#include "BreakpointPrinter.h"
#include "NewPMDriver.h"
#include "PassPrinters.h"
#include "llvm/ADT/StringSet.h"
#include "llvm/ADT/Triple.h"
#include "llvm/Analysis/CallGraph.h"
#include "llvm/Analysis/CallGraphSCCPass.h"
#include "llvm/Analysis/LoopPass.h"
#include "llvm/Analysis/RegionPass.h"
#include "llvm/Bitcode/BitcodeWriterPass.h"
#include "llvm/CodeGen/CommandFlags.h"
#include "llvm/DebugInfo.h"
#include "llvm/IR/DataLayout.h"
#include "llvm/IR/IRPrintingPasses.h"
#include "llvm/IR/LLVMContext.h"
Expand Down Expand Up @@ -184,65 +183,7 @@ DefaultDataLayout("default-data-layout",
cl::desc("data layout string to use if not specified by module"),
cl::value_desc("layout-string"), cl::init(""));

namespace {

struct BreakpointPrinter : public ModulePass {
raw_ostream &Out;
static char ID;
DITypeIdentifierMap TypeIdentifierMap;

BreakpointPrinter(raw_ostream &out)
: ModulePass(ID), Out(out) {
}

void getContextName(DIDescriptor Context, std::string &N) {
if (Context.isNameSpace()) {
DINameSpace NS(Context);
if (!NS.getName().empty()) {
getContextName(NS.getContext(), N);
N = N + NS.getName().str() + "::";
}
} else if (Context.isType()) {
DIType TY(Context);
if (!TY.getName().empty()) {
getContextName(TY.getContext().resolve(TypeIdentifierMap), N);
N = N + TY.getName().str() + "::";
}
}
}

virtual bool runOnModule(Module &M) {
TypeIdentifierMap.clear();
NamedMDNode *CU_Nodes = M.getNamedMetadata("llvm.dbg.cu");
if (CU_Nodes)
TypeIdentifierMap = generateDITypeIdentifierMap(CU_Nodes);

StringSet<> Processed;
if (NamedMDNode *NMD = M.getNamedMetadata("llvm.dbg.sp"))
for (unsigned i = 0, e = NMD->getNumOperands(); i != e; ++i) {
std::string Name;
DISubprogram SP(NMD->getOperand(i));
assert((!SP || SP.isSubprogram()) &&
"A MDNode in llvm.dbg.sp should be null or a DISubprogram.");
if (!SP)
continue;
getContextName(SP.getContext().resolve(TypeIdentifierMap), Name);
Name = Name + SP.getDisplayName().str();
if (!Name.empty() && Processed.insert(Name)) {
Out << Name << "\n";
}
}
return false;
}

virtual void getAnalysisUsage(AnalysisUsage &AU) const {
AU.setPreservesAll();
}
};

} // anonymous namespace

char BreakpointPrinter::ID = 0;

static inline void addPass(PassManagerBase &PM, Pass *P) {
// Add the pass to the pass manager...
Expand Down Expand Up @@ -550,7 +491,7 @@ int main(int argc, char **argv) {
return 1;
}
}
Passes.add(new BreakpointPrinter(Out->os()));
Passes.add(createBreakpointPrinter(Out->os()));
NoOutput = true;
}

Expand Down

0 comments on commit a0ea8fa

Please sign in to comment.