cps2ds.h

Go to the documentation of this file.

#pragma once
 
#include <vector>
 
#include <mim/def.h>
#include <mim/pass/pass.h>
 
namespace mim::plug::direct {
 
/// This is the second part of ds2cps.
/// We replace all ds call sites of cps (or ds converted) functions with the cps calls.
/// `b = f args` becomes `f (args,cont)` with a newly introduced continuation `cont: .Cn b`.
class CPS2DS : public RWPass<CPS2DS, Lam> {
public:
    CPS2DS(PassMan& man)
        : RWPass(man, "cps2ds") {}
 
    void enter() override;
 
private:
    Def2Def rewritten_;
    DefSet rewritten_lams;
    std::vector<Lam*> lam_stack;
    Lam* curr_lam_ = nullptr;
 
    void rewrite_lam(Lam* lam);
    const Def* rewrite_body(const Def*);
    const Def* rewrite_body_(const Def*);
};
 
} // namespace mim::plug::direct