1 files changed, 26 insertions, 0 deletions

diff --git a/src/convert.rs b/src/convert.rs
new file mode 100644
index 0000000..368b2ef
--- /dev/null
+++ b/src/convert.rs
@@ -0,0 +1,26 @@
+use crate::{
+    error::{OBIError, OBIResult},
+    CompressionType, Image,
+};
+
+use std::io::Write;
+
+use png::{BitDepth, ColorType, Encoder, Writer};
+
+pub fn to_png<W: Write>(writer: W, img: &Image) -> Writer<W> {
+    let mut encoder = Encoder::new(writer, img.width(), img.height());
+    encoder.set_color(ColorType::RGBA);
+    encoder.set_depth(BitDepth::Eight);
+    let mut writer = encoder.write_header().unwrap();
+    writer
+        .write_image_data(
+            &img.data
+                .iter()
+                .take((img.width() * img.height()) as usize)
+                .map(|x| if *x { vec![255; 4] } else { vec![0, 0, 0, 255] })
+                .flatten()
+                .collect::<Vec<_>>()[..],
+        )
+        .unwrap();
+    writer
+}