summaryrefslogtreecommitdiff
path: root/media/libjxl/src/tools/viewer/load_jxl.cc
diff options
context:
space:
mode:
authorJob Bautista <jobbautista9@protonmail.com>2022-06-19 15:34:18 +0800
committerJob Bautista <jobbautista9@protonmail.com>2022-06-29 13:53:09 +0800
commit73e06b6df9e396ef080fefcaa3196265bff723c4 (patch)
tree47caa76143ab4bdf70c2a0eb5f4657d2fe0ee22b /media/libjxl/src/tools/viewer/load_jxl.cc
parent8c1064bc1cb494e445a6447587afb8942a6b783b (diff)
downloaduxp-73e06b6df9e396ef080fefcaa3196265bff723c4.tar.gz
Issue #1769 - Part 1: Add vendored libjxl and highway sources.
Used old-configure to add the build option for enabling JPEG-XL support. Highway version: 0.17.0 libjxl version: tree of commit 318c592d98b97d103941b90d47107f06a10c71da
Diffstat (limited to 'media/libjxl/src/tools/viewer/load_jxl.cc')
-rw-r--r--media/libjxl/src/tools/viewer/load_jxl.cc174
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