1use std::ops::{Deref, DerefMut};
37
38pub struct BorrowMut<'a, T: 'a> {
39 inner: &'a mut Vec<T>
40}
41
42impl<'a, T> Deref for BorrowMut<'a, T> {
43 type Target = Vec<T>;
44 fn deref(&self) -> &Self::Target { self.inner }
45}
46
47impl<'a, T> DerefMut for BorrowMut<'a, T> {
48 fn deref_mut(&mut self) -> &mut Self::Target { self.inner }
49}
50
51impl<'a, T> Drop for BorrowMut<'a, T> {
52 fn drop(&mut self){
53 self.inner.clear();
54 }
55}
56
57pub struct TmpVec<U> {
58 inner: Vec<U>
59}
60
61impl<U> Default for TmpVec<U> {
62 fn default() -> Self { Self::new() }
63}
64
65impl<U> TmpVec<U> {
66 pub fn new() -> Self { TmpVec{ inner: Vec::new() } }
67 pub fn with_capacity(cap: usize) -> Self { TmpVec{ inner: Vec::with_capacity(cap) } }
68
69 pub fn borrow_mut<T>(&mut self) -> BorrowMut<T> {
70 use std::mem;
71 assert_eq!(mem::size_of::<T>(), mem::size_of::<U>());
72 assert_eq!(mem::align_of::<T>(), mem::align_of::<U>());
73 unsafe { self.inner.set_len(0); } BorrowMut{ inner: unsafe { mem::transmute(&mut self.inner) } }
75 }
76}
77
78impl<U> Drop for TmpVec<U> {
79 fn drop(&mut self){
80 unsafe { self.inner.set_len(0); } }
82}
83