diff options
Diffstat (limited to 'apps/mupdfdraw.c')
-rw-r--r-- | apps/mupdfdraw.c | 495 |
1 files changed, 495 insertions, 0 deletions
diff --git a/apps/mupdfdraw.c b/apps/mupdfdraw.c new file mode 100644 index 00000000..552fe31a --- /dev/null +++ b/apps/mupdfdraw.c @@ -0,0 +1,495 @@ +/* + * pdfdraw -- command line tool for drawing pdf documents + */ + +#include "fitz.h" +#include "mupdf.h" + +#ifdef _MSC_VER +#include <winsock2.h> +#else +#include <sys/time.h> +#endif + +static char *output = NULL; +static float resolution = 72; +static float rotation = 0; + +static int showxml = 0; +static int showtext = 0; +static int showtime = 0; +static int showmd5 = 0; +static int showoutline = 0; +static int savealpha = 0; +static int uselist = 1; +static int alphabits = 8; +static float gamma_value = 1; +static int invert = 0; + +static fz_colorspace *colorspace; +static char *filename; + +static struct { + int count, total; + int min, max; + int minpage, maxpage; +} timing; + +static void usage(void) +{ + fprintf(stderr, + "usage: pdfdraw [options] input.pdf [pages]\n" + "\t-o -\toutput filename (%%d for page number)\n" + "\t\tsupported formats: pgm, ppm, pam, png, pbm\n" + "\t-p -\tpassword\n" + "\t-r -\tresolution in dpi (default: 72)\n" + "\t-A\tdisable accelerated functions\n" + "\t-a\tsave alpha channel (only pam and png)\n" + "\t-b -\tnumber of bits of antialiasing (0 to 8)\n" + "\t-g\trender in grayscale\n" + "\t-m\tshow timing information\n" + "\t-t\tshow text (-tt for xml)\n" + "\t-x\tshow display list\n" + "\t-d\tdisable use of display list\n" + "\t-5\tshow md5 checksums\n" + "\t-R -\trotate clockwise by given number of degrees\n" + "\t-G gamma\tgamma correct output\n" + "\t-I\tinvert output\n" + "\t-l\tprint outline\n" + "\tpages\tcomma separated list of ranges\n"); + exit(1); +} + +static int gettime(void) +{ + static struct timeval first; + static int once = 1; + struct timeval now; + if (once) + { + gettimeofday(&first, NULL); + once = 0; + } + gettimeofday(&now, NULL); + return (now.tv_sec - first.tv_sec) * 1000 + (now.tv_usec - first.tv_usec) / 1000; +} + +static int isrange(char *s) +{ + while (*s) + { + if ((*s < '0' || *s > '9') && *s != '-' && *s != ',') + return 0; + s++; + } + return 1; +} + +static void drawpage(pdf_document *doc, int pagenum) +{ + pdf_page *page; + fz_display_list *list = NULL; + fz_device *dev = NULL; + int start; + fz_context *ctx = doc->ctx; + + fz_var(list); + fz_var(dev); + + if (showtime) + { + start = gettime(); + } + + fz_try(ctx) + { + page = pdf_load_page(doc, pagenum - 1); + } + fz_catch(ctx) + { + fz_throw(ctx, "cannot load page %d in file '%s'", pagenum, filename); + } + + if (uselist) + { + fz_try(ctx) + { + list = fz_new_display_list(ctx); + dev = fz_new_list_device(ctx, list); + pdf_run_page(doc, page, dev, fz_identity, NULL); + } + fz_catch(ctx) + { + fz_free_device(dev); + fz_free_display_list(ctx, list); + pdf_free_page(ctx, page); + fz_throw(ctx, "cannot draw page %d in file '%s'", pagenum, filename); + } + fz_free_device(dev); + dev = NULL; + } + + if (showxml) + { + fz_try(ctx) + { + dev = fz_new_trace_device(ctx); + printf("<page number=\"%d\">\n", pagenum); + if (list) + fz_execute_display_list(list, dev, fz_identity, fz_infinite_bbox, NULL); + else + pdf_run_page(doc, page, dev, fz_identity, NULL); + printf("</page>\n"); + } + fz_catch(ctx) + { + fz_free_device(dev); + fz_free_display_list(ctx, list); + pdf_free_page(ctx, page); + fz_rethrow(ctx); + } + fz_free_device(dev); + dev = NULL; + } + + if (showtext) + { + fz_text_span *text = NULL; + + fz_var(text); + + fz_try(ctx) + { + text = fz_new_text_span(ctx); + dev = fz_new_text_device(ctx, text); + if (list) + fz_execute_display_list(list, dev, fz_identity, fz_infinite_bbox, NULL); + else + pdf_run_page(doc, page, dev, fz_identity, NULL); + fz_free_device(dev); + dev = NULL; + printf("[Page %d]\n", pagenum); + if (showtext > 1) + fz_debug_text_span_xml(text); + else + fz_debug_text_span(text); + printf("\n"); + } + fz_catch(ctx) + { + fz_free_device(dev); + fz_free_text_span(ctx, text); + fz_free_display_list(ctx, list); + pdf_free_page(ctx, page); + fz_rethrow(ctx); + } + fz_free_text_span(ctx, text); + } + + if (showmd5 || showtime) + printf("page %s %d", filename, pagenum); + + if (output || showmd5 || showtime) + { + float zoom; + fz_matrix ctm; + fz_rect bounds; + fz_bbox bbox; + fz_pixmap *pix = NULL; + + fz_var(pix); + + bounds = pdf_bound_page(doc, page); + zoom = resolution / 72; + ctm = fz_scale(zoom, zoom); + ctm = fz_concat(ctm, fz_rotate(rotation)); + bbox = fz_round_rect(fz_transform_rect(ctm, bounds)); + + /* TODO: banded rendering and multi-page ppm */ + + fz_try(ctx) + { + pix = fz_new_pixmap_with_rect(ctx, colorspace, bbox); + + if (savealpha) + fz_clear_pixmap(pix); + else + fz_clear_pixmap_with_color(pix, 255); + + dev = fz_new_draw_device(ctx, pix); + if (list) + fz_execute_display_list(list, dev, ctm, bbox, NULL); + else + pdf_run_page(doc, page, dev, ctm, NULL); + fz_free_device(dev); + dev = NULL; + + if (invert) + fz_invert_pixmap(pix); + if (gamma_value != 1) + fz_gamma_pixmap(pix, gamma_value); + + if (savealpha) + fz_unmultiply_pixmap(pix); + + if (output) + { + char buf[512]; + sprintf(buf, output, pagenum); + if (strstr(output, ".pgm") || strstr(output, ".ppm") || strstr(output, ".pnm")) + fz_write_pnm(ctx, pix, buf); + else if (strstr(output, ".pam")) + fz_write_pam(ctx, pix, buf, savealpha); + else if (strstr(output, ".png")) + fz_write_png(ctx, pix, buf, savealpha); + else if (strstr(output, ".pbm")) { + fz_halftone *ht = fz_get_default_halftone(ctx, 1); + fz_bitmap *bit = fz_halftone_pixmap(ctx, pix, ht); + fz_write_pbm(ctx, bit, buf); + fz_drop_bitmap(ctx, bit); + fz_drop_halftone(ctx, ht); + } + } + + if (showmd5) + { + fz_md5 md5; + unsigned char digest[16]; + int i; + + fz_md5_init(&md5); + fz_md5_update(&md5, pix->samples, pix->w * pix->h * pix->n); + fz_md5_final(&md5, digest); + + printf(" "); + for (i = 0; i < 16; i++) + printf("%02x", digest[i]); + } + + fz_drop_pixmap(ctx, pix); + } + fz_catch(ctx) + { + fz_free_device(dev); + fz_drop_pixmap(ctx, pix); + fz_free_display_list(ctx, list); + pdf_free_page(ctx, page); + fz_rethrow(ctx); + } + } + + if (list) + fz_free_display_list(ctx, list); + + pdf_free_page(ctx, page); + + if (showtime) + { + int end = gettime(); + int diff = end - start; + + if (diff < timing.min) + { + timing.min = diff; + timing.minpage = pagenum; + } + if (diff > timing.max) + { + timing.max = diff; + timing.maxpage = pagenum; + } + timing.total += diff; + timing.count ++; + + printf(" %dms", diff); + } + + if (showmd5 || showtime) + printf("\n"); + + fz_flush_warnings(ctx); +} + +static void drawrange(pdf_document *doc, char *range) +{ + int page, spage, epage; + char *spec, *dash; + + spec = fz_strsep(&range, ","); + while (spec) + { + dash = strchr(spec, '-'); + + if (dash == spec) + spage = epage = pdf_count_pages(doc); + else + spage = epage = atoi(spec); + + if (dash) + { + if (strlen(dash) > 1) + epage = atoi(dash + 1); + else + epage = pdf_count_pages(doc); + } + + spage = CLAMP(spage, 1, pdf_count_pages(doc)); + epage = CLAMP(epage, 1, pdf_count_pages(doc)); + + if (spage < epage) + for (page = spage; page <= epage; page++) + drawpage(doc, page); + else + for (page = spage; page >= epage; page--) + drawpage(doc, page); + + spec = fz_strsep(&range, ","); + } +} + +static void drawoutline(pdf_document *doc) +{ + fz_outline *outline = pdf_load_outline(doc); + if (showoutline > 1) + fz_debug_outline_xml(outline, 0); + else + fz_debug_outline(outline, 0); + fz_free_outline(outline); +} + +#ifdef MUPDF_COMBINED_EXE +int pdfdraw_main(int argc, char **argv) +#else +int main(int argc, char **argv) +#endif +{ + char *password = ""; + int grayscale = 0; + int accelerate = 1; + pdf_document *doc = NULL; + int c; + fz_context *ctx; + + fz_var(doc); + + while ((c = fz_getopt(argc, argv, "lo:p:r:R:Aab:dgmtx5G:I")) != -1) + { + switch (c) + { + case 'o': output = fz_optarg; break; + case 'p': password = fz_optarg; break; + case 'r': resolution = atof(fz_optarg); break; + case 'R': rotation = atof(fz_optarg); break; + case 'A': accelerate = 0; break; + case 'a': savealpha = 1; break; + case 'b': alphabits = atoi(fz_optarg); break; + case 'l': showoutline++; break; + case 'm': showtime++; break; + case 't': showtext++; break; + case 'x': showxml++; break; + case '5': showmd5++; break; + case 'g': grayscale++; break; + case 'd': uselist = 0; break; + case 'G': gamma_value = atof(fz_optarg); break; + case 'I': invert++; break; + default: usage(); break; + } + } + + if (fz_optind == argc) + usage(); + + if (!showtext && !showxml && !showtime && !showmd5 && !showoutline && !output) + { + printf("nothing to do\n"); + exit(0); + } + + if (accelerate) + fz_accelerate(); + + ctx = fz_new_context(NULL, FZ_STORE_DEFAULT); + if (!ctx) + { + fprintf(stderr, "cannot initialise context\n"); + exit(1); + } + + fz_set_aa_level(ctx, alphabits); + + colorspace = fz_device_rgb; + if (grayscale) + colorspace = fz_device_gray; + if (output && strstr(output, ".pgm")) + colorspace = fz_device_gray; + if (output && strstr(output, ".ppm")) + colorspace = fz_device_rgb; + if (output && strstr(output, ".pbm")) + colorspace = fz_device_gray; + + timing.count = 0; + timing.total = 0; + timing.min = 1 << 30; + timing.max = 0; + timing.minpage = 0; + timing.maxpage = 0; + + if (showxml) + printf("<?xml version=\"1.0\"?>\n"); + + fz_try(ctx) + { + while (fz_optind < argc) + { + filename = argv[fz_optind++]; + + fz_try(ctx) + { + doc = pdf_open_document(ctx, filename); + } + fz_catch(ctx) + { + fz_throw(ctx, "cannot open document: %s", filename); + } + + if (pdf_needs_password(doc)) + if (!pdf_authenticate_password(doc, password)) + fz_throw(ctx, "cannot authenticate password: %s", filename); + + if (showxml) + printf("<document name=\"%s\">\n", filename); + + if (showoutline) + drawoutline(doc); + + if (showtext || showxml || showtime || showmd5 || output) + { + if (fz_optind == argc || !isrange(argv[fz_optind])) + drawrange(doc, "1-"); + if (fz_optind < argc && isrange(argv[fz_optind])) + drawrange(doc, argv[fz_optind++]); + } + + if (showxml) + printf("</document>\n"); + + pdf_close_document(doc); + doc = NULL; + } + } + fz_catch(ctx) + { + pdf_close_document(doc); + } + + if (showtime) + { + printf("total %dms / %d pages for an average of %dms\n", + timing.total, timing.count, timing.total / timing.count); + printf("fastest page %d: %dms\n", timing.minpage, timing.min); + printf("slowest page %d: %dms\n", timing.maxpage, timing.max); + } + + fz_free_context(ctx); + return 0; +} |