mirrors/Odin
1
0
Fork 0
mirror of https://github.com/odin-lang/Odin.git synced 2025-12-31 02:12:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

Improve behaviour of return with named results to aid with defer statements

Browse Source
This commit is contained in:
gingerBill
2020-06-10 14:53:35 +01:00
parent a9295d33ab
commit 99944f3b02
2 changed files with 51 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
src/llvm_backend.cpp
View File

@@ -2434,7 +2434,6 @@ void lb_begin_procedure_body(lbProcedure *p) {
if (p->type->Proc.has_named_results) {
GB_ASSERT(p->type->Proc.result_count > 0);
TypeTuple *results = &p->type->Proc.results->Tuple;
LLVMValueRef return_ptr = LLVMGetParam(p->value, 0);
isize result_index = 0;
@@ -4192,6 +4191,15 @@ void lb_build_stmt(lbProcedure *p, Ast *node) {
res = lb_build_expr(p, rs->results[0]);
res = lb_emit_conv(p, res, e->type);
}
if (p->type->Proc.has_named_results) {
// NOTE(bill): store the named values before returning
if (e->token.string != "") {
lbValue *found = map_get(&p->module->values, hash_entity(e));
GB_ASSERT(found != nullptr);
lb_emit_store(p, *found, res);
}
}
} else {
auto results = array_make<lbValue>(heap_allocator(), 0, return_count);
@@ -4221,6 +4229,23 @@ void lb_build_stmt(lbProcedure *p, Ast *node) {
GB_ASSERT(results.count == return_count);
if (p->type->Proc.has_named_results) {
// NOTE(bill): store the named values before returning
for_array(i, p->type->Proc.results->Tuple.variables) {
Entity *e = p->type->Proc.results->Tuple.variables[i];
if (e->kind != Entity_Variable) {
continue;
}
if (e->token.string == "") {
continue;
}
lbValue *found = map_get(&p->module->values, hash_entity(e));
GB_ASSERT(found != nullptr);
lb_emit_store(p, *found, results[i]);
}
}
Type *ret_type = p->type->Proc.results;
// NOTE(bill): Doesn't need to be zero because it will be initialized in the loops
res = lb_add_local_generated(p, ret_type, false).addr;