From 38e77214558f3a5f917896df575de1c01ee60f47 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Fri, 19 Dec 2008 18:37:59 +0000 Subject: Add support for writing LLVM IR to a specified BitstreamWriter. Patch by Lukasz Janyst! git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@61251 91177308-0d34-0410-b5e6-96231b3b80d8 --- lib/Bitcode/Writer/BitcodeWriter.cpp | 29 +++++++++++++++++------------ 1 file changed, 17 insertions(+), 12 deletions(-) (limited to 'lib/Bitcode') diff --git a/lib/Bitcode/Writer/BitcodeWriter.cpp b/lib/Bitcode/Writer/BitcodeWriter.cpp index adf49a5..b98dab7 100644 --- a/lib/Bitcode/Writer/BitcodeWriter.cpp +++ b/lib/Bitcode/Writer/BitcodeWriter.cpp @@ -1355,7 +1355,23 @@ void llvm::WriteBitcodeToFile(const Module *M, raw_ostream &Out) { BitstreamWriter Stream(Buffer); Buffer.reserve(256*1024); + + WriteBitcodeToStream( M, Stream ); + + // If writing to stdout, set binary mode. + if (&llvm::outs() == &Out) + sys::Program::ChangeStdoutToBinary(); + + // Write the generated bitstream to "Out". + Out.write((char*)&Buffer.front(), Buffer.size()); + // Make sure it hits disk now. + Out.flush(); +} + +/// WriteBitcodeToStream - Write the specified module to the specified output +/// stream. +void llvm::WriteBitcodeToStream(const Module *M, BitstreamWriter &Stream) { // If this is darwin, emit a file header and trailer if needed. bool isDarwin = M->getTargetTriple().find("-darwin") != std::string::npos; if (isDarwin) @@ -1373,16 +1389,5 @@ void llvm::WriteBitcodeToFile(const Module *M, raw_ostream &Out) { WriteModule(M, Stream); if (isDarwin) - EmitDarwinBCTrailer(Stream, Buffer.size()); - - - // If writing to stdout, set binary mode. - if (&llvm::outs() == &Out) - sys::Program::ChangeStdoutToBinary(); - - // Write the generated bitstream to "Out". - Out.write((char*)&Buffer.front(), Buffer.size()); - - // Make sure it hits disk now. - Out.flush(); + EmitDarwinBCTrailer(Stream, Stream.getBuffer().size()); } -- cgit v1.1