lime/project/src/graphics/format/PNG.cpp

extern "C" {
	
	#include <png.h>
	#include <pngstruct.h>
	#define PNG_SIG_SIZE 8
	
}

#include <setjmp.h>
#include <graphics/format/PNG.h>
#include <utils/FileIO.h>
#include <utils/QuickVec.h>


namespace lime {
	
	
	struct ReadBuf {
		
		
		ReadBuf (const uint8 *inData, int inLen) : mData (inData), mLen (inLen) {}
		
		bool Read (uint8 *outBuffer, int inN) {
			
			if (inN > mLen) {
				
				memset (outBuffer, 0, inN);
				return false;
				
			}
			
			memcpy (outBuffer, mData, inN);
			mData += inN;
			mLen -= inN;
			return true;
			
		}
		
		const uint8 *mData;
		int mLen;
		
		
	};
	
	
	static void user_error_fn (png_structp png_ptr, png_const_charp error_msg) {
		
		longjmp (png_ptr->jmp_buf_local, 1);
		
	}
	
	
	static void user_warning_fn (png_structp png_ptr, png_const_charp warning_msg) {}
	
	
	static void user_read_data_fn (png_structp png_ptr, png_bytep data, png_size_t length) {
		
		png_voidp buffer = png_get_io_ptr (png_ptr);
		((ReadBuf *)buffer)->Read (data, length);
		
	}
	
	
	void user_write_data (png_structp png_ptr, png_bytep data, png_size_t length) {
		
		QuickVec<unsigned char> *buffer = (QuickVec<unsigned char> *)png_get_io_ptr (png_ptr);
		buffer->append ((unsigned char *)data,(int)length);
		
	}
	
	
	void user_flush_data (png_structp png_ptr) {}
	
	
	bool PNG::Decode (Resource *resource, ImageBuffer *imageBuffer) {
		
		unsigned char png_sig[PNG_SIG_SIZE];
		png_structp png_ptr;
		png_infop info_ptr;
		png_uint_32 width, height;
		int bit_depth, color_type, interlace_type;
		
		FILE *file = NULL;
		
		if (resource->path) {
			
			file = lime::fopen (resource->path, "rb");
			if (!file) return false;
			
			// verify the PNG signature
			int read = lime::fread (png_sig, PNG_SIG_SIZE, 1, file);
			if (png_sig_cmp (png_sig, 0, PNG_SIG_SIZE)) {
				
				lime::fclose (file);
				return false;
				
			} else {
				
				lime::fseek (file, 0, 0);
				
			}
			
		} else {
			
			// TODO: optimize ByteArray Format check?
			
		}
		
		if ((png_ptr = png_create_read_struct (PNG_LIBPNG_VER_STRING, NULL, NULL, NULL)) == NULL) {
			
			if (file) lime::fclose (file);
			return false;
			
		}
		
		if ((info_ptr = png_create_info_struct (png_ptr)) == NULL) {
			
			png_destroy_read_struct (&png_ptr, (png_infopp)NULL, (png_infopp)NULL);
			if (file) lime::fclose (file);
			return false;
			
		}
		
		// sets the point which libpng will jump back to in the case of an error
		if (setjmp (png_jmpbuf (png_ptr))) {
			
			png_destroy_read_struct (&png_ptr, &info_ptr, (png_infopp)NULL);
			if (file) lime::fclose (file);
			return false;
			
		}
		
		if (file) {
			
			png_init_io (png_ptr, file);
			
		} else {
			
			ReadBuf buffer (resource->data->Bytes (), resource->data->Size ());
			png_set_read_fn (png_ptr, (void *)&buffer, user_read_data_fn);
			
		}
		
		png_read_info (png_ptr, info_ptr);
		
		png_get_IHDR(png_ptr, info_ptr, &width, &height, &bit_depth, &color_type, &interlace_type, NULL, NULL);
		
		bool has_alpha = (color_type == PNG_COLOR_TYPE_GRAY_ALPHA || color_type == PNG_COLOR_TYPE_RGB_ALPHA || png_get_valid (png_ptr, info_ptr, PNG_INFO_tRNS));
		
		png_set_expand (png_ptr);
		png_set_filler (png_ptr, 0xff, PNG_FILLER_AFTER);
		//png_set_gray_1_2_4_to_8 (png_ptr);
		png_set_palette_to_rgb (png_ptr);
		png_set_gray_to_rgb (png_ptr);
		
		if (bit_depth == 16)
			png_set_strip_16 (png_ptr);
		
		//png_set_bgr (png_ptr);
		
		int bpp = 4;
		const unsigned int stride = width * bpp;
		imageBuffer->Resize (width, height, bpp);
		
		unsigned char *bytes = imageBuffer->data->Bytes ();
		
		int number_of_passes = png_set_interlace_handling (png_ptr);
		
		for (int pass = 0; pass < number_of_passes; pass++) {
			
			for (int i = 0; i < height; i++) {
				
				png_bytep anAddr = (png_bytep)(bytes + i * stride);
				png_read_rows (png_ptr, (png_bytepp) &anAddr, NULL, 1);
				
			}
			
		}
		
		png_read_end (png_ptr, info_ptr);
		png_destroy_read_struct (&png_ptr, &info_ptr, (png_infopp)NULL);
		
		return true;
		
	}
	
	
	static bool Encode (ImageBuffer *imageBuffer, ByteArray *bytes) {
		
		return true;
		
		/*png_structp png_ptr = png_create_write_struct(PNG_LIBPNG_VER_STRING, NULL, user_error_fn, user_warning_fn);
		
		if (!png_ptr)
			return false;
		
		png_infop info_ptr = png_create_info_struct(png_ptr);
		if (!info_ptr)
			return false;
		
		if (setjmp(png_jmpbuf(png_ptr)))
		{
			png_destroy_write_struct(&png_ptr, &info_ptr );
			return false;
		}
		
		QuickVec<uint8> out_buffer;
		
		png_set_write_fn(png_ptr, &out_buffer, user_write_data, user_flush_data);
		
		int w = inSurface->Width();
		int h = inSurface->Height();
		
		int bit_depth = 8;
		int color_type = (inSurface->Format()&pfHasAlpha) ?
		PNG_COLOR_TYPE_RGB_ALPHA :
		PNG_COLOR_TYPE_RGB;
		png_set_IHDR(png_ptr, info_ptr, w, h,
		bit_depth, color_type, PNG_INTERLACE_NONE,
		PNG_COMPRESSION_TYPE_BASE, PNG_FILTER_TYPE_BASE);
		
		png_write_info(png_ptr, info_ptr);
		
		bool do_alpha = color_type==PNG_COLOR_TYPE_RGBA;
		
		{
		QuickVec<uint8> row_data(w*4);
		png_bytep row = &row_data[0];
		for(int y=0;y<h;y++)
		{
		uint8 *buf = &row_data[0];
		const uint8 *src = (const uint8 *)inSurface->Row(y);
		for(int x=0;x<w;x++)
		{
		buf[0] = src[2];
		buf[1] = src[1];
		buf[2] = src[0];
		src+=3;
		buf+=3;
		if (do_alpha)
		*buf++ = *src;
		src++;
		}
		png_write_rows(png_ptr, &row, 1);
		}
		}
		
		png_write_end(png_ptr, NULL);
		
		*outBytes = ByteArray(out_buffer);
		
		return true;*/
		
	}
	
	
}