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

remove allocation

Browse source
This commit is contained in:
Folkert 2021-03-20 23:51:55 +01:00
parent 03fab9297c
commit 8b8afec50e
1 changed files with 5 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

15
compiler/can/src/operator.rs
View file

@ -502,8 +502,8 @@ fn desugar_bin_ops<'a>(
lefts: &'a [(Located<Expr<'_>>, Located<BinOp>)], lefts: &'a [(Located<Expr<'_>>, Located<BinOp>)],
right: &'a Located<Expr<'_>>, right: &'a Located<Expr<'_>>,
) -> &'a Located<Expr<'a>> { ) -> &'a Located<Expr<'a>> {
let mut arg_stack: Vec<&'a Located<Expr>> = Vec::new_in(arena); let mut arg_stack: Vec<&'a Located<Expr>> = Vec::with_capacity_in(lefts.len() + 1, arena);
let mut op_stack: Vec<Located<BinOp>> = Vec::new_in(arena); let mut op_stack: Vec<Located<BinOp>> = Vec::with_capacity_in(lefts.len(), arena);
for (loc_expr, loc_op) in lefts { for (loc_expr, loc_op) in lefts {
arg_stack.push(desugar_expr(arena, loc_expr)); arg_stack.push(desugar_expr(arena, loc_expr));
@ -524,7 +524,7 @@ fn desugar_bin_ops<'a>(
Pizza => { Pizza => {
// Rewrite the Pizza operator into an Apply // Rewrite the Pizza operator into an Apply
match &right.value { match right.value {
Apply(function, arguments, _called_via) => { Apply(function, arguments, _called_via) => {
let mut args = Vec::with_capacity_in(1 + arguments.len(), arena); let mut args = Vec::with_capacity_in(1 + arguments.len(), arena);
@ -538,20 +538,15 @@ fn desugar_bin_ops<'a>(
Apply(function, args, CalledVia::BinOp(Pizza)) Apply(function, args, CalledVia::BinOp(Pizza))
} }
expr => { _ => {
// e.g. `1 |> (if b then (\a -> a) else (\c -> c))` // e.g. `1 |> (if b then (\a -> a) else (\c -> c))`
let mut args = Vec::with_capacity_in(1, arena); let mut args = Vec::with_capacity_in(1, arena);
args.push(left); args.push(left);
let function = arena.alloc(Located {
value: Nested(expr),
region: right.region,
});
let args = args.into_bump_slice(); let args = args.into_bump_slice();
Apply(function, args, CalledVia::BinOp(Pizza)) Apply(right, args, CalledVia::BinOp(Pizza))
} }
} }
} }