?  search:
summary | log | graphiclog1 | graphiclog2 | refs | edit | fork

c2ac33b8cafc04e4bc4102daa14ffef57d12ebda
[llpp.git] / link.c
blobc2ac33b8cafc04e4bc4102daa14ffef57d12ebda
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>
9
10 #include <unistd.h>
11 #include <pthread.h>
12 #include <sys/time.h>
13 #include <sys/types.h>
14 #include <sys/ioctl.h>
15
16 #ifdef __CYGWIN__
17 #include <cygwin/socket.h> /* FIONREAD */
18 #else
19 #include <spawn.h>
20 #endif
21
22 #include <regex.h>
23 #include <ctype.h>
24 #include <stdarg.h>
25 #include <limits.h>
26
27 #include <GL/gl.h>
28
29 #include <caml/fail.h>
30 #include <caml/alloc.h>
31 #include <caml/memory.h>
32 #include <caml/unixsupport.h>
33
34 #include <fitz.h>
35 #include <mupdf.h>
36 #include <mupdf-internal.h>
37 #include <muxps.h>
38 #include <muxps-internal.h>
39 #include <mucbz.h>
40
41 #include FT_FREETYPE_H
42
43 #define PIGGYBACK
44
45 #ifndef __USE_GNU
46 extern char **environ;
47 #endif
48
49 #define MIN(a,b) ((a) < (b) ? (a) : (b))
50 #define MAX(a,b) ((a) > (b) ? (a) : (b))
51
52 #if defined __GNUC__
53 #define NORETURN_ATTR __attribute__ ((noreturn))
54 #define UNUSED_ATTR __attribute__ ((unused))
55 #define OPTIMIZE_ATTR(n) __attribute__ ((optimize ("O"#n)))
56 #define GCC_FMT_ATTR(a, b) __attribute__ ((format (printf, a, b)))
57 #else
58 #define NORETURN_ATTR
59 #define UNUSED_ATTR
60 #define OPTIMIZE_ATTR(n)
61 #define GCC_FMT_ATTR(a, b)
62 #endif
63
64 #define FMT_s "zu"
65
66 #define FMT_ptr "p"
67 #define FMT_ptr_cast(p) (p)
68 #define FMT_ptr_cast2(p) (p)
69
70 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
71 err (int exitcode, const char *fmt, ...)
72 {
73 va_list ap;
74 int savederrno;
75
76 savederrno = errno;
77 va_start (ap, fmt);
78 vfprintf (stderr, fmt, ap);
79 va_end (ap);
80 fprintf (stderr, ": %s\n", strerror (savederrno));
81 fflush (stderr);
82 _exit (exitcode);
83 }
84
85 static void NORETURN_ATTR GCC_FMT_ATTR (2, 3)
86 errx (int exitcode, const char *fmt, ...)
87 {
88 va_list ap;
89
90 va_start (ap, fmt);
91 vfprintf (stderr, fmt, ap);
92 va_end (ap);
93 fputc ('\n', stderr);
94 fflush (stderr);
95 _exit (exitcode);
96 }
97
98 #ifndef GL_TEXTURE_RECTANGLE_ARB
99 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
100 #endif
101
102 #ifndef GL_BGRA
103 #define GL_BGRA 0x80E1
104 #endif
105
106 #ifndef GL_UNSIGNED_INT_8_8_8_8
107 #define GL_UNSIGNED_INT_8_8_8_8 0x8035
108 #endif
109
110 #ifndef GL_UNSIGNED_INT_8_8_8_8_REV
111 #define GL_UNSIGNED_INT_8_8_8_8_REV 0x8367
112 #endif
113
114 #if 0
115 #define lprintf printf
116 #else
117 #define lprintf(...)
118 #endif
119
120 #define ARSERT(cond) for (;;) { \
121 if (!(cond)) { \
122 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
123 } \
124 break; \
125 }
126
127 struct slice {
128 int h;
129 int texindex;
130 };
131
132 struct tile {
133 int x, y, w, h;
134 int slicecount;
135 int sliceheight;
136 struct pbo *pbo;
137 fz_pixmap *pixmap;
138 struct slice slices[1];
139 };
140
141 struct pagedim {
142 int pageno;
143 int rotate;
144 int left;
145 int tctmready;
146 fz_bbox bounds;
147 fz_rect pagebox;
148 fz_rect mediabox;
149 fz_matrix ctm, zoomctm, lctm, tctm;
150 };
151
152 struct slink {
153 fz_bbox bbox;
154 fz_link *link;
155 };
156
157 enum { DPDF, DXPS, DCBZ };
158
159 struct page {
160 int tgen;
161 int sgen;
162 int type;
163 int pageno;
164 int pdimno;
165 fz_text_page *text;
166 fz_text_sheet *sheet;
167 union {
168 void *ptr;
169 pdf_page *pdfpage;
170 xps_page *xpspage;
171 cbz_page *cbzpage;
172 } u;
173 fz_display_list *dlist;
174 int slinkcount;
175 struct slink *slinks;
176 struct mark {
177 int i;
178 fz_text_span *span;
179 } fmark, lmark;
180 void (*freepage) (void *);
181 };
182
183 struct {
184 int type;
185 int sliceheight;
186 struct pagedim *pagedims;
187 int pagecount;
188 int pagedimcount;
189 union {
190 pdf_document *pdf;
191 xps_document *xps;
192 cbz_document *cbz;
193 } u;
194 fz_context *ctx;
195 int w, h;
196
197 int texindex;
198 int texcount;
199 GLuint *texids;
200
201 GLenum texiform;
202 GLenum texform;
203 GLenum texty;
204
205 fz_colorspace *colorspace;
206
207 struct {
208 int w, h;
209 struct slice *slice;
210 } *texowners;
211
212 int rotate;
213 int proportional;
214 int trimmargins;
215 int needoutline;
216 int gen;
217 int aalevel;
218
219 int trimanew;
220 fz_bbox trimfuzz;
221 fz_pixmap *pig;
222
223 pthread_t thread;
224 int cr, cw;
225 FT_Face face;
226
227 void (*closedoc) (void);
228 void (*freepage) (void *);
229
230 char *trimcachepath;
231
232 int pbo_usable;
233 void (*glBindBufferARB) (GLenum, GLuint);
234 void (*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;
240
241 struct pbo {
242 GLuint id;
243 void *ptr;
244 size_t size;
245 };
246
247 static void UNUSED_ATTR debug_rect (const char *cap, fz_rect r)
248 {
249 printf ("%s(rect) %.2f,%.2f,%.2f,%.2f\n", cap, r.x0, r.y0, r.x1, r.y1);
250 }
251
252 static void UNUSED_ATTR debug_bbox (const char *cap, fz_bbox r)
253 {
254 printf ("%s(bbox) %d,%d,%d,%d\n", cap, r.x0, r.y0, r.x1, r.y1);
255 }
256
257 static void UNUSED_ATTR debug_matrix (const char *cap, fz_matrix m)
258 {
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);
261 }
262
263 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
264
265 static void lock (const char *cap)
266 {
267 int ret = pthread_mutex_lock (&mutex);
268 if (ret) {
269 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
270 }
271 }
272
273 static void unlock (const char *cap)
274 {
275 int ret = pthread_mutex_unlock (&mutex);
276 if (ret) {
277 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
278 }
279 }
280
281 static int trylock (const char *cap)
282 {
283 int ret = pthread_mutex_trylock (&mutex);
284 if (ret && ret != EBUSY) {
285 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
286 }
287 return ret == EBUSY;
288 }
289
290 static void *parse_pointer (const char *cap, const char *s)
291 {
292 int ret;
293 void *ptr;
294
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);
298 }
299 return ptr;
300 }
301
302 static double now (void)
303 {
304 struct timeval tv;
305
306 if (gettimeofday (&tv, NULL)) {
307 err (1, "gettimeofday");
308 }
309 return tv.tv_sec + tv.tv_usec*1e-6;
310 }
311
312 static int hasdata (void)
313 {
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;
318 }
319
320 CAMLprim value ml_hasdata (value fd_v)
321 {
322 CAMLparam1 (fd_v);
323 int ret, avail;
324
325 ret = ioctl (Int_val (fd_v), FIONREAD, &avail);
326 if (ret) uerror ("ioctl (FIONREAD)", Nothing);
327 CAMLreturn (Val_bool (avail > 0));
328 }
329
330 static void readdata (void *p, int size)
331 {
332 ssize_t n;
333
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);
339 }
340 if (n - size) {
341 if (!n) errx (1, "EOF while reading");
342 errx (1, "read (req %d, ret %zd)", size, n);
343 }
344 }
345
346 static void writedata (char *p, int size)
347 {
348 char buf[4];
349 ssize_t n;
350
351 buf[0] = (size >> 24) & 0xff;
352 buf[1] = (size >> 16) & 0xff;
353 buf[2] = (size >> 8) & 0xff;
354 buf[3] = (size >> 0) & 0xff;
355
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);
360 }
361
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);
366 }
367 }
368
369 static int readlen (void)
370 {
371 unsigned char p[4];
372
373 readdata (p, 4);
374 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
375 }
376
377 static void GCC_FMT_ATTR (1, 2) printd (const char *fmt, ...)
378 {
379 int size = 200, len;
380 va_list ap;
381 char *buf;
382
383 buf = malloc (size);
384 for (;;) {
385 if (!buf) err (1, "malloc for temp buf (%d bytes) failed", size);
386
387 va_start (ap, fmt);
388 len = vsnprintf (buf, size, fmt, ap);
389 va_end (ap);
390
391 if (len > -1 && len < size) {
392 writedata (buf, len);
393 break;
394 }
395
396 if (len > -1) {
397 size = len + 1;
398 }
399 else {
400 size *= 2;
401 }
402 buf = realloc (buf, size);
403 }
404 free (buf);
405 }
406
407 static void closepdf (void)
408 {
409 if (state.u.pdf) {
410 pdf_close_document (state.u.pdf);
411 state.u.pdf = NULL;
412 }
413 }
414
415 static void closexps (void)
416 {
417 if (state.u.xps) {
418 xps_close_document (state.u.xps);
419 state.u.xps = NULL;
420 }
421 }
422
423 static void closecbz (void)
424 {
425 if (state.u.cbz) {
426 cbz_close_document (state.u.cbz);
427 state.u.cbz = NULL;
428 }
429 }
430
431 static void freepdfpage (void *ptr)
432 {
433 pdf_free_page (state.u.pdf, ptr);
434 }
435
436 static void freeemptypdfpage (void *ptr)
437 {
438 (void) ptr;
439 }
440
441 static void freexpspage (void *ptr)
442 {
443 xps_free_page (state.u.xps, ptr);
444 }
445
446 static void freecbzpage (void *ptr)
447 {
448 cbz_free_page (state.u.cbz, ptr);
449 }
450
451 static void openxref (char *filename, char *password)
452 {
453 int i, len;
454
455 for (i = 0; i < state.texcount; ++i) {
456 state.texowners[i].w = -1;
457 state.texowners[i].slice = NULL;
458 }
459
460 if (state.closedoc) state.closedoc ();
461
462 len = strlen (filename);
463
464 state.type = DPDF;
465 if (len > 4) {
466 char ext[4];
467
468 ext[0] = tolower ((int) filename[len-3]);
469 ext[1] = tolower ((int) filename[len-2]);
470 ext[2] = tolower ((int) filename[len-1]);
471
472 /**/ if (ext[0] == 'x' && ext[1] == 'p' && ext[2] == 's') {
473 state.type = DXPS;
474 }
475 else if (ext[0] == 'c' && ext[1] == 'b' && ext[2] == 'z') {
476 state.type = DCBZ;
477 }
478 }
479
480 if (state.pagedims) {
481 free (state.pagedims);
482 state.pagedims = NULL;
483 }
484 state.pagedimcount = 0;
485
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");
494 }
495 }
496 state.pagecount = pdf_count_pages (state.u.pdf);
497 state.closedoc = closepdf;
498 state.freepage = freepdfpage;
499 break;
500
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;
507
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;
514 }
515 }
516
517 static void pdfinfo (void)
518 {
519 if (state.type == DPDF) {
520 pdf_obj *infoobj;
521
522 printd ("info PDF version\t%d.%d",
523 state.u.pdf->version / 10, state.u.pdf->version % 10);
524
525 infoobj = pdf_dict_gets (state.u.pdf->trailer, "Info");
526 if (infoobj) {
527 int i;
528 char *s;
529 char *items[] = { "Title", "Author", "Creator",
530 "Producer", "CreationDate" };
531
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);
538 }
539 printd ("info %s\t%s", items[i], s);
540 }
541 fz_free (state.ctx, s);
542 }
543 }
544 printd ("infoend");
545 }
546 }
547
548 static void unlinktile (struct tile *tile)
549 {
550 int i;
551
552 for (i = 0; i < tile->slicecount; ++i) {
553 struct slice *s = &tile->slices[i];
554
555 if (s->texindex != -1) {
556 if (state.texowners[s->texindex].slice == s) {
557 state.texowners[s->texindex].slice = NULL;
558 }
559 }
560 }
561 }
562
563 static void freepage (struct page *page)
564 {
565 if (page->text) {
566 fz_free_text_page (state.ctx, page->text);
567 }
568 if (page->sheet) {
569 fz_free_text_sheet (state.ctx, page->sheet);
570 }
571 if (page->slinks) {
572 free (page->slinks);
573 }
574 page->freepage (page->u.ptr);
575 fz_free_display_list (state.ctx, page->dlist);
576 free (page);
577 }
578
579 static void freetile (struct tile *tile)
580 {
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);
588 }
589 state.pig = tile->pixmap;
590 #endif
591 }
592 free (tile);
593 }
594
595 #ifdef __ALTIVEC__
596 #include <altivec.h>
597
598 static int cacheline32bytes;
599
600 static void __attribute__ ((constructor)) clcheck (void)
601 {
602 char **envp = environ;
603 unsigned long *auxv;
604
605 while (*envp++);
606
607 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
608 if (*auxv == 19) {
609 cacheline32bytes = auxv[1] == 32;
610 return;
611 }
612 }
613 }
614
615 static void OPTIMIZE_ATTR (3) clearpixmap (fz_pixmap *pixmap)
616 {
617 size_t size = pixmap->w * pixmap->h * pixmap->n;
618 if (cacheline32bytes && size > 32) {
619 intptr_t a1, a2, diff;
620 size_t sizea, i;
621 vector unsigned char v = vec_splat_u8 (-1);
622 vector unsigned char *p;
623
624 a1 = a2 = (intptr_t) pixmap->samples;
625 a2 = (a1 + 31) & ~31;
626 diff = a2 - a1;
627 sizea = size - diff;
628 p = (void *) a2;
629
630 while (a1 != a2) *(char *) a1++ = 0xff;
631 for (i = 0; i < (sizea & ~31); i += 32) {
632 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
633 vec_st (v, i, p);
634 vec_st (v, i + 16, p);
635 }
636 while (i < sizea) *((char *) a1 + i++) = 0xff;
637 }
638 else fz_clear_pixmap_with_value (state.ctx, pixmap, 0xff);
639 }
640 #else
641 #define clearpixmap(p) fz_clear_pixmap_with_value (state.ctx, p, 0xff)
642 #endif
643
644 static fz_matrix trimctm (pdf_page *page, int pindex)
645 {
646 fz_matrix ctm;
647 struct pagedim *pdim = &state.pagedims[pindex];
648
649 if (!pdim->tctmready) {
650 if (state.trimmargins) {
651 fz_rect realbox;
652
653 ctm = fz_concat (fz_rotate (-pdim->rotate), fz_scale (1, -1));
654 realbox = fz_transform_rect (ctm, pdim->mediabox);
655 ctm = fz_concat (ctm, fz_translate (-realbox.x0, -realbox.y0));
656 ctm = fz_concat (fz_invert_matrix (page->ctm), ctm);
657 }
658 else {
659 ctm = fz_identity;
660 }
661 pdim->tctm = ctm;
662 pdim->tctmready = 1;
663 }
664 return pdim->tctm;
665 }
666
667 static fz_matrix pagectm (struct page *page)
668 {
669 if (page->type == DPDF) {
670 return fz_concat (trimctm (page->u.pdfpage, page->pdimno),
671 state.pagedims[page->pdimno].ctm);
672 }
673 else {
674 fz_matrix ctm;
675 struct pagedim *pdim = &state.pagedims[page->pdimno];
676
677 ctm = state.pagedims[page->pdimno].ctm;
678 ctm = fz_concat (fz_translate (-pdim->mediabox.x0,
679 -pdim->mediabox.y0), ctm);
680 return ctm;
681 }
682 }
683
684 static void *loadpage (int pageno, int pindex)
685 {
686 fz_device *dev;
687 struct page *page = NULL;
688
689 page = calloc (sizeof (struct page), 1);
690 if (!page) {
691 err (1, "calloc page %d", pageno);
692 }
693
694 page->dlist = fz_new_display_list (state.ctx);
695 dev = fz_new_list_device (state.ctx, page->dlist);
696 switch (state.type) {
697 case DPDF:
698 fz_try (state.ctx) {
699 page->u.pdfpage = pdf_load_page (state.u.pdf, pageno);
700 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, fz_identity, NULL);
701 page->freepage = freepdfpage;
702 }
703 fz_catch (state.ctx) {
704 page->u.ptr = NULL;
705 page->freepage = freeemptypdfpage;
706 }
707 break;
708
709 case DXPS:
710 page->u.xpspage = xps_load_page (state.u.xps, pageno);
711 xps_run_page (state.u.xps, page->u.xpspage, dev, fz_identity, NULL);
712 page->freepage = freexpspage;
713 break;
714
715 case DCBZ:
716 page->u.cbzpage = cbz_load_page (state.u.cbz, pageno);
717 cbz_run_page (state.u.cbz, page->u.cbzpage, dev, fz_identity, NULL);
718 page->freepage = freecbzpage;
719 break;
720 }
721 fz_free_device (dev);
722
723 page->pdimno = pindex;
724 page->pageno = pageno;
725 page->sgen = state.gen;
726 page->tgen = state.gen;
727 page->type = state.type;
728
729 return page;
730 }
731
732 static struct tile *alloctile (int h)
733 {
734 int i;
735 int slicecount;
736 size_t tilesize;
737 struct tile *tile;
738
739 slicecount = (h + state.sliceheight - 1) / state.sliceheight;
740 tilesize = sizeof (*tile) + ((slicecount - 1) * sizeof (struct slice));
741 tile = calloc (tilesize, 1);
742 if (!tile) {
743 err (1, "can not allocate tile (%" FMT_s " bytes)", tilesize);
744 }
745 for (i = 0; i < slicecount; ++i) {
746 int sh = MIN (h, state.sliceheight);
747 tile->slices[i].h = sh;
748 tile->slices[i].texindex = -1;
749 h -= sh;
750 }
751 tile->slicecount = slicecount;
752 tile->sliceheight = state.sliceheight;
753 return tile;
754 }
755
756 #ifdef OBSCURED_OPT
757 struct obs {
758 int cured;
759 fz_bbox b;
760 };
761
762 static void obs_fill_image (fz_device *dev, fz_image *image, fz_matrix ctm,
763 float alpha)
764 {
765 struct obs *obs = dev->user;
766
767 if (!obs->cured && fabs (1.0 - alpha) < 1e6) {
768 fz_bbox b = fz_round_rect (fz_transform_rect (ctm, fz_unit_rect));
769 b = fz_intersect_bbox (b, obs->b);
770 obs->cured = b.x0 == obs->b.x0
771 && b.x1 == obs->b.x1
772 && b.y0 == obs->b.y0
773 && b.y1 == obs->b.y1;
774 }
775 }
776
777 static int obscured (struct page *page, fz_bbox bbox)
778 {
779 fz_device dev;
780 struct obs obs;
781
782 memset (&dev, 0, sizeof (dev));
783 memset (&obs, 0, sizeof (obs));
784 dev.hints = 0;
785 dev.flags = 0;
786 dev.user = &obs;
787 dev.ctx = state.ctx;
788 dev.fill_image = obs_fill_image;
789 obs.b = bbox;
790 fz_run_display_list (page->dlist, &dev, pagectm (page), bbox, NULL);
791 return obs.cured;
792 }
793 #define OBSCURED obscured
794 #else
795 #define OBSCURED(a, b) 0
796 #endif
797
798 static struct tile *rendertile (struct page *page, int x, int y, int w, int h,
799 struct pbo *pbo)
800 {
801 fz_bbox bbox;
802 fz_device *dev;
803 struct tile *tile;
804 struct pagedim *pdim;
805
806 tile = alloctile (h);
807 pdim = &state.pagedims[page->pdimno];
808
809 bbox = pdim->bounds;
810 bbox.x0 += x;
811 bbox.y0 += y;
812 bbox.x1 = bbox.x0 + w;
813 bbox.y1 = bbox.y0 + h;
814
815 if (state.pig) {
816 if (state.pig->w == w
817 && state.pig->h == h
818 && state.pig->colorspace == state.colorspace) {
819 tile->pixmap = state.pig;
820 tile->pixmap->x = bbox.x0;
821 tile->pixmap->y = bbox.y0;
822 }
823 else {
824 fz_drop_pixmap (state.ctx, state.pig);
825 }
826 state.pig = NULL;
827 }
828 if (!tile->pixmap) {
829 if (pbo) {
830 tile->pixmap =
831 fz_new_pixmap_with_bbox_and_data (state.ctx, state.colorspace,
832 bbox, pbo->ptr);
833 tile->pbo = pbo;
834 }
835 else {
836 tile->pixmap =
837 fz_new_pixmap_with_bbox (state.ctx, state.colorspace, bbox);
838 }
839 }
840
841 tile->w = w;
842 tile->h = h;
843 if (!page->u.ptr || ((w < 128 && h < 128) || !OBSCURED (page, bbox))) {
844 clearpixmap (tile->pixmap);
845 }
846 dev = fz_new_draw_device (state.ctx, tile->pixmap);
847 fz_run_display_list (page->dlist, dev, pagectm (page), bbox, NULL);
848 fz_free_device (dev);
849
850 return tile;
851 }
852
853 static void initpdims (void)
854 {
855 double start, end;
856 int pageno, trim, show;
857 FILE *trimf = NULL;
858 int trimw = 0;
859
860 start = now ();
861
862 if (state.trimmargins && state.trimcachepath) {
863 trimf = fopen (state.trimcachepath, "rb");
864 if (!trimf) {
865 trimf = fopen (state.trimcachepath, "wb");
866 trimw = 1;
867 }
868 }
869 for (pageno = 0; pageno < state.pagecount; ++pageno) {
870 int rotate = 0;
871 struct pagedim *p;
872 fz_rect mediabox;
873
874 switch (state.type) {
875 case DPDF: {
876 pdf_obj *pageobj = state.u.pdf->page_objs[pageno];
877
878 if (state.trimmargins) {
879 pdf_obj *obj;
880 pdf_page *page;
881
882 fz_try (state.ctx) {
883 page = pdf_load_page (state.u.pdf, pageno);
884 obj = pdf_dict_gets (pageobj, "llpp.TrimBox");
885 trim = state.trimanew || !obj;
886 if (trim) {
887 fz_rect rect;
888 fz_bbox bbox;
889 fz_matrix ctm;
890 fz_device *dev;
891
892 dev = fz_new_bbox_device (state.ctx, &bbox);
893 dev->hints |= FZ_IGNORE_SHADE;
894 ctm = fz_invert_matrix (page->ctm);
895 pdf_run_page (state.u.pdf, page, dev, fz_identity, NULL);
896 fz_free_device (dev);
897
898 rect.x0 = bbox.x0 + state.trimfuzz.x0;
899 rect.x1 = bbox.x1 + state.trimfuzz.x1;
900 rect.y0 = bbox.y0 + state.trimfuzz.y0;
901 rect.y1 = bbox.y1 + state.trimfuzz.y1;
902 rect = fz_transform_rect (ctm, rect);
903 rect = fz_intersect_rect (rect, page->mediabox);
904
905 if (fz_is_empty_rect (rect)) {
906 mediabox = page->mediabox;
907 }
908 else {
909 mediabox = rect;
910 }
911
912 obj = pdf_new_array (state.ctx, 4);
913 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x0));
914 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y0));
915 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.x1));
916 pdf_array_push (obj, pdf_new_real (state.ctx, mediabox.y1));
917 pdf_dict_puts (pageobj, "llpp.TrimBox", obj);
918 }
919 else {
920 mediabox.x0 = pdf_to_real (pdf_array_get (obj, 0));
921 mediabox.y0 = pdf_to_real (pdf_array_get (obj, 1));
922 mediabox.x1 = pdf_to_real (pdf_array_get (obj, 2));
923 mediabox.y1 = pdf_to_real (pdf_array_get (obj, 3));
924 }
925
926 rotate = page->rotate;
927 pdf_free_page (state.u.pdf, page);
928
929 show = trim ? pageno % 5 == 0 : pageno % 20 == 0;
930 if (show) {
931 printd ("progress %f Trimming %d",
932 (double) (pageno + 1) / state.pagecount,
933 pageno + 1);
934 }
935 }
936 fz_catch (state.ctx) {
937 }
938 }
939 else {
940 fz_rect cropbox;
941
942 mediabox = pdf_to_rect (state.ctx,
943 pdf_dict_gets (pageobj, "MediaBox"));
944 if (fz_is_empty_rect (mediabox)) {
945 fprintf (stderr, "cannot find page size for page %d\n",
946 pageno+1);
947 mediabox.x0 = 0;
948 mediabox.y0 = 0;
949 mediabox.x1 = 612;
950 mediabox.y1 = 792;
951 }
952
953 cropbox = pdf_to_rect (state.ctx,
954 pdf_dict_gets (pageobj, "CropBox"));
955 if (!fz_is_empty_rect (cropbox)) {
956 mediabox = fz_intersect_rect (mediabox, cropbox);
957 }
958 rotate = pdf_to_int (pdf_dict_gets (pageobj, "Rotate"));
959 }
960 break;
961 }
962
963 case DXPS:
964 {
965 xps_page *page;
966
967 page = xps_load_page (state.u.xps, pageno);
968 mediabox = xps_bound_page (state.u.xps, page);
969 rotate = 0;
970 if (state.trimmargins) {
971 fz_rect rect;
972 fz_bbox bbox;
973 fz_device *dev;
974
975 dev = fz_new_bbox_device (state.ctx, &bbox);
976 dev->hints |= FZ_IGNORE_SHADE;
977 xps_run_page (state.u.xps, page, dev, fz_identity, NULL);
978 fz_free_device (dev);
979
980 rect.x0 = bbox.x0 + state.trimfuzz.x0;
981 rect.x1 = bbox.x1 + state.trimfuzz.x1;
982 rect.y0 = bbox.y0 + state.trimfuzz.y0;
983 rect.y1 = bbox.y1 + state.trimfuzz.y1;
984 rect = fz_intersect_rect (rect, mediabox);
985
986 if (!fz_is_empty_rect (rect)) {
987 mediabox = rect;
988 }
989 }
990 xps_free_page (state.u.xps, page);
991 printd ("progress %f loading %d",
992 (double) (pageno + 1) / state.pagecount,
993 pageno + 1);
994 }
995 break;
996
997 case DCBZ:
998 {
999 rotate = 0;
1000 if (state.trimmargins && trimw) {
1001 cbz_page *page;
1002
1003 page = cbz_load_page (state.u.cbz, pageno);
1004 mediabox = cbz_bound_page (state.u.cbz, page);
1005 cbz_free_page (state.u.cbz, page);
1006 printd ("progress %f Trimming %d",
1007 (double) (pageno + 1) / state.pagecount,
1008 pageno + 1);
1009 if (trimf) {
1010 int n = fwrite (&mediabox, sizeof (mediabox), 1, trimf);
1011 if (n - 1) {
1012 err (1, "fwrite trim mediabox");
1013 }
1014 }
1015 }
1016 else {
1017 if (trimf) {
1018 int n = fread (&mediabox, sizeof (mediabox), 1, trimf);
1019 if (n - 1) {
1020 err (1, "fread trim mediabox");
1021 }
1022 }
1023 else {
1024 mediabox.x0 = mediabox.y0 = 0;
1025 mediabox.x1 = 900;
1026 mediabox.y1 = 900;
1027 }
1028 }
1029 }
1030 break;
1031
1032 default:
1033 ARSERT (0 && state.type);
1034 }
1035
1036 if (state.pagedimcount == 0
1037 || (p = &state.pagedims[state.pagedimcount-1], p->rotate != rotate)
1038 || memcmp (&p->mediabox, &mediabox, sizeof (mediabox))) {
1039 size_t size;
1040
1041 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
1042 state.pagedims = realloc (state.pagedims, size);
1043 if (!state.pagedims) {
1044 err (1, "realloc pagedims to %" FMT_s " (%d elems)",
1045 size, state.pagedimcount + 1);
1046 }
1047
1048 p = &state.pagedims[state.pagedimcount++];
1049 p->rotate = rotate;
1050 p->mediabox = mediabox;
1051 p->pageno = pageno;
1052 }
1053 }
1054 end = now ();
1055 if (state.trimmargins) {
1056 printd ("progress 1 Trimmed %d pages in %f seconds",
1057 state.pagecount, end - start);
1058 }
1059 else {
1060 printd ("vmsg Processed %d pages in %f seconds",
1061 state.pagecount, end - start);
1062 }
1063 state.trimanew = 0;
1064 if (trimf) {
1065 if (fclose (trimf)) {
1066 err (1, "fclose");
1067 }
1068 }
1069 }
1070
1071 static void layout (void)
1072 {
1073 int pindex;
1074 fz_rect box;
1075 fz_matrix ctm;
1076 double zoom, w, maxw = 0;
1077 struct pagedim *p = state.pagedims;
1078
1079 if (state.proportional) {
1080 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
1081 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
1082 p->mediabox);
1083 w = box.x1 - box.x0;
1084 maxw = MAX (w, maxw);
1085 }
1086 }
1087
1088 p = state.pagedims;
1089 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
1090 fz_bbox bbox;
1091
1092 ctm = fz_rotate (state.rotate);
1093 box = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
1094 p->mediabox);
1095 w = box.x1 - box.x0;
1096
1097 if (state.proportional) {
1098 double scale = w / maxw;
1099 zoom = (state.w / w) * scale;
1100 }
1101 else {
1102 zoom = state.w / w;
1103 }
1104
1105 p->zoomctm = fz_scale (zoom, zoom);
1106 ctm = fz_concat (p->zoomctm, ctm);
1107
1108 p->pagebox = fz_transform_rect (fz_rotate (p->rotate), p->mediabox);
1109 p->pagebox.x1 -= p->pagebox.x0;
1110 p->pagebox.y1 -= p->pagebox.y0;
1111 p->pagebox.x0 = 0;
1112 p->pagebox.y0 = 0;
1113 bbox = fz_round_rect (fz_transform_rect (ctm, p->pagebox));
1114
1115 p->bounds = bbox;
1116 p->left = state.proportional ? ((maxw - w) * zoom) / 2.0 : 0;
1117 p->ctm = ctm;
1118
1119 ctm = fz_identity;
1120 ctm = fz_concat (ctm, fz_translate (0, -p->mediabox.y1));
1121 ctm = fz_concat (ctm, fz_scale (zoom, -zoom));
1122 ctm = fz_concat (ctm, fz_rotate (p->rotate + state.rotate));
1123 p->lctm = ctm;
1124
1125 p->tctmready = 0;
1126 }
1127
1128 while (p-- != state.pagedims) {
1129 int x0 = MIN (p->bounds.x0, p->bounds.x1);
1130 int y0 = MIN (p->bounds.y0, p->bounds.y1);
1131 int x1 = MAX (p->bounds.x0, p->bounds.x1);
1132 int y1 = MAX (p->bounds.y0, p->bounds.y1);
1133 int w = x1 - x0;
1134 int h = y1 - y0;
1135
1136 printd ("pdim %d %d %d %d", p->pageno, w, h, p->left);
1137 }
1138 }
1139
1140 static void recurse_outline (fz_outline *outline, int level)
1141 {
1142 while (outline) {
1143 fz_link_dest *dest;
1144 int i, top = 0;
1145 struct pagedim *pdim = state.pagedims;
1146
1147 dest = &outline->dest;
1148 for (i = 0; i < state.pagedimcount; ++i) {
1149 if (state.pagedims[i].pageno > dest->ld.gotor.page)
1150 break;
1151 pdim = &state.pagedims[i];
1152 }
1153 if (dest->ld.gotor.flags & fz_link_flag_t_valid) {
1154 fz_point p;
1155 p.x = 0;
1156 p.y = dest->ld.gotor.lt.y;
1157 p = fz_transform_point (pdim->lctm, p);
1158 top = p.y;
1159 }
1160 if (dest->ld.gotor.page >= 0 && dest->ld.gotor.page < 1<<30) {
1161 int h;
1162 double y0, y1;
1163
1164 y0 = MIN (pdim->bounds.y0, pdim->bounds.y1);
1165 y1 = MAX (pdim->bounds.y0, pdim->bounds.y1);
1166 h = y1 - y0;
1167 printd ("o %d %d %d %d %s",
1168 level, dest->ld.gotor.page, top, h, outline->title);
1169 }
1170 if (outline->down) {
1171 recurse_outline (outline->down, level + 1);
1172 }
1173 outline = outline->next;
1174 }
1175 }
1176
1177 static void process_outline (void)
1178 {
1179 fz_outline *outline;
1180
1181 if (!state.needoutline) return;
1182
1183 state.needoutline = 0;
1184 switch (state.type) {
1185 case DPDF:
1186 outline = pdf_load_outline (state.u.pdf);
1187 break;
1188 case DXPS:
1189 outline = xps_load_outline (state.u.xps);
1190 break;
1191 default:
1192 outline = NULL;
1193 break;
1194 }
1195 if (outline) {
1196 recurse_outline (outline, 0);
1197 fz_free_outline (state.ctx, outline);
1198 }
1199 }
1200
1201 static char *strofspan (fz_text_span *span)
1202 {
1203 char *p;
1204 char utf8[10];
1205 fz_text_char *ch;
1206 size_t size = 0, cap = 80;
1207
1208 p = malloc (cap + 1);
1209 if (!p) return NULL;
1210
1211 for (ch = span->text; ch < span->text + span->len; ++ch) {
1212 int n = fz_runetochar (utf8, ch->c);
1213 if (size + n > cap) {
1214 cap *= 2;
1215 p = realloc (p, cap + 1);
1216 if (!p) return NULL;
1217 }
1218
1219 memcpy (p + size, utf8, n);
1220 size += n;
1221 }
1222 p[size] = 0;
1223 return p;
1224 }
1225
1226 static int matchspan (regex_t *re, fz_text_span *span, fz_matrix ctm,
1227 int stop, int pageno, double start)
1228 {
1229 int ret;
1230 char *p;
1231 regmatch_t rm;
1232 int a, b, c;
1233 fz_rect *sb, *eb;
1234 fz_point p1, p2, p3, p4;
1235
1236 p = strofspan (span);
1237 if (!p) return -1;
1238
1239 ret = regexec (re, p, 1, &rm, 0);
1240 if (ret) {
1241 free (p);
1242 if (ret != REG_NOMATCH) {
1243 size_t size;
1244 char errbuf[80];
1245 size = regerror (ret, re, errbuf, sizeof (errbuf));
1246 printd ("msg regexec error `%.*s'",
1247 (int) size, errbuf);
1248 return -1;
1249 }
1250 return 0;
1251 }
1252 else {
1253 int l = span->len;
1254 for (a = 0, c = 0; c < rm.rm_so && a < l; a++) {
1255 c += fz_runelen (span->text[a].c);
1256 }
1257 for (b = a; c < rm.rm_eo - 1 && b < l; b++) {
1258 c += fz_runelen (span->text[b].c);
1259 }
1260
1261 if (fz_runelen (span->text[b].c) > 1) {
1262 b = MAX (0, b-1);
1263 }
1264 sb = &span->text[MIN (a, l-1)].bbox;
1265 eb = &span->text[MIN (b, l-1)].bbox;
1266
1267 p1.x = sb->x0;
1268 p1.y = sb->y0;
1269 p2.x = eb->x1;
1270 p2.y = sb->y0;
1271 p3.x = eb->x1;
1272 p3.y = eb->y1;
1273 p4.x = sb->x0;
1274 p4.y = eb->y1;
1275
1276 if (!stop) {
1277 printd ("firstmatch %d %d %f %f %f %f %f %f %f %f",
1278 pageno, 1,
1279 p1.x, p1.y,
1280 p2.x, p2.y,
1281 p3.x, p3.y,
1282 p4.x, p4.y);
1283
1284 printd ("progress 1 found at %d `%.*s' in %f sec",
1285 pageno + 1, (int) (rm.rm_eo - rm.rm_so), &p[rm.rm_so],
1286 now () - start);
1287 }
1288 else {
1289 printd ("match %d %d %f %f %f %f %f %f %f %f",
1290 pageno, 2,
1291 p1.x, p1.y,
1292 p2.x, p2.y,
1293 p3.x, p3.y,
1294 p4.x, p4.y);
1295 }
1296 free (p);
1297 return 1;
1298 }
1299 }
1300
1301 static int compareblocks (const void *l, const void *r)
1302 {
1303 fz_text_block const *ls = l;
1304 fz_text_block const* rs = r;
1305 return ls->bbox.y0 - rs->bbox.y0;
1306 }
1307
1308 /* wishful thinking function */
1309 static void search (regex_t *re, int pageno, int y, int forward)
1310 {
1311 int i, j;
1312 fz_matrix ctm;
1313 fz_device *tdev;
1314 union { void *ptr; pdf_page *pdfpage; xps_page *xpspage; } u;
1315 fz_text_page *text;
1316 fz_text_sheet *sheet;
1317 struct pagedim *pdim, *pdimprev;
1318 int stop = 0, niters = 0;
1319 double start, end;
1320
1321 if (!(state.type == DPDF || state.type == DXPS))
1322 return;
1323
1324 start = now ();
1325 while (pageno >= 0 && pageno < state.pagecount && !stop) {
1326 if (niters++ == 5) {
1327 niters = 0;
1328 if (hasdata ()) {
1329 printd ("progress 1 attention requested aborting search at %d",
1330 pageno);
1331 stop = 1;
1332 }
1333 else {
1334 printd ("progress %f searching in page %d",
1335 (double) (pageno + 1) / state.pagecount,
1336 pageno);
1337 }
1338 }
1339 pdimprev = NULL;
1340 for (i = 0; i < state.pagedimcount; ++i) {
1341 pdim = &state.pagedims[i];
1342 if (pdim->pageno == pageno) {
1343 goto found;
1344 }
1345 if (pdim->pageno > pageno) {
1346 pdim = pdimprev;
1347 goto found;
1348 }
1349 pdimprev = pdim;
1350 }
1351 pdim = pdimprev;
1352 found:
1353
1354 sheet = fz_new_text_sheet (state.ctx);
1355 text = fz_new_text_page (state.ctx, fz_infinite_rect);
1356 tdev = fz_new_text_device (state.ctx, sheet, text);
1357
1358 switch (state.type) {
1359 case DPDF:
1360 u.ptr = NULL;
1361 fz_try (state.ctx) {
1362 u.pdfpage = pdf_load_page (state.u.pdf, pageno);
1363 trimctm (u.pdfpage, pdim - state.pagedims);
1364 ctm = fz_concat (pdim->tctm, pdim->zoomctm);
1365 pdf_run_page (state.u.pdf, u.pdfpage, tdev, ctm, NULL);
1366 }
1367 fz_catch (state.ctx) {
1368 fz_free_device (tdev);
1369 u.ptr = NULL;
1370 goto nextiter;
1371 }
1372 break;
1373
1374 case DXPS:
1375 u.xpspage = xps_load_page (state.u.xps, pageno);
1376 ctm = pdim->ctm;
1377 xps_run_page (state.u.xps, u.xpspage, tdev, ctm, NULL);
1378 break;
1379
1380 default:
1381 ARSERT (0 && state.type);
1382 }
1383
1384 qsort (text->blocks, text->len, sizeof (*text->blocks), compareblocks);
1385 fz_free_device (tdev);
1386
1387 for (j = 0; j < text->len; ++j) {
1388 int k;
1389 fz_text_block *block;
1390
1391 block = &text->blocks[forward ? j : text->len - 1 - j];
1392
1393 for (k = 0; k < block->len; ++k) {
1394 fz_text_line *line;
1395 fz_text_span *span;
1396
1397 if (forward) {
1398 line = &block->lines[k];
1399 if (line->bbox.y0 < y + 1) continue;
1400 }
1401 else {
1402 line = &block->lines[block->len - 1 - k];
1403 if (line->bbox.y0 > y - 1) continue;
1404 }
1405
1406 for (span = line->spans;
1407 span < line->spans + line->len;
1408 ++span) {
1409
1410 switch (matchspan (re, span, ctm, stop, pageno, start)) {
1411 case 0: break;
1412 case 1: stop = 1; break;
1413 case -1: stop = 1; goto endloop;
1414 }
1415 }
1416 }
1417 }
1418 nextiter:
1419 if (forward) {
1420 pageno += 1;
1421 y = 0;
1422 }
1423 else {
1424 pageno -= 1;
1425 y = INT_MAX;
1426 }
1427 endloop:
1428 fz_free_text_page (state.ctx, text);
1429 fz_free_text_sheet (state.ctx, sheet);
1430 if (u.ptr) {
1431 state.freepage (u.ptr);
1432 }
1433 }
1434 end = now ();
1435 if (!stop) {
1436 printd ("progress 1 no matches %f sec", end - start);
1437 }
1438 printd ("clearrects");
1439 }
1440
1441 static void set_tex_params (int colorspace)
1442 {
1443 union {
1444 unsigned char b;
1445 unsigned int s;
1446 } endianness = {1};
1447
1448 switch (colorspace) {
1449 case 0:
1450 state.texiform = GL_RGBA8;
1451 state.texform = GL_RGBA;
1452 state.texty = GL_UNSIGNED_BYTE;
1453 state.colorspace = fz_device_rgb;
1454 break;
1455 case 1:
1456 state.texiform = GL_RGBA8;
1457 state.texform = GL_BGRA;
1458 state.texty = endianness.s > 1
1459 ? GL_UNSIGNED_INT_8_8_8_8
1460 : GL_UNSIGNED_INT_8_8_8_8_REV;
1461 state.colorspace = fz_device_bgr;
1462 break;
1463 case 2:
1464 state.texiform = GL_LUMINANCE_ALPHA;
1465 state.texform = GL_LUMINANCE_ALPHA;
1466 state.texty = GL_UNSIGNED_BYTE;
1467 state.colorspace = fz_device_gray;
1468 break;
1469 default:
1470 errx (1, "invalid colorspce %d", colorspace);
1471 }
1472 }
1473
1474 static void realloctexts (int texcount)
1475 {
1476 size_t size;
1477
1478 if (texcount == state.texcount) return;
1479
1480 if (texcount < state.texcount) {
1481 glDeleteTextures (state.texcount - texcount,
1482 state.texids + texcount);
1483 }
1484
1485 size = texcount * sizeof (*state.texids);
1486 state.texids = realloc (state.texids, size);
1487 if (!state.texids) {
1488 err (1, "realloc texids %" FMT_s, size);
1489 }
1490
1491 size = texcount * sizeof (*state.texowners);
1492 state.texowners = realloc (state.texowners, size);
1493 if (!state.texowners) {
1494 err (1, "realloc texowners %" FMT_s, size);
1495 }
1496 if (texcount > state.texcount) {
1497 int i;
1498
1499 glGenTextures (texcount - state.texcount,
1500 state.texids + state.texcount);
1501 for (i = state.texcount; i < texcount; ++i) {
1502 state.texowners[i].w = -1;
1503 state.texowners[i].slice = NULL;
1504 }
1505 }
1506 state.texcount = texcount;
1507 state.texindex = 0;
1508 }
1509
1510 static char *mbtoutf8 (char *s)
1511 {
1512 char *p, *r;
1513 wchar_t *tmp;
1514 size_t i, ret, len;
1515
1516 len = mbstowcs (NULL, s, strlen (s));
1517 if (len == 0) {
1518 return s;
1519 }
1520 else {
1521 if (len == (size_t) -1) {
1522 return s;
1523 }
1524 }
1525
1526 tmp = malloc (len * sizeof (wchar_t));
1527 if (!tmp) {
1528 return s;
1529 }
1530
1531 ret = mbstowcs (tmp, s, len);
1532 if (ret == (size_t) -1) {
1533 free (tmp);
1534 return s;
1535 }
1536
1537 len = 0;
1538 for (i = 0; i < ret; ++i) {
1539 len += fz_runelen (tmp[i]);
1540 }
1541
1542 p = r = malloc (len + 1);
1543 if (!r) {
1544 free (tmp);
1545 return s;
1546 }
1547
1548 for (i = 0; i < ret; ++i) {
1549 p += fz_runetochar (p, tmp[i]);
1550 }
1551 *p = 0;
1552 free (tmp);
1553 return r;
1554 }
1555
1556 CAMLprim value ml_mbtoutf8 (value s_v)
1557 {
1558 CAMLparam1 (s_v);
1559 CAMLlocal1 (ret_v);
1560 char *s, *r;
1561
1562 s = String_val (s_v);
1563 r = mbtoutf8 (s);
1564 if (r == s) {
1565 ret_v = s_v;
1566 }
1567 else {
1568 ret_v = caml_copy_string (r);
1569 free (r);
1570 }
1571 CAMLreturn (ret_v);
1572 }
1573
1574 static void * mainloop (void *unused)
1575 {
1576 char *p = NULL;
1577 int len, ret, oldlen = 0;
1578
1579 for (;;) {
1580 len = readlen ();
1581 if (len == 0) {
1582 errx (1, "readlen returned 0");
1583 }
1584
1585 if (oldlen < len + 1) {
1586 p = realloc (p, len + 1);
1587 if (!p) {
1588 err (1, "realloc %d failed", len + 1);
1589 }
1590 oldlen = len + 1;
1591 }
1592 readdata (p, len);
1593 p[len] = 0;
1594
1595 if (!strncmp ("open", p, 4)) {
1596 size_t filenamelen;
1597 char *password;
1598 char *filename = p + 5;
1599 char *utf8filename;
1600
1601 filenamelen = strlen (filename);
1602 password = filename + filenamelen + 1;
1603
1604 openxref (filename, password);
1605 pdfinfo ();
1606 initpdims ();
1607 utf8filename = mbtoutf8 (filename);
1608 printd ("msg Opened %s (press h/F1 to get help)", utf8filename);
1609 if (utf8filename != filename) {
1610 free (utf8filename);
1611 }
1612 state.needoutline = 1;
1613 }
1614 else if (!strncmp ("cs", p, 2)) {
1615 int i, colorspace;
1616
1617 ret = sscanf (p + 2, " %d", &colorspace);
1618 if (ret != 1) {
1619 errx (1, "malformed cs `%.*s' ret=%d", len, p, ret);
1620 }
1621 lock ("cs");
1622 set_tex_params (colorspace);
1623 for (i = 0; i < state.texcount; ++i) {
1624 state.texowners[i].w = -1;
1625 state.texowners[i].slice = NULL;
1626 }
1627 unlock ("cs");
1628 }
1629 else if (!strncmp ("freepage", p, 8)) {
1630 void *ptr;
1631
1632 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1633 if (ret != 1) {
1634 errx (1, "malformed freepage `%.*s' ret=%d", len, p, ret);
1635 }
1636 freepage (ptr);
1637 }
1638 else if (!strncmp ("freetile", p, 8)) {
1639 void *ptr;
1640
1641 ret = sscanf (p + 8, " %" FMT_ptr, FMT_ptr_cast (&ptr));
1642 if (ret != 1) {
1643 errx (1, "malformed freetile `%.*s' ret=%d", len, p, ret);
1644 }
1645 freetile (ptr);
1646 }
1647 else if (!strncmp ("search", p, 6)) {
1648 int icase, pageno, y, ret, len2, forward;
1649 char *pattern;
1650 regex_t re;
1651
1652 ret = sscanf (p + 6, " %d %d %d %d,%n",
1653 &icase, &pageno, &y, &forward, &len2);
1654 if (ret != 4) {
1655 errx (1, "malformed search `%s' ret=%d", p, ret);
1656 }
1657
1658 pattern = p + 6 + len2;
1659 ret = regcomp (&re, pattern,
1660 REG_EXTENDED | (icase ? REG_ICASE : 0));
1661 if (ret) {
1662 char errbuf[80];
1663 size_t size;
1664
1665 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1666 printd ("msg regcomp failed `%.*s'", (int) size, errbuf);
1667 }
1668 else {
1669 search (&re, pageno, y, forward);
1670 regfree (&re);
1671 }
1672 }
1673 else if (!strncmp ("geometry", p, 8)) {
1674 int w, h;
1675
1676 printd ("clear");
1677 ret = sscanf (p + 8, " %d %d", &w, &h);
1678 if (ret != 2) {
1679 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1680 }
1681
1682 lock ("geometry");
1683 state.h = h;
1684 if (w != state.w) {
1685 int i;
1686 state.w = w;
1687 for (i = 0; i < state.texcount; ++i) {
1688 state.texowners[i].slice = NULL;
1689 }
1690 }
1691 layout ();
1692 process_outline ();
1693 state.gen++;
1694 unlock ("geometry");
1695 printd ("continue %d", state.pagecount);
1696 }
1697 else if (!strncmp ("reqlayout", p, 9)) {
1698 int rotate, proportional;
1699
1700 printd ("clear");
1701 ret = sscanf (p + 9, " %d %d", &rotate, &proportional);
1702 if (ret != 2) {
1703 errx (1, "bad reqlayout line `%.*s' ret=%d", len, p, ret);
1704 }
1705 lock ("reqlayout");
1706 if (state.rotate != rotate || state.proportional != proportional) {
1707 state.gen += 1;
1708 }
1709 state.rotate = rotate;
1710 state.proportional = proportional;
1711 layout ();
1712 process_outline ();
1713 state.gen++;
1714 unlock ("reqlayout");
1715 printd ("continue %d", state.pagecount);
1716 }
1717 else if (!strncmp ("page", p, 4)) {
1718 double a, b;
1719 struct page *page;
1720 int pageno, pindex, ret;
1721
1722 ret = sscanf (p + 4, " %d %d", &pageno, &pindex);
1723 if (ret != 2) {
1724 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1725 }
1726
1727 lock ("page");
1728 a = now ();
1729 page = loadpage (pageno, pindex);
1730 b = now ();
1731 unlock ("page");
1732
1733 printd ("page %" FMT_ptr " %f", FMT_ptr_cast2 (page), b - a);
1734 }
1735 else if (!strncmp ("tile", p, 4)) {
1736 int x, y, w, h, ret;
1737 struct page *page;
1738 struct tile *tile;
1739 double a, b;
1740 void *data;
1741
1742 ret = sscanf (p + 4, " %" FMT_ptr " %d %d %d %d %" FMT_ptr,
1743 FMT_ptr_cast (&page), &x, &y, &w, &h, &data);
1744 if (ret != 6) {
1745 errx (1, "bad tile line `%.*s' ret=%d", len, p, ret);
1746 }
1747
1748 lock ("tile");
1749 a = now ();
1750 tile = rendertile (page, x, y, w, h, data);
1751 b = now ();
1752 unlock ("tile");
1753
1754 printd ("tile %d %d %" FMT_ptr " %u %f",
1755 x, y,
1756 FMT_ptr_cast2 (tile),
1757 tile->w * tile->h * tile->pixmap->n,
1758 b - a);
1759 }
1760 else if (!strncmp ("settrim", p, 7)) {
1761 int trimmargins;
1762 fz_bbox fuzz;
1763
1764 ret = sscanf (p + 7, " %d %d %d %d %d", &trimmargins,
1765 &fuzz.x0, &fuzz.y0, &fuzz.x1, &fuzz.y1);
1766 if (ret != 5) {
1767 errx (1, "malformed settrim `%.*s' ret=%d", len, p, ret);
1768 }
1769 printd ("clear");
1770 lock ("settrim");
1771 state.trimmargins = trimmargins;
1772 state.needoutline = 1;
1773 if (memcmp (&fuzz, &state.trimfuzz, sizeof (fuzz))) {
1774 state.trimanew = 1;
1775 state.trimfuzz = fuzz;
1776 }
1777 state.pagedimcount = 0;
1778 free (state.pagedims);
1779 state.pagedims = NULL;
1780 initpdims ();
1781 layout ();
1782 process_outline ();
1783 unlock ("settrim");
1784 printd ("continue %d", state.pagecount);
1785 }
1786 else if (!strncmp ("sliceh", p, 6)) {
1787 int h;
1788
1789 ret = sscanf (p + 6, " %d", &h);
1790 if (ret != 1) {
1791 errx (1, "malformed sliceh `%.*s' ret=%d", len, p, ret);
1792 }
1793 if (h != state.sliceheight) {
1794 int i;
1795
1796 state.sliceheight = h;
1797 for (i = 0; i < state.texcount; ++i) {
1798 state.texowners[i].w = -1;
1799 state.texowners[i].h = -1;
1800 state.texowners[i].slice = NULL;
1801 }
1802 }
1803 }
1804 else if (!strncmp ("interrupt", p, 9)) {
1805 printd ("vmsg interrupted");
1806 }
1807 else {
1808 errx (1, "unknown command %.*s", len, p);
1809 }
1810 }
1811 return 0;
1812 }
1813
1814 CAMLprim value ml_realloctexts (value texcount_v)
1815 {
1816 CAMLparam1 (texcount_v);
1817 int ok;
1818
1819 if (trylock ("ml_realloctexts")) {
1820 ok = 0;
1821 goto done;
1822 }
1823 realloctexts (Int_val (texcount_v));
1824 ok = 1;
1825 unlock ("ml_realloctexts");
1826
1827 done:
1828 CAMLreturn (Val_bool (ok));
1829 }
1830
1831 static void showsel (struct page *page, int ox, int oy)
1832 {
1833 int seen = 0;
1834 fz_bbox bbox;
1835 fz_text_line *line;
1836 fz_text_span *span;
1837 fz_text_block *block;
1838 struct mark first, last;
1839
1840 first = page->fmark;
1841 last = page->lmark;
1842
1843 if (!first.span || !last.span) return;
1844
1845 glEnable (GL_BLEND);
1846 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1847 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1848
1849 ox += state.pagedims[page->pdimno].bounds.x0;
1850 oy += state.pagedims[page->pdimno].bounds.y0;
1851 for (block = page->text->blocks;
1852 block < page->text->blocks + page->text->len;
1853 ++block) {
1854 for (line = block->lines;
1855 line < block->lines + block->len;
1856 ++line) {
1857 for (span = line->spans;
1858 span < line->spans + line->len;
1859 ++span) {
1860 int i, j, k;
1861
1862 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
1863
1864 j = 0;
1865 k = span->len - 1;
1866
1867 if (span == page->fmark.span && span == page->lmark.span) {
1868 seen = 1;
1869 j = MIN (first.i, last.i);
1870 k = MAX (first.i, last.i);
1871 }
1872 else if (span == first.span) {
1873 seen = 1;
1874 j = first.i;
1875 }
1876 else if (span == last.span) {
1877 seen = 1;
1878 k = last.i;
1879 }
1880
1881 if (seen) {
1882 for (i = j; i <= k; ++i) {
1883 fz_bbox b = fz_round_rect (span->text[i].bbox);
1884 bbox = fz_union_bbox (bbox, b);
1885 }
1886 lprintf ("%d %d %d %d oy=%d ox=%d\n",
1887 bbox.x0,
1888 bbox.y0,
1889 bbox.x1,
1890 bbox.y1,
1891 oy, ox);
1892
1893 glRecti (bbox.x0 + ox, bbox.y0 + oy,
1894 bbox.x1 + ox, bbox.y1 + oy);
1895 if (span == last.span) {
1896 goto done;
1897 }
1898 }
1899 }
1900 }
1901 }
1902 done:
1903 glDisable (GL_BLEND);
1904 }
1905
1906 #include "glfont.c"
1907
1908 static void highlightlinks (struct page *page, int xoff, int yoff)
1909 {
1910 fz_matrix ctm;
1911 fz_link *link, *links;
1912
1913 switch (page->type) {
1914 case DPDF:
1915 links = page->u.pdfpage->links;
1916 break;
1917
1918 case DXPS:
1919 links = page->u.xpspage->links;
1920 break;
1921
1922 default:
1923 return;
1924 }
1925
1926 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1927 glEnable (GL_LINE_STIPPLE);
1928 glLineStipple (0.5, 0xcccc);
1929
1930 xoff -= state.pagedims[page->pdimno].bounds.x0;
1931 yoff -= state.pagedims[page->pdimno].bounds.y0;
1932 ctm = fz_concat (pagectm (page), fz_translate (xoff, yoff));
1933
1934 glBegin (GL_QUADS);
1935 for (link = links; link; link = link->next) {
1936 fz_point p1, p2, p3, p4;
1937
1938 p1.x = link->rect.x0;
1939 p1.y = link->rect.y0;
1940
1941 p2.x = link->rect.x1;
1942 p2.y = link->rect.y0;
1943
1944 p3.x = link->rect.x1;
1945 p3.y = link->rect.y1;
1946
1947 p4.x = link->rect.x0;
1948 p4.y = link->rect.y1;
1949
1950 p1 = fz_transform_point (ctm, p1);
1951 p2 = fz_transform_point (ctm, p2);
1952 p3 = fz_transform_point (ctm, p3);
1953 p4 = fz_transform_point (ctm, p4);
1954
1955 switch (link->dest.kind) {
1956 case FZ_LINK_GOTO: glColor3ub (255, 0, 0); break;
1957 case FZ_LINK_URI: glColor3ub (0, 0, 255); break;
1958 default: glColor3ub (0, 0, 0); break;
1959 }
1960
1961 glVertex2f (p1.x, p1.y);
1962 glVertex2f (p2.x, p2.y);
1963 glVertex2f (p3.x, p3.y);
1964 glVertex2f (p4.x, p4.y);
1965 }
1966 glEnd ();
1967
1968 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1969 glDisable (GL_LINE_STIPPLE);
1970 }
1971
1972 static int compareslinks (const void *l, const void *r)
1973 {
1974 struct slink const *ls = l;
1975 struct slink const *rs = r;
1976 if (ls->bbox.y0 == rs->bbox.y0) {
1977 return rs->bbox.x0 - rs->bbox.x0;
1978 }
1979 return ls->bbox.y0 - rs->bbox.y0;
1980 }
1981
1982 static void droptext (struct page *page)
1983 {
1984 if (page->text) {
1985 fz_free_text_page (state.ctx, page->text);
1986 page->fmark.i = -1;
1987 page->lmark.i = -1;
1988 page->fmark.span = NULL;
1989 page->lmark.span = NULL;
1990 page->text = NULL;
1991 }
1992 if (page->sheet) {
1993 fz_free_text_sheet (state.ctx, page->sheet);
1994 }
1995 }
1996
1997 static void dropslinks (struct page *page)
1998 {
1999 if (page->slinks) {
2000 free (page->slinks);
2001 page->slinks = NULL;
2002 page->slinkcount = 0;
2003 }
2004 }
2005
2006 static void ensureslinks (struct page *page)
2007 {
2008 fz_matrix ctm;
2009 int i, count = 0;
2010 size_t slinksize = sizeof (*page->slinks);
2011 fz_link *link, *links;
2012
2013 if (state.gen != page->sgen) {
2014 dropslinks (page);
2015 page->sgen = state.gen;
2016 }
2017 if (page->slinks) return;
2018
2019 switch (page->type) {
2020 case DPDF:
2021 links = page->u.pdfpage->links;
2022 ctm = fz_concat (trimctm (page->u.pdfpage, page->pdimno),
2023 state.pagedims[page->pdimno].ctm);
2024 break;
2025
2026 case DXPS:
2027 links = page->u.xpspage->links;
2028 ctm = state.pagedims[page->pdimno].ctm;
2029 break;
2030
2031 default:
2032 return;
2033 }
2034
2035 for (link = links; link; link = link->next) {
2036 count++;
2037 }
2038 if (count > 0) {
2039 page->slinkcount = count;
2040 page->slinks = calloc (count, slinksize);
2041 if (!page->slinks) {
2042 err (1, "realloc slinks %d", count);
2043 }
2044
2045 for (i = 0, link = links; link; ++i, link = link->next) {
2046 page->slinks[i].link = link;
2047 page->slinks[i].bbox =
2048 fz_round_rect (fz_transform_rect (ctm, link->rect));
2049 }
2050 qsort (page->slinks, count, slinksize, compareslinks);
2051 }
2052 }
2053
2054 /* slightly tweaked fmt_ulong by D.J. Bernstein */
2055 static void fmt_linkn (char *s, unsigned int u)
2056 {
2057 unsigned int len; unsigned int q;
2058 int zma = 'z' - 'a' + 1;
2059 len = 1; q = u;
2060 while (q > zma - 1) { ++len; q /= zma; }
2061 if (s) {
2062 s += len;
2063 do { *--s = 'a' + (u % zma) - (u < zma && len > 1); u /= zma; } while(u);
2064 /* handles u == 0 */
2065 }
2066 s[len] = 0;
2067 }
2068
2069 static void highlightslinks (struct page *page, int xoff, int yoff,
2070 int noff, char *targ, int tlen, int hfsize)
2071 {
2072 int i;
2073 char buf[40];
2074 struct slink *slink;
2075 double x0, y0, x1, y1, w;
2076
2077 ensureslinks (page);
2078 glColor3ub (0xc3, 0xb0, 0x91);
2079 for (i = 0; i < page->slinkcount; ++i) {
2080 fmt_linkn (buf, i + noff);
2081 if (!tlen || !strncmp (targ, buf, tlen)) {
2082 slink = &page->slinks[i];
2083
2084 x0 = slink->bbox.x0 + xoff - 5;
2085 y1 = slink->bbox.y0 + yoff - 5;
2086 y0 = y1 + 10 + hfsize;
2087 w = measure_string (state.face, hfsize, buf);
2088 x1 = x0 + w + 10;
2089 glRectd (x0, y0, x1, y1);
2090 }
2091 }
2092
2093 glEnable (GL_BLEND);
2094 glBlendFunc (GL_SRC_ALPHA, GL_ONE_MINUS_SRC_ALPHA);
2095 glEnable (GL_TEXTURE_2D);
2096 glColor3ub (0, 0, 0);
2097 for (i = 0; i < page->slinkcount; ++i) {
2098 fmt_linkn (buf, i + noff);
2099 if (!tlen || !strncmp (targ, buf, tlen)) {
2100 slink = &page->slinks[i];
2101
2102 x0 = slink->bbox.x0 + xoff;
2103 y0 = slink->bbox.y0 + yoff + hfsize;
2104 draw_string (state.face, hfsize, x0, y0, buf);
2105 }
2106 }
2107 glDisable (GL_TEXTURE_2D);
2108 glDisable (GL_BLEND);
2109 }
2110
2111
2112 static void uploadslice (struct tile *tile, struct slice *slice)
2113 {
2114 int offset;
2115 struct slice *slice1;
2116 unsigned char *texdata;
2117
2118 offset = 0;
2119 for (slice1 = tile->slices; slice != slice1; slice1++) {
2120 offset += slice1->h * tile->w * tile->pixmap->n;
2121 }
2122 if (slice->texindex != -1 && slice->texindex < state.texcount
2123 && state.texowners[slice->texindex].slice == slice) {
2124 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
2125 }
2126 else {
2127 int subimage = 0;
2128 int texindex = state.texindex++ % state.texcount;
2129
2130 if (state.texowners[texindex].w == tile->w) {
2131 if (state.texowners[texindex].h >= slice->h) {
2132 subimage = 1;
2133 }
2134 else {
2135 state.texowners[texindex].h = slice->h;
2136 }
2137 }
2138 else {
2139 state.texowners[texindex].h = slice->h;
2140 }
2141
2142 state.texowners[texindex].w = tile->w;
2143 state.texowners[texindex].slice = slice;
2144 slice->texindex = texindex;
2145
2146 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[texindex]);
2147 if (tile->pbo) {
2148 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
2149 texdata = 0;
2150 }
2151 else {
2152 texdata = tile->pixmap->samples;
2153 }
2154 if (subimage) {
2155 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
2156 0,
2157 0,
2158 0,
2159 tile->w,
2160 slice->h,
2161 state.texform,
2162 state.texty,
2163 texdata+offset
2164 );
2165 }
2166 else {
2167 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
2168 0,
2169 state.texiform,
2170 tile->w,
2171 slice->h,
2172 0,
2173 state.texform,
2174 state.texty,
2175 texdata+offset
2176 );
2177 }
2178 if (tile->pbo) {
2179 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
2180 }
2181 }
2182 }
2183
2184 CAMLprim value ml_drawtile (value args_v, value ptr_v)
2185 {
2186 CAMLparam2 (args_v, ptr_v);
2187 int dispx = Int_val (Field (args_v, 0));
2188 int dispy = Int_val (Field (args_v, 1));
2189 int dispw = Int_val (Field (args_v, 2));
2190 int disph = Int_val (Field (args_v, 3));
2191 int tilex = Int_val (Field (args_v, 4));
2192 int tiley = Int_val (Field (args_v, 5));
2193 char *s = String_val (ptr_v);
2194 struct tile *tile = parse_pointer ("ml_drawtile", s);
2195
2196 glEnable (GL_TEXTURE_RECTANGLE_ARB);
2197 {
2198 int slicey, firstslice;
2199 struct slice *slice;
2200
2201 firstslice = tiley / tile->sliceheight;
2202 slice = &tile->slices[firstslice];
2203 slicey = tiley % tile->sliceheight;
2204
2205 while (disph > 0) {
2206 int dh;
2207
2208 dh = slice->h - slicey;
2209 dh = MIN (disph, dh);
2210 uploadslice (tile, slice);
2211
2212 glBegin (GL_QUADS);
2213 {
2214 glTexCoord2i (tilex, slicey);
2215 glVertex2i (dispx, dispy);
2216
2217 glTexCoord2i (tilex+dispw, slicey);
2218 glVertex2i (dispx+dispw, dispy);
2219
2220 glTexCoord2i (tilex+dispw, slicey+dh);
2221 glVertex2i (dispx+dispw, dispy+dh);
2222
2223 glTexCoord2i (tilex, slicey+dh);
2224 glVertex2i (dispx, dispy+dh);
2225 }
2226 glEnd ();
2227
2228 dispy += dh;
2229 disph -= dh;
2230 slice++;
2231 ARSERT (!(slice - tile->slices >= tile->slicecount && disph > 0));
2232 slicey = 0;
2233 }
2234 }
2235 glDisable (GL_TEXTURE_RECTANGLE_ARB);
2236 CAMLreturn (Val_unit);
2237 }
2238
2239 CAMLprim value ml_postprocess (value ptr_v, value hlinks_v,
2240 value xoff_v, value yoff_v,
2241 value li_v)
2242 {
2243 CAMLparam5 (ptr_v, hlinks_v, xoff_v, yoff_v, li_v);
2244 int xoff = Int_val (xoff_v);
2245 int yoff = Int_val (yoff_v);
2246 int noff = Int_val (Field (li_v, 0));
2247 char *targ = String_val (Field (li_v, 1));
2248 int tlen = caml_string_length (Field (li_v, 1));
2249 int hfsize = Int_val (Field (li_v, 2));
2250 char *s = String_val (ptr_v);
2251 int hlmask = Int_val (hlinks_v);
2252 struct page *page = parse_pointer ("ml_postprocess", s);
2253
2254 if (!page->u.ptr) {
2255 /* deal with loadpage failed pages */
2256 goto done;
2257 }
2258
2259 if (hlmask & 1) highlightlinks (page, xoff, yoff);
2260 if (trylock ("ml_postprocess")) {
2261 noff = 0;
2262 goto done;
2263 }
2264 if (hlmask & 2) {
2265 highlightslinks (page, xoff, yoff, noff, targ, tlen, hfsize);
2266 noff = page->slinkcount;
2267 }
2268 showsel (page, xoff, yoff);
2269 unlock ("ml_postprocess");
2270
2271 done:
2272 CAMLreturn (Val_int (noff));
2273 }
2274
2275 static fz_link *getlink (struct page *page, int x, int y)
2276 {
2277 fz_point p;
2278 fz_matrix ctm;
2279 fz_link *link, *links;
2280
2281 switch (page->type) {
2282 case DPDF:
2283 ctm = trimctm (page->u.pdfpage, page->pdimno);
2284 links = page->u.pdfpage->links;
2285 break;
2286
2287 case DXPS:
2288 ctm = fz_identity;
2289 links = page->u.xpspage->links;
2290 break;
2291
2292 default:
2293 return NULL;
2294 }
2295 p.x = x;
2296 p.y = y;
2297
2298 ctm = fz_concat (ctm, state.pagedims[page->pdimno].ctm);
2299 ctm = fz_invert_matrix (ctm);
2300 p = fz_transform_point (ctm, p);
2301
2302 for (link = links; link; link = link->next) {
2303 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
2304 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
2305 return link;
2306 }
2307 }
2308 }
2309 return NULL;
2310 }
2311
2312 static void ensuretext (struct page *page)
2313 {
2314 if (state.gen != page->tgen) {
2315 droptext (page);
2316 page->tgen = state.gen;
2317 }
2318 if (!page->text) {
2319 fz_device *tdev;
2320
2321 page->text = fz_new_text_page (state.ctx, fz_infinite_rect);
2322 page->sheet = fz_new_text_sheet (state.ctx);
2323 tdev = fz_new_text_device (state.ctx, page->sheet, page->text);
2324 fz_run_display_list (page->dlist,
2325 tdev,
2326 pagectm (page),
2327 fz_infinite_bbox, NULL);
2328 qsort (page->text->blocks, page->text->len,
2329 sizeof (*page->text->blocks), compareblocks);
2330 fz_free_device (tdev);
2331 }
2332 }
2333
2334 CAMLprim value ml_find_page_with_links (value start_page_v, value dir_v)
2335 {
2336 CAMLparam2 (start_page_v, dir_v);
2337 CAMLlocal1 (ret_v);
2338 int i, dir = Int_val (dir_v);
2339 int start_page = Int_val (start_page_v);
2340 int end_page = dir > 0 ? state.pagecount : -1;
2341
2342 ret_v = Val_int (0);
2343 if (!(state.type == DPDF || state.type == DXPS)) {
2344 goto done;
2345 }
2346
2347 lock ("ml_findpage_with_links");
2348 for (i = start_page + dir; i != end_page; i += dir) {
2349 int found;
2350
2351 switch (state.type) {
2352 case DPDF:
2353 {
2354 pdf_page *page = NULL;
2355
2356 fz_try (state.ctx) {
2357 page = pdf_load_page (state.u.pdf, i);
2358 found = !!page->links;
2359 }
2360 fz_catch (state.ctx) {
2361 found = 0;
2362 }
2363 if (page) {
2364 freepdfpage (page);
2365 }
2366 }
2367 break;
2368 case DXPS:
2369 {
2370 xps_page *page = xps_load_page (state.u.xps, i);
2371 found = !!page->links;
2372 freexpspage (page);
2373 }
2374 break;
2375
2376 default:
2377 ARSERT ("invalid document type");
2378 }
2379
2380 if (found) {
2381 ret_v = caml_alloc_small (1, 1);
2382 Field (ret_v, 0) = Val_int (i);
2383 goto unlock;
2384 }
2385 }
2386 unlock:
2387 unlock ("ml_findpage_with_links");
2388
2389 done:
2390 CAMLreturn (ret_v);
2391 }
2392
2393 enum { dir_first, dir_last};
2394 enum { dir_first_visible, dir_left, dir_right, dir_down, dir_up };
2395
2396 CAMLprim value ml_findlink (value ptr_v, value dir_v)
2397 {
2398 CAMLparam2 (ptr_v, dir_v);
2399 CAMLlocal2 (ret_v, pos_v);
2400 struct page *page;
2401 int dirtag, i, slinkindex;
2402 struct slink *found = NULL ,*slink;
2403 char *s = String_val (ptr_v);
2404
2405 page = parse_pointer ("ml_findlink", s);
2406 ret_v = Val_int (0);
2407 if (trylock ("ml_findlink")) {
2408 goto done;
2409 }
2410
2411 ensureslinks (page);
2412
2413 if (Is_block (dir_v)) {
2414 dirtag = Tag_val (dir_v);
2415 switch (dirtag) {
2416 case dir_first_visible:
2417 {
2418 int x0, y0, dir, first_index, last_index;
2419
2420 pos_v = Field (dir_v, 0);
2421 x0 = Int_val (Field (pos_v, 0));
2422 y0 = Int_val (Field (pos_v, 1));
2423 dir = Int_val (Field (pos_v, 2));
2424
2425 if (dir >= 0) {
2426 dir = 1;
2427 first_index = 0;
2428 last_index = page->slinkcount;
2429 }
2430 else {
2431 first_index = page->slinkcount - 1;
2432 last_index = -1;
2433 }
2434
2435 for (i = first_index; i != last_index; i += dir) {
2436 slink = &page->slinks[i];
2437 if (slink->bbox.y0 >= y0 && slink->bbox.x0 >= x0) {
2438 found = slink;
2439 break;
2440 }
2441 }
2442 }
2443 break;
2444
2445 case dir_left:
2446 slinkindex = Int_val (Field (dir_v, 0));
2447 found = &page->slinks[slinkindex];
2448 for (i = slinkindex - 1; i >= 0; --i) {
2449 slink = &page->slinks[i];
2450 if (slink->bbox.x0 < found->bbox.x0) {
2451 found = slink;
2452 break;
2453 }
2454 }
2455 break;
2456
2457 case dir_right:
2458 slinkindex = Int_val (Field (dir_v, 0));
2459 found = &page->slinks[slinkindex];
2460 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2461 slink = &page->slinks[i];
2462 if (slink->bbox.x0 > found->bbox.x0) {
2463 found = slink;
2464 break;
2465 }
2466 }
2467 break;
2468
2469 case dir_down:
2470 slinkindex = Int_val (Field (dir_v, 0));
2471 found = &page->slinks[slinkindex];
2472 for (i = slinkindex + 1; i < page->slinkcount; ++i) {
2473 slink = &page->slinks[i];
2474 if (slink->bbox.y0 >= found->bbox.y0) {
2475 found = slink;
2476 break;
2477 }
2478 }
2479 break;
2480
2481 case dir_up:
2482 slinkindex = Int_val (Field (dir_v, 0));
2483 found = &page->slinks[slinkindex];
2484 for (i = slinkindex - 1; i >= 0; --i) {
2485 slink = &page->slinks[i];
2486 if (slink->bbox.y0 <= found->bbox.y0) {
2487 found = slink;
2488 break;
2489 }
2490 }
2491 break;
2492 }
2493 }
2494 else {
2495 dirtag = Int_val (dir_v);
2496 switch (dirtag) {
2497 case dir_first:
2498 found = page->slinks;
2499 break;
2500
2501 case dir_last:
2502 if (page->slinks) {
2503 found = page->slinks + (page->slinkcount - 1);
2504 }
2505 break;
2506 }
2507 }
2508 if (found) {
2509 ret_v = caml_alloc_small (2, 1);
2510 Field (ret_v, 0) = Val_int (found - page->slinks);
2511 }
2512
2513 unlock ("ml_findlink");
2514 done:
2515 CAMLreturn (ret_v);
2516 }
2517
2518 enum { uuri, ugoto, utext, uunexpected, ulaunch, unamed, uremote };
2519
2520 #define LINKTOVAL \
2521 { \
2522 int pageno; \
2523 \
2524 switch (link->dest.kind) { \
2525 case FZ_LINK_GOTO: \
2526 { \
2527 fz_point p; \
2528 \
2529 pageno = link->dest.ld.gotor.page; \
2530 p.x = 0; \
2531 p.y = 0; \
2532 \
2533 if (link->dest.ld.gotor.flags & fz_link_flag_t_valid) { \
2534 p.y = link->dest.ld.gotor.lt.y; \
2535 p = fz_transform_point (pdim->lctm, p); \
2536 } \
2537 tup_v = caml_alloc_tuple (2); \
2538 ret_v = caml_alloc_small (1, ugoto); \
2539 Field (tup_v, 0) = Val_int (pageno); \
2540 Field (tup_v, 1) = Val_int (p.y); \
2541 Field (ret_v, 0) = tup_v; \
2542 } \
2543 break; \
2544 \
2545 case FZ_LINK_URI: \
2546 str_v = caml_copy_string (link->dest.ld.uri.uri); \
2547 ret_v = caml_alloc_small (1, uuri); \
2548 Field (ret_v, 0) = str_v; \
2549 break; \
2550 \
2551 case FZ_LINK_LAUNCH: \
2552 str_v = caml_copy_string (link->dest.ld.launch.file_spec); \
2553 ret_v = caml_alloc_small (1, ulaunch); \
2554 Field (ret_v, 0) = str_v; \
2555 break; \
2556 \
2557 case FZ_LINK_NAMED: \
2558 str_v = caml_copy_string (link->dest.ld.named.named); \
2559 ret_v = caml_alloc_small (1, unamed); \
2560 Field (ret_v, 0) = str_v; \
2561 break; \
2562 \
2563 case FZ_LINK_GOTOR: \
2564 str_v = caml_copy_string (link->dest.ld.gotor.file_spec); \
2565 pageno = link->dest.ld.gotor.page; \
2566 tup_v = caml_alloc_tuple (2); \
2567 ret_v = caml_alloc_small (1, uremote); \
2568 Field (tup_v, 0) = str_v; \
2569 Field (tup_v, 1) = Val_int (pageno); \
2570 Field (ret_v, 0) = tup_v; \
2571 break; \
2572 \
2573 default: \
2574 { \
2575 char buf[80]; \
2576 \
2577 snprintf (buf, sizeof (buf), \
2578 "unhandled link kind %d", link->dest.kind); \
2579 str_v = caml_copy_string (buf); \
2580 ret_v = caml_alloc_small (1, uunexpected); \
2581 Field (ret_v, 0) = str_v; \
2582 } \
2583 break; \
2584 } \
2585 }
2586
2587 CAMLprim value ml_getlink (value ptr_v, value n_v)
2588 {
2589 CAMLparam2 (ptr_v, n_v);
2590 CAMLlocal3 (ret_v, tup_v, str_v);
2591 fz_link *link;
2592 struct page *page;
2593 struct pagedim *pdim;
2594 char *s = String_val (ptr_v);
2595
2596 ret_v = Val_int (0);
2597 if (trylock ("ml_getlink")) {
2598 goto done;
2599 }
2600
2601 page = parse_pointer ("ml_getlink", s);
2602 ensureslinks (page);
2603 pdim = &state.pagedims[page->pdimno];
2604 link = page->slinks[Int_val (n_v)].link;
2605 LINKTOVAL;
2606
2607 unlock ("ml_getlink");
2608 done:
2609 CAMLreturn (ret_v);
2610 }
2611
2612 CAMLprim value ml_getlinkcount (value ptr_v)
2613 {
2614 CAMLparam1 (ptr_v);
2615 struct page *page;
2616 char *s = String_val (ptr_v);
2617
2618 page = parse_pointer ("ml_getlinkcount", s);
2619 CAMLreturn (Val_int (page->slinkcount));
2620 }
2621
2622 CAMLprim value ml_getlinkrect (value ptr_v, value n_v)
2623 {
2624 CAMLparam2 (ptr_v, n_v);
2625 CAMLlocal1 (ret_v);
2626 struct page *page;
2627 struct slink *slink;
2628 char *s = String_val (ptr_v);
2629
2630 page = parse_pointer ("ml_getlinkrect", s);
2631 ret_v = caml_alloc_tuple (4);
2632 if (trylock ("ml_getlinkrect")) {
2633 Field (ret_v, 0) = Val_int (0);
2634 Field (ret_v, 1) = Val_int (0);
2635 Field (ret_v, 2) = Val_int (0);
2636 Field (ret_v, 3) = Val_int (0);
2637 goto done;
2638 }
2639 ensureslinks (page);
2640
2641 slink = &page->slinks[Int_val (n_v)];
2642 Field (ret_v, 0) = Val_int (slink->bbox.x0);
2643 Field (ret_v, 1) = Val_int (slink->bbox.y0);
2644 Field (ret_v, 2) = Val_int (slink->bbox.x1);
2645 Field (ret_v, 3) = Val_int (slink->bbox.y1);
2646 unlock ("ml_getlinkrect");
2647
2648 done:
2649 CAMLreturn (ret_v);
2650 }
2651
2652 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
2653 {
2654 CAMLparam3 (ptr_v, x_v, y_v);
2655 CAMLlocal3 (ret_v, tup_v, str_v);
2656 fz_link *link;
2657 struct page *page;
2658 char *s = String_val (ptr_v);
2659 int x = Int_val (x_v), y = Int_val (y_v);
2660 struct pagedim *pdim;
2661
2662 ret_v = Val_int (0);
2663 if (trylock ("ml_whatsunder")) {
2664 goto done;
2665 }
2666
2667 page = parse_pointer ("ml_whatsunder", s);
2668 pdim = &state.pagedims[page->pdimno];
2669 x += pdim->bounds.x0;
2670 y += pdim->bounds.y0;
2671 link = getlink (page, x, y);
2672 if (link) {
2673 LINKTOVAL;
2674 }
2675 else {
2676 fz_rect *b;
2677 fz_text_block *block;
2678
2679 ensuretext (page);
2680 for (block = page->text->blocks;
2681 block < page->text->blocks + page->text->len;
2682 ++block) {
2683 fz_text_line *line;
2684
2685 b = &block->bbox;
2686 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2687 continue;
2688
2689 for (line = block->lines;
2690 line < block->lines + block->len;
2691 ++line) {
2692 fz_text_span *span;
2693
2694 b = &line->bbox;
2695 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2696 continue;
2697
2698 for (span = line->spans;
2699 span < line->spans + line->len;
2700 ++span) {
2701 fz_text_char *ch;
2702
2703 b = &span->bbox;
2704 if (!(x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1))
2705 continue;
2706
2707 for (ch = span->text; ch < span->text + span->len; ++ch) {
2708 b = &ch->bbox;
2709
2710 if (x >= b->x0 && x <= b->x1
2711 && y >= b->y0 && y <= b->y1) {
2712 const char *n2 =
2713 span->style->font && span->style->font->name
2714 ? span->style->font->name
2715 : "Span has no font name"
2716 ;
2717 FT_FaceRec *face = span->style->font->ft_face;
2718 if (face && face->family_name) {
2719 char *s;
2720 char *n1 = face->family_name;
2721 size_t l1 = strlen (n1);
2722 size_t l2 = strlen (n2);
2723
2724 if (l1 != l2 || memcmp (n1, n2, l1)) {
2725 s = malloc (l1 + l2 + 2);
2726 if (s) {
2727 memcpy (s, n2, l2);
2728 s[l2] = '=';
2729 memcpy (s + l2 + 1, n1, l1 + 1);
2730 str_v = caml_copy_string (s);
2731 free (s);
2732 }
2733 }
2734 }
2735 if (str_v == 0) {
2736 str_v = caml_copy_string (n2);
2737 }
2738 ret_v = caml_alloc_small (1, utext);
2739 Field (ret_v, 0) = str_v;
2740 goto unlock;
2741 }
2742 }
2743 }
2744 }
2745 }
2746 }
2747 unlock:
2748 unlock ("ml_whatsunder");
2749
2750 done:
2751 CAMLreturn (ret_v);
2752 }
2753
2754 CAMLprim value ml_seltext (value ptr_v, value rect_v)
2755 {
2756 CAMLparam2 (ptr_v, rect_v);
2757 fz_rect *b;
2758 struct page *page;
2759 struct pagedim *pdim;
2760 int i, x0, x1, y0, y1;
2761 char *s = String_val (ptr_v);
2762 int fi = 0, li = 0;
2763 fz_text_block *block;
2764 fz_text_span *span, *fspan, *lspan;
2765 fz_text_line *line, *fline = NULL, *lline = NULL;
2766
2767 if (trylock ("ml_seltext")) {
2768 goto done;
2769 }
2770
2771 page = parse_pointer ("ml_seltext", s);
2772 ensuretext (page);
2773
2774 pdim = &state.pagedims[page->pdimno];
2775 x0 = Int_val (Field (rect_v, 0)) + pdim->bounds.x0;;
2776 y0 = Int_val (Field (rect_v, 1)) + pdim->bounds.y0;
2777 x1 = Int_val (Field (rect_v, 2)) + pdim->bounds.x0;
2778 y1 = Int_val (Field (rect_v, 3)) + pdim->bounds.y0;
2779
2780 if (0) {
2781 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2782 glColor3ub (128, 128, 128);
2783 glRecti (x0, y0, x1, y1);
2784 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2785 }
2786
2787 fspan = lspan = NULL;
2788
2789 for (block = page->text->blocks;
2790 block < page->text->blocks + page->text->len;
2791 ++block) {
2792 for (line = block->lines;
2793 line < block->lines + block->len;
2794 ++line) {
2795 for (span = line->spans;
2796 span < line->spans + line->len;
2797 ++span) {
2798 for (i = 0; i < span->len; ++i) {
2799 b = &span->text[i].bbox;
2800 int selected = 0;
2801
2802 if (x0 >= b->x0 && x0 <= b->x1
2803 && y0 >= b->y0 && y0 <= b->y1) {
2804 fspan = span;
2805 fline = line;
2806 fi = i;
2807 selected = 1;
2808 }
2809 if (x1 >= b->x0 && x1 <= b->x1
2810 && y1 >= b->y0 && y1 <= b->y1) {
2811 lspan = span;
2812 lline = line;
2813 li = i;
2814 selected = 1;
2815 }
2816 if (0 && selected) {
2817 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
2818 glColor3ub (128, 128, 128);
2819 glRecti (b->x0, b->y0, b->x1, b->y1);
2820 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
2821 }
2822 }
2823 }
2824 }
2825 }
2826 if (y1 < y0 || x1 < x0) {
2827 int swap = 0;
2828
2829 if (fspan == lspan) {
2830 swap = 1;
2831 }
2832 else {
2833 if (y1 < y0) {
2834 if (fline != lline) {
2835 swap = 1;
2836 }
2837 }
2838 }
2839
2840 if (swap) {
2841 i = fi;
2842 span = fspan;
2843
2844 fi = li;
2845 fspan = lspan;
2846
2847 li = i;
2848 lspan = span;
2849 }
2850 }
2851
2852 page->fmark.i = fi;
2853 page->fmark.span = fspan;
2854
2855 page->lmark.i = li;
2856 page->lmark.span = lspan;
2857
2858 unlock ("ml_seltext");
2859
2860 done:
2861 CAMLreturn (Val_unit);
2862 }
2863
2864 static int UNUSED_ATTR pipespan (FILE *f, fz_text_span *span, int a, int b)
2865 {
2866 char buf[4];
2867 int i, len, ret;
2868
2869 for (i = a; i <= b; ++i) {
2870 len = fz_runetochar (buf, span->text[i].c);
2871 ret = fwrite (buf, len, 1, f);
2872
2873 if (ret != 1) {
2874 fprintf (stderr, "failed to write %d bytes ret=%d: %s\n",
2875 len, ret, strerror (errno));
2876 return -1;
2877 }
2878 }
2879 return 0;
2880 }
2881
2882 #ifdef __CYGWIN__
2883 value ml_popen (value UNUSED_ATTR u1, value UNUSED_ATTR u2)
2884 {
2885 caml_failwith ("ml_popen not implemented under Cygwin");
2886 }
2887 #else
2888 CAMLprim value ml_popen (value command_v, value fds_v)
2889 {
2890 CAMLparam2 (command_v, fds_v);
2891 CAMLlocal2 (l_v, tup_v);
2892 int ret;
2893 char *msg = NULL;
2894 value earg_v = Nothing;
2895 posix_spawnattr_t attr;
2896 posix_spawn_file_actions_t fa;
2897 char *argv[] = { "/bin/sh", "-c", String_val (command_v), NULL };
2898
2899 if ((ret = posix_spawn_file_actions_init (&fa)) != 0) {
2900 unix_error (ret, "posix_spawn_file_actions_init", Nothing);
2901 }
2902
2903 if ((ret = posix_spawnattr_init (&attr)) != 0) {
2904 msg = "posix_spawnattr_init";
2905 goto fail1;
2906 }
2907
2908 #ifdef POSIX_SPAWN_USEVFORK
2909 if ((ret = posix_spawnattr_setflags (&attr, POSIX_SPAWN_USEVFORK)) != 0) {
2910 msg = "posix_spawnattr_setflags POSIX_SPAWN_USEVFORK";
2911 goto fail;
2912 }
2913 #endif
2914
2915 for (l_v = fds_v; l_v != Val_int (0); l_v = Field (l_v, 1)) {
2916 int fd1, fd2;
2917
2918 tup_v = Field (l_v, 0);
2919 fd1 = Int_val (Field (tup_v, 0));
2920 fd2 = Int_val (Field (tup_v, 1));
2921 if (fd2 < 0) {
2922 if ((ret = posix_spawn_file_actions_addclose (&fa, fd1)) != 0) {
2923 msg = "posix_spawn_file_actions_addclose";
2924 earg_v = tup_v;
2925 goto fail;
2926 }
2927 }
2928 else {
2929 if ((ret = posix_spawn_file_actions_adddup2 (&fa, fd1, fd2)) != 0) {
2930 msg = "posix_spawn_file_actions_adddup2";
2931 earg_v = tup_v;
2932 goto fail;
2933 }
2934 }
2935 }
2936
2937 if ((ret = posix_spawn (NULL, "/bin/sh", &fa, &attr, argv, environ))) {
2938 msg = "posix_spawn";
2939 goto fail;
2940 }
2941
2942 fail:
2943 if ((ret = posix_spawnattr_destroy (&attr)) != 0) {
2944 fprintf (stderr, "posix_spawnattr_destroy: %s\n", strerror (ret));
2945 }
2946
2947 fail1:
2948 if ((ret = posix_spawn_file_actions_destroy (&fa)) != 0) {
2949 fprintf (stderr, "posix_spawn_file_actions_destroy: %s\n",
2950 strerror (ret));
2951 }
2952
2953 if (msg)
2954 unix_error (ret, msg, earg_v);
2955
2956 CAMLreturn (Val_unit);
2957 }
2958 #endif
2959
2960 CAMLprim value ml_copysel (value fd_v, value ptr_v)
2961 {
2962 CAMLparam1 (ptr_v);
2963 FILE *f;
2964 int seen = 0;
2965 struct page *page;
2966 fz_text_line *line;
2967 fz_text_span *span;
2968 fz_text_block *block;
2969 int fd = Int_val (fd_v);
2970 char *s = String_val (ptr_v);
2971
2972 if (trylock ("ml_copysel")) {
2973 goto done;
2974 }
2975
2976 page = parse_pointer ("ml_sopysel", s);
2977
2978 if (!page->fmark.span || !page->lmark.span) {
2979 fprintf (stderr, "nothing to copy\n");
2980 goto unlock;
2981 }
2982
2983 f = fdopen (fd, "w");
2984 if (!f) {
2985 fprintf (stderr, "failed to fopen sel pipe: %s\n",
2986 strerror (errno));
2987 f = stdout;
2988 }
2989
2990 for (block = page->text->blocks;
2991 block < page->text->blocks + page->text->len;
2992 ++block) {
2993 for (line = block->lines;
2994 line < block->lines + block->len;
2995 ++line) {
2996 for (span = line->spans;
2997 span < line->spans + line->len;
2998 ++span) {
2999 int a, b;
3000
3001 seen |= span == page->fmark.span || span == page->lmark.span;
3002 a = span == page->fmark.span ? page->fmark.i : 0;
3003 b = span == page->lmark.span ? page->lmark.i : span->len - 1;
3004
3005 if (seen) {
3006 if (pipespan (f, span, a, b)) {
3007 goto close;
3008 }
3009 if (span == line->spans + line->len - 1) {
3010 if (putc ('\n', f) == EOF) {
3011 fprintf (stderr,
3012 "failed break line on sel pipe: %s\n",
3013 strerror (errno));
3014 goto close;
3015 }
3016 }
3017 if (span == page->lmark.span) {
3018 goto endloop;
3019 }
3020 }
3021 }
3022 }
3023 }
3024 endloop:
3025 page->lmark.span = NULL;
3026 page->fmark.span = NULL;
3027
3028 close:
3029 if (f != stdout) {
3030 int ret = fclose (f);
3031 fd = -1;
3032 if (ret == -1) {
3033 if (errno != ECHILD) {
3034 fprintf (stderr, "failed to close sel pipe: %s\n",
3035 strerror (errno));
3036 }
3037 }
3038 }
3039 unlock:
3040 unlock ("ml_copysel");
3041
3042 done:
3043 if (fd >= 0) {
3044 if (close (fd)) {
3045 fprintf (stderr, "failed to close sel pipe: %s\n",
3046 strerror (errno));
3047 }
3048 }
3049 CAMLreturn (Val_unit);
3050 }
3051
3052 CAMLprim value ml_getpdimrect (value pagedimno_v)
3053 {
3054 CAMLparam1 (pagedimno_v);
3055 CAMLlocal1 (ret_v);
3056 int pagedimno = Int_val (pagedimno_v);
3057 fz_rect box;
3058
3059 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
3060 if (trylock ("ml_getpdimrect")) {
3061 box = fz_empty_rect;
3062 }
3063 else {
3064 box = state.pagedims[pagedimno].mediabox;
3065 unlock ("ml_getpdimrect");
3066 }
3067
3068 Store_double_field (ret_v, 0, box.x0);
3069 Store_double_field (ret_v, 1, box.x1);
3070 Store_double_field (ret_v, 2, box.y0);
3071 Store_double_field (ret_v, 3, box.y1);
3072
3073 CAMLreturn (ret_v);
3074 }
3075
3076 static double getmaxw (void)
3077 {
3078 int i;
3079 struct pagedim *p;
3080 double maxw = 0.0;
3081
3082 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
3083 double x0, x1, w;
3084
3085 x0 = MIN (p->mediabox.x0, p->mediabox.x1);
3086 x1 = MAX (p->mediabox.x0, p->mediabox.x1);
3087
3088 w = x1 - x0;
3089 maxw = MAX (w, maxw);
3090 }
3091 return maxw;
3092 }
3093
3094 CAMLprim value ml_getmaxw (value unit_v)
3095 {
3096 CAMLparam1 (unit_v);
3097 CAMLlocal1 (ret_v);
3098 double maxw = 0.0;
3099
3100 if (trylock ("ml_getmaxw")) {
3101 goto done;
3102 }
3103 maxw = getmaxw ();
3104 unlock ("ml_getmaxw");
3105 done:
3106 ret_v = caml_copy_double (maxw);
3107 CAMLreturn (ret_v);
3108 }
3109
3110 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v,
3111 value dw_v, value cols_v)
3112 {
3113 CAMLparam3 (winw_v, winh_v, dw_v);
3114 CAMLlocal1 (ret_v);
3115 int i;
3116 double zoom = 1.0;
3117 double maxw = 0.0, maxh = 0.0;
3118 struct pagedim *p;
3119 double winw = Int_val (winw_v);
3120 double winh = Int_val (winh_v);
3121 double dw = Int_val (dw_v);
3122 double cols = Int_val (cols_v);
3123 double pw = 1.0, ph = 1.0, aspect;
3124
3125 if (trylock ("ml_zoom_for_height")) {
3126 goto done;
3127 }
3128
3129 if (state.proportional) {
3130 maxw = getmaxw () / cols;
3131 }
3132
3133 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
3134 double x0, x1, y0, y1, w, h, scaledh, scale;
3135 fz_rect rect;
3136
3137 rect = fz_transform_rect (fz_rotate (p->rotate + state.rotate),
3138 p->mediabox);
3139 x0 = MIN (rect.x0, rect.x1);
3140 x1 = MAX (rect.x0, rect.x1);
3141 y0 = MIN (rect.y0, rect.y1);
3142 y1 = MAX (rect.y0, rect.y1);
3143
3144 w = (x1 - x0) / cols;
3145 h = y1 - y0;
3146
3147 if (state.proportional) {
3148 scale = w / maxw;
3149 scaledh = h * scale;
3150 }
3151 else {
3152 scale = 1.0;
3153 scaledh = h;
3154 }
3155
3156 if (scaledh > maxh) {
3157 maxh = scaledh;
3158 ph = scaledh;
3159 pw = w * scale;
3160 }
3161 }
3162
3163 aspect = pw / ph;
3164 zoom = (winh * aspect + dw) / winw;
3165
3166 unlock ("ml_zoom_for_height");
3167 done:
3168 ret_v = caml_copy_double (zoom);
3169 CAMLreturn (ret_v);
3170 }
3171
3172 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
3173 {
3174 CAMLparam4 (pt_v, x_v, y_v, string_v);
3175 CAMLlocal1 (ret_v);
3176 int pt = Int_val(pt_v);
3177 int x = Int_val (x_v);
3178 int y = Int_val (y_v);
3179 double w;
3180
3181 w = draw_string (state.face, pt, x, y, String_val (string_v));
3182 ret_v = caml_copy_double (w);
3183 CAMLreturn (ret_v);
3184 }
3185
3186 CAMLprim value ml_measure_string (value pt_v, value string_v)
3187 {
3188 CAMLparam2 (pt_v, string_v);
3189 CAMLlocal1 (ret_v);
3190 int pt = Int_val (pt_v);
3191 double w;
3192
3193 w = measure_string (state.face, pt, String_val (string_v));
3194 ret_v = caml_copy_double (w);
3195 CAMLreturn (ret_v);
3196 }
3197
3198 CAMLprim value ml_getpagebox (value opaque_v)
3199 {
3200 CAMLparam1 (opaque_v);
3201 CAMLlocal1 (ret_v);
3202 fz_bbox bbox;
3203 fz_device *dev;
3204 char *s = String_val (opaque_v);
3205 struct page *page = parse_pointer ("ml_getpagebox", s);
3206
3207 ret_v = caml_alloc_tuple (4);
3208 dev = fz_new_bbox_device (state.ctx, &bbox);
3209 dev->hints |= FZ_IGNORE_SHADE;
3210
3211 switch (page->type) {
3212 case DPDF:
3213 pdf_run_page (state.u.pdf, page->u.pdfpage, dev, pagectm (page), NULL);
3214 break;
3215
3216 case DXPS:
3217 xps_run_page (state.u.xps, page->u.xpspage, dev, pagectm (page), NULL);
3218 break;
3219
3220 default:
3221 bbox = fz_infinite_bbox;
3222 break;
3223 }
3224
3225 fz_free_device (dev);
3226 Field (ret_v, 0) = Val_int (bbox.x0);
3227 Field (ret_v, 1) = Val_int (bbox.y0);
3228 Field (ret_v, 2) = Val_int (bbox.x1);
3229 Field (ret_v, 3) = Val_int (bbox.y1);
3230
3231 CAMLreturn (ret_v);
3232 }
3233
3234 CAMLprim value ml_setaalevel (value level_v)
3235 {
3236 CAMLparam1 (level_v);
3237
3238 state.aalevel = Int_val (level_v);
3239 CAMLreturn (Val_unit);
3240 }
3241
3242 #undef pixel
3243 #include <X11/Xlib.h>
3244 #include <GL/glx.h>
3245
3246 static struct {
3247 Display *dpy;
3248 GLXContext ctx;
3249 GLXDrawable drawable;
3250 } glx;
3251
3252 #include "keysym2ucs.c"
3253
3254 CAMLprim value ml_keysymtoutf8 (value keysym_v)
3255 {
3256 CAMLparam1 (keysym_v);
3257 CAMLlocal1 (str_v);
3258 KeySym keysym = Int_val (keysym_v);
3259 Rune rune;
3260 int len;
3261 char buf[5];
3262
3263 rune = keysym2ucs (keysym);
3264 len = fz_runetochar (buf, rune);
3265 buf[len] = 0;
3266 str_v = caml_copy_string (buf);
3267 CAMLreturn (str_v);
3268 }
3269
3270 CAMLprim value ml_glx (value win_v)
3271 {
3272 CAMLparam1 (win_v);
3273 XVisualInfo *visual;
3274 int screen, wid = Int_val (win_v);
3275 int attributes[] = { GLX_RGBA, GLX_DOUBLEBUFFER, None };
3276
3277 glx.dpy = XOpenDisplay (NULL);
3278 if (!glx.dpy) {
3279 caml_failwith ("XOpenDisplay");
3280 }
3281
3282 screen = DefaultScreen (glx.dpy);
3283 visual = glXChooseVisual (glx.dpy, screen, attributes);
3284 if (!visual) {
3285 XCloseDisplay (glx.dpy);
3286 glx.dpy = NULL;
3287 caml_failwith ("glXChooseVisual");
3288 }
3289
3290 glx.ctx = glXCreateContext (glx.dpy, visual, NULL, True);
3291 XFree (visual);
3292 if (!glx.ctx) {
3293 XCloseDisplay (glx.dpy);
3294 glx.dpy = NULL;
3295 caml_failwith ("glXCreateContext");
3296 }
3297
3298 if (!glXMakeCurrent (glx.dpy, wid, glx.ctx)) {
3299 glXDestroyContext (glx.dpy, glx.ctx);
3300 XCloseDisplay (glx.dpy);
3301 glx.dpy = NULL;
3302 glx.ctx = NULL;
3303 caml_failwith ("glXMakeCurrent");
3304 }
3305 glx.drawable = wid;
3306 CAMLreturn (Val_unit);
3307 }
3308
3309 CAMLprim value ml_swapb (value unit_v)
3310 {
3311 CAMLparam1 (unit_v);
3312 glXSwapBuffers (glx.dpy, glx.drawable);
3313 CAMLreturn (Val_unit);
3314 }
3315
3316 CAMLprim value ml_glxsync (value unit_v)
3317 {
3318 CAMLparam1 (unit_v);
3319 if (glx.dpy && glx.ctx) {
3320 glXWaitX ();
3321 glXWaitGL ();
3322 }
3323 CAMLreturn (Val_unit);
3324 }
3325
3326 enum { piunknown, pilinux, piosx, pisun, pifreebsd,
3327 pidragonflybsd, piopenbsd, pinetbsd, picygwin };
3328
3329 CAMLprim value ml_platform (value unit_v)
3330 {
3331 CAMLparam1 (unit_v);
3332 int platid = piunknown;
3333
3334 #if defined __linux__
3335 platid = pilinux;
3336 #elif defined __CYGWIN__
3337 platid = picygwin;
3338 #elif defined __DragonFly__
3339 platid = pidragonflybsd;
3340 #elif defined __FreeBSD__
3341 platid = pifreebsd;
3342 #elif defined __OpenBSD__
3343 platid = piopenbsd;
3344 #elif defined __NetBSD__
3345 platid = pinetbsd;
3346 #elif defined __sun__
3347 platid = pisun;
3348 #elif defined __APPLE__
3349 platid = piosx;
3350 #endif
3351 CAMLreturn (Val_int (platid));
3352 }
3353
3354 CAMLprim value ml_cloexec (value fd_v)
3355 {
3356 CAMLparam1 (fd_v);
3357 int fd = Int_val (fd_v);
3358
3359 if (fcntl (fd, F_SETFD, FD_CLOEXEC, 1)) {
3360 uerror ("fcntl", Nothing);
3361 }
3362 CAMLreturn (Val_unit);
3363 }
3364
3365 CAMLprim value ml_getpbo (value w_v, value h_v, value cs_v)
3366 {
3367 CAMLparam2 (w_v, h_v);
3368 CAMLlocal1 (ret_v);
3369 struct pbo *pbo;
3370 int w = Int_val (w_v);
3371 int h = Int_val (h_v);
3372 int cs = Int_val (cs_v);
3373
3374 if (state.pbo_usable) {
3375 pbo = calloc (sizeof (*pbo), 1);
3376 if (!pbo) {
3377 err (1, "calloc pbo");
3378 }
3379
3380 switch (cs) {
3381 case 0:
3382 case 1:
3383 pbo->size = w*h*4;
3384 break;
3385 case 2:
3386 pbo->size = w*h*2;
3387 break;
3388 default:
3389 errx (1, "ml_getpbo: invalid colorspace %d", cs);
3390 }
3391
3392 state.glGenBuffersARB (1, &pbo->id);
3393 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->id);
3394 state.glBufferDataARB (GL_PIXEL_UNPACK_BUFFER_ARB, pbo->size,
3395 NULL, GL_STREAM_DRAW);
3396 pbo->ptr = state.glMapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB,
3397 GL_READ_WRITE);
3398 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3399 if (!pbo->ptr) {
3400 fprintf (stderr, "glMapBufferARB failed: %#x", glGetError ());
3401 state.glDeleteBuffersARB (1, &pbo->id);
3402 free (pbo);
3403 ret_v = caml_copy_string ("0");
3404 }
3405 else {
3406 int res;
3407 char *s;
3408
3409 res = snprintf (NULL, 0, "%" FMT_ptr, pbo);
3410 if (res < 0) {
3411 err (1, "snprintf %" FMT_ptr " failed", pbo);
3412 }
3413 s = malloc (res+1);
3414 if (!s) {
3415 err (1, "malloc %d bytes failed", res+1);
3416 }
3417 res = sprintf (s, "%" FMT_ptr, pbo);
3418 if (res < 0) {
3419 err (1, "sprintf %" FMT_ptr " failed", pbo);
3420 }
3421 ret_v = caml_copy_string (s);
3422 free (s);
3423 }
3424 }
3425 else {
3426 ret_v = caml_copy_string ("0");
3427 }
3428 CAMLreturn (ret_v);
3429 }
3430
3431 CAMLprim value ml_freepbo (value s_v)
3432 {
3433 CAMLparam1 (s_v);
3434 char *s = String_val (s_v);
3435 struct tile *tile = parse_pointer ("ml_freepbo", s);
3436
3437 if (tile->pbo) {
3438 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
3439 state.glUnmapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB);
3440 state.glDeleteBuffersARB (1, &tile->pbo->id);
3441 tile->pbo->id = -1;
3442 tile->pbo->ptr = NULL;
3443 tile->pbo->size = -1;
3444 }
3445 CAMLreturn (Val_unit);
3446 }
3447
3448 CAMLprim value ml_unmappbo (value s_v)
3449 {
3450 CAMLparam1 (s_v);
3451 char *s = String_val (s_v);
3452 struct tile *tile = parse_pointer ("ml_unmappbo", s);
3453
3454 if (tile->pbo) {
3455 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, tile->pbo->id);
3456 state.glUnmapBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB);
3457 tile->pbo->ptr = NULL;
3458 state.glBindBufferARB (GL_PIXEL_UNPACK_BUFFER_ARB, 0);
3459 }
3460 CAMLreturn (Val_unit);
3461 }
3462
3463 static void setuppbo (void)
3464 {
3465 #define GGPA(n) *(void (**) ()) &state.n = glXGetProcAddress ((GLubyte *) #n)
3466 GGPA (glBindBufferARB);
3467 if (state.glBindBufferARB) {
3468 GGPA (glUnmapBufferARB);
3469 if (state.glUnmapBufferARB) {
3470 GGPA (glMapBufferARB);
3471 if (state.glMapBufferARB) {
3472 GGPA (glBufferDataARB);
3473 if (state.glBufferDataARB) {
3474 GGPA (glGenBuffersARB);
3475 if (state.glGenBuffersARB) {
3476 GGPA (glDeleteBuffersARB);
3477 if (state.glDeleteBuffersARB) {
3478 state.pbo_usable = 1;
3479 }
3480 }
3481 }
3482 }
3483 }
3484 }
3485 #undef GGPA
3486 }
3487
3488 CAMLprim value ml_pbo_usable (value unit_v)
3489 {
3490 CAMLparam1 (unit_v);
3491 CAMLreturn (Val_bool (state.pbo_usable));
3492 }
3493
3494 CAMLprim value ml_init (value pipe_v, value params_v)
3495 {
3496 CAMLparam2 (pipe_v, params_v);
3497 CAMLlocal2 (trim_v, fuzz_v);
3498 int ret;
3499 int texcount;
3500 char *fontpath;
3501 int colorspace;
3502 int mustoresize;
3503 int haspboext;
3504 struct sigaction sa;
3505
3506 state.cr = Int_val (Field (pipe_v, 0));
3507 state.cw = Int_val (Field (pipe_v, 1));
3508 state.rotate = Int_val (Field (params_v, 0));
3509 state.proportional = Bool_val (Field (params_v, 1));
3510 trim_v = Field (params_v, 2);
3511 texcount = Int_val (Field (params_v, 3));
3512 state.sliceheight = Int_val (Field (params_v, 4));
3513 mustoresize = Int_val (Field (params_v, 5));
3514 colorspace = Int_val (Field (params_v, 6));
3515 fontpath = String_val (Field (params_v, 7));
3516 state.trimcachepath = strdup (String_val (Field (params_v, 8)));
3517 if (!state.trimcachepath) {
3518 fprintf (stderr, "failed to strdup trimcachepath: %s\n",
3519 strerror (errno));
3520 }
3521 haspboext = Bool_val (Field (params_v, 9));
3522
3523 state.ctx = fz_new_context (NULL, NULL, mustoresize);
3524
3525 state.trimmargins = Bool_val (Field (trim_v, 0));
3526 fuzz_v = Field (trim_v, 1);
3527 state.trimfuzz.x0 = Int_val (Field (fuzz_v, 0));
3528 state.trimfuzz.y0 = Int_val (Field (fuzz_v, 1));
3529 state.trimfuzz.x1 = Int_val (Field (fuzz_v, 2));
3530 state.trimfuzz.y1 = Int_val (Field (fuzz_v, 3));
3531
3532 set_tex_params (colorspace);
3533
3534 if (*fontpath) {
3535 state.face = load_font (fontpath);
3536 }
3537 else {
3538 unsigned int len;
3539 void *base = pdf_lookup_substitute_font (0, 0, 0, 0, &len);
3540
3541 state.face = load_builtin_font (base, len);
3542 }
3543 if (!state.face) _exit (1);
3544
3545 realloctexts (texcount);
3546
3547 if (haspboext) {
3548 setuppbo ();
3549 }
3550
3551 #ifdef __CYGWIN__
3552 sa.sa_handler = SIG_IGN;
3553 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP;
3554 #else
3555 sa.sa_handler = SIG_DFL;
3556 sa.sa_flags = SA_RESTART | SA_NOCLDSTOP | SA_NOCLDWAIT;
3557 #endif
3558 if (sigemptyset (&sa.sa_mask)) {
3559 err (1, "sigemptyset");
3560 }
3561 if (sigaction (SIGCHLD, &sa, NULL)) {
3562 err (1, "sigaction");
3563 }
3564
3565 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
3566 if (ret) {
3567 errx (1, "pthread_create: %s", strerror (ret));
3568 }
3569
3570 CAMLreturn (Val_unit);
3571 }
PDF pager