Bruce/tomo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Activity

Add early outs for (or) and (and) reducers

Browse Source
This commit is contained in:
Bruce Hill 2024-09-05 11:51:07 -04:00
parent 0d3c8e16bc
commit d801681026
2 changed files with 22 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
compile.c
View File

@ -757,7 +757,7 @@ CORD compile_statement(env_t *env, ast_t *ast)
else if (target)
code_err(ast, "No loop target named '%s' was found", target);
else
code_err(ast, "I couldn't figure out how to make this skip work!");
return "continue;";
}
case Stop: {
const char *target = Match(ast, Stop)->target;
@ -783,7 +783,7 @@ CORD compile_statement(env_t *env, ast_t *ast)
else if (target)
code_err(ast, "No loop target named '%s' was found", target);
else
code_err(ast, "I couldn't figure out how to make this stop work!");
return "break;";
}
case Pass: return ";";
case Defer: {
@ -2676,12 +2676,24 @@ CORD compile(env_t *env, ast_t *ast)
ast_t *body = FakeAST(InlineCCode, .code="{}"); // placeholder
ast_t *loop = FakeAST(For, .vars=new(ast_list_t, .ast=item), .iter=reduction->iter, .body=body, .empty=empty);
env_t *body_scope = for_scope(scope, loop);
// For the special case of (or)/(and), we need to early out if we can:
CORD early_out = CORD_EMPTY;
if (t->tag == BoolType && reduction->combination->tag == BinaryOp) {
auto binop = Match(reduction->combination, BinaryOp);
if (binop->op == BINOP_AND)
early_out = "if (!reduction) break;";
else if (binop->op == BINOP_OR)
early_out = "if (reduction) break;";
}
body->__data.InlineCCode.code = CORD_all(
"if (is_first) {\n"
" reduction = ", compile(body_scope, item), ";\n"
" is_first = no;\n"
"} else {\n"
" reduction = ", compile(body_scope, reduction->combination), ";\n"
" reduction = ", compile(body_scope, reduction->combination), ";\n",
early_out,
"}\n");
code = CORD_all(code, compile_statement(scope, loop), "\nreduction;})");
return code;

7
test/reductions.tm
View File

@ -16,3 +16,10 @@ func main():
= Foo(x=0, y=10)
>> (_max_.y:abs()) [Foo(0, 0), Foo(1, 0), Foo(0, 10), Foo(0, -999)]
= Foo(x=0, y=-999)
!! (or) and (and) have early out behavior:
>> (or) i == 3 for i in 99999999999999999
= yes
>> (and) i < 10 for i in 99999999999999999
= no