mem.cpp

Go to the documentation of this file.

#include "mim/plug/mem/mem.h"
 
#include <mim/config.h>
#include <mim/phase.h>
 
#include "mim/plug/mem/phase/add_mem.h"
#include "mim/plug/mem/phase/free_locals.h"
#include "mim/plug/mem/phase/heap2stack.h"
#include "mim/plug/mem/phase/seo.h"
 
using namespace mim;
using namespace mim::plug;
 
static void reg_phases(Flags2Phases& phases) {
    MIM_REPL(phases, mem::alloc2malloc_repl, {
        if (auto alloc = Axm::isa<mem::alloc>(def)) {
            auto [pointee, addr_space] = alloc->decurry()->args<2>();
            return mem::op_malloc(pointee, addr_space, alloc->arg());
        } else if (auto slot = Axm::isa<mem::slot>(def)) {
            auto [Ta, mr]              = slot->uncurry_args<2>();
            auto [pointee, addr_space] = Ta->projs<2>();
            auto [mem, ret]            = mr->projs<2>();
            return mem::op_mslot(pointee, addr_space, mem, ret);
        }
        return {};
    });
 
    // clang-format off
    Phase::hook<mem::add_mem,     mem::phase::AddMem    >(phases);
    Phase::hook<mem::free_locals, mem::phase::FreeLocals>(phases);
    Phase::hook<mem::heap2stack,  mem::phase::Heap2Stack>(phases);
    Phase::hook<mem::seo,         mem::phase::SEO       >(phases);
    // clang-format on
}
 
MIM_PLUGIN_ENTRY(mem) {
    plugin.register_normalizers = mem::register_normalizers;
    plugin.register_phases      = reg_phases;
}