PassManager.hpp

#pragma once

#include "Module.hpp"

#include <memory>
#include <vector>

class Pass {
  public:
    Pass(Module *m) : m_(m) {}
    virtual ~Pass() = default;
    virtual void run() = 0;

  protected:
    Module *m_;
};

class PassManager {
  public:
    PassManager(Module *m) : m_(m) {}

    template <typename PassType, typename... Args>
    void add_pass(Args &&...args) {
        passes_.emplace_back(new PassType(m_, std::forward<Args>(args)...));
    }

    void run() {
        for (auto &pass : passes_) {
            pass->run();
        }
    }

  private:
    std::vector<std::unique_ptr<Pass>> passes_;
    Module *m_;
};