autd3-protobuf 26.0.0

Protocol Buffer
Documentation
use std::time::Duration;

use crate::{
    pb::*,
    traits::{FromMessage, ToMessage},
    AUTDProtoBufError,
};

impl ToMessage for autd3_driver::firmware::fpga::SamplingConfig {
    type Message = SamplingConfig;

    fn to_msg(&self, _: Option<&autd3_driver::geometry::Geometry>) -> Self::Message {
        Self::Message {
            config: Some(match *self {
                autd3::derive::SamplingConfig::Freq(value) => {
                    sampling_config::Config::Freq(SamplingConfigFreq { value: value.hz() })
                }
                autd3::derive::SamplingConfig::FreqNearest(value) => {
                    sampling_config::Config::FreqNearest(SamplingConfigFreqNearest {
                        value: value.hz(),
                    })
                }
                autd3::derive::SamplingConfig::Period(value) => {
                    sampling_config::Config::Period(SamplingConfigPeriod {
                        value: value.as_nanos() as _,
                    })
                }
                autd3::derive::SamplingConfig::PeriodNearest(value) => {
                    sampling_config::Config::PeriodNearest(SamplingConfigPeriodNearest {
                        value: value.as_nanos() as _,
                    })
                }
                autd3::derive::SamplingConfig::DivisionRaw(value) => {
                    sampling_config::Config::DivisionRaw(SamplingConfigDivisionRaw { value })
                }
                autd3::derive::SamplingConfig::Division(value) => {
                    sampling_config::Config::Division(SamplingConfigDivision { value })
                }
                _ => unimplemented!(),
            }),
        }
    }
}

impl FromMessage<SamplingConfig> for autd3_driver::firmware::fpga::SamplingConfig {
    fn from_msg(msg: &SamplingConfig) -> Result<Self, AUTDProtoBufError> {
        msg.config
            .as_ref()
            .ok_or(AUTDProtoBufError::DataParseError)
            .map(|config| match *config {
                sampling_config::Config::Freq(SamplingConfigFreq { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::Freq(
                        value * autd3_driver::defined::Hz,
                    )
                }
                sampling_config::Config::FreqNearest(SamplingConfigFreqNearest { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::FreqNearest(
                        value * autd3_driver::defined::Hz,
                    )
                }
                sampling_config::Config::Period(SamplingConfigPeriod { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::Period(Duration::from_nanos(
                        value,
                    ))
                }
                sampling_config::Config::PeriodNearest(SamplingConfigPeriodNearest { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::PeriodNearest(
                        Duration::from_nanos(value),
                    )
                }
                sampling_config::Config::Division(SamplingConfigDivision { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::Division(value)
                }
                sampling_config::Config::DivisionRaw(SamplingConfigDivisionRaw { value }) => {
                    autd3_driver::firmware::fpga::SamplingConfig::DivisionRaw(value)
                }
            })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use autd3_driver::{
        derive::SamplingConfig,
        firmware::fpga::{SAMPLING_FREQ_DIV_MAX, SAMPLING_FREQ_DIV_MIN},
    };
    use rand::Rng;

    #[test]
    fn test_sampling_config() {
        let mut rng = rand::thread_rng();
        let v = SamplingConfig::DivisionRaw(
            rng.gen_range(SAMPLING_FREQ_DIV_MIN..SAMPLING_FREQ_DIV_MAX),
        );
        let msg = v.to_msg(None);
        let v2 = SamplingConfig::from_msg(&msg).unwrap();
        assert_eq!(v, v2);
    }
}