mirror of
https://github.com/godotengine/godot.git
synced 2024-12-09 10:09:20 +08:00
174 lines
3.5 KiB
C++
174 lines
3.5 KiB
C++
#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();
|
|
DVector<uint8_t> src=p_img->get_data();
|
|
DVector<uint8_t> dst;
|
|
|
|
DVector<uint8_t>::Read r = src.read();
|
|
|
|
int mmc=p_img->get_mipmaps();
|
|
|
|
|
|
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)];
|
|
DVector<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=DVector<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->get_mipmaps(),Image::FORMAT_RGB,dst);
|
|
if (p_img->get_mipmaps())
|
|
p_img->generate_mipmaps(-1,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_RGB)
|
|
img.convert(Image::FORMAT_RGB);
|
|
|
|
|
|
int mmc=img.get_mipmaps();
|
|
if (mmc==0)
|
|
img.generate_mipmaps(); // force mipmaps, so it works on most hardware
|
|
|
|
|
|
DVector<uint8_t> res_data;
|
|
DVector<uint8_t> dst_data;
|
|
DVector<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);
|
|
DVector<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,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;
|
|
|
|
|
|
}
|