mod colorspace; mod decode; mod encode; mod error; mod header; mod pixel; mod utils; pub mod canonical; pub mod consts; pub use crate::colorspace::ColorSpace; pub use crate::decode::qoi_decode_to_vec; pub use crate::encode::qoi_encode_to_vec; pub use crate::error::{Error, Result}; pub use crate::header::Header; #[cfg(test)] mod tests { use std::fs::File; use std::path::PathBuf; use crate::{consts::QOI_MAGIC, qoi_decode_to_vec, qoi_encode_to_vec}; fn read_png(rel_path: &str) -> (u32, u32, u8, Vec) { let get_path = || -> Option { Some(PathBuf::from(file!()).parent()?.parent()?.join(rel_path)) }; let decoder = png::Decoder::new(File::open(get_path().unwrap()).unwrap()); let mut reader = decoder.read_info().unwrap(); let mut buf = vec![0; reader.output_buffer_size()]; let info = reader.next_frame(&mut buf).unwrap(); let bytes = &buf[..info.buffer_size()]; (info.width, info.height, info.color_type.samples() as u8, bytes.to_vec()) } #[test] fn kodim_01() { let (w, h, c, v) = read_png("assets/kodim01.png"); let q = qoi_encode_to_vec(&v, w, h, c, 0).unwrap(); std::fs::write("kodim01.qoi", q.as_slice()).unwrap(); } #[test] fn wikipedia() { let (w, h, c, v) = read_png("assets/en.wikipedia.org.png"); let q = qoi_encode_to_vec(&v, w, h, c, 0).unwrap(); std::fs::write("wikipedia.qoi", q.as_slice()).unwrap(); } #[test] fn roundtrip_3() { let three_raw = include_bytes!("../assets/three.raw").to_vec(); let (w, h, c) = (572, 354, 3); let three_qoi = qoi_encode_to_vec(&three_raw, w, h, c, 0).unwrap(); let (header, three_rtp) = qoi_decode_to_vec(&three_qoi, c).unwrap(); assert_eq!(header.magic, QOI_MAGIC); assert_eq!(header.width, w); assert_eq!(header.height, h); assert_eq!(header.channels, c); assert_eq!(three_rtp.len(), (w as usize) * (h as usize) * 3); assert_eq!(three_raw, three_rtp.as_slice()); } }