Robustness
[llpp.git] / link.c
blobe567fb0e1e84d29d0032f0c8b48b590d389428c9
1 /* lots of code c&p-ed directly from mupdf */
2 #include <errno.h>
3 #include <stdio.h>
4 #include <ctype.h>
5 #include <string.h>
6 #include <stdlib.h>
7 #include <signal.h>
9 #include <unistd.h>
10 #include <pthread.h>
11 #include <sys/time.h>
12 #include <sys/types.h>
13 #include <sys/ioctl.h>
15 #ifdef __CYGWIN__
16 #include <cygwin/socket.h> /* FIONREAD */
17 #else
18 #include <spawn.h>
19 #endif
21 #include <regex.h>
22 #include <ctype.h>
23 #include <stdarg.h>
24 #include <limits.h>
26 #include <GL/gl.h>
28 #include <caml/fail.h>
29 #include <caml/alloc.h>
30 #include <caml/memory.h>
31 #include <caml/unixsupport.h>
33 #include <fitz.h>
34 #include <mupdf.h>
35 #include <mupdf-internal.h>
36 #include <muxps.h>
37 #include <muxps-internal.h>
38 #include <mucbz.h>
40 #include FT_FREETYPE_H
42 #define PIGGYBACK
44 #ifndef __USE_GNU
45 extern char **environ;
46 #endif
48 #if defined __GNUC__
49 #define NORETURN __attribute__ ((noreturn))
50 #define UNUSED __attribute__ ((unused))
51 #define OPTIMIZE(n) __attribute__ ((optimize ("O"#n)))
52 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
53 #else
54 #define NORETURN
55 #define UNUSED
56 #define OPTIMIZE(n)
57 #define GCC_FMT_ATTR(a, b)
58 #endif
60 #define FMT_s "zu"
62 #define FMT_ptr "p"
63 #define FMT_ptr_cast(p) (p)
64 #define FMT_ptr_cast2(p) (p)
66 static void NORETURN GCC_FMT_ATTR (2, 3)
67 err (int exitcode, const char *fmt, ...)
69 va_list ap;
70 int savederrno;
72 savederrno = errno;
73 va_start (ap, fmt);
74 vfprintf (stderr, fmt, ap);
75 va_end (ap);
76 fprintf (stderr, ": %s\n", strerror (savederrno));
77 fflush (stderr);
78 _exit (exitcode);
81 static void NORETURN GCC_FMT_ATTR (2, 3)
82 errx (int exitcode, const char *fmt, ...)
84 va_list ap;
86 va_start (ap, fmt);
87 vfprintf (stderr, fmt, ap);
88 va_end (ap);
89 fputc ('\n', stderr);
90 fflush (stderr);
91 _exit (exitcode);
94 #ifndef GL_TEXTURE_RECTANGLE_ARB
95 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
96 #endif
98 #ifndef GL_BGRA
99 #define GL_BGRA 0x80E1
100 #endif
102 #ifndef GL_UNSIGNED_INT_8_8_8_8
103 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
104 #endif
106 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
107 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
108 #endif
110 #if 0
111 #define lprintf printf
112 #else
113 #define lprintf(...)
114 #endif
116 #define ARSERT(cond) for (;;) { \
117 if (!(cond)) { \
118 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
120 break; \
123 struct slice {
124 int h;
125 int texindex;
128 struct tile {
129 int x, y, w, h;
130 int slicecount;
131 int sliceheight;
132 fz_pixmap *pixmap;
133 struct slice slices[1];
136 struct pagedim {
137 int pageno;
138 int rotate;
139 int left;
140 int tctmready;
141 fz_bbox bounds;
142 fz_rect pagebox;
143 fz_rect mediabox;
144 fz_matrix ctm, zoomctm, lctm, tctm;
147 struct slink {
148 fz_bbox bbox;
149 fz_link *link;
152 enum { DPDF, DXPS, DCBZ };
154 struct page {
155 int tgen;
156 int sgen;
157 int type;
158 int pageno;
159 int pdimno;
160 fz_text_page *text;
161 fz_text_sheet *sheet;
162 union {
163 void *ptr;
164 pdf_page *pdfpage;
165 xps_page *xpspage;
166 cbz_page *cbzpage;
167 } u;
168 fz_display_list *dlist;
169 int slinkcount;
170 struct slink *slinks;
171 struct mark {
172 int i;
173 fz_text_span *span;
174 } fmark, lmark;
175 void (*freepage) (void *);
178 struct {
179 int type;
180 int sliceheight;
181 struct pagedim *pagedims;
182 int pagecount;
183 int pagedimcount;
184 union {
185 pdf_document *pdf;
186 xps_document *xps;
187 cbz_document *cbz;
188 } u;
189 fz_context *ctx;
190 int w, h;
192 int texindex;
193 int texcount;
194 GLuint *texids;
196 GLenum texiform;
197 GLenum texform;
198 GLenum texty;
200 fz_colorspace *colorspace;
202 struct {
203 int w, h;
204 struct slice *slice;
205 } *texowners;
207 int rotate;
208 int proportional;
209 int trimmargins;
210 int needoutline;
211 int gen;
212 int aalevel;
214 int trimanew;
215 fz_bbox trimfuzz;
216 fz_pixmap *pig;
218 pthread_t thread;
219 int cr, cw;
220 FT_Face face;
222 void (*closedoc) (void);
223 void (*freepage) (void *);
224 } state;
226 static void UNUSED debug_rect (const char *cap, fz_rect r)
228 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
231 static void UNUSED debug_bbox (const char *cap, fz_bbox r)
233 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
236 static void UNUSED debug_matrix (const char *cap, fz_matrix m)
238 printf ("%s(matrix) %.2f,%.2f,%.2f,%.2f %.2f %.2f\n", cap,
239 m.a, m.b, m.c, m.d, m.e, m.f);
242 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
244 static void lock (const char *cap)
246 int ret = pthread_mutex_lock (&mutex);
247 if (ret) {
248 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
252 static void unlock (const char *cap)
254 int ret = pthread_mutex_unlock (&mutex);
255 if (ret) {
256 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
260 static int trylock (const char *cap)
262 int ret = pthread_mutex_trylock (&mutex);
264 if (ret && ret != EBUSY) {
265 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
267 return ret == EBUSY;
270 static void *parse_pointer (const char *cap, const char *s)
272 int ret;
273 void *ptr;
275 ret = sscanf (s, "%" FMT_ptr, FMT_ptr_cast (&ptr));
276 if (ret != 1) {
277 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
279 return ptr;
282 static double now (void)
284 struct timeval tv;
286 if (gettimeofday (&tv, NULL)) {
287 err (1, "gettimeofday");
289 return tv.tv_sec + tv.tv_usec*1e-6;
292 static int hasdata (void)
294 int ret, avail;
295 ret = ioctl (state.cr, FIONREAD, &avail);
296 if (ret) err (1, "hasdata: FIONREAD error ret=%d", ret);
297 return avail > 0;
300 CAMLprim value ml_hasdata (value fd_v)
302 CAMLparam1 (fd_v);
303 int ret, avail;
305 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
306 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
307 CAMLreturn (Val_bool (avail > 0));
310 static void readdata (void *p, int size)
312 ssize_t n;
314 n = read (state.cr, p, size);
315 if (n - size) {
316 if (!n) errx (1, "EOF while reading");
317 err (1, "read (req %d, ret %zd)", size, n);
321 static void writedata (char *p, int size)
323 char buf[4];
324 ssize_t n;
326 buf[0] = (size >> 24) & 0xff;
327 buf[1] = (size >> 16) & 0xff;
328 buf[2] = (size >> 8) & 0xff;
329 buf[3] = (size >> 0) & 0xff;
331 n = write (state.cw, buf, 4);
332 if (n != 4) {
333 if (!n) errx (1, "EOF while writing length");
334 err (1, "write %zd", n);
337 n = write (state.cw, p, size);
338 if (n - size) {
339 if (!n) errx (1, "EOF while writing data");
340 err (1, "write (req %d, ret %zd)", size, n);
344 static int readlen (void)
346 unsigned char p[4];
348 readdata (p, 4);
349 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
352 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
354 int size = 200, len;
355 va_list ap;
356 char *buf;
358 buf = malloc (size);
359 for (;;) {
360 if (!buf) err (errno, "malloc for temp buf (%d bytes) failed", size);
362 va_start (ap, fmt);
363 len = vsnprintf (buf, size, fmt, ap);
364 va_end (ap);
366 if (len > -1 && len < size) {
367 writedata (buf, len);
368 break;
371 if (len > -1) {
372 size = len + 1;
374 else {
375 size *= 2;
377 buf = realloc (buf, size);
379 free (buf);
382 static void closepdf (void)
384 if (state.u.pdf) {
385 pdf_close_document (state.u.pdf);
386 state.u.pdf = NULL;
390 static void closexps (void)
392 if (state.u.xps) {
393 xps_close_document (state.u.xps);
394 state.u.xps = NULL;
398 static void closecbz (void)
400 if (state.u.cbz) {
401 cbz_close_document (state.u.cbz);
402 state.u.cbz = NULL;
406 static void freepdfpage (void *ptr)
408 pdf_free_page (state.u.pdf, ptr);
411 static void freeemptypdfpage (void *ptr)
413 (void) ptr;
416 static void freexpspage (void *ptr)
418 xps_free_page (state.u.xps, ptr);
421 static void freecbzpage (void *ptr)
423 cbz_free_page (state.u.cbz, ptr);
426 static void openxref (char *filename, char *password)
428 int i, len;
430 for (i = 0; i < state.texcount; ++i) {
431 state.texowners[i].w = -1;
432 state.texowners[i].slice = NULL;
435 if (state.closedoc) state.closedoc ();
437 len = strlen (filename);
439 state.type = DPDF;
440 if (len > 4) {
441 char ext[4];
443 ext[0] = tolower ((int) filename[len-3]);
444 ext[1] = tolower ((int) filename[len-2]);
445 ext[2] = tolower ((int) filename[len-1]);
447 /**/ if (ext[0] == 'x' && ext[1] == 'p' && ext[2] == 's') {
448 state.type = DXPS;
450 else if (ext[0] == 'c' && ext[1] == 'b' && ext[2] == 'z') {
451 state.type = DCBZ;
455 if (state.pagedims) {
456 free (state.pagedims);
457 state.pagedims = NULL;
459 state.pagedimcount = 0;
461 fz_set_aa_level (state.ctx, state.aalevel);
462 switch (state.type) {
463 case DPDF:
464 state.u.pdf = pdf_open_document (state.ctx, filename);
465 if (pdf_needs_password (state.u.pdf)) {
466 int okay = pdf_authenticate_password (state.u.pdf, password);
467 if (!okay) {
468 errx (1, "invalid password");
471 state.pagecount = pdf_count_pages (state.u.pdf);
472 state.closedoc = closepdf;
473 state.freepage = freepdfpage;
474 break;
476 case DXPS:
477 state.u.xps = xps_open_document (state.ctx, filename);
478 state.pagecount = xps_count_pages (state.u.xps);
479 state.closedoc = closexps;
480 state.freepage = freexpspage;
481 break;
483 case DCBZ:
484 state.u.cbz = cbz_open_document (state.ctx, filename);
485 state.pagecount = cbz_count_pages (state.u.cbz);
486 state.closedoc = closecbz;
487 state.freepage = freecbzpage;
488 break;
492 static void pdfinfo (void)
494 if (state.type == DPDF) {
495 pdf_obj *infoobj;
497 printd ("info PDF version\t%d.%d",
498 state.u.pdf->version / 10, state.u.pdf->version % 10);
500 infoobj = pdf_dict_gets (state.u.pdf->trailer, "Info");
501 if (infoobj) {
502 int i;
503 char *s;
504 char *items[] = { "Title", "Author", "Creator",
505 "Producer", "CreationDate" };
507 for (i = 0; i < sizeof (items) / sizeof (*items); ++i) {
508 pdf_obj *obj = pdf_dict_gets (infoobj, items[i]);
509 s = pdf_to_utf8 (state.ctx, obj);
510 if (*s) {
511 if (i == 0) {
512 printd ("title %s", s);
514 printd ("info %s\t%s", items[i], s);
516 fz_free (state.ctx, s);
519 printd ("infoend");
523 static void unlinktile (struct tile *tile)
525 int i;
527 for (i = 0; i < tile->slicecount; ++i) {
528 struct slice *s = &tile->slices[i];
530 if (s->texindex != -1) {
531 if (state.texowners[s->texindex].slice == s) {
532 state.texowners[s->texindex].slice = NULL;
538 static void freepage (struct page *page)
540 if (page->text) {
541 fz_free_text_page (state.ctx, page->text);
543 if (page->sheet) {
544 fz_free_text_sheet (state.ctx, page->sheet);
546 if (page->slinks) {
547 free (page->slinks);
549 page->freepage (page->u.ptr);
550 fz_free_display_list (state.ctx, page->dlist);
551 free (page);
554 static void freetile (struct tile *tile)
556 unlinktile (tile);
557 #ifndef PIGGYBACK
558 fz_drop_pixmap (state.ctx, tile->pixmap);
559 #else
560 if (state.pig) {
561 fz_drop_pixmap (state.ctx, state.pig);
563 state.pig = tile->pixmap;
564 #endif
565 free (tile);
568 #ifdef __ALTIVEC__
569 #include <altivec.h>
571 static int cacheline32bytes;
573 static void __attribute__ ((constructor)) clcheck (void)
575 char **envp = environ;
576 unsigned long *auxv;
578 while (*envp++);
580 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
581 if (*auxv == 19) {
582 cacheline32bytes = auxv[1] == 32;
583 return;
588 static void OPTIMIZE (3) clearpixmap (fz_pixmap *pixmap)
590 if (cacheline32bytes) {
591 intptr_t a1, a2, diff;
592 size_t sizea, i, size = pixmap->w * pixmap->h * pixmap->n;
593 vector unsigned char v = vec_splat_u8 (-1);
594 vector unsigned char *p;
596 a1 = a2 = (intptr_t) pixmap->samples;
597 a2 = (a1 + 31) & ~31;
598 diff = a2 - a1;
599 sizea = size - diff;
600 p = (void *) a2;
602 while (a1 != a2) *(char *) a1++ = 0xff;
603 for (i = 0; i < (sizea & ~31); i += 32) {
604 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
605 vec_st (v, i, p);
606 vec_st (v, i + 16, p);
608 while (i < sizea) *((char *) a1 + i++) = 0xff;
610 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
612 #else
613 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
614 #endif
616 static fz_matrix trimctm (pdf_page *page, int pindex)
618 fz_matrix ctm;
619 struct pagedim *pdim = &state.pagedims[pindex];
621 if (!pdim->tctmready) {
622 if (state.trimmargins) {
623 fz_rect realbox;
625 ctm = fz_concat (fz_rotate (-pdim->rotate), fz_scale (1, -1));
626 realbox = fz_transform_rect (ctm, pdim->mediabox);
627 ctm = fz_concat (ctm, fz_translate (-realbox.x0, -realbox.y0));
628 ctm = fz_concat (fz_invert_matrix (page->ctm), ctm);
630 else {
631 ctm = fz_identity;
633 pdim->tctm = ctm;
634 pdim->tctmready = 1;
636 return pdim->tctm;
639 static fz_matrix pagectm (struct page *page)
641 if (page->type == DPDF) {
642 return fz_concat (trimctm (page->u.pdfpage, page->pdimno),
643 state.pagedims[page->pdimno].ctm);
645 else {
646 fz_matrix ctm;
647 struct pagedim *pdim = &state.pagedims[page->pdimno];
649 ctm = state.pagedims[page->pdimno].ctm;
650 ctm = fz_concat (fz_translate (-pdim->mediabox.x0,
651 -pdim->mediabox.y0), ctm);
652 return ctm;
656 static void *loadpage (int pageno, int pindex)
658 fz_device *dev;
659 struct page *page = NULL;
661 page = calloc (sizeof (struct page), 1);
662 if (!page) {
663 err (1, "calloc page %d", pageno);
666 page->dlist = fz_new_display_list (state.ctx);
667 dev = fz_new_list_device (state.ctx, page->dlist);
668 switch (state.type) {
669 case DPDF:
670 fz_try (state.ctx) {
671 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
672 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, fz_identity, NULL);
673 page->freepage = freepdfpage;
675 fz_catch (state.ctx) {
676 page->u.ptr = NULL;
677 page->freepage = freeemptypdfpage;
679 break;
681 case DXPS:
682 page->u.xpspage = xps_load_page (state.u.xps, pageno);
683 xps_run_page (state.u.xps, page->u.xpspage, dev, fz_identity, NULL);
684 page->freepage = freexpspage;
685 break;
687 case DCBZ:
688 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
689 cbz_run_page (state.u.cbz, page->u.cbzpage, dev, fz_identity, NULL);
690 page->freepage = freecbzpage;
691 break;
693 fz_free_device (dev);
695 page->pdimno = pindex;
696 page->pageno = pageno;
697 page->sgen = state.gen;
698 page->tgen = state.gen;
699 page->type = state.type;
701 return page;
704 static struct tile *alloctile (int h)
706 int i;
707 int slicecount;
708 size_t tilesize;
709 struct tile *tile;
711 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
712 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
713 tile = calloc (tilesize, 1);
714 if (!tile) {
715 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
717 for (i = 0; i < slicecount; ++i) {
718 int sh = MIN (h, state.sliceheight);
719 tile->slices[i].h = sh;
720 tile->slices[i].texindex = -1;
721 h -= sh;
723 tile->slicecount = slicecount;
724 tile->sliceheight = state.sliceheight;
725 return tile;
728 struct obs {
729 int cured;
730 fz_bbox b;
733 static void obs_fill_image (fz_device *dev, fz_image *image, fz_matrix ctm,
734 float alpha)
736 struct obs *obs = dev->user;
738 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
739 fz_bbox b = fz_round_rect (fz_transform_rect (ctm, fz_unit_rect));
740 b = fz_intersect_bbox (b, obs->b);
741 obs->cured = b.x0 == obs->b.x0
742 && b.x1 == obs->b.x1
743 && b.y0 == obs->b.y0
744 && b.y1 == obs->b.y1;
748 static int obscured (struct page *page, fz_bbox bbox)
750 fz_device dev;
751 struct obs obs;
753 memset (&dev, 0, sizeof (dev));
754 memset (&obs, 0, sizeof (obs));
755 dev.hints = 0;
756 dev.flags = 0;
757 dev.user = &obs;
758 dev.ctx = state.ctx;
759 dev.fill_image = obs_fill_image;
760 obs.b = bbox;
761 fz_run_display_list (page->dlist, &dev, pagectm (page), bbox, NULL);
762 return obs.cured;
765 static struct tile *rendertile (struct page *page, int x, int y, int w, int h)
767 fz_bbox bbox;
768 fz_device *dev;
769 struct tile *tile;
770 struct pagedim *pdim;
772 tile = alloctile (h);
773 pdim = &state.pagedims[page->pdimno];
775 bbox = pdim->bounds;
776 bbox.x0 += x;
777 bbox.y0 += y;
778 bbox.x1 = bbox.x0 + w;
779 bbox.y1 = bbox.y0 + h;
781 if (state.pig) {
782 if (state.pig->w == w
783 && state.pig->h == h
784 && state.pig->colorspace == state.colorspace) {
785 tile->pixmap = state.pig;
786 tile->pixmap->x = bbox.x0;
787 tile->pixmap->y = bbox.y0;
789 else {
790 fz_drop_pixmap (state.ctx, state.pig);
792 state.pig = NULL;
794 if (!tile->pixmap) {
795 tile->pixmap =
796 fz_new_pixmap_with_bbox (state.ctx, state.colorspace, bbox);
799 tile->w = w;
800 tile->h = h;
801 if (!page->u.ptr || ((w < 128 && h < 128) || !obscured (page, bbox))) {
802 clearpixmap (tile->pixmap);
804 dev = fz_new_draw_device (state.ctx, tile->pixmap);
805 fz_run_display_list (page->dlist, dev, pagectm (page), bbox, NULL);
806 fz_free_device (dev);
808 return tile;
811 static void initpdims (void)
813 double start, end;
814 int pageno, trim, show;
816 start = now ();
817 for (pageno = 0; pageno < state.pagecount; ++pageno) {
818 int rotate = 0;
819 struct pagedim *p;
820 fz_rect mediabox;
822 switch (state.type) {
823 case DPDF: {
824 pdf_obj *pageobj = state.u.pdf->page_objs[pageno];
826 if (state.trimmargins) {
827 pdf_obj *obj;
828 pdf_page *page;
830 fz_try (state.ctx) {
831 page = pdf_load_page (state.u.pdf, pageno);
832 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
833 trim = state.trimanew || !obj;
834 if (trim) {
835 fz_rect rect;
836 fz_bbox bbox;
837 fz_matrix ctm;
838 fz_device *dev;
840 dev = fz_new_bbox_device (state.ctx, &bbox);
841 dev->hints |= FZ_IGNORE_SHADE;
842 ctm = fz_invert_matrix (page->ctm);
843 pdf_run_page (state.u.pdf, page, dev, fz_identity, NULL);
844 fz_free_device (dev);
846 rect.x0 = bbox.x0 + state.trimfuzz.x0;
847 rect.x1 = bbox.x1 + state.trimfuzz.x1;
848 rect.y0 = bbox.y0 + state.trimfuzz.y0;
849 rect.y1 = bbox.y1 + state.trimfuzz.y1;
850 rect = fz_transform_rect (ctm, rect);
851 rect = fz_intersect_rect (rect, page->mediabox);
853 if (fz_is_empty_rect (rect)) {
854 mediabox = page->mediabox;
856 else {
857 mediabox = rect;
860 obj = pdf_new_array (state.ctx, 4);
861 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x0));
862 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y0));
863 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x1));
864 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y1));
865 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
867 else {
868 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
869 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
870 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
871 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
874 rotate = page->rotate;
875 pdf_free_page (state.u.pdf, page);
877 show = trim ? pageno % 5 == 0 : pageno % 20 == 0;
878 if (show) {
879 printd ("progress %f Trimming %d",
880 (double) (pageno + 1) / state.pagecount,
881 pageno + 1);
884 fz_catch (state.ctx) {
887 else {
888 fz_rect cropbox;
890 mediabox = pdf_to_rect (state.ctx,
891 pdf_dict_gets (pageobj, "MediaBox"));
892 if (fz_is_empty_rect (mediabox)) {
893 fprintf (stderr, "cannot find page size for page %d\n",
894 pageno+1);
895 mediabox.x0 = 0;
896 mediabox.y0 = 0;
897 mediabox.x1 = 612;
898 mediabox.y1 = 792;
901 cropbox = pdf_to_rect (state.ctx,
902 pdf_dict_gets (pageobj, "CropBox"));
903 if (!fz_is_empty_rect (cropbox)) {
904 mediabox = fz_intersect_rect (mediabox, cropbox);
906 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
908 break;
911 case DXPS:
913 xps_page *page;
915 page = xps_load_page (state.u.xps, pageno);
916 mediabox = xps_bound_page (state.u.xps, page);
917 rotate = 0;
918 if (state.trimmargins) {
919 fz_rect rect;
920 fz_bbox bbox;
921 fz_device *dev;
923 dev = fz_new_bbox_device (state.ctx, &bbox);
924 dev->hints |= FZ_IGNORE_SHADE;
925 xps_run_page (state.u.xps, page, dev, fz_identity, NULL);
926 fz_free_device (dev);
928 rect.x0 = bbox.x0 + state.trimfuzz.x0;
929 rect.x1 = bbox.x1 + state.trimfuzz.x1;
930 rect.y0 = bbox.y0 + state.trimfuzz.y0;
931 rect.y1 = bbox.y1 + state.trimfuzz.y1;
932 rect = fz_intersect_rect (rect, mediabox);
934 if (!fz_is_empty_rect (rect)) {
935 mediabox = rect;
938 xps_free_page (state.u.xps, page);
939 printd ("progress %f loading %d",
940 (double) (pageno + 1) / state.pagecount,
941 pageno + 1);
943 break;
945 case DCBZ:
947 rotate = 0;
948 if (state.trimmargins) {
949 cbz_page *page;
951 page = cbz_load_page (state.u.cbz, pageno);
952 mediabox = cbz_bound_page (state.u.cbz, page);
953 cbz_free_page (state.u.cbz, page);
954 printd ("progress %f Trimming %d",
955 (double) (pageno + 1) / state.pagecount,
956 pageno + 1);
958 else {
959 mediabox.x0 = mediabox.y0 = 0;
960 mediabox.x1 = 900;
961 mediabox.y1 = 900;
964 break;
966 default:
967 ARSERT (0 && state.type);
970 if (state.pagedimcount == 0
971 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
972 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
973 size_t size;
975 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
976 state.pagedims = realloc (state.pagedims, size);
977 if (!state.pagedims) {
978 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
979 size, state.pagedimcount + 1);
982 p = &state.pagedims[state.pagedimcount++];
983 p->rotate = rotate;
984 p->mediabox = mediabox;
985 p->pageno = pageno;
988 end = now ();
989 if (state.trimmargins) {
990 printd ("progress 1 Trimmed %d pages in %f seconds",
991 state.pagecount, end - start);
993 else {
994 printd ("vmsg Processed %d pages in %f seconds",
995 state.pagecount, end - start);
997 state.trimanew = 0;
1000 static void layout (void)
1002 int pindex;
1003 fz_rect box;
1004 fz_matrix ctm;
1005 double zoom, w, maxw = 0;
1006 struct pagedim *p = state.pagedims;
1008 if (state.proportional) {
1009 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
1010 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
1011 p->mediabox);
1012 w = box.x1 - box.x0;
1013 maxw = MAX (w, maxw);
1017 p = state.pagedims;
1018 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
1019 fz_bbox bbox;
1021 ctm = fz_rotate (state.rotate);
1022 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
1023 p->mediabox);
1024 w = box.x1 - box.x0;
1026 if (state.proportional) {
1027 double scale = w / maxw;
1028 zoom = (state.w / w) * scale;
1030 else {
1031 zoom = state.w / w;
1034 p->zoomctm = fz_scale (zoom, zoom);
1035 ctm = fz_concat (p->zoomctm, ctm);
1037 p->pagebox = fz_transform_rect (fz_rotate (p->rotate), p->mediabox);
1038 p->pagebox.x1 -= p->pagebox.x0;
1039 p->pagebox.y1 -= p->pagebox.y0;
1040 p->pagebox.x0 = 0;
1041 p->pagebox.y0 = 0;
1042 bbox = fz_round_rect (fz_transform_rect (ctm, p->pagebox));
1044 p->bounds = bbox;
1045 p->left = state.proportional ? ((maxw - w) * zoom) / 2.0 : 0;
1046 p->ctm = ctm;
1048 ctm = fz_identity;
1049 ctm = fz_concat (ctm, fz_translate (0, -p->mediabox.y1));
1050 ctm = fz_concat (ctm, fz_scale (zoom, -zoom));
1051 ctm = fz_concat (ctm, fz_rotate (p->rotate + state.rotate));
1052 p->lctm = ctm;
1054 p->tctmready = 0;
1057 while (p-- != state.pagedims) {
1058 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1059 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1060 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1061 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1062 int w = x1 - x0;
1063 int h = y1 - y0;
1065 printd ("pdim %d %d %d %d", p->pageno, w, h, p->left);
1069 static void recurse_outline (fz_outline *outline, int level)
1071 while (outline) {
1072 fz_link_dest *dest;
1073 int i, top = 0;
1074 struct pagedim *pdim = state.pagedims;
1076 dest = &outline->dest;
1077 for (i = 0; i < state.pagedimcount; ++i) {
1078 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1079 break;
1080 pdim = &state.pagedims[i];
1082 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1083 fz_point p;
1084 p.x = 0;
1085 p.y = dest->ld.gotor.lt.y;
1086 p = fz_transform_point (pdim->lctm, p);
1087 top = p.y;
1089 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1090 int h;
1091 double y0, y1;
1093 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1094 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1095 h = y1 - y0;
1096 printd ("o %d %d %d %d %s",
1097 level, dest->ld.gotor.page, top, h, outline->title);
1099 if (outline->down) {
1100 recurse_outline (outline->down, level + 1);
1102 outline = outline->next;
1106 static void process_outline (void)
1108 fz_outline *outline;
1110 if (!state.needoutline) return;
1112 state.needoutline = 0;
1113 switch (state.type) {
1114 case DPDF:
1115 outline = pdf_load_outline (state.u.pdf);
1116 break;
1117 case DXPS:
1118 outline = xps_load_outline (state.u.xps);
1119 break;
1120 default:
1121 outline = NULL;
1122 break;
1124 if (outline) {
1125 recurse_outline (outline, 0);
1126 fz_free_outline (state.ctx, outline);
1130 static char *strofspan (fz_text_span *span)
1132 char *p;
1133 char utf8[10];
1134 fz_text_char *ch;
1135 size_t size = 0, cap = 80;
1137 p = malloc (cap + 1);
1138 if (!p) return NULL;
1140 for (ch = span->text; ch < span->text + span->len; ++ch) {
1141 int n = fz_runetochar (utf8, ch->c);
1142 if (size + n > cap) {
1143 cap *= 2;
1144 p = realloc (p, cap + 1);
1145 if (!p) return NULL;
1148 memcpy (p + size, utf8, n);
1149 size += n;
1151 p[size] = 0;
1152 return p;
1155 static int matchspan (regex_t *re, fz_text_span *span, fz_matrix ctm,
1156 int stop, int pageno, double start)
1158 int ret;
1159 char *p;
1160 regmatch_t rm;
1161 int a, b, c;
1162 fz_rect *sb, *eb;
1163 fz_point p1, p2, p3, p4;
1165 p = strofspan (span);
1166 if (!p) return -1;
1168 ret = regexec (re, p, 1, &rm, 0);
1169 if (ret) {
1170 free (p);
1171 if (ret != REG_NOMATCH) {
1172 size_t size;
1173 char errbuf[80];
1174 size = regerror (ret, re, errbuf, sizeof (errbuf));
1175 printd ("msg regexec error `%.*s'",
1176 (int) size, errbuf);
1177 return -1;
1179 return 0;
1181 else {
1182 int l = span->len;
1183 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1184 c += fz_runelen (span->text[a].c);
1186 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1187 c += fz_runelen (span->text[b].c);
1190 if (fz_runelen (span->text[b].c) > 1) {
1191 b = MAX (0, b-1);
1193 sb = &span->text[MIN (a, l-1)].bbox;
1194 eb = &span->text[MIN (b, l-1)].bbox;
1196 p1.x = sb->x0;
1197 p1.y = sb->y0;
1198 p2.x = eb->x1;
1199 p2.y = sb->y0;
1200 p3.x = eb->x1;
1201 p3.y = eb->y1;
1202 p4.x = sb->x0;
1203 p4.y = eb->y1;
1205 if (!stop) {
1206 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1207 pageno, 1,
1208 p1.x, p1.y,
1209 p2.x, p2.y,
1210 p3.x, p3.y,
1211 p4.x, p4.y);
1213 printd ("progress 1 found at %d `%.*s' in %f sec",
1214 pageno + 1, (int) (rm.rm_eo - rm.rm_so), &p[rm.rm_so],
1215 now () - start);
1217 else {
1218 printd ("match %d %d %f %f %f %f %f %f %f %f",
1219 pageno, 2,
1220 p1.x, p1.y,
1221 p2.x, p2.y,
1222 p3.x, p3.y,
1223 p4.x, p4.y);
1225 free (p);
1226 return 1;
1230 static int compareblocks (const void *l, const void *r)
1232 fz_text_block const *ls = l;
1233 fz_text_block const* rs = r;
1234 return ls->bbox.y0 - rs->bbox.y0;
1237 /* wishful thinking function */
1238 static void search (regex_t *re, int pageno, int y, int forward)
1240 int i, j;
1241 fz_matrix ctm;
1242 fz_device *tdev;
1243 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1244 fz_text_page *text;
1245 fz_text_sheet *sheet;
1246 struct pagedim *pdim, *pdimprev;
1247 int stop = 0, niters = 0;
1248 double start, end;
1250 if (!(state.type == DPDF || state.type == DXPS))
1251 return;
1253 start = now ();
1254 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1255 if (niters++ == 5) {
1256 niters = 0;
1257 if (hasdata ()) {
1258 printd ("progress 1 attention requested aborting search at %d",
1259 pageno);
1260 stop = 1;
1262 else {
1263 printd ("progress %f searching in page %d",
1264 (double) (pageno + 1) / state.pagecount,
1265 pageno);
1268 pdimprev = NULL;
1269 for (i = 0; i < state.pagedimcount; ++i) {
1270 pdim = &state.pagedims[i];
1271 if (pdim->pageno == pageno) {
1272 goto found;
1274 if (pdim->pageno > pageno) {
1275 pdim = pdimprev;
1276 goto found;
1278 pdimprev = pdim;
1280 pdim = pdimprev;
1281 found:
1283 sheet = fz_new_text_sheet (state.ctx);
1284 text = fz_new_text_page (state.ctx, fz_infinite_rect);
1285 tdev = fz_new_text_device (state.ctx, sheet, text);
1287 switch (state.type) {
1288 case DPDF:
1289 u.ptr = NULL;
1290 fz_try (state.ctx) {
1291 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1292 trimctm (u.pdfpage, pdim - state.pagedims);
1293 ctm = fz_concat (pdim->tctm, pdim->zoomctm);
1294 pdf_run_page (state.u.pdf, u.pdfpage, tdev, ctm, NULL);
1296 fz_catch (state.ctx) {
1297 fz_free_device (tdev);
1298 u.ptr = NULL;
1299 goto nextiter;
1301 break;
1303 case DXPS:
1304 u.xpspage = xps_load_page (state.u.xps, pageno);
1305 ctm = pdim->ctm;
1306 xps_run_page (state.u.xps, u.xpspage, tdev, ctm, NULL);
1307 break;
1309 default:
1310 ARSERT (0 && state.type);
1313 qsort (text->blocks, text->len, sizeof (*text->blocks), compareblocks);
1314 fz_free_device (tdev);
1316 for (j = 0; j < text->len; ++j) {
1317 int k;
1318 fz_text_block *block;
1320 block = &text->blocks[forward ? j : text->len - 1 - j];
1322 for (k = 0; k < block->len; ++k) {
1323 fz_text_line *line;
1324 fz_text_span *span;
1326 if (forward) {
1327 line = &block->lines[k];
1328 if (line->bbox.y0 < y + 1) continue;
1330 else {
1331 line = &block->lines[block->len - 1 - k];
1332 if (line->bbox.y0 > y - 1) continue;
1335 for (span = line->spans;
1336 span < line->spans + line->len;
1337 ++span) {
1339 switch (matchspan (re, span, ctm, stop, pageno, start)) {
1340 case 0: break;
1341 case 1: stop = 1; break;
1342 case -1: stop = 1; goto endloop;
1347 nextiter:
1348 if (forward) {
1349 pageno += 1;
1350 y = 0;
1352 else {
1353 pageno -= 1;
1354 y = INT_MAX;
1356 endloop:
1357 fz_free_text_page (state.ctx, text);
1358 fz_free_text_sheet (state.ctx, sheet);
1359 if (u.ptr) {
1360 state.freepage (u.ptr);
1363 end = now ();
1364 if (!stop) {
1365 printd ("progress 1 no matches %f sec", end - start);
1367 printd ("clearrects");
1370 static void set_tex_params (int colorspace)
1372 union {
1373 unsigned int s;
1374 unsigned char b[4];
1375 } endianness = {0};
1376 endianness.s = 0;
1377 endianness.b[0] = 1;
1379 switch (colorspace) {
1380 case 0:
1381 state.texiform = GL_RGBA8;
1382 state.texform = GL_RGBA;
1383 state.texty = GL_UNSIGNED_BYTE;
1384 state.colorspace = fz_device_rgb;
1385 break;
1386 case 1:
1387 state.texiform = GL_RGBA8;
1388 state.texform = GL_BGRA;
1389 state.texty = endianness.s > 0xff
1390 ? GL_UNSIGNED_INT_8_8_8_8
1391 : GL_UNSIGNED_INT_8_8_8_8_REV;
1392 state.colorspace = fz_device_bgr;
1393 break;
1394 case 2:
1395 state.texiform = GL_LUMINANCE_ALPHA;
1396 state.texform = GL_LUMINANCE_ALPHA;
1397 state.texty = GL_UNSIGNED_BYTE;
1398 state.colorspace = fz_device_gray;
1399 break;
1400 default:
1401 errx (1, "invalid colorspce %d", colorspace);
1405 static void realloctexts (int texcount)
1407 size_t size;
1409 if (texcount == state.texcount) return;
1411 if (texcount < state.texcount) {
1412 glDeleteTextures (state.texcount - texcount,
1413 state.texids + texcount);
1416 size = texcount * sizeof (*state.texids);
1417 state.texids = realloc (state.texids, size);
1418 if (!state.texids) {
1419 err (1, "realloc texids %" FMT_s, size);
1422 size = texcount * sizeof (*state.texowners);
1423 state.texowners = realloc (state.texowners, size);
1424 if (!state.texowners) {
1425 err (1, "realloc texowners %" FMT_s, size);
1427 if (texcount > state.texcount) {
1428 int i;
1430 glGenTextures (texcount - state.texcount,
1431 state.texids + state.texcount);
1432 for (i = state.texcount; i < texcount; ++i) {
1433 state.texowners[i].w = -1;
1434 state.texowners[i].slice = NULL;
1437 state.texcount = texcount;
1438 state.texindex = 0;
1441 static void * mainloop (void *unused)
1443 char *p = NULL;
1444 int len, ret, oldlen = 0;
1446 for (;;) {
1447 len = readlen ();
1448 if (len == 0) {
1449 errx (1, "readlen returned 0");
1452 if (oldlen < len + 1) {
1453 p = realloc (p, len + 1);
1454 if (!p) {
1455 err (1, "realloc %d failed", len + 1);
1457 oldlen = len + 1;
1459 readdata (p, len);
1460 p[len] = 0;
1462 if (!strncmp ("open", p, 4)) {
1463 size_t filenamelen;
1464 char *password;
1465 char *filename = p + 5;
1467 filenamelen = strlen (filename);
1468 password = filename + filenamelen + 1;
1470 openxref (filename, password);
1471 pdfinfo ();
1472 initpdims ();
1473 printd ("msg Opened %s (press h/F1 to get help)", filename);
1474 state.needoutline = 1;
1476 else if (!strncmp ("cs", p, 2)) {
1477 int i, colorspace;
1479 ret = sscanf (p + 2, " %d", &colorspace);
1480 if (ret != 1) {
1481 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
1483 lock ("cs");
1484 set_tex_params (colorspace);
1485 for (i = 0; i < state.texcount; ++i) {
1486 state.texowners[i].w = -1;
1487 state.texowners[i].slice = NULL;
1489 unlock ("cs");
1491 else if (!strncmp ("freepage", p, 8)) {
1492 void *ptr;
1494 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1495 if (ret != 1) {
1496 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
1498 freepage (ptr);
1500 else if (!strncmp ("freetile", p, 8)) {
1501 void *ptr;
1503 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1504 if (ret != 1) {
1505 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
1507 freetile (ptr);
1509 else if (!strncmp ("search", p, 6)) {
1510 int icase, pageno, y, ret, len2, forward;
1511 char *pattern;
1512 regex_t re;
1514 ret = sscanf (p + 6, " %d %d %d %d,%n",
1515 &icase, &pageno, &y, &forward, &len2);
1516 if (ret != 4) {
1517 errx (1, "malformed search `%s' ret=%d", p, ret);
1520 pattern = p + 6 + len2;
1521 ret = regcomp (&re, pattern,
1522 REG_EXTENDED | (icase ? REG_ICASE : 0));
1523 if (ret) {
1524 char errbuf[80];
1525 size_t size;
1527 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1528 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
1530 else {
1531 search (&re, pageno, y, forward);
1532 regfree (&re);
1535 else if (!strncmp ("geometry", p, 8)) {
1536 int w, h;
1538 printd ("clear");
1539 ret = sscanf (p + 8, " %d %d", &w, &h);
1540 if (ret != 2) {
1541 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1544 lock ("geometry");
1545 state.h = h;
1546 if (w != state.w) {
1547 int i;
1548 state.w = w;
1549 for (i = 0; i < state.texcount; ++i) {
1550 state.texowners[i].slice = NULL;
1553 layout ();
1554 process_outline ();
1555 state.gen++;
1556 unlock ("geometry");
1557 printd ("continue %d", state.pagecount);
1559 else if (!strncmp ("reqlayout", p, 9)) {
1560 int rotate, proportional;
1562 printd ("clear");
1563 ret = sscanf (p + 9, " %d %d", &rotate, &proportional);
1564 if (ret != 2) {
1565 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
1567 lock ("reqlayout");
1568 if (state.rotate != rotate || state.proportional != proportional) {
1569 state.gen += 1;
1571 state.rotate = rotate;
1572 state.proportional = proportional;
1573 layout ();
1574 process_outline ();
1575 state.gen++;
1576 unlock ("reqlayout");
1577 printd ("continue %d", state.pagecount);
1579 else if (!strncmp ("page", p, 4)) {
1580 double a, b;
1581 struct page *page;
1582 int pageno, pindex, ret;
1584 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
1585 if (ret != 2) {
1586 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1589 lock ("page");
1590 a = now ();
1591 page = loadpage (pageno, pindex);
1592 b = now ();
1593 unlock ("page");
1595 printd ("page %" FMT_ptr " %f", FMT_ptr_cast2 (page), b - a);
1597 else if (!strncmp ("tile", p, 4)) {
1598 int x, y, w, h, ret;
1599 struct page *page;
1600 struct tile *tile;
1601 double a, b;
1603 ret = sscanf (p + 4, " %" FMT_ptr " %d %d %d %d",
1604 FMT_ptr_cast (&page), &x, &y, &w, &h);
1605 if (ret != 5) {
1606 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
1609 lock ("tile");
1610 a = now ();
1611 tile = rendertile (page, x, y, w, h);
1612 b = now ();
1613 unlock ("tile");
1615 printd ("tile %d %d %" FMT_ptr " %u %f",
1616 x, y,
1617 FMT_ptr_cast2 (tile),
1618 tile->w * tile->h * tile->pixmap->n,
1619 b - a);
1621 else if (!strncmp ("settrim", p, 7)) {
1622 int trimmargins;
1623 fz_bbox fuzz;
1625 ret = sscanf (p + 7, " %d %d %d %d %d", &trimmargins,
1626 &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
1627 if (ret != 5) {
1628 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
1630 printd ("clear");
1631 lock ("settrim");
1632 state.trimmargins = trimmargins;
1633 state.needoutline = 1;
1634 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
1635 state.trimanew = 1;
1636 state.trimfuzz = fuzz;
1638 state.pagedimcount = 0;
1639 free (state.pagedims);
1640 state.pagedims = NULL;
1641 initpdims ();
1642 layout ();
1643 process_outline ();
1644 unlock ("settrim");
1645 printd ("continue %d", state.pagecount);
1647 else if (!strncmp ("sliceh", p, 6)) {
1648 int h;
1650 ret = sscanf (p + 6, " %d", &h);
1651 if (ret != 1) {
1652 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
1654 if (h != state.sliceheight) {
1655 int i;
1657 state.sliceheight = h;
1658 for (i = 0; i < state.texcount; ++i) {
1659 state.texowners[i].w = -1;
1660 state.texowners[i].h = -1;
1661 state.texowners[i].slice = NULL;
1665 else if (!strncmp ("interrupt", p, 9)) {
1666 printd ("vmsg interrupted");
1668 else {
1669 errx (1, "unknown command %.*s", len, p);
1672 return 0;
1675 CAMLprim value ml_realloctexts (value texcount_v)
1677 CAMLparam1 (texcount_v);
1678 int ok;
1680 if (trylock ("ml_realloctexts")) {
1681 ok = 0;
1682 goto done;
1684 realloctexts (Int_val (texcount_v));
1685 ok = 1;
1686 unlock ("ml_realloctexts");
1688 done:
1689 CAMLreturn (Val_bool (ok));
1692 static void showsel (struct page *page, int ox, int oy)
1694 int seen = 0;
1695 fz_bbox bbox;
1696 fz_text_line *line;
1697 fz_text_span *span;
1698 fz_text_block *block;
1699 struct mark first, last;
1701 first = page->fmark;
1702 last = page->lmark;
1704 if (!first.span || !last.span) return;
1706 glEnable (GL_BLEND);
1707 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1708 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1710 ox += state.pagedims[page->pdimno].bounds.x0;
1711 oy += state.pagedims[page->pdimno].bounds.y0;
1712 for (block = page->text->blocks;
1713 block < page->text->blocks + page->text->len;
1714 ++block) {
1715 for (line = block->lines;
1716 line < block->lines + block->len;
1717 ++line) {
1718 for (span = line->spans;
1719 span < line->spans + line->len;
1720 ++span) {
1721 int i, j, k;
1723 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
1725 j = 0;
1726 k = span->len - 1;
1728 if (span == page->fmark.span && span == page->lmark.span) {
1729 seen = 1;
1730 j = MIN (first.i, last.i);
1731 k = MAX (first.i, last.i);
1733 else if (span == first.span) {
1734 seen = 1;
1735 j = first.i;
1737 else if (span == last.span) {
1738 seen = 1;
1739 k = last.i;
1742 if (seen) {
1743 for (i = j; i <= k; ++i) {
1744 fz_bbox b = fz_round_rect (span->text[i].bbox);
1745 bbox = fz_union_bbox (bbox, b);
1747 lprintf ("%d %d %d %d oy=%d ox=%d\n",
1748 bbox.x0,
1749 bbox.y0,
1750 bbox.x1,
1751 bbox.y1,
1752 oy, ox);
1754 glRecti (bbox.x0 + ox, bbox.y0 + oy,
1755 bbox.x1 + ox, bbox.y1 + oy);
1756 if (span == last.span) {
1757 goto done;
1763 done:
1764 glDisable (GL_BLEND);
1767 #include "glfont.c"
1769 static void highlightlinks (struct page *page, int xoff, int yoff)
1771 fz_matrix ctm;
1772 fz_link *link, *links;
1774 switch (page->type) {
1775 case DPDF:
1776 links = page->u.pdfpage->links;
1777 break;
1779 case DXPS:
1780 links = page->u.xpspage->links;
1781 break;
1783 default:
1784 return;
1787 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1788 glEnable (GL_LINE_STIPPLE);
1789 glLineStipple (0.5, 0xcccc);
1791 xoff -= state.pagedims[page->pdimno].bounds.x0;
1792 yoff -= state.pagedims[page->pdimno].bounds.y0;
1793 ctm = fz_concat (pagectm (page), fz_translate (xoff, yoff));
1795 glBegin (GL_QUADS);
1796 for (link = links; link; link = link->next) {
1797 fz_point p1, p2, p3, p4;
1799 p1.x = link->rect.x0;
1800 p1.y = link->rect.y0;
1802 p2.x = link->rect.x1;
1803 p2.y = link->rect.y0;
1805 p3.x = link->rect.x1;
1806 p3.y = link->rect.y1;
1808 p4.x = link->rect.x0;
1809 p4.y = link->rect.y1;
1811 p1 = fz_transform_point (ctm, p1);
1812 p2 = fz_transform_point (ctm, p2);
1813 p3 = fz_transform_point (ctm, p3);
1814 p4 = fz_transform_point (ctm, p4);
1816 switch (link->dest.kind) {
1817 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
1818 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
1819 default: glColor3ub (0, 0, 0); break;
1822 glVertex2f (p1.x, p1.y);
1823 glVertex2f (p2.x, p2.y);
1824 glVertex2f (p3.x, p3.y);
1825 glVertex2f (p4.x, p4.y);
1827 glEnd ();
1829 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1830 glDisable (GL_LINE_STIPPLE);
1833 static int compareslinks (const void *l, const void *r)
1835 struct slink const *ls = l;
1836 struct slink const *rs = r;
1837 if (ls->bbox.y0 == rs->bbox.y0) {
1838 return rs->bbox.x0 - rs->bbox.x0;
1840 return ls->bbox.y0 - rs->bbox.y0;
1843 static void droptext (struct page *page)
1845 if (page->text) {
1846 fz_free_text_page (state.ctx, page->text);
1847 page->fmark.i = -1;
1848 page->lmark.i = -1;
1849 page->fmark.span = NULL;
1850 page->lmark.span = NULL;
1851 page->text = NULL;
1853 if (page->sheet) {
1854 fz_free_text_sheet (state.ctx, page->sheet);
1858 static void dropslinks (struct page *page)
1860 if (page->slinks) {
1861 free (page->slinks);
1862 page->slinks = NULL;
1863 page->slinkcount = 0;
1867 static void ensureslinks (struct page *page)
1869 fz_matrix ctm;
1870 int i, count = 0;
1871 size_t slinksize = sizeof (*page->slinks);
1872 fz_link *link, *links;
1874 if (state.gen != page->sgen) {
1875 dropslinks (page);
1876 page->sgen = state.gen;
1878 if (page->slinks) return;
1880 switch (page->type) {
1881 case DPDF:
1882 links = page->u.pdfpage->links;
1883 ctm = fz_concat (trimctm (page->u.pdfpage, page->pdimno),
1884 state.pagedims[page->pdimno].ctm);
1885 break;
1887 case DXPS:
1888 links = page->u.xpspage->links;
1889 ctm = state.pagedims[page->pdimno].ctm;
1890 break;
1892 default:
1893 return;
1896 for (link = links; link; link = link->next) {
1897 count++;
1899 if (count > 0) {
1900 page->slinkcount = count;
1901 page->slinks = calloc (count, slinksize);
1902 if (!page->slinks) {
1903 err (1, "realloc slinks %d", count);
1906 for (i = 0, link = links; link; ++i, link = link->next) {
1907 page->slinks[i].link = link;
1908 page->slinks[i].bbox =
1909 fz_round_rect (fz_transform_rect (ctm, link->rect));
1911 qsort (page->slinks, count, slinksize, compareslinks);
1915 /* slightly tweaked fmt_ulong by D.J. Bernstein */
1916 static void fmt_linkn (char *s, unsigned int u)
1918 unsigned int len; unsigned int q;
1919 int zma = 'z' - 'a' + 1;
1920 len = 1; q = u;
1921 while (q > zma - 1) { ++len; q /= zma; }
1922 if (s) {
1923 s += len;
1924 do { *--s = 'a' + (u % zma) - (u < zma && len > 1); u /= zma; } while(u);
1925 /* handles u == 0 */
1927 s[len] = 0;
1930 static void highlightslinks (struct page *page, int xoff, int yoff,
1931 int noff, char *targ, int tlen, int hfsize)
1933 int i;
1934 char buf[40];
1935 struct slink *slink;
1936 double x0, y0, x1, y1, w;
1938 ensureslinks (page);
1939 glColor3ub (0xc3, 0xb0, 0x91);
1940 for (i = 0; i < page->slinkcount; ++i) {
1941 fmt_linkn (buf, i + noff);
1942 if (!tlen || !strncmp (targ, buf, tlen)) {
1943 slink = &page->slinks[i];
1945 x0 = slink->bbox.x0 + xoff - 5;
1946 y1 = slink->bbox.y0 + yoff - 5;
1947 y0 = y1 + 10 + hfsize;
1948 w = measure_string (state.face, hfsize, buf);
1949 x1 = x0 + w + 10;
1950 glRectd (x0, y0, x1, y1);
1954 glEnable (GL_BLEND);
1955 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
1956 glEnable (GL_TEXTURE_2D);
1957 glColor3ub (0, 0, 0);
1958 for (i = 0; i < page->slinkcount; ++i) {
1959 fmt_linkn (buf, i + noff);
1960 if (!tlen || !strncmp (targ, buf, tlen)) {
1961 slink = &page->slinks[i];
1963 x0 = slink->bbox.x0 + xoff;
1964 y0 = slink->bbox.y0 + yoff + hfsize;
1965 draw_string (state.face, hfsize, x0, y0, buf);
1968 glDisable (GL_TEXTURE_2D);
1969 glDisable (GL_BLEND);
1973 static void uploadslice (struct tile *tile, struct slice *slice)
1975 int offset;
1976 struct slice *slice1;
1978 offset = 0;
1979 for (slice1 = tile->slices; slice != slice1; slice1++) {
1980 offset += slice1->h * tile->w * tile->pixmap->n;
1982 if (slice->texindex != -1 && slice->texindex < state.texcount
1983 && state.texowners[slice->texindex].slice == slice) {
1984 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1986 else {
1987 int subimage = 0;
1988 int texindex = state.texindex++ % state.texcount;
1990 if (state.texowners[texindex].w == tile->w) {
1991 if (state.texowners[texindex].h >= slice->h) {
1992 subimage = 1;
1994 else {
1995 state.texowners[texindex].h = slice->h;
1998 else {
1999 state.texowners[texindex].h = slice->h;
2002 state.texowners[texindex].w = tile->w;
2003 state.texowners[texindex].slice = slice;
2004 slice->texindex = texindex;
2006 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
2007 if (subimage) {
2008 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
2012 tile->w,
2013 slice->h,
2014 state.texform,
2015 state.texty,
2016 tile->pixmap->samples+offset
2019 else {
2020 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
2022 state.texiform,
2023 tile->w,
2024 slice->h,
2026 state.texform,
2027 state.texty,
2028 tile->pixmap->samples+offset
2034 CAMLprim value ml_drawtile (value args_v, value ptr_v)
2036 CAMLparam2 (args_v, ptr_v);
2037 int dispx = Int_val (Field (args_v, 0));
2038 int dispy = Int_val (Field (args_v, 1));
2039 int dispw = Int_val (Field (args_v, 2));
2040 int disph = Int_val (Field (args_v, 3));
2041 int tilex = Int_val (Field (args_v, 4));
2042 int tiley = Int_val (Field (args_v, 5));
2043 char *s = String_val (ptr_v);
2044 struct tile *tile = parse_pointer ("ml_drawtile", s);
2046 glEnable (GL_TEXTURE_RECTANGLE_ARB);
2048 int slicey, firstslice;
2049 struct slice *slice;
2051 firstslice = tiley / tile->sliceheight;
2052 slice = &tile->slices[firstslice];
2053 slicey = tiley % tile->sliceheight;
2055 while (disph > 0) {
2056 int dh;
2058 dh = slice->h - slicey;
2059 dh = MIN (disph, dh);
2060 uploadslice (tile, slice);
2062 glBegin (GL_QUADS);
2064 glTexCoord2i (tilex, slicey);
2065 glVertex2i (dispx, dispy);
2067 glTexCoord2i (tilex+dispw, slicey);
2068 glVertex2i (dispx+dispw, dispy);
2070 glTexCoord2i (tilex+dispw, slicey+dh);
2071 glVertex2i (dispx+dispw, dispy+dh);
2073 glTexCoord2i (tilex, slicey+dh);
2074 glVertex2i (dispx, dispy+dh);
2076 glEnd ();
2078 dispy += dh;
2079 disph -= dh;
2080 slice++;
2081 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
2082 slicey = 0;
2085 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2086 CAMLreturn (Val_unit);
2089 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2090 value xoff_v, value yoff_v,
2091 value li_v)
2093 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, li_v);
2094 int xoff = Int_val (xoff_v);
2095 int yoff = Int_val (yoff_v);
2096 int noff = Int_val (Field (li_v, 0));
2097 char *targ = String_val (Field (li_v, 1));
2098 int tlen = caml_string_length (Field (li_v, 1));
2099 int hfsize = Int_val (Field (li_v, 2));
2100 char *s = String_val (ptr_v);
2101 int hlmask = Int_val (hlinks_v);
2102 struct page *page = parse_pointer ("ml_postprocess", s);
2104 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2105 if (trylock ("ml_postprocess")) {
2106 noff = 0;
2107 goto done;
2109 if (hlmask & 2) {
2110 highlightslinks (page, xoff, yoff, noff, targ, tlen, hfsize);
2111 noff = page->slinkcount;
2113 showsel (page, xoff, yoff);
2114 unlock ("ml_postprocess");
2116 done:
2117 CAMLreturn (Val_int (noff));
2120 static fz_link *getlink (struct page *page, int x, int y)
2122 fz_point p;
2123 fz_matrix ctm;
2124 fz_link *link, *links;
2126 switch (page->type) {
2127 case DPDF:
2128 ctm = trimctm (page->u.pdfpage, page->pdimno);
2129 links = page->u.pdfpage->links;
2130 break;
2132 case DXPS:
2133 ctm = fz_identity;
2134 links = page->u.xpspage->links;
2135 break;
2137 default:
2138 return NULL;
2140 p.x = x;
2141 p.y = y;
2143 ctm = fz_concat (ctm, state.pagedims[page->pdimno].ctm);
2144 ctm = fz_invert_matrix (ctm);
2145 p = fz_transform_point (ctm, p);
2147 for (link = links; link; link = link->next) {
2148 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2149 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2150 return link;
2154 return NULL;
2157 static void ensuretext (struct page *page)
2159 if (state.gen != page->tgen) {
2160 droptext (page);
2161 page->tgen = state.gen;
2163 if (!page->text) {
2164 fz_device *tdev;
2166 page->text = fz_new_text_page (state.ctx, fz_infinite_rect);
2167 page->sheet = fz_new_text_sheet (state.ctx);
2168 tdev = fz_new_text_device (state.ctx, page->sheet, page->text);
2169 fz_run_display_list (page->dlist,
2170 tdev,
2171 pagectm (page),
2172 fz_infinite_bbox, NULL);
2173 qsort (page->text->blocks, page->text->len,
2174 sizeof (*page->text->blocks), compareblocks);
2175 fz_free_device (tdev);
2179 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2181 CAMLparam2 (start_page_v, dir_v);
2182 CAMLlocal1 (ret_v);
2183 int i, dir = Int_val (dir_v);
2184 int start_page = Int_val (start_page_v);
2185 int end_page = dir > 0 ? state.pagecount : -1;
2187 ret_v = Val_int (0);
2188 if (!(state.type == DPDF || state.type == DXPS)) {
2189 goto done;
2192 lock ("ml_findpage_with_links");
2193 for (i = start_page + dir; i != end_page; i += dir) {
2194 int found;
2196 switch (state.type) {
2197 case DPDF:
2199 pdf_page *page = NULL;
2201 fz_try (state.ctx) {
2202 page = pdf_load_page (state.u.pdf, i);
2203 found = !!page->links;
2205 fz_catch (state.ctx) {
2206 found = 0;
2208 if (page) {
2209 freepdfpage (page);
2212 break;
2213 case DXPS:
2215 xps_page *page = xps_load_page (state.u.xps, i);
2216 found = !!page->links;
2217 freexpspage (page);
2219 break;
2221 default:
2222 ARSERT ("invalid document type");
2225 if (found) {
2226 ret_v = caml_alloc_small (1, 1);
2227 Field (ret_v, 0) = Val_int (i);
2228 goto unlock;
2231 unlock:
2232 unlock ("ml_findpage_with_links");
2234 done:
2235 CAMLreturn (ret_v);
2238 enum { dir_first, dir_last};
2239 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
2241 CAMLprim value ml_findlink (value ptr_v, value dir_v)
2243 CAMLparam2 (ptr_v, dir_v);
2244 CAMLlocal2 (ret_v, pos_v);
2245 struct page *page;
2246 int dirtag, i, slinkindex;
2247 struct slink *found = NULL ,*slink;
2248 char *s = String_val (ptr_v);
2250 page = parse_pointer ("ml_findlink", s);
2251 ret_v = Val_int (0);
2252 if (trylock ("ml_findlink")) {
2253 goto done;
2256 ensureslinks (page);
2258 if (Is_block (dir_v)) {
2259 dirtag = Tag_val (dir_v);
2260 switch (dirtag) {
2261 case dir_first_visible:
2263 int x0, y0, dir, first_index, last_index;
2265 pos_v = Field (dir_v, 0);
2266 x0 = Int_val (Field (pos_v, 0));
2267 y0 = Int_val (Field (pos_v, 1));
2268 dir = Int_val (Field (pos_v, 2));
2270 if (dir >= 0) {
2271 dir = 1;
2272 first_index = 0;
2273 last_index = page->slinkcount;
2275 else {
2276 first_index = page->slinkcount - 1;
2277 last_index = -1;
2280 for (i = first_index; i != last_index; i += dir) {
2281 slink = &page->slinks[i];
2282 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
2283 found = slink;
2284 break;
2288 break;
2290 case dir_left:
2291 slinkindex = Int_val (Field (dir_v, 0));
2292 found = &page->slinks[slinkindex];
2293 for (i = slinkindex - 1; i >= 0; --i) {
2294 slink = &page->slinks[i];
2295 if (slink->bbox.x0 < found->bbox.x0) {
2296 found = slink;
2297 break;
2300 break;
2302 case dir_right:
2303 slinkindex = Int_val (Field (dir_v, 0));
2304 found = &page->slinks[slinkindex];
2305 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2306 slink = &page->slinks[i];
2307 if (slink->bbox.x0 > found->bbox.x0) {
2308 found = slink;
2309 break;
2312 break;
2314 case dir_down:
2315 slinkindex = Int_val (Field (dir_v, 0));
2316 found = &page->slinks[slinkindex];
2317 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2318 slink = &page->slinks[i];
2319 if (slink->bbox.y0 >= found->bbox.y0) {
2320 found = slink;
2321 break;
2324 break;
2326 case dir_up:
2327 slinkindex = Int_val (Field (dir_v, 0));
2328 found = &page->slinks[slinkindex];
2329 for (i = slinkindex - 1; i >= 0; --i) {
2330 slink = &page->slinks[i];
2331 if (slink->bbox.y0 <= found->bbox.y0) {
2332 found = slink;
2333 break;
2336 break;
2339 else {
2340 dirtag = Int_val (dir_v);
2341 switch (dirtag) {
2342 case dir_first:
2343 found = page->slinks;
2344 break;
2346 case dir_last:
2347 if (page->slinks) {
2348 found = page->slinks + (page->slinkcount - 1);
2350 break;
2353 if (found) {
2354 ret_v = caml_alloc_small (2, 1);
2355 Field (ret_v, 0) = Val_int (found - page->slinks);
2358 unlock ("ml_findlink");
2359 done:
2360 CAMLreturn (ret_v);
2363 enum { uuri, ugoto, utext, uunexpected, ulaunch, unamed, uremote };
2365 #define LINKTOVAL \
2367 int pageno; \
2369 switch (link->dest.kind) { \
2370 case FZ_LINK_GOTO: \
2372 fz_point p; \
2374 pageno = link->dest.ld.gotor.page; \
2375 p.x = 0; \
2376 p.y = 0; \
2378 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
2379 p.y = link->dest.ld.gotor.lt.y; \
2380 p = fz_transform_point (pdim->lctm, p); \
2382 tup_v = caml_alloc_tuple (2); \
2383 ret_v = caml_alloc_small (1, ugoto); \
2384 Field (tup_v, 0) = Val_int (pageno); \
2385 Field (tup_v, 1) = Val_int (p.y); \
2386 Field (ret_v, 0) = tup_v; \
2388 break; \
2390 case FZ_LINK_URI: \
2391 str_v = caml_copy_string (link->dest.ld.uri.uri); \
2392 ret_v = caml_alloc_small (1, uuri); \
2393 Field (ret_v, 0) = str_v; \
2394 break; \
2396 case FZ_LINK_LAUNCH: \
2397 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
2398 ret_v = caml_alloc_small (1, ulaunch); \
2399 Field (ret_v, 0) = str_v; \
2400 break; \
2402 case FZ_LINK_NAMED: \
2403 str_v = caml_copy_string (link->dest.ld.named.named); \
2404 ret_v = caml_alloc_small (1, unamed); \
2405 Field (ret_v, 0) = str_v; \
2406 break; \
2408 case FZ_LINK_GOTOR: \
2409 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
2410 pageno = link->dest.ld.gotor.page; \
2411 tup_v = caml_alloc_tuple (2); \
2412 ret_v = caml_alloc_small (1, uremote); \
2413 Field (tup_v, 0) = str_v; \
2414 Field (tup_v, 1) = Val_int (pageno); \
2415 Field (ret_v, 0) = tup_v; \
2416 break; \
2418 default: \
2420 char buf[80]; \
2422 snprintf (buf, sizeof (buf), \
2423 "unhandled link kind %d", link->dest.kind); \
2424 str_v = caml_copy_string (buf); \
2425 ret_v = caml_alloc_small (1, uunexpected); \
2426 Field (ret_v, 0) = str_v; \
2428 break; \
2432 CAMLprim value ml_getlink (value ptr_v, value n_v)
2434 CAMLparam2 (ptr_v, n_v);
2435 CAMLlocal3 (ret_v, tup_v, str_v);
2436 fz_link *link;
2437 struct page *page;
2438 struct pagedim *pdim;
2439 char *s = String_val (ptr_v);
2441 ret_v = Val_int (0);
2442 if (trylock ("ml_getlink")) {
2443 goto done;
2446 page = parse_pointer ("ml_getlink", s);
2447 ensureslinks (page);
2448 pdim = &state.pagedims[page->pdimno];
2449 link = page->slinks[Int_val (n_v)].link;
2450 LINKTOVAL;
2452 unlock ("ml_getlink");
2453 done:
2454 CAMLreturn (ret_v);
2457 CAMLprim value ml_getlinkcount (value ptr_v)
2459 CAMLparam1 (ptr_v);
2460 struct page *page;
2461 char *s = String_val (ptr_v);
2463 page = parse_pointer ("ml_getlinkcount", s);
2464 CAMLreturn (Val_int (page->slinkcount));
2467 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
2469 CAMLparam2 (ptr_v, n_v);
2470 CAMLlocal1 (ret_v);
2471 struct page *page;
2472 struct slink *slink;
2473 char *s = String_val (ptr_v);
2475 page = parse_pointer ("ml_getlinkrect", s);
2476 ret_v = caml_alloc_tuple (4);
2477 if (trylock ("ml_getlinkrect")) {
2478 Field (ret_v, 0) = Val_int (0);
2479 Field (ret_v, 1) = Val_int (0);
2480 Field (ret_v, 2) = Val_int (0);
2481 Field (ret_v, 3) = Val_int (0);
2482 goto done;
2484 ensureslinks (page);
2486 slink = &page->slinks[Int_val (n_v)];
2487 Field (ret_v, 0) = Val_int (slink->bbox.x0);
2488 Field (ret_v, 1) = Val_int (slink->bbox.y0);
2489 Field (ret_v, 2) = Val_int (slink->bbox.x1);
2490 Field (ret_v, 3) = Val_int (slink->bbox.y1);
2491 unlock ("ml_getlinkrect");
2493 done:
2494 CAMLreturn (ret_v);
2497 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
2499 CAMLparam3 (ptr_v, x_v, y_v);
2500 CAMLlocal3 (ret_v, tup_v, str_v);
2501 fz_link *link;
2502 struct page *page;
2503 char *s = String_val (ptr_v);
2504 int x = Int_val (x_v), y = Int_val (y_v);
2505 struct pagedim *pdim;
2507 ret_v = Val_int (0);
2508 if (trylock ("ml_whatsunder")) {
2509 goto done;
2512 page = parse_pointer ("ml_whatsunder", s);
2513 pdim = &state.pagedims[page->pdimno];
2514 x += pdim->bounds.x0;
2515 y += pdim->bounds.y0;
2516 link = getlink (page, x, y);
2517 if (link) {
2518 LINKTOVAL;
2520 else {
2521 fz_rect *b;
2522 fz_text_block *block;
2524 ensuretext (page);
2525 for (block = page->text->blocks;
2526 block < page->text->blocks + page->text->len;
2527 ++block) {
2528 fz_text_line *line;
2530 b = &block->bbox;
2531 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2532 continue;
2534 for (line = block->lines;
2535 line < block->lines + block->len;
2536 ++line) {
2537 fz_text_span *span;
2539 b = &line->bbox;
2540 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2541 continue;
2543 for (span = line->spans;
2544 span < line->spans + line->len;
2545 ++span) {
2546 fz_text_char *ch;
2548 b = &span->bbox;
2549 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2550 continue;
2552 for (ch = span->text; ch < span->text + span->len; ++ch) {
2553 b = &ch->bbox;
2555 if (x >= b->x0 && x <= b->x1
2556 && y >= b->y0 && y <= b->y1) {
2557 const char *n2 =
2558 span->style->font && span->style->font->name
2559 ? span->style->font->name
2560 : "Span has no font name"
2562 FT_FaceRec *face = span->style->font->ft_face;
2563 if (face && face->family_name) {
2564 char *s;
2565 char *n1 = face->family_name;
2566 size_t l1 = strlen (n1);
2567 size_t l2 = strlen (n2);
2569 if (l1 != l2 || memcmp (n1, n2, l1)) {
2570 s = malloc (l1 + l2 + 2);
2571 if (s) {
2572 memcpy (s, n2, l2);
2573 s[l2] = '=';
2574 memcpy (s + l2 + 1, n1, l1 + 1);
2575 str_v = caml_copy_string (s);
2576 free (s);
2580 if (str_v == 0) {
2581 str_v = caml_copy_string (n2);
2583 ret_v = caml_alloc_small (1, utext);
2584 Field (ret_v, 0) = str_v;
2585 goto unlock;
2592 unlock:
2593 unlock ("ml_whatsunder");
2595 done:
2596 CAMLreturn (ret_v);
2599 CAMLprim value ml_seltext (value ptr_v, value rect_v)
2601 CAMLparam2 (ptr_v, rect_v);
2602 fz_rect *b;
2603 struct page *page;
2604 struct pagedim *pdim;
2605 int i, x0, x1, y0, y1;
2606 char *s = String_val (ptr_v);
2607 int fi = 0, li = 0;
2608 fz_text_block *block;
2609 fz_text_span *span, *fspan, *lspan;
2610 fz_text_line *line, *fline = NULL, *lline = NULL;
2612 if (trylock ("ml_seltext")) {
2613 goto done;
2616 page = parse_pointer ("ml_seltext", s);
2617 ensuretext (page);
2619 pdim = &state.pagedims[page->pdimno];
2620 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;;
2621 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
2622 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
2623 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
2625 if (0) {
2626 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2627 glColor3ub (128, 128, 128);
2628 glRecti (x0, y0, x1, y1);
2629 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2632 fspan = lspan = NULL;
2634 for (block = page->text->blocks;
2635 block < page->text->blocks + page->text->len;
2636 ++block) {
2637 for (line = block->lines;
2638 line < block->lines + block->len;
2639 ++line) {
2640 for (span = line->spans;
2641 span < line->spans + line->len;
2642 ++span) {
2643 for (i = 0; i < span->len; ++i) {
2644 b = &span->text[i].bbox;
2645 int selected = 0;
2647 if (x0 >= b->x0 && x0 <= b->x1
2648 && y0 >= b->y0 && y0 <= b->y1) {
2649 fspan = span;
2650 fline = line;
2651 fi = i;
2652 selected = 1;
2654 if (x1 >= b->x0 && x1 <= b->x1
2655 && y1 >= b->y0 && y1 <= b->y1) {
2656 lspan = span;
2657 lline = line;
2658 li = i;
2659 selected = 1;
2661 if (0 && selected) {
2662 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2663 glColor3ub (128, 128, 128);
2664 glRecti (b->x0, b->y0, b->x1, b->y1);
2665 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2671 if (y1 < y0 || x1 < x0) {
2672 int swap = 0;
2674 if (fspan == lspan) {
2675 swap = 1;
2677 else {
2678 if (y1 < y0) {
2679 if (fline != lline) {
2680 swap = 1;
2685 if (swap) {
2686 i = fi;
2687 span = fspan;
2689 fi = li;
2690 fspan = lspan;
2692 li = i;
2693 lspan = span;
2697 page->fmark.i = fi;
2698 page->fmark.span = fspan;
2700 page->lmark.i = li;
2701 page->lmark.span = lspan;
2703 unlock ("ml_seltext");
2705 done:
2706 CAMLreturn (Val_unit);
2709 static int UNUSED pipespan (FILE *f, fz_text_span *span, int a, int b)
2711 char buf[4];
2712 int i, len, ret;
2714 for (i = a; i <= b; ++i) {
2715 len = fz_runetochar (buf, span->text[i].c);
2716 ret = fwrite (buf, len, 1, f);
2718 if (ret != 1) {
2719 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
2720 len, ret, strerror (errno));
2721 return -1;
2724 return 0;
2727 #ifdef __CYGWIN__
2728 value ml_popen (value UNUSED u1, value UNUSED u2)
2730 caml_failwith ("ml_popen not implemented under Cygwin");
2732 #else
2733 CAMLprim value ml_popen (value command_v, value fds_v)
2735 CAMLparam2 (command_v, fds_v);
2736 CAMLlocal2 (l_v, tup_v);
2737 int ret;
2738 char *msg = NULL;
2739 value earg_v = Nothing;
2740 posix_spawnattr_t attr;
2741 posix_spawn_file_actions_t fa;
2742 char *argv[] = { "/bin/sh", "-c", String_val (command_v), NULL };
2744 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
2745 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
2748 if ((ret = posix_spawnattr_init (&attr)) != 0) {
2749 msg = "posix_spawnattr_init";
2750 goto fail;
2753 #ifdef POSIX_SPAWN_USEVFORK
2754 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
2755 msg = "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK";
2756 goto fail;
2758 #endif
2760 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
2761 int fd1, fd2;
2763 tup_v = Field (l_v, 0);
2764 fd1 = Int_val (Field (tup_v, 0));
2765 fd2 = Int_val (Field (tup_v, 1));
2766 if (fd2 < 0) {
2767 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
2768 msg = "posix_spawn_file_actions_addclose";
2769 earg_v = tup_v;
2770 goto fail;
2773 else {
2774 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
2775 msg = "posix_spawn_file_actions_adddup2";
2776 earg_v = tup_v;
2777 goto fail;
2782 if ((ret = posix_spawn (NULL, "/bin/sh", &fa, &attr, argv, environ))) {
2783 msg = "posix_spawn";
2784 goto fail;
2787 fail:
2788 if ((ret = posix_spawnattr_destroy (&attr)) != 0) {
2789 fprintf (stderr, "posix_spawnattr_destroy: %s\n", strerror (ret));
2792 if ((ret = posix_spawn_file_actions_destroy (&fa)) != 0) {
2793 fprintf (stderr, "posix_spawn_file_actions_destroy: %s\n",
2794 strerror (ret));
2797 if (msg)
2798 unix_error (ret, msg, earg_v);
2800 CAMLreturn (Val_unit);
2802 #endif
2804 CAMLprim value ml_copysel (value fd_v, value ptr_v)
2806 CAMLparam1 (ptr_v);
2807 FILE *f;
2808 int seen = 0;
2809 struct page *page;
2810 fz_text_line *line;
2811 fz_text_span *span;
2812 fz_text_block *block;
2813 int fd = Int_val (fd_v);
2814 char *s = String_val (ptr_v);
2816 if (trylock ("ml_copysel")) {
2817 goto done;
2820 page = parse_pointer ("ml_sopysel", s);
2822 if (!page->fmark.span || !page->lmark.span) {
2823 fprintf (stderr, "nothing to copy\n");
2824 goto unlock;
2827 f = fdopen (fd, "w");
2828 if (!f) {
2829 fprintf (stderr, "failed to fopen sel pipe: %s\n",
2830 strerror (errno));
2831 f = stdout;
2834 for (block = page->text->blocks;
2835 block < page->text->blocks + page->text->len;
2836 ++block) {
2837 for (line = block->lines;
2838 line < block->lines + block->len;
2839 ++line) {
2840 for (span = line->spans;
2841 span < line->spans + line->len;
2842 ++span) {
2843 int a, b;
2845 seen |= span == page->fmark.span || span == page->lmark.span;
2846 a = span == page->fmark.span ? page->fmark.i : 0;
2847 b = span == page->lmark.span ? page->lmark.i : span->len - 1;
2849 if (seen) {
2850 if (pipespan (f, span, a, b)) {
2851 goto close;
2853 if (span == line->spans + line->len - 1) {
2854 if (putc ('\n', f) == EOF) {
2855 fprintf (stderr,
2856 "failed break line on sel pipe: %s\n",
2857 strerror (errno));
2858 goto close;
2861 if (span == page->lmark.span) {
2862 goto endloop;
2868 endloop:
2869 page->lmark.span = NULL;
2870 page->fmark.span = NULL;
2872 close:
2873 if (f != stdout) {
2874 int ret = fclose (f);
2875 fd = -1;
2876 if (ret == -1) {
2877 if (errno != ECHILD) {
2878 fprintf (stderr, "failed to close sel pipe: %s\n",
2879 strerror (errno));
2883 unlock:
2884 unlock ("ml_copysel");
2886 done:
2887 if (fd >= 0) {
2888 if (close (fd)) {
2889 fprintf (stderr, "failed to close sel pipe: %s\n",
2890 strerror (errno));
2893 CAMLreturn (Val_unit);
2896 CAMLprim value ml_getpdimrect (value pagedimno_v)
2898 CAMLparam1 (pagedimno_v);
2899 CAMLlocal1 (ret_v);
2900 int pagedimno = Int_val (pagedimno_v);
2901 fz_rect box;
2903 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
2904 if (trylock ("ml_getpdimrect")) {
2905 box = fz_empty_rect;
2907 else {
2908 box = state.pagedims[pagedimno].mediabox;
2909 unlock ("ml_getpdimrect");
2912 Store_double_field (ret_v, 0, box.x0);
2913 Store_double_field (ret_v, 1, box.x1);
2914 Store_double_field (ret_v, 2, box.y0);
2915 Store_double_field (ret_v, 3, box.y1);
2917 CAMLreturn (ret_v);
2920 static double getmaxw (void)
2922 int i;
2923 struct pagedim *p;
2924 double maxw = 0.0;
2926 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
2927 double x0, x1, w;
2929 x0 = MIN (p->mediabox.x0, p->mediabox.x1);
2930 x1 = MAX (p->mediabox.x0, p->mediabox.x1);
2932 w = x1 - x0;
2933 maxw = MAX (w, maxw);
2935 return maxw;
2938 CAMLprim value ml_getmaxw (value unit_v)
2940 CAMLparam1 (unit_v);
2941 CAMLlocal1 (ret_v);
2942 double maxw = 0.0;
2944 if (trylock ("ml_getmaxw")) {
2945 goto done;
2947 maxw = getmaxw ();
2948 unlock ("ml_getmaxw");
2949 done:
2950 ret_v = caml_copy_double (maxw);
2951 CAMLreturn (ret_v);
2954 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v, value dw_v)
2956 CAMLparam3 (winw_v, winh_v, dw_v);
2957 CAMLlocal1 (ret_v);
2958 int i;
2959 double zoom = 1.0;
2960 double maxw = 0.0, maxh = 0.0;
2961 struct pagedim *p;
2962 double winw = Int_val (winw_v);
2963 double winh = Int_val (winh_v);
2964 double dw = Int_val (dw_v);
2965 double pw = 1.0, ph = 1.0, num, den;
2967 if (trylock ("ml_zoom_for_height")) {
2968 goto done;
2971 if (state.proportional) {
2972 maxw = getmaxw ();
2975 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
2976 double x0, x1, y0, y1, w, h, scaledh, scale;
2978 x0 = MIN (p->mediabox.x0, p->mediabox.x1);
2979 x1 = MAX (p->mediabox.x0, p->mediabox.x1);
2980 y0 = MIN (p->mediabox.y0, p->mediabox.y1);
2981 y1 = MAX (p->mediabox.y0, p->mediabox.y1);
2983 w = x1 - x0;
2984 h = y1 - y0;
2986 if (state.proportional) {
2987 scale = w / maxw;
2988 scaledh = h * scale;
2990 else {
2991 scale = 1.0;
2992 scaledh = h;
2995 if (scaledh > maxh) {
2996 maxh = scaledh;
2997 ph = scaledh;
2998 pw = w * scale;
3002 num = (winh * pw) + (ph * dw);
3003 den = ph * winw;
3004 zoom = num / den;
3006 unlock ("ml_zoom_for_height");
3007 done:
3008 ret_v = caml_copy_double (zoom);
3009 CAMLreturn (ret_v);
3012 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
3014 CAMLparam4 (pt_v, x_v, y_v, string_v);
3015 CAMLlocal1 (ret_v);
3016 int pt = Int_val(pt_v);
3017 int x = Int_val (x_v);
3018 int y = Int_val (y_v);
3019 double w;
3021 w = draw_string (state.face, pt, x, y, String_val (string_v));
3022 ret_v = caml_copy_double (w);
3023 CAMLreturn (ret_v);
3026 CAMLprim value ml_measure_string (value pt_v, value string_v)
3028 CAMLparam2 (pt_v, string_v);
3029 CAMLlocal1 (ret_v);
3030 int pt = Int_val (pt_v);
3031 double w;
3033 w = measure_string (state.face, pt, String_val (string_v));
3034 ret_v = caml_copy_double (w);
3035 CAMLreturn (ret_v);
3038 CAMLprim value ml_getpagebox (value opaque_v)
3040 CAMLparam1 (opaque_v);
3041 CAMLlocal1 (ret_v);
3042 fz_bbox bbox;
3043 fz_device *dev;
3044 char *s = String_val (opaque_v);
3045 struct page *page = parse_pointer ("ml_getpagebox", s);
3047 ret_v = caml_alloc_tuple (4);
3048 dev = fz_new_bbox_device (state.ctx, &bbox);
3049 dev->hints |= FZ_IGNORE_SHADE;
3051 switch (page->type) {
3052 case DPDF:
3053 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, pagectm (page), NULL);
3054 break;
3056 case DXPS:
3057 xps_run_page (state.u.xps, page->u.xpspage, dev, pagectm (page), NULL);
3058 break;
3060 default:
3061 bbox = fz_infinite_bbox;
3062 break;
3065 fz_free_device (dev);
3066 Field (ret_v, 0) = Val_int (bbox.x0);
3067 Field (ret_v, 1) = Val_int (bbox.y0);
3068 Field (ret_v, 2) = Val_int (bbox.x1);
3069 Field (ret_v, 3) = Val_int (bbox.y1);
3071 CAMLreturn (ret_v);
3074 CAMLprim value ml_setaalevel (value level_v)
3076 CAMLparam1 (level_v);
3078 state.aalevel = Int_val (level_v);
3079 CAMLreturn (Val_unit);
3082 #undef pixel
3083 #include <X11/Xlib.h>
3084 #include <GL/glx.h>
3086 static struct {
3087 Display *dpy;
3088 GLXContext ctx;
3089 GLXDrawable drawable;
3090 } glx;
3092 #include "keysym2ucs.c"
3094 CAMLprim value ml_keysymtoutf8 (value keysym_v)
3096 CAMLparam1 (keysym_v);
3097 CAMLlocal1 (str_v);
3098 KeySym keysym = Int_val (keysym_v);
3099 Rune rune;
3100 int len;
3101 char buf[5];
3103 rune = keysym2ucs (keysym);
3104 len = fz_runetochar (buf, rune);
3105 buf[len] = 0;
3106 str_v = caml_copy_string (buf);
3107 CAMLreturn (str_v);
3110 CAMLprim value ml_glx (value win_v)
3112 CAMLparam1 (win_v);
3113 XVisualInfo *visual;
3114 int screen, wid = Int_val (win_v);
3115 int attributes[] = { GLX_RGBA, GLX_DOUBLEBUFFER, None };
3117 glx.dpy = XOpenDisplay (NULL);
3118 if (!glx.dpy) {
3119 caml_failwith ("XOpenDisplay failed");
3122 screen = DefaultScreen (glx.dpy);
3123 visual = glXChooseVisual (glx.dpy, screen, attributes);
3124 if (!visual) {
3125 XCloseDisplay (glx.dpy);
3126 glx.dpy = NULL;
3127 caml_failwith ("glXChooseVisual");
3130 glx.ctx = glXCreateContext (glx.dpy, visual, NULL, True);
3131 if (!glx.ctx) {
3132 XCloseDisplay (glx.dpy);
3133 XFree (visual);
3134 glx.dpy = NULL;
3135 caml_failwith ("glXCreateContext");
3138 XFree (visual);
3139 if (!glXMakeCurrent (glx.dpy, wid, glx.ctx)) {
3140 glXDestroyContext (glx.dpy, glx.ctx);
3141 XCloseDisplay (glx.dpy);
3142 glx.dpy = NULL;
3143 glx.ctx = NULL;
3144 caml_failwith ("glXMakeCurrent");
3146 glx.drawable = wid;
3148 CAMLreturn (Val_unit);
3151 CAMLprim value ml_swapb (value unit_v)
3153 CAMLparam1 (unit_v);
3154 glXSwapBuffers (glx.dpy, glx.drawable);
3155 CAMLreturn (Val_unit);
3158 CAMLprim value ml_glxsync (value unit_v)
3160 CAMLparam1 (unit_v);
3161 if (glx.dpy && glx.ctx) {
3162 glXWaitX ();
3163 glXWaitGL ();
3165 CAMLreturn (Val_unit);
3168 enum { piunknown, pilinux, piosx, pisun, pifreebsd,
3169 pidragonflybsd, piopenbsd, pinetbsd, picygwin };
3171 CAMLprim value ml_platform (value unit_v)
3173 CAMLparam1 (unit_v);
3174 int platid = piunknown;
3176 #if defined __linux__
3177 platid = pilinux;
3178 #elif defined __CYGWIN__
3179 platid = picygwin;
3180 #elif defined __DragonFly__
3181 platid = pidragonflybsd;
3182 #elif defined __FreeBSD__
3183 platid = pifreebsd;
3184 #elif defined __OpenBSD__
3185 platid = piopenbsd;
3186 #elif defined __NetBSD__
3187 platid = pinetbsd;
3188 #elif defined __sun__
3189 platid = pisun;
3190 #elif defined __APPLE__
3191 platid = piosx;
3192 #endif
3193 CAMLreturn (Val_int (platid));
3196 CAMLprim value ml_cloexec (value fd_v)
3198 CAMLparam1 (fd_v);
3199 int fd = Int_val (fd_v);
3201 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
3202 uerror ("fcntl", Nothing);
3204 CAMLreturn (Val_unit);
3207 CAMLprim value ml_init (value pipe_v, value params_v)
3209 CAMLparam2 (pipe_v, params_v);
3210 CAMLlocal2 (trim_v, fuzz_v);
3211 int ret;
3212 int texcount;
3213 char *fontpath;
3214 int colorspace;
3215 int mustoresize;
3216 struct sigaction sa;
3218 state.cr = Int_val (Field (pipe_v, 0));
3219 state.cw = Int_val (Field (pipe_v, 1));
3220 state.rotate = Int_val (Field (params_v, 0));
3221 state.proportional = Bool_val (Field (params_v, 1));
3222 trim_v = Field (params_v, 2);
3223 texcount = Int_val (Field (params_v, 3));
3224 state.sliceheight = Int_val (Field (params_v, 4));
3225 mustoresize = Int_val (Field (params_v, 5));
3226 colorspace = Int_val (Field (params_v, 6));
3227 fontpath = String_val (Field (params_v, 7));
3229 state.ctx = fz_new_context (NULL, NULL, mustoresize);
3231 state.trimmargins = Bool_val (Field (trim_v, 0));
3232 fuzz_v = Field (trim_v, 1);
3233 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
3234 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
3235 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
3236 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
3238 set_tex_params (colorspace);
3240 if (*fontpath) {
3241 state.face = load_font (fontpath);
3243 else {
3244 unsigned int len;
3245 void *base = pdf_lookup_substitute_font (0, 0, 0, 0, &len);
3247 state.face = load_builtin_font (base, len);
3249 if (!state.face) _exit (1);
3251 realloctexts (texcount);
3253 #ifdef __CYGWIN__
3254 sa.sa_handler = SIG_IGN;
3255 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
3256 #else
3257 sa.sa_handler = SIG_DFL;
3258 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP | SA_NOCLDWAIT;
3259 #endif
3260 if (sigemptyset (&sa.sa_mask)) {
3261 err (1, "sigemptyset");
3263 if (sigaction (SIGCHLD, &sa, NULL)) {
3264 err (1, "sigaction");
3267 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
3268 if (ret) {
3269 errx (1, "pthread_create: %s", strerror (ret));
3272 CAMLreturn (Val_unit);