Merge pull request #200 from WebAssembly/print-pass

Move printing to a pass

1 files changed, 24 insertions, 2 deletions

diff --git a/src/pass.h b/src/pass.h
index a95be9850..53b486e41 100644
--- a/src/pass.h
+++ b/src/pass.h
@@ -71,10 +71,21 @@ struct PassRunner {
 
   PassRunner(MixedArena* allocator) : allocator(allocator) {}
 
-  void add(std::string passName);
+  void add(std::string passName) {
+    auto pass = PassRegistry::get()->createPass(passName);
+    assert(pass);
+    passes.push_back(pass);
+  }
 
   template<class P>
-  void add();
+  void add() {
+    passes.push_back(new P());
+  }
+
+  template<class P, class Arg>
+  void add(Arg& arg){
+    passes.push_back(new P(arg));
+  }
 
   void run(Module* module);
 
@@ -143,6 +154,17 @@ private:
   size_t counter = 0;
 };
 
+// Prints out a module
+class Printer : public Pass {
+  std::ostream& o;
+
+public:
+  Printer() : o(std::cout) {}
+  Printer(std::ostream& o) : o(o) {}
+
+  void run(PassRunner* runner, Module* module) override;
+};
+
 } // namespace wasm
 
 #endif // wasm_pass_h