2023-08-30 18:06:26 +00:00
|
|
|
#include "Compiler.hpp"
|
2023-08-31 12:41:43 +00:00
|
|
|
#include "lib/Node.hpp"
|
2023-08-30 18:06:26 +00:00
|
|
|
#include "lib/opcodes.hpp"
|
2023-08-31 12:41:43 +00:00
|
|
|
#include "StaticPass.hpp"
|
2023-08-30 18:06:26 +00:00
|
|
|
|
|
|
|
namespace roza
|
|
|
|
{
|
|
|
|
/*explicit*/ Compiler::Compiler(StatusLog& log)
|
|
|
|
: m_log { log }
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
/*virtual*/ Compiler::~Compiler()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
std::shared_ptr<Program> Compiler::compile(std::shared_ptr<Node> root)
|
|
|
|
{
|
|
|
|
auto program = std::make_shared<Program>();
|
|
|
|
compile_node(root, program);
|
|
|
|
return program;
|
|
|
|
}
|
|
|
|
|
|
|
|
void Compiler::compile_node(std::shared_ptr<Node> root, std::shared_ptr<Program> prog)
|
|
|
|
{
|
|
|
|
switch (root->type())
|
|
|
|
{
|
2023-08-31 12:41:43 +00:00
|
|
|
case NODE_ASSERT: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_ASSERT);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_ASSERT_STATIC_FAIL: {
|
|
|
|
bool failed = false;
|
|
|
|
|
|
|
|
try
|
|
|
|
{
|
|
|
|
StaticPass static_pass {m_log};
|
|
|
|
static_pass.check_children(root);
|
|
|
|
compile_children(root, prog);
|
|
|
|
|
|
|
|
failed = true;
|
|
|
|
}
|
|
|
|
catch(...)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
if (failed)
|
|
|
|
{
|
|
|
|
m_log.fatal(root->loc(), "static assertion failed");
|
|
|
|
}
|
|
|
|
|
|
|
|
} break;
|
|
|
|
|
2023-08-30 18:06:26 +00:00
|
|
|
case NODE_INT: {
|
|
|
|
auto value = std::make_shared<Value>(std::stoi(root->repr()), root->loc());
|
|
|
|
prog->push_instr(OP_PUSH_CONST, prog->push_value(value));
|
|
|
|
} break;
|
|
|
|
|
2023-08-31 09:07:03 +00:00
|
|
|
case NODE_BOOL: {
|
|
|
|
auto value = std::make_shared<Value>(root->repr() == "true", root->loc());
|
|
|
|
prog->push_instr(OP_PUSH_CONST, prog->push_value(value));
|
|
|
|
} break;
|
|
|
|
|
2023-08-31 09:37:13 +00:00
|
|
|
case NODE_EQ: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_EQ);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_NE: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_EQ);
|
|
|
|
prog->push_instr(OP_NOT);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_LT: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_ILT);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_GT: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IGT);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_LE: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IGT);
|
|
|
|
prog->push_instr(OP_NOT);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_GE: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_ILT);
|
|
|
|
prog->push_instr(OP_NOT);
|
|
|
|
} break;
|
|
|
|
|
2023-08-31 09:07:03 +00:00
|
|
|
case NODE_IMP: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IMP);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_AND: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_AND);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_OR: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_OR);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_NOT: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_NOT);
|
|
|
|
} break;
|
|
|
|
|
2023-08-30 22:31:19 +00:00
|
|
|
case NODE_ADD: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IADD);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_SUB: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_ISUB);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_MUL: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IMUL);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_DIV: {
|
2023-08-30 18:06:26 +00:00
|
|
|
compile_children(root, prog);
|
2023-08-30 22:31:19 +00:00
|
|
|
prog->push_instr(OP_IDIV);
|
2023-08-30 18:06:26 +00:00
|
|
|
} break;
|
|
|
|
|
2023-08-30 22:31:19 +00:00
|
|
|
case NODE_MOD: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IMOD);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_POW: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IPOW);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_UADD: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IUADD);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_USUB: {
|
|
|
|
compile_children(root, prog);
|
|
|
|
prog->push_instr(OP_IUSUB);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
case NODE_PROG: {
|
2023-08-30 18:06:26 +00:00
|
|
|
compile_children(root, prog);
|
|
|
|
} break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
m_log.fatal(root->loc(), "cannot compile node '" + root->string() + "'.");
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void Compiler::compile_children(std::shared_ptr<Node> root, std::shared_ptr<Program> prog)
|
|
|
|
{
|
|
|
|
for (size_t i=0; i<root->size(); i++)
|
|
|
|
{
|
|
|
|
compile_node(root->child(i), prog);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|