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
#[macro_export]
macro_rules! _probor_index {
($n: expr) => { $n }
}
#[macro_export]
macro_rules! _probor_max {
() => { 0 };
($n:expr, $($tail:expr,)*) => { ::std::cmp::max(_probor_max!($($tail,)*), $n) };
}
#[macro_export]
macro_rules! _probor_count {
() => { 0 };
($item:ident $($tail:ident)*) => { _probor_count!($($tail)*) + 1 };
}
#[macro_export]
macro_rules! _probor_skip_to {
($encoder:expr, $idx:expr, $n:expr) => {
debug_assert!($idx < $n);
$idx += 1;
while $idx < $n {
try!($encoder.null());
$idx += 1;
}
}
}
#[macro_export]
macro_rules! _probor_encode_pos_field {
($encoder:expr, $me:expr, $idx:expr, $field:ident #$n:tt optional) => {
_probor_skip_to!($encoder, $idx, $n);
match $me.$field {
Some(ref x) => try!($crate::Encodable::encode(x, $encoder)),
None => try!($encoder.null()),
}
};
($encoder:expr, $me:expr, $idx:expr, $field:ident #$n:tt) => {
_probor_skip_to!($encoder, $idx, $n);
try!($crate::Encodable::encode(&$me.$field, $encoder));
};
($encoder:expr, $me:expr, $idx:expr, $field:ident optional) => {
_probor_skip_to!($encoder, $idx, $n);
match $me.$field {
Some(ref x) => try!($crate::Encodable::encode(x, $encoder)),
None => try!($encoder.null()),
}
};
($encoder:expr, $me:expr, $idx:expr, $field:ident) => {
_probor_skip_to!($encoder, $idx, $n);
try!($crate::Encodable::encode(&$me.$field, $encoder));
};
}
#[macro_export]
macro_rules! _probor_encode_field {
($encoder:expr, $me:expr, $idx:expr, $field:ident #$n:tt optional) => {
try!($encoder.u64(_probor_index!($n)));
match $me.$field {
Some(ref x) => try!($crate::Encodable::encode(x, $encoder)),
None => try!($encoder.null()),
}
};
($encoder:expr, $me:expr, $idx:expr, $field:ident #$n:tt) => {
try!($encoder.u64(_probor_index!($n)));
try!($crate::Encodable::encode(&$me.$field, $encoder));
};
($encoder:expr, $me:expr, $idx:expr, $field:ident optional) => {
try!($encoder.text(stringify!($field)));
match $me.$field {
Some(ref x) => try!($crate::Encodable::encode(x, $encoder)),
None => try!($encoder.null()),
}
};
($encoder:expr, $me:expr, $idx:expr, $field:ident) => {
try!($encoder.text(stringify!($field)));
try!($crate::Encodable::encode(&$me.$field, $encoder));
};
}
#[macro_export]
macro_rules! probor_enc_struct {
($encoder:expr, $me:expr, { $( $item:ident => (#$n:tt $($props:tt)* ), )* } )
=> {{
try!($encoder.array(_probor_max!( $($n,)* ) + 1));
let mut idx = -1;
$(
_probor_encode_pos_field!($encoder, $me, idx, $item #$n $($props)*);
)*
}};
($encoder:expr, $me:expr, { $( $item:ident => ( $($props:tt)* ), )* } )
=> {{
try!($encoder.object(_probor_count!( $($item)* )));
$(
_probor_encode_field!($encoder, $me, idx, $item $($props)*);
)*
}};
}