Simplify JxlDecoder freeing

Use unique_ptr with custom deleter to enforce consistent deallocation, no matter how we leave the function.
This commit is contained in:
Ingvar Stepanyan
2020-10-30 12:53:26 +00:00
parent 5df81ad062
commit 93474b025e
2 changed files with 18 additions and 23 deletions

View File

@@ -11,47 +11,42 @@ using namespace emscripten;
thread_local const val Uint8ClampedArray = val::global("Uint8ClampedArray"); thread_local const val Uint8ClampedArray = val::global("Uint8ClampedArray");
thread_local const val ImageData = val::global("ImageData"); thread_local const val ImageData = val::global("ImageData");
#define EXPECT_EQ(a, b) \ #define EXPECT_TRUE(a) \
if ((a) != (b)) { \ if (!(a)) \
JxlDecoderDestroy(dec); \ return val::null();
return val::null(); \
} #define EXPECT_EQ(a, b) EXPECT_TRUE((a) == (b));
val decode(std::string data) { val decode(std::string data) {
JxlDecoder* dec = JxlDecoderCreate(nullptr); std::unique_ptr<JxlDecoder,
std::integral_constant<decltype(&JxlDecoderDestroy), JxlDecoderDestroy>>
dec(JxlDecoderCreate(nullptr));
EXPECT_EQ(JXL_DEC_SUCCESS, EXPECT_EQ(JXL_DEC_SUCCESS,
JxlDecoderSubscribeEvents( JxlDecoderSubscribeEvents(
dec, JXL_DEC_BASIC_INFO | JXL_DEC_COLOR_ENCODING | JXL_DEC_FULL_IMAGE)); dec.get(), JXL_DEC_BASIC_INFO | JXL_DEC_COLOR_ENCODING | JXL_DEC_FULL_IMAGE));
auto next_in = (const uint8_t*)data.c_str(); auto next_in = (const uint8_t*)data.c_str();
auto avail_in = data.size(); auto avail_in = data.size();
EXPECT_EQ(JXL_DEC_BASIC_INFO, JxlDecoderProcessInput(dec, &next_in, &avail_in)); EXPECT_EQ(JXL_DEC_BASIC_INFO, JxlDecoderProcessInput(dec.get(), &next_in, &avail_in));
size_t buffer_size; size_t buffer_size;
const JxlPixelFormat format = {4, JXL_LITTLE_ENDIAN, JXL_TYPE_FLOAT}; const JxlPixelFormat format = {4, JXL_LITTLE_ENDIAN, JXL_TYPE_FLOAT};
EXPECT_EQ(JXL_DEC_SUCCESS, JxlDecoderImageOutBufferSize(dec, &format, &buffer_size)); EXPECT_EQ(JXL_DEC_SUCCESS, JxlDecoderImageOutBufferSize(dec.get(), &format, &buffer_size));
JxlBasicInfo info; JxlBasicInfo info;
EXPECT_EQ(JXL_DEC_SUCCESS, JxlDecoderGetBasicInfo(dec, &info)); EXPECT_EQ(JXL_DEC_SUCCESS, JxlDecoderGetBasicInfo(dec.get(), &info));
EXPECT_EQ(JXL_DEC_COLOR_ENCODING, JxlDecoderProcessInput(dec, &next_in, &avail_in)); EXPECT_EQ(JXL_DEC_COLOR_ENCODING, JxlDecoderProcessInput(dec.get(), &next_in, &avail_in));
size_t icc_size; size_t icc_size;
EXPECT_EQ(JXL_DEC_SUCCESS, EXPECT_EQ(JXL_DEC_SUCCESS,
JxlDecoderGetICCProfileSize(dec, JXL_COLOR_PROFILE_TARGET_DATA, &icc_size)); JxlDecoderGetICCProfileSize(dec.get(), JXL_COLOR_PROFILE_TARGET_DATA, &icc_size));
std::vector<uint8_t> icc_profile(icc_size); std::vector<uint8_t> icc_profile(icc_size);
EXPECT_EQ(JXL_DEC_SUCCESS, EXPECT_EQ(JXL_DEC_SUCCESS,
JxlDecoderGetColorAsICCProfile(dec, JXL_COLOR_PROFILE_TARGET_DATA, icc_profile.data(), JxlDecoderGetColorAsICCProfile(dec.get(), JXL_COLOR_PROFILE_TARGET_DATA,
icc_profile.size())); icc_profile.data(), icc_profile.size()));
std::unique_ptr<float[]> float_pixels(new float[(buffer_size + 3) / 4]); std::unique_ptr<float[]> float_pixels(new float[(buffer_size + 3) / 4]);
EXPECT_EQ(JXL_DEC_SUCCESS, EXPECT_EQ(JXL_DEC_SUCCESS,
JxlDecoderSetImageOutBuffer( JxlDecoderSetImageOutBuffer(
dec, &format, reinterpret_cast<uint8_t*>(float_pixels.get()), buffer_size)); dec.get(), &format, reinterpret_cast<uint8_t*>(float_pixels.get()), buffer_size));
EXPECT_EQ(JXL_DEC_FULL_IMAGE, JxlDecoderProcessInput(dec, &next_in, &avail_in)); EXPECT_EQ(JXL_DEC_FULL_IMAGE, JxlDecoderProcessInput(dec.get(), &next_in, &avail_in));
JxlDecoderDestroy(dec);
#undef EXPECT_EQ
#define EXPECT_TRUE(a) \
if (!(a)) { \
return val::null(); \
}
std::unique_ptr<uint8_t[]> pixels(new uint8_t[info.xsize * info.ysize * 4]); std::unique_ptr<uint8_t[]> pixels(new uint8_t[info.xsize * info.ysize * 4]);
// Convert to sRGB. // Convert to sRGB.

Binary file not shown.