Patches to make the LLVM sources more -pedantic clean. Patch provided
[oota-llvm.git] / include / llvm / Assembly / PrintModulePass.h
index 08de60363f48bb62623b6ecd0033dbca3b382e31..fab59cbc1885a1f477c9401d2caa6865e8850ede 100644 (file)
@@ -1,4 +1,11 @@
-//===- llvm/Assembly/PrintModulePass.h - Printing Pass -----------*- C++ -*--=//
+//===- llvm/Assembly/PrintModulePass.h - Printing Pass ----------*- C++ -*-===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file was developed by the LLVM research group and is distributed under
+// the University of Illinois Open Source License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
 //
 // This file defines two passes to print out a module.  The PrintModulePass pass
 // simply prints out the entire module when it is executed.  The
 #define LLVM_ASSEMBLY_PRINTMODULEPASS_H
 
 #include "llvm/Pass.h"
-#include "llvm/Value.h"
+#include "llvm/Module.h"
 #include <iostream>
 
-class PrintModulePass : public Pass {
+namespace llvm {
+
+class PrintModulePass : public ModulePass {
   std::ostream *Out;      // ostream to print on
   bool DeleteStream;      // Delete the ostream in our dtor?
 public:
@@ -27,9 +36,9 @@ public:
   ~PrintModulePass() {
     if (DeleteStream) delete Out;
   }
-  
-  bool run(Module &M) {
-    (*Out) << (Value&)M;
+
+  bool runOnModule(Module &M) {
+    (*Out) << M << std::flush;
     return false;
   }
 
@@ -52,7 +61,7 @@ public:
   inline ~PrintFunctionPass() {
     if (DeleteStream) delete Out;
   }
-  
+
   // runOnFunction - This pass just prints a banner followed by the function as
   // it's processed.
   //
@@ -60,10 +69,12 @@ public:
     (*Out) << Banner << (Value&)F;
     return false;
   }
-  
+
   virtual void getAnalysisUsage(AnalysisUsage &AU) const {
     AU.setPreservesAll();
   }
 };
 
+} // End llvm namespace
+
 #endif