Robustness
[llpp.git] / link.c
blob78b05017b808dbfe2109f487eef9e2178f34d484
1 /* lots of code c&p-ed directly from mupdf */
2 #include <errno.h>
3 #include <stdio.h>
4 #include <ctype.h>
5 #include <spawn.h>
6 #include <string.h>
7 #include <stdlib.h>
8 #include <signal.h>
10 #include <unistd.h>
11 #include <pthread.h>
12 #include <sys/time.h>
13 #include <sys/types.h>
14 #include <sys/ioctl.h>
16 #ifdef __CYGWIN__
17 #include <cygwin/socket.h> /* FIONREAD */
18 #endif
20 #include <regex.h>
21 #include <ctype.h>
22 #include <stdarg.h>
23 #include <limits.h>
25 #include <GL/gl.h>
27 #include <caml/fail.h>
28 #include <caml/alloc.h>
29 #include <caml/memory.h>
30 #include <caml/unixsupport.h>
32 #include <fitz.h>
33 #include <mupdf.h>
34 #include <muxps.h>
35 #include <mucbz.h>
37 #include FT_FREETYPE_H
39 #define PIGGYBACK
41 #if defined __GNUC__
42 #define NORETURN __attribute__ ((noreturn))
43 #define UNUSED __attribute__ ((unused))
44 #define OPTIMIZE(n) __attribute__ ((optimize ("O"#n)))
45 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
46 #else
47 #define NORETURN
48 #define UNUSED
49 #define OPTIMIZE(n)
50 #define GCC_FMT_ATTR(a, b)
51 #endif
53 #define FMT_s "zu"
55 #define FMT_ptr "p"
56 #define FMT_ptr_cast(p) (p)
57 #define FMT_ptr_cast2(p) (p)
59 static void NORETURN GCC_FMT_ATTR (2, 3)
60 err (int exitcode, const char *fmt, ...)
62 va_list ap;
63 int savederrno;
65 savederrno = errno;
66 va_start (ap, fmt);
67 vfprintf (stderr, fmt, ap);
68 va_end (ap);
69 fprintf (stderr, ": %s\n", strerror (savederrno));
70 fflush (stderr);
71 _exit (exitcode);
74 static void NORETURN GCC_FMT_ATTR (2, 3)
75 errx (int exitcode, const char *fmt, ...)
77 va_list ap;
79 va_start (ap, fmt);
80 vfprintf (stderr, fmt, ap);
81 va_end (ap);
82 fputc ('\n', stderr);
83 fflush (stderr);
84 _exit (exitcode);
87 #ifndef GL_TEXTURE_RECTANGLE_ARB
88 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
89 #endif
91 #ifndef GL_BGRA
92 #define GL_BGRA 0x80E1
93 #endif
95 #ifndef GL_UNSIGNED_INT_8_8_8_8
96 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
97 #endif
99 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
100 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
101 #endif
103 #if 0
104 #define lprintf printf
105 #else
106 #define lprintf(...)
107 #endif
109 #define ARSERT(cond) for (;;) { \
110 if (!(cond)) { \
111 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
113 break; \
116 struct slice {
117 int h;
118 int texindex;
121 struct tile {
122 int x, y, w, h;
123 int slicecount;
124 int sliceheight;
125 fz_pixmap *pixmap;
126 struct slice slices[1];
129 struct pagedim {
130 int pageno;
131 int rotate;
132 int left;
133 int tctmready;
134 fz_bbox bounds;
135 fz_rect pagebox;
136 fz_rect mediabox;
137 fz_matrix ctm, zoomctm, lctm, tctm;
140 struct slink {
141 fz_bbox bbox;
142 fz_link *link;
145 enum { DPDF, DXPS, DCBZ };
147 struct page {
148 int gen;
149 int type;
150 int pageno;
151 int pdimno;
152 fz_text_span *text;
153 union {
154 void *ptr;
155 pdf_page *pdfpage;
156 xps_page *xpspage;
157 cbz_page *cbzpage;
158 } u;
159 fz_display_list *dlist;
160 int slinkcount;
161 struct slink *slinks;
162 struct mark {
163 int i;
164 fz_text_span *span;
165 } fmark, lmark;
166 void (*freepage) (void *);
169 struct {
170 int type;
171 int sliceheight;
172 struct pagedim *pagedims;
173 int pagecount;
174 int pagedimcount;
175 union {
176 pdf_document *pdf;
177 xps_document *xps;
178 cbz_document *cbz;
179 } u;
180 fz_context *ctx;
181 int w, h;
183 int texindex;
184 int texcount;
185 GLuint *texids;
187 GLenum texiform;
188 GLenum texform;
189 GLenum texty;
191 fz_colorspace *colorspace;
193 struct {
194 int w, h;
195 struct slice *slice;
196 } *texowners;
198 int rotate;
199 int proportional;
200 int trimmargins;
201 int needoutline;
202 int gen;
203 int aalevel;
205 int trimanew;
206 fz_bbox trimfuzz;
207 fz_pixmap *pig;
209 pthread_t thread;
210 int cr, cw;
211 FT_Face face;
213 void (*closedoc) (void);
214 void (*freepage) (void *);
215 } state;
217 static void UNUSED debug_rect (const char *cap, fz_rect r)
219 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
222 static void UNUSED debug_bbox (const char *cap, fz_bbox r)
224 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
227 static void UNUSED debug_matrix (const char *cap, fz_matrix m)
229 printf ("%s(matrix) %.2f,%.2f,%.2f,%.2f %.2f %.2f\n", cap,
230 m.a, m.b, m.c, m.d, m.e, m.f);
233 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
235 static void lock (const char *cap)
237 int ret = pthread_mutex_lock (&mutex);
238 if (ret) {
239 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
243 static void unlock (const char *cap)
245 int ret = pthread_mutex_unlock (&mutex);
246 if (ret) {
247 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
251 static int trylock (const char *cap)
253 int ret = pthread_mutex_trylock (&mutex);
255 if (ret && ret != EBUSY) {
256 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
258 return ret == EBUSY;
261 static void *parse_pointer (const char *cap, const char *s)
263 int ret;
264 void *ptr;
266 ret = sscanf (s, "%" FMT_ptr, FMT_ptr_cast (&ptr));
267 if (ret != 1) {
268 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
270 return ptr;
273 static double now (void)
275 struct timeval tv;
277 if (gettimeofday (&tv, NULL)) {
278 err (1, "gettimeofday");
280 return tv.tv_sec + tv.tv_usec*1e-6;
283 static int hasdata (void)
285 int ret, avail;
286 ret = ioctl (state.cr, FIONREAD, &avail);
287 if (ret) err (1, "hasdata: FIONREAD error ret=%d", ret);
288 return avail > 0;
291 CAMLprim value ml_hasdata (value fd_v)
293 CAMLparam1 (fd_v);
294 int ret, avail;
296 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
297 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
298 CAMLreturn (Val_bool (avail > 0));
301 static void readdata (void *p, int size)
303 ssize_t n;
305 n = read (state.cr, p, size);
306 if (n - size) {
307 if (!n) errx (1, "EOF while reading");
308 err (1, "read (req %d, ret %zd)", size, n);
312 static void writedata (char *p, int size)
314 char buf[4];
315 ssize_t n;
317 buf[0] = (size >> 24) & 0xff;
318 buf[1] = (size >> 16) & 0xff;
319 buf[2] = (size >> 8) & 0xff;
320 buf[3] = (size >> 0) & 0xff;
322 n = write (state.cw, buf, 4);
323 if (n != 4) {
324 if (!n) errx (1, "EOF while writing length");
325 err (1, "write %zd", n);
328 n = write (state.cw, p, size);
329 if (n - size) {
330 if (!n) errx (1, "EOF while writing data");
331 err (1, "write (req %d, ret %zd)", size, n);
335 static int readlen (void)
337 unsigned char p[4];
339 readdata (p, 4);
340 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
343 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
345 int size = 200, len;
346 va_list ap;
347 char *buf;
349 buf = malloc (size);
350 for (;;) {
351 if (!buf) err (errno, "malloc for temp buf (%d bytes) failed", size);
353 va_start (ap, fmt);
354 len = vsnprintf (buf, size, fmt, ap);
355 va_end (ap);
357 if (len > -1 && len < size) {
358 writedata (buf, len);
359 break;
362 if (len > -1) {
363 size = len + 1;
365 else {
366 size *= 2;
368 buf = realloc (buf, size);
370 free (buf);
373 static void closepdf (void)
375 if (state.u.pdf) {
376 pdf_close_document (state.u.pdf);
377 state.u.pdf = NULL;
381 static void closexps (void)
383 if (state.u.xps) {
384 xps_close_document (state.u.xps);
385 state.u.xps = NULL;
389 static void closecbz (void)
391 if (state.u.cbz) {
392 cbz_close_document (state.u.cbz);
393 state.u.cbz = NULL;
397 static void freepdfpage (void *ptr)
399 pdf_free_page (state.u.pdf, ptr);
402 static void freexpspage (void *ptr)
404 xps_free_page (state.u.xps, ptr);
407 static void freecbzpage (void *ptr)
409 cbz_free_page (state.u.cbz, ptr);
412 static void openxref (char *filename, char *password)
414 int i, len;
416 for (i = 0; i < state.texcount; ++i) {
417 state.texowners[i].w = -1;
418 state.texowners[i].slice = NULL;
421 if (state.closedoc) state.closedoc ();
423 len = strlen (filename);
425 state.type = DPDF;
426 if (len > 4) {
427 char ext[4];
429 ext[0] = tolower ((int) filename[len-3]);
430 ext[1] = tolower ((int) filename[len-2]);
431 ext[2] = tolower ((int) filename[len-1]);
433 /**/ if (ext[0] == 'x' && ext[1] == 'p' && ext[2] == 's') {
434 state.type = DXPS;
436 else if (ext[0] == 'c' && ext[1] == 'b' && ext[2] == 'z') {
437 state.type = DCBZ;
441 if (state.pagedims) {
442 free (state.pagedims);
443 state.pagedims = NULL;
445 state.pagedimcount = 0;
447 fz_set_aa_level (state.ctx, state.aalevel);
448 switch (state.type) {
449 case DPDF:
450 state.u.pdf = pdf_open_document (state.ctx, filename);
451 if (pdf_needs_password (state.u.pdf)) {
452 int okay = pdf_authenticate_password (state.u.pdf, password);
453 if (!okay) {
454 errx (1, "invalid password");
457 state.pagecount = pdf_count_pages (state.u.pdf);
458 state.closedoc = closepdf;
459 state.freepage = freepdfpage;
460 break;
462 case DXPS:
463 state.u.xps = xps_open_document (state.ctx, filename);
464 state.pagecount = xps_count_pages (state.u.xps);
465 state.closedoc = closexps;
466 state.freepage = freexpspage;
467 break;
469 case DCBZ:
470 state.u.cbz = cbz_open_document (state.ctx, filename);
471 state.pagecount = cbz_count_pages (state.u.cbz);
472 state.closedoc = closecbz;
473 state.freepage = freecbzpage;
474 break;
478 static void pdfinfo (void)
480 if (state.type == DPDF) {
481 pdf_obj *infoobj;
483 printd ("info PDF version\t%d.%d",
484 state.u.pdf->version / 10, state.u.pdf->version % 10);
486 infoobj = pdf_dict_gets (state.u.pdf->trailer, "Info");
487 if (infoobj) {
488 int i;
489 char *s;
490 char *items[] = { "Title", "Author", "Creator",
491 "Producer", "CreationDate" };
493 for (i = 0; i < sizeof (items) / sizeof (*items); ++i) {
494 pdf_obj *obj = pdf_dict_gets (infoobj, items[i]);
495 s = pdf_to_utf8 (state.ctx, obj);
496 if (*s) {
497 if (i == 0) {
498 printd ("title %s", s);
500 printd ("info %s\t%s", items[i], s);
502 fz_free (state.ctx, s);
505 printd ("infoend");
509 static void unlinktile (struct tile *tile)
511 int i;
513 for (i = 0; i < tile->slicecount; ++i) {
514 struct slice *s = &tile->slices[i];
516 if (s->texindex != -1) {
517 if (state.texowners[s->texindex].slice == s) {
518 state.texowners[s->texindex].slice = NULL;
524 static void freepage (struct page *page)
526 if (page->text) {
527 fz_free_text_span (state.ctx, page->text);
529 if (page->slinks) {
530 free (page->slinks);
532 page->freepage (page->u.ptr);
533 fz_free_display_list (state.ctx, page->dlist);
534 free (page);
537 static void freetile (struct tile *tile)
539 unlinktile (tile);
540 #ifndef PIGGYBACK
541 fz_drop_pixmap (state.ctx, tile->pixmap);
542 #else
543 if (state.pig) {
544 fz_drop_pixmap (state.ctx, state.pig);
546 state.pig = tile->pixmap;
547 #endif
548 free (tile);
551 #ifdef __ALTIVEC__
552 #include <altivec.h>
554 static int cacheline32bytes;
555 extern char **environ;
557 static void __attribute__ ((constructor)) clcheck (void)
559 char **envp = environ;
560 unsigned long *auxv;
562 while (*envp++);
564 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
565 if (*auxv == 19) {
566 cacheline32bytes = auxv[1] == 32;
567 return;
572 static void OPTIMIZE (3) clearpixmap (fz_pixmap *pixmap)
574 if (cacheline32bytes) {
575 intptr_t a1, a2, diff;
576 size_t sizea, i, size = pixmap->w * pixmap->h * pixmap->n;
577 vector unsigned char v = vec_splat_u8 (-1);
578 vector unsigned char *p;
580 a1 = a2 = (intptr_t) pixmap->samples;
581 a2 = (a1 + 31) & ~31;
582 diff = a2 - a1;
583 sizea = size - diff;
584 p = (void *) a2;
586 while (a1 != a2) *(char *) a1++ = 0xff;
587 for (i = 0; i < (sizea & ~31); i += 32) {
588 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
589 vec_st (v, i, p);
590 vec_st (v, i + 16, p);
592 while (i < sizea) *((char *) a1 + i++) = 0xff;
594 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
596 #else
597 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
598 #endif
600 static fz_matrix trimctm (pdf_page *page, int pindex)
602 fz_matrix ctm;
603 struct pagedim *pdim = &state.pagedims[pindex];
605 if (!pdim->tctmready) {
606 if (state.trimmargins) {
607 fz_rect realbox;
609 ctm = fz_concat (fz_rotate (-pdim->rotate), fz_scale (1, -1));
610 realbox = fz_transform_rect (ctm, pdim->mediabox);
611 ctm = fz_concat (ctm, fz_translate (-realbox.x0, -realbox.y0));
612 ctm = fz_concat (fz_invert_matrix (page->ctm), ctm);
614 else {
615 ctm = fz_identity;
617 pdim->tctm = ctm;
618 pdim->tctmready = 1;
620 return pdim->tctm;
623 static fz_matrix pagectm (struct page *page)
625 if (page->type == DPDF) {
626 return fz_concat (trimctm (page->u.pdfpage, page->pdimno),
627 state.pagedims[page->pdimno].ctm);
629 else {
630 fz_matrix ctm;
631 struct pagedim *pdim = &state.pagedims[page->pdimno];
633 ctm = state.pagedims[page->pdimno].ctm;
634 ctm = fz_concat (fz_translate (-pdim->mediabox.x0,
635 -pdim->mediabox.y0), ctm);
636 return ctm;
640 static void *loadpage (int pageno, int pindex)
642 fz_device *dev;
643 struct page *page = NULL;
645 page = calloc (sizeof (struct page), 1);
646 if (!page) {
647 err (1, "calloc page %d", pageno);
650 page->dlist = fz_new_display_list (state.ctx);
651 dev = fz_new_list_device (state.ctx, page->dlist);
652 switch (state.type) {
653 case DPDF:
654 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
655 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, fz_identity, NULL);
656 page->freepage = freepdfpage;
657 break;
659 case DXPS:
660 page->u.xpspage = xps_load_page (state.u.xps, pageno);
661 xps_run_page (state.u.xps, page->u.xpspage, dev, fz_identity, NULL);
662 page->freepage = freexpspage;
663 break;
665 case DCBZ:
666 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
667 cbz_run_page (state.u.cbz, page->u.cbzpage, dev, fz_identity, NULL);
668 page->freepage = freecbzpage;
669 break;
671 fz_free_device (dev);
673 page->pdimno = pindex;
674 page->pageno = pageno;
675 page->gen = state.gen;
676 page->type = state.type;
678 return page;
681 static struct tile *alloctile (int h)
683 int i;
684 int slicecount;
685 size_t tilesize;
686 struct tile *tile;
688 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
689 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
690 tile = calloc (tilesize, 1);
691 if (!tile) {
692 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
694 for (i = 0; i < slicecount; ++i) {
695 int sh = MIN (h, state.sliceheight);
696 tile->slices[i].h = sh;
697 tile->slices[i].texindex = -1;
698 h -= sh;
700 tile->slicecount = slicecount;
701 tile->sliceheight = state.sliceheight;
702 return tile;
705 struct obs {
706 int cured;
707 fz_bbox b;
710 static void obs_fill_image (fz_device *dev, fz_image *image, fz_matrix ctm,
711 float alpha)
713 struct obs *obs = dev->user;
715 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
716 fz_bbox b = fz_round_rect (fz_transform_rect (ctm, fz_unit_rect));
717 b = fz_intersect_bbox (b, obs->b);
718 obs->cured = b.x0 == obs->b.x0
719 && b.x1 == obs->b.x1
720 && b.y0 == obs->b.y0
721 && b.y1 == obs->b.y1;
725 static int obscured (struct page *page, fz_bbox bbox)
727 fz_device dev;
728 struct obs obs;
730 memset (&dev, 0, sizeof (dev));
731 memset (&obs, 0, sizeof (obs));
732 dev.hints = 0;
733 dev.flags = 0;
734 dev.user = &obs;
735 dev.fill_image = obs_fill_image;
736 obs.b = bbox;
737 fz_run_display_list (page->dlist, &dev, pagectm (page), bbox, NULL);
738 return obs.cured;
741 static struct tile *rendertile (struct page *page, int x, int y, int w, int h)
743 fz_bbox bbox;
744 fz_device *dev;
745 struct tile *tile;
746 struct pagedim *pdim;
748 tile = alloctile (h);
749 pdim = &state.pagedims[page->pdimno];
751 bbox = pdim->bounds;
752 bbox.x0 += x;
753 bbox.y0 += y;
754 bbox.x1 = bbox.x0 + w;
755 bbox.y1 = bbox.y0 + h;
757 if (state.pig) {
758 if (state.pig->w == w
759 && state.pig->h == h
760 && state.pig->colorspace == state.colorspace) {
761 tile->pixmap = state.pig;
762 tile->pixmap->x = bbox.x0;
763 tile->pixmap->y = bbox.y0;
765 else {
766 fz_drop_pixmap (state.ctx, state.pig);
768 state.pig = NULL;
770 if (!tile->pixmap) {
771 tile->pixmap =
772 fz_new_pixmap_with_rect (state.ctx, state.colorspace, bbox);
775 tile->w = w;
776 tile->h = h;
777 if ((w < 128 && h < 128) || !obscured (page, bbox)) {
778 clearpixmap (tile->pixmap);
780 dev = fz_new_draw_device (state.ctx, tile->pixmap);
781 fz_run_display_list (page->dlist, dev, pagectm (page), bbox, NULL);
782 fz_free_device (dev);
784 return tile;
787 static void initpdims (void)
789 int pageno;
790 double start, end;
792 start = now ();
793 for (pageno = 0; pageno < state.pagecount; ++pageno) {
794 int rotate;
795 struct pagedim *p;
796 fz_rect mediabox;
798 switch (state.type) {
799 case DPDF: {
800 pdf_obj *pageobj = state.u.pdf->page_objs[pageno];
802 if (state.trimmargins) {
803 pdf_obj *obj;
804 pdf_page *page;
806 page = pdf_load_page (state.u.pdf, pageno);
807 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
808 if (state.trimanew || !obj) {
809 fz_rect rect;
810 fz_bbox bbox;
811 fz_matrix ctm;
812 fz_device *dev;
814 dev = fz_new_bbox_device (state.ctx, &bbox);
815 dev->hints |= FZ_IGNORE_SHADE;
816 ctm = fz_invert_matrix (page->ctm);
817 pdf_run_page (state.u.pdf, page, dev, fz_identity, NULL);
818 fz_free_device (dev);
820 rect.x0 = bbox.x0 + state.trimfuzz.x0;
821 rect.x1 = bbox.x1 + state.trimfuzz.x1;
822 rect.y0 = bbox.y0 + state.trimfuzz.y0;
823 rect.y1 = bbox.y1 + state.trimfuzz.y1;
824 rect = fz_transform_rect (ctm, rect);
825 rect = fz_intersect_rect (rect, page->mediabox);
827 if (fz_is_empty_rect (rect)) {
828 mediabox = page->mediabox;
830 else {
831 mediabox = rect;
834 obj = pdf_new_array (state.ctx, 4);
835 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x0));
836 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y0));
837 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x1));
838 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y1));
839 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
841 else {
842 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
843 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
844 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
845 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
848 rotate = page->rotate;
849 pdf_free_page (state.u.pdf, page);
851 printd ("progress %f Trimming %d",
852 (double) (pageno + 1) / state.pagecount,
853 pageno + 1);
855 else {
856 fz_rect cropbox;
858 mediabox = pdf_to_rect (state.ctx,
859 pdf_dict_gets (pageobj, "MediaBox"));
860 if (fz_is_empty_rect (mediabox)) {
861 fprintf (stderr, "cannot find page size for page %d\n",
862 pageno+1);
863 mediabox.x0 = 0;
864 mediabox.y0 = 0;
865 mediabox.x1 = 612;
866 mediabox.y1 = 792;
869 cropbox = pdf_to_rect (state.ctx,
870 pdf_dict_gets (pageobj, "CropBox"));
871 if (!fz_is_empty_rect (cropbox)) {
872 mediabox = fz_intersect_rect (mediabox, cropbox);
874 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
876 break;
879 case DXPS:
881 xps_page *page;
883 page = xps_load_page (state.u.xps, pageno);
884 mediabox = xps_bound_page (state.u.xps, page);
885 rotate = 0;
886 if (state.trimmargins) {
887 fz_rect rect;
888 fz_bbox bbox;
889 fz_device *dev;
891 dev = fz_new_bbox_device (state.ctx, &bbox);
892 dev->hints |= FZ_IGNORE_SHADE;
893 xps_run_page (state.u.xps, page, dev, fz_identity, NULL);
894 fz_free_device (dev);
896 rect.x0 = bbox.x0 + state.trimfuzz.x0;
897 rect.x1 = bbox.x1 + state.trimfuzz.x1;
898 rect.y0 = bbox.y0 + state.trimfuzz.y0;
899 rect.y1 = bbox.y1 + state.trimfuzz.y1;
900 rect = fz_intersect_rect (rect, mediabox);
902 if (!fz_is_empty_rect (rect)) {
903 mediabox = rect;
906 xps_free_page (state.u.xps, page);
907 printd ("progress %f loading %d",
908 (double) (pageno + 1) / state.pagecount,
909 pageno + 1);
911 break;
913 case DCBZ:
915 rotate = 0;
916 if (state.trimmargins) {
917 cbz_page *page;
919 page = cbz_load_page (state.u.cbz, pageno);
920 mediabox = cbz_bound_page (state.u.cbz, page);
921 cbz_free_page (state.u.cbz, page);
922 printd ("progress %f Trimming %d",
923 (double) (pageno + 1) / state.pagecount,
924 pageno + 1);
926 else {
927 mediabox.x0 = mediabox.y0 = 0;
928 mediabox.x1 = 900;
929 mediabox.y1 = 900;
932 break;
934 default:
935 ARSERT (0 && state.type);
938 if (state.pagedimcount == 0
939 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
940 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
941 size_t size;
943 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
944 state.pagedims = realloc (state.pagedims, size);
945 if (!state.pagedims) {
946 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
947 size, state.pagedimcount + 1);
950 p = &state.pagedims[state.pagedimcount++];
951 p->rotate = rotate;
952 p->mediabox = mediabox;
953 p->pageno = pageno;
956 end = now ();
957 if (state.trimmargins) {
958 printd ("progress 1 Trimmed %d pages in %f seconds",
959 state.pagecount, end - start);
961 else {
962 printd ("vmsg Processed %d pages in %f seconds",
963 state.pagecount, end - start);
965 state.trimanew = 0;
968 static void layout (void)
970 int pindex;
971 fz_rect box;
972 fz_matrix ctm;
973 double zoom, w, maxw = 0;
974 struct pagedim *p = state.pagedims;
976 if (state.proportional) {
977 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
978 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
979 p->mediabox);
980 w = box.x1 - box.x0;
981 maxw = MAX (w, maxw);
985 p = state.pagedims;
986 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
987 fz_bbox bbox;
989 ctm = fz_rotate (state.rotate);
990 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
991 p->mediabox);
992 w = box.x1 - box.x0;
994 if (state.proportional) {
995 double scale = w / maxw;
996 zoom = (state.w / w) * scale;
998 else {
999 zoom = state.w / w;
1002 p->zoomctm = fz_scale (zoom, zoom);
1003 ctm = fz_concat (p->zoomctm, ctm);
1005 p->pagebox = fz_transform_rect (fz_rotate (p->rotate), p->mediabox);
1006 p->pagebox.x1 -= p->pagebox.x0;
1007 p->pagebox.y1 -= p->pagebox.y0;
1008 p->pagebox.x0 = 0;
1009 p->pagebox.y0 = 0;
1010 bbox = fz_round_rect (fz_transform_rect (ctm, p->pagebox));
1012 p->bounds = bbox;
1013 p->left = state.proportional ? ((maxw - w) * zoom) / 2.0 : 0;
1014 p->ctm = ctm;
1016 ctm = fz_identity;
1017 ctm = fz_concat (ctm, fz_translate (0, -p->mediabox.y1));
1018 ctm = fz_concat (ctm, fz_scale (zoom, -zoom));
1019 ctm = fz_concat (ctm, fz_rotate (p->rotate + state.rotate));
1020 p->lctm = ctm;
1022 p->tctmready = 0;
1025 while (p-- != state.pagedims) {
1026 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1027 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1028 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1029 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1030 int w = x1 - x0;
1031 int h = y1 - y0;
1033 printd ("pdim %d %d %d %d", p->pageno, w, h, p->left);
1037 static void recurse_outline (fz_outline *outline, int level)
1039 while (outline) {
1040 fz_link_dest *dest;
1041 int i, top = 0;
1042 struct pagedim *pdim = state.pagedims;
1044 dest = &outline->dest;
1045 for (i = 0; i < state.pagedimcount; ++i) {
1046 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1047 break;
1048 pdim = &state.pagedims[i];
1050 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1051 fz_point p;
1052 p.x = 0;
1053 p.y = dest->ld.gotor.lt.y;
1054 p = fz_transform_point (pdim->lctm, p);
1055 top = p.y;
1057 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1058 int h;
1059 double y0, y1;
1061 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1062 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1063 h = y1 - y0;
1064 printd ("o %d %d %d %d %s",
1065 level, dest->ld.gotor.page, top, h, outline->title);
1067 if (outline->down) {
1068 recurse_outline (outline->down, level + 1);
1070 outline = outline->next;
1074 static void process_outline (void)
1076 fz_outline *outline;
1078 if (!state.needoutline) return;
1080 state.needoutline = 0;
1081 switch (state.type) {
1082 case DPDF:
1083 outline = pdf_load_outline (state.u.pdf);
1084 break;
1085 case DXPS:
1086 outline = xps_load_outline (state.u.xps);
1087 break;
1088 default:
1089 outline = NULL;
1090 break;
1092 if (outline) {
1093 recurse_outline (outline, 0);
1094 fz_free_outline (state.ctx, outline);
1098 static int comparespans (const void *l, const void *r)
1100 fz_text_span const *const*ls = l;
1101 fz_text_span const *const*rs = r;
1102 return (*ls)->text->bbox.y0 - (*rs)->text->bbox.y0;
1105 /* wishful thinking function */
1106 static void search (regex_t *re, int pageno, int y, int forward)
1108 int i, j;
1109 int ret;
1110 char *p;
1111 char buf[256];
1112 fz_matrix ctm;
1113 fz_device *tdev;
1114 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1115 fz_text_span *text, *span, **pspan;
1116 struct pagedim *pdim, *pdimprev;
1117 int stop = 0;
1118 int niters = 0;
1119 int nspans;
1120 double start, end;
1122 if (!(state.type == DPDF || state.type == DXPS))
1123 return;
1125 start = now ();
1126 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1127 if (niters++ == 5) {
1128 niters = 0;
1129 if (hasdata ()) {
1130 printd ("progress 1 attention requested aborting search at %d",
1131 pageno);
1132 stop = 1;
1134 else {
1135 printd ("progress %f searching in page %d",
1136 (double) (pageno + 1) / state.pagecount,
1137 pageno);
1140 pdimprev = NULL;
1141 for (i = 0; i < state.pagedimcount; ++i) {
1142 pdim = &state.pagedims[i];
1143 if (pdim->pageno == pageno) {
1144 goto found;
1146 if (pdim->pageno > pageno) {
1147 pdim = pdimprev;
1148 goto found;
1150 pdimprev = pdim;
1152 pdim = pdimprev;
1153 found:
1155 text = fz_new_text_span (state.ctx);
1156 tdev = fz_new_text_device (state.ctx, text);
1158 switch (state.type) {
1159 case DPDF:
1160 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1161 pdf_run_page (state.u.pdf, u.pdfpage, tdev, fz_identity, NULL);
1162 break;
1164 case DXPS:
1165 u.xpspage = xps_load_page (state.u.xps, pageno);
1166 xps_run_page (state.u.xps, u.xpspage, tdev, fz_identity, NULL);
1167 break;
1169 default:
1170 ARSERT (0 && state.type);
1173 fz_free_device (tdev);
1175 nspans = 0;
1176 for (span = text; span; span = span->next) {
1177 nspans++;
1179 pspan = malloc (sizeof (void *) * nspans);
1180 if (!pspan) {
1181 err (1, "malloc span pointers %" FMT_s, sizeof (void *) * nspans);
1183 for (i = 0, span = text; span; span = span->next, ++i) {
1184 pspan[i] = span;
1186 qsort (pspan, nspans, sizeof (fz_text_span *), comparespans);
1188 j = forward ? 0 : nspans - 1;
1189 while (nspans--) {
1190 regmatch_t rm;
1192 span = pspan[j];
1193 j += forward ? 1 : -1;
1194 p = buf;
1195 for (i = 0; i < span->len; ++i) {
1196 int len;
1197 char cbuf[4];
1199 if (forward) {
1200 if (span->text[i].bbox.y0 < y + 1) {
1201 continue;
1204 else {
1205 if (span->text[i].bbox.y0 > y - 1) {
1206 continue;
1209 len = runetochar (cbuf, &span->text[i].c);
1210 if (sizeof (buf) - 1 - (p - buf) > len) {
1211 int k;
1212 for (k = 0; k < len; ++k)
1213 *p++ = cbuf[k];
1215 else {
1216 break;
1219 if (p == buf) {
1220 continue;
1222 *p++ = 0;
1224 ret = regexec (re, buf, 1, &rm, 0);
1225 if (ret) {
1226 if (ret != REG_NOMATCH) {
1227 size_t size;
1228 char errbuf[80];
1229 size = regerror (ret, re, errbuf, sizeof (errbuf));
1230 printd ("msg regexec error `%.*s'",
1231 (int) size, errbuf);
1232 fz_free_text_span (state.ctx, text);
1233 state.freepage (u.ptr);
1234 free (pspan);
1235 return;
1238 else {
1239 fz_bbox *sb, *eb;
1240 fz_point p1, p2, p3, p4;
1241 int a, b, c, l;
1243 l = span->len;
1244 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1245 c += runelen (span->text[a].c);
1247 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1248 c += runelen (span->text[b].c);
1251 if (runelen (span->text[b].c) > 1) {
1252 b = MAX (0, b-1);
1254 sb = &span->text[MIN (a, l-1)].bbox;
1255 eb = &span->text[MIN (b, l-1)].bbox;
1257 p1.x = sb->x0;
1258 p1.y = sb->y0;
1259 p2.x = eb->x1;
1260 p2.y = sb->y0;
1261 p3.x = eb->x1;
1262 p3.y = eb->y1;
1263 p4.x = sb->x0;
1264 p4.y = eb->y1;
1266 switch (state.type) {
1267 case DPDF:
1268 trimctm (u.pdfpage, pdim - state.pagedims);
1269 ctm = fz_concat (pdim->tctm, pdim->zoomctm);
1270 break;
1272 case DXPS:
1273 ctm = pdim->ctm;
1274 break;
1277 p1 = fz_transform_point (ctm, p1);
1278 p2 = fz_transform_point (ctm, p2);
1279 p3 = fz_transform_point (ctm, p3);
1280 p4 = fz_transform_point (ctm, p4);
1282 if (!stop) {
1283 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1284 pageno, 1,
1285 p1.x, p1.y,
1286 p2.x, p2.y,
1287 p3.x, p3.y,
1288 p4.x, p4.y);
1290 printd ("progress 1 found at %d `%.*s' in %f sec",
1291 pageno, (int) (rm.rm_eo - rm.rm_so), &buf[rm.rm_so],
1292 now () - start);
1294 else {
1295 printd ("match %d %d %f %f %f %f %f %f %f %f",
1296 pageno, 2,
1297 p1.x, p1.y,
1298 p2.x, p2.y,
1299 p3.x, p3.y,
1300 p4.x, p4.y);
1302 stop = 1;
1305 if (forward) {
1306 pageno += 1;
1307 y = 0;
1309 else {
1310 pageno -= 1;
1311 y = INT_MAX;
1313 fz_free_text_span (state.ctx, text);
1314 state.freepage (u.ptr);
1315 free (pspan);
1317 end = now ();
1318 if (!stop) {
1319 printd ("progress 1 no matches %f sec", end - start);
1321 printd ("clearrects");
1324 static void set_tex_params (int colorspace)
1326 switch (colorspace) {
1327 case 0:
1328 state.texiform = GL_RGBA8;
1329 state.texform = GL_RGBA;
1330 state.texty = GL_UNSIGNED_BYTE;
1331 state.colorspace = fz_device_rgb;
1332 break;
1333 case 1:
1334 state.texiform = GL_RGBA8;
1335 state.texform = GL_BGRA;
1336 state.texty = fz_is_big_endian ()
1337 ? GL_UNSIGNED_INT_8_8_8_8
1338 : GL_UNSIGNED_INT_8_8_8_8_REV;
1339 state.colorspace = fz_device_bgr;
1340 break;
1341 case 2:
1342 state.texiform = GL_LUMINANCE_ALPHA;
1343 state.texform = GL_LUMINANCE_ALPHA;
1344 state.texty = GL_UNSIGNED_BYTE;
1345 state.colorspace = fz_device_gray;
1346 break;
1347 default:
1348 errx (1, "invalid colorspce %d", colorspace);
1352 static void realloctexts (int texcount)
1354 size_t size;
1356 if (texcount == state.texcount) return;
1358 if (texcount < state.texcount) {
1359 glDeleteTextures (state.texcount - texcount,
1360 state.texids + texcount);
1363 size = texcount * sizeof (*state.texids);
1364 state.texids = realloc (state.texids, size);
1365 if (!state.texids) {
1366 err (1, "realloc texids %" FMT_s, size);
1369 size = texcount * sizeof (*state.texowners);
1370 state.texowners = realloc (state.texowners, size);
1371 if (!state.texowners) {
1372 err (1, "realloc texowners %" FMT_s, size);
1374 if (texcount > state.texcount) {
1375 int i;
1377 glGenTextures (texcount - state.texcount,
1378 state.texids + state.texcount);
1379 for (i = state.texcount; i < texcount; ++i) {
1380 state.texowners[i].w = -1;
1381 state.texowners[i].slice = NULL;
1384 state.texcount = texcount;
1385 state.texindex = 0;
1388 static void * mainloop (void *unused)
1390 char *p = NULL;
1391 int len, ret, oldlen = 0;
1393 for (;;) {
1394 len = readlen ();
1395 if (len == 0) {
1396 errx (1, "readlen returned 0");
1399 if (oldlen < len + 1) {
1400 p = realloc (p, len + 1);
1401 if (!p) {
1402 err (1, "realloc %d failed", len + 1);
1404 oldlen = len + 1;
1406 readdata (p, len);
1407 p[len] = 0;
1409 if (!strncmp ("open", p, 4)) {
1410 size_t filenamelen;
1411 char *password;
1412 char *filename = p + 5;
1414 filenamelen = strlen (filename);
1415 password = filename + filenamelen + 1;
1417 openxref (filename, password);
1418 pdfinfo ();
1419 initpdims ();
1420 printd ("msg Opened %s (press h/F1 to get help)", filename);
1421 state.needoutline = 1;
1423 else if (!strncmp ("cs", p, 2)) {
1424 int i, colorspace;
1426 ret = sscanf (p + 2, " %d", &colorspace);
1427 if (ret != 1) {
1428 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
1430 lock ("cs");
1431 set_tex_params (colorspace);
1432 for (i = 0; i < state.texcount; ++i) {
1433 state.texowners[i].w = -1;
1434 state.texowners[i].slice = NULL;
1436 unlock ("cs");
1438 else if (!strncmp ("freepage", p, 8)) {
1439 void *ptr;
1441 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1442 if (ret != 1) {
1443 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
1445 freepage (ptr);
1447 else if (!strncmp ("freetile", p, 8)) {
1448 void *ptr;
1450 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1451 if (ret != 1) {
1452 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
1454 freetile (ptr);
1456 else if (!strncmp ("search", p, 6)) {
1457 int icase, pageno, y, ret, len2, forward;
1458 char *pattern;
1459 regex_t re;
1461 ret = sscanf (p + 6, " %d %d %d %d,%n",
1462 &icase, &pageno, &y, &forward, &len2);
1463 if (ret != 4) {
1464 errx (1, "malformed search `%s' ret=%d", p, ret);
1467 pattern = p + 6 + len2;
1468 ret = regcomp (&re, pattern,
1469 REG_EXTENDED | (icase ? REG_ICASE : 0));
1470 if (ret) {
1471 char errbuf[80];
1472 size_t size;
1474 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1475 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
1477 else {
1478 search (&re, pageno, y, forward);
1479 regfree (&re);
1482 else if (!strncmp ("geometry", p, 8)) {
1483 int w, h;
1485 printd ("clear");
1486 ret = sscanf (p + 8, " %d %d", &w, &h);
1487 if (ret != 2) {
1488 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1491 lock ("geometry");
1492 state.h = h;
1493 if (w != state.w) {
1494 int i;
1495 state.w = w;
1496 for (i = 0; i < state.texcount; ++i) {
1497 state.texowners[i].slice = NULL;
1500 layout ();
1501 process_outline ();
1502 state.gen++;
1503 unlock ("geometry");
1504 printd ("continue %d", state.pagecount);
1506 else if (!strncmp ("reqlayout", p, 9)) {
1507 int rotate, proportional;
1509 printd ("clear");
1510 ret = sscanf (p + 9, " %d %d", &rotate, &proportional);
1511 if (ret != 2) {
1512 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
1514 lock ("reqlayout");
1515 if (state.rotate != rotate || state.proportional != proportional) {
1516 state.gen += 1;
1518 state.rotate = rotate;
1519 state.proportional = proportional;
1520 layout ();
1521 process_outline ();
1522 state.gen++;
1523 unlock ("reqlayout");
1524 printd ("continue %d", state.pagecount);
1526 else if (!strncmp ("page", p, 4)) {
1527 double a, b;
1528 struct page *page;
1529 int pageno, pindex, ret;
1531 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
1532 if (ret != 2) {
1533 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1536 lock ("page");
1537 a = now ();
1538 page = loadpage (pageno, pindex);
1539 b = now ();
1540 unlock ("page");
1542 printd ("page %" FMT_ptr " %f", FMT_ptr_cast2 (page), b - a);
1544 else if (!strncmp ("tile", p, 4)) {
1545 int x, y, w, h, ret;
1546 struct page *page;
1547 struct tile *tile;
1548 double a, b;
1550 ret = sscanf (p + 4, " %" FMT_ptr " %d %d %d %d",
1551 FMT_ptr_cast (&page), &x, &y, &w, &h);
1552 if (ret != 5) {
1553 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
1556 lock ("tile");
1557 a = now ();
1558 tile = rendertile (page, x, y, w, h);
1559 b = now ();
1560 unlock ("tile");
1562 printd ("tile %d %d %" FMT_ptr " %u %f",
1563 x, y,
1564 FMT_ptr_cast2 (tile),
1565 tile->w * tile->h * tile->pixmap->n,
1566 b - a);
1568 else if (!strncmp ("settrim", p, 7)) {
1569 int trimmargins;
1570 fz_bbox fuzz;
1572 ret = sscanf (p + 7, " %d %d %d %d %d", &trimmargins,
1573 &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
1574 if (ret != 5) {
1575 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
1577 printd ("clear");
1578 lock ("settrim");
1579 state.trimmargins = trimmargins;
1580 state.needoutline = 1;
1581 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
1582 state.trimanew = 1;
1583 state.trimfuzz = fuzz;
1585 state.pagedimcount = 0;
1586 free (state.pagedims);
1587 state.pagedims = NULL;
1588 initpdims ();
1589 layout ();
1590 process_outline ();
1591 unlock ("settrim");
1592 printd ("continue %d", state.pagecount);
1594 else if (!strncmp ("sliceh", p, 6)) {
1595 int h;
1597 ret = sscanf (p + 6, " %d", &h);
1598 if (ret != 1) {
1599 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
1601 if (h != state.sliceheight) {
1602 int i;
1604 state.sliceheight = h;
1605 for (i = 0; i < state.texcount; ++i) {
1606 state.texowners[i].w = -1;
1607 state.texowners[i].h = -1;
1608 state.texowners[i].slice = NULL;
1612 else if (!strncmp ("interrupt", p, 9)) {
1613 printd ("vmsg interrupted");
1615 else {
1616 errx (1, "unknown command %.*s", len, p);
1619 return 0;
1622 CAMLprim value ml_realloctexts (value texcount_v)
1624 CAMLparam1 (texcount_v);
1625 int ok;
1627 if (trylock ("ml_realloctexts")) {
1628 ok = 0;
1629 goto done;
1631 realloctexts (Int_val (texcount_v));
1632 ok = 1;
1633 unlock ("ml_realloctexts");
1635 done:
1636 CAMLreturn (Val_bool (ok));
1639 static void showsel (struct page *page, int ox, int oy)
1641 fz_bbox bbox;
1642 fz_text_span *span;
1643 struct mark first, last;
1645 first = page->fmark;
1646 last = page->lmark;
1648 if (!first.span || !last.span) return;
1650 glEnable (GL_BLEND);
1651 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1652 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1654 ox += state.pagedims[page->pdimno].bounds.x0;
1655 oy += state.pagedims[page->pdimno].bounds.y0;
1656 for (span = first.span; span; span = span->next) {
1657 int i, j, k;
1659 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
1661 j = 0;
1662 k = span->len - 1;
1664 if (span == page->fmark.span && span == page->lmark.span) {
1665 j = MIN (first.i, last.i);
1666 k = MAX (first.i, last.i);
1668 else if (span == first.span) {
1669 j = first.i;
1671 else if (span == last.span) {
1672 k = last.i;
1675 for (i = j; i <= k; ++i) {
1676 bbox = fz_union_bbox (bbox, span->text[i].bbox);
1678 lprintf ("%d %d %d %d oy=%d ox=%d\n",
1679 bbox.x0,
1680 bbox.y0,
1681 bbox.x1,
1682 bbox.y1,
1683 oy, ox);
1685 glRecti (bbox.x0 + ox, bbox.y0 + oy, bbox.x1 + ox, bbox.y1 + oy);
1687 if (span == last.span) break;
1689 glDisable (GL_BLEND);
1692 #include "glfont.c"
1694 static void highlightlinks (struct page *page, int xoff, int yoff)
1696 fz_matrix ctm;
1697 fz_link *link, *links;
1699 switch (page->type) {
1700 case DPDF:
1701 links = page->u.pdfpage->links;
1702 break;
1704 case DXPS:
1705 links = page->u.xpspage->links;
1706 break;
1708 default:
1709 return;
1712 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1713 glEnable (GL_LINE_STIPPLE);
1714 glLineStipple (0.5, 0xcccc);
1716 xoff -= state.pagedims[page->pdimno].bounds.x0;
1717 yoff -= state.pagedims[page->pdimno].bounds.y0;
1718 ctm = fz_concat (pagectm (page), fz_translate (xoff, yoff));
1720 glBegin (GL_QUADS);
1721 for (link = links; link; link = link->next) {
1722 fz_point p1, p2, p3, p4;
1724 p1.x = link->rect.x0;
1725 p1.y = link->rect.y0;
1727 p2.x = link->rect.x1;
1728 p2.y = link->rect.y0;
1730 p3.x = link->rect.x1;
1731 p3.y = link->rect.y1;
1733 p4.x = link->rect.x0;
1734 p4.y = link->rect.y1;
1736 p1 = fz_transform_point (ctm, p1);
1737 p2 = fz_transform_point (ctm, p2);
1738 p3 = fz_transform_point (ctm, p3);
1739 p4 = fz_transform_point (ctm, p4);
1741 switch (link->dest.kind) {
1742 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
1743 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
1744 default: glColor3ub (0, 0, 0); break;
1747 glVertex2f (p1.x, p1.y);
1748 glVertex2f (p2.x, p2.y);
1749 glVertex2f (p3.x, p3.y);
1750 glVertex2f (p4.x, p4.y);
1752 glEnd ();
1754 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1755 glDisable (GL_LINE_STIPPLE);
1758 static int compareslinks (const void *l, const void *r)
1760 struct slink const *ls = l;
1761 struct slink const *rs = r;
1762 if (ls->bbox.y0 == rs->bbox.y0) {
1763 return rs->bbox.x0 - rs->bbox.x0;
1765 return ls->bbox.y0 - rs->bbox.y0;
1768 static void droptext (struct page *page)
1770 if (page->text) {
1771 fz_free_text_span (state.ctx, page->text);
1772 page->fmark.i = -1;
1773 page->lmark.i = -1;
1774 page->fmark.span = NULL;
1775 page->lmark.span = NULL;
1776 page->text = NULL;
1780 static void dropslinks (struct page *page)
1782 if (page->slinks) {
1783 free (page->slinks);
1784 page->slinks = NULL;
1785 page->slinkcount = 0;
1789 static void ensureslinks (struct page *page)
1791 fz_matrix ctm;
1792 int i, count = 0;
1793 size_t slinksize = sizeof (*page->slinks);
1794 fz_link *link, *links;
1796 if (state.gen != page->gen) {
1797 droptext (page);
1798 dropslinks (page);
1799 page->gen = state.gen;
1801 if (page->slinks) return;
1803 switch (page->type) {
1804 case DPDF:
1805 links = page->u.pdfpage->links;
1806 ctm = fz_concat (trimctm (page->u.pdfpage, page->pdimno),
1807 state.pagedims[page->pdimno].ctm);
1808 break;
1810 case DXPS:
1811 links = page->u.xpspage->links;
1812 ctm = state.pagedims[page->pdimno].ctm;
1813 break;
1815 default:
1816 return;
1819 for (link = links; link; link = link->next) {
1820 count++;
1822 if (count > 0) {
1823 page->slinkcount = count;
1824 page->slinks = calloc (count, slinksize);
1825 if (!page->slinks) {
1826 err (1, "realloc slinks %d", count);
1829 for (i = 0, link = links; link; ++i, link = link->next) {
1830 page->slinks[i].link = link;
1831 page->slinks[i].bbox =
1832 fz_round_rect (fz_transform_rect (ctm, link->rect));
1834 qsort (page->slinks, count, slinksize, compareslinks);
1838 /* slightly tweaked fmt_ulong by D.J. Bernstein */
1839 static void fmt_linkn (char *s, unsigned int u)
1841 unsigned int len; unsigned int q;
1842 int zma = 'z' - 'a' + 1;
1843 len = 1; q = u;
1844 while (q > zma - 1) { ++len; q /= zma; }
1845 if (s) {
1846 s += len;
1847 do { *--s = 'a' + (u % zma); u /= zma; } while(u); /* handles u == 0 */
1849 s[len] = 0;
1852 static void highlightslinks (struct page *page, int xoff, int yoff, int noff)
1854 int i;
1855 char buf[40];
1856 struct slink *slink;
1857 double x0, y0, x1, y1, w;
1859 ensureslinks (page);
1860 glColor3ub (0xc3, 0xb0, 0x91);
1861 for (i = 0; i < page->slinkcount; ++i) {
1862 slink = &page->slinks[i];
1864 fmt_linkn (buf, i + noff);
1865 x0 = slink->bbox.x0 + xoff - 5;
1866 y1 = slink->bbox.y0 + yoff - 5;
1867 y0 = y1 + 22;
1868 w = measure_string (state.face, 12, buf);
1869 x1 = x0 + w + 10;
1870 glRectd (x0, y0, x1, y1);
1873 glEnable (GL_BLEND);
1874 glEnable (GL_TEXTURE_2D);
1875 glColor3ub (0, 0, 0);
1876 for (i = 0; i < page->slinkcount; ++i) {
1877 slink = &page->slinks[i];
1879 fmt_linkn (buf, i + noff);
1880 x0 = slink->bbox.x0 + xoff;
1881 y0 = slink->bbox.y0 + yoff + 12;
1882 draw_string (state.face, 12, x0, y0, buf);
1884 glDisable (GL_TEXTURE_2D);
1885 glDisable (GL_BLEND);
1888 static void uploadslice (struct tile *tile, struct slice *slice)
1890 int offset;
1891 struct slice *slice1;
1893 offset = 0;
1894 for (slice1 = tile->slices; slice != slice1; slice1++) {
1895 offset += slice1->h * tile->w * tile->pixmap->n;
1897 if (slice->texindex != -1 && slice->texindex < state.texcount
1898 && state.texowners[slice->texindex].slice == slice) {
1899 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1901 else {
1902 int subimage = 0;
1903 int texindex = state.texindex++ % state.texcount;
1905 if (state.texowners[texindex].w == tile->w) {
1906 if (state.texowners[texindex].h >= slice->h) {
1907 subimage = 1;
1909 else {
1910 state.texowners[texindex].h = slice->h;
1913 else {
1914 state.texowners[texindex].h = slice->h;
1917 state.texowners[texindex].w = tile->w;
1918 state.texowners[texindex].slice = slice;
1919 slice->texindex = texindex;
1921 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
1922 if (subimage) {
1923 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
1927 tile->w,
1928 slice->h,
1929 state.texform,
1930 state.texty,
1931 tile->pixmap->samples+offset
1934 else {
1935 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
1937 state.texiform,
1938 tile->w,
1939 slice->h,
1941 state.texform,
1942 state.texty,
1943 tile->pixmap->samples+offset
1949 CAMLprim value ml_drawtile (value args_v, value ptr_v)
1951 CAMLparam2 (args_v, ptr_v);
1952 int dispx = Int_val (Field (args_v, 0));
1953 int dispy = Int_val (Field (args_v, 1));
1954 int dispw = Int_val (Field (args_v, 2));
1955 int disph = Int_val (Field (args_v, 3));
1956 int tilex = Int_val (Field (args_v, 4));
1957 int tiley = Int_val (Field (args_v, 5));
1958 char *s = String_val (ptr_v);
1959 struct tile *tile = parse_pointer ("ml_drawtile", s);
1961 glEnable (GL_TEXTURE_RECTANGLE_ARB);
1963 int slicey, firstslice;
1964 struct slice *slice;
1966 firstslice = tiley / tile->sliceheight;
1967 slice = &tile->slices[firstslice];
1968 slicey = tiley % tile->sliceheight;
1970 while (disph > 0) {
1971 int dh;
1973 dh = slice->h - slicey;
1974 dh = MIN (disph, dh);
1975 uploadslice (tile, slice);
1977 glBegin (GL_QUADS);
1979 glTexCoord2i (tilex, slicey);
1980 glVertex2i (dispx, dispy);
1982 glTexCoord2i (tilex+dispw, slicey);
1983 glVertex2i (dispx+dispw, dispy);
1985 glTexCoord2i (tilex+dispw, slicey+dh);
1986 glVertex2i (dispx+dispw, dispy+dh);
1988 glTexCoord2i (tilex, slicey+dh);
1989 glVertex2i (dispx, dispy+dh);
1991 glEnd ();
1993 dispy += dh;
1994 disph -= dh;
1995 slice++;
1996 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
1997 slicey = 0;
2000 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2001 CAMLreturn (Val_unit);
2004 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2005 value xoff_v, value yoff_v,
2006 value noff_v)
2008 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, noff_v);
2009 int xoff = Int_val (xoff_v);
2010 int yoff = Int_val (yoff_v);
2011 int noff = Int_val (noff_v);
2012 char *s = String_val (ptr_v);
2013 int hlmask = Int_val (hlinks_v);
2014 struct page *page = parse_pointer ("ml_postprocess", s);
2016 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2017 if (trylock ("ml_postprocess")) {
2018 noff = 0;
2019 goto done;
2021 if (hlmask & 2) {
2022 highlightslinks (page, xoff, yoff, noff);
2023 noff = page->slinkcount;
2025 showsel (page, xoff, yoff);
2026 unlock ("ml_postprocess");
2028 done:
2029 CAMLreturn (Val_int (noff));
2032 static fz_link *getlink (struct page *page, int x, int y)
2034 fz_point p;
2035 fz_matrix ctm;
2036 fz_link *link, *links;
2038 switch (page->type) {
2039 case DPDF:
2040 ctm = trimctm (page->u.pdfpage, page->pdimno);
2041 links = page->u.pdfpage->links;
2042 break;
2044 case DXPS:
2045 ctm = fz_identity;
2046 links = page->u.xpspage->links;
2047 break;
2049 default:
2050 return NULL;
2052 p.x = x;
2053 p.y = y;
2055 ctm = fz_concat (ctm, state.pagedims[page->pdimno].ctm);
2056 ctm = fz_invert_matrix (ctm);
2057 p = fz_transform_point (ctm, p);
2059 for (link = links; link; link = link->next) {
2060 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2061 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2062 return link;
2066 return NULL;
2069 static void ensuretext (struct page *page)
2071 if (state.gen != page->gen) {
2072 droptext (page);
2073 dropslinks (page);
2074 page->gen = state.gen;
2076 if (!page->text) {
2077 fz_device *tdev;
2079 page->text = fz_new_text_span (state.ctx);
2080 tdev = fz_new_text_device (state.ctx, page->text);
2081 fz_run_display_list (page->dlist,
2082 tdev,
2083 pagectm (page),
2084 fz_infinite_bbox, NULL);
2085 fz_free_device (tdev);
2089 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2091 CAMLparam2 (start_page_v, dir_v);
2092 CAMLlocal1 (ret_v);
2093 int i, dir = Int_val (dir_v);
2094 int start_page = Int_val (start_page_v);
2095 int end_page = dir > 0 ? state.pagecount : -1;
2097 ret_v = Val_int (0);
2098 if (!(state.type == DPDF || state.type == DXPS)) {
2099 goto done;
2102 lock ("ml_findpage_with_links");
2103 for (i = start_page + dir; i != end_page; i += dir) {
2104 int found;
2106 switch (state.type) {
2107 case DPDF:
2109 pdf_page *page = pdf_load_page (state.u.pdf, i);
2110 found = !!page->links;
2111 freepdfpage (page);
2113 break;
2114 case DXPS:
2116 xps_page *page = xps_load_page (state.u.xps, i);
2117 found = !!page->links;
2118 freexpspage (page);
2120 break;
2122 default:
2123 ARSERT ("invalid document type");
2126 if (found) {
2127 ret_v = caml_alloc_small (1, 1);
2128 Field (ret_v, 0) = Val_int (i);
2129 goto unlock;
2132 unlock:
2133 unlock ("ml_findpage_with_links");
2135 done:
2136 CAMLreturn (ret_v);
2139 enum { dir_first, dir_last};
2140 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
2142 CAMLprim value ml_findlink (value ptr_v, value dir_v)
2144 CAMLparam2 (ptr_v, dir_v);
2145 CAMLlocal2 (ret_v, pos_v);
2146 struct page *page;
2147 int dirtag, i, slinkindex;
2148 struct slink *found = NULL ,*slink;
2149 char *s = String_val (ptr_v);
2151 page = parse_pointer ("ml_findlink", s);
2152 ensureslinks (page);
2154 ret_v = Val_int (0);
2155 if (Is_block (dir_v)) {
2156 dirtag = Tag_val (dir_v);
2157 switch (dirtag) {
2158 case dir_first_visible:
2160 int x0, y0, dir, first_index, last_index;
2162 pos_v = Field (dir_v, 0);
2163 x0 = Int_val (Field (pos_v, 0));
2164 y0 = Int_val (Field (pos_v, 1));
2165 dir = Int_val (Field (pos_v, 2));
2167 if (dir >= 0) {
2168 dir = 1;
2169 first_index = 0;
2170 last_index = page->slinkcount;
2172 else {
2173 first_index = page->slinkcount - 1;
2174 last_index = -1;
2177 for (i = first_index; i != last_index; i += dir) {
2178 slink = &page->slinks[i];
2179 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
2180 found = slink;
2181 break;
2185 break;
2187 case dir_left:
2188 slinkindex = Int_val (Field (dir_v, 0));
2189 found = &page->slinks[slinkindex];
2190 for (i = slinkindex - 1; i >= 0; --i) {
2191 slink = &page->slinks[i];
2192 if (slink->bbox.x0 < found->bbox.x0) {
2193 found = slink;
2194 break;
2197 break;
2199 case dir_right:
2200 slinkindex = Int_val (Field (dir_v, 0));
2201 found = &page->slinks[slinkindex];
2202 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2203 slink = &page->slinks[i];
2204 if (slink->bbox.x0 > found->bbox.x0) {
2205 found = slink;
2206 break;
2209 break;
2211 case dir_down:
2212 slinkindex = Int_val (Field (dir_v, 0));
2213 found = &page->slinks[slinkindex];
2214 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2215 slink = &page->slinks[i];
2216 if (slink->bbox.y0 >= found->bbox.y0) {
2217 found = slink;
2218 break;
2221 break;
2223 case dir_up:
2224 slinkindex = Int_val (Field (dir_v, 0));
2225 found = &page->slinks[slinkindex];
2226 for (i = slinkindex - 1; i >= 0; --i) {
2227 slink = &page->slinks[i];
2228 if (slink->bbox.y0 <= found->bbox.y0) {
2229 found = slink;
2230 break;
2233 break;
2236 else {
2237 dirtag = Int_val (dir_v);
2238 switch (dirtag) {
2239 case dir_first:
2240 found = page->slinks;
2241 break;
2243 case dir_last:
2244 if (page->slinks) {
2245 found = page->slinks + (page->slinkcount - 1);
2247 break;
2250 if (found) {
2251 ret_v = caml_alloc_small (2, 1);
2252 Field (ret_v, 0) = Val_int (found - page->slinks);
2255 CAMLreturn (ret_v);
2258 enum { uuri, ugoto, utext, uunexpected, ulaunch, unamed, uremote };
2260 #define LINKTOVAL \
2262 int pageno; \
2264 switch (link->dest.kind) { \
2265 case FZ_LINK_GOTO: \
2267 fz_point p; \
2269 pageno = link->dest.ld.gotor.page; \
2270 p.x = 0; \
2271 p.y = 0; \
2273 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
2274 p.y = link->dest.ld.gotor.lt.y; \
2275 p = fz_transform_point (pdim->lctm, p); \
2277 tup_v = caml_alloc_tuple (2); \
2278 ret_v = caml_alloc_small (1, ugoto); \
2279 Field (tup_v, 0) = Val_int (pageno); \
2280 Field (tup_v, 1) = Val_int (p.y); \
2281 Field (ret_v, 0) = tup_v; \
2283 break; \
2285 case FZ_LINK_URI: \
2286 str_v = caml_copy_string (link->dest.ld.uri.uri); \
2287 ret_v = caml_alloc_small (1, uuri); \
2288 Field (ret_v, 0) = str_v; \
2289 break; \
2291 case FZ_LINK_LAUNCH: \
2292 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
2293 ret_v = caml_alloc_small (1, ulaunch); \
2294 Field (ret_v, 0) = str_v; \
2295 break; \
2297 case FZ_LINK_NAMED: \
2298 str_v = caml_copy_string (link->dest.ld.named.named); \
2299 ret_v = caml_alloc_small (1, unamed); \
2300 Field (ret_v, 0) = str_v; \
2301 break; \
2303 case FZ_LINK_GOTOR: \
2304 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
2305 pageno = link->dest.ld.gotor.page; \
2306 tup_v = caml_alloc_tuple (2); \
2307 ret_v = caml_alloc_small (1, uremote); \
2308 Field (tup_v, 0) = str_v; \
2309 Field (tup_v, 1) = Val_int (pageno); \
2310 Field (ret_v, 0) = tup_v; \
2311 break; \
2313 default: \
2315 char buf[80]; \
2317 snprintf (buf, sizeof (buf), \
2318 "unhandled link kind %d", link->dest.kind); \
2319 str_v = caml_copy_string (buf); \
2320 ret_v = caml_alloc_small (1, uunexpected); \
2321 Field (ret_v, 0) = str_v; \
2323 break; \
2327 CAMLprim value ml_getlink (value ptr_v, value n_v)
2329 CAMLparam2 (ptr_v, n_v);
2330 CAMLlocal3 (ret_v, tup_v, str_v);
2331 fz_link *link;
2332 struct page *page;
2333 struct pagedim *pdim;
2334 char *s = String_val (ptr_v);
2336 ret_v = Val_int (0);
2337 page = parse_pointer ("ml_getlink", s);
2338 ensureslinks (page);
2339 pdim = &state.pagedims[page->pdimno];
2340 link = page->slinks[Int_val (n_v)].link;
2341 LINKTOVAL;
2342 CAMLreturn (ret_v);
2345 CAMLprim value ml_getlinkcount (value ptr_v)
2347 CAMLparam1 (ptr_v);
2348 struct page *page;
2349 char *s = String_val (ptr_v);
2351 page = parse_pointer ("ml_getlinkcount", s);
2352 CAMLreturn (Val_int (page->slinkcount));
2355 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
2357 CAMLparam2 (ptr_v, n_v);
2358 CAMLlocal1 (ret_v);
2359 struct page *page;
2360 struct slink *slink;
2361 char *s = String_val (ptr_v);
2363 page = parse_pointer ("ml_getlinkrect", s);
2364 ensureslinks (page);
2366 slink = &page->slinks[Int_val (n_v)];
2367 ret_v = caml_alloc_tuple (4);
2368 Field (ret_v, 0) = Val_int (slink->bbox.x0);
2369 Field (ret_v, 1) = Val_int (slink->bbox.y0);
2370 Field (ret_v, 2) = Val_int (slink->bbox.x1);
2371 Field (ret_v, 3) = Val_int (slink->bbox.y1);
2372 CAMLreturn (ret_v);
2375 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
2377 CAMLparam3 (ptr_v, x_v, y_v);
2378 CAMLlocal3 (ret_v, tup_v, str_v);
2379 fz_link *link;
2380 struct page *page;
2381 char *s = String_val (ptr_v);
2382 int x = Int_val (x_v), y = Int_val (y_v);
2383 struct pagedim *pdim;
2385 ret_v = Val_int (0);
2386 if (trylock ("ml_whatsunder")) {
2387 goto done;
2390 page = parse_pointer ("ml_whatsunder", s);
2391 pdim = &state.pagedims[page->pdimno];
2392 x += pdim->bounds.x0;
2393 y += pdim->bounds.y0;
2394 link = getlink (page, x, y);
2395 if (link) {
2396 LINKTOVAL;
2398 else {
2399 int i;
2400 fz_text_span *span;
2402 ensuretext (page);
2403 for (span = page->text; span; span = span->next) {
2404 for (i = 0; i < span->len; ++i) {
2405 fz_bbox *b;
2406 b = &span->text[i].bbox;
2407 if ((x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1)) {
2408 const char *n2 =
2409 span->font && span->font->name
2410 ? span->font->name
2411 : "Span has no font name"
2413 FT_FaceRec *face = span->font->ft_face;
2414 if (face && face->family_name) {
2415 char *s;
2416 char *n1 = face->family_name;
2417 size_t l1 = strlen (n1);
2418 size_t l2 = strlen (n2);
2420 if (l1 != l2 || memcmp (n1, n2, l1)) {
2421 s = malloc (l1 + l2 + 2);
2422 if (s) {
2423 memcpy (s, n2, l2);
2424 s[l2] = '=';
2425 memcpy (s + l2 + 1, n1, l1 + 1);
2426 str_v = caml_copy_string (s);
2427 free (s);
2431 if (str_v == 0) {
2432 str_v = caml_copy_string (n2);
2434 ret_v = caml_alloc_small (1, utext);
2435 Field (ret_v, 0) = str_v;
2436 goto unlock;
2441 unlock:
2442 unlock ("ml_whatsunder");
2444 done:
2445 CAMLreturn (ret_v);
2448 CAMLprim value ml_seltext (value ptr_v, value rect_v)
2450 CAMLparam2 (ptr_v, rect_v);
2451 fz_bbox *b;
2452 struct page *page;
2453 fz_text_span *span;
2454 struct mark first, last;
2455 struct pagedim *pdim;
2456 int i, x0, x1, y0, y1;
2457 char *s = String_val (ptr_v);
2459 if (trylock ("ml_seltext")) {
2460 goto done;
2463 page = parse_pointer ("ml_seltext", s);
2464 ensuretext (page);
2466 pdim = &state.pagedims[page->pdimno];
2467 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;;
2468 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
2469 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
2470 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
2472 if (0) {
2473 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2474 glColor3ub (128, 128, 128);
2475 glRecti (x0, y0, x1, y1);
2476 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2479 first.span = NULL;
2480 last.span = NULL;
2482 last.i = first.i = 0;
2483 first.span = page->text;
2484 for (span = page->text; span; span = span->next) {
2485 for (i = 0; i < span->len; ++i) {
2486 b = &span->text[i].bbox;
2487 int selected = 0;
2489 if (x0 >= b->x0 && x0 <= b->x1 && y0 >= b->y0 && y0 <= b->y1) {
2490 first.i = i;
2491 first.span = span;
2492 selected = 1;
2494 if (x1 >= b->x0 && x1 <= b->x1 && y1 >= b->y0 && y1 <= b->y1) {
2495 last.i = i;
2496 last.span = span;
2497 selected = 1;
2499 if (0 && selected) {
2500 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2501 glColor3ub (128, 128, 128);
2502 glRecti (b->x0, b->y0, b->x1, b->y1);
2503 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2508 if (y1 < y0 || x1 < x0) {
2509 int swap = 0;
2511 if (first.span == last.span) {
2512 swap = 1;
2514 else {
2515 if (y1 < y0) {
2516 for (span = first.span; span && span != last.span;
2517 span = span->next) {
2518 if (span->eol) {
2519 swap = 1;
2520 break;
2526 if (swap) {
2527 i = first.i;
2528 span = first.span;
2529 first.i = last.i;
2530 first.span = last.span;
2531 last.i = i;
2532 last.span = span;
2536 page->fmark = first;
2537 page->lmark = last;
2539 unlock ("ml_seltext");
2541 done:
2542 CAMLreturn (Val_unit);
2545 static int pipespan (FILE *f, fz_text_span *span, int a, int b)
2547 char buf[4];
2548 int i, len, ret;
2550 for (i = a; i <= b; ++i) {
2551 len = runetochar (buf, &span->text[i].c);
2552 ret = fwrite (buf, len, 1, f);
2554 if (ret != 1) {
2555 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
2556 len, ret, strerror (errno));
2557 return -1;
2560 return 0;
2563 CAMLprim value ml_popen (value command_v, value fds_v)
2565 CAMLparam2 (command_v, fds_v);
2566 CAMLlocal2 (l_v, tup_v);
2567 int ret;
2568 posix_spawnattr_t attr;
2569 posix_spawn_file_actions_t fa;
2570 char *command = String_val (command_v);
2571 char *argv[] = { "/bin/sh", "-c", NULL, NULL };
2573 argv[2] = command;
2574 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
2575 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
2578 if ((ret = posix_spawnattr_init (&attr)) != 0) {
2579 unix_error (ret, "posix_spawnattr_init", Nothing);
2582 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
2583 unix_error (ret, "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK",
2584 Nothing);
2587 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
2588 int fd1, fd2;
2590 tup_v = Field (l_v, 0);
2591 fd1 = Int_val (Field (tup_v, 0));
2592 fd2 = Int_val (Field (tup_v, 1));
2593 if (fd2 < 0) {
2594 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
2595 unix_error (ret, "posix_spawn_file_actions_addclose", tup_v);
2598 else {
2599 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
2600 unix_error (ret, "posix_spawn_file_actions_adddup2", tup_v);
2605 if ((ret = posix_spawn (NULL, "/bin/sh", &fa, &attr, argv, environ))) {
2606 unix_error (ret, "posix_spawn", command_v);
2608 CAMLreturn (Val_unit);
2611 CAMLprim value ml_copysel (value fd_v, value ptr_v)
2613 CAMLparam1 (ptr_v);
2614 FILE *f;
2615 struct page *page;
2616 fz_text_span *span;
2617 char *s = String_val (ptr_v);
2619 if (trylock ("ml_copysel")) {
2620 goto done;
2623 page = parse_pointer ("ml_sopysel", s);
2625 if (!page->fmark.span || !page->lmark.span) {
2626 fprintf (stderr, "nothing to copy\n");
2627 goto unlock;
2630 f = fdopen (Int_val (fd_v), "w");
2631 if (!f) {
2632 fprintf (stderr, "failed to fopen sel pipe: %s\n",
2633 strerror (errno));
2634 f = stdout;
2635 if (close (Int_val (fd_v))) {
2636 fprintf (stderr, "failed to close sel pipe: %s\n",
2637 strerror (errno));
2641 for (span = page->fmark.span;
2642 span && span != page->lmark.span->next;
2643 span = span->next) {
2644 int a = span == page->fmark.span ? page->fmark.i : 0;
2645 int b = span == page->lmark.span ? page->lmark.i : span->len - 1;
2646 if (pipespan (f, span, a, b)) {
2647 goto close;
2649 if (span->eol) {
2650 if (putc ('\n', f) == EOF) {
2651 fprintf (stderr, "failed break line on sel pipe: %s\n",
2652 strerror (errno));
2653 goto close;
2657 page->lmark.span = NULL;
2658 page->fmark.span = NULL;
2660 close:
2661 if (f != stdout) {
2662 int ret = fclose (f);
2663 if (ret == -1) {
2664 if (errno != ECHILD) {
2665 fprintf (stderr, "failed to close sel pipe: %s\n",
2666 strerror (errno));
2670 unlock:
2671 unlock ("ml_copysel");
2673 done:
2674 CAMLreturn (Val_unit);
2677 CAMLprim value ml_getpdimrect (value pagedimno_v)
2679 CAMLparam1 (pagedimno_v);
2680 CAMLlocal1 (ret_v);
2681 int pagedimno = Int_val (pagedimno_v);
2682 fz_rect box;
2684 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
2685 if (trylock ("ml_getpdimrect")) {
2686 box = fz_empty_rect;
2688 else {
2689 box = state.pagedims[pagedimno].mediabox;
2690 unlock ("ml_getpdimrect");
2693 Store_double_field (ret_v, 0, box.x0);
2694 Store_double_field (ret_v, 1, box.x1);
2695 Store_double_field (ret_v, 2, box.y0);
2696 Store_double_field (ret_v, 3, box.y1);
2698 CAMLreturn (ret_v);
2701 static double getmaxw (void)
2703 int i;
2704 struct pagedim *p;
2705 double maxw = 0.0;
2707 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
2708 double x0, x1, w;
2710 x0 = MIN (p->mediabox.x0, p->mediabox.x1);
2711 x1 = MAX (p->mediabox.x0, p->mediabox.x1);
2713 w = x1 - x0;
2714 maxw = MAX (w, maxw);
2716 return maxw;
2719 CAMLprim value ml_getmaxw (value unit_v)
2721 CAMLparam1 (unit_v);
2722 CAMLlocal1 (ret_v);
2723 double maxw = 0.0;
2725 if (trylock ("ml_getmaxw")) {
2726 goto done;
2728 maxw = getmaxw ();
2729 unlock ("ml_getmaxw");
2730 done:
2731 ret_v = caml_copy_double (maxw);
2732 CAMLreturn (ret_v);
2735 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v, value dw_v)
2737 CAMLparam3 (winw_v, winh_v, dw_v);
2738 CAMLlocal1 (ret_v);
2739 int i;
2740 double zoom = 1.0;
2741 double maxw = 0.0, maxh = 0.0;
2742 struct pagedim *p;
2743 double winw = Int_val (winw_v);
2744 double winh = Int_val (winh_v);
2745 double dw = Int_val (dw_v);
2746 double pw = 1.0, ph = 1.0, num, den;
2748 if (trylock ("ml_zoom_for_height")) {
2749 goto done;
2752 if (state.proportional) {
2753 maxw = getmaxw ();
2756 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
2757 double x0, x1, y0, y1, w, h, scaledh, scale;
2759 x0 = MIN (p->mediabox.x0, p->mediabox.x1);
2760 x1 = MAX (p->mediabox.x0, p->mediabox.x1);
2761 y0 = MIN (p->mediabox.y0, p->mediabox.y1);
2762 y1 = MAX (p->mediabox.y0, p->mediabox.y1);
2764 w = x1 - x0;
2765 h = y1 - y0;
2767 if (state.proportional) {
2768 scale = w / maxw;
2769 scaledh = h * scale;
2771 else {
2772 scale = 1.0;
2773 scaledh = h;
2776 if (scaledh > maxh) {
2777 maxh = scaledh;
2778 ph = scaledh;
2779 pw = w * scale;
2783 num = (winh * pw) + (ph * dw);
2784 den = ph * winw;
2785 zoom = num / den;
2787 unlock ("ml_zoom_for_height");
2788 done:
2789 ret_v = caml_copy_double (zoom);
2790 CAMLreturn (ret_v);
2793 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
2795 CAMLparam4 (pt_v, x_v, y_v, string_v);
2796 CAMLlocal1 (ret_v);
2797 int pt = Int_val(pt_v);
2798 int x = Int_val (x_v);
2799 int y = Int_val (y_v);
2800 double w;
2802 w = draw_string (state.face, pt, x, y, String_val (string_v));
2803 ret_v = caml_copy_double (w);
2804 CAMLreturn (ret_v);
2807 CAMLprim value ml_measure_string (value pt_v, value string_v)
2809 CAMLparam2 (pt_v, string_v);
2810 CAMLlocal1 (ret_v);
2811 int pt = Int_val (pt_v);
2812 double w;
2814 w = measure_string (state.face, pt, String_val (string_v));
2815 ret_v = caml_copy_double (w);
2816 CAMLreturn (ret_v);
2819 CAMLprim value ml_getpagebox (value opaque_v)
2821 CAMLparam1 (opaque_v);
2822 CAMLlocal1 (ret_v);
2823 fz_bbox bbox;
2824 fz_device *dev;
2825 char *s = String_val (opaque_v);
2826 struct page *page = parse_pointer ("ml_getpagebox", s);
2828 ret_v = caml_alloc_tuple (4);
2829 dev = fz_new_bbox_device (state.ctx, &bbox);
2830 dev->hints |= FZ_IGNORE_SHADE;
2832 switch (page->type) {
2833 case DPDF:
2834 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, pagectm (page), NULL);
2835 break;
2837 case DXPS:
2838 xps_run_page (state.u.xps, page->u.xpspage, dev, pagectm (page), NULL);
2839 break;
2841 default:
2842 bbox = fz_infinite_bbox;
2843 break;
2846 fz_free_device (dev);
2847 Field (ret_v, 0) = Val_int (bbox.x0);
2848 Field (ret_v, 1) = Val_int (bbox.y0);
2849 Field (ret_v, 2) = Val_int (bbox.x1);
2850 Field (ret_v, 3) = Val_int (bbox.y1);
2852 CAMLreturn (ret_v);
2855 CAMLprim value ml_setaalevel (value level_v)
2857 CAMLparam1 (level_v);
2859 state.aalevel = Int_val (level_v);
2860 CAMLreturn (Val_unit);
2863 #undef pixel
2864 #include <X11/Xlib.h>
2865 #include <GL/glx.h>
2867 static struct {
2868 Display *dpy;
2869 GLXContext ctx;
2870 GLXDrawable drawable;
2871 } glx;
2873 #include "keysym2ucs.c"
2875 CAMLprim value ml_keysymtoutf8 (value keysym_v)
2877 CAMLparam1 (keysym_v);
2878 CAMLlocal1 (str_v);
2879 KeySym keysym = Int_val (keysym_v);
2880 Rune rune;
2881 int len;
2882 char buf[5];
2884 rune = keysym2ucs (keysym);
2885 len = runetochar (buf, &rune);
2886 buf[len] = 0;
2887 str_v = caml_copy_string (buf);
2888 CAMLreturn (str_v);
2891 CAMLprim value ml_glx (value win_v)
2893 CAMLparam1 (win_v);
2894 XVisualInfo *visual;
2895 int screen, wid = Int_val (win_v);
2896 int attributes[] = { GLX_RGBA, GLX_DOUBLEBUFFER, None };
2898 glx.dpy = XOpenDisplay (NULL);
2899 if (!glx.dpy) {
2900 caml_failwith ("XOpenDisplay failed");
2903 screen = DefaultScreen (glx.dpy);
2904 visual = glXChooseVisual (glx.dpy, screen, attributes);
2905 if (!visual) {
2906 XCloseDisplay (glx.dpy);
2907 glx.dpy = NULL;
2908 caml_failwith ("glXChooseVisual");
2911 glx.ctx = glXCreateContext (glx.dpy, visual, NULL, True);
2912 if (!glx.ctx) {
2913 XCloseDisplay (glx.dpy);
2914 XFree (visual);
2915 glx.dpy = NULL;
2916 caml_failwith ("glXCreateContext");
2919 XFree (visual);
2920 if (!glXMakeCurrent (glx.dpy, wid, glx.ctx)) {
2921 glXDestroyContext (glx.dpy, glx.ctx);
2922 XCloseDisplay (glx.dpy);
2923 glx.dpy = NULL;
2924 glx.ctx = NULL;
2925 caml_failwith ("glXMakeCurrent");
2927 glx.drawable = wid;
2929 CAMLreturn (Val_unit);
2932 CAMLprim value ml_swapb (value unit_v)
2934 CAMLparam1 (unit_v);
2935 glXSwapBuffers (glx.dpy, glx.drawable);
2936 CAMLreturn (Val_unit);
2939 enum { piunknown, pilinux, piosx, pisun, pifreebsd,
2940 pidragonflybsd, piopenbsd, pinetbsd, picygwin };
2942 CAMLprim value ml_platform (value unit_v)
2944 CAMLparam1 (unit_v);
2945 int platid = piunknown;
2947 #if defined __linux__
2948 platid = pilinux;
2949 #elif defined __CYGWIN__
2950 platid = picygwin;
2951 #elif defined __DragonFly__
2952 platid = pidragonflybsd;
2953 #elif defined __FreeBSD__
2954 platid = pifreebsd;
2955 #elif defined __OpenBSD__
2956 platid = piopenbsd;
2957 #elif defined __NetBSD__
2958 platid = pinetbsd;
2959 #elif defined __sun__
2960 platid = pisun;
2961 #elif defined __APPLE__
2962 platid = piosx;
2963 #endif
2964 CAMLreturn (Val_int (platid));
2967 CAMLprim value ml_cloexec (value fd_v)
2969 CAMLparam1 (fd_v);
2970 int fd = Int_val (fd_v);
2972 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
2973 uerror ("fcntl", Nothing);
2975 CAMLreturn (Val_unit);
2978 CAMLprim value ml_init (value pipe_v, value params_v)
2980 CAMLparam2 (pipe_v, params_v);
2981 CAMLlocal2 (trim_v, fuzz_v);
2982 int ret;
2983 int texcount;
2984 char *fontpath;
2985 int colorspace;
2986 int mustoresize;
2987 struct sigaction sa;
2989 state.cr = Int_val (Field (pipe_v, 0));
2990 state.cw = Int_val (Field (pipe_v, 1));
2991 state.rotate = Int_val (Field (params_v, 0));
2992 state.proportional = Bool_val (Field (params_v, 1));
2993 trim_v = Field (params_v, 2);
2994 texcount = Int_val (Field (params_v, 3));
2995 state.sliceheight = Int_val (Field (params_v, 4));
2996 mustoresize = Int_val (Field (params_v, 5));
2997 colorspace = Int_val (Field (params_v, 6));
2998 fontpath = String_val (Field (params_v, 7));
3000 state.ctx = fz_new_context (NULL, NULL, mustoresize);
3002 state.trimmargins = Bool_val (Field (trim_v, 0));
3003 fuzz_v = Field (trim_v, 1);
3004 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
3005 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
3006 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
3007 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
3009 set_tex_params (colorspace);
3011 if (*fontpath) {
3012 state.face = load_font (fontpath);
3014 else {
3015 unsigned int len;
3016 void *base = pdf_find_substitute_font (0, 0, 0, 0, &len);
3018 state.face = load_builtin_font (base, len);
3020 if (!state.face) _exit (1);
3022 realloctexts (texcount);
3024 #ifdef __CYGWIN__
3025 sa.sa_handler = SIG_IGN;
3026 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
3027 #else
3028 sa.sa_handler = SIG_DFL;
3029 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP | SA_NOCLDWAIT;
3030 #endif
3031 if (sigemptyset (&sa.sa_mask)) {
3032 err (1, "sigemptyset");
3034 if (sigaction (SIGCHLD, &sa, NULL)) {
3035 err (1, "sigaction");
3038 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
3039 if (ret) {
3040 errx (1, "pthread_create: %s", strerror (ret));
3043 CAMLreturn (Val_unit);