ast_base.cpp

#include "ast/ast_node_type.h"
#include "ast/ast_base.h"
#include "ast/context.h"
#include <iostream>
 
using namespace tanlang;
 
ASTBase::ASTBase(ASTNodeType node_type, TokenizedSourceFile *src, int bp)
    : SourceTraceable(src), _node_type(node_type), _bp(bp) {}
 
ASTNodeType ASTBase::get_node_type() const { return _node_type; }
 
void ASTBase::set_node_type(ASTNodeType node_type) { _node_type = node_type; }
 
int ASTBase::get_bp() const { return _bp; }
 
Context *ASTBase::ctx() {
  if (!_ctx)
    _ctx = new Context((ASTBase *)this); // context <-> AST node mapping
  return _ctx;
}
 
str ASTBase::repr(const str &prefix) const {
  str ret = fmt::format("{} {}\n", prefix, this->to_string(true));
 
  vector<ASTBase *> children = get_children();
  size_t n_children = children.size();
  for (size_t i = 0; i < n_children; ++i) {
    auto *c = children[i];
    if (c) {
      ret += c->repr(prefix + "-");
    }
  }
 
  return ret;
}
 
str ASTBase::to_string(bool include_source_code) const {
  str ret = ASTTypeNames[_node_type];
 
  if (include_source_code) {
    str code = src()->get_source_code(start(), end());
    ret = fmt::format("{}: `{}`", ret, code);
  }
 
  return ret;
}
 
ASTBase *ASTBase::get() const { return const_cast<ASTBase *>(this); }
 
vector<ASTBase *> ASTBase::get_children() const {
  TAN_ASSERT(false);
  return {};
}
 
#define MAKE_ASTTYPE_NAME_PAIR(t) \
  { ASTNodeType::t, #t }
 
umap<ASTNodeType, str> ASTBase::ASTTypeNames = {
    MAKE_ASTTYPE_NAME_PAIR(PROGRAM),
    MAKE_ASTTYPE_NAME_PAIR(FUNC_CALL),
    MAKE_ASTTYPE_NAME_PAIR(FUNC_DECL),
    MAKE_ASTTYPE_NAME_PAIR(ARG_DECL),
    MAKE_ASTTYPE_NAME_PAIR(VAR_DECL),
    MAKE_ASTTYPE_NAME_PAIR(STRUCT_DECL),
    MAKE_ASTTYPE_NAME_PAIR(COMPOUND_STATEMENT),
    MAKE_ASTTYPE_NAME_PAIR(BOP),
    MAKE_ASTTYPE_NAME_PAIR(UOP),
    MAKE_ASTTYPE_NAME_PAIR(BOP_OR_UOP),
    MAKE_ASTTYPE_NAME_PAIR(ASSIGN),
    MAKE_ASTTYPE_NAME_PAIR(CAST),
    MAKE_ASTTYPE_NAME_PAIR(ID),
    MAKE_ASTTYPE_NAME_PAIR(LOOP),
    MAKE_ASTTYPE_NAME_PAIR(CONTINUE),
    MAKE_ASTTYPE_NAME_PAIR(BREAK),
    MAKE_ASTTYPE_NAME_PAIR(PARENTHESIS),
    MAKE_ASTTYPE_NAME_PAIR(RET),
    MAKE_ASTTYPE_NAME_PAIR(IF),
    MAKE_ASTTYPE_NAME_PAIR(IMPORT),
    MAKE_ASTTYPE_NAME_PAIR(VAR_REF),
    MAKE_ASTTYPE_NAME_PAIR(INTRINSIC),
 
    MAKE_ASTTYPE_NAME_PAIR(BOOL_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(INTEGER_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(FLOAT_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(CHAR_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(STRING_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(ARRAY_LITERAL),
    MAKE_ASTTYPE_NAME_PAIR(NULLPTR_LITERAL),
};
 
#undef MAKE_ASTTYPE_NAME_PAIR
 
umap<ASTNodeType, int> ASTBase::OpPrecedence = {
    {ASTNodeType::PROGRAM,            PREC_LOWEST },
    {ASTNodeType::COMPOUND_STATEMENT, PREC_LOWEST },
    {ASTNodeType::PARENTHESIS,        PREC_CALL   },
    {ASTNodeType::RET,                PREC_LOWEST },
    {ASTNodeType::IF,                 PREC_LOWEST },
    {ASTNodeType::STRING_LITERAL,     PREC_LITERAL},
    {ASTNodeType::CAST,               PREC_CAST   },
    {ASTNodeType::ASSIGN,             PREC_ASSIGN }
};