axiom.cpp

Go to the documentation of this file.

#include "thorin/axiom.h"
 
#include "thorin/world.h"
 
namespace thorin {
 
Axiom::Axiom(NormalizeFn normalizer, u8 curry, u8 trip, const Def* type, plugin_t plugin, tag_t tag, sub_t sub)
    : Def(Node, type, Defs{}, plugin | (flags_t(tag) << 8_u64) | flags_t(sub)) {
    normalizer_ = normalizer;
    curry_      = curry;
    trip_       = trip;
}
 
std::pair<u8, u8> Axiom::infer_curry_and_trip(const Def* type) {
    u8 curry = 0;
    u8 trip  = 0;
    MutSet done;
    while (auto pi = type->isa<Pi>()) {
        if (auto mut = pi->isa_mut()) {
            if (auto [_, ins] = done.emplace(mut); !ins) {
                // infer trip
                auto curr = pi;
                do {
                    ++trip;
                    curr = curr->codom()->as<Pi>();
                } while (curr != mut);
                break;
            }
        }
 
        ++curry;
        type = pi->codom();
    }
 
    return {curry, trip};
}
 
std::tuple<const Axiom*, u8, u8> Axiom::get(const Def* def) {
    if (auto axiom = def->isa<Axiom>()) return {axiom, axiom->curry(), axiom->trip()};
    if (auto app = def->isa<App>()) return {app->axiom(), app->curry(), app->trip()};
    return {nullptr, 0, 0};
}
 
} // namespace thorin