43 lines
1.2 KiB
C
43 lines
1.2 KiB
C
#include "codegen.h"
|
|
|
|
extern const struct code_generator unit_generator;
|
|
extern const struct code_generator expr_generator;
|
|
extern const struct code_generator var_generator;
|
|
extern const struct code_generator msg_generator;
|
|
|
|
static const struct code_generator *code_generators[] = {
|
|
[CODE_GENERATOR_UNIT] = &unit_generator,
|
|
[CODE_GENERATOR_EXPR] = &expr_generator,
|
|
[CODE_GENERATOR_VAR] = &var_generator,
|
|
[CODE_GENERATOR_MSG] = &msg_generator,
|
|
};
|
|
static const size_t nr_code_generators
|
|
= sizeof code_generators / sizeof code_generators[0];
|
|
|
|
static const struct code_generator *root_code_generators[] = {
|
|
[IVY_AST_UNIT] = &unit_generator,
|
|
[IVY_AST_OP] = &expr_generator,
|
|
[IVY_AST_INT] = &expr_generator,
|
|
[IVY_AST_IDENT] = &expr_generator,
|
|
};
|
|
static const size_t nr_root_code_generators
|
|
= sizeof root_code_generators / sizeof root_code_generators[0];
|
|
|
|
const struct code_generator *get_code_generator(enum code_generator_type type)
|
|
{
|
|
if (type >= nr_code_generators) {
|
|
return NULL;
|
|
}
|
|
|
|
return code_generators[type];
|
|
}
|
|
|
|
const struct code_generator *get_root_code_generator(enum ivy_ast_node_type type)
|
|
{
|
|
if (type >= nr_root_code_generators) {
|
|
return NULL;
|
|
}
|
|
|
|
return root_code_generators[type];
|
|
}
|