Blame


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 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" },
45 24c02865 2005-01-04 devnull };
46 24c02865 2005-01-04 devnull
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);
55 24c02865 2005-01-04 devnull
56 24c02865 2005-01-04 devnull static char*
57 24c02865 2005-01-04 devnull gfxpagename(Document *doc, int page)
58 24c02865 2005-01-04 devnull {
59 24c02865 2005-01-04 devnull GfxInfo *gfx = doc->extra;
60 24c02865 2005-01-04 devnull return gfx->g[page].name;
61 24c02865 2005-01-04 devnull }
62 24c02865 2005-01-04 devnull
63 24c02865 2005-01-04 devnull static Image*
64 24c02865 2005-01-04 devnull gfxdrawpage(Document *doc, int page)
65 24c02865 2005-01-04 devnull {
66 24c02865 2005-01-04 devnull GfxInfo *gfx = doc->extra;
67 24c02865 2005-01-04 devnull return convert(gfx->g+page);
68 24c02865 2005-01-04 devnull }
69 24c02865 2005-01-04 devnull
70 24c02865 2005-01-04 devnull Document*
71 24c02865 2005-01-04 devnull initgfx(Biobuf *b, int argc, char **argv, uchar *buf, int nbuf)
72 24c02865 2005-01-04 devnull {
73 24c02865 2005-01-04 devnull GfxInfo *gfx;
74 24c02865 2005-01-04 devnull Document *doc;
75 24c02865 2005-01-04 devnull int i;
76 24c02865 2005-01-04 devnull
77 24c02865 2005-01-04 devnull USED(b);
78 17157e4a 2006-03-20 devnull
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;
82 24c02865 2005-01-04 devnull
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;
90 24c02865 2005-01-04 devnull
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);
94 24c02865 2005-01-04 devnull else{
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]);
98 24c02865 2005-01-04 devnull }
99 24c02865 2005-01-04 devnull
100 24c02865 2005-01-04 devnull return doc;
101 24c02865 2005-01-04 devnull }
102 24c02865 2005-01-04 devnull
103 24c02865 2005-01-04 devnull static int
104 24c02865 2005-01-04 devnull genaddpage(Document *doc, char *name, uchar *buf, int nbuf)
105 24c02865 2005-01-04 devnull {
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;
111 24c02865 2005-01-04 devnull
112 24c02865 2005-01-04 devnull l = 0;
113 24c02865 2005-01-04 devnull gfx = doc->extra;
114 24c02865 2005-01-04 devnull
115 24c02865 2005-01-04 devnull assert((name == nil) ^ (buf == nil));
116 24c02865 2005-01-04 devnull assert(name != nil || doc->npage == 0);
117 24c02865 2005-01-04 devnull
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;
121 24c02865 2005-01-04 devnull
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;
127 24c02865 2005-01-04 devnull }
128 24c02865 2005-01-04 devnull
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;
132 24c02865 2005-01-04 devnull }
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;
136 24c02865 2005-01-04 devnull }
137 24c02865 2005-01-04 devnull
138 24c02865 2005-01-04 devnull
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];
141 24c02865 2005-01-04 devnull
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;
177 24c02865 2005-01-04 devnull else
178 24c02865 2005-01-04 devnull g->type = Icvt2pic;
179 24c02865 2005-01-04 devnull
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;
183 05a4d855 2007-03-26 devnull }else{
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);
186 24c02865 2005-01-04 devnull }
187 24c02865 2005-01-04 devnull
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++;
190 24c02865 2005-01-04 devnull }
191 24c02865 2005-01-04 devnull
192 24c02865 2005-01-04 devnull static int
193 24c02865 2005-01-04 devnull addpage(Document *doc, char *name)
194 24c02865 2005-01-04 devnull {
195 24c02865 2005-01-04 devnull return genaddpage(doc, name, nil, 0);
196 24c02865 2005-01-04 devnull }
197 24c02865 2005-01-04 devnull
198 24c02865 2005-01-04 devnull static int
199 24c02865 2005-01-04 devnull rmpage(Document *doc, int n)
200 24c02865 2005-01-04 devnull {
201 24c02865 2005-01-04 devnull int i;
202 24c02865 2005-01-04 devnull GfxInfo *gfx;
203 24c02865 2005-01-04 devnull
204 24c02865 2005-01-04 devnull if(n < 0 || n >= doc->npage)
205 24c02865 2005-01-04 devnull return -1;
206 24c02865 2005-01-04 devnull
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);
210 24c02865 2005-01-04 devnull
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];
213 24c02865 2005-01-04 devnull
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;
219 24c02865 2005-01-04 devnull }
220 24c02865 2005-01-04 devnull
221 24c02865 2005-01-04 devnull
222 05a4d855 2007-03-26 devnull Image*
223 24c02865 2005-01-04 devnull convert(Graphic *g)
224 24c02865 2005-01-04 devnull {
225 24c02865 2005-01-04 devnull int fd;
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;
231 24c02865 2005-01-04 devnull
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");
240 24c02865 2005-01-04 devnull }
241 24c02865 2005-01-04 devnull }else
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;
247 24c02865 2005-01-04 devnull
248 05a4d855 2007-03-26 devnull if(g->fd >= 0) /* is pipe */
249 24c02865 2005-01-04 devnull name = "";
250 24c02865 2005-01-04 devnull else
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");
255 24c02865 2005-01-04 devnull }
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");
263 24c02865 2005-01-04 devnull }
264 24c02865 2005-01-04 devnull }
265 24c02865 2005-01-04 devnull
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");
269 24c02865 2005-01-04 devnull }
270 24c02865 2005-01-04 devnull
271 05a4d855 2007-03-26 devnull close(fd);
272 24c02865 2005-01-04 devnull return im;
273 24c02865 2005-01-04 devnull }
274 24c02865 2005-01-04 devnull
275 24c02865 2005-01-04 devnull static int
276 05a4d855 2007-03-26 devnull spawnrc(char *cmd, Graphic *g)
277 24c02865 2005-01-04 devnull {
278 24c02865 2005-01-04 devnull int pfd[2];
279 05a4d855 2007-03-26 devnull int fd[3];
280 24c02865 2005-01-04 devnull
281 24c02865 2005-01-04 devnull if(chatty) fprint(2, "spawning(%s)...", cmd);
282 24c02865 2005-01-04 devnull
283 24c02865 2005-01-04 devnull if(pipe(pfd) < 0)
284 24c02865 2005-01-04 devnull return -1;
285 05a4d855 2007-03-26 devnull
286 05a4d855 2007-03-26 devnull if(g->fd > 0)
287 05a4d855 2007-03-26 devnull fd[0] = dup(g->fd, -1);
288 05a4d855 2007-03-26 devnull else
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);
292 05a4d855 2007-03-26 devnull
293 05a4d855 2007-03-26 devnull if(threadspawnl(fd, "rc", "rc", "-c", cmd, nil) == -1)
294 24c02865 2005-01-04 devnull return -1;
295 24c02865 2005-01-04 devnull
296 05a4d855 2007-03-26 devnull return pfd[0];
297 24c02865 2005-01-04 devnull }