fieldset 0.1.4

Struct to enum derivation
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

#[derive(Clone, Copy, Debug)]
pub struct BitSet<const N_32: usize> {
    bits: [u32; N_32],
}

pub struct BitSetOffsetted<'a> {
    bits: &'a mut [u32],
    offset: usize,
}

impl<'a> BitSetOffsetted<'a> {
    pub fn test(&self, i: usize) -> bool {
        self.bits[self.idx(i) / 32] & (1 << (self.idx(i) % 32)) != 0
    }

    pub fn set(&mut self, i: usize) {
        self.bits[self.idx(i) / 32] |= 1 << (self.idx(i) % 32);
    }

    pub fn clear(&mut self, i: usize) {
        self.bits[self.idx(i) / 32] &= !(1 << (self.idx(i) % 32));
    }

    pub fn offset(&mut self, offset: usize) -> BitSetOffsetted {
        BitSetOffsetted {
            bits: &mut self.bits,
            offset: self.offset + offset,
        }
    }

    fn idx(&self, index: usize) -> usize {
        self.offset + index
    }
}

impl<const N_32: usize> BitSet<N_32> {
    pub fn new() -> Self {
        Self { bits: [0; N_32] }
    }

    pub fn test(&self, i: usize) -> bool {
        self.bits[i / 32] & (1 << (i % 32)) != 0
    }

    pub fn set(&mut self, i: usize, x: bool) {
        if x {
            self.bits[i / 32] |= 1 << (i % 32);
        } else {
            self.bits[i / 32] &= !(1 << (i % 32));
        }
    }

    pub fn offset(&mut self, offset: usize) -> BitSetOffsetted {
        BitSetOffsetted {
            bits: &mut self.bits,
            offset,
        }
    }
}