Move files into src/ and build into build/

1 files changed, 129 insertions, 0 deletions

diff --git a/src/enums.c b/src/enums.c
new file mode 100644
index 00000000..06af0df3
--- /dev/null
+++ b/src/enums.c
@@ -0,0 +1,129 @@
+// Logic for compiling tagged unions (enums)
+#include <ctype.h>
+#include <gc/cord.h>
+#include <gc.h>
+#include <stdio.h>
+
+#include "ast.h"
+#include "stdlib/text.h"
+#include "compile.h"
+#include "cordhelpers.h"
+#include "structs.h"
+#include "environment.h"
+#include "typecheck.h"
+#include "stdlib/util.h"
+
+CORD compile_enum_typeinfo(env_t *env, ast_t *ast)
+{
+    auto def = Match(ast, EnumDef);
+    CORD full_name = CORD_cat(namespace_prefix(env, env->namespace), def->name);
+
+    // Compile member types and constructors:
+    CORD member_typeinfos = CORD_EMPTY;
+    for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+        if (!tag->fields) continue;
+
+        const char *tag_name = heap_strf("%s$%s", def->name, tag->name);
+        type_t *tag_type = Table$str_get(*env->types, tag_name);
+        assert(tag_type && tag_type->tag == StructType);
+        member_typeinfos = CORD_all(
+            member_typeinfos, compile_struct_typeinfo(env, tag_type, tag_name, tag->fields, tag->secret, false));
+    }
+
+    int num_tags = 0;
+    for (tag_ast_t *t = def->tags; t; t = t->next)
+        num_tags += 1;
+
+    type_t *t = Table$str_get(*env->types, def->name);
+    const char *metamethods = is_packed_data(t) ? "PackedDataEnum$metamethods" : "Enum$metamethods";
+    CORD typeinfo = CORD_asprintf("public const TypeInfo_t %r$$info = {%zu, %zu, .metamethods=%s, {.tag=EnumInfo, .EnumInfo={.name=\"%s\", "
+                                  ".num_tags=%d, .tags=(NamedType_t[]){",
+                                  full_name, type_size(t), type_align(t), metamethods, def->name, num_tags);
+
+    for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+        const char *tag_type_name = heap_strf("%s$%s", def->name, tag->name);
+        type_t *tag_type = Table$str_get(*env->types, tag_type_name);
+        if (tag_type && Match(tag_type, StructType)->fields)
+            typeinfo = CORD_all(typeinfo, "{\"", tag->name, "\", ", compile_type_info(tag_type), "}, ");
+        else
+            typeinfo = CORD_all(typeinfo, "{\"", tag->name, "\"}, ");
+    }
+    typeinfo = CORD_all(typeinfo, "}}}};\n");
+    return CORD_all(member_typeinfos, typeinfo);
+}
+
+CORD compile_enum_constructors(env_t *env, ast_t *ast)
+{
+    auto def = Match(ast, EnumDef);
+    CORD full_name = CORD_cat(namespace_prefix(env, env->namespace), def->name);
+
+    CORD constructors = CORD_EMPTY;
+    for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+        if (!tag->fields) continue;
+
+        CORD arg_sig = CORD_EMPTY;
+        for (arg_ast_t *field = tag->fields; field; field = field->next) {
+            type_t *field_t = get_arg_ast_type(env, field);
+            arg_sig = CORD_all(arg_sig, compile_declaration(field_t, CORD_all("$", field->name)));
+            if (field->next) arg_sig = CORD_cat(arg_sig, ", ");
+        }
+        if (arg_sig == CORD_EMPTY) arg_sig = "void";
+        CORD constructor_impl = CORD_all("public inline ", full_name, "$$type ", full_name, "$tagged$", tag->name, "(", arg_sig, ") { return (",
+                                         full_name, "$$type){.$tag=", full_name, "$tag$", tag->name, ", .", tag->name, "={");
+        for (arg_ast_t *field = tag->fields; field; field = field->next) {
+            constructor_impl = CORD_all(constructor_impl, "$", field->name);
+            if (field->next) constructor_impl = CORD_cat(constructor_impl, ", ");
+        }
+        constructor_impl = CORD_cat(constructor_impl, "}}; }\n");
+        constructors = CORD_cat(constructors, constructor_impl);
+    }
+    return constructors;
+}
+
+CORD compile_enum_header(env_t *env, ast_t *ast)
+{
+    auto def = Match(ast, EnumDef);
+    CORD full_name = CORD_all(namespace_prefix(env, env->namespace), def->name);
+    CORD all_defs = CORD_EMPTY;
+    CORD enum_def = CORD_all("struct ", full_name, "$$struct {\n"
+                             "\tenum { ", full_name, "$null=0, ");
+
+    bool has_any_tags_with_fields = false;
+    for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+        enum_def = CORD_all(enum_def, full_name, "$tag$", tag->name);
+        if (tag->next) enum_def = CORD_all(enum_def, ", ");
+        has_any_tags_with_fields = has_any_tags_with_fields || (tag->fields != NULL);
+    }
+    enum_def = CORD_all(enum_def, "} $tag;\n");
+
+    if (has_any_tags_with_fields) {
+        enum_def = CORD_all(enum_def, "union {\n");
+        for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+            if (!tag->fields) continue;
+            CORD field_def = compile_struct_header(env, WrapAST(ast, StructDef, .name=CORD_to_const_char_star(CORD_all(def->name, "$", tag->name)), .fields=tag->fields));
+            all_defs = CORD_all(all_defs, field_def);
+            enum_def = CORD_all(enum_def, full_name, "$", tag->name, "$$type ", tag->name, ";\n");
+        }
+        enum_def = CORD_all(enum_def, "};\n");
+    }
+    enum_def = CORD_all(enum_def, "};\n");
+    all_defs = CORD_all(all_defs, enum_def);
+
+    all_defs = CORD_all(all_defs, "extern const TypeInfo_t ", full_name, "$$info;\n");
+    for (tag_ast_t *tag = def->tags; tag; tag = tag->next) {
+        if (!tag->fields) continue;
+
+        CORD arg_sig = CORD_EMPTY;
+        for (arg_ast_t *field = tag->fields; field; field = field->next) {
+            type_t *field_t = get_arg_ast_type(env, field);
+            arg_sig = CORD_all(arg_sig, compile_declaration(field_t, CORD_all("$", field->name)));
+            if (field->next) arg_sig = CORD_all(arg_sig, ", ");
+        }
+        if (arg_sig == CORD_EMPTY) arg_sig = "void";
+        CORD constructor_def = CORD_all(full_name, "$$type ", full_name, "$tagged$", tag->name, "(", arg_sig, ");\n");
+        all_defs = CORD_all(all_defs, constructor_def);
+    }
+    return all_defs;
+}
+
+// vim: ts=4 sw=0 et cino=L2,l1,(0,W4,m1,\:0