autocommit 2022-02-28 00:14:11 CET

src/utils/vec_ext.rs

@@ -4,6 +4,7 @@ use std::slice;
 
 pub trait VecExt<T> {
     fn split_at_spare_mut_ext(&mut self) -> (&mut [T], &mut [MaybeUninit<T>]);
+    fn split_at_spare_mut_bytes_ext(&mut self) -> (&mut [T], &mut [MaybeUninit<u8>]);
 }
 
 impl<T> VecExt<T> for Vec<T> {
@@ -20,4 +21,22 @@ impl<T> VecExt<T> for Vec<T> {
             (initialized, spare)
         }
     }
+
+    fn split_at_spare_mut_bytes_ext(&mut self) -> (&mut [T], &mut [MaybeUninit<u8>]) {
+        let (l, r) = self.split_at_spare_mut_ext();
+        unsafe { (l, uapi::as_maybe_uninit_bytes_mut2(r)) }
+    }
+}
+
+pub trait UninitVecExt<T> {
+    fn set_len_safe(&mut self, n: usize);
+}
+
+impl<T> UninitVecExt<T> for Vec<MaybeUninit<T>> {
+    fn set_len_safe(&mut self, n: usize) {
+        assert!(n <= self.capacity());
+        unsafe {
+            self.set_len(n);
+        }
+    }
 }