diff options
| -rw-r--r-- | src/codegen.c | 97 |
1 files changed, 57 insertions, 40 deletions
diff --git a/src/codegen.c b/src/codegen.c index 1db5625..a3a1b87 100644 --- a/src/codegen.c +++ b/src/codegen.c @@ -10,6 +10,7 @@ #include "common.h" #include "errors.h" #include "parser.h" +#include "types.h" static size_t codegenstmt(LLVMBuilderRef, struct ast, struct lexemes, size_t); @@ -24,48 +25,64 @@ codegen(struct ast ast, struct lexemes toks) LLVMBuilderRef builder = LLVMCreateBuilder(); for (size_t i = 0; i < ast.len;) { - if (ast.kinds[i] != ASTCDECL) - err("codegen: Expected constant declaration"); - - size_t expr = ast.kids[i].rhs; - if (ast.kinds[expr] != ASTFN) { - assert(!"not implemented"); - __builtin_unreachable(); + switch (ast.kinds[i]) { + case ASTDECL: { + /* TODO: Temporary allocator */ + struct strview sv = toks.strs[ast.lexemes[i]]; + char *name = bufalloc(NULL, sv.len + 1, 1); + ((uchar *)memcpy(name, sv.p, sv.len))[sv.len] = 0; + + LLVMValueRef globl, val; + globl = LLVMAddGlobal(mod, LLVMInt64Type(), name); + i = codegenexpr(builder, ast, toks, ast.kids[i].rhs, &val); + LLVMSetInitializer(globl, val); + free(name); + break; } - - size_t proto = ast.kids[expr].lhs, body = ast.kids[expr].rhs; - - LLVMTypeRef ret; - LLVMTypeRef params[] = {0}; - - if (ast.kids[proto].rhs == AST_EMPTY) - ret = LLVMVoidType(); - else { - size_t type = ast.kids[proto].rhs; - struct strview sv = toks.strs[ast.lexemes[type]]; - - /* TODO: Make int 32bit on 32bit platforms */ - if (strncmp("int", sv.p, sv.len) == 0) - ret = LLVMInt64Type(); - else - err("codegen: Unknown type: %.*s", (int)sv.len, sv.p); + case ASTCDECL: { + idx_t_ expr = ast.kids[i].rhs; + if (ast.kinds[expr] != ASTFN) { + assert(!"not implemented"); + __builtin_unreachable(); + } + + idx_t_ proto = ast.kids[expr].lhs, body = ast.kids[expr].rhs; + + LLVMTypeRef ret; + LLVMTypeRef params[] = {0}; + + if (ast.kids[proto].rhs == AST_EMPTY) + ret = LLVMVoidType(); + else { + size_t type = ast.kids[proto].rhs; + struct strview sv = toks.strs[ast.lexemes[type]]; + + /* TODO: Make int 32bit on 32bit platforms */ + if (strncmp("int", sv.p, sv.len) == 0) + ret = LLVMInt64Type(); + else + err("codegen: Unknown type: %.*s", (int)sv.len, sv.p); + } + + LLVMTypeRef fnproto = LLVMFunctionType(ret, params, 0, false); + + struct strview sv = toks.strs[ast.lexemes[i]]; + char *fnname = bufalloc(NULL, sv.len + 1, 1); + ((char *)memcpy(fnname, sv.p, sv.len))[sv.len] = 0; + + LLVMValueRef fn = LLVMAddFunction(mod, fnname, fnproto); + LLVMBasicBlockRef entry = LLVMAppendBasicBlock(fn, "entry"); + LLVMPositionBuilderAtEnd(builder, entry); + + free(fnname); + + for (i = ast.kids[body].lhs; i <= ast.kids[body].rhs;) + i = codegenstmt(builder, ast, toks, i); + break; + } + default: + err("codegen: Expected declaration"); } - - LLVMTypeRef fnproto = LLVMFunctionType(ret, params, 0, false); - - struct strview sv = toks.strs[ast.lexemes[i]]; - char *fnname = bufalloc(NULL, sv.len + 1, 1); - ((char *)memcpy(fnname, sv.p, sv.len))[sv.len] = 0; - - LLVMValueRef fn = LLVMAddFunction(mod, fnname, fnproto); - LLVMBasicBlockRef entry = LLVMAppendBasicBlock(fn, "entry"); - LLVMPositionBuilderAtEnd(builder, entry); - - free(fnname); - - for (i = ast.kids[body].lhs; i <= ast.kids[body].rhs;) - i = codegenstmt(builder, ast, toks, i); - } LLVMDisposeBuilder(builder); |