autd3-protobuf 27.0.0

Protocol Buffer
Documentation 
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

use autd3_driver::derive::EmitIntensity;

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

impl ToMessage for autd3_gain_holo::EmissionConstraint {
    type Message = EmissionConstraint;

    fn to_msg(&self, _: Option<&autd3_driver::geometry::Geometry>) -> Self::Message {
        match self {
            autd3_gain_holo::EmissionConstraint::Normalize => Self::Message {
                constraint: Some(emission_constraint::Constraint::Normalize(
                    NormalizeConstraint {},
                )),
            },
            autd3_gain_holo::EmissionConstraint::Multiply(value) => Self::Message {
                constraint: Some(emission_constraint::Constraint::Multiply(
                    MultiplyConstraint { value: *value as _ },
                )),
            },
            autd3_gain_holo::EmissionConstraint::Uniform(value) => Self::Message {
                constraint: Some(emission_constraint::Constraint::Uniform(
                    UniformConstraint {
                        value: Some(value.to_msg(None)),
                    },
                )),
            },
            autd3_gain_holo::EmissionConstraint::Clamp(min, max) => Self::Message {
                constraint: Some(emission_constraint::Constraint::Clamp(ClampConstraint {
                    min: Some(min.to_msg(None)),
                    max: Some(max.to_msg(None)),
                })),
            },
            _ => unimplemented!(),
        }
    }
}

impl FromMessage<EmissionConstraint> for autd3_gain_holo::EmissionConstraint {
    fn from_msg(msg: &EmissionConstraint) -> Result<Self, AUTDProtoBufError> {
        match msg.constraint {
            Some(emission_constraint::Constraint::Normalize(_)) => {
                Ok(autd3_gain_holo::EmissionConstraint::Normalize)
            }
            Some(emission_constraint::Constraint::Multiply(ref v)) => {
                Ok(autd3_gain_holo::EmissionConstraint::Multiply(v.value as _))
            }
            Some(emission_constraint::Constraint::Uniform(ref v)) => Ok(
                autd3_gain_holo::EmissionConstraint::Uniform(EmitIntensity::from_msg(
                    v.value.as_ref().ok_or(AUTDProtoBufError::DataParseError)?,
                )?),
            ),
            Some(emission_constraint::Constraint::Clamp(ref v)) => {
                Ok(autd3_gain_holo::EmissionConstraint::Clamp(
                    EmitIntensity::from_msg(
                        v.min.as_ref().ok_or(AUTDProtoBufError::DataParseError)?,
                    )?,
                    EmitIntensity::from_msg(
                        v.max.as_ref().ok_or(AUTDProtoBufError::DataParseError)?,
                    )?,
                ))
            }
            None => Err(AUTDProtoBufError::DataParseError),
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use autd3_driver::derive::EmitIntensity;
    use rand::Rng;

    #[test]
    fn test_emission_constraint_normalize() {
        let v = autd3_gain_holo::EmissionConstraint::Normalize;
        let msg = v.to_msg(None);
        let v2 = autd3_gain_holo::EmissionConstraint::from_msg(&msg).unwrap();
        assert_eq!(v, v2);
    }

    #[test]
    fn test_emission_constraint_uniform() {
        let mut rng = rand::thread_rng();
        let v = autd3_gain_holo::EmissionConstraint::Uniform(EmitIntensity::new(rng.gen()));
        let msg = v.to_msg(None);
        let v2 = autd3_gain_holo::EmissionConstraint::from_msg(&msg).unwrap();
        assert_eq!(v, v2);
    }

    #[test]
    fn test_emission_constraint_clamp() {
        let mut rng = rand::thread_rng();
        let v = autd3_gain_holo::EmissionConstraint::Clamp(
            EmitIntensity::new(rng.gen()),
            EmitIntensity::new(rng.gen()),
        );
        let msg = v.to_msg(None);
        let v2 = autd3_gain_holo::EmissionConstraint::from_msg(&msg).unwrap();
        assert_eq!(v, v2);
    }
}