1 /* lot's of code c&p-ed directly from mupdf */
16 /* fugly as hell and GCC specific but... */
18 #define GL_GLEXT_PROTOTYPES
24 #include <caml/fail.h>
25 #include <caml/alloc.h>
26 #include <caml/memory.h>
27 #include <caml/unixsupport.h>
33 #define lprintf printf
38 #define ARSERT(cond) for (;;) { \
40 errx (1, "%s:%d " #cond, __FILE__, __LINE__); \
56 struct pagedim
*pagedim
;
58 struct slice slices
[];
74 struct pagedim
*pagedims
;
99 static pthread_mutex_t mutex
= PTHREAD_MUTEX_INITIALIZER
;
101 static void lock (const char *cap
)
103 int ret
= pthread_mutex_lock (&mutex
);
105 errx (1, "%s: pthread_mutex_lock: %s", cap
, strerror (ret
));
109 static void unlock (const char *cap
)
111 int ret
= pthread_mutex_unlock (&mutex
);
113 errx (1, "%s: pthread_mutex_unlock: %s", cap
, strerror (ret
));
117 static int trylock (const char *cap
)
119 int ret
= pthread_mutex_trylock (&mutex
);
121 if (ret
&& ret
!= EBUSY
) {
122 errx (1, "%s: pthread_mutex_trylock: %s", cap
, strerror (ret
));
127 static void *parse_pointer (const char *cap
, const char *s
)
132 ret
= sscanf (s
, "%p", &ptr
);
134 errx (1, "%s: cannot parse pointer in `%s'", cap
, s
);
139 static int hasdata (int sock
)
146 ret
= poll (&pfd
, 1, 0);
153 return pfd
.revents
& POLLIN
;
156 static double now (void)
160 if (gettimeofday (&tv
, NULL
)) {
161 err (1, "gettimeofday");
163 return tv
.tv_sec
+ tv
.tv_usec
*1e-6;
166 static void readdata (int fd
, char *p
, int size
)
170 n
= read (fd
, p
, size
);
172 err (1, "read (req %d, ret %zd)", size
, n
);
176 static void writedata (int fd
, char *p
, int size
)
181 buf
[0] = (size
>> 24) & 0xff;
182 buf
[1] = (size
>> 16) & 0xff;
183 buf
[2] = (size
>> 8) & 0xff;
184 buf
[3] = (size
>> 0) & 0xff;
186 n
= write (fd
, buf
, 4);
188 err (1, "write %zd", n
);
191 n
= write (fd
, p
, size
);
193 err (1, "write (req %d, ret %zd)", size
, n
);
197 static void __attribute__ ((format (printf
, 2, 3)))
198 printd (int fd
, const char *fmt
, ...)
205 len
= vsnprintf (buf
, sizeof (buf
), fmt
, ap
);
207 writedata (fd
, buf
, len
);
210 static void die (fz_error error
)
212 fz_catch (error
, "aborting");
214 pdf_closexref (state
.xref
);
218 static void openxref (char *filename
)
223 fd
= open (filename
, O_BINARY
| O_RDONLY
, 0666);
225 die (fz_throw ("cannot open file '%s'", filename
));
227 file
= fz_openfile (fd
);
228 state
.xref
= pdf_openxref (file
);
230 die (fz_throw ("cannot open PDF file '%s'", filename
));
231 fz_dropstream (file
);
233 if (pdf_needspassword (state
.xref
)) {
234 die (fz_throw ("password protected"));
237 state
.pagecount
= pdf_getpagecount (state
.xref
);
238 state
.pagetbl
= stat_alloc (state
.pagecount
* sizeof (*state
.pagetbl
));
241 static int readlen (int fd
)
248 err (1, "read %zd", n
);
251 return (p
[0] << 24) | (p
[1] << 16) | (p
[2] << 8) | p
[3];
254 static void freepage (struct page
*page
)
259 fz_droppixmap (page
->pixmap
);
260 for (p
= state
.pages
; p
; p
= p
->prev
) {
261 if (p
->prev
== page
) {
262 p
->prev
= page
->prev
;
266 for (i
= 0; i
< page
->slicecount
; ++i
) {
267 struct slice
*s
= &page
->slices
[i
];
268 if (s
->texindex
!= -1) {
269 if (state
.texowners
[s
->texindex
].slice
== s
) {
270 state
.texowners
[s
->texindex
].slice
= NULL
;
271 ARSERT (state
.texowners
[s
->texindex
].w
== s
->w
);
272 ARSERT (state
.texowners
[s
->texindex
].h
>= s
->h
);
277 fz_freetextspan (page
->text
);
279 if (page
->drawpage
) {
280 pdf_droppage (page
->drawpage
);
286 static void subdivide (struct page
*p
)
289 int h
= p
->pixmap
->h
;
290 int th
= MIN (h
, state
.sliceheight
);
292 for (i
= 0; i
< p
->slicecount
; ++i
) {
293 struct slice
*s
= &p
->slices
[i
];
301 static void *render (int pageno
, int pindex
)
309 fz_displaylist
*list
;
310 fz_device
*idev
, *mdev
;
311 struct pagedim
*pagedim
;
314 /* printd (state.sock, "T rendering %d", pageno); */
315 pdf_flushxref (state
.xref
, 0);
317 pagedim
= &state
.pagedims
[pindex
];
318 slicecount
= (pagedim
->bbox
.y1
- pagedim
->bbox
.y0
319 + state
.sliceheight
- 1) / state
.sliceheight
;
320 slicecount
+= slicecount
== 0;
322 page
= calloc (sizeof (*page
)
323 + (slicecount
* sizeof (struct slice
)), 1);
325 err (1, "calloc page %d\n", pageno
);
327 page
->slicecount
= slicecount
;
328 page
->prev
= state
.pages
;
331 pageobj
= pdf_getpageobject (state
.xref
, pageno
);
333 die (fz_throw ("cannot retrieve info from page %d", pageno
));
335 error
= pdf_loadpage (&drawpage
, state
.xref
, pageobj
);
339 page
->pixmap
= fz_newpixmapwithrect (pdf_devicergb
, pagedim
->bbox
);
342 fz_clearpixmap (page
->pixmap
, 0xFF);
344 list
= fz_newdisplaylist ();
346 die (fz_throw ("fz_newdisplaylist failed"));
348 mdev
= fz_newlistdevice (list
);
349 error
= pdf_runcontentstream (mdev
, fz_identity (), state
.xref
,
354 fz_freedevice (mdev
);
356 idev
= fz_newdrawdevice (state
.cache
, page
->pixmap
);
358 die (fz_throw ("fz_newdrawdevice failed"));
359 fz_executedisplaylist (list
, idev
, pagedim
->ctm
);
360 fz_freedevice (idev
);
362 fz_freedisplaylist (list
);
364 page
->drawpage
= drawpage
;
365 page
->pagedim
= pagedim
;
366 page
->pageno
= pageno
;
370 if (!state
.lotsamemory
) {
371 pdf_agestoreditems (state
.xref
->store
);
372 pdf_evictageditems (state
.xref
->store
);
375 /* printd (state.sock, "T rendering %d took %f sec", pageno, end - start); */
379 /* almost verbatim copy of pdf_getpagecountimp */
389 recurse_page (fz_obj
*node
, int bias
, int *pagesp
, struct stuff inherit
)
398 if (!fz_isdict(node
))
400 fz_warn("pagetree node is missing, igoring missing pages...");
404 type
= fz_dictgets(node
, "Type");
405 kids
= fz_dictgets(node
, "Kids");
406 count
= fz_dictgets(node
, "Count");
409 typestr
= fz_toname(type
);
412 fz_warn("pagetree node (%d %d R) lacks required type", fz_tonum(node
), fz_togen(node
));
414 kids
= fz_dictgets(node
, "Kids");
417 fz_warn("guessing it may be a pagetree node, continuing...");
422 fz_warn("guessing it may be a page, continuing...");
427 if (!strcmp(typestr
, "Page")) {
432 int pageno
= *pagesp
;
434 state
.pagetbl
[pageno
+ bias
] = fz_tonum (node
);
436 obj
= fz_dictgets (node
, "CropBox");
437 if (!obj
) obj
= inherit
.cropbox
;
438 if (!fz_isarray (obj
)) {
439 obj
= fz_dictgets (node
, "MediaBox");
440 if (!obj
) obj
= inherit
.mediabox
;
442 if (!fz_isarray (obj
)) {
443 die (fz_throw ("cannot find page bounds %d (%d Rd)",
444 fz_tonum (node
), fz_togen (node
)));
447 box
= pdf_torect (obj
);
449 obj
= fz_dictgets (node
, "Rotate");
450 if (!obj
) obj
= inherit
.rotate
;
451 if (fz_isint (obj
)) {
452 rotate
= fz_toint (obj
);
458 p
= &state
.pagedims
[state
.pagedimcount
- 1];
459 if ((state
.pagedimcount
== 0)
460 || (p
->rotate
!= rotate
|| memcmp (&p
->box
, &box
, sizeof (box
)))) {
463 size
= (state
.pagedimcount
+ 1) * sizeof (*state
.pagedims
);
464 state
.pagedims
= realloc (state
.pagedims
, size
);
465 if (!state
.pagedims
) {
466 err (1, "realloc pagedims to %zu (%d elems)",
467 size
, state
.pagedimcount
+ 1);
469 p
= &state
.pagedims
[state
.pagedimcount
++];
472 p
->pageno
= pageno
+ bias
;
476 else if (!strcmp(typestr
, "Pages"))
480 if (!fz_isarray(kids
))
481 fz_warn("page tree node contains no pages");
483 pdf_logpage("subtree (%d %d R) {\n", fz_tonum(node
), fz_togen(node
));
485 inh
= fz_dictgets(node
, "Resources");
486 if (inh
) inherit
.resources
= inh
;
488 inh
= fz_dictgets(node
, "MediaBox");
489 if (inh
) inherit
.mediabox
= inh
;
491 inh
= fz_dictgets(node
, "CropBox");
492 if (inh
) inherit
.cropbox
= inh
;
494 inh
= fz_dictgets(node
, "Rotate");
495 if (inh
) inherit
.rotate
= inh
;
497 for (i
= 0; i
< fz_arraylen(kids
); i
++)
499 fz_obj
*obj
= fz_arrayget(kids
, i
);
501 /* prevent infinite recursion possible in maliciously crafted PDFs */
504 fz_warn("cyclic page tree");
508 recurse_page (obj
, *pagesp
+ bias
, &pages
, inherit
);
511 if (pages
!= fz_toint(count
))
513 fz_warn("page tree node contains incorrect number of pages, continuing...");
514 count
= fz_newint(pages
);
515 fz_dictputs(node
, "Count", count
);
519 pdf_logpage("%d pages\n", pages
);
527 static void initpdims (void)
533 struct stuff inherit
;
536 catalog
= fz_dictgets (state
.xref
->trailer
, "Root");
537 pages
= fz_dictgets (catalog
, "Pages");
539 inherit
.resources
= nil
;
540 inherit
.mediabox
= nil
;
541 inherit
.cropbox
= nil
;
542 inherit
.rotate
= nil
;
545 recurse_page (pages
, 0, &count
, inherit
);
547 printd (state
.sock
, "T Processed %d pages in %f seconds",
551 static void layout (void)
557 struct pagedim
*p
= state
.pagedims
;
560 printd (state
.sock
, "c");
561 for (pindex
= 0; pindex
< state
.pagedimcount
; ++pindex
, ++p
) {
562 box
.x0
= MIN (p
->box
.x0
, p
->box
.x1
);
563 box
.y0
= MIN (p
->box
.y0
, p
->box
.y1
);
564 box
.x1
= MAX (p
->box
.x0
, p
->box
.x1
);
565 box
.y1
= MAX (p
->box
.y0
, p
->box
.y1
);
567 ctm
= fz_identity ();
568 ctm
= fz_concat (ctm
, fz_translate (0, -box
.y1
));
569 ctm
= fz_concat (ctm
, fz_rotate (p
->rotate
));
570 box2
= fz_transformrect (ctm
, box
);
571 w
= box2
.x1
- box2
.x0
;
573 zoom
= (state
.w
/ w
);
574 ctm
= fz_identity ();
575 ctm
= fz_concat (ctm
, fz_translate (0, -box
.y1
));
576 ctm
= fz_concat (ctm
, fz_scale (zoom
, -zoom
));
577 ctm
= fz_concat (ctm
, fz_rotate (p
->rotate
));
578 p
->bbox
= fz_roundrect (fz_transformrect (ctm
, box
));
579 memcpy (&p
->ctm
, &ctm
, sizeof (ctm
));
582 while (p
-- != state
.pagedims
) {
583 printd (state
.sock
, "l %d %d %d",
584 p
->pageno
, p
->bbox
.x1
- p
->bbox
.x0
, p
->bbox
.y1
- p
->bbox
.y0
);
587 printd (state
.sock
, "C %d", state
.pagecount
);
590 static void recurse_outline (pdf_outline
*outline
, int level
)
597 if (!outline
->link
) goto next
;
599 obj
= outline
->link
->dest
;
600 if (fz_isarray (obj
)) {
604 struct pagedim
*pagedim
= state
.pagedims
;
606 obj2
= fz_arrayget (obj
, 0);
607 if (fz_isint (obj2
)) {
608 pageno
= fz_toint (obj2
);
611 num
= fz_tonum (obj2
);
612 for (i
= 0; i
< state
.pagecount
; ++i
) {
613 if (state
.pagetbl
[i
] == num
) {
620 for (i
= 0; i
< state
.pagedimcount
; ++i
) {
621 if (state
.pagedims
[i
].pageno
> pageno
)
623 pagedim
= &state
.pagedims
[i
];
626 if (fz_arraylen (obj
) > 3) {
629 p
.x
= fz_toint (fz_arrayget (obj
, 2));
630 p
.y
= fz_toint (fz_arrayget (obj
, 3));
631 p
= fz_transformpoint (pagedim
->ctm
, p
);
636 lprintf ("%*c%s %d\n", level
, ' ', outline
->title
, pageno
);
637 printd (state
.sock
, "o %d %d %d %s",
638 level
, pageno
, top
, outline
->title
);
640 if (outline
->child
) {
641 recurse_outline (outline
->child
, level
+ 1);
643 outline
= outline
->next
;
647 static void process_outline (void)
649 pdf_outline
*outline
;
651 outline
= pdf_loadoutline (state
.xref
);
653 recurse_outline (outline
, 0);
654 pdf_dropoutline (outline
);
658 static int comparespans (const void *l
, const void *r
)
660 fz_textspan
*const*ls
= l
;
661 fz_textspan
*const*rs
= r
;
663 return (*ls
)->text
->bbox
.y0
- (*rs
)->text
->bbox
.y0
;
666 /* wishful thinking function */
667 static void search (regex_t
*re
, int pageno
, int y
, int forward
)
677 fz_textspan
*text
, *span
, **pspan
;
678 struct pagedim
*pdim
, *pdimprev
;
685 while (pageno
>= 0 && pageno
< state
.pagecount
&& !stop
) {
688 if (hasdata (state
.sock
)) {
689 printd (state
.sock
, "T attention requested aborting search at %d",
694 printd (state
.sock
, "T searching in page %d", pageno
);
698 for (i
= 0; i
< state
.pagedimcount
; ++i
) {
699 pdim
= &state
.pagedims
[i
];
700 if (pdim
->pageno
== pageno
) {
703 if (pdim
->pageno
> pageno
) {
712 pageobj
= pdf_getpageobject (state
.xref
, pageno
+ 1);
714 die (fz_throw ("cannot retrieve info from page %d", pageno
));
716 error
= pdf_loadpage (&drawpage
, state
.xref
, pageobj
);
720 text
= fz_newtextspan ();
721 tdev
= fz_newtextdevice (text
);
722 error
= pdf_runcontentstream (tdev
, pdim
->ctm
, state
.xref
,
725 if (error
) die (error
);
726 fz_freedevice (tdev
);
729 for (span
= text
; span
; span
= span
->next
) {
732 pspan
= malloc (sizeof (void *) * nspans
);
734 err (1, "malloc span pointers %zu", sizeof (void *) * nspans
);
736 for (i
= 0, span
= text
; span
; span
= span
->next
, ++i
) {
739 qsort (pspan
, nspans
, sizeof (fz_textspan
*), comparespans
);
741 j
= forward
? 0 : nspans
- 1;
746 j
+= forward
? 1 : -1;
748 /* XXX: spans are not sorted "visually" */
749 for (i
= 0; i
< MIN (span
->len
, sizeof (buf
) - 1); ++i
) {
751 if (span
->text
[i
].bbox
.y0
< y
+ 1) {
756 if (span
->text
[i
].bbox
.y0
> y
- 1) {
760 if (span
->text
[i
].c
< 256) {
761 *p
++ = span
->text
[i
].c
;
772 ret
= regexec (re
, buf
, 1, &rm
, 0);
774 if (ret
!= REG_NOMATCH
) {
777 size
= regerror (ret
, re
, errbuf
, sizeof (errbuf
));
779 "T regexec error `%.*s'",
786 r
.x0
= span
->text
[rm
.rm_so
].bbox
.x0
- pdim
->bbox
.x0
;
787 r
.y0
= span
->text
[rm
.rm_so
].bbox
.y0
;
788 r
.x1
= span
->text
[rm
.rm_eo
- 1].bbox
.x1
- pdim
->bbox
.x0
;
789 r
.y1
= span
->text
[rm
.rm_eo
- 1].bbox
.y1
;
792 printd (state
.sock
, "F %d %d %f %f %f %f",
798 printd (state
.sock
, "R %d %d %f %f %f %f",
803 printd (state
.sock
, "T found at %d `%.*s' %f in %f sec",
804 pageno
, rm
.rm_eo
- rm
.rm_so
, &buf
[rm
.rm_so
],
805 span
->text
[0].bbox
.y0
- drawpage
->mediabox
.y0
,
818 fz_freetextspan (text
);
819 pdf_droppage (drawpage
);
824 printd (state
.sock
, "T no matches %f sec", end
- start
);
826 printd (state
.sock
, "d");
829 static void *mainloop (void *unused
)
832 int len
, ret
, oldlen
= 0;
835 len
= readlen (state
.sock
);
837 errx (1, "readlen returned 0");
840 if (oldlen
< len
+ 1) {
841 p
= realloc (p
, len
+ 1);
843 err (1, "realloc %d failed", len
+ 1);
847 readdata (state
.sock
, p
, len
);
850 if (!strncmp ("open", p
, 4)) {
851 char *filename
= p
+ 5;
856 else if (!strncmp ("free", p
, 4)) {
859 ret
= sscanf (p
+ 4, " %p", &ptr
);
861 errx (1, "malformed free `%.*s' ret=%d", len
, p
, ret
);
866 printd (state
.sock
, "d");
868 else if (!strncmp ("search", p
, 6)) {
869 int icase
, pageno
, y
, ret
, len2
, forward
;
873 ret
= sscanf (p
+ 6, " %d %d %d %d,%n",
874 &icase
, &pageno
, &y
, &forward
, &len2
);
876 errx (1, "malformed search `%s' ret=%d", p
, ret
);
879 pattern
= p
+ 6 + len2
;
880 ret
= regcomp (&re
, pattern
,
881 REG_EXTENDED
| (icase
? REG_ICASE
: 0));
886 size
= regerror (ret
, &re
, errbuf
, sizeof (errbuf
));
887 printd (state
.sock
, "T regcomp failed `%.*s'",
891 search (&re
, pageno
, y
, forward
);
895 else if (!strncmp ("geometry", p
, 8)) {
898 ret
= sscanf (p
+ 8, " %d %d", &w
, &h
);
900 errx (1, "malformed geometry `%.*s' ret=%d", len
, p
, ret
);
906 for (i
= 0; i
< state
.texcount
; ++i
) {
907 state
.texowners
[i
].slice
= NULL
;
914 printd (state
.sock
, "d");
916 else if (!strncmp ("render", p
, 6)) {
917 int pageno
, pindex
, w
, h
, ret
;
920 ret
= sscanf (p
+ 6, " %d %d %d %d", &pageno
, &pindex
, &w
, &h
);
922 errx (1, "bad render line `%.*s' ret=%d", len
, p
, ret
);
925 page
= render (pageno
, pindex
);
926 printd (state
.sock
, "r %d %d %d %p\n",
933 errx (1, "unknown command %.*s", len
, p
);
939 static void upload2 (struct page
*page
, int slicenum
, const char *cap
)
944 struct slice
*slice
= &page
->slices
[slicenum
];
949 ARSERT (w
== slice
->w
);
950 if (slice
->texindex
!= -1
951 && state
.texowners
[slice
->texindex
].slice
== slice
) {
952 glBindTexture (GL_TEXTURE_RECTANGLE_ARB
, state
.texids
[slice
->texindex
]);
956 int index
= (state
.texindex
++ % state
.texcount
);
959 for (i
= 0; i
< slicenum
; ++i
) {
960 offset
+= w
* page
->slices
[i
].h
* 4;
963 if (state
.texowners
[index
].w
== slice
->w
) {
964 if (state
.texowners
[index
].h
>= slice
->h
) {
968 state
.texowners
[index
].h
= slice
->h
;
972 state
.texowners
[index
].h
= slice
->h
;
975 state
.texowners
[index
].slice
= slice
;
976 state
.texowners
[index
].w
= slice
->w
;
977 slice
->texindex
= index
;
979 glBindTexture (GL_TEXTURE_RECTANGLE_ARB
, state
.texids
[slice
->texindex
]);
983 GLenum err
= glGetError ();
984 if (err
!= GL_NO_ERROR
) {
985 printf ("\e[0;31mERROR1 %d %d %#x\e[0m\n", w
, slice
->h
, err
);
989 glTexSubImage2D (GL_TEXTURE_RECTANGLE_ARB
,
997 page
->pixmap
->samples
+ offset
1000 GLenum err
= glGetError ();
1001 if (err
!= GL_NO_ERROR
) {
1002 printf ("\e[0;31mERROR %d %d %#x\e[0m\n", w
, slice
->h
, err
);
1008 glTexImage2D (GL_TEXTURE_RECTANGLE_ARB
,
1016 page
->pixmap
->samples
+ offset
1021 lprintf ("%s[%d] slice=%d(%d,%d) texid=%d %f sec\n",
1022 subimage
? "sub" : "img",
1023 page
->pageno
, slicenum
,
1025 state
.texids
[slice
->texindex
],
1030 CAMLprim value
ml_preload (value ptr_v
)
1036 char *s
= String_val (ptr_v
);
1039 if (trylock ("ml_preload")) {
1042 ret
= sscanf (s
, "%p", &ptr
);
1044 errx (1, "cannot parse pointer `%s'", s
);
1048 for (i
= 0; i
< page
->slicecount
; ++i
) {
1049 upload2 (ptr
, i
, "preload");
1052 unlock ("ml_preload");
1054 CAMLreturn (Val_unit
);
1057 CAMLprim value
ml_draw (value dispy_v
, value w_v
, value h_v
,
1058 value py_v
, value ptr_v
)
1060 CAMLparam5 (dispy_v
, w_v
, h_v
, py_v
, ptr_v
);
1061 int dispy
= Int_val (dispy_v
);
1062 int w
= Int_val (w_v
);
1063 int h
= Int_val (h_v
);
1064 int py
= Int_val (py_v
);
1065 char *s
= String_val (ptr_v
);
1071 if (trylock ("ml_draw")) {
1075 ret
= sscanf (s
, "%p", &ptr
);
1077 errx (1, "cannot parse pointer `%s'", s
);
1081 w
= page
->pixmap
->w
;
1083 ARSERT (h
>= 0 && "ml_draw wrong h");
1085 glEnable (GL_TEXTURE_RECTANGLE_ARB
);
1086 if (state
.useatifs
) {
1087 glEnable (GL_FRAGMENT_SHADER_ATI
);
1090 for (slicenum
= 0; slicenum
< page
->slicecount
; ++slicenum
) {
1091 struct slice
*slice
= &page
->slices
[slicenum
];
1092 if (slice
->h
> py
) {
1098 h
= MIN (state
.h
, h
);
1101 struct slice
*slice
= &page
->slices
[slicenum
];
1103 ARSERT (slicenum
< page
->slicecount
&& "ml_draw wrong slicenum");
1105 th
= MIN (h
, slice
->h
- py
);
1106 upload2 (page
, slicenum
, "upload");
1110 glTexCoord2i (0, py
);
1111 glVertex2i (0, dispy
);
1113 glTexCoord2i (w
, py
);
1114 glVertex2i (w
, dispy
);
1116 glTexCoord2i (w
, py
+th
);
1117 glVertex2i (w
, dispy
+ th
);
1119 glTexCoord2i (0, py
+th
);
1120 glVertex2i (0, dispy
+ th
);
1130 glDisable (GL_TEXTURE_RECTANGLE_ARB
);
1131 if (state
.useatifs
) {
1132 glDisable (GL_FRAGMENT_SHADER_ATI
);
1137 CAMLreturn (Val_unit
);
1140 static pdf_link
*getlink (struct page
*page
, int x
, int y
)
1149 ctm
= fz_invertmatrix (page
->pagedim
->ctm
);
1150 p
= fz_transformpoint (ctm
, p
);
1152 for (link
= page
->drawpage
->links
; link
; link
= link
->next
) {
1153 if (p
.x
>= link
->rect
.x0
&& p
.x
<= link
->rect
.x1
) {
1154 if (p
.y
>= link
->rect
.y0
&& p
.y
<= link
->rect
.y1
) {
1155 if (link
->kind
== PDF_LGOTO
) {
1164 CAMLprim value
ml_checklink (value ptr_v
, value x_v
, value y_v
)
1166 CAMLparam3 (ptr_v
, x_v
, y_v
);
1167 char *s
= String_val (ptr_v
);
1170 if (trylock ("ml_checklink")) {
1174 ret
= NULL
!= getlink (parse_pointer ("ml_checklink", s
),
1175 Int_val (x_v
), Int_val (y_v
));
1176 unlock ("ml_checklink");
1178 CAMLreturn (Val_bool (ret
));
1181 CAMLprim value
ml_getlink (value ptr_v
, value x_v
, value y_v
)
1183 CAMLparam3 (ptr_v
, x_v
, y_v
);
1184 CAMLlocal2 (ret_v
, tup_v
);
1187 char *s
= String_val (ptr_v
);
1189 if (trylock ("ml_gettext")) {
1190 ret_v
= Val_int (0);
1194 page
= parse_pointer ("ml_getlink", s
);
1196 link
= getlink (page
, Int_val (x_v
), Int_val (y_v
));
1203 obj
= fz_arrayget (link
->dest
, 0);
1204 if (fz_isindirect (obj
)) {
1205 pageno
= pdf_findpageobject (state
.xref
, obj
) - 1;
1207 else if (fz_isint (obj
)) {
1208 pageno
= fz_toint (obj
);
1211 if (fz_arraylen (link
->dest
) > 3) {
1212 p
.x
= fz_toint (fz_arrayget (link
->dest
, 2));
1213 p
.y
= fz_toint (fz_arrayget (link
->dest
, 3));
1214 p
= fz_transformpoint (page
->pagedim
->ctm
, p
);
1221 tup_v
= caml_alloc_tuple (2);
1222 ret_v
= caml_alloc_small (1, 1);
1223 Field (tup_v
, 0) = Val_int (pageno
);
1224 Field (tup_v
, 1) = Val_int (p
.y
);
1225 Field (ret_v
, 0) = tup_v
;
1228 ret_v
= Val_int (0);
1230 unlock ("ml_getlink");
1236 CAMLprim value
ml_gettext (value ptr_v
, value rect_v
, value oy_v
, value rectsel_v
)
1238 CAMLparam4 (ptr_v
, rect_v
, oy_v
, rect_v
);
1243 char *s
= String_val (ptr_v
);
1244 int rectsel
= Bool_val (rectsel_v
);
1245 int i
, bx0
, bx1
, by0
, by1
, x0
, x1
, y0
, y1
, oy
;
1247 /* stop GCC from complaining about uninitialized variables */
1248 int rx0
= rx0
, rx1
= rx1
, ry0
= ry0
, ry1
= ry1
;
1250 if (trylock ("ml_gettext")) {
1254 page
= parse_pointer ("ml_gettext", s
);
1256 oy
= Int_val (oy_v
);
1257 p1
.x
= Int_val (Field (rect_v
, 0));
1258 p1
.y
= Int_val (Field (rect_v
, 1));
1259 p2
.x
= Int_val (Field (rect_v
, 2));
1260 p2
.y
= Int_val (Field (rect_v
, 3));
1263 glEnable (GL_BLEND
);
1264 glPolygonMode (GL_FRONT_AND_BACK
, GL_LINE
);
1265 glBlendFunc (GL_DST_ALPHA
, GL_SRC_ALPHA
);
1266 glColor4f (0, 0, 0, 0.2);
1267 glRecti (p1
.x
, p1
.y
, p2
.x
, p2
.y
);
1268 glPolygonMode (GL_FRONT_AND_BACK
, GL_FILL
);
1269 glDisable (GL_BLEND
);
1272 ctm
= page
->pagedim
->ctm
;
1277 page
->text
= fz_newtextspan ();
1278 tdev
= fz_newtextdevice (page
->text
);
1279 error
= pdf_runcontentstream (tdev
, page
->pagedim
->ctm
, state
.xref
,
1280 page
->drawpage
->resources
,
1281 page
->drawpage
->contents
);
1282 if (error
) die (error
);
1283 fz_freedevice (tdev
);
1288 printf ("BBox %f %f %f %f\n", p1
.x
, p1
.y
, p2
.x
, p2
.y
);
1289 p1
.x
+= page
->pixmap
->x
;
1290 p1
.y
+= page
->pixmap
->y
;
1291 p2
.x
+= page
->pixmap
->x
;
1292 p2
.y
+= page
->pixmap
->y
;
1297 printf ("BBox %d %d %d %d %d %d\n", x0
, y0
, x1
, y1
, oy
, page
->pageno
);
1299 for (span
= page
->text
; span
; span
= span
->next
) {
1302 /* fz_debugtextspanxml (span); */
1303 for (i
= 0; i
< span
->len
; ++i
) {
1306 bx0
= span
->text
[i
].bbox
.x0
;
1307 bx1
= span
->text
[i
].bbox
.x1
;
1308 by0
= span
->text
[i
].bbox
.y0
+ oy
;
1309 by1
= span
->text
[i
].bbox
.y1
+ oy
;
1311 if ((bx1
>= x0
&& bx0
<= x1
&& by1
>= y0
&& by0
<= y1
)) {
1313 rx0
= bx0
- page
->pixmap
->x
;
1314 rx1
= bx1
- page
->pixmap
->x
;
1320 c
= span
->text
[i
].c
;
1322 if ((isprint (c
) && !isspace (c
))) {
1324 bx0
-= page
->pixmap
->x
;
1325 bx1
-= page
->pixmap
->x
;
1326 glEnable (GL_BLEND
);
1327 glPolygonMode (GL_FRONT_AND_BACK
, GL_FILL
);
1328 glBlendFunc (GL_DST_ALPHA
, GL_SRC_ALPHA
);
1329 glColor4f (0.5, 0.5, 0.0, 0.6);
1330 glRecti (bx0
, by0
, bx1
, by1
);
1331 glPolygonMode (GL_FRONT_AND_BACK
, GL_FILL
);
1332 glDisable (GL_BLEND
);
1334 if (isprint (c
) || c
==' ') {
1349 glEnable (GL_BLEND
);
1350 glPolygonMode (GL_FRONT_AND_BACK
, GL_FILL
);
1351 glBlendFunc (GL_DST_ALPHA
, GL_SRC_ALPHA
);
1352 glColor4f (0.5, 0.5, 0.0, 0.6);
1353 glRecti (rx0
, ry0
, rx1
, ry1
);
1354 glPolygonMode (GL_FRONT_AND_BACK
, GL_FILL
);
1355 glDisable (GL_BLEND
);
1359 if (seen
&& span
->eol
) {
1360 x0
= page
->pixmap
->x
;
1361 putc ('\n', stdout
);
1364 unlock ("ml_gettext");
1367 CAMLreturn (Val_unit
);
1370 CAMLprim value
ml_getpagewh (value pagedimno_v
)
1372 CAMLparam1 (pagedimno_v
);
1374 int pagedimno
= Int_val (pagedimno_v
);
1376 ret_v
= caml_alloc_small (4 * Double_wosize
, Double_array_tag
);
1377 Store_double_field (ret_v
, 0, state
.pagedims
[pagedimno
].box
.x0
);
1378 Store_double_field (ret_v
, 1, state
.pagedims
[pagedimno
].box
.x1
);
1379 Store_double_field (ret_v
, 2, state
.pagedims
[pagedimno
].box
.y0
);
1380 Store_double_field (ret_v
, 3, state
.pagedims
[pagedimno
].box
.y1
);
1384 static void initgl (void)
1387 if (strstr ((char *) glGetString (GL_EXTENSIONS
),
1388 "GL_ATI_fragment_shader")) {
1389 /* Here, with MESA, rv280, powerpc32: BGRA(rev) is slow while
1390 ABGR is fast, so fix things in the shader */
1391 state
.texform
= GL_ABGR_EXT
;
1392 state
.texty
= GL_UNSIGNED_INT_8_8_8_8
;
1394 glBindFragmentShaderATI (1);
1395 glBeginFragmentShaderATI ();
1397 glSampleMapATI (GL_REG_0_ATI
, GL_TEXTURE0_ARB
, GL_SWIZZLE_STR_ATI
);
1399 glColorFragmentOp1ATI (GL_MOV_ATI
,
1400 GL_REG_1_ATI
, GL_RED_BIT_ATI
, GL_NONE
,
1401 GL_REG_0_ATI
, GL_BLUE
, GL_NONE
);
1402 glColorFragmentOp1ATI (GL_MOV_ATI
,
1403 GL_REG_1_ATI
, GL_BLUE_BIT_ATI
, GL_NONE
,
1404 GL_REG_0_ATI
, GL_RED
, GL_NONE
);
1405 glColorFragmentOp1ATI (
1407 GL_REG_0_ATI
, GL_RED_BIT_ATI
| GL_BLUE_BIT_ATI
, GL_NONE
,
1408 GL_REG_1_ATI
, GL_NONE
, GL_NONE
1411 glEndFragmentShaderATI ();
1415 state
.texform
= GL_BGRA_EXT
;
1416 state
.texty
= GL_UNSIGNED_INT_8_8_8_8_REV
;
1419 state
.texform
= GL_BGRA_EXT
;
1420 state
.texty
= GL_UNSIGNED_INT_8_8_8_8
;
1424 CAMLprim value
ml_init (value sock_v
)
1427 CAMLparam1 (sock_v
);
1429 state
.texcount
= 128;
1430 state
.sliceheight
= 64;
1432 state
.texids
= calloc (state
.texcount
* sizeof (*state
.texids
), 1);
1433 if (!state
.texids
) {
1434 err (1, "calloc texids %zu", state
.texcount
* sizeof (*state
.texids
));
1437 state
.texowners
= calloc (state
.texcount
* sizeof (*state
.texowners
), 1);
1438 if (!state
.texowners
) {
1439 err (1, "calloc texowners %zu",
1440 state
.texcount
* sizeof (*state
.texowners
));
1443 glGenTextures (state
.texcount
, state
.texids
);
1445 state
.sock
= Int_val (sock_v
);
1448 state
.cache
= fz_newglyphcache ();
1450 errx (1, "fz_newglyphcache failed");
1453 ret
= pthread_create (&state
.thread
, NULL
, mainloop
, NULL
);
1455 errx (1, "pthread_create: %s", strerror (errno
));
1458 CAMLreturn (Val_unit
);