431 lines
12 KiB
C++
431 lines
12 KiB
C++
/*
|
|
Copyright (C) 2015, SiPlus, Chasseur de bots.
|
|
All Rights Reserved.
|
|
|
|
This file is part of GtkRadiant.
|
|
|
|
GtkRadiant is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 2 of the License, or
|
|
(at your option) any later version.
|
|
|
|
GtkRadiant is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with GtkRadiant; if not, write to the Free Software
|
|
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*/
|
|
|
|
#include "ktx.h"
|
|
|
|
#include <string.h>
|
|
|
|
#include "bytestreamutils.h"
|
|
#include "etclib.h"
|
|
#include "ifilesystem.h"
|
|
#include "imagelib.h"
|
|
|
|
|
|
const int KTX_TYPE_UNSIGNED_BYTE = 0x1401;
|
|
const int KTX_TYPE_UNSIGNED_SHORT_4_4_4_4 = 0x8033;
|
|
const int KTX_TYPE_UNSIGNED_SHORT_5_5_5_1 = 0x8034;
|
|
const int KTX_TYPE_UNSIGNED_SHORT_5_6_5 = 0x8363;
|
|
|
|
const int KTX_FORMAT_ALPHA = 0x1906;
|
|
const int KTX_FORMAT_RGB = 0x1907;
|
|
const int KTX_FORMAT_RGBA = 0x1908;
|
|
const int KTX_FORMAT_LUMINANCE = 0x1909;
|
|
const int KTX_FORMAT_LUMINANCE_ALPHA = 0x190A;
|
|
const int KTX_FORMAT_BGR = 0x80E0;
|
|
const int KTX_FORMAT_BGRA = 0x80E1;
|
|
|
|
const int KTX_FORMAT_ETC1_RGB8 = 0x8D64;
|
|
|
|
class KTX_Decoder {
|
|
public:
|
|
virtual ~KTX_Decoder() = default;
|
|
|
|
virtual void Decode(PointerInputStream &istream, byte *out) = 0;
|
|
|
|
virtual unsigned int GetPixelSize() = 0;
|
|
};
|
|
|
|
class KTX_Decoder_A8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
out[0] = out[1] = out[2] = 0;
|
|
out[3] = istream_read_byte(istream);
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 1;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_RGB8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
istream.read(out, 3);
|
|
out[3] = 255;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_RGBA8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
istream.read(out, 4);
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 4;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_L8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
byte l = istream_read_byte(istream);
|
|
out[0] = out[1] = out[2] = l;
|
|
out[3] = 255;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 1;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_LA8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
byte la[2];
|
|
istream.read(la, 2);
|
|
out[0] = out[1] = out[2] = la[0];
|
|
out[3] = la[1];
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_BGR8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
byte bgr[3];
|
|
istream.read(bgr, 3);
|
|
out[0] = bgr[2];
|
|
out[1] = bgr[1];
|
|
out[2] = bgr[0];
|
|
out[3] = 255;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 3;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_BGRA8 : public KTX_Decoder {
|
|
public:
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
byte bgra[4];
|
|
istream.read(bgra, 4);
|
|
out[0] = bgra[2];
|
|
out[1] = bgra[1];
|
|
out[2] = bgra[0];
|
|
out[3] = bgra[3];
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 4;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_RGBA4 : public KTX_Decoder {
|
|
protected:
|
|
bool m_bigEndian;
|
|
public:
|
|
KTX_Decoder_RGBA4(bool bigEndian) : m_bigEndian(bigEndian)
|
|
{}
|
|
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
uint16_t rgba;
|
|
if (m_bigEndian) {
|
|
rgba = istream_read_uint16_be(istream);
|
|
} else {
|
|
rgba = istream_read_uint16_le(istream);
|
|
}
|
|
int r = (rgba >> 12) & 0xf;
|
|
int g = (rgba >> 8) & 0xf;
|
|
int b = (rgba >> 4) & 0xf;
|
|
int a = rgba & 0xf;
|
|
out[0] = (r << 4) | r;
|
|
out[1] = (g << 4) | g;
|
|
out[2] = (b << 4) | b;
|
|
out[3] = (a << 4) | a;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_RGBA5 : public KTX_Decoder {
|
|
protected:
|
|
bool m_bigEndian;
|
|
public:
|
|
KTX_Decoder_RGBA5(bool bigEndian) : m_bigEndian(bigEndian)
|
|
{}
|
|
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
uint16_t rgba;
|
|
if (m_bigEndian) {
|
|
rgba = istream_read_uint16_be(istream);
|
|
} else {
|
|
rgba = istream_read_uint16_le(istream);
|
|
}
|
|
int r = (rgba >> 11) & 0x1f;
|
|
int g = (rgba >> 6) & 0x1f;
|
|
int b = (rgba >> 1) & 0x1f;
|
|
out[0] = (r << 3) | (r >> 2);
|
|
out[1] = (g << 3) | (g >> 2);
|
|
out[2] = (b << 3) | (b >> 2);
|
|
out[3] = (rgba & 1) * 255;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
class KTX_Decoder_RGB5 : public KTX_Decoder {
|
|
protected:
|
|
bool m_bigEndian;
|
|
public:
|
|
KTX_Decoder_RGB5(bool bigEndian) : m_bigEndian(bigEndian)
|
|
{}
|
|
|
|
virtual void Decode(PointerInputStream &istream, byte *out)
|
|
{
|
|
uint16_t rgb;
|
|
if (m_bigEndian) {
|
|
rgb = istream_read_uint16_be(istream);
|
|
} else {
|
|
rgb = istream_read_uint16_le(istream);
|
|
}
|
|
int r = (rgb >> 11) & 0x1f;
|
|
int g = (rgb >> 5) & 0x3f;
|
|
int b = rgb & 0x1f;
|
|
out[0] = (r << 3) | (r >> 2);
|
|
out[1] = (g << 2) | (g >> 4);
|
|
out[2] = (b << 3) | (b >> 2);
|
|
out[3] = 255;
|
|
}
|
|
|
|
virtual unsigned int GetPixelSize()
|
|
{
|
|
return 2;
|
|
}
|
|
};
|
|
|
|
static void KTX_DecodeETC1(PointerInputStream &istream, Image &image)
|
|
{
|
|
unsigned int width = image.getWidth(), height = image.getHeight();
|
|
unsigned int stride = width * 4;
|
|
byte *pixbuf = image.getRGBAPixels();
|
|
byte etc[8], rgba[64];
|
|
|
|
for (unsigned int y = 0; y < height; y += 4, pixbuf += stride * 4) {
|
|
unsigned int blockrows = height - y;
|
|
if (blockrows > 4) {
|
|
blockrows = 4;
|
|
}
|
|
|
|
byte *p = pixbuf;
|
|
for (unsigned int x = 0; x < width; x += 4, p += 16) {
|
|
istream.read(etc, 8);
|
|
ETC_DecodeETC1Block(etc, rgba, qtrue);
|
|
|
|
unsigned int blockrowsize = width - x;
|
|
if (blockrowsize > 4) {
|
|
blockrowsize = 4;
|
|
}
|
|
blockrowsize *= 4;
|
|
for (unsigned int blockrow = 0; blockrow < blockrows; blockrow++) {
|
|
memcpy(p + blockrow * stride, rgba + blockrow * 16, blockrowsize);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
Image *LoadKTXBuff(PointerInputStream &istream)
|
|
{
|
|
byte identifier[12];
|
|
istream.read(identifier, 12);
|
|
if (memcmp(identifier, "\xABKTX 11\xBB\r\n\x1A\n", 12)) {
|
|
globalErrorStream() << "LoadKTX: Image has the wrong identifier\n";
|
|
return 0;
|
|
}
|
|
|
|
bool bigEndian = (istream_read_uint32_le(istream) == 0x01020304);
|
|
|
|
unsigned int type;
|
|
if (bigEndian) {
|
|
type = istream_read_uint32_be(istream);
|
|
} else {
|
|
type = istream_read_uint32_le(istream);
|
|
}
|
|
|
|
// For compressed textures, the format is in glInternalFormat.
|
|
// For uncompressed textures, it's in glBaseInternalFormat.
|
|
istream.seek((type ? 3 : 2) * sizeof(uint32_t));
|
|
unsigned int format;
|
|
if (bigEndian) {
|
|
format = istream_read_uint32_be(istream);
|
|
} else {
|
|
format = istream_read_uint32_le(istream);
|
|
}
|
|
if (!type) {
|
|
istream.seek(sizeof(uint32_t));
|
|
}
|
|
|
|
unsigned int width, height;
|
|
if (bigEndian) {
|
|
width = istream_read_uint32_be(istream);
|
|
height = istream_read_uint32_be(istream);
|
|
} else {
|
|
width = istream_read_uint32_le(istream);
|
|
height = istream_read_uint32_le(istream);
|
|
}
|
|
if (!width) {
|
|
globalErrorStream() << "LoadKTX: Image has zero width\n";
|
|
return 0;
|
|
}
|
|
if (!height) {
|
|
height = 1;
|
|
}
|
|
|
|
// Skip the key/values and load the first 2D image in the texture.
|
|
// Since KTXorientation is only a hint and has no effect on the texture data and coordinates, it must be ignored.
|
|
istream.seek(4 * sizeof(uint32_t));
|
|
unsigned int bytesOfKeyValueData;
|
|
if (bigEndian) {
|
|
bytesOfKeyValueData = istream_read_uint32_be(istream);
|
|
} else {
|
|
bytesOfKeyValueData = istream_read_uint32_le(istream);
|
|
}
|
|
istream.seek(bytesOfKeyValueData + sizeof(uint32_t));
|
|
|
|
RGBAImage *image = new RGBAImage(width, height);
|
|
|
|
if (type) {
|
|
KTX_Decoder *decoder = NULL;
|
|
switch (type) {
|
|
case KTX_TYPE_UNSIGNED_BYTE:
|
|
switch (format) {
|
|
case KTX_FORMAT_ALPHA:
|
|
decoder = new KTX_Decoder_A8();
|
|
break;
|
|
case KTX_FORMAT_RGB:
|
|
decoder = new KTX_Decoder_RGB8();
|
|
break;
|
|
case KTX_FORMAT_RGBA:
|
|
decoder = new KTX_Decoder_RGBA8();
|
|
break;
|
|
case KTX_FORMAT_LUMINANCE:
|
|
decoder = new KTX_Decoder_L8();
|
|
break;
|
|
case KTX_FORMAT_LUMINANCE_ALPHA:
|
|
decoder = new KTX_Decoder_LA8();
|
|
break;
|
|
case KTX_FORMAT_BGR:
|
|
decoder = new KTX_Decoder_BGR8();
|
|
break;
|
|
case KTX_FORMAT_BGRA:
|
|
decoder = new KTX_Decoder_BGRA8();
|
|
break;
|
|
}
|
|
break;
|
|
case KTX_TYPE_UNSIGNED_SHORT_4_4_4_4:
|
|
if (format == KTX_FORMAT_RGBA) {
|
|
decoder = new KTX_Decoder_RGBA4(bigEndian);
|
|
}
|
|
break;
|
|
case KTX_TYPE_UNSIGNED_SHORT_5_5_5_1:
|
|
if (format == KTX_FORMAT_RGBA) {
|
|
decoder = new KTX_Decoder_RGBA5(bigEndian);
|
|
}
|
|
break;
|
|
case KTX_TYPE_UNSIGNED_SHORT_5_6_5:
|
|
if (format == KTX_FORMAT_RGB) {
|
|
decoder = new KTX_Decoder_RGB5(bigEndian);
|
|
}
|
|
break;
|
|
}
|
|
|
|
if (!decoder) {
|
|
globalErrorStream() << "LoadKTX: Image has an unsupported pixel type " << type << " or format " << format
|
|
<< "\n";
|
|
image->release();
|
|
return 0;
|
|
}
|
|
|
|
unsigned int inRowLength = width * decoder->GetPixelSize();
|
|
unsigned int inPadding = ((inRowLength + 3) & ~3) - inRowLength;
|
|
byte *out = image->getRGBAPixels();
|
|
for (unsigned int y = 0; y < height; y++) {
|
|
for (unsigned int x = 0; x < width; x++, out += 4) {
|
|
decoder->Decode(istream, out);
|
|
}
|
|
|
|
if (inPadding) {
|
|
istream.seek(inPadding);
|
|
}
|
|
}
|
|
|
|
delete decoder;
|
|
} else {
|
|
switch (format) {
|
|
case KTX_FORMAT_ETC1_RGB8:
|
|
KTX_DecodeETC1(istream, *image);
|
|
break;
|
|
default:
|
|
globalErrorStream() << "LoadKTX: Image has an unsupported compressed format " << format << "\n";
|
|
image->release();
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
return image;
|
|
}
|
|
|
|
Image *LoadKTX(ArchiveFile &file)
|
|
{
|
|
ScopedArchiveBuffer buffer(file);
|
|
PointerInputStream istream(buffer.buffer);
|
|
return LoadKTXBuff(istream);
|
|
}
|