branch_normalize.cpp

Go to the documentation of this file.

#include "mim/phase/branch_normalize.h"
 
#include "mim/lam.h"
 
namespace mim {
 
const Def* BranchNormalize::rewrite_mut_Lam(Lam* lam) {
    if (lam->is_set()) {
        if (auto br = Branch(lam->body())) {
            auto tt = br.tt()->isa<Lam>() ? br.tt() : Lam::eta_expand(br.tt());
            auto ff = br.ff()->isa<Lam>() ? br.ff() : Lam::eta_expand(br.ff());
 
            if (tt != br.tt() || ff != br.ff()) {
                log().d("branch-normalize: tt {} → {}, ff {} → {}", br.tt(), tt, br.ff(), ff);
                lam->branch(lam->filter(), br.cond(), tt, ff, br.arg());
                // if we have this pattern nested, it will work recursively due to the rewrite of the new stuff below
            }
        }
    }
 
    return RWPhase::rewrite_mut_Lam(lam);
}
 
} // namespace mim