55 lines
1.9 KiB
C++
55 lines
1.9 KiB
C++
#ifndef wg_NODE_HPP
|
|
#define wg_NODE_HPP
|
|
|
|
#include "commons.hpp"
|
|
#include "Loc.hpp"
|
|
|
|
#define NODE_TYPES(G) \
|
|
G(NODE_PROG), \
|
|
G(NODE_IDENT), \
|
|
G(NODE_HASH), \
|
|
G(NODE_DIR), \
|
|
G(NODE_INT), \
|
|
G(NODE_ADD), G(NODE_SUB), \
|
|
G(NODE_MUL),G(NODE_DIV), \
|
|
G(NODE_MOD), G(NODE_OPAR), G(NODE_CPAR), \
|
|
G(NODE_SEMICOLON), G(NODE_COMMA), G(NODE_CALL), \
|
|
G(NODE_ARGS), G(NODE_TYPE), G(NODE_RETURN), \
|
|
G(NODE_FUN), G(NODE_PARAMS), G(NODE_BLOCK), \
|
|
G(NODE_OBRACE), G(NODE_CBRACE), G(NODE_FUNDECL), \
|
|
G(NODE_EXTERN), G(NODE_RET), G(NODE_DOT), G(NODE_NS), \
|
|
G(NODE_AS), G(NODE_BOOL), G(NODE_AND), G(NODE_OR), \
|
|
G(NODE_NOT), G(NODE_EQ), G(NODE_NE), G(NODE_LT), G(NODE_GT), \
|
|
G(NODE_LE), G(NODE_GE)
|
|
|
|
namespace wg
|
|
{
|
|
WG_ENUM(NodeType, NODE_TYPES);
|
|
|
|
class Node
|
|
{
|
|
public:
|
|
explicit Node(NodeType type, std::string const& repr, Loc const& loc);
|
|
virtual ~Node();
|
|
|
|
NodeType type() const { return m_type; }
|
|
std::string repr() const { return m_repr; }
|
|
Loc loc() const { return m_loc; }
|
|
|
|
size_t size() const { return m_children.size(); }
|
|
|
|
void add_child(std::shared_ptr<Node> child);
|
|
std::shared_ptr<Node> child(size_t index) const;
|
|
|
|
std::string string() const;
|
|
|
|
private:
|
|
NodeType m_type;
|
|
std::string m_repr;
|
|
Loc m_loc;
|
|
std::vector<std::shared_ptr<Node>> m_children;
|
|
};
|
|
}
|
|
|
|
#endif
|