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

pub mod eluv;
pub mod extended_line;

/// Macro for implementing the [Encoder](crate::encoder::Encoder) trait for
/// complex methods. It assumes that method is only composed by other methods.
///
/// Requires the method to have `methods` field being vector containing type [Encoder](crate::encoder::Encoder).
#[macro_export]
macro_rules! impl_complex_encoder {
    ($t:ident, $c:ident) => {
        impl crate::encoder::Encoder<$c> for $t {
            fn partial_encode(
                &self,
                context: &mut $c,
                data: &mut dyn Iterator<Item = crate::binary::Bit>,
            ) -> Result<crate::encoder::EncoderResult, Box<dyn std::error::Error>> {
                let mut is_data_still_available = crate::encoder::EncoderResult::Success;
                for encoder in &self.methods {
                    if let crate::encoder::EncoderResult::NoDataLeft =
                        encoder.partial_encode(context, data)?
                    {
                        is_data_still_available = crate::encoder::EncoderResult::NoDataLeft;
                        break;
                    }
                }
                Ok(is_data_still_available)
            }
        }

        impl crate::encoder::Capacity for $t {
            fn bitrate(&self) -> usize {
                self.methods
                    .iter()
                    .fold(0, |acc, encoder| acc + encoder.bitrate())
            }
        }
    };
}

/// Macro for implementing the [Decoder](crate::decoder::Decoder) trait for
/// complex methods. It assumes that method is only composed by other methods.
///
/// Requires the method to have `methods` field being vector containing type [Decoder](crate::decoder::Decoder).
#[macro_export]
macro_rules! impl_complex_decoder {
    ($t:ident, $c:ident) => {
        impl crate::decoder::Decoder<$c> for $t {
            fn partial_decode(
                &self,
                context: &$c,
            ) -> Result<Vec<crate::binary::Bit>, crate::context::ContextError> {
                let mut secret_data = Vec::default();
                for decoder in &self.methods {
                    let mut result = decoder.partial_decode(context)?;
                    secret_data.append(&mut result);
                }
                Ok(secret_data)
            }
        }
    };
}