From 3633f9a1f4f7638355beba85c53a2bb2ab0e3976 Mon Sep 17 00:00:00 2001 From: nsfisis Date: Thu, 5 Feb 2026 23:51:26 +0900 Subject: feat: implement more complex initializer --- src/codegen.c | 104 +++++++++++++++++++++++++--------------------------------- 1 file changed, 44 insertions(+), 60 deletions(-) (limited to 'src/codegen.c') diff --git a/src/codegen.c b/src/codegen.c index 2247c6c..b19dc25 100644 --- a/src/codegen.c +++ b/src/codegen.c @@ -2,6 +2,7 @@ #include #include #include "common.h" +#include "parse.h" #include "preprocess.h" typedef enum { @@ -10,6 +11,7 @@ typedef enum { } GenMode; typedef struct { + Program* prog; FILE* out; int next_label; int* loop_labels; @@ -17,8 +19,9 @@ typedef struct { int switch_label; } CodeGen; -static CodeGen* codegen_new(FILE* out) { +static CodeGen* codegen_new(Program* prog, FILE* out) { CodeGen* g = calloc(1, sizeof(CodeGen)); + g->prog = prog; g->out = out; g->next_label = 1; g->loop_labels = calloc(1024, sizeof(int)); @@ -816,8 +819,46 @@ static void codegen_func(CodeGen* g, AstNode* ast) { g->current_func = NULL; } + +static void codegen_global_var(CodeGen* g, AstNode* var) { + fprintf(g->out, " %s:\n", var->as.gvar_decl->name); + if (!var->as.gvar_decl->expr) { + fprintf(g->out, " .zero %d\n", type_sizeof(var->ty)); + return; + } + + if (var->ty->kind == TypeKind_ptr) { + if (var->as.gvar_decl->expr->kind == AstNodeKind_ref_expr) { + if (var->as.gvar_decl->expr->as.ref_expr->operand->kind != AstNodeKind_gvar) { + unimplemented(); + } + fprintf(g->out, " .quad %s\n", var->as.gvar_decl->expr->as.ref_expr->operand->as.gvar->name); + } else if (var->as.gvar_decl->expr->kind == AstNodeKind_gvar) { + fprintf(g->out, " .quad %s\n", var->as.gvar_decl->expr->as.gvar->name); + } else { + unimplemented(); + } + return; + } + if (var->ty->kind == TypeKind_array && var->as.gvar_decl->expr->kind == AstNodeKind_str_expr) { + const char* str = g->prog->str_literals[var->as.gvar_decl->expr->as.str_expr->idx - 1]; + fprintf(g->out, " .string \"%s\"\n", str); + return; + } + + StrBuilder* data_buf = calloc(1, sizeof(StrBuilder)); + strbuilder_init(data_buf); + strbuilder_reserve(data_buf, type_sizeof(var->ty)); + + eval_init_expr(data_buf, var->as.gvar_decl->expr, var->ty); + + for (size_t i = 0; i < data_buf->len; ++i) { + fprintf(g->out, " .byte %d\n", (int)data_buf->buf[i]); + } +} + void codegen(Program* prog, FILE* out) { - CodeGen* g = codegen_new(out); + CodeGen* g = codegen_new(prog, out); fprintf(g->out, ".intel_syntax noprefix\n\n"); @@ -833,64 +874,7 @@ void codegen(Program* prog, FILE* out) { fprintf(g->out, ".data\n\n"); for (int i = 0; i < prog->vars->as.list->len; ++i) { - AstNode* var = prog->vars->as.list->items + i; - fprintf(g->out, " %s:\n", var->as.gvar_decl->name); - if (var->as.gvar_decl->expr) { - if (type_sizeof(var->ty) == 1) { - if (var->as.gvar_decl->expr->kind != AstNodeKind_int_expr) - unimplemented(); - fprintf(g->out, " .byte %d\n", var->as.gvar_decl->expr->as.int_expr->value); - } else if (type_sizeof(var->ty) == 2) { - if (var->as.gvar_decl->expr->kind != AstNodeKind_int_expr) - unimplemented(); - fprintf(g->out, " .word %d\n", var->as.gvar_decl->expr->as.int_expr->value); - } else if (type_sizeof(var->ty) == 4) { - if (var->as.gvar_decl->expr->kind != AstNodeKind_int_expr) - unimplemented(); - fprintf(g->out, " .long %d\n", var->as.gvar_decl->expr->as.int_expr->value); - } else if (var->ty->kind == TypeKind_ptr) { - if (var->as.gvar_decl->expr->kind == AstNodeKind_ref_expr) { - if (var->as.gvar_decl->expr->as.ref_expr->operand->kind != AstNodeKind_gvar) { - unimplemented(); - } - fprintf(g->out, " .quad %s\n", var->as.gvar_decl->expr->as.ref_expr->operand->as.gvar->name); - } else if (var->as.gvar_decl->expr->kind == AstNodeKind_gvar) { - fprintf(g->out, " .quad %s\n", var->as.gvar_decl->expr->as.gvar->name); - } else { - unimplemented(); - } - } else if (var->ty->kind == TypeKind_array) { - if (var->as.gvar_decl->expr->kind == AstNodeKind_str_expr) { - const char* str = prog->str_literals[var->as.gvar_decl->expr->as.str_expr->idx - 1]; - fprintf(g->out, " .string \"%s\"\n", str); - } else if (var->as.gvar_decl->expr->kind == AstNodeKind_array_initializer) { - AstNode* init = var->as.gvar_decl->expr->as.array_initializer->list; - int elem_size = type_sizeof(var->ty->base); - const char* directive; - if (elem_size == 1) { - directive = ".byte"; - } else if (elem_size == 2) { - directive = ".word"; - } else if (elem_size == 4) { - directive = ".long"; - } else { - unimplemented(); - } - for (int j = 0; j < init->as.list->len; ++j) { - AstNode* elem = &init->as.list->items[j]; - if (elem->kind != AstNodeKind_int_expr) - unimplemented(); - fprintf(g->out, " %s %d\n", directive, elem->as.int_expr->value); - } - } else { - unimplemented(); - } - } else { - unimplemented(); - } - } else { - fprintf(g->out, " .zero %d\n", type_sizeof(var->ty)); - } + codegen_global_var(g, &prog->vars->as.list->items[i]); } fprintf(g->out, ".text\n\n"); -- cgit v1.3-1-g0d28