gimp/plug-ins/common/file-jpegxl.c

2168 lines
76 KiB
C
Raw Normal View History

2021-08-26 20:24:21 +08:00
/* GIMP - The GNU Image Manipulation Program
* Copyright (C) 1995 Spencer Kimball and Peter Mattis
*
* file-jpegxl - JPEG XL file format plug-in for the GIMP
* Copyright (C) 2021 Daniel Novomesky
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include "config.h"
#include <gexiv2/gexiv2.h>
2021-08-26 20:24:21 +08:00
#include <glib/gstdio.h>
#include <jxl/decode.h>
#include <jxl/encode.h>
#include <jxl/thread_parallel_runner.h>
#include <libgimp/gimp.h>
#include <libgimp/gimpui.h>
#include "libgimp/stdplugins-intl.h"
#define LOAD_PROC "file-jpegxl-load"
#define EXPORT_PROC "file-jpegxl-export"
2021-08-26 20:24:21 +08:00
#define PLUG_IN_BINARY "file-jpegxl"
typedef struct _JpegXL JpegXL;
typedef struct _JpegXLClass JpegXLClass;
struct _JpegXL
{
GimpPlugIn parent_instance;
};
struct _JpegXLClass
{
GimpPlugInClass parent_class;
};
#define JPEGXL_TYPE (jpegxl_get_type ())
#define JPEGXL(obj) (G_TYPE_CHECK_INSTANCE_CAST ((obj), JPEGXL_TYPE, JpegXL))
2021-08-26 20:24:21 +08:00
GType jpegxl_get_type (void) G_GNUC_CONST;
static GList *jpegxl_query_procedures (GimpPlugIn *plug_in);
static GimpProcedure *jpegxl_create_procedure (GimpPlugIn *plug_in,
const gchar *name);
static GimpValueArray *jpegxl_load (GimpProcedure *procedure,
GimpRunMode run_mode,
GFile *file,
GimpMetadata *metadata,
GimpMetadataLoadFlags *flags,
GimpProcedureConfig *config,
gpointer run_data);
static GimpValueArray *jpegxl_export (GimpProcedure *procedure,
GimpRunMode run_mode,
GimpImage *image,
GFile *file,
GimpMetadata *metadata,
GimpProcedureConfig *config,
gpointer run_data);
static void create_cmyk_layer (GimpImage *image,
GimpLayer *layer,
const gchar *type_format,
const Babl *space,
gpointer picture_buffer,
gpointer key_buffer,
gint bit_depth,
gboolean has_alpha);
static void extract_cmyk (GeglBuffer *buffer,
gpointer *cmy_data,
gpointer *key_data,
gpointer *alpha_data,
const Babl *type,
const Babl *space,
gint width,
gint height,
gint bit_depth,
gboolean has_alpha);
2021-08-26 20:24:21 +08:00
G_DEFINE_TYPE (JpegXL, jpegxl, GIMP_TYPE_PLUG_IN)
GIMP_MAIN (JPEGXL_TYPE)
DEFINE_STD_SET_I18N
2021-08-26 20:24:21 +08:00
static void
jpegxl_class_init (JpegXLClass *klass)
{
GimpPlugInClass *plug_in_class = GIMP_PLUG_IN_CLASS (klass);
plug_in_class->query_procedures = jpegxl_query_procedures;
plug_in_class->create_procedure = jpegxl_create_procedure;
plug_in_class->set_i18n = STD_SET_I18N;
2021-08-26 20:24:21 +08:00
}
static void
jpegxl_init (JpegXL *jpeg_xl)
{
}
static GList *
jpegxl_query_procedures (GimpPlugIn *plug_in)
{
GList *list = NULL;
list = g_list_append (list, g_strdup (LOAD_PROC));
list = g_list_append (list, g_strdup (EXPORT_PROC));
2021-08-26 20:24:21 +08:00
return list;
}
static GimpProcedure *
jpegxl_create_procedure (GimpPlugIn *plug_in,
const gchar *name)
{
GimpProcedure *procedure = NULL;
if (! strcmp (name, LOAD_PROC))
{
procedure = gimp_load_procedure_new (plug_in, name,
GIMP_PDB_PROC_TYPE_PLUGIN,
jpegxl_load, NULL, NULL);
2021-08-26 20:24:21 +08:00
gimp_procedure_set_menu_label (procedure, _("JPEG XL image"));
2021-08-26 20:24:21 +08:00
gimp_procedure_set_documentation (procedure,
_("Loads files in the JPEG XL file format"),
_("Loads files in the JPEG XL file format"),
2021-08-26 20:24:21 +08:00
name);
gimp_procedure_set_attribution (procedure,
"Daniel Novomesky",
"(C) 2021 Daniel Novomesky",
"2021");
gimp_file_procedure_set_mime_types (GIMP_FILE_PROCEDURE (procedure),
"image/jxl");
gimp_file_procedure_set_extensions (GIMP_FILE_PROCEDURE (procedure),
"jxl");
gimp_file_procedure_set_magics (GIMP_FILE_PROCEDURE (procedure),
"0,string,\xFF\x0A,0,string,\\000\\000\\000\x0CJXL\\040\\015\\012\x87\\012");
2021-08-26 20:24:21 +08:00
}
else if (! strcmp (name, EXPORT_PROC))
2021-08-26 20:24:21 +08:00
{
procedure = gimp_export_procedure_new (plug_in, name,
GIMP_PDB_PROC_TYPE_PLUGIN,
FALSE, jpegxl_export, NULL, NULL);
2021-08-26 20:24:21 +08:00
gimp_procedure_set_image_types (procedure, "RGB*, GRAY*");
gimp_procedure_set_menu_label (procedure, _("JPEG XL image"));
2021-08-26 20:24:21 +08:00
gimp_procedure_set_documentation (procedure,
_("Saves files in the JPEG XL file format"),
_("Saves files in the JPEG XL file format"),
2021-08-26 20:24:21 +08:00
name);
gimp_procedure_set_attribution (procedure,
"Daniel Novomesky",
"(C) 2021 Daniel Novomesky",
"2021");
gimp_file_procedure_set_format_name (GIMP_FILE_PROCEDURE (procedure),
"JPEG XL");
2021-08-26 20:24:21 +08:00
gimp_file_procedure_set_mime_types (GIMP_FILE_PROCEDURE (procedure),
"image/jxl");
gimp_file_procedure_set_extensions (GIMP_FILE_PROCEDURE (procedure),
"jxl");
gimp_procedure_add_boolean_argument (procedure, "lossless",
_("L_ossless"),
_("Use lossless compression"),
FALSE,
G_PARAM_READWRITE);
gimp_procedure_add_double_argument (procedure, "compression",
_("Co_mpression/maxError"),
_("Max. butteraugli distance, lower = higher quality. Range: 0 .. 15. 1.0 = visually lossless."),
0.1, 15, 1,
G_PARAM_READWRITE);
gimp_procedure_add_int_argument (procedure, "save-bit-depth",
_("_Bit depth"),
_("Bit depth of exported image"),
8, 16, 8,
G_PARAM_READWRITE);
gimp_procedure_add_int_argument (procedure, "speed",
_("Effort/S_peed"),
_("Encoder effort setting"),
1, 9,
7,
G_PARAM_READWRITE);
gimp_procedure_add_boolean_argument (procedure, "uses-original-profile",
_("Save ori_ginal profile"),
_("Store ICC profile to exported JXL file"),
FALSE,
G_PARAM_READWRITE);
gimp_procedure_add_boolean_argument (procedure, "cmyk",
_("Export as CMY_K"),
_("Create a CMYK JPEG XL image using the soft-proofing color profile"),
FALSE,
G_PARAM_READWRITE);
gimp_procedure_add_boolean_argument (procedure, "save-exif",
_("Save Exi_f"),
_("Toggle saving Exif data"),
gimp_export_exif (),
G_PARAM_READWRITE);
gimp_procedure_add_boolean_argument (procedure, "save-xmp",
_("Save _XMP"),
_("Toggle saving XMP data"),
gimp_export_xmp (),
G_PARAM_READWRITE);
2021-08-26 20:24:21 +08:00
}
return procedure;
}
/* The Key data is stored in a separate extra
* channel. We combine the CMY values from the
* main image with the K values to create
* the final layer buffer.
*/
static void
create_cmyk_layer (GimpImage *image,
GimpLayer *layer,
const gchar *type_format,
const Babl *space,
gpointer cmy_data,
gpointer key_data,
gint bit_depth,
gboolean has_alpha)
{
const Babl *cmy_format = NULL;
const Babl *cmyka_format = NULL;
const Babl *key_format = NULL;
const Babl *rgb_format = NULL;
GeglBuffer *output_buffer;
GeglBuffer *picture_buffer;
GeglBuffer *cmy_buffer;
GeglBuffer *key_buffer;
GeglBufferIterator *iter;
GeglColor *fill_color = gegl_color_new ("rgba(0.0,0.0,0.0,0.0)");
gint width;
gint height;
gint n_components = 3;
const Babl *type = babl_type (type_format);
GString *rgb_type = g_string_new (NULL);
GString *cmy_type = g_string_new (NULL);
GString *cmyka_type = g_string_new (NULL);
width = gimp_image_get_width (image);
height = gimp_image_get_height (image);
if (has_alpha)
n_components = 4;
gimp_image_insert_layer (image, layer, NULL, 0);
output_buffer = gimp_drawable_get_buffer (GIMP_DRAWABLE (layer));
g_string_printf (rgb_type, has_alpha ? "R'G'B'A %s" : "R'G'B' %s", type_format);
g_string_printf (cmy_type, "cmyk %s", type_format);
g_string_printf (cmyka_type, "cmykA %s", type_format);
rgb_format = babl_format (rgb_type->str);
cmy_format = babl_format_with_space ( cmy_type->str, space);
cmyka_format = babl_format_with_space ( cmyka_type->str, space);
key_format = babl_format_new (babl_model ("Y"),
type,
babl_component ("Y"),
NULL);
cmy_format = babl_format_with_space (babl_format_get_encoding (cmy_format),
space);
cmyka_format = babl_format_with_space (babl_format_get_encoding (cmyka_format),
space);
key_format = babl_format_with_space (babl_format_get_encoding (key_format),
space);
picture_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
has_alpha ? cmyka_format : cmy_format);
gegl_buffer_set_color (picture_buffer, NULL, fill_color);
cmy_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
babl_format_n (type, n_components));
key_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
key_format);
gegl_buffer_set (cmy_buffer, GEGL_RECTANGLE (0, 0, width, height), 0,
babl_format_n (type, n_components), cmy_data,
GEGL_AUTO_ROWSTRIDE);
gegl_buffer_set (key_buffer, GEGL_RECTANGLE (0, 0, width, height), 0,
key_format, key_data, GEGL_AUTO_ROWSTRIDE);
iter = gegl_buffer_iterator_new (picture_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
has_alpha ? cmyka_format : cmy_format,
GEGL_BUFFER_READWRITE,
GEGL_ABYSS_NONE, 4);
gegl_buffer_iterator_add (iter, cmy_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
babl_format_n (type, n_components),
GEGL_ACCESS_READ, GEGL_ABYSS_NONE);
gegl_buffer_iterator_add (iter, key_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
key_format, GEGL_ACCESS_READ, GEGL_ABYSS_NONE);
gegl_buffer_iterator_add (iter, output_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
rgb_format, GEGL_BUFFER_READWRITE, GEGL_ABYSS_NONE);
while (gegl_buffer_iterator_next (iter))
{
guchar *pixel = iter->items[0].data;
guchar *cmy = iter->items[1].data;
guchar *k = iter->items[2].data;
guchar *output = iter->items[3].data;
gint length = iter->length;
gint row = length;
while (length--)
{
gint i;
for (i = 0; i < 3 * bit_depth; i++)
pixel[i] = cmy[i];
for (i = 0; i < bit_depth; i++)
pixel[i + (3 * bit_depth)] = k[i];
if (has_alpha)
{
for (i = 0; i < bit_depth; i++)
pixel[i + (4 * bit_depth)] = cmy[i + (3 * bit_depth)];
}
pixel += 4 * bit_depth;
cmy += 3 * bit_depth;
k += bit_depth;
if (has_alpha)
{
pixel += bit_depth;
cmy += bit_depth;
}
output += 4 * bit_depth;
}
/* Convert row from CMYK/A to RGB, due to layer buffers
* having a maximum of 4 colors currently */
pixel -= (4 * bit_depth) * row;
if (has_alpha)
pixel -= row * bit_depth;
output -= (4 * bit_depth) * row;
babl_process (babl_fish (has_alpha ? cmyka_format : cmy_format, rgb_format),
pixel, output, row);
}
g_object_unref (output_buffer);
g_object_unref (picture_buffer);
g_object_unref (cmy_buffer);
g_object_unref (key_buffer);
g_free (key_data);
g_string_free (cmyka_type, TRUE);
g_string_free (cmy_type, TRUE);
g_string_free (rgb_type, TRUE);
}
static GimpImage *
load_image (GFile *file,
GimpRunMode runmode,
GimpMetadata *metadata,
GimpMetadataLoadFlags *flags,
GError **error)
2021-08-26 20:24:21 +08:00
{
FILE *inputFile = g_fopen (g_file_peek_path (file), "rb");
2021-08-26 20:24:21 +08:00
gsize inputFileSize;
gpointer memory;
JxlSignature signature;
JxlDecoder *decoder;
void *runner;
JxlBasicInfo basicinfo;
JxlDecoderStatus status;
JxlPixelFormat pixel_format;
JxlColorEncoding color_encoding;
size_t icc_size = 0;
GimpColorProfile *profile = NULL;
gboolean loadlinear = FALSE;
size_t channel_depth;
2021-08-26 20:24:21 +08:00
size_t result_size;
size_t extra_channel_size = 0;
2021-08-26 20:24:21 +08:00
gpointer picture_buffer;
gpointer key_buffer = NULL;
gboolean is_cmyk = FALSE;
gboolean has_alpha = FALSE;
gint cmyk_channel_id = -1;
2021-08-26 20:24:21 +08:00
GimpImage *image;
GimpLayer *layer;
GeglBuffer *buffer;
const Babl *space = NULL;
const gchar *type;
GimpPrecision precision_linear;
GimpPrecision precision_non_linear;
uint32_t i;
2021-08-26 20:24:21 +08:00
if (!inputFile)
{
g_set_error (error, G_FILE_ERROR, 0,
"Cannot open file for read: %s\n",
g_file_peek_path (file));
2021-08-26 20:24:21 +08:00
return NULL;
}
fseek (inputFile, 0, SEEK_END);
inputFileSize = ftell (inputFile);
fseek (inputFile, 0, SEEK_SET);
if (inputFileSize < 1)
{
g_set_error (error, G_FILE_ERROR, 0,
"File too small: %s\n",
g_file_peek_path (file));
2021-08-26 20:24:21 +08:00
fclose (inputFile);
return NULL;
}
memory = g_malloc (inputFileSize);
if (fread (memory, 1, inputFileSize, inputFile) != inputFileSize)
{
g_set_error (error, G_FILE_ERROR, 0,
"Failed to read %zu bytes: %s\n", inputFileSize,
g_file_peek_path (file));
2021-08-26 20:24:21 +08:00
fclose (inputFile);
g_free (memory);
return NULL;
}
fclose (inputFile);
signature = JxlSignatureCheck (memory, inputFileSize);
if (signature != JXL_SIG_CODESTREAM && signature != JXL_SIG_CONTAINER)
{
g_set_error (error, G_FILE_ERROR, 0,
"File %s is probably not in JXL format!\n",
g_file_peek_path (file));
2021-08-26 20:24:21 +08:00
g_free (memory);
return NULL;
}
decoder = JxlDecoderCreate (NULL);
if (!decoder)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderCreate failed");
g_free (memory);
return NULL;
}
runner = JxlThreadParallelRunnerCreate (NULL, gimp_get_num_processors ());
if (JxlDecoderSetParallelRunner (decoder, JxlThreadParallelRunner, runner) != JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderSetParallelRunner failed");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (JxlDecoderSetInput (decoder, memory, inputFileSize) != JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderSetInput failed");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
JxlDecoderCloseInput (decoder);
2021-08-26 20:24:21 +08:00
if (JxlDecoderSubscribeEvents (decoder, JXL_DEC_BASIC_INFO | JXL_DEC_COLOR_ENCODING | JXL_DEC_FULL_IMAGE) != JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderSubscribeEvents failed");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
status = JxlDecoderProcessInput (decoder);
if (status == JXL_DEC_ERROR)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JXL decoding failed");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (status == JXL_DEC_NEED_MORE_INPUT)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JXL data incomplete");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
status = JxlDecoderGetBasicInfo (decoder, &basicinfo);
if (status != JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JXL basic info not available");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (basicinfo.xsize == 0 || basicinfo.ysize == 0)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JXL image has zero dimensions");
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
status = JxlDecoderProcessInput (decoder);
if (status != JXL_DEC_COLOR_ENCODING)
{
g_set_error (error, G_FILE_ERROR, 0,
"Unexpected event %d instead of JXL_DEC_COLOR_ENCODING", status);
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (basicinfo.uses_original_profile == JXL_FALSE)
{
if (basicinfo.num_color_channels == 3)
{
JxlColorEncodingSetToSRGB (&color_encoding, JXL_FALSE);
JxlDecoderSetPreferredColorProfile (decoder, &color_encoding);
}
else if (basicinfo.num_color_channels == 1)
{
JxlColorEncodingSetToSRGB (&color_encoding, JXL_TRUE);
JxlDecoderSetPreferredColorProfile (decoder, &color_encoding);
}
}
pixel_format.endianness = JXL_NATIVE_ENDIAN;
pixel_format.align = 0;
if (basicinfo.uses_original_profile == JXL_FALSE || basicinfo.bits_per_sample > 16)
{
pixel_format.data_type = JXL_TYPE_FLOAT;
channel_depth = 4;
precision_linear = GIMP_PRECISION_FLOAT_LINEAR;
precision_non_linear = GIMP_PRECISION_FLOAT_NON_LINEAR;
type = "float";
}
else if (basicinfo.bits_per_sample <= 8)
{
pixel_format.data_type = JXL_TYPE_UINT8;
channel_depth = 1;
precision_linear = GIMP_PRECISION_U8_LINEAR;
precision_non_linear = GIMP_PRECISION_U8_NON_LINEAR;
type = "u8";
}
else if (basicinfo.exponent_bits_per_sample == 0)
{
pixel_format.data_type = JXL_TYPE_UINT16;
channel_depth = 2;
precision_linear = GIMP_PRECISION_U16_LINEAR;
precision_non_linear = GIMP_PRECISION_U16_NON_LINEAR;
type = "u16";
}
else
{
pixel_format.data_type = JXL_TYPE_FLOAT16;
channel_depth = 2;
precision_linear = GIMP_PRECISION_HALF_LINEAR;
precision_non_linear = GIMP_PRECISION_HALF_NON_LINEAR;
type = "half";
}
2021-08-26 20:24:21 +08:00
if (basicinfo.num_color_channels == 1) /* grayscale */
{
if (basicinfo.alpha_bits > 0)
{
pixel_format.num_channels = 2;
}
else
{
pixel_format.num_channels = 1;
}
}
else /* RGB */
{
if (basicinfo.alpha_bits > 0) /* RGB with alpha */
{
pixel_format.num_channels = 4;
}
else /* RGB no alpha */
{
pixel_format.num_channels = 3;
}
}
/* Check for extra channels */
for (i = 0; i < basicinfo.num_extra_channels; i++)
{
JxlExtraChannelInfo extra;
if (JXL_DEC_SUCCESS != JxlDecoderGetExtraChannelInfo (decoder, i, &extra))
break;
/* K channel for CMYK images */
if (extra.type == JXL_CHANNEL_BLACK)
{
is_cmyk = TRUE;
cmyk_channel_id = i;
if (pixel_format.num_channels < 3)
pixel_format.num_channels = 3;
}
/* Confirm presence of alpha channel */
if (extra.type == JXL_CHANNEL_ALPHA)
{
has_alpha = TRUE;
pixel_format.num_channels = 4;
}
}
result_size = channel_depth * pixel_format.num_channels * (size_t) basicinfo.xsize * (size_t) basicinfo.ysize;
extra_channel_size = channel_depth * basicinfo.num_extra_channels * (size_t) basicinfo.xsize * (size_t) basicinfo.ysize;
result_size += extra_channel_size;
2021-08-26 20:24:21 +08:00
if (JxlDecoderGetColorAsEncodedProfile (decoder,
#if JPEGXL_NUMERIC_VERSION < JPEGXL_COMPUTE_NUMERIC_VERSION(0,9,0)
&pixel_format,
#endif
JXL_COLOR_PROFILE_TARGET_DATA,
&color_encoding) == JXL_DEC_SUCCESS)
2021-08-26 20:24:21 +08:00
{
if (color_encoding.white_point == JXL_WHITE_POINT_D65)
{
switch (color_encoding.transfer_function)
{
case JXL_TRANSFER_FUNCTION_LINEAR:
loadlinear = TRUE;
switch (color_encoding.color_space)
{
case JXL_COLOR_SPACE_RGB:
profile = gimp_color_profile_new_rgb_srgb_linear ();
break;
case JXL_COLOR_SPACE_GRAY:
profile = gimp_color_profile_new_d65_gray_linear ();
break;
default:
break;
}
break;
case JXL_TRANSFER_FUNCTION_SRGB:
switch (color_encoding.color_space)
{
case JXL_COLOR_SPACE_RGB:
profile = gimp_color_profile_new_rgb_srgb ();
break;
case JXL_COLOR_SPACE_GRAY:
profile = gimp_color_profile_new_d65_gray_srgb_trc ();
break;
default:
break;
}
break;
default:
break;
}
}
}
if (! profile)
2021-08-26 20:24:21 +08:00
{
if (JxlDecoderGetICCProfileSize (decoder,
#if JPEGXL_NUMERIC_VERSION < JPEGXL_COMPUTE_NUMERIC_VERSION(0,9,0)
&pixel_format,
#endif
JXL_COLOR_PROFILE_TARGET_DATA,
&icc_size) == JXL_DEC_SUCCESS)
2021-08-26 20:24:21 +08:00
{
if (icc_size > 0)
{
gpointer raw_icc_profile = g_malloc (icc_size);
if (JxlDecoderGetColorAsICCProfile (decoder,
#if JPEGXL_NUMERIC_VERSION < JPEGXL_COMPUTE_NUMERIC_VERSION(0,9,0)
&pixel_format,
#endif
JXL_COLOR_PROFILE_TARGET_DATA,
raw_icc_profile, icc_size)
2021-08-26 20:24:21 +08:00
== JXL_DEC_SUCCESS)
{
profile = gimp_color_profile_new_from_icc_profile (raw_icc_profile, icc_size, error);
if (profile)
{
loadlinear = gimp_color_profile_is_linear (profile);
}
else
{
g_printerr ("%s: Failed to read ICC profile: %s\n", G_STRFUNC, (*error)->message);
g_clear_error (error);
}
}
else
{
g_printerr ("Failed to obtain data from JPEG XL decoder");
}
g_free (raw_icc_profile);
}
else
{
g_printerr ("Empty ICC data");
}
}
else
{
g_message ("no ICC, other color profile");
}
}
status = JxlDecoderProcessInput (decoder);
if (status != JXL_DEC_NEED_IMAGE_OUT_BUFFER)
{
g_set_error (error, G_FILE_ERROR, 0,
"Unexpected event %d instead of JXL_DEC_NEED_IMAGE_OUT_BUFFER", status);
if (profile)
{
g_object_unref (profile);
}
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
picture_buffer = g_try_malloc (result_size);
if (! picture_buffer)
{
g_set_error (error, G_FILE_ERROR, 0, "Memory could not be allocated.");
if (profile)
{
g_object_unref (profile);
}
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
2021-08-26 20:24:21 +08:00
if (JxlDecoderSetImageOutBuffer (decoder, &pixel_format, picture_buffer, result_size) != JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderSetImageOutBuffer failed");
if (profile)
{
g_object_unref (profile);
}
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
/* Loading key channel buffer data */
if (is_cmyk)
{
if (JxlDecoderExtraChannelBufferSize (decoder, &pixel_format,
&result_size, cmyk_channel_id)
!= JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderExtraChannelBufferSize failed");
if (profile)
g_object_unref (profile);
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
key_buffer = g_try_malloc (result_size);
if (! key_buffer)
{
g_set_error (error, G_FILE_ERROR, 0, "Memory could not be allocated.");
if (profile)
g_object_unref (profile);
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (JxlDecoderSetExtraChannelBuffer (decoder, &pixel_format, key_buffer,
result_size, cmyk_channel_id)
!= JXL_DEC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"ERROR: JxlDecoderSetExtraChannelBuffer failed");
if (profile)
g_object_unref (profile);
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
}
2021-08-26 20:24:21 +08:00
status = JxlDecoderProcessInput (decoder);
if (status != JXL_DEC_FULL_IMAGE)
{
g_set_error (error, G_FILE_ERROR, 0,
"Unexpected event %d instead of JXL_DEC_FULL_IMAGE", status);
g_free (picture_buffer);
if (profile)
{
g_object_unref (profile);
}
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return NULL;
}
if (basicinfo.num_color_channels == 1) /* grayscale */
{
image = gimp_image_new_with_precision (basicinfo.xsize, basicinfo.ysize, GIMP_GRAY,
loadlinear ? precision_linear : precision_non_linear);
2021-08-26 20:24:21 +08:00
if (profile)
{
if (gimp_color_profile_is_gray (profile))
{
gimp_image_set_color_profile (image, profile);
}
}
layer = gimp_layer_new (image, "Background",
basicinfo.xsize, basicinfo.ysize,
(basicinfo.alpha_bits > 0) ? GIMP_GRAYA_IMAGE : GIMP_GRAY_IMAGE, 100,
gimp_image_get_default_new_layer_mode (image));
}
else /* RGB or CMYK */
2021-08-26 20:24:21 +08:00
{
image = gimp_image_new_with_precision (basicinfo.xsize, basicinfo.ysize, GIMP_RGB,
loadlinear ? precision_linear : precision_non_linear);
2021-08-26 20:24:21 +08:00
if (profile)
{
if (gimp_color_profile_is_rgb (profile))
{
gimp_image_set_color_profile (image, profile);
}
else if (is_cmyk && gimp_color_profile_is_cmyk (profile))
{
gimp_image_set_simulation_profile (image, profile);
space = gimp_color_profile_get_space (profile,
GIMP_COLOR_RENDERING_INTENT_RELATIVE_COLORIMETRIC,
NULL);
}
2021-08-26 20:24:21 +08:00
}
layer = gimp_layer_new (image, "Background",
basicinfo.xsize, basicinfo.ysize,
(basicinfo.alpha_bits > 0) ? GIMP_RGBA_IMAGE : GIMP_RGB_IMAGE, 100,
gimp_image_get_default_new_layer_mode (image));
}
if (is_cmyk)
{
create_cmyk_layer (image, layer, type, space, picture_buffer,
key_buffer, channel_depth, has_alpha);
}
else
{
gimp_image_insert_layer (image, layer, NULL, 0);
2021-08-26 20:24:21 +08:00
buffer = gimp_drawable_get_buffer (GIMP_DRAWABLE (layer));
2021-08-26 20:24:21 +08:00
gegl_buffer_set (buffer, GEGL_RECTANGLE (0, 0, basicinfo.xsize, basicinfo.ysize), 0,
NULL, picture_buffer, GEGL_AUTO_ROWSTRIDE);
2021-08-26 20:24:21 +08:00
g_object_unref (buffer);
}
2021-08-26 20:24:21 +08:00
g_free (picture_buffer);
if (profile)
{
g_object_unref (profile);
}
if (metadata)
{
if (basicinfo.have_container)
{
JxlDecoderReleaseInput (decoder);
JxlDecoderRewind (decoder);
if (JxlDecoderSetInput (decoder, memory, inputFileSize) != JXL_DEC_SUCCESS)
{
g_printerr ("%s: JxlDecoderSetInput failed after JxlDecoderRewind\n", G_STRFUNC);
}
else
{
JxlDecoderCloseInput (decoder);
if (JxlDecoderSubscribeEvents (decoder, JXL_DEC_BOX) != JXL_DEC_SUCCESS)
{
g_printerr ("%s: JxlDecoderSubscribeEvents for JXL_DEC_BOX failed\n", G_STRFUNC);
}
else
{
gboolean search_exif = TRUE;
gboolean search_xmp = TRUE;
gboolean success_exif = FALSE;
gboolean success_xmp = FALSE;
JxlBoxType box_type = { 0, 0, 0, 0 };
GByteArray *exif_box = NULL;
GByteArray *xml_box = NULL;
size_t exif_remains = 0;
size_t xml_remains = 0;
while (search_exif || search_xmp)
{
status = JxlDecoderProcessInput (decoder);
switch (status)
{
case JXL_DEC_SUCCESS:
if (box_type[0] == 'E' && box_type[1] == 'x' && box_type[2] == 'i' && box_type[3] == 'f' && search_exif)
{
exif_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (exif_box, exif_box->len - exif_remains);
success_exif = TRUE;
}
else if (box_type[0] == 'x' && box_type[1] == 'm' && box_type[2] == 'l' && box_type[3] == ' ' && search_xmp)
{
xml_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (xml_box, xml_box->len - xml_remains);
success_xmp = TRUE;
}
search_exif = FALSE;
search_xmp = FALSE;
break;
case JXL_DEC_ERROR:
search_exif = FALSE;
search_xmp = FALSE;
g_printerr ("%s: Metadata decoding error\n", G_STRFUNC);
break;
case JXL_DEC_NEED_MORE_INPUT:
search_exif = FALSE;
search_xmp = FALSE;
g_printerr ("%s: JXL metadata are probably incomplete\n", G_STRFUNC);
break;
case JXL_DEC_BOX:
JxlDecoderSetDecompressBoxes (decoder, JXL_TRUE);
if (box_type[0] == 'E' && box_type[1] == 'x' && box_type[2] == 'i' && box_type[3] == 'f' && search_exif && exif_box)
{
exif_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (exif_box, exif_box->len - exif_remains);
search_exif = FALSE;
success_exif = TRUE;
}
else if (box_type[0] == 'x' && box_type[1] == 'm' && box_type[2] == 'l' && box_type[3] == ' ' && search_xmp && xml_box)
{
xml_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (xml_box, xml_box->len - xml_remains);
search_xmp = FALSE;
success_xmp = TRUE;
}
if (JxlDecoderGetBoxType (decoder, box_type, JXL_TRUE) == JXL_DEC_SUCCESS)
{
if (box_type[0] == 'E' && box_type[1] == 'x' && box_type[2] == 'i' && box_type[3] == 'f' && search_exif)
{
exif_box = g_byte_array_sized_new (4096);
g_byte_array_set_size (exif_box, 4096);
JxlDecoderSetBoxBuffer (decoder, exif_box->data, exif_box->len);
}
else if (box_type[0] == 'x' && box_type[1] == 'm' && box_type[2] == 'l' && box_type[3] == ' ' && search_xmp)
{
xml_box = g_byte_array_sized_new (4096);
g_byte_array_set_size (xml_box, 4096);
JxlDecoderSetBoxBuffer (decoder, xml_box->data, xml_box->len);
}
}
else
{
search_exif = FALSE;
search_xmp = FALSE;
g_printerr ("%s: Error in JxlDecoderGetBoxType\n", G_STRFUNC);
}
break;
case JXL_DEC_BOX_NEED_MORE_OUTPUT:
if (box_type[0] == 'E' && box_type[1] == 'x' && box_type[2] == 'i' && box_type[3] == 'f' && search_exif)
{
exif_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (exif_box, exif_box->len + 4096);
JxlDecoderSetBoxBuffer (decoder, exif_box->data + exif_box->len - (4096 + exif_remains), 4096 + exif_remains);
}
else if (box_type[0] == 'x' && box_type[1] == 'm' && box_type[2] == 'l' && box_type[3] == ' ' && search_xmp)
{
xml_remains = JxlDecoderReleaseBoxBuffer (decoder);
g_byte_array_set_size (xml_box, xml_box->len + 4096);
JxlDecoderSetBoxBuffer (decoder, xml_box->data + xml_box->len - (4096 + xml_remains), 4096 + xml_remains);
}
else
{
search_exif = FALSE;
search_xmp = FALSE;
}
break;
default:
break;
}
}
if (success_exif || success_xmp)
{
gexiv2_metadata_clear (GEXIV2_METADATA (metadata));
if (success_exif && exif_box)
{
const guint8 tiffHeaderBE[4] = { 'M', 'M', 0, 42 };
const guint8 tiffHeaderLE[4] = { 'I', 'I', 42, 0 };
const guint8 *tiffheader = exif_box->data;
glong new_exif_size = exif_box->len;
while (new_exif_size >= 4) /*Searching for TIFF Header*/
{
if (tiffheader[0] == tiffHeaderBE[0] && tiffheader[1] == tiffHeaderBE[1] &&
tiffheader[2] == tiffHeaderBE[2] && tiffheader[3] == tiffHeaderBE[3])
{
break;
}
if (tiffheader[0] == tiffHeaderLE[0] && tiffheader[1] == tiffHeaderLE[1] &&
tiffheader[2] == tiffHeaderLE[2] && tiffheader[3] == tiffHeaderLE[3])
{
break;
}
new_exif_size--;
tiffheader++;
}
if (new_exif_size > 4) /* TIFF header + some data found*/
{
if (! gexiv2_metadata_open_buf (GEXIV2_METADATA (metadata), tiffheader, new_exif_size, error))
{
g_printerr ("%s: Failed to set EXIF metadata: %s\n", G_STRFUNC, (*error)->message);
g_clear_error (error);
}
}
else
{
g_printerr ("%s: EXIF metadata not set\n", G_STRFUNC);
}
}
if (success_xmp && xml_box)
{
if (! gimp_metadata_set_from_xmp (metadata, xml_box->data, xml_box->len, error))
{
g_printerr ("%s: Failed to set XMP metadata: %s\n", G_STRFUNC, (*error)->message);
g_clear_error (error);
}
}
}
if (exif_box)
{
g_byte_array_free (exif_box, TRUE);
}
if (xml_box)
{
g_byte_array_free (xml_box, TRUE);
}
}
}
}
gexiv2_metadata_try_set_orientation (GEXIV2_METADATA (metadata),
GEXIV2_ORIENTATION_NORMAL, NULL);
gexiv2_metadata_try_set_metadata_pixel_width (GEXIV2_METADATA (metadata),
basicinfo.xsize, NULL);
gexiv2_metadata_try_set_metadata_pixel_height (GEXIV2_METADATA (metadata),
basicinfo.ysize, NULL);
*flags = GIMP_METADATA_LOAD_COMMENT | GIMP_METADATA_LOAD_RESOLUTION;
}
2021-08-26 20:24:21 +08:00
JxlThreadParallelRunnerDestroy (runner);
JxlDecoderDestroy (decoder);
g_free (memory);
return image;
}
static GimpValueArray *
jpegxl_load (GimpProcedure *procedure,
GimpRunMode run_mode,
GFile *file,
GimpMetadata *metadata,
GimpMetadataLoadFlags *flags,
GimpProcedureConfig *config,
gpointer run_data)
2021-08-26 20:24:21 +08:00
{
GimpValueArray *return_vals;
GimpImage *image;
GError *error = NULL;
2021-08-26 20:24:21 +08:00
gegl_init (NULL, NULL);
image = load_image (file, run_mode, metadata, flags, &error);
2021-08-26 20:24:21 +08:00
if (! image)
return gimp_procedure_new_return_values (procedure,
GIMP_PDB_EXECUTION_ERROR,
error);
2021-08-26 20:24:21 +08:00
return_vals = gimp_procedure_new_return_values (procedure,
GIMP_PDB_SUCCESS,
NULL);
GIMP_VALUES_SET_IMAGE (return_vals, 1, image);
return return_vals;
}
static void
extract_cmyk (GeglBuffer *buffer,
gpointer *cmy_data,
gpointer *key_data,
gpointer *alpha_data,
const Babl *type,
const Babl *space,
gint width,
gint height,
gint bit_depth,
gboolean has_alpha)
{
GeglBuffer *cmy_buffer;
GeglBuffer *key_buffer;
GeglBuffer *alpha_buffer = NULL;
const Babl *format;
const Babl *key_format = NULL;
GeglBufferIterator *iter;
GeglColor *white = gegl_color_new ("white");
if (has_alpha)
{
format = babl_format_new (babl_model ("cmykA"),
type,
babl_component ("cyan"),
babl_component ("magenta"),
babl_component ("yellow"),
babl_component ("key"),
babl_component ("A"),
NULL);
}
else
{
format = babl_format_new (babl_model ("cmyk"),
type,
babl_component ("cyan"),
babl_component ("magenta"),
babl_component ("yellow"),
babl_component ("key"),
NULL);
}
format = babl_format_with_space (babl_format_get_encoding (format), space);
key_format = babl_format_new (babl_model ("Y"),
type,
babl_component ("Y"),
NULL);
key_format = babl_format_with_space (babl_format_get_encoding (key_format),
space);
cmy_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
babl_format_n (type, 3));
gegl_buffer_set_color (cmy_buffer, NULL, white);
key_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
key_format);
gegl_buffer_set_color (key_buffer, NULL, white);
if (has_alpha)
{
alpha_buffer = gegl_buffer_new (GEGL_RECTANGLE (0, 0, width, height),
babl_format_n (type, 1));
gegl_buffer_set_color (alpha_buffer, NULL, white);
}
g_object_unref (white);
iter = gegl_buffer_iterator_new (buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
format,
GEGL_BUFFER_READWRITE,
GEGL_ABYSS_NONE, 4);
gegl_buffer_iterator_add (iter, cmy_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
babl_format_n (type, 3), GEGL_BUFFER_READWRITE,
GEGL_ABYSS_NONE);
gegl_buffer_iterator_add (iter, key_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
key_format, GEGL_BUFFER_READWRITE,
GEGL_ABYSS_NONE);
if (has_alpha)
gegl_buffer_iterator_add (iter, alpha_buffer,
GEGL_RECTANGLE (0, 0, width, height), 0,
babl_format_n (type, 1), GEGL_BUFFER_READWRITE,
GEGL_ABYSS_NONE);
while (gegl_buffer_iterator_next (iter))
{
guchar *pixel = iter->items[0].data;
guchar *cmy = iter->items[1].data;
guchar *k = iter->items[2].data;
guchar *a = NULL;
gint length = iter->length;
if (has_alpha)
a = iter->items[3].data;
while (length--)
{
gint range = 1;
gint i;
if (bit_depth > 8)
range = 2;
for (i = 0; i < (3 * range); i++)
cmy[i] = pixel[i];
for (i = 0; i < range; i++)
k[i] = pixel[i + (3 * range)];
if (has_alpha)
{
for (i = 0; i < range; i++)
a[i] = pixel[i + (4 * range)];
}
pixel += 4 * range;
cmy += 3 * range;
k += range;
if (has_alpha)
{
a += range;
pixel += range;
}
}
}
gegl_buffer_get (cmy_buffer, GEGL_RECTANGLE (0, 0,
width, height), 1.0,
babl_format_n (type, 3), *cmy_data,
GEGL_AUTO_ROWSTRIDE, GEGL_ABYSS_NONE);
gegl_buffer_get (key_buffer, GEGL_RECTANGLE (0, 0,
width, height), 1.0,
key_format, *key_data,
GEGL_AUTO_ROWSTRIDE, GEGL_ABYSS_NONE);
if (has_alpha)
gegl_buffer_get (alpha_buffer, GEGL_RECTANGLE (0, 0,
width, height), 1.0,
babl_format_n (type, 1), *alpha_data,
GEGL_AUTO_ROWSTRIDE, GEGL_ABYSS_NONE);
g_object_unref (cmy_buffer);
g_object_unref (key_buffer);
if (has_alpha)
g_object_unref (alpha_buffer);
}
static gboolean
export_image (GFile *file,
GimpProcedureConfig *config,
GimpImage *image,
GimpDrawable *drawable,
GimpMetadata *metadata,
GError **error)
2021-08-26 20:24:21 +08:00
{
JxlEncoder *encoder;
void *runner;
JxlEncoderFrameSettings *encoder_options;
JxlPixelFormat pixel_format;
JxlBasicInfo output_info;
JxlColorEncoding color_profile;
JxlEncoderStatus status;
size_t buffer_size;
size_t total_channel_size = 0;
size_t channel_buffer_size = 0;
GByteArray *compressed;
FILE *outfile;
GeglBuffer *buffer;
GimpImageType drawable_type;
gint drawable_width;
gint drawable_height;
gpointer picture_buffer;
gpointer cmy_buffer = NULL;
gpointer key_buffer = NULL;
gpointer alpha_buffer = NULL;
GimpColorProfile *profile = NULL;
const Babl *file_format = NULL;
const Babl *space = NULL;
const Babl *type = NULL;
gboolean out_linear = FALSE;
size_t offset = 0;
uint8_t *next_out;
size_t avail_out;
gdouble compression = 1.0;
gboolean lossless = FALSE;
gint speed = 7;
gint bit_depth = 8;
gboolean cmyk = FALSE;
gboolean uses_original_profile = FALSE;
gboolean save_exif = FALSE;
gboolean save_xmp = FALSE;
2021-08-26 20:24:21 +08:00
gimp_progress_init_printf (_("Exporting '%s'"),
gimp_file_get_utf8_name (file));
2021-08-26 20:24:21 +08:00
g_object_get (config,
"lossless", &lossless,
"compression", &compression,
"speed", &speed,
"save-bit-depth", &bit_depth,
"cmyk", &cmyk,
2021-08-26 20:24:21 +08:00
"uses-original-profile", &uses_original_profile,
"save-exif", &save_exif,
"save-xmp", &save_xmp,
2021-08-26 20:24:21 +08:00
NULL);
if (lossless || cmyk)
{
/* JPEG XL developers recommend enabling uses_original_profile
* for better lossless compression efficiency.
* Profile must be saved for CMYK export */
uses_original_profile = TRUE;
}
else
{
/* 0.1 is actually minimal value for lossy in libjxl 0.5
* 0.01 is allowed in libjxl 0.6 but
* using too low value with lossy compression is not wise */
if (compression < 0.1)
{
compression = 0.1;
}
}
2021-08-26 20:24:21 +08:00
drawable_type = gimp_drawable_type (drawable);
drawable_width = gimp_drawable_get_width (drawable);
drawable_height = gimp_drawable_get_height (drawable);
JxlEncoderInitBasicInfo(&output_info);
2021-08-26 20:24:21 +08:00
if (uses_original_profile)
{
output_info.uses_original_profile = JXL_TRUE;
if (cmyk)
profile = gimp_image_get_simulation_profile (image);
else
profile = gimp_image_get_effective_color_profile (image);
/* CMYK profile is required for export. If not assigned,
* disable CMYK flag and revert to RGB */
if (cmyk && ! profile)
{
cmyk = FALSE;
profile = gimp_image_get_effective_color_profile (image);
}
2021-08-26 20:24:21 +08:00
out_linear = gimp_color_profile_is_linear (profile);
space = gimp_color_profile_get_space (profile,
GIMP_COLOR_RENDERING_INTENT_RELATIVE_COLORIMETRIC,
error);
if (error && *error)
{
g_printerr ("%s: error getting the profile space: %s\n",
G_STRFUNC, (*error)->message);
2021-08-26 20:24:21 +08:00
return FALSE;
}
}
else
{
output_info.uses_original_profile = JXL_FALSE;
2021-08-26 20:24:21 +08:00
space = babl_space ("sRGB");
out_linear = FALSE;
}
if (bit_depth > 8)
{
pixel_format.data_type = JXL_TYPE_UINT16;
output_info.bits_per_sample = 16;
}
else
{
pixel_format.data_type = JXL_TYPE_UINT8;
output_info.bits_per_sample = 8;
}
2021-08-26 20:24:21 +08:00
pixel_format.endianness = JXL_NATIVE_ENDIAN;
pixel_format.align = 0;
output_info.xsize = drawable_width;
output_info.ysize = drawable_height;
output_info.exponent_bits_per_sample = 0;
output_info.orientation = JXL_ORIENT_IDENTITY;
output_info.animation.tps_numerator = 10;
output_info.animation.tps_denominator = 1;
output_info.num_extra_channels = 0;
2021-08-26 20:24:21 +08:00
if (cmyk)
2021-08-26 20:24:21 +08:00
{
if (bit_depth > 8)
2021-08-26 20:24:21 +08:00
{
file_format = babl_format_with_space (gimp_drawable_has_alpha (drawable) ?
"cmykA u16" : "cmyk u16", space);
type = babl_type ("u16");
2021-08-26 20:24:21 +08:00
}
else
{
file_format = babl_format_with_space (gimp_drawable_has_alpha (drawable) ?
"cmykA u8" : "cmyk u8", space);
type = babl_type ("u8");
2021-08-26 20:24:21 +08:00
}
output_info.num_color_channels = 3;
pixel_format.num_channels = 3;
JxlColorEncodingSetToSRGB (&color_profile, JXL_FALSE);
2021-08-26 20:24:21 +08:00
if (gimp_drawable_has_alpha (drawable))
{
output_info.num_extra_channels = 2;
output_info.alpha_bits = 8;
if (bit_depth > 8)
output_info.alpha_bits = 16;
output_info.alpha_exponent_bits = 0;
}
else
{
output_info.num_extra_channels = 1;
output_info.alpha_bits = 0;
}
}
else /* For RGB and grayscale export */
{
switch (drawable_type)
2021-08-26 20:24:21 +08:00
{
case GIMP_GRAYA_IMAGE:
if (uses_original_profile && out_linear)
{
file_format = babl_format ( (bit_depth > 8) ? "YA u16" : "YA u8");
JxlColorEncodingSetToLinearSRGB (&color_profile, JXL_TRUE);
}
else
{
file_format = babl_format ( (bit_depth > 8) ? "Y'A u16" : "Y'A u8");
JxlColorEncodingSetToSRGB (&color_profile, JXL_TRUE);
}
pixel_format.num_channels = 2;
output_info.num_color_channels = 1;
output_info.alpha_bits = (bit_depth > 8) ? 16 : 8;
output_info.alpha_exponent_bits = 0;
output_info.num_extra_channels = 1;
uses_original_profile = FALSE;
break;
case GIMP_GRAY_IMAGE:
if (uses_original_profile && out_linear)
{
file_format = babl_format ( (bit_depth > 8) ? "Y u16" : "Y u8");
JxlColorEncodingSetToLinearSRGB (&color_profile, JXL_TRUE);
}
else
{
file_format = babl_format ( (bit_depth > 8) ? "Y' u16" : "Y' u8");
JxlColorEncodingSetToSRGB (&color_profile, JXL_TRUE);
}
pixel_format.num_channels = 1;
output_info.num_color_channels = 1;
output_info.alpha_bits = 0;
uses_original_profile = FALSE;
break;
case GIMP_RGBA_IMAGE:
if (bit_depth > 8)
{
file_format = babl_format_with_space (out_linear ? "RGBA u16" : "R'G'B'A u16", space);
output_info.alpha_bits = 16;
}
else
{
file_format = babl_format_with_space (out_linear ? "RGBA u8" : "R'G'B'A u8", space);
output_info.alpha_bits = 8;
}
pixel_format.num_channels = 4;
JxlColorEncodingSetToSRGB (&color_profile, JXL_FALSE);
output_info.num_color_channels = 3;
output_info.alpha_exponent_bits = 0;
output_info.num_extra_channels = 1;
break;
case GIMP_RGB_IMAGE:
if (bit_depth > 8)
{
file_format = babl_format_with_space (out_linear ? "RGB u16" : "R'G'B' u16", space);
}
else
{
file_format = babl_format_with_space (out_linear ? "RGB u8" : "R'G'B' u8", space);
}
pixel_format.num_channels = 3;
JxlColorEncodingSetToSRGB (&color_profile, JXL_FALSE);
output_info.num_color_channels = 3;
output_info.alpha_bits = 0;
break;
default:
if (profile)
{
g_object_unref (profile);
}
return FALSE;
break;
2021-08-26 20:24:21 +08:00
}
}
if (bit_depth > 8)
{
buffer_size = 2 * pixel_format.num_channels * (size_t) output_info.xsize * (size_t) output_info.ysize;
total_channel_size =
2 * output_info.num_extra_channels * (size_t) output_info.xsize * (size_t) output_info.ysize;
}
else
{
buffer_size = pixel_format.num_channels * (size_t) output_info.xsize * (size_t) output_info.ysize;
total_channel_size =
output_info.num_extra_channels * (size_t) output_info.xsize * (size_t) output_info.ysize;
}
/* The maximum number of channels for JPEG XL is 3.
The image may have more (e.g. CMYKA), so we'll combine them
for the initial GeglBuffer loading */
total_channel_size += buffer_size;
picture_buffer = g_malloc (total_channel_size);
2021-08-26 20:24:21 +08:00
gimp_progress_update (0.3);
buffer = gimp_drawable_get_buffer (drawable);
gegl_buffer_get (buffer, GEGL_RECTANGLE (0, 0,
drawable_width, drawable_height), 1.0,
file_format, picture_buffer,
GEGL_AUTO_ROWSTRIDE, GEGL_ABYSS_NONE);
/* Copying K value to buffer */
if (cmyk)
{
channel_buffer_size = (size_t) output_info.xsize * (size_t) output_info.ysize;
if (bit_depth > 8)
channel_buffer_size *= 2;
cmy_buffer = g_malloc (buffer_size);
key_buffer = g_malloc (channel_buffer_size);
if (gimp_drawable_has_alpha (drawable))
alpha_buffer = g_malloc (channel_buffer_size);
extract_cmyk (buffer, &cmy_buffer, &key_buffer, &alpha_buffer,
type, space, output_info.xsize, output_info.ysize,
bit_depth, gimp_drawable_has_alpha (drawable));
}
2021-08-26 20:24:21 +08:00
g_object_unref (buffer);
gimp_progress_update (0.4);
encoder = JxlEncoderCreate (NULL);
if (!encoder)
{
g_set_error (error, G_FILE_ERROR, 0,
"Failed to create Jxl encoder");
g_free (picture_buffer);
if (profile)
{
g_object_unref (profile);
}
return FALSE;
}
if ( (output_info.bits_per_sample > 12 && (output_info.uses_original_profile || output_info.alpha_bits > 12)) || (metadata && (save_exif || save_xmp)))
{
output_info.have_container = JXL_TRUE;
JxlEncoderUseContainer (encoder, JXL_TRUE);
if (output_info.bits_per_sample > 12 && (output_info.uses_original_profile || output_info.alpha_bits > 12))
{
JxlEncoderSetCodestreamLevel (encoder, 10);
}
if (metadata && (save_exif || save_xmp))
{
JxlEncoderUseBoxes (encoder);
}
}
2021-08-26 20:24:21 +08:00
runner = JxlThreadParallelRunnerCreate (NULL, gimp_get_num_processors ());
if (JxlEncoderSetParallelRunner (encoder, JxlThreadParallelRunner, runner) != JXL_ENC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderSetParallelRunner failed");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
if (profile)
{
g_object_unref (profile);
}
return FALSE;
}
status = JxlEncoderSetBasicInfo (encoder, &output_info);
if (status != JXL_ENC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderSetBasicInfo failed!");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
if (profile)
{
g_object_unref (profile);
}
return FALSE;
}
if (uses_original_profile)
{
const uint8_t *icc_data = NULL;
size_t icc_length = 0;
icc_data = gimp_color_profile_get_icc_profile (profile, &icc_length);
status = JxlEncoderSetICCProfile (encoder, icc_data, icc_length);
g_object_unref (profile);
profile = NULL;
if (status != JXL_ENC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderSetICCProfile failed!");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
2021-10-01 18:04:52 +08:00
return FALSE;
2021-08-26 20:24:21 +08:00
}
}
else
{
if (profile)
{
g_object_unref (profile);
profile = NULL;
}
status = JxlEncoderSetColorEncoding (encoder, &color_profile);
if (status != JXL_ENC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderSetColorEncoding failed!");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
return FALSE;
}
}
encoder_options = JxlEncoderFrameSettingsCreate (encoder, NULL);
2021-08-26 20:24:21 +08:00
if (lossless)
{
JxlEncoderSetFrameDistance (encoder_options, 0);
JxlEncoderSetFrameLossless (encoder_options, JXL_TRUE);
2021-08-26 20:24:21 +08:00
}
else
{
JxlEncoderSetFrameDistance (encoder_options, compression);
JxlEncoderSetFrameLossless (encoder_options, JXL_FALSE);
2021-08-26 20:24:21 +08:00
}
status = JxlEncoderFrameSettingsSetOption (encoder_options, JXL_ENC_FRAME_SETTING_EFFORT, speed);
2021-08-26 20:24:21 +08:00
if (status != JXL_ENC_SUCCESS)
{
g_printerr ("JxlEncoderFrameSettingsSetOption failed to set effort %d", speed);
2021-08-26 20:24:21 +08:00
}
gimp_progress_update (0.5);
status = JxlEncoderAddImageFrame (encoder_options, &pixel_format,
(cmyk) ? cmy_buffer : picture_buffer,
buffer_size);
2021-08-26 20:24:21 +08:00
if (status != JXL_ENC_SUCCESS)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderAddImageFrame failed!");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
return FALSE;
}
if (cmyk)
{
JxlExtraChannelInfo extra;
JxlExtraChannelInfo extra_alpha;
JxlEncoderInitExtraChannelInfo (JXL_CHANNEL_BLACK, &extra);
extra.bits_per_sample = output_info.bits_per_sample;
extra.exponent_bits_per_sample = output_info.exponent_bits_per_sample;
/* Key Channel */
status = JxlEncoderSetExtraChannelInfo (encoder, 0, &extra);
if (status != JXL_ENC_SUCCESS)
g_printerr ("JxlEncoderSetExtraChannelInfo failed");
status = JxlEncoderSetExtraChannelBuffer (encoder_options, &pixel_format,
key_buffer, channel_buffer_size,
0);
if (status != JXL_ENC_SUCCESS)
g_printerr ("JxlEncoderSetExtraChannelBuffer failed");
if (alpha_buffer)
{
/* Alpha Channel */
JxlEncoderInitExtraChannelInfo (JXL_CHANNEL_ALPHA, &extra_alpha);
extra_alpha.bits_per_sample = output_info.bits_per_sample;
extra_alpha.exponent_bits_per_sample = output_info.exponent_bits_per_sample;
status = JxlEncoderSetExtraChannelInfo (encoder, 1, &extra_alpha);
if (status != JXL_ENC_SUCCESS)
g_printerr ("JxlEncoderSetExtraChannelInfo failed");
status = JxlEncoderSetExtraChannelBuffer (encoder_options,
&pixel_format,
alpha_buffer,
channel_buffer_size, 1);
if (status != JXL_ENC_SUCCESS)
g_printerr ("JxlEncoderSetExtraChannelBuffer failed");
g_free (alpha_buffer);
}
g_free (key_buffer);
}
gimp_progress_update (0.65);
if (metadata && (save_exif || save_xmp))
{
GimpMetadata *filtered_metadata;
GimpMetadataSaveFlags metadata_flags = 0;
if (save_exif)
{
metadata_flags |= GIMP_METADATA_SAVE_EXIF;
}
if (save_xmp)
{
metadata_flags |= GIMP_METADATA_SAVE_XMP;
}
filtered_metadata = gimp_image_metadata_save_filter (image, "image/jxl", metadata, metadata_flags, NULL, error);
if (! filtered_metadata)
{
if (error && *error)
{
g_printerr ("%s: error filtering metadata: %s",
G_STRFUNC, (*error)->message);
g_clear_error (error);
}
}
else
{
GExiv2Metadata *filtered_g2metadata = GEXIV2_METADATA (filtered_metadata);
/* EXIF metadata */
if (save_exif && gexiv2_metadata_has_exif (filtered_g2metadata))
{
GBytes *raw_exif_data;
raw_exif_data = gexiv2_metadata_get_exif_data (filtered_g2metadata, GEXIV2_BYTE_ORDER_LITTLE, error);
if (raw_exif_data)
{
gsize exif_size = 0;
gconstpointer exif_buffer = g_bytes_get_data (raw_exif_data, &exif_size);
if (exif_size >= 4)
{
const JxlBoxType exif_box_type = { 'E', 'x', 'i', 'f' };
uint8_t *content = g_new (uint8_t, exif_size + 4);
content[0] = 0;
content[1] = 0;
content[2] = 0;
content[3] = 0;
memcpy (content + 4, exif_buffer, exif_size);
if (JxlEncoderAddBox (encoder, exif_box_type, content, exif_size + 4, JXL_FALSE) != JXL_ENC_SUCCESS)
{
g_printerr ("%s: Failed to save EXIF metadata.\n", G_STRFUNC);
}
g_free (content);
}
g_bytes_unref (raw_exif_data);
}
else
{
if (error && *error)
{
g_printerr ("%s: error preparing EXIF metadata: %s",
G_STRFUNC, (*error)->message);
g_clear_error (error);
}
}
}
/* XMP metadata */
if (save_xmp && gexiv2_metadata_has_xmp (filtered_g2metadata))
{
gchar *xmp_packet;
xmp_packet = gexiv2_metadata_try_generate_xmp_packet (filtered_g2metadata, GEXIV2_USE_COMPACT_FORMAT | GEXIV2_OMIT_ALL_FORMATTING, 0, NULL);
if (xmp_packet)
{
int xmp_size = strlen (xmp_packet);
if (xmp_size > 0)
{
const JxlBoxType xml_box_type = { 'x', 'm', 'l', ' ' };
if (JxlEncoderAddBox (encoder, xml_box_type, (const uint8_t *) xmp_packet, xmp_size, JXL_FALSE) != JXL_ENC_SUCCESS)
{
g_printerr ("%s: Failed to save XMP metadata.\n", G_STRFUNC);
}
}
g_free (xmp_packet);
}
}
g_object_unref (filtered_metadata);
}
}
2021-08-26 20:24:21 +08:00
JxlEncoderCloseInput (encoder);
gimp_progress_update (0.7);
compressed = g_byte_array_sized_new (4096);
g_byte_array_set_size (compressed, 4096);
do
{
next_out = compressed->data + offset;
avail_out = compressed->len - offset;
status = JxlEncoderProcessOutput (encoder, &next_out, &avail_out);
if (status == JXL_ENC_NEED_MORE_OUTPUT)
{
offset = next_out - compressed->data;
g_byte_array_set_size (compressed, compressed->len * 2);
}
else if (status == JXL_ENC_ERROR)
{
g_set_error (error, G_FILE_ERROR, 0,
"JxlEncoderProcessOutput failed!");
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
return FALSE;
}
}
while (status != JXL_ENC_SUCCESS);
JxlThreadParallelRunnerDestroy (runner);
JxlEncoderDestroy (encoder);
g_free (picture_buffer);
g_byte_array_set_size (compressed, next_out - compressed->data);
gimp_progress_update (0.8);
if (compressed->len > 0)
{
outfile = g_fopen (g_file_peek_path (file), "wb");
2021-08-26 20:24:21 +08:00
if (!outfile)
{
g_set_error (error, G_FILE_ERROR, 0,
"Could not open '%s' for writing!\n",
g_file_peek_path (file));
2021-08-26 20:24:21 +08:00
g_byte_array_free (compressed, TRUE);
return FALSE;
}
fwrite (compressed->data, 1, compressed->len, outfile);
fclose (outfile);
gimp_progress_update (1.0);
g_byte_array_free (compressed, TRUE);
return TRUE;
}
g_set_error (error, G_FILE_ERROR, 0,
"No data to write");
g_byte_array_free (compressed, TRUE);
return FALSE;
}
static gboolean
save_dialog (GimpImage *image,
GimpProcedure *procedure,
GObject *config)
{
GtkWidget *dialog;
GtkListStore *store;
GtkWidget *compression_scale;
GtkWidget *orig_profile_check;
GtkWidget *profile_label;
GimpColorProfile *cmyk_profile = NULL;
gboolean run;
2021-08-26 20:24:21 +08:00
dialog = gimp_export_procedure_dialog_new (GIMP_EXPORT_PROCEDURE (procedure),
GIMP_PROCEDURE_CONFIG (config),
image);
2021-08-26 20:24:21 +08:00
gimp_procedure_dialog_get_widget (GIMP_PROCEDURE_DIALOG (dialog),
"lossless", GTK_TYPE_CHECK_BUTTON);
compression_scale = gimp_procedure_dialog_get_widget (GIMP_PROCEDURE_DIALOG (dialog),
"compression",
GIMP_TYPE_SCALE_ENTRY);
g_object_bind_property (config, "lossless",
compression_scale, "sensitive",
G_BINDING_SYNC_CREATE |
G_BINDING_INVERT_BOOLEAN);
store = gimp_int_store_new (_("lightning (fastest)"), 1,
_("thunder"), 2,
_("falcon (faster)"), 3,
_("cheetah"), 4,
_("hare"), 5,
_("wombat"), 6,
_("squirrel"), 7,
_("kitten"), 8,
_("tortoise (slower)"), 9,
2021-08-26 20:24:21 +08:00
NULL);
gimp_procedure_dialog_get_int_combo (GIMP_PROCEDURE_DIALOG (dialog),
"speed", GIMP_INT_STORE (store));
store = gimp_int_store_new (_("8 bit/channel"), 8,
_("16 bit/channel"), 16,
NULL);
gimp_procedure_dialog_get_int_combo (GIMP_PROCEDURE_DIALOG (dialog),
"save-bit-depth", GIMP_INT_STORE (store));
/* Profile label */
profile_label = gimp_procedure_dialog_get_label (GIMP_PROCEDURE_DIALOG (dialog),
"profile-label", _("CMYK profile required for export"),
FALSE, FALSE);
gtk_label_set_xalign (GTK_LABEL (profile_label), 0.0);
gtk_label_set_ellipsize (GTK_LABEL (profile_label), PANGO_ELLIPSIZE_END);
gimp_label_set_attributes (GTK_LABEL (profile_label),
PANGO_ATTR_STYLE, PANGO_STYLE_ITALIC,
-1);
gimp_help_set_help_data (profile_label,
_("Name of the color profile used for CMYK export."), NULL);
gimp_procedure_dialog_fill_frame (GIMP_PROCEDURE_DIALOG (dialog),
"cmyk-frame", "cmyk", FALSE,
"profile-label");
cmyk_profile = gimp_image_get_simulation_profile (image);
if (! cmyk_profile)
{
g_object_set (config,
"cmyk", FALSE,
NULL);
}
if (cmyk_profile)
{
if (gimp_color_profile_is_cmyk (cmyk_profile))
{
gchar *label_text;
label_text = g_strdup_printf (_("Profile: %s"),
gimp_color_profile_get_label (cmyk_profile));
gtk_label_set_text (GTK_LABEL (profile_label), label_text);
gimp_label_set_attributes (GTK_LABEL (profile_label),
PANGO_ATTR_STYLE, PANGO_STYLE_NORMAL,
-1);
g_free (label_text);
}
g_object_unref (cmyk_profile);
}
/* JPEG XL requires a CMYK profile if exporting as CMYK */
gimp_procedure_dialog_set_sensitive (GIMP_PROCEDURE_DIALOG (dialog),
"cmyk",
cmyk_profile != NULL,
NULL, NULL, FALSE);
orig_profile_check = gimp_procedure_dialog_get_widget (GIMP_PROCEDURE_DIALOG (dialog),
"uses-original-profile",
GTK_TYPE_CHECK_BUTTON);
g_object_bind_property (config, "lossless",
orig_profile_check, "sensitive",
G_BINDING_SYNC_CREATE |
G_BINDING_INVERT_BOOLEAN);
2021-08-26 20:24:21 +08:00
gimp_procedure_dialog_fill (GIMP_PROCEDURE_DIALOG (dialog),
"lossless", "compression",
"speed", "save-bit-depth",
"cmyk-frame",
"uses-original-profile",
"save-exif", "save-xmp",
2021-08-26 20:24:21 +08:00
NULL);
run = gimp_procedure_dialog_run (GIMP_PROCEDURE_DIALOG (dialog));
gtk_widget_destroy (dialog);
return run;
}
static GimpValueArray *
jpegxl_export (GimpProcedure *procedure,
GimpRunMode run_mode,
GimpImage *image,
GFile *file,
GimpMetadata *metadata,
GimpProcedureConfig *config,
gpointer run_data)
2021-08-26 20:24:21 +08:00
{
GimpPDBStatusType status = GIMP_PDB_SUCCESS;
GimpExportReturn export = GIMP_EXPORT_IGNORE;
GList *drawables;
GError *error = NULL;
2021-08-26 20:24:21 +08:00
gegl_init (NULL, NULL);
if (run_mode == GIMP_RUN_INTERACTIVE)
2021-08-26 20:24:21 +08:00
{
gimp_ui_init (PLUG_IN_BINARY);
if (! save_dialog (image, procedure, G_OBJECT (config)))
{
status = GIMP_PDB_CANCEL;
}
}
export = gimp_export_image (&image,
GIMP_EXPORT_CAN_HANDLE_RGB |
GIMP_EXPORT_CAN_HANDLE_GRAY |
GIMP_EXPORT_CAN_HANDLE_ALPHA);
drawables = gimp_image_list_layers (image);
2021-08-26 20:24:21 +08:00
if (status == GIMP_PDB_SUCCESS)
{
GimpMetadataSaveFlags metadata_flags;
GimpMetadata *metadata = gimp_image_metadata_save_prepare (image, "image/jxl", &metadata_flags);
if (! export_image (file, config, image, drawables->data, metadata, &error))
2021-08-26 20:24:21 +08:00
{
status = GIMP_PDB_EXECUTION_ERROR;
}
if (metadata)
{
g_object_unref (metadata);
}
2021-08-26 20:24:21 +08:00
}
if (export == GIMP_EXPORT_EXPORT)
gimp_image_delete (image);
2021-08-26 20:24:21 +08:00
g_list_free (drawables);
2021-08-26 20:24:21 +08:00
return gimp_procedure_new_return_values (procedure, status, error);
}