1 files changed, 110 insertions, 0 deletions
diff --git a/src/pass.h b/src/pass.h
new file mode 100644
index 000000000..638052c70
--- /dev/null
+++ b/src/pass.h
@@ -0,0 +1,110 @@
+
+#include <functional>
+
+#include "wasm.h"
+#include "mixed_arena.h"
+
+namespace wasm {
+
+class Pass;
+
+//
+// Global registry of all passes in /passes/
+//
+struct PassRegistry {
+  static PassRegistry* get();
+
+  typedef std::function<Pass* ()> Creator;
+
+  void registerPass(const char* name, Creator create);
+  Pass* createPass(std::string name);
+  std::vector<std::string> getRegisteredNames();
+
+private:
+  std::map<std::string, Creator> passCreatorMap;
+};
+
+//
+// Utility class to register a pass. See pass files for usage.
+//
+template<class P>
+struct RegisterPass {
+  RegisterPass() {
+    PassRegistry::get()->registerPass(P().name, []() {
+      return new P();
+    });
+  }
+};
+
+//
+// Runs a set of passes, in order
+//
+struct PassRunner {
+  MixedArena* allocator;
+  std::vector<Pass*> passes;
+  Pass* currPass;
+
+  PassRunner(MixedArena* allocator) : allocator(allocator) {}
+
+  void add(std::string passName);
+
+  template<class P>
+  void add();
+
+  void run(Module* module);
+
+  // Get the last pass that was already executed of a certain type.
+  template<class P>
+  P* getLast();
+
+  ~PassRunner();
+};
+
+//
+// Core pass class
+//
+struct Pass : public WasmWalker {
+  const char* name;
+
+  Pass(const char* name) : name(name) {}
+
+  // Override this to perform preparation work before the pass runs
+  virtual void prepare(PassRunner* runner, Module *module) {}
+
+  void run(PassRunner* runner, Module *module) {
+    prepare(runner, module);
+    startWalk(module);
+  }
+};
+
+// Standard passes. All passes in /passes/ are runnable from the shell,
+// but registering them here in addition allows them to communicate
+// e.g. through PassRunner::getLast
+
+// Handles names in a module, in particular adding names without duplicates
+struct NameManager : public Pass {
+  NameManager() : Pass("name-manager") {}
+
+  Name getUnique(std::string prefix);
+  // TODO: getUniqueInFunction
+
+  // visitors
+  void visitBlock(Block* curr) override;
+  void visitLoop(Loop* curr) override;
+  void visitLabel(Label* curr) override;
+  void visitBreak(Break* curr) override;
+  void visitSwitch(Switch* curr) override;
+  void visitCall(Call* curr) override;
+  void visitCallImport(CallImport* curr) override;
+  void visitFunctionType(FunctionType* curr) override;
+  void visitFunction(Function* curr) override;
+  void visitImport(Import* curr) override;
+  void visitExport(Export* curr) override;
+
+private:
+  std::set<Name> names;
+  size_t counter = 0;
+};
+
+} // namespace wasm
+