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

use crate::{datagram::*, defined::DEFAULT_TIMEOUT, firmware::fpga::ULTRASOUND_PERIOD};

const PULSE_WIDTH_MAX: u16 = ULTRASOUND_PERIOD as u16 / 2;

#[derive(Debug, Clone)]
pub struct PulseWidthEncoder {
    buf: Vec<u16>,
}

impl PulseWidthEncoder {
    /// constructor
    pub fn new(buf: Vec<u16>) -> Result<Self, AUTDInternalError> {
        if buf.iter().any(|&v| v > PULSE_WIDTH_MAX)
            || buf
                != buf
                    .iter()
                    .scan(buf[0], |state, &x| {
                        *state = (*state).max(x);
                        Some(*state)
                    })
                    .collect::<Vec<u16>>()
        {
            return Err(AUTDInternalError::InvalidPulseWidthEncoderData);
        }
        Ok(Self { buf })
    }

    pub fn buf(&self) -> &[u16] {
        &self.buf
    }
}

impl Default for PulseWidthEncoder {
    fn default() -> Self {
        Self::new(
            include_bytes!("asin.dat")
                .iter()
                .enumerate()
                .map(|(i, &v)| {
                    if i >= 0xFF * 0xFF {
                        PULSE_WIDTH_MAX
                    } else {
                        v as u16
                    }
                })
                .collect(),
        )
        .unwrap()
    }
}

impl Datagram for PulseWidthEncoder {
    type O1 = crate::firmware::operation::PulseWidthEncoderOp;
    type O2 = crate::firmware::operation::NullOp;

    fn operation(self) -> (Self::O1, Self::O2) {
        (Self::O1::new(self.buf), Self::O2::default())
    }

    fn timeout(&self) -> Option<Duration> {
        Some(DEFAULT_TIMEOUT)
    }
}

#[cfg(test)]
mod tests {
    use std::f64::consts::PI;

    use crate::derive::EmitIntensity;

    use super::*;

    #[rstest::rstest]
    #[test]
    #[case(Ok(vec![0, 0]), vec![0, 0])]
    #[case(Ok(vec![0, PULSE_WIDTH_MAX]), vec![0, PULSE_WIDTH_MAX])]
    #[case(Ok(vec![PULSE_WIDTH_MAX, PULSE_WIDTH_MAX]), vec![PULSE_WIDTH_MAX, PULSE_WIDTH_MAX])]
    #[case(Err(AUTDInternalError::InvalidPulseWidthEncoderData), vec![0, PULSE_WIDTH_MAX + 1])]
    #[case(Err(AUTDInternalError::InvalidPulseWidthEncoderData), vec![PULSE_WIDTH_MAX, 0])]
    fn new(#[case] expected: Result<Vec<u16>, AUTDInternalError>, #[case] buf: Vec<u16>) {
        assert_eq!(expected, PulseWidthEncoder::new(buf).map(|d| d.buf));
    }

    #[test]
    fn default() {
        let datagram = PulseWidthEncoder::default();
        assert_eq!(Some(DEFAULT_TIMEOUT), datagram.timeout());
        datagram
            .buf()
            .iter()
            .enumerate()
            .filter(|&(i, _)| {
                i <= EmitIntensity::MAX.value() as usize * EmitIntensity::MAX.value() as usize
            })
            .for_each(|(i, &v)| {
                assert_eq!(
                    ((i as f64
                        / EmitIntensity::MAX.value() as f64
                        / EmitIntensity::MAX.value() as f64)
                        .asin()
                        / PI
                        * PULSE_WIDTH_MAX as f64
                        * 2.)
                        .round() as u16,
                    v
                );
            });
        let _ = datagram.operation();
    }
}