diff --git a/include/llvm/Bitcode/ReaderWriter.h b/include/llvm/Bitcode/ReaderWriter.h index a37cc7bf4c4..78983857ae6 100644 --- a/include/llvm/Bitcode/ReaderWriter.h +++ b/include/llvm/Bitcode/ReaderWriter.h @@ -21,6 +21,7 @@ namespace llvm { class Module; class ModuleProvider; class MemoryBuffer; + class ModulePass; /// getBitcodeModuleProvider - Read the header of the specified bitcode buffer /// and prepare for lazy deserialization of function bodies. If successful, @@ -38,6 +39,10 @@ namespace llvm { /// WriteBitcodeToFile - Write the specified module to the specified output /// stream. void WriteBitcodeToFile(const Module *M, std::ostream &Out); + + /// CreateBitcodeWriterPass - Create and return a pass that writes the module + /// to the specified ostream. + ModulePass *CreateBitcodeWriterPass(std::ostream &Str); } // End llvm namespace #endif diff --git a/lib/Bitcode/Writer/BitcodeWriterPass.cpp b/lib/Bitcode/Writer/BitcodeWriterPass.cpp new file mode 100644 index 00000000000..74123116da8 --- /dev/null +++ b/lib/Bitcode/Writer/BitcodeWriterPass.cpp @@ -0,0 +1,43 @@ +//===--- Bitcode/Writer/BitcodeWriterPass.cpp - Bitcode Writer ------------===// +// +// The LLVM Compiler Infrastructure +// +// This file was developed by Chris Lattner and is distributed under +// the University of Illinois Open Source License. See LICENSE.TXT for details. +// +//===----------------------------------------------------------------------===// +// +// BitcodeWriterPass implementation. +// +//===----------------------------------------------------------------------===// + +#include "llvm/Bitcode/ReaderWriter.h" +#include "llvm/Pass.h" +using namespace llvm; + +namespace { + class WriteBitcodePass : public ModulePass { + std::ostream *Out; // ostream to print on + public: + static char ID; // Pass identifcation, replacement for typeid + WriteBitcodePass() : ModulePass((intptr_t) &ID), Out(0) { } + WriteBitcodePass(std::ostream &o) : ModulePass((intptr_t) &ID), Out(&o) {} + + bool runOnModule(Module &M) { + if (Out) + WriteBitcodeToFile(&M, *Out); + return false; + } + }; +} + +char WriteBitcodePass::ID = 0; +static RegisterPass X("emitbitcode", "Bitcode Writer"); + +/// CreateBitcodeWriterPass - Create and return a pass that writes the module +/// to the specified ostream. +ModulePass *llvm::CreateBitcodeWriterPass(std::ostream &Str) { + return new WriteBitcodePass(Str); +} + +