summaryrefslogtreecommitdiff
path: root/source/fitz/output-cbz.c
blob: df5c06400fc81b4aaf959e4c27df05390ec145f6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
#include "mupdf/fitz.h"

#include <zlib.h>

typedef struct fz_cbz_writer_s fz_cbz_writer;

struct fz_cbz_writer_s
{
	fz_document_writer super;
	fz_zip_writer *zip;
	int resolution;
	fz_pixmap *pixmap;
	int count;
};

static fz_device *
cbz_begin_page(fz_context *ctx, fz_document_writer *wri_, const fz_rect *mediabox, fz_matrix *ctm)
{
	fz_cbz_writer *wri = (fz_cbz_writer*)wri_;
	fz_rect bbox;
	fz_irect ibbox;

	fz_scale(ctm, wri->resolution / 72, wri->resolution / 72);
	bbox = *mediabox;
	fz_transform_rect(&bbox, ctm);
	fz_round_rect(&ibbox, &bbox);

	wri->pixmap = fz_new_pixmap_with_bbox(ctx, fz_device_rgb(ctx), &ibbox);
	fz_clear_pixmap_with_value(ctx, wri->pixmap, 0xFF);

	return fz_new_draw_device(ctx, 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;
	char name[40];

	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_try(ctx)
		fz_write_zip_entry(ctx, wri->zip, name, buffer, 0);
	fz_always(ctx)
		fz_drop_buffer(ctx, buffer);
	fz_catch(ctx)
		fz_rethrow(ctx);

	fz_drop_pixmap(ctx, wri->pixmap);
	wri->pixmap = NULL;
}

static void
cbz_drop_imp(fz_context *ctx, fz_document_writer *wri_)
{
	fz_cbz_writer *wri = (fz_cbz_writer*)wri_;
	fz_try(ctx)
		fz_drop_zip_writer(ctx, wri->zip);
	fz_always(ctx)
		fz_drop_pixmap(ctx, wri->pixmap);
	fz_catch(ctx)
		fz_rethrow(ctx);
}

fz_document_writer *
fz_new_cbz_writer(fz_context *ctx, const char *path, const char *options)
{
	fz_cbz_writer *wri = fz_malloc_struct(ctx, fz_cbz_writer);

	wri->super.begin_page = cbz_begin_page;
	wri->super.end_page = cbz_end_page;
	wri->super.drop_imp = cbz_drop_imp;

	fz_try(ctx)
		wri->zip = fz_new_zip_writer(ctx, path);
	fz_catch(ctx)
	{
		fz_free(ctx, wri);
		fz_rethrow(ctx);
	}

	// TODO: getopt-like comma separated list of options
	if (options)
		wri->resolution = atoi(options);

	if (wri->resolution == 0)
		wri->resolution = 96;

	return (fz_document_writer*)wri;
}