From d1ce3b48e24b9e7d78cac590d1e4590a68c2e2ce Mon Sep 17 00:00:00 2001 From: Daniel Dunbar Date: Wed, 22 Oct 2008 17:39:14 +0000 Subject: Add raw_ostream versions of WriteBitcodeToFile and BitcodeWriterPass. - The old versions are still hanging around, but should be migrated away from. git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@57989 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Bitcode/Writer/BitcodeWriterPass.cpp | 21 +++++++++++++++++---- 1 file changed, 17 insertions(+), 4 deletions(-) (limited to 'lib/Bitcode/Writer/BitcodeWriterPass.cpp') diff --git a/lib/Bitcode/Writer/BitcodeWriterPass.cpp b/lib/Bitcode/Writer/BitcodeWriterPass.cpp index dd16fabbd7..209cf0980d 100644 --- a/lib/Bitcode/Writer/BitcodeWriterPass.cpp +++ b/lib/Bitcode/Writer/BitcodeWriterPass.cpp @@ -17,16 +17,24 @@ using namespace llvm; namespace { class WriteBitcodePass : public ModulePass { - std::ostream &Out; // ostream to print on + // FIXME: Kill off std::ostream + std::ostream *Out; + raw_ostream *RawOut; // raw_ostream to print on public: - static char ID; // Pass identifcation, replacement for typeid + static char ID; // Pass identification, replacement for typeid explicit WriteBitcodePass(std::ostream &o) - : ModulePass(&ID), Out(o) {} + : ModulePass(&ID), Out(&o), RawOut(0) {} + explicit WriteBitcodePass(raw_ostream &o) + : ModulePass(&ID), Out(0), RawOut(&o) {} const char *getPassName() const { return "Bitcode Writer"; } bool runOnModule(Module &M) { - WriteBitcodeToFile(&M, Out); + if (Out) { + WriteBitcodeToFile(&M, *Out); + } else { + WriteBitcodeToFile(&M, *RawOut); + } return false; } }; @@ -41,3 +49,8 @@ ModulePass *llvm::CreateBitcodeWriterPass(std::ostream &Str) { } +/// createBitcodeWriterPass - Create and return a pass that writes the module +/// to the specified ostream. +ModulePass *llvm::createBitcodeWriterPass(raw_ostream &Str) { + return new WriteBitcodePass(Str); +} -- cgit v1.2.3-18-g5258