Add base piping for list.Replace

compiler/gen_llvm/src/llvm/build.rs

@@ -13,8 +13,8 @@ use crate::llvm::build_list::{
     self, allocate_list, empty_polymorphic_list, list_all, list_any, list_append, list_concat,
     list_contains, list_drop_at, list_find_unsafe, list_get_unsafe, list_join, list_keep_errs,
     list_keep_if, list_keep_oks, list_len, list_map, list_map2, list_map3, list_map4,
-    list_map_with_index, list_prepend, list_range, list_repeat, list_reverse, list_set,
-    list_single, list_sort_with, list_sublist, list_swap,
+    list_map_with_index, list_prepend, list_range, list_repeat, list_replace, list_reverse,
+    list_set, list_single, list_sort_with, list_sublist, list_swap,
 };
 use crate::llvm::build_str::{
     str_concat, str_count_graphemes, str_ends_with, str_from_float, str_from_int, str_from_utf8,
@@ -5653,6 +5653,21 @@ fn run_low_level<'a, 'ctx, 'env>(
                 wrapper_struct,
             )
         }
+        ListReplace => {
+            let list = load_symbol(scope, &args[0]);
+            let index = load_symbol(scope, &args[1]);
+            let (element, element_layout) = load_symbol_and_layout(scope, &args[2]);
+
+            list_replace(
+                env,
+                layout_ids,
+                list,
+                index.into_int_value(),
+                element,
+                element_layout,
+                update_mode,
+            )
+        }
         ListSet => {
             let list = load_symbol(scope, &args[0]);
             let index = load_symbol(scope, &args[1]);

compiler/gen_llvm/src/llvm/build_list.rs

@@ -291,6 +291,50 @@ pub fn list_drop_at<'a, 'ctx, 'env>(
     )
 }
 
+/// List.replace : List elem, Nat, elem -> List elem
+pub fn list_replace<'a, 'ctx, 'env>(
+    env: &Env<'a, 'ctx, 'env>,
+    layout_ids: &mut LayoutIds<'a>,
+    list: BasicValueEnum<'ctx>,
+    index: IntValue<'ctx>,
+    element: BasicValueEnum<'ctx>,
+    element_layout: &Layout<'a>,
+    update_mode: UpdateMode,
+) -> BasicValueEnum<'ctx> {
+    let (length, bytes) = load_list(
+        env.builder,
+        list.into_struct_value(),
+        env.context.i8_type().ptr_type(AddressSpace::Generic),
+    );
+
+    let new_bytes = match update_mode {
+        UpdateMode::InPlace => call_bitcode_fn(
+            env,
+            &[
+                bytes.into(),
+                index.into(),
+                pass_element_as_opaque(env, element, *element_layout),
+                layout_width(env, element_layout),
+            ],
+            bitcode::LIST_REPLACE_IN_PLACE,
+        ),
+        UpdateMode::Immutable => call_bitcode_fn(
+            env,
+            &[
+                bytes.into(),
+                length.into(),
+                env.alignment_intvalue(element_layout),
+                index.into(),
+                pass_element_as_opaque(env, element, *element_layout),
+                layout_width(env, element_layout),
+            ],
+            bitcode::LIST_REPLACE,
+        ),
+    };
+
+    store_list(env, new_bytes.into_pointer_value(), length)
+}
+
 /// List.set : List elem, Nat, elem -> List elem
 pub fn list_set<'a, 'ctx, 'env>(
     env: &Env<'a, 'ctx, 'env>,