- class PrintModulePass : public ModulePass {
- std::string Banner;
- raw_ostream *Out; // raw_ostream to print on
- bool DeleteStream; // Delete the ostream in our dtor?
- public:
- static char ID;
- PrintModulePass() : ModulePass(ID), Out(&dbgs()),
- DeleteStream(false) {}
- PrintModulePass(const std::string &B, raw_ostream *o, bool DS)
- : ModulePass(ID), Banner(B), Out(o), DeleteStream(DS) {}
-
- ~PrintModulePass() {
- if (DeleteStream) delete Out;
- }
-
- bool runOnModule(Module &M) {
- (*Out) << Banner << M;
- return false;
- }
-
- virtual void getAnalysisUsage(AnalysisUsage &AU) const {
- AU.setPreservesAll();
- }
- };
-
- class PrintFunctionPass : public FunctionPass {
- std::string Banner; // String to print before each function
- raw_ostream *Out; // raw_ostream to print on
- bool DeleteStream; // Delete the ostream in our dtor?
- public:
- static char ID;
- PrintFunctionPass() : FunctionPass(ID), Banner(""), Out(&dbgs()),
- DeleteStream(false) {}
- PrintFunctionPass(const std::string &B, raw_ostream *o, bool DS)
- : FunctionPass(ID), Banner(B), Out(o), DeleteStream(DS) {}
-
- ~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 << static_cast<Value&>(F);
- return false;
- }
-
- virtual void getAnalysisUsage(AnalysisUsage &AU) const {
- AU.setPreservesAll();
- }
- };
-
- class PrintBasicBlockPass : public BasicBlockPass {
- std::string Banner;
- raw_ostream *Out; // raw_ostream to print on
- bool DeleteStream; // Delete the ostream in our dtor?
- public:
- static char ID;
- PrintBasicBlockPass() : BasicBlockPass(ID), Out(&dbgs()),
- DeleteStream(false) {}
- PrintBasicBlockPass(const std::string &B, raw_ostream *o, bool DS)
- : BasicBlockPass(ID), Banner(B), Out(o), DeleteStream(DS) {}
-
- ~PrintBasicBlockPass() {
- if (DeleteStream) delete Out;
- }
-
- bool runOnBasicBlock(BasicBlock &BB) {
- (*Out) << Banner << BB;
- return false;
- }
-
- virtual void getAnalysisUsage(AnalysisUsage &AU) const {
- AU.setPreservesAll();
- }
- };
+class PrintModulePassWrapper : public ModulePass {
+ PrintModulePass P;
+
+public:
+ static char ID;
+ PrintModulePassWrapper() : ModulePass(ID) {}
+ PrintModulePassWrapper(raw_ostream &OS, const std::string &Banner)
+ : ModulePass(ID), P(OS, Banner) {}
+
+ bool runOnModule(Module &M) {
+ P.run(&M);
+ return false;
+ }
+
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.setPreservesAll();
+ }
+};
+
+class PrintFunctionPassWrapper : public FunctionPass {
+ PrintFunctionPass P;
+
+public:
+ static char ID;
+ PrintFunctionPassWrapper() : FunctionPass(ID) {}
+ PrintFunctionPassWrapper(raw_ostream &OS, const std::string &Banner)
+ : FunctionPass(ID), P(OS, Banner) {}
+
+ // This pass just prints a banner followed by the function as it's processed.
+ bool runOnFunction(Function &F) {
+ P.run(&F);
+ return false;
+ }
+
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.setPreservesAll();
+ }
+};
+
+class PrintBasicBlockPass : public BasicBlockPass {
+ raw_ostream &Out;
+ std::string Banner;
+
+public:
+ static char ID;
+ PrintBasicBlockPass() : BasicBlockPass(ID), Out(dbgs()) {}
+ PrintBasicBlockPass(raw_ostream &Out, const std::string &Banner)
+ : BasicBlockPass(ID), Out(Out), Banner(Banner) {}
+
+ bool runOnBasicBlock(BasicBlock &BB) {
+ Out << Banner << BB;
+ return false;
+ }
+
+ virtual void getAnalysisUsage(AnalysisUsage &AU) const {
+ AU.setPreservesAll();
+ }
+};
+