1 files changed, 16 insertions, 3 deletions
diff --git a/include/llvm/Transforms/PrintModulePass.h b/include/llvm/Transforms/PrintModulePass.h
index 98948d8844..c8fb08f10a 100644
--- a/include/llvm/Transforms/PrintModulePass.h
+++ b/include/llvm/Transforms/PrintModulePass.h
@@ -10,14 +10,17 @@
 
 #include "llvm/Transforms/Pass.h"
 #include "llvm/Assembly/Writer.h"
+#include "llvm/Bytecode/Writer.h"
 
 class PrintModulePass : public Pass {
   string Banner;          // String to print before each method
   ostream *Out;           // ostream to print on
   bool DeleteStream;      // Delete the ostream in our dtor?
+  bool PrintAsBytecode;   // Print as bytecode rather than assembly?
 public:
-  inline PrintModulePass(const string &B, ostream *o = &cout, bool DS = false) 
-    : Banner(B), Out(o), DeleteStream(DS) {}
+  inline PrintModulePass(const string &B, ostream *o = &cout, bool DS = false,
+                         bool printAsBytecode = false)
+    : Banner(B), Out(o), DeleteStream(DS), PrintAsBytecode(printAsBytecode) {}
 
   ~PrintModulePass() {
     if (DeleteStream) delete Out;
@@ -27,7 +30,17 @@ public:
   // it's processed.
   //
   bool doPerMethodWork(Method *M) {
-    (*Out) << Banner << M;
+    if (! PrintAsBytecode)
+      (*Out) << Banner << M;
+    return false;
+  }
+  
+  // doPassFinalization - Virtual method overriden by subclasses to do any post
+  // processing needed after all passes have run.
+  //
+  bool doPassFinalization(Module *M) {
+    if (PrintAsBytecode)
+      WriteBytecodeToFile(M, *Out);
     return false;
   }
 };