rtps-rs 0.1.2

Real Time Publish Subscribe protocol
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
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92

use bit_set::BitSet;
use bit_vec::BitVec;
use speedy::{Context, Readable, Reader, Writable, Writer};
use std::ops::{Deref, DerefMut};

#[derive(Debug, PartialEq)]
pub struct BitSetRef(BitSet);

impl BitSetRef {
    pub fn new() -> BitSetRef {
        BitSetRef(BitSet::with_capacity(0))
    }

    pub fn into_bit_set(self) -> BitSet {
        self.0
    }
}

impl Deref for BitSetRef {
    type Target = BitSet;

    fn deref(&self) -> &BitSet {
        &self.0
    }
}

impl DerefMut for BitSetRef {
    fn deref_mut(&mut self) -> &mut BitSet {
        &mut self.0
    }
}

impl<'a, C: Context> Readable<'a, C> for BitSetRef {
    #[inline]
    fn read_from<R: Reader<'a, C>>(reader: &mut R) -> Result<Self, C::Error> {
        let number_of_bits = reader.read_u32()?;
        let mut bit_vec = BitVec::with_capacity(number_of_bits as usize);
        unsafe {
            let inner = bit_vec.storage_mut();
            for _ in 0..(number_of_bits / 32) {
                inner.push(reader.read_u32()?);
            }
        }
        Ok(BitSetRef(BitSet::from_bit_vec(bit_vec)))
    }

    #[inline]
    fn minimum_bytes_needed() -> usize {
        4
    }
}

impl<C: Context> Writable<C> for BitSetRef {
    #[inline]
    fn write_to<T: ?Sized + Writer<C>>(&self, writer: &mut T) -> Result<(), C::Error> {
        let bytes = self.0.get_ref().storage();
        writer.write_u32((bytes.len() * 32) as u32)?;
        for byte in bytes {
            writer.write_u32(*byte)?;
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    serialization_test!( type = BitSetRef,
    {
        bit_set_empty,
        BitSetRef::new(),
        le = [0x00, 0x00, 0x00, 0x00],
        be = [0x00, 0x00, 0x00, 0x00]
    },
    {
        bit_set_non_zero_size,
        (|| {
            let mut set = BitSetRef::new();
            set.insert(0);
            set.insert(42);
            set.insert(7);
            set
        })(),
        le = [0x40, 0x00, 0x00, 0x00,
              0x81, 0x00, 0x00, 0x00,
              0x00, 0x04, 0x00, 0x00],
        be = [0x00, 0x00, 0x00, 0x40,
              0x00, 0x00, 0x00, 0x81,
              0x00, 0x00, 0x04, 0x00]
    });
}