ach_util/
refer.rs

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

ach_util/
refer.rs
