byte_trie 0.3.0

A compressed trie based on bytes with some weird child node bucket sizes.
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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

pub use crate::BytesKey;
use std::fmt;

pub enum KeyMatch {
    /// The keys are **exactly** the same
    Exact,
    /// The original key is fully matched, new key has more bytes
    FullSelf(usize),
    /// The new key is fully matched, the original has more bytes
    FullOther(usize),
    /// Both keys match partially, and have additional bytes
    Partial(usize),
    /// No parts of the new keys match
    None,
}

#[derive(Debug)]
pub struct ByteKey(Vec<u8>);

impl BytesKey for ByteKey {
    fn new(vec: Vec<u8>) -> Self {
        ByteKey(vec)
    }

    fn from_bytes(bytes: &[u8]) -> Self {
        ByteKey(bytes.to_vec())
    }

    fn get(&self) -> &[u8] {
        &self.0
    }

    fn get_mut(&mut self) -> &mut Vec<u8> {
        &mut self.0
    }
}

impl fmt::Display for ByteKey {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.iter().try_for_each(|b| write!(f, "{:02x}", b))
    }
}

#[derive(Debug)]
pub struct NibbleKey(Vec<u8>);

impl BytesKey for NibbleKey {
    fn new(nibble_vec: Vec<u8>) -> Self {
        NibbleKey(nibble_vec)
    }

    fn from_bytes(bytes: &[u8]) -> Self {
        NibbleKey(
            bytes
                .iter()
                .fold(Vec::with_capacity(bytes.len() * 2), |mut vec, byte| {
                    vec.push(byte >> 4);
                    vec.push(byte & 0x0F);
                    vec
                }),
        )
    }

    fn get(&self) -> &[u8] {
        &self.0
    }

    fn get_mut(&mut self) -> &mut Vec<u8> {
        &mut self.0
    }
}

impl fmt::Display for NibbleKey {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.iter().try_for_each(|b| write!(f, "{:x}", b))
    }
}

#[derive(Debug)]
pub struct BitKey(Vec<u8>);

impl BytesKey for BitKey {
    fn new(bit_vec: Vec<u8>) -> Self {
        BitKey(bit_vec)
    }

    fn from_bytes(bytes: &[u8]) -> Self {
        BitKey(
            bytes
                .iter()
                .fold(Vec::with_capacity(bytes.len() * 8), |mut vec, byte| {
                    vec.push(byte & 0x1);
                    vec.push(byte & 0x2);
                    vec.push(byte & 0x4);
                    vec.push(byte & 0x8);
                    vec.push(byte & 0x10);
                    vec.push(byte & 0x20);
                    vec.push(byte & 0x40);
                    vec.push(byte & 0x80);
                    vec
                }),
        )
    }

    fn get(&self) -> &[u8] {
        &self.0
    }

    fn get_mut(&mut self) -> &mut Vec<u8> {
        &mut self.0
    }
}

impl fmt::Display for BitKey {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        self.0.iter().try_for_each(|b| write!(f, "{:b}", b))
    }
}