diff --git a/lang/codegen/codegen.h b/lang/codegen/codegen.h index adfa6c2..16149b4 100644 --- a/lang/codegen/codegen.h +++ b/lang/codegen/codegen.h @@ -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 { diff --git a/lang/codegen/expr.c b/lang/codegen/expr.c index 556e149..7198f1e 100644 --- a/lang/codegen/expr.c +++ b/lang/codegen/expr.c @@ -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), }, }; diff --git a/lang/codegen/for-loop.c b/lang/codegen/for-loop.c new file mode 100644 index 0000000..d0a568c --- /dev/null +++ b/lang/codegen/for-loop.c @@ -0,0 +1,70 @@ +#include "codegen.h" + +#include +#include +#include + +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), + }, +}; diff --git a/lang/codegen/generator.c b/lang/codegen/generator.c index 4297812..fba7e19 100644 --- a/lang/codegen/generator.c +++ b/lang/codegen/generator.c @@ -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];