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

use crate::{
    derive::TransitionMode,
    error::AUTDInternalError,
    firmware::{
        fpga::Segment,
        operation::{cast, Remains, SwapSegmentOperation, TypeTag},
    },
    geometry::{Device, Geometry},
};

#[repr(C, align(2))]
struct GainUpdate {
    tag: TypeTag,
    segment: u8,
}

pub struct GainSwapSegmentOp {
    segment: Segment,
    remains: Remains,
}

impl SwapSegmentOperation for GainSwapSegmentOp {
    fn new(segment: Segment, _: TransitionMode) -> Self {
        Self {
            segment,
            remains: Default::default(),
        }
    }

    fn pack(&mut self, device: &Device, tx: &mut [u8]) -> Result<usize, AUTDInternalError> {
        *cast::<GainUpdate>(tx) = GainUpdate {
            tag: TypeTag::GainSwapSegment,
            segment: self.segment as u8,
        };

        self.remains[device] -= 1;
        Ok(std::mem::size_of::<GainUpdate>())
    }

    fn required_size(&self, _: &Device) -> usize {
        std::mem::size_of::<GainUpdate>()
    }

    fn init(&mut self, geometry: &Geometry) -> Result<(), AUTDInternalError> {
        self.remains.init(geometry, |_| 1);
        Ok(())
    }

    fn is_done(&self, device: &Device) -> bool {
        self.remains.is_done(device)
    }
}