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

// "aci_png" crate - Licensed under the MIT LICENSE
//  * Copyright (c) 2017-2018  Jeron A. Lau <jeron.lau@plopgrizzly.com>
//
//! Aldaron's Codec Interface / PNG is a library developed by Plop Grizzly to
//! encode(TODO) and decode png and apng(TODO) image files.

#![warn(missing_docs)]
#![doc(html_logo_url = "http://plopgrizzly.com/icon.png",
	html_favicon_url = "http://plopgrizzly.com/icon.png",
	html_root_url = "http://plopgrizzly.com/aci_png/")]

extern crate afi;
extern crate byteorder;

use byteorder::ReadBytesExt;

mod png;

use afi::GraphicBuilder;
pub use afi::{ Graphic, GraphicDecodeErr };

/// Encode PNG data.  Set `alpha` to `true` if you want to save the alpha
/// channel.
pub fn encode(mut graphic: Graphic, alpha: bool) -> Vec<u8> {
	let mut out = vec![];
	graphic.rgba();
	let graphic = graphic.as_bytes();

	png::write(&mut out, graphic.0, graphic.1, graphic.2, alpha).unwrap();

	out
}

/// Decode PNG data.  On success, returns the png as a `Graphic`.
pub fn decode(png: &[u8]) -> Result<Graphic, GraphicDecodeErr> {
	let mut reader = ::std::io::Cursor::new(png);
	let i: Result<png::Image<u8>, png::Error> = png::read(&mut reader);
	let image = if let Result::Ok(p) = i { p }
		else { return Err(GraphicDecodeErr::IncorrectFormat) };
	let bytes: Vec<u8> = image.buf;

	let size = (image.w * image.h) as usize;
	let mut out: Vec<u32> = Vec::with_capacity(size);

	for i in 0..size {
		out.push((&bytes[i * 4 .. i * 4 + 4])
			.read_u32::<byteorder::LittleEndian>()
			.unwrap()
		);
	}

	Ok(GraphicBuilder::new().rgba(image.w, image.h, out))
}