Comment key
[llpp.git] / link.c
blob2ecc43c58995740eeabf949961657b9a769104bb
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 <mupdf/fitz.h>
34 #include <mupdf/cbz.h>
35 #include <mupdf/pdf.h>
36 #include <mupdf/xps.h>
38 #include FT_FREETYPE_H
40 #define PIGGYBACK
42 #ifndef __USE_GNU
43 extern char **environ;
44 #endif
46 #define MIN(a,b) ((a) < (b) ? (a) : (b))
47 #define MAX(a,b) ((a) > (b) ? (a) : (b))
49 #if defined __GNUC__
50 #define NORETURN_ATTR __attribute__ ((noreturn))
51 #define UNUSED_ATTR __attribute__ ((unused))
52 #define OPTIMIZE_ATTR(n) __attribute__ ((optimize ("O"#n)))
53 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
54 #else
55 #define NORETURN_ATTR
56 #define UNUSED_ATTR
57 #define OPTIMIZE_ATTR(n)
58 #define GCC_FMT_ATTR(a, b)
59 #endif
61 #define FMT_s "zu"
63 #define FMT_ptr "p"
64 #define FMT_ptr_cast(p) (p)
65 #define FMT_ptr_cast2(p) (p)
67 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
68 err (int exitcode, const char *fmt, ...)
70 va_list ap;
71 int savederrno;
73 savederrno = errno;
74 va_start (ap, fmt);
75 vfprintf (stderr, fmt, ap);
76 va_end (ap);
77 fprintf (stderr, ": %s\n", strerror (savederrno));
78 fflush (stderr);
79 _exit (exitcode);
82 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
83 errx (int exitcode, const char *fmt, ...)
85 va_list ap;
87 va_start (ap, fmt);
88 vfprintf (stderr, fmt, ap);
89 va_end (ap);
90 fputc ('\n', stderr);
91 fflush (stderr);
92 _exit (exitcode);
95 #ifndef GL_TEXTURE_RECTANGLE_ARB
96 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
97 #endif
99 #ifndef GL_BGRA
100 #define GL_BGRA 0x80E1
101 #endif
103 #ifndef GL_UNSIGNED_INT_8_8_8_8
104 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
105 #endif
107 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
108 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
109 #endif
111 #if 0
112 #define lprintf printf
113 #else
114 #define lprintf(...)
115 #endif
117 #define ARSERT(cond) for (;;) { \
118 if (!(cond)) { \
119 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
121 break; \
124 struct slice {
125 int h;
126 int texindex;
129 struct tile {
130 int w, h;
131 int slicecount;
132 int sliceheight;
133 struct pbo *pbo;
134 fz_pixmap *pixmap;
135 struct slice slices[1];
138 struct pagedim {
139 int pageno;
140 int rotate;
141 int left;
142 int tctmready;
143 fz_irect bounds;
144 fz_rect pagebox;
145 fz_rect mediabox;
146 fz_matrix ctm, zoomctm, lctm, tctm;
149 struct slink {
150 fz_irect bbox;
151 fz_link *link;
154 enum { DPDF, DXPS, DCBZ };
156 struct page {
157 int tgen;
158 int sgen;
159 int type;
160 int pageno;
161 int pdimno;
162 fz_text_page *text;
163 fz_text_sheet *sheet;
164 union {
165 void *ptr;
166 pdf_page *pdfpage;
167 xps_page *xpspage;
168 cbz_page *cbzpage;
169 } u;
170 fz_display_list *dlist;
171 int slinkcount;
172 struct slink *slinks;
173 struct mark {
174 int i;
175 fz_text_span *span;
176 } fmark, lmark;
177 void (*freepage) (void *);
180 struct {
181 int type;
182 int sliceheight;
183 struct pagedim *pagedims;
184 int pagecount;
185 int pagedimcount;
186 union {
187 pdf_document *pdf;
188 xps_document *xps;
189 cbz_document *cbz;
190 } u;
191 fz_context *ctx;
192 int w, h;
194 int texindex;
195 int texcount;
196 GLuint *texids;
198 GLenum texiform;
199 GLenum texform;
200 GLenum texty;
202 fz_colorspace *colorspace;
204 struct {
205 int w, h;
206 struct slice *slice;
207 } *texowners;
209 int rotate;
210 enum { FitWidth, FitProportional, FitPage } fitmodel;
211 int trimmargins;
212 int needoutline;
213 int gen;
214 int aalevel;
216 int trimanew;
217 fz_irect trimfuzz;
218 fz_pixmap *pig;
220 pthread_t thread;
221 int cr, cw;
222 FT_Face face;
224 void (*closedoc) (void);
225 void (*freepage) (void *);
227 char *trimcachepath;
228 int cxack;
230 GLuint stid;
232 int pbo_usable;
233 void (*glBindBufferARB) (GLenum, GLuint);
234 GLboolean (*glUnmapBufferARB) (GLenum);
235 void *(*glMapBufferARB) (GLenum, GLenum);
236 void (*glBufferDataARB) (GLenum, GLsizei, void *, GLenum);
237 void (*glGenBuffersARB) (GLsizei, GLuint *);
238 void (*glDeleteBuffersARB) (GLsizei, GLuint *);
239 } state;
241 struct pbo {
242 GLuint id;
243 void *ptr;
244 size_t size;
247 static void UNUSED_ATTR debug_rect (const char *cap, fz_rect r)
249 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
252 static void UNUSED_ATTR debug_bbox (const char *cap, fz_irect r)
254 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
257 static void UNUSED_ATTR debug_matrix (const char *cap, fz_matrix m)
259 printf ("%s(matrix) %.2f,%.2f,%.2f,%.2f %.2f %.2f\n", cap,
260 m.a, m.b, m.c, m.d, m.e, m.f);
263 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
265 static void lock (const char *cap)
267 int ret = pthread_mutex_lock (&mutex);
268 if (ret) {
269 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
273 static void unlock (const char *cap)
275 int ret = pthread_mutex_unlock (&mutex);
276 if (ret) {
277 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
281 static int trylock (const char *cap)
283 int ret = pthread_mutex_trylock (&mutex);
284 if (ret && ret != EBUSY) {
285 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
287 return ret == EBUSY;
290 static void *parse_pointer (const char *cap, const char *s)
292 int ret;
293 void *ptr;
295 ret = sscanf (s, "%" FMT_ptr, FMT_ptr_cast (&ptr));
296 if (ret != 1) {
297 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
299 return ptr;
302 static double now (void)
304 struct timeval tv;
306 if (gettimeofday (&tv, NULL)) {
307 err (1, "gettimeofday");
309 return tv.tv_sec + tv.tv_usec*1e-6;
312 static int hasdata (void)
314 int ret, avail;
315 ret = ioctl (state.cr, FIONREAD, &avail);
316 if (ret) err (1, "hasdata: FIONREAD error ret=%d", ret);
317 return avail > 0;
320 CAMLprim value ml_hasdata (value fd_v)
322 CAMLparam1 (fd_v);
323 int ret, avail;
325 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
326 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
327 CAMLreturn (Val_bool (avail > 0));
330 static void readdata (void *p, int size)
332 ssize_t n;
334 again:
335 n = read (state.cr, p, size);
336 if (n < 0) {
337 if (errno == EINTR) goto again;
338 err (1, "read (req %d, ret %zd)", size, n);
340 if (n - size) {
341 if (!n) errx (1, "EOF while reading");
342 errx (1, "read (req %d, ret %zd)", size, n);
346 static void writedata (char *p, int size)
348 char buf[4];
349 ssize_t n;
351 buf[0] = (size >> 24) & 0xff;
352 buf[1] = (size >> 16) & 0xff;
353 buf[2] = (size >> 8) & 0xff;
354 buf[3] = (size >> 0) & 0xff;
356 n = write (state.cw, buf, 4);
357 if (n != 4) {
358 if (!n) errx (1, "EOF while writing length");
359 err (1, "write %zd", n);
362 n = write (state.cw, p, size);
363 if (n - size) {
364 if (!n) errx (1, "EOF while writing data");
365 err (1, "write (req %d, ret %zd)", size, n);
369 static int readlen (void)
371 unsigned char p[4];
373 readdata (p, 4);
374 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
377 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
379 int size = 200, len;
380 va_list ap;
381 char *buf;
383 buf = malloc (size);
384 for (;;) {
385 if (!buf) err (1, "malloc for temp buf (%d bytes) failed", size);
387 va_start (ap, fmt);
388 len = vsnprintf (buf, size, fmt, ap);
389 va_end (ap);
391 if (len > -1 && len < size) {
392 writedata (buf, len);
393 break;
396 if (len > -1) {
397 size = len + 1;
399 else {
400 size *= 2;
402 buf = realloc (buf, size);
404 free (buf);
407 static void closepdf (void)
409 if (state.u.pdf) {
410 pdf_close_document (state.u.pdf);
411 state.u.pdf = NULL;
415 static void closexps (void)
417 if (state.u.xps) {
418 xps_close_document (state.u.xps);
419 state.u.xps = NULL;
423 static void closecbz (void)
425 if (state.u.cbz) {
426 cbz_close_document (state.u.cbz);
427 state.u.cbz = NULL;
431 static void freepdfpage (void *ptr)
433 pdf_free_page (state.u.pdf, ptr);
436 static void freeemptyxxxpage (void *ptr)
438 (void) ptr;
441 static void freexpspage (void *ptr)
443 xps_free_page (state.u.xps, ptr);
446 static void freecbzpage (void *ptr)
448 cbz_free_page (state.u.cbz, ptr);
451 static void openxref (char *filename, char *password)
453 int i, len;
455 for (i = 0; i < state.texcount; ++i) {
456 state.texowners[i].w = -1;
457 state.texowners[i].slice = NULL;
460 if (state.closedoc) state.closedoc ();
462 len = strlen (filename);
464 state.type = DPDF;
465 if (len > 4) {
466 char ext[4];
468 ext[0] = tolower ((int) filename[len-3]);
469 ext[1] = tolower ((int) filename[len-2]);
470 ext[2] = tolower ((int) filename[len-1]);
472 /**/ if (ext[0] == 'x' && ext[1] == 'p' && ext[2] == 's') {
473 state.type = DXPS;
475 else if (ext[0] == 'c' && ext[1] == 'b' && ext[2] == 'z') {
476 state.type = DCBZ;
480 if (state.pagedims) {
481 free (state.pagedims);
482 state.pagedims = NULL;
484 state.pagedimcount = 0;
486 fz_set_aa_level (state.ctx, state.aalevel);
487 switch (state.type) {
488 case DPDF:
489 state.u.pdf = pdf_open_document (state.ctx, filename);
490 if (pdf_needs_password (state.u.pdf)) {
491 int okay = pdf_authenticate_password (state.u.pdf, password);
492 if (!okay) {
493 errx (1, "invalid password");
496 state.pagecount = pdf_count_pages (state.u.pdf);
497 state.closedoc = closepdf;
498 state.freepage = freepdfpage;
499 break;
501 case DXPS:
502 state.u.xps = xps_open_document (state.ctx, filename);
503 state.pagecount = xps_count_pages (state.u.xps);
504 state.closedoc = closexps;
505 state.freepage = freexpspage;
506 break;
508 case DCBZ:
509 state.u.cbz = cbz_open_document (state.ctx, filename);
510 state.pagecount = cbz_count_pages (state.u.cbz);
511 state.closedoc = closecbz;
512 state.freepage = freecbzpage;
513 break;
517 static void pdfinfo (void)
519 if (state.type == DPDF) {
520 pdf_obj *infoobj;
522 printd ("info PDF version\t%d.%d",
523 state.u.pdf->version / 10, state.u.pdf->version % 10);
525 infoobj = pdf_dict_gets (pdf_trailer (state.u.pdf), "Info");
526 if (infoobj) {
527 int i;
528 char *s;
529 char *items[] = { "Title", "Author", "Creator",
530 "Producer", "CreationDate" };
532 for (i = 0; i < sizeof (items) / sizeof (*items); ++i) {
533 pdf_obj *obj = pdf_dict_gets (infoobj, items[i]);
534 s = pdf_to_utf8 (state.u.pdf, obj);
535 if (*s) {
536 if (i == 0) {
537 printd ("title %s", s);
539 printd ("info %s\t%s", items[i], s);
541 fz_free (state.ctx, s);
544 printd ("infoend");
548 static void unlinktile (struct tile *tile)
550 int i;
552 for (i = 0; i < tile->slicecount; ++i) {
553 struct slice *s = &tile->slices[i];
555 if (s->texindex != -1) {
556 if (state.texowners[s->texindex].slice == s) {
557 state.texowners[s->texindex].slice = NULL;
563 static void freepage (struct page *page)
565 if (page->text) {
566 fz_free_text_page (state.ctx, page->text);
568 if (page->sheet) {
569 fz_free_text_sheet (state.ctx, page->sheet);
571 if (page->slinks) {
572 free (page->slinks);
574 page->freepage (page->u.ptr);
575 fz_drop_display_list (state.ctx, page->dlist);
576 free (page);
579 static void freetile (struct tile *tile)
581 unlinktile (tile);
582 if (!tile->pbo) {
583 #ifndef PIGGYBACK
584 fz_drop_pixmap (state.ctx, tile->pixmap);
585 #else
586 if (state.pig) {
587 fz_drop_pixmap (state.ctx, state.pig);
589 state.pig = tile->pixmap;
590 #endif
592 else {
593 free (tile->pbo);
594 fz_drop_pixmap (state.ctx, tile->pixmap);
596 free (tile);
599 #ifdef __ALTIVEC__
600 #include <stdint.h>
601 #include <altivec.h>
603 static int cacheline32bytes;
605 static void __attribute__ ((constructor)) clcheck (void)
607 char **envp = environ;
608 unsigned long *auxv;
610 while (*envp++);
612 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
613 if (*auxv == 19) {
614 cacheline32bytes = auxv[1] == 32;
615 return;
620 static void OPTIMIZE_ATTR (3) clearpixmap (fz_pixmap *pixmap)
622 size_t size = pixmap->w * pixmap->h * pixmap->n;
623 if (cacheline32bytes && size > 32) {
624 intptr_t a1, a2, diff;
625 size_t sizea, i;
626 vector unsigned char v = vec_splat_u8 (-1);
627 vector unsigned char *p;
629 a1 = a2 = (intptr_t) pixmap->samples;
630 a2 = (a1 + 31) & ~31;
631 diff = a2 - a1;
632 sizea = size - diff;
633 p = (void *) a2;
635 while (a1 != a2) *(char *) a1++ = 0xff;
636 for (i = 0; i < (sizea & ~31); i += 32) {
637 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
638 vec_st (v, i, p);
639 vec_st (v, i + 16, p);
641 while (i < sizea) *((char *) a1 + i++) = 0xff;
643 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
645 #else
646 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
647 #endif
649 static void trimctm (pdf_page *page, int pindex)
651 fz_matrix ctm;
652 struct pagedim *pdim = &state.pagedims[pindex];
654 if (!pdim->tctmready) {
655 if (state.trimmargins) {
656 fz_rect realbox;
657 fz_matrix rm, sm, tm, im, ctm1;
659 fz_rotate (&rm, -pdim->rotate);
660 fz_scale (&sm, 1, -1);
661 fz_concat (&ctm, &rm, &sm);
662 realbox = pdim->mediabox;
663 fz_transform_rect (&realbox, &ctm);
664 fz_translate (&tm, -realbox.x0, -realbox.y0);
665 fz_concat (&ctm1, &ctm, &tm);
666 fz_invert_matrix (&im, &page->ctm);
667 fz_concat (&ctm, &im, &ctm1);
669 else {
670 ctm = fz_identity;
672 pdim->tctm = ctm;
673 pdim->tctmready = 1;
677 static fz_matrix pagectm (struct page *page)
679 fz_matrix ctm, tm;
681 if (page->type == DPDF) {
682 trimctm (page->u.pdfpage, page->pdimno);
683 fz_concat (&ctm,
684 &state.pagedims[page->pdimno].tctm,
685 &state.pagedims[page->pdimno].ctm);
687 else {
688 struct pagedim *pdim = &state.pagedims[page->pdimno];
690 fz_translate (&tm, -pdim->mediabox.x0, -pdim->mediabox.y0);
691 fz_concat (&ctm, &tm, &state.pagedims[page->pdimno].ctm);
693 return ctm;
696 static void *loadpage (int pageno, int pindex)
698 fz_device *dev;
699 struct page *page;
701 page = calloc (sizeof (struct page), 1);
702 if (!page) {
703 err (1, "calloc page %d", pageno);
706 page->dlist = fz_new_display_list (state.ctx);
707 dev = fz_new_list_device (state.ctx, page->dlist);
708 fz_try (state.ctx) {
709 switch (state.type) {
710 case DPDF:
711 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
712 pdf_run_page (state.u.pdf, page->u.pdfpage, dev,
713 &fz_identity, NULL);
714 page->freepage = freepdfpage;
715 break;
717 case DXPS:
718 page->u.xpspage = xps_load_page (state.u.xps, pageno);
719 xps_run_page (state.u.xps, page->u.xpspage, dev,
720 &fz_identity, NULL);
721 page->freepage = freexpspage;
722 break;
724 case DCBZ:
725 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
726 cbz_run_page (state.u.cbz, page->u.cbzpage, dev,
727 &fz_identity, NULL);
728 page->freepage = freecbzpage;
729 break;
732 fz_catch (state.ctx) {
733 page->u.ptr = NULL;
734 page->freepage = freeemptyxxxpage;
736 fz_free_device (dev);
738 page->pdimno = pindex;
739 page->pageno = pageno;
740 page->sgen = state.gen;
741 page->tgen = state.gen;
742 page->type = state.type;
744 return page;
747 static struct tile *alloctile (int h)
749 int i;
750 int slicecount;
751 size_t tilesize;
752 struct tile *tile;
754 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
755 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
756 tile = calloc (tilesize, 1);
757 if (!tile) {
758 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
760 for (i = 0; i < slicecount; ++i) {
761 int sh = MIN (h, state.sliceheight);
762 tile->slices[i].h = sh;
763 tile->slices[i].texindex = -1;
764 h -= sh;
766 tile->slicecount = slicecount;
767 tile->sliceheight = state.sliceheight;
768 return tile;
771 #ifdef OBSCURED_OPT
772 struct obs {
773 int cured;
774 fz_irect b;
777 static void obs_fill_image (fz_device *dev, fz_image *image,
778 const fz_matrix *ctm, float alpha)
780 struct obs *obs = dev->user;
782 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
783 fz_irect b;
784 fz_rect rect = fz_unit_rect;
786 fz_transform_rect (&rect, ctm);
787 fz_round_rect (&b, &rect);
788 fz_intersect_irect (&b, &obs->b);
789 obs->cured = b.x0 == obs->b.x0
790 && b.x1 == obs->b.x1
791 && b.y0 == obs->b.y0
792 && b.y1 == obs->b.y1;
796 static int obscured (struct page *page, fz_irect bbox)
798 fz_rect rect;
799 fz_matrix ctm;
800 fz_device dev;
801 struct obs obs;
803 memset (&dev, 0, sizeof (dev));
804 memset (&obs, 0, sizeof (obs));
805 dev.hints = 0;
806 dev.flags = 0;
807 dev.user = &obs;
808 dev.ctx = state.ctx;
809 dev.fill_image = obs_fill_image;
810 obs.b = bbox;
811 fz_rect_from_irect (&rect, &bbox);
812 ctm = pagectm (page);
813 fz_run_display_list (page->dlist, &dev, &ctm, &rect, NULL);
814 return obs.cured;
816 #define OBSCURED obscured
817 #else
818 #define OBSCURED(a, b) 0
819 #endif
821 static struct tile *rendertile (struct page *page, int x, int y, int w, int h,
822 struct pbo *pbo)
824 fz_rect rect;
825 fz_irect bbox;
826 fz_matrix ctm;
827 fz_device *dev;
828 struct tile *tile;
829 struct pagedim *pdim;
831 tile = alloctile (h);
832 pdim = &state.pagedims[page->pdimno];
834 bbox = pdim->bounds;
835 bbox.x0 += x;
836 bbox.y0 += y;
837 bbox.x1 = bbox.x0 + w;
838 bbox.y1 = bbox.y0 + h;
840 if (state.pig) {
841 if (state.pig->w == w
842 && state.pig->h == h
843 && state.pig->colorspace == state.colorspace) {
844 tile->pixmap = state.pig;
845 tile->pixmap->x = bbox.x0;
846 tile->pixmap->y = bbox.y0;
848 else {
849 fz_drop_pixmap (state.ctx, state.pig);
851 state.pig = NULL;
853 if (!tile->pixmap) {
854 if (pbo) {
855 tile->pixmap =
856 fz_new_pixmap_with_bbox_and_data (state.ctx, state.colorspace,
857 &bbox, pbo->ptr);
858 tile->pbo = pbo;
860 else {
861 tile->pixmap =
862 fz_new_pixmap_with_bbox (state.ctx, state.colorspace, &bbox);
866 tile->w = w;
867 tile->h = h;
868 if (!page->u.ptr || ((w < 128 && h < 128) || !OBSCURED (page, bbox))) {
869 clearpixmap (tile->pixmap);
871 dev = fz_new_draw_device (state.ctx, tile->pixmap);
872 ctm = pagectm (page);
873 fz_rect_from_irect (&rect, &bbox);
874 fz_run_display_list (page->dlist, dev, &ctm, &rect, NULL);
875 fz_free_device (dev);
877 return tile;
880 static void initpdims (void)
882 double start, end;
883 int pageno, trim, show;
884 FILE *trimf = NULL;
885 int trimw = 0, cxcount;
887 start = now ();
889 if (state.trimmargins && state.trimcachepath) {
890 trimf = fopen (state.trimcachepath, "rb");
891 if (!trimf) {
892 trimf = fopen (state.trimcachepath, "wb");
893 trimw = 1;
897 if (state.trimmargins || state.type == DPDF || !state.cxack)
898 cxcount = state.pagecount;
899 else
900 cxcount = MIN (state.pagecount, 1);
902 for (pageno = 0; pageno < cxcount; ++pageno) {
903 int rotate = 0;
904 struct pagedim *p;
905 fz_rect mediabox;
907 switch (state.type) {
908 case DPDF: {
909 pdf_obj *pageref = pdf_lookup_page_obj (state.u.pdf, pageno);
910 pdf_obj *pageobj = pdf_resolve_indirect (pageref);
912 if (state.trimmargins) {
913 pdf_obj *obj;
914 pdf_page *page;
916 fz_try (state.ctx) {
917 page = pdf_load_page (state.u.pdf, pageno);
918 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
919 trim = state.trimanew || !obj;
920 if (trim) {
921 fz_rect rect;
922 fz_matrix ctm;
923 fz_device *dev;
925 dev = fz_new_bbox_device (state.ctx, &rect);
926 dev->hints |= FZ_IGNORE_SHADE;
927 fz_invert_matrix (&ctm, &page->ctm);
928 pdf_run_page (state.u.pdf, page, dev,
929 &fz_identity, NULL);
930 fz_free_device (dev);
932 rect.x0 += state.trimfuzz.x0;
933 rect.x1 += state.trimfuzz.x1;
934 rect.y0 += state.trimfuzz.y0;
935 rect.y1 += state.trimfuzz.y1;
936 fz_transform_rect (&rect, &ctm);
937 fz_intersect_rect (&rect, &page->mediabox);
939 if (fz_is_empty_rect (&rect)) {
940 mediabox = page->mediabox;
942 else {
943 mediabox = rect;
946 obj = pdf_new_array (state.u.pdf, 4);
947 pdf_array_push (obj, pdf_new_real (state.u.pdf,
948 mediabox.x0));
949 pdf_array_push (obj, pdf_new_real (state.u.pdf,
950 mediabox.y0));
951 pdf_array_push (obj, pdf_new_real (state.u.pdf,
952 mediabox.x1));
953 pdf_array_push (obj, pdf_new_real (state.u.pdf,
954 mediabox.y1));
955 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
957 else {
958 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
959 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
960 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
961 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
964 rotate = page->rotate;
965 pdf_free_page (state.u.pdf, page);
967 show = trim ? pageno % 5 == 0 : pageno % 20 == 0;
968 if (show) {
969 printd ("progress %f Trimming %d",
970 (double) (pageno + 1) / state.pagecount,
971 pageno + 1);
974 fz_catch (state.ctx) {
975 fprintf (stderr, "failed to load page %d\n", pageno+1);
978 else {
979 fz_rect cropbox;
981 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "MediaBox"),
982 &mediabox);
983 if (fz_is_empty_rect (&mediabox)) {
984 fprintf (stderr, "cannot find page size for page %d\n",
985 pageno+1);
986 mediabox.x0 = 0;
987 mediabox.y0 = 0;
988 mediabox.x1 = 612;
989 mediabox.y1 = 792;
992 pdf_to_rect (state.ctx, pdf_dict_gets (pageobj, "CropBox"),
993 &cropbox);
994 if (!fz_is_empty_rect (&cropbox)) {
995 fz_intersect_rect (&mediabox, &cropbox);
997 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
999 break;
1002 case DXPS:
1004 xps_page *page;
1006 fz_try (state.ctx) {
1007 page = xps_load_page (state.u.xps, pageno);
1008 xps_bound_page (state.u.xps, page, &mediabox);
1009 rotate = 0;
1010 if (state.trimmargins) {
1011 fz_rect rect;
1012 fz_device *dev;
1014 dev = fz_new_bbox_device (state.ctx, &rect);
1015 dev->hints |= FZ_IGNORE_SHADE;
1016 xps_run_page (state.u.xps, page, dev,
1017 &fz_identity, NULL);
1018 fz_free_device (dev);
1020 rect.x0 += state.trimfuzz.x0;
1021 rect.x1 += state.trimfuzz.x1;
1022 rect.y0 += state.trimfuzz.y0;
1023 rect.y1 += state.trimfuzz.y1;
1024 fz_intersect_rect (&rect, &mediabox);
1026 if (!fz_is_empty_rect (&rect)) {
1027 mediabox = rect;
1030 xps_free_page (state.u.xps, page);
1031 if (!state.cxack) {
1032 printd ("progress %f loading %d",
1033 (double) (pageno + 1) / state.pagecount,
1034 pageno + 1);
1037 fz_catch (state.ctx) {
1040 break;
1042 case DCBZ:
1044 rotate = 0;
1045 mediabox.x0 = 0;
1046 mediabox.y0 = 0;
1047 mediabox.x1 = 900;
1048 mediabox.y1 = 900;
1049 if (state.trimmargins && trimw) {
1050 cbz_page *page;
1052 fz_try (state.ctx) {
1053 page = cbz_load_page (state.u.cbz, pageno);
1054 cbz_bound_page (state.u.cbz, page, &mediabox);
1055 cbz_free_page (state.u.cbz, page);
1056 printd ("progress %f Trimming %d",
1057 (double) (pageno + 1) / state.pagecount,
1058 pageno + 1);
1060 fz_catch (state.ctx) {
1061 fprintf (stderr, "failed to load page %d\n", pageno+1);
1063 if (trimf) {
1064 int n = fwrite (&mediabox, sizeof (mediabox), 1, trimf);
1065 if (n - 1) {
1066 err (1, "fwrite trim mediabox");
1070 else {
1071 if (trimf) {
1072 int n = fread (&mediabox, sizeof (mediabox), 1, trimf);
1073 if (n - 1) {
1074 err (1, "fread trim mediabox");
1077 else {
1078 if (state.cxack) {
1079 cbz_page *page;
1080 fz_try (state.ctx) {
1081 page = cbz_load_page (state.u.cbz, pageno);
1082 cbz_bound_page (state.u.cbz, page, &mediabox);
1083 cbz_free_page (state.u.cbz, page);
1085 fz_catch (state.ctx) {
1086 fprintf (stderr, "failed to load cbz page\n");
1092 break;
1094 default:
1095 ARSERT (0 && state.type);
1098 if (state.pagedimcount == 0
1099 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
1100 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
1101 size_t size;
1103 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
1104 state.pagedims = realloc (state.pagedims, size);
1105 if (!state.pagedims) {
1106 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
1107 size, state.pagedimcount + 1);
1110 p = &state.pagedims[state.pagedimcount++];
1111 p->rotate = rotate;
1112 p->mediabox = mediabox;
1113 p->pageno = pageno;
1116 end = now ();
1117 if (state.trimmargins) {
1118 printd ("progress 1 Trimmed %d pages in %f seconds",
1119 state.pagecount, end - start);
1121 else {
1122 printd ("vmsg Processed %d pages in %f seconds",
1123 state.pagecount, end - start);
1125 state.trimanew = 0;
1126 if (trimf) {
1127 if (fclose (trimf)) {
1128 err (1, "fclose");
1133 static void layout (void)
1135 int pindex;
1136 fz_rect box;
1137 fz_matrix ctm, rm;
1138 struct pagedim *p = p;
1139 double zw, w, maxw = 0.0, zoom = zoom;
1141 if (state.pagedimcount == 0) return;
1143 switch (state.fitmodel) {
1144 case FitProportional:
1145 for (pindex = 0; pindex < state.pagedimcount; ++pindex) {
1146 double x0, x1;
1148 p = &state.pagedims[pindex];
1149 fz_rotate (&rm, p->rotate + state.rotate);
1150 box = p->mediabox;
1151 fz_transform_rect (&box, &rm);
1153 x0 = MIN (box.x0, box.x1);
1154 x1 = MAX (box.x0, box.x1);
1156 w = x1 - x0;
1157 maxw = MAX (w, maxw);
1158 zoom = state.w / maxw;
1160 break;
1162 case FitPage:
1163 maxw = state.w;
1164 break;
1166 case FitWidth:
1167 break;
1169 default:
1170 ARSERT (0 && state.fitmodel);
1173 for (pindex = 0; pindex < state.pagedimcount; ++pindex) {
1174 fz_rect rect;
1175 fz_matrix tm, sm;
1177 p = &state.pagedims[pindex];
1178 fz_rotate (&ctm, state.rotate);
1179 fz_rotate (&rm, p->rotate + state.rotate);
1180 box = p->mediabox;
1181 fz_transform_rect (&box, &rm);
1182 w = box.x1 - box.x0;
1183 switch (state.fitmodel) {
1184 case FitProportional:
1185 p->left = ((maxw - w) * zoom) / 2.0;
1186 break;
1187 case FitPage:
1189 double zh, h;
1190 zw = maxw / w;
1191 h = box.y1 - box.y0;
1192 zh = state.h / h;
1193 zoom = MIN (zw, zh);
1194 p->left = (maxw - (w * zoom)) / 2.0;
1196 break;
1197 case FitWidth:
1198 p->left = 0;
1199 zoom = state.w / w;
1200 break;
1203 fz_scale (&p->zoomctm, zoom, zoom);
1204 fz_concat (&ctm, &p->zoomctm, &ctm);
1206 fz_rotate (&rm, p->rotate);
1207 p->pagebox = p->mediabox;
1208 fz_transform_rect (&p->pagebox, &rm);
1209 p->pagebox.x1 -= p->pagebox.x0;
1210 p->pagebox.y1 -= p->pagebox.y0;
1211 p->pagebox.x0 = 0;
1212 p->pagebox.y0 = 0;
1213 rect = p->pagebox;
1214 fz_transform_rect (&rect, &ctm);
1215 fz_round_rect (&p->bounds, &rect);
1216 p->ctm = ctm;
1218 fz_translate (&tm, 0, -p->mediabox.y1);
1219 fz_scale (&sm, zoom, -zoom);
1220 fz_concat (&ctm, &tm, &sm);
1221 fz_concat (&p->lctm, &ctm, &rm);
1223 p->tctmready = 0;
1226 do {
1227 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1228 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1229 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1230 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1231 int w = x1 - x0;
1232 int h = y1 - y0;
1234 printd ("pdim %d %d %d %d", p->pageno, w, h, p->left);
1235 } while (p-- != state.pagedims);
1238 static
1239 struct anchor { int n; int x; int y; int w; int h; }
1240 desttoanchor (fz_link_dest *dest)
1242 int i;
1243 struct anchor a;
1244 struct pagedim *pdim = state.pagedims;
1246 a.n = -1;
1247 a.x = 0;
1248 a.y = 0;
1249 for (i = 0; i < state.pagedimcount; ++i) {
1250 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1251 break;
1252 pdim = &state.pagedims[i];
1254 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1255 fz_point p;
1256 if (dest->ld.gotor.flags & fz_link_flag_l_valid)
1257 p.x = dest->ld.gotor.lt.x;
1258 else
1259 p.x = 0.0;
1260 p.y = dest->ld.gotor.lt.y;
1261 fz_transform_point (&p, &pdim->lctm);
1262 a.y = p.y;
1263 a.x = p.x;
1265 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1266 double x0, x1, y0, y1;
1268 x0 = MIN (pdim->bounds.x0, pdim->bounds.x1);
1269 x1 = MAX (pdim->bounds.x0, pdim->bounds.x1);
1270 a.w = x1 - x0;
1271 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1272 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1273 a.h = y1 - y0;
1274 a.n = dest->ld.gotor.page;
1276 return a;
1279 static void recurse_outline (fz_outline *outline, int level)
1281 while (outline) {
1282 if (outline->dest.kind == FZ_LINK_GOTO) {
1283 struct anchor a = desttoanchor (&outline->dest);
1285 if (a.n >= 0) {
1286 printd ("o %d %d %d %d %s",
1287 level, a.n, a.y, a.h, outline->title);
1290 else {
1291 printd ("emsg Unhandled outline kind %d for %s\n",
1292 outline->dest.kind, outline->title);
1294 if (outline->down) {
1295 recurse_outline (outline->down, level + 1);
1297 outline = outline->next;
1301 static void process_outline (void)
1303 fz_outline *outline;
1305 if (!state.needoutline) return;
1307 state.needoutline = 0;
1308 switch (state.type) {
1309 case DPDF:
1310 outline = pdf_load_outline (state.u.pdf);
1311 break;
1312 case DXPS:
1313 outline = xps_load_outline (state.u.xps);
1314 break;
1315 default:
1316 outline = NULL;
1317 break;
1319 if (outline) {
1320 recurse_outline (outline, 0);
1321 fz_free_outline (state.ctx, outline);
1325 static char *strofspan (fz_text_span *span)
1327 char *p;
1328 char utf8[10];
1329 fz_text_char *ch;
1330 size_t size = 0, cap = 80;
1332 p = malloc (cap + 1);
1333 if (!p) return NULL;
1335 for (ch = span->text; ch < span->text + span->len; ++ch) {
1336 int n = fz_runetochar (utf8, ch->c);
1337 if (size + n > cap) {
1338 cap *= 2;
1339 p = realloc (p, cap + 1);
1340 if (!p) return NULL;
1343 memcpy (p + size, utf8, n);
1344 size += n;
1346 p[size] = 0;
1347 return p;
1350 static int matchspan (regex_t *re, fz_text_page *page,
1351 fz_text_span *span, fz_matrix ctm,
1352 int stop, int pageno, double start)
1354 int ret;
1355 char *p;
1356 regmatch_t rm;
1357 int a, b, c;
1358 fz_rect sb, eb;
1359 fz_point p1, p2, p3, p4;
1361 p = strofspan (span);
1362 if (!p) return -1;
1364 ret = regexec (re, p, 1, &rm, 0);
1365 if (ret) {
1366 free (p);
1367 if (ret != REG_NOMATCH) {
1368 size_t size;
1369 char errbuf[80];
1370 size = regerror (ret, re, errbuf, sizeof (errbuf));
1371 printd ("msg regexec error `%.*s'",
1372 (int) size, errbuf);
1373 return -1;
1375 return 0;
1377 else {
1378 int l = span->len;
1380 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1381 c += fz_runelen (span->text[a].c);
1383 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1384 c += fz_runelen (span->text[b].c);
1387 if (fz_runelen (span->text[b].c) > 1) {
1388 b = MAX (0, b-1);
1391 fz_text_char_bbox (&sb, span, a);
1392 fz_text_char_bbox (&eb, span, b);
1394 p1.x = sb.x0;
1395 p1.y = sb.y0;
1396 p2.x = eb.x1;
1397 p2.y = sb.y0;
1398 p3.x = eb.x1;
1399 p3.y = eb.y1;
1400 p4.x = sb.x0;
1401 p4.y = eb.y1;
1403 if (!stop) {
1404 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1405 pageno, 1,
1406 p1.x, p1.y,
1407 p2.x, p2.y,
1408 p3.x, p3.y,
1409 p4.x, p4.y);
1411 printd ("progress 1 found at %d `%.*s' in %f sec",
1412 pageno + 1, (int) (rm.rm_eo - rm.rm_so), &p[rm.rm_so],
1413 now () - start);
1415 else {
1416 printd ("match %d %d %f %f %f %f %f %f %f %f",
1417 pageno, 2,
1418 p1.x, p1.y,
1419 p2.x, p2.y,
1420 p3.x, p3.y,
1421 p4.x, p4.y);
1423 free (p);
1424 return 1;
1428 static int compareblocks (const void *l, const void *r)
1430 fz_text_block const *ls = l;
1431 fz_text_block const* rs = r;
1432 return ls->bbox.y0 - rs->bbox.y0;
1435 /* wishful thinking function */
1436 static void search (regex_t *re, int pageno, int y, int forward)
1438 int i, j;
1439 fz_matrix ctm;
1440 fz_device *tdev;
1441 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1442 fz_text_page *text;
1443 fz_text_sheet *sheet;
1444 struct pagedim *pdim, *pdimprev;
1445 int stop = 0, niters = 0;
1446 double start, end;
1448 if (!(state.type == DPDF || state.type == DXPS))
1449 return;
1451 start = now ();
1452 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1453 if (niters++ == 5) {
1454 niters = 0;
1455 if (hasdata ()) {
1456 printd ("progress 1 attention requested aborting search at %d",
1457 pageno);
1458 stop = 1;
1460 else {
1461 printd ("progress %f searching in page %d",
1462 (double) (pageno + 1) / state.pagecount,
1463 pageno);
1466 pdimprev = NULL;
1467 for (i = 0; i < state.pagedimcount; ++i) {
1468 pdim = &state.pagedims[i];
1469 if (pdim->pageno == pageno) {
1470 goto found;
1472 if (pdim->pageno > pageno) {
1473 pdim = pdimprev;
1474 goto found;
1476 pdimprev = pdim;
1478 pdim = pdimprev;
1479 found:
1481 sheet = fz_new_text_sheet (state.ctx);
1482 text = fz_new_text_page (state.ctx);
1483 tdev = fz_new_text_device (state.ctx, sheet, text);
1485 switch (state.type) {
1486 case DPDF:
1487 u.ptr = NULL;
1488 fz_try (state.ctx) {
1489 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1490 trimctm (u.pdfpage, pdim - state.pagedims);
1491 fz_concat (&ctm, &pdim->tctm, &pdim->zoomctm);
1492 fz_begin_page (tdev, &fz_infinite_rect, &ctm);
1493 pdf_run_page (state.u.pdf, u.pdfpage, tdev, &ctm, NULL);
1494 fz_end_page (tdev);
1496 fz_catch (state.ctx) {
1497 fz_free_device (tdev);
1498 u.ptr = NULL;
1499 goto nextiter;
1501 break;
1503 case DXPS:
1504 u.xpspage = xps_load_page (state.u.xps, pageno);
1505 xps_run_page (state.u.xps, u.xpspage, tdev, &pdim->ctm, NULL);
1506 break;
1508 default:
1509 ARSERT (0 && state.type);
1512 qsort (text->blocks, text->len, sizeof (*text->blocks), compareblocks);
1513 fz_free_device (tdev);
1515 for (j = 0; j < text->len; ++j) {
1516 int k;
1517 fz_page_block *pb;
1518 fz_text_block *block;
1520 pb = &text->blocks[forward ? j : text->len - 1 - j];
1521 if (pb->type != FZ_PAGE_BLOCK_TEXT) continue;
1522 block = pb->u.text;
1524 for (k = 0; k < block->len; ++k) {
1525 fz_text_line *line;
1526 fz_text_span *span;
1528 if (forward) {
1529 line = &block->lines[k];
1530 if (line->bbox.y0 < y + 1) continue;
1532 else {
1533 line = &block->lines[block->len - 1 - k];
1534 if (line->bbox.y0 > y - 1) continue;
1537 for (span = line->first_span; span; span = span->next) {
1538 switch (matchspan (re, text, span, ctm,
1539 stop, pageno, start)) {
1540 case 0: break;
1541 case 1: stop = 1; break;
1542 case -1: stop = 1; goto endloop;
1547 nextiter:
1548 if (forward) {
1549 pageno += 1;
1550 y = 0;
1552 else {
1553 pageno -= 1;
1554 y = INT_MAX;
1556 endloop:
1557 fz_free_text_page (state.ctx, text);
1558 fz_free_text_sheet (state.ctx, sheet);
1559 if (u.ptr) {
1560 state.freepage (u.ptr);
1563 end = now ();
1564 if (!stop) {
1565 printd ("progress 1 no matches %f sec", end - start);
1567 printd ("clearrects");
1570 static void set_tex_params (int colorspace)
1572 union {
1573 unsigned char b;
1574 unsigned int s;
1575 } endianness = {1};
1577 switch (colorspace) {
1578 case 0:
1579 state.texiform = GL_RGBA8;
1580 state.texform = GL_RGBA;
1581 state.texty = GL_UNSIGNED_BYTE;
1582 state.colorspace = fz_device_rgb (state.ctx);
1583 break;
1584 case 1:
1585 state.texiform = GL_RGBA8;
1586 state.texform = GL_BGRA;
1587 state.texty = endianness.s > 1
1588 ? GL_UNSIGNED_INT_8_8_8_8
1589 : GL_UNSIGNED_INT_8_8_8_8_REV;
1590 state.colorspace = fz_device_bgr (state.ctx);
1591 break;
1592 case 2:
1593 state.texiform = GL_LUMINANCE_ALPHA;
1594 state.texform = GL_LUMINANCE_ALPHA;
1595 state.texty = GL_UNSIGNED_BYTE;
1596 state.colorspace = fz_device_gray (state.ctx);
1597 break;
1598 default:
1599 errx (1, "invalid colorspce %d", colorspace);
1603 static void realloctexts (int texcount)
1605 size_t size;
1607 if (texcount == state.texcount) return;
1609 if (texcount < state.texcount) {
1610 glDeleteTextures (state.texcount - texcount,
1611 state.texids + texcount);
1614 size = texcount * sizeof (*state.texids);
1615 state.texids = realloc (state.texids, size);
1616 if (!state.texids) {
1617 err (1, "realloc texids %" FMT_s, size);
1620 size = texcount * sizeof (*state.texowners);
1621 state.texowners = realloc (state.texowners, size);
1622 if (!state.texowners) {
1623 err (1, "realloc texowners %" FMT_s, size);
1625 if (texcount > state.texcount) {
1626 int i;
1628 glGenTextures (texcount - state.texcount,
1629 state.texids + state.texcount);
1630 for (i = state.texcount; i < texcount; ++i) {
1631 state.texowners[i].w = -1;
1632 state.texowners[i].slice = NULL;
1635 state.texcount = texcount;
1636 state.texindex = 0;
1639 static char *mbtoutf8 (char *s)
1641 char *p, *r;
1642 wchar_t *tmp;
1643 size_t i, ret, len;
1645 len = mbstowcs (NULL, s, strlen (s));
1646 if (len == 0) {
1647 return s;
1649 else {
1650 if (len == (size_t) -1) {
1651 return s;
1655 tmp = malloc (len * sizeof (wchar_t));
1656 if (!tmp) {
1657 return s;
1660 ret = mbstowcs (tmp, s, len);
1661 if (ret == (size_t) -1) {
1662 free (tmp);
1663 return s;
1666 len = 0;
1667 for (i = 0; i < ret; ++i) {
1668 len += fz_runelen (tmp[i]);
1671 p = r = malloc (len + 1);
1672 if (!r) {
1673 free (tmp);
1674 return s;
1677 for (i = 0; i < ret; ++i) {
1678 p += fz_runetochar (p, tmp[i]);
1680 *p = 0;
1681 free (tmp);
1682 return r;
1685 CAMLprim value ml_mbtoutf8 (value s_v)
1687 CAMLparam1 (s_v);
1688 CAMLlocal1 (ret_v);
1689 char *s, *r;
1691 s = String_val (s_v);
1692 r = mbtoutf8 (s);
1693 if (r == s) {
1694 ret_v = s_v;
1696 else {
1697 ret_v = caml_copy_string (r);
1698 free (r);
1700 CAMLreturn (ret_v);
1703 static void * mainloop (void *unused)
1705 char *p = NULL;
1706 int len, ret, oldlen = 0;
1708 for (;;) {
1709 len = readlen ();
1710 if (len == 0) {
1711 errx (1, "readlen returned 0");
1714 if (oldlen < len + 1) {
1715 p = realloc (p, len + 1);
1716 if (!p) {
1717 err (1, "realloc %d failed", len + 1);
1719 oldlen = len + 1;
1721 readdata (p, len);
1722 p[len] = 0;
1724 if (!strncmp ("open", p, 4)) {
1725 int wthack, off;
1726 char *password;
1727 char *filename;
1728 char *utf8filename;
1729 size_t filenamelen;
1731 ret = sscanf (p + 5, " %d %d %n", &wthack, &state.cxack, &off);
1732 if (ret != 2) {
1733 errx (1, "malformed open `%.*s' ret=%d", len, p, ret);
1736 filename = p + 5 + off;
1737 filenamelen = strlen (filename);
1738 password = filename + filenamelen + 1;
1740 lock ("open");
1741 openxref (filename, password);
1742 pdfinfo ();
1743 initpdims ();
1744 unlock ("open");
1746 if (!wthack) {
1747 utf8filename = mbtoutf8 (filename);
1748 printd ("msg Opened %s (press h/F1 to get help)", utf8filename);
1749 if (utf8filename != filename) {
1750 free (utf8filename);
1753 state.needoutline = 1;
1755 else if (!strncmp ("cs", p, 2)) {
1756 int i, colorspace;
1758 ret = sscanf (p + 2, " %d", &colorspace);
1759 if (ret != 1) {
1760 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
1762 lock ("cs");
1763 set_tex_params (colorspace);
1764 for (i = 0; i < state.texcount; ++i) {
1765 state.texowners[i].w = -1;
1766 state.texowners[i].slice = NULL;
1768 unlock ("cs");
1770 else if (!strncmp ("freepage", p, 8)) {
1771 void *ptr;
1773 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1774 if (ret != 1) {
1775 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
1777 freepage (ptr);
1779 else if (!strncmp ("freetile", p, 8)) {
1780 void *ptr;
1782 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1783 if (ret != 1) {
1784 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
1786 freetile (ptr);
1788 else if (!strncmp ("search", p, 6)) {
1789 int icase, pageno, y, ret, len2, forward;
1790 char *pattern;
1791 regex_t re;
1793 ret = sscanf (p + 6, " %d %d %d %d,%n",
1794 &icase, &pageno, &y, &forward, &len2);
1795 if (ret != 4) {
1796 errx (1, "malformed search `%s' ret=%d", p, ret);
1799 pattern = p + 6 + len2;
1800 ret = regcomp (&re, pattern,
1801 REG_EXTENDED | (icase ? REG_ICASE : 0));
1802 if (ret) {
1803 char errbuf[80];
1804 size_t size;
1806 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1807 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
1809 else {
1810 search (&re, pageno, y, forward);
1811 regfree (&re);
1814 else if (!strncmp ("geometry", p, 8)) {
1815 int w, h, fitmodel;
1817 printd ("clear");
1818 ret = sscanf (p + 8, " %d %d %d", &w, &h, &fitmodel);
1819 if (ret != 3) {
1820 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1823 lock ("geometry");
1824 state.h = h;
1825 if (w != state.w) {
1826 int i;
1827 state.w = w;
1828 for (i = 0; i < state.texcount; ++i) {
1829 state.texowners[i].slice = NULL;
1832 state.fitmodel = fitmodel;
1833 layout ();
1834 process_outline ();
1836 state.gen++;
1837 unlock ("geometry");
1838 printd ("continue %d", state.pagecount);
1840 else if (!strncmp ("reqlayout", p, 9)) {
1841 char *nameddest;
1842 int rotate, fitmodel, off, h;
1844 printd ("clear");
1845 ret = sscanf (p + 9, " %d %d %d %n",
1846 &rotate, &fitmodel, &h, &off);
1847 if (ret != 3) {
1848 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
1850 lock ("reqlayout");
1851 if (state.rotate != rotate || state.fitmodel != fitmodel) {
1852 state.gen += 1;
1854 state.rotate = rotate;
1855 state.fitmodel = fitmodel;
1856 state.h = h;
1857 layout ();
1858 process_outline ();
1860 nameddest = p + 9 + off;
1861 if (state.type == DPDF && nameddest && *nameddest) {
1862 struct anchor a;
1863 fz_link_dest dest;
1864 pdf_obj *needle, *obj;
1866 needle = pdf_new_string (state.u.pdf, nameddest,
1867 strlen (nameddest));
1868 obj = pdf_lookup_dest (state.u.pdf, needle);
1869 if (obj) {
1870 dest = pdf_parse_link_dest (state.u.pdf, FZ_LINK_GOTO, obj);
1872 a = desttoanchor (&dest);
1873 if (a.n >= 0) {
1874 printd ("a %d %d %d", a.n, a.x, a.y);
1876 else {
1877 printd ("emsg failed to parse destination `%s'\n",
1878 nameddest);
1881 else {
1882 printd ("emsg destination `%s' not found\n",
1883 nameddest);
1885 pdf_drop_obj (needle);
1888 state.gen++;
1889 unlock ("reqlayout");
1890 printd ("continue %d", state.pagecount);
1892 else if (!strncmp ("page", p, 4)) {
1893 double a, b;
1894 struct page *page;
1895 int pageno, pindex, ret;
1897 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
1898 if (ret != 2) {
1899 errx (1, "bad page line `%.*s' ret=%d", len, p, ret);
1902 lock ("page");
1903 a = now ();
1904 page = loadpage (pageno, pindex);
1905 b = now ();
1906 unlock ("page");
1908 printd ("page %" FMT_ptr " %f", FMT_ptr_cast2 (page), b - a);
1910 else if (!strncmp ("tile", p, 4)) {
1911 int x, y, w, h, ret;
1912 struct page *page;
1913 struct tile *tile;
1914 double a, b;
1915 void *data;
1917 ret = sscanf (p + 4, " %" FMT_ptr " %d %d %d %d %" FMT_ptr,
1918 FMT_ptr_cast (&page), &x, &y, &w, &h, &data);
1919 if (ret != 6) {
1920 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
1923 lock ("tile");
1924 a = now ();
1925 tile = rendertile (page, x, y, w, h, data);
1926 b = now ();
1927 unlock ("tile");
1929 printd ("tile %d %d %" FMT_ptr " %u %f",
1930 x, y,
1931 FMT_ptr_cast2 (tile),
1932 tile->w * tile->h * tile->pixmap->n,
1933 b - a);
1935 else if (!strncmp ("settrim", p, 7)) {
1936 fz_irect fuzz;
1937 int trimmargins;
1939 ret = sscanf (p + 7, " %d %d %d %d %d",
1940 &trimmargins, &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
1941 if (ret != 5) {
1942 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
1944 printd ("clear");
1945 lock ("settrim");
1946 state.trimmargins = trimmargins;
1947 state.needoutline = 1;
1948 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
1949 state.trimanew = 1;
1950 state.trimfuzz = fuzz;
1952 state.pagedimcount = 0;
1953 free (state.pagedims);
1954 state.pagedims = NULL;
1955 initpdims ();
1956 layout ();
1957 process_outline ();
1958 unlock ("settrim");
1959 printd ("continue %d", state.pagecount);
1961 else if (!strncmp ("sliceh", p, 6)) {
1962 int h;
1964 ret = sscanf (p + 6, " %d", &h);
1965 if (ret != 1) {
1966 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
1968 if (h != state.sliceheight) {
1969 int i;
1971 state.sliceheight = h;
1972 for (i = 0; i < state.texcount; ++i) {
1973 state.texowners[i].w = -1;
1974 state.texowners[i].h = -1;
1975 state.texowners[i].slice = NULL;
1979 else if (!strncmp ("interrupt", p, 9)) {
1980 printd ("vmsg interrupted");
1982 else {
1983 errx (1, "unknown command %.*s", len, p);
1986 return 0;
1989 CAMLprim value ml_realloctexts (value texcount_v)
1991 CAMLparam1 (texcount_v);
1992 int ok;
1994 if (trylock ("ml_realloctexts")) {
1995 ok = 0;
1996 goto done;
1998 realloctexts (Int_val (texcount_v));
1999 ok = 1;
2000 unlock ("ml_realloctexts");
2002 done:
2003 CAMLreturn (Val_bool (ok));
2006 static void showsel (struct page *page, int ox, int oy)
2008 int seen = 0;
2009 fz_irect bbox;
2010 fz_rect rect;
2011 fz_text_line *line;
2012 fz_page_block *pageb;
2013 fz_text_block *block;
2014 struct mark first, last;
2016 first = page->fmark;
2017 last = page->lmark;
2019 if (!first.span || !last.span) return;
2021 glEnable (GL_BLEND);
2022 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
2023 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
2025 ox += state.pagedims[page->pdimno].bounds.x0;
2026 oy += state.pagedims[page->pdimno].bounds.y0;
2027 for (pageb = page->text->blocks;
2028 pageb < page->text->blocks + page->text->len;
2029 ++pageb) {
2030 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
2031 block = pageb->u.text;
2033 for (line = block->lines;
2034 line < block->lines + block->len;
2035 ++line) {
2036 fz_text_span *span;
2037 rect = fz_empty_rect;
2039 for (span = line->first_span; span; span = span->next) {
2040 int i, j, k;
2041 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
2043 j = 0;
2044 k = span->len - 1;
2046 if (span == page->fmark.span && span == page->lmark.span) {
2047 seen = 1;
2048 j = MIN (first.i, last.i);
2049 k = MAX (first.i, last.i);
2051 else if (span == first.span) {
2052 seen = 1;
2053 j = first.i;
2055 else if (span == last.span) {
2056 seen = 1;
2057 k = last.i;
2060 if (seen) {
2061 for (i = j; i <= k; ++i) {
2062 fz_rect bbox;
2063 fz_union_rect (&rect,
2064 fz_text_char_bbox (&bbox, span, i));
2066 fz_round_rect (&bbox, &rect);
2067 lprintf ("%d %d %d %d oy=%d ox=%d\n",
2068 bbox.x0,
2069 bbox.y0,
2070 bbox.x1,
2071 bbox.y1,
2072 oy, ox);
2074 glRecti (bbox.x0 + ox, bbox.y0 + oy,
2075 bbox.x1 + ox, bbox.y1 + oy);
2076 if (span == last.span) {
2077 goto done;
2083 done:
2084 glDisable (GL_BLEND);
2087 #include "glfont.c"
2089 static void highlightlinks (struct page *page, int xoff, int yoff)
2091 fz_matrix ctm, tm, pm;
2092 fz_link *link, *links;
2094 switch (page->type) {
2095 case DPDF:
2096 links = page->u.pdfpage->links;
2097 break;
2099 case DXPS:
2100 links = page->u.xpspage->links;
2101 break;
2103 default:
2104 return;
2107 glEnable (GL_TEXTURE_1D);
2108 glEnable (GL_BLEND);
2109 glBindTexture (GL_TEXTURE_1D, state.stid);
2111 xoff -= state.pagedims[page->pdimno].bounds.x0;
2112 yoff -= state.pagedims[page->pdimno].bounds.y0;
2113 fz_translate (&tm, xoff, yoff);
2114 pm = pagectm (page);
2115 fz_concat (&ctm, &pm, &tm);
2117 glBegin (GL_LINES);
2118 for (link = links; link; link = link->next) {
2119 fz_point p1, p2, p3, p4;
2121 p1.x = link->rect.x0;
2122 p1.y = link->rect.y0;
2124 p2.x = link->rect.x1;
2125 p2.y = link->rect.y0;
2127 p3.x = link->rect.x1;
2128 p3.y = link->rect.y1;
2130 p4.x = link->rect.x0;
2131 p4.y = link->rect.y1;
2133 fz_transform_point (&p1, &ctm);
2134 fz_transform_point (&p2, &ctm);
2135 fz_transform_point (&p3, &ctm);
2136 fz_transform_point (&p4, &ctm);
2138 switch (link->dest.kind) {
2139 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
2140 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
2141 case FZ_LINK_LAUNCH: glColor3ub (0, 255, 0); break;
2142 default: glColor3ub (0, 0, 0); break;
2146 float w, h, s, t;
2148 w = p2.x - p1.x;
2149 h = p2.y - p1.y;
2150 t = sqrtf (w*w + h*h) * .25f;
2152 w = p3.x - p2.x;
2153 h = p3.y - p2.y;
2154 s = sqrtf (w*w + h*h) * .25f;
2156 glTexCoord1i (0);
2157 glVertex2f (p1.x, p1.y);
2158 glTexCoord1f (t);
2159 glVertex2f (p2.x, p2.y);
2161 glTexCoord1i (0);
2162 glVertex2f (p2.x, p2.y);
2163 glTexCoord1f (s);
2164 glVertex2f (p3.x, p3.y);
2166 glTexCoord1i (0);
2167 glVertex2f (p3.x, p3.y);
2168 glTexCoord1f (t);
2169 glVertex2f (p4.x, p4.y);
2171 glTexCoord1i (0);
2172 glVertex2f (p4.x, p4.y);
2173 glTexCoord1f (s);
2174 glVertex2f (p1.x, p1.y);
2177 glEnd ();
2179 glDisable (GL_BLEND);
2180 glDisable (GL_TEXTURE_1D);
2183 static int compareslinks (const void *l, const void *r)
2185 struct slink const *ls = l;
2186 struct slink const *rs = r;
2187 if (ls->bbox.y0 == rs->bbox.y0) {
2188 return rs->bbox.x0 - rs->bbox.x0;
2190 return ls->bbox.y0 - rs->bbox.y0;
2193 static void droptext (struct page *page)
2195 if (page->text) {
2196 fz_free_text_page (state.ctx, page->text);
2197 page->fmark.i = -1;
2198 page->lmark.i = -1;
2199 page->fmark.span = NULL;
2200 page->lmark.span = NULL;
2201 page->text = NULL;
2203 if (page->sheet) {
2204 fz_free_text_sheet (state.ctx, page->sheet);
2205 page->sheet = NULL;
2209 static void dropslinks (struct page *page)
2211 if (page->slinks) {
2212 free (page->slinks);
2213 page->slinks = NULL;
2214 page->slinkcount = 0;
2218 static void ensureslinks (struct page *page)
2220 fz_matrix ctm;
2221 int i, count = 0;
2222 size_t slinksize = sizeof (*page->slinks);
2223 fz_link *link, *links;
2225 if (state.gen != page->sgen) {
2226 dropslinks (page);
2227 page->sgen = state.gen;
2229 if (page->slinks) return;
2231 switch (page->type) {
2232 case DPDF:
2233 links = page->u.pdfpage->links;
2234 trimctm (page->u.pdfpage, page->pdimno);
2235 fz_concat (&ctm,
2236 &state.pagedims[page->pdimno].tctm,
2237 &state.pagedims[page->pdimno].ctm);
2238 break;
2240 case DXPS:
2241 links = page->u.xpspage->links;
2242 ctm = state.pagedims[page->pdimno].ctm;
2243 break;
2245 default:
2246 return;
2249 for (link = links; link; link = link->next) {
2250 count++;
2252 if (count > 0) {
2253 page->slinkcount = count;
2254 page->slinks = calloc (count, slinksize);
2255 if (!page->slinks) {
2256 err (1, "realloc slinks %d", count);
2259 for (i = 0, link = links; link; ++i, link = link->next) {
2260 fz_rect rect;
2262 rect = link->rect;
2263 fz_transform_rect (&rect, &ctm);
2264 page->slinks[i].link = link;
2265 fz_round_rect (&page->slinks[i].bbox, &rect);
2267 qsort (page->slinks, count, slinksize, compareslinks);
2271 /* slightly tweaked fmt_ulong by D.J. Bernstein */
2272 static void fmt_linkn (char *s, unsigned int u)
2274 unsigned int len; unsigned int q;
2275 int zma = 'z' - 'a' + 1;
2276 len = 1; q = u;
2277 while (q > zma - 1) { ++len; q /= zma; }
2278 if (s) {
2279 s += len;
2280 do { *--s = 'a' + (u % zma) - (u < zma && len > 1); u /= zma; } while(u);
2281 /* handles u == 0 */
2283 s[len] = 0;
2286 static void highlightslinks (struct page *page, int xoff, int yoff,
2287 int noff, char *targ, int tlen, int hfsize)
2289 int i;
2290 char buf[40];
2291 struct slink *slink;
2292 double x0, y0, x1, y1, w;
2294 ensureslinks (page);
2295 glColor3ub (0xc3, 0xb0, 0x91);
2296 for (i = 0; i < page->slinkcount; ++i) {
2297 fmt_linkn (buf, i + noff);
2298 if (!tlen || !strncmp (targ, buf, tlen)) {
2299 slink = &page->slinks[i];
2301 x0 = slink->bbox.x0 + xoff - 5;
2302 y1 = slink->bbox.y0 + yoff - 5;
2303 y0 = y1 + 10 + hfsize;
2304 w = measure_string (state.face, hfsize, buf);
2305 x1 = x0 + w + 10;
2306 glRectd (x0, y0, x1, y1);
2310 glEnable (GL_BLEND);
2311 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2312 glEnable (GL_TEXTURE_2D);
2313 glColor3ub (0, 0, 0);
2314 for (i = 0; i < page->slinkcount; ++i) {
2315 fmt_linkn (buf, i + noff);
2316 if (!tlen || !strncmp (targ, buf, tlen)) {
2317 slink = &page->slinks[i];
2319 x0 = slink->bbox.x0 + xoff;
2320 y0 = slink->bbox.y0 + yoff + hfsize;
2321 draw_string (state.face, hfsize, x0, y0, buf);
2324 glDisable (GL_TEXTURE_2D);
2325 glDisable (GL_BLEND);
2329 static void uploadslice (struct tile *tile, struct slice *slice)
2331 int offset;
2332 struct slice *slice1;
2333 unsigned char *texdata;
2335 offset = 0;
2336 for (slice1 = tile->slices; slice != slice1; slice1++) {
2337 offset += slice1->h * tile->w * tile->pixmap->n;
2339 if (slice->texindex != -1 && slice->texindex < state.texcount
2340 && state.texowners[slice->texindex].slice == slice) {
2341 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
2343 else {
2344 int subimage = 0;
2345 int texindex = state.texindex++ % state.texcount;
2347 if (state.texowners[texindex].w == tile->w) {
2348 if (state.texowners[texindex].h >= slice->h) {
2349 subimage = 1;
2351 else {
2352 state.texowners[texindex].h = slice->h;
2355 else {
2356 state.texowners[texindex].h = slice->h;
2359 state.texowners[texindex].w = tile->w;
2360 state.texowners[texindex].slice = slice;
2361 slice->texindex = texindex;
2363 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
2364 if (tile->pbo) {
2365 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
2366 texdata = 0;
2368 else {
2369 texdata = tile->pixmap->samples;
2371 if (subimage) {
2372 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
2376 tile->w,
2377 slice->h,
2378 state.texform,
2379 state.texty,
2380 texdata+offset
2383 else {
2384 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
2386 state.texiform,
2387 tile->w,
2388 slice->h,
2390 state.texform,
2391 state.texty,
2392 texdata+offset
2395 if (tile->pbo) {
2396 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
2401 CAMLprim value ml_drawtile (value args_v, value ptr_v)
2403 CAMLparam2 (args_v, ptr_v);
2404 int dispx = Int_val (Field (args_v, 0));
2405 int dispy = Int_val (Field (args_v, 1));
2406 int dispw = Int_val (Field (args_v, 2));
2407 int disph = Int_val (Field (args_v, 3));
2408 int tilex = Int_val (Field (args_v, 4));
2409 int tiley = Int_val (Field (args_v, 5));
2410 char *s = String_val (ptr_v);
2411 struct tile *tile = parse_pointer ("ml_drawtile", s);
2413 glEnable (GL_TEXTURE_RECTANGLE_ARB);
2415 int slicey, firstslice;
2416 struct slice *slice;
2418 firstslice = tiley / tile->sliceheight;
2419 slice = &tile->slices[firstslice];
2420 slicey = tiley % tile->sliceheight;
2422 while (disph > 0) {
2423 int dh;
2425 dh = slice->h - slicey;
2426 dh = MIN (disph, dh);
2427 uploadslice (tile, slice);
2429 glBegin (GL_QUADS);
2431 glTexCoord2i (tilex, slicey);
2432 glVertex2i (dispx, dispy);
2434 glTexCoord2i (tilex+dispw, slicey);
2435 glVertex2i (dispx+dispw, dispy);
2437 glTexCoord2i (tilex+dispw, slicey+dh);
2438 glVertex2i (dispx+dispw, dispy+dh);
2440 glTexCoord2i (tilex, slicey+dh);
2441 glVertex2i (dispx, dispy+dh);
2443 glEnd ();
2445 dispy += dh;
2446 disph -= dh;
2447 slice++;
2448 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
2449 slicey = 0;
2452 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2453 CAMLreturn (Val_unit);
2456 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2457 value xoff_v, value yoff_v,
2458 value li_v)
2460 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, li_v);
2461 int xoff = Int_val (xoff_v);
2462 int yoff = Int_val (yoff_v);
2463 int noff = Int_val (Field (li_v, 0));
2464 char *targ = String_val (Field (li_v, 1));
2465 int tlen = caml_string_length (Field (li_v, 1));
2466 int hfsize = Int_val (Field (li_v, 2));
2467 char *s = String_val (ptr_v);
2468 int hlmask = Int_val (hlinks_v);
2469 struct page *page = parse_pointer ("ml_postprocess", s);
2471 if (!page->u.ptr) {
2472 /* deal with loadpage failed pages */
2473 goto done;
2476 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2477 if (trylock ("ml_postprocess")) {
2478 noff = 0;
2479 goto done;
2481 if (hlmask & 2) {
2482 highlightslinks (page, xoff, yoff, noff, targ, tlen, hfsize);
2483 noff = page->slinkcount;
2485 if (page->tgen == state.gen) {
2486 showsel (page, xoff, yoff);
2488 unlock ("ml_postprocess");
2490 done:
2491 CAMLreturn (Val_int (noff));
2494 static fz_link *getlink (struct page *page, int x, int y)
2496 fz_point p;
2497 fz_matrix ctm;
2498 const fz_matrix *tctm;
2499 fz_link *link, *links;
2501 switch (page->type) {
2502 case DPDF:
2503 trimctm (page->u.pdfpage, page->pdimno);
2504 tctm = &state.pagedims[page->pdimno].tctm;
2505 links = page->u.pdfpage->links;
2506 break;
2508 case DXPS:
2509 tctm = &fz_identity;
2510 links = page->u.xpspage->links;
2511 break;
2513 default:
2514 return NULL;
2516 p.x = x;
2517 p.y = y;
2519 fz_concat (&ctm, tctm, &state.pagedims[page->pdimno].ctm);
2520 fz_invert_matrix (&ctm, &ctm);
2521 fz_transform_point (&p, &ctm);
2523 for (link = links; link; link = link->next) {
2524 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2525 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2526 return link;
2530 return NULL;
2533 static void ensuretext (struct page *page)
2535 if (state.gen != page->tgen) {
2536 droptext (page);
2537 page->tgen = state.gen;
2539 if (!page->text) {
2540 fz_matrix ctm;
2541 fz_device *tdev;
2543 page->text = fz_new_text_page (state.ctx);
2544 page->sheet = fz_new_text_sheet (state.ctx);
2545 tdev = fz_new_text_device (state.ctx, page->sheet, page->text);
2546 ctm = pagectm (page);
2547 fz_begin_page (tdev, &fz_infinite_rect, &ctm);
2548 fz_run_display_list (page->dlist, tdev, &ctm, &fz_infinite_rect, NULL);
2549 qsort (page->text->blocks, page->text->len,
2550 sizeof (*page->text->blocks), compareblocks);
2551 fz_end_page (tdev);
2552 fz_free_device (tdev);
2556 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2558 CAMLparam2 (start_page_v, dir_v);
2559 CAMLlocal1 (ret_v);
2560 int i, dir = Int_val (dir_v);
2561 int start_page = Int_val (start_page_v);
2562 int end_page = dir > 0 ? state.pagecount : -1;
2564 ret_v = Val_int (0);
2565 if (!(state.type == DPDF || state.type == DXPS)) {
2566 goto done;
2569 lock ("ml_findpage_with_links");
2570 for (i = start_page + dir; i != end_page; i += dir) {
2571 int found;
2573 switch (state.type) {
2574 case DPDF:
2576 pdf_page *page = NULL;
2578 fz_try (state.ctx) {
2579 page = pdf_load_page (state.u.pdf, i);
2580 found = !!page->links;
2582 fz_catch (state.ctx) {
2583 found = 0;
2585 if (page) {
2586 freepdfpage (page);
2589 break;
2590 case DXPS:
2592 xps_page *page = xps_load_page (state.u.xps, i);
2593 found = !!page->links;
2594 freexpspage (page);
2596 break;
2598 default:
2599 ARSERT (0 && "invalid document type");
2602 if (found) {
2603 ret_v = caml_alloc_small (1, 1);
2604 Field (ret_v, 0) = Val_int (i);
2605 goto unlock;
2608 unlock:
2609 unlock ("ml_findpage_with_links");
2611 done:
2612 CAMLreturn (ret_v);
2615 enum { dir_first, dir_last};
2616 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
2618 CAMLprim value ml_findlink (value ptr_v, value dir_v)
2620 CAMLparam2 (ptr_v, dir_v);
2621 CAMLlocal2 (ret_v, pos_v);
2622 struct page *page;
2623 int dirtag, i, slinkindex;
2624 struct slink *found = NULL ,*slink;
2625 char *s = String_val (ptr_v);
2627 page = parse_pointer ("ml_findlink", s);
2628 ret_v = Val_int (0);
2629 if (trylock ("ml_findlink")) {
2630 goto done;
2633 ensureslinks (page);
2635 if (Is_block (dir_v)) {
2636 dirtag = Tag_val (dir_v);
2637 switch (dirtag) {
2638 case dir_first_visible:
2640 int x0, y0, dir, first_index, last_index;
2642 pos_v = Field (dir_v, 0);
2643 x0 = Int_val (Field (pos_v, 0));
2644 y0 = Int_val (Field (pos_v, 1));
2645 dir = Int_val (Field (pos_v, 2));
2647 if (dir >= 0) {
2648 dir = 1;
2649 first_index = 0;
2650 last_index = page->slinkcount;
2652 else {
2653 first_index = page->slinkcount - 1;
2654 last_index = -1;
2657 for (i = first_index; i != last_index; i += dir) {
2658 slink = &page->slinks[i];
2659 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
2660 found = slink;
2661 break;
2665 break;
2667 case dir_left:
2668 slinkindex = Int_val (Field (dir_v, 0));
2669 found = &page->slinks[slinkindex];
2670 for (i = slinkindex - 1; i >= 0; --i) {
2671 slink = &page->slinks[i];
2672 if (slink->bbox.x0 < found->bbox.x0) {
2673 found = slink;
2674 break;
2677 break;
2679 case dir_right:
2680 slinkindex = Int_val (Field (dir_v, 0));
2681 found = &page->slinks[slinkindex];
2682 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2683 slink = &page->slinks[i];
2684 if (slink->bbox.x0 > found->bbox.x0) {
2685 found = slink;
2686 break;
2689 break;
2691 case dir_down:
2692 slinkindex = Int_val (Field (dir_v, 0));
2693 found = &page->slinks[slinkindex];
2694 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2695 slink = &page->slinks[i];
2696 if (slink->bbox.y0 >= found->bbox.y0) {
2697 found = slink;
2698 break;
2701 break;
2703 case dir_up:
2704 slinkindex = Int_val (Field (dir_v, 0));
2705 found = &page->slinks[slinkindex];
2706 for (i = slinkindex - 1; i >= 0; --i) {
2707 slink = &page->slinks[i];
2708 if (slink->bbox.y0 <= found->bbox.y0) {
2709 found = slink;
2710 break;
2713 break;
2716 else {
2717 dirtag = Int_val (dir_v);
2718 switch (dirtag) {
2719 case dir_first:
2720 found = page->slinks;
2721 break;
2723 case dir_last:
2724 if (page->slinks) {
2725 found = page->slinks + (page->slinkcount - 1);
2727 break;
2730 if (found) {
2731 ret_v = caml_alloc_small (2, 1);
2732 Field (ret_v, 0) = Val_int (found - page->slinks);
2735 unlock ("ml_findlink");
2736 done:
2737 CAMLreturn (ret_v);
2740 enum { uuri, ugoto, utext, uunexpected,
2741 ulaunch, unamed, uremote, uremotedest };
2743 #define LINKTOVAL \
2745 int pageno; \
2747 switch (link->dest.kind) { \
2748 case FZ_LINK_GOTO: \
2750 fz_point p; \
2752 pageno = link->dest.ld.gotor.page; \
2753 p.x = 0; \
2754 p.y = 0; \
2756 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
2757 p.y = link->dest.ld.gotor.lt.y; \
2758 fz_transform_point (&p, &pdim->lctm); \
2760 tup_v = caml_alloc_tuple (2); \
2761 ret_v = caml_alloc_small (1, ugoto); \
2762 Field (tup_v, 0) = Val_int (pageno); \
2763 Field (tup_v, 1) = Val_int (p.y); \
2764 Field (ret_v, 0) = tup_v; \
2766 break; \
2768 case FZ_LINK_URI: \
2769 str_v = caml_copy_string (link->dest.ld.uri.uri); \
2770 ret_v = caml_alloc_small (1, uuri); \
2771 Field (ret_v, 0) = str_v; \
2772 break; \
2774 case FZ_LINK_LAUNCH: \
2775 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
2776 ret_v = caml_alloc_small (1, ulaunch); \
2777 Field (ret_v, 0) = str_v; \
2778 break; \
2780 case FZ_LINK_NAMED: \
2781 str_v = caml_copy_string (link->dest.ld.named.named); \
2782 ret_v = caml_alloc_small (1, unamed); \
2783 Field (ret_v, 0) = str_v; \
2784 break; \
2786 case FZ_LINK_GOTOR: \
2788 int rty; \
2790 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
2791 pageno = link->dest.ld.gotor.page; \
2792 if (pageno == -1) { \
2793 gr_v = caml_copy_string (link->dest.ld.gotor.dest); \
2794 rty = uremotedest; \
2796 else { \
2797 gr_v = Val_int (pageno); \
2798 rty = uremote; \
2800 tup_v = caml_alloc_tuple (2); \
2801 ret_v = caml_alloc_small (1, rty); \
2802 Field (tup_v, 0) = str_v; \
2803 Field (tup_v, 1) = gr_v; \
2804 Field (ret_v, 0) = tup_v; \
2806 break; \
2808 default: \
2810 char buf[80]; \
2812 snprintf (buf, sizeof (buf), \
2813 "unhandled link kind %d", link->dest.kind); \
2814 str_v = caml_copy_string (buf); \
2815 ret_v = caml_alloc_small (1, uunexpected); \
2816 Field (ret_v, 0) = str_v; \
2818 break; \
2822 CAMLprim value ml_getlink (value ptr_v, value n_v)
2824 CAMLparam2 (ptr_v, n_v);
2825 CAMLlocal4 (ret_v, tup_v, str_v, gr_v);
2826 fz_link *link;
2827 struct page *page;
2828 struct pagedim *pdim;
2829 char *s = String_val (ptr_v);
2831 ret_v = Val_int (0);
2832 if (trylock ("ml_getlink")) {
2833 goto done;
2836 page = parse_pointer ("ml_getlink", s);
2837 ensureslinks (page);
2838 pdim = &state.pagedims[page->pdimno];
2839 link = page->slinks[Int_val (n_v)].link;
2840 LINKTOVAL;
2842 unlock ("ml_getlink");
2843 done:
2844 CAMLreturn (ret_v);
2847 CAMLprim value ml_getlinkcount (value ptr_v)
2849 CAMLparam1 (ptr_v);
2850 struct page *page;
2851 char *s = String_val (ptr_v);
2853 page = parse_pointer ("ml_getlinkcount", s);
2854 CAMLreturn (Val_int (page->slinkcount));
2857 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
2859 CAMLparam2 (ptr_v, n_v);
2860 CAMLlocal1 (ret_v);
2861 struct page *page;
2862 struct slink *slink;
2863 char *s = String_val (ptr_v);
2865 page = parse_pointer ("ml_getlinkrect", s);
2866 ret_v = caml_alloc_tuple (4);
2867 if (trylock ("ml_getlinkrect")) {
2868 Field (ret_v, 0) = Val_int (0);
2869 Field (ret_v, 1) = Val_int (0);
2870 Field (ret_v, 2) = Val_int (0);
2871 Field (ret_v, 3) = Val_int (0);
2872 goto done;
2874 ensureslinks (page);
2876 slink = &page->slinks[Int_val (n_v)];
2877 Field (ret_v, 0) = Val_int (slink->bbox.x0);
2878 Field (ret_v, 1) = Val_int (slink->bbox.y0);
2879 Field (ret_v, 2) = Val_int (slink->bbox.x1);
2880 Field (ret_v, 3) = Val_int (slink->bbox.y1);
2881 unlock ("ml_getlinkrect");
2883 done:
2884 CAMLreturn (ret_v);
2887 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
2889 CAMLparam3 (ptr_v, x_v, y_v);
2890 CAMLlocal4 (ret_v, tup_v, str_v, gr_v);
2891 fz_link *link;
2892 struct page *page;
2893 char *s = String_val (ptr_v);
2894 int x = Int_val (x_v), y = Int_val (y_v);
2895 struct pagedim *pdim;
2897 ret_v = Val_int (0);
2898 if (trylock ("ml_whatsunder")) {
2899 goto done;
2902 page = parse_pointer ("ml_whatsunder", s);
2903 pdim = &state.pagedims[page->pdimno];
2904 x += pdim->bounds.x0;
2905 y += pdim->bounds.y0;
2906 link = getlink (page, x, y);
2907 if (link) {
2908 LINKTOVAL;
2910 else {
2911 fz_rect *b;
2912 fz_page_block *pageb;
2913 fz_text_block *block;
2915 ensuretext (page);
2916 for (pageb = page->text->blocks;
2917 pageb < page->text->blocks + page->text->len;
2918 ++pageb) {
2919 fz_text_line *line;
2920 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
2921 block = pageb->u.text;
2923 b = &block->bbox;
2924 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2925 continue;
2927 for (line = block->lines;
2928 line < block->lines + block->len;
2929 ++line) {
2930 fz_text_span *span;
2932 b = &line->bbox;
2933 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2934 continue;
2936 for (span = line->first_span; span; span = span->next) {
2937 int charnum;
2939 b = &span->bbox;
2940 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2941 continue;
2943 for (charnum = 0; charnum < span->len; ++charnum) {
2944 fz_rect bbox;
2945 fz_text_char_bbox (&bbox, span, charnum);
2946 b = &bbox;
2948 if (x >= b->x0 && x <= b->x1
2949 && y >= b->y0 && y <= b->y1) {
2950 fz_text_style *style = span->text->style;
2951 const char *n2 =
2952 style->font && style->font->name
2953 ? style->font->name
2954 : "Span has no font name"
2956 FT_FaceRec *face = style->font->ft_face;
2957 if (face && face->family_name) {
2958 char *s;
2959 char *n1 = face->family_name;
2960 size_t l1 = strlen (n1);
2961 size_t l2 = strlen (n2);
2963 if (l1 != l2 || memcmp (n1, n2, l1)) {
2964 s = malloc (l1 + l2 + 2);
2965 if (s) {
2966 memcpy (s, n2, l2);
2967 s[l2] = '=';
2968 memcpy (s + l2 + 1, n1, l1 + 1);
2969 str_v = caml_copy_string (s);
2970 free (s);
2974 if (str_v == 0) {
2975 str_v = caml_copy_string (n2);
2977 ret_v = caml_alloc_small (1, utext);
2978 Field (ret_v, 0) = str_v;
2979 goto unlock;
2986 unlock:
2987 unlock ("ml_whatsunder");
2989 done:
2990 CAMLreturn (ret_v);
2993 enum { mark_page, mark_block, mark_line, mark_word };
2995 static int uninteresting (int c)
2997 return c == ' ' || c == '\n' || c == '\t' || c == '\n' || c == '\r'
2998 || ispunct (c);
3001 CAMLprim value ml_markunder (value ptr_v, value x_v, value y_v, value mark_v)
3003 CAMLparam4 (ptr_v, x_v, y_v, mark_v);
3004 CAMLlocal1 (ret_v);
3005 fz_rect *b;
3006 struct page *page;
3007 fz_text_line *line;
3008 fz_page_block *pageb;
3009 fz_text_block *block;
3010 struct pagedim *pdim;
3011 int mark = Int_val (mark_v);
3012 char *s = String_val (ptr_v);
3013 int x = Int_val (x_v), y = Int_val (y_v);
3015 ret_v = Val_bool (0);
3016 if (trylock ("ml_markunder")) {
3017 goto done;
3020 page = parse_pointer ("ml_markunder", s);
3021 pdim = &state.pagedims[page->pdimno];
3022 x += pdim->bounds.x0;
3023 y += pdim->bounds.y0;
3025 ensuretext (page);
3027 if (mark == mark_page) {
3028 int i;
3030 for (i = page->text->len - 1; i >= 0; --i) {
3031 pageb = &page->text->blocks[i];
3032 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3034 if (i < 0) goto unlock;
3036 block = pageb->u.text;
3038 page->fmark.i = 0;
3039 page->fmark.span = block->lines->first_span;
3041 line = &block->lines[block->len - 1];
3042 page->lmark.i = line->last_span->len - 1;
3043 page->lmark.span = line->last_span;
3044 ret_v = Val_bool (1);
3045 goto unlock;
3048 for (pageb = page->text->blocks;
3049 pageb < page->text->blocks + page->text->len;
3050 ++pageb) {
3051 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3052 block = pageb->u.text;
3054 b = &block->bbox;
3055 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3056 continue;
3058 if (mark == mark_block) {
3059 page->fmark.i = 0;
3060 page->fmark.span = block->lines->first_span;
3062 line = &block->lines[block->len - 1];
3063 page->lmark.i = line->last_span->len - 1;
3064 page->lmark.span = line->last_span;
3065 ret_v = Val_bool (1);
3066 goto unlock;
3069 for (line = block->lines;
3070 line < block->lines + block->len;
3071 ++line) {
3072 fz_text_span *span;
3074 b = &line->bbox;
3075 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3076 continue;
3078 if (mark == mark_line) {
3079 page->fmark.i = 0;
3080 page->fmark.span = line->first_span;
3082 page->lmark.i = line->last_span->len - 1;
3083 page->lmark.span = line->last_span;
3084 ret_v = Val_bool (1);
3085 goto unlock;
3088 for (span = line->first_span; span; span = span->next) {
3089 int charnum;
3091 b = &span->bbox;
3092 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
3093 continue;
3095 for (charnum = 0; charnum < span->len; ++charnum) {
3096 fz_rect bbox;
3097 fz_text_char_bbox (&bbox, span, charnum);
3098 b = &bbox;
3100 if (x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1) {
3101 /* unicode ftw */
3102 int charnum2, charnum3 = -1, charnum4 = -1;
3104 if (uninteresting (span->text[charnum].c)) goto unlock;
3106 for (charnum2 = charnum; charnum2 >= 0; --charnum2) {
3107 if (uninteresting (span->text[charnum2].c)) {
3108 charnum3 = charnum2 + 1;
3109 break;
3112 if (charnum3 == -1) charnum3 = 0;
3114 for (charnum2 = charnum + 1;
3115 charnum2 < span->len;
3116 ++charnum2) {
3117 if (uninteresting (span->text[charnum2].c)) break;
3118 charnum4 = charnum2;
3120 if (charnum4 == -1) goto unlock;
3122 page->fmark.i = charnum3;
3123 page->fmark.span = span;
3125 page->lmark.i = charnum4;
3126 page->lmark.span = span;
3127 ret_v = Val_bool (1);
3128 goto unlock;
3134 unlock:
3135 if (!Bool_val (ret_v)) {
3136 page->fmark.span = NULL;
3137 page->lmark.span = NULL;
3138 page->fmark.i = 0;
3139 page->lmark.i = 0;
3141 unlock ("ml_markunder");
3143 done:
3144 CAMLreturn (ret_v);
3147 CAMLprim value ml_rectofblock (value ptr_v, value x_v, value y_v)
3149 CAMLparam3 (ptr_v, x_v, y_v);
3150 CAMLlocal2 (ret_v, res_v);
3151 fz_rect *b = NULL;
3152 struct page *page;
3153 fz_page_block *pageb;
3154 struct pagedim *pdim;
3155 char *s = String_val (ptr_v);
3156 int x = Int_val (x_v), y = Int_val (y_v);
3158 ret_v = Val_int (0);
3159 if (trylock ("ml_rectofblock")) {
3160 goto done;
3163 page = parse_pointer ("ml_rectofblock", s);
3164 pdim = &state.pagedims[page->pdimno];
3165 x += pdim->bounds.x0;
3166 y += pdim->bounds.y0;
3168 ensuretext (page);
3170 for (pageb = page->text->blocks;
3171 pageb < page->text->blocks + page->text->len;
3172 ++pageb) {
3173 switch (pageb->type) {
3174 case FZ_PAGE_BLOCK_TEXT:
3175 b = &pageb->u.text->bbox;
3176 break;
3178 case FZ_PAGE_BLOCK_IMAGE:
3179 b = &pageb->u.image->bbox;
3180 break;
3182 default:
3183 continue;
3186 if (x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1)
3187 break;
3188 b = NULL;
3190 if (b) {
3191 res_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
3192 ret_v = caml_alloc_small (1, 1);
3193 Store_double_field (res_v, 0, b->x0);
3194 Store_double_field (res_v, 1, b->x1);
3195 Store_double_field (res_v, 2, b->y0);
3196 Store_double_field (res_v, 3, b->y1);
3197 Field (ret_v, 0) = res_v;
3199 unlock ("ml_rectofblock");
3201 done:
3202 CAMLreturn (ret_v);
3205 CAMLprim value ml_seltext (value ptr_v, value rect_v)
3207 CAMLparam2 (ptr_v, rect_v);
3208 fz_rect b;
3209 struct page *page;
3210 struct pagedim *pdim;
3211 int i, x0, x1, y0, y1;
3212 char *s = String_val (ptr_v);
3213 int fi = 0, li = 0;
3214 fz_page_block *pageb;
3215 fz_text_block *block;
3216 fz_text_span *span, *fspan, *lspan;
3217 fz_text_line *line, *fline = NULL, *lline = NULL;
3219 if (trylock ("ml_seltext")) {
3220 goto done;
3223 page = parse_pointer ("ml_seltext", s);
3224 ensuretext (page);
3226 pdim = &state.pagedims[page->pdimno];
3227 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;;
3228 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
3229 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
3230 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
3232 if (0) {
3233 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
3234 glColor3ub (128, 128, 128);
3235 glRecti (x0, y0, x1, y1);
3236 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
3239 fspan = lspan = NULL;
3241 for (pageb= page->text->blocks;
3242 pageb < page->text->blocks + page->text->len;
3243 ++pageb) {
3244 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3245 block = pageb->u.text;
3246 for (line = block->lines;
3247 line < block->lines + block->len;
3248 ++line) {
3249 fz_text_span *span;
3251 for (span = line->first_span; span; span = span->next) {
3252 for (i = 0; i < span->len; ++i) {
3253 int selected = 0;
3255 fz_text_char_bbox (&b, span, i);
3257 if (x0 >= b.x0 && x0 <= b.x1
3258 && y0 >= b.y0 && y0 <= b.y1) {
3259 fspan = span;
3260 fline = line;
3261 fi = i;
3262 selected = 1;
3264 if (x1 >= b.x0 && x1 <= b.x1
3265 && y1 >= b.y0 && y1 <= b.y1) {
3266 lspan = span;
3267 lline = line;
3268 li = i;
3269 selected = 1;
3271 if (0 && selected) {
3272 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
3273 glColor3ub (128, 128, 128);
3274 glRecti (b.x0, b.y0, b.x1, b.y1);
3275 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
3281 if (y1 < y0 || x1 < x0) {
3282 int swap = 0;
3284 if (fspan == lspan) {
3285 swap = 1;
3287 else {
3288 if (y1 < y0) {
3289 if (fline != lline) {
3290 swap = 1;
3295 if (swap) {
3296 i = fi;
3297 span = fspan;
3299 fi = li;
3300 fspan = lspan;
3302 li = i;
3303 lspan = span;
3307 page->fmark.i = fi;
3308 page->fmark.span = fspan;
3310 page->lmark.i = li;
3311 page->lmark.span = lspan;
3313 unlock ("ml_seltext");
3315 done:
3316 CAMLreturn (Val_unit);
3319 static int UNUSED_ATTR pipespan (FILE *f, fz_text_span *span, int a, int b)
3321 char buf[4];
3322 int i, len, ret;
3324 for (i = a; i <= b; ++i) {
3325 len = fz_runetochar (buf, span->text[i].c);
3326 ret = fwrite (buf, len, 1, f);
3328 if (ret != 1) {
3329 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
3330 len, ret, strerror (errno));
3331 return -1;
3334 return 0;
3337 #ifdef __CYGWIN__
3338 value ml_popen (value UNUSED_ATTR u1, value UNUSED_ATTR u2)
3340 caml_failwith ("ml_popen not implemented under Cygwin");
3342 #else
3343 CAMLprim value ml_popen (value command_v, value fds_v)
3345 CAMLparam2 (command_v, fds_v);
3346 CAMLlocal2 (l_v, tup_v);
3347 int ret;
3348 char *msg = NULL;
3349 value earg_v = Nothing;
3350 posix_spawnattr_t attr;
3351 posix_spawn_file_actions_t fa;
3352 char *argv[] = { "/bin/sh", "-c", String_val (command_v), NULL };
3354 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
3355 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
3358 if ((ret = posix_spawnattr_init (&attr)) != 0) {
3359 msg = "posix_spawnattr_init";
3360 goto fail1;
3363 #ifdef POSIX_SPAWN_USEVFORK
3364 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
3365 msg = "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK";
3366 goto fail;
3368 #endif
3370 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
3371 int fd1, fd2;
3373 tup_v = Field (l_v, 0);
3374 fd1 = Int_val (Field (tup_v, 0));
3375 fd2 = Int_val (Field (tup_v, 1));
3376 if (fd2 < 0) {
3377 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
3378 msg = "posix_spawn_file_actions_addclose";
3379 earg_v = tup_v;
3380 goto fail;
3383 else {
3384 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
3385 msg = "posix_spawn_file_actions_adddup2";
3386 earg_v = tup_v;
3387 goto fail;
3392 if ((ret = posix_spawn (NULL, "/bin/sh", &fa, &attr, argv, environ))) {
3393 msg = "posix_spawn";
3394 goto fail;
3397 fail:
3398 if ((ret = posix_spawnattr_destroy (&attr)) != 0) {
3399 fprintf (stderr, "posix_spawnattr_destroy: %s\n", strerror (ret));
3402 fail1:
3403 if ((ret = posix_spawn_file_actions_destroy (&fa)) != 0) {
3404 fprintf (stderr, "posix_spawn_file_actions_destroy: %s\n",
3405 strerror (ret));
3408 if (msg)
3409 unix_error (ret, msg, earg_v);
3411 CAMLreturn (Val_unit);
3413 #endif
3415 CAMLprim value ml_copysel (value fd_v, value ptr_v, value clear_v)
3417 CAMLparam3 (fd_v, ptr_v, clear_v);
3418 FILE *f;
3419 int seen = 0;
3420 struct page *page;
3421 fz_text_line *line;
3422 fz_page_block *pageb;
3423 fz_text_block *block;
3424 int fd = Int_val (fd_v);
3425 char *s = String_val (ptr_v);
3426 int clear = Bool_val (clear_v);
3428 if (trylock ("ml_copysel")) {
3429 goto done;
3432 page = parse_pointer ("ml_sopysel", s);
3434 if (!page->fmark.span || !page->lmark.span) {
3435 fprintf (stderr, "nothing to copy\n");
3436 goto unlock;
3439 f = fdopen (fd, "w");
3440 if (!f) {
3441 fprintf (stderr, "failed to fopen sel pipe: %s\n",
3442 strerror (errno));
3443 f = stdout;
3446 for (pageb = page->text->blocks;
3447 pageb < page->text->blocks + page->text->len;
3448 ++pageb) {
3449 if (pageb->type != FZ_PAGE_BLOCK_TEXT) continue;
3450 block = pageb->u.text;
3451 for (line = block->lines;
3452 line < block->lines + block->len;
3453 ++line) {
3454 fz_text_span *span;
3456 for (span = line->first_span; span; span = span->next) {
3457 int a, b;
3459 seen |= span == page->fmark.span || span == page->lmark.span;
3460 a = span == page->fmark.span ? page->fmark.i : 0;
3461 b = span == page->lmark.span ? page->lmark.i : span->len - 1;
3463 if (seen) {
3464 if (pipespan (f, span, a, b)) {
3465 goto close;
3467 if (span == line->last_span) {
3468 if (putc ('\n', f) == EOF) {
3469 fprintf (stderr,
3470 "failed break line on sel pipe: %s\n",
3471 strerror (errno));
3472 goto close;
3475 if (span == page->lmark.span) {
3476 goto endloop;
3482 endloop:
3483 if (clear) {
3484 page->lmark.span = NULL;
3485 page->fmark.span = NULL;
3488 close:
3489 if (f != stdout) {
3490 int ret = fclose (f);
3491 fd = -1;
3492 if (ret == -1) {
3493 if (errno != ECHILD) {
3494 fprintf (stderr, "failed to close sel pipe: %s\n",
3495 strerror (errno));
3499 unlock:
3500 unlock ("ml_copysel");
3502 done:
3503 if (fd >= 0) {
3504 if (close (fd)) {
3505 fprintf (stderr, "failed to close sel pipe: %s\n",
3506 strerror (errno));
3509 CAMLreturn (Val_unit);
3512 CAMLprim value ml_getpdimrect (value pagedimno_v)
3514 CAMLparam1 (pagedimno_v);
3515 CAMLlocal1 (ret_v);
3516 int pagedimno = Int_val (pagedimno_v);
3517 fz_rect box;
3519 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
3520 if (trylock ("ml_getpdimrect")) {
3521 box = fz_empty_rect;
3523 else {
3524 box = state.pagedims[pagedimno].mediabox;
3525 unlock ("ml_getpdimrect");
3528 Store_double_field (ret_v, 0, box.x0);
3529 Store_double_field (ret_v, 1, box.x1);
3530 Store_double_field (ret_v, 2, box.y0);
3531 Store_double_field (ret_v, 3, box.y1);
3533 CAMLreturn (ret_v);
3536 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v,
3537 value dw_v, value cols_v)
3539 CAMLparam3 (winw_v, winh_v, dw_v);
3540 CAMLlocal1 (ret_v);
3541 int i;
3542 double zoom = -1.;
3543 double maxh = 0.0;
3544 struct pagedim *p;
3545 double winw = Int_val (winw_v);
3546 double winh = Int_val (winh_v);
3547 double dw = Int_val (dw_v);
3548 double cols = Int_val (cols_v);
3549 double pw = 1.0, ph = 1.0;
3551 if (trylock ("ml_zoom_for_height")) {
3552 goto done;
3555 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
3556 double w = p->pagebox.x1 / cols;
3557 double h = p->pagebox.y1;
3558 if (h > maxh) {
3559 maxh = h;
3560 ph = h;
3561 if (state.fitmodel != FitProportional) pw = w;
3563 if ((state.fitmodel == FitProportional) && w > pw) pw = w;
3566 zoom = (((winh / ph) * pw) + dw) / winw;
3567 unlock ("ml_zoom_for_height");
3568 done:
3569 ret_v = caml_copy_double (zoom);
3570 CAMLreturn (ret_v);
3573 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
3575 CAMLparam4 (pt_v, x_v, y_v, string_v);
3576 CAMLlocal1 (ret_v);
3577 int pt = Int_val(pt_v);
3578 int x = Int_val (x_v);
3579 int y = Int_val (y_v);
3580 double w;
3582 w = draw_string (state.face, pt, x, y, String_val (string_v));
3583 ret_v = caml_copy_double (w);
3584 CAMLreturn (ret_v);
3587 CAMLprim value ml_measure_string (value pt_v, value string_v)
3589 CAMLparam2 (pt_v, string_v);
3590 CAMLlocal1 (ret_v);
3591 int pt = Int_val (pt_v);
3592 double w;
3594 w = measure_string (state.face, pt, String_val (string_v));
3595 ret_v = caml_copy_double (w);
3596 CAMLreturn (ret_v);
3599 CAMLprim value ml_getpagebox (value opaque_v)
3601 CAMLparam1 (opaque_v);
3602 CAMLlocal1 (ret_v);
3603 fz_rect rect;
3604 fz_irect bbox;
3605 fz_matrix ctm;
3606 fz_device *dev;
3607 char *s = String_val (opaque_v);
3608 struct page *page = parse_pointer ("ml_getpagebox", s);
3610 ret_v = caml_alloc_tuple (4);
3611 dev = fz_new_bbox_device (state.ctx, &rect);
3612 dev->hints |= FZ_IGNORE_SHADE;
3614 switch (page->type) {
3615 case DPDF:
3616 ctm = pagectm (page);
3617 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, &ctm, NULL);
3618 break;
3620 case DXPS:
3621 ctm = pagectm (page);
3622 xps_run_page (state.u.xps, page->u.xpspage, dev, &ctm, NULL);
3623 break;
3625 default:
3626 rect = fz_infinite_rect;
3627 break;
3630 fz_free_device (dev);
3631 fz_round_rect (&bbox, &rect);
3632 Field (ret_v, 0) = Val_int (bbox.x0);
3633 Field (ret_v, 1) = Val_int (bbox.y0);
3634 Field (ret_v, 2) = Val_int (bbox.x1);
3635 Field (ret_v, 3) = Val_int (bbox.y1);
3637 CAMLreturn (ret_v);
3640 CAMLprim value ml_setaalevel (value level_v)
3642 CAMLparam1 (level_v);
3644 state.aalevel = Int_val (level_v);
3645 CAMLreturn (Val_unit);
3648 #undef pixel
3649 #include <X11/Xlib.h>
3650 #include <GL/glx.h>
3652 static struct {
3653 Display *dpy;
3654 GLXContext ctx;
3655 GLXDrawable drawable;
3656 } glx;
3658 #include "keysym2ucs.c"
3660 CAMLprim value ml_keysymtoutf8 (value keysym_v)
3662 CAMLparam1 (keysym_v);
3663 CAMLlocal1 (str_v);
3664 KeySym keysym = Int_val (keysym_v);
3665 Rune rune;
3666 int len;
3667 char buf[5];
3669 rune = keysym2ucs (keysym);
3670 len = fz_runetochar (buf, rune);
3671 buf[len] = 0;
3672 str_v = caml_copy_string (buf);
3673 CAMLreturn (str_v);
3676 CAMLprim value ml_glx (value win_v)
3678 CAMLparam1 (win_v);
3679 XVisualInfo *visual;
3680 int screen, wid = Int_val (win_v);
3681 int attributes[] = { GLX_RGBA, GLX_DOUBLEBUFFER, None };
3683 glx.dpy = XOpenDisplay (NULL);
3684 if (!glx.dpy) {
3685 caml_failwith ("XOpenDisplay");
3688 screen = DefaultScreen (glx.dpy);
3689 visual = glXChooseVisual (glx.dpy, screen, attributes);
3690 if (!visual) {
3691 XCloseDisplay (glx.dpy);
3692 glx.dpy = NULL;
3693 caml_failwith ("glXChooseVisual");
3696 glx.ctx = glXCreateContext (glx.dpy, visual, NULL, True);
3697 XFree (visual);
3698 if (!glx.ctx) {
3699 XCloseDisplay (glx.dpy);
3700 glx.dpy = NULL;
3701 caml_failwith ("glXCreateContext");
3704 if (!glXMakeCurrent (glx.dpy, wid, glx.ctx)) {
3705 glXDestroyContext (glx.dpy, glx.ctx);
3706 XCloseDisplay (glx.dpy);
3707 glx.dpy = NULL;
3708 glx.ctx = NULL;
3709 caml_failwith ("glXMakeCurrent");
3711 glx.drawable = wid;
3712 CAMLreturn (Val_unit);
3715 CAMLprim value ml_swapb (value unit_v)
3717 CAMLparam1 (unit_v);
3718 glXSwapBuffers (glx.dpy, glx.drawable);
3719 CAMLreturn (Val_unit);
3722 CAMLprim value ml_glxsync (value unit_v)
3724 CAMLparam1 (unit_v);
3725 if (glx.dpy && glx.ctx) {
3726 glXWaitX ();
3727 glXWaitGL ();
3729 CAMLreturn (Val_unit);
3732 enum { piunknown, pilinux, piosx, pisun, pifreebsd,
3733 pidragonflybsd, piopenbsd, pinetbsd, picygwin };
3735 CAMLprim value ml_platform (value unit_v)
3737 CAMLparam1 (unit_v);
3738 int platid = piunknown;
3740 #if defined __linux__
3741 platid = pilinux;
3742 #elif defined __CYGWIN__
3743 platid = picygwin;
3744 #elif defined __DragonFly__
3745 platid = pidragonflybsd;
3746 #elif defined __FreeBSD__
3747 platid = pifreebsd;
3748 #elif defined __OpenBSD__
3749 platid = piopenbsd;
3750 #elif defined __NetBSD__
3751 platid = pinetbsd;
3752 #elif defined __sun__
3753 platid = pisun;
3754 #elif defined __APPLE__
3755 platid = piosx;
3756 #endif
3757 CAMLreturn (Val_int (platid));
3760 CAMLprim value ml_cloexec (value fd_v)
3762 CAMLparam1 (fd_v);
3763 int fd = Int_val (fd_v);
3765 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
3766 uerror ("fcntl", Nothing);
3768 CAMLreturn (Val_unit);
3771 CAMLprim value ml_getpbo (value w_v, value h_v, value cs_v)
3773 CAMLparam2 (w_v, h_v);
3774 CAMLlocal1 (ret_v);
3775 struct pbo *pbo;
3776 int w = Int_val (w_v);
3777 int h = Int_val (h_v);
3778 int cs = Int_val (cs_v);
3780 if (state.pbo_usable) {
3781 pbo = calloc (sizeof (*pbo), 1);
3782 if (!pbo) {
3783 err (1, "calloc pbo");
3786 switch (cs) {
3787 case 0:
3788 case 1:
3789 pbo->size = w*h*4;
3790 break;
3791 case 2:
3792 pbo->size = w*h*2;
3793 break;
3794 default:
3795 errx (1, "ml_getpbo: invalid colorspace %d", cs);
3798 state.glGenBuffersARB (1, &pbo->id);
3799 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->id);
3800 state.glBufferDataARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->size,
3801 NULL, GL_STREAM_DRAW);
3802 pbo->ptr = state.glMapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB,
3803 GL_READ_WRITE);
3804 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3805 if (!pbo->ptr) {
3806 fprintf (stderr, "glMapBufferARB failed: %#x\n", glGetError ());
3807 state.glDeleteBuffersARB (1, &pbo->id);
3808 free (pbo);
3809 ret_v = caml_copy_string ("0");
3811 else {
3812 int res;
3813 char *s;
3815 res = snprintf (NULL, 0, "%" FMT_ptr, pbo);
3816 if (res < 0) {
3817 err (1, "snprintf %" FMT_ptr " failed", pbo);
3819 s = malloc (res+1);
3820 if (!s) {
3821 err (1, "malloc %d bytes failed", res+1);
3823 res = sprintf (s, "%" FMT_ptr, pbo);
3824 if (res < 0) {
3825 err (1, "sprintf %" FMT_ptr " failed", pbo);
3827 ret_v = caml_copy_string (s);
3828 free (s);
3831 else {
3832 ret_v = caml_copy_string ("0");
3834 CAMLreturn (ret_v);
3837 CAMLprim value ml_freepbo (value s_v)
3839 CAMLparam1 (s_v);
3840 char *s = String_val (s_v);
3841 struct tile *tile = parse_pointer ("ml_freepbo", s);
3843 if (tile->pbo) {
3844 state.glDeleteBuffersARB (1, &tile->pbo->id);
3845 tile->pbo->id = -1;
3846 tile->pbo->ptr = NULL;
3847 tile->pbo->size = -1;
3849 CAMLreturn (Val_unit);
3852 CAMLprim value ml_unmappbo (value s_v)
3854 CAMLparam1 (s_v);
3855 char *s = String_val (s_v);
3856 struct tile *tile = parse_pointer ("ml_unmappbo", s);
3858 if (tile->pbo) {
3859 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
3860 if (state.glUnmapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB) == GL_FALSE) {
3861 errx (1, "glUnmapBufferARB failed: %#x\n", glGetError ());
3863 tile->pbo->ptr = NULL;
3864 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3866 CAMLreturn (Val_unit);
3869 static void setuppbo (void)
3871 #define GGPA(n) (*(void (**) ()) &state.n = glXGetProcAddress ((GLubyte *) #n))
3872 state.pbo_usable = GGPA (glBindBufferARB)
3873 && GGPA (glUnmapBufferARB)
3874 && GGPA (glMapBufferARB)
3875 && GGPA (glBufferDataARB)
3876 && GGPA (glGenBuffersARB)
3877 && GGPA (glDeleteBuffersARB);
3878 #undef GGPA
3881 CAMLprim value ml_pbo_usable (value unit_v)
3883 CAMLparam1 (unit_v);
3884 CAMLreturn (Val_bool (state.pbo_usable));
3887 CAMLprim value ml_unproject (value ptr_v, value x_v, value y_v)
3889 CAMLparam3 (ptr_v, x_v, y_v);
3890 CAMLlocal2 (ret_v, tup_v);
3891 struct page *page;
3892 char *s = String_val (ptr_v);
3893 int x = Int_val (x_v), y = Int_val (y_v);
3894 struct pagedim *pdim;
3895 fz_point p;
3896 fz_matrix ctm;
3898 page = parse_pointer ("ml_unproject", s);
3899 pdim = &state.pagedims[page->pdimno];
3901 ret_v = Val_int (0);
3902 if (trylock ("ml_unproject")) {
3903 goto done;
3906 switch (page->type) {
3907 case DPDF:
3908 trimctm (page->u.pdfpage, page->pdimno);
3909 break;
3911 default:
3912 break;
3914 p.x = x;
3915 p.y = y;
3917 fz_concat (&ctm, &pdim->tctm, &pdim->ctm);
3918 fz_invert_matrix (&ctm, &ctm);
3919 fz_transform_point (&p, &ctm);
3921 tup_v = caml_alloc_tuple (2);
3922 ret_v = caml_alloc_small (1, 1);
3923 Field (tup_v, 0) = Val_int (p.x);
3924 Field (tup_v, 1) = Val_int (p.y);
3925 Field (ret_v, 0) = tup_v;
3927 unlock ("ml_unproject");
3928 done:
3929 CAMLreturn (ret_v);
3932 static void makestippletex (void)
3934 const char pixels[] = "\xff\xff\0\0";
3935 glGenTextures (1, &state.stid);
3936 glBindTexture (GL_TEXTURE_1D, state.stid);
3937 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
3938 glTexParameteri (GL_TEXTURE_1D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
3939 glTexImage1D (
3940 GL_TEXTURE_1D,
3942 GL_ALPHA,
3945 GL_ALPHA,
3946 GL_UNSIGNED_BYTE,
3947 pixels
3951 CAMLprim value ml_init (value pipe_v, value params_v)
3953 CAMLparam2 (pipe_v, params_v);
3954 CAMLlocal2 (trim_v, fuzz_v);
3955 int ret;
3956 int texcount;
3957 char *fontpath;
3958 int colorspace;
3959 int mustoresize;
3960 int haspboext;
3961 struct sigaction sa;
3963 state.cr = Int_val (Field (pipe_v, 0));
3964 state.cw = Int_val (Field (pipe_v, 1));
3965 state.rotate = Int_val (Field (params_v, 0));
3966 state.fitmodel = Int_val (Field (params_v, 1));
3967 trim_v = Field (params_v, 2);
3968 texcount = Int_val (Field (params_v, 3));
3969 state.sliceheight = Int_val (Field (params_v, 4));
3970 mustoresize = Int_val (Field (params_v, 5));
3971 colorspace = Int_val (Field (params_v, 6));
3972 fontpath = String_val (Field (params_v, 7));
3974 state.trimcachepath = strdup (String_val (Field (params_v, 8)));
3975 if (!state.trimcachepath) {
3976 fprintf (stderr, "failed to strdup trimcachepath: %s\n",
3977 strerror (errno));
3979 haspboext = Bool_val (Field (params_v, 9));
3981 state.ctx = fz_new_context (NULL, NULL, mustoresize);
3983 state.trimmargins = Bool_val (Field (trim_v, 0));
3984 fuzz_v = Field (trim_v, 1);
3985 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
3986 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
3987 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
3988 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
3990 set_tex_params (colorspace);
3992 if (*fontpath) {
3993 state.face = load_font (fontpath);
3995 else {
3996 unsigned int len;
3997 void *base = pdf_lookup_substitute_font (0, 0, 0, 0, &len);
3999 state.face = load_builtin_font (base, len);
4001 if (!state.face) _exit (1);
4003 realloctexts (texcount);
4005 if (haspboext) {
4006 setuppbo ();
4009 makestippletex ();
4010 #ifdef __CYGWIN__
4011 sa.sa_handler = SIG_IGN;
4012 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
4013 #else
4014 sa.sa_handler = SIG_DFL;
4015 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP | SA_NOCLDWAIT;
4016 #endif
4017 if (sigemptyset (&sa.sa_mask)) {
4018 err (1, "sigemptyset");
4020 if (sigaction (SIGCHLD, &sa, NULL)) {
4021 err (1, "sigaction");
4024 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
4025 if (ret) {
4026 errx (1, "pthread_create: %s", strerror (ret));
4029 CAMLreturn (Val_unit);