autd3-rs 0.3.0

Core async client library for the AUTD3 phased-array kit.
Documentation
use super::Command;
use crate::datagram::DatagramBuilder;
use crate::operation::{ChangePatternBank, ConfigPattern, WritePatternBuffer};
use crate::value::{Emission, LoopBehavior, PatternBank, SamplingConfig, TransitionMode};
use core::num::NonZeroU16;

#[derive(Clone, Copy, Debug)]
pub struct Pattern<'a> {
    pub bank: PatternBank,
    pub emissions: &'a [Vec<Emission>],
}

impl<'a> Pattern<'a> {
    #[must_use]
    pub fn new(emissions: &'a [Vec<Emission>]) -> Self {
        Self::with_bank(PatternBank::B0, emissions)
    }

    #[must_use]
    pub fn with_bank(bank: PatternBank, emissions: &'a [Vec<Emission>]) -> Self {
        Self { bank, emissions }
    }
}

impl<'a> Command<'a> for Pattern<'a> {
    fn expand(self, builder: &mut DatagramBuilder<'a>) {
        builder
            .push(WritePatternBuffer {
                bank: self.bank,
                index: 0,
                emissions: self.emissions,
            })
            .push(ConfigPattern {
                bank: self.bank,
                config: SamplingConfig::new(NonZeroU16::MAX),
                size: 1,
                loop_behavior: LoopBehavior::Infinite,
            })
            .push(ChangePatternBank {
                bank: self.bank,
                transition_mode: TransitionMode::Immediate,
            });
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::geometry::Autd3;
    use crate::mirror::FREQ_DIV_NO_LIMIT;
    use crate::protocol::Cmd;

    #[test]
    fn pattern_expands_to_write_config_then_change_bank() {
        let patterns = vec![vec![Emission::default(); Autd3::NUM_TRANSDUCERS]; 2];
        let mut b = DatagramBuilder::new(2);
        b.push(Pattern::new(&patterns));
        let datagrams = b.build().unwrap();

        assert_eq!(datagrams.len(), 3);
        assert_eq!(
            datagrams.frame(0).unwrap().datagrams()[0].cmd,
            Cmd::WritePatternBuffer
        );
        let cfg = datagrams.frame(1).unwrap();
        assert_eq!(cfg.datagrams()[0].cmd, Cmd::ConfigPattern);
        assert_eq!(
            &cfg.datagrams()[0].payload[2..4],
            &FREQ_DIV_NO_LIMIT.to_le_bytes()
        );

        let chg = datagrams.frame(2).unwrap();
        assert_eq!(chg.datagrams()[0].cmd, Cmd::ChangePatternBank);
        assert_eq!(chg.datagrams()[0].payload[0], 0, "bank B0");
        assert_eq!(chg.datagrams()[0].payload[1], 0xFF, "IMMEDIATE");
    }
}