pass support

1 files changed, 75 insertions, 0 deletions

diff --git a/src/pass.cpp b/src/pass.cpp
new file mode 100644
index 000000000..25e7a41e5
--- /dev/null
+++ b/src/pass.cpp
@@ -0,0 +1,75 @@
+
+#include <pass.h>
+
+namespace wasm {
+
+// PassRegistry
+
+PassRegistry* PassRegistry::get() {
+  static PassRegistry* manager = nullptr;
+  if (!manager) {
+    manager = new PassRegistry();
+  }
+  return manager;
+}
+
+void PassRegistry::registerPass(const char* name, Creator create) {
+  assert(passCreatorMap.find(name) == passCreatorMap.end());
+  passCreatorMap[name] = create;
+}
+
+Pass* PassRegistry::createPass(std::string name) {
+  Creator creator = passCreatorMap[name];
+  if (!creator) return nullptr;
+  return creator();
+}
+
+std::vector<std::string> PassRegistry::getRegisteredNames() {
+  std::vector<std::string> ret;
+  for (auto pair : passCreatorMap) {
+    ret.push_back(pair.first);
+  }
+  return ret;
+}
+
+// PassRunner
+
+void PassRunner::add(std::string passName) {
+  passes.push_back(PassRegistry::get()->createPass(passName));
+}
+
+template<class P>
+void PassRunner::add() {
+  passes.push_back(new P());
+}
+
+void PassRunner::run(Module* module) {
+  for (auto pass : passes) {
+    currPass = pass;
+    pass->run(this, module);
+  }
+}
+
+template<class P>
+P* PassRunner::getLast() {
+  bool found = false;
+  P* ret;
+  for (int i = passes.size() - 1; i >= 0; i--) {
+    if (found && (ret = dynamic_cast<P*>(passes[i]))) {
+      return ret;
+    }
+    if (passes[i] == currPass) {
+      found = true;
+    }
+  }
+  return nullptr;
+}
+
+PassRunner::~PassRunner() {
+  for (auto pass : passes) {
+    delete(pass);
+  }
+}
+
+} // namespace wasm
+