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
macro_rules! enum_builder {
(
$(#[$comment:meta])*
@U8
EnumName: $enum_name: ident;
EnumVal { $( $enum_var: ident => $enum_val: expr ),* }
) => {
$(#[$comment])*
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
pub enum $enum_name {
$( $enum_var),*
,Unknown(u8)
}
impl $enum_name {
pub fn get_u8(&self) -> u8 {
let x = self.clone();
match x {
$( $enum_name::$enum_var => $enum_val),*
,$enum_name::Unknown(x) => x
}
}
}
impl Codec for $enum_name {
fn encode(&self, bytes: &mut Vec<u8>) {
self.get_u8().encode(bytes);
}
fn read(r: &mut Reader) -> Option<Self> {
Some(match u8::read(r) {
None => return None,
$( Some($enum_val) => $enum_name::$enum_var),*
,Some(x) => $enum_name::Unknown(x)
})
}
}
};
(
$(#[$comment:meta])*
@U16
EnumName: $enum_name: ident;
EnumVal { $( $enum_var: ident => $enum_val: expr ),* }
) => {
$(#[$comment])*
#[derive(Debug, PartialEq, Eq, Clone, Copy)]
pub enum $enum_name {
$( $enum_var),*
,Unknown(u16)
}
impl $enum_name {
pub fn get_u16(&self) -> u16 {
let x = self.clone();
match x {
$( $enum_name::$enum_var => $enum_val),*
,$enum_name::Unknown(x) => x
}
}
}
impl Codec for $enum_name {
fn encode(&self, bytes: &mut Vec<u8>) {
self.get_u16().encode(bytes);
}
fn read(r: &mut Reader) -> Option<Self> {
Some(match u16::read(r) {
None => return None,
$( Some($enum_val) => $enum_name::$enum_var),*
,Some(x) => $enum_name::Unknown(x)
})
}
}
};
}