1 24c02865 2005-01-04 devnull /*
  2 24c02865 2005-01-04 devnull  * graphics file reading for page
  3 24c02865 2005-01-04 devnull  */
  4 24c02865 2005-01-04 devnull 
  5 24c02865 2005-01-04 devnull #include <u.h>
  6 24c02865 2005-01-04 devnull #include <libc.h>
  7 24c02865 2005-01-04 devnull #include <draw.h>
  8 05a4d855 2007-03-26 devnull #include <thread.h>
  9 24c02865 2005-01-04 devnull #include <bio.h>
 10 05a4d855 2007-03-26 devnull #include <cursor.h>
 11 24c02865 2005-01-04 devnull #include "page.h"
 12 24c02865 2005-01-04 devnull 
 13 24c02865 2005-01-04 devnull typedef struct Convert	Convert;
 14 24c02865 2005-01-04 devnull typedef struct GfxInfo	GfxInfo;
 15 24c02865 2005-01-04 devnull 
 16 24c02865 2005-01-04 devnull struct Convert {
 17 24c02865 2005-01-04 devnull 	char *name;
 18 24c02865 2005-01-04 devnull 	char *cmd;
 19 24c02865 2005-01-04 devnull 	char *truecmd;	/* cmd for true color */
 20 24c02865 2005-01-04 devnull };
 21 24c02865 2005-01-04 devnull 
 22 24c02865 2005-01-04 devnull struct GfxInfo {
 23 24c02865 2005-01-04 devnull 	Graphic *g;
 24 24c02865 2005-01-04 devnull };
 25 24c02865 2005-01-04 devnull 
 26 24c02865 2005-01-04 devnull /*
 27 24c02865 2005-01-04 devnull  * N.B. These commands need to read stdin if %a is replaced
 28 24c02865 2005-01-04 devnull  * with an empty string.
 29 24c02865 2005-01-04 devnull  */
 30 24c02865 2005-01-04 devnull Convert cvt[] = {
 31 8f651420 2008-12-02 trisk 	{ "plan9",	"fb/3to1 rgbv %a |fb/pcp -tplan9" },
 32 8f651420 2008-12-02 trisk 	{ "tiff",	"fb/tiff2pic %a | fb/3to1 rgbv | fb/pcp -tplan9" },
 33 8f651420 2008-12-02 trisk 	{ "jpeg",	"jpg -9 %a", "jpg -t9 %a" },
 34 8f651420 2008-12-02 trisk 	{ "gif",	"gif -9 %a", "gif -t9 %a" },
 35 8f651420 2008-12-02 trisk 	{ "inferno",	nil },
 36 8f651420 2008-12-02 trisk 	{ "fax",	"aux/g3p9bit -g %a" },
 37 8f651420 2008-12-02 trisk 	{ "unknown",	"fb/cvt2pic %a |fb/3to1 rgbv" },
 38 e9a2f2e3 2008-12-06 rsc 	{ "plan9bm",	nil },
 39 8f651420 2008-12-02 trisk 	{ "ppm",	"ppm -9 %a", "ppm -t9 %a" },
 40 8f651420 2008-12-02 trisk 	{ "png",	"png -9 %a", "png -t9 %a" },
 41 8f651420 2008-12-02 trisk 	{ "yuv",	"yuv -9 %a", "yuv -t9 %a"  },
 42 8f651420 2008-12-02 trisk 	{ "bmp",	"bmp -9 %a", "bmp -t9 %a"  },
 43 24c02865 2005-01-04 devnull };
 44 24c02865 2005-01-04 devnull 
 45 24c02865 2005-01-04 devnull static Image*	gfxdrawpage(Document *d, int page);
 46 24c02865 2005-01-04 devnull static char*	gfxpagename(Document*, int);
 47 05a4d855 2007-03-26 devnull static int	spawnrc(char*, Graphic*);
 48 05a4d855 2007-03-26 devnull //static void	waitrc(void);
 49 05a4d855 2007-03-26 devnull //static int	spawnpost(int);
 50 24c02865 2005-01-04 devnull static int	addpage(Document*, char*);
 51 24c02865 2005-01-04 devnull static int	rmpage(Document*, int);
 52 24c02865 2005-01-04 devnull static int	genaddpage(Document*, char*, uchar*, int);
 53 24c02865 2005-01-04 devnull 
 54 24c02865 2005-01-04 devnull static char*
 55 24c02865 2005-01-04 devnull gfxpagename(Document *doc, int page)
 56 24c02865 2005-01-04 devnull {
 57 24c02865 2005-01-04 devnull 	GfxInfo *gfx = doc->extra;
 58 24c02865 2005-01-04 devnull 	return gfx->g[page].name;
 59 24c02865 2005-01-04 devnull }
 60 24c02865 2005-01-04 devnull 
 61 24c02865 2005-01-04 devnull static Image*
 62 24c02865 2005-01-04 devnull gfxdrawpage(Document *doc, int page)
 63 24c02865 2005-01-04 devnull {
 64 24c02865 2005-01-04 devnull 	GfxInfo *gfx = doc->extra;
 65 24c02865 2005-01-04 devnull 	return convert(gfx->g+page);
 66 24c02865 2005-01-04 devnull }
 67 24c02865 2005-01-04 devnull 
 68 24c02865 2005-01-04 devnull Document*
 69 24c02865 2005-01-04 devnull initgfx(Biobuf *b, int argc, char **argv, uchar *buf, int nbuf)
 70 24c02865 2005-01-04 devnull {
 71 24c02865 2005-01-04 devnull 	GfxInfo *gfx;
 72 24c02865 2005-01-04 devnull 	Document *doc;
 73 24c02865 2005-01-04 devnull 	int i;
 74 24c02865 2005-01-04 devnull 
 75 24c02865 2005-01-04 devnull 	USED(b);
 76 17157e4a 2006-03-20 devnull 
 77 24c02865 2005-01-04 devnull 	doc = emalloc(sizeof(*doc));
 78 24c02865 2005-01-04 devnull 	gfx = emalloc(sizeof(*gfx));
 79 24c02865 2005-01-04 devnull 	gfx->g = nil;
 80 fa325e9b 2020-01-10 cross 
 81 24c02865 2005-01-04 devnull 	doc->npage = 0;
 82 24c02865 2005-01-04 devnull 	doc->drawpage = gfxdrawpage;
 83 24c02865 2005-01-04 devnull 	doc->pagename = gfxpagename;
 84 24c02865 2005-01-04 devnull 	doc->addpage = addpage;
 85 24c02865 2005-01-04 devnull 	doc->rmpage = rmpage;
 86 24c02865 2005-01-04 devnull 	doc->extra = gfx;
 87 24c02865 2005-01-04 devnull 	doc->fwdonly = 0;
 88 d9e047e5 2010-02-04 rsc 	doc->type = Tgfx;
 89 24c02865 2005-01-04 devnull 
 90 24c02865 2005-01-04 devnull 	fprint(2, "reading through graphics...\n");
 91 24c02865 2005-01-04 devnull 	if(argc==0 && buf)
 92 24c02865 2005-01-04 devnull 		genaddpage(doc, nil, buf, nbuf);
 93 24c02865 2005-01-04 devnull 	else{
 94 24c02865 2005-01-04 devnull 		for(i=0; i<argc; i++)
 95 24c02865 2005-01-04 devnull 			if(addpage(doc, argv[i]) < 0)
 96 24c02865 2005-01-04 devnull 				fprint(2, "warning: not including %s: %r\n", argv[i]);
 97 24c02865 2005-01-04 devnull 	}
 98 24c02865 2005-01-04 devnull 
 99 24c02865 2005-01-04 devnull 	return doc;
100 24c02865 2005-01-04 devnull }
101 24c02865 2005-01-04 devnull 
102 24c02865 2005-01-04 devnull static int
103 24c02865 2005-01-04 devnull genaddpage(Document *doc, char *name, uchar *buf, int nbuf)
104 24c02865 2005-01-04 devnull {
105 24c02865 2005-01-04 devnull 	Graphic *g;
106 24c02865 2005-01-04 devnull 	GfxInfo *gfx;
107 24c02865 2005-01-04 devnull 	Biobuf *b;
108 24c02865 2005-01-04 devnull 	uchar xbuf[32];
109 24c02865 2005-01-04 devnull 	int i, l;
110 24c02865 2005-01-04 devnull 
111 24c02865 2005-01-04 devnull 	l = 0;
112 24c02865 2005-01-04 devnull 	gfx = doc->extra;
113 24c02865 2005-01-04 devnull 
114 24c02865 2005-01-04 devnull 	assert((name == nil) ^ (buf == nil));
115 24c02865 2005-01-04 devnull 	assert(name != nil || doc->npage == 0);
116 24c02865 2005-01-04 devnull 
117 24c02865 2005-01-04 devnull 	for(i=0; i<doc->npage; i++)
118 24c02865 2005-01-04 devnull 		if(strcmp(gfx->g[i].name, name) == 0)
119 24c02865 2005-01-04 devnull 			return i;
120 24c02865 2005-01-04 devnull 
121 24c02865 2005-01-04 devnull 	if(name){
122 24c02865 2005-01-04 devnull 		l = strlen(name);
123 24c02865 2005-01-04 devnull 		if((b = Bopen(name, OREAD)) == nil) {
124 24c02865 2005-01-04 devnull 			werrstr("Bopen: %r");
125 24c02865 2005-01-04 devnull 			return -1;
126 24c02865 2005-01-04 devnull 		}
127 24c02865 2005-01-04 devnull 
128 24c02865 2005-01-04 devnull 		if(Bread(b, xbuf, sizeof xbuf) != sizeof xbuf) {
129 24c02865 2005-01-04 devnull 			werrstr("short read: %r");
130 24c02865 2005-01-04 devnull 			return -1;
131 24c02865 2005-01-04 devnull 		}
132 24c02865 2005-01-04 devnull 		Bterm(b);
133 24c02865 2005-01-04 devnull 		buf = xbuf;
134 24c02865 2005-01-04 devnull 		nbuf = sizeof xbuf;
135 24c02865 2005-01-04 devnull 	}
136 24c02865 2005-01-04 devnull 
137 24c02865 2005-01-04 devnull 
138 24c02865 2005-01-04 devnull 	gfx->g = erealloc(gfx->g, (doc->npage+1)*(sizeof(*gfx->g)));
139 24c02865 2005-01-04 devnull 	g = &gfx->g[doc->npage];
140 24c02865 2005-01-04 devnull 
141 24c02865 2005-01-04 devnull 	memset(g, 0, sizeof *g);
142 24c02865 2005-01-04 devnull 	if(memcmp(buf, "GIF", 3) == 0)
143 24c02865 2005-01-04 devnull 		g->type = Igif;
144 fa325e9b 2020-01-10 cross 	else if(memcmp(buf, "\111\111\052\000", 4) == 0)
145 24c02865 2005-01-04 devnull 		g->type = Itiff;
146 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "\115\115\000\052", 4) == 0)
147 24c02865 2005-01-04 devnull 		g->type = Itiff;
148 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "\377\330\377", 3) == 0)
149 24c02865 2005-01-04 devnull 		g->type = Ijpeg;
150 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "\211PNG\r\n\032\n", 3) == 0)
151 24c02865 2005-01-04 devnull 		g->type = Ipng;
152 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "compressed\n", 11) == 0)
153 24c02865 2005-01-04 devnull 		g->type = Iinferno;
154 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "\0PC Research, Inc", 17) == 0)
155 24c02865 2005-01-04 devnull 		g->type = Ifax;
156 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "TYPE=ccitt-g31", 14) == 0)
157 24c02865 2005-01-04 devnull 		g->type = Ifax;
158 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "II*", 3) == 0)
159 24c02865 2005-01-04 devnull 		g->type = Ifax;
160 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "TYPE=", 5) == 0)
161 24c02865 2005-01-04 devnull 		g->type = Ipic;
162 24c02865 2005-01-04 devnull 	else if(buf[0] == 'P' && '0' <= buf[1] && buf[1] <= '9')
163 24c02865 2005-01-04 devnull 		g->type = Ippm;
164 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "BM", 2) == 0)
165 24c02865 2005-01-04 devnull 		g->type = Ibmp;
166 24c02865 2005-01-04 devnull 	else if(memcmp(buf, "          ", 10) == 0 &&
167 24c02865 2005-01-04 devnull 		'0' <= buf[10] && buf[10] <= '9' &&
168 24c02865 2005-01-04 devnull 		buf[11] == ' ')
169 24c02865 2005-01-04 devnull 		g->type = Iplan9bm;
170 24c02865 2005-01-04 devnull 	else if(strtochan((char*)buf) != 0)
171 24c02865 2005-01-04 devnull 		g->type = Iplan9bm;
172 24c02865 2005-01-04 devnull 	else if (l > 4 && strcmp(name + l -4, ".yuv") == 0)
173 24c02865 2005-01-04 devnull 		g->type = Iyuv;
174 24c02865 2005-01-04 devnull 	else
175 24c02865 2005-01-04 devnull 		g->type = Icvt2pic;
176 24c02865 2005-01-04 devnull 
177 05a4d855 2007-03-26 devnull 	if(name){
178 24c02865 2005-01-04 devnull 		g->name = estrdup(name);
179 05a4d855 2007-03-26 devnull 		g->fd = -1;
180 05a4d855 2007-03-26 devnull 	}else{
181 24c02865 2005-01-04 devnull 		g->name = estrdup("stdin");	/* so it can be freed */
182 05a4d855 2007-03-26 devnull 		g->fd = stdinpipe(buf, nbuf);
183 24c02865 2005-01-04 devnull 	}
184 24c02865 2005-01-04 devnull 
185 24c02865 2005-01-04 devnull 	if(chatty) fprint(2, "classified \"%s\" as \"%s\"\n", g->name, cvt[g->type].name);
186 24c02865 2005-01-04 devnull 	return doc->npage++;
187 24c02865 2005-01-04 devnull }
188 24c02865 2005-01-04 devnull 
189 fa325e9b 2020-01-10 cross static int
190 24c02865 2005-01-04 devnull addpage(Document *doc, char *name)
191 24c02865 2005-01-04 devnull {
192 24c02865 2005-01-04 devnull 	return genaddpage(doc, name, nil, 0);
193 24c02865 2005-01-04 devnull }
194 24c02865 2005-01-04 devnull 
195 24c02865 2005-01-04 devnull static int
196 24c02865 2005-01-04 devnull rmpage(Document *doc, int n)
197 24c02865 2005-01-04 devnull {
198 24c02865 2005-01-04 devnull 	int i;
199 24c02865 2005-01-04 devnull 	GfxInfo *gfx;
200 24c02865 2005-01-04 devnull 
201 24c02865 2005-01-04 devnull 	if(n < 0 || n >= doc->npage)
202 24c02865 2005-01-04 devnull 		return -1;
203 24c02865 2005-01-04 devnull 
204 24c02865 2005-01-04 devnull 	gfx = doc->extra;
205 24c02865 2005-01-04 devnull 	doc->npage--;
206 24c02865 2005-01-04 devnull 	free(gfx->g[n].name);
207 24c02865 2005-01-04 devnull 
208 24c02865 2005-01-04 devnull 	for(i=n; i<doc->npage; i++)
209 24c02865 2005-01-04 devnull 		gfx->g[i] = gfx->g[i+1];
210 24c02865 2005-01-04 devnull 
211 24c02865 2005-01-04 devnull 	if(n < doc->npage)
212 24c02865 2005-01-04 devnull 		return n;
213 24c02865 2005-01-04 devnull 	if(n == 0)
214 24c02865 2005-01-04 devnull 		return 0;
215 24c02865 2005-01-04 devnull 	return n-1;
216 24c02865 2005-01-04 devnull }
217 24c02865 2005-01-04 devnull 
218 24c02865 2005-01-04 devnull 
219 05a4d855 2007-03-26 devnull Image*
220 24c02865 2005-01-04 devnull convert(Graphic *g)
221 24c02865 2005-01-04 devnull {
222 24c02865 2005-01-04 devnull 	int fd;
223 24c02865 2005-01-04 devnull 	Convert c;
224 24c02865 2005-01-04 devnull 	char *cmd;
225 24c02865 2005-01-04 devnull 	char *name, buf[1000];
226 24c02865 2005-01-04 devnull 	Image *im;
227 24c02865 2005-01-04 devnull 	int rcspawned = 0;
228 24c02865 2005-01-04 devnull 
229 24c02865 2005-01-04 devnull 	c = cvt[g->type];
230 24c02865 2005-01-04 devnull 	if(c.cmd == nil) {
231 24c02865 2005-01-04 devnull 		if(chatty) fprint(2, "no conversion for bitmap \"%s\"...\n", g->name);
232 05a4d855 2007-03-26 devnull 		if(g->fd < 0){	/* not stdin */
233 24c02865 2005-01-04 devnull 			fd = open(g->name, OREAD);
234 24c02865 2005-01-04 devnull 			if(fd < 0) {
235 24c02865 2005-01-04 devnull 				fprint(2, "cannot open file: %r\n");
236 24c02865 2005-01-04 devnull 				wexits("open");
237 24c02865 2005-01-04 devnull 			}
238 24c02865 2005-01-04 devnull 		}else
239 05a4d855 2007-03-26 devnull 			fd = g->fd;
240 24c02865 2005-01-04 devnull 	} else {
241 24c02865 2005-01-04 devnull 		cmd = c.cmd;
242 24c02865 2005-01-04 devnull 		if(truecolor && c.truecmd)
243 24c02865 2005-01-04 devnull 			cmd = c.truecmd;
244 24c02865 2005-01-04 devnull 
245 05a4d855 2007-03-26 devnull 		if(g->fd >= 0)	/* is pipe */
246 24c02865 2005-01-04 devnull 			name = "";
247 24c02865 2005-01-04 devnull 		else
248 24c02865 2005-01-04 devnull 			name = g->name;
249 24c02865 2005-01-04 devnull 		if(strlen(cmd)+strlen(name) > sizeof buf) {
250 24c02865 2005-01-04 devnull 			fprint(2, "command too long\n");
251 24c02865 2005-01-04 devnull 			wexits("convert");
252 24c02865 2005-01-04 devnull 		}
253 24c02865 2005-01-04 devnull 		snprint(buf, sizeof buf, cmd, name);
254 24c02865 2005-01-04 devnull 		if(chatty) fprint(2, "using \"%s\" to convert \"%s\"...\n", buf, g->name);
255 05a4d855 2007-03-26 devnull 		fd = spawnrc(buf, g);
256 24c02865 2005-01-04 devnull 		rcspawned++;
257 24c02865 2005-01-04 devnull 		if(fd < 0) {
258 24c02865 2005-01-04 devnull 			fprint(2, "cannot spawn converter: %r\n");
259 24c02865 2005-01-04 devnull 			wexits("convert");
260 fa325e9b 2020-01-10 cross 		}
261 24c02865 2005-01-04 devnull 	}
262 24c02865 2005-01-04 devnull 
263 24c02865 2005-01-04 devnull 	im = readimage(display, fd, 0);
264 24c02865 2005-01-04 devnull 	if(im == nil) {
265 24c02865 2005-01-04 devnull 		fprint(2, "warning: couldn't read image: %r\n");
266 24c02865 2005-01-04 devnull 	}
267 24c02865 2005-01-04 devnull 
268 05a4d855 2007-03-26 devnull 	close(fd);
269 24c02865 2005-01-04 devnull 	return im;
270 24c02865 2005-01-04 devnull }
271 24c02865 2005-01-04 devnull 
272 24c02865 2005-01-04 devnull static int
273 05a4d855 2007-03-26 devnull spawnrc(char *cmd, Graphic *g)
274 24c02865 2005-01-04 devnull {
275 24c02865 2005-01-04 devnull 	int pfd[2];
276 05a4d855 2007-03-26 devnull 	int fd[3];
277 24c02865 2005-01-04 devnull 
278 24c02865 2005-01-04 devnull 	if(chatty) fprint(2, "spawning(%s)...", cmd);
279 24c02865 2005-01-04 devnull 
280 24c02865 2005-01-04 devnull 	if(pipe(pfd) < 0)
281 24c02865 2005-01-04 devnull 		return -1;
282 05a4d855 2007-03-26 devnull 
283 05a4d855 2007-03-26 devnull 	if(g->fd > 0)
284 05a4d855 2007-03-26 devnull 		fd[0] = dup(g->fd, -1);
285 05a4d855 2007-03-26 devnull 	else
286 05a4d855 2007-03-26 devnull 		fd[0] = open("/dev/null", OREAD);
287 05a4d855 2007-03-26 devnull 	fd[1] = pfd[1];
288 05a4d855 2007-03-26 devnull 	fd[2] = dup(2, -1);
289 05a4d855 2007-03-26 devnull 
290 05a4d855 2007-03-26 devnull 	if(threadspawnl(fd, "rc", "rc", "-c", cmd, nil) == -1)
291 24c02865 2005-01-04 devnull 		return -1;
292 24c02865 2005-01-04 devnull 
293 05a4d855 2007-03-26 devnull 	return pfd[0];
294 24c02865 2005-01-04 devnull }