rlvgl_core/plugins/
jpeg.rs

1//! JPEG decoder for Color pixel arrays.
2use crate::widget::Color;
3use alloc::vec::Vec;
4use jpeg_decoder::{Decoder, Error, PixelFormat};
5use std::io::Cursor;
6
7/// Decode a JPEG image into a vector of RGB [`Color`]s.
8pub fn decode(data: &[u8]) -> Result<(Vec<Color>, u16, u16), Error> {
9    let mut decoder = Decoder::new(Cursor::new(data));
10    let pixels_raw = decoder.decode()?;
11    let info = decoder
12        .info()
13        .ok_or_else(|| Error::Format("missing image info".into()))?;
14    let mut pixels = Vec::with_capacity(info.width as usize * info.height as usize);
15    match info.pixel_format {
16        PixelFormat::L8 => {
17            for &v in &pixels_raw {
18                pixels.push(Color(v, v, v));
19            }
20        }
21        PixelFormat::L16 => {
22            for chunk in pixels_raw.chunks_exact(2) {
23                let val = u16::from_be_bytes([chunk[0], chunk[1]]);
24                let v = (val / 257) as u8;
25                pixels.push(Color(v, v, v));
26            }
27        }
28        PixelFormat::RGB24 => {
29            for chunk in pixels_raw.chunks_exact(3) {
30                pixels.push(Color(chunk[0], chunk[1], chunk[2]));
31            }
32        }
33        PixelFormat::CMYK32 => {
34            for chunk in pixels_raw.chunks_exact(4) {
35                let c = chunk[0] as f32 / 255.0;
36                let m = chunk[1] as f32 / 255.0;
37                let y = chunk[2] as f32 / 255.0;
38                let k = chunk[3] as f32 / 255.0;
39                let r = (1.0 - (c * (1.0 - k) + k)) * 255.0;
40                let g = (1.0 - (m * (1.0 - k) + k)) * 255.0;
41                let b = (1.0 - (y * (1.0 - k) + k)) * 255.0;
42                pixels.push(Color(r as u8, g as u8, b as u8));
43            }
44        }
45    }
46    Ok((pixels, info.width, info.height))
47}
48
49#[cfg(test)]
50mod tests {
51    use super::*;
52    use base64::Engine;
53
54    const RED_DOT_JPEG: &str = "/9j/4AAQSkZJRgABAQAAAQABAAD/2wBDAAgGBgcGBQgHBwcJCQgKDBQNDAsLDBkSEw8UHRofHh0aHBwgJC4nICIsIxwcKDcpLDAxNDQ0Hyc5PTgyPC4zNDL/2wBDAQkJCQwLDBgNDRgyIRwhMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjIyMjL/wAARCAABAAEDASIAAhEBAxEB/8QAHwAAAQUBAQEBAQEAAAAAAAAAAAECAwQFBgcICQoL/8QAtRAAAgEDAwIEAwUFBAQAAAF9AQIDAAQRBRIhMUEGE1FhByJxFDKBkaEII0KxwRVS0fAkM2JyggkKFhcYGRolJicoKSo0NTY3ODk6Q0RFRkdISUpTVFVWV1hZWmNkZWZnaGlqc3R1dnd4eXqDhIWGh4iJipKTlJWWl5iZmqKjpKWmp6ipqrKztLW2t7i5usLDxMXGx8jJytLT1NXW19jZ2uHi4+Tl5ufo6erx8vP09fb3+Pn6/8QAHwEAAwEBAQEBAQEBAQAAAAAAAAECAwQFBgcICQoL/8QAtREAAgECBAQDBAcFBAQAAQJ3AAECAxEEBSExBhJBUQdhcRMiMoEIFEKRobHBCSMzUvAVYnLRChYkNOEl8RcYGRomJygpKjU2Nzg5OkNERUZHSElKU1RVVldYWVpjZGVmZ2hpanN0dXZ3eHl6goOEhYaHiImKkpOUlZaXmJmaoqOkpaanqKmqsrO0tba3uLm6wsPExcbHyMnK0tPU1dbX2Nna4uPk5ebn6Onq8vP09fb3+Pn6/9oADAMBAAIRAxEAPwDi6KKK+ZP3E//Z";
55
56    #[test]
57    fn decode_red_dot() {
58        let data = base64::engine::general_purpose::STANDARD
59            .decode(RED_DOT_JPEG)
60            .unwrap();
61        let (pixels, w, h) = decode(&data).unwrap();
62        assert_eq!((w, h), (1, 1));
63        assert!(pixels[0].0 >= 250 && pixels[0].1 == 0 && pixels[0].2 == 0);
64    }
65}
rlvgl_core/plugins/jpeg.rs

rlvgl_core/plugins/
jpeg.rs
