More aggressive safety net
[llpp.git] / link.c
blob7d7fefb2347337fcf353649eff1c11ced35a8084
1 /* lots of code c&p-ed directly from mupdf */
2 #include <errno.h>
3 #include <stdio.h>
4 #include <string.h>
5 #include <stdlib.h>
7 #ifdef _WIN32
8 #define WIN32_LEAN_AND_MEAN
9 #include <windows.h>
10 #include <winsock2.h>
11 #define fionread_arg long
12 #define ssize_t int
13 #define FMT_ss "%d"
14 #ifdef _WIN64
15 #define FMT_s "%i64u"
16 #else
17 #define FMT_s "%u"
18 #endif
19 #pragma warning (disable:4244)
20 #pragma warning (disable:4996)
21 #pragma warning (disable:4995)
22 #endif
24 #ifdef _MSC_VER
25 #define NORETURN __declspec (noreturn)
26 #define UNUSED
27 #elif defined __GNUC__
28 #define NORETURN __attribute__ ((noreturn))
29 #define UNUSED __attribute__ ((unused))
30 #else
31 #define NORETURN
32 #define UNUSED
33 #endif
35 #ifdef _WIN32
36 static void __declspec (noreturn) sockerr (int exitcode, const char *fmt, ...)
38 va_list ap;
40 va_start (ap, fmt);
41 vfprintf (stderr, fmt, ap);
42 va_end (ap);
43 fprintf (stderr, ": wsaerror 0x%x\n", WSAGetLastError ());
44 _exit (exitcode);
46 #else
47 #define FMT_ss "%zd"
48 #define FMT_s "%zu"
49 #define fionread_arg int
50 #define ioctlsocket ioctl
51 #define sockerr err
52 #include <unistd.h>
53 #endif
55 #include <regex.h>
56 #include <ctype.h>
57 #include <stdarg.h>
58 #include <limits.h>
60 #ifndef _WIN32
61 #include <pthread.h>
62 #include <sys/time.h>
63 #include <sys/types.h>
64 #include <sys/socket.h>
65 #include <sys/ioctl.h>
66 #endif
68 static void NORETURN 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 errx (int exitcode, const char *fmt, ...)
84 va_list ap;
86 va_start (ap, fmt);
87 vfprintf (stderr, fmt, ap);
88 va_end (ap);
89 fputc ('\n', stderr);
90 fflush (stderr);
91 _exit (exitcode);
94 #ifdef __APPLE__
95 #include <OpenGL/gl.h>
96 #else
97 #include <GL/gl.h>
98 #endif
100 #ifndef GL_TEXTURE_RECTANGLE_ARB
101 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
102 #endif
104 #include <caml/fail.h>
105 #include <caml/alloc.h>
106 #include <caml/memory.h>
107 #include <caml/unixsupport.h>
109 #include <fitz.h>
110 #include <mupdf.h>
112 #if 0
113 #define lprintf printf
114 #else
115 #define lprintf(...)
116 #endif
118 #ifdef FT_FREETYPE_H
119 #include FT_FREETYPE_H
120 #endif
122 #define ARSERT(cond) for (;;) { \
123 if (!(cond)) { \
124 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
126 break; \
129 struct slice {
130 int texindex;
131 int w, h;
134 struct pagedim {
135 int pageno;
136 int rotate;
137 int left;
138 fz_rect box;
139 fz_bbox bbox;
140 fz_matrix ctm, ctm1;
143 struct page {
144 int pageno;
145 int slicecount;
146 fz_text_span *text;
147 fz_pixmap *pixmap;
148 pdf_page *drawpage;
149 struct pagedim pagedim;
150 struct mark {
151 int i;
152 fz_text_span *span;
153 } fmark, lmark;
154 struct slice slices[];
157 #if !defined _WIN32 && !defined __APPLE__
158 #define USE_XSEL
159 #endif
161 struct {
162 int sock;
163 int sliceheight;
164 struct page *pig;
165 struct pagedim *pagedims;
166 int pagecount;
167 int pagedimcount;
168 pdf_xref *xref;
169 fz_glyph_cache *cache;
170 int w, h;
172 int texindex;
173 int texcount;
174 GLuint *texids;
176 GLenum texform;
177 GLenum texty;
179 struct {
180 int w, h;
181 struct slice *slice;
182 } *texowners;
184 int rotate;
185 int proportional;
186 int needoutline;
188 #ifdef _WIN32
189 HANDLE thread;
190 #else
191 pthread_t thread;
192 #endif
193 FILE *xselpipe;
195 FT_Face face;
196 } state;
198 #ifdef _WIN32
199 static CRITICAL_SECTION critsec;
201 static void lock (void *unused)
203 (void) unused;
204 EnterCriticalSection (&critsec);
207 static void unlock (void *unused)
209 (void) unused;
210 LeaveCriticalSection (&critsec);
213 static int trylock (void *unused)
215 return TryEnterCriticalSection (&critsec) == 0;
217 #else
218 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
220 static void lock (const char *cap)
222 int ret = pthread_mutex_lock (&mutex);
223 if (ret) {
224 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
228 static void unlock (const char *cap)
230 int ret = pthread_mutex_unlock (&mutex);
231 if (ret) {
232 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
236 static int trylock (const char *cap)
238 int ret = pthread_mutex_trylock (&mutex);
240 if (ret && ret != EBUSY) {
241 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
243 return ret == EBUSY;
245 #endif
247 static void *parse_pointer (const char *cap, const char *s)
249 int ret;
250 void *ptr;
252 ret = sscanf (s, "%p", &ptr);
253 if (ret != 1) {
254 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
256 return ptr;
259 static int hasdata (int sock)
261 int ret;
262 fionread_arg avail;
263 ret = ioctlsocket (sock, FIONREAD, &avail);
264 if (ret) sockerr (1, "hasdata: FIONREAD error ret=%d", ret);
265 return avail > 0;
268 static double now (void)
270 struct timeval tv;
272 if (gettimeofday (&tv, NULL)) {
273 err (1, "gettimeofday");
275 return tv.tv_sec + tv.tv_usec*1e-6;
278 static void readdata (int fd, char *p, int size)
280 ssize_t n;
282 n = recv (fd, p, size, 0);
283 if (n - size) {
284 if (!n) errx (1, "EOF while reading");
285 sockerr (1, "recv (req %d, ret " FMT_ss ")", size, n);
289 static void writedata (int fd, char *p, int size)
291 char buf[4];
292 ssize_t n;
294 buf[0] = (size >> 24) & 0xff;
295 buf[1] = (size >> 16) & 0xff;
296 buf[2] = (size >> 8) & 0xff;
297 buf[3] = (size >> 0) & 0xff;
299 n = send (fd, buf, 4, 0);
300 if (n != 4) {
301 if (!n) errx (1, "EOF while writing length");
302 sockerr (1, "send " FMT_ss, n);
305 n = send (fd, p, size, 0);
306 if (n - size) {
307 if (!n) errx (1, "EOF while writing data");
308 sockerr (1, "send (req %d, ret " FMT_ss ")", size, n);
312 static void
313 #ifdef __GNUC__
314 __attribute__ ((format (printf, 2, 3)))
315 #endif
316 printd (int fd, const char *fmt, ...)
318 int size = 200, len;
319 va_list ap;
320 char *buf;
322 buf = malloc (size);
323 for (;;) {
324 if (!buf) err (errno, "malloc for temp buf (%d bytes) failed", size);
326 va_start (ap, fmt);
327 len = vsnprintf (buf, size, fmt, ap);
328 va_end (ap);
330 if (len > -1 && len < size) {
331 writedata (fd, buf, len);
332 break;
335 if (len > -1) {
336 size = len + 1;
338 else {
339 size *= 2;
341 buf = realloc (buf, size);
343 free (buf);
346 static void die (fz_error error)
348 fz_catch (error, "aborting");
349 if (state.xref)
350 pdf_free_xref (state.xref);
351 _exit (1);
354 static void openxref (char *filename, char *password)
356 int i;
357 fz_error error;
359 for (i = 0; i < state.texcount; ++i) {
360 state.texowners[i].slice = NULL;
363 if (state.cache) {
364 fz_free_glyph_cache (state.cache);
367 state.cache = fz_new_glyph_cache ();
368 if (!state.cache) {
369 errx (1, "fz_newglyph_cache failed");
372 if (state.xref) {
373 if (state.xref->store) {
374 pdf_free_store (state.xref->store);
375 state.xref->store = NULL;
377 pdf_free_xref (state.xref);
378 state.xref = NULL;
381 if (state.pagedims) {
382 free (state.pagedims);
383 state.pagedims = NULL;
385 state.pagedimcount = 0;
387 error = pdf_open_xref (&state.xref, filename, password);
388 if (error) {
389 die (error);
392 error = pdf_load_page_tree (state.xref);
393 if (error) {
394 die (error);
397 state.pagecount = pdf_count_pages (state.xref);
400 static void pdfinfo (void)
402 printd (state.sock, "i PDF version %d.%d\n",
403 state.xref->version / 10, state.xref->version % 10);
405 #if _XOPEN_SOURCE >= 700 || _POSIX_C_SOURCE >= 200809L
407 FILE *f;
408 char *buf;
409 size_t size;
410 fz_obj *obj;
412 f = open_memstream (&buf, &size);
413 if (f) {
414 obj = fz_dict_gets (state.xref->trailer, "Info");
415 fz_fprint_obj (f, fz_resolve_indirect (obj), 0);
416 if (fclose (f)) err (1, "fclose on memstream failed");
417 if (size > 64*1024) size = 64*1024;
418 printd (state.sock, "i %.*s", (int) size, buf);
419 free (buf);
421 else {
422 printd (state.sock, "i error opening memstream: %s\n",
423 strerror (errno));
426 #endif
429 static int readlen (int fd)
431 ssize_t n;
432 unsigned char p[4];
434 n = recv (fd, p, 4, 0);
435 if (n != 4) {
436 if (!n) errx (1, "EOF while reading length");
437 sockerr (1, "recv " FMT_ss, n);
440 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
443 static void unlinkpage (struct page *page)
445 int i;
447 for (i = 0; i < page->slicecount; ++i) {
448 struct slice *s = &page->slices[i];
449 if (s->texindex != -1) {
450 if (state.texowners[s->texindex].slice == s) {
451 state.texowners[s->texindex].slice = NULL;
452 ARSERT (state.texowners[s->texindex].w == s->w);
453 ARSERT (state.texowners[s->texindex].h >= s->h);
459 static void freepage (struct page *page)
461 fz_drop_pixmap (page->pixmap);
463 unlinkpage (page);
465 if (page->text) {
466 fz_free_text_span (page->text);
468 if (page->drawpage) {
469 pdf_free_page (page->drawpage);
472 free (page);
475 static void subdivide (struct page *p)
477 int i;
478 int h = p->pixmap->h;
479 int th = MIN (h, state.sliceheight);
481 for (i = 0; i < p->slicecount; ++i) {
482 struct slice *s = &p->slices[i];
483 s->texindex = -1;
484 s->h = MIN (th, h);
485 s->w = p->pixmap->w;
486 h -= th;
490 static int compatpdims (struct pagedim *p1, struct pagedim *p2)
492 return p1->rotate == p2->rotate
493 && !memcmp (&p1->bbox, &p2->bbox, sizeof (p1->bbox))
494 && !memcmp (&p1->ctm, &p2->ctm, sizeof (p1->ctm));
497 #ifdef __ALTIVEC__
498 #include <altivec.h>
500 static int cacheline32bytes;
501 extern char **environ;
503 static void __attribute__ ((constructor)) clcheck (void)
505 char **envp = environ;
506 unsigned long *auxv;
508 while (*envp++);
510 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
511 if (*auxv == 19) {
512 cacheline32bytes = auxv[1] == 32;
513 return;
518 static void __attribute__ ((optimize ("O"))) clearpixmap (fz_pixmap *pixmap)
520 if (cacheline32bytes) {
521 intptr_t a1, a2, diff;
522 size_t sizea, i, size = pixmap->w * pixmap->h * pixmap->n;
523 vector unsigned char v = vec_splat_u8 (-1);
524 vector unsigned char *p;
526 a1 = a2 = (intptr_t) pixmap->samples;
527 a2 = (a1 + 31) & ~31;
528 diff = a2 - a1;
529 sizea = size - diff;
530 p = (void *) a2;
532 while (a1 != a2) *(char *) a1++ = 0xff;
533 for (i = 0; i < (sizea & ~31); i += 32) {
534 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
535 vec_st (v, i, p);
536 vec_st (v, i + 16, p);
538 while (i < sizea) *((char *) a1 + i++) = 0xff;
540 else fz_clear_pixmap_with_color (pixmap, 0xff);
542 #else
543 #define clearpixmap(p) fz_clear_pixmap_with_color (p, 0xff)
544 #endif
546 static void *render (int pageno, int pindex)
548 fz_error error;
549 int slicecount;
550 struct page *page = NULL;
551 double start, end;
552 pdf_page *drawpage;
553 fz_device *idev;
554 struct pagedim *pagedim;
556 start = now ();
557 printd (state.sock, "V rendering %d", pageno);
559 pagedim = &state.pagedims[pindex];
560 slicecount = (pagedim->bbox.y1 - pagedim->bbox.y0
561 + state.sliceheight - 1) / state.sliceheight;
562 slicecount += slicecount == 0;
564 if (state.pig) {
565 if (compatpdims (&state.pig->pagedim, pagedim)) {
566 page = state.pig;
567 if (page->text) {
568 fz_free_text_span (page->text);
569 page->text = NULL;
571 if (page->drawpage) {
572 pdf_free_page (page->drawpage);
573 page->drawpage = NULL;
576 else {
577 freepage (state.pig);
580 if (!page) {
581 page = calloc (sizeof (*page)
582 + (slicecount * sizeof (struct slice)), 1);
583 if (!page) {
584 err (1, "calloc page %d\n", pageno);
586 page->pixmap = fz_new_pixmap_with_rect (fz_device_rgb, pagedim->bbox);
589 page->slicecount = slicecount;
591 error = pdf_load_page (&drawpage, state.xref, pageno - 1);
592 if (error)
593 die (error);
595 clearpixmap (page->pixmap);
597 idev = fz_new_draw_device (state.cache, page->pixmap);
598 if (!idev)
599 die (fz_throw ("fz_newdrawdevice failed"));
600 error = pdf_run_page (state.xref, drawpage, idev, pagedim->ctm);
601 if (error)
602 die (fz_rethrow (error, "pdf_runpage failed"));
603 fz_free_device (idev);
605 page->drawpage = drawpage;
606 page->pagedim = *pagedim;
607 page->pageno = pageno;
608 subdivide (page);
609 end = now ();
611 pdf_age_store (state.xref->store, 3);
613 printd (state.sock, "V rendering %d took %f sec", pageno, end - start);
614 state.pig = NULL;
615 return page;
618 static void initpdims (void)
620 int pageno;
621 double start, end;
623 start = now ();
624 for (pageno = 0; pageno < state.pagecount; ++pageno) {
625 int rotate;
626 fz_rect box;
627 struct pagedim *p;
628 fz_obj *obj, *pageobj;
630 pageobj = state.xref->page_objs[pageno];
632 obj = fz_dict_gets (pageobj, "CropBox");
633 if (!fz_is_array (obj)) {
634 obj = fz_dict_gets (pageobj, "MediaBox");
635 if (!fz_is_array (obj)) {
636 die (fz_throw ("cannot find page bounds %d (%d Rd)",
637 fz_to_num (pageobj), fz_to_gen (pageobj)));
640 box = pdf_to_rect (obj);
642 obj = fz_dict_gets (pageobj, "Rotate");
643 if (fz_is_int (obj)) {
644 rotate = fz_to_int (obj);
646 else {
647 rotate = 0;
649 rotate += state.rotate;
651 p = &state.pagedims[state.pagedimcount - 1];
652 if ((state.pagedimcount == 0)
653 || (p->rotate != rotate || memcmp (&p->box, &box, sizeof (box)))) {
654 size_t size;
656 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
657 state.pagedims = realloc (state.pagedims, size);
658 if (!state.pagedims) {
659 err (1, "realloc pagedims to " FMT_s " (%d elems)",
660 size, state.pagedimcount + 1);
662 p = &state.pagedims[state.pagedimcount++];
663 p->rotate = rotate;
664 p->box = box;
665 p->pageno = pageno;
668 end = now ();
669 printd (state.sock, "T Processed %d pages in %f seconds",
670 state.pagecount, end - start);
673 static void layout (void)
675 int pindex;
676 fz_matrix ctm;
677 fz_rect box, box2;
678 double zoom, w, maxw = 0;
679 struct pagedim *p = state.pagedims;
681 if (state.proportional) {
682 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
683 box.x0 = MIN (p->box.x0, p->box.x1);
684 box.y0 = MIN (p->box.y0, p->box.y1);
685 box.x1 = MAX (p->box.x0, p->box.x1);
686 box.y1 = MAX (p->box.y0, p->box.y1);
688 ctm = fz_identity;
689 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
690 ctm = fz_concat (ctm, fz_rotate (p->rotate));
691 box2 = fz_transform_rect (ctm, box);
692 w = box2.x1 - box2.x0;
693 maxw = MAX (w, maxw);
697 p = state.pagedims;
698 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
699 box.x0 = MIN (p->box.x0, p->box.x1);
700 box.y0 = MIN (p->box.y0, p->box.y1);
701 box.x1 = MAX (p->box.x0, p->box.x1);
702 box.y1 = MAX (p->box.y0, p->box.y1);
704 ctm = fz_identity;
705 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
706 ctm = fz_concat (ctm, fz_rotate (p->rotate));
707 box2 = fz_transform_rect (ctm, box);
708 w = box2.x1 - box2.x0;
710 if (state.proportional) {
711 double scale = w / maxw;
712 zoom = (state.w / w) * scale;
714 else {
715 zoom = state.w / w;
717 ctm = fz_identity;
718 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
719 ctm = fz_concat (ctm, fz_scale (zoom, -zoom));
720 memcpy (&p->ctm1, &ctm, sizeof (ctm));
721 ctm = fz_concat (ctm, fz_rotate (p->rotate));
722 p->bbox = fz_round_rect (fz_transform_rect (ctm, box));
723 p->left = state.proportional ? ((maxw - w) * zoom) / 2.0 : 0;
724 memcpy (&p->ctm, &ctm, sizeof (ctm));
727 while (p-- != state.pagedims) {
728 printd (state.sock, "l %d %d %d %d",
729 p->pageno, p->bbox.x1 - p->bbox.x0, p->bbox.y1 - p->bbox.y0,
730 p->left);
734 static void recurse_outline (pdf_outline *outline, int level)
736 while (outline) {
737 int i;
738 fz_obj *obj;
739 int pageno = -1;
740 int top = 0, h = 0;
742 if (!outline->link) goto next;
744 obj = outline->link->dest;
745 if (fz_is_indirect (obj)) {
746 obj = fz_resolve_indirect (obj);
748 if (fz_is_array (obj)) {
749 fz_obj *obj2;
751 obj2 = fz_array_get (obj, 0);
752 if (fz_is_int (obj2)) {
753 pageno = fz_to_int (obj2);
755 else {
756 pageno = pdf_find_page_number (state.xref, obj2);
759 if (fz_array_len (obj) > 3) {
760 fz_point p;
761 fz_obj *xo, *yo;
763 xo = fz_array_get (obj, 2);
764 yo = fz_array_get (obj, 3);
765 if (!fz_is_null (xo) && !fz_is_null (yo)) {
766 struct pagedim *pagedim = state.pagedims;
768 for (i = 0; i < state.pagedimcount; ++i) {
769 if (state.pagedims[i].pageno > pageno)
770 break;
771 pagedim = &state.pagedims[i];
773 p.x = fz_to_int (xo);
774 p.y = fz_to_int (yo);
775 p = fz_transform_point (pagedim->ctm, p);
776 h = pagedim->bbox.y1 - pagedim->bbox.y0;
777 top = p.y;
781 else {
782 pageno = pdf_find_page_number (state.xref, outline->link->dest);
785 lprintf ("%*c%s %d\n", level, ' ', outline->title, pageno);
786 if (pageno > 0) {
787 printd (state.sock, "o %d %d %d %d %s",
788 level, pageno, top, h, outline->title);
790 next:
791 if (outline->child) {
792 recurse_outline (outline->child, level + 1);
794 outline = outline->next;
798 static void process_outline (void)
800 pdf_outline *outline;
802 if (!state.needoutline) return;
804 state.needoutline = 0;
805 outline = pdf_load_outline (state.xref);
806 if (outline) {
807 recurse_outline (outline, 0);
808 pdf_free_outline (outline);
812 static int comparespans (const void *l, const void *r)
814 fz_text_span const *const*ls = l;
815 fz_text_span const *const*rs = r;
816 return (*ls)->text->bbox.y0 - (*rs)->text->bbox.y0;
819 /* wishful thinking function */
820 static void search (regex_t *re, int pageno, int y, int forward)
822 int i, j;
823 int ret;
824 char *p;
825 char buf[256];
826 fz_matrix ctm;
827 fz_error error;
828 fz_device *tdev;
829 pdf_page *drawpage;
830 fz_text_span *text, *span, **pspan;
831 struct pagedim *pdim, *pdimprev;
832 int stop = 0;
833 int niters = 0;
834 int nspans;
835 double start, end;
837 start = now ();
838 while (pageno >= 0 && pageno < state.pagecount && !stop) {
839 if (niters++ == 5) {
840 pdf_age_store (state.xref->store, 3);
841 niters = 0;
842 if (hasdata (state.sock)) {
843 printd (state.sock, "T attention requested aborting search at %d",
844 pageno);
845 stop = 1;
847 else {
848 printd (state.sock, "T searching in page %d", pageno);
851 pdimprev = NULL;
852 for (i = 0; i < state.pagedimcount; ++i) {
853 pdim = &state.pagedims[i];
854 if (pdim->pageno == pageno) {
855 goto found;
857 if (pdim->pageno > pageno) {
858 pdim = pdimprev;
859 goto found;
861 pdimprev = pdim;
863 pdim = pdimprev;
864 found:
866 error = pdf_load_page (&drawpage, state.xref, pageno);
867 if (error)
868 die (error);
870 ctm = fz_rotate (pdim->rotate);
872 text = fz_new_text_span ();
873 tdev = fz_new_text_device (text);
874 error = pdf_run_page (state.xref, drawpage, tdev, pdim->ctm1);
875 if (error) die (error);
876 fz_free_device (tdev);
878 nspans = 0;
879 for (span = text; span; span = span->next) {
880 nspans++;
882 pspan = malloc (sizeof (void *) * nspans);
883 if (!pspan) {
884 err (1, "malloc span pointers " FMT_s, sizeof (void *) * nspans);
886 for (i = 0, span = text; span; span = span->next, ++i) {
887 pspan[i] = span;
889 qsort (pspan, nspans, sizeof (fz_text_span *), comparespans);
891 j = forward ? 0 : nspans - 1;
892 while (nspans--) {
893 regmatch_t rm;
895 span = pspan[j];
896 j += forward ? 1 : -1;
897 p = buf;
898 for (i = 0; i < MIN (span->len, (int) sizeof (buf) - 1); ++i) {
899 if (forward) {
900 if (span->text[i].bbox.y0 < y + 1) {
901 continue;
904 else {
905 if (span->text[i].bbox.y0 > y - 1) {
906 continue;
909 if (span->text[i].c < 256) {
910 *p++ = span->text[i].c;
912 else {
913 *p++ = '?';
916 if (p == buf) {
917 continue;
919 *p++ = 0;
921 ret = regexec (re, buf, 1, &rm, 0);
922 if (ret) {
923 if (ret != REG_NOMATCH) {
924 size_t size;
925 char errbuf[80];
926 size = regerror (ret, re, errbuf, sizeof (errbuf));
927 printd (state.sock,
928 "T regexec error `%.*s'",
929 (int) size, errbuf);
930 fz_free_text_span (text);
931 pdf_free_page (drawpage);
932 free (pspan);
933 return;
936 else {
937 int xoff, yoff;
938 fz_bbox *sb, *eb;
939 fz_point p1, p2, p3, p4;
941 xoff = pdim->left - pdim->bbox.x0;
942 yoff = -pdim->bbox.y0;
944 sb = &span->text[rm.rm_so].bbox;
945 eb = &span->text[rm.rm_eo - 1].bbox;
947 p1.x = sb->x0;
948 p1.y = sb->y0;
949 p2.x = eb->x1;
950 p2.y = sb->y0;
951 p3.x = eb->x1;
952 p3.y = eb->y1;
953 p4.x = sb->x0;
954 p4.y = eb->y1;
956 p1 = fz_transform_point (ctm, p1);
957 p2 = fz_transform_point (ctm, p2);
958 p3 = fz_transform_point (ctm, p3);
959 p4 = fz_transform_point (ctm, p4);
961 if (!stop) {
962 printd (state.sock, "F %d %d %f %f %f %f %f %f %f %f",
963 pageno, 1,
964 p1.x + xoff, p1.y + yoff,
965 p2.x + xoff, p2.y + yoff,
966 p3.x + xoff, p3.y + yoff,
967 p4.x + xoff, p4.y + yoff);
969 printd (state.sock, "T found at %d `%.*s' in %f sec",
970 pageno, rm.rm_eo - rm.rm_so, &buf[rm.rm_so],
971 now () - start);
973 else {
974 printd (state.sock, "R %d %d %f %f %f %f %f %f %f %f",
975 pageno, 2,
976 p1.x + xoff, p1.y + yoff,
977 p2.x + xoff, p2.y + yoff,
978 p3.x + xoff, p3.y + yoff,
979 p4.x + xoff, p4.y + yoff);
981 stop = 1;
984 if (forward) {
985 pageno += 1;
986 y = 0;
988 else {
989 pageno -= 1;
990 y = INT_MAX;
992 fz_free_text_span (text);
993 pdf_free_page (drawpage);
994 free (pspan);
996 end = now ();
997 if (!stop) {
998 printd (state.sock, "T no matches %f sec", end - start);
1000 printd (state.sock, "D");
1003 static
1004 #ifdef _WIN32
1005 DWORD _stdcall
1006 #else
1007 void *
1008 #endif
1009 mainloop (void *unused)
1011 char *p = NULL;
1012 int len, ret, oldlen = 0;
1014 for (;;) {
1015 len = readlen (state.sock);
1016 if (len == 0) {
1017 errx (1, "readlen returned 0");
1020 if (oldlen < len + 1) {
1021 p = realloc (p, len + 1);
1022 if (!p) {
1023 err (1, "realloc %d failed", len + 1);
1025 oldlen = len + 1;
1027 readdata (state.sock, p, len);
1028 p[len] = 0;
1030 if (!strncmp ("open", p, 4)) {
1031 fz_obj *obj;
1032 size_t filenamelen;
1033 char *password;
1034 char *filename = p + 5;
1036 filenamelen = strlen (filename);
1037 password = filename + filenamelen + 1;
1039 openxref (filename, password);
1040 initpdims ();
1042 obj = fz_dict_gets (state.xref->trailer, "Info");
1043 if (obj) {
1044 char *s;
1046 obj = fz_dict_gets (obj, "Title");
1047 s = pdf_to_utf8 (obj);
1048 if (*s) {
1049 printd (state.sock, "t %s", s);
1051 fz_free (s);
1054 state.needoutline = 1;
1056 else if (!strncmp ("free", p, 4)) {
1057 void *ptr;
1059 ret = sscanf (p + 4, " %p", &ptr);
1060 if (ret != 1) {
1061 errx (1, "malformed free `%.*s' ret=%d", len, p, ret);
1063 unlinkpage (ptr);
1064 state.pig = ptr;
1066 else if (!strncmp ("search", p, 6)) {
1067 int icase, pageno, y, ret, len2, forward;
1068 char *pattern;
1069 regex_t re;
1071 ret = sscanf (p + 6, " %d %d %d %d,%n",
1072 &icase, &pageno, &y, &forward, &len2);
1073 if (ret != 4) {
1074 errx (1, "malformed search `%s' ret=%d", p, ret);
1077 pattern = p + 6 + len2;
1078 ret = regcomp (&re, pattern,
1079 REG_EXTENDED | (icase ? REG_ICASE : 0));
1080 if (ret) {
1081 char errbuf[80];
1082 size_t size;
1084 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1085 printd (state.sock, "T regcomp failed `%.*s'",
1086 (int) size, errbuf);
1088 else {
1089 search (&re, pageno, y, forward);
1090 regfree (&re);
1093 else if (!strncmp ("geometry", p, 8)) {
1094 int w, h;
1096 printd (state.sock, "c");
1097 ret = sscanf (p + 8, " %d %d", &w, &h);
1098 if (ret != 2) {
1099 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1102 lock ("geometry");
1103 state.h = h;
1104 if (w != state.w) {
1105 int i;
1106 state.w = w;
1107 for (i = 0; i < state.texcount; ++i) {
1108 state.texowners[i].slice = NULL;
1111 layout ();
1112 process_outline ();
1113 unlock ("geometry");
1114 printd (state.sock, "C %d", state.pagecount);
1116 else if (!strncmp ("reinit", p, 6)) {
1117 float rotate;
1118 int proportional;
1120 printd (state.sock, "c");
1121 ret = sscanf (p + 6, " %f %d", &rotate, &proportional);
1122 if (ret != 2) {
1123 errx (1, "bad rotate line `%.*s' ret=%d", len, p, ret);
1125 lock ("reinit");
1126 state.rotate = rotate;
1127 state.proportional = proportional;
1128 state.pagedimcount = 0;
1129 free (state.pagedims);
1130 state.pagedims = NULL;
1131 initpdims ();
1132 layout ();
1133 process_outline ();
1134 if (state.pig) {
1135 freepage (state.pig);
1136 state.pig = NULL;
1138 unlock ("reinit");
1139 printd (state.sock, "C %d", state.pagecount);
1141 else if (!strncmp ("render", p, 6)) {
1142 int pageno, pindex, w, h, ret;
1143 struct page *page;
1145 ret = sscanf (p + 6, " %d %d %d %d", &pageno, &pindex, &w, &h);
1146 if (ret != 4) {
1147 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1150 lock ("render");
1151 page = render (pageno, pindex);
1152 unlock ("render");
1154 printd (state.sock, "r %d %d %d %d %d %d %p",
1155 pageno,
1156 state.w,
1157 state.h,
1158 state.rotate,
1159 state.proportional,
1160 w * h * 4,
1161 page);
1163 else if (!strncmp ("info", p, 4)) {
1164 pdfinfo ();
1166 else if (!strncmp ("interrupt", p, 9)) {
1167 printd (state.sock, "V interrupted");
1169 else {
1170 errx (1, "unknown command %.*s", len, p);
1173 return 0;
1176 static void showsel (struct page *page, int oy)
1178 int ox;
1179 fz_bbox bbox;
1180 fz_text_span *span;
1181 struct mark first, last;
1183 first = page->fmark;
1184 last = page->lmark;
1186 if (!first.span || !last.span) return;
1188 glEnable (GL_BLEND);
1189 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1190 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1192 ox = -page->pixmap->x + page->pagedim.left;
1193 oy = -page->pixmap->y + oy;
1194 for (span = first.span; span; span = span->next) {
1195 int i, j, k;
1197 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
1199 j = 0;
1200 k = span->len - 1;
1202 if (span == page->fmark.span && span == page->lmark.span) {
1203 j = MIN (first.i, last.i);
1204 k = MAX (first.i, last.i);
1206 else if (span == first.span) {
1207 j = first.i;
1209 else if (span == last.span) {
1210 k = last.i;
1213 for (i = j; i <= k; ++i) {
1214 bbox = fz_union_bbox (bbox, span->text[i].bbox);
1216 lprintf ("%d %d %d %d oy=%d ox=%d\n",
1217 bbox.x0,
1218 bbox.y0,
1219 bbox.x1,
1220 bbox.y1,
1221 oy, ox);
1223 glRecti (bbox.x0 + ox, bbox.y0 + oy, bbox.x1 + ox, bbox.y1 + oy);
1225 if (span == last.span) break;
1227 glDisable (GL_BLEND);
1230 static void highlightlinks (struct page *page, int yoff)
1232 pdf_link *link;
1233 int xoff;
1235 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1236 glEnable (GL_LINE_STIPPLE);
1237 glLineStipple (0.5, 0xcccc);
1239 xoff = page->pagedim.left - page->pixmap->x;
1240 yoff -= page->pixmap->y;
1242 glBegin (GL_QUADS);
1243 for (link = page->drawpage->links; link; link = link->next) {
1244 fz_point p1, p2, p3, p4;
1245 fz_matrix ctm = page->pagedim.ctm;
1247 p1.x = link->rect.x0;
1248 p1.y = link->rect.y0;
1250 p2.x = link->rect.x1;
1251 p2.y = link->rect.y0;
1253 p3.x = link->rect.x1;
1254 p3.y = link->rect.y1;
1256 p4.x = link->rect.x0;
1257 p4.y = link->rect.y1;
1259 p1 = fz_transform_point (ctm, p1);
1260 p2 = fz_transform_point (ctm, p2);
1261 p3 = fz_transform_point (ctm, p3);
1262 p4 = fz_transform_point (ctm, p4);
1264 switch (link->kind) {
1265 case PDF_LINK_GOTO: glColor3ub (255, 0, 0); break;
1266 case PDF_LINK_URI: glColor3ub (0, 0, 255); break;
1267 default: glColor3ub (0, 0, 0); break;
1270 glVertex2f (p1.x + xoff, p1.y + yoff);
1271 glVertex2f (p2.x + xoff, p2.y + yoff);
1272 glVertex2f (p3.x + xoff, p3.y + yoff);
1273 glVertex2f (p4.x + xoff, p4.y + yoff);
1275 glEnd ();
1277 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1278 glDisable (GL_LINE_STIPPLE);
1281 static void upload2 (struct page *page, int slicenum, const char *cap)
1283 int i;
1284 int w;
1285 double start, end;
1286 struct slice *slice = &page->slices[slicenum];
1288 w = page->pixmap->w;
1290 ARSERT (w == slice->w);
1291 if (slice->texindex != -1
1292 && state.texowners[slice->texindex].slice == slice) {
1293 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1295 else {
1296 int subimage = 0;
1297 int index = (state.texindex++ % state.texcount);
1298 size_t offset = 0;
1300 for (i = 0; i < slicenum; ++i) {
1301 offset += w * page->slices[i].h * 4;
1304 if (state.texowners[index].w == slice->w) {
1305 if (state.texowners[index].h >= slice->h ) {
1306 subimage = 1;
1308 else {
1309 state.texowners[index].h = slice->h;
1312 else {
1313 state.texowners[index].h = slice->h;
1316 state.texowners[index].slice = slice;
1317 state.texowners[index].w = slice->w;
1318 slice->texindex = index;
1320 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1321 start = now ();
1322 if (subimage) {
1324 GLenum err = glGetError ();
1325 if (err != GL_NO_ERROR) {
1326 printf ("\033[0;31mERROR1 %d %d %#x\033[0m\n", w, slice->h, err);
1327 abort ();
1330 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
1335 slice->h,
1336 state.texform,
1337 state.texty,
1338 page->pixmap->samples + offset
1341 GLenum err = glGetError ();
1342 if (err != GL_NO_ERROR) {
1343 printf ("\033[0;31mERROR %d %d %#x\033[0m\n", w, slice->h, err);
1344 abort ();
1348 else {
1349 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
1351 GL_RGBA8,
1353 slice->h,
1355 state.texform,
1356 state.texty,
1357 page->pixmap->samples + offset
1361 end = now ();
1362 (void) start;
1363 (void) end;
1364 lprintf ("%s[%d] slice=%d(%d,%d) texid=%d %f sec\n",
1365 subimage ? "sub" : "img",
1366 page->pageno, slicenum,
1367 slice->w, slice->h,
1368 state.texids[slice->texindex],
1369 end - start);
1373 CAMLprim value ml_draw (value args_v, value ptr_v)
1375 CAMLparam2 (args_v, ptr_v);
1376 int dispy = Int_val (Field (args_v, 0));
1377 int w = Int_val (Field (args_v, 1));
1378 int h = Int_val (Field (args_v, 2));
1379 int py = Int_val (Field (args_v, 3));
1380 int hlinks = Bool_val (Field (args_v, 4));
1381 char *s = String_val (ptr_v);
1382 int ret;
1383 void *ptr;
1384 struct page *page;
1385 int slicenum = 0;
1386 int yoff = dispy - py;
1388 ret = sscanf (s, "%p", &ptr);
1389 if (ret != 1) {
1390 errx (1, "cannot parse pointer `%s'", s);
1392 page = ptr;
1394 w = page->pixmap->w;
1396 ARSERT (h >= 0 && "ml_draw wrong h");
1397 ARSERT (py <= page->pixmap->h && "ml_draw wrong py");
1399 glEnable (GL_TEXTURE_RECTANGLE_ARB);
1401 for (slicenum = 0; slicenum < page->slicecount; ++slicenum) {
1402 struct slice *slice = &page->slices[slicenum];
1403 if (slice->h > py) {
1404 break;
1406 py -= slice->h;
1409 h = MIN (state.h, h);
1410 while (h) {
1411 int th, left;
1412 struct slice *slice = &page->slices[slicenum];
1414 ARSERT (slicenum < page->slicecount && "ml_draw wrong slicenum");
1416 th = MIN (h, slice->h - py);
1417 upload2 (page, slicenum, "upload");
1419 left = page->pagedim.left;
1420 glBegin (GL_QUADS);
1422 glTexCoord2i (0, py);
1423 glVertex2i (left, dispy);
1425 glTexCoord2i (w, py);
1426 glVertex2i (left+w, dispy);
1428 glTexCoord2i (w, py+th);
1429 glVertex2i (left+w, dispy + th);
1431 glTexCoord2i (0, py+th);
1432 glVertex2i (left, dispy + th);
1434 glEnd ();
1436 h -= th;
1437 py = 0;
1438 dispy += th;
1439 slicenum += 1;
1442 showsel (page, yoff);
1443 if (hlinks) highlightlinks (page, yoff);
1444 glDisable (GL_TEXTURE_RECTANGLE_ARB);
1446 CAMLreturn (Val_unit);
1449 static pdf_link *getlink (struct page *page, int x, int y)
1451 fz_point p;
1452 fz_matrix ctm;
1453 pdf_link *link;
1455 p.x = x + page->pixmap->x;
1456 p.y = y + page->pixmap->y;
1458 ctm = fz_invert_matrix (page->pagedim.ctm);
1459 p = fz_transform_point (ctm, p);
1461 for (link = page->drawpage->links; link; link = link->next) {
1462 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
1463 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
1464 return link;
1468 return NULL;
1471 static void ensuretext (struct page *page)
1473 if (!page->text) {
1474 fz_error error;
1475 fz_device *tdev;
1477 page->text = fz_new_text_span ();
1478 tdev = fz_new_text_device (page->text);
1479 error = pdf_run_page (state.xref, page->drawpage, tdev,
1480 page->pagedim.ctm);
1481 if (error) die (error);
1482 fz_free_device (tdev);
1486 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
1488 CAMLparam3 (ptr_v, x_v, y_v);
1489 CAMLlocal3 (ret_v, tup_v, str_v);
1490 pdf_link *link;
1491 struct page *page;
1492 char *s = String_val (ptr_v);
1494 ret_v = Val_int (0);
1495 if (trylock ("ml_whatsunder")) {
1496 goto done;
1499 page = parse_pointer ("ml_whatsunder", s);
1500 link = getlink (page, Int_val (x_v), Int_val (y_v));
1501 if (link) {
1502 switch (link->kind) {
1503 case PDF_LINK_GOTO:
1505 int pageno;
1506 fz_point p;
1507 fz_obj *obj;
1509 pageno = -1;
1510 p.x = 0;
1511 p.y = 0;
1513 if (fz_is_array (link->dest)) {
1514 obj = fz_array_get (link->dest, 0);
1515 if (fz_is_indirect (obj)) {
1516 pageno = pdf_find_page_number (state.xref, obj);
1518 else if (fz_is_int (obj)) {
1519 pageno = fz_to_int (obj);
1522 if (fz_array_len (link->dest) > 3) {
1523 fz_obj *xo, *yo;
1525 xo = fz_array_get (link->dest, 2);
1526 yo = fz_array_get (link->dest, 3);
1527 if (!fz_is_null (xo) && !fz_is_null (yo)) {
1528 p.x = fz_to_int (xo);
1529 p.y = fz_to_int (yo);
1530 p = fz_transform_point (page->pagedim.ctm, p);
1534 else {
1535 pageno = pdf_find_page_number (state.xref, link->dest);
1537 tup_v = caml_alloc_tuple (2);
1538 ret_v = caml_alloc_small (1, 1);
1539 Field (tup_v, 0) = Val_int (pageno);
1540 Field (tup_v, 1) = Val_int (p.y);
1541 Field (ret_v, 0) = tup_v;
1543 break;
1545 case PDF_LINK_URI:
1546 str_v = caml_copy_string (fz_to_str_buf (link->dest));
1547 ret_v = caml_alloc_small (1, 0);
1548 Field (ret_v, 0) = str_v;
1549 break;
1551 default:
1552 printd (state.sock, "T unhandled link kind %d", link->kind);
1553 break;
1556 else {
1557 int i, x, y;
1558 fz_text_span *span;
1560 ensuretext (page);
1561 x = Int_val (x_v) + page->pixmap->x;
1562 y = Int_val (y_v) + page->pixmap->y;
1564 for (span = page->text; span; span = span->next) {
1565 for (i = 0; i < span->len; ++i) {
1566 fz_bbox *b;
1567 b = &span->text[i].bbox;
1568 if ((x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1)) {
1569 const char *n2 =
1570 span->font && span->font->name
1571 ? span->font->name
1572 : "Span has no font name"
1574 #ifdef FT_FREETYPE_H
1575 FT_FaceRec *face = span->font->ft_face;
1576 if (face && face->family_name) {
1577 char *s;
1578 char *n1 = face->family_name;
1579 size_t l1 = strlen (n1);
1580 size_t l2 = strlen (n2);
1582 if (l1 != l2 || memcmp (n1, n2, l1)) {
1583 s = malloc (l1 + l2 + 2);
1584 if (s) {
1585 memcpy (s, n2, l2);
1586 s[l2] = '=';
1587 memcpy (s + l2 + 1, n1, l1 + 1);
1588 str_v = caml_copy_string (s);
1589 free (s);
1593 if (str_v == 0) {
1594 str_v = caml_copy_string (n2);
1596 #else
1597 str_v = caml_copy_string (n2);
1598 #endif
1599 ret_v = caml_alloc_small (1, 2);
1600 Field (ret_v, 0) = str_v;
1601 goto unlock;
1606 unlock:
1607 unlock ("ml_whatsunder");
1609 done:
1610 CAMLreturn (ret_v);
1613 CAMLprim value ml_seltext (value ptr_v, value rect_v, value oy_v)
1615 CAMLparam4 (ptr_v, rect_v, oy_v, rect_v);
1616 fz_bbox *b;
1617 struct page *page;
1618 fz_text_span *span;
1619 struct mark first, last;
1620 int i, x0, x1, y0, y1, oy, left;
1621 char *s = String_val (ptr_v);
1623 if (trylock ("ml_seltext")) {
1624 goto done;
1627 page = parse_pointer ("ml_seltext", s);
1628 ensuretext (page);
1630 oy = Int_val (oy_v);
1631 x0 = Int_val (Field (rect_v, 0));
1632 y0 = Int_val (Field (rect_v, 1));
1633 x1 = Int_val (Field (rect_v, 2));
1634 y1 = Int_val (Field (rect_v, 3));
1636 left = page->pagedim.left;
1637 if (0) {
1638 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1639 glColor3ub (128, 128, 128);
1640 glRecti (x0, y0, x1, y1);
1641 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1644 x0 += page->pixmap->x - left;
1645 y0 += page->pixmap->y - oy;
1646 x1 += page->pixmap->x - left;
1647 y1 += page->pixmap->y - oy;
1649 first.span = NULL;
1650 last.span = NULL;
1652 last.i = first.i = 0;
1653 first.span = page->text;
1654 for (span = page->text; span; span = span->next) {
1655 for (i = 0; i < span->len; ++i) {
1656 b = &span->text[i].bbox;
1657 int selected = 0;
1659 if (x0 >= b->x0 && x0 <= b->x1 && y0 >= b->y0 && y0 <= b->y1) {
1660 first.i = i;
1661 first.span = span;
1662 selected = 1;
1664 if (x1 >= b->x0 && x1 <= b->x1 && y1 >= b->y0 && y1 <= b->y1) {
1665 last.i = i;
1666 last.span = span;
1667 selected = 1;
1669 if (0 && selected) {
1670 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1671 glColor3ub (128, 128, 128);
1672 glRecti (b->x0+left, b->y0, b->x1+left, b->y1);
1673 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1678 if (y1 < y0 || x1 < x0) {
1679 int swap = 0;
1681 if (first.span == last.span) {
1682 swap = 1;
1684 else {
1685 if (y1 < y0) {
1686 for (span = first.span; span && span != last.span;
1687 span = span->next) {
1688 if (span->eol) {
1689 swap = 1;
1690 break;
1696 if (swap) {
1697 i = first.i;
1698 span = first.span;
1699 first.i = last.i;
1700 first.span = last.span;
1701 last.i = i;
1702 last.span = span;
1706 page->fmark = first;
1707 page->lmark = last;
1709 unlock ("ml_seltext");
1711 done:
1712 CAMLreturn (Val_unit);
1715 static int pipespan (FILE *f, fz_text_span *span, int a, int b)
1717 char buf[4];
1718 int i, len, ret;
1720 for (i = a; i <= b; ++i) {
1721 len = runetochar (buf, &span->text[i].c);
1722 ret = fwrite (buf, len, 1, f);
1724 if (ret != 1) {
1725 printd (state.sock, "T failed to write %d bytes ret=%d: %s",
1726 len, ret, strerror (errno));
1727 return -1;
1730 return 0;
1733 CAMLprim value ml_copysel (value ptr_v)
1735 CAMLparam1 (ptr_v);
1736 FILE *f;
1737 struct page *page;
1738 char *s = String_val (ptr_v);
1740 if (trylock ("ml_copysel")) {
1741 goto done;
1744 if (!*s) {
1745 close:
1746 #ifdef USE_XSEL
1747 if (state.xselpipe) {
1748 int ret = pclose (state.xselpipe);
1749 if (ret) {
1750 printd (state.sock, "T failed to close xsel pipe `%s'",
1751 strerror (errno));
1753 state.xselpipe = NULL;
1755 #else
1756 printf ("========================================\n");
1757 #endif
1759 else {
1760 fz_text_span *span;
1762 page = parse_pointer ("ml_sopysel", s);
1764 if (!page->fmark.span || !page->lmark.span) {
1765 printd (state.sock, "T nothing to copy");
1766 goto unlock;
1769 f = stdout;
1770 #ifdef USE_XSEL
1771 if (!state.xselpipe) {
1772 state.xselpipe = popen ("xsel -i", "w");
1773 if (!state.xselpipe) {
1774 printd (state.sock, "T failed to open xsel pipe `%s'",
1775 strerror (errno));
1777 else {
1778 f = state.xselpipe;
1781 else {
1782 f = state.xselpipe;
1784 #endif
1786 for (span = page->fmark.span;
1787 span && span != page->lmark.span->next;
1788 span = span->next) {
1789 int a = span == page->fmark.span ? page->fmark.i : 0;
1790 int b = span == page->lmark.span ? page->lmark.i : span->len - 1;
1791 if (pipespan (f, span, a, b)) {
1792 goto close;
1794 if (span->eol) {
1795 if (putc ('\n', f) == EOF) {
1796 printd (state.sock, "T failed break line on xsel pipe `%s'",
1797 strerror (errno));
1798 goto close;
1802 page->lmark.span = NULL;
1803 page->fmark.span = NULL;
1806 unlock:
1807 unlock ("ml_copysel");
1809 done:
1810 CAMLreturn (Val_unit);
1813 CAMLprim value ml_getpdimrect (value pagedimno_v)
1815 CAMLparam1 (pagedimno_v);
1816 CAMLlocal1 (ret_v);
1817 int pagedimno = Int_val (pagedimno_v);
1818 fz_rect box;
1820 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
1821 if (trylock ("ml_getpdimrect")) {
1822 box = fz_empty_rect;
1824 else {
1825 box = state.pagedims[pagedimno].box;
1826 unlock ("ml_getpdimrect");
1829 Store_double_field (ret_v, 0, box.x0);
1830 Store_double_field (ret_v, 1, box.x1);
1831 Store_double_field (ret_v, 2, box.y0);
1832 Store_double_field (ret_v, 3, box.y1);
1834 CAMLreturn (ret_v);
1837 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v, value dw_v)
1839 CAMLparam3 (winw_v, winh_v, dw_v);
1840 CAMLlocal1 (ret_v);
1841 int i;
1842 double zoom = 1.0;
1843 double maxw = 0.0, maxh = 0.0;
1844 struct pagedim *p;
1845 double winw = Int_val (winw_v);
1846 double winh = Int_val (winh_v);
1847 double dw = Int_val (dw_v);
1848 double pw = 1.0, ph = 1.0, num, den;
1850 if (trylock ("ml_zoom_for_height")) {
1851 goto done;
1854 if (state.proportional) {
1855 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
1856 double x0, x1, w;
1858 x0 = MIN (p->box.x0, p->box.x1);
1859 x1 = MAX (p->box.x0, p->box.x1);
1861 w = x1 - x0;
1862 maxw = MAX (w, maxw);
1866 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
1867 double x0, x1, y0, y1, w, h, scaledh, scale;
1869 x0 = MIN (p->box.x0, p->box.x1);
1870 y0 = MIN (p->box.y0, p->box.y1);
1871 x1 = MAX (p->box.x0, p->box.x1);
1872 y1 = MAX (p->box.y0, p->box.y1);
1874 w = x1 - x0;
1875 h = y1 - y0;
1877 if (state.proportional) {
1878 scale = w / maxw;
1879 scaledh = h * scale;
1881 else {
1882 scale = 1.0;
1883 scaledh = h;
1886 if (scaledh > maxh) {
1887 maxh = scaledh;
1888 ph = scaledh;
1889 pw = w * scale;
1893 num = (winh * pw) + (ph * dw);
1894 den = ph * winw;
1895 zoom = num / den;
1897 unlock ("ml_zoom_for_height");
1898 done:
1899 ret_v = caml_copy_double (zoom);
1900 CAMLreturn (ret_v);
1903 #include "glfont.c"
1905 CAMLprim value ml_draw_string (value pt_v, value x_v, value y_v, value string_v)
1907 CAMLparam4 (pt_v, x_v, y_v, string_v);
1908 int pt = Int_val(pt_v);
1909 int x = Int_val (x_v);
1910 int y = Int_val (y_v);
1911 double w;
1913 w = draw_string (state.face, pt, x, y, String_val (string_v));
1914 CAMLreturn (caml_copy_double (w));
1917 CAMLprim value ml_measure_string (value pt_v, value string_v)
1919 CAMLparam2 (pt_v, string_v);
1920 CAMLlocal1 (ret_v);
1921 int pt = Int_val (pt_v);
1922 double w;
1924 w = measure_string (state.face, pt, String_val (string_v));
1925 CAMLreturn (caml_copy_double (w));
1928 CAMLprim value ml_init (value sock_v, value params_v)
1930 CAMLparam2 (sock_v, params_v);
1931 #ifndef _WIN32
1932 int ret;
1933 #endif
1934 char *fontpath;
1936 state.rotate = Int_val (Field (params_v, 0));
1937 state.proportional = Bool_val (Field (params_v, 1));
1938 state.texcount = Int_val (Field (params_v, 2));
1939 state.sliceheight = Int_val (Field (params_v, 3));
1940 fontpath = String_val (Field (params_v, 4));
1941 state.texform = GL_RGBA;
1942 state.texty = GL_UNSIGNED_BYTE;
1944 if (*fontpath) {
1945 state.face = load_font (fontpath);
1947 else {
1948 unsigned int len;
1949 void *base = pdf_find_builtin_font ("Courier-Bold", &len);
1951 state.face = load_builtin_font (base, len);
1953 if (!state.face) _exit (1);
1955 fz_accelerate ();
1956 state.texids = calloc (state.texcount * sizeof (*state.texids), 1);
1957 if (!state.texids) {
1958 err (1, "calloc texids " FMT_s,
1959 state.texcount * sizeof (*state.texids));
1962 state.texowners = calloc (state.texcount * sizeof (*state.texowners), 1);
1963 if (!state.texowners) {
1964 err (1, "calloc texowners " FMT_s,
1965 state.texcount * sizeof (*state.texowners));
1968 glGenTextures (state.texcount, state.texids);
1970 #ifdef _WIN32
1971 state.sock = Socket_val (sock_v);
1972 #else
1973 state.sock = Int_val (sock_v);
1974 #endif
1976 #ifdef _WIN32
1977 InitializeCriticalSection (&critsec);
1978 state.thread = CreateThread (NULL, 0, mainloop, NULL, 0, NULL);
1979 if (state.thread == INVALID_HANDLE_VALUE) {
1980 errx (1, "CreateThread failed: %lx", GetLastError ());
1982 #else
1983 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
1984 if (ret) {
1985 errx (1, "pthread_create: %s", strerror (ret));
1987 #endif
1989 CAMLreturn (Val_unit);