From 7773bfb4c63ab3ea49b428e20ef0946b713fd5f5 Mon Sep 17 00:00:00 2001
From: Akshay <nerdy@peppe.rs>
Date: Fri, 23 Apr 2021 17:16:33 +0530
Subject: add basic obi to png conversions

---
 src/convert.rs | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)
 create mode 100644 src/convert.rs

(limited to 'src/convert.rs')

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
+}
-- 
cgit v1.2.3