language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-09-26 21:39:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

decref lists

Browse source
This commit is contained in:
J.Teeuwissen 2023-04-05 14:45:02 +02:00
parent bd8f3ae929
commit ccb2e85607
No known key found for this signature in database
GPG key ID: DB5F7A1ED8D478AD
1 changed files with 23 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

27
crates/compiler/mono/src/perceus.rs
View file

@ -961,10 +961,23 @@ fn insert_refcount_operations_binding<'a>(
// This should always be true, not sure where this could be set to false.
debug_assert!(passed_function.owns_captured_environment);
// define macro that inserts a decref statement for a variable amount of symbols
macro_rules! decref_lists {
($stmt:expr, $symbol:expr) => {
arena.alloc(Stmt::Refcounting(ModifyRc::DecRef($symbol), $stmt))
};
($stmt:expr, $symbol:expr, $($symbols:expr),+) => {{
decref_lists!(decref_lists!($stmt, $symbol), $($symbols),+)
}};
}
match operator {
HigherOrder::ListMap { xs } => {
if let [_xs_symbol, _function_symbol, closure_symbol] = &arguments {
let new_let = new_let!(stmt);
let new_stmt = decref_lists!(stmt, *xs);
let new_let = new_let!(new_stmt);
inc_owned!([*xs, *closure_symbol].into_iter(), new_let)
} else {
@ -975,7 +988,9 @@ fn insert_refcount_operations_binding<'a>(
if let [_xs_symbol, _ys_symbol, _function_symbol, closure_symbol] =
&arguments
{
let new_let = new_let!(stmt);
let new_stmt = decref_lists!(stmt, *xs, *ys);
let new_let = new_let!(new_stmt);
inc_owned!([*xs, *ys, *closure_symbol].into_iter(), new_let)
} else {
@ -986,7 +1001,9 @@ fn insert_refcount_operations_binding<'a>(
if let [_xs_symbol, _ys_symbol, _zs_symbol, _function_symbol, closure_symbol] =
&arguments
{
let new_let = new_let!(stmt);
let new_stmt = decref_lists!(stmt, *xs, *ys, *zs);
let new_let = new_let!(new_stmt);
inc_owned!([*xs, *ys, *zs, *closure_symbol].into_iter(), new_let)
} else {
@ -997,7 +1014,9 @@ fn insert_refcount_operations_binding<'a>(
if let [_xs_symbol, _ys_symbol, _zs_symbol, _ws_symbol, _function_symbol, closure_symbol] =
&arguments
{
let new_let = new_let!(stmt);
let new_stmt = decref_lists!(stmt, *xs, *ys, *zs, *ws);
let new_let = new_let!(new_stmt);
inc_owned!([*xs, *ys, *zs, *ws, *closure_symbol].into_iter(), new_let)
} else {