51 lines
1.7 KiB
C++
51 lines
1.7 KiB
C++
#include <emscripten/bind.h>
|
|
#include <emscripten/val.h>
|
|
#include "avif/avif.h"
|
|
|
|
using namespace emscripten;
|
|
|
|
thread_local const val Uint8ClampedArray = val::global("Uint8ClampedArray");
|
|
thread_local const val ImageData = val::global("ImageData");
|
|
|
|
val decode(std::string avifimage) {
|
|
// point raw.data and raw.size to the contents of an .avif(s)
|
|
avifROData raw = {
|
|
.data = (uint8_t*)avifimage.c_str(),
|
|
.size = avifimage.length()
|
|
};
|
|
|
|
avifImage* image = avifImageCreateEmpty();
|
|
avifDecoder* decoder = avifDecoderCreate();
|
|
avifResult decodeResult = avifDecoderRead(decoder, image, &raw);
|
|
// image is an independent copy of decoded data, decoder may be destroyed here
|
|
avifDecoderDestroy(decoder);
|
|
|
|
val result = val::null();
|
|
|
|
if (decodeResult == AVIF_RESULT_OK) {
|
|
// Convert to interleaved RGB(A)/BGR(A) using a libavif-allocated buffer.
|
|
avifRGBImage rgb;
|
|
avifRGBImageSetDefaults(&rgb, image); // Defaults to AVIF_RGB_FORMAT_RGBA which is what we want.
|
|
rgb.depth = 8; // Does not need to match image->depth. We always want 8-bit pixels.
|
|
|
|
avifRGBImageAllocatePixels(&rgb);
|
|
avifImageYUVToRGB(image, &rgb);
|
|
|
|
// We want to create a *copy* of the decoded data to be owned by the JavaScript side.
|
|
// For that, we perform `new Uint8Array(wasmMemBuffer, wasmPtr, wasmSize).slice()`:
|
|
result = ImageData.new_(Uint8ClampedArray.new_(typed_memory_view(rgb.rowBytes * rgb.height, rgb.pixels)), rgb.width, rgb.height);
|
|
|
|
// Now we can safely free the RGB pixels:
|
|
avifRGBImageFreePixels(&rgb);
|
|
}
|
|
|
|
// Image has been converted to RGB, we don't need the original anymore.
|
|
avifImageDestroy(image);
|
|
|
|
return result;
|
|
}
|
|
|
|
EMSCRIPTEN_BINDINGS(my_module) {
|
|
function("decode", &decode);
|
|
}
|