tok.cpp

Go to the documentation of this file.

#include "thorin/fe/tok.h"
 
#include <fe/assert.h>
 
#include "thorin/lam.h"
#include "thorin/tuple.h"
 
namespace thorin {
 
std::string_view Tok::tag2str(Tok::Tag tag) {
    switch (tag) {
#define CODE(t, str) \
    case Tok::Tag::t: return str;
        THORIN_KEY(CODE)
        THORIN_TOK(CODE)
#undef CODE
        case Tag::Nil: fe::unreachable();
    }
    fe::unreachable();
}
 
/// @name std::ostream operator
///@{
std::ostream& operator<<(std::ostream& os, Tok tok) {
    if (tok.isa(Tok::Tag::M_anx) || tok.isa(Tok::Tag::M_id) || tok.isa(Tok::Tag::M_str)) return os << tok.sym();
    return os << Tok::tag2str(tok.tag());
}
///@}
 
// clang-format off
Tok::Prec Tok::prec(const Def* def) {
    if (def->isa<Pi     >()) return Tok::Prec::Arrow;
    if (def->isa<App    >()) return Tok::Prec::App;
    if (def->isa<Extract>()) return Tok::Prec::Extract;
    if (def->isa<Lit    >()) return Tok::Prec::Lit;
    return Tok::Prec::Bot;
}
// clang-format on
 
} // namespace thorin