2 24c02865 2005-01-04 devnull * graphics file reading for page
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"
13 24c02865 2005-01-04 devnull typedef struct Convert Convert;
14 24c02865 2005-01-04 devnull typedef struct GfxInfo GfxInfo;
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 */
22 24c02865 2005-01-04 devnull struct GfxInfo {
23 24c02865 2005-01-04 devnull Graphic *g;
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.
30 24c02865 2005-01-04 devnull Convert cvt[] = {
31 24c02865 2005-01-04 devnull [Ipic] { "plan9", "fb/3to1 rgbv %a |fb/pcp -tplan9" },
32 24c02865 2005-01-04 devnull [Itiff] { "tiff", "fb/tiff2pic %a | fb/3to1 rgbv | fb/pcp -tplan9" },
33 24c02865 2005-01-04 devnull [Iplan9bm] { "plan9bm", nil },
34 24c02865 2005-01-04 devnull [Ijpeg] { "jpeg", "jpg -9 %a", "jpg -t9 %a" },
35 24c02865 2005-01-04 devnull [Igif] { "gif", "gif -9 %a", "gif -t9 %a" },
36 24c02865 2005-01-04 devnull [Iinferno] { "inferno", nil },
37 24c02865 2005-01-04 devnull [Ifax] { "fax", "aux/g3p9bit -g %a" },
38 24c02865 2005-01-04 devnull [Icvt2pic] { "unknown", "fb/cvt2pic %a |fb/3to1 rgbv" },
39 24c02865 2005-01-04 devnull [Ippm] { "ppm", "ppm -9 %a", "ppm -t9 %a" },
40 24c02865 2005-01-04 devnull /* ``temporary'' hack for hobby */
41 24c02865 2005-01-04 devnull [Iccittg4] { "ccitt-g4", "cat %a|rx nslocum /usr/lib/ocr/bin/bcp -M|fb/pcp -tcompressed -l0" },
42 24c02865 2005-01-04 devnull [Ipng] { "png", "png -9 %a", "png -t9 %a" },
43 24c02865 2005-01-04 devnull [Iyuv] { "yuv", "yuv -9 %a", "yuv -t9 %a" },
44 05a4d855 2007-03-26 devnull [Ibmp] { "bmp", "bmp -9 %a", "bmp -t9 %a" },
47 24c02865 2005-01-04 devnull static Image* gfxdrawpage(Document *d, int page);
48 24c02865 2005-01-04 devnull static char* gfxpagename(Document*, int);
49 05a4d855 2007-03-26 devnull static int spawnrc(char*, Graphic*);
50 05a4d855 2007-03-26 devnull //static void waitrc(void);
51 05a4d855 2007-03-26 devnull //static int spawnpost(int);
52 24c02865 2005-01-04 devnull static int addpage(Document*, char*);
53 24c02865 2005-01-04 devnull static int rmpage(Document*, int);
54 24c02865 2005-01-04 devnull static int genaddpage(Document*, char*, uchar*, int);
56 24c02865 2005-01-04 devnull static char*
57 24c02865 2005-01-04 devnull gfxpagename(Document *doc, int page)
59 24c02865 2005-01-04 devnull GfxInfo *gfx = doc->extra;
60 24c02865 2005-01-04 devnull return gfx->g[page].name;
63 24c02865 2005-01-04 devnull static Image*
64 24c02865 2005-01-04 devnull gfxdrawpage(Document *doc, int page)
66 24c02865 2005-01-04 devnull GfxInfo *gfx = doc->extra;
67 24c02865 2005-01-04 devnull return convert(gfx->g+page);
70 24c02865 2005-01-04 devnull Document*
71 24c02865 2005-01-04 devnull initgfx(Biobuf *b, int argc, char **argv, uchar *buf, int nbuf)
73 24c02865 2005-01-04 devnull GfxInfo *gfx;
74 24c02865 2005-01-04 devnull Document *doc;
79 24c02865 2005-01-04 devnull doc = emalloc(sizeof(*doc));
80 24c02865 2005-01-04 devnull gfx = emalloc(sizeof(*gfx));
81 24c02865 2005-01-04 devnull gfx->g = nil;
83 24c02865 2005-01-04 devnull doc->npage = 0;
84 24c02865 2005-01-04 devnull doc->drawpage = gfxdrawpage;
85 24c02865 2005-01-04 devnull doc->pagename = gfxpagename;
86 24c02865 2005-01-04 devnull doc->addpage = addpage;
87 24c02865 2005-01-04 devnull doc->rmpage = rmpage;
88 24c02865 2005-01-04 devnull doc->extra = gfx;
89 24c02865 2005-01-04 devnull doc->fwdonly = 0;
91 24c02865 2005-01-04 devnull fprint(2, "reading through graphics...\n");
92 24c02865 2005-01-04 devnull if(argc==0 && buf)
93 24c02865 2005-01-04 devnull genaddpage(doc, nil, buf, nbuf);
95 24c02865 2005-01-04 devnull for(i=0; i<argc; i++)
96 24c02865 2005-01-04 devnull if(addpage(doc, argv[i]) < 0)
97 24c02865 2005-01-04 devnull fprint(2, "warning: not including %s: %r\n", argv[i]);
100 24c02865 2005-01-04 devnull return doc;
103 24c02865 2005-01-04 devnull static int
104 24c02865 2005-01-04 devnull genaddpage(Document *doc, char *name, uchar *buf, int nbuf)
106 24c02865 2005-01-04 devnull Graphic *g;
107 24c02865 2005-01-04 devnull GfxInfo *gfx;
108 24c02865 2005-01-04 devnull Biobuf *b;
109 24c02865 2005-01-04 devnull uchar xbuf[32];
110 24c02865 2005-01-04 devnull int i, l;
113 24c02865 2005-01-04 devnull gfx = doc->extra;
115 24c02865 2005-01-04 devnull assert((name == nil) ^ (buf == nil));
116 24c02865 2005-01-04 devnull assert(name != nil || doc->npage == 0);
118 24c02865 2005-01-04 devnull for(i=0; i<doc->npage; i++)
119 24c02865 2005-01-04 devnull if(strcmp(gfx->g[i].name, name) == 0)
120 24c02865 2005-01-04 devnull return i;
122 24c02865 2005-01-04 devnull if(name){
123 24c02865 2005-01-04 devnull l = strlen(name);
124 24c02865 2005-01-04 devnull if((b = Bopen(name, OREAD)) == nil) {
125 24c02865 2005-01-04 devnull werrstr("Bopen: %r");
126 24c02865 2005-01-04 devnull return -1;
129 24c02865 2005-01-04 devnull if(Bread(b, xbuf, sizeof xbuf) != sizeof xbuf) {
130 24c02865 2005-01-04 devnull werrstr("short read: %r");
131 24c02865 2005-01-04 devnull return -1;
133 24c02865 2005-01-04 devnull Bterm(b);
134 24c02865 2005-01-04 devnull buf = xbuf;
135 24c02865 2005-01-04 devnull nbuf = sizeof xbuf;
139 24c02865 2005-01-04 devnull gfx->g = erealloc(gfx->g, (doc->npage+1)*(sizeof(*gfx->g)));
140 24c02865 2005-01-04 devnull g = &gfx->g[doc->npage];
142 24c02865 2005-01-04 devnull memset(g, 0, sizeof *g);
143 24c02865 2005-01-04 devnull if(memcmp(buf, "GIF", 3) == 0)
144 24c02865 2005-01-04 devnull g->type = Igif;
145 24c02865 2005-01-04 devnull else if(memcmp(buf, "\111\111\052\000", 4) == 0)
146 24c02865 2005-01-04 devnull g->type = Itiff;
147 24c02865 2005-01-04 devnull else if(memcmp(buf, "\115\115\000\052", 4) == 0)
148 24c02865 2005-01-04 devnull g->type = Itiff;
149 24c02865 2005-01-04 devnull else if(memcmp(buf, "\377\330\377", 3) == 0)
150 24c02865 2005-01-04 devnull g->type = Ijpeg;
151 24c02865 2005-01-04 devnull else if(memcmp(buf, "\211PNG\r\n\032\n", 3) == 0)
152 24c02865 2005-01-04 devnull g->type = Ipng;
153 24c02865 2005-01-04 devnull else if(memcmp(buf, "compressed\n", 11) == 0)
154 24c02865 2005-01-04 devnull g->type = Iinferno;
155 24c02865 2005-01-04 devnull else if(memcmp(buf, "\0PC Research, Inc", 17) == 0)
156 24c02865 2005-01-04 devnull g->type = Ifax;
157 24c02865 2005-01-04 devnull else if(memcmp(buf, "TYPE=ccitt-g31", 14) == 0)
158 24c02865 2005-01-04 devnull g->type = Ifax;
159 24c02865 2005-01-04 devnull else if(memcmp(buf, "II*", 3) == 0)
160 24c02865 2005-01-04 devnull g->type = Ifax;
161 24c02865 2005-01-04 devnull else if(memcmp(buf, "TYPE=ccitt-g4", 13) == 0)
162 24c02865 2005-01-04 devnull g->type = Iccittg4;
163 24c02865 2005-01-04 devnull else if(memcmp(buf, "TYPE=", 5) == 0)
164 24c02865 2005-01-04 devnull g->type = Ipic;
165 24c02865 2005-01-04 devnull else if(buf[0] == 'P' && '0' <= buf[1] && buf[1] <= '9')
166 24c02865 2005-01-04 devnull g->type = Ippm;
167 24c02865 2005-01-04 devnull else if(memcmp(buf, "BM", 2) == 0)
168 24c02865 2005-01-04 devnull g->type = Ibmp;
169 24c02865 2005-01-04 devnull else if(memcmp(buf, " ", 10) == 0 &&
170 24c02865 2005-01-04 devnull '0' <= buf[10] && buf[10] <= '9' &&
171 24c02865 2005-01-04 devnull buf[11] == ' ')
172 24c02865 2005-01-04 devnull g->type = Iplan9bm;
173 24c02865 2005-01-04 devnull else if(strtochan((char*)buf) != 0)
174 24c02865 2005-01-04 devnull g->type = Iplan9bm;
175 24c02865 2005-01-04 devnull else if (l > 4 && strcmp(name + l -4, ".yuv") == 0)
176 24c02865 2005-01-04 devnull g->type = Iyuv;
178 24c02865 2005-01-04 devnull g->type = Icvt2pic;
180 05a4d855 2007-03-26 devnull if(name){
181 24c02865 2005-01-04 devnull g->name = estrdup(name);
182 05a4d855 2007-03-26 devnull g->fd = -1;
184 24c02865 2005-01-04 devnull g->name = estrdup("stdin"); /* so it can be freed */
185 05a4d855 2007-03-26 devnull g->fd = stdinpipe(buf, nbuf);
188 24c02865 2005-01-04 devnull if(chatty) fprint(2, "classified \"%s\" as \"%s\"\n", g->name, cvt[g->type].name);
189 24c02865 2005-01-04 devnull return doc->npage++;
192 24c02865 2005-01-04 devnull static int
193 24c02865 2005-01-04 devnull addpage(Document *doc, char *name)
195 24c02865 2005-01-04 devnull return genaddpage(doc, name, nil, 0);
198 24c02865 2005-01-04 devnull static int
199 24c02865 2005-01-04 devnull rmpage(Document *doc, int n)
202 24c02865 2005-01-04 devnull GfxInfo *gfx;
204 24c02865 2005-01-04 devnull if(n < 0 || n >= doc->npage)
205 24c02865 2005-01-04 devnull return -1;
207 24c02865 2005-01-04 devnull gfx = doc->extra;
208 24c02865 2005-01-04 devnull doc->npage--;
209 24c02865 2005-01-04 devnull free(gfx->g[n].name);
211 24c02865 2005-01-04 devnull for(i=n; i<doc->npage; i++)
212 24c02865 2005-01-04 devnull gfx->g[i] = gfx->g[i+1];
214 24c02865 2005-01-04 devnull if(n < doc->npage)
215 24c02865 2005-01-04 devnull return n;
216 24c02865 2005-01-04 devnull if(n == 0)
217 24c02865 2005-01-04 devnull return 0;
218 24c02865 2005-01-04 devnull return n-1;
223 24c02865 2005-01-04 devnull convert(Graphic *g)
226 24c02865 2005-01-04 devnull Convert c;
227 24c02865 2005-01-04 devnull char *cmd;
228 24c02865 2005-01-04 devnull char *name, buf[1000];
229 24c02865 2005-01-04 devnull Image *im;
230 24c02865 2005-01-04 devnull int rcspawned = 0;
232 24c02865 2005-01-04 devnull c = cvt[g->type];
233 24c02865 2005-01-04 devnull if(c.cmd == nil) {
234 24c02865 2005-01-04 devnull if(chatty) fprint(2, "no conversion for bitmap \"%s\"...\n", g->name);
235 05a4d855 2007-03-26 devnull if(g->fd < 0){ /* not stdin */
236 24c02865 2005-01-04 devnull fd = open(g->name, OREAD);
237 24c02865 2005-01-04 devnull if(fd < 0) {
238 24c02865 2005-01-04 devnull fprint(2, "cannot open file: %r\n");
239 24c02865 2005-01-04 devnull wexits("open");
242 05a4d855 2007-03-26 devnull fd = g->fd;
243 24c02865 2005-01-04 devnull } else {
244 24c02865 2005-01-04 devnull cmd = c.cmd;
245 24c02865 2005-01-04 devnull if(truecolor && c.truecmd)
246 24c02865 2005-01-04 devnull cmd = c.truecmd;
248 05a4d855 2007-03-26 devnull if(g->fd >= 0) /* is pipe */
249 24c02865 2005-01-04 devnull name = "";
251 24c02865 2005-01-04 devnull name = g->name;
252 24c02865 2005-01-04 devnull if(strlen(cmd)+strlen(name) > sizeof buf) {
253 24c02865 2005-01-04 devnull fprint(2, "command too long\n");
254 24c02865 2005-01-04 devnull wexits("convert");
256 24c02865 2005-01-04 devnull snprint(buf, sizeof buf, cmd, name);
257 24c02865 2005-01-04 devnull if(chatty) fprint(2, "using \"%s\" to convert \"%s\"...\n", buf, g->name);
258 05a4d855 2007-03-26 devnull fd = spawnrc(buf, g);
259 24c02865 2005-01-04 devnull rcspawned++;
260 24c02865 2005-01-04 devnull if(fd < 0) {
261 24c02865 2005-01-04 devnull fprint(2, "cannot spawn converter: %r\n");
262 24c02865 2005-01-04 devnull wexits("convert");
266 24c02865 2005-01-04 devnull im = readimage(display, fd, 0);
267 24c02865 2005-01-04 devnull if(im == nil) {
268 24c02865 2005-01-04 devnull fprint(2, "warning: couldn't read image: %r\n");
271 05a4d855 2007-03-26 devnull close(fd);
272 24c02865 2005-01-04 devnull return im;
275 24c02865 2005-01-04 devnull static int
276 05a4d855 2007-03-26 devnull spawnrc(char *cmd, Graphic *g)
278 24c02865 2005-01-04 devnull int pfd[2];
279 05a4d855 2007-03-26 devnull int fd[3];
281 24c02865 2005-01-04 devnull if(chatty) fprint(2, "spawning(%s)...", cmd);
283 24c02865 2005-01-04 devnull if(pipe(pfd) < 0)
284 24c02865 2005-01-04 devnull return -1;
286 05a4d855 2007-03-26 devnull if(g->fd > 0)
287 05a4d855 2007-03-26 devnull fd[0] = dup(g->fd, -1);
289 05a4d855 2007-03-26 devnull fd[0] = open("/dev/null", OREAD);
290 05a4d855 2007-03-26 devnull fd[1] = pfd[1];
291 05a4d855 2007-03-26 devnull fd[2] = dup(2, -1);
293 05a4d855 2007-03-26 devnull if(threadspawnl(fd, "rc", "rc", "-c", cmd, nil) == -1)
294 24c02865 2005-01-04 devnull return -1;
296 05a4d855 2007-03-26 devnull return pfd[0];