autd3-protobuf 27.0.0

Protocol Buffer
Documentation
use autd3_driver::ethercat::{DcSysTime, ECAT_DC_SYS_TIME_BASE};

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

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

    fn to_msg(&self, _: Option<&autd3_driver::geometry::Geometry>) -> Self::Message {
        Self::Message {
            mode: Some(match *self {
                autd3::derive::TransitionMode::SyncIdx => {
                    transition_mode::Mode::SyncIdx(TransitionModeSyncIdx {})
                }
                autd3::derive::TransitionMode::SysTime(value) => {
                    transition_mode::Mode::SysTime(TransitionModeSysTime {
                        value: value.sys_time(),
                    })
                }
                autd3::derive::TransitionMode::GPIO(value) => {
                    transition_mode::Mode::Gpio(TransitionModeGpio { value: value as _ })
                }
                autd3::derive::TransitionMode::Ext => {
                    transition_mode::Mode::Ext(TransitionModeExt {})
                }
                autd3::derive::TransitionMode::Immediate => {
                    transition_mode::Mode::Immediate(TransitionModeImmediate {})
                }
                _ => unimplemented!(),
            }),
        }
    }
}

impl FromMessage<TransitionMode> for autd3_driver::firmware::fpga::TransitionMode {
    fn from_msg(msg: &TransitionMode) -> Result<Self, AUTDProtoBufError> {
        let mode = msg.mode.as_ref().ok_or(AUTDProtoBufError::DataParseError)?;
        Ok(match *mode {
            transition_mode::Mode::SyncIdx(TransitionModeSyncIdx {}) => {
                autd3_driver::firmware::fpga::TransitionMode::SyncIdx
            }
            transition_mode::Mode::SysTime(TransitionModeSysTime { value }) => {
                autd3_driver::firmware::fpga::TransitionMode::SysTime(
                    DcSysTime::from_utc(
                        ECAT_DC_SYS_TIME_BASE + std::time::Duration::from_nanos(value),
                    )
                    .unwrap(),
                )
            }
            transition_mode::Mode::Gpio(TransitionModeGpio { value }) => {
                autd3_driver::firmware::fpga::TransitionMode::GPIO(
                    autd3_driver::firmware::fpga::GPIOIn::from(GpioIn::try_from(value)?),
                )
            }
            transition_mode::Mode::Ext(TransitionModeExt {}) => {
                autd3_driver::firmware::fpga::TransitionMode::Ext
            }
            transition_mode::Mode::Immediate(TransitionModeImmediate {}) => {
                autd3_driver::firmware::fpga::TransitionMode::Immediate
            }
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[rstest::rstest]
    #[test]
    #[case(autd3_driver::firmware::fpga::TransitionMode::SyncIdx)]
    #[case(
        autd3_driver::firmware::fpga::TransitionMode::SysTime(
            DcSysTime::from_utc(ECAT_DC_SYS_TIME_BASE).unwrap()
                + std::time::Duration::from_nanos(1)
        ),
    )]
    #[case(autd3_driver::firmware::fpga::TransitionMode::GPIO(
        autd3_driver::firmware::fpga::GPIOIn::I0
    ))]
    #[case(autd3_driver::firmware::fpga::TransitionMode::Ext)]
    fn test_transition_mode(#[case] expect: autd3_driver::firmware::fpga::TransitionMode) {
        let msg = expect.to_msg(None);
        assert_eq!(
            expect,
            autd3_driver::firmware::fpga::TransitionMode::from_msg(&msg).unwrap()
        );
    }
}