-//===- 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 <iostream>
+#include "llvm/Module.h"
+#include "llvm/Support/Streams.h"
+
+namespace llvm {
-class PrintModulePass : public Pass {
- std::ostream *Out; // ostream to print on
+class PrintModulePass : public ModulePass {
+ OStream *Out; // ostream to print on
bool DeleteStream; // Delete the ostream in our dtor?
public:
- PrintModulePass() : Out(&std::cerr), DeleteStream(false) {}
- PrintModulePass(std::ostream *o, bool DS = false)
- : Out(o), DeleteStream(DS) {
- }
+ static char ID;
+ PrintModulePass() : ModulePass((intptr_t)&ID), Out(&cerr), DeleteStream(false) {}
+ PrintModulePass(OStream *o, bool DS = false)
+ : ModulePass((intptr_t)&ID), Out(o), DeleteStream(DS) {}
~PrintModulePass() {
if (DeleteStream) delete Out;
}
-
- bool run(Module &M) {
- (*Out) << (Value&)M;
+
+ bool runOnModule(Module &M) {
+ (*Out) << M << std::flush;
return false;
}
class PrintFunctionPass : public FunctionPass {
std::string Banner; // String to print before each function
- std::ostream *Out; // ostream to print on
+ OStream *Out; // ostream to print on
bool DeleteStream; // Delete the ostream in our dtor?
public:
- PrintFunctionPass() : Banner(""), Out(&std::cerr), DeleteStream(false) {}
- PrintFunctionPass(const std::string &B, std::ostream *o = &std::cout,
+ static char ID;
+ PrintFunctionPass() : FunctionPass((intptr_t)&ID), Banner(""), Out(&cerr),
+ DeleteStream(false) {}
+ PrintFunctionPass(const std::string &B, OStream *o = &cout,
bool DS = false)
- : Banner(B), Out(o), DeleteStream(DS) {
- }
+ : FunctionPass((intptr_t)&ID), Banner(B), Out(o), DeleteStream(DS) {}
inline ~PrintFunctionPass() {
if (DeleteStream) delete Out;
}
-
+
// runOnFunction - This pass just prints a banner followed by the function as
// it's processed.
//
bool runOnFunction(Function &F) {
- (*Out) << Banner << (Value&)F;
+ (*Out) << Banner << static_cast<Value&>(F);
return false;
}
-
+
virtual void getAnalysisUsage(AnalysisUsage &AU) const {
AU.setPreservesAll();
}
};
+} // End llvm namespace
+
#endif