wash/ivy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lang: codegen: add for-loop generator

Browse Source
This commit is contained in:
max 🍓
2025-05-08 10:53:00 +01:00
parent c31638c3de
commit 475575f76f
4 changed files with 82 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lang/codegen/codegen.h
View File

@@ -47,6 +47,7 @@ enum code_generator_type {
CODE_GENERATOR_RETURN,
CODE_GENERATOR_COND_GROUP,
CODE_GENERATOR_COND,
CODE_GENERATOR_FOR_LOOP,
};
enum code_generator_scope_type {

9
lang/codegen/expr.c
View File

@@ -229,6 +229,14 @@ static struct code_generator_result gen_cond_group(
return CODEGEN_RESULT_OK(CODEGEN_R_REPEAT_NODE);
}
static struct code_generator_result gen_for_loop(
struct ivy_codegen *gen, struct code_generator_state *state,
struct ivy_ast_node *node, size_t depth)
{
codegen_push_generator(gen, CODE_GENERATOR_FOR_LOOP, 0, NULL);
return CODEGEN_RESULT_OK(CODEGEN_R_REPEAT_NODE);
}
#if 0
static struct code_generator_result gen_var_ref(
struct ivy_codegen *gen, struct code_generator_state *state,
@@ -504,5 +512,6 @@ struct code_generator expr_generator = {
NODE_CODEGEN(FSTRING, gen_fstring),
NODE_CODEGEN(IDENT, gen_var_reference),
NODE_CODEGEN(COND_GROUP, gen_cond_group),
NODE_CODEGEN(FOR_LOOP, gen_for_loop),
},
};

70
lang/codegen/for-loop.c Normal file
View File

@@ -0,0 +1,70 @@
#include "codegen.h"
#include <mie/block.h>
#include <mie/func.h>
#include <mie/module.h>
struct for_codegen_state {
struct code_generator_state s_base;
struct mie_value *s_iterator;
struct mie_value *s_container;
};
static struct code_generator_result gen_expr(
struct ivy_codegen *gen, struct code_generator_state *state,
struct ivy_ast_node *node, size_t depth)
{
struct for_codegen_state *for_loop = (struct for_codegen_state *)state;
codegen_push_generator(gen, CODE_GENERATOR_EXPR, 0, NULL);
return CODEGEN_RESULT_OK(CODEGEN_R_REPEAT_NODE);
}
static struct code_generator_result gen_block(
struct ivy_codegen *gen, struct code_generator_state *state,
struct ivy_ast_node *node, size_t depth)
{
codegen_push_generator(gen, CODE_GENERATOR_BLOCK, 0, NULL);
return CODEGEN_RESULT_OK(0);
}
static enum ivy_status state_init(
struct ivy_codegen *gen, struct code_generator_state *state,
uintptr_t argv, void *argp)
{
return IVY_OK;
}
static enum ivy_status state_fini(
struct ivy_codegen *gen, struct code_generator_state *state,
struct mie_value **result)
{
return IVY_OK;
}
static struct code_generator_result value_received(
struct ivy_codegen *gen, struct code_generator_state *state,
struct mie_value *value)
{
struct for_codegen_state *for_loop = (struct for_codegen_state *)state;
if (!for_loop->s_iterator) {
for_loop->s_iterator = value;
} else if (!for_loop->s_container) {
for_loop->s_container = value;
}
return CODEGEN_RESULT_OK(0);
}
struct code_generator for_loop_generator = {
.g_type = CODE_GENERATOR_FOR_LOOP,
.g_state_size = sizeof(struct for_codegen_state),
.g_state_init = state_init,
.g_state_fini = state_fini,
.g_value_received = value_received,
.g_expr_generator = gen_expr,
.g_node_generators = {
NODE_CODEGEN(BLOCK, gen_block),
},
};

2
lang/codegen/generator.c
View File

@@ -12,6 +12,7 @@ extern const struct code_generator lambda_generator;
extern const struct code_generator return_generator;
extern const struct code_generator cond_generator;
extern const struct code_generator cond_group_generator;
extern const struct code_generator for_loop_generator;
static const struct code_generator *code_generators[] = {
[CODE_GENERATOR_UNIT] = &unit_generator,
@@ -24,6 +25,7 @@ static const struct code_generator *code_generators[] = {
[CODE_GENERATOR_RETURN] = &return_generator,
[CODE_GENERATOR_COND] = &cond_generator,
[CODE_GENERATOR_COND_GROUP] = &cond_group_generator,
[CODE_GENERATOR_FOR_LOOP] = &for_loop_generator,
};
static const size_t nr_code_generators
= sizeof code_generators / sizeof code_generators[0];