From 0084f362a29363ae25fdf36f414d5233ccae1dcd Mon Sep 17 00:00:00 2001 From: Thomas Voss Date: Fri, 21 Jun 2024 02:25:24 +0200 Subject: Cleanup codegen.c --- make.c | 2 - src/codegen.c | 208 ++++++---------------------------------------------------- src/codegen.h | 3 +- src/main.c | 2 +- 4 files changed, 23 insertions(+), 192 deletions(-) diff --git a/make.c b/make.c index 65d84d1..4a21735 100644 --- a/make.c +++ b/make.c @@ -147,8 +147,6 @@ main(int argc, char **argv) void cc(void *arg) { - if (strcmp(arg, "src/codegen.c") == 0) - return; if (!tagvalid(arg)) return; diff --git a/src/codegen.c b/src/codegen.c index f04be03..145ce6e 100644 --- a/src/codegen.c +++ b/src/codegen.c @@ -1,238 +1,70 @@ -#include #include -#include -#include #include -#include #include #include #include "alloc.h" -#include "codegen.h" +#include "analyzer.h" #include "common.h" #include "errors.h" -#include "parser.h" -#include "types.h" +#include "strview.h" -typedef mpq_t constval; - -typedef struct constmap { - struct constmap *child[4]; - struct strview key; - constval val; -} constmap; +#define lengthof(xs) (sizeof(xs) / sizeof(*(xs))) /* A context structure we can pass to all the codegen functions just so they have easy access to everything */ struct cgctx { + arena a; LLVMContextRef ctx; LLVMModuleRef mod; LLVMBuilderRef bob; struct strview namespace; - constmap *consts; - arena a; }; -static void codegenprog(struct cgctx, struct type *, struct ast, - struct lexemes) - __attribute__((nonnull)); -// static size_t codegenexpr(struct cgctx, struct type *, struct ast, -// struct lexemes, size_t, LLVMValueRef *) -// __attribute__((nonnull)); - -static LLVMTypeRef type2llvm(struct type); - -static bool constmap_equals(struct strview, struct strview); -static uint64_t constmap_hash(struct strview); -static constval *constmap_upsert(constmap **, struct strview, arena *) - __attribute__((nonnull(1))); - -/* TODO: Don’t do this? */ -#define lengthof(xs) (sizeof(xs) / sizeof(*(xs))) +static LLVMTypeRef type2llvm(struct cgctx, struct type); void codegen(const char *file, struct type *types, struct ast ast, struct lexemes toks) { - struct cgctx ctx = {0}; - ctx.ctx = LLVMContextCreate(); - ctx.mod = LLVMModuleCreateWithNameInContext("oryx", ctx.ctx); - ctx.bob = LLVMCreateBuilderInContext(ctx.ctx); + struct cgctx ctx; + ctx.a = NULL; + ctx.namespace.p = NULL; + ctx.ctx = LLVMContextCreate(); + ctx.mod = LLVMModuleCreateWithNameInContext("oryx", ctx.ctx); + ctx.bob = LLVMCreateBuilderInContext(ctx.ctx); LLVMSetSourceFileName(ctx.mod, file, strlen(file)); - codegenprog(ctx, types, ast, toks); - arena_free(&ctx.a); LLVMDisposeBuilder(ctx.bob); char *error = NULL; - LLVMVerifyModule(ctx.mod, LLVMAbortProcessAction, &error); - LLVMDisposeMessage(error); + if (LLVMVerifyModule(ctx.mod, LLVMReturnStatusAction, &error) == 1) + err("codegen: %s", error); LLVMDumpModule(ctx.mod); LLVMDisposeModule(ctx.mod); LLVMContextDispose(ctx.ctx); } -void -interp_const_expr(struct cgctx ctx, struct type *types, struct ast ast, - struct lexemes toks, idx_t_ i, constval v) -{ - switch (ast.kinds[i]) { - case ASTNUMLIT: { - mpf_t f; - mpq_init(v); - - struct strview sv = toks.strs[ast.lexemes[i]]; - char *buf = bufalloc(NULL, sv.len + 1, 1); - memcpy(buf, sv.p, sv.len); - buf[sv.len] = 0; - - if (mpf_init_set_str(f, buf, 10) == -1) - err("mpf_init_set_str: Invalid input ‘%s’", buf); - free(buf); - - mpq_set_f(v, f); - mpf_clear(f); - break; - } - case ASTIDENT: { - struct strview sv = toks.strs[ast.lexemes[i]]; - constval *w = constmap_upsert(&ctx.consts, sv, NULL); - v = *w; - break; - } - default: - err("codegen: Not implemented"); - } -} - -void -register_const(struct cgctx ctx, struct type *types, struct ast ast, - struct lexemes toks, idx_t_ i) -{ - struct pair p = ast.kids[i]; - assert(p.rhs != AST_EMPTY); - - constval v; - interp_const_expr(ctx, types, ast, toks, p.rhs, v); - - struct strview sv = toks.strs[ast.lexemes[i]]; - *constmap_upsert(&ctx.consts, sv, &ctx.a) = v; - - /* TODO: Remove */ - printf("%.*s = ", (int)sv.len, sv.p); - mpq_out_str(stdout, 10, v); - putchar('\n'); - mpq_clear(v); -} - -void -codegenprog(struct cgctx ctx, struct type *types, struct ast ast, - struct lexemes toks) -{ - for (size_t i = 0; i < ast.len; i = fwdnode(ast, i)) { - if ((ast.kinds[i] | 1) == ASTPCDECL) - register_const(ctx, types, ast, toks, i); - - // mpq_t v = interp_const_expr(ctx, ast, toks, p.rhs); - // constmap_upsert(&ctx.consts, ast.lexemes[p.lhs], &ctx.a); - } - - // for (size_t i = 0; i < ast.len;) { - // assert(ast.kinds[i] <= _AST_DECLS_END); - // i = codegenprog(ctx, types, ast, toks, i); - // } -} - -// size_t -// codegenexpr(struct cgctx ctx, struct type *types, struct ast ast, -// struct lexemes toks, size_t i, LLVMValueRef *v) -// { -// (void)ctx; -// switch (ast.kinds[i]) { -// case ASTNUMLIT: { -// /* TODO: Arbitrary precision? */ -// struct strview sv = toks.strs[ast.lexemes[i]]; -// -// bool has_sep = memchr(sv.p, '\'', sv.len) != NULL; -// -// /* TODO: Temporary one-time-use allocator? */ -// if (has_sep) { -// size_t len = 0; -// char *p = bufalloc(NULL, sv.len, 1); -// for (size_t i = 0; i < sv.len; i++) { -// if (sv.p[i] != '\'') -// p[len++] = sv.p[i]; -// } -// -// *v = LLVMConstIntOfStringAndSize(type2llvm(types[i]), p, len, 10); -// free(p); -// } else { -// *v = LLVMConstIntOfStringAndSize(type2llvm(types[i]), sv.p, sv.len, -// 10); -// } -// return i + 1; -// } -// case ASTIDENT: -// err("codegen: %s: Not implemented", __func__); -// default: -// assert(!"unreachable"); -// __builtin_unreachable(); -// } -// } - LLVMTypeRef -type2llvm(struct type t) +type2llvm(struct cgctx ctx, struct type t) { switch (t.kind) { - case TYPE_UNSET: - case TYPE_CHECKING: - assert(!"codegen: Hit TYPE_UNSET or TYPE_CHECKING"); - __builtin_unreachable(); case TYPE_FN: - err("codegen: %s: Not implemented", __func__); + err("codegen: %s: Not implemented for function types", __func__); case TYPE_NUM: /* TODO: Floats */ + if (t.isfloat) + err("codegen: %s: Not implemented for floats", __func__); /* TODO: Arbitrary precision */ if (t.size == 0) - return LLVMInt64Type(); - return LLVMIntType((unsigned)t.size * 8); + return LLVMInt64TypeInContext(ctx.ctx); + assert((unsigned)t.size * 8 <= UINT8_MAX); + return LLVMIntTypeInContext(ctx.ctx, t.size * 8); default: __builtin_unreachable(); } } - -bool -constmap_equals(struct strview x, struct strview y) -{ - return x.len == y.len && memcmp(x.p, y.p, x.len) == 0; -} - -uint64_t -constmap_hash(struct strview sv) -{ - uint64_t h = 0x100; - for (size_t i = 0; i < sv.len; i++) { - h ^= sv.p[i]; - h *= 1111111111111111111u; - } - return h; -} - -constval * -constmap_upsert(constmap **m, struct strview key, arena *a) -{ - for (uint64_t h = constmap_hash(key); *m; h <<= 2) { - if (constmap_equals(key, (*m)->key)) - return &(*m)->val; - m = &(*m)->child[h >> 62]; - } - if (a == NULL) - return NULL; - *m = arena_new(a, constmap, 1); - (*m)->key = key; - return &(*m)->val; -} diff --git a/src/codegen.h b/src/codegen.h index 74bd4fb..eda799d 100644 --- a/src/codegen.h +++ b/src/codegen.h @@ -5,7 +5,8 @@ #include "lexer.h" #include "parser.h" -void codegen(const char *, struct type *, struct ast, struct lexemes) +void codegen(const char *, struct scope *, struct type *, struct ast, + struct lexemes) __attribute__((nonnull)); #endif /* !ORYX_CODEGEN_H */ diff --git a/src/main.c b/src/main.c index 05228d7..bc7cc7d 100644 --- a/src/main.c +++ b/src/main.c @@ -34,7 +34,7 @@ main(int argc, char **argv) struct lexemes toks = lexstring(src, srclen); struct ast ast = parsetoks(toks); analyzeprog(ast, toks, &a, &types, &scps); - // codegen(argv[1], types, ast, toks); + codegen(argv[1], scps, types, ast, toks); #if DEBUG free(scps); -- cgit v1.2.3