1 24c02865 2005-01-04 devnull /*
  2 24c02865 2005-01-04 devnull  * pdf.c
  3 24c02865 2005-01-04 devnull  * 
  4 24c02865 2005-01-04 devnull  * pdf file support for page
  5 24c02865 2005-01-04 devnull  */
  6 24c02865 2005-01-04 devnull 
  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 24c02865 2005-01-04 devnull #include <event.h>
 11 24c02865 2005-01-04 devnull #include <bio.h>
 12 24c02865 2005-01-04 devnull #include "page.h"
 13 24c02865 2005-01-04 devnull 
 14 24c02865 2005-01-04 devnull typedef struct PDFInfo	PDFInfo;
 15 24c02865 2005-01-04 devnull struct PDFInfo {
 16 24c02865 2005-01-04 devnull 	GSInfo gs;
 17 24c02865 2005-01-04 devnull 	Rectangle *pagebbox;
 18 24c02865 2005-01-04 devnull };
 19 24c02865 2005-01-04 devnull 
 20 24c02865 2005-01-04 devnull static Image*	pdfdrawpage(Document *d, int page);
 21 24c02865 2005-01-04 devnull static char*	pdfpagename(Document*, int);
 22 24c02865 2005-01-04 devnull 
 23 24c02865 2005-01-04 devnull char *pdfprolog = 
 24 24c02865 2005-01-04 devnull #include "pdfprolog.c"
 25 24c02865 2005-01-04 devnull 	;
 26 24c02865 2005-01-04 devnull 
 27 24c02865 2005-01-04 devnull Rectangle
 28 24c02865 2005-01-04 devnull pdfbbox(GSInfo *gs)
 29 24c02865 2005-01-04 devnull {
 30 24c02865 2005-01-04 devnull 	char *p;
 31 24c02865 2005-01-04 devnull 	char *f[4];
 32 24c02865 2005-01-04 devnull 	Rectangle r;
 33 24c02865 2005-01-04 devnull 	
 34 24c02865 2005-01-04 devnull 	r = Rect(0,0,0,0);
 35 24c02865 2005-01-04 devnull 	waitgs(gs);
 36 24c02865 2005-01-04 devnull 	gscmd(gs, "/CropBox knownoget {} {[0 0 0 0]} ifelse PAGE==\n");
 37 24c02865 2005-01-04 devnull 	p = Brdline(&gs->gsrd, '\n');
 38 24c02865 2005-01-04 devnull 	p[Blinelen(&gs->gsrd)-1] ='\0';
 39 24c02865 2005-01-04 devnull 	if(p[0] != '[')
 40 24c02865 2005-01-04 devnull 		return r;
 41 24c02865 2005-01-04 devnull 	if(tokenize(p+1, f, 4) != 4)
 42 24c02865 2005-01-04 devnull 		return r;
 43 24c02865 2005-01-04 devnull 	r = Rect(atoi(f[0]), atoi(f[1]), atoi(f[2]), atoi(f[3]));
 44 24c02865 2005-01-04 devnull 	waitgs(gs);
 45 24c02865 2005-01-04 devnull 	return r;
 46 24c02865 2005-01-04 devnull }
 47 24c02865 2005-01-04 devnull 
 48 24c02865 2005-01-04 devnull Document*
 49 24c02865 2005-01-04 devnull initpdf(Biobuf *b, int argc, char **argv, uchar *buf, int nbuf)
 50 24c02865 2005-01-04 devnull {
 51 24c02865 2005-01-04 devnull 	Document *d;
 52 24c02865 2005-01-04 devnull 	PDFInfo *pdf;
 53 24c02865 2005-01-04 devnull 	char *p;
 54 24c02865 2005-01-04 devnull 	char *fn;
 55 24c02865 2005-01-04 devnull 	char fdbuf[20];
 56 24c02865 2005-01-04 devnull 	int fd;
 57 24c02865 2005-01-04 devnull 	int i, npage;
 58 24c02865 2005-01-04 devnull 	Rectangle bbox;
 59 24c02865 2005-01-04 devnull 
 60 24c02865 2005-01-04 devnull 	if(argc > 1) {
 61 24c02865 2005-01-04 devnull 		fprint(2, "can only view one pdf file at a time\n");
 62 24c02865 2005-01-04 devnull 		return nil;
 63 24c02865 2005-01-04 devnull 	}
 64 24c02865 2005-01-04 devnull 
 65 24c02865 2005-01-04 devnull 	fprint(2, "reading through pdf...\n");
 66 24c02865 2005-01-04 devnull 	if(b == nil){	/* standard input; spool to disk (ouch) */
 67 24c02865 2005-01-04 devnull 		fd = spooltodisk(buf, nbuf, &fn);
 68 24c02865 2005-01-04 devnull 		sprint(fdbuf, "/fd/%d", fd);
 69 24c02865 2005-01-04 devnull 		b = Bopen(fdbuf, OREAD);
 70 24c02865 2005-01-04 devnull 		if(b == nil){
 71 24c02865 2005-01-04 devnull 			fprint(2, "cannot open disk spool file\n");
 72 24c02865 2005-01-04 devnull 			wexits("Bopen temp");
 73 24c02865 2005-01-04 devnull 		}
 74 24c02865 2005-01-04 devnull 	}else
 75 24c02865 2005-01-04 devnull 		fn = argv[0];
 76 24c02865 2005-01-04 devnull 
 77 24c02865 2005-01-04 devnull 	/* sanity check */
 78 24c02865 2005-01-04 devnull 	Bseek(b, 0, 0);
 79 24c02865 2005-01-04 devnull 	if(!(p = Brdline(b, '\n')) && !(p = Brdline(b, '\r'))) {
 80 24c02865 2005-01-04 devnull 		fprint(2, "cannot find end of first line\n");
 81 24c02865 2005-01-04 devnull 		wexits("initps");
 82 24c02865 2005-01-04 devnull 	}
 83 24c02865 2005-01-04 devnull 	if(strncmp(p, "%PDF-", 5) != 0) {
 84 24c02865 2005-01-04 devnull 		werrstr("not pdf");
 85 24c02865 2005-01-04 devnull 		return nil;
 86 24c02865 2005-01-04 devnull 	}
 87 24c02865 2005-01-04 devnull 
 88 24c02865 2005-01-04 devnull 	/* setup structures so one free suffices */
 89 24c02865 2005-01-04 devnull 	p = emalloc(sizeof(*d) + sizeof(*pdf));
 90 24c02865 2005-01-04 devnull 	d = (Document*) p;
 91 24c02865 2005-01-04 devnull 	p += sizeof(*d);
 92 24c02865 2005-01-04 devnull 	pdf = (PDFInfo*) p;
 93 24c02865 2005-01-04 devnull 
 94 24c02865 2005-01-04 devnull 	d->extra = pdf;
 95 24c02865 2005-01-04 devnull 	d->b = b;
 96 24c02865 2005-01-04 devnull 	d->drawpage = pdfdrawpage;
 97 24c02865 2005-01-04 devnull 	d->pagename = pdfpagename;
 98 24c02865 2005-01-04 devnull 	d->fwdonly = 0;
 99 24c02865 2005-01-04 devnull 
100 24c02865 2005-01-04 devnull 	if(spawngs(&pdf->gs) < 0)
101 24c02865 2005-01-04 devnull 		return nil;
102 24c02865 2005-01-04 devnull 
103 24c02865 2005-01-04 devnull 	gscmd(&pdf->gs, "%s", pdfprolog);
104 24c02865 2005-01-04 devnull 	waitgs(&pdf->gs);
105 24c02865 2005-01-04 devnull 
106 24c02865 2005-01-04 devnull 	setdim(&pdf->gs, Rect(0,0,0,0), ppi, 0);
107 24c02865 2005-01-04 devnull 	gscmd(&pdf->gs, "(%s) (r) file pdfopen begin\n", fn);
108 24c02865 2005-01-04 devnull 	gscmd(&pdf->gs, "pdfpagecount PAGE==\n");
109 24c02865 2005-01-04 devnull 	p = Brdline(&pdf->gs.gsrd, '\n');
110 24c02865 2005-01-04 devnull 	npage = atoi(p);
111 24c02865 2005-01-04 devnull 	if(npage < 1) {
112 24c02865 2005-01-04 devnull 		fprint(2, "no pages?\n");
113 24c02865 2005-01-04 devnull 		return nil;
114 24c02865 2005-01-04 devnull 	}
115 24c02865 2005-01-04 devnull 	d->npage = npage;
116 24c02865 2005-01-04 devnull 	d->docname = argv[0];
117 24c02865 2005-01-04 devnull 
118 24c02865 2005-01-04 devnull 	gscmd(&pdf->gs, "Trailer\n");
119 24c02865 2005-01-04 devnull 	bbox = pdfbbox(&pdf->gs);
120 24c02865 2005-01-04 devnull 
121 24c02865 2005-01-04 devnull 	pdf->pagebbox = emalloc(sizeof(Rectangle)*npage);
122 24c02865 2005-01-04 devnull 	for(i=0; i<npage; i++) {
123 24c02865 2005-01-04 devnull 		gscmd(&pdf->gs, "%d pdfgetpage\n", i+1);
124 24c02865 2005-01-04 devnull 		pdf->pagebbox[i] = pdfbbox(&pdf->gs);
125 24c02865 2005-01-04 devnull 		if(Dx(pdf->pagebbox[i]) <= 0)
126 24c02865 2005-01-04 devnull 			pdf->pagebbox[i] = bbox;
127 24c02865 2005-01-04 devnull 	}
128 24c02865 2005-01-04 devnull 
129 24c02865 2005-01-04 devnull 	return d;
130 24c02865 2005-01-04 devnull }
131 24c02865 2005-01-04 devnull 
132 24c02865 2005-01-04 devnull static Image*
133 24c02865 2005-01-04 devnull pdfdrawpage(Document *doc, int page)
134 24c02865 2005-01-04 devnull {
135 24c02865 2005-01-04 devnull 	PDFInfo *pdf = doc->extra;
136 24c02865 2005-01-04 devnull 	Image *im;
137 24c02865 2005-01-04 devnull 
138 24c02865 2005-01-04 devnull 	gscmd(&pdf->gs, "%d DoPDFPage\n", page+1);
139 24c02865 2005-01-04 devnull 	im = readimage(display, pdf->gs.gsdfd, 0);
140 24c02865 2005-01-04 devnull 	if(im == nil) {
141 24c02865 2005-01-04 devnull 		fprint(2, "fatal: readimage error %r\n");
142 24c02865 2005-01-04 devnull 		wexits("readimage");
143 24c02865 2005-01-04 devnull 	}
144 24c02865 2005-01-04 devnull 	waitgs(&pdf->gs);
145 24c02865 2005-01-04 devnull 	return im;
146 24c02865 2005-01-04 devnull }
147 24c02865 2005-01-04 devnull 
148 24c02865 2005-01-04 devnull static char*
149 24c02865 2005-01-04 devnull pdfpagename(Document *d, int page)
150 24c02865 2005-01-04 devnull {
151 24c02865 2005-01-04 devnull 	static char str[15];
152 24c02865 2005-01-04 devnull 	USED(d);
153 24c02865 2005-01-04 devnull 	sprint(str, "p %d", page+1);
154 24c02865 2005-01-04 devnull 	return str;
155 24c02865 2005-01-04 devnull }