autd3-protobuf 30.0.1

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

impl ToMessage for autd3::gain::Uniform {
    type Message = Datagram;

    fn to_msg(
        &self,
        _: Option<&autd3_core::geometry::Geometry>,
    ) -> Result<Self::Message, AUTDProtoBufError> {
        Ok(Self::Message {
            datagram: Some(datagram::Datagram::Gain(Gain {
                gain: Some(gain::Gain::Uniform(Uniform {
                    intensity: Some(self.intensity.to_msg(None)?),
                    phase: Some(self.phase.to_msg(None)?),
                })),
            })),
        })
    }
}

impl FromMessage<Uniform> for autd3::gain::Uniform {
    fn from_msg(msg: Uniform) -> Result<Self, AUTDProtoBufError> {
        Ok(Self {
            phase: autd3_driver::firmware::fpga::Phase::from_msg(
                msg.phase.ok_or(AUTDProtoBufError::DataParseError)?,
            )?,
            intensity: autd3_driver::firmware::fpga::EmitIntensity::from_msg(
                msg.intensity.ok_or(AUTDProtoBufError::DataParseError)?,
            )?,
        })
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use autd3_driver::firmware::fpga::{EmitIntensity, Phase};
    use rand::Rng;

    #[test]
    fn test_phase() {
        let mut rng = rand::rng();

        let g = autd3::gain::Uniform {
            intensity: EmitIntensity(rng.random()),
            phase: Phase(rng.random()),
        };
        let msg = g.to_msg(None).unwrap();
        match msg.datagram {
            Some(datagram::Datagram::Gain(Gain {
                gain: Some(gain::Gain::Uniform(gain)),
                ..
            })) => {
                let g2 = autd3::gain::Uniform::from_msg(gain).unwrap();
                assert_eq!(g.intensity, g2.intensity);
                assert_eq!(g.phase, g2.phase);
            }
            _ => panic!("unexpected datagram type"),
        }
    }
}