Patches to make the LLVM sources more -pedantic clean. Patch provided

[oota-llvm.git] / include / llvm / Assembly / PrintModulePass.h

diff --git a/include/llvm/Assembly/PrintModulePass.h b/include/llvm/Assembly/PrintModulePass.h
index 181ce98227e99d800668f63eba1f6d39b160b097..fab59cbc1885a1f477c9401d2caa6865e8850ede 100644 (file)
--- a/include/llvm/Assembly/PrintModulePass.h
+++ b/include/llvm/Assembly/PrintModulePass.h
@@ -1,9 +1,17 @@
-//===- llvm/Assembly/PrintModulePass.h - Printing Pass -----------*- C++ -*--=//
+//===- llvm/Assembly/PrintModulePass.h - Printing Pass ----------*- C++ -*-===//
 //
-// This file defines two passes to print out a module.  The PrintModulePass
-// pass simply prints out the entire module when it is executed.  The
-// PrintMethodPass class is designed to be pipelined with other MethodPass's,
-// and prints out the methods of the class as they are processed.
+//                     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
+// PrintFunctionPass class is designed to be pipelined with other
+// FunctionPass's, and prints out the functions of the class as they are
+// processed.
 //
 //===----------------------------------------------------------------------===//
 
@@ -11,48 +19,62 @@
 #define LLVM_ASSEMBLY_PRINTMODULEPASS_H
 
 #include "llvm/Pass.h"
-#include "llvm/Assembly/Writer.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:
-  inline PrintModulePass(std::ostream *o = &std::cout, bool DS = false)
+  PrintModulePass() : Out(&std::cerr), DeleteStream(false) {}
+  PrintModulePass(std::ostream *o, bool DS = false)
     : Out(o), DeleteStream(DS) {
   }
-  
-  inline ~PrintModulePass() {
+
+  ~PrintModulePass() {
     if (DeleteStream) delete Out;
   }
-  
-  bool run(Module *M) {
-    (*Out) << M;
+
+  bool runOnModule(Module &M) {
+    (*Out) << M << std::flush;
     return false;
   }
+
+  virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+    AU.setPreservesAll();
+  }
 };
 
-class PrintMethodPass : public MethodPass {
-  std::string Banner;     // String to print before each method
+class PrintFunctionPass : public FunctionPass {
+  std::string Banner;     // String to print before each function
   std::ostream *Out;      // ostream to print on
   bool DeleteStream;      // Delete the ostream in our dtor?
 public:
-  inline PrintMethodPass(const std::string &B, std::ostream *o = &std::cout,
-                         bool DS = false)
+  PrintFunctionPass() : Banner(""), Out(&std::cerr), DeleteStream(false) {}
+  PrintFunctionPass(const std::string &B, std::ostream *o = &std::cout,
+                    bool DS = false)
     : Banner(B), Out(o), DeleteStream(DS) {
   }
-  
-  inline ~PrintMethodPass() {
+
+  inline ~PrintFunctionPass() {
     if (DeleteStream) delete Out;
   }
-  
-  // runOnMethod - This pass just prints a banner followed by the method as
+
+  // runOnFunction - This pass just prints a banner followed by the function as
   // it's processed.
   //
-  bool runOnMethod(Method *M) {
-    (*Out) << Banner << M;
+  bool runOnFunction(Function &F) {
+    (*Out) << Banner << (Value&)F;
     return false;
   }
+
+  virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+    AU.setPreservesAll();
+  }
 };
 
+} // End llvm namespace
+
 #endif