tilde.cpp 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842
  1. #include "tilde.hpp"
  2. gb_global Slice<TB_Arena> global_tb_arenas;
  3. gb_internal TB_Arena *cg_arena(void) {
  4. return &global_tb_arenas[current_thread_index()];
  5. }
  6. gb_internal void cg_global_arena_init(void) {
  7. global_tb_arenas = slice_make<TB_Arena>(permanent_allocator(), global_thread_pool.threads.count);
  8. for_array(i, global_tb_arenas) {
  9. tb_arena_create(&global_tb_arenas[i], 2ull<<20);
  10. }
  11. }
  12. // returns TB_TYPE_VOID if not trivially possible
  13. gb_internal TB_DataType cg_data_type(Type *t) {
  14. GB_ASSERT(t != nullptr);
  15. t = core_type(t);
  16. i64 sz = type_size_of(t);
  17. switch (t->kind) {
  18. case Type_Basic:
  19. switch (t->Basic.kind) {
  20. case Basic_bool:
  21. case Basic_b8:
  22. case Basic_b16:
  23. case Basic_b32:
  24. case Basic_b64:
  25. case Basic_i8:
  26. case Basic_u8:
  27. case Basic_i16:
  28. case Basic_u16:
  29. case Basic_i32:
  30. case Basic_u32:
  31. case Basic_i64:
  32. case Basic_u64:
  33. case Basic_i128:
  34. case Basic_u128:
  35. case Basic_rune:
  36. case Basic_int:
  37. case Basic_uint:
  38. case Basic_uintptr:
  39. case Basic_typeid:
  40. return TB_TYPE_INTN(cast(u16)gb_min(8*sz, 64));
  41. case Basic_f16: return TB_TYPE_F16;
  42. case Basic_f32: return TB_TYPE_F32;
  43. case Basic_f64: return TB_TYPE_F64;
  44. case Basic_rawptr: return TB_TYPE_PTR;
  45. case Basic_cstring: return TB_TYPE_PTR;
  46. // Endian Specific Types
  47. case Basic_i16le:
  48. case Basic_u16le:
  49. case Basic_i32le:
  50. case Basic_u32le:
  51. case Basic_i64le:
  52. case Basic_u64le:
  53. case Basic_i128le:
  54. case Basic_u128le:
  55. case Basic_i16be:
  56. case Basic_u16be:
  57. case Basic_i32be:
  58. case Basic_u32be:
  59. case Basic_i64be:
  60. case Basic_u64be:
  61. case Basic_i128be:
  62. case Basic_u128be:
  63. return TB_TYPE_INTN(cast(u16)gb_min(8*sz, 64));
  64. case Basic_f16le: return TB_TYPE_F16;
  65. case Basic_f32le: return TB_TYPE_F32;
  66. case Basic_f64le: return TB_TYPE_F64;
  67. case Basic_f16be: return TB_TYPE_F16;
  68. case Basic_f32be: return TB_TYPE_F32;
  69. case Basic_f64be: return TB_TYPE_F64;
  70. }
  71. break;
  72. case Type_Pointer:
  73. case Type_MultiPointer:
  74. case Type_Proc:
  75. return TB_TYPE_PTR;
  76. case Type_BitSet:
  77. return cg_data_type(bit_set_to_int(t));
  78. case Type_RelativePointer:
  79. return cg_data_type(t->RelativePointer.base_integer);
  80. }
  81. // unknown
  82. return {};
  83. }
  84. gb_internal cgValue cg_value(TB_Global *g, Type *type) {
  85. return cg_value((TB_Symbol *)g, type);
  86. }
  87. gb_internal cgValue cg_value(TB_External *e, Type *type) {
  88. return cg_value((TB_Symbol *)e, type);
  89. }
  90. gb_internal cgValue cg_value(TB_Function *f, Type *type) {
  91. return cg_value((TB_Symbol *)f, type);
  92. }
  93. gb_internal cgValue cg_value(TB_Symbol *s, Type *type) {
  94. cgValue v = {};
  95. v.kind = cgValue_Symbol;
  96. v.type = type;
  97. v.symbol = s;
  98. return v;
  99. }
  100. gb_internal cgValue cg_value(TB_Node *node, Type *type) {
  101. cgValue v = {};
  102. v.kind = cgValue_Value;
  103. v.type = type;
  104. v.node = node;
  105. return v;
  106. }
  107. gb_internal cgValue cg_lvalue_addr(TB_Node *node, Type *type) {
  108. GB_ASSERT(node->dt.type == TB_PTR);
  109. cgValue v = {};
  110. v.kind = cgValue_Addr;
  111. v.type = type;
  112. v.node = node;
  113. return v;
  114. }
  115. gb_internal cgValue cg_lvalue_addr_to_value(cgValue v) {
  116. if (v.kind == cgValue_Value) {
  117. GB_ASSERT(is_type_pointer(v.type));
  118. GB_ASSERT(v.node->dt.type == TB_PTR);
  119. } else {
  120. GB_ASSERT(v.kind == cgValue_Addr);
  121. GB_ASSERT(v.node->dt.type == TB_PTR);
  122. v.kind = cgValue_Value;
  123. v.type = alloc_type_pointer(v.type);
  124. }
  125. return v;
  126. }
  127. gb_internal cgValue cg_value_multi(cgValueMulti *multi, Type *type) {
  128. GB_ASSERT(type->kind == Type_Tuple);
  129. GB_ASSERT(multi != nullptr);
  130. GB_ASSERT(type->Tuple.variables.count > 1);
  131. GB_ASSERT(multi->values.count == type->Tuple.variables.count);
  132. cgValue v = {};
  133. v.kind = cgValue_Multi;
  134. v.type = type;
  135. v.multi = multi;
  136. return v;
  137. }
  138. gb_internal cgValue cg_value_multi(Slice<cgValue> const &values, Type *type) {
  139. cgValueMulti *multi = gb_alloc_item(permanent_allocator(), cgValueMulti);
  140. multi->values = values;
  141. return cg_value_multi(multi, type);
  142. }
  143. gb_internal cgValue cg_value_multi2(cgValue const &x, cgValue const &y, Type *type) {
  144. GB_ASSERT(type->kind == Type_Tuple);
  145. GB_ASSERT(type->Tuple.variables.count == 2);
  146. cgValueMulti *multi = gb_alloc_item(permanent_allocator(), cgValueMulti);
  147. multi->values = slice_make<cgValue>(permanent_allocator(), 2);
  148. multi->values[0] = x;
  149. multi->values[1] = y;
  150. return cg_value_multi(multi, type);
  151. }
  152. gb_internal cgAddr cg_addr(cgValue const &value) {
  153. GB_ASSERT(value.kind != cgValue_Multi);
  154. cgAddr addr = {};
  155. addr.kind = cgAddr_Default;
  156. addr.addr = value;
  157. if (addr.addr.kind == cgValue_Addr) {
  158. GB_ASSERT(addr.addr.node != nullptr);
  159. addr.addr.kind = cgValue_Value;
  160. addr.addr.type = alloc_type_pointer(addr.addr.type);
  161. }
  162. return addr;
  163. }
  164. gb_internal cgAddr cg_addr_map(cgValue addr, cgValue map_key, Type *map_type, Type *map_result) {
  165. GB_ASSERT(is_type_pointer(addr.type));
  166. Type *mt = type_deref(addr.type);
  167. GB_ASSERT(is_type_map(mt));
  168. cgAddr v = {cgAddr_Map, addr};
  169. v.map.key = map_key;
  170. v.map.type = map_type;
  171. v.map.result = map_result;
  172. return v;
  173. }
  174. gb_internal cgAddr cg_addr_soa_variable(cgValue addr, cgValue index, Ast *index_expr) {
  175. cgAddr v = {cgAddr_SoaVariable, addr};
  176. v.soa.index = index;
  177. v.soa.index_expr = index_expr;
  178. return v;
  179. }
  180. gb_internal void cg_set_debug_pos_from_node(cgProcedure *p, Ast *node) {
  181. if (node) {
  182. TokenPos pos = ast_token(node).pos;
  183. TB_SourceFile **file = map_get(&p->module->file_id_map, cast(uintptr)pos.file_id);
  184. if (file) {
  185. tb_inst_location(p->func, *file, pos.line, pos.column);
  186. }
  187. }
  188. }
  189. gb_internal void cg_add_symbol(cgModule *m, Entity *e, TB_Symbol *symbol) {
  190. if (e) {
  191. rw_mutex_lock(&m->values_mutex);
  192. map_set(&m->symbols, e, symbol);
  193. rw_mutex_unlock(&m->values_mutex);
  194. }
  195. }
  196. gb_internal void cg_add_entity(cgModule *m, Entity *e, cgValue const &val) {
  197. if (e) {
  198. rw_mutex_lock(&m->values_mutex);
  199. GB_ASSERT(val.node != nullptr);
  200. map_set(&m->values, e, val);
  201. rw_mutex_unlock(&m->values_mutex);
  202. }
  203. }
  204. gb_internal void cg_add_member(cgModule *m, String const &name, cgValue const &val) {
  205. if (name.len > 0) {
  206. rw_mutex_lock(&m->values_mutex);
  207. string_map_set(&m->members, name, val);
  208. rw_mutex_unlock(&m->values_mutex);
  209. }
  210. }
  211. gb_internal void cg_add_procedure_value(cgModule *m, cgProcedure *p) {
  212. rw_mutex_lock(&m->values_mutex);
  213. if (p->entity != nullptr) {
  214. map_set(&m->procedure_values, p->func, p->entity);
  215. if (p->symbol != nullptr) {
  216. map_set(&m->symbols, p->entity, p->symbol);
  217. }
  218. }
  219. string_map_set(&m->procedures, p->name, p);
  220. rw_mutex_unlock(&m->values_mutex);
  221. }
  222. gb_internal TB_Symbol *cg_find_symbol_from_entity(cgModule *m, Entity *e) {
  223. GB_ASSERT(e != nullptr);
  224. rw_mutex_lock(&m->values_mutex);
  225. TB_Symbol **found = map_get(&m->symbols, e);
  226. if (found) {
  227. rw_mutex_unlock(&m->values_mutex);
  228. return *found;
  229. }
  230. String link_name = cg_get_entity_name(m, e);
  231. cgProcedure **proc_found = string_map_get(&m->procedures, link_name);
  232. if (proc_found) {
  233. TB_Symbol *symbol = (*proc_found)->symbol;
  234. map_set(&m->symbols, e, symbol);
  235. rw_mutex_unlock(&m->values_mutex);
  236. return symbol;
  237. }
  238. rw_mutex_unlock(&m->values_mutex);
  239. if (e->kind == Entity_Procedure) {
  240. debugf("[Tilde] try to generate procedure %.*s as it was not in the minimum_dependency_set", LIT(e->token.string));
  241. // IMPORTANT TODO(bill): This is an utter bodge, try and fix this shit
  242. cgProcedure *p = cg_procedure_create(m, e);
  243. if (p != nullptr) {
  244. GB_ASSERT(p->symbol != nullptr);
  245. cg_add_procedure_to_queue(p);
  246. return p->symbol;
  247. }
  248. }
  249. GB_PANIC("could not find entity's symbol %.*s", LIT(e->token.string));
  250. return nullptr;
  251. }
  252. struct cgGlobalVariable {
  253. cgValue var;
  254. cgValue init;
  255. DeclInfo *decl;
  256. bool is_initialized;
  257. };
  258. // Returns already_has_entry_point
  259. gb_internal bool cg_global_variables_create(cgModule *m, Array<cgGlobalVariable> *global_variables) {
  260. isize global_variable_max_count = 0;
  261. bool already_has_entry_point = false;
  262. for (Entity *e : m->info->entities) {
  263. String name = e->token.string;
  264. if (e->kind == Entity_Variable) {
  265. global_variable_max_count++;
  266. } else if (e->kind == Entity_Procedure) {
  267. if ((e->scope->flags&ScopeFlag_Init) && name == "main") {
  268. GB_ASSERT(e == m->info->entry_point);
  269. }
  270. if (build_context.command_kind == Command_test &&
  271. (e->Procedure.is_export || e->Procedure.link_name.len > 0)) {
  272. String link_name = e->Procedure.link_name;
  273. if (e->pkg->kind == Package_Runtime) {
  274. if (link_name == "main" ||
  275. link_name == "DllMain" ||
  276. link_name == "WinMain" ||
  277. link_name == "wWinMain" ||
  278. link_name == "mainCRTStartup" ||
  279. link_name == "_start") {
  280. already_has_entry_point = true;
  281. }
  282. }
  283. }
  284. }
  285. }
  286. *global_variables = array_make<cgGlobalVariable>(permanent_allocator(), 0, global_variable_max_count);
  287. auto *min_dep_set = &m->info->minimum_dependency_set;
  288. for (DeclInfo *d : m->info->variable_init_order) {
  289. Entity *e = d->entity;
  290. if ((e->scope->flags & ScopeFlag_File) == 0) {
  291. continue;
  292. }
  293. if (!ptr_set_exists(min_dep_set, e)) {
  294. continue;
  295. }
  296. DeclInfo *decl = decl_info_of_entity(e);
  297. if (decl == nullptr) {
  298. continue;
  299. }
  300. GB_ASSERT(e->kind == Entity_Variable);
  301. bool is_foreign = e->Variable.is_foreign;
  302. bool is_export = e->Variable.is_export;
  303. String name = cg_get_entity_name(m, e);
  304. TB_Linkage linkage = TB_LINKAGE_PRIVATE;
  305. if (is_foreign) {
  306. linkage = TB_LINKAGE_PUBLIC;
  307. // lb_add_foreign_library_path(m, e->Variable.foreign_library);
  308. // lb_set_wasm_import_attributes(g.value, e, name);
  309. } else if (is_export) {
  310. linkage = TB_LINKAGE_PUBLIC;
  311. }
  312. // lb_set_linkage_from_entity_flags(m, g.value, e->flags);
  313. TB_DebugType *debug_type = cg_debug_type(m, e->type);
  314. TB_Global *global = tb_global_create(m->mod, name.len, cast(char const *)name.text, debug_type, linkage);
  315. cgValue g = cg_value(global, alloc_type_pointer(e->type));
  316. TB_ModuleSectionHandle section = tb_module_get_data(m->mod);
  317. if (e->Variable.thread_local_model != "") {
  318. section = tb_module_get_tls(m->mod);
  319. }
  320. if (e->Variable.link_section.len > 0) {
  321. // TODO(bill): custom module sections
  322. // LLVMSetSection(g.value, alloc_cstring(permanent_allocator(), e->Variable.link_section));
  323. }
  324. cgGlobalVariable var = {};
  325. var.var = g;
  326. var.decl = decl;
  327. if (decl->init_expr != nullptr) {
  328. TypeAndValue tav = type_and_value_of_expr(decl->init_expr);
  329. isize max_regions = cg_global_const_calculate_region_count(tav.value, e->type);
  330. tb_global_set_storage(m->mod, section, global, type_size_of(e->type), type_align_of(e->type), max_regions);
  331. if (tav.mode == Addressing_Constant &&
  332. tav.value.kind != ExactValue_Invalid) {
  333. cg_global_const_add_region(m, tav.value, e->type, global, 0);
  334. var.is_initialized = true;
  335. }
  336. if (!var.is_initialized && is_type_untyped_nil(tav.type)) {
  337. var.is_initialized = true;
  338. }
  339. } else {
  340. var.is_initialized = true;
  341. // TODO(bill): is this even needed;
  342. i64 max_regions = cg_global_const_calculate_region_count_from_basic_type(e->type);
  343. tb_global_set_storage(m->mod, section, global, type_size_of(e->type), type_align_of(e->type), max_regions);
  344. }
  345. array_add(global_variables, var);
  346. cg_add_symbol(m, e, cast(TB_Symbol *)global);
  347. cg_add_entity(m, e, g);
  348. cg_add_member(m, name, g);
  349. }
  350. cg_setup_type_info_data(m);
  351. return already_has_entry_point;
  352. }
  353. gb_internal void cg_global_variables_initialize(cgProcedure *p, Array<cgGlobalVariable> *global_variables) {
  354. for (cgGlobalVariable &var : *global_variables) {
  355. if (var.is_initialized) {
  356. continue;
  357. }
  358. cgValue src = cg_build_expr(p, var.decl->init_expr);
  359. cgValue dst = cg_flatten_value(p, var.var);
  360. cg_emit_store(p, dst, src);
  361. }
  362. }
  363. gb_internal cgModule *cg_module_create(Checker *c) {
  364. cgModule *m = gb_alloc_item(permanent_allocator(), cgModule);
  365. m->checker = c;
  366. m->info = &c->info;
  367. TB_FeatureSet feature_set = {};
  368. bool is_jit = false;
  369. m->mod = tb_module_create(TB_ARCH_X86_64, TB_SYSTEM_WINDOWS, &feature_set, is_jit);
  370. tb_module_set_tls_index(m->mod, 10, "_tls_index");
  371. map_init(&m->values);
  372. map_init(&m->symbols);
  373. map_init(&m->file_id_map);
  374. map_init(&m->debug_type_map);
  375. map_init(&m->proc_debug_type_map);
  376. map_init(&m->proc_proto_map);
  377. map_init(&m->anonymous_proc_lits_map);
  378. map_init(&m->equal_procs);
  379. map_init(&m->hasher_procs);
  380. map_init(&m->map_get_procs);
  381. map_init(&m->map_set_procs);
  382. map_init(&m->map_info_map);
  383. map_init(&m->map_cell_info_map);
  384. array_init(&m->single_threaded_procedure_queue, heap_allocator());
  385. for_array(id, global_files) {
  386. if (AstFile *f = global_files[id]) {
  387. char const *path = alloc_cstring(temporary_allocator(), f->fullpath);
  388. TB_SourceFile *file = tb_get_source_file(m->mod, path);
  389. map_set(&m->file_id_map, cast(uintptr)id, file);
  390. }
  391. }
  392. return m;
  393. }
  394. gb_internal void cg_module_destroy(cgModule *m) {
  395. map_destroy(&m->values);
  396. map_destroy(&m->symbols);
  397. map_destroy(&m->file_id_map);
  398. map_destroy(&m->debug_type_map);
  399. map_destroy(&m->proc_debug_type_map);
  400. map_destroy(&m->proc_proto_map);
  401. map_destroy(&m->anonymous_proc_lits_map);
  402. map_destroy(&m->equal_procs);
  403. map_destroy(&m->hasher_procs);
  404. map_destroy(&m->map_get_procs);
  405. map_destroy(&m->map_set_procs);
  406. map_destroy(&m->map_info_map);
  407. map_destroy(&m->map_cell_info_map);
  408. array_free(&m->single_threaded_procedure_queue);
  409. tb_module_destroy(m->mod);
  410. }
  411. gb_internal String cg_set_nested_type_name_ir_mangled_name(Entity *e, cgProcedure *p) {
  412. // NOTE(bill, 2020-03-08): A polymorphic procedure may take a nested type declaration
  413. // and as a result, the declaration does not have time to determine what it should be
  414. GB_ASSERT(e != nullptr && e->kind == Entity_TypeName);
  415. if (e->TypeName.ir_mangled_name.len != 0) {
  416. return e->TypeName.ir_mangled_name;
  417. }
  418. GB_ASSERT((e->scope->flags & ScopeFlag_File) == 0);
  419. if (p == nullptr) {
  420. Entity *proc = nullptr;
  421. if (e->parent_proc_decl != nullptr) {
  422. proc = e->parent_proc_decl->entity;
  423. } else {
  424. Scope *scope = e->scope;
  425. while (scope != nullptr && (scope->flags & ScopeFlag_Proc) == 0) {
  426. scope = scope->parent;
  427. }
  428. GB_ASSERT(scope != nullptr);
  429. GB_ASSERT(scope->flags & ScopeFlag_Proc);
  430. proc = scope->procedure_entity;
  431. }
  432. GB_ASSERT(proc->kind == Entity_Procedure);
  433. if (proc->cg_procedure != nullptr) {
  434. p = proc->cg_procedure;
  435. }
  436. }
  437. // NOTE(bill): Generate a new name
  438. // parent_proc.name-guid
  439. String ts_name = e->token.string;
  440. if (p != nullptr) {
  441. isize name_len = p->name.len + 1 + ts_name.len + 1 + 10 + 1;
  442. char *name_text = gb_alloc_array(permanent_allocator(), char, name_len);
  443. u32 guid = 1+p->module->nested_type_name_guid.fetch_add(1);
  444. name_len = gb_snprintf(name_text, name_len, "%.*s" ABI_PKG_NAME_SEPARATOR "%.*s-%u", LIT(p->name), LIT(ts_name), guid);
  445. String name = make_string(cast(u8 *)name_text, name_len-1);
  446. e->TypeName.ir_mangled_name = name;
  447. return name;
  448. } else {
  449. // NOTE(bill): a nested type be required before its parameter procedure exists. Just give it a temp name for now
  450. isize name_len = 9 + 1 + ts_name.len + 1 + 10 + 1;
  451. char *name_text = gb_alloc_array(permanent_allocator(), char, name_len);
  452. static std::atomic<u32> guid;
  453. name_len = gb_snprintf(name_text, name_len, "_internal" ABI_PKG_NAME_SEPARATOR "%.*s-%u", LIT(ts_name), 1+guid.fetch_add(1));
  454. String name = make_string(cast(u8 *)name_text, name_len-1);
  455. e->TypeName.ir_mangled_name = name;
  456. return name;
  457. }
  458. }
  459. gb_internal String cg_mangle_name(cgModule *m, Entity *e) {
  460. String name = e->token.string;
  461. AstPackage *pkg = e->pkg;
  462. GB_ASSERT_MSG(pkg != nullptr, "Missing package for '%.*s'", LIT(name));
  463. String pkgn = pkg->name;
  464. GB_ASSERT(!rune_is_digit(pkgn[0]));
  465. if (pkgn == "llvm") {
  466. GB_PANIC("llvm. entities are not allowed with the tilde backend");
  467. }
  468. isize max_len = pkgn.len + 1 + name.len + 1;
  469. bool require_suffix_id = is_type_polymorphic(e->type, true);
  470. if ((e->scope->flags & (ScopeFlag_File | ScopeFlag_Pkg)) == 0) {
  471. require_suffix_id = true;
  472. } else if (is_blank_ident(e->token)) {
  473. require_suffix_id = true;
  474. }if (e->flags & EntityFlag_NotExported) {
  475. require_suffix_id = true;
  476. }
  477. if (require_suffix_id) {
  478. max_len += 21;
  479. }
  480. char *new_name = gb_alloc_array(permanent_allocator(), char, max_len);
  481. isize new_name_len = gb_snprintf(
  482. new_name, max_len,
  483. "%.*s" ABI_PKG_NAME_SEPARATOR "%.*s", LIT(pkgn), LIT(name)
  484. );
  485. if (require_suffix_id) {
  486. char *str = new_name + new_name_len-1;
  487. isize len = max_len-new_name_len;
  488. isize extra = gb_snprintf(str, len, "-%llu", cast(unsigned long long)e->id);
  489. new_name_len += extra-1;
  490. }
  491. String mangled_name = make_string((u8 const *)new_name, new_name_len-1);
  492. return mangled_name;
  493. }
  494. gb_internal String cg_get_entity_name(cgModule *m, Entity *e) {
  495. if (e != nullptr && e->kind == Entity_TypeName && e->TypeName.ir_mangled_name.len != 0) {
  496. return e->TypeName.ir_mangled_name;
  497. }
  498. GB_ASSERT(e != nullptr);
  499. if (e->pkg == nullptr) {
  500. return e->token.string;
  501. }
  502. if (e->kind == Entity_TypeName && (e->scope->flags & ScopeFlag_File) == 0) {
  503. return cg_set_nested_type_name_ir_mangled_name(e, nullptr);
  504. }
  505. String name = {};
  506. bool no_name_mangle = false;
  507. if (e->kind == Entity_Variable) {
  508. bool is_foreign = e->Variable.is_foreign;
  509. bool is_export = e->Variable.is_export;
  510. no_name_mangle = e->Variable.link_name.len > 0 || is_foreign || is_export;
  511. if (e->Variable.link_name.len > 0) {
  512. return e->Variable.link_name;
  513. }
  514. } else if (e->kind == Entity_Procedure && e->Procedure.link_name.len > 0) {
  515. return e->Procedure.link_name;
  516. } else if (e->kind == Entity_Procedure && e->Procedure.is_export) {
  517. no_name_mangle = true;
  518. }
  519. if (!no_name_mangle) {
  520. name = cg_mangle_name(m, e);
  521. }
  522. if (name.len == 0) {
  523. name = e->token.string;
  524. }
  525. if (e->kind == Entity_TypeName) {
  526. e->TypeName.ir_mangled_name = name;
  527. } else if (e->kind == Entity_Procedure) {
  528. e->Procedure.link_name = name;
  529. }
  530. return name;
  531. }
  532. #include "tilde_const.cpp"
  533. #include "tilde_debug.cpp"
  534. #include "tilde_expr.cpp"
  535. #include "tilde_builtin.cpp"
  536. #include "tilde_type_info.cpp"
  537. #include "tilde_proc.cpp"
  538. #include "tilde_stmt.cpp"
  539. gb_internal String cg_filepath_obj_for_module(cgModule *m, bool use_assembly) {
  540. String path = concatenate3_strings(permanent_allocator(),
  541. build_context.build_paths[BuildPath_Output].basename,
  542. STR_LIT("/"),
  543. build_context.build_paths[BuildPath_Output].name
  544. );
  545. // if (m->file) {
  546. // char buf[32] = {};
  547. // isize n = gb_snprintf(buf, gb_size_of(buf), "-%u", m->file->id);
  548. // String suffix = make_string((u8 *)buf, n-1);
  549. // path = concatenate_strings(permanent_allocator(), path, suffix);
  550. // } else if (m->pkg) {
  551. // path = concatenate3_strings(permanent_allocator(), path, STR_LIT("-"), m->pkg->name);
  552. // }
  553. String ext = {};
  554. if (use_assembly) {
  555. ext = STR_LIT(".S");
  556. } else {
  557. if (is_arch_wasm()) {
  558. ext = STR_LIT(".wasm.o");
  559. } else {
  560. switch (build_context.metrics.os) {
  561. case TargetOs_windows:
  562. ext = STR_LIT(".obj");
  563. break;
  564. default:
  565. case TargetOs_darwin:
  566. case TargetOs_linux:
  567. case TargetOs_essence:
  568. ext = STR_LIT(".o");
  569. break;
  570. case TargetOs_freestanding:
  571. switch (build_context.metrics.abi) {
  572. default:
  573. case TargetABI_Default:
  574. case TargetABI_SysV:
  575. ext = STR_LIT(".o");
  576. break;
  577. case TargetABI_Win64:
  578. ext = STR_LIT(".obj");
  579. break;
  580. }
  581. break;
  582. }
  583. }
  584. }
  585. return concatenate_strings(permanent_allocator(), path, ext);
  586. }
  587. gb_internal WORKER_TASK_PROC(cg_procedure_generate_worker_proc) {
  588. cgProcedure *p = cast(cgProcedure *)data;
  589. cg_procedure_generate(p);
  590. return 0;
  591. }
  592. gb_internal void cg_add_procedure_to_queue(cgProcedure *p) {
  593. if (p == nullptr) {
  594. return;
  595. }
  596. cgModule *m = p->module;
  597. if (m->do_threading) {
  598. thread_pool_add_task(cg_procedure_generate_worker_proc, p);
  599. } else {
  600. array_add(&m->single_threaded_procedure_queue, p);
  601. }
  602. }
  603. gb_internal bool cg_generate_code(Checker *c, LinkerData *linker_data) {
  604. TIME_SECTION("Tilde Module Initializtion");
  605. CheckerInfo *info = &c->info;
  606. linker_data_init(linker_data, info, c->parser->init_fullpath);
  607. #if defined(GB_SYSTEM_OSX)
  608. linker_enable_system_library_linking(linker_data);
  609. #endif
  610. cg_global_arena_init();
  611. cgModule *m = cg_module_create(c);
  612. defer (cg_module_destroy(m));
  613. m->do_threading = false;
  614. TIME_SECTION("Tilde Global Variables");
  615. Array<cgGlobalVariable> global_variables = {};
  616. bool already_has_entry_point = cg_global_variables_create(m, &global_variables);
  617. gb_unused(already_has_entry_point);
  618. if (true) {
  619. Type *proc_type = alloc_type_proc(nullptr, nullptr, 0, nullptr, 0, false, ProcCC_Odin);
  620. cgProcedure *p = cg_procedure_create_dummy(m, str_lit(CG_STARTUP_RUNTIME_PROC_NAME), proc_type);
  621. p->is_startup = true;
  622. cg_startup_runtime_proc = p;
  623. }
  624. if (true) {
  625. Type *proc_type = alloc_type_proc(nullptr, nullptr, 0, nullptr, 0, false, ProcCC_Odin);
  626. cgProcedure *p = cg_procedure_create_dummy(m, str_lit(CG_CLEANUP_RUNTIME_PROC_NAME), proc_type);
  627. p->is_startup = true;
  628. cg_cleanup_runtime_proc = p;
  629. }
  630. auto *min_dep_set = &info->minimum_dependency_set;
  631. Array<cgProcedure *> procedures_to_generate = {};
  632. array_init(&procedures_to_generate, heap_allocator());
  633. defer (array_free(&procedures_to_generate));
  634. for (Entity *e : info->entities) {
  635. String name = e->token.string;
  636. Scope *scope = e->scope;
  637. if ((scope->flags & ScopeFlag_File) == 0) {
  638. continue;
  639. }
  640. Scope *package_scope = scope->parent;
  641. GB_ASSERT(package_scope->flags & ScopeFlag_Pkg);
  642. if (e->kind != Entity_Procedure) {
  643. continue;
  644. }
  645. if (!ptr_set_exists(min_dep_set, e)) {
  646. // NOTE(bill): Nothing depends upon it so doesn't need to be built
  647. continue;
  648. }
  649. if (cgProcedure *p = cg_procedure_create(m, e)) {
  650. array_add(&procedures_to_generate, p);
  651. }
  652. }
  653. for (cgProcedure *p : procedures_to_generate) {
  654. cg_add_procedure_to_queue(p);
  655. }
  656. if (!m->do_threading) {
  657. for (isize i = 0; i < m->single_threaded_procedure_queue.count; i++) {
  658. cgProcedure *p = m->single_threaded_procedure_queue[i];
  659. cg_procedure_generate(p);
  660. }
  661. }
  662. thread_pool_wait();
  663. {
  664. cgProcedure *p = cg_startup_runtime_proc;
  665. cg_procedure_begin(p);
  666. cg_global_variables_initialize(p, &global_variables);
  667. tb_inst_ret(p->func, 0, nullptr);
  668. cg_procedure_end(p);
  669. }
  670. {
  671. cgProcedure *p = cg_cleanup_runtime_proc;
  672. cg_procedure_begin(p);
  673. tb_inst_ret(p->func, 0, nullptr);
  674. cg_procedure_end(p);
  675. }
  676. TB_DebugFormat debug_format = TB_DEBUGFMT_NONE;
  677. if (build_context.ODIN_DEBUG) {
  678. switch (build_context.metrics.os) {
  679. case TargetOs_windows:
  680. debug_format = TB_DEBUGFMT_CODEVIEW;
  681. break;
  682. case TargetOs_darwin:
  683. case TargetOs_linux:
  684. case TargetOs_essence:
  685. case TargetOs_freebsd:
  686. case TargetOs_openbsd:
  687. debug_format = TB_DEBUGFMT_DWARF;
  688. break;
  689. }
  690. }
  691. TB_ExportBuffer export_buffer = tb_module_object_export(m->mod, debug_format);
  692. defer (tb_export_buffer_free(export_buffer));
  693. String filepath_obj = cg_filepath_obj_for_module(m, false);
  694. array_add(&linker_data->output_object_paths, filepath_obj);
  695. GB_ASSERT(tb_export_buffer_to_file(export_buffer, cast(char const *)filepath_obj.text));
  696. return true;
  697. }
  698. #undef ABI_PKG_NAME_SEPARATOR