aboutsummaryrefslogtreecommitdiff
path: root/fiv-io-cmm.c
diff options
context:
space:
mode:
Diffstat (limited to 'fiv-io-cmm.c')
-rw-r--r--fiv-io-cmm.c467
1 files changed, 467 insertions, 0 deletions
diff --git a/fiv-io-cmm.c b/fiv-io-cmm.c
new file mode 100644
index 0000000..1336a22
--- /dev/null
+++ b/fiv-io-cmm.c
@@ -0,0 +1,467 @@
+//
+// fiv-io-cmm.c: colour management
+//
+// Copyright (c) 2024, Přemysl Eric Janouch <p@janouch.name>
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+// WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+// MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
+// SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+// WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION
+// OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
+// CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+//
+
+#include "config.h"
+
+#include <glib.h>
+#include <stdbool.h>
+
+#include "fiv-io.h"
+
+// Colour management must be handled before RGB conversions.
+// TODO(p): Make it also possible to use Skia's skcms.
+#ifdef HAVE_LCMS2
+#include <lcms2.h>
+#endif // HAVE_LCMS2
+#ifdef HAVE_LCMS2_FAST_FLOAT
+#include <lcms2_fast_float.h>
+#endif // HAVE_LCMS2_FAST_FLOAT
+
+// --- Profiles ----------------------------------------------------------------
+
+struct _FivIoProfile {
+ FivIoCmm *cmm;
+#ifdef HAVE_LCMS2
+ cmsHPROFILE profile;
+#endif
+};
+
+GBytes *
+fiv_io_profile_to_bytes(FivIoProfile *profile)
+{
+#ifdef HAVE_LCMS2
+ cmsUInt32Number len = 0;
+ (void) cmsSaveProfileToMem(profile, NULL, &len);
+ gchar *data = g_malloc0(len);
+ if (!cmsSaveProfileToMem(profile, data, &len)) {
+ g_free(data);
+ return NULL;
+ }
+ return g_bytes_new_take(data, len);
+#else
+ (void) profile;
+ return NULL;
+#endif
+}
+
+#ifdef HAVE_LCMS2
+
+static FivIoProfile *
+fiv_io_profile_new(FivIoCmm *cmm, cmsHPROFILE profile)
+{
+ FivIoProfile *self = g_new0(FivIoProfile, 1);
+ self->cmm = g_object_ref(cmm);
+ self->profile = profile;
+ return self;
+}
+
+#endif
+
+void
+fiv_io_profile_free(FivIoProfile *self)
+{
+#ifdef HAVE_LCMS2
+ cmsCloseProfile(self->profile);
+#endif
+ g_clear_object(&self->cmm);
+ g_free(self);
+}
+
+// --- Contexts ----------------------------------------------------------------
+
+struct _FivIoCmm {
+ GObject parent_instance;
+#ifdef HAVE_LCMS2
+ cmsContext context;
+
+ // https://github.com/mm2/Little-CMS/issues/430
+ gboolean broken_premul;
+#endif
+};
+
+G_DEFINE_TYPE(FivIoCmm, fiv_io_cmm, G_TYPE_OBJECT)
+
+static void
+fiv_io_cmm_finalize(GObject *gobject)
+{
+#ifdef HAVE_LCMS2
+ FivIoCmm *self = FIV_IO_CMM(gobject);
+ cmsDeleteContext(self->context);
+#endif
+
+ G_OBJECT_CLASS(fiv_io_cmm_parent_class)->finalize(gobject);
+}
+
+static void
+fiv_io_cmm_class_init(FivIoCmmClass *klass)
+{
+ GObjectClass *object_class = G_OBJECT_CLASS(klass);
+ object_class->finalize = fiv_io_cmm_finalize;
+}
+
+static void
+fiv_io_cmm_init(FivIoCmm *self)
+{
+ (void) self;
+
+#ifdef HAVE_LCMS2
+ self->context = cmsCreateContext(NULL, self);
+#endif
+#ifdef HAVE_LCMS2_FAST_FLOAT
+ if (cmsPluginTHR(self->context, cmsFastFloatExtensions()))
+ self->broken_premul = LCMS_VERSION <= 2160;
+#endif // HAVE_LCMS2_FAST_FLOAT
+}
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+FivIoCmm *
+fiv_io_cmm_get_default()
+{
+ static gsize initialization_value = 0;
+ static FivIoCmm *default_ = NULL;
+ if (g_once_init_enter(&initialization_value)) {
+ gsize setup_value = 1;
+ default_ = g_object_new(FIV_TYPE_IO_CMM, NULL);
+ g_once_init_leave(&initialization_value, setup_value);
+ }
+ return default_;
+}
+
+FivIoProfile *
+fiv_io_cmm_get_profile(FivIoCmm *self, const void *data, size_t len)
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+#ifdef HAVE_LCMS2
+ return fiv_io_profile_new(self,
+ cmsOpenProfileFromMemTHR(self->context, data, len));
+#else
+ (void) data;
+ (void) len;
+ return NULL;
+#endif
+}
+
+FivIoProfile *
+fiv_io_cmm_get_profile_sRGB(FivIoCmm *self)
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+#ifdef HAVE_LCMS2
+ return fiv_io_profile_new(self,
+ cmsCreate_sRGBProfileTHR(self->context));
+#else
+ return NULL;
+#endif
+}
+
+FivIoProfile *
+fiv_io_cmm_get_profile_parametric(FivIoCmm *self,
+ double gamma, double whitepoint[2], double primaries[6])
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+#ifdef HAVE_LCMS2
+ const cmsCIExyY cmsWP = {whitepoint[0], whitepoint[1], 1.0};
+ const cmsCIExyYTRIPLE cmsP = {
+ {primaries[0], primaries[1], 1.0},
+ {primaries[2], primaries[3], 1.0},
+ {primaries[4], primaries[5], 1.0},
+ };
+
+ cmsToneCurve *curve = cmsBuildGamma(self->context, gamma);
+ if (!curve)
+ return NULL;
+
+ cmsHPROFILE profile = cmsCreateRGBProfileTHR(self->context,
+ &cmsWP, &cmsP, (cmsToneCurve *[3]){curve, curve, curve});
+ cmsFreeToneCurve(curve);
+ return fiv_io_profile_new(self, profile);
+#else
+ (void) gamma;
+ (void) whitepoint;
+ (void) primaries;
+ return NULL;
+#endif
+}
+
+FivIoProfile *
+fiv_io_cmm_get_profile_sRGB_gamma(FivIoCmm *self, double gamma)
+{
+ return fiv_io_cmm_get_profile_parametric(self, gamma,
+ (double[2]){0.3127, 0.3290},
+ (double[6]){0.6400, 0.3300, 0.3000, 0.6000, 0.1500, 0.0600});
+}
+
+FivIoProfile *
+fiv_io_cmm_get_profile_from_bytes(FivIoCmm *self, GBytes *bytes)
+{
+ gsize len = 0;
+ gconstpointer p = g_bytes_get_data(bytes, &len);
+ return fiv_io_cmm_get_profile(self, p, len);
+}
+
+// --- Image loading -----------------------------------------------------------
+
+// TODO(p): In general, try to use CAIRO_FORMAT_RGB30 or CAIRO_FORMAT_RGBA128F.
+#ifndef HAVE_LCMS2
+#define FIV_IO_PROFILE_ARGB32 0
+#define FIV_IO_PROFILE_4X16LE 0
+#else
+#define FIV_IO_PROFILE_ARGB32 \
+ (G_BYTE_ORDER == G_LITTLE_ENDIAN ? TYPE_BGRA_8 : TYPE_ARGB_8)
+#define FIV_IO_PROFILE_4X16LE \
+ (G_BYTE_ORDER == G_LITTLE_ENDIAN ? TYPE_BGRA_16 : TYPE_BGRA_16_SE)
+#endif
+
+// CAIRO_STRIDE_ALIGNMENT is 4 bytes, so there will be no padding with
+// ARGB/BGRA/XRGB/BGRX.
+static void
+trivial_cmyk_to_host_byte_order_argb(unsigned char *p, int len)
+{
+ // This CMYK handling has been seen in gdk-pixbuf/JPEG, GIMP/JPEG, skcms.
+ // It will typically produce horribly oversaturated results.
+ // Assume that all YCCK/CMYK JPEG files use inverted CMYK, as Photoshop
+ // does, see https://bugzilla.gnome.org/show_bug.cgi?id=618096
+ while (len--) {
+ int c = p[0], m = p[1], y = p[2], k = p[3];
+#if G_BYTE_ORDER == G_LITTLE_ENDIAN
+ p[0] = k * y / 255;
+ p[1] = k * m / 255;
+ p[2] = k * c / 255;
+ p[3] = 255;
+#else
+ p[3] = k * y / 255;
+ p[2] = k * m / 255;
+ p[1] = k * c / 255;
+ p[0] = 255;
+#endif
+ p += 4;
+ }
+}
+
+void
+fiv_io_cmm_cmyk(FivIoCmm *self,
+ FivIoImage *image, FivIoProfile *source, FivIoProfile *target)
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+#ifndef HAVE_LCMS2
+ (void) self;
+ (void) source;
+ (void) target;
+#else
+ cmsHTRANSFORM transform = NULL;
+ if (source && target) {
+ transform = cmsCreateTransformTHR(self->context,
+ source->profile, TYPE_CMYK_8_REV,
+ target->profile, FIV_IO_PROFILE_ARGB32, INTENT_PERCEPTUAL, 0);
+ }
+ if (transform) {
+ cmsDoTransform(
+ transform, image->data, image->data, image->width * image->height);
+ cmsDeleteTransform(transform);
+ return;
+ }
+#endif
+ trivial_cmyk_to_host_byte_order_argb(
+ image->data, image->width * image->height);
+}
+
+static bool
+fiv_io_cmm_rgb_direct(FivIoCmm *self, unsigned char *data, int w, int h,
+ FivIoProfile *source, FivIoProfile *target,
+ uint32_t source_format, uint32_t target_format)
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+#ifndef HAVE_LCMS2
+ (void) self;
+ (void) data;
+ (void) w;
+ (void) h;
+ (void) source;
+ (void) source_format;
+ (void) target;
+ (void) target_format;
+ return false;
+#else
+ // TODO(p): We should make this optional.
+ FivIoProfile *src_fallback = NULL;
+ if (target && !source)
+ source = src_fallback = fiv_io_cmm_get_profile_sRGB(self);
+
+ cmsHTRANSFORM transform = NULL;
+ if (source && target) {
+ transform = cmsCreateTransformTHR(self->context,
+ source->profile, source_format,
+ target->profile, target_format, INTENT_PERCEPTUAL, 0);
+ }
+ if (transform) {
+ cmsDoTransform(transform, data, data, w * h);
+ cmsDeleteTransform(transform);
+ }
+ if (src_fallback)
+ fiv_io_profile_free(src_fallback);
+ return transform != NULL;
+#endif
+}
+
+static void
+fiv_io_cmm_xrgb32(FivIoCmm *self,
+ FivIoImage *image, FivIoProfile *source, FivIoProfile *target)
+{
+ fiv_io_cmm_rgb_direct(self, image->data, image->width, image->height,
+ source, target, FIV_IO_PROFILE_ARGB32, FIV_IO_PROFILE_ARGB32);
+}
+
+void
+fiv_io_cmm_4x16le_direct(FivIoCmm *self, unsigned char *data,
+ int w, int h, FivIoProfile *source, FivIoProfile *target)
+{
+ fiv_io_cmm_rgb_direct(self, data, w, h, source, target,
+ FIV_IO_PROFILE_4X16LE, FIV_IO_PROFILE_4X16LE);
+}
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+void
+fiv_io_cmm_page(FivIoCmm *self, FivIoImage *page, FivIoProfile *target,
+ void (*frame_cb) (FivIoCmm *, FivIoImage *, FivIoProfile *, FivIoProfile *))
+{
+ FivIoProfile *source = NULL;
+ if (page->icc)
+ source = fiv_io_cmm_get_profile_from_bytes(self, page->icc);
+
+ // TODO(p): All animations need to be composited in a linear colour space.
+ for (FivIoImage *frame = page; frame != NULL; frame = frame->frame_next)
+ frame_cb(self, frame, source, target);
+
+ if (source)
+ fiv_io_profile_free(source);
+}
+
+// From libwebp, verified to exactly match [x * a / 255].
+#define PREMULTIPLY8(a, x) (((uint32_t) (x) * (uint32_t) (a) * 32897U) >> 23)
+
+void
+fiv_io_premultiply_argb32(FivIoImage *image)
+{
+ if (image->format != CAIRO_FORMAT_ARGB32)
+ return;
+
+ for (uint32_t y = 0; y < image->height; y++) {
+ uint32_t *dstp = (uint32_t *) (image->data + image->stride * y);
+ for (uint32_t x = 0; x < image->width; x++) {
+ uint32_t argb = dstp[x], a = argb >> 24;
+ dstp[x] = a << 24 |
+ PREMULTIPLY8(a, 0xFF & (argb >> 16)) << 16 |
+ PREMULTIPLY8(a, 0xFF & (argb >> 8)) << 8 |
+ PREMULTIPLY8(a, 0xFF & argb);
+ }
+ }
+}
+
+#if defined HAVE_LCMS2 && LCMS_VERSION >= 2130
+
+#define FIV_IO_PROFILE_ARGB32_PREMUL \
+ (G_BYTE_ORDER == G_LITTLE_ENDIAN ? TYPE_BGRA_8_PREMUL : TYPE_ARGB_8_PREMUL)
+
+static void
+fiv_io_cmm_argb32(FivIoCmm *self, FivIoImage *image,
+ FivIoProfile *source, FivIoProfile *target)
+{
+ g_return_if_fail(image->format == CAIRO_FORMAT_ARGB32);
+
+ // TODO: With self->broken_premul,
+ // this probably also needs to be wrapped in un-premultiplication.
+ fiv_io_cmm_rgb_direct(self, image->data, image->width, image->height,
+ source, target,
+ FIV_IO_PROFILE_ARGB32_PREMUL, FIV_IO_PROFILE_ARGB32_PREMUL);
+}
+
+void
+fiv_io_cmm_argb32_premultiply(FivIoCmm *self,
+ FivIoImage *image, FivIoProfile *source, FivIoProfile *target)
+{
+ if (!self)
+ self = fiv_io_cmm_get_default();
+
+ if (image->format != CAIRO_FORMAT_ARGB32) {
+ fiv_io_cmm_xrgb32(self, image, source, target);
+ } else if (self->broken_premul) {
+ fiv_io_cmm_xrgb32(self, image, source, target);
+ fiv_io_premultiply_argb32(image);
+ } else if (!fiv_io_cmm_rgb_direct(self, image->data,
+ image->width, image->height, source, target,
+ FIV_IO_PROFILE_ARGB32, FIV_IO_PROFILE_ARGB32_PREMUL)) {
+ g_debug("failed to create a premultiplying transform");
+ fiv_io_premultiply_argb32(image);
+ }
+}
+
+#else // ! HAVE_LCMS2 || LCMS_VERSION < 2130
+
+// TODO(p): Unpremultiply, transform, repremultiply. Or require lcms2>=2.13.
+#define fiv_io_cmm_argb32(self, image, source, target)
+
+void
+fiv_io_cmm_argb32_premultiply(FivIoCmm *self,
+ FivIoImage *image, FivIoProfile *source, FivIoProfile *target)
+{
+ fiv_io_cmm_xrgb32(self, image, source, target);
+ fiv_io_premultiply_argb32(image);
+}
+
+#endif // ! HAVE_LCMS2 || LCMS_VERSION < 2130
+
+// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
+
+void
+fiv_io_cmm_any(FivIoCmm *self,
+ FivIoImage *image, FivIoProfile *source, FivIoProfile *target)
+{
+ // TODO(p): Ensure we do colour management early enough, so that
+ // no avoidable increase of quantization error occurs beforehands,
+ // and also for correct alpha compositing.
+ switch (image->format) {
+ break; case CAIRO_FORMAT_RGB24:
+ fiv_io_cmm_xrgb32(self, image, source, target);
+ break; case CAIRO_FORMAT_ARGB32:
+ fiv_io_cmm_argb32(self, image, source, target);
+ break; default:
+ g_debug("CM attempted on an unsupported surface format");
+ }
+}
+
+// TODO(p): Offer better integration, upgrade the bit depth if appropriate.
+FivIoImage *
+fiv_io_cmm_finish(FivIoCmm *self, FivIoImage *image, FivIoProfile *target)
+{
+ if (!target)
+ return image;
+
+ for (FivIoImage *page = image; page != NULL; page = page->page_next)
+ fiv_io_cmm_page(self, page, target, fiv_io_cmm_any);
+ return image;
+}