Martin Michelsen
103 lines
2.9 KiB
C++
103 lines
2.9 KiB
C++
#include "GVMEncoder.hh"
|
|
|
|
#include <phosg/Encoding.hh>
|
|
#include <phosg/Image.hh>
|
|
#include <phosg/Strings.hh>
|
|
|
|
#include "Text.hh"
|
|
|
|
using namespace std;
|
|
|
|
struct GVMFileEntry {
|
|
be_uint16_t file_num;
|
|
pstring<TextEncoding::ASCII, 0x1C> name;
|
|
parray<be_uint32_t, 2> unknown_a1;
|
|
} __packed_ws__(GVMFileEntry, 0x26);
|
|
|
|
struct GVMFileHeader {
|
|
be_uint32_t magic; // 'GVMH'
|
|
le_uint32_t header_size;
|
|
be_uint16_t flags;
|
|
be_uint16_t num_files;
|
|
} __packed_ws__(GVMFileHeader, 0x0C);
|
|
|
|
struct GVRHeader {
|
|
be_uint32_t magic; // 'GVRT'
|
|
le_uint32_t data_size;
|
|
be_uint16_t unknown;
|
|
uint8_t format_flags; // High 4 bits are pixel format, low 4 are data flags
|
|
GVRDataFormat data_format;
|
|
be_uint16_t width;
|
|
be_uint16_t height;
|
|
} __packed_ws__(GVRHeader, 0x10);
|
|
|
|
string encode_gvm(const Image& img, GVRDataFormat data_format) {
|
|
if (img.get_width() > 0xFFFF) {
|
|
throw runtime_error("image is too wide to be encoded as a GVR texture");
|
|
}
|
|
if (img.get_height() > 0xFFFF) {
|
|
throw runtime_error("image is too tall to be encoded as a GVR texture");
|
|
}
|
|
if (img.get_width() & 3) {
|
|
throw runtime_error("image width is not a multiple of 4");
|
|
}
|
|
if (img.get_height() & 3) {
|
|
throw runtime_error("image height is not a multiple of 4");
|
|
}
|
|
size_t pixel_count = img.get_width() * img.get_height();
|
|
size_t pixel_bytes = 0;
|
|
switch (data_format) {
|
|
case GVRDataFormat::RGB565:
|
|
case GVRDataFormat::RGB5A3:
|
|
pixel_bytes = pixel_count * 2;
|
|
break;
|
|
case GVRDataFormat::ARGB8888:
|
|
pixel_bytes = pixel_count * 2;
|
|
break;
|
|
default:
|
|
throw invalid_argument("cannot encode pixel format");
|
|
}
|
|
|
|
StringWriter w;
|
|
w.put<GVMFileHeader>({.magic = 0x47564D48, .header_size = 0x48, .flags = 0x010F, .num_files = 1});
|
|
GVMFileEntry file_entry;
|
|
file_entry.file_num = 0;
|
|
file_entry.name.encode("img", 1);
|
|
file_entry.unknown_a1.clear(0);
|
|
w.put(file_entry);
|
|
w.extend_to(0x50, 0x00);
|
|
w.put<GVRHeader>({.magic = 0x47565254,
|
|
.data_size = pixel_bytes + 8,
|
|
.unknown = 0,
|
|
.format_flags = 0,
|
|
.data_format = data_format,
|
|
.width = img.get_width(),
|
|
.height = img.get_height()});
|
|
|
|
for (size_t y = 0; y < img.get_height(); y += 4) {
|
|
for (size_t x = 0; x < img.get_width(); x += 4) {
|
|
for (size_t yy = 0; yy < 4; yy++) {
|
|
for (size_t xx = 0; xx < 4; xx++) {
|
|
uint64_t a, r, g, b;
|
|
img.read_pixel(x + xx, y + yy, &r, &g, &b, &a);
|
|
switch (data_format) {
|
|
case GVRDataFormat::RGB565:
|
|
w.put_u16b(encode_rgb565(r, g, b));
|
|
break;
|
|
case GVRDataFormat::RGB5A3:
|
|
w.put_u16b(encode_rgb5a3(r, g, b, a));
|
|
break;
|
|
case GVRDataFormat::ARGB8888:
|
|
w.put_u32b(encode_argb8888(r, g, b, a));
|
|
break;
|
|
default:
|
|
throw logic_error("cannot encode pixel format");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return std::move(w.str());
|
|
}
|