1 | use core::cell::UnsafeCell; |
2 | use core::mem::MaybeUninit; |
3 | use core::ptr; |
4 | |
5 | pub(crate) struct UninitCell<T>(MaybeUninit<UnsafeCell<T>>); |
6 | impl<T> UninitCell<T> { |
7 | pub const fn uninit() -> Self { |
8 | Self(MaybeUninit::uninit()) |
9 | } |
10 | |
11 | pub unsafe fn as_mut_ptr(&self) -> *mut T { |
12 | (*self.0.as_ptr()).get() |
13 | } |
14 | |
15 | #[allow (clippy::mut_from_ref)] |
16 | pub unsafe fn as_mut(&self) -> &mut T { |
17 | &mut *self.as_mut_ptr() |
18 | } |
19 | |
20 | #[inline (never)] |
21 | pub unsafe fn write_in_place(&self, func: impl FnOnce() -> T) { |
22 | ptr::write(self.as_mut_ptr(), src:func()) |
23 | } |
24 | |
25 | pub unsafe fn drop_in_place(&self) { |
26 | ptr::drop_in_place(self.as_mut_ptr()) |
27 | } |
28 | } |
29 | |
30 | unsafe impl<T> Sync for UninitCell<T> {} |
31 | |
32 | #[repr (transparent)] |
33 | pub struct SyncUnsafeCell<T> { |
34 | value: UnsafeCell<T>, |
35 | } |
36 | |
37 | unsafe impl<T: Sync> Sync for SyncUnsafeCell<T> {} |
38 | |
39 | impl<T> SyncUnsafeCell<T> { |
40 | #[inline ] |
41 | pub const fn new(value: T) -> Self { |
42 | Self { |
43 | value: UnsafeCell::new(value), |
44 | } |
45 | } |
46 | |
47 | pub unsafe fn set(&self, value: T) { |
48 | *self.value.get() = value; |
49 | } |
50 | |
51 | pub unsafe fn get(&self) -> T |
52 | where |
53 | T: Copy, |
54 | { |
55 | *self.value.get() |
56 | } |
57 | } |
58 | |