language-servers/roc
1
0
Fork 0
mirror of https://github.com/roc-lang/roc.git synced 2025-08-03 11:52:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

simplified insert_refcount_stmt

Browse source
This commit is contained in:
J.Teeuwissen 2023-02-10 13:12:10 +01:00
parent ae499137e4
commit 26a9b919d9
No known key found for this signature in database
GPG key ID: DB5F7A1ED8D478AD
1 changed files with 14 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -129,22 +129,23 @@ fn insert_refcount_stmt<'a>(
mut usage: VariableUsage,
continuation: &'a Stmt<'a>,
) -> &'a Stmt<'a> {
usage.iter_mut().for_each(|(symbol, mut usage_count)| {
match consume_variable(environment, *symbol) {
// If the variable is borrowed, we need to increment the reference count for each usage.
Ownership::Borrowed => {}
// If the variable is owned, we need to increment the reference count for each usage except one.
Ownership::Owned => *usage_count -= 1,
}
});
usage
.iter()
.fold(continuation, |continuation, (symbol, usage_count)| {
arena.alloc(Stmt::Refcounting(
ModifyRc::Inc(*symbol, *usage_count),
continuation,
))
let new_count = match consume_variable(environment, *symbol) {
// If the variable is borrowed, we need to increment the reference count for each usage.
Ownership::Borrowed => *usage_count,
// If the variable is owned, we need to increment the reference count for each usage except one.
Ownership::Owned => *usage_count - 1,
};
match new_count {
0 => continuation,
_ => arena.alloc(Stmt::Refcounting(
ModifyRc::Inc(*symbol, *usage_count),
continuation,
)),
}
})
}