diff mupdf-source/source/fitz/output-ps.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-ps.c	Mon Sep 15 11:43:07 2025 +0200
@@ -0,0 +1,411 @@
+// Copyright (C) 2004-2021 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 "z-imp.h"
+
+#include <limits.h>
+
+typedef struct ps_band_writer_s
+{
+	fz_band_writer super;
+	z_stream stream;
+	int stream_started;
+	int stream_ended;
+	size_t input_size;
+	unsigned char *input;
+	size_t output_size;
+	unsigned char *output;
+} ps_band_writer;
+
+void
+fz_write_ps_file_header(fz_context *ctx, fz_output *out)
+{
+	fz_write_printf(ctx, out,
+		"%%!PS-Adobe-3.0\n"
+		//"%%%%BoundingBox: 0 0 612 792\n"
+		//"%%%%HiResBoundingBox: 0 0 612 792\n"
+		"%%%%Creator: MuPDF\n"
+		"%%%%LanguageLevel: 2\n"
+		"%%%%CreationDate: D:20160318101706Z00'00'\n"
+		"%%%%DocumentData: Binary\n"
+		"%%%%Pages: (atend)\n"
+		"%%%%EndComments\n"
+		"\n"
+		"%%%%BeginProlog\n"
+		"%%%%EndProlog\n"
+		"\n"
+		"%%%%BeginSetup\n"
+		"%%%%EndSetup\n"
+		"\n"
+		);
+}
+
+void
+fz_write_ps_file_trailer(fz_context *ctx, fz_output *out, int pages)
+{
+	fz_write_printf(ctx, out, "%%%%Trailer\n%%%%Pages: %d\n%%%%EOF\n", pages);
+}
+
+static void
+ps_write_header(fz_context *ctx, fz_band_writer *writer_, fz_colorspace *cs)
+{
+	ps_band_writer *writer = (ps_band_writer *)writer_;
+	fz_output *out = writer->super.out;
+	int w = writer->super.w;
+	int h = writer->super.h;
+	int n = writer->super.n;
+	int alpha = writer->super.alpha;
+	int xres = writer->super.xres;
+	int yres = writer->super.yres;
+	int pagenum = writer->super.pagenum;
+	int w_points = (w * 72 + (xres>>1)) / xres;
+	int h_points = (h * 72 + (yres>>1)) / yres;
+	float sx = (float) w / w_points;
+	float sy = (float) h / h_points;
+	int err;
+
+	if (writer->super.s != 0)
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "Postscript writer cannot cope with spot colors");
+
+	if (alpha != 0)
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "Postscript output cannot have alpha");
+
+	writer->super.w = w;
+	writer->super.h = h;
+	writer->super.n = n;
+
+	writer->stream.zalloc = fz_zlib_alloc;
+	writer->stream.zfree = fz_zlib_free;
+	writer->stream.opaque = ctx;
+	writer->stream_started = 1;
+
+	err = deflateInit(&writer->stream, Z_DEFAULT_COMPRESSION);
+	if (err != Z_OK)
+		fz_throw(ctx, FZ_ERROR_LIBRARY, "compression error %d", err);
+
+	fz_write_printf(ctx, out, "%%%%Page: %d %d\n", pagenum, pagenum);
+	fz_write_printf(ctx, out, "%%%%PageBoundingBox: 0 0 %d %d\n", w_points, h_points);
+	fz_write_printf(ctx, out, "%%%%BeginPageSetup\n");
+	fz_write_printf(ctx, out, "<</PageSize [%d %d]>> setpagedevice\n", w_points, h_points);
+	fz_write_printf(ctx, out, "%%%%EndPageSetup\n\n");
+	fz_write_printf(ctx, out, "/DataFile currentfile /FlateDecode filter def\n\n");
+	switch(n)
+	{
+	case 1:
+		fz_write_string(ctx, out, "/DeviceGray setcolorspace\n");
+		break;
+	case 3:
+		fz_write_string(ctx, out, "/DeviceRGB setcolorspace\n");
+		break;
+	case 4:
+		fz_write_string(ctx, out, "/DeviceCMYK setcolorspace\n");
+		break;
+	default:
+		fz_throw(ctx, FZ_ERROR_ARGUMENT, "Unexpected colorspace for ps output");
+	}
+	fz_write_printf(ctx, out,
+		"<<\n"
+		"/ImageType 1\n"
+		"/Width %d\n"
+		"/Height %d\n"
+		"/ImageMatrix [ %g 0 0 -%g 0 %d ]\n"
+		"/MultipleDataSources false\n"
+		"/DataSource DataFile\n"
+		"/BitsPerComponent 8\n"
+		//"/Decode [0 1]\n"
+		"/Interpolate false\n"
+		">>\n"
+		"image\n"
+		, w, h, sx, sy, h);
+}
+
+static void
+ps_write_trailer(fz_context *ctx, fz_band_writer *writer_)
+{
+	ps_band_writer *writer = (ps_band_writer *)writer_;
+	fz_output *out = writer->super.out;
+	int err;
+
+	writer->stream_ended = 1;
+	err = deflateEnd(&writer->stream);
+	if (err != Z_OK)
+		fz_throw(ctx, FZ_ERROR_LIBRARY, "compression error %d", err);
+
+	fz_write_data(ctx, out, writer->output, writer->output_size - writer->stream.avail_out);
+	fz_write_string(ctx, out, "\nshowpage\n%%%%PageTrailer\n%%%%EndPageTrailer\n\n");
+}
+
+static void
+ps_drop_band_writer(fz_context *ctx, fz_band_writer *writer_)
+{
+	ps_band_writer *writer = (ps_band_writer *)writer_;
+
+	if (writer->stream_started && !writer->stream_ended)
+	{
+		int err = deflateEnd(&writer->stream);
+		if (err != Z_OK)
+			fz_warn(ctx, "ignoring compression error %d", err);
+	}
+
+	fz_free(ctx, writer->input);
+	fz_free(ctx, writer->output);
+}
+
+void fz_write_pixmap_as_ps(fz_context *ctx, fz_output *out, const fz_pixmap *pixmap)
+{
+	fz_band_writer *writer;
+
+	fz_write_ps_file_header(ctx, out);
+
+	writer = fz_new_ps_band_writer(ctx, out);
+
+	fz_try(ctx)
+	{
+		fz_write_header(ctx, writer, pixmap->w, pixmap->h, pixmap->n, pixmap->alpha, pixmap->xres, pixmap->yres, 0, pixmap->colorspace, pixmap->seps);
+		fz_write_band(ctx, writer, pixmap->stride, pixmap->h, pixmap->samples);
+		fz_close_band_writer(ctx, writer);
+	}
+	fz_always(ctx)
+	{
+		fz_drop_band_writer(ctx, writer);
+	}
+	fz_catch(ctx)
+	{
+		fz_rethrow(ctx);
+	}
+
+	fz_write_ps_file_trailer(ctx, out, 1);
+}
+
+void fz_save_pixmap_as_ps(fz_context *ctx, fz_pixmap *pixmap, char *filename, int append)
+{
+	fz_output *out = fz_new_output_with_path(ctx, filename, append);
+	fz_try(ctx)
+	{
+		fz_write_pixmap_as_ps(ctx, out, pixmap);
+		fz_close_output(ctx, out);
+	}
+	fz_always(ctx)
+		fz_drop_output(ctx, out);
+	fz_catch(ctx)
+		fz_rethrow(ctx);
+}
+
+static void
+ps_write_band(fz_context *ctx, fz_band_writer *writer_, int stride, int band_start, int band_height, const unsigned char *samples)
+{
+	ps_band_writer *writer = (ps_band_writer *)writer_;
+	fz_output *out = writer->super.out;
+	int w = writer->super.w;
+	int h = writer->super.h;
+	int n = writer->super.n;
+	int x, y, i, err, finalband;
+	size_t required_input;
+	size_t required_output;
+	size_t remain;
+	unsigned char *o;
+
+	if (!out)
+		return;
+
+
+	finalband = (band_start+band_height >= h);
+	if (finalband)
+		band_height = h - band_start;
+
+	required_input = w;
+	if (required_input > SIZE_MAX / n)
+		fz_throw(ctx, FZ_ERROR_LIMIT, "ps data too large.");
+	required_input = required_input * n;
+	if (required_input > SIZE_MAX / band_height)
+		fz_throw(ctx, FZ_ERROR_LIMIT, "ps data too large.");
+	required_input *= band_height;
+	required_output = required_input >= UINT_MAX ? UINT_MAX : deflateBound(&writer->stream, (uLong)required_input);
+	if (required_output < required_input || required_output > UINT_MAX)
+		required_output = UINT_MAX;
+
+	if (writer->input == NULL || writer->input_size < required_input)
+	{
+		fz_free(ctx, writer->input);
+		writer->input = NULL;
+		writer->input = Memento_label(fz_malloc(ctx, required_input), "pswriter_input");
+		writer->input_size = required_input;
+	}
+
+	if (writer->output == NULL || writer->output_size < required_output)
+	{
+		fz_free(ctx, writer->output);
+		writer->output = NULL;
+		writer->output = Memento_label(fz_malloc(ctx, required_output), "pswriter_output");
+		writer->output_size = required_output;
+	}
+
+	o = writer->input;
+	for (y = 0; y < band_height; y++)
+	{
+		for (x = 0; x < w; x++)
+		{
+			for (i = n; i > 0; i--)
+				*o++ = *samples++;
+		}
+		samples += stride - w*n;
+	}
+
+	remain = o - writer->input;
+	o = writer->input;
+
+	do
+	{
+		size_t eaten;
+
+		writer->stream.next_in = o;
+		writer->stream.avail_in = (uInt)(remain <= UINT_MAX ? remain : UINT_MAX);
+		writer->stream.next_out = writer->output;
+		writer->stream.avail_out = writer->output_size <= UINT_MAX ? (uInt)writer->output_size : UINT_MAX;
+
+		err = deflate(&writer->stream, (finalband && remain == writer->stream.avail_in) ? Z_FINISH : Z_NO_FLUSH);
+		if (err != Z_OK && err != Z_STREAM_END)
+			fz_throw(ctx, FZ_ERROR_LIBRARY, "compression error %d", err);
+
+		/* We are guaranteed that writer->stream.next_in will have been updated for the
+		 * data that has been eaten. */
+		eaten = (writer->stream.next_in - o);
+		remain -= eaten;
+		o += eaten;
+
+		/* We are guaranteed that writer->stream.next_out will have been updated for the
+		 * data that has been written. */
+		if (writer->stream.next_out != writer->output)
+			fz_write_data(ctx, out, writer->output, writer->output_size - writer->stream.avail_out);
+
+		/* Zlib only guarantees to have finished when we have no more data to feed in, and
+		 * the last call to deflate did not return with avail_out == 0. (i.e. no more is
+		 * buffered internally.) */
+	}
+	while (remain != 0 || writer->stream.avail_out == 0);
+}
+
+fz_band_writer *fz_new_ps_band_writer(fz_context *ctx, fz_output *out)
+{
+	ps_band_writer *writer = fz_new_band_writer(ctx, ps_band_writer, out);
+
+	writer->super.header = ps_write_header;
+	writer->super.band = ps_write_band;
+	writer->super.trailer = ps_write_trailer;
+	writer->super.drop = ps_drop_band_writer;
+
+	return &writer->super;
+}
+
+/* High-level document writer interface */
+
+typedef struct
+{
+	fz_document_writer super;
+	fz_draw_options draw;
+	fz_pixmap *pixmap;
+	fz_output *out;
+	int count;
+} fz_ps_writer;
+
+static fz_device *
+ps_begin_page(fz_context *ctx, fz_document_writer *wri_, fz_rect mediabox)
+{
+	fz_ps_writer *wri = (fz_ps_writer*)wri_;
+	wri->count++;
+	return fz_new_draw_device_with_options(ctx, &wri->draw, mediabox, &wri->pixmap);
+}
+
+static void
+ps_end_page(fz_context *ctx, fz_document_writer *wri_, fz_device *dev)
+{
+	fz_ps_writer *wri = (fz_ps_writer*)wri_;
+	fz_pixmap *pix = wri->pixmap;
+	fz_band_writer *bw;
+
+	fz_try(ctx)
+	{
+		fz_close_device(ctx, dev);
+		bw = fz_new_ps_band_writer(ctx, wri->out);
+		fz_write_header(ctx, bw, pix->w, pix->h, pix->n, pix->alpha, pix->xres, pix->yres, 0, pix->colorspace, pix->seps);
+		fz_write_band(ctx, bw, pix->stride, pix->h, pix->samples);
+		fz_close_band_writer(ctx, bw);
+	}
+	fz_always(ctx)
+	{
+		fz_drop_device(ctx, dev);
+		fz_drop_band_writer(ctx, bw);
+		fz_drop_pixmap(ctx, wri->pixmap);
+		wri->pixmap = NULL;
+	}
+	fz_catch(ctx)
+		fz_rethrow(ctx);
+}
+
+static void
+ps_close_writer(fz_context *ctx, fz_document_writer *wri_)
+{
+	fz_ps_writer *wri = (fz_ps_writer*)wri_;
+	fz_write_ps_file_trailer(ctx, wri->out, wri->count);
+	fz_close_output(ctx, wri->out);
+}
+
+static void
+ps_drop_writer(fz_context *ctx, fz_document_writer *wri_)
+{
+	fz_ps_writer *wri = (fz_ps_writer*)wri_;
+	fz_drop_pixmap(ctx, wri->pixmap);
+	fz_drop_output(ctx, wri->out);
+}
+
+fz_document_writer *
+fz_new_ps_writer_with_output(fz_context *ctx, fz_output *out, const char *options)
+{
+	fz_ps_writer *wri = NULL;
+
+	fz_var(wri);
+
+	fz_try(ctx)
+	{
+		wri = fz_new_derived_document_writer(ctx, fz_ps_writer, ps_begin_page, ps_end_page, ps_close_writer, ps_drop_writer);
+		fz_parse_draw_options(ctx, &wri->draw, options);
+		wri->out = out;
+		fz_write_ps_file_header(ctx, wri->out);
+	}
+	fz_catch(ctx)
+	{
+		fz_drop_output(ctx, out);
+		fz_free(ctx, wri);
+		fz_rethrow(ctx);
+	}
+
+	return (fz_document_writer*)wri;
+}
+
+fz_document_writer *
+fz_new_ps_writer(fz_context *ctx, const char *path, const char *options)
+{
+	fz_output *out = fz_new_output_with_path(ctx, path ? path : "out.ps", 0);
+	return fz_new_ps_writer_with_output(ctx, out, options);
+}