open_pql/vm/
store_var_idx.rs

1use super::*;
2
3#[derive(Debug, Clone, Copy, From, PartialEq, Eq)]
4pub struct VmStoreVarIdx(u8);
5
6impl Add<u8> for VmStoreVarIdx {
7    type Output = Self;
8
9    fn add(self, rhs: u8) -> Self::Output {
10        Self(self.0 + rhs)
11    }
12}
13
14impl VmStoreVarIdx {
15    pub const fn to_usize(self) -> usize {
16        self.0 as usize
17    }
18}
19
20impl TryFrom<usize> for VmStoreVarIdx {
21    type Error = PQLError;
22
23    fn try_from(value: usize) -> Result<Self, Self::Error> {
24        if value > u8::MAX as usize {
25            return Err(PQLError::TooManyVariables);
26        }
27
28        Ok(Self(value.to_le_bytes()[0]))
29    }
30}
31
32impl PartialOrd for VmStoreVarIdx {
33    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
34        if self.0 == other.0 {
35            Some(Ordering::Equal)
36        } else {
37            None
38        }
39    }
40}
41
42#[cfg_attr(coverage_nightly, coverage(off))]
43#[cfg(test)]
44mod tests {
45    use super::*;
46
47    impl Arbitrary for VmStoreVarIdx {
48        fn arbitrary(g: &mut quickcheck::Gen) -> Self {
49            Self(u8::arbitrary(g))
50        }
51    }
52
53    #[test]
54    fn test_add() {
55        let v = VmStoreVarIdx(254);
56
57        assert_eq!(v + 1, 255.into());
58    }
59
60    #[test]
61    fn test_ord() {
62        use VmStoreVarIdx as I;
63        assert_eq!(Some(Ordering::Equal), I(0).partial_cmp(&I(0)));
64        assert_eq!(None, I(1).partial_cmp(&I(0)));
65    }
66
67    #[test]
68    fn test_cast() {
69        assert_eq!(
70            VmStoreVarIdx(100),
71            VmStoreVarIdx::try_from(100usize).unwrap()
72        );
73        assert_eq!(100, VmStoreVarIdx(100).to_usize());
74    }
75}
open_pql/vm/store_var_idx.rs

open_pql/vm/
store_var_idx.rs
