diff options
Diffstat (limited to 'media/libjxl/src/tools/viewer/load_jxl.cc')
| -rw-r--r-- | media/libjxl/src/tools/viewer/load_jxl.cc | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/media/libjxl/src/tools/viewer/load_jxl.cc b/media/libjxl/src/tools/viewer/load_jxl.cc new file mode 100644 index 0000000000..7fd35d8224 --- /dev/null +++ b/media/libjxl/src/tools/viewer/load_jxl.cc @@ -0,0 +1,174 @@ +// Copyright (c) the JPEG XL Project Authors. All rights reserved. +// +// Use of this source code is governed by a BSD-style +// license that can be found in the LICENSE file. + +#include "tools/viewer/load_jxl.h" + +#include <stdint.h> + +#include <QElapsedTimer> +#include <QFile> + +#include "jxl/decode.h" +#include "jxl/decode_cxx.h" +#include "jxl/thread_parallel_runner_cxx.h" +#include "jxl/types.h" +#include "lcms2.h" + +namespace jxl { + +namespace { + +struct CmsProfileCloser { + void operator()(const cmsHPROFILE profile) const { + if (profile != nullptr) { + cmsCloseProfile(profile); + } + } +}; +using CmsProfileUniquePtr = + std::unique_ptr<std::remove_pointer<cmsHPROFILE>::type, CmsProfileCloser>; + +struct CmsTransformDeleter { + void operator()(const cmsHTRANSFORM transform) const { + if (transform != nullptr) { + cmsDeleteTransform(transform); + } + } +}; +using CmsTransformUniquePtr = + std::unique_ptr<std::remove_pointer<cmsHTRANSFORM>::type, + CmsTransformDeleter>; + +} // namespace + +QImage loadJxlImage(const QString& filename, const QByteArray& targetIccProfile, + qint64* elapsed_ns, bool* usedRequestedProfile) { + auto runner = JxlThreadParallelRunnerMake( + nullptr, JxlThreadParallelRunnerDefaultNumWorkerThreads()); + + auto dec = JxlDecoderMake(nullptr); + +#define EXPECT_TRUE(a) \ + do { \ + if (!(a)) { \ + fprintf(stderr, "Assertion failure (%d): %s\n", __LINE__, #a); \ + return QImage(); \ + } \ + } while (false) +#define EXPECT_EQ(a, b) \ + do { \ + int a_ = a; \ + int b_ = b; \ + if (a_ != b_) { \ + fprintf(stderr, "Assertion failure (%d): %s (%d) != %s (%d)\n", \ + __LINE__, #a, a_, #b, b_); \ + return QImage(); \ + } \ + } while (false) + + EXPECT_EQ(JXL_DEC_SUCCESS, + JxlDecoderSubscribeEvents(dec.get(), JXL_DEC_BASIC_INFO | + JXL_DEC_COLOR_ENCODING | + JXL_DEC_FULL_IMAGE)); + QFile jpegXlFile(filename); + if (!jpegXlFile.open(QIODevice::ReadOnly)) { + return QImage(); + } + const QByteArray jpegXlData = jpegXlFile.readAll(); + if (jpegXlData.size() < 4) { + return QImage(); + } + + QElapsedTimer timer; + timer.start(); + const uint8_t* jxl_data = reinterpret_cast<const uint8_t*>(jpegXlData.data()); + size_t jxl_size = jpegXlData.size(); + JxlDecoderSetInput(dec.get(), jxl_data, jxl_size); + EXPECT_EQ(JXL_DEC_BASIC_INFO, JxlDecoderProcessInput(dec.get())); + JxlBasicInfo info; + EXPECT_EQ(JXL_DEC_SUCCESS, JxlDecoderGetBasicInfo(dec.get(), &info)); + size_t pixel_count = info.xsize * info.ysize; + + EXPECT_EQ(JXL_DEC_COLOR_ENCODING, JxlDecoderProcessInput(dec.get())); + static const JxlPixelFormat format = {4, JXL_TYPE_FLOAT, JXL_NATIVE_ENDIAN, + 0}; + size_t icc_size; + EXPECT_EQ(JXL_DEC_SUCCESS, + JxlDecoderGetICCProfileSize( + dec.get(), &format, JXL_COLOR_PROFILE_TARGET_DATA, &icc_size)); + std::vector<uint8_t> icc_profile(icc_size); + EXPECT_EQ(JXL_DEC_SUCCESS, + JxlDecoderGetColorAsICCProfile( + dec.get(), &format, JXL_COLOR_PROFILE_TARGET_DATA, + icc_profile.data(), icc_profile.size())); + + std::vector<float> float_pixels(pixel_count * 4); + EXPECT_EQ(JXL_DEC_NEED_IMAGE_OUT_BUFFER, JxlDecoderProcessInput(dec.get())); + EXPECT_EQ(JXL_DEC_SUCCESS, + JxlDecoderSetImageOutBuffer(dec.get(), &format, float_pixels.data(), + pixel_count * 4 * sizeof(float))); + EXPECT_EQ(JXL_DEC_FULL_IMAGE, JxlDecoderProcessInput(dec.get())); + + std::vector<uint16_t> uint16_pixels(pixel_count * 4); + const thread_local cmsContext context = cmsCreateContext(nullptr, nullptr); + EXPECT_TRUE(context != nullptr); + const CmsProfileUniquePtr jxl_profile(cmsOpenProfileFromMemTHR( + context, icc_profile.data(), icc_profile.size())); + EXPECT_TRUE(jxl_profile != nullptr); + CmsProfileUniquePtr target_profile(cmsOpenProfileFromMemTHR( + context, targetIccProfile.data(), targetIccProfile.size())); + if (usedRequestedProfile != nullptr) { + *usedRequestedProfile = (target_profile != nullptr); + } + if (target_profile == nullptr) { + target_profile.reset(cmsCreate_sRGBProfileTHR(context)); + } + EXPECT_TRUE(target_profile != nullptr); + CmsTransformUniquePtr transform(cmsCreateTransformTHR( + context, jxl_profile.get(), TYPE_RGBA_FLT, target_profile.get(), + TYPE_RGBA_16, INTENT_RELATIVE_COLORIMETRIC, cmsFLAGS_COPY_ALPHA)); + EXPECT_TRUE(transform != nullptr); + cmsDoTransform(transform.get(), float_pixels.data(), uint16_pixels.data(), + pixel_count); + if (elapsed_ns != nullptr) *elapsed_ns = timer.nsecsElapsed(); + + QImage result(info.xsize, info.ysize, +#if QT_VERSION >= QT_VERSION_CHECK(5, 12, 0) + info.alpha_premultiplied ? QImage::Format_RGBA64_Premultiplied + : QImage::Format_RGBA64 +#else + info.alpha_premultiplied ? QImage::Format_ARGB32_Premultiplied + : QImage::Format_ARGB32 +#endif + ); + + for (int y = 0; y < result.height(); ++y) { +#if QT_VERSION >= QT_VERSION_CHECK(5, 12, 0) + QRgba64* const row = reinterpret_cast<QRgba64*>(result.scanLine(y)); +#else + QRgb* const row = reinterpret_cast<QRgb*>(result.scanLine(y)); +#endif + const uint16_t* const data = uint16_pixels.data() + result.width() * y * 4; + for (int x = 0; x < result.width(); ++x) { +#if QT_VERSION >= QT_VERSION_CHECK(5, 6, 0) + row[x] = qRgba64(data[4 * x + 0], data[4 * x + 1], data[4 * x + 2], + data[4 * x + 3]) +#if QT_VERSION < QT_VERSION_CHECK(5, 12, 0) + .toArgb32() +#endif + ; +#else + // Qt version older than 5.6 doesn't have a qRgba64. + row[x] = qRgba(data[4 * x + 0] * (255.f / 65535) + .5f, + data[4 * x + 1] * (255.f / 65535) + .5f, + data[4 * x + 2] * (255.f / 65535) + .5f, + data[4 * x + 3] * (255.f / 65535) + .5f); +#endif + } + } + return result; +} + +} // namespace jxl |