Mercurial > hgrepos > Python2 > PyMuPDF
diff mupdf-source/source/fitz/output-cbz.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-cbz.c Mon Sep 15 11:43:07 2025 +0200 @@ -0,0 +1,204 @@ +// Copyright (C) 2004-2024 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 <zlib.h> + +#include <limits.h> + +typedef struct +{ + fz_document_writer super; + fz_draw_options options; + fz_pixmap *pixmap; + int count; + fz_zip_writer *zip; +} fz_cbz_writer; + +static fz_device * +cbz_begin_page(fz_context *ctx, fz_document_writer *wri_, fz_rect mediabox) +{ + fz_cbz_writer *wri = (fz_cbz_writer*)wri_; + return fz_new_draw_device_with_options(ctx, &wri->options, mediabox, &wri->pixmap); +} + +static void +cbz_end_page(fz_context *ctx, fz_document_writer *wri_, fz_device *dev) +{ + fz_cbz_writer *wri = (fz_cbz_writer*)wri_; + fz_buffer *buffer = NULL; + char name[40]; + + fz_var(buffer); + + fz_try(ctx) + { + fz_close_device(ctx, dev); + wri->count += 1; + fz_snprintf(name, sizeof name, "p%04d.png", wri->count); + buffer = fz_new_buffer_from_pixmap_as_png(ctx, wri->pixmap, fz_default_color_params); + fz_write_zip_entry(ctx, wri->zip, name, buffer, 0); + } + fz_always(ctx) + { + fz_drop_device(ctx, dev); + fz_drop_buffer(ctx, buffer); + fz_drop_pixmap(ctx, wri->pixmap); + wri->pixmap = NULL; + } + fz_catch(ctx) + fz_rethrow(ctx); +} + +static void +cbz_close_writer(fz_context *ctx, fz_document_writer *wri_) +{ + fz_cbz_writer *wri = (fz_cbz_writer*)wri_; + fz_close_zip_writer(ctx, wri->zip); +} + +static void +cbz_drop_writer(fz_context *ctx, fz_document_writer *wri_) +{ + fz_cbz_writer *wri = (fz_cbz_writer*)wri_; + fz_drop_zip_writer(ctx, wri->zip); + fz_drop_pixmap(ctx, wri->pixmap); +} + +fz_document_writer * +fz_new_cbz_writer_with_output(fz_context *ctx, fz_output *out, const char *options) +{ + fz_cbz_writer *wri = NULL; + + fz_var(wri); + fz_var(out); + + fz_try(ctx) + { + fz_output *out_temp = out; + wri = fz_new_derived_document_writer(ctx, fz_cbz_writer, cbz_begin_page, cbz_end_page, cbz_close_writer, cbz_drop_writer); + fz_parse_draw_options(ctx, &wri->options, options); + out = NULL; + wri->zip = fz_new_zip_writer_with_output(ctx, out_temp); + } + 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_cbz_writer(fz_context *ctx, const char *path, const char *options) +{ + fz_output *out = fz_new_output_with_path(ctx, path ? path : "out.cbz", 0); + fz_document_writer *wri = NULL; + fz_try(ctx) + wri = fz_new_cbz_writer_with_output(ctx, out, options); + fz_catch(ctx) + { + fz_drop_output(ctx, out); + fz_rethrow(ctx); + } + return wri; +} + +/* generic image file output writer */ + +typedef struct +{ + fz_document_writer super; + fz_draw_options options; + fz_pixmap *pixmap; + void (*save)(fz_context *ctx, fz_pixmap *pix, const char *filename); + int count; + char *path; +} fz_pixmap_writer; + +static fz_device * +pixmap_begin_page(fz_context *ctx, fz_document_writer *wri_, fz_rect mediabox) +{ + fz_pixmap_writer *wri = (fz_pixmap_writer*)wri_; + return fz_new_draw_device_with_options(ctx, &wri->options, mediabox, &wri->pixmap); +} + +static void +pixmap_end_page(fz_context *ctx, fz_document_writer *wri_, fz_device *dev) +{ + fz_pixmap_writer *wri = (fz_pixmap_writer*)wri_; + char path[PATH_MAX]; + + fz_try(ctx) + { + fz_close_device(ctx, dev); + wri->count += 1; + fz_format_output_path(ctx, path, sizeof path, wri->path, wri->count); + wri->save(ctx, wri->pixmap, path); + } + fz_always(ctx) + { + fz_drop_device(ctx, dev); + fz_drop_pixmap(ctx, wri->pixmap); + wri->pixmap = NULL; + } + fz_catch(ctx) + fz_rethrow(ctx); +} + +static void +pixmap_drop_writer(fz_context *ctx, fz_document_writer *wri_) +{ + fz_pixmap_writer *wri = (fz_pixmap_writer*)wri_; + fz_drop_pixmap(ctx, wri->pixmap); + fz_free(ctx, wri->path); +} + +fz_document_writer * +fz_new_pixmap_writer(fz_context *ctx, const char *path, const char *options, + const char *default_path, int n, + void (*save)(fz_context *ctx, fz_pixmap *pix, const char *filename)) +{ + fz_pixmap_writer *wri = fz_new_derived_document_writer(ctx, fz_pixmap_writer, pixmap_begin_page, pixmap_end_page, NULL, pixmap_drop_writer); + + fz_try(ctx) + { + fz_parse_draw_options(ctx, &wri->options, options); + wri->path = fz_strdup(ctx, path ? path : default_path); + wri->save = save; + switch (n) + { + case 1: wri->options.colorspace = fz_device_gray(ctx); break; + case 3: wri->options.colorspace = fz_device_rgb(ctx); break; + case 4: wri->options.colorspace = fz_device_cmyk(ctx); break; + } + } + fz_catch(ctx) + { + fz_free(ctx, wri); + fz_rethrow(ctx); + } + + return (fz_document_writer*)wri; +}
