Kill stray debugging stuff
[llpp.git] / link.c
bloba738d01f406c8b8d694313544ff1df1ff49277ad
1 /* lots of code c&p-ed directly from mupdf */
2 #define CAML_NAME_SPACE
4 #include <errno.h>
5 #include <stdio.h>
6 #include <ctype.h>
7 #include <string.h>
8 #include <stdlib.h>
9 #include <signal.h>
10 #include <wchar.h>
12 #include <unistd.h>
13 #include <pthread.h>
14 #include <sys/time.h>
15 #include <sys/types.h>
16 #include <sys/ioctl.h>
17 #include <sys/utsname.h>
19 #ifdef __CYGWIN__
20 #include <cygwin/socket.h> /* FIONREAD */
21 #else
22 #include <spawn.h>
23 #endif
25 #include <regex.h>
26 #include <stdarg.h>
27 #include <limits.h>
28 #include <inttypes.h>
30 #include <GL/gl.h>
32 #include <caml/fail.h>
33 #include <caml/alloc.h>
34 #include <caml/memory.h>
35 #include <caml/unixsupport.h>
37 #include <mupdf/fitz.h>
38 #include <mupdf/cbz.h>
39 #include <mupdf/pdf.h>
40 #include <mupdf/xps.h>
41 #include <mupdf/img.h>
43 #include <ft2build.h>
44 #include FT_FREETYPE_H
46 #ifdef USE_FONTCONFIG
47 #include <fontconfig/fontconfig.h>
48 #endif
50 #define PIGGYBACK
51 #define CACHE_PAGEREFS
53 #ifndef __USE_GNU
54 extern char **environ;
55 #endif
57 #define MIN(a,b) ((a) < (b) ? (a) : (b))
58 #define MAX(a,b) ((a) > (b) ? (a) : (b))
60 #if defined __GNUC__
61 #define NORETURN_ATTR __attribute__ ((noreturn))
62 #define UNUSED_ATTR __attribute__ ((unused))
63 #if !defined __clang__
64 #define OPTIMIZE_ATTR(n) __attribute__ ((optimize ("O"#n)))
65 #else
66 #define OPTIMIZE_ATTR(n)
67 #endif
68 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
69 #else
70 #define NORETURN_ATTR
71 #define UNUSED_ATTR
72 #define OPTIMIZE_ATTR(n)
73 #define GCC_FMT_ATTR(a, b)
74 #endif
76 #define FMT_s "zu"
78 #define FMT_ptr PRIxPTR
79 #define SCN_ptr SCNxPTR
80 #define FMT_ptr_cast(p) ((uintptr_t) (p))
81 #define SCN_ptr_cast(p) ((uintptr_t *) (p))
83 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
84 err (int exitcode, const char *fmt, ...)
86 va_list ap;
87 int savederrno;
89 savederrno = errno;
90 va_start (ap, fmt);
91 vfprintf (stderr, fmt, ap);
92 va_end (ap);
93 fprintf (stderr, ": %s\n", strerror (savederrno));
94 fflush (stderr);
95 _exit (exitcode);
98 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
99 errx (int exitcode, const char *fmt, ...)
101 va_list ap;
103 va_start (ap, fmt);
104 vfprintf (stderr, fmt, ap);
105 va_end (ap);
106 fputc ('\n', stderr);
107 fflush (stderr);
108 _exit (exitcode);
111 #ifndef GL_TEXTURE_RECTANGLE_ARB
112 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
113 #endif
115 #ifndef GL_BGRA
116 #define GL_BGRA 0x80E1
117 #endif
119 #ifndef GL_UNSIGNED_INT_8_8_8_8
120 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
121 #endif
123 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
124 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
125 #endif
127 #if 0
128 #define lprintf printf
129 #else
130 #define lprintf(...)
131 #endif
133 #define ARSERT(cond) for (;;) { \
134 if (!(cond)) { \
135 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
137 break; \
140 struct slice {
141 int h;
142 int texindex;
145 struct tile {
146 int w, h;
147 int slicecount;
148 int sliceheight;
149 struct pbo *pbo;
150 fz_pixmap *pixmap;
151 struct slice slices[1];
154 struct pagedim {
155 int pageno;
156 int rotate;
157 int left;
158 int tctmready;
159 fz_irect bounds;
160 fz_rect pagebox;
161 fz_rect mediabox;
162 fz_matrix ctm, zoomctm, lctm, tctm;
165 struct slink {
166 enum { SLINK, SANNOT } tag;
167 fz_irect bbox;
168 union {
169 fz_link *link;
170 pdf_annot *annot;
171 } u;
174 struct annot {
175 fz_irect bbox;
176 pdf_annot *annot;
179 enum { DNONE, DPDF, DXPS, DCBZ, DIMG };
181 struct page {
182 int tgen;
183 int sgen;
184 int agen;
185 int type;
186 int pageno;
187 int pdimno;
188 fz_text_page *text;
189 fz_text_sheet *sheet;
190 union {
191 void *ptr;
192 pdf_page *pdfpage;
193 xps_page *xpspage;
194 cbz_page *cbzpage;
195 image_page *imgpage;
196 } u;
197 fz_display_list *dlist;
198 int slinkcount;
199 struct slink *slinks;
200 int annotcount;
201 struct annot *annots;
202 struct mark {
203 int i;
204 fz_text_span *span;
205 } fmark, lmark;
206 void (*freepage) (void *);
209 struct {
210 int type;
211 int sliceheight;
212 struct pagedim *pagedims;
213 int pagecount;
214 int pagedimcount;
215 union {
216 pdf_document *pdf;
217 xps_document *xps;
218 cbz_document *cbz;
219 image_document *img;
220 } u;
221 fz_context *ctx;
222 int w, h;
224 int texindex;
225 int texcount;
226 GLuint *texids;
228 GLenum texiform;
229 GLenum texform;
230 GLenum texty;
232 fz_colorspace *colorspace;
234 struct {
235 int w, h;
236 struct slice *slice;
237 } *texowners;
239 int rotate;
240 enum { FitWidth, FitProportional, FitPage } fitmodel;
241 int trimmargins;
242 int needoutline;
243 int gen;
244 int aalevel;
246 int trimanew;
247 fz_irect trimfuzz;
248 fz_pixmap *pig;
250 pthread_t thread;
251 int csock;
252 FT_Face face;
254 void (*closedoc) (void);
255 void (*freepage) (void *);
257 char *trimcachepath;
258 int cxack;
260 GLuint stid;
262 int pbo_usable;
263 void (*glBindBufferARB) (GLenum, GLuint);
264 GLboolean (*glUnmapBufferARB) (GLenum);
265 void *(*glMapBufferARB) (GLenum, GLenum);
266 void (*glBufferDataARB) (GLenum, GLsizei, void *, GLenum);
267 void (*glGenBuffersARB) (GLsizei, GLuint *);
268 void (*glDeleteBuffersARB) (GLsizei, GLuint *);
270 GLfloat texcoords[8];
271 GLfloat vertices[16];
273 #ifdef CACHE_PAGEREFS
274 struct {
275 int idx;
276 int count;
277 pdf_obj **objs;
278 pdf_document *pdf;
279 } pdflut;
280 #endif
281 } state;
283 struct pbo {
284 GLuint id;
285 void *ptr;
286 size_t size;
289 static void UNUSED_ATTR debug_rect (const char *cap, fz_rect r)
291 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
294 static void UNUSED_ATTR debug_bbox (const char *cap, fz_irect r)
296 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
299 static void UNUSED_ATTR debug_matrix (const char *cap, fz_matrix m)
301 printf ("%s(matrix) %.2f,%.2f,%.2f,%.2f %.2f %.2f\n", cap,
302 m.a, m.b, m.c, m.d, m.e, m.f);
305 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
307 static void lock (const char *cap)
309 int ret = pthread_mutex_lock (&mutex);
310 if (ret) {
311 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
315 static void unlock (const char *cap)
317 int ret = pthread_mutex_unlock (&mutex);
318 if (ret) {
319 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
323 static int trylock (const char *cap)
325 int ret = pthread_mutex_trylock (&mutex);
326 if (ret && ret != EBUSY) {
327 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
329 return ret == EBUSY;
332 static void *parse_pointer (const char *cap, const char *s)
334 int ret;
335 void *ptr;
337 ret = sscanf (s, "%" SCN_ptr, SCN_ptr_cast (&ptr));
338 if (ret != 1) {
339 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
341 return ptr;
344 static double now (void)
346 struct timeval tv;
348 if (gettimeofday (&tv, NULL)) {
349 err (1, "gettimeofday");
351 return tv.tv_sec + tv.tv_usec*1e-6;
354 static int hasdata (void)
356 int ret, avail;
357 ret = ioctl (state.csock, FIONREAD, &avail);
358 if (ret) err (1, "hasdata: FIONREAD error ret=%d", ret);
359 return avail > 0;
362 CAMLprim value ml_hasdata (value fd_v)
364 CAMLparam1 (fd_v);
365 int ret, avail;
367 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
368 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
369 CAMLreturn (Val_bool (avail > 0));
372 static void readdata (void *p, int size)
374 ssize_t n;
376 again:
377 n = read (state.csock, p, size);
378 if (n < 0) {
379 if (errno == EINTR) goto again;
380 err (1, "read (req %d, ret %zd)", size, n);
382 if (n - size) {
383 if (!n) errx (1, "EOF while reading");
384 errx (1, "read (req %d, ret %zd)", size, n);
388 static void writedata (char *p, int size)
390 ssize_t n;
392 p[0] = (size >> 24) & 0xff;
393 p[1] = (size >> 16) & 0xff;
394 p[2] = (size >> 8) & 0xff;
395 p[3] = (size >> 0) & 0xff;
397 n = write (state.csock, p, size + 4);
398 if (n - size - 4) {
399 if (!n) errx (1, "EOF while writing data");
400 err (1, "write (req %d, ret %zd)", size + 4, n);
404 static int readlen (void)
406 unsigned char p[4];
408 readdata (p, 4);
409 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
412 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
414 int size = 200, len;
415 va_list ap;
416 char *buf;
418 buf = malloc (size);
419 for (;;) {
420 if (!buf) err (1, "malloc for temp buf (%d bytes) failed", size);
422 va_start (ap, fmt);
423 len = vsnprintf (buf + 4, size - 4, fmt, ap);
424 va_end (ap);
426 if (len > -1) {
427 if (len < size - 4) {
428 writedata (buf, len);
429 break;
431 else size = len + 5;
433 else {
434 err (1, "vsnprintf for `%s' failed", fmt);
436 buf = realloc (buf, size);
438 free (buf);
441 static void closepdf (void)
443 #ifdef CACHE_PAGEREFS
444 if (state.pdflut.objs) {
445 int i;
447 for (i = 0; i < state.pdflut.count; ++i) {
448 pdf_drop_obj (state.pdflut.objs[i]);
450 free (state.pdflut.objs);
451 state.pdflut.objs = NULL;
452 state.pdflut.idx = 0;
454 #endif
456 if (state.u.pdf) {
457 pdf_close_document (state.u.pdf);
458 state.u.pdf = NULL;
462 static void closexps (void)
464 if (state.u.xps) {
465 xps_close_document (state.u.xps);
466 state.u.xps = NULL;
470 static void closecbz (void)
472 if (state.u.cbz) {
473 cbz_close_document (state.u.cbz);
474 state.u.cbz = NULL;
478 static void closeimg (void)
480 if (state.u.img) {
481 image_close_document (state.u.img);
482 state.u.img = NULL;
486 static void freepdfpage (void *ptr)
488 pdf_free_page (state.u.pdf, ptr);
491 static void freeemptyxxxpage (void *ptr)
493 (void) ptr;
496 static void freexpspage (void *ptr)
498 xps_free_page (state.u.xps, ptr);
501 static void freecbzpage (void *ptr)
503 cbz_free_page (state.u.cbz, ptr);
506 static void freeimgpage (void *ptr)
508 image_free_page (state.u.img, ptr);
511 static int openxref (char *filename, char *password)
513 int i, len;
515 for (i = 0; i < state.texcount; ++i) {
516 state.texowners[i].w = -1;
517 state.texowners[i].slice = NULL;
520 if (state.closedoc) state.closedoc ();
522 len = strlen (filename);
524 state.type = DPDF;
526 size_t j;
527 struct {
528 char *ext;
529 int type;
530 } tbl[] = {
531 { "pdf", DPDF },
532 { "xps", DXPS },
533 { "zip", DCBZ },
534 { "cbz", DCBZ },
535 { "jpg", DIMG },
536 { "jpeg", DIMG },
537 { "png", DIMG }
540 for (j = 0; j < sizeof (tbl) / sizeof (*tbl); ++j) {
541 const char *ext = tbl[j].ext;
542 int len2 = strlen (ext);
544 if (len2 < len && !strcasecmp (ext, filename + len - len2)) {
545 state.type = tbl[j].type;
546 break;
551 if (state.pagedims) {
552 free (state.pagedims);
553 state.pagedims = NULL;
555 state.pagedimcount = 0;
557 fz_set_aa_level (state.ctx, state.aalevel);
558 switch (state.type) {
559 case DPDF:
560 state.u.pdf = pdf_open_document (state.ctx, filename);
561 if (pdf_needs_password (state.u.pdf)) {
562 if (password && !*password) {
563 printd ("pass");
564 return 0;
566 else {
567 int ok = pdf_authenticate_password (state.u.pdf, password);
568 if (!ok) {
569 printd ("pass fail");
570 return 0;
574 state.pagecount = pdf_count_pages (state.u.pdf);
575 state.closedoc = closepdf;
576 state.freepage = freepdfpage;
577 break;
579 case DXPS:
580 state.u.xps = xps_open_document (state.ctx, filename);
581 state.pagecount = xps_count_pages (state.u.xps);
582 state.closedoc = closexps;
583 state.freepage = freexpspage;
584 break;
586 case DCBZ:
587 state.u.cbz = cbz_open_document (state.ctx, filename);
588 state.pagecount = cbz_count_pages (state.u.cbz);
589 state.closedoc = closecbz;
590 state.freepage = freecbzpage;
591 break;
593 case DIMG:
594 state.u.img = image_open_document (state.ctx, filename);
595 state.pagecount = 1;
596 state.closedoc = closeimg;
597 state.freepage = freeimgpage;
598 break;
600 return 1;
603 static void pdfinfo (void)
605 if (state.type == DPDF) {
606 pdf_obj *infoobj;
608 printd ("info PDF version\t%d.%d",
609 state.u.pdf->version / 10, state.u.pdf->version % 10);
611 infoobj = pdf_dict_gets (pdf_trailer (state.u.pdf), "Info");
612 if (infoobj) {
613 unsigned int i;
614 char *s;
615 char *items[] = { "Title", "Author", "Creator",
616 "Producer", "CreationDate" };
618 for (i = 0; i < sizeof (items) / sizeof (*items); ++i) {
619 pdf_obj *obj = pdf_dict_gets (infoobj, items[i]);
620 s = pdf_to_utf8 (state.u.pdf, obj);
621 if (*s) printd ("info %s\t%s", items[i], s);
622 fz_free (state.ctx, s);
625 printd ("infoend");
629 static void unlinktile (struct tile *tile)
631 int i;
633 for (i = 0; i < tile->slicecount; ++i) {
634 struct slice *s = &tile->slices[i];
636 if (s->texindex != -1) {
637 if (state.texowners[s->texindex].slice == s) {
638 state.texowners[s->texindex].slice = NULL;
644 static void freepage (struct page *page)
646 if (page->text) {
647 fz_free_text_page (state.ctx, page->text);
649 if (page->sheet) {
650 fz_free_text_sheet (state.ctx, page->sheet);
652 if (page->slinks) {
653 free (page->slinks);
655 fz_drop_display_list (state.ctx, page->dlist);
656 page->freepage (page->u.ptr);
657 free (page);
660 static void freetile (struct tile *tile)
662 unlinktile (tile);
663 if (!tile->pbo) {
664 #ifndef PIGGYBACK
665 fz_drop_pixmap (state.ctx, tile->pixmap);
666 #else
667 if (state.pig) {
668 fz_drop_pixmap (state.ctx, state.pig);
670 state.pig = tile->pixmap;
671 #endif
673 else {
674 free (tile->pbo);
675 fz_drop_pixmap (state.ctx, tile->pixmap);
677 free (tile);
680 #ifdef __ALTIVEC__
681 #include <stdint.h>
682 #include <altivec.h>
684 static int cacheline32bytes;
686 static void __attribute__ ((constructor)) clcheck (void)
688 char **envp = environ;
689 unsigned long *auxv;
691 while (*envp++);
693 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
694 if (*auxv == 19) {
695 cacheline32bytes = auxv[1] == 32;
696 return;
701 static void OPTIMIZE_ATTR (3) clearpixmap (fz_pixmap *pixmap)
703 size_t size = pixmap->w * pixmap->h * pixmap->n;
704 if (cacheline32bytes && size > 32) {
705 intptr_t a1, a2, diff;
706 size_t sizea, i;
707 vector unsigned char v = vec_splat_u8 (-1);
708 vector unsigned char *p;
710 a1 = a2 = (intptr_t) pixmap->samples;
711 a2 = (a1 + 31) & ~31;
712 diff = a2 - a1;
713 sizea = size - diff;
714 p = (void *) a2;
716 while (a1 != a2) *(char *) a1++ = 0xff;
717 for (i = 0; i < (sizea & ~31); i += 32) {
718 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
719 vec_st (v, i, p);
720 vec_st (v, i + 16, p);
722 while (i < sizea) *((char *) a1 + i++) = 0xff;
724 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
726 #else
727 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
728 #endif
730 static void trimctm (pdf_page *page, int pindex)
732 fz_matrix ctm;
733 struct pagedim *pdim = &state.pagedims[pindex];
735 if (!pdim->tctmready) {
736 if (state.trimmargins) {
737 fz_rect realbox;
738 fz_matrix rm, sm, tm, im, ctm1;
740 fz_rotate (&rm, -pdim->rotate);
741 fz_scale (&sm, 1, -1);
742 fz_concat (&ctm, &rm, &sm);
743 realbox = pdim->mediabox;
744 fz_transform_rect (&realbox, &ctm);
745 fz_translate (&tm, -realbox.x0, -realbox.y0);
746 fz_concat (&ctm1, &ctm, &tm);
747 fz_invert_matrix (&im, &page->ctm);
748 fz_concat (&ctm, &im, &ctm1);
750 else {
751 ctm = fz_identity;
753 pdim->tctm = ctm;
754 pdim->tctmready = 1;
758 static fz_matrix pagectm (struct page *page)
760 fz_matrix ctm, tm;
762 if (page->type == DPDF) {
763 trimctm (page->u.pdfpage, page->pdimno);
764 fz_concat (&ctm,
765 &state.pagedims[page->pdimno].tctm,
766 &state.pagedims[page->pdimno].ctm);
768 else {
769 struct pagedim *pdim = &state.pagedims[page->pdimno];
771 fz_translate (&tm, -pdim->mediabox.x0, -pdim->mediabox.y0);
772 fz_concat (&ctm, &tm, &state.pagedims[page->pdimno].ctm);
774 return ctm;
777 static void *loadpage (int pageno, int pindex)
779 fz_device *dev;
780 struct page *page;
782 page = calloc (sizeof (struct page), 1);
783 if (!page) {
784 err (1, "calloc page %d", pageno);
787 page->dlist = fz_new_display_list (state.ctx);
788 dev = fz_new_list_device (state.ctx, page->dlist);
789 fz_try (state.ctx) {
790 switch (state.type) {
791 case DPDF:
792 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
793 pdf_run_page (state.u.pdf, page->u.pdfpage, dev,
794 &fz_identity, NULL);
795 page->freepage = freepdfpage;
796 break;
798 case DXPS:
799 page->u.xpspage = xps_load_page (state.u.xps, pageno);
800 xps_run_page (state.u.xps, page->u.xpspage, dev,
801 &fz_identity, NULL);
802 page->freepage = freexpspage;
803 break;
805 case DCBZ:
806 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
807 cbz_run_page (state.u.cbz, page->u.cbzpage, dev,
808 &fz_identity, NULL);
809 page->freepage = freecbzpage;
810 break;
812 case DIMG:
813 page->u.imgpage = image_load_page (state.u.img, pageno);
814 image_run_page (state.u.img, page->u.imgpage, dev,
815 &fz_identity, NULL);
816 page->freepage = freeimgpage;
817 break;
820 fz_catch (state.ctx) {
821 page->u.ptr = NULL;
822 page->freepage = freeemptyxxxpage;
824 fz_free_device (dev);
826 page->pdimno = pindex;
827 page->pageno = pageno;
828 page->sgen = state.gen;
829 page->agen = state.gen;
830 page->tgen = state.gen;
831 page->type = state.type;
833 return page;
836 static struct tile *alloctile (int h)
838 int i;
839 int slicecount;
840 size_t tilesize;
841 struct tile *tile;
843 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
844 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
845 tile = calloc (tilesize, 1);
846 if (!tile) {
847 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
849 for (i = 0; i < slicecount; ++i) {
850 int sh = MIN (h, state.sliceheight);
851 tile->slices[i].h = sh;
852 tile->slices[i].texindex = -1;
853 h -= sh;
855 tile->slicecount = slicecount;
856 tile->sliceheight = state.sliceheight;
857 return tile;
860 #ifdef OBSCURED_OPT
861 struct obs {
862 int cured;
863 fz_irect b;
866 static void obs_fill_image (fz_device *dev, fz_image UNUSED_ATTR *image,
867 const fz_matrix *ctm, float alpha)
869 struct obs *obs = dev->user;
871 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
872 fz_irect b;
873 fz_rect rect = fz_unit_rect;
875 fz_transform_rect (&rect, ctm);
876 fz_round_rect (&b, &rect);
877 fz_intersect_irect (&b, &obs->b);
878 obs->cured = b.x0 == obs->b.x0
879 && b.x1 == obs->b.x1
880 && b.y0 == obs->b.y0
881 && b.y1 == obs->b.y1;
885 static int obscured (struct page *page, fz_irect bbox)
887 fz_rect rect;
888 fz_matrix ctm;
889 fz_device dev;
890 struct obs obs;
892 memset (&dev, 0, sizeof (dev));
893 memset (&obs, 0, sizeof (obs));
894 dev.hints = 0;
895 dev.flags = 0;
896 dev.user = &obs;
897 dev.ctx = state.ctx;
898 dev.fill_image = obs_fill_image;
899 obs.b = bbox;
900 fz_rect_from_irect (&rect, &bbox);
901 ctm = pagectm (page);
902 fz_run_display_list (page->dlist, &dev, &ctm, &rect, NULL);
903 return obs.cured;
905 #define OBSCURED obscured
906 #else
907 #define OBSCURED(a, b) 0
908 #endif
910 static struct tile *rendertile (struct page *page, int x, int y, int w, int h,
911 struct pbo *pbo)
913 fz_rect rect;
914 fz_irect bbox;
915 fz_matrix ctm;
916 fz_device *dev;
917 struct tile *tile;
918 struct pagedim *pdim;
920 tile = alloctile (h);
921 pdim = &state.pagedims[page->pdimno];
923 bbox = pdim->bounds;
924 bbox.x0 += x;
925 bbox.y0 += y;
926 bbox.x1 = bbox.x0 + w;
927 bbox.y1 = bbox.y0 + h;
929 if (state.pig) {
930 if (state.pig->w == w
931 && state.pig->h == h
932 && state.pig->colorspace == state.colorspace) {
933 tile->pixmap = state.pig;
934 tile->pixmap->x = bbox.x0;
935 tile->pixmap->y = bbox.y0;
937 else {
938 fz_drop_pixmap (state.ctx, state.pig);
940 state.pig = NULL;
942 if (!tile->pixmap) {
943 if (pbo) {
944 tile->pixmap =
945 fz_new_pixmap_with_bbox_and_data (state.ctx, state.colorspace,
946 &bbox, pbo->ptr);
947 tile->pbo = pbo;
949 else {
950 tile->pixmap =
951 fz_new_pixmap_with_bbox (state.ctx, state.colorspace, &bbox);
955 tile->w = w;
956 tile->h = h;
957 if (!page->u.ptr || ((w < 128 && h < 128) || !OBSCURED (page, bbox))) {
958 clearpixmap (tile->pixmap);
960 dev = fz_new_draw_device (state.ctx, tile->pixmap);
961 ctm = pagectm (page);
962 fz_rect_from_irect (&rect, &bbox);
963 fz_run_display_list (page->dlist, dev, &ctm, &rect, NULL);
964 fz_free_device (dev);
966 return tile;
969 #ifdef CACHE_PAGEREFS
970 /* modified mupdf/source/pdf/pdf-page.c:pdf_lookup_page_loc_imp
971 thanks to Robin Watts */
972 static void
973 pdf_collect_pages(pdf_document *doc, pdf_obj *node)
975 fz_context *ctx = doc->ctx;
976 pdf_obj *kids;
977 int i, len;
979 if (state.pdflut.idx == state.pagecount) return;
981 kids = pdf_dict_gets (node, "Kids");
982 len = pdf_array_len (kids);
984 if (len == 0)
985 fz_throw (ctx, FZ_ERROR_GENERIC, "Malformed pages tree");
987 if (pdf_mark_obj (node))
988 fz_throw (ctx, FZ_ERROR_GENERIC, "cycle in page tree");
989 for (i = 0; i < len; i++) {
990 pdf_obj *kid = pdf_array_get (kids, i);
991 char *type = pdf_to_name (pdf_dict_gets (kid, "Type"));
992 if (*type
993 ? !strcmp (type, "Pages")
994 : pdf_dict_gets (kid, "Kids")
995 && !pdf_dict_gets (kid, "MediaBox")) {
996 pdf_collect_pages (doc, kid);
998 else {
999 if (*type
1000 ? strcmp (type, "Page") != 0
1001 : !pdf_dict_gets (kid, "MediaBox"))
1002 fz_warn (ctx, "non-page object in page tree (%s)", type);
1003 state.pdflut.objs[state.pdflut.idx++] = pdf_keep_obj (kid);
1006 pdf_unmark_obj (node);
1009 static void
1010 pdf_load_page_objs (pdf_document *doc)
1012 pdf_obj *root = pdf_dict_gets (pdf_trailer (doc), "Root");
1013 pdf_obj *node = pdf_dict_gets (root, "Pages");
1015 if (!node)
1016 fz_throw (doc->ctx, FZ_ERROR_GENERIC, "cannot find page tree");
1018 state.pdflut.idx = 0;
1019 pdf_collect_pages (doc, node);
1021 #endif
1023 static void initpdims (void)
1025 double start, end;
1026 FILE *trimf = NULL;
1027 fz_rect rootmediabox;
1028 int pageno, trim, show;
1029 int trimw = 0, cxcount;
1031 start = now ();
1033 if (state.trimmargins && state.trimcachepath) {
1034 trimf = fopen (state.trimcachepath, "rb");
1035 if (!trimf) {
1036 trimf = fopen (state.trimcachepath, "wb");
1037 trimw = 1;
1041 if (state.trimmargins || state.type == DPDF || !state.cxack)
1042 cxcount = state.pagecount;
1043 else
1044 cxcount = MIN (state.pagecount, 1);
1046 if (state.type == DPDF) {
1047 pdf_obj *obj;
1048 obj = pdf_dict_getp (pdf_trailer (state.u.pdf),
1049 "Root/Pages/MediaBox");
1050 pdf_to_rect (state.ctx, obj, &rootmediabox);
1053 #ifdef CACHE_PAGEREFS
1054 if (state.type == DPDF
1055 && (!state.pdflut.objs || state.pdflut.pdf != state.u.pdf)) {
1056 state.pdflut.objs = calloc (sizeof (*state.pdflut.objs), cxcount);
1057 if (!state.pdflut.objs) {
1058 err (1, "malloc pageobjs %zu %d %zu failed",
1059 sizeof (*state.pdflut.objs), cxcount,
1060 sizeof (*state.pdflut.objs) * cxcount);
1062 state.pdflut.count = cxcount;
1063 pdf_load_page_objs (state.u.pdf);
1064 state.pdflut.pdf = state.u.pdf;
1066 #endif
1068 for (pageno = 0; pageno < cxcount; ++pageno) {
1069 int rotate = 0;
1070 struct pagedim *p;
1071 fz_rect mediabox;
1073 switch (state.type) {
1074 case DPDF: {
1075 pdf_document *pdf = state.u.pdf;
1076 pdf_obj *pageref, *pageobj;
1078 #ifdef CACHE_PAGEREFS
1079 pageref = state.pdflut.objs[pageno];
1080 #else
1081 pageref = pdf_lookup_page_obj (pdf, pageno);
1082 show = !state.trimmargins && pageno % 20 == 0;
1083 if (show) {
1084 printd ("progress %f Gathering dimensions %d",
1085 (double) (pageno) / state.pagecount,
1086 pageno);
1088 #endif
1089 pageobj = pdf_resolve_indirect (pageref);
1091 if (state.trimmargins) {
1092 pdf_obj *obj;
1093 pdf_page *page;
1095 fz_try (state.ctx) {
1096 page = pdf_load_page (pdf, pageno);
1097 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
1098 trim = state.trimanew || !obj;
1099 if (trim) {
1100 fz_rect rect;
1101 fz_matrix ctm;
1102 fz_device *dev;
1104 dev = fz_new_bbox_device (state.ctx, &rect);
1105 dev->hints |= FZ_IGNORE_SHADE;
1106 fz_invert_matrix (&ctm, &page->ctm);
1107 pdf_run_page (pdf, page, dev, &fz_identity, NULL);
1108 fz_free_device (dev);
1110 rect.x0 += state.trimfuzz.x0;
1111 rect.x1 += state.trimfuzz.x1;
1112 rect.y0 += state.trimfuzz.y0;
1113 rect.y1 += state.trimfuzz.y1;
1114 fz_transform_rect (&rect, &ctm);
1115 fz_intersect_rect (&rect, &page->mediabox);
1117 if (fz_is_empty_rect (&rect)) {
1118 mediabox = page->mediabox;
1120 else {
1121 mediabox = rect;
1124 obj = pdf_new_array (pdf, 4);
1125 pdf_array_push (obj, pdf_new_real (pdf, mediabox.x0));
1126 pdf_array_push (obj, pdf_new_real (pdf, mediabox.y0));
1127 pdf_array_push (obj, pdf_new_real (pdf, mediabox.x1));
1128 pdf_array_push (obj, pdf_new_real (pdf, mediabox.y1));
1129 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
1131 else {
1132 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
1133 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
1134 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
1135 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
1138 rotate = page->rotate;
1139 pdf_free_page (pdf, page);
1141 show = trim ? pageno % 5 == 0 : pageno % 20 == 0;
1142 if (show) {
1143 printd ("progress %f Trimming %d",
1144 (double) (pageno + 1) / state.pagecount,
1145 pageno + 1);
1148 fz_catch (state.ctx) {
1149 fprintf (stderr, "failed to load page %d\n", pageno+1);
1152 else {
1153 int empty = 0;
1154 fz_rect cropbox;
1156 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "MediaBox"),
1157 &mediabox);
1158 if (fz_is_empty_rect (&mediabox)) {
1159 mediabox.x0 = 0;
1160 mediabox.y0 = 0;
1161 mediabox.x1 = 612;
1162 mediabox.y1 = 792;
1163 empty = 1;
1166 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "CropBox"),
1167 &cropbox);
1168 if (!fz_is_empty_rect (&cropbox)) {
1169 if (empty) {
1170 mediabox = cropbox;
1172 else {
1173 fz_intersect_rect (&mediabox, &cropbox);
1176 else {
1177 if (empty) {
1178 if (fz_is_empty_rect (&rootmediabox)) {
1179 fprintf (stderr,
1180 "cannot find page size for page %d\n",
1181 pageno+1);
1183 else {
1184 mediabox = rootmediabox;
1188 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
1190 break;
1193 case DXPS:
1195 xps_page *page;
1197 fz_try (state.ctx) {
1198 page = xps_load_page (state.u.xps, pageno);
1199 xps_bound_page (state.u.xps, page, &mediabox);
1200 rotate = 0;
1201 if (state.trimmargins) {
1202 fz_rect rect;
1203 fz_device *dev;
1205 dev = fz_new_bbox_device (state.ctx, &rect);
1206 dev->hints |= FZ_IGNORE_SHADE;
1207 xps_run_page (state.u.xps, page, dev,
1208 &fz_identity, NULL);
1209 fz_free_device (dev);
1211 rect.x0 += state.trimfuzz.x0;
1212 rect.x1 += state.trimfuzz.x1;
1213 rect.y0 += state.trimfuzz.y0;
1214 rect.y1 += state.trimfuzz.y1;
1215 fz_intersect_rect (&rect, &mediabox);
1217 if (!fz_is_empty_rect (&rect)) {
1218 mediabox = rect;
1221 xps_free_page (state.u.xps, page);
1222 if (!state.cxack) {
1223 printd ("progress %f loading %d",
1224 (double) (pageno + 1) / state.pagecount,
1225 pageno + 1);
1228 fz_catch (state.ctx) {
1231 break;
1233 case DCBZ:
1235 rotate = 0;
1236 mediabox.x0 = 0;
1237 mediabox.y0 = 0;
1238 mediabox.x1 = 900;
1239 mediabox.y1 = 900;
1240 if (state.trimmargins && trimw) {
1241 cbz_page *page;
1243 fz_try (state.ctx) {
1244 page = cbz_load_page (state.u.cbz, pageno);
1245 cbz_bound_page (state.u.cbz, page, &mediabox);
1246 cbz_free_page (state.u.cbz, page);
1247 printd ("progress %f Trimming %d",
1248 (double) (pageno + 1) / state.pagecount,
1249 pageno + 1);
1251 fz_catch (state.ctx) {
1252 fprintf (stderr, "failed to load page %d\n", pageno+1);
1254 if (trimf) {
1255 int n = fwrite (&mediabox, sizeof (mediabox), 1, trimf);
1256 if (n - 1) {
1257 err (1, "fwrite trim mediabox");
1261 else {
1262 if (trimf) {
1263 int n = fread (&mediabox, sizeof (mediabox), 1, trimf);
1264 if (n - 1) {
1265 err (1, "fread trim mediabox");
1268 else {
1269 if (state.cxack) {
1270 cbz_page *page;
1271 fz_try (state.ctx) {
1272 page = cbz_load_page (state.u.cbz, pageno);
1273 cbz_bound_page (state.u.cbz, page, &mediabox);
1274 cbz_free_page (state.u.cbz, page);
1276 fz_catch (state.ctx) {
1277 fprintf (stderr, "failed to load cbz page\n");
1283 break;
1285 case DIMG:
1287 image_page *page;
1289 rotate = 0;
1290 mediabox.x0 = 0;
1291 mediabox.y0 = 0;
1292 mediabox.x1 = 900;
1293 mediabox.y1 = 900;
1295 fz_try (state.ctx) {
1296 page = image_load_page (state.u.img, pageno);
1297 image_bound_page (state.u.img, page, &mediabox);
1298 image_free_page (state.u.img, page);
1300 fz_catch (state.ctx) {
1301 fprintf (stderr, "failed to load page %d\n", pageno+1);
1304 break;
1306 default:
1307 ARSERT (0 && state.type);
1310 if (state.pagedimcount == 0
1311 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
1312 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
1313 size_t size;
1315 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
1316 state.pagedims = realloc (state.pagedims, size);
1317 if (!state.pagedims) {
1318 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
1319 size, state.pagedimcount + 1);
1322 p = &state.pagedims[state.pagedimcount++];
1323 p->rotate = rotate;
1324 p->mediabox = mediabox;
1325 p->pageno = pageno;
1328 end = now ();
1329 if (state.trimmargins) {
1330 printd ("progress 1 Trimmed %d pages in %f seconds",
1331 state.pagecount, end - start);
1333 else {
1334 if (state.type == DPDF)
1335 printd ("progress 1 Processed %d pages in %f seconds",
1336 state.pagecount, end - start);
1337 else
1338 printd ("vmsg Processed %d pages in %f seconds",
1339 state.pagecount, end - start);
1341 state.trimanew = 0;
1342 if (trimf) {
1343 if (fclose (trimf)) {
1344 err (1, "fclose");
1349 static void layout (void)
1351 int pindex;
1352 fz_rect box;
1353 fz_matrix ctm, rm;
1354 struct pagedim *p = p;
1355 double zw, w, maxw = 0.0, zoom = zoom;
1357 if (state.pagedimcount == 0) return;
1359 switch (state.fitmodel) {
1360 case FitProportional:
1361 for (pindex = 0; pindex < state.pagedimcount; ++pindex) {
1362 double x0, x1;
1364 p = &state.pagedims[pindex];
1365 fz_rotate (&rm, p->rotate + state.rotate);
1366 box = p->mediabox;
1367 fz_transform_rect (&box, &rm);
1369 x0 = MIN (box.x0, box.x1);
1370 x1 = MAX (box.x0, box.x1);
1372 w = x1 - x0;
1373 maxw = MAX (w, maxw);
1374 zoom = state.w / maxw;
1376 break;
1378 case FitPage:
1379 maxw = state.w;
1380 break;
1382 case FitWidth:
1383 break;
1385 default:
1386 ARSERT (0 && state.fitmodel);
1389 for (pindex = 0; pindex < state.pagedimcount; ++pindex) {
1390 fz_rect rect;
1391 fz_matrix tm, sm;
1393 p = &state.pagedims[pindex];
1394 fz_rotate (&ctm, state.rotate);
1395 fz_rotate (&rm, p->rotate + state.rotate);
1396 box = p->mediabox;
1397 fz_transform_rect (&box, &rm);
1398 w = box.x1 - box.x0;
1399 switch (state.fitmodel) {
1400 case FitProportional:
1401 p->left = ((maxw - w) * zoom) / 2.0;
1402 break;
1403 case FitPage:
1405 double zh, h;
1406 zw = maxw / w;
1407 h = box.y1 - box.y0;
1408 zh = state.h / h;
1409 zoom = MIN (zw, zh);
1410 p->left = (maxw - (w * zoom)) / 2.0;
1412 break;
1413 case FitWidth:
1414 p->left = 0;
1415 zoom = state.w / w;
1416 break;
1419 fz_scale (&p->zoomctm, zoom, zoom);
1420 fz_concat (&ctm, &p->zoomctm, &ctm);
1422 fz_rotate (&rm, p->rotate);
1423 p->pagebox = p->mediabox;
1424 fz_transform_rect (&p->pagebox, &rm);
1425 p->pagebox.x1 -= p->pagebox.x0;
1426 p->pagebox.y1 -= p->pagebox.y0;
1427 p->pagebox.x0 = 0;
1428 p->pagebox.y0 = 0;
1429 rect = p->pagebox;
1430 fz_transform_rect (&rect, &ctm);
1431 fz_round_rect (&p->bounds, &rect);
1432 p->ctm = ctm;
1434 fz_translate (&tm, 0, -p->mediabox.y1);
1435 fz_scale (&sm, zoom, -zoom);
1436 fz_concat (&ctm, &tm, &sm);
1437 fz_concat (&p->lctm, &ctm, &rm);
1439 p->tctmready = 0;
1442 do {
1443 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1444 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1445 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1446 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1447 int boundw = x1 - x0;
1448 int boundh = y1 - y0;
1450 printd ("pdim %d %d %d %d", p->pageno, boundw, boundh, p->left);
1451 } while (p-- != state.pagedims);
1454 static
1455 struct anchor { int n; int x; int y; int w; int h; }
1456 desttoanchor (fz_link_dest *dest)
1458 int i;
1459 struct anchor a;
1460 struct pagedim *pdim = state.pagedims;
1462 a.n = -1;
1463 a.x = 0;
1464 a.y = 0;
1465 for (i = 0; i < state.pagedimcount; ++i) {
1466 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1467 break;
1468 pdim = &state.pagedims[i];
1470 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1471 fz_point p;
1472 if (dest->ld.gotor.flags & fz_link_flag_l_valid)
1473 p.x = dest->ld.gotor.lt.x;
1474 else
1475 p.x = 0.0;
1476 p.y = dest->ld.gotor.lt.y;
1477 fz_transform_point (&p, &pdim->lctm);
1478 a.x = p.x;
1479 a.y = p.y;
1481 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1482 double x0, x1, y0, y1;
1484 x0 = MIN (pdim->bounds.x0, pdim->bounds.x1);
1485 x1 = MAX (pdim->bounds.x0, pdim->bounds.x1);
1486 a.w = x1 - x0;
1487 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1488 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1489 a.h = y1 - y0;
1490 a.n = dest->ld.gotor.page;
1492 return a;
1495 static void recurse_outline (fz_outline *outline, int level)
1497 while (outline) {
1498 switch (outline->dest.kind) {
1499 case FZ_LINK_GOTO:
1501 struct anchor a = desttoanchor (&outline->dest);
1503 if (a.n >= 0) {
1504 printd ("o %d %d %d %d %s",
1505 level, a.n, a.y, a.h, outline->title);
1508 break;
1510 case FZ_LINK_URI:
1511 printd ("ou %d %" FMT_s " %s %s", level,
1512 strlen (outline->title), outline->title,
1513 outline->dest.ld.uri.uri);
1514 break;
1516 case FZ_LINK_NONE:
1517 printd ("on %d %s", level, outline->title);
1518 break;
1520 default:
1521 printd ("emsg Unhandled outline kind %d for %s\n",
1522 outline->dest.kind, outline->title);
1523 break;
1525 if (outline->down) {
1526 recurse_outline (outline->down, level + 1);
1528 outline = outline->next;
1532 static void process_outline (void)
1534 fz_outline *outline;
1536 if (!state.needoutline || !state.pagedimcount) return;
1538 state.needoutline = 0;
1539 switch (state.type) {
1540 case DPDF:
1541 outline = pdf_load_outline (state.u.pdf);
1542 break;
1543 case DXPS:
1544 outline = xps_load_outline (state.u.xps);
1545 break;
1546 default:
1547 outline = NULL;
1548 break;
1550 if (outline) {
1551 recurse_outline (outline, 0);
1552 fz_free_outline (state.ctx, outline);
1556 static char *strofspan (fz_text_span *span)
1558 char *p;
1559 char utf8[10];
1560 fz_text_char *ch;
1561 size_t size = 0, cap = 80;
1563 p = malloc (cap + 1);
1564 if (!p) return NULL;
1566 for (ch = span->text; ch < span->text + span->len; ++ch) {
1567 int n = fz_runetochar (utf8, ch->c);
1568 if (size + n > cap) {
1569 cap *= 2;
1570 p = realloc (p, cap + 1);
1571 if (!p) return NULL;
1574 memcpy (p + size, utf8, n);
1575 size += n;
1577 p[size] = 0;
1578 return p;
1581 static int matchspan (regex_t *re, fz_text_span *span,
1582 int stop, int pageno, double start)
1584 int ret;
1585 char *p;
1586 regmatch_t rm;
1587 int a, b, c;
1588 fz_rect sb, eb;
1589 fz_point p1, p2, p3, p4;
1591 p = strofspan (span);
1592 if (!p) return -1;
1594 ret = regexec (re, p, 1, &rm, 0);
1595 if (ret) {
1596 free (p);
1597 if (ret != REG_NOMATCH) {
1598 size_t size;
1599 char errbuf[80];
1600 size = regerror (ret, re, errbuf, sizeof (errbuf));
1601 printd ("msg regexec error `%.*s'",
1602 (int) size, errbuf);
1603 return -1;
1605 return 0;
1607 else {
1608 int l = span->len;
1610 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1611 c += fz_runelen (span->text[a].c);
1613 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1614 c += fz_runelen (span->text[b].c);
1617 if (fz_runelen (span->text[b].c) > 1) {
1618 b = MAX (0, b-1);
1621 fz_text_char_bbox (&sb, span, a);
1622 fz_text_char_bbox (&eb, span, b);
1624 p1.x = sb.x0;
1625 p1.y = sb.y0;
1626 p2.x = eb.x1;
1627 p2.y = sb.y0;
1628 p3.x = eb.x1;
1629 p3.y = eb.y1;
1630 p4.x = sb.x0;
1631 p4.y = eb.y1;
1633 if (!stop) {
1634 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1635 pageno, 1,
1636 p1.x, p1.y,
1637 p2.x, p2.y,
1638 p3.x, p3.y,
1639 p4.x, p4.y);
1641 printd ("progress 1 found at %d `%.*s' in %f sec",
1642 pageno + 1, (int) (rm.rm_eo - rm.rm_so), &p[rm.rm_so],
1643 now () - start);
1645 else {
1646 printd ("match %d %d %f %f %f %f %f %f %f %f",
1647 pageno, 2,
1648 p1.x, p1.y,
1649 p2.x, p2.y,
1650 p3.x, p3.y,
1651 p4.x, p4.y);
1653 free (p);
1654 return 1;
1658 static int compareblocks (const void *l, const void *r)
1660 fz_text_block const *ls = l;
1661 fz_text_block const* rs = r;
1662 return ls->bbox.y0 - rs->bbox.y0;
1665 /* wishful thinking function */
1666 static void search (regex_t *re, int pageno, int y, int forward)
1668 int i, j;
1669 fz_matrix ctm;
1670 fz_device *tdev;
1671 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1672 fz_text_page *text;
1673 fz_text_sheet *sheet;
1674 struct pagedim *pdim, *pdimprev;
1675 int stop = 0, niters = 0;
1676 double start, end;
1678 if (!(state.type == DPDF || state.type == DXPS))
1679 return;
1681 start = now ();
1682 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1683 if (niters++ == 5) {
1684 niters = 0;
1685 if (hasdata ()) {
1686 printd ("progress 1 attention requested aborting search at %d",
1687 pageno);
1688 stop = 1;
1690 else {
1691 printd ("progress %f searching in page %d",
1692 (double) (pageno + 1) / state.pagecount,
1693 pageno);
1696 pdimprev = NULL;
1697 for (i = 0; i < state.pagedimcount; ++i) {
1698 pdim = &state.pagedims[i];
1699 if (pdim->pageno == pageno) {
1700 goto found;
1702 if (pdim->pageno > pageno) {
1703 pdim = pdimprev;
1704 goto found;
1706 pdimprev = pdim;
1708 pdim = pdimprev;
1709 found:
1711 sheet = fz_new_text_sheet (state.ctx);
1712 text = fz_new_text_page (state.ctx);
1713 tdev = fz_new_text_device (state.ctx, sheet, text);
1715 switch (state.type) {
1716 case DPDF:
1717 u.ptr = NULL;
1718 fz_try (state.ctx) {
1719 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1720 trimctm (u.pdfpage, pdim - state.pagedims);
1721 fz_concat (&ctm, &pdim->tctm, &pdim->zoomctm);
1722 fz_begin_page (tdev, &fz_infinite_rect, &ctm);
1723 pdf_run_page (state.u.pdf, u.pdfpage, tdev, &ctm, NULL);
1724 fz_end_page (tdev);
1726 fz_catch (state.ctx) {
1727 fz_free_device (tdev);
1728 u.ptr = NULL;
1729 goto nextiter;
1731 break;
1733 case DXPS:
1734 u.xpspage = xps_load_page (state.u.xps, pageno);
1735 xps_run_page (state.u.xps, u.xpspage, tdev, &pdim->ctm, NULL);
1736 break;
1738 default:
1739 ARSERT (0 && state.type);
1742 qsort (text->blocks, text->len, sizeof (*text->blocks), compareblocks);
1743 fz_free_device (tdev);
1745 for (j = 0; j < text->len; ++j) {
1746 int k;
1747 fz_page_block *pb;
1748 fz_text_block *block;
1750 pb = &text->blocks[forward ? j : text->len - 1 - j];
1751 if (pb->type != FZ_PAGE_BLOCK_TEXT) continue;
1752 block = pb->u.text;
1754 for (k = 0; k < block->len; ++k) {
1755 fz_text_line *line;
1756 fz_text_span *span;
1758 if (forward) {
1759 line = &block->lines[k];
1760 if (line->bbox.y0 < y + 1) continue;
1762 else {
1763 line = &block->lines[block->len - 1 - k];
1764 if (line->bbox.y0 > y - 1) continue;
1767 for (span = line->first_span; span; span = span->next) {
1768 switch (matchspan (re, span, stop, pageno, start)) {
1769 case 0: break;
1770 case 1: stop = 1; break;
1771 case -1: stop = 1; goto endloop;
1776 nextiter:
1777 if (forward) {
1778 pageno += 1;
1779 y = 0;
1781 else {
1782 pageno -= 1;
1783 y = INT_MAX;
1785 endloop:
1786 fz_free_text_page (state.ctx, text);
1787 fz_free_text_sheet (state.ctx, sheet);
1788 if (u.ptr) {
1789 state.freepage (u.ptr);
1792 end = now ();
1793 if (!stop) {
1794 printd ("progress 1 no matches %f sec", end - start);
1796 printd ("clearrects");
1799 static void set_tex_params (int colorspace)
1801 union {
1802 unsigned char b;
1803 unsigned int s;
1804 } endianness = {1};
1806 switch (colorspace) {
1807 case 0:
1808 state.texiform = GL_RGBA8;
1809 state.texform = GL_RGBA;
1810 state.texty = GL_UNSIGNED_BYTE;
1811 state.colorspace = fz_device_rgb (state.ctx);
1812 break;
1813 case 1:
1814 state.texiform = GL_RGBA8;
1815 state.texform = GL_BGRA;
1816 state.texty = endianness.s > 1
1817 ? GL_UNSIGNED_INT_8_8_8_8
1818 : GL_UNSIGNED_INT_8_8_8_8_REV;
1819 state.colorspace = fz_device_bgr (state.ctx);
1820 break;
1821 case 2:
1822 state.texiform = GL_LUMINANCE_ALPHA;
1823 state.texform = GL_LUMINANCE_ALPHA;
1824 state.texty = GL_UNSIGNED_BYTE;
1825 state.colorspace = fz_device_gray (state.ctx);
1826 break;
1827 default:
1828 errx (1, "invalid colorspce %d", colorspace);
1832 static void realloctexts (int texcount)
1834 size_t size;
1836 if (texcount == state.texcount) return;
1838 if (texcount < state.texcount) {
1839 glDeleteTextures (state.texcount - texcount,
1840 state.texids + texcount);
1843 size = texcount * sizeof (*state.texids);
1844 state.texids = realloc (state.texids, size);
1845 if (!state.texids) {
1846 err (1, "realloc texids %" FMT_s, size);
1849 size = texcount * sizeof (*state.texowners);
1850 state.texowners = realloc (state.texowners, size);
1851 if (!state.texowners) {
1852 err (1, "realloc texowners %" FMT_s, size);
1854 if (texcount > state.texcount) {
1855 int i;
1857 glGenTextures (texcount - state.texcount,
1858 state.texids + state.texcount);
1859 for (i = state.texcount; i < texcount; ++i) {
1860 state.texowners[i].w = -1;
1861 state.texowners[i].slice = NULL;
1864 state.texcount = texcount;
1865 state.texindex = 0;
1868 static char *mbtoutf8 (char *s)
1870 char *p, *r;
1871 wchar_t *tmp;
1872 size_t i, ret, len;
1874 len = mbstowcs (NULL, s, strlen (s));
1875 if (len == 0) {
1876 return s;
1878 else {
1879 if (len == (size_t) -1) {
1880 return s;
1884 tmp = malloc (len * sizeof (wchar_t));
1885 if (!tmp) {
1886 return s;
1889 ret = mbstowcs (tmp, s, len);
1890 if (ret == (size_t) -1) {
1891 free (tmp);
1892 return s;
1895 len = 0;
1896 for (i = 0; i < ret; ++i) {
1897 len += fz_runelen (tmp[i]);
1900 p = r = malloc (len + 1);
1901 if (!r) {
1902 free (tmp);
1903 return s;
1906 for (i = 0; i < ret; ++i) {
1907 p += fz_runetochar (p, tmp[i]);
1909 *p = 0;
1910 free (tmp);
1911 return r;
1914 CAMLprim value ml_mbtoutf8 (value s_v)
1916 CAMLparam1 (s_v);
1917 CAMLlocal1 (ret_v);
1918 char *s, *r;
1920 s = String_val (s_v);
1921 r = mbtoutf8 (s);
1922 if (r == s) {
1923 ret_v = s_v;
1925 else {
1926 ret_v = caml_copy_string (r);
1927 free (r);
1929 CAMLreturn (ret_v);
1932 static void * mainloop (void UNUSED_ATTR *unused)
1934 char *p = NULL;
1935 int len, ret, oldlen = 0;
1937 for (;;) {
1938 len = readlen ();
1939 if (len == 0) {
1940 errx (1, "readlen returned 0");
1943 if (oldlen < len + 1) {
1944 p = realloc (p, len + 1);
1945 if (!p) {
1946 err (1, "realloc %d failed", len + 1);
1948 oldlen = len + 1;
1950 readdata (p, len);
1951 p[len] = 0;
1953 if (!strncmp ("open", p, 4)) {
1954 int wthack, off, ok = 0;
1955 char *password;
1956 char *filename;
1957 char *utf8filename;
1958 size_t filenamelen;
1960 ret = sscanf (p + 5, " %d %d %n", &wthack, &state.cxack, &off);
1961 if (ret != 2) {
1962 errx (1, "malformed open `%.*s' ret=%d", len, p, ret);
1965 filename = p + 5 + off;
1966 filenamelen = strlen (filename);
1967 password = filename + filenamelen + 1;
1969 lock ("open");
1970 fz_try (state.ctx) {
1971 ok = openxref (filename, password);
1973 fz_catch (state.ctx) {
1974 utf8filename = mbtoutf8 (filename);
1975 printd ("msg Could not open %s", utf8filename);
1977 if (ok) {
1978 pdfinfo ();
1979 initpdims ();
1981 unlock ("open");
1983 if (ok) {
1984 if (!wthack) {
1985 utf8filename = mbtoutf8 (filename);
1986 printd ("msg Opened %s (press h/F1 to get help)",
1987 utf8filename);
1988 if (utf8filename != filename) {
1989 free (utf8filename);
1992 state.needoutline = 1;
1995 else if (!strncmp ("cs", p, 2)) {
1996 int i, colorspace;
1998 ret = sscanf (p + 2, " %d", &colorspace);
1999 if (ret != 1) {
2000 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
2002 lock ("cs");
2003 set_tex_params (colorspace);
2004 for (i = 0; i < state.texcount; ++i) {
2005 state.texowners[i].w = -1;
2006 state.texowners[i].slice = NULL;
2008 unlock ("cs");
2010 else if (!strncmp ("freepage", p, 8)) {
2011 void *ptr;
2013 ret = sscanf (p + 8, " %" SCN_ptr, SCN_ptr_cast (&ptr));
2014 if (ret != 1) {
2015 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
2017 freepage (ptr);
2019 else if (!strncmp ("freetile", p, 8)) {
2020 void *ptr;
2022 ret = sscanf (p + 8, " %" SCN_ptr, SCN_ptr_cast (&ptr));
2023 if (ret != 1) {
2024 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
2026 freetile (ptr);
2028 else if (!strncmp ("search", p, 6)) {
2029 int icase, pageno, y, len2, forward;
2030 char *pattern;
2031 regex_t re;
2033 ret = sscanf (p + 6, " %d %d %d %d,%n",
2034 &icase, &pageno, &y, &forward, &len2);
2035 if (ret != 4) {
2036 errx (1, "malformed search `%s' ret=%d", p, ret);
2039 pattern = p + 6 + len2;
2040 ret = regcomp (&re, pattern,
2041 REG_EXTENDED | (icase ? REG_ICASE : 0));
2042 if (ret) {
2043 char errbuf[80];
2044 size_t size;
2046 size = regerror (ret, &re, errbuf, sizeof (errbuf));
2047 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
2049 else {
2050 search (&re, pageno, y, forward);
2051 regfree (&re);
2054 else if (!strncmp ("geometry", p, 8)) {
2055 int w, h, fitmodel;
2057 printd ("clear");
2058 ret = sscanf (p + 8, " %d %d %d", &w, &h, &fitmodel);
2059 if (ret != 3) {
2060 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
2063 lock ("geometry");
2064 state.h = h;
2065 if (w != state.w) {
2066 int i;
2067 state.w = w;
2068 for (i = 0; i < state.texcount; ++i) {
2069 state.texowners[i].slice = NULL;
2072 state.fitmodel = fitmodel;
2073 layout ();
2074 process_outline ();
2076 state.gen++;
2077 unlock ("geometry");
2078 printd ("continue %d", state.pagecount);
2080 else if (!strncmp ("reqlayout", p, 9)) {
2081 char *nameddest;
2082 int rotate, off, h;
2083 unsigned int fitmodel;
2085 printd ("clear");
2086 ret = sscanf (p + 9, " %d %u %d %n",
2087 &rotate, &fitmodel, &h, &off);
2088 if (ret != 3) {
2089 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
2091 lock ("reqlayout");
2092 if (state.rotate != rotate || state.fitmodel != fitmodel) {
2093 state.gen += 1;
2095 state.rotate = rotate;
2096 state.fitmodel = fitmodel;
2097 state.h = h;
2098 layout ();
2099 process_outline ();
2101 nameddest = p + 9 + off;
2102 if (state.type == DPDF && nameddest && *nameddest) {
2103 struct anchor a;
2104 fz_link_dest dest;
2105 pdf_obj *needle, *obj;
2107 needle = pdf_new_string (state.u.pdf, nameddest,
2108 strlen (nameddest));
2109 obj = pdf_lookup_dest (state.u.pdf, needle);
2110 if (obj) {
2111 dest = pdf_parse_link_dest (state.u.pdf, FZ_LINK_GOTO, obj);
2113 a = desttoanchor (&dest);
2114 if (a.n >= 0) {
2115 printd ("a %d %d %d", a.n, a.x, a.y);
2117 else {
2118 printd ("emsg failed to parse destination `%s'\n",
2119 nameddest);
2122 else {
2123 printd ("emsg destination `%s' not found\n",
2124 nameddest);
2126 pdf_drop_obj (needle);
2129 state.gen++;
2130 unlock ("reqlayout");
2131 printd ("continue %d", state.pagecount);
2133 else if (!strncmp ("page", p, 4)) {
2134 double a, b;
2135 struct page *page;
2136 int pageno, pindex;
2138 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
2139 if (ret != 2) {
2140 errx (1, "bad page line `%.*s' ret=%d", len, p, ret);
2143 lock ("page");
2144 a = now ();
2145 page = loadpage (pageno, pindex);
2146 b = now ();
2147 unlock ("page");
2149 printd ("page %" FMT_ptr " %f", FMT_ptr_cast (page), b - a);
2151 else if (!strncmp ("tile", p, 4)) {
2152 int x, y, w, h;
2153 struct page *page;
2154 struct tile *tile;
2155 double a, b;
2156 void *data;
2158 ret = sscanf (p + 4, " %" SCN_ptr " %d %d %d %d %" SCN_ptr,
2159 SCN_ptr_cast (&page), &x, &y, &w, &h,
2160 SCN_ptr_cast (&data));
2161 if (ret != 6) {
2162 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
2165 lock ("tile");
2166 a = now ();
2167 tile = rendertile (page, x, y, w, h, data);
2168 b = now ();
2169 unlock ("tile");
2171 printd ("tile %d %d %" FMT_ptr " %u %f",
2172 x, y,
2173 FMT_ptr_cast (tile),
2174 tile->w * tile->h * tile->pixmap->n,
2175 b - a);
2177 else if (!strncmp ("trimset", p, 7)) {
2178 fz_irect fuzz;
2179 int trimmargins;
2181 ret = sscanf (p + 7, " %d %d %d %d %d",
2182 &trimmargins, &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
2183 if (ret != 5) {
2184 errx (1, "malformed trimset `%.*s' ret=%d", len, p, ret);
2186 lock ("trimset");
2187 state.trimmargins = trimmargins;
2188 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
2189 state.trimanew = 1;
2190 state.trimfuzz = fuzz;
2192 unlock ("trimset");
2194 else if (!strncmp ("settrim", p, 7)) {
2195 fz_irect fuzz;
2196 int trimmargins;
2198 ret = sscanf (p + 7, " %d %d %d %d %d",
2199 &trimmargins, &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
2200 if (ret != 5) {
2201 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
2203 printd ("clear");
2204 lock ("settrim");
2205 state.trimmargins = trimmargins;
2206 state.needoutline = 1;
2207 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
2208 state.trimanew = 1;
2209 state.trimfuzz = fuzz;
2211 state.pagedimcount = 0;
2212 free (state.pagedims);
2213 state.pagedims = NULL;
2214 initpdims ();
2215 layout ();
2216 process_outline ();
2217 unlock ("settrim");
2218 printd ("continue %d", state.pagecount);
2220 else if (!strncmp ("sliceh", p, 6)) {
2221 int h;
2223 ret = sscanf (p + 6, " %d", &h);
2224 if (ret != 1) {
2225 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
2227 if (h != state.sliceheight) {
2228 int i;
2230 state.sliceheight = h;
2231 for (i = 0; i < state.texcount; ++i) {
2232 state.texowners[i].w = -1;
2233 state.texowners[i].h = -1;
2234 state.texowners[i].slice = NULL;
2238 else if (!strncmp ("interrupt", p, 9)) {
2239 printd ("vmsg interrupted");
2241 else {
2242 errx (1, "unknown command %.*s", len, p);
2245 return 0;
2248 CAMLprim value ml_realloctexts (value texcount_v)
2250 CAMLparam1 (texcount_v);
2251 int ok;
2253 if (trylock ("ml_realloctexts")) {
2254 ok = 0;
2255 goto done;
2257 realloctexts (Int_val (texcount_v));
2258 ok = 1;
2259 unlock ("ml_realloctexts");
2261 done:
2262 CAMLreturn (Val_bool (ok));
2265 static void recti (int x0, int y0, int x1, int y1)
2267 GLfloat *v = state.vertices;
2269 glVertexPointer (2, GL_FLOAT, 0, v);
2270 v[0] = x0; v[1] = y0;
2271 v[2] = x1; v[3] = y0;
2272 v[4] = x0; v[5] = y1;
2273 v[6] = x1; v[7] = y1;
2274 glDrawArrays (GL_TRIANGLE_STRIP, 0, 4);
2277 static void showsel (struct page *page, int ox, int oy)
2279 int seen = 0;
2280 fz_irect bbox;
2281 fz_rect rect;
2282 fz_text_line *line;
2283 fz_page_block *pageb;
2284 fz_text_block *block;
2285 struct mark first, last;
2286 unsigned char selcolor[] = {15,15,15,140};
2288 first = page->fmark;
2289 last = page->lmark;
2291 if (!first.span || !last.span) return;
2293 glEnable (GL_BLEND);
2294 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
2295 glColor4ubv (selcolor);
2297 ox += state.pagedims[page->pdimno].bounds.x0;
2298 oy += state.pagedims[page->pdimno].bounds.y0;
2299 for (pageb = page->text->blocks;
2300 pageb < page->text->blocks + page->text->len;
2301 ++pageb) {
2302 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
2303 block = pageb->u.text;
2305 for (line = block->lines;
2306 line < block->lines + block->len;
2307 ++line) {
2308 fz_text_span *span;
2309 rect = fz_empty_rect;
2311 for (span = line->first_span; span; span = span->next) {
2312 int i, j, k;
2313 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
2315 j = 0;
2316 k = span->len - 1;
2318 if (span == page->fmark.span && span == page->lmark.span) {
2319 seen = 1;
2320 j = MIN (first.i, last.i);
2321 k = MAX (first.i, last.i);
2323 else {
2324 if (span == first.span) {
2325 seen = 1;
2326 j = first.i;
2328 else if (span == last.span) {
2329 seen = 1;
2330 k = last.i;
2334 if (seen) {
2335 for (i = j; i <= k; ++i) {
2336 fz_rect bbox1;
2337 fz_union_rect (&rect,
2338 fz_text_char_bbox (&bbox1, span, i));
2340 fz_round_rect (&bbox, &rect);
2341 lprintf ("%d %d %d %d oy=%d ox=%d\n",
2342 bbox.x0,
2343 bbox.y0,
2344 bbox.x1,
2345 bbox.y1,
2346 oy, ox);
2348 recti (bbox.x0 + ox, bbox.y0 + oy,
2349 bbox.x1 + ox, bbox.y1 + oy);
2350 if (span == last.span) {
2351 goto done;
2353 rect = fz_empty_rect;
2358 done:
2359 glDisable (GL_BLEND);
2362 #include "glfont.c"
2364 static void stipplerect (fz_matrix *m,
2365 fz_point *p1,
2366 fz_point *p2,
2367 fz_point *p3,
2368 fz_point *p4,
2369 GLfloat *texcoords,
2370 GLfloat *vertices)
2372 fz_transform_point (p1, m);
2373 fz_transform_point (p2, m);
2374 fz_transform_point (p3, m);
2375 fz_transform_point (p4, m);
2377 float w, h, s, t;
2379 w = p2->x - p1->x;
2380 h = p2->y - p1->y;
2381 t = sqrtf (w*w + h*h) * .25f;
2383 w = p3->x - p2->x;
2384 h = p3->y - p2->y;
2385 s = sqrtf (w*w + h*h) * .25f;
2387 texcoords[0] = 0; vertices[0] = p1->x; vertices[1] = p1->y;
2388 texcoords[1] = t; vertices[2] = p2->x; vertices[3] = p2->y;
2390 texcoords[2] = 0; vertices[4] = p2->x; vertices[5] = p2->y;
2391 texcoords[3] = s; vertices[6] = p3->x; vertices[7] = p3->y;
2393 texcoords[4] = 0; vertices[8] = p3->x; vertices[9] = p3->y;
2394 texcoords[5] = t; vertices[10] = p4->x; vertices[11] = p4->y;
2396 texcoords[6] = 0; vertices[12] = p4->x; vertices[13] = p4->y;
2397 texcoords[7] = s; vertices[14] = p1->x; vertices[15] = p1->y;
2399 glDrawArrays (GL_LINES, 0, 8);
2402 static void highlightlinks (struct page *page, int xoff, int yoff)
2404 int i;
2405 fz_matrix ctm, tm, pm;
2406 fz_link *link, *links;
2407 GLfloat *texcoords = state.texcoords;
2408 GLfloat *vertices = state.vertices;
2410 switch (page->type) {
2411 case DPDF:
2412 links = page->u.pdfpage->links;
2413 break;
2415 case DXPS:
2416 links = page->u.xpspage->links;
2417 break;
2419 default:
2420 return;
2423 glEnable (GL_TEXTURE_1D);
2424 glEnable (GL_BLEND);
2425 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2426 glBindTexture (GL_TEXTURE_1D, state.stid);
2428 xoff -= state.pagedims[page->pdimno].bounds.x0;
2429 yoff -= state.pagedims[page->pdimno].bounds.y0;
2430 fz_translate (&tm, xoff, yoff);
2431 pm = pagectm (page);
2432 fz_concat (&ctm, &pm, &tm);
2434 glTexCoordPointer (1, GL_FLOAT, 0, texcoords);
2435 glVertexPointer (2, GL_FLOAT, 0, vertices);
2437 for (link = links; link; link = link->next) {
2438 fz_point p1, p2, p3, p4;
2440 p1.x = link->rect.x0;
2441 p1.y = link->rect.y0;
2443 p2.x = link->rect.x1;
2444 p2.y = link->rect.y0;
2446 p3.x = link->rect.x1;
2447 p3.y = link->rect.y1;
2449 p4.x = link->rect.x0;
2450 p4.y = link->rect.y1;
2452 switch (link->dest.kind) {
2453 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
2454 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
2455 case FZ_LINK_LAUNCH: glColor3ub (0, 255, 0); break;
2456 default: glColor3ub (0, 0, 0); break;
2458 stipplerect (&ctm, &p1, &p2, &p3, &p4, texcoords, vertices);
2461 for (i = 0; i < page->annotcount; ++i) {
2462 fz_point p1, p2, p3, p4;
2463 struct annot *annot = &page->annots[i];
2465 p1.x = annot->bbox.x0;
2466 p1.y = annot->bbox.y0;
2468 p2.x = annot->bbox.x1;
2469 p2.y = annot->bbox.y0;
2471 p3.x = annot->bbox.x1;
2472 p3.y = annot->bbox.y1;
2474 p4.x = annot->bbox.x0;
2475 p4.y = annot->bbox.y1;
2477 glColor3ub (0, 0, 128);
2478 stipplerect (&ctm, &p1, &p2, &p3, &p4, texcoords, vertices);
2481 glDisable (GL_BLEND);
2482 glDisable (GL_TEXTURE_1D);
2485 static int compareslinks (const void *l, const void *r)
2487 struct slink const *ls = l;
2488 struct slink const *rs = r;
2489 if (ls->bbox.y0 == rs->bbox.y0) {
2490 return rs->bbox.x0 - rs->bbox.x0;
2492 return ls->bbox.y0 - rs->bbox.y0;
2495 static void droptext (struct page *page)
2497 if (page->text) {
2498 fz_free_text_page (state.ctx, page->text);
2499 page->fmark.i = -1;
2500 page->lmark.i = -1;
2501 page->fmark.span = NULL;
2502 page->lmark.span = NULL;
2503 page->text = NULL;
2505 if (page->sheet) {
2506 fz_free_text_sheet (state.ctx, page->sheet);
2507 page->sheet = NULL;
2511 static void dropanots (struct page *page)
2513 if (page->annots) {
2514 free (page->annots);
2515 page->annots = NULL;
2516 page->annotcount = 0;
2520 static void ensureanots (struct page *page)
2522 fz_matrix ctm;
2523 int i, count = 0;
2524 size_t anotsize = sizeof (*page->annots);
2525 pdf_annot *annot;
2527 if (state.gen != page->agen) {
2528 dropanots (page);
2529 page->agen = state.gen;
2531 if (page->annots) return;
2533 switch (page->type) {
2534 case DPDF:
2535 trimctm (page->u.pdfpage, page->pdimno);
2536 fz_concat (&ctm,
2537 &state.pagedims[page->pdimno].tctm,
2538 &state.pagedims[page->pdimno].ctm);
2539 break;
2541 default:
2542 return;
2545 for (annot = pdf_first_annot (state.u.pdf, page->u.pdfpage);
2546 annot;
2547 annot = pdf_next_annot (state.u.pdf, annot)) {
2548 count++;
2551 if (count > 0) {
2552 page->annotcount = count;
2553 page->annots = calloc (count, anotsize);
2554 if (!page->annots) {
2555 err (1, "calloc annots %d", count);
2558 for (annot = pdf_first_annot (state.u.pdf, page->u.pdfpage), i = 0;
2559 annot;
2560 annot = pdf_next_annot (state.u.pdf, annot), i++) {
2561 fz_rect rect;
2563 pdf_bound_annot (state.u.pdf, annot, &rect);
2564 fz_transform_rect (&rect, &ctm);
2565 page->annots[i].annot = annot;
2566 fz_round_rect (&page->annots[i].bbox, &annot->pagerect);
2571 static void dropslinks (struct page *page)
2573 if (page->slinks) {
2574 free (page->slinks);
2575 page->slinks = NULL;
2576 page->slinkcount = 0;
2580 static void ensureslinks (struct page *page)
2582 fz_matrix ctm;
2583 int i, count;
2584 size_t slinksize = sizeof (*page->slinks);
2585 fz_link *link, *links;
2587 ensureanots (page);
2588 if (state.gen != page->sgen) {
2589 dropslinks (page);
2590 page->sgen = state.gen;
2592 if (page->slinks) return;
2594 switch (page->type) {
2595 case DPDF:
2596 links = page->u.pdfpage->links;
2597 trimctm (page->u.pdfpage, page->pdimno);
2598 fz_concat (&ctm,
2599 &state.pagedims[page->pdimno].tctm,
2600 &state.pagedims[page->pdimno].ctm);
2601 break;
2603 case DXPS:
2604 links = page->u.xpspage->links;
2605 ctm = state.pagedims[page->pdimno].ctm;
2606 break;
2608 default:
2609 return;
2612 count = page->annotcount;
2613 for (link = links; link; link = link->next) {
2614 count++;
2616 if (count > 0) {
2617 int j;
2619 page->slinkcount = count;
2620 page->slinks = calloc (count, slinksize);
2621 if (!page->slinks) {
2622 err (1, "calloc slinks %d", count);
2625 for (i = 0, link = links; link; ++i, link = link->next) {
2626 fz_rect rect;
2628 rect = link->rect;
2629 fz_transform_rect (&rect, &ctm);
2630 page->slinks[i].tag = SLINK;
2631 page->slinks[i].u.link = link;
2632 fz_round_rect (&page->slinks[i].bbox, &rect);
2634 for (j = 0; j < page->annotcount; ++j, ++i) {
2635 fz_rect rect;
2637 rect = page->annots[j].annot->pagerect;
2638 fz_transform_rect (&rect, &ctm);
2639 fz_round_rect (&page->slinks[i].bbox, &rect);
2641 page->slinks[i].tag = SANNOT;
2642 page->slinks[i].u.annot = page->annots[j].annot;
2644 qsort (page->slinks, count, slinksize, compareslinks);
2648 /* slightly tweaked fmt_ulong by D.J. Bernstein */
2649 static void fmt_linkn (char *s, unsigned int u)
2651 unsigned int len; unsigned int q;
2652 unsigned int zma = 'z' - 'a' + 1;
2653 len = 1; q = u;
2654 while (q > zma - 1) { ++len; q /= zma; }
2655 if (s) {
2656 s += len;
2657 do { *--s = 'a' + (u % zma) - (u < zma && len > 1); u /= zma; } while(u);
2658 /* handles u == 0 */
2660 s[len] = 0;
2663 static void highlightslinks (struct page *page, int xoff, int yoff,
2664 int noff, char *targ, int tlen, int hfsize)
2666 int i;
2667 char buf[40];
2668 struct slink *slink;
2669 double x0, y0, x1, y1, w;
2671 ensureslinks (page);
2672 glColor3ub (0xc3, 0xb0, 0x91);
2673 for (i = 0; i < page->slinkcount; ++i) {
2674 fmt_linkn (buf, i + noff);
2675 if (!tlen || !strncmp (targ, buf, tlen)) {
2676 slink = &page->slinks[i];
2678 x0 = slink->bbox.x0 + xoff - 5;
2679 y1 = slink->bbox.y0 + yoff - 5;
2680 y0 = y1 + 10 + hfsize;
2681 w = measure_string (state.face, hfsize, buf);
2682 x1 = x0 + w + 10;
2683 recti (x0, y0, x1, y1);
2687 glEnable (GL_BLEND);
2688 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2689 glEnable (GL_TEXTURE_2D);
2690 glColor3ub (0, 0, 0);
2691 for (i = 0; i < page->slinkcount; ++i) {
2692 fmt_linkn (buf, i + noff);
2693 if (!tlen || !strncmp (targ, buf, tlen)) {
2694 slink = &page->slinks[i];
2696 x0 = slink->bbox.x0 + xoff;
2697 y0 = slink->bbox.y0 + yoff + hfsize;
2698 draw_string (state.face, hfsize, x0, y0, buf);
2701 glDisable (GL_TEXTURE_2D);
2702 glDisable (GL_BLEND);
2705 static void uploadslice (struct tile *tile, struct slice *slice)
2707 int offset;
2708 struct slice *slice1;
2709 unsigned char *texdata;
2711 offset = 0;
2712 for (slice1 = tile->slices; slice != slice1; slice1++) {
2713 offset += slice1->h * tile->w * tile->pixmap->n;
2715 if (slice->texindex != -1 && slice->texindex < state.texcount
2716 && state.texowners[slice->texindex].slice == slice) {
2717 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
2719 else {
2720 int subimage = 0;
2721 int texindex = state.texindex++ % state.texcount;
2723 if (state.texowners[texindex].w == tile->w) {
2724 if (state.texowners[texindex].h >= slice->h) {
2725 subimage = 1;
2727 else {
2728 state.texowners[texindex].h = slice->h;
2731 else {
2732 state.texowners[texindex].h = slice->h;
2735 state.texowners[texindex].w = tile->w;
2736 state.texowners[texindex].slice = slice;
2737 slice->texindex = texindex;
2739 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
2740 if (tile->pbo) {
2741 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
2742 texdata = 0;
2744 else {
2745 texdata = tile->pixmap->samples;
2747 if (subimage) {
2748 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
2752 tile->w,
2753 slice->h,
2754 state.texform,
2755 state.texty,
2756 texdata+offset
2759 else {
2760 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
2762 state.texiform,
2763 tile->w,
2764 slice->h,
2766 state.texform,
2767 state.texty,
2768 texdata+offset
2771 if (tile->pbo) {
2772 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
2777 CAMLprim value ml_begintiles (value unit_v)
2779 CAMLparam1 (unit_v);
2780 glEnable (GL_TEXTURE_RECTANGLE_ARB);
2781 glTexCoordPointer (2, GL_FLOAT, 0, state.texcoords);
2782 glVertexPointer (2, GL_FLOAT, 0, state.vertices);
2783 CAMLreturn (unit_v);
2786 CAMLprim value ml_endtiles (value unit_v)
2788 CAMLparam1 (unit_v);
2789 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2790 CAMLreturn (unit_v);
2793 CAMLprim value ml_drawtile (value args_v, value ptr_v)
2795 CAMLparam2 (args_v, ptr_v);
2796 int dispx = Int_val (Field (args_v, 0));
2797 int dispy = Int_val (Field (args_v, 1));
2798 int dispw = Int_val (Field (args_v, 2));
2799 int disph = Int_val (Field (args_v, 3));
2800 int tilex = Int_val (Field (args_v, 4));
2801 int tiley = Int_val (Field (args_v, 5));
2802 char *s = String_val (ptr_v);
2803 struct tile *tile = parse_pointer ("ml_drawtile", s);
2804 int slicey, firstslice;
2805 struct slice *slice;
2806 GLfloat *texcoords = state.texcoords;
2807 GLfloat *vertices = state.vertices;
2809 firstslice = tiley / tile->sliceheight;
2810 slice = &tile->slices[firstslice];
2811 slicey = tiley % tile->sliceheight;
2813 while (disph > 0) {
2814 int dh;
2816 dh = slice->h - slicey;
2817 dh = MIN (disph, dh);
2818 uploadslice (tile, slice);
2820 texcoords[0] = tilex; texcoords[1] = slicey;
2821 texcoords[2] = tilex+dispw; texcoords[3] = slicey;
2822 texcoords[4] = tilex; texcoords[5] = slicey+dh;
2823 texcoords[6] = tilex+dispw; texcoords[7] = slicey+dh;
2825 vertices[0] = dispx; vertices[1] = dispy;
2826 vertices[2] = dispx+dispw; vertices[3] = dispy;
2827 vertices[4] = dispx; vertices[5] = dispy+dh;
2828 vertices[6] = dispx+dispw; vertices[7] = dispy+dh;
2830 glDrawArrays (GL_TRIANGLE_STRIP, 0, 4);
2831 dispy += dh;
2832 disph -= dh;
2833 slice++;
2834 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
2835 slicey = 0;
2837 CAMLreturn (Val_unit);
2840 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2841 value xoff_v, value yoff_v,
2842 value li_v)
2844 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, li_v);
2845 int xoff = Int_val (xoff_v);
2846 int yoff = Int_val (yoff_v);
2847 int noff = Int_val (Field (li_v, 0));
2848 char *targ = String_val (Field (li_v, 1));
2849 int tlen = caml_string_length (Field (li_v, 1));
2850 int hfsize = Int_val (Field (li_v, 2));
2851 char *s = String_val (ptr_v);
2852 int hlmask = Int_val (hlinks_v);
2853 struct page *page = parse_pointer ("ml_postprocess", s);
2855 if (!page->u.ptr) {
2856 /* deal with loadpage failed pages */
2857 goto done;
2860 ensureanots (page);
2862 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2863 if (trylock ("ml_postprocess")) {
2864 noff = 0;
2865 goto done;
2867 if (hlmask & 2) {
2868 highlightslinks (page, xoff, yoff, noff, targ, tlen, hfsize);
2869 noff = page->slinkcount;
2871 if (page->tgen == state.gen) {
2872 showsel (page, xoff, yoff);
2874 unlock ("ml_postprocess");
2876 done:
2877 CAMLreturn (Val_int (noff));
2880 static struct annot *getannot (struct page *page, int x, int y)
2882 int i;
2883 fz_point p;
2884 fz_matrix ctm;
2885 const fz_matrix *tctm;
2887 if (!page->annots) return NULL;
2889 switch (page->type) {
2890 case DPDF:
2891 trimctm (page->u.pdfpage, page->pdimno);
2892 tctm = &state.pagedims[page->pdimno].tctm;
2893 break;
2895 case DXPS:
2896 tctm = &fz_identity;
2897 break;
2899 default:
2900 return NULL;
2903 p.x = x;
2904 p.y = y;
2906 fz_concat (&ctm, tctm, &state.pagedims[page->pdimno].ctm);
2907 fz_invert_matrix (&ctm, &ctm);
2908 fz_transform_point (&p, &ctm);
2910 for (i = 0; i < page->annotcount; ++i) {
2911 struct annot *a = &page->annots[i];
2912 if (p.x >= a->annot->pagerect.x0 && p.x <= a->annot->pagerect.x1) {
2913 if (p.y >= a->annot->pagerect.y0 && p.y <= a->annot->pagerect.y1) {
2914 return a;
2918 return NULL;
2921 static fz_link *getlink (struct page *page, int x, int y)
2923 fz_point p;
2924 fz_matrix ctm;
2925 const fz_matrix *tctm;
2926 fz_link *link, *links;
2928 switch (page->type) {
2929 case DPDF:
2930 trimctm (page->u.pdfpage, page->pdimno);
2931 tctm = &state.pagedims[page->pdimno].tctm;
2932 links = page->u.pdfpage->links;
2933 break;
2935 case DXPS:
2936 tctm = &fz_identity;
2937 links = page->u.xpspage->links;
2938 break;
2940 default:
2941 return NULL;
2943 p.x = x;
2944 p.y = y;
2946 fz_concat (&ctm, tctm, &state.pagedims[page->pdimno].ctm);
2947 fz_invert_matrix (&ctm, &ctm);
2948 fz_transform_point (&p, &ctm);
2950 for (link = links; link; link = link->next) {
2951 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2952 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2953 return link;
2957 return NULL;
2960 static void ensuretext (struct page *page)
2962 if (state.gen != page->tgen) {
2963 droptext (page);
2964 page->tgen = state.gen;
2966 if (!page->text) {
2967 fz_matrix ctm;
2968 fz_device *tdev;
2970 page->text = fz_new_text_page (state.ctx);
2971 page->sheet = fz_new_text_sheet (state.ctx);
2972 tdev = fz_new_text_device (state.ctx, page->sheet, page->text);
2973 ctm = pagectm (page);
2974 fz_begin_page (tdev, &fz_infinite_rect, &ctm);
2975 fz_run_display_list (page->dlist, tdev, &ctm, &fz_infinite_rect, NULL);
2976 qsort (page->text->blocks, page->text->len,
2977 sizeof (*page->text->blocks), compareblocks);
2978 fz_end_page (tdev);
2979 fz_free_device (tdev);
2983 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2985 CAMLparam2 (start_page_v, dir_v);
2986 CAMLlocal1 (ret_v);
2987 int i, dir = Int_val (dir_v);
2988 int start_page = Int_val (start_page_v);
2989 int end_page = dir > 0 ? state.pagecount : -1;
2991 ret_v = Val_int (0);
2992 if (!(state.type == DPDF || state.type == DXPS)) {
2993 goto done;
2996 lock ("ml_findpage_with_links");
2997 for (i = start_page + dir; i != end_page; i += dir) {
2998 int found;
3000 switch (state.type) {
3001 case DPDF:
3003 pdf_page *page = NULL;
3005 fz_try (state.ctx) {
3006 page = pdf_load_page (state.u.pdf, i);
3007 found = !!page->links || !!page->annots;
3009 fz_catch (state.ctx) {
3010 found = 0;
3012 if (page) {
3013 freepdfpage (page);
3016 break;
3017 case DXPS:
3019 xps_page *page = xps_load_page (state.u.xps, i);
3020 found = !!page->links;
3021 freexpspage (page);
3023 break;
3025 default:
3026 ARSERT (0 && "invalid document type");
3029 if (found) {
3030 ret_v = caml_alloc_small (1, 1);
3031 Field (ret_v, 0) = Val_int (i);
3032 goto unlock;
3035 unlock:
3036 unlock ("ml_findpage_with_links");
3038 done:
3039 CAMLreturn (ret_v);
3042 enum { dir_first, dir_last };
3043 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
3045 CAMLprim value ml_findlink (value ptr_v, value dir_v)
3047 CAMLparam2 (ptr_v, dir_v);
3048 CAMLlocal2 (ret_v, pos_v);
3049 struct page *page;
3050 int dirtag, i, slinkindex;
3051 struct slink *found = NULL ,*slink;
3052 char *s = String_val (ptr_v);
3054 page = parse_pointer ("ml_findlink", s);
3055 ret_v = Val_int (0);
3056 /* This is scary we are not taking locks here ensureslinks does
3057 not modify state and given that we obtained the page it can not
3058 disappear under us either */
3059 lock ("ml_findlink");
3060 ensureslinks (page);
3062 if (Is_block (dir_v)) {
3063 dirtag = Tag_val (dir_v);
3064 switch (dirtag) {
3065 case dir_first_visible:
3067 int x0, y0, dir, first_index, last_index;
3069 pos_v = Field (dir_v, 0);
3070 x0 = Int_val (Field (pos_v, 0));
3071 y0 = Int_val (Field (pos_v, 1));
3072 dir = Int_val (Field (pos_v, 2));
3074 if (dir >= 0) {
3075 dir = 1;
3076 first_index = 0;
3077 last_index = page->slinkcount;
3079 else {
3080 first_index = page->slinkcount - 1;
3081 last_index = -1;
3084 for (i = first_index; i != last_index; i += dir) {
3085 slink = &page->slinks[i];
3086 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
3087 found = slink;
3088 break;
3092 break;
3094 case dir_left:
3095 slinkindex = Int_val (Field (dir_v, 0));
3096 found = &page->slinks[slinkindex];
3097 for (i = slinkindex - 1; i >= 0; --i) {
3098 slink = &page->slinks[i];
3099 if (slink->bbox.x0 < found->bbox.x0) {
3100 found = slink;
3101 break;
3104 break;
3106 case dir_right:
3107 slinkindex = Int_val (Field (dir_v, 0));
3108 found = &page->slinks[slinkindex];
3109 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
3110 slink = &page->slinks[i];
3111 if (slink->bbox.x0 > found->bbox.x0) {
3112 found = slink;
3113 break;
3116 break;
3118 case dir_down:
3119 slinkindex = Int_val (Field (dir_v, 0));
3120 found = &page->slinks[slinkindex];
3121 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
3122 slink = &page->slinks[i];
3123 if (slink->bbox.y0 >= found->bbox.y0) {
3124 found = slink;
3125 break;
3128 break;
3130 case dir_up:
3131 slinkindex = Int_val (Field (dir_v, 0));
3132 found = &page->slinks[slinkindex];
3133 for (i = slinkindex - 1; i >= 0; --i) {
3134 slink = &page->slinks[i];
3135 if (slink->bbox.y0 <= found->bbox.y0) {
3136 found = slink;
3137 break;
3140 break;
3143 else {
3144 dirtag = Int_val (dir_v);
3145 switch (dirtag) {
3146 case dir_first:
3147 found = page->slinks;
3148 break;
3150 case dir_last:
3151 if (page->slinks) {
3152 found = page->slinks + (page->slinkcount - 1);
3154 break;
3157 if (found) {
3158 ret_v = caml_alloc_small (2, 1);
3159 Field (ret_v, 0) = Val_int (found - page->slinks);
3162 unlock ("ml_findlink");
3163 CAMLreturn (ret_v);
3166 enum { uuri, ugoto, utext, uunexpected, ulaunch,
3167 unamed, uremote, uremotedest, uannot };
3169 #define LINKTOVAL \
3171 int pageno; \
3173 switch (link->dest.kind) { \
3174 case FZ_LINK_GOTO: \
3176 fz_point p; \
3178 pageno = link->dest.ld.gotor.page; \
3179 p.x = 0; \
3180 p.y = 0; \
3182 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
3183 p.y = link->dest.ld.gotor.lt.y; \
3184 fz_transform_point (&p, &pdim->lctm); \
3185 if (p.y < 0) p.y = 0; \
3187 tup_v = caml_alloc_tuple (2); \
3188 ret_v = caml_alloc_small (1, ugoto); \
3189 Field (tup_v, 0) = Val_int (pageno); \
3190 Field (tup_v, 1) = Val_int (p.y); \
3191 Field (ret_v, 0) = tup_v; \
3193 break; \
3195 case FZ_LINK_URI: \
3196 str_v = caml_copy_string (link->dest.ld.uri.uri); \
3197 ret_v = caml_alloc_small (1, uuri); \
3198 Field (ret_v, 0) = str_v; \
3199 break; \
3201 case FZ_LINK_LAUNCH: \
3202 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
3203 ret_v = caml_alloc_small (1, ulaunch); \
3204 Field (ret_v, 0) = str_v; \
3205 break; \
3207 case FZ_LINK_NAMED: \
3208 str_v = caml_copy_string (link->dest.ld.named.named); \
3209 ret_v = caml_alloc_small (1, unamed); \
3210 Field (ret_v, 0) = str_v; \
3211 break; \
3213 case FZ_LINK_GOTOR: \
3215 int rty; \
3217 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
3218 pageno = link->dest.ld.gotor.page; \
3219 if (pageno == -1) { \
3220 gr_v = caml_copy_string (link->dest.ld.gotor.dest); \
3221 rty = uremotedest; \
3223 else { \
3224 gr_v = Val_int (pageno); \
3225 rty = uremote; \
3227 tup_v = caml_alloc_tuple (2); \
3228 ret_v = caml_alloc_small (1, rty); \
3229 Field (tup_v, 0) = str_v; \
3230 Field (tup_v, 1) = gr_v; \
3231 Field (ret_v, 0) = tup_v; \
3233 break; \
3235 default: \
3237 char buf[80]; \
3239 snprintf (buf, sizeof (buf), \
3240 "unhandled link kind %d", link->dest.kind); \
3241 str_v = caml_copy_string (buf); \
3242 ret_v = caml_alloc_small (1, uunexpected); \
3243 Field (ret_v, 0) = str_v; \
3245 break; \
3249 CAMLprim value ml_getlink (value ptr_v, value n_v)
3251 CAMLparam2 (ptr_v, n_v);
3252 CAMLlocal4 (ret_v, tup_v, str_v, gr_v);
3253 fz_link *link;
3254 struct page *page;
3255 struct pagedim *pdim;
3256 char *s = String_val (ptr_v);
3257 struct slink *slink;
3259 /* See ml_findlink for caveat */
3261 ret_v = Val_int (0);
3262 page = parse_pointer ("ml_getlink", s);
3263 ensureslinks (page);
3264 pdim = &state.pagedims[page->pdimno];
3265 slink = &page->slinks[Int_val (n_v)];
3266 if (slink->tag == SLINK) {
3267 link = slink->u.link;
3268 LINKTOVAL;
3270 else {
3271 ret_v = caml_alloc_small (1, uannot);
3272 tup_v = caml_alloc_tuple (2);
3273 Field (ret_v, 0) = tup_v;
3274 Field (tup_v, 0) = ptr_v;
3275 Field (tup_v, 1) = n_v;
3278 unlock ("ml_getlink");
3279 CAMLreturn (ret_v);
3282 CAMLprim value ml_getannotcontents (value ptr_v, value n_v)
3284 CAMLparam2 (ptr_v, n_v);
3285 char *s = String_val (ptr_v);
3286 struct page *page;
3287 struct slink *slink;
3289 page = parse_pointer ("ml_getannotcontent", s);
3290 slink = &page->slinks[Int_val (n_v)];
3291 CAMLreturn (caml_copy_string (pdf_annot_contents (state.u.pdf,
3292 slink->u.annot)));
3295 CAMLprim value ml_getlinkcount (value ptr_v)
3297 CAMLparam1 (ptr_v);
3298 struct page *page;
3299 char *s = String_val (ptr_v);
3301 page = parse_pointer ("ml_getlinkcount", s);
3302 CAMLreturn (Val_int (page->slinkcount));
3305 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
3307 CAMLparam2 (ptr_v, n_v);
3308 CAMLlocal1 (ret_v);
3309 struct page *page;
3310 struct slink *slink;
3311 char *s = String_val (ptr_v);
3312 /* See ml_findlink for caveat */
3314 page = parse_pointer ("ml_getlinkrect", s);
3315 ret_v = caml_alloc_tuple (4);
3316 ensureslinks (page);
3318 slink = &page->slinks[Int_val (n_v)];
3319 Field (ret_v, 0) = Val_int (slink->bbox.x0);
3320 Field (ret_v, 1) = Val_int (slink->bbox.y0);
3321 Field (ret_v, 2) = Val_int (slink->bbox.x1);
3322 Field (ret_v, 3) = Val_int (slink->bbox.y1);
3323 unlock ("ml_getlinkrect");
3324 CAMLreturn (ret_v);
3327 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
3329 CAMLparam3 (ptr_v, x_v, y_v);
3330 CAMLlocal4 (ret_v, tup_v, str_v, gr_v);
3331 fz_link *link;
3332 struct annot *annot;
3333 struct page *page;
3334 char *ptr = String_val (ptr_v);
3335 int x = Int_val (x_v), y = Int_val (y_v);
3336 struct pagedim *pdim;
3338 ret_v = Val_int (0);
3339 if (trylock ("ml_whatsunder")) {
3340 goto done;
3343 page = parse_pointer ("ml_whatsunder", ptr);
3344 pdim = &state.pagedims[page->pdimno];
3345 x += pdim->bounds.x0;
3346 y += pdim->bounds.y0;
3348 if (state.type == DPDF) {
3349 annot = getannot (page, x, y);
3350 if (annot) {
3351 int i, n = -1;
3353 ensureslinks (page);
3354 for (i = 0; i < page->slinkcount; ++i) {
3355 if (page->slinks[i].tag == SANNOT
3356 && page->slinks[i].u.annot == annot->annot) {
3357 n = i;
3358 break;
3361 ret_v = caml_alloc_small (1, uannot);
3362 tup_v = caml_alloc_tuple (2);
3363 Field (ret_v, 0) = tup_v;
3364 Field (tup_v, 0) = ptr_v;
3365 Field (tup_v, 1) = Val_int (n);
3366 goto unlock;
3370 link = getlink (page, x, y);
3371 if (link) {
3372 LINKTOVAL;
3374 else {
3375 fz_rect *b;
3376 fz_page_block *pageb;
3377 fz_text_block *block;
3379 ensuretext (page);
3380 for (pageb = page->text->blocks;
3381 pageb < page->text->blocks + page->text->len;
3382 ++pageb) {
3383 fz_text_line *line;
3384 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3385 block = pageb->u.text;
3387 b = &block->bbox;
3388 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3389 continue;
3391 for (line = block->lines;
3392 line < block->lines + block->len;
3393 ++line) {
3394 fz_text_span *span;
3396 b = &line->bbox;
3397 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3398 continue;
3400 for (span = line->first_span; span; span = span->next) {
3401 int charnum;
3403 b = &span->bbox;
3404 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3405 continue;
3407 for (charnum = 0; charnum < span->len; ++charnum) {
3408 fz_rect bbox;
3409 fz_text_char_bbox (&bbox, span, charnum);
3410 b = &bbox;
3412 if (x >= b->x0 && x <= b->x1
3413 && y >= b->y0 && y <= b->y1) {
3414 fz_text_style *style = span->text->style;
3415 const char *n2 =
3416 style->font && style->font->name
3417 ? style->font->name
3418 : "Span has no font name"
3420 FT_FaceRec *face = style->font->ft_face;
3421 if (face && face->family_name) {
3422 char *s;
3423 char *n1 = face->family_name;
3424 size_t l1 = strlen (n1);
3425 size_t l2 = strlen (n2);
3427 if (l1 != l2 || memcmp (n1, n2, l1)) {
3428 s = malloc (l1 + l2 + 2);
3429 if (s) {
3430 memcpy (s, n2, l2);
3431 s[l2] = '=';
3432 memcpy (s + l2 + 1, n1, l1 + 1);
3433 str_v = caml_copy_string (s);
3434 free (s);
3438 if (str_v == Val_unit) {
3439 str_v = caml_copy_string (n2);
3441 ret_v = caml_alloc_small (1, utext);
3442 Field (ret_v, 0) = str_v;
3443 goto unlock;
3450 unlock:
3451 unlock ("ml_whatsunder");
3453 done:
3454 CAMLreturn (ret_v);
3457 enum { mark_page, mark_block, mark_line, mark_word };
3459 static int uninteresting (int c)
3461 return c == ' ' || c == '\n' || c == '\t' || c == '\n' || c == '\r'
3462 || ispunct (c);
3465 CAMLprim value ml_clearmark (value ptr_v)
3467 CAMLparam1 (ptr_v);
3468 char *s = String_val (ptr_v);
3469 struct page *page;
3471 if (trylock ("ml_clearmark")) {
3472 goto done;
3475 page = parse_pointer ("ml_clearmark", s);
3476 page->fmark.span = NULL;
3477 page->lmark.span = NULL;
3478 page->fmark.i = 0;
3479 page->lmark.i = 0;
3481 unlock ("ml_clearmark");
3482 done:
3483 CAMLreturn (Val_unit);
3486 CAMLprim value ml_markunder (value ptr_v, value x_v, value y_v, value mark_v)
3488 CAMLparam4 (ptr_v, x_v, y_v, mark_v);
3489 CAMLlocal1 (ret_v);
3490 fz_rect *b;
3491 struct page *page;
3492 fz_text_line *line;
3493 fz_page_block *pageb;
3494 fz_text_block *block;
3495 struct pagedim *pdim;
3496 int mark = Int_val (mark_v);
3497 char *s = String_val (ptr_v);
3498 int x = Int_val (x_v), y = Int_val (y_v);
3500 ret_v = Val_bool (0);
3501 if (trylock ("ml_markunder")) {
3502 goto done;
3505 page = parse_pointer ("ml_markunder", s);
3506 pdim = &state.pagedims[page->pdimno];
3508 ensuretext (page);
3510 if (mark == mark_page) {
3511 int i;
3512 fz_page_block *pb1 = NULL, *pb2 = NULL;
3514 for (i = 0; i < page->text->len; ++i) {
3515 if (page->text->blocks[i].type == FZ_PAGE_BLOCK_TEXT) {
3516 pb1 = &page->text->blocks[i];
3517 break;
3520 if (!pb1) goto unlock;
3522 for (i = page->text->len - 1; i >= 0; --i) {
3523 if (page->text->blocks[i].type == FZ_PAGE_BLOCK_TEXT) {
3524 pb2 = &page->text->blocks[i];
3525 break;
3528 if (!pb2) goto unlock;
3530 block = pb1->u.text;
3532 page->fmark.i = 0;
3533 page->fmark.span = block->lines->first_span;
3535 block = pb2->u.text;
3536 line = &block->lines[block->len - 1];
3537 page->lmark.i = line->last_span->len - 1;
3538 page->lmark.span = line->last_span;
3539 ret_v = Val_bool (1);
3540 goto unlock;
3543 x += pdim->bounds.x0;
3544 y += pdim->bounds.y0;
3546 for (pageb = page->text->blocks;
3547 pageb < page->text->blocks + page->text->len;
3548 ++pageb) {
3549 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3550 block = pageb->u.text;
3552 b = &block->bbox;
3553 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3554 continue;
3556 if (mark == mark_block) {
3557 page->fmark.i = 0;
3558 page->fmark.span = block->lines->first_span;
3560 line = &block->lines[block->len - 1];
3561 page->lmark.i = line->last_span->len - 1;
3562 page->lmark.span = line->last_span;
3563 ret_v = Val_bool (1);
3564 goto unlock;
3567 for (line = block->lines;
3568 line < block->lines + block->len;
3569 ++line) {
3570 fz_text_span *span;
3572 b = &line->bbox;
3573 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3574 continue;
3576 if (mark == mark_line) {
3577 page->fmark.i = 0;
3578 page->fmark.span = line->first_span;
3580 page->lmark.i = line->last_span->len - 1;
3581 page->lmark.span = line->last_span;
3582 ret_v = Val_bool (1);
3583 goto unlock;
3586 for (span = line->first_span; span; span = span->next) {
3587 int charnum;
3589 b = &span->bbox;
3590 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3591 continue;
3593 for (charnum = 0; charnum < span->len; ++charnum) {
3594 fz_rect bbox;
3595 fz_text_char_bbox (&bbox, span, charnum);
3596 b = &bbox;
3598 if (x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1) {
3599 /* unicode ftw */
3600 int charnum2, charnum3 = -1, charnum4 = -1;
3602 if (uninteresting (span->text[charnum].c)) goto unlock;
3604 for (charnum2 = charnum; charnum2 >= 0; --charnum2) {
3605 if (uninteresting (span->text[charnum2].c)) {
3606 charnum3 = charnum2 + 1;
3607 break;
3610 if (charnum3 == -1) charnum3 = 0;
3612 for (charnum2 = charnum + 1;
3613 charnum2 < span->len;
3614 ++charnum2) {
3615 if (uninteresting (span->text[charnum2].c)) break;
3616 charnum4 = charnum2;
3618 if (charnum4 == -1) goto unlock;
3620 page->fmark.i = charnum3;
3621 page->fmark.span = span;
3623 page->lmark.i = charnum4;
3624 page->lmark.span = span;
3625 ret_v = Val_bool (1);
3626 goto unlock;
3632 unlock:
3633 if (!Bool_val (ret_v)) {
3634 page->fmark.span = NULL;
3635 page->lmark.span = NULL;
3636 page->fmark.i = 0;
3637 page->lmark.i = 0;
3639 unlock ("ml_markunder");
3641 done:
3642 CAMLreturn (ret_v);
3645 CAMLprim value ml_rectofblock (value ptr_v, value x_v, value y_v)
3647 CAMLparam3 (ptr_v, x_v, y_v);
3648 CAMLlocal2 (ret_v, res_v);
3649 fz_rect *b = NULL;
3650 struct page *page;
3651 fz_page_block *pageb;
3652 struct pagedim *pdim;
3653 char *s = String_val (ptr_v);
3654 int x = Int_val (x_v), y = Int_val (y_v);
3656 ret_v = Val_int (0);
3657 if (trylock ("ml_rectofblock")) {
3658 goto done;
3661 page = parse_pointer ("ml_rectofblock", s);
3662 pdim = &state.pagedims[page->pdimno];
3663 x += pdim->bounds.x0;
3664 y += pdim->bounds.y0;
3666 ensuretext (page);
3668 for (pageb = page->text->blocks;
3669 pageb < page->text->blocks + page->text->len;
3670 ++pageb) {
3671 switch (pageb->type) {
3672 case FZ_PAGE_BLOCK_TEXT:
3673 b = &pageb->u.text->bbox;
3674 break;
3676 case FZ_PAGE_BLOCK_IMAGE:
3677 b = &pageb->u.image->bbox;
3678 break;
3680 default:
3681 continue;
3684 if (x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1)
3685 break;
3686 b = NULL;
3688 if (b) {
3689 res_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
3690 ret_v = caml_alloc_small (1, 1);
3691 Store_double_field (res_v, 0, b->x0);
3692 Store_double_field (res_v, 1, b->x1);
3693 Store_double_field (res_v, 2, b->y0);
3694 Store_double_field (res_v, 3, b->y1);
3695 Field (ret_v, 0) = res_v;
3697 unlock ("ml_rectofblock");
3699 done:
3700 CAMLreturn (ret_v);
3703 CAMLprim value ml_seltext (value ptr_v, value rect_v)
3705 CAMLparam2 (ptr_v, rect_v);
3706 fz_rect b;
3707 struct page *page;
3708 struct pagedim *pdim;
3709 char *s = String_val (ptr_v);
3710 int i, x0, x1, y0, y1, fi, li;
3711 fz_text_line *line;
3712 fz_page_block *pageb;
3713 fz_text_block *block;
3714 fz_text_span *span, *fspan, *lspan;
3716 if (trylock ("ml_seltext")) {
3717 goto done;
3720 page = parse_pointer ("ml_seltext", s);
3721 ensuretext (page);
3723 pdim = &state.pagedims[page->pdimno];
3724 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;
3725 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
3726 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
3727 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
3729 if (y0 > y1) {
3730 int t = y0;
3731 y0 = y1;
3732 y1 = t;
3733 x0 = x1;
3734 x1 = t;
3737 if (0) {
3738 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
3739 glColor3ub (128, 128, 128);
3740 recti (x0, y0, x1, y1);
3741 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
3744 fi = page->fmark.i;
3745 fspan = page->fmark.span;
3747 li = page->lmark.i;
3748 lspan = page->lmark.span;
3750 for (pageb = page->text->blocks;
3751 pageb < page->text->blocks + page->text->len;
3752 ++pageb) {
3753 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3754 block = pageb->u.text;
3755 for (line = block->lines;
3756 line < block->lines + block->len;
3757 ++line) {
3759 for (span = line->first_span; span; span = span->next) {
3760 for (i = 0; i < span->len; ++i) {
3761 int selected = 0;
3763 fz_text_char_bbox (&b, span, i);
3765 if (x0 >= b.x0 && x0 <= b.x1
3766 && y0 >= b.y0 && y0 <= b.y1) {
3767 fspan = span;
3768 fi = i;
3769 selected = 1;
3771 if (x1 >= b.x0 && x1 <= b.x1
3772 && y1 >= b.y0 && y1 <= b.y1) {
3773 lspan = span;
3774 li = i;
3775 selected = 1;
3777 if (0 && selected) {
3778 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
3779 glColor3ub (128, 128, 128);
3780 recti (b.x0, b.y0, b.x1, b.y1);
3781 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
3787 if (x1 < x0 && fspan == lspan) {
3788 i = fi;
3789 span = fspan;
3791 fi = li;
3792 fspan = lspan;
3794 li = i;
3795 lspan = span;
3798 page->fmark.i = fi;
3799 page->fmark.span = fspan;
3801 page->lmark.i = li;
3802 page->lmark.span = lspan;
3804 unlock ("ml_seltext");
3806 done:
3807 CAMLreturn (Val_unit);
3810 static int UNUSED_ATTR pipespan (FILE *f, fz_text_span *span, int a, int b)
3812 char buf[4];
3813 int i, len, ret;
3815 for (i = a; i <= b; ++i) {
3816 len = fz_runetochar (buf, span->text[i].c);
3817 ret = fwrite (buf, len, 1, f);
3819 if (ret != 1) {
3820 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
3821 len, ret, strerror (errno));
3822 return -1;
3825 return 0;
3828 #ifdef __CYGWIN__
3829 CAMLprim value ml_popen (value UNUSED_ATTR u1, value UNUSED_ATTR u2)
3831 caml_failwith ("ml_popen not implemented under Cygwin");
3833 #else
3834 CAMLprim value ml_popen (value command_v, value fds_v)
3836 CAMLparam2 (command_v, fds_v);
3837 CAMLlocal2 (l_v, tup_v);
3838 int ret;
3839 pid_t pid;
3840 char *msg = NULL;
3841 value earg_v = Nothing;
3842 posix_spawnattr_t attr;
3843 posix_spawn_file_actions_t fa;
3844 char *argv[] = { "/bin/sh", "-c", NULL, NULL };
3846 argv[2] = String_val (command_v);
3848 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
3849 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
3852 if ((ret = posix_spawnattr_init (&attr)) != 0) {
3853 msg = "posix_spawnattr_init";
3854 goto fail1;
3857 #ifdef POSIX_SPAWN_USEVFORK
3858 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
3859 msg = "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK";
3860 goto fail;
3862 #endif
3864 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
3865 int fd1, fd2;
3867 tup_v = Field (l_v, 0);
3868 fd1 = Int_val (Field (tup_v, 0));
3869 fd2 = Int_val (Field (tup_v, 1));
3870 if (fd2 < 0) {
3871 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
3872 msg = "posix_spawn_file_actions_addclose";
3873 earg_v = tup_v;
3874 goto fail;
3877 else {
3878 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
3879 msg = "posix_spawn_file_actions_adddup2";
3880 earg_v = tup_v;
3881 goto fail;
3886 if ((ret = posix_spawn (&pid, "/bin/sh", &fa, &attr, argv, environ))) {
3887 msg = "posix_spawn";
3888 goto fail;
3891 fail:
3892 if ((ret = posix_spawnattr_destroy (&attr)) != 0) {
3893 fprintf (stderr, "posix_spawnattr_destroy: %s\n", strerror (ret));
3896 fail1:
3897 if ((ret = posix_spawn_file_actions_destroy (&fa)) != 0) {
3898 fprintf (stderr, "posix_spawn_file_actions_destroy: %s\n",
3899 strerror (ret));
3902 if (msg)
3903 unix_error (ret, msg, earg_v);
3905 CAMLreturn (Val_int (pid));
3907 #endif
3909 CAMLprim value ml_hassel (value ptr_v)
3911 CAMLparam1 (ptr_v);
3912 CAMLlocal1 (ret_v);
3913 struct page *page;
3914 char *s = String_val (ptr_v);
3916 ret_v = Val_bool (0);
3917 if (trylock ("ml_hassel")) {
3918 goto done;
3921 page = parse_pointer ("ml_hassel", s);
3922 ret_v = Val_bool (page->fmark.span && page->lmark.span);
3923 unlock ("ml_hassel");
3924 done:
3925 CAMLreturn (ret_v);
3928 CAMLprim value ml_copysel (value fd_v, value ptr_v)
3930 CAMLparam2 (fd_v, ptr_v);
3931 FILE *f;
3932 int seen = 0;
3933 struct page *page;
3934 fz_text_line *line;
3935 fz_page_block *pageb;
3936 fz_text_block *block;
3937 int fd = Int_val (fd_v);
3938 char *s = String_val (ptr_v);
3940 if (trylock ("ml_copysel")) {
3941 goto done;
3944 page = parse_pointer ("ml_copysel", s);
3946 if (!page->fmark.span || !page->lmark.span) {
3947 fprintf (stderr, "nothing to copy on page %d\n", page->pageno);
3948 goto unlock;
3951 f = fdopen (fd, "w");
3952 if (!f) {
3953 fprintf (stderr, "failed to fdopen sel pipe (from fd %d): %s\n",
3954 fd, strerror (errno));
3955 f = stdout;
3958 for (pageb = page->text->blocks;
3959 pageb < page->text->blocks + page->text->len;
3960 ++pageb) {
3961 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3962 block = pageb->u.text;
3963 for (line = block->lines;
3964 line < block->lines + block->len;
3965 ++line) {
3966 fz_text_span *span;
3968 for (span = line->first_span; span; span = span->next) {
3969 int a, b;
3971 seen |= span == page->fmark.span || span == page->lmark.span;
3972 a = span == page->fmark.span ? page->fmark.i : 0;
3973 b = span == page->lmark.span ? page->lmark.i : span->len - 1;
3975 if (seen) {
3976 if (pipespan (f, span, a, b)) {
3977 goto close;
3979 if (span == page->lmark.span) {
3980 goto close;
3982 if (span == line->last_span) {
3983 if (putc ('\n', f) == EOF) {
3984 fprintf (stderr,
3985 "failed break line on sel pipe: %s\n",
3986 strerror (errno));
3987 goto close;
3994 close:
3995 if (f != stdout) {
3996 int ret = fclose (f);
3997 fd = -1;
3998 if (ret == -1) {
3999 if (errno != ECHILD) {
4000 fprintf (stderr, "failed to close sel pipe: %s\n",
4001 strerror (errno));
4005 unlock:
4006 unlock ("ml_copysel");
4008 done:
4009 if (fd >= 0) {
4010 if (close (fd)) {
4011 fprintf (stderr, "failed to close sel pipe: %s\n",
4012 strerror (errno));
4015 CAMLreturn (Val_unit);
4018 CAMLprim value ml_getpdimrect (value pagedimno_v)
4020 CAMLparam1 (pagedimno_v);
4021 CAMLlocal1 (ret_v);
4022 int pagedimno = Int_val (pagedimno_v);
4023 fz_rect box;
4025 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
4026 if (trylock ("ml_getpdimrect")) {
4027 box = fz_empty_rect;
4029 else {
4030 box = state.pagedims[pagedimno].mediabox;
4031 unlock ("ml_getpdimrect");
4034 Store_double_field (ret_v, 0, box.x0);
4035 Store_double_field (ret_v, 1, box.x1);
4036 Store_double_field (ret_v, 2, box.y0);
4037 Store_double_field (ret_v, 3, box.y1);
4039 CAMLreturn (ret_v);
4042 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v,
4043 value dw_v, value cols_v)
4045 CAMLparam3 (winw_v, winh_v, dw_v);
4046 CAMLlocal1 (ret_v);
4047 int i;
4048 double zoom = -1.;
4049 double maxh = 0.0;
4050 struct pagedim *p;
4051 double winw = Int_val (winw_v);
4052 double winh = Int_val (winh_v);
4053 double dw = Int_val (dw_v);
4054 double cols = Int_val (cols_v);
4055 double pw = 1.0, ph = 1.0;
4057 if (trylock ("ml_zoom_for_height")) {
4058 goto done;
4061 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
4062 double w = p->pagebox.x1 / cols;
4063 double h = p->pagebox.y1;
4064 if (h > maxh) {
4065 maxh = h;
4066 ph = h;
4067 if (state.fitmodel != FitProportional) pw = w;
4069 if ((state.fitmodel == FitProportional) && w > pw) pw = w;
4072 zoom = (((winh / ph) * pw) + dw) / winw;
4073 unlock ("ml_zoom_for_height");
4074 done:
4075 ret_v = caml_copy_double (zoom);
4076 CAMLreturn (ret_v);
4079 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
4081 CAMLparam4 (pt_v, x_v, y_v, string_v);
4082 CAMLlocal1 (ret_v);
4083 int pt = Int_val(pt_v);
4084 int x = Int_val (x_v);
4085 int y = Int_val (y_v);
4086 double w;
4088 w = draw_string (state.face, pt, x, y, String_val (string_v));
4089 ret_v = caml_copy_double (w);
4090 CAMLreturn (ret_v);
4093 CAMLprim value ml_measure_string (value pt_v, value string_v)
4095 CAMLparam2 (pt_v, string_v);
4096 CAMLlocal1 (ret_v);
4097 int pt = Int_val (pt_v);
4098 double w;
4100 w = measure_string (state.face, pt, String_val (string_v));
4101 ret_v = caml_copy_double (w);
4102 CAMLreturn (ret_v);
4105 CAMLprim value ml_getpagebox (value opaque_v)
4107 CAMLparam1 (opaque_v);
4108 CAMLlocal1 (ret_v);
4109 fz_rect rect;
4110 fz_irect bbox;
4111 fz_matrix ctm;
4112 fz_device *dev;
4113 char *s = String_val (opaque_v);
4114 struct page *page = parse_pointer ("ml_getpagebox", s);
4116 ret_v = caml_alloc_tuple (4);
4117 dev = fz_new_bbox_device (state.ctx, &rect);
4118 dev->hints |= FZ_IGNORE_SHADE;
4120 switch (page->type) {
4121 case DPDF:
4122 ctm = pagectm (page);
4123 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, &ctm, NULL);
4124 break;
4126 case DXPS:
4127 ctm = pagectm (page);
4128 xps_run_page (state.u.xps, page->u.xpspage, dev, &ctm, NULL);
4129 break;
4131 default:
4132 rect = fz_infinite_rect;
4133 break;
4136 fz_free_device (dev);
4137 fz_round_rect (&bbox, &rect);
4138 Field (ret_v, 0) = Val_int (bbox.x0);
4139 Field (ret_v, 1) = Val_int (bbox.y0);
4140 Field (ret_v, 2) = Val_int (bbox.x1);
4141 Field (ret_v, 3) = Val_int (bbox.y1);
4143 CAMLreturn (ret_v);
4146 CAMLprim value ml_setaalevel (value level_v)
4148 CAMLparam1 (level_v);
4150 state.aalevel = Int_val (level_v);
4151 CAMLreturn (Val_unit);
4154 #pragma GCC diagnostic push
4155 #pragma GCC diagnostic ignored "-Wvariadic-macros"
4156 #include <X11/Xlib.h>
4157 #pragma GCC diagnostic pop
4159 #include <GL/glx.h>
4161 static struct {
4162 Window wid;
4163 Display *dpy;
4164 GLXContext ctx;
4165 XVisualInfo *visual;
4166 } glx;
4168 CAMLprim value ml_glxinit (value display_v, value wid_v, value screen_v)
4170 CAMLparam3 (display_v, wid_v, screen_v);
4171 int attribs[] = {GLX_RGBA, GLX_DOUBLEBUFFER, None};
4173 glx.dpy = XOpenDisplay (String_val (display_v));
4174 if (!glx.dpy) {
4175 caml_failwith ("XOpenDisplay");
4178 glx.visual = glXChooseVisual (glx.dpy, Int_val (screen_v), attribs);
4179 if (!glx.visual) {
4180 XCloseDisplay (glx.dpy);
4181 caml_failwith ("glXChooseVisual");
4184 glx.wid = Int_val (wid_v);
4185 CAMLreturn (Val_int (glx.visual->visualid));
4188 CAMLprim value ml_glxcompleteinit (value unit_v)
4190 CAMLparam1 (unit_v);
4192 glx.ctx = glXCreateContext (glx.dpy, glx.visual, NULL, True);
4193 if (!glx.ctx) {
4194 caml_failwith ("glXCreateContext");
4197 XFree (glx.visual);
4198 glx.visual = NULL;
4200 if (!glXMakeCurrent (glx.dpy, glx.wid, glx.ctx)) {
4201 glXDestroyContext (glx.dpy, glx.ctx);
4202 glx.ctx = NULL;
4203 caml_failwith ("glXMakeCurrent");
4205 CAMLreturn (Val_unit);
4208 CAMLprim value ml_swapb (value unit_v)
4210 CAMLparam1 (unit_v);
4211 glXSwapBuffers (glx.dpy, glx.wid);
4212 CAMLreturn (Val_unit);
4215 #include "keysym2ucs.c"
4217 CAMLprim value ml_keysymtoutf8 (value keysym_v)
4219 CAMLparam1 (keysym_v);
4220 CAMLlocal1 (str_v);
4221 KeySym keysym = Int_val (keysym_v);
4222 Rune rune;
4223 int len;
4224 char buf[5];
4226 rune = keysym2ucs (keysym);
4227 len = fz_runetochar (buf, rune);
4228 buf[len] = 0;
4229 str_v = caml_copy_string (buf);
4230 CAMLreturn (str_v);
4233 enum { piunknown, pilinux, piosx, pisun, pibsd, picygwin };
4235 CAMLprim value ml_platform (value unit_v)
4237 CAMLparam1 (unit_v);
4238 CAMLlocal2 (tup_v, arr_v);
4239 int platid = piunknown;
4240 struct utsname buf;
4242 #if defined __linux__
4243 platid = pilinux;
4244 #elif defined __CYGWIN__
4245 platid = picygwin;
4246 #elif defined __DragonFly__ || defined __FreeBSD__
4247 || defined __OpenBSD__ || defined __NetBSD__
4248 platid = pibsd;
4249 #elif defined __sun__
4250 platid = pisun;
4251 #elif defined __APPLE__
4252 platid = piosx;
4253 #endif
4254 if (uname (&buf)) err (1, "uname");
4256 tup_v = caml_alloc_tuple (2);
4258 char const *sar[] = {
4259 buf.sysname,
4260 buf.release,
4261 buf.version,
4262 buf.machine,
4263 NULL
4265 arr_v = caml_copy_string_array (sar);
4267 Field (tup_v, 0) = Val_int (platid);
4268 Field (tup_v, 1) = arr_v;
4269 CAMLreturn (tup_v);
4272 CAMLprim value ml_cloexec (value fd_v)
4274 CAMLparam1 (fd_v);
4275 int fd = Int_val (fd_v);
4277 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
4278 uerror ("fcntl", Nothing);
4280 CAMLreturn (Val_unit);
4283 CAMLprim value ml_getpbo (value w_v, value h_v, value cs_v)
4285 CAMLparam2 (w_v, h_v);
4286 CAMLlocal1 (ret_v);
4287 struct pbo *pbo;
4288 int w = Int_val (w_v);
4289 int h = Int_val (h_v);
4290 int cs = Int_val (cs_v);
4292 if (state.pbo_usable) {
4293 pbo = calloc (sizeof (*pbo), 1);
4294 if (!pbo) {
4295 err (1, "calloc pbo");
4298 switch (cs) {
4299 case 0:
4300 case 1:
4301 pbo->size = w*h*4;
4302 break;
4303 case 2:
4304 pbo->size = w*h*2;
4305 break;
4306 default:
4307 errx (1, "ml_getpbo: invalid colorspace %d", cs);
4310 state.glGenBuffersARB (1, &pbo->id);
4311 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->id);
4312 state.glBufferDataARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->size,
4313 NULL, GL_STREAM_DRAW);
4314 pbo->ptr = state.glMapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB,
4315 GL_READ_WRITE);
4316 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
4317 if (!pbo->ptr) {
4318 fprintf (stderr, "glMapBufferARB failed: %#x\n", glGetError ());
4319 state.glDeleteBuffersARB (1, &pbo->id);
4320 free (pbo);
4321 ret_v = caml_copy_string ("0");
4323 else {
4324 int res;
4325 char *s;
4327 res = snprintf (NULL, 0, "%" FMT_ptr, FMT_ptr_cast (pbo));
4328 if (res < 0) {
4329 err (1, "snprintf %" FMT_ptr " failed", FMT_ptr_cast (pbo));
4331 s = malloc (res+1);
4332 if (!s) {
4333 err (1, "malloc %d bytes failed", res+1);
4335 res = sprintf (s, "%" FMT_ptr, FMT_ptr_cast (pbo));
4336 if (res < 0) {
4337 err (1, "sprintf %" FMT_ptr " failed", FMT_ptr_cast (pbo));
4339 ret_v = caml_copy_string (s);
4340 free (s);
4343 else {
4344 ret_v = caml_copy_string ("0");
4346 CAMLreturn (ret_v);
4349 CAMLprim value ml_freepbo (value s_v)
4351 CAMLparam1 (s_v);
4352 char *s = String_val (s_v);
4353 struct tile *tile = parse_pointer ("ml_freepbo", s);
4355 if (tile->pbo) {
4356 state.glDeleteBuffersARB (1, &tile->pbo->id);
4357 tile->pbo->id = -1;
4358 tile->pbo->ptr = NULL;
4359 tile->pbo->size = -1;
4361 CAMLreturn (Val_unit);
4364 CAMLprim value ml_unmappbo (value s_v)
4366 CAMLparam1 (s_v);
4367 char *s = String_val (s_v);
4368 struct tile *tile = parse_pointer ("ml_unmappbo", s);
4370 if (tile->pbo) {
4371 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
4372 if (state.glUnmapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB) == GL_FALSE) {
4373 errx (1, "glUnmapBufferARB failed: %#x\n", glGetError ());
4375 tile->pbo->ptr = NULL;
4376 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
4378 CAMLreturn (Val_unit);
4381 static void setuppbo (void)
4383 #define GGPA(n) (*(void (**) ()) &state.n = glXGetProcAddress ((GLubyte *) #n))
4384 state.pbo_usable = GGPA (glBindBufferARB)
4385 && GGPA (glUnmapBufferARB)
4386 && GGPA (glMapBufferARB)
4387 && GGPA (glBufferDataARB)
4388 && GGPA (glGenBuffersARB)
4389 && GGPA (glDeleteBuffersARB);
4390 #undef GGPA
4393 CAMLprim value ml_pbo_usable (value unit_v)
4395 CAMLparam1 (unit_v);
4396 CAMLreturn (Val_bool (state.pbo_usable));
4399 CAMLprim value ml_unproject (value ptr_v, value x_v, value y_v)
4401 CAMLparam3 (ptr_v, x_v, y_v);
4402 CAMLlocal2 (ret_v, tup_v);
4403 struct page *page;
4404 char *s = String_val (ptr_v);
4405 int x = Int_val (x_v), y = Int_val (y_v);
4406 struct pagedim *pdim;
4407 fz_point p;
4408 fz_matrix ctm;
4410 page = parse_pointer ("ml_unproject", s);
4411 pdim = &state.pagedims[page->pdimno];
4413 ret_v = Val_int (0);
4414 if (trylock ("ml_unproject")) {
4415 goto done;
4418 switch (page->type) {
4419 case DPDF:
4420 trimctm (page->u.pdfpage, page->pdimno);
4421 break;
4423 default:
4424 break;
4426 p.x = x + pdim->bounds.x0;
4427 p.y = y + pdim->bounds.y0;
4429 fz_concat (&ctm, &pdim->tctm, &pdim->ctm);
4430 fz_invert_matrix (&ctm, &ctm);
4431 fz_transform_point (&p, &ctm);
4433 tup_v = caml_alloc_tuple (2);
4434 ret_v = caml_alloc_small (1, 1);
4435 Field (tup_v, 0) = Val_int (p.x);
4436 Field (tup_v, 1) = Val_int (p.y);
4437 Field (ret_v, 0) = tup_v;
4439 unlock ("ml_unproject");
4440 done:
4441 CAMLreturn (ret_v);
4444 CAMLprim value ml_addannot (value ptr_v, value x_v, value y_v,
4445 value contents_v)
4447 CAMLparam4 (ptr_v, x_v, y_v, contents_v);
4449 if (state.type == DPDF) {
4450 pdf_document *pdf;
4451 pdf_annot *annot;
4452 struct page *page;
4453 fz_point p;
4454 char *s = String_val (ptr_v);
4456 pdf = state.u.pdf;
4457 page = parse_pointer ("ml_addannot", s);
4458 annot = pdf_create_annot (pdf, page->u.pdfpage, FZ_ANNOT_TEXT);
4459 p.x = Int_val (x_v);
4460 p.y = Int_val (y_v);
4461 pdf_set_annot_contents (pdf, annot, String_val (contents_v));
4462 pdf_set_text_annot_position (pdf, annot, p);
4464 CAMLreturn (Val_unit);
4467 CAMLprim value ml_delannot (value ptr_v, value n_v)
4469 CAMLparam2 (ptr_v, n_v);
4471 if (state.type == DPDF) {
4472 struct page *page;
4473 char *s = String_val (ptr_v);
4474 struct slink *slink;
4476 page = parse_pointer ("ml_delannot", s);
4477 slink = &page->slinks[Int_val (n_v)];
4478 pdf_delete_annot (state.u.pdf, page->u.pdfpage, slink->u.annot);
4480 CAMLreturn (Val_unit);
4483 CAMLprim value ml_modannot (value ptr_v, value n_v, value str_v)
4485 CAMLparam3 (ptr_v, n_v, str_v);
4487 if (state.type == DPDF) {
4488 struct page *page;
4489 char *s = String_val (ptr_v);
4490 struct slink *slink;
4492 page = parse_pointer ("ml_modannot", s);
4493 slink = &page->slinks[Int_val (n_v)];
4494 pdf_set_annot_contents (state.u.pdf, slink->u.annot,
4495 String_val (str_v));
4496 /* Erm... dirty */
4497 state.u.pdf->dirty = 1;
4499 CAMLreturn (Val_unit);
4502 CAMLprim value ml_hasunsavedchanges (value unit_v)
4504 CAMLparam1 (unit_v);
4505 int modified = 0;
4507 if (state.type == DPDF && state.u.pdf) {
4508 modified = pdf_has_unsaved_changes (state.u.pdf);
4510 CAMLreturn (Val_bool (modified));
4513 CAMLprim value ml_savedoc (value path_v)
4515 CAMLparam1 (path_v);
4517 if (state.type == DPDF) {
4518 pdf_write_document (state.u.pdf, String_val (path_v), NULL);
4520 CAMLreturn (Val_unit);
4523 static void makestippletex (void)
4525 const char pixels[] = "\xff\xff\0\0";
4526 glGenTextures (1, &state.stid);
4527 glBindTexture (GL_TEXTURE_1D, state.stid);
4528 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
4529 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
4530 glTexImage1D (
4531 GL_TEXTURE_1D,
4533 GL_ALPHA,
4536 GL_ALPHA,
4537 GL_UNSIGNED_BYTE,
4538 pixels
4542 CAMLprim value ml_fz_version (value UNUSED_ATTR unit_v)
4544 return caml_copy_string (FZ_VERSION);
4547 #ifdef USE_FONTCONFIG
4548 static struct {
4549 int inited;
4550 FcConfig *config;
4551 } fc;
4553 static fz_font *fc_load_system_font_func (fz_context *ctx,
4554 const char *name,
4555 int bold,
4556 int italic,
4557 int UNUSED_ATTR needs_exact_metrics)
4559 char *buf;
4560 size_t i, size;
4561 fz_font *font;
4562 FcChar8 *path;
4563 FcResult result;
4564 FcPattern *pat, *pat1;
4566 lprintf ("looking up %s bold:%d italic:%d needs_exact_metrics:%d\n",
4567 name, bold, italic, needs_exact_metrics);
4568 if (!fc.inited) {
4569 fc.inited = 1;
4570 fc.config = FcInitLoadConfigAndFonts ();
4571 if (!fc.config) {
4572 lprintf ("FcInitLoadConfigAndFonts failed\n");
4573 return NULL;
4576 if (!fc.config) return NULL;
4578 size = strlen (name);
4579 if (bold) size += sizeof (":bold") - 1;
4580 if (italic) size += sizeof (":italic") - 1;
4581 size += 1;
4583 buf = malloc (size);
4584 if (!buf) {
4585 err (1, "malloc %zu failed", size);
4588 strcpy (buf, name);
4589 if (bold && italic) {
4590 strcat (buf, ":bold:italic");
4592 else {
4593 if (bold) strcat (buf, ":bold");
4594 if (italic) strcat (buf, ":italic");
4596 for (i = 0; i < size; ++i) {
4597 if (buf[i] == ',' || buf[i] == '-') buf[i] = ':';
4600 lprintf ("fcbuf=%s\n", buf);
4601 pat = FcNameParse ((FcChar8 *) buf);
4602 if (!pat) {
4603 printd ("emsg FcNameParse failed\n");
4604 free (buf);
4605 return NULL;
4608 if (!FcConfigSubstitute (fc.config, pat, FcMatchPattern)) {
4609 printd ("emsg FcConfigSubstitute failed\n");
4610 free (buf);
4611 return NULL;
4613 FcDefaultSubstitute (pat);
4615 pat1 = FcFontMatch (fc.config, pat, &result);
4616 if (!pat1) {
4617 printd ("emsg FcFontMatch failed\n");
4618 FcPatternDestroy (pat);
4619 free (buf);
4620 return NULL;
4623 if (FcPatternGetString (pat1, FC_FILE, 0, &path) != FcResultMatch) {
4624 printd ("emsg FcPatternGetString failed\n");
4625 FcPatternDestroy (pat);
4626 FcPatternDestroy (pat1);
4627 free (buf);
4628 return NULL;
4631 #if 0
4632 printd ("emsg name=%s path=%s\n", name, path);
4633 #endif
4634 font = fz_new_font_from_file (ctx, name, (char *) path, 0, 0);
4635 FcPatternDestroy (pat);
4636 FcPatternDestroy (pat1);
4637 free (buf);
4638 return font;
4640 #endif
4642 CAMLprim value ml_init (value csock_v, value params_v)
4644 CAMLparam2 (csock_v, params_v);
4645 CAMLlocal2 (trim_v, fuzz_v);
4646 int ret;
4647 int texcount;
4648 char *fontpath;
4649 int colorspace;
4650 int mustoresize;
4651 int haspboext;
4653 state.csock = Int_val (csock_v);
4654 state.rotate = Int_val (Field (params_v, 0));
4655 state.fitmodel = Int_val (Field (params_v, 1));
4656 trim_v = Field (params_v, 2);
4657 texcount = Int_val (Field (params_v, 3));
4658 state.sliceheight = Int_val (Field (params_v, 4));
4659 mustoresize = Int_val (Field (params_v, 5));
4660 colorspace = Int_val (Field (params_v, 6));
4661 fontpath = String_val (Field (params_v, 7));
4663 if (caml_string_length (Field (params_v, 8)) > 0) {
4664 state.trimcachepath = strdup (String_val (Field (params_v, 8)));
4666 if (!state.trimcachepath) {
4667 fprintf (stderr, "failed to strdup trimcachepath: %s\n",
4668 strerror (errno));
4671 haspboext = Bool_val (Field (params_v, 9));
4673 state.ctx = fz_new_context (NULL, NULL, mustoresize);
4675 #ifdef USE_FONTCONFIG
4676 if (Bool_val (Field (params_v, 10))) {
4677 fz_install_load_system_font_funcs (
4678 state.ctx, fc_load_system_font_func, NULL
4681 #endif
4683 state.trimmargins = Bool_val (Field (trim_v, 0));
4684 fuzz_v = Field (trim_v, 1);
4685 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
4686 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
4687 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
4688 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
4690 set_tex_params (colorspace);
4692 if (*fontpath) {
4693 #ifndef USE_FONTCONFIG
4694 state.face = load_font (fontpath);
4695 #else
4696 FcChar8 *path;
4697 FcResult result;
4698 char *buf = fontpath;
4699 FcPattern *pat, *pat1;
4701 fc.inited = 1;
4702 fc.config = FcInitLoadConfigAndFonts ();
4703 if (!fc.config) {
4704 errx (1, "FcInitLoadConfigAndFonts failed");
4707 pat = FcNameParse ((FcChar8 *) buf);
4708 if (!pat) {
4709 errx (1, "FcNameParse failed");
4712 if (!FcConfigSubstitute (fc.config, pat, FcMatchPattern)) {
4713 errx (1, "FcConfigSubstitute failed");
4715 FcDefaultSubstitute (pat);
4717 pat1 = FcFontMatch (fc.config, pat, &result);
4718 if (!pat1) {
4719 errx (1, "FcFontMatch failed");
4722 if (FcPatternGetString (pat1, FC_FILE, 0, &path) != FcResultMatch) {
4723 errx (1, "FcPatternGetString failed");
4726 state.face = load_font ((char *) path);
4727 FcPatternDestroy (pat);
4728 FcPatternDestroy (pat1);
4729 #endif
4731 else {
4732 unsigned int len;
4733 void *base = pdf_lookup_substitute_font (0, 0, 0, 0, &len);
4735 state.face = load_builtin_font (base, len);
4737 if (!state.face) _exit (1);
4739 realloctexts (texcount);
4741 if (haspboext) {
4742 setuppbo ();
4745 makestippletex ();
4747 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
4748 if (ret) {
4749 errx (1, "pthread_create: %s", strerror (ret));
4752 CAMLreturn (Val_unit);