-//===- 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 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 module as they are
+// processed.
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_ASSEMBLY_PRINTMODULEPASS_H
#define LLVM_ASSEMBLY_PRINTMODULEPASS_H
-#include "llvm/Pass.h"
-#include "llvm/Assembly/Writer.h"
-#include <iostream>
+#include <string>
-class PrintModulePass : public Pass {
- 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)
- : Out(o), DeleteStream(DS) {
- }
+namespace llvm {
+ class FunctionPass;
+ class ModulePass;
+ class raw_ostream;
- inline ~PrintModulePass() {
- if (DeleteStream) delete Out;
- }
+ /// createPrintModulePass - Create and return a pass that writes the
+ /// module to the specified raw_ostream.
+ ModulePass *createPrintModulePass(raw_ostream *OS,
+ bool DeleteStream=false,
+ const std::string &Banner = "");
- bool run(Module *M) {
- (*Out) << M;
- return false;
- }
-};
+ /// createPrintFunctionPass - Create and return a pass that prints
+ /// functions to the specified raw_ostream as they are processed.
+ FunctionPass *createPrintFunctionPass(const std::string &Banner,
+ raw_ostream *OS,
+ bool DeleteStream=false);
-class PrintMethodPass : public MethodPass {
- std::string Banner; // String to print before each method
- 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)
- : Banner(B), Out(o), DeleteStream(DS) {
- }
-
- inline ~PrintMethodPass() {
- if (DeleteStream) delete Out;
- }
-
- // runOnMethod - This pass just prints a banner followed by the method as
- // it's processed.
- //
- bool runOnMethod(Method *M) {
- (*Out) << Banner << M;
- return false;
- }
-};
+} // End llvm namespace
#endif