85 lines
2.2 KiB
C++
85 lines
2.2 KiB
C++
#include "macro.hpp"
|
|
#include "src/Compiler.hpp"
|
|
|
|
namespace fkstd
|
|
{
|
|
void assert_static_fail(Compiler& compiler,
|
|
std::shared_ptr<Node> node,
|
|
std::shared_ptr<Program> program)
|
|
{
|
|
try
|
|
{
|
|
compiler.compile_prog(node->child(1), program);
|
|
}
|
|
catch(std::exception const&)
|
|
{
|
|
program->load_const(std::make_shared<Constant>(TYPE_BOOL,
|
|
true,
|
|
node->loc()));
|
|
return;
|
|
}
|
|
|
|
std::stringstream ss;
|
|
ss << "assertion failed";
|
|
node->loc().error<assert_error>(LOG_ERROR, ss.str());
|
|
}
|
|
|
|
void decl(Compiler& compiler,
|
|
std::shared_ptr<Node> node,
|
|
std::shared_ptr<Program> program)
|
|
{
|
|
static addr_t addr = 0;
|
|
addr++;
|
|
|
|
std::string ident = node->child(1)->repr();
|
|
auto rhs = node->child(2);
|
|
|
|
compiler.compile_prog(rhs, program);
|
|
|
|
auto entry = compiler.sym()->declare_local(ident,
|
|
addr,
|
|
node->loc())
|
|
.set_node(rhs);
|
|
|
|
program->add(OP_STORE_LOCAL, addr);
|
|
}
|
|
|
|
void block(Compiler& compiler,
|
|
std::shared_ptr<Node> node,
|
|
std::shared_ptr<Program> program)
|
|
{
|
|
compiler.sym()->enter_scope();
|
|
|
|
for (size_t i=1; i<node->size(); i++)
|
|
{
|
|
compiler.compile_prog(node->child(i), program);
|
|
|
|
if (i != node->size() - 1)
|
|
{
|
|
program->add(OP_POP);
|
|
}
|
|
}
|
|
|
|
compiler.sym()->leave_scope();
|
|
}
|
|
|
|
void if_macro(Compiler& compiler,
|
|
std::shared_ptr<Node> node,
|
|
std::shared_ptr<Program> program)
|
|
{
|
|
auto if_expr = node->child(1);
|
|
auto then_expr = node->child(2);
|
|
auto else_expr = node->child(3);
|
|
|
|
compiler.compile_prog(if_expr, program);
|
|
size_t to_else = program->add(OP_BNE, 0 /*else*/);
|
|
|
|
compiler.compile_prog(then_expr, program);
|
|
size_t to_end = program->add(OP_BR, 0 /* end */);
|
|
|
|
program->set_param(to_else, program->size());
|
|
compiler.compile_prog(else_expr, program);
|
|
program->set_param(to_end, program->size());
|
|
}
|
|
}
|