diff mupdf-source/source/fitz/output-jpeg.c @ 2:b50eed0cc0ef upstream

ADD: MuPDF v1.26.7: the MuPDF source as downloaded by a default build of PyMuPDF 1.26.4. The directory name has changed: no version number in the expanded directory now.
author Franz Glasner <fzglas.hg@dom66.de>
date Mon, 15 Sep 2025 11:43:07 +0200
parents
children
line wrap: on
line diff
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/mupdf-source/source/fitz/output-jpeg.c	Mon Sep 15 11:43:07 2025 +0200
@@ -0,0 +1,316 @@
+// Copyright (C) 2004-2023 Artifex Software, Inc.
+//
+// This file is part of MuPDF.
+//
+// MuPDF is free software: you can redistribute it and/or modify it under the
+// terms of the GNU Affero General Public License as published by the Free
+// Software Foundation, either version 3 of the License, or (at your option)
+// any later version.
+//
+// MuPDF 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 Affero General Public License for more
+// details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with MuPDF. If not, see <https://www.gnu.org/licenses/agpl-3.0.en.html>
+//
+// Alternative licensing terms are available from the licensor.
+// For commercial licensing, see <https://www.artifex.com/> or contact
+// Artifex Software, Inc., 39 Mesa Street, Suite 108A, San Francisco,
+// CA 94129, USA, for further information.
+
+#include "mupdf/fitz.h"
+
+#include <jpeglib.h>
+
+#ifdef SHARE_JPEG
+
+#define JZ_CTX_FROM_CINFO(c) (fz_context *)((c)->client_data)
+
+static void fz_jpg_mem_init(j_common_ptr cinfo, fz_context *ctx)
+{
+	cinfo->client_data = ctx;
+}
+
+#define fz_jpg_mem_term(cinfo)
+
+#else /* SHARE_JPEG */
+
+typedef void * backing_store_ptr;
+
+#include "jmemcust.h"
+
+#define JZ_CTX_FROM_CINFO(c) (fz_context *)(GET_CUST_MEM_DATA(c)->priv)
+
+static void *
+fz_jpg_mem_alloc(j_common_ptr cinfo, size_t size)
+{
+	fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+	return fz_malloc_no_throw(ctx, size);
+}
+
+static void
+fz_jpg_mem_free(j_common_ptr cinfo, void *object, size_t size)
+{
+	fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+	fz_free(ctx, object);
+}
+
+static void
+fz_jpg_mem_init(j_common_ptr cinfo, fz_context *ctx)
+{
+	jpeg_cust_mem_data *custmptr;
+	custmptr = fz_malloc_struct(ctx, jpeg_cust_mem_data);
+	if (!jpeg_cust_mem_init(custmptr, (void *) ctx, NULL, NULL, NULL,
+				fz_jpg_mem_alloc, fz_jpg_mem_free,
+				fz_jpg_mem_alloc, fz_jpg_mem_free, NULL))
+	{
+		fz_free(ctx, custmptr);
+		fz_throw(ctx, FZ_ERROR_LIBRARY, "cannot initialize custom JPEG memory handler");
+	}
+	cinfo->client_data = custmptr;
+}
+
+static void
+fz_jpg_mem_term(j_common_ptr cinfo)
+{
+	if (cinfo->client_data)
+	{
+		fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+		fz_free(ctx, cinfo->client_data);
+		cinfo->client_data = NULL;
+	}
+}
+
+#endif
+
+#define OUTPUT_BUF_SIZE (16<<10)
+
+typedef struct {
+	struct jpeg_destination_mgr pub;
+	fz_output *out;
+	JOCTET buffer[OUTPUT_BUF_SIZE];
+} my_destination_mgr;
+
+typedef my_destination_mgr * my_dest_ptr;
+
+static void error_exit(j_common_ptr cinfo)
+{
+	char msg[JMSG_LENGTH_MAX];
+	fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+	cinfo->err->format_message(cinfo, msg);
+	fz_throw(ctx, FZ_ERROR_LIBRARY, "jpeg error: %s", msg);
+}
+
+static void init_destination(j_compress_ptr cinfo)
+{
+	my_dest_ptr dest = (my_dest_ptr) cinfo->dest;
+	dest->pub.next_output_byte = dest->buffer;
+	dest->pub.free_in_buffer = OUTPUT_BUF_SIZE;
+}
+
+static boolean empty_output_buffer(j_compress_ptr cinfo)
+{
+	fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+	my_dest_ptr dest = (my_dest_ptr) cinfo->dest;
+	fz_output *out = dest->out;
+
+	fz_write_data(ctx, out, dest->buffer, OUTPUT_BUF_SIZE);
+
+	dest->pub.next_output_byte = dest->buffer;
+	dest->pub.free_in_buffer = OUTPUT_BUF_SIZE;
+
+	return TRUE;
+}
+
+static void term_destination(j_compress_ptr cinfo)
+{
+	fz_context *ctx = JZ_CTX_FROM_CINFO(cinfo);
+	my_dest_ptr dest = (my_dest_ptr) cinfo->dest;
+	fz_output *out = dest->out;
+	size_t datacount = OUTPUT_BUF_SIZE - dest->pub.free_in_buffer;
+
+	fz_write_data(ctx, out, dest->buffer, datacount);
+}
+
+void
+fz_write_pixmap_as_jpeg(fz_context *ctx, fz_output *out, fz_pixmap *pix, int quality, int invert_cmyk)
+{
+	struct jpeg_compress_struct cinfo;
+	struct jpeg_error_mgr err;
+	my_destination_mgr dest;
+	JSAMPROW row_pointer[1];
+	unsigned char *outbuffer = NULL;
+	size_t outsize = 0;
+	fz_colorspace *cs = pix->colorspace;
+	int n = pix->n;
+	int alpha = pix->alpha;
+
+	if (pix->s > 0)
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "pixmap may not have separations to save as JPEG");
+	if (cs && !fz_colorspace_is_gray(ctx, cs) && !fz_colorspace_is_rgb(ctx, cs) && !fz_colorspace_is_cmyk(ctx, cs))
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "pixmap must be Grayscale, RGB, or CMYK to save as JPEG");
+
+	/* Treat alpha only as greyscale */
+	if (n == 1 && alpha)
+		alpha = 0;
+	n -= alpha;
+
+	if (alpha > 0)
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "pixmap may not have alpha to save as JPEG");
+
+	cinfo.mem = NULL;
+	cinfo.global_state = 0;
+	cinfo.err = jpeg_std_error(&err);
+	err.error_exit = error_exit;
+
+	cinfo.client_data = NULL;
+	fz_jpg_mem_init((j_common_ptr)&cinfo, ctx);
+
+	fz_try(ctx)
+	{
+		jpeg_create_compress(&cinfo);
+
+		cinfo.dest = (void*) &dest;
+		dest.pub.init_destination = init_destination;
+		dest.pub.empty_output_buffer = empty_output_buffer;
+		dest.pub.term_destination = term_destination;
+		dest.out = out;
+
+		cinfo.image_width = pix->w;
+		cinfo.image_height = pix->h;
+		cinfo.input_components = n;
+		switch (n) {
+		case 1:
+			cinfo.in_color_space = JCS_GRAYSCALE;
+			break;
+		case 3:
+			cinfo.in_color_space = JCS_RGB;
+			break;
+		case 4:
+			cinfo.in_color_space = JCS_CMYK;
+			break;
+		}
+
+		jpeg_set_defaults(&cinfo);
+		jpeg_set_quality(&cinfo, quality, FALSE);
+
+		/* Write image resolution */
+		cinfo.density_unit = 1; /* dots/inch */
+		cinfo.X_density = pix->xres;
+		cinfo.Y_density = pix->yres;
+
+		/* Disable chroma subsampling */
+		cinfo.comp_info[0].h_samp_factor = 1;
+		cinfo.comp_info[0].v_samp_factor = 1;
+
+		/* Progressive JPEGs are smaller */
+		jpeg_simple_progression(&cinfo);
+
+		jpeg_start_compress(&cinfo, TRUE);
+
+		if (fz_colorspace_is_cmyk(ctx, pix->colorspace) && invert_cmyk)
+			fz_invert_pixmap_raw(ctx, pix);
+
+		while (cinfo.next_scanline < cinfo.image_height) {
+			row_pointer[0] = &pix->samples[cinfo.next_scanline * pix->stride];
+			(void) jpeg_write_scanlines(&cinfo, row_pointer, 1);
+		}
+
+		if (fz_colorspace_is_cmyk(ctx, pix->colorspace) && invert_cmyk)
+			fz_invert_pixmap_raw(ctx, pix);
+
+		jpeg_finish_compress(&cinfo);
+
+		fz_write_data(ctx, out, outbuffer, outsize);
+	}
+	fz_always(ctx)
+	{
+		jpeg_destroy_compress(&cinfo);
+		fz_jpg_mem_term((j_common_ptr)&cinfo);
+		fz_free(ctx, outbuffer);
+	}
+	fz_catch(ctx)
+	{
+		fz_rethrow(ctx);
+	}
+}
+
+void
+fz_save_pixmap_as_jpeg(fz_context *ctx, fz_pixmap *pixmap, const char *filename, int quality)
+{
+	fz_output *out = fz_new_output_with_path(ctx, filename, 0);
+	fz_try(ctx)
+	{
+		fz_write_pixmap_as_jpeg(ctx, out, pixmap, quality, 1);
+		fz_close_output(ctx, out);
+	}
+	fz_always(ctx)
+	{
+		fz_drop_output(ctx, out);
+	}
+	fz_catch(ctx)
+	{
+		fz_rethrow(ctx);
+	}
+}
+
+static fz_buffer *
+jpeg_from_pixmap(fz_context *ctx, fz_pixmap *pix, fz_color_params color_params, int quality, int invert_cmyk, int drop)
+{
+	fz_buffer *buf = NULL;
+	fz_output *out = NULL;
+	fz_pixmap *pix2 = NULL;
+
+	fz_var(buf);
+	fz_var(out);
+	fz_var(pix2);
+
+	if (pix->w == 0 || pix->h == 0)
+	{
+		if (drop)
+			fz_drop_pixmap(ctx, pix);
+		return NULL;
+	}
+
+	fz_try(ctx)
+	{
+		if (pix->colorspace && pix->colorspace != fz_device_gray(ctx) && pix->colorspace != fz_device_rgb(ctx) && pix->colorspace != fz_device_cmyk(ctx))
+		{
+			pix2 = fz_convert_pixmap(ctx, pix, fz_device_rgb(ctx), NULL, NULL, color_params, 1);
+			if (drop)
+				fz_drop_pixmap(ctx, pix);
+			pix = pix2;
+		}
+		buf = fz_new_buffer(ctx, 1024);
+		out = fz_new_output_with_buffer(ctx, buf);
+		fz_write_pixmap_as_jpeg(ctx, out, pix, quality, invert_cmyk);
+		fz_close_output(ctx, out);
+	}
+	fz_always(ctx)
+	{
+		if (drop)
+			fz_drop_pixmap(ctx, pix);
+		fz_drop_output(ctx, out);
+	}
+	fz_catch(ctx)
+	{
+		fz_drop_buffer(ctx, buf);
+		fz_rethrow(ctx);
+	}
+	return buf;
+}
+
+fz_buffer *
+fz_new_buffer_from_image_as_jpeg(fz_context *ctx, fz_image *image, fz_color_params color_params, int quality, int invert_cmyk)
+{
+	fz_pixmap *pix = fz_get_pixmap_from_image(ctx, image, NULL, NULL, NULL, NULL);
+	return jpeg_from_pixmap(ctx, pix, color_params, quality, 1, invert_cmyk);
+}
+
+fz_buffer *
+fz_new_buffer_from_pixmap_as_jpeg(fz_context *ctx, fz_pixmap *pix, fz_color_params color_params, int quality, int invert_cmyk)
+{
+	return jpeg_from_pixmap(ctx, pix, color_params, quality, 0, invert_cmyk);
+}