diff --git a/compile.c b/compile.c index d2554ed..154a9e0 100644 --- a/compile.c +++ b/compile.c @@ -4389,7 +4389,6 @@ CORD compile_statement_type_header(env_t *env, ast_t *ast) auto def = Match(ast, LangDef); CORD full_name = CORD_cat(namespace_prefix(env, env->namespace), def->name); return CORD_all( - "typedef Text_t ", namespace_prefix(env, env->namespace), def->name, "_t;\n" // Constructor macro: "#define ", namespace_prefix(env, env->namespace), def->name, "(text) ((", namespace_prefix(env, env->namespace), def->name, "_t){.length=sizeof(text)-1, .tag=TEXT_ASCII, .ascii=\"\" text})\n" @@ -4494,7 +4493,28 @@ typedef struct { CORD *header; } compile_typedef_info_t; -static void _visit_statement(compile_typedef_info_t *info, ast_t *ast) +static void _make_typedefs(compile_typedef_info_t *info, ast_t *ast) +{ + if (ast->tag == StructDef) { + auto def = Match(ast, StructDef); + CORD full_name = CORD_cat(namespace_prefix(info->env, info->env->namespace), def->name); + *info->header = CORD_all(*info->header, "typedef struct ", full_name, "_s ", full_name, "_t;\n"); + } else if (ast->tag == EnumDef) { + auto def = Match(ast, EnumDef); + CORD full_name = CORD_cat(namespace_prefix(info->env, info->env->namespace), def->name); + *info->header = CORD_all(*info->header, "typedef struct ", full_name, "_s ", full_name, "_t;\n"); + + for (tag_ast_t *tag = def->tags; tag; tag = tag->next) { + if (!tag->fields) continue; + *info->header = CORD_all(*info->header, "typedef struct ", full_name, "$", tag->name, "_s ", full_name, "$", tag->name, "_t;\n"); + } + } else if (ast->tag == LangDef) { + auto def = Match(ast, LangDef); + *info->header = CORD_all(*info->header, "typedef Text_t ", namespace_prefix(info->env, info->env->namespace), def->name, "_t;\n"); + } +} + +static void _define_types(compile_typedef_info_t *info, ast_t *ast) { *info->header = CORD_all(*info->header, compile_statement_type_header(info->env, ast), @@ -4509,7 +4529,8 @@ CORD compile_file_header(env_t *env, ast_t *ast) "#include \n"); compile_typedef_info_t info = {.env=env, .header=&header}; - visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_visit_statement, &info}); + visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_make_typedefs, &info}); + visit_topologically(Match(ast, Block)->statements, (Closure_t){.fn=(void*)_define_types, &info}); header = CORD_all(header, "void _$", env->namespace->name, "$$initialize(void);\n"); return header; diff --git a/enums.c b/enums.c index 354d975..25e5533 100644 --- a/enums.c +++ b/enums.c @@ -68,7 +68,7 @@ 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_all("typedef struct ", full_name, "_s ", full_name, "_t;\n"); + CORD all_defs = CORD_EMPTY; CORD enum_def = CORD_all("struct ", full_name, "_s {\n" "\tenum { ", full_name, "$null=0, "); diff --git a/structs.c b/structs.c index 94a04de..7c65267 100644 --- a/structs.c +++ b/structs.c @@ -62,7 +62,6 @@ CORD compile_struct_header(env_t *env, ast_t *ast) struct_code = CORD_all(struct_code, "};\n"); type_t *t = Table$str_get(*env->types, def->name); return CORD_all( - "typedef struct ", full_name, "_s ", full_name, "_t;\n", "struct ", full_name, "_s {\n", fields, "};\n",