diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.rs | 3 | ||||
| -rw-r--r-- | src/view.rs | 12 |
2 files changed, 14 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs index 3f3b866..3c67e9e 100644 --- a/src/main.rs +++ b/src/main.rs @@ -18,7 +18,8 @@ fn main() { let view = View { display_buffer: &display_buffer, - viewer: render_viewer::ppm_exporter, + // viewer: render_viewer::ppm_exporter, + viewer: render_viewer::png_exporter, }; view.display(); } diff --git a/src/view.rs b/src/view.rs index 28ce849..bf3a356 100644 --- a/src/view.rs +++ b/src/view.rs @@ -22,6 +22,7 @@ impl<'a> View<'a> { } pub mod render_viewer { + use crate::global::Pixel; use super::*; pub fn ppm_exporter(data: &DisplayBuffer) -> Result<String, std::io::Error> { @@ -39,4 +40,15 @@ pub mod render_viewer { file.write_all(txt_data.as_bytes())?; Ok(format!("Output rendered to file {}", file_name)) } + + pub fn png_exporter(data: &DisplayBuffer) -> Result<String, std::io::Error> { + let file_name = "output.png"; + let mut imgbuf = image::ImageBuffer::new(IMG_WIDTH as u32, IMG_HEIGHT as u32); + for (x, y, pixel) in imgbuf.enumerate_pixels_mut() { + let Pixel{r, g, b} = data[y as usize][x as usize]; + *pixel = image::Rgb([r, g, b]); + } + imgbuf.save(file_name).unwrap(); + Ok(format!("Output rendered to file {}", file_name)) + } } |