Remove handle #7

Merged
CHatingPython merged 11 commits from remove-handle into master 2026-06-03 12:14:38 +00:00
8 changed files with 162 additions and 67 deletions
Showing only changes of commit 52537e0955 - Show all commits
+15 -5
View File
@@ -19,7 +19,15 @@ enum class declaration_node_t {
/** /**
* @brief Declaration AST node interface. * @brief Declaration AST node interface.
*/ */
class declaration_node : public statement_node { class declaration_node : public statement_node, public abstract_node {
public:
/**
* @brief Construct a new declaration AST node.
*
* @param location Node location.
*/
declaration_node(struct location location)
: abstract_node(location) {}
public: public:
/** /**
* @brief Returns this node's category. * @brief Returns this node's category.
@@ -53,10 +61,11 @@ public:
/** /**
* @brief Construct a new function declaration node object from name token. * @brief Construct a new function declaration node object from name token.
* *
* @param location Node location.
* @param name Name of the function. * @param name Name of the function.
*/ */
function_declaration_node(front::token name) function_declaration_node(struct location location, front::token name)
: p_name(name) {} : declaration_node(location), p_name(name) {}
public: public:
/** /**
* @brief Returns this node's declaration type. * @brief Returns this node's declaration type.
@@ -92,11 +101,12 @@ public:
/** /**
* @brief Construct a new function definition node object from name and body. * @brief Construct a new function definition node object from name and body.
* *
* @param location Node location.
* @param name Name of the function. * @param name Name of the function.
* @param body Body of the function. * @param body Body of the function.
*/ */
function_definition_node(front::token name, body_r&& body) function_definition_node(struct location location, front::token name, body_r&& body)
: function_declaration_node(name), m_body(std::move(body)) {} : function_declaration_node(location, name), m_body(std::move(body)) {}
public: public:
/** /**
* @brief Returns this node's declaration type. * @brief Returns this node's declaration type.
+33 -11
View File
@@ -21,7 +21,15 @@ enum class expression_node_t {
/** /**
* @brief Expression AST node. * @brief Expression AST node.
*/ */
class expression_node : public statement_node { class expression_node : public statement_node, public abstract_node {
public:
/**
* @brief Construct a new expression AST node.
*
* @param location Node location.
*/
expression_node(struct location location)
: abstract_node(location) {}
public: public:
/** /**
* @brief Returns this node's category. * @brief Returns this node's category.
@@ -57,10 +65,11 @@ public:
/** /**
* @brief Construct a new var read expression node object from a name handle. * @brief Construct a new var read expression node object from a name handle.
* *
* @param location Node location.
* @param name Handle to the name. * @param name Handle to the name.
*/ */
var_read_expression_node(name_type&& name) var_read_expression_node(struct location location, name_type&& name)
: m_name(std::move(name)) {} : expression_node(location), m_name(std::move(name)) {}
/** /**
* @brief Returns the variable's name. * @brief Returns the variable's name.
@@ -114,11 +123,12 @@ public:
/** /**
* @brief Construct a new unaryop expression node object from type and expression node handle. * @brief Construct a new unaryop expression node object from type and expression node handle.
* *
* @param location Node location.
* @param type Operation type. * @param type Operation type.
* @param node Handle to the inner expression node. * @param node Handle to the inner expression node.
*/ */
unaryop_expression_node(unaryop_expression_node_t type, expression_node_r&& node) unaryop_expression_node(struct location location, unaryop_expression_node_t type, expression_node_r&& node)
: m_type(type), m_node(std::move(node)) {} : expression_node(location), m_type(type), m_node(std::move(node)) {}
/** /**
* @brief Sets this node's inner expression. * @brief Sets this node's inner expression.
@@ -199,12 +209,16 @@ public:
/** /**
* @brief Construct a new binary operation expression AST node. * @brief Construct a new binary operation expression AST node.
* *
* @param location Node location.
* @param type Binary operation type. * @param type Binary operation type.
* @param lhs Left-hand-side expression. * @param lhs Left-hand-side expression.
* @param rhs Right-hand-side expression. * @param rhs Right-hand-side expression.
*/ */
binop_expression_node(binop_expression_node_t type, expression_node_r&& lhs, expression_node_r&& rhs) binop_expression_node(struct location location,
: m_type(type), m_lhs(std::move(lhs)), m_rhs(std::move(rhs)) {} binop_expression_node_t type,
expression_node_r&& lhs,
expression_node_r&& rhs)
: expression_node(location), m_type(type), m_lhs(std::move(lhs)), m_rhs(std::move(rhs)) {}
/** /**
* @brief Returns this node's binary operation type. * @brief Returns this node's binary operation type.
@@ -276,21 +290,29 @@ public:
/** /**
* @brief Construct a new variable assignment expression AST node. * @brief Construct a new variable assignment expression AST node.
* *
* @param location Node location.
* @param lhs Left-hand-side expression handle. * @param lhs Left-hand-side expression handle.
* @param rhs Right-hand-side expression handle. * @param rhs Right-hand-side expression handle.
*/ */
var_assign_expression_node(expression_node_r&& lhs, expression_node_r&& rhs) var_assign_expression_node(struct location location, expression_node_r&& lhs, expression_node_r&& rhs)
: m_compound(binop_expression_node_t::None), m_lhs(std::move(lhs)), m_rhs(std::move(rhs)) {} : expression_node(location),
m_compound(binop_expression_node_t::None),
m_lhs(std::move(lhs)),
m_rhs(std::move(rhs)) {}
/** /**
* @brief Construct a new compound variable assignment expression AST node. * @brief Construct a new compound variable assignment expression AST node.
* *
* @param location Node location.
* @param compound Compound operation type. * @param compound Compound operation type.
* @param lhs Left-hand-side expression handle. * @param lhs Left-hand-side expression handle.
* @param rhs Right-hand-side expression handle. * @param rhs Right-hand-side expression handle.
*/ */
var_assign_expression_node(binop_expression_node_t compound, expression_node_r&& lhs, expression_node_r&& rhs) var_assign_expression_node(struct location location,
: m_compound(compound), m_lhs(std::move(lhs)), m_rhs(std::move(rhs)) {} binop_expression_node_t compound,
expression_node_r&& lhs,
expression_node_r&& rhs)
: expression_node(location), m_compound(compound), m_lhs(std::move(lhs)), m_rhs(std::move(rhs)) {}
/** /**
* @brief Returns this node's compound operation type. * @brief Returns this node's compound operation type.
+14 -4
View File
@@ -20,6 +20,14 @@ enum class literal_node_t {
* @brief Literal AST node. * @brief Literal AST node.
*/ */
class literal_node : public expression_node { class literal_node : public expression_node {
public:
/**
* @brief Construct a new literal AST node.
*
* @param location Node location.
*/
literal_node(struct location location)
: expression_node(location) {}
public: public:
/** /**
* @brief Returns this node's category. * @brief Returns this node's category.
@@ -55,10 +63,11 @@ public:
/** /**
* @brief Construct a new string literal node object from a handle. * @brief Construct a new string literal node object from a handle.
* *
* @param location Node location.
* @param value A string view result. * @param value A string view result.
*/ */
string_literal_node(value_type&& value) string_literal_node(struct location location, value_type&& value)
: m_value(std::move(value)) {} : literal_node(location), m_value(std::move(value)) {}
public: public:
/** /**
* @brief Returns this node's literal type. * @brief Returns this node's literal type.
@@ -93,10 +102,11 @@ public:
/** /**
* @brief Construct a new integer literal node object from a handle. * @brief Construct a new integer literal node object from a handle.
* *
* @param location Node location.
* @param value An integer result. * @param value An integer result.
*/ */
integer_literal_node(value_type&& value) integer_literal_node(struct location location, value_type&& value)
: m_value(std::move(value)) {} : literal_node(location), m_value(std::move(value)) {}
public: public:
/** /**
* @brief Returns this node's literal type. * @brief Returns this node's literal type.
+30
View File
@@ -71,6 +71,14 @@ public:
* @return The node category. * @return The node category.
*/ */
virtual node_t category() const = 0; virtual node_t category() const = 0;
/**
* @brief Returns the location of this AST node.
* @see locaiton
*
* @return The location.
*/
virtual location location() const = 0;
public: public:
/** /**
* @brief Compares two nodes for equality. * @brief Compares two nodes for equality.
@@ -128,6 +136,28 @@ protected:
virtual bool equal(const node& rhs) const = 0; virtual bool equal(const node& rhs) const = 0;
}; };
/**
* @brief An abstract AST node.
* @see node
*
* Implements location().
*/
class abstract_node : public virtual node {
public:
abstract_node(struct location location)
: p_location(location) {}
public:
/**
* @brief Returns the location of this AST node.
* @see locaiton
*
* @return The location.
*/
struct location location() const override { return p_location; }
protected:
struct location p_location; /**< Node location. */
};
} // namespace ast } // namespace ast
} // namespace furc } // namespace furc
+8 -2
View File
@@ -12,9 +12,15 @@ namespace ast {
/** /**
* @brief Program AST node. * @brief Program AST node.
*/ */
class program_node final : public node { class program_node final : public abstract_node {
public: public:
program_node() = default; /**
* @brief Construct a new program AST node.
*
* @param location Node location.
*/
program_node(struct location location)
: abstract_node(location) {}
node_t category() const override { return node_t::Program; } node_t category() const override { return node_t::Program; }
public: public:
+24 -13
View File
@@ -22,7 +22,7 @@ enum class statement_node_t {
/** /**
* @brief Statement AST node. * @brief Statement AST node.
*/ */
class statement_node : public node { class statement_node : public virtual node {
public: public:
/** /**
* @brief Returns this node's category. * @brief Returns this node's category.
@@ -44,19 +44,24 @@ protected:
/** /**
* @brief Return statement AST node. * @brief Return statement AST node.
*/ */
class return_statement_node final : public statement_node { class return_statement_node final : public statement_node, public abstract_node {
public: public:
using value_type = std::optional<expression_node_r>; /**< Value type. */ using value_type = std::optional<expression_node_r>; /**< Value type. */
public: public:
return_statement_node() = default; /**
* @brief Construct a new return statement AST node.
*/
return_statement_node(struct location location)
: abstract_node(location) {}
/** /**
* @brief Construct a new return statement AST node. * @brief Construct a new return statement AST node.
* *
* @param location Node location.
* @param value Return value handle. * @param value Return value handle.
*/ */
return_statement_node(expression_node_r&& value) return_statement_node(struct location location, expression_node_r&& value)
: m_value(std::move(value)) {} : abstract_node(location), m_value(std::move(value)) {}
public: public:
/** /**
* @brief Returns this node's return value handle. * @brief Returns this node's return value handle.
@@ -84,26 +89,31 @@ private:
/** /**
* @brief If statement AST node. * @brief If statement AST node.
*/ */
class if_statement_node final : public statement_node { class if_statement_node final : public statement_node, public abstract_node {
public: public:
/** /**
* @brief Construct a new if statement AST node. * @brief Construct a new if statement AST node.
* *
* @param location Node location.
* @param cond Condition expression handle. * @param cond Condition expression handle.
* @param then Then statement handle. * @param then Then statement handle.
*/ */
if_statement_node(expression_node_r&& cond, statement_node_r&& then) if_statement_node(struct location location, expression_node_r&& cond, statement_node_r&& then)
: m_cond(std::move(cond)), m_then(std::move(then)) {} : abstract_node(location), m_cond(std::move(cond)), m_then(std::move(then)) {}
/** /**
* @brief Construct a new if statement AST node. * @brief Construct a new if statement AST node.
* *
* @param location Node location.
* @param cond Condition expression handle. * @param cond Condition expression handle.
* @param then Then statement handle. * @param then Then statement handle.
* @param elze Else statement handle. * @param elze Else statement handle.
*/ */
if_statement_node(expression_node_r&& cond, statement_node_r&& then, statement_node_r&& elze) if_statement_node(struct location location,
: m_cond(std::move(cond)), m_then(std::move(then)), m_else(std::move(elze)) {} expression_node_r&& cond,
statement_node_r&& then,
statement_node_r&& elze)
: abstract_node(location), m_cond(std::move(cond)), m_then(std::move(then)), m_else(std::move(elze)) {}
public: public:
/** /**
* @brief Returns this node's condition expression handle. * @brief Returns this node's condition expression handle.
@@ -147,15 +157,16 @@ private:
/** /**
* @brief Compound statement AST node. * @brief Compound statement AST node.
*/ */
class compound_statement_node final : public statement_node { class compound_statement_node final : public statement_node, public abstract_node {
public: public:
/** /**
* @brief Construct a new compound statement AST node. * @brief Construct a new compound statement AST node.
* *
* @param location Node location.
* @param body Body handle. * @param body Body handle.
*/ */
compound_statement_node(body_r&& body) compound_statement_node(struct location location, body_r&& body)
: m_body(std::move(body)) {} : abstract_node(location), m_body(std::move(body)) {}
public: public:
/** /**
* @brief Returns this node's body handle. * @brief Returns this node's body handle.
+16 -17
View File
@@ -13,7 +13,7 @@ std::ostream& operator<<(std::ostream& os, const error& error) {
} }
bool literal_node::equal(const node& rhs) const { bool literal_node::equal(const node& rhs) const {
return literal_type() == reinterpret_cast<const literal_node&>(rhs).literal_type(); return literal_type() == dynamic_cast<const literal_node&>(rhs).literal_type();
} }
void string_literal_node::accept(visitor& visitor) const { void string_literal_node::accept(visitor& visitor) const {
@@ -26,7 +26,7 @@ std::ostream& string_literal_node::print(std::ostream& os) const {
} }
bool string_literal_node::equal(const node& rhs) const { bool string_literal_node::equal(const node& rhs) const {
return literal_node::equal(rhs) && m_value == reinterpret_cast<const string_literal_node&>(rhs).m_value; return literal_node::equal(rhs) && m_value == dynamic_cast<const string_literal_node&>(rhs).m_value;
} }
void integer_literal_node::accept(visitor& visitor) const { void integer_literal_node::accept(visitor& visitor) const {
@@ -39,11 +39,11 @@ std::ostream& integer_literal_node::print(std::ostream& os) const {
} }
bool integer_literal_node::equal(const node& rhs) const { bool integer_literal_node::equal(const node& rhs) const {
return literal_node::equal(rhs) && m_value == reinterpret_cast<const integer_literal_node&>(rhs).m_value; return literal_node::equal(rhs) && m_value == dynamic_cast<const integer_literal_node&>(rhs).m_value;
} }
bool expression_node::equal(const node& rhs) const { bool expression_node::equal(const node& rhs) const {
return expression_type() == reinterpret_cast<const expression_node&>(rhs).expression_type(); return expression_type() == dynamic_cast<const expression_node&>(rhs).expression_type();
} }
void var_read_expression_node::accept(visitor& visitor) const { void var_read_expression_node::accept(visitor& visitor) const {
@@ -56,7 +56,7 @@ std::ostream& var_read_expression_node::print(std::ostream& os) const {
} }
bool var_read_expression_node::equal(const node& rhsNode) const { bool var_read_expression_node::equal(const node& rhsNode) const {
const auto& rhs = reinterpret_cast<const var_read_expression_node&>(rhsNode); const auto& rhs = dynamic_cast<const var_read_expression_node&>(rhsNode);
return expression_node::equal(rhsNode) && m_name == rhs.m_name; return expression_node::equal(rhsNode) && m_name == rhs.m_name;
} }
@@ -90,7 +90,7 @@ std::ostream& unaryop_expression_node::print(std::ostream& os) const {
} }
bool unaryop_expression_node::equal(const node& rhsNode) const { bool unaryop_expression_node::equal(const node& rhsNode) const {
const auto& rhs = reinterpret_cast<const unaryop_expression_node&>(rhsNode); const auto& rhs = dynamic_cast<const unaryop_expression_node&>(rhsNode);
return expression_node::equal(rhsNode) && m_type == rhs.m_type && m_node == rhs.m_node; return expression_node::equal(rhsNode) && m_type == rhs.m_type && m_node == rhs.m_node;
} }
@@ -122,7 +122,7 @@ std::ostream& binop_expression_node::print(std::ostream& os) const {
} }
bool binop_expression_node::equal(const node& rhsNode) const { bool binop_expression_node::equal(const node& rhsNode) const {
const auto& rhs = reinterpret_cast<const binop_expression_node&>(rhsNode); const auto& rhs = dynamic_cast<const binop_expression_node&>(rhsNode);
return expression_node::equal(rhsNode) && m_type == rhs.m_type && m_lhs == rhs.m_lhs && m_rhs == rhs.m_rhs; return expression_node::equal(rhsNode) && m_type == rhs.m_type && m_lhs == rhs.m_lhs && m_rhs == rhs.m_rhs;
} }
@@ -135,12 +135,12 @@ std::ostream& var_assign_expression_node::print(std::ostream& os) const {
} }
bool var_assign_expression_node::equal(const node& rhsNode) const { bool var_assign_expression_node::equal(const node& rhsNode) const {
const auto& rhs = reinterpret_cast<const var_assign_expression_node&>(rhsNode); const auto& rhs = dynamic_cast<const var_assign_expression_node&>(rhsNode);
return expression_node::equal(rhsNode) && m_compound == rhs.m_compound && m_lhs == rhs.m_lhs && m_rhs == rhs.m_rhs; return expression_node::equal(rhsNode) && m_compound == rhs.m_compound && m_lhs == rhs.m_lhs && m_rhs == rhs.m_rhs;
} }
bool declaration_node::equal(const node& rhs) const { bool declaration_node::equal(const node& rhs) const {
return declaration_type() == reinterpret_cast<const declaration_node&>(rhs).declaration_type(); return declaration_type() == dynamic_cast<const declaration_node&>(rhs).declaration_type();
} }
void function_declaration_node::accept(visitor& visitor) const { void function_declaration_node::accept(visitor& visitor) const {
@@ -152,7 +152,7 @@ std::ostream& function_declaration_node::print(std::ostream& os) const {
} }
bool function_declaration_node::equal(const node& rhs) const { bool function_declaration_node::equal(const node& rhs) const {
return declaration_node::equal(rhs) && p_name == reinterpret_cast<const function_declaration_node&>(rhs).p_name; return declaration_node::equal(rhs) && p_name == dynamic_cast<const function_declaration_node&>(rhs).p_name;
} }
void function_definition_node::accept(visitor& visitor) const { void function_definition_node::accept(visitor& visitor) const {
@@ -171,12 +171,11 @@ std::ostream& function_definition_node::print(std::ostream& os) const {
} }
bool function_definition_node::equal(const node& rhs) const { bool function_definition_node::equal(const node& rhs) const {
return function_declaration_node::equal(rhs) && return function_declaration_node::equal(rhs) && m_body == dynamic_cast<const function_definition_node&>(rhs).m_body;
m_body == reinterpret_cast<const function_definition_node&>(rhs).m_body;
} }
bool statement_node::equal(const node& rhs) const { bool statement_node::equal(const node& rhs) const {
return statement_type() == reinterpret_cast<const statement_node&>(rhs).statement_type(); return statement_type() == dynamic_cast<const statement_node&>(rhs).statement_type();
} }
void return_statement_node::accept(visitor& visitor) const { void return_statement_node::accept(visitor& visitor) const {
@@ -190,7 +189,7 @@ std::ostream& return_statement_node::print(std::ostream& os) const {
} }
bool return_statement_node::equal(const node& rhs) const { bool return_statement_node::equal(const node& rhs) const {
return statement_node::equal(rhs) && m_value == reinterpret_cast<const return_statement_node&>(rhs).m_value; return statement_node::equal(rhs) && m_value == dynamic_cast<const return_statement_node&>(rhs).m_value;
} }
void if_statement_node::accept(visitor& visitor) const { void if_statement_node::accept(visitor& visitor) const {
@@ -205,7 +204,7 @@ std::ostream& if_statement_node::print(std::ostream& os) const {
} }
bool if_statement_node::equal(const node& rhsNode) const { bool if_statement_node::equal(const node& rhsNode) const {
const auto& rhs = reinterpret_cast<const if_statement_node&>(rhsNode); const auto& rhs = dynamic_cast<const if_statement_node&>(rhsNode);
return statement_node::equal(rhs) && m_cond == rhs.m_cond && m_then == rhs.m_then && m_else == rhs.m_else; return statement_node::equal(rhs) && m_cond == rhs.m_cond && m_then == rhs.m_then && m_else == rhs.m_else;
} }
@@ -218,7 +217,7 @@ std::ostream& compound_statement_node::print(std::ostream& os) const {
} }
bool compound_statement_node::equal(const node& rhs) const { bool compound_statement_node::equal(const node& rhs) const {
return statement_node::equal(rhs) && m_body == reinterpret_cast<const compound_statement_node&>(rhs).m_body; return statement_node::equal(rhs) && m_body == dynamic_cast<const compound_statement_node&>(rhs).m_body;
} }
void program_node::accept(visitor& visitor) const { void program_node::accept(visitor& visitor) const {
@@ -240,7 +239,7 @@ std::ostream& program_node::print(std::ostream& os) const {
} }
bool program_node::equal(const node& rhs) const { bool program_node::equal(const node& rhs) const {
return m_declarations == reinterpret_cast<const program_node&>(rhs).m_declarations; return m_declarations == dynamic_cast<const program_node&>(rhs).m_declarations;
} }
std::ostream& operator<<(std::ostream& os, const body& body) { std::ostream& operator<<(std::ostream& os, const body& body) {
+22 -15
View File
@@ -31,7 +31,7 @@ parser::parser(std::string_view filename)
} }
ast::program_node_r parser::parse() & { ast::program_node_r parser::parse() & {
auto program = m_arena.allocate_shared<ast::program_node>(); auto program = m_arena.allocate_shared<ast::program_node>(location{ m_filename });
while (peek_token().has_value()) { while (peek_token().has_value()) {
program->push(std::move(parse_declaration())); program->push(std::move(parse_declaration()));
@@ -61,11 +61,11 @@ ast::declaration_node_r parser::parse_declaration() {
case token_t::LBrace: { case token_t::LBrace: {
ast::body_r body = parse_body(); ast::body_r body = parse_body();
if (body.has_error()) return ast::declaration_node_r(ast::error{ body.error().location }); if (body.has_error()) return ast::declaration_node_r(ast::error{ body.error().location });
return m_arena.allocate_shared<ast::function_definition_node>(*name, std::move(body)); return m_arena.allocate_shared<ast::function_definition_node>(first->location, *name, std::move(body));
} }
case token_t::Semicolon: { case token_t::Semicolon: {
m_peekBuffer.clear(); m_peekBuffer.clear();
return m_arena.allocate_shared<ast::function_declaration_node>(*name); return m_arena.allocate_shared<ast::function_declaration_node>(first->location, *name);
} }
default: return ast::declaration_node_r(ast::error{ tok->location }); default: return ast::declaration_node_r(ast::error{ tok->location });
} }
@@ -101,13 +101,13 @@ ast::statement_node_r parser::parse_statement() {
auto tok = next_token(); auto tok = next_token();
if (peek_token()->type == token_t::Semicolon) { if (peek_token()->type == token_t::Semicolon) {
next_token(); next_token();
return m_arena.allocate_shared<ast::return_statement_node>(); return m_arena.allocate_shared<ast::return_statement_node>(location);
} }
auto value = parse_expression(); auto value = parse_expression();
auto err = eat_token(token_t::Semicolon); auto err = eat_token(token_t::Semicolon);
if (err.has_error()) return ast::statement_node_r(ast::error{ err.error().location }); if (err.has_error()) return ast::statement_node_r(ast::error{ err.error().location });
return m_arena.allocate_shared<ast::return_statement_node>(std::move(value)); return m_arena.allocate_shared<ast::return_statement_node>(location, std::move(value));
} }
case keyword_token::If: { case keyword_token::If: {
auto tok = next_token(); auto tok = next_token();
@@ -126,19 +126,20 @@ ast::statement_node_r parser::parse_statement() {
peek_token()->value.keyword == keyword_token::Else) { peek_token()->value.keyword == keyword_token::Else) {
next_token(); next_token();
return m_arena.allocate_shared<ast::if_statement_node>(std::move(cond), return m_arena.allocate_shared<ast::if_statement_node>(location,
std::move(cond),
std::move(then), std::move(then),
std::move(parse_statement())); std::move(parse_statement()));
} }
return m_arena.allocate_shared<ast::if_statement_node>(std::move(cond), std::move(then)); return m_arena.allocate_shared<ast::if_statement_node>(location, std::move(cond), std::move(then));
} }
case keyword_token::None: case keyword_token::None:
case keyword_token::Func: case keyword_token::Func:
default: break; default: break;
} }
} }
case token_t::LBrace: return m_arena.allocate_shared<ast::compound_statement_node>(parse_body()); case token_t::LBrace: return m_arena.allocate_shared<ast::compound_statement_node>(location, parse_body());
default: break; default: break;
} }
@@ -165,12 +166,12 @@ ast::literal_node_r parser::parse_literal() {
case token_t::String: { case token_t::String: {
auto tok = next_token(); auto tok = next_token();
if (tok.has_error()) return ast::literal_node_r(ast::error{ tok.error().location }); if (tok.has_error()) return ast::literal_node_r(ast::error{ tok.error().location });
return m_arena.allocate_shared<ast::string_literal_node>((*tok)->string); return m_arena.allocate_shared<ast::string_literal_node>(tok->location, (*tok)->string);
} }
case token_t::Integer: { case token_t::Integer: {
auto tok = next_token(); auto tok = next_token();
if (tok.has_error()) return ast::literal_node_r(ast::error{ tok.error().location }); if (tok.has_error()) return ast::literal_node_r(ast::error{ tok.error().location });
return m_arena.allocate_shared<ast::integer_literal_node>((*tok)->integer); return m_arena.allocate_shared<ast::integer_literal_node>(tok->location, (*tok)->integer);
} }
default: break; default: break;
} }
@@ -184,7 +185,7 @@ ast::expression_node_r parser::parse_expression_primary() {
case token_t::Identifier: { case token_t::Identifier: {
auto tok = next_token(); auto tok = next_token();
if (tok.has_error()) return ast::expression_node_r(ast::error{ tok.error().location }); if (tok.has_error()) return ast::expression_node_r(ast::error{ tok.error().location });
return m_arena.allocate_shared<ast::var_read_expression_node>((*tok)->string); return m_arena.allocate_shared<ast::var_read_expression_node>(tok->location, (*tok)->string);
} }
case token_t::LParen: { case token_t::LParen: {
auto tok = next_token(); auto tok = next_token();
@@ -230,7 +231,8 @@ ast::expression_node_r parser::parse_expression_unary(std::uint32_t precedence)
expression = parse_expression_unary(current.precedence + 1); expression = parse_expression_unary(current.precedence + 1);
} }
result = m_arena.allocate_shared<ast::unaryop_expression_node>(current.type, std::move(expression)); result =
m_arena.allocate_shared<ast::unaryop_expression_node>(token->location, current.type, std::move(expression));
} }
if (result == nullptr) return parse_expression_primary(); if (result == nullptr) return parse_expression_primary();
@@ -341,13 +343,18 @@ ast::expression_node_r parser::parse_expression_rhs(ast::expression_node_r&& ini
switch (current.type) { switch (current.type) {
case rhsop_info_t::Unaryop: case rhsop_info_t::Unaryop:
lhs = m_arena.allocate_shared<ast::unaryop_expression_node>(current.unary, std::move(lhs)); lhs =
m_arena.allocate_shared<ast::unaryop_expression_node>(opToken->location, current.unary, std::move(lhs));
break; break;
case rhsop_info_t::Binop: case rhsop_info_t::Binop:
lhs = m_arena.allocate_shared<ast::binop_expression_node>(current.binary, std::move(lhs), std::move(rhs)); lhs = m_arena.allocate_shared<ast::binop_expression_node>(opToken->location,
current.binary,
std::move(lhs),
std::move(rhs));
break; break;
case rhsop_info_t::Assignment: case rhsop_info_t::Assignment:
lhs = m_arena.allocate_shared<ast::var_assign_expression_node>(current.assignment, lhs = m_arena.allocate_shared<ast::var_assign_expression_node>(opToken->location,
current.assignment,
std::move(lhs), std::move(lhs),
std::move(rhs)); std::move(rhs));
break; break;