| 1 | /// Takes an `Option<&mut Vec<T>>` style buffer and gets its pointer. |
| 2 | macro_rules! map_ptr { |
| 3 | ($buffer:expr) => { |
| 4 | match $buffer { |
| 5 | Some(b) => b.as_ptr() as _, |
| 6 | None => 0 as _, |
| 7 | } |
| 8 | }; |
| 9 | } |
| 10 | |
| 11 | /// Takes an `Option<&mut Vec<T>>` style buffer and gets its allocated length. |
| 12 | macro_rules! map_len { |
| 13 | ($buffer:expr) => { |
| 14 | match $buffer { |
| 15 | Some(b) => b.capacity() as _, |
| 16 | None => 0, |
| 17 | } |
| 18 | }; |
| 19 | } |
| 20 | |
| 21 | /// Takes an `Option<&mut Vec<T>>` style buffer and reserves space. |
| 22 | macro_rules! map_reserve { |
| 23 | ($buffer:expr, $size:expr) => { |
| 24 | match $buffer { |
| 25 | Some(ref mut b) => crate::utils::map_reserve_inner(b, $size), |
| 26 | _ => (), |
| 27 | } |
| 28 | }; |
| 29 | } |
| 30 | |
| 31 | pub(crate) fn map_reserve_inner<T>(b: &mut Vec<T>, size: usize) { |
| 32 | let old_len: usize = b.len(); |
| 33 | if size <= old_len { |
| 34 | return; |
| 35 | } |
| 36 | b.reserve_exact(additional:size - old_len); |
| 37 | |
| 38 | // `memset` to 0, at least so Valgrind doesn't complain |
| 39 | unsafe { |
| 40 | let ptr: *mut u8 = b.as_mut_ptr().add(count:old_len) as *mut u8; |
| 41 | ptr.write_bytes(val:0, (size - old_len) * std::mem::size_of::<T>()); |
| 42 | } |
| 43 | } |
| 44 | |
| 45 | /// Takes an `Option<&mut Vec<T>>` style buffer and shrinks it. |
| 46 | macro_rules! map_set { |
| 47 | ($buffer:expr, $min:expr) => { |
| 48 | match $buffer { |
| 49 | Some(ref mut b) => unsafe { b.set_len($min) }, |
| 50 | _ => (), |
| 51 | } |
| 52 | }; |
| 53 | } |
| 54 | |