ach_util/
refer.rs

1use super::state::MemoryState;
2use bytemuck::NoUninit;
3
4pub type AtomicMemoryRefer = atomic::Atomic<MemoryRefer>;
5
6#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, NoUninit)]
7#[repr(C)]
8pub struct MemoryRefer(u32);
9impl Default for MemoryRefer {
10    fn default() -> Self {
11        Self::new()
12    }
13}
14impl MemoryRefer {
15    /// Uninitialized
16    pub const fn new() -> Self {
17        Self(0)
18    }
19    pub fn state(&self) -> MemoryState {
20        ((self.0 >> 24) as u8).into()
21    }
22    pub fn set_state(&mut self, val: MemoryState) {
23        self.0 = (self.0 & 0x00FF_FFFF) | ((u8::from(val) as u32) << 24);
24    }
25    pub const fn max_refer() -> usize {
26        0x00FF_FFFF
27    }
28    pub fn ref_num(&self) -> Result<usize, MemoryState> {
29        let state = self.state();
30        if state.is_initialized() || state.is_regaining() || state.is_erasing() {
31            Ok((self.0 as usize) & 0x00FF_FFFF)
32        } else {
33            Err(state)
34        }
35    }
36    pub fn ref_add(&mut self) -> Result<(), MemoryState> {
37        let ref_num = self.ref_num()?;
38        if ref_num == Self::max_refer() {
39            return Ok(());
40        }
41        let state = self.state();
42        if state.is_initialized() {
43            self.0 += 1;
44            Ok(())
45        } else {
46            Err(state)
47        }
48    }
49    pub fn ref_sub(&mut self) -> Result<(), MemoryState> {
50        let ref_num = self.ref_num()?;
51        if ref_num == Self::max_refer() || ref_num == 0 {
52            return Ok(());
53        }
54        let state = self.state();
55        if state.is_initialized() || state.is_regaining() {
56            self.0 -= 1;
57            Ok(())
58        } else {
59            Err(state)
60        }
61    }
62}
63impl From<MemoryState> for MemoryRefer {
64    fn from(s: MemoryState) -> Self {
65        let mut refer = MemoryRefer::new();
66        refer.set_state(s);
67        refer
68    }
69}
70impl From<u32> for MemoryRefer {
71    fn from(s: u32) -> Self {
72        Self(s)
73    }
74}
75impl From<MemoryRefer> for u32 {
76    fn from(s: MemoryRefer) -> Self {
77        s.0
78    }
79}
ach_util/refer.rs

ach_util/
refer.rs
