lang: codegen: implement return generation

This commit is contained in:
2025-04-24 09:25:36 +01:00
parent da4aa7df78
commit d37c07e5e4
5 changed files with 132 additions and 0 deletions

View File

@@ -183,6 +183,27 @@ static struct code_generator_result gen_var_declaration(
return CODEGEN_RESULT_OK(CODEGEN_R_REPEAT_NODE);
}
static struct code_generator_result gen_return(
struct ivy_codegen *gen, struct code_generator_state *state,
struct ivy_ast_node *node, size_t depth)
{
struct unit_codegen_state *unit = (struct unit_codegen_state *)state;
struct mie_builder *builder = gen->c_builder;
struct mie_func *current = mie_builder_get_current_func(builder);
enum ivy_status status = IVY_OK;
if (!current || current != unit->s_immediate) {
status = switch_to_immediate_func(gen, unit);
}
if (status != IVY_OK) {
return CODEGEN_RESULT_ERR(status);
}
codegen_push_generator(gen, CODE_GENERATOR_RETURN, 0, NULL);
return CODEGEN_RESULT_OK(CODEGEN_R_REPEAT_NODE);
}
#if 0
static struct code_generator_result gen_var_declaration(
struct ivy_codegen *gen, struct code_generator_state *state,
@@ -272,6 +293,7 @@ struct code_generator unit_generator = {
.g_state_fini = state_fini,
.g_node_generators = {
NODE_CODEGEN(VAR, gen_var_declaration),
NODE_CODEGEN(RETURN, gen_return),
NODE_CODEGEN(UNIT_PACKAGE, gen_unit_package),
NODE_CODEGEN(UNIT_IMPORT, gen_unit_import),
},