diff options
author | Chandler Carruth <chandlerc@gmail.com> | 2013-01-02 09:10:48 +0000 |
---|---|---|
committer | Chandler Carruth <chandlerc@gmail.com> | 2013-01-02 09:10:48 +0000 |
commit | c2c50cdcdc19a1bca993c06d13d8cdca87083ce4 (patch) | |
tree | e46b4b6d4d44228df1e870f690506dbf296954c1 /lib/IR/PrintModulePass.cpp | |
parent | 3af932322859285988a4c53551540fc0658b2116 (diff) |
Rename VMCore directory to IR.
Aside from moving the actual files, this patch only updates the build
system and the source file comments under lib/... that are relevant.
I'll be updating other docs and other files in smaller subsequnet
commits.
While I've tried to test this, but it is entirely possible that there
will still be some build system fallout.
Also, note that I've not changed the library name itself: libLLVMCore.a
is still the library name. I'd be interested in others' opinions about
whether we should rename this as well (I think we should, just not sure
what it might break)
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@171359 91177308-0d34-0410-b5e6-96231b3b80d8
Diffstat (limited to 'lib/IR/PrintModulePass.cpp')
-rw-r--r-- | lib/IR/PrintModulePass.cpp | 100 |
1 files changed, 100 insertions, 0 deletions
diff --git a/lib/IR/PrintModulePass.cpp b/lib/IR/PrintModulePass.cpp new file mode 100644 index 0000000000..422aa0917c --- /dev/null +++ b/lib/IR/PrintModulePass.cpp @@ -0,0 +1,100 @@ +//===--- IR/PrintModulePass.cpp - Module/Function Printer -----------------===// +// +// The LLVM Compiler Infrastructure +// +// This file is distributed under the University of Illinois Open Source +// License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// PrintModulePass and PrintFunctionPass implementations. +// +//===----------------------------------------------------------------------===// + +#include "llvm/Assembly/PrintModulePass.h" +#include "llvm/Function.h" +#include "llvm/Module.h" +#include "llvm/Pass.h" +#include "llvm/Support/Debug.h" +#include "llvm/Support/raw_ostream.h" +using namespace llvm; + +namespace { + + class PrintModulePass : public ModulePass { + std::string Banner; + raw_ostream *Out; // raw_ostream to print on + bool DeleteStream; // Delete the ostream in our dtor? + public: + static char ID; + PrintModulePass() : ModulePass(ID), Out(&dbgs()), + DeleteStream(false) {} + PrintModulePass(const std::string &B, raw_ostream *o, bool DS) + : ModulePass(ID), Banner(B), Out(o), DeleteStream(DS) {} + + ~PrintModulePass() { + if (DeleteStream) delete Out; + } + + bool runOnModule(Module &M) { + (*Out) << Banner << M; + return false; + } + + virtual void getAnalysisUsage(AnalysisUsage &AU) const { + AU.setPreservesAll(); + } + }; + + class PrintFunctionPass : public FunctionPass { + std::string Banner; // String to print before each function + raw_ostream *Out; // raw_ostream to print on + bool DeleteStream; // Delete the ostream in our dtor? + public: + static char ID; + PrintFunctionPass() : FunctionPass(ID), Banner(""), Out(&dbgs()), + DeleteStream(false) {} + PrintFunctionPass(const std::string &B, raw_ostream *o, bool DS) + : FunctionPass(ID), Banner(B), Out(o), DeleteStream(DS) {} + + ~PrintFunctionPass() { + if (DeleteStream) delete Out; + } + + // runOnFunction - This pass just prints a banner followed by the + // function as it's processed. + // + bool runOnFunction(Function &F) { + (*Out) << Banner << static_cast<Value&>(F); + return false; + } + + virtual void getAnalysisUsage(AnalysisUsage &AU) const { + AU.setPreservesAll(); + } + }; +} + +char PrintModulePass::ID = 0; +INITIALIZE_PASS(PrintModulePass, "print-module", + "Print module to stderr", false, false) +char PrintFunctionPass::ID = 0; +INITIALIZE_PASS(PrintFunctionPass, "print-function", + "Print function to stderr", false, false) + +/// createPrintModulePass - Create and return a pass that writes the +/// module to the specified raw_ostream. +ModulePass *llvm::createPrintModulePass(llvm::raw_ostream *OS, + bool DeleteStream, + const std::string &Banner) { + return new PrintModulePass(Banner, OS, DeleteStream); +} + +/// createPrintFunctionPass - Create and return a pass that prints +/// functions to the specified raw_ostream as they are processed. +FunctionPass *llvm::createPrintFunctionPass(const std::string &Banner, + llvm::raw_ostream *OS, + bool DeleteStream) { + return new PrintFunctionPass(Banner, OS, DeleteStream); +} + |