Add -v command line switch
[llpp.git] / link.c
blob27f93b92a55b54c1851791b90b05dc627c4b84f3
1 /* lots of code c&p-ed directly from mupdf */
2 #ifdef _WIN32
3 #define WIN32_LEAN_AND_MEAN
4 #include <windows.h>
5 #include <winsock2.h>
6 #define fionread_arg long
7 #define ssize_t int
8 #define FMT_ss "%d"
9 #ifdef _WIN64
10 #define FMT_s "%i64u"
11 #else
12 #define FMT_s "%u"
13 #endif
14 #pragma warning (disable:4244)
15 #pragma warning (disable:4996)
16 #pragma warning (disable:4995)
17 #endif
19 #ifdef _MSC_VER
20 #include <errno.h>
21 #include <stdio.h>
22 #include <string.h>
23 #include <stdlib.h>
24 static void __declspec (noreturn) err (int exitcode, const char *fmt, ...)
26 va_list ap;
27 int errcode;
29 errcode = errno;
30 va_start (ap, fmt);
31 vfprintf (stderr, fmt, ap);
32 va_end (ap);
33 fprintf (stderr, ": %s\n", strerror (errno));
34 exit (exitcode);
36 static void __declspec (noreturn) errx (int exitcode, const char *fmt, ...)
38 va_list ap;
40 va_start (ap, fmt);
41 vfprintf (stderr, fmt, ap);
42 va_end (ap);
43 fputc ('\n', stderr);
44 exit (exitcode);
46 static void __declspec (noreturn) sockerr (int exitcode, const char *fmt, ...)
48 va_list ap;
50 va_start (ap, fmt);
51 vfprintf (stderr, fmt, ap);
52 va_end (ap);
53 fprintf (stderr, ": wsaerror 0x%x\n", WSAGetLastError ());
54 exit (exitcode);
56 #else
57 #define FMT_ss "%zd"
58 #define FMT_s "%zu"
59 #define fionread_arg int
60 #define ioctlsocket ioctl
61 #define _GNU_SOURCE
62 #include <err.h>
63 #define sockerr err
64 #endif
65 #include <regex.h>
66 #include <errno.h>
67 #include <ctype.h>
68 #include <stdio.h>
69 #include <stdarg.h>
70 #include <stdlib.h>
71 #include <string.h>
72 #include <limits.h>
73 #ifndef _WIN32
74 #include <pthread.h>
75 #include <sys/time.h>
76 #include <sys/types.h>
77 #include <sys/socket.h>
78 #include <sys/ioctl.h>
79 #endif
81 #ifdef __APPLE__
82 #include <OpenGL/gl.h>
83 #else
84 #include <GL/gl.h>
85 #endif
87 #ifndef GL_TEXTURE_RECTANGLE_ARB
88 #define GL_TEXTURE_RECTANGLE_ARB 0x84F5
89 #endif
91 #include <caml/fail.h>
92 #include <caml/alloc.h>
93 #include <caml/memory.h>
94 #include <caml/unixsupport.h>
96 #include <fitz.h>
97 #include <mupdf.h>
99 #if 0
100 #define lprintf printf
101 #else
102 #define lprintf(...)
103 #endif
105 #ifdef FT_FREETYPE_H
106 #include FT_FREETYPE_H
107 #endif
109 #define ARSERT(cond) for (;;) { \
110 if (!(cond)) { \
111 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
113 break; \
116 struct slice {
117 int texindex;
118 int w, h;
121 struct pagedim {
122 int pageno;
123 int rotate;
124 int left;
125 fz_rect box;
126 fz_bbox bbox;
127 fz_matrix ctm, ctm1;
130 struct page {
131 int pageno;
132 int slicecount;
133 fz_text_span *text;
134 fz_pixmap *pixmap;
135 pdf_page *drawpage;
136 struct pagedim pagedim;
137 struct mark {
138 int i;
139 fz_text_span *span;
140 } fmark, lmark;
141 struct slice slices[];
144 #if !defined _WIN32 && !defined __APPLE__
145 #define USE_XSEL
146 #endif
148 struct {
149 int sock;
150 int sliceheight;
151 struct page *pig;
152 struct pagedim *pagedims;
153 int pagecount;
154 int pagedimcount;
155 pdf_xref *xref;
156 fz_glyph_cache *cache;
157 int w, h;
159 int texindex;
160 int texcount;
161 GLuint *texids;
163 GLenum texform;
164 GLenum texty;
166 struct {
167 int w, h;
168 struct slice *slice;
169 } *texowners;
171 int rotate;
172 int proportional;
173 int needoutline;
175 #ifdef _WIN32
176 HANDLE thread;
177 #else
178 pthread_t thread;
179 #endif
180 FILE *xselpipe;
181 } state;
183 #ifdef _WIN32
184 static CRITICAL_SECTION critsec;
186 static void lock (void *unused)
188 (void) unused;
189 EnterCriticalSection (&critsec);
192 static void unlock (void *unused)
194 (void) unused;
195 LeaveCriticalSection (&critsec);
198 static int trylock (void *unused)
200 return TryEnterCriticalSection (&critsec) == 0;
202 #else
203 static pthread_mutex_t mutex = PTHREAD_MUTEX_INITIALIZER;
205 static void lock (const char *cap)
207 int ret = pthread_mutex_lock (&mutex);
208 if (ret) {
209 errx (1, "%s: pthread_mutex_lock: %s", cap, strerror (ret));
213 static void unlock (const char *cap)
215 int ret = pthread_mutex_unlock (&mutex);
216 if (ret) {
217 errx (1, "%s: pthread_mutex_unlock: %s", cap, strerror (ret));
221 static int trylock (const char *cap)
223 int ret = pthread_mutex_trylock (&mutex);
225 if (ret && ret != EBUSY) {
226 errx (1, "%s: pthread_mutex_trylock: %s", cap, strerror (ret));
228 return ret == EBUSY;
230 #endif
232 static void *parse_pointer (const char *cap, const char *s)
234 int ret;
235 void *ptr;
237 ret = sscanf (s, "%p", &ptr);
238 if (ret != 1) {
239 errx (1, "%s: cannot parse pointer in `%s'", cap, s);
241 return ptr;
244 static int hasdata (int sock)
246 int ret;
247 fionread_arg avail;
248 ret = ioctlsocket (sock, FIONREAD, &avail);
249 if (ret) sockerr (1, "hasdata: FIONREAD error ret=%d", ret);
250 return avail > 0;
253 static double now (void)
255 struct timeval tv;
257 if (gettimeofday (&tv, NULL)) {
258 err (1, "gettimeofday");
260 return tv.tv_sec + tv.tv_usec*1e-6;
263 static void readdata (int fd, char *p, int size)
265 ssize_t n;
267 n = recv (fd, p, size, 0);
268 if (n - size) {
269 if (!n) errx (1, "EOF while reading");
270 sockerr (1, "recv (req %d, ret " FMT_ss ")", size, n);
274 static void writedata (int fd, char *p, int size)
276 char buf[4];
277 ssize_t n;
279 buf[0] = (size >> 24) & 0xff;
280 buf[1] = (size >> 16) & 0xff;
281 buf[2] = (size >> 8) & 0xff;
282 buf[3] = (size >> 0) & 0xff;
284 n = send (fd, buf, 4, 0);
285 if (n != 4) {
286 if (!n) errx (1, "EOF while writing length");
287 sockerr (1, "send " FMT_ss, n);
290 n = send (fd, p, size, 0);
291 if (n - size) {
292 if (!n) errx (1, "EOF while writing data");
293 sockerr (1, "send (req %d, ret " FMT_ss ")", size, n);
297 static void
298 #ifdef __GNUC__
299 __attribute__ ((format (printf, 2, 3)))
300 #endif
301 printd (int fd, const char *fmt, ...)
303 int size = 200, len;
304 va_list ap;
305 char *buf;
307 buf = malloc (size);
308 for (;;) {
309 if (!buf) err (errno, "malloc for temp buf (%d bytes) failed", size);
311 va_start (ap, fmt);
312 len = vsnprintf (buf, size, fmt, ap);
313 va_end (ap);
315 if (len > -1 && len < size) {
316 writedata (fd, buf, len);
317 break;
320 if (len > -1) {
321 size = len + 1;
323 else {
324 size *= 2;
326 buf = realloc (buf, size);
328 free (buf);
331 static void die (fz_error error)
333 fz_catch (error, "aborting");
334 if (state.xref)
335 pdf_free_xref (state.xref);
336 exit (1);
339 static void openxref (char *filename, char *password)
341 int i;
342 fz_error error;
344 for (i = 0; i < state.texcount; ++i) {
345 state.texowners[i].slice = NULL;
348 if (state.cache) {
349 fz_free_glyph_cache (state.cache);
352 state.cache = fz_new_glyph_cache ();
353 if (!state.cache) {
354 errx (1, "fz_newglyph_cache failed");
357 if (state.xref) {
358 if (state.xref->store) {
359 pdf_free_store (state.xref->store);
360 state.xref->store = NULL;
362 pdf_free_xref (state.xref);
363 state.xref = NULL;
366 if (state.pagedims) {
367 free (state.pagedims);
368 state.pagedims = NULL;
370 state.pagedimcount = 0;
372 error = pdf_open_xref (&state.xref, filename, password);
373 if (error) {
374 die (error);
377 error = pdf_load_page_tree (state.xref);
378 if (error) {
379 die (error);
382 state.pagecount = pdf_count_pages (state.xref);
385 static int readlen (int fd)
387 ssize_t n;
388 unsigned char p[4];
390 n = recv (fd, p, 4, 0);
391 if (n != 4) {
392 if (!n) errx (1, "EOF while reading length");
393 sockerr (1, "recv " FMT_ss, n);
396 return (p[0] << 24) | (p[1] << 16) | (p[2] << 8) | p[3];
399 static void unlinkpage (struct page *page)
401 int i;
403 for (i = 0; i < page->slicecount; ++i) {
404 struct slice *s = &page->slices[i];
405 if (s->texindex != -1) {
406 if (state.texowners[s->texindex].slice == s) {
407 state.texowners[s->texindex].slice = NULL;
408 ARSERT (state.texowners[s->texindex].w == s->w);
409 ARSERT (state.texowners[s->texindex].h >= s->h);
415 static void freepage (struct page *page)
417 fz_drop_pixmap (page->pixmap);
419 unlinkpage (page);
421 if (page->text) {
422 fz_free_text_span (page->text);
424 if (page->drawpage) {
425 pdf_free_page (page->drawpage);
428 free (page);
431 static void subdivide (struct page *p)
433 int i;
434 int h = p->pixmap->h;
435 int th = MIN (h, state.sliceheight);
437 for (i = 0; i < p->slicecount; ++i) {
438 struct slice *s = &p->slices[i];
439 s->texindex = -1;
440 s->h = MIN (th, h);
441 s->w = p->pixmap->w;
442 h -= th;
446 static int compatpdims (struct pagedim *p1, struct pagedim *p2)
448 return p1->rotate == p2->rotate
449 && !memcmp (&p1->bbox, &p2->bbox, sizeof (p1->bbox))
450 && !memcmp (&p1->ctm, &p2->ctm, sizeof (p1->ctm));
453 #ifdef __ALTIVEC__
454 #include <altivec.h>
456 static int cacheline32bytes;
458 static void __attribute__ ((constructor)) clcheck (void)
460 char **envp = environ;
461 unsigned long *auxv;
463 while (*envp++);
465 for (auxv = (unsigned long *) envp; *auxv != 0; auxv += 2) {
466 if (*auxv == 19) {
467 cacheline32bytes = auxv[1] == 32;
468 return;
473 static void __attribute__ ((optimize ("O"))) clearpixmap (fz_pixmap *pixmap)
475 if (cacheline32bytes) {
476 intptr_t a1, a2, diff;
477 size_t sizea, i, size = pixmap->w * pixmap->h * pixmap->n;
478 vector unsigned char v = vec_splat_u8 (-1);
479 vector unsigned char *p;
481 a1 = a2 = (intptr_t) pixmap->samples;
482 a2 = (a1 + 31) & ~31;
483 diff = a2 - a1;
484 sizea = size - diff;
485 p = (void *) a2;
487 while (a1 != a2) *(char *) a1++ = 0xff;
488 for (i = 0; i < (sizea & ~31); i += 32) {
489 __asm volatile ("dcbz %0, %1"::"b"(a2),"r"(i));
490 vec_st (v, i, p);
491 vec_st (v, i + 16, p);
493 while (i < sizea) *((char *) a1 + i++) = 0xff;
495 else fz_clear_pixmap_with_color (pixmap, 0xff);
497 #else
498 #define clearpixmap(p) fz_clear_pixmap_with_color (p, 0xff)
499 #endif
501 static void *render (int pageno, int pindex)
503 fz_error error;
504 int slicecount;
505 struct page *page = NULL;
506 double start, end;
507 pdf_page *drawpage;
508 fz_device *idev;
509 struct pagedim *pagedim;
511 start = now ();
512 printd (state.sock, "V rendering %d", pageno);
514 pagedim = &state.pagedims[pindex];
515 slicecount = (pagedim->bbox.y1 - pagedim->bbox.y0
516 + state.sliceheight - 1) / state.sliceheight;
517 slicecount += slicecount == 0;
519 if (state.pig) {
520 if (compatpdims (&state.pig->pagedim, pagedim)) {
521 page = state.pig;
522 if (page->text) {
523 fz_free_text_span (page->text);
524 page->text = NULL;
526 if (page->drawpage) {
527 pdf_free_page (page->drawpage);
528 page->drawpage = NULL;
531 else {
532 freepage (state.pig);
535 if (!page) {
536 page = calloc (sizeof (*page)
537 + (slicecount * sizeof (struct slice)), 1);
538 if (!page) {
539 err (1, "calloc page %d\n", pageno);
541 page->pixmap = fz_new_pixmap_with_rect (fz_device_rgb, pagedim->bbox);
544 page->slicecount = slicecount;
546 error = pdf_load_page (&drawpage, state.xref, pageno - 1);
547 if (error)
548 die (error);
550 clearpixmap (page->pixmap);
552 idev = fz_new_draw_device (state.cache, page->pixmap);
553 if (!idev)
554 die (fz_throw ("fz_newdrawdevice failed"));
555 error = pdf_run_page (state.xref, drawpage, idev, pagedim->ctm);
556 if (error)
557 die (fz_rethrow (error, "pdf_runpage failed"));
558 fz_free_device (idev);
560 page->drawpage = drawpage;
561 page->pagedim = *pagedim;
562 page->pageno = pageno;
563 subdivide (page);
564 end = now ();
566 pdf_age_store (state.xref->store, 3);
568 printd (state.sock, "V rendering %d took %f sec", pageno, end - start);
569 state.pig = NULL;
570 return page;
573 static void initpdims (void)
575 int pageno;
576 double start, end;
578 start = now ();
579 for (pageno = 0; pageno < state.pagecount; ++pageno) {
580 int rotate;
581 fz_rect box;
582 struct pagedim *p;
583 fz_obj *obj, *pageobj;
585 pageobj = state.xref->page_objs[pageno];
587 obj = fz_dict_gets (pageobj, "CropBox");
588 if (!fz_is_array (obj)) {
589 obj = fz_dict_gets (pageobj, "MediaBox");
590 if (!fz_is_array (obj)) {
591 die (fz_throw ("cannot find page bounds %d (%d Rd)",
592 fz_to_num (pageobj), fz_to_gen (pageobj)));
595 box = pdf_to_rect (obj);
597 obj = fz_dict_gets (pageobj, "Rotate");
598 if (fz_is_int (obj)) {
599 rotate = fz_to_int (obj);
601 else {
602 rotate = 0;
604 rotate += state.rotate;
606 p = &state.pagedims[state.pagedimcount - 1];
607 if ((state.pagedimcount == 0)
608 || (p->rotate != rotate || memcmp (&p->box, &box, sizeof (box)))) {
609 size_t size;
611 size = (state.pagedimcount + 1) * sizeof (*state.pagedims);
612 state.pagedims = realloc (state.pagedims, size);
613 if (!state.pagedims) {
614 err (1, "realloc pagedims to " FMT_s " (%d elems)",
615 size, state.pagedimcount + 1);
617 p = &state.pagedims[state.pagedimcount++];
618 p->rotate = rotate;
619 p->box = box;
620 p->pageno = pageno;
623 end = now ();
624 printd (state.sock, "T Processed %d pages in %f seconds",
625 state.pagecount, end - start);
628 static void layout (void)
630 int pindex;
631 fz_matrix ctm;
632 fz_rect box, box2;
633 double zoom, w, maxw = 0;
634 struct pagedim *p = state.pagedims;
636 if (state.proportional) {
637 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
638 box.x0 = MIN (p->box.x0, p->box.x1);
639 box.y0 = MIN (p->box.y0, p->box.y1);
640 box.x1 = MAX (p->box.x0, p->box.x1);
641 box.y1 = MAX (p->box.y0, p->box.y1);
643 ctm = fz_identity;
644 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
645 ctm = fz_concat (ctm, fz_rotate (p->rotate));
646 box2 = fz_transform_rect (ctm, box);
647 w = box2.x1 - box2.x0;
648 maxw = MAX (w, maxw);
652 p = state.pagedims;
653 for (pindex = 0; pindex < state.pagedimcount; ++pindex, ++p) {
654 box.x0 = MIN (p->box.x0, p->box.x1);
655 box.y0 = MIN (p->box.y0, p->box.y1);
656 box.x1 = MAX (p->box.x0, p->box.x1);
657 box.y1 = MAX (p->box.y0, p->box.y1);
659 ctm = fz_identity;
660 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
661 ctm = fz_concat (ctm, fz_rotate (p->rotate));
662 box2 = fz_transform_rect (ctm, box);
663 w = box2.x1 - box2.x0;
665 if (state.proportional) {
666 double scale = w / maxw;
667 zoom = (state.w / w) * scale;
669 else {
670 zoom = state.w / w;
672 ctm = fz_identity;
673 ctm = fz_concat (ctm, fz_translate (0, -box.y1));
674 ctm = fz_concat (ctm, fz_scale (zoom, -zoom));
675 memcpy (&p->ctm1, &ctm, sizeof (ctm));
676 ctm = fz_concat (ctm, fz_rotate (p->rotate));
677 p->bbox = fz_round_rect (fz_transform_rect (ctm, box));
678 p->left = state.proportional ? ((maxw - w) * zoom) / 2.0 : 0;
679 memcpy (&p->ctm, &ctm, sizeof (ctm));
682 while (p-- != state.pagedims) {
683 printd (state.sock, "l %d %d %d %d",
684 p->pageno, p->bbox.x1 - p->bbox.x0, p->bbox.y1 - p->bbox.y0,
685 p->left);
689 static void recurse_outline (pdf_outline *outline, int level)
691 while (outline) {
692 int i;
693 fz_obj *obj;
694 int pageno = -1;
695 int top = 0, h = 0;
697 if (!outline->link) goto next;
699 obj = outline->link->dest;
700 if (fz_is_indirect (obj)) {
701 obj = fz_resolve_indirect (obj);
703 if (fz_is_array (obj)) {
704 fz_obj *obj2;
706 obj2 = fz_array_get (obj, 0);
707 if (fz_is_int (obj2)) {
708 pageno = fz_to_int (obj2);
710 else {
711 pageno = pdf_find_page_number (state.xref, obj2);
714 if (fz_array_len (obj) > 3) {
715 fz_point p;
716 fz_obj *xo, *yo;
718 xo = fz_array_get (obj, 2);
719 yo = fz_array_get (obj, 3);
720 if (!fz_is_null (xo) && !fz_is_null (yo)) {
721 struct pagedim *pagedim = state.pagedims;
723 for (i = 0; i < state.pagedimcount; ++i) {
724 if (state.pagedims[i].pageno > pageno)
725 break;
726 pagedim = &state.pagedims[i];
728 p.x = fz_to_int (xo);
729 p.y = fz_to_int (yo);
730 p = fz_transform_point (pagedim->ctm, p);
731 h = pagedim->bbox.y1 - pagedim->bbox.y0;
732 top = p.y;
736 else {
737 pageno = pdf_find_page_number (state.xref, outline->link->dest);
740 lprintf ("%*c%s %d\n", level, ' ', outline->title, pageno);
741 if (pageno > 0) {
742 printd (state.sock, "o %d %d %d %d %s",
743 level, pageno, top, h, outline->title);
745 next:
746 if (outline->child) {
747 recurse_outline (outline->child, level + 1);
749 outline = outline->next;
753 static void process_outline (void)
755 pdf_outline *outline;
757 if (!state.needoutline) return;
759 state.needoutline = 0;
760 outline = pdf_load_outline (state.xref);
761 if (outline) {
762 recurse_outline (outline, 0);
763 pdf_free_outline (outline);
767 static int comparespans (const void *l, const void *r)
769 fz_text_span const *const*ls = l;
770 fz_text_span const *const*rs = r;
771 return (*ls)->text->bbox.y0 - (*rs)->text->bbox.y0;
774 /* wishful thinking function */
775 static void search (regex_t *re, int pageno, int y, int forward)
777 int i, j;
778 int ret;
779 char *p;
780 char buf[256];
781 fz_matrix ctm;
782 fz_error error;
783 fz_device *tdev;
784 pdf_page *drawpage;
785 fz_text_span *text, *span, **pspan;
786 struct pagedim *pdim, *pdimprev;
787 int stop = 0;
788 int niters = 0;
789 int nspans;
790 double start, end;
792 start = now ();
793 while (pageno >= 0 && pageno < state.pagecount && !stop) {
794 if (niters++ == 5) {
795 pdf_age_store (state.xref->store, 3);
796 niters = 0;
797 if (hasdata (state.sock)) {
798 printd (state.sock, "T attention requested aborting search at %d",
799 pageno);
800 stop = 1;
802 else {
803 printd (state.sock, "T searching in page %d", pageno);
806 pdimprev = NULL;
807 for (i = 0; i < state.pagedimcount; ++i) {
808 pdim = &state.pagedims[i];
809 if (pdim->pageno == pageno) {
810 goto found;
812 if (pdim->pageno > pageno) {
813 pdim = pdimprev;
814 goto found;
816 pdimprev = pdim;
818 pdim = pdimprev;
819 found:
821 error = pdf_load_page (&drawpage, state.xref, pageno);
822 if (error)
823 die (error);
825 ctm = fz_rotate (pdim->rotate);
827 text = fz_new_text_span ();
828 tdev = fz_new_text_device (text);
829 error = pdf_run_page (state.xref, drawpage, tdev, pdim->ctm1);
830 if (error) die (error);
831 fz_free_device (tdev);
833 nspans = 0;
834 for (span = text; span; span = span->next) {
835 nspans++;
837 pspan = malloc (sizeof (void *) * nspans);
838 if (!pspan) {
839 err (1, "malloc span pointers " FMT_s, sizeof (void *) * nspans);
841 for (i = 0, span = text; span; span = span->next, ++i) {
842 pspan[i] = span;
844 qsort (pspan, nspans, sizeof (fz_text_span *), comparespans);
846 j = forward ? 0 : nspans - 1;
847 while (nspans--) {
848 regmatch_t rm;
850 span = pspan[j];
851 j += forward ? 1 : -1;
852 p = buf;
853 for (i = 0; i < MIN (span->len, (int) sizeof (buf) - 1); ++i) {
854 if (forward) {
855 if (span->text[i].bbox.y0 < y + 1) {
856 continue;
859 else {
860 if (span->text[i].bbox.y0 > y - 1) {
861 continue;
864 if (span->text[i].c < 256) {
865 *p++ = span->text[i].c;
867 else {
868 *p++ = '?';
871 if (p == buf) {
872 continue;
874 *p++ = 0;
876 ret = regexec (re, buf, 1, &rm, 0);
877 if (ret) {
878 if (ret != REG_NOMATCH) {
879 size_t size;
880 char errbuf[80];
881 size = regerror (ret, re, errbuf, sizeof (errbuf));
882 printd (state.sock,
883 "T regexec error `%.*s'",
884 (int) size, errbuf);
885 fz_free_text_span (text);
886 pdf_free_page (drawpage);
887 free (pspan);
888 return;
891 else {
892 int xoff, yoff;
893 fz_bbox *sb, *eb;
894 fz_point p1, p2, p3, p4;
896 xoff = pdim->left - pdim->bbox.x0;
897 yoff = -pdim->bbox.y0;
899 sb = &span->text[rm.rm_so].bbox;
900 eb = &span->text[rm.rm_eo - 1].bbox;
902 p1.x = sb->x0;
903 p1.y = sb->y0;
904 p2.x = eb->x1;
905 p2.y = sb->y0;
906 p3.x = eb->x1;
907 p3.y = eb->y1;
908 p4.x = sb->x0;
909 p4.y = eb->y1;
911 p1 = fz_transform_point (ctm, p1);
912 p2 = fz_transform_point (ctm, p2);
913 p3 = fz_transform_point (ctm, p3);
914 p4 = fz_transform_point (ctm, p4);
916 if (!stop) {
917 printd (state.sock, "F %d %d %f %f %f %f %f %f %f %f",
918 pageno, 1,
919 p1.x + xoff, p1.y + yoff,
920 p2.x + xoff, p2.y + yoff,
921 p3.x + xoff, p3.y + yoff,
922 p4.x + xoff, p4.y + yoff);
924 printd (state.sock, "T found at %d `%.*s' in %f sec",
925 pageno, rm.rm_eo - rm.rm_so, &buf[rm.rm_so],
926 now () - start);
928 else {
929 printd (state.sock, "R %d %d %f %f %f %f %f %f %f %f",
930 pageno, 2,
931 p1.x + xoff, p1.y + yoff,
932 p2.x + xoff, p2.y + yoff,
933 p3.x + xoff, p3.y + yoff,
934 p4.x + xoff, p4.y + yoff);
936 stop = 1;
939 if (forward) {
940 pageno += 1;
941 y = 0;
943 else {
944 pageno -= 1;
945 y = INT_MAX;
947 fz_free_text_span (text);
948 pdf_free_page (drawpage);
949 free (pspan);
951 end = now ();
952 if (!stop) {
953 printd (state.sock, "T no matches %f sec", end - start);
955 printd (state.sock, "D");
958 static
959 #ifdef _WIN32
960 DWORD _stdcall
961 #else
962 void *
963 #endif
964 mainloop (void *unused)
966 char *p = NULL;
967 int len, ret, oldlen = 0;
969 for (;;) {
970 len = readlen (state.sock);
971 if (len == 0) {
972 errx (1, "readlen returned 0");
975 if (oldlen < len + 1) {
976 p = realloc (p, len + 1);
977 if (!p) {
978 err (1, "realloc %d failed", len + 1);
980 oldlen = len + 1;
982 readdata (state.sock, p, len);
983 p[len] = 0;
985 if (!strncmp ("open", p, 4)) {
986 fz_obj *obj;
987 size_t filenamelen;
988 char *password;
989 char *filename = p + 5;
991 filenamelen = strlen (filename);
992 password = filename + filenamelen + 1;
994 openxref (filename, password);
995 initpdims ();
997 obj = fz_dict_gets (state.xref->trailer, "Info");
998 if (obj) {
999 char *s;
1001 obj = fz_dict_gets (obj, "Title");
1002 s = pdf_to_utf8 (obj);
1003 if (*s) {
1004 printd (state.sock, "t %s", s);
1006 fz_free (s);
1009 state.needoutline = 1;
1011 else if (!strncmp ("free", p, 4)) {
1012 void *ptr;
1014 ret = sscanf (p + 4, " %p", &ptr);
1015 if (ret != 1) {
1016 errx (1, "malformed free `%.*s' ret=%d", len, p, ret);
1018 unlinkpage (ptr);
1019 state.pig = ptr;
1021 else if (!strncmp ("search", p, 6)) {
1022 int icase, pageno, y, ret, len2, forward;
1023 char *pattern;
1024 regex_t re;
1026 ret = sscanf (p + 6, " %d %d %d %d,%n",
1027 &icase, &pageno, &y, &forward, &len2);
1028 if (ret != 4) {
1029 errx (1, "malformed search `%s' ret=%d", p, ret);
1032 pattern = p + 6 + len2;
1033 ret = regcomp (&re, pattern,
1034 REG_EXTENDED | (icase ? REG_ICASE : 0));
1035 if (ret) {
1036 char errbuf[80];
1037 size_t size;
1039 size = regerror (ret, &re, errbuf, sizeof (errbuf));
1040 printd (state.sock, "T regcomp failed `%.*s'",
1041 (int) size, errbuf);
1043 else {
1044 search (&re, pageno, y, forward);
1045 regfree (&re);
1048 else if (!strncmp ("geometry", p, 8)) {
1049 int w, h;
1051 printd (state.sock, "c");
1052 ret = sscanf (p + 8, " %d %d", &w, &h);
1053 if (ret != 2) {
1054 errx (1, "malformed geometry `%.*s' ret=%d", len, p, ret);
1057 lock ("geometry");
1058 state.h = h;
1059 if (w != state.w) {
1060 int i;
1061 state.w = w;
1062 for (i = 0; i < state.texcount; ++i) {
1063 state.texowners[i].slice = NULL;
1066 layout ();
1067 process_outline ();
1068 unlock ("geometry");
1069 printd (state.sock, "C %d", state.pagecount);
1071 else if (!strncmp ("reinit", p, 6)) {
1072 float rotate;
1073 int proportional;
1075 printd (state.sock, "c");
1076 ret = sscanf (p + 6, " %f %d", &rotate, &proportional);
1077 if (ret != 2) {
1078 errx (1, "bad rotate line `%.*s' ret=%d", len, p, ret);
1080 lock ("reinit");
1081 state.rotate = rotate;
1082 state.proportional = proportional;
1083 state.pagedimcount = 0;
1084 free (state.pagedims);
1085 state.pagedims = NULL;
1086 initpdims ();
1087 layout ();
1088 process_outline ();
1089 if (state.pig) {
1090 freepage (state.pig);
1091 state.pig = NULL;
1093 unlock ("reinit");
1094 printd (state.sock, "C %d", state.pagecount);
1096 else if (!strncmp ("render", p, 6)) {
1097 int pageno, pindex, w, h, ret;
1098 struct page *page;
1100 ret = sscanf (p + 6, " %d %d %d %d", &pageno, &pindex, &w, &h);
1101 if (ret != 4) {
1102 errx (1, "bad render line `%.*s' ret=%d", len, p, ret);
1105 lock ("render");
1106 page = render (pageno, pindex);
1107 unlock ("render");
1109 printd (state.sock, "r %d %d %d %d %d %d %p",
1110 pageno,
1111 state.w,
1112 state.h,
1113 state.rotate,
1114 state.proportional,
1115 w * h * 4,
1116 page);
1118 else if (!strncmp ("interrupt", p, 9)) {
1119 printd (state.sock, "V interrupted");
1121 else {
1122 errx (1, "unknown command %.*s", len, p);
1125 return 0;
1128 static void showsel (struct page *page, int oy)
1130 int ox;
1131 fz_bbox bbox;
1132 fz_text_span *span;
1133 struct mark first, last;
1135 first = page->fmark;
1136 last = page->lmark;
1138 if (!first.span || !last.span) return;
1140 glEnable (GL_BLEND);
1141 glBlendFunc (GL_SRC_ALPHA, GL_SRC_ALPHA);
1142 glColor4f (0.5f, 0.5f, 0.0f, 0.6f);
1144 ox = -page->pixmap->x + page->pagedim.left;
1145 oy = -page->pixmap->y + oy;
1146 for (span = first.span; span; span = span->next) {
1147 int i, j, k;
1149 bbox.x0 = bbox.y0 = bbox.x1 = bbox.y1 = 0;
1151 j = 0;
1152 k = span->len - 1;
1154 if (span == page->fmark.span && span == page->lmark.span) {
1155 j = MIN (first.i, last.i);
1156 k = MAX (first.i, last.i);
1158 else if (span == first.span) {
1159 j = first.i;
1161 else if (span == last.span) {
1162 k = last.i;
1165 for (i = j; i <= k; ++i) {
1166 bbox = fz_union_bbox (bbox, span->text[i].bbox);
1168 lprintf ("%d %d %d %d oy=%d ox=%d\n",
1169 bbox.x0,
1170 bbox.y0,
1171 bbox.x1,
1172 bbox.y1,
1173 oy, ox);
1175 glRecti (bbox.x0 + ox, bbox.y0 + oy, bbox.x1 + ox, bbox.y1 + oy);
1177 if (span == last.span) break;
1179 glDisable (GL_BLEND);
1182 static void highlightlinks (struct page *page, int yoff)
1184 pdf_link *link;
1185 int xoff;
1187 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1188 glEnable (GL_LINE_STIPPLE);
1189 glLineStipple (0.5, 0xcccc);
1191 xoff = page->pagedim.left - page->pixmap->x;
1192 yoff -= page->pixmap->y;
1194 glBegin (GL_QUADS);
1195 for (link = page->drawpage->links; link; link = link->next) {
1196 fz_point p1, p2, p3, p4;
1197 fz_matrix ctm = page->pagedim.ctm;
1199 p1.x = link->rect.x0;
1200 p1.y = link->rect.y0;
1202 p2.x = link->rect.x1;
1203 p2.y = link->rect.y0;
1205 p3.x = link->rect.x1;
1206 p3.y = link->rect.y1;
1208 p4.x = link->rect.x0;
1209 p4.y = link->rect.y1;
1211 p1 = fz_transform_point (ctm, p1);
1212 p2 = fz_transform_point (ctm, p2);
1213 p3 = fz_transform_point (ctm, p3);
1214 p4 = fz_transform_point (ctm, p4);
1216 switch (link->kind) {
1217 case PDF_LINK_GOTO: glColor3ub (255, 0, 0); break;
1218 case PDF_LINK_URI: glColor3ub (0, 0, 255); break;
1219 default: glColor3ub (0, 0, 0); break;
1222 glVertex2f (p1.x + xoff, p1.y + yoff);
1223 glVertex2f (p2.x + xoff, p2.y + yoff);
1224 glVertex2f (p3.x + xoff, p3.y + yoff);
1225 glVertex2f (p4.x + xoff, p4.y + yoff);
1227 glEnd ();
1229 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1230 glDisable (GL_LINE_STIPPLE);
1233 static void upload2 (struct page *page, int slicenum, const char *cap)
1235 int i;
1236 int w;
1237 double start, end;
1238 struct slice *slice = &page->slices[slicenum];
1240 w = page->pixmap->w;
1242 ARSERT (w == slice->w);
1243 if (slice->texindex != -1
1244 && state.texowners[slice->texindex].slice == slice) {
1245 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1247 else {
1248 int subimage = 0;
1249 int index = (state.texindex++ % state.texcount);
1250 size_t offset = 0;
1252 for (i = 0; i < slicenum; ++i) {
1253 offset += w * page->slices[i].h * 4;
1256 if (state.texowners[index].w == slice->w) {
1257 if (state.texowners[index].h >= slice->h ) {
1258 subimage = 1;
1260 else {
1261 state.texowners[index].h = slice->h;
1264 else {
1265 state.texowners[index].h = slice->h;
1268 state.texowners[index].slice = slice;
1269 state.texowners[index].w = slice->w;
1270 slice->texindex = index;
1272 glBindTexture (GL_TEXTURE_RECTANGLE_ARB, state.texids[slice->texindex]);
1273 start = now ();
1274 if (subimage) {
1276 GLenum err = glGetError ();
1277 if (err != GL_NO_ERROR) {
1278 printf ("\033[0;31mERROR1 %d %d %#x\033[0m\n", w, slice->h, err);
1279 abort ();
1282 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB,
1287 slice->h,
1288 state.texform,
1289 state.texty,
1290 page->pixmap->samples + offset
1293 GLenum err = glGetError ();
1294 if (err != GL_NO_ERROR) {
1295 printf ("\033[0;31mERROR %d %d %#x\033[0m\n", w, slice->h, err);
1296 abort ();
1300 else {
1301 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB,
1303 GL_RGBA8,
1305 slice->h,
1307 state.texform,
1308 state.texty,
1309 page->pixmap->samples + offset
1313 end = now ();
1314 (void) start;
1315 (void) end;
1316 lprintf ("%s[%d] slice=%d(%d,%d) texid=%d %f sec\n",
1317 subimage ? "sub" : "img",
1318 page->pageno, slicenum,
1319 slice->w, slice->h,
1320 state.texids[slice->texindex],
1321 end - start);
1325 CAMLprim value ml_draw (value args_v, value ptr_v)
1327 CAMLparam2 (args_v, ptr_v);
1328 int dispy = Int_val (Field (args_v, 0));
1329 int w = Int_val (Field (args_v, 1));
1330 int h = Int_val (Field (args_v, 2));
1331 int py = Int_val (Field (args_v, 3));
1332 int hlinks = Bool_val (Field (args_v, 4));
1333 char *s = String_val (ptr_v);
1334 int ret;
1335 void *ptr;
1336 struct page *page;
1337 int slicenum = 0;
1338 int yoff = dispy - py;
1340 ret = sscanf (s, "%p", &ptr);
1341 if (ret != 1) {
1342 errx (1, "cannot parse pointer `%s'", s);
1344 page = ptr;
1346 w = page->pixmap->w;
1348 ARSERT (h >= 0 && "ml_draw wrong h");
1349 ARSERT (py <= page->pixmap->h && "ml_draw wrong py");
1351 glEnable (GL_TEXTURE_RECTANGLE_ARB);
1353 for (slicenum = 0; slicenum < page->slicecount; ++slicenum) {
1354 struct slice *slice = &page->slices[slicenum];
1355 if (slice->h > py) {
1356 break;
1358 py -= slice->h;
1361 h = MIN (state.h, h);
1362 while (h) {
1363 int th, left;
1364 struct slice *slice = &page->slices[slicenum];
1366 ARSERT (slicenum < page->slicecount && "ml_draw wrong slicenum");
1368 th = MIN (h, slice->h - py);
1369 upload2 (page, slicenum, "upload");
1371 left = page->pagedim.left;
1372 glBegin (GL_QUADS);
1374 glTexCoord2i (0, py);
1375 glVertex2i (left, dispy);
1377 glTexCoord2i (w, py);
1378 glVertex2i (left+w, dispy);
1380 glTexCoord2i (w, py+th);
1381 glVertex2i (left+w, dispy + th);
1383 glTexCoord2i (0, py+th);
1384 glVertex2i (left, dispy + th);
1386 glEnd ();
1388 h -= th;
1389 py = 0;
1390 dispy += th;
1391 slicenum += 1;
1394 showsel (page, yoff);
1395 if (hlinks) highlightlinks (page, yoff);
1396 glDisable (GL_TEXTURE_RECTANGLE_ARB);
1398 CAMLreturn (Val_unit);
1401 static pdf_link *getlink (struct page *page, int x, int y)
1403 fz_point p;
1404 fz_matrix ctm;
1405 pdf_link *link;
1407 p.x = x + page->pixmap->x;
1408 p.y = y + page->pixmap->y;
1410 ctm = fz_invert_matrix (page->pagedim.ctm);
1411 p = fz_transform_point (ctm, p);
1413 for (link = page->drawpage->links; link; link = link->next) {
1414 if (p.x >= link->rect.x0 && p.x <= link->rect.x1) {
1415 if (p.y >= link->rect.y0 && p.y <= link->rect.y1) {
1416 return link;
1420 return NULL;
1423 static void ensuretext (struct page *page)
1425 if (!page->text) {
1426 fz_error error;
1427 fz_device *tdev;
1429 page->text = fz_new_text_span ();
1430 tdev = fz_new_text_device (page->text);
1431 error = pdf_run_page (state.xref, page->drawpage, tdev,
1432 page->pagedim.ctm);
1433 if (error) die (error);
1434 fz_free_device (tdev);
1438 CAMLprim value ml_whatsunder (value ptr_v, value x_v, value y_v)
1440 CAMLparam3 (ptr_v, x_v, y_v);
1441 CAMLlocal3 (ret_v, tup_v, str_v);
1442 pdf_link *link;
1443 struct page *page;
1444 char *s = String_val (ptr_v);
1446 ret_v = Val_int (0);
1447 if (trylock ("ml_whatsunder")) {
1448 goto done;
1451 page = parse_pointer ("ml_whatsunder", s);
1452 link = getlink (page, Int_val (x_v), Int_val (y_v));
1453 if (link) {
1454 switch (link->kind) {
1455 case PDF_LINK_GOTO:
1457 int pageno;
1458 fz_point p;
1459 fz_obj *obj;
1461 pageno = -1;
1462 p.x = 0;
1463 p.y = 0;
1465 if (fz_is_array (link->dest)) {
1466 obj = fz_array_get (link->dest, 0);
1467 if (fz_is_indirect (obj)) {
1468 pageno = pdf_find_page_number (state.xref, obj);
1470 else if (fz_is_int (obj)) {
1471 pageno = fz_to_int (obj);
1474 if (fz_array_len (link->dest) > 3) {
1475 fz_obj *xo, *yo;
1477 xo = fz_array_get (link->dest, 2);
1478 yo = fz_array_get (link->dest, 3);
1479 if (!fz_is_null (xo) && !fz_is_null (yo)) {
1480 p.x = fz_to_int (xo);
1481 p.y = fz_to_int (yo);
1482 p = fz_transform_point (page->pagedim.ctm, p);
1486 else {
1487 pageno = pdf_find_page_number (state.xref, link->dest);
1489 tup_v = caml_alloc_tuple (2);
1490 ret_v = caml_alloc_small (1, 1);
1491 Field (tup_v, 0) = Val_int (pageno);
1492 Field (tup_v, 1) = Val_int (p.y);
1493 Field (ret_v, 0) = tup_v;
1495 break;
1497 case PDF_LINK_URI:
1498 str_v = caml_copy_string (fz_to_str_buf (link->dest));
1499 ret_v = caml_alloc_small (1, 0);
1500 Field (ret_v, 0) = str_v;
1501 break;
1503 default:
1504 printd (state.sock, "T unhandled link kind %d", link->kind);
1505 break;
1508 else {
1509 int i, x, y;
1510 fz_text_span *span;
1512 ensuretext (page);
1513 x = Int_val (x_v) + page->pixmap->x;
1514 y = Int_val (y_v) + page->pixmap->y;
1516 for (span = page->text; span; span = span->next) {
1517 for (i = 0; i < span->len; ++i) {
1518 fz_bbox *b;
1519 b = &span->text[i].bbox;
1520 if ((x >= b->x0 && x <= b->x1 && y >= b->y0 && y <= b->y1)) {
1521 const char *n2 =
1522 span->font && span->font->name
1523 ? span->font->name
1524 : "Span has no font name"
1526 #ifdef FT_FREETYPE_H
1527 FT_FaceRec *face = span->font->ft_face;
1528 if (face && face->family_name) {
1529 char *s;
1530 char *n1 = face->family_name;
1531 size_t l1 = strlen (n1);
1532 size_t l2 = strlen (n2);
1534 if (l1 != l2 || memcmp (n1, n2, l1)) {
1535 s = malloc (l1 + l2 + 2);
1536 if (s) {
1537 memcpy (s, n2, l2);
1538 s[l2] = '=';
1539 memcpy (s + l2 + 1, n1, l1 + 1);
1540 str_v = caml_copy_string (s);
1541 free (s);
1545 if (str_v == 0) {
1546 str_v = caml_copy_string (n2);
1548 #else
1549 str_v = caml_copy_string (n2);
1550 #endif
1551 ret_v = caml_alloc_small (1, 2);
1552 Field (ret_v, 0) = str_v;
1553 goto unlock;
1558 unlock:
1559 unlock ("ml_whatsunder");
1561 done:
1562 CAMLreturn (ret_v);
1565 CAMLprim value ml_seltext (value ptr_v, value rect_v, value oy_v)
1567 CAMLparam4 (ptr_v, rect_v, oy_v, rect_v);
1568 fz_bbox *b;
1569 struct page *page;
1570 fz_text_span *span;
1571 struct mark first, last;
1572 int i, x0, x1, y0, y1, oy, left;
1573 char *s = String_val (ptr_v);
1575 if (trylock ("ml_seltext")) {
1576 goto done;
1579 page = parse_pointer ("ml_seltext", s);
1580 ensuretext (page);
1582 oy = Int_val (oy_v);
1583 x0 = Int_val (Field (rect_v, 0));
1584 y0 = Int_val (Field (rect_v, 1));
1585 x1 = Int_val (Field (rect_v, 2));
1586 y1 = Int_val (Field (rect_v, 3));
1588 left = page->pagedim.left;
1589 if (0) {
1590 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1591 glColor3ub (128, 128, 128);
1592 glRecti (x0, y0, x1, y1);
1593 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1596 x0 += page->pixmap->x - left;
1597 y0 += page->pixmap->y - oy;
1598 x1 += page->pixmap->x - left;
1599 y1 += page->pixmap->y - oy;
1601 first.span = NULL;
1602 last.span = NULL;
1604 last.i = first.i = 0;
1605 first.span = page->text;
1606 for (span = page->text; span; span = span->next) {
1607 for (i = 0; i < span->len; ++i) {
1608 b = &span->text[i].bbox;
1609 int selected = 0;
1611 if (x0 >= b->x0 && x0 <= b->x1 && y0 >= b->y0 && y0 <= b->y1) {
1612 first.i = i;
1613 first.span = span;
1614 selected = 1;
1616 if (x1 >= b->x0 && x1 <= b->x1 && y1 >= b->y0 && y1 <= b->y1) {
1617 last.i = i;
1618 last.span = span;
1619 selected = 1;
1621 if (0 && selected) {
1622 glPolygonMode (GL_FRONT_AND_BACK, GL_LINE);
1623 glColor3ub (128, 128, 128);
1624 glRecti (b->x0+left, b->y0, b->x1+left, b->y1);
1625 glPolygonMode (GL_FRONT_AND_BACK, GL_FILL);
1630 if (y1 < y0 || x1 < x0) {
1631 int swap = 0;
1633 if (first.span == last.span) {
1634 swap = 1;
1636 else {
1637 if (y1 < y0) {
1638 for (span = first.span; span && span != last.span;
1639 span = span->next) {
1640 if (span->eol) {
1641 swap = 1;
1642 break;
1648 if (swap) {
1649 i = first.i;
1650 span = first.span;
1651 first.i = last.i;
1652 first.span = last.span;
1653 last.i = i;
1654 last.span = span;
1658 page->fmark = first;
1659 page->lmark = last;
1661 unlock ("ml_seltext");
1663 done:
1664 CAMLreturn (Val_unit);
1667 static int pipespan (FILE *f, fz_text_span *span, int a, int b)
1669 char buf[4];
1670 int i, len, ret;
1672 for (i = a; i <= b; ++i) {
1673 len = runetochar (buf, &span->text[i].c);
1674 ret = fwrite (buf, len, 1, f);
1676 if (ret != 1) {
1677 printd (state.sock, "T failed to write %d bytes ret=%d: %s",
1678 len, ret, strerror (errno));
1679 return -1;
1682 return 0;
1685 CAMLprim value ml_copysel (value ptr_v)
1687 CAMLparam1 (ptr_v);
1688 FILE *f;
1689 struct page *page;
1690 char *s = String_val (ptr_v);
1692 if (trylock ("ml_copysel")) {
1693 goto done;
1696 if (!*s) {
1697 close:
1698 #ifdef USE_XSEL
1699 if (state.xselpipe) {
1700 int ret = pclose (state.xselpipe);
1701 if (ret) {
1702 printd (state.sock, "T failed to close xsel pipe `%s'",
1703 strerror (errno));
1705 state.xselpipe = NULL;
1707 #else
1708 printf ("========================================\n");
1709 #endif
1711 else {
1712 fz_text_span *span;
1714 page = parse_pointer ("ml_sopysel", s);
1716 if (!page->fmark.span || !page->lmark.span) {
1717 printd (state.sock, "T nothing to copy");
1718 goto unlock;
1721 f = stdout;
1722 #ifdef USE_XSEL
1723 if (!state.xselpipe) {
1724 state.xselpipe = popen ("xsel -i", "w");
1725 if (!state.xselpipe) {
1726 printd (state.sock, "T failed to open xsel pipe `%s'",
1727 strerror (errno));
1729 else {
1730 f = state.xselpipe;
1733 else {
1734 f = state.xselpipe;
1736 #endif
1738 for (span = page->fmark.span;
1739 span && span != page->lmark.span->next;
1740 span = span->next) {
1741 int a = span == page->fmark.span ? page->fmark.i : 0;
1742 int b = span == page->lmark.span ? page->lmark.i : span->len - 1;
1743 if (pipespan (f, span, a, b)) {
1744 goto close;
1746 if (span->eol) {
1747 if (putc ('\n', f) == EOF) {
1748 printd (state.sock, "T failed break line on xsel pipe `%s'",
1749 strerror (errno));
1750 goto close;
1754 page->lmark.span = NULL;
1755 page->fmark.span = NULL;
1758 unlock:
1759 unlock ("ml_copysel");
1761 done:
1762 CAMLreturn (Val_unit);
1765 CAMLprim value ml_getpdimrect (value pagedimno_v)
1767 CAMLparam1 (pagedimno_v);
1768 CAMLlocal1 (ret_v);
1769 int pagedimno = Int_val (pagedimno_v);
1770 fz_rect box;
1772 ret_v = caml_alloc_small (4 * Double_wosize, Double_array_tag);
1773 if (trylock ("ml_getpdimrect")) {
1774 box = fz_empty_rect;
1776 else {
1777 box = state.pagedims[pagedimno].box;
1778 unlock ("ml_getpdimrect");
1781 Store_double_field (ret_v, 0, box.x0);
1782 Store_double_field (ret_v, 1, box.x1);
1783 Store_double_field (ret_v, 2, box.y0);
1784 Store_double_field (ret_v, 3, box.y1);
1786 CAMLreturn (ret_v);
1789 CAMLprim value ml_zoom_for_height (value winw_v, value winh_v, value dw_v)
1791 CAMLparam3 (winw_v, winh_v, dw_v);
1792 CAMLlocal1 (ret_v);
1793 int i;
1794 double zoom = 1.0;
1795 double maxw = 0.0, maxh = 0.0;
1796 struct pagedim *p;
1797 double winw = Int_val (winw_v);
1798 double winh = Int_val (winh_v);
1799 double dw = Int_val (dw_v);
1800 double pw = 1.0, ph = 1.0, num, den;
1802 if (trylock ("ml_zoom_for_height")) {
1803 goto done;
1806 if (state.proportional) {
1807 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
1808 double x0, x1, w;
1810 x0 = MIN (p->box.x0, p->box.x1);
1811 x1 = MAX (p->box.x0, p->box.x1);
1813 w = x1 - x0;
1814 maxw = MAX (w, maxw);
1818 for (i = 0, p = state.pagedims; i < state.pagedimcount; ++i, ++p) {
1819 double x0, x1, y0, y1, w, h, scaledh, scale;
1821 x0 = MIN (p->box.x0, p->box.x1);
1822 y0 = MIN (p->box.y0, p->box.y1);
1823 x1 = MAX (p->box.x0, p->box.x1);
1824 y1 = MAX (p->box.y0, p->box.y1);
1826 w = x1 - x0;
1827 h = y1 - y0;
1829 if (state.proportional) {
1830 scale = w / maxw;
1831 scaledh = h * scale;
1833 else {
1834 scale = 1.0;
1835 scaledh = h;
1838 if (scaledh > maxh) {
1839 maxh = scaledh;
1840 ph = scaledh;
1841 pw = w * scale;
1845 num = (winh * pw) + (ph * dw);
1846 den = ph * winw;
1847 zoom = num / den;
1849 unlock ("ml_zoom_for_height");
1850 done:
1851 ret_v = caml_copy_double (zoom);
1852 CAMLreturn (ret_v);
1855 CAMLprim value ml_init (value sock_v, value params_v)
1857 #ifndef _WIN32
1858 int ret;
1859 #endif
1860 CAMLparam2 (sock_v, params_v);
1862 state.rotate = Int_val (Field (params_v, 0));
1863 state.proportional = Bool_val (Field (params_v, 1));
1864 state.texcount = Int_val (Field (params_v, 2));
1865 state.sliceheight = Int_val (Field (params_v, 3));
1866 state.texform = GL_RGBA;
1867 state.texty = GL_UNSIGNED_BYTE;
1869 fz_accelerate ();
1870 state.texids = calloc (state.texcount * sizeof (*state.texids), 1);
1871 if (!state.texids) {
1872 err (1, "calloc texids " FMT_s,
1873 state.texcount * sizeof (*state.texids));
1876 state.texowners = calloc (state.texcount * sizeof (*state.texowners), 1);
1877 if (!state.texowners) {
1878 err (1, "calloc texowners " FMT_s,
1879 state.texcount * sizeof (*state.texowners));
1882 glGenTextures (state.texcount, state.texids);
1884 #ifdef _WIN32
1885 state.sock = Socket_val (sock_v);
1886 #else
1887 state.sock = Int_val (sock_v);
1888 #endif
1890 #ifdef _WIN32
1891 InitializeCriticalSection (&critsec);
1892 state.thread = CreateThread (NULL, 0, mainloop, NULL, 0, NULL);
1893 if (state.thread == INVALID_HANDLE_VALUE) {
1894 errx (1, "CreateThread failed: %lx", GetLastError ());
1896 #else
1897 ret = pthread_create (&state.thread, NULL, mainloop, NULL);
1898 if (ret) {
1899 errx (1, "pthread_create: %s", strerror (ret));
1901 #endif
1903 CAMLreturn (Val_unit);