ast_base.h

#ifndef __TAN_SRC_AST_AST_BASE_H__
#define __TAN_SRC_AST_AST_BASE_H__
#include "base.h"
#include "source_traceable.h"
#include "precedence.h"
#include "ast_node_type.h"
#include <typeinfo>
 
namespace tanlang {
 
class Context;
 
class ASTBase : public SourceTraceable {
public:
  /// string representation of ASTNodeType
  static umap<ASTNodeType, str> ASTTypeNames;
 
  /// operator precedence of tokens
  static umap<ASTNodeType, int> OpPrecedence;
 
public:
  ASTBase() = delete;
  ASTBase(ASTNodeType node_type, TokenizedSourceFile *src, int bp);
  virtual ~ASTBase() = default;
 
public:
  [[nodiscard]] ASTNodeType get_node_type() const;
  void set_node_type(ASTNodeType node_type);
  [[nodiscard]] int get_bp() const;
  [[nodiscard]] Context *ctx();
 
  /**
   * \brief Get a ordered list of child nodes
   */
  [[nodiscard]] virtual vector<ASTBase *> get_children() const;
 
  /// AST tree string representation
  str repr(const str &prefix = "-") const;
 
public:
  virtual bool is_stmt() const = 0;
  virtual bool is_expr() const = 0;
 
  /// Which terminal token is expected immediately after this node
  virtual str terminal_token() const { return ";"; }
 
protected:
  /// Different from repr, to_string output doesn't include child nodes
  [[nodiscard]] virtual str to_string(bool include_source_code = false) const;
 
  /**
   * \brief Get the "actual" this. Used for implementing proxy classes.
   */
  [[nodiscard]] virtual ASTBase *get() const;
 
private:
  ASTNodeType _node_type;
  int _bp = 0; /// binding power
  Context *_ctx = nullptr;
};
 
} // namespace tanlang
 
#endif //__TAN_SRC_AST_AST_BASE_H__