/*************************************************************************/
/*  image_etc.cpp                                                        */
/*************************************************************************/
/*                       This file is part of:                           */
/*                           GODOT ENGINE                                */
/*                    http://www.godotengine.org                         */
/*************************************************************************/
/* Copyright (c) 2007-2017 Juan Linietsky, Ariel Manzur.                 */
/*                                                                       */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the       */
/* "Software"), to deal in the Software without restriction, including   */
/* without limitation the rights to use, copy, modify, merge, publish,   */
/* distribute, sublicense, and/or sell copies of the Software, and to    */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions:                                             */
/*                                                                       */
/* The above copyright notice and this permission notice shall be        */
/* included in all copies or substantial portions of the Software.       */
/*                                                                       */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,       */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF    */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY  */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,  */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE     */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.                */
/*************************************************************************/
#include "image_etc.h"
#include "image.h"
#include "rg_etc1.h"
#include "print_string.h"
#include "os/copymem.h"
static void _decompress_etc(Image *p_img) {

	ERR_FAIL_COND(p_img->get_format()!=Image::FORMAT_ETC);

	int imgw = p_img->get_width();
	int imgh = p_img->get_height();
	PoolVector<uint8_t> src=p_img->get_data();
	PoolVector<uint8_t> dst;

	PoolVector<uint8_t>::Read r = src.read();

	int mmc=p_img->get_mipmap_count();


	for(int i=0;i<=mmc;i++) {

		dst.resize(dst.size()+imgw*imgh*3);
		const uint8_t *srcbr=&r[p_img->get_mipmap_offset(i)];
		PoolVector<uint8_t>::Write w = dst.write();

		uint8_t *wptr = &w[dst.size()-imgw*imgh*3];

		int bw=MAX(imgw/4,1);
		int bh=MAX(imgh/4,1);

		for(int y=0;y<bh;y++) {

			for(int x=0;x<bw;x++) {

				uint8_t block[4*4*4];


				rg_etc1::unpack_etc1_block(srcbr,(unsigned int*)block);
				srcbr+=8;

				int maxx=MIN(imgw,4);
				int maxy=MIN(imgh,4);

				for(int yy=0;yy<maxy;yy++) {

					for(int xx=0;xx<maxx;xx++) {

						uint32_t src_ofs = (yy*4+xx)*4;
						uint32_t dst_ofs = ((y*4+yy)*imgw+x*4+xx)*3;
						wptr[dst_ofs+0]=block[src_ofs+0];
						wptr[dst_ofs+1]=block[src_ofs+1];
						wptr[dst_ofs+2]=block[src_ofs+2];

					}
				}

			}

		}

		imgw=MAX(1,imgw/2);
		imgh=MAX(1,imgh/2);
	}


	r=PoolVector<uint8_t>::Read();
	//print_line("Re Creating ETC into regular image: w "+itos(p_img->get_width())+" h "+itos(p_img->get_height())+" mm "+itos(p_img->get_mipmaps()));
	*p_img=Image(p_img->get_width(),p_img->get_height(),p_img->has_mipmaps(),Image::FORMAT_RGB8,dst);
	if (p_img->has_mipmaps())
		p_img->generate_mipmaps(true);


}

static void _compress_etc(Image *p_img) {

	Image img = *p_img;

	int imgw=img.get_width(),imgh=img.get_height();

	ERR_FAIL_COND( nearest_power_of_2(imgw)!=imgw || nearest_power_of_2(imgh)!=imgh );

	if (img.get_format()!=Image::FORMAT_RGB8)
		img.convert(Image::FORMAT_RGB8);


	int mmc=img.get_mipmap_count();
	if (mmc==0)
		img.generate_mipmaps(); // force mipmaps, so it works on most hardware


	PoolVector<uint8_t> res_data;
	PoolVector<uint8_t> dst_data;
	PoolVector<uint8_t>::Read r = img.get_data().read();

	int mc=0;


	rg_etc1::etc1_pack_params pp;
	pp.m_quality=rg_etc1::cLowQuality;
	for(int i=0;i<=mmc;i++) {


		int bw=MAX(imgw/4,1);
		int bh=MAX(imgh/4,1);
		const uint8_t *src = &r[img.get_mipmap_offset(i)];
		int mmsize = MAX(bw,1)*MAX(bh,1)*8;
		dst_data.resize(dst_data.size()+mmsize);
		PoolVector<uint8_t>::Write w=dst_data.write();
		uint8_t *dst = &w[dst_data.size()-mmsize];


//		print_line("bh: "+itos(bh)+" bw: "+itos(bw));

		for(int y=0;y<bh;y++) {

			for(int x=0;x<bw;x++) {

//				print_line("x: "+itos(x)+" y: "+itos(y));

				uint8_t block[4*4*4];
				zeromem(block,4*4*4);
				uint8_t cblock[8];

				int maxy = MIN(imgh,4);
				int maxx = MIN(imgw,4);


				for(int yy=0;yy<maxy;yy++) {

					for(int xx=0;xx<maxx;xx++) {


						uint32_t dst_ofs = (yy*4+xx)*4;
						uint32_t src_ofs = ((y*4+yy)*imgw+x*4+xx)*3;
						block[dst_ofs+0]=src[src_ofs+0];
						block[dst_ofs+1]=src[src_ofs+1];
						block[dst_ofs+2]=src[src_ofs+2];
						block[dst_ofs+3]=255;

					}
				}

				rg_etc1::pack_etc1_block(cblock, (const unsigned int*)block, pp);
				for(int j=0;j<8;j++) {

					dst[j]=cblock[j];
				}

				dst+=8;
			}

		}

		imgw=MAX(1,imgw/2);
		imgh=MAX(1,imgh/2);
		mc++;

	}

	*p_img=Image(p_img->get_width(),p_img->get_height(),(mc-1)?true:false,Image::FORMAT_ETC,dst_data);


}

void _register_etc1_compress_func() {

	rg_etc1::pack_etc1_block_init();
	Image::_image_compress_etc_func=_compress_etc;
	Image::_image_decompress_etc=_decompress_etc;


}