summaryrefslogtreecommitdiff
path: root/tools/hdr/tone_map.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tools/hdr/tone_map.cc')
-rw-r--r--tools/hdr/tone_map.cc171
1 files changed, 89 insertions, 82 deletions
diff --git a/tools/hdr/tone_map.cc b/tools/hdr/tone_map.cc
index 951b119..1ef3823 100644
--- a/tools/hdr/tone_map.cc
+++ b/tools/hdr/tone_map.cc
@@ -1,82 +1,89 @@
-// 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 <stdio.h>
-#include <stdlib.h>
-
-#include "lib/extras/codec.h"
-#include "lib/extras/tone_mapping.h"
-#include "lib/jxl/base/thread_pool_internal.h"
-#include "tools/args.h"
-#include "tools/cmdline.h"
-
-int main(int argc, const char** argv) {
- jxl::ThreadPoolInternal pool;
-
- jpegxl::tools::CommandLineParser parser;
- float max_nits = 0;
- parser.AddOptionValue('m', "max_nits", "nits",
- "maximum luminance in the image", &max_nits,
- &jpegxl::tools::ParseFloat, 0);
- float target_nits = 0;
- auto target_nits_option = parser.AddOptionValue(
- 't', "target_nits", "nits",
- "peak luminance of the display for which to tone map", &target_nits,
- &jpegxl::tools::ParseFloat, 0);
- bool pq = false;
- parser.AddOptionFlag('p', "pq",
- "write the output with absolute luminance using PQ", &pq,
- &jpegxl::tools::SetBooleanTrue, 0);
- const char* input_filename = nullptr;
- auto input_filename_option = parser.AddPositionalOption(
- "input", true, "input image", &input_filename, 0);
- const char* output_filename = nullptr;
- auto output_filename_option = parser.AddPositionalOption(
- "output", true, "output image", &output_filename, 0);
-
- if (!parser.Parse(argc, argv)) {
- fprintf(stderr, "See -h for help.\n");
- return EXIT_FAILURE;
- }
-
- if (parser.HelpFlagPassed()) {
- parser.PrintHelp();
- return EXIT_SUCCESS;
- }
-
- if (!parser.GetOption(target_nits_option)->matched()) {
- fprintf(stderr,
- "Missing required argument --target_nits.\nSee -h for help.\n");
- return EXIT_FAILURE;
- }
- if (!parser.GetOption(input_filename_option)->matched()) {
- fprintf(stderr, "Missing input filename.\nSee -h for help.\n");
- return EXIT_FAILURE;
- }
- if (!parser.GetOption(output_filename_option)->matched()) {
- fprintf(stderr, "Missing output filename.\nSee -h for help.\n");
- return EXIT_FAILURE;
- }
-
- jxl::CodecInOut image;
- jxl::ColorHints color_hints;
- color_hints.Add("color_space", "RGB_D65_202_Rel_PeQ");
- JXL_CHECK(jxl::SetFromFile(input_filename, color_hints, &image, &pool));
- if (max_nits > 0) {
- image.metadata.m.SetIntensityTarget(max_nits);
- }
- JXL_CHECK(jxl::ToneMapTo({0, target_nits}, &image, &pool));
-
- jxl::ColorEncoding c_out = image.metadata.m.color_encoding;
- if (pq) {
- c_out.tf.SetTransferFunction(jxl::TransferFunction::kPQ);
- } else {
- c_out.tf.SetTransferFunction(jxl::TransferFunction::kSRGB);
- }
- JXL_CHECK(c_out.CreateICC());
- JXL_CHECK(image.TransformTo(c_out, &pool));
- image.metadata.m.color_encoding = c_out;
- JXL_CHECK(jxl::EncodeToFile(image, output_filename, &pool));
-}
+// 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 <stdio.h>
+#include <stdlib.h>
+
+#include "lib/extras/codec.h"
+#include "lib/extras/tone_mapping.h"
+#include "lib/jxl/base/thread_pool_internal.h"
+#include "lib/jxl/enc_color_management.h"
+#include "tools/args.h"
+#include "tools/cmdline.h"
+
+int main(int argc, const char** argv) {
+ jxl::ThreadPoolInternal pool;
+
+ jpegxl::tools::CommandLineParser parser;
+ float max_nits = 0;
+ parser.AddOptionValue('m', "max_nits", "nits",
+ "maximum luminance in the image", &max_nits,
+ &jpegxl::tools::ParseFloat, 0);
+ float target_nits = 0;
+ auto target_nits_option = parser.AddOptionValue(
+ 't', "target_nits", "nits",
+ "peak luminance of the display for which to tone map", &target_nits,
+ &jpegxl::tools::ParseFloat, 0);
+ float preserve_saturation = .1f;
+ parser.AddOptionValue(
+ 's', "preserve_saturation", "0..1",
+ "to what extent to try and preserve saturation over luminance",
+ &preserve_saturation, &jpegxl::tools::ParseFloat, 0);
+ bool pq = false;
+ parser.AddOptionFlag('p', "pq",
+ "write the output with absolute luminance using PQ", &pq,
+ &jpegxl::tools::SetBooleanTrue, 0);
+ const char* input_filename = nullptr;
+ auto input_filename_option = parser.AddPositionalOption(
+ "input", true, "input image", &input_filename, 0);
+ const char* output_filename = nullptr;
+ auto output_filename_option = parser.AddPositionalOption(
+ "output", true, "output image", &output_filename, 0);
+
+ if (!parser.Parse(argc, argv)) {
+ fprintf(stderr, "See -h for help.\n");
+ return EXIT_FAILURE;
+ }
+
+ if (parser.HelpFlagPassed()) {
+ parser.PrintHelp();
+ return EXIT_SUCCESS;
+ }
+
+ if (!parser.GetOption(target_nits_option)->matched()) {
+ fprintf(stderr,
+ "Missing required argument --target_nits.\nSee -h for help.\n");
+ return EXIT_FAILURE;
+ }
+ if (!parser.GetOption(input_filename_option)->matched()) {
+ fprintf(stderr, "Missing input filename.\nSee -h for help.\n");
+ return EXIT_FAILURE;
+ }
+ if (!parser.GetOption(output_filename_option)->matched()) {
+ fprintf(stderr, "Missing output filename.\nSee -h for help.\n");
+ return EXIT_FAILURE;
+ }
+
+ jxl::CodecInOut image;
+ jxl::extras::ColorHints color_hints;
+ color_hints.Add("color_space", "RGB_D65_202_Rel_PeQ");
+ JXL_CHECK(jxl::SetFromFile(input_filename, color_hints, &image, &pool));
+ if (max_nits > 0) {
+ image.metadata.m.SetIntensityTarget(max_nits);
+ }
+ JXL_CHECK(jxl::ToneMapTo({0, target_nits}, &image, &pool));
+ JXL_CHECK(jxl::GamutMap(&image, preserve_saturation, &pool));
+
+ jxl::ColorEncoding c_out = image.metadata.m.color_encoding;
+ if (pq) {
+ c_out.tf.SetTransferFunction(jxl::TransferFunction::kPQ);
+ } else {
+ c_out.tf.SetTransferFunction(jxl::TransferFunction::k709);
+ }
+ JXL_CHECK(c_out.CreateICC());
+ JXL_CHECK(image.TransformTo(c_out, jxl::GetJxlCms(), &pool));
+ image.metadata.m.color_encoding = c_out;
+ JXL_CHECK(jxl::EncodeToFile(image, output_filename, &pool));
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-26 03:04:17 +0000