From ba8ec1910bf2527eb290c128a04c23f42ec6a3bb Mon Sep 17 00:00:00 2001 From: Griffin Smith Date: Sun, 9 Oct 2022 12:59:23 -0400 Subject: feat(tvix/eval): Implement builtins.foldl' Change-Id: Ibc97db4343cb3a1a1677f69fb6c3518c61978aad Reviewed-on: https://cl.tvl.fyi/c/depot/+/6906 Autosubmit: grfn Reviewed-by: tazjin Tested-by: BuildkiteCI --- tvix/eval/src/builtins/mod.rs | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) (limited to 'tvix/eval/src/builtins/mod.rs') diff --git a/tvix/eval/src/builtins/mod.rs b/tvix/eval/src/builtins/mod.rs index 93b04bb37b..f32a29a206 100644 --- a/tvix/eval/src/builtins/mod.rs +++ b/tvix/eval/src/builtins/mod.rs @@ -222,6 +222,24 @@ fn pure_builtins() -> Vec { .map(|list| Value::List(NixList::from(list))) .map_err(Into::into) }), + Builtin::new( + "foldl'", + &[true, false, true], + |mut args: Vec, vm: &mut VM| { + let list = args.pop().unwrap().to_list()?; + let mut res = args.pop().unwrap(); + let op = args.pop().unwrap(); + for val in list { + val.force(vm)?; + vm.push(val); + vm.push(res); + let partial = vm.call_value(&op)?; + res = vm.call_value(&partial)?; + } + + Ok(res) + }, + ), Builtin::new("genList", &[true, true], |args: Vec, vm: &mut VM| { let len = args[1].as_int()?; (0..len) -- cgit 1.4.1