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

first pass for List.keep

Browse source
This commit is contained in:
Folkert 2021-05-19 21:30:43 +02:00
parent e3b102e0c3
commit d01d539d6b
5 changed files with 387 additions and 195 deletions
Download patch file Download diff file Expand all files Collapse all files

38
compiler/mono/src/borrow.rs
View file

@ -425,7 +425,7 @@ impl<'a> BorrowInfState<'a> {
debug_assert!(op.is_higher_order());
match op {
ListMap | ListSortWith => {
ListMap | ListKeepIf | ListKeepOks | ListKeepErrs => {
match self.param_map.get_symbol(arguments[1], *closure_layout) {
Some(function_ps) => {
// own the list if the function wants to own the element
@ -481,7 +481,6 @@ impl<'a> BorrowInfState<'a> {
if !function_ps[0].borrow {
self.own_var(arguments[0]);
}
if !function_ps[1].borrow {
self.own_var(arguments[1]);
}
@ -496,6 +495,41 @@ impl<'a> BorrowInfState<'a> {
}
None => unreachable!(),
},
ListSortWith => {
match self.param_map.get_symbol(arguments[1], *closure_layout) {
Some(function_ps) => {
// always own the input list
self.own_var(arguments[0]);
// own the closure environment if the function needs to own it
if let Some(false) = function_ps.get(2).map(|p| p.borrow) {
self.own_var(arguments[2]);
}
}
None => unreachable!(),
}
}
ListWalk | ListWalkUntil | ListWalkBackwards | DictWalk => {
match self.param_map.get_symbol(arguments[2], *closure_layout) {
Some(function_ps) => {
// own the data structure if the function wants to own the element
if !function_ps[0].borrow {
self.own_var(arguments[0]);
}
// own the default value if the function wants to own it
if !function_ps[1].borrow {
self.own_var(arguments[1]);
}
// own the closure environment if the function needs to own it
if let Some(false) = function_ps.get(2).map(|p| p.borrow) {
self.own_var(arguments[3]);
}
}
None => unreachable!(),
}
}
_ => {
// very unsure what demand RunLowLevel should place upon its arguments
self.own_var(z);