Rename path to the cache directory
[llpp.git] / link.c
blobb4673c1a4a0f13c772935392a016f32ff45cba61
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>
8 #include <wchar.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 #else
19 #include <spawn.h>
20 #endif
22 #include <regex.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 #define MIN(a,b) ((a) < (b) ? (a) : (b))
49 #define MAX(a,b) ((a) > (b) ? (a) : (b))
51 #if defined __GNUC__
52 #define NORETURN_ATTR __attribute__ ((noreturn))
53 #define UNUSED_ATTR __attribute__ ((unused))
54 #define OPTIMIZE_ATTR(n) __attribute__ ((optimize ("O"#n)))
55 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
56 #else
57 #define NORETURN_ATTR
58 #define UNUSED_ATTR
59 #define OPTIMIZE_ATTR(n)
60 #define GCC_FMT_ATTR(a, b)
61 #endif
63 #define FMT_s "zu"
65 #define FMT_ptr "p"
66 #define FMT_ptr_cast(p) (p)
67 #define FMT_ptr_cast2(p) (p)
69 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
70 err (int exitcode, const char *fmt, ...)
72 va_list ap;
73 int savederrno;
75 savederrno = errno;
76 va_start (ap, fmt);
77 vfprintf (stderr, fmt, ap);
78 va_end (ap);
79 fprintf (stderr, ": %s\n", strerror (savederrno));
80 fflush (stderr);
81 _exit (exitcode);
84 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
85 errx (int exitcode, const char *fmt, ...)
87 va_list ap;
89 va_start (ap, fmt);
90 vfprintf (stderr, fmt, ap);
91 va_end (ap);
92 fputc ('\n', stderr);
93 fflush (stderr);
94 _exit (exitcode);
97 #ifndef GL_TEXTURE_RECTANGLE_ARB
98 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
99 #endif
101 #ifndef GL_BGRA
102 #define GL_BGRA 0x80E1
103 #endif
105 #ifndef GL_UNSIGNED_INT_8_8_8_8
106 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
107 #endif
109 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
110 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
111 #endif
113 #if 0
114 #define lprintf printf
115 #else
116 #define lprintf(...)
117 #endif
119 #define ARSERT(cond) for (;;) { \
120 if (!(cond)) { \
121 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
123 break; \
126 struct slice {
127 int h;
128 int texindex;
131 struct tile {
132 int x, y, w, h;
133 int slicecount;
134 int sliceheight;
135 struct pbo *pbo;
136 fz_pixmap *pixmap;
137 struct slice slices[1];
140 struct pagedim {
141 int pageno;
142 int rotate;
143 int left;
144 int tctmready;
145 fz_irect bounds;
146 fz_rect pagebox;
147 fz_rect mediabox;
148 fz_matrix ctm, zoomctm, lctm, tctm;
151 struct slink {
152 fz_irect bbox;
153 fz_link *link;
156 enum { DPDF, DXPS, DCBZ };
158 struct page {
159 int tgen;
160 int sgen;
161 int type;
162 int pageno;
163 int pdimno;
164 fz_text_page *text;
165 fz_text_sheet *sheet;
166 union {
167 void *ptr;
168 pdf_page *pdfpage;
169 xps_page *xpspage;
170 cbz_page *cbzpage;
171 } u;
172 fz_display_list *dlist;
173 int slinkcount;
174 struct slink *slinks;
175 struct mark {
176 int i;
177 fz_text_span *span;
178 } fmark, lmark;
179 void (*freepage) (void *);
182 struct {
183 int type;
184 int sliceheight;
185 struct pagedim *pagedims;
186 int pagecount;
187 int pagedimcount;
188 union {
189 pdf_document *pdf;
190 xps_document *xps;
191 cbz_document *cbz;
192 } u;
193 fz_context *ctx;
194 int w, h;
196 int texindex;
197 int texcount;
198 GLuint *texids;
200 GLenum texiform;
201 GLenum texform;
202 GLenum texty;
204 fz_colorspace *colorspace;
206 struct {
207 int w, h;
208 struct slice *slice;
209 } *texowners;
211 int rotate;
212 enum { FitWidth, FitProportional, FitPage } fitmodel;
213 int trimmargins;
214 int needoutline;
215 int gen;
216 int aalevel;
218 int trimanew;
219 fz_irect trimfuzz;
220 fz_pixmap *pig;
222 pthread_t thread;
223 int cr, cw;
224 FT_Face face;
226 void (*closedoc) (void);
227 void (*freepage) (void *);
229 char *trimcachepath;
231 GLuint stid;
233 int pbo_usable;
234 void (*glBindBufferARB) (GLenum, GLuint);
235 GLboolean (*glUnmapBufferARB) (GLenum);
236 void *(*glMapBufferARB) (GLenum, GLenum);
237 void (*glBufferDataARB) (GLenum, GLsizei, void *, GLenum);
238 void (*glGenBuffersARB) (GLsizei, GLuint *);
239 void (*glDeleteBuffersARB) (GLsizei, GLuint *);
240 } state;
242 struct pbo {
243 GLuint id;
244 void *ptr;
245 size_t size;
248 static void UNUSED_ATTR debug_rect (const char *cap, fz_rect r)
250 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
253 static void UNUSED_ATTR debug_bbox (const char *cap, fz_irect r)
255 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
258 static void UNUSED_ATTR debug_matrix (const char *cap, fz_matrix m)
260 printf ("%s(matrix) %.2f,%.2f,%.2f,%.2f %.2f %.2f\n", cap,
261 m.a, m.b, m.c, m.d, m.e, m.f);
264 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
266 static void lock (const char *cap)
268 int ret = pthread_mutex_lock (&mutex);
269 if (ret) {
270 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
274 static void unlock (const char *cap)
276 int ret = pthread_mutex_unlock (&mutex);
277 if (ret) {
278 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
282 static int trylock (const char *cap)
284 int ret = pthread_mutex_trylock (&mutex);
285 if (ret && ret != EBUSY) {
286 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
288 return ret == EBUSY;
291 static void *parse_pointer (const char *cap, const char *s)
293 int ret;
294 void *ptr;
296 ret = sscanf (s, "%" FMT_ptr, FMT_ptr_cast (&ptr));
297 if (ret != 1) {
298 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
300 return ptr;
303 static double now (void)
305 struct timeval tv;
307 if (gettimeofday (&tv, NULL)) {
308 err (1, "gettimeofday");
310 return tv.tv_sec + tv.tv_usec*1e-6;
313 static int hasdata (void)
315 int ret, avail;
316 ret = ioctl (state.cr, FIONREAD, &avail);
317 if (ret) err (1, "hasdata: FIONREAD error ret=%d", ret);
318 return avail > 0;
321 CAMLprim value ml_hasdata (value fd_v)
323 CAMLparam1 (fd_v);
324 int ret, avail;
326 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
327 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
328 CAMLreturn (Val_bool (avail > 0));
331 static void readdata (void *p, int size)
333 ssize_t n;
335 again:
336 n = read (state.cr, p, size);
337 if (n < 0) {
338 if (errno == EINTR) goto again;
339 err (1, "read (req %d, ret %zd)", size, n);
341 if (n - size) {
342 if (!n) errx (1, "EOF while reading");
343 errx (1, "read (req %d, ret %zd)", size, n);
347 static void writedata (char *p, int size)
349 char buf[4];
350 ssize_t n;
352 buf[0] = (size >> 24) & 0xff;
353 buf[1] = (size >> 16) & 0xff;
354 buf[2] = (size >> 8) & 0xff;
355 buf[3] = (size >> 0) & 0xff;
357 n = write (state.cw, buf, 4);
358 if (n != 4) {
359 if (!n) errx (1, "EOF while writing length");
360 err (1, "write %zd", n);
363 n = write (state.cw, p, size);
364 if (n - size) {
365 if (!n) errx (1, "EOF while writing data");
366 err (1, "write (req %d, ret %zd)", size, n);
370 static int readlen (void)
372 unsigned char p[4];
374 readdata (p, 4);
375 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
378 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
380 int size = 200, len;
381 va_list ap;
382 char *buf;
384 buf = malloc (size);
385 for (;;) {
386 if (!buf) err (1, "malloc for temp buf (%d bytes) failed", size);
388 va_start (ap, fmt);
389 len = vsnprintf (buf, size, fmt, ap);
390 va_end (ap);
392 if (len > -1 && len < size) {
393 writedata (buf, len);
394 break;
397 if (len > -1) {
398 size = len + 1;
400 else {
401 size *= 2;
403 buf = realloc (buf, size);
405 free (buf);
408 static void closepdf (void)
410 if (state.u.pdf) {
411 pdf_close_document (state.u.pdf);
412 state.u.pdf = NULL;
416 static void closexps (void)
418 if (state.u.xps) {
419 xps_close_document (state.u.xps);
420 state.u.xps = NULL;
424 static void closecbz (void)
426 if (state.u.cbz) {
427 cbz_close_document (state.u.cbz);
428 state.u.cbz = NULL;
432 static void freepdfpage (void *ptr)
434 pdf_free_page (state.u.pdf, ptr);
437 static void freeemptyxxxpage (void *ptr)
439 (void) ptr;
442 static void freexpspage (void *ptr)
444 xps_free_page (state.u.xps, ptr);
447 static void freecbzpage (void *ptr)
449 cbz_free_page (state.u.cbz, ptr);
452 static void openxref (char *filename, char *password)
454 int i, len;
456 for (i = 0; i < state.texcount; ++i) {
457 state.texowners[i].w = -1;
458 state.texowners[i].slice = NULL;
461 if (state.closedoc) state.closedoc ();
463 len = strlen (filename);
465 state.type = DPDF;
466 if (len > 4) {
467 char ext[4];
469 ext[0] = tolower ((int) filename[len-3]);
470 ext[1] = tolower ((int) filename[len-2]);
471 ext[2] = tolower ((int) filename[len-1]);
473 /**/ if (ext[0] == 'x' && ext[1] == 'p' && ext[2] == 's') {
474 state.type = DXPS;
476 else if (ext[0] == 'c' && ext[1] == 'b' && ext[2] == 'z') {
477 state.type = DCBZ;
481 if (state.pagedims) {
482 free (state.pagedims);
483 state.pagedims = NULL;
485 state.pagedimcount = 0;
487 fz_set_aa_level (state.ctx, state.aalevel);
488 switch (state.type) {
489 case DPDF:
490 state.u.pdf = pdf_open_document (state.ctx, filename);
491 if (pdf_needs_password (state.u.pdf)) {
492 int okay = pdf_authenticate_password (state.u.pdf, password);
493 if (!okay) {
494 errx (1, "invalid password");
497 state.pagecount = pdf_count_pages (state.u.pdf);
498 state.closedoc = closepdf;
499 state.freepage = freepdfpage;
500 break;
502 case DXPS:
503 state.u.xps = xps_open_document (state.ctx, filename);
504 state.pagecount = xps_count_pages (state.u.xps);
505 state.closedoc = closexps;
506 state.freepage = freexpspage;
507 break;
509 case DCBZ:
510 state.u.cbz = cbz_open_document (state.ctx, filename);
511 state.pagecount = cbz_count_pages (state.u.cbz);
512 state.closedoc = closecbz;
513 state.freepage = freecbzpage;
514 break;
518 static void pdfinfo (void)
520 if (state.type == DPDF) {
521 pdf_obj *infoobj;
523 printd ("info PDF version\t%d.%d",
524 state.u.pdf->version / 10, state.u.pdf->version % 10);
526 infoobj = pdf_dict_gets (state.u.pdf->trailer, "Info");
527 if (infoobj) {
528 int i;
529 char *s;
530 char *items[] = { "Title", "Author", "Creator",
531 "Producer", "CreationDate" };
533 for (i = 0; i < sizeof (items) / sizeof (*items); ++i) {
534 pdf_obj *obj = pdf_dict_gets (infoobj, items[i]);
535 s = pdf_to_utf8 (state.u.pdf, obj);
536 if (*s) {
537 if (i == 0) {
538 printd ("title %s", s);
540 printd ("info %s\t%s", items[i], s);
542 fz_free (state.ctx, s);
545 printd ("infoend");
549 static void unlinktile (struct tile *tile)
551 int i;
553 for (i = 0; i < tile->slicecount; ++i) {
554 struct slice *s = &tile->slices[i];
556 if (s->texindex != -1) {
557 if (state.texowners[s->texindex].slice == s) {
558 state.texowners[s->texindex].slice = NULL;
564 static void freepage (struct page *page)
566 if (page->text) {
567 fz_free_text_page (state.ctx, page->text);
569 if (page->sheet) {
570 fz_free_text_sheet (state.ctx, page->sheet);
572 if (page->slinks) {
573 free (page->slinks);
575 page->freepage (page->u.ptr);
576 fz_free_display_list (state.ctx, page->dlist);
577 free (page);
580 static void freetile (struct tile *tile)
582 unlinktile (tile);
583 if (!tile->pbo) {
584 #ifndef PIGGYBACK
585 fz_drop_pixmap (state.ctx, tile->pixmap);
586 #else
587 if (state.pig) {
588 fz_drop_pixmap (state.ctx, state.pig);
590 state.pig = tile->pixmap;
591 #endif
593 else {
594 free (tile->pbo);
595 fz_drop_pixmap (state.ctx, tile->pixmap);
597 free (tile);
600 #ifdef __ALTIVEC__
601 #include <stdint.h>
602 #include <altivec.h>
604 static int cacheline32bytes;
606 static void __attribute__ ((constructor)) clcheck (void)
608 char **envp = environ;
609 unsigned long *auxv;
611 while (*envp++);
613 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
614 if (*auxv == 19) {
615 cacheline32bytes = auxv[1] == 32;
616 return;
621 static void OPTIMIZE_ATTR (3) clearpixmap (fz_pixmap *pixmap)
623 size_t size = pixmap->w * pixmap->h * pixmap->n;
624 if (cacheline32bytes && size > 32) {
625 intptr_t a1, a2, diff;
626 size_t sizea, i;
627 vector unsigned char v = vec_splat_u8 (-1);
628 vector unsigned char *p;
630 a1 = a2 = (intptr_t) pixmap->samples;
631 a2 = (a1 + 31) & ~31;
632 diff = a2 - a1;
633 sizea = size - diff;
634 p = (void *) a2;
636 while (a1 != a2) *(char *) a1++ = 0xff;
637 for (i = 0; i < (sizea & ~31); i += 32) {
638 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
639 vec_st (v, i, p);
640 vec_st (v, i + 16, p);
642 while (i < sizea) *((char *) a1 + i++) = 0xff;
644 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
646 #else
647 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
648 #endif
650 static void trimctm (pdf_page *page, int pindex)
652 fz_matrix ctm;
653 struct pagedim *pdim = &state.pagedims[pindex];
655 if (!pdim->tctmready) {
656 if (state.trimmargins) {
657 fz_rect realbox;
658 fz_matrix rm, sm, tm, im, ctm1;
660 fz_rotate (&rm, -pdim->rotate);
661 fz_scale (&sm, 1, -1);
662 fz_concat (&ctm, &rm, &sm);
663 realbox = pdim->mediabox;
664 fz_transform_rect (&realbox, &ctm);
665 fz_translate (&tm, -realbox.x0, -realbox.y0);
666 fz_concat (&ctm1, &ctm, &tm);
667 fz_invert_matrix (&im, &page->ctm);
668 fz_concat (&ctm, &im, &ctm1);
670 else {
671 ctm = fz_identity;
673 pdim->tctm = ctm;
674 pdim->tctmready = 1;
678 static fz_matrix pagectm (struct page *page)
680 fz_matrix ctm, tm;
682 if (page->type == DPDF) {
683 trimctm (page->u.pdfpage, page->pdimno);
684 fz_concat (&ctm,
685 &state.pagedims[page->pdimno].tctm,
686 &state.pagedims[page->pdimno].ctm);
688 else {
689 struct pagedim *pdim = &state.pagedims[page->pdimno];
691 fz_translate (&tm, -pdim->mediabox.x0, -pdim->mediabox.y0);
692 fz_concat (&ctm, &tm, &state.pagedims[page->pdimno].ctm);
694 return ctm;
697 static void *loadpage (int pageno, int pindex)
699 fz_device *dev;
700 struct page *page = NULL;
702 page = calloc (sizeof (struct page), 1);
703 if (!page) {
704 err (1, "calloc page %d", pageno);
707 page->dlist = fz_new_display_list (state.ctx);
708 dev = fz_new_list_device (state.ctx, page->dlist);
709 fz_try (state.ctx) {
710 switch (state.type) {
711 case DPDF:
712 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
713 pdf_run_page (state.u.pdf, page->u.pdfpage, dev,
714 &fz_identity, NULL);
715 page->freepage = freepdfpage;
716 break;
718 case DXPS:
719 page->u.xpspage = xps_load_page (state.u.xps, pageno);
720 xps_run_page (state.u.xps, page->u.xpspage, dev,
721 &fz_identity, NULL);
722 page->freepage = freexpspage;
723 break;
725 case DCBZ:
726 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
727 cbz_run_page (state.u.cbz, page->u.cbzpage, dev,
728 &fz_identity, NULL);
729 page->freepage = freecbzpage;
730 break;
733 fz_catch (state.ctx) {
734 page->u.ptr = NULL;
735 page->freepage = freeemptyxxxpage;
737 fz_free_device (dev);
739 page->pdimno = pindex;
740 page->pageno = pageno;
741 page->sgen = state.gen;
742 page->tgen = state.gen;
743 page->type = state.type;
745 return page;
748 static struct tile *alloctile (int h)
750 int i;
751 int slicecount;
752 size_t tilesize;
753 struct tile *tile;
755 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
756 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
757 tile = calloc (tilesize, 1);
758 if (!tile) {
759 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
761 for (i = 0; i < slicecount; ++i) {
762 int sh = MIN (h, state.sliceheight);
763 tile->slices[i].h = sh;
764 tile->slices[i].texindex = -1;
765 h -= sh;
767 tile->slicecount = slicecount;
768 tile->sliceheight = state.sliceheight;
769 return tile;
772 #ifdef OBSCURED_OPT
773 struct obs {
774 int cured;
775 fz_irect b;
778 static void obs_fill_image (fz_device *dev, fz_image *image,
779 const fz_matrix *ctm, float alpha)
781 struct obs *obs = dev->user;
783 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
784 fz_irect b;
785 fz_rect rect = fz_unit_rect;
787 fz_transform_rect (&rect, ctm);
788 fz_round_rect (&b, &rect);
789 fz_intersect_irect (&b, &obs->b);
790 obs->cured = b.x0 == obs->b.x0
791 && b.x1 == obs->b.x1
792 && b.y0 == obs->b.y0
793 && b.y1 == obs->b.y1;
797 static int obscured (struct page *page, fz_irect bbox)
799 fz_rect rect;
800 fz_matrix ctm;
801 fz_device dev;
802 struct obs obs;
804 memset (&dev, 0, sizeof (dev));
805 memset (&obs, 0, sizeof (obs));
806 dev.hints = 0;
807 dev.flags = 0;
808 dev.user = &obs;
809 dev.ctx = state.ctx;
810 dev.fill_image = obs_fill_image;
811 obs.b = bbox;
812 fz_rect_from_irect (&rect, &bbox);
813 ctm = pagectm (page);
814 fz_run_display_list (page->dlist, &dev, &ctm, &rect, NULL);
815 return obs.cured;
817 #define OBSCURED obscured
818 #else
819 #define OBSCURED(a, b) 0
820 #endif
822 static struct tile *rendertile (struct page *page, int x, int y, int w, int h,
823 struct pbo *pbo)
825 fz_rect rect;
826 fz_irect bbox;
827 fz_matrix ctm;
828 fz_device *dev;
829 struct tile *tile;
830 struct pagedim *pdim;
832 tile = alloctile (h);
833 pdim = &state.pagedims[page->pdimno];
835 bbox = pdim->bounds;
836 bbox.x0 += x;
837 bbox.y0 += y;
838 bbox.x1 = bbox.x0 + w;
839 bbox.y1 = bbox.y0 + h;
841 if (state.pig) {
842 if (state.pig->w == w
843 && state.pig->h == h
844 && state.pig->colorspace == state.colorspace) {
845 tile->pixmap = state.pig;
846 tile->pixmap->x = bbox.x0;
847 tile->pixmap->y = bbox.y0;
849 else {
850 fz_drop_pixmap (state.ctx, state.pig);
852 state.pig = NULL;
854 if (!tile->pixmap) {
855 if (pbo) {
856 tile->pixmap =
857 fz_new_pixmap_with_bbox_and_data (state.ctx, state.colorspace,
858 &bbox, pbo->ptr);
859 tile->pbo = pbo;
861 else {
862 tile->pixmap =
863 fz_new_pixmap_with_bbox (state.ctx, state.colorspace, &bbox);
867 tile->w = w;
868 tile->h = h;
869 if (!page->u.ptr || ((w < 128 && h < 128) || !OBSCURED (page, bbox))) {
870 clearpixmap (tile->pixmap);
872 dev = fz_new_draw_device (state.ctx, tile->pixmap);
873 ctm = pagectm (page);
874 fz_rect_from_irect (&rect, &bbox);
875 fz_run_display_list (page->dlist, dev, &ctm, &rect, NULL);
876 fz_free_device (dev);
878 return tile;
881 static void initpdims (void)
883 double start, end;
884 int pageno, trim, show;
885 FILE *trimf = NULL;
886 int trimw = 0;
888 start = now ();
890 if (state.trimmargins && state.trimcachepath) {
891 trimf = fopen (state.trimcachepath, "rb");
892 if (!trimf) {
893 trimf = fopen (state.trimcachepath, "wb");
894 trimw = 1;
897 for (pageno = 0; pageno < state.pagecount; ++pageno) {
898 int rotate = 0;
899 struct pagedim *p;
900 fz_rect mediabox;
902 switch (state.type) {
903 case DPDF: {
904 pdf_obj *pageobj = state.u.pdf->page_objs[pageno];
906 if (state.trimmargins) {
907 pdf_obj *obj;
908 pdf_page *page;
910 fz_try (state.ctx) {
911 page = pdf_load_page (state.u.pdf, pageno);
912 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
913 trim = state.trimanew || !obj;
914 if (trim) {
915 fz_rect rect;
916 fz_matrix ctm;
917 fz_device *dev;
919 dev = fz_new_bbox_device (state.ctx, &rect);
920 dev->hints |= FZ_IGNORE_SHADE;
921 fz_invert_matrix (&ctm, &page->ctm);
922 pdf_run_page (state.u.pdf, page, dev, &fz_identity, NULL);
923 fz_free_device (dev);
925 rect.x0 += state.trimfuzz.x0;
926 rect.x1 += state.trimfuzz.x1;
927 rect.y0 += state.trimfuzz.y0;
928 rect.y1 += state.trimfuzz.y1;
929 fz_transform_rect (&rect, &ctm);
930 fz_intersect_rect (&rect, &page->mediabox);
932 if (fz_is_empty_rect (&rect)) {
933 mediabox = page->mediabox;
935 else {
936 mediabox = rect;
939 obj = pdf_new_array (state.ctx, 4);
940 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x0));
941 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y0));
942 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x1));
943 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y1));
944 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
946 else {
947 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
948 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
949 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
950 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
953 rotate = page->rotate;
954 pdf_free_page (state.u.pdf, page);
956 show = trim ? pageno % 5 == 0 : pageno % 20 == 0;
957 if (show) {
958 printd ("progress %f Trimming %d",
959 (double) (pageno + 1) / state.pagecount,
960 pageno + 1);
963 fz_catch (state.ctx) {
964 fprintf (stderr, "failed to load page %d\n", pageno+1);
967 else {
968 fz_rect cropbox;
970 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "MediaBox"),
971 &mediabox);
972 if (fz_is_empty_rect (&mediabox)) {
973 fprintf (stderr, "cannot find page size for page %d\n",
974 pageno+1);
975 mediabox.x0 = 0;
976 mediabox.y0 = 0;
977 mediabox.x1 = 612;
978 mediabox.y1 = 792;
981 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "CropBox"),
982 &cropbox);
983 if (!fz_is_empty_rect (&cropbox)) {
984 fz_intersect_rect (&mediabox, &cropbox);
986 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
988 break;
991 case DXPS:
993 xps_page *page;
995 fz_try (state.ctx) {
996 page = xps_load_page (state.u.xps, pageno);
997 xps_bound_page (state.u.xps, page, &mediabox);
998 rotate = 0;
999 if (state.trimmargins) {
1000 fz_rect rect;
1001 fz_device *dev;
1003 dev = fz_new_bbox_device (state.ctx, &rect);
1004 dev->hints |= FZ_IGNORE_SHADE;
1005 xps_run_page (state.u.xps, page, dev,
1006 &fz_identity, NULL);
1007 fz_free_device (dev);
1009 rect.x0 += state.trimfuzz.x0;
1010 rect.x1 += state.trimfuzz.x1;
1011 rect.y0 += state.trimfuzz.y0;
1012 rect.y1 += state.trimfuzz.y1;
1013 fz_intersect_rect (&rect, &mediabox);
1015 if (!fz_is_empty_rect (&rect)) {
1016 mediabox = rect;
1019 xps_free_page (state.u.xps, page);
1020 printd ("progress %f loading %d",
1021 (double) (pageno + 1) / state.pagecount,
1022 pageno + 1);
1024 fz_catch (state.ctx) {
1027 break;
1029 case DCBZ:
1031 rotate = 0;
1032 if (state.trimmargins && trimw) {
1033 cbz_page *page;
1035 fz_try (state.ctx) {
1036 page = cbz_load_page (state.u.cbz, pageno);
1037 cbz_bound_page (state.u.cbz, page, &mediabox);
1038 cbz_free_page (state.u.cbz, page);
1039 printd ("progress %f Trimming %d",
1040 (double) (pageno + 1) / state.pagecount,
1041 pageno + 1);
1043 fz_catch (state.ctx) {
1044 fprintf (stderr, "failed to load page %d\n", pageno+1);
1045 mediabox.x0 = 0;
1046 mediabox.y0 = 0;
1047 mediabox.x1 = 900;
1048 mediabox.y1 = 900;
1050 if (trimf) {
1051 int n = fwrite (&mediabox, sizeof (mediabox), 1, trimf);
1052 if (n - 1) {
1053 err (1, "fwrite trim mediabox");
1057 else {
1058 if (trimf) {
1059 int n = fread (&mediabox, sizeof (mediabox), 1, trimf);
1060 if (n - 1) {
1061 err (1, "fread trim mediabox");
1064 else {
1065 mediabox.x0 = mediabox.y0 = 0;
1066 mediabox.x1 = 900;
1067 mediabox.y1 = 900;
1071 break;
1073 default:
1074 ARSERT (0 && state.type);
1077 if (state.pagedimcount == 0
1078 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
1079 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
1080 size_t size;
1082 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
1083 state.pagedims = realloc (state.pagedims, size);
1084 if (!state.pagedims) {
1085 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
1086 size, state.pagedimcount + 1);
1089 p = &state.pagedims[state.pagedimcount++];
1090 p->rotate = rotate;
1091 p->mediabox = mediabox;
1092 p->pageno = pageno;
1095 end = now ();
1096 if (state.trimmargins) {
1097 printd ("progress 1 Trimmed %d pages in %f seconds",
1098 state.pagecount, end - start);
1100 else {
1101 printd ("vmsg Processed %d pages in %f seconds",
1102 state.pagecount, end - start);
1104 state.trimanew = 0;
1105 if (trimf) {
1106 if (fclose (trimf)) {
1107 err (1, "fclose");
1112 static void layout (double zoom1)
1114 int pindex;
1115 fz_rect box;
1116 fz_matrix ctm;
1117 double zoom, w, maxw = 0.0;
1118 struct pagedim *p = state.pagedims;
1120 if (state.fitmodel == FitProportional) {
1121 int i;
1122 struct pagedim *p;
1124 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
1125 fz_rect rect;
1126 fz_matrix rm;
1127 double x0, x1;
1129 fz_rotate (&rm, p->rotate + state.rotate);
1130 rect = p->mediabox;
1131 fz_transform_rect (&rect, &rm);
1133 x0 = MIN (rect.x0, rect.x1);
1134 x1 = MAX (rect.x0, rect.x1);
1136 w = x1 - x0;
1137 maxw = MAX (w, maxw);
1141 p = state.pagedims;
1142 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
1143 fz_rect rect;
1144 fz_matrix tm, sm, rm;
1146 fz_rotate (&ctm, state.rotate);
1147 fz_rotate (&rm, p->rotate + state.rotate);
1148 box = p->mediabox;
1149 fz_transform_rect (&box, &rm);
1150 w = box.x1 - box.x0;
1151 zoom = state.w / w;
1152 switch (state.fitmodel) {
1153 case FitProportional:
1154 zoom *= w / maxw;
1155 break;
1156 case FitPage:
1158 double z1, h;
1159 h = box.y1 - box.y0;
1160 z1 = state.h / h;
1161 if (z1 < zoom) {
1162 zoom = z1;
1164 zoom *= zoom1;
1166 break;
1167 case FitWidth:
1168 break;
1169 default:
1170 ARSERT (0 && state.fitmodel);
1173 fz_scale (&p->zoomctm, zoom, zoom);
1174 fz_concat (&ctm, &p->zoomctm, &ctm);
1176 fz_rotate (&rm, p->rotate);
1177 p->pagebox = p->mediabox;
1178 fz_transform_rect (&p->pagebox, &rm);
1179 p->pagebox.x1 -= p->pagebox.x0;
1180 p->pagebox.y1 -= p->pagebox.y0;
1181 p->pagebox.x0 = 0;
1182 p->pagebox.y0 = 0;
1183 rect = p->pagebox;
1184 fz_transform_rect (&rect, &ctm);
1185 fz_round_rect (&p->bounds, &rect);
1187 p->left = (state.fitmodel == FitProportional)
1188 ? ((maxw - w) * zoom) / 2.0
1189 : 0;
1190 p->ctm = ctm;
1192 fz_translate (&tm, 0, -p->mediabox.y1);
1193 fz_scale (&sm, zoom, -zoom);
1194 fz_concat (&ctm, &tm, &sm);
1195 fz_concat (&p->lctm, &ctm, &rm);
1197 p->tctmready = 0;
1200 while (p-- != state.pagedims) {
1201 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1202 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1203 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1204 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1205 int w = x1 - x0;
1206 int h = y1 - y0;
1208 printd ("pdim %d %d %d %d", p->pageno, w, h, p->left);
1212 static
1213 struct anchor { int n; int x; int y; int w; int h; }
1214 desttoanchor (fz_link_dest *dest)
1216 int i;
1217 struct anchor a;
1218 struct pagedim *pdim = state.pagedims;
1220 a.n = -1;
1221 a.x = 0;
1222 a.y = 0;
1223 for (i = 0; i < state.pagedimcount; ++i) {
1224 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1225 break;
1226 pdim = &state.pagedims[i];
1228 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1229 fz_point p;
1230 if (dest->ld.gotor.flags & fz_link_flag_l_valid)
1231 p.x = dest->ld.gotor.lt.x;
1232 p.y = dest->ld.gotor.lt.y;
1233 fz_transform_point (&p, &pdim->lctm);
1234 a.y = p.y;
1235 a.x = p.x;
1237 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1238 double x0, x1, y0, y1;
1240 x0 = MIN (pdim->bounds.x0, pdim->bounds.x1);
1241 x1 = MAX (pdim->bounds.x0, pdim->bounds.x1);
1242 a.w = x1 - x0;
1243 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1244 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1245 a.h = y1 - y0;
1246 a.n = dest->ld.gotor.page;
1248 return a;
1251 static void recurse_outline (fz_outline *outline, int level)
1253 while (outline) {
1254 struct anchor a = desttoanchor (&outline->dest);
1256 if (a.n >= 0) {
1257 printd ("o %d %d %d %d %s",
1258 level, a.n, a.y, a.h, outline->title);
1260 if (outline->down) {
1261 recurse_outline (outline->down, level + 1);
1263 outline = outline->next;
1267 static void process_outline (void)
1269 fz_outline *outline;
1271 if (!state.needoutline) return;
1273 state.needoutline = 0;
1274 switch (state.type) {
1275 case DPDF:
1276 outline = pdf_load_outline (state.u.pdf);
1277 break;
1278 case DXPS:
1279 outline = xps_load_outline (state.u.xps);
1280 break;
1281 default:
1282 outline = NULL;
1283 break;
1285 if (outline) {
1286 recurse_outline (outline, 0);
1287 fz_free_outline (state.ctx, outline);
1291 static char *strofspan (fz_text_span *span)
1293 char *p;
1294 char utf8[10];
1295 fz_text_char *ch;
1296 size_t size = 0, cap = 80;
1298 p = malloc (cap + 1);
1299 if (!p) return NULL;
1301 for (ch = span->text; ch < span->text + span->len; ++ch) {
1302 int n = fz_runetochar (utf8, ch->c);
1303 if (size + n > cap) {
1304 cap *= 2;
1305 p = realloc (p, cap + 1);
1306 if (!p) return NULL;
1309 memcpy (p + size, utf8, n);
1310 size += n;
1312 p[size] = 0;
1313 return p;
1316 static int matchspan (regex_t *re, fz_text_page *page,
1317 fz_text_span *span, fz_matrix ctm,
1318 int stop, int pageno, double start)
1320 int ret;
1321 char *p;
1322 regmatch_t rm;
1323 int a, b, c;
1324 fz_rect sb, eb;
1325 fz_point p1, p2, p3, p4;
1327 p = strofspan (span);
1328 if (!p) return -1;
1330 ret = regexec (re, p, 1, &rm, 0);
1331 if (ret) {
1332 free (p);
1333 if (ret != REG_NOMATCH) {
1334 size_t size;
1335 char errbuf[80];
1336 size = regerror (ret, re, errbuf, sizeof (errbuf));
1337 printd ("msg regexec error `%.*s'",
1338 (int) size, errbuf);
1339 return -1;
1341 return 0;
1343 else {
1344 int l = span->len;
1346 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1347 c += fz_runelen (span->text[a].c);
1349 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1350 c += fz_runelen (span->text[b].c);
1353 if (fz_runelen (span->text[b].c) > 1) {
1354 b = MAX (0, b-1);
1357 fz_text_char_bbox (&sb, span, a);
1358 fz_text_char_bbox (&eb, span, b);
1360 p1.x = sb.x0;
1361 p1.y = sb.y0;
1362 p2.x = eb.x1;
1363 p2.y = sb.y0;
1364 p3.x = eb.x1;
1365 p3.y = eb.y1;
1366 p4.x = sb.x0;
1367 p4.y = eb.y1;
1369 if (!stop) {
1370 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1371 pageno, 1,
1372 p1.x, p1.y,
1373 p2.x, p2.y,
1374 p3.x, p3.y,
1375 p4.x, p4.y);
1377 printd ("progress 1 found at %d `%.*s' in %f sec",
1378 pageno + 1, (int) (rm.rm_eo - rm.rm_so), &p[rm.rm_so],
1379 now () - start);
1381 else {
1382 printd ("match %d %d %f %f %f %f %f %f %f %f",
1383 pageno, 2,
1384 p1.x, p1.y,
1385 p2.x, p2.y,
1386 p3.x, p3.y,
1387 p4.x, p4.y);
1389 free (p);
1390 return 1;
1394 static int compareblocks (const void *l, const void *r)
1396 fz_text_block const *ls = l;
1397 fz_text_block const* rs = r;
1398 return ls->bbox.y0 - rs->bbox.y0;
1401 /* wishful thinking function */
1402 static void search (regex_t *re, int pageno, int y, int forward)
1404 int i, j;
1405 fz_matrix ctm;
1406 fz_device *tdev;
1407 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1408 fz_text_page *text;
1409 fz_text_sheet *sheet;
1410 struct pagedim *pdim, *pdimprev;
1411 int stop = 0, niters = 0;
1412 double start, end;
1414 if (!(state.type == DPDF || state.type == DXPS))
1415 return;
1417 start = now ();
1418 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1419 if (niters++ == 5) {
1420 niters = 0;
1421 if (hasdata ()) {
1422 printd ("progress 1 attention requested aborting search at %d",
1423 pageno);
1424 stop = 1;
1426 else {
1427 printd ("progress %f searching in page %d",
1428 (double) (pageno + 1) / state.pagecount,
1429 pageno);
1432 pdimprev = NULL;
1433 for (i = 0; i < state.pagedimcount; ++i) {
1434 pdim = &state.pagedims[i];
1435 if (pdim->pageno == pageno) {
1436 goto found;
1438 if (pdim->pageno > pageno) {
1439 pdim = pdimprev;
1440 goto found;
1442 pdimprev = pdim;
1444 pdim = pdimprev;
1445 found:
1447 sheet = fz_new_text_sheet (state.ctx);
1448 text = fz_new_text_page (state.ctx, &fz_infinite_rect);
1449 tdev = fz_new_text_device (state.ctx, sheet, text);
1451 switch (state.type) {
1452 case DPDF:
1453 u.ptr = NULL;
1454 fz_try (state.ctx) {
1455 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1456 trimctm (u.pdfpage, pdim - state.pagedims);
1457 fz_concat (&ctm, &pdim->tctm, &pdim->zoomctm);
1458 pdf_run_page (state.u.pdf, u.pdfpage, tdev, &ctm, NULL);
1460 fz_catch (state.ctx) {
1461 fz_free_device (tdev);
1462 u.ptr = NULL;
1463 goto nextiter;
1465 break;
1467 case DXPS:
1468 u.xpspage = xps_load_page (state.u.xps, pageno);
1469 xps_run_page (state.u.xps, u.xpspage, tdev, &pdim->ctm, NULL);
1470 break;
1472 default:
1473 ARSERT (0 && state.type);
1476 qsort (text->blocks, text->len, sizeof (*text->blocks), compareblocks);
1477 fz_free_device (tdev);
1479 for (j = 0; j < text->len; ++j) {
1480 int k;
1481 fz_text_block *block;
1483 block = &text->blocks[forward ? j : text->len - 1 - j];
1485 for (k = 0; k < block->len; ++k) {
1486 int spannum;
1487 fz_text_line *line;
1489 if (forward) {
1490 line = &block->lines[k];
1491 if (line->bbox.y0 < y + 1) continue;
1493 else {
1494 line = &block->lines[block->len - 1 - k];
1495 if (line->bbox.y0 > y - 1) continue;
1498 for (spannum = 0; spannum < line->len; ++spannum) {
1499 fz_text_span *span = line->spans[spannum];
1501 switch (matchspan (re, text, span, ctm,
1502 stop, pageno, start)) {
1503 case 0: break;
1504 case 1: stop = 1; break;
1505 case -1: stop = 1; goto endloop;
1510 nextiter:
1511 if (forward) {
1512 pageno += 1;
1513 y = 0;
1515 else {
1516 pageno -= 1;
1517 y = INT_MAX;
1519 endloop:
1520 fz_free_text_page (state.ctx, text);
1521 fz_free_text_sheet (state.ctx, sheet);
1522 if (u.ptr) {
1523 state.freepage (u.ptr);
1526 end = now ();
1527 if (!stop) {
1528 printd ("progress 1 no matches %f sec", end - start);
1530 printd ("clearrects");
1533 static void set_tex_params (int colorspace)
1535 union {
1536 unsigned char b;
1537 unsigned int s;
1538 } endianness = {1};
1540 switch (colorspace) {
1541 case 0:
1542 state.texiform = GL_RGBA8;
1543 state.texform = GL_RGBA;
1544 state.texty = GL_UNSIGNED_BYTE;
1545 state.colorspace = fz_device_rgb;
1546 break;
1547 case 1:
1548 state.texiform = GL_RGBA8;
1549 state.texform = GL_BGRA;
1550 state.texty = endianness.s > 1
1551 ? GL_UNSIGNED_INT_8_8_8_8
1552 : GL_UNSIGNED_INT_8_8_8_8_REV;
1553 state.colorspace = fz_device_bgr;
1554 break;
1555 case 2:
1556 state.texiform = GL_LUMINANCE_ALPHA;
1557 state.texform = GL_LUMINANCE_ALPHA;
1558 state.texty = GL_UNSIGNED_BYTE;
1559 state.colorspace = fz_device_gray;
1560 break;
1561 default:
1562 errx (1, "invalid colorspce %d", colorspace);
1566 static void realloctexts (int texcount)
1568 size_t size;
1570 if (texcount == state.texcount) return;
1572 if (texcount < state.texcount) {
1573 glDeleteTextures (state.texcount - texcount,
1574 state.texids + texcount);
1577 size = texcount * sizeof (*state.texids);
1578 state.texids = realloc (state.texids, size);
1579 if (!state.texids) {
1580 err (1, "realloc texids %" FMT_s, size);
1583 size = texcount * sizeof (*state.texowners);
1584 state.texowners = realloc (state.texowners, size);
1585 if (!state.texowners) {
1586 err (1, "realloc texowners %" FMT_s, size);
1588 if (texcount > state.texcount) {
1589 int i;
1591 glGenTextures (texcount - state.texcount,
1592 state.texids + state.texcount);
1593 for (i = state.texcount; i < texcount; ++i) {
1594 state.texowners[i].w = -1;
1595 state.texowners[i].slice = NULL;
1598 state.texcount = texcount;
1599 state.texindex = 0;
1602 static char *mbtoutf8 (char *s)
1604 char *p, *r;
1605 wchar_t *tmp;
1606 size_t i, ret, len;
1608 len = mbstowcs (NULL, s, strlen (s));
1609 if (len == 0) {
1610 return s;
1612 else {
1613 if (len == (size_t) -1) {
1614 return s;
1618 tmp = malloc (len * sizeof (wchar_t));
1619 if (!tmp) {
1620 return s;
1623 ret = mbstowcs (tmp, s, len);
1624 if (ret == (size_t) -1) {
1625 free (tmp);
1626 return s;
1629 len = 0;
1630 for (i = 0; i < ret; ++i) {
1631 len += fz_runelen (tmp[i]);
1634 p = r = malloc (len + 1);
1635 if (!r) {
1636 free (tmp);
1637 return s;
1640 for (i = 0; i < ret; ++i) {
1641 p += fz_runetochar (p, tmp[i]);
1643 *p = 0;
1644 free (tmp);
1645 return r;
1648 CAMLprim value ml_mbtoutf8 (value s_v)
1650 CAMLparam1 (s_v);
1651 CAMLlocal1 (ret_v);
1652 char *s, *r;
1654 s = String_val (s_v);
1655 r = mbtoutf8 (s);
1656 if (r == s) {
1657 ret_v = s_v;
1659 else {
1660 ret_v = caml_copy_string (r);
1661 free (r);
1663 CAMLreturn (ret_v);
1666 static void * mainloop (void *unused)
1668 char *p = NULL;
1669 int len, ret, oldlen = 0;
1671 for (;;) {
1672 len = readlen ();
1673 if (len == 0) {
1674 errx (1, "readlen returned 0");
1677 if (oldlen < len + 1) {
1678 p = realloc (p, len + 1);
1679 if (!p) {
1680 err (1, "realloc %d failed", len + 1);
1682 oldlen = len + 1;
1684 readdata (p, len);
1685 p[len] = 0;
1687 if (!strncmp ("open", p, 4)) {
1688 size_t filenamelen;
1689 int wthack, off;
1690 char *password;
1691 char *filename;
1692 char *utf8filename;
1694 ret = sscanf (p + 5, " %d %n", &wthack, &off);
1695 if (ret != 1) {
1696 errx (1, "malformed open `%.*s' ret=%d", len, p, ret);
1699 filename = p + 5 + off;
1700 filenamelen = strlen (filename);
1701 password = filename + filenamelen + 1;
1703 lock ("open");
1704 openxref (filename, password);
1705 pdfinfo ();
1706 initpdims ();
1707 unlock ("open");
1709 if (!wthack) {
1710 utf8filename = mbtoutf8 (filename);
1711 printd ("msg Opened %s (press h/F1 to get help)", utf8filename);
1712 if (utf8filename != filename) {
1713 free (utf8filename);
1716 state.needoutline = 1;
1718 else if (!strncmp ("cs", p, 2)) {
1719 int i, colorspace;
1721 ret = sscanf (p + 2, " %d", &colorspace);
1722 if (ret != 1) {
1723 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
1725 lock ("cs");
1726 set_tex_params (colorspace);
1727 for (i = 0; i < state.texcount; ++i) {
1728 state.texowners[i].w = -1;
1729 state.texowners[i].slice = NULL;
1731 unlock ("cs");
1733 else if (!strncmp ("freepage", p, 8)) {
1734 void *ptr;
1736 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1737 if (ret != 1) {
1738 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
1740 freepage (ptr);
1742 else if (!strncmp ("freetile", p, 8)) {
1743 void *ptr;
1745 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1746 if (ret != 1) {
1747 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
1749 freetile (ptr);
1751 else if (!strncmp ("search", p, 6)) {
1752 int icase, pageno, y, ret, len2, forward;
1753 char *pattern;
1754 regex_t re;
1756 ret = sscanf (p + 6, " %d %d %d %d,%n",
1757 &icase, &pageno, &y, &forward, &len2);
1758 if (ret != 4) {
1759 errx (1, "malformed search `%s' ret=%d", p, ret);
1762 pattern = p + 6 + len2;
1763 ret = regcomp (&re, pattern,
1764 REG_EXTENDED | (icase ? REG_ICASE : 0));
1765 if (ret) {
1766 char errbuf[80];
1767 size_t size;
1769 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1770 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
1772 else {
1773 search (&re, pageno, y, forward);
1774 regfree (&re);
1777 else if (!strncmp ("geometry", p, 8)) {
1778 float zoom;
1779 int w, h, fitmodel;
1781 printd ("clear");
1782 ret = sscanf (p + 8, " %d %d %f %d", &w, &h, &zoom, &fitmodel);
1783 if (ret != 4) {
1784 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1787 lock ("geometry");
1788 state.h = h;
1789 if (w != state.w) {
1790 int i;
1791 state.w = w;
1792 for (i = 0; i < state.texcount; ++i) {
1793 state.texowners[i].slice = NULL;
1796 state.fitmodel = fitmodel;
1797 layout (zoom);
1798 process_outline ();
1800 state.gen++;
1801 unlock ("geometry");
1802 printd ("continue %d", state.pagecount);
1804 else if (!strncmp ("reqlayout", p, 9)) {
1805 float zoom;
1806 char *nameddest;
1807 int rotate, fitmodel, off;
1809 printd ("clear");
1810 ret = sscanf (p + 9, " %d %d %f %n",
1811 &rotate, &fitmodel, &zoom, &off);
1812 if (ret != 3) {
1813 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
1815 lock ("reqlayout");
1816 if (state.rotate != rotate || state.fitmodel != fitmodel) {
1817 state.gen += 1;
1819 state.rotate = rotate;
1820 state.fitmodel = fitmodel;
1821 layout (zoom);
1822 process_outline ();
1824 nameddest = p + 9 + off;
1825 if (state.type == DPDF && nameddest && *nameddest) {
1826 struct anchor a;
1827 fz_link_dest dest;
1828 pdf_obj *needle, *obj;
1830 needle = pdf_new_string (state.ctx, nameddest,
1831 strlen (nameddest));
1832 obj = pdf_lookup_dest (state.u.pdf, needle);
1833 if (obj) {
1834 dest = pdf_parse_link_dest (state.u.pdf, obj);
1836 a = desttoanchor (&dest);
1837 if (a.n >= 0) {
1838 printd ("a %d %d %d", a.n, a.x, a.y);
1840 else {
1841 printd ("emsg failed to parse destination `%s'\n",
1842 nameddest);
1845 else {
1846 printd ("emsg destination `%s' not found\n",
1847 nameddest);
1849 pdf_drop_obj (needle);
1852 state.gen++;
1853 unlock ("reqlayout");
1854 printd ("continue %d", state.pagecount);
1856 else if (!strncmp ("page", p, 4)) {
1857 double a, b;
1858 struct page *page;
1859 int pageno, pindex, ret;
1861 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
1862 if (ret != 2) {
1863 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1866 lock ("page");
1867 a = now ();
1868 page = loadpage (pageno, pindex);
1869 b = now ();
1870 unlock ("page");
1872 printd ("page %" FMT_ptr " %f", FMT_ptr_cast2 (page), b - a);
1874 else if (!strncmp ("tile", p, 4)) {
1875 int x, y, w, h, ret;
1876 struct page *page;
1877 struct tile *tile;
1878 double a, b;
1879 void *data;
1881 ret = sscanf (p + 4, " %" FMT_ptr " %d %d %d %d %" FMT_ptr,
1882 FMT_ptr_cast (&page), &x, &y, &w, &h, &data);
1883 if (ret != 6) {
1884 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
1887 lock ("tile");
1888 a = now ();
1889 tile = rendertile (page, x, y, w, h, data);
1890 b = now ();
1891 unlock ("tile");
1893 printd ("tile %d %d %" FMT_ptr " %u %f",
1894 x, y,
1895 FMT_ptr_cast2 (tile),
1896 tile->w * tile->h * tile->pixmap->n,
1897 b - a);
1899 else if (!strncmp ("settrim", p, 7)) {
1900 float zoom;
1901 fz_irect fuzz;
1902 int trimmargins;
1904 ret = sscanf (p + 7, " %d %d %d %d %d %f",
1905 &trimmargins, &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1,
1906 &zoom);
1907 if (ret != 6) {
1908 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
1910 printd ("clear");
1911 lock ("settrim");
1912 state.trimmargins = trimmargins;
1913 state.needoutline = 1;
1914 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
1915 state.trimanew = 1;
1916 state.trimfuzz = fuzz;
1918 state.pagedimcount = 0;
1919 free (state.pagedims);
1920 state.pagedims = NULL;
1921 initpdims ();
1922 layout (zoom);
1923 process_outline ();
1924 unlock ("settrim");
1925 printd ("continue %d", state.pagecount);
1927 else if (!strncmp ("sliceh", p, 6)) {
1928 int h;
1930 ret = sscanf (p + 6, " %d", &h);
1931 if (ret != 1) {
1932 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
1934 if (h != state.sliceheight) {
1935 int i;
1937 state.sliceheight = h;
1938 for (i = 0; i < state.texcount; ++i) {
1939 state.texowners[i].w = -1;
1940 state.texowners[i].h = -1;
1941 state.texowners[i].slice = NULL;
1945 else if (!strncmp ("interrupt", p, 9)) {
1946 printd ("vmsg interrupted");
1948 else {
1949 errx (1, "unknown command %.*s", len, p);
1952 return 0;
1955 CAMLprim value ml_realloctexts (value texcount_v)
1957 CAMLparam1 (texcount_v);
1958 int ok;
1960 if (trylock ("ml_realloctexts")) {
1961 ok = 0;
1962 goto done;
1964 realloctexts (Int_val (texcount_v));
1965 ok = 1;
1966 unlock ("ml_realloctexts");
1968 done:
1969 CAMLreturn (Val_bool (ok));
1972 static void showsel (struct page *page, int ox, int oy)
1974 int seen = 0;
1975 fz_irect bbox;
1976 fz_rect rect;
1977 fz_text_line *line;
1978 fz_text_block *block;
1979 struct mark first, last;
1981 first = page->fmark;
1982 last = page->lmark;
1984 if (!first.span || !last.span) return;
1986 glEnable (GL_BLEND);
1987 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1988 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1990 ox += state.pagedims[page->pdimno].bounds.x0;
1991 oy += state.pagedims[page->pdimno].bounds.y0;
1992 for (block = page->text->blocks;
1993 block < page->text->blocks + page->text->len;
1994 ++block) {
1995 for (line = block->lines;
1996 line < block->lines + block->len;
1997 ++line) {
1998 int spannum;
1999 rect = fz_empty_rect;
2000 for (spannum = 0; spannum < line->len; ++spannum) {
2001 int i, j, k;
2002 fz_text_span *span = line->spans[spannum];
2004 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
2006 j = 0;
2007 k = span->len - 1;
2009 if (span == page->fmark.span && span == page->lmark.span) {
2010 seen = 1;
2011 j = MIN (first.i, last.i);
2012 k = MAX (first.i, last.i);
2014 else if (span == first.span) {
2015 seen = 1;
2016 j = first.i;
2018 else if (span == last.span) {
2019 seen = 1;
2020 k = last.i;
2023 if (seen) {
2024 for (i = j; i <= k; ++i) {
2025 fz_rect bbox;
2026 fz_union_rect (&rect,
2027 fz_text_char_bbox (&bbox, span, i));
2029 fz_round_rect (&bbox, &rect);
2030 lprintf ("%d %d %d %d oy=%d ox=%d\n",
2031 bbox.x0,
2032 bbox.y0,
2033 bbox.x1,
2034 bbox.y1,
2035 oy, ox);
2037 glRecti (bbox.x0 + ox, bbox.y0 + oy,
2038 bbox.x1 + ox, bbox.y1 + oy);
2039 if (span == last.span) {
2040 goto done;
2046 done:
2047 glDisable (GL_BLEND);
2050 #include "glfont.c"
2052 static void highlightlinks (struct page *page, int xoff, int yoff)
2054 fz_matrix ctm, tm, pm;
2055 fz_link *link, *links;
2057 switch (page->type) {
2058 case DPDF:
2059 links = page->u.pdfpage->links;
2060 break;
2062 case DXPS:
2063 links = page->u.xpspage->links;
2064 break;
2066 default:
2067 return;
2070 glEnable (GL_TEXTURE_1D);
2071 glEnable (GL_BLEND);
2072 glBindTexture (GL_TEXTURE_1D, state.stid);
2074 xoff -= state.pagedims[page->pdimno].bounds.x0;
2075 yoff -= state.pagedims[page->pdimno].bounds.y0;
2076 fz_translate (&tm, xoff, yoff);
2077 pm = pagectm (page);
2078 fz_concat (&ctm, &pm, &tm);
2080 glBegin (GL_LINES);
2081 for (link = links; link; link = link->next) {
2082 fz_point p1, p2, p3, p4;
2084 p1.x = link->rect.x0;
2085 p1.y = link->rect.y0;
2087 p2.x = link->rect.x1;
2088 p2.y = link->rect.y0;
2090 p3.x = link->rect.x1;
2091 p3.y = link->rect.y1;
2093 p4.x = link->rect.x0;
2094 p4.y = link->rect.y1;
2096 fz_transform_point (&p1, &ctm);
2097 fz_transform_point (&p2, &ctm);
2098 fz_transform_point (&p3, &ctm);
2099 fz_transform_point (&p4, &ctm);
2101 switch (link->dest.kind) {
2102 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
2103 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
2104 case FZ_LINK_LAUNCH: glColor3ub (0, 255, 0); break;
2105 default: glColor3ub (0, 0, 0); break;
2109 float w, h, s, t;
2111 w = p2.x - p1.x;
2112 h = p2.y - p1.y;
2113 t = sqrtf (w*w + h*h) * .25f;
2115 w = p3.x - p2.x;
2116 h = p3.y - p2.y;
2117 s = sqrtf (w*w + h*h) * .25f;
2119 glTexCoord1i (0);
2120 glVertex2f (p1.x, p1.y);
2121 glTexCoord1f (t);
2122 glVertex2f (p2.x, p2.y);
2124 glTexCoord1i (0);
2125 glVertex2f (p2.x, p2.y);
2126 glTexCoord1f (s);
2127 glVertex2f (p3.x, p3.y);
2129 glTexCoord1i (0);
2130 glVertex2f (p3.x, p3.y);
2131 glTexCoord1f (t);
2132 glVertex2f (p4.x, p4.y);
2134 glTexCoord1i (0);
2135 glVertex2f (p4.x, p4.y);
2136 glTexCoord1f (s);
2137 glVertex2f (p1.x, p1.y);
2140 glEnd ();
2142 glDisable (GL_BLEND);
2143 glDisable (GL_TEXTURE_1D);
2146 static int compareslinks (const void *l, const void *r)
2148 struct slink const *ls = l;
2149 struct slink const *rs = r;
2150 if (ls->bbox.y0 == rs->bbox.y0) {
2151 return rs->bbox.x0 - rs->bbox.x0;
2153 return ls->bbox.y0 - rs->bbox.y0;
2156 static void droptext (struct page *page)
2158 if (page->text) {
2159 fz_free_text_page (state.ctx, page->text);
2160 page->fmark.i = -1;
2161 page->lmark.i = -1;
2162 page->fmark.span = NULL;
2163 page->lmark.span = NULL;
2164 page->text = NULL;
2166 if (page->sheet) {
2167 fz_free_text_sheet (state.ctx, page->sheet);
2171 static void dropslinks (struct page *page)
2173 if (page->slinks) {
2174 free (page->slinks);
2175 page->slinks = NULL;
2176 page->slinkcount = 0;
2180 static void ensureslinks (struct page *page)
2182 fz_matrix ctm;
2183 int i, count = 0;
2184 size_t slinksize = sizeof (*page->slinks);
2185 fz_link *link, *links;
2187 if (state.gen != page->sgen) {
2188 dropslinks (page);
2189 page->sgen = state.gen;
2191 if (page->slinks) return;
2193 switch (page->type) {
2194 case DPDF:
2195 links = page->u.pdfpage->links;
2196 trimctm (page->u.pdfpage, page->pdimno);
2197 fz_concat (&ctm,
2198 &state.pagedims[page->pdimno].tctm,
2199 &state.pagedims[page->pdimno].ctm);
2200 break;
2202 case DXPS:
2203 links = page->u.xpspage->links;
2204 ctm = state.pagedims[page->pdimno].ctm;
2205 break;
2207 default:
2208 return;
2211 for (link = links; link; link = link->next) {
2212 count++;
2214 if (count > 0) {
2215 page->slinkcount = count;
2216 page->slinks = calloc (count, slinksize);
2217 if (!page->slinks) {
2218 err (1, "realloc slinks %d", count);
2221 for (i = 0, link = links; link; ++i, link = link->next) {
2222 fz_rect rect;
2224 rect = link->rect;
2225 fz_transform_rect (&rect, &ctm);
2226 page->slinks[i].link = link;
2227 fz_round_rect (&page->slinks[i].bbox, &rect);
2229 qsort (page->slinks, count, slinksize, compareslinks);
2233 /* slightly tweaked fmt_ulong by D.J. Bernstein */
2234 static void fmt_linkn (char *s, unsigned int u)
2236 unsigned int len; unsigned int q;
2237 int zma = 'z' - 'a' + 1;
2238 len = 1; q = u;
2239 while (q > zma - 1) { ++len; q /= zma; }
2240 if (s) {
2241 s += len;
2242 do { *--s = 'a' + (u % zma) - (u < zma && len > 1); u /= zma; } while(u);
2243 /* handles u == 0 */
2245 s[len] = 0;
2248 static void highlightslinks (struct page *page, int xoff, int yoff,
2249 int noff, char *targ, int tlen, int hfsize)
2251 int i;
2252 char buf[40];
2253 struct slink *slink;
2254 double x0, y0, x1, y1, w;
2256 ensureslinks (page);
2257 glColor3ub (0xc3, 0xb0, 0x91);
2258 for (i = 0; i < page->slinkcount; ++i) {
2259 fmt_linkn (buf, i + noff);
2260 if (!tlen || !strncmp (targ, buf, tlen)) {
2261 slink = &page->slinks[i];
2263 x0 = slink->bbox.x0 + xoff - 5;
2264 y1 = slink->bbox.y0 + yoff - 5;
2265 y0 = y1 + 10 + hfsize;
2266 w = measure_string (state.face, hfsize, buf);
2267 x1 = x0 + w + 10;
2268 glRectd (x0, y0, x1, y1);
2272 glEnable (GL_BLEND);
2273 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2274 glEnable (GL_TEXTURE_2D);
2275 glColor3ub (0, 0, 0);
2276 for (i = 0; i < page->slinkcount; ++i) {
2277 fmt_linkn (buf, i + noff);
2278 if (!tlen || !strncmp (targ, buf, tlen)) {
2279 slink = &page->slinks[i];
2281 x0 = slink->bbox.x0 + xoff;
2282 y0 = slink->bbox.y0 + yoff + hfsize;
2283 draw_string (state.face, hfsize, x0, y0, buf);
2286 glDisable (GL_TEXTURE_2D);
2287 glDisable (GL_BLEND);
2291 static void uploadslice (struct tile *tile, struct slice *slice)
2293 int offset;
2294 struct slice *slice1;
2295 unsigned char *texdata;
2297 offset = 0;
2298 for (slice1 = tile->slices; slice != slice1; slice1++) {
2299 offset += slice1->h * tile->w * tile->pixmap->n;
2301 if (slice->texindex != -1 && slice->texindex < state.texcount
2302 && state.texowners[slice->texindex].slice == slice) {
2303 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
2305 else {
2306 int subimage = 0;
2307 int texindex = state.texindex++ % state.texcount;
2309 if (state.texowners[texindex].w == tile->w) {
2310 if (state.texowners[texindex].h >= slice->h) {
2311 subimage = 1;
2313 else {
2314 state.texowners[texindex].h = slice->h;
2317 else {
2318 state.texowners[texindex].h = slice->h;
2321 state.texowners[texindex].w = tile->w;
2322 state.texowners[texindex].slice = slice;
2323 slice->texindex = texindex;
2325 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
2326 if (tile->pbo) {
2327 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
2328 texdata = 0;
2330 else {
2331 texdata = tile->pixmap->samples;
2333 if (subimage) {
2334 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
2338 tile->w,
2339 slice->h,
2340 state.texform,
2341 state.texty,
2342 texdata+offset
2345 else {
2346 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
2348 state.texiform,
2349 tile->w,
2350 slice->h,
2352 state.texform,
2353 state.texty,
2354 texdata+offset
2357 if (tile->pbo) {
2358 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
2363 CAMLprim value ml_drawtile (value args_v, value ptr_v)
2365 CAMLparam2 (args_v, ptr_v);
2366 int dispx = Int_val (Field (args_v, 0));
2367 int dispy = Int_val (Field (args_v, 1));
2368 int dispw = Int_val (Field (args_v, 2));
2369 int disph = Int_val (Field (args_v, 3));
2370 int tilex = Int_val (Field (args_v, 4));
2371 int tiley = Int_val (Field (args_v, 5));
2372 char *s = String_val (ptr_v);
2373 struct tile *tile = parse_pointer ("ml_drawtile", s);
2375 glEnable (GL_TEXTURE_RECTANGLE_ARB);
2377 int slicey, firstslice;
2378 struct slice *slice;
2380 firstslice = tiley / tile->sliceheight;
2381 slice = &tile->slices[firstslice];
2382 slicey = tiley % tile->sliceheight;
2384 while (disph > 0) {
2385 int dh;
2387 dh = slice->h - slicey;
2388 dh = MIN (disph, dh);
2389 uploadslice (tile, slice);
2391 glBegin (GL_QUADS);
2393 glTexCoord2i (tilex, slicey);
2394 glVertex2i (dispx, dispy);
2396 glTexCoord2i (tilex+dispw, slicey);
2397 glVertex2i (dispx+dispw, dispy);
2399 glTexCoord2i (tilex+dispw, slicey+dh);
2400 glVertex2i (dispx+dispw, dispy+dh);
2402 glTexCoord2i (tilex, slicey+dh);
2403 glVertex2i (dispx, dispy+dh);
2405 glEnd ();
2407 dispy += dh;
2408 disph -= dh;
2409 slice++;
2410 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
2411 slicey = 0;
2414 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2415 CAMLreturn (Val_unit);
2418 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2419 value xoff_v, value yoff_v,
2420 value li_v)
2422 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, li_v);
2423 int xoff = Int_val (xoff_v);
2424 int yoff = Int_val (yoff_v);
2425 int noff = Int_val (Field (li_v, 0));
2426 char *targ = String_val (Field (li_v, 1));
2427 int tlen = caml_string_length (Field (li_v, 1));
2428 int hfsize = Int_val (Field (li_v, 2));
2429 char *s = String_val (ptr_v);
2430 int hlmask = Int_val (hlinks_v);
2431 struct page *page = parse_pointer ("ml_postprocess", s);
2433 if (!page->u.ptr) {
2434 /* deal with loadpage failed pages */
2435 goto done;
2438 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2439 if (trylock ("ml_postprocess")) {
2440 noff = 0;
2441 goto done;
2443 if (hlmask & 2) {
2444 highlightslinks (page, xoff, yoff, noff, targ, tlen, hfsize);
2445 noff = page->slinkcount;
2447 showsel (page, xoff, yoff);
2448 unlock ("ml_postprocess");
2450 done:
2451 CAMLreturn (Val_int (noff));
2454 static fz_link *getlink (struct page *page, int x, int y)
2456 fz_point p;
2457 fz_matrix ctm;
2458 const fz_matrix *tctm;
2459 fz_link *link, *links;
2461 switch (page->type) {
2462 case DPDF:
2463 trimctm (page->u.pdfpage, page->pdimno);
2464 tctm = &state.pagedims[page->pdimno].tctm;
2465 links = page->u.pdfpage->links;
2466 break;
2468 case DXPS:
2469 tctm = &fz_identity;
2470 links = page->u.xpspage->links;
2471 break;
2473 default:
2474 return NULL;
2476 p.x = x;
2477 p.y = y;
2479 fz_concat (&ctm, tctm, &state.pagedims[page->pdimno].ctm);
2480 fz_invert_matrix (&ctm, &ctm);
2481 fz_transform_point (&p, &ctm);
2483 for (link = links; link; link = link->next) {
2484 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2485 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2486 return link;
2490 return NULL;
2493 static void ensuretext (struct page *page)
2495 if (state.gen != page->tgen) {
2496 droptext (page);
2497 page->tgen = state.gen;
2499 if (!page->text) {
2500 fz_matrix ctm;
2501 fz_device *tdev;
2503 page->text = fz_new_text_page (state.ctx, &fz_infinite_rect);
2504 page->sheet = fz_new_text_sheet (state.ctx);
2505 tdev = fz_new_text_device (state.ctx, page->sheet, page->text);
2506 ctm = pagectm (page);
2507 fz_run_display_list (page->dlist, tdev, &ctm, &fz_infinite_rect, NULL);
2508 qsort (page->text->blocks, page->text->len,
2509 sizeof (*page->text->blocks), compareblocks);
2510 fz_free_device (tdev);
2514 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2516 CAMLparam2 (start_page_v, dir_v);
2517 CAMLlocal1 (ret_v);
2518 int i, dir = Int_val (dir_v);
2519 int start_page = Int_val (start_page_v);
2520 int end_page = dir > 0 ? state.pagecount : -1;
2522 ret_v = Val_int (0);
2523 if (!(state.type == DPDF || state.type == DXPS)) {
2524 goto done;
2527 lock ("ml_findpage_with_links");
2528 for (i = start_page + dir; i != end_page; i += dir) {
2529 int found;
2531 switch (state.type) {
2532 case DPDF:
2534 pdf_page *page = NULL;
2536 fz_try (state.ctx) {
2537 page = pdf_load_page (state.u.pdf, i);
2538 found = !!page->links;
2540 fz_catch (state.ctx) {
2541 found = 0;
2543 if (page) {
2544 freepdfpage (page);
2547 break;
2548 case DXPS:
2550 xps_page *page = xps_load_page (state.u.xps, i);
2551 found = !!page->links;
2552 freexpspage (page);
2554 break;
2556 default:
2557 ARSERT ("invalid document type");
2560 if (found) {
2561 ret_v = caml_alloc_small (1, 1);
2562 Field (ret_v, 0) = Val_int (i);
2563 goto unlock;
2566 unlock:
2567 unlock ("ml_findpage_with_links");
2569 done:
2570 CAMLreturn (ret_v);
2573 enum { dir_first, dir_last};
2574 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
2576 CAMLprim value ml_findlink (value ptr_v, value dir_v)
2578 CAMLparam2 (ptr_v, dir_v);
2579 CAMLlocal2 (ret_v, pos_v);
2580 struct page *page;
2581 int dirtag, i, slinkindex;
2582 struct slink *found = NULL ,*slink;
2583 char *s = String_val (ptr_v);
2585 page = parse_pointer ("ml_findlink", s);
2586 ret_v = Val_int (0);
2587 if (trylock ("ml_findlink")) {
2588 goto done;
2591 ensureslinks (page);
2593 if (Is_block (dir_v)) {
2594 dirtag = Tag_val (dir_v);
2595 switch (dirtag) {
2596 case dir_first_visible:
2598 int x0, y0, dir, first_index, last_index;
2600 pos_v = Field (dir_v, 0);
2601 x0 = Int_val (Field (pos_v, 0));
2602 y0 = Int_val (Field (pos_v, 1));
2603 dir = Int_val (Field (pos_v, 2));
2605 if (dir >= 0) {
2606 dir = 1;
2607 first_index = 0;
2608 last_index = page->slinkcount;
2610 else {
2611 first_index = page->slinkcount - 1;
2612 last_index = -1;
2615 for (i = first_index; i != last_index; i += dir) {
2616 slink = &page->slinks[i];
2617 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
2618 found = slink;
2619 break;
2623 break;
2625 case dir_left:
2626 slinkindex = Int_val (Field (dir_v, 0));
2627 found = &page->slinks[slinkindex];
2628 for (i = slinkindex - 1; i >= 0; --i) {
2629 slink = &page->slinks[i];
2630 if (slink->bbox.x0 < found->bbox.x0) {
2631 found = slink;
2632 break;
2635 break;
2637 case dir_right:
2638 slinkindex = Int_val (Field (dir_v, 0));
2639 found = &page->slinks[slinkindex];
2640 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2641 slink = &page->slinks[i];
2642 if (slink->bbox.x0 > found->bbox.x0) {
2643 found = slink;
2644 break;
2647 break;
2649 case dir_down:
2650 slinkindex = Int_val (Field (dir_v, 0));
2651 found = &page->slinks[slinkindex];
2652 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2653 slink = &page->slinks[i];
2654 if (slink->bbox.y0 >= found->bbox.y0) {
2655 found = slink;
2656 break;
2659 break;
2661 case dir_up:
2662 slinkindex = Int_val (Field (dir_v, 0));
2663 found = &page->slinks[slinkindex];
2664 for (i = slinkindex - 1; i >= 0; --i) {
2665 slink = &page->slinks[i];
2666 if (slink->bbox.y0 <= found->bbox.y0) {
2667 found = slink;
2668 break;
2671 break;
2674 else {
2675 dirtag = Int_val (dir_v);
2676 switch (dirtag) {
2677 case dir_first:
2678 found = page->slinks;
2679 break;
2681 case dir_last:
2682 if (page->slinks) {
2683 found = page->slinks + (page->slinkcount - 1);
2685 break;
2688 if (found) {
2689 ret_v = caml_alloc_small (2, 1);
2690 Field (ret_v, 0) = Val_int (found - page->slinks);
2693 unlock ("ml_findlink");
2694 done:
2695 CAMLreturn (ret_v);
2698 enum { uuri, ugoto, utext, uunexpected, ulaunch, unamed, uremote };
2700 #define LINKTOVAL \
2702 int pageno; \
2704 switch (link->dest.kind) { \
2705 case FZ_LINK_GOTO: \
2707 fz_point p; \
2709 pageno = link->dest.ld.gotor.page; \
2710 p.x = 0; \
2711 p.y = 0; \
2713 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
2714 p.y = link->dest.ld.gotor.lt.y; \
2715 fz_transform_point (&p, &pdim->lctm); \
2717 tup_v = caml_alloc_tuple (2); \
2718 ret_v = caml_alloc_small (1, ugoto); \
2719 Field (tup_v, 0) = Val_int (pageno); \
2720 Field (tup_v, 1) = Val_int (p.y); \
2721 Field (ret_v, 0) = tup_v; \
2723 break; \
2725 case FZ_LINK_URI: \
2726 str_v = caml_copy_string (link->dest.ld.uri.uri); \
2727 ret_v = caml_alloc_small (1, uuri); \
2728 Field (ret_v, 0) = str_v; \
2729 break; \
2731 case FZ_LINK_LAUNCH: \
2732 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
2733 ret_v = caml_alloc_small (1, ulaunch); \
2734 Field (ret_v, 0) = str_v; \
2735 break; \
2737 case FZ_LINK_NAMED: \
2738 str_v = caml_copy_string (link->dest.ld.named.named); \
2739 ret_v = caml_alloc_small (1, unamed); \
2740 Field (ret_v, 0) = str_v; \
2741 break; \
2743 case FZ_LINK_GOTOR: \
2744 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
2745 pageno = link->dest.ld.gotor.page; \
2746 tup_v = caml_alloc_tuple (2); \
2747 ret_v = caml_alloc_small (1, uremote); \
2748 Field (tup_v, 0) = str_v; \
2749 Field (tup_v, 1) = Val_int (pageno); \
2750 Field (ret_v, 0) = tup_v; \
2751 break; \
2753 default: \
2755 char buf[80]; \
2757 snprintf (buf, sizeof (buf), \
2758 "unhandled link kind %d", link->dest.kind); \
2759 str_v = caml_copy_string (buf); \
2760 ret_v = caml_alloc_small (1, uunexpected); \
2761 Field (ret_v, 0) = str_v; \
2763 break; \
2767 CAMLprim value ml_getlink (value ptr_v, value n_v)
2769 CAMLparam2 (ptr_v, n_v);
2770 CAMLlocal3 (ret_v, tup_v, str_v);
2771 fz_link *link;
2772 struct page *page;
2773 struct pagedim *pdim;
2774 char *s = String_val (ptr_v);
2776 ret_v = Val_int (0);
2777 if (trylock ("ml_getlink")) {
2778 goto done;
2781 page = parse_pointer ("ml_getlink", s);
2782 ensureslinks (page);
2783 pdim = &state.pagedims[page->pdimno];
2784 link = page->slinks[Int_val (n_v)].link;
2785 LINKTOVAL;
2787 unlock ("ml_getlink");
2788 done:
2789 CAMLreturn (ret_v);
2792 CAMLprim value ml_getlinkcount (value ptr_v)
2794 CAMLparam1 (ptr_v);
2795 struct page *page;
2796 char *s = String_val (ptr_v);
2798 page = parse_pointer ("ml_getlinkcount", s);
2799 CAMLreturn (Val_int (page->slinkcount));
2802 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
2804 CAMLparam2 (ptr_v, n_v);
2805 CAMLlocal1 (ret_v);
2806 struct page *page;
2807 struct slink *slink;
2808 char *s = String_val (ptr_v);
2810 page = parse_pointer ("ml_getlinkrect", s);
2811 ret_v = caml_alloc_tuple (4);
2812 if (trylock ("ml_getlinkrect")) {
2813 Field (ret_v, 0) = Val_int (0);
2814 Field (ret_v, 1) = Val_int (0);
2815 Field (ret_v, 2) = Val_int (0);
2816 Field (ret_v, 3) = Val_int (0);
2817 goto done;
2819 ensureslinks (page);
2821 slink = &page->slinks[Int_val (n_v)];
2822 Field (ret_v, 0) = Val_int (slink->bbox.x0);
2823 Field (ret_v, 1) = Val_int (slink->bbox.y0);
2824 Field (ret_v, 2) = Val_int (slink->bbox.x1);
2825 Field (ret_v, 3) = Val_int (slink->bbox.y1);
2826 unlock ("ml_getlinkrect");
2828 done:
2829 CAMLreturn (ret_v);
2832 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
2834 CAMLparam3 (ptr_v, x_v, y_v);
2835 CAMLlocal3 (ret_v, tup_v, str_v);
2836 fz_link *link;
2837 struct page *page;
2838 char *s = String_val (ptr_v);
2839 int x = Int_val (x_v), y = Int_val (y_v);
2840 struct pagedim *pdim;
2842 ret_v = Val_int (0);
2843 if (trylock ("ml_whatsunder")) {
2844 goto done;
2847 page = parse_pointer ("ml_whatsunder", s);
2848 pdim = &state.pagedims[page->pdimno];
2849 x += pdim->bounds.x0;
2850 y += pdim->bounds.y0;
2851 link = getlink (page, x, y);
2852 if (link) {
2853 LINKTOVAL;
2855 else {
2856 fz_rect *b;
2857 fz_text_block *block;
2859 ensuretext (page);
2860 for (block = page->text->blocks;
2861 block < page->text->blocks + page->text->len;
2862 ++block) {
2863 fz_text_line *line;
2865 b = &block->bbox;
2866 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2867 continue;
2869 for (line = block->lines;
2870 line < block->lines + block->len;
2871 ++line) {
2872 int spannum;
2874 b = &line->bbox;
2875 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2876 continue;
2878 for (spannum = 0; spannum < line->len; ++spannum) {
2879 int charnum;
2880 fz_text_span *span = line->spans[spannum];
2882 b = &span->bbox;
2883 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2884 continue;
2886 for (charnum = 0; charnum < span->len; ++charnum) {
2887 fz_rect bbox;
2888 fz_text_char_bbox (&bbox, span, charnum);
2889 b = &bbox;
2891 if (x >= b->x0 && x <= b->x1
2892 && y >= b->y0 && y <= b->y1) {
2893 fz_text_style *style = span->text->style;
2894 const char *n2 =
2895 style->font && style->font->name
2896 ? style->font->name
2897 : "Span has no font name"
2899 FT_FaceRec *face = style->font->ft_face;
2900 if (face && face->family_name) {
2901 char *s;
2902 char *n1 = face->family_name;
2903 size_t l1 = strlen (n1);
2904 size_t l2 = strlen (n2);
2906 if (l1 != l2 || memcmp (n1, n2, l1)) {
2907 s = malloc (l1 + l2 + 2);
2908 if (s) {
2909 memcpy (s, n2, l2);
2910 s[l2] = '=';
2911 memcpy (s + l2 + 1, n1, l1 + 1);
2912 str_v = caml_copy_string (s);
2913 free (s);
2917 if (str_v == 0) {
2918 str_v = caml_copy_string (n2);
2920 ret_v = caml_alloc_small (1, utext);
2921 Field (ret_v, 0) = str_v;
2922 goto unlock;
2929 unlock:
2930 unlock ("ml_whatsunder");
2932 done:
2933 CAMLreturn (ret_v);
2936 CAMLprim value ml_seltext (value ptr_v, value rect_v)
2938 CAMLparam2 (ptr_v, rect_v);
2939 fz_rect b;
2940 struct page *page;
2941 struct pagedim *pdim;
2942 int i, x0, x1, y0, y1;
2943 char *s = String_val (ptr_v);
2944 int fi = 0, li = 0;
2945 fz_text_block *block;
2946 fz_text_span *span, *fspan, *lspan;
2947 fz_text_line *line, *fline = NULL, *lline = NULL;
2949 if (trylock ("ml_seltext")) {
2950 goto done;
2953 page = parse_pointer ("ml_seltext", s);
2954 ensuretext (page);
2956 pdim = &state.pagedims[page->pdimno];
2957 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;;
2958 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
2959 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
2960 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
2962 if (0) {
2963 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2964 glColor3ub (128, 128, 128);
2965 glRecti (x0, y0, x1, y1);
2966 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2969 fspan = lspan = NULL;
2971 for (block = page->text->blocks;
2972 block < page->text->blocks + page->text->len;
2973 ++block) {
2974 for (line = block->lines;
2975 line < block->lines + block->len;
2976 ++line) {
2977 int spannum;
2979 for (spannum = 0; spannum < line->len; ++spannum) {
2980 span = line->spans[spannum];
2982 for (i = 0; i < span->len; ++i) {
2983 int selected = 0;
2985 fz_text_char_bbox (&b, span, i);
2987 if (x0 >= b.x0 && x0 <= b.x1
2988 && y0 >= b.y0 && y0 <= b.y1) {
2989 fspan = span;
2990 fline = line;
2991 fi = i;
2992 selected = 1;
2994 if (x1 >= b.x0 && x1 <= b.x1
2995 && y1 >= b.y0 && y1 <= b.y1) {
2996 lspan = span;
2997 lline = line;
2998 li = i;
2999 selected = 1;
3001 if (0 && selected) {
3002 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
3003 glColor3ub (128, 128, 128);
3004 glRecti (b.x0, b.y0, b.x1, b.y1);
3005 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
3011 if (y1 < y0 || x1 < x0) {
3012 int swap = 0;
3014 if (fspan == lspan) {
3015 swap = 1;
3017 else {
3018 if (y1 < y0) {
3019 if (fline != lline) {
3020 swap = 1;
3025 if (swap) {
3026 i = fi;
3027 span = fspan;
3029 fi = li;
3030 fspan = lspan;
3032 li = i;
3033 lspan = span;
3037 page->fmark.i = fi;
3038 page->fmark.span = fspan;
3040 page->lmark.i = li;
3041 page->lmark.span = lspan;
3043 unlock ("ml_seltext");
3045 done:
3046 CAMLreturn (Val_unit);
3049 static int UNUSED_ATTR pipespan (FILE *f, fz_text_span *span, int a, int b)
3051 char buf[4];
3052 int i, len, ret;
3054 for (i = a; i <= b; ++i) {
3055 len = fz_runetochar (buf, span->text[i].c);
3056 ret = fwrite (buf, len, 1, f);
3058 if (ret != 1) {
3059 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
3060 len, ret, strerror (errno));
3061 return -1;
3064 return 0;
3067 #ifdef __CYGWIN__
3068 value ml_popen (value UNUSED_ATTR u1, value UNUSED_ATTR u2)
3070 caml_failwith ("ml_popen not implemented under Cygwin");
3072 #else
3073 CAMLprim value ml_popen (value command_v, value fds_v)
3075 CAMLparam2 (command_v, fds_v);
3076 CAMLlocal2 (l_v, tup_v);
3077 int ret;
3078 char *msg = NULL;
3079 value earg_v = Nothing;
3080 posix_spawnattr_t attr;
3081 posix_spawn_file_actions_t fa;
3082 char *argv[] = { "/bin/sh", "-c", String_val (command_v), NULL };
3084 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
3085 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
3088 if ((ret = posix_spawnattr_init (&attr)) != 0) {
3089 msg = "posix_spawnattr_init";
3090 goto fail1;
3093 #ifdef POSIX_SPAWN_USEVFORK
3094 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
3095 msg = "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK";
3096 goto fail;
3098 #endif
3100 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
3101 int fd1, fd2;
3103 tup_v = Field (l_v, 0);
3104 fd1 = Int_val (Field (tup_v, 0));
3105 fd2 = Int_val (Field (tup_v, 1));
3106 if (fd2 < 0) {
3107 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
3108 msg = "posix_spawn_file_actions_addclose";
3109 earg_v = tup_v;
3110 goto fail;
3113 else {
3114 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
3115 msg = "posix_spawn_file_actions_adddup2";
3116 earg_v = tup_v;
3117 goto fail;
3122 if ((ret = posix_spawn (NULL, "/bin/sh", &fa, &attr, argv, environ))) {
3123 msg = "posix_spawn";
3124 goto fail;
3127 fail:
3128 if ((ret = posix_spawnattr_destroy (&attr)) != 0) {
3129 fprintf (stderr, "posix_spawnattr_destroy: %s\n", strerror (ret));
3132 fail1:
3133 if ((ret = posix_spawn_file_actions_destroy (&fa)) != 0) {
3134 fprintf (stderr, "posix_spawn_file_actions_destroy: %s\n",
3135 strerror (ret));
3138 if (msg)
3139 unix_error (ret, msg, earg_v);
3141 CAMLreturn (Val_unit);
3143 #endif
3145 CAMLprim value ml_copysel (value fd_v, value ptr_v)
3147 CAMLparam1 (ptr_v);
3148 FILE *f;
3149 int seen = 0;
3150 struct page *page;
3151 fz_text_line *line;
3152 fz_text_block *block;
3153 int fd = Int_val (fd_v);
3154 char *s = String_val (ptr_v);
3156 if (trylock ("ml_copysel")) {
3157 goto done;
3160 page = parse_pointer ("ml_sopysel", s);
3162 if (!page->fmark.span || !page->lmark.span) {
3163 fprintf (stderr, "nothing to copy\n");
3164 goto unlock;
3167 f = fdopen (fd, "w");
3168 if (!f) {
3169 fprintf (stderr, "failed to fopen sel pipe: %s\n",
3170 strerror (errno));
3171 f = stdout;
3174 for (block = page->text->blocks;
3175 block < page->text->blocks + page->text->len;
3176 ++block) {
3177 for (line = block->lines;
3178 line < block->lines + block->len;
3179 ++line) {
3180 int spannum;
3182 for (spannum = 0; spannum < line->len; ++spannum) {
3183 int a, b;
3184 fz_text_span *span = line->spans[spannum];
3186 seen |= span == page->fmark.span || span == page->lmark.span;
3187 a = span == page->fmark.span ? page->fmark.i : 0;
3188 b = span == page->lmark.span ? page->lmark.i : span->len - 1;
3190 if (seen) {
3191 if (pipespan (f, span, a, b)) {
3192 goto close;
3194 if (spannum == line->len - 1) {
3195 if (putc ('\n', f) == EOF) {
3196 fprintf (stderr,
3197 "failed break line on sel pipe: %s\n",
3198 strerror (errno));
3199 goto close;
3202 if (span == page->lmark.span) {
3203 goto endloop;
3209 endloop:
3210 page->lmark.span = NULL;
3211 page->fmark.span = NULL;
3213 close:
3214 if (f != stdout) {
3215 int ret = fclose (f);
3216 fd = -1;
3217 if (ret == -1) {
3218 if (errno != ECHILD) {
3219 fprintf (stderr, "failed to close sel pipe: %s\n",
3220 strerror (errno));
3224 unlock:
3225 unlock ("ml_copysel");
3227 done:
3228 if (fd >= 0) {
3229 if (close (fd)) {
3230 fprintf (stderr, "failed to close sel pipe: %s\n",
3231 strerror (errno));
3234 CAMLreturn (Val_unit);
3237 CAMLprim value ml_getpdimrect (value pagedimno_v)
3239 CAMLparam1 (pagedimno_v);
3240 CAMLlocal1 (ret_v);
3241 int pagedimno = Int_val (pagedimno_v);
3242 fz_rect box;
3244 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
3245 if (trylock ("ml_getpdimrect")) {
3246 box = fz_empty_rect;
3248 else {
3249 box = state.pagedims[pagedimno].mediabox;
3250 unlock ("ml_getpdimrect");
3253 Store_double_field (ret_v, 0, box.x0);
3254 Store_double_field (ret_v, 1, box.x1);
3255 Store_double_field (ret_v, 2, box.y0);
3256 Store_double_field (ret_v, 3, box.y1);
3258 CAMLreturn (ret_v);
3261 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v,
3262 value dw_v, value cols_v)
3264 CAMLparam3 (winw_v, winh_v, dw_v);
3265 CAMLlocal1 (ret_v);
3266 int i;
3267 double zoom = -1.;
3268 double maxh = 0.0;
3269 struct pagedim *p;
3270 double winw = Int_val (winw_v);
3271 double winh = Int_val (winh_v);
3272 double dw = Int_val (dw_v);
3273 double cols = Int_val (cols_v);
3274 double pw = 1.0, ph = 1.0;
3276 if (trylock ("ml_zoom_for_height")) {
3277 goto done;
3280 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
3281 double w = p->pagebox.x1 / cols;
3282 double h = p->pagebox.y1;
3283 if (h > maxh) {
3284 maxh = h;
3285 ph = h;
3286 if (state.fitmodel != FitProportional) pw = w;
3288 if ((state.fitmodel == FitProportional) && w > pw) pw = w;
3291 zoom = (((winh / ph) * pw) + dw) / winw;
3292 unlock ("ml_zoom_for_height");
3293 done:
3294 ret_v = caml_copy_double (zoom);
3295 CAMLreturn (ret_v);
3298 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
3300 CAMLparam4 (pt_v, x_v, y_v, string_v);
3301 CAMLlocal1 (ret_v);
3302 int pt = Int_val(pt_v);
3303 int x = Int_val (x_v);
3304 int y = Int_val (y_v);
3305 double w;
3307 w = draw_string (state.face, pt, x, y, String_val (string_v));
3308 ret_v = caml_copy_double (w);
3309 CAMLreturn (ret_v);
3312 CAMLprim value ml_measure_string (value pt_v, value string_v)
3314 CAMLparam2 (pt_v, string_v);
3315 CAMLlocal1 (ret_v);
3316 int pt = Int_val (pt_v);
3317 double w;
3319 w = measure_string (state.face, pt, String_val (string_v));
3320 ret_v = caml_copy_double (w);
3321 CAMLreturn (ret_v);
3324 CAMLprim value ml_getpagebox (value opaque_v)
3326 CAMLparam1 (opaque_v);
3327 CAMLlocal1 (ret_v);
3328 fz_rect rect;
3329 fz_irect bbox;
3330 fz_matrix ctm;
3331 fz_device *dev;
3332 char *s = String_val (opaque_v);
3333 struct page *page = parse_pointer ("ml_getpagebox", s);
3335 ret_v = caml_alloc_tuple (4);
3336 dev = fz_new_bbox_device (state.ctx, &rect);
3337 dev->hints |= FZ_IGNORE_SHADE;
3339 switch (page->type) {
3340 case DPDF:
3341 ctm = pagectm (page);
3342 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, &ctm, NULL);
3343 break;
3345 case DXPS:
3346 ctm = pagectm (page);
3347 xps_run_page (state.u.xps, page->u.xpspage, dev, &ctm, NULL);
3348 break;
3350 default:
3351 rect = fz_infinite_rect;
3352 break;
3355 fz_free_device (dev);
3356 fz_round_rect (&bbox, &rect);
3357 Field (ret_v, 0) = Val_int (bbox.x0);
3358 Field (ret_v, 1) = Val_int (bbox.y0);
3359 Field (ret_v, 2) = Val_int (bbox.x1);
3360 Field (ret_v, 3) = Val_int (bbox.y1);
3362 CAMLreturn (ret_v);
3365 CAMLprim value ml_setaalevel (value level_v)
3367 CAMLparam1 (level_v);
3369 state.aalevel = Int_val (level_v);
3370 CAMLreturn (Val_unit);
3373 #undef pixel
3374 #include <X11/Xlib.h>
3375 #include <GL/glx.h>
3377 static struct {
3378 Display *dpy;
3379 GLXContext ctx;
3380 GLXDrawable drawable;
3381 } glx;
3383 #include "keysym2ucs.c"
3385 CAMLprim value ml_keysymtoutf8 (value keysym_v)
3387 CAMLparam1 (keysym_v);
3388 CAMLlocal1 (str_v);
3389 KeySym keysym = Int_val (keysym_v);
3390 Rune rune;
3391 int len;
3392 char buf[5];
3394 rune = keysym2ucs (keysym);
3395 len = fz_runetochar (buf, rune);
3396 buf[len] = 0;
3397 str_v = caml_copy_string (buf);
3398 CAMLreturn (str_v);
3401 CAMLprim value ml_glx (value win_v)
3403 CAMLparam1 (win_v);
3404 XVisualInfo *visual;
3405 int screen, wid = Int_val (win_v);
3406 int attributes[] = { GLX_RGBA, GLX_DOUBLEBUFFER, None };
3408 glx.dpy = XOpenDisplay (NULL);
3409 if (!glx.dpy) {
3410 caml_failwith ("XOpenDisplay");
3413 screen = DefaultScreen (glx.dpy);
3414 visual = glXChooseVisual (glx.dpy, screen, attributes);
3415 if (!visual) {
3416 XCloseDisplay (glx.dpy);
3417 glx.dpy = NULL;
3418 caml_failwith ("glXChooseVisual");
3421 glx.ctx = glXCreateContext (glx.dpy, visual, NULL, True);
3422 XFree (visual);
3423 if (!glx.ctx) {
3424 XCloseDisplay (glx.dpy);
3425 glx.dpy = NULL;
3426 caml_failwith ("glXCreateContext");
3429 if (!glXMakeCurrent (glx.dpy, wid, glx.ctx)) {
3430 glXDestroyContext (glx.dpy, glx.ctx);
3431 XCloseDisplay (glx.dpy);
3432 glx.dpy = NULL;
3433 glx.ctx = NULL;
3434 caml_failwith ("glXMakeCurrent");
3436 glx.drawable = wid;
3437 CAMLreturn (Val_unit);
3440 CAMLprim value ml_swapb (value unit_v)
3442 CAMLparam1 (unit_v);
3443 glXSwapBuffers (glx.dpy, glx.drawable);
3444 CAMLreturn (Val_unit);
3447 CAMLprim value ml_glxsync (value unit_v)
3449 CAMLparam1 (unit_v);
3450 if (glx.dpy && glx.ctx) {
3451 glXWaitX ();
3452 glXWaitGL ();
3454 CAMLreturn (Val_unit);
3457 enum { piunknown, pilinux, piosx, pisun, pifreebsd,
3458 pidragonflybsd, piopenbsd, pinetbsd, picygwin };
3460 CAMLprim value ml_platform (value unit_v)
3462 CAMLparam1 (unit_v);
3463 int platid = piunknown;
3465 #if defined __linux__
3466 platid = pilinux;
3467 #elif defined __CYGWIN__
3468 platid = picygwin;
3469 #elif defined __DragonFly__
3470 platid = pidragonflybsd;
3471 #elif defined __FreeBSD__
3472 platid = pifreebsd;
3473 #elif defined __OpenBSD__
3474 platid = piopenbsd;
3475 #elif defined __NetBSD__
3476 platid = pinetbsd;
3477 #elif defined __sun__
3478 platid = pisun;
3479 #elif defined __APPLE__
3480 platid = piosx;
3481 #endif
3482 CAMLreturn (Val_int (platid));
3485 CAMLprim value ml_cloexec (value fd_v)
3487 CAMLparam1 (fd_v);
3488 int fd = Int_val (fd_v);
3490 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
3491 uerror ("fcntl", Nothing);
3493 CAMLreturn (Val_unit);
3496 CAMLprim value ml_getpbo (value w_v, value h_v, value cs_v)
3498 CAMLparam2 (w_v, h_v);
3499 CAMLlocal1 (ret_v);
3500 struct pbo *pbo;
3501 int w = Int_val (w_v);
3502 int h = Int_val (h_v);
3503 int cs = Int_val (cs_v);
3505 if (state.pbo_usable) {
3506 pbo = calloc (sizeof (*pbo), 1);
3507 if (!pbo) {
3508 err (1, "calloc pbo");
3511 switch (cs) {
3512 case 0:
3513 case 1:
3514 pbo->size = w*h*4;
3515 break;
3516 case 2:
3517 pbo->size = w*h*2;
3518 break;
3519 default:
3520 errx (1, "ml_getpbo: invalid colorspace %d", cs);
3523 state.glGenBuffersARB (1, &pbo->id);
3524 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->id);
3525 state.glBufferDataARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->size,
3526 NULL, GL_STREAM_DRAW);
3527 pbo->ptr = state.glMapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB,
3528 GL_READ_WRITE);
3529 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3530 if (!pbo->ptr) {
3531 fprintf (stderr, "glMapBufferARB failed: %#x\n", glGetError ());
3532 state.glDeleteBuffersARB (1, &pbo->id);
3533 free (pbo);
3534 ret_v = caml_copy_string ("0");
3536 else {
3537 int res;
3538 char *s;
3540 res = snprintf (NULL, 0, "%" FMT_ptr, pbo);
3541 if (res < 0) {
3542 err (1, "snprintf %" FMT_ptr " failed", pbo);
3544 s = malloc (res+1);
3545 if (!s) {
3546 err (1, "malloc %d bytes failed", res+1);
3548 res = sprintf (s, "%" FMT_ptr, pbo);
3549 if (res < 0) {
3550 err (1, "sprintf %" FMT_ptr " failed", pbo);
3552 ret_v = caml_copy_string (s);
3553 free (s);
3556 else {
3557 ret_v = caml_copy_string ("0");
3559 CAMLreturn (ret_v);
3562 CAMLprim value ml_freepbo (value s_v)
3564 CAMLparam1 (s_v);
3565 char *s = String_val (s_v);
3566 struct tile *tile = parse_pointer ("ml_freepbo", s);
3568 if (tile->pbo) {
3569 state.glDeleteBuffersARB (1, &tile->pbo->id);
3570 tile->pbo->id = -1;
3571 tile->pbo->ptr = NULL;
3572 tile->pbo->size = -1;
3574 CAMLreturn (Val_unit);
3577 CAMLprim value ml_unmappbo (value s_v)
3579 CAMLparam1 (s_v);
3580 char *s = String_val (s_v);
3581 struct tile *tile = parse_pointer ("ml_unmappbo", s);
3583 if (tile->pbo) {
3584 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
3585 if (state.glUnmapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB) == GL_FALSE) {
3586 errx (1, "glUnmapBufferARB failed: %#x\n", glGetError ());
3588 tile->pbo->ptr = NULL;
3589 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3591 CAMLreturn (Val_unit);
3594 static void setuppbo (void)
3596 #define GGPA(n) *(void (**) ()) &state.n = glXGetProcAddress ((GLubyte *) #n)
3597 GGPA (glBindBufferARB);
3598 if (state.glBindBufferARB) {
3599 GGPA (glUnmapBufferARB);
3600 if (state.glUnmapBufferARB) {
3601 GGPA (glMapBufferARB);
3602 if (state.glMapBufferARB) {
3603 GGPA (glBufferDataARB);
3604 if (state.glBufferDataARB) {
3605 GGPA (glGenBuffersARB);
3606 if (state.glGenBuffersARB) {
3607 GGPA (glDeleteBuffersARB);
3608 if (state.glDeleteBuffersARB) {
3609 state.pbo_usable = 1;
3616 #undef GGPA
3619 CAMLprim value ml_pbo_usable (value unit_v)
3621 CAMLparam1 (unit_v);
3622 CAMLreturn (Val_bool (state.pbo_usable));
3625 CAMLprim value ml_unproject (value ptr_v, value x_v, value y_v)
3627 CAMLparam3 (ptr_v, x_v, y_v);
3628 CAMLlocal2 (ret_v, tup_v);
3629 struct page *page;
3630 char *s = String_val (ptr_v);
3631 int x = Int_val (x_v), y = Int_val (y_v);
3632 struct pagedim *pdim;
3633 fz_point p;
3634 fz_matrix ctm;
3636 page = parse_pointer ("ml_unproject", s);
3637 pdim = &state.pagedims[page->pdimno];
3639 ret_v = Val_int (0);
3640 if (trylock ("ml_unproject")) {
3641 goto done;
3644 switch (page->type) {
3645 case DPDF:
3646 trimctm (page->u.pdfpage, page->pdimno);
3647 break;
3649 default:
3650 break;
3652 p.x = x;
3653 p.y = y;
3655 fz_concat (&ctm, &pdim->tctm, &pdim->ctm);
3656 fz_invert_matrix (&ctm, &ctm);
3657 fz_transform_point (&p, &ctm);
3659 tup_v = caml_alloc_tuple (2);
3660 ret_v = caml_alloc_small (1, 1);
3661 Field (tup_v, 0) = Val_int (p.x);
3662 Field (tup_v, 1) = Val_int (p.y);
3663 Field (ret_v, 0) = tup_v;
3665 unlock ("ml_unproject");
3666 done:
3667 CAMLreturn (ret_v);
3670 static void makestippletex (void)
3672 const char *pixels = "\xff\xff\0\0";
3673 glGenTextures (1, &state.stid);
3674 glBindTexture (GL_TEXTURE_1D, state.stid);
3675 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
3676 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
3677 glTexImage1D (
3678 GL_TEXTURE_1D,
3680 GL_ALPHA,
3683 GL_ALPHA,
3684 GL_UNSIGNED_BYTE,
3685 pixels
3689 CAMLprim value ml_init (value pipe_v, value params_v)
3691 CAMLparam2 (pipe_v, params_v);
3692 CAMLlocal2 (trim_v, fuzz_v);
3693 int ret;
3694 int texcount;
3695 char *fontpath;
3696 int colorspace;
3697 int mustoresize;
3698 int haspboext;
3699 struct sigaction sa;
3701 state.cr = Int_val (Field (pipe_v, 0));
3702 state.cw = Int_val (Field (pipe_v, 1));
3703 state.rotate = Int_val (Field (params_v, 0));
3704 state.fitmodel = Int_val (Field (params_v, 1));
3705 trim_v = Field (params_v, 2);
3706 texcount = Int_val (Field (params_v, 3));
3707 state.sliceheight = Int_val (Field (params_v, 4));
3708 mustoresize = Int_val (Field (params_v, 5));
3709 colorspace = Int_val (Field (params_v, 6));
3710 fontpath = String_val (Field (params_v, 7));
3711 state.trimcachepath = strdup (String_val (Field (params_v, 8)));
3712 if (!state.trimcachepath) {
3713 fprintf (stderr, "failed to strdup trimcachepath: %s\n",
3714 strerror (errno));
3716 haspboext = Bool_val (Field (params_v, 9));
3718 state.ctx = fz_new_context (NULL, NULL, mustoresize);
3720 state.trimmargins = Bool_val (Field (trim_v, 0));
3721 fuzz_v = Field (trim_v, 1);
3722 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
3723 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
3724 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
3725 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
3727 set_tex_params (colorspace);
3729 if (*fontpath) {
3730 state.face = load_font (fontpath);
3732 else {
3733 unsigned int len;
3734 void *base = pdf_lookup_substitute_font (0, 0, 0, 0, &len);
3736 state.face = load_builtin_font (base, len);
3738 if (!state.face) _exit (1);
3740 realloctexts (texcount);
3742 if (haspboext) {
3743 setuppbo ();
3746 makestippletex ();
3747 #ifdef __CYGWIN__
3748 sa.sa_handler = SIG_IGN;
3749 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
3750 #else
3751 sa.sa_handler = SIG_DFL;
3752 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP | SA_NOCLDWAIT;
3753 #endif
3754 if (sigemptyset (&sa.sa_mask)) {
3755 err (1, "sigemptyset");
3757 if (sigaction (SIGCHLD, &sa, NULL)) {
3758 err (1, "sigaction");
3761 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
3762 if (ret) {
3763 errx (1, "pthread_create: %s", strerror (ret));
3766 CAMLreturn (Val_unit);