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
use std::ops::Deref;
use evcxr_runtime;
use image;
pub trait ImageDisplay {
fn evcxr_display(&self);
}
impl<P: image::Pixel<Subpixel = u8> + 'static, C> ImageDisplay for image::ImageBuffer<P, C>
where
C: Deref<Target = [P::Subpixel]>,
{
fn evcxr_display(&self) {
let mut buffer = Vec::new();
image::png::PngEncoder::new(&mut buffer)
.encode(&**self, self.width(), self.height(), P::COLOR_TYPE)
.unwrap();
evcxr_runtime::mime_type("image/png").bytes(&buffer);
}
}
#[cfg(test)]
mod tests {
#[test]
fn rgb_image() {
let img = image::ImageBuffer::from_fn(10, 10, |x, y| {
if (x as i32 - y as i32).abs() < 3 {
image::Rgb([0, 0, 255])
} else {
image::Rgb([0, 0, 0])
}
});
use super::ImageDisplay;
img.evcxr_display();
}
#[test]
fn gray_image() {
let img = image::ImageBuffer::from_fn(10, 10, |x, y| {
if (x as i32 - y as i32).abs() < 3 {
image::Luma([255])
} else {
image::Luma([0])
}
});
use super::ImageDisplay;
img.evcxr_display();
}
}