How can I write data from a slice to the same slice?
To copy data from one range inside a slice to another in general (allowing overlap), we can't even use .split_at_mut()
.
I would use .split_at_mut()
primarily otherwise. (Is there anything that makes you think the bounds check is not going to be optimized out? Also, are you copying enough data that it's a small effect in comparison?)
Anyway, this is how you could wrap std::ptr::copy
(overlap-allowing copy, a.k.a memmove) in a safe or an unsafe
function.
use std::ptr::copy;
use std::ops::Range;
/// Copy the range `data[from]` onto the index `to` and following
///
/// **Panics** if `from` or `to` is out of bounds
pub fn move_memory<T: Copy>(data: &mut [T], from: Range<usize>, to: usize) {
assert!(from.start <= from.end);
assert!(from.end <= data.len());
assert!(to <= data.len() - (from.end - from.start));
unsafe {
move_memory_unchecked(data, from, to);
}
}
pub unsafe fn move_memory_unchecked<T: Copy>(data: &mut [T], from: Range<usize>, to: usize) {
debug_assert!(from.start <= from.end);
debug_assert!(from.end <= data.len());
debug_assert!(to <= data.len() - (from.end - from.start));
let ptr = data.as_mut_ptr();
copy(ptr.offset(from.start as isize),
ptr.offset(to as isize),
from.end - from.start)
}
fn main() {
let mut data = [0, 1, 2, 3, 4, 5, 6, 7];
move_memory(&mut data, 2..6, 0);
println!("{:?}", data);
move_memory(&mut data, 0..3, 5);
println!("{:?}", data);
}
Playground link
If your types implement Copy
and the subslices are not overlapping:
fn main() {
let mut v = [1, 2, 3, 4, 5, 6];
{
let (left, right) = v.split_at_mut(3);
// Perform further work to ensure slices are the same length, as needed
left.copy_from_slice(right);
}
assert!(v == [4, 5, 6, 4, 5, 6]);
}
Rust 1.37 (2019-08-15) adds the library function slice::copy_within
which does exactly what you want:
let mut foo = [1, 2, 3, 4, 5];
foo.copy_within(3 .. 5, 0); // <-- THIS
assert_eq!(foo, [4, 5, 3, 4, 5]);