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

#![allow(dead_code)]

use skia_safe::Surface;

use std::fs::File;
use std::io::Write;

use std::cell::Cell;

use crate::drawing::scene::{draw_scene, DrawScene};
use crate::engine::scene::Scene;
use crate::engine::NodeRef;

/// A scene renderer that renders to an image file.
/// Image encoding is currently hard-coded to PNG.
pub struct SkiaImageRenderer {
    pub surface: Surface,
    pub filename: String,
    pub image_format: skia_safe::EncodedImageFormat,
}
impl SkiaImageRenderer {
    pub fn new(height: i32, width: i32, filename: impl Into<String>) -> Self {
        let filename = filename.into();
        let surface = skia_safe::surfaces::raster_n32_premul((width, height)).expect("no surface!");
        let image_format = skia_safe::EncodedImageFormat::PNG;
        Self {
            surface,
            filename,
            image_format,
        }
    }

    pub fn create<S: Into<String>>(width: i32, height: i32, filename: S) -> Cell<Self> {
        Cell::new(Self::new(width, height, filename.into()))
    }

    pub fn surface(&self) -> Surface {
        self.surface.clone()
    }

    pub fn save(&mut self) {
        let mut file = File::create(&self.filename).expect("no file!");
        let image = self.surface.image_snapshot();
        let data = image.encode(None, self.image_format, None).unwrap();
        file.write_all(&data).expect("no write!");
    }
}

impl DrawScene for SkiaImageRenderer {
    fn draw_scene(&self, scene: &Scene, root_id: NodeRef, _damage: Option<skia_safe::Rect>) {
        let mut surface = self.surface();

        let c = surface.canvas();
        draw_scene(c, scene, root_id);
    }
}