4 24c02865 2005-01-04 devnull * pdf file support for page
7 24c02865 2005-01-04 devnull #include <u.h>
8 24c02865 2005-01-04 devnull #include <libc.h>
9 24c02865 2005-01-04 devnull #include <draw.h>
10 17157e4a 2006-03-20 devnull #include <cursor.h>
11 05a4d855 2007-03-26 devnull #include <thread.h>
12 24c02865 2005-01-04 devnull #include <bio.h>
13 24c02865 2005-01-04 devnull #include "page.h"
15 24c02865 2005-01-04 devnull static Image* pdfdrawpage(Document *d, int page);
16 24c02865 2005-01-04 devnull static char* pdfpagename(Document*, int);
18 24c02865 2005-01-04 devnull char *pdfprolog =
19 24c02865 2005-01-04 devnull #include "pdfprolog.c"
22 24c02865 2005-01-04 devnull Rectangle
23 24c02865 2005-01-04 devnull pdfbbox(GSInfo *gs)
26 24c02865 2005-01-04 devnull char *f[4];
27 24c02865 2005-01-04 devnull Rectangle r;
29 24c02865 2005-01-04 devnull r = Rect(0,0,0,0);
30 24c02865 2005-01-04 devnull waitgs(gs);
31 24c02865 2005-01-04 devnull gscmd(gs, "/CropBox knownoget {} {[0 0 0 0]} ifelse PAGE==\n");
32 24c02865 2005-01-04 devnull p = Brdline(&gs->gsrd, '\n');
33 24c02865 2005-01-04 devnull p[Blinelen(&gs->gsrd)-1] ='\0';
34 24c02865 2005-01-04 devnull if(p[0] != '[')
35 24c02865 2005-01-04 devnull return r;
36 24c02865 2005-01-04 devnull if(tokenize(p+1, f, 4) != 4)
37 24c02865 2005-01-04 devnull return r;
38 24c02865 2005-01-04 devnull r = Rect(atoi(f[0]), atoi(f[1]), atoi(f[2]), atoi(f[3]));
39 24c02865 2005-01-04 devnull waitgs(gs);
40 24c02865 2005-01-04 devnull return r;
43 24c02865 2005-01-04 devnull Document*
44 24c02865 2005-01-04 devnull initpdf(Biobuf *b, int argc, char **argv, uchar *buf, int nbuf)
46 24c02865 2005-01-04 devnull Document *d;
47 24c02865 2005-01-04 devnull PDFInfo *pdf;
49 24c02865 2005-01-04 devnull char *fn;
50 24c02865 2005-01-04 devnull char fdbuf[20];
52 24c02865 2005-01-04 devnull int i, npage;
53 24c02865 2005-01-04 devnull Rectangle bbox;
55 24c02865 2005-01-04 devnull if(argc > 1) {
56 24c02865 2005-01-04 devnull fprint(2, "can only view one pdf file at a time\n");
57 24c02865 2005-01-04 devnull return nil;
60 24c02865 2005-01-04 devnull fprint(2, "reading through pdf...\n");
61 24c02865 2005-01-04 devnull if(b == nil){ /* standard input; spool to disk (ouch) */
62 24c02865 2005-01-04 devnull fd = spooltodisk(buf, nbuf, &fn);
63 05a4d855 2007-03-26 devnull sprint(fdbuf, "/dev/fd/%d", fd);
64 24c02865 2005-01-04 devnull b = Bopen(fdbuf, OREAD);
65 24c02865 2005-01-04 devnull if(b == nil){
66 24c02865 2005-01-04 devnull fprint(2, "cannot open disk spool file\n");
67 24c02865 2005-01-04 devnull wexits("Bopen temp");
70 24c02865 2005-01-04 devnull fn = argv[0];
72 24c02865 2005-01-04 devnull /* sanity check */
73 24c02865 2005-01-04 devnull Bseek(b, 0, 0);
74 24c02865 2005-01-04 devnull if(!(p = Brdline(b, '\n')) && !(p = Brdline(b, '\r'))) {
75 24c02865 2005-01-04 devnull fprint(2, "cannot find end of first line\n");
76 24c02865 2005-01-04 devnull wexits("initps");
78 24c02865 2005-01-04 devnull if(strncmp(p, "%PDF-", 5) != 0) {
79 24c02865 2005-01-04 devnull werrstr("not pdf");
80 24c02865 2005-01-04 devnull return nil;
83 24c02865 2005-01-04 devnull /* setup structures so one free suffices */
84 24c02865 2005-01-04 devnull p = emalloc(sizeof(*d) + sizeof(*pdf));
85 24c02865 2005-01-04 devnull d = (Document*) p;
86 24c02865 2005-01-04 devnull p += sizeof(*d);
87 24c02865 2005-01-04 devnull pdf = (PDFInfo*) p;
89 24c02865 2005-01-04 devnull d->extra = pdf;
90 24c02865 2005-01-04 devnull d->b = b;
91 24c02865 2005-01-04 devnull d->drawpage = pdfdrawpage;
92 24c02865 2005-01-04 devnull d->pagename = pdfpagename;
93 24c02865 2005-01-04 devnull d->fwdonly = 0;
94 d9e047e5 2010-02-04 rsc d->type = Tpdf;
96 17157e4a 2006-03-20 devnull if(spawngs(&pdf->gs, "-dDELAYSAFER") < 0)
97 24c02865 2005-01-04 devnull return nil;
99 24c02865 2005-01-04 devnull gscmd(&pdf->gs, "%s", pdfprolog);
100 24c02865 2005-01-04 devnull waitgs(&pdf->gs);
102 24c02865 2005-01-04 devnull setdim(&pdf->gs, Rect(0,0,0,0), ppi, 0);
103 17157e4a 2006-03-20 devnull gscmd(&pdf->gs, "(%s) (r) file { DELAYSAFER { .setsafe } if } stopped pop pdfopen begin\n", fn);
104 24c02865 2005-01-04 devnull gscmd(&pdf->gs, "pdfpagecount PAGE==\n");
105 24c02865 2005-01-04 devnull p = Brdline(&pdf->gs.gsrd, '\n');
106 24c02865 2005-01-04 devnull npage = atoi(p);
107 24c02865 2005-01-04 devnull if(npage < 1) {
108 24c02865 2005-01-04 devnull fprint(2, "no pages?\n");
109 24c02865 2005-01-04 devnull return nil;
111 24c02865 2005-01-04 devnull d->npage = npage;
112 24c02865 2005-01-04 devnull d->docname = argv[0];
114 24c02865 2005-01-04 devnull gscmd(&pdf->gs, "Trailer\n");
115 24c02865 2005-01-04 devnull bbox = pdfbbox(&pdf->gs);
117 24c02865 2005-01-04 devnull pdf->pagebbox = emalloc(sizeof(Rectangle)*npage);
118 24c02865 2005-01-04 devnull for(i=0; i<npage; i++) {
119 24c02865 2005-01-04 devnull gscmd(&pdf->gs, "%d pdfgetpage\n", i+1);
120 05a4d855 2007-03-26 devnull pdf->pagebbox[i] = pdfbbox(&pdf->gs);
121 24c02865 2005-01-04 devnull if(Dx(pdf->pagebbox[i]) <= 0)
122 24c02865 2005-01-04 devnull pdf->pagebbox[i] = bbox;
124 24c02865 2005-01-04 devnull return d;
127 24c02865 2005-01-04 devnull static Image*
128 24c02865 2005-01-04 devnull pdfdrawpage(Document *doc, int page)
130 24c02865 2005-01-04 devnull PDFInfo *pdf = doc->extra;
131 24c02865 2005-01-04 devnull Image *im;
133 24c02865 2005-01-04 devnull gscmd(&pdf->gs, "%d DoPDFPage\n", page+1);
134 05a4d855 2007-03-26 devnull im = convert(&pdf->gs.g);
135 24c02865 2005-01-04 devnull if(im == nil) {
136 24c02865 2005-01-04 devnull fprint(2, "fatal: readimage error %r\n");
137 24c02865 2005-01-04 devnull wexits("readimage");
139 24c02865 2005-01-04 devnull waitgs(&pdf->gs);
140 24c02865 2005-01-04 devnull return im;
143 24c02865 2005-01-04 devnull static char*
144 24c02865 2005-01-04 devnull pdfpagename(Document *d, int page)
146 24c02865 2005-01-04 devnull static char str[15];
148 24c02865 2005-01-04 devnull USED(d);
149 24c02865 2005-01-04 devnull sprint(str, "p %d", page+1);
150 24c02865 2005-01-04 devnull return str;
