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