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

use joycon_sys::mcu::{ir::Resolution, *};
use joycon_sys::*;

pub struct Image {
    buffer: Box<[[u8; 300]; 0x100]>,
    resolution: ir::Resolution,
    prev_fragment_id: u8,
    changing_resolution: bool,
    pub last_image: Option<image::GrayImage>,
}

impl Image {
    pub fn new() -> Image {
        Image {
            buffer: Box::new([[0; 300]; 0x100]),
            resolution: Resolution::default(),
            prev_fragment_id: 0,
            changing_resolution: false,
            last_image: None,
        }
    }

    pub fn change_resolution(&mut self, resolution: ir::Resolution) {
        self.resolution = resolution;
        self.changing_resolution = true;
    }

    pub fn handle(&mut self, report: &MCUReport) -> [Option<OutputReport>; 2] {
        // TODO: handle lossed packets
        if let Some(packet) = report.ir_data() {
            if self.changing_resolution {
                if packet.frag_number != 0 {
                    return [Some(OutputReport::ir_ack(packet.frag_number)), None];
                }
                self.changing_resolution = false;
            }

            self.buffer[packet.frag_number as usize] = packet.img_fragment;
            let resend = if packet.frag_number > 0
                && self.prev_fragment_id > 0
                && packet.frag_number - 1 > self.prev_fragment_id
            {
                println!("requesting again packet {}", packet.frag_number - 1);
                Some(OutputReport::ir_resend(packet.frag_number - 1))
            } else {
                None
            };
            let (width, height) = self.resolution.size();
            let mut buffer = Vec::with_capacity((width * height) as usize);
            for fragment in self
                .buffer
                .iter()
                .take(self.resolution.max_fragment_id() as usize + 1)
            {
                buffer.extend(fragment.iter());
            }
            self.last_image = Some(image::imageops::rotate90(
                &image::GrayImage::from_raw(width, height, buffer).unwrap(),
            ));
            //println!("got packet {}", packet.frag_number);
            if packet.frag_number == self.resolution.max_fragment_id() {
                self.prev_fragment_id = 0;
            } else {
                self.prev_fragment_id = packet.frag_number;
            }
            [Some(OutputReport::ir_ack(packet.frag_number)), resend]
        } else if report.id() == MCUReportId::Empty {
            [
                Some(OutputReport::ir_resend(self.prev_fragment_id + 1)),
                None,
            ]
        } else if report.id() == MCUReportId::EmptyAwaitingCmd {
            [Some(OutputReport::ir_ack(self.prev_fragment_id)), None]
        } else {
            [None, None]
        }
    }
}

impl Default for Image {
    fn default() -> Self {
        Self::new()
    }
}