1 #include <u.h>
2 #include <libc.h>
3 #include <bio.h>
4 #include <draw.h>
5 #include <event.h>
6 #include "imagefile.h"
7 
8 int		cflag = 0;
9 int		dflag = 0;
10 int		eflag = 0;
11 int		jflag = 0;
12 int		fflag = 0;
13 int		Fflag = 0;
14 int		nineflag = 0;
15 int		threeflag = 0;
16 int		colorspace = CYCbCr;	/* default for 8-bit displays: combine color rotation with dither */
17 int		output = 0;
18 ulong	outchan = CMAP8;
19 Image	*image;
20 int		defaultcolor = 1;
21 
22 enum{
23 	Border	= 2,
24 	Edge		= 5
25 };
26 
27 char	*show(int, char*, int);
28 
29 void
30 eresized(int new)
31 {
32 	Rectangle r;
33 
34 	if(new && getwindow(display, Refnone) < 0){
35 		fprint(2, "jpg: can't reattach to window\n");
36 		exits("resize");
37 	}
38 	if(image == nil)
39 		return;
40 	r = insetrect(screen->clipr, Edge+Border);
41 	r.max.x = r.min.x+Dx(image->r);
42 	r.max.y = r.min.y+Dy(image->r);
43 	border(screen, r, -Border, nil, ZP);
44 	draw(screen, r, image, nil, image->r.min);
45 	flushimage(display, 1);
46 }
47 
48 void
49 usage(void)
50 {
51 	fprint(2, "usage: jpg -39cdefFkJrtv -W winsize [file.jpg ...]\n");
52 	exits("usage");
53 }
54 
55 void
56 main(int argc, char *argv[])
57 {
58 	int fd, i, yflag;
59 	char *err;
60 	char buf[12+1];
61 
62 	yflag = 0;
63 	ARGBEGIN{
64 	case 'W':
65 		winsize = EARGF(usage());
66 		break;
67 	case 'c':		/* produce encoded, compressed, bitmap file; no display by default */
68 		cflag++;
69 		dflag++;
70 		output++;
71 		if(defaultcolor)
72 			outchan = CMAP8;
73 		break;
74 	case 'd':		/* suppress display of image */
75 		dflag++;
76 		break;
77 	case 'e':		/* disable floyd-steinberg error diffusion */
78 		eflag++;
79 		break;
80 	case 'F':
81 		Fflag++;	/* make a movie */
82 		fflag++;	/* merge two fields per image */
83 		break;
84 	case 'f':
85 		fflag++;	/* merge two fields per image */
86 		break;
87 	case 'J':		/* decode jpeg only; no display or remap (for debugging, etc.) */
88 		jflag++;
89 		break;
90 	case 'k':		/* force black and white */
91 		defaultcolor = 0;
92 		outchan = GREY8;
93 		break;
94 	case 'r':
95 		colorspace = CRGB;
96 		break;
97 	case '3':		/* produce encoded, compressed, three-color bitmap file; no display by default */
98 		threeflag++;
99 		/* fall through */
100 	case 't':		/* produce encoded, compressed, true-color bitmap file; no display by default */
101 		cflag++;
102 		dflag++;
103 		output++;
104 		defaultcolor = 0;
105 		outchan = RGB24;
106 		break;
107 	case 'v':		/* force RGBV */
108 		defaultcolor = 0;
109 		outchan = CMAP8;
110 		break;
111 	case 'y':	/* leave it in CYCbCr; for debugging only */
112 		yflag = 1;
113 		colorspace = CYCbCr;
114 		break;
115 	case '9':		/* produce plan 9, uncompressed, bitmap file; no display by default */
116 		nineflag++;
117 		dflag++;
118 		output++;
119 		if(defaultcolor)
120 			outchan = CMAP8;
121 		break;
122 	default:
123 		usage();
124 	}ARGEND;
125 
126 	if(yflag==0 && dflag==0 && colorspace==CYCbCr){	/* see if we should convert right to RGB */
127 		fd = open("/dev/screen", OREAD);
128 		if(fd > 0){
129 			buf[12] = '\0';
130 			if(read(fd, buf, 12)==12 && chantodepth(strtochan(buf))>8)
131 				colorspace = CRGB;
132 			close(fd);
133 		}
134 	}
135 
136 	err = nil;
137 	if(argc == 0)
138 		err = show(0, "<stdin>", outchan);
139 	else{
140 		for(i=0; i<argc; i++){
141 			fd = open(argv[i], OREAD);
142 			if(fd < 0){
143 				fprint(2, "jpg: can't open %s: %r\n", argv[i]);
144 				err = "open";
145 			}else{
146 				err = show(fd, argv[i], outchan);
147 				close(fd);
148 			}
149 			if((nineflag || cflag) && argc>1 && err==nil){
150 				fprint(2, "jpg: exiting after one file\n");
151 				break;
152 			}
153 		}
154 	}
155 	exits(err);
156 }
157 
158 Rawimage**
159 vidmerge(Rawimage **aa1, Rawimage **aa2)
160 {
161 	Rawimage **aao, *ao, *a1, *a2;
162 	int i, c, row, col;
163 
164 	aao = nil;
165 	for (i = 0; aa1[i]; i++) {
166 
167 		a1 = aa1[i];
168 		a2 = aa2[i];
169 		if (a2 == nil){
170 			fprint(2, "jpg: vidmerge: unequal lengths\n");
171 			return nil;
172 		}
173 		aao = realloc(aao, (i+2)*sizeof(Rawimage *));
174 		if (aao == nil){
175 			fprint(2, "jpg: vidmerge: realloc\n");
176 			return nil;
177 		}
178 		aao[i+1] = nil;
179 		ao = aao[i] = malloc(sizeof(Rawimage));
180 		if (ao == nil){
181 			fprint(2, "jpg: vidmerge: realloc\n");
182 			return nil;
183 		}
184 		memcpy(ao, a1, sizeof(Rawimage));
185 		if (!eqrect(a1->r , a2->r)){
186 			fprint(2, "jpg: vidmerge: rects different in img %d\n", i);
187 			return nil;
188 		}
189 		if (a1->cmaplen != a2->cmaplen){
190 			fprint(2, "jpg: vidmerge: cmaplen different in img %d\n", i);
191 			return nil;
192 		}
193 		if (a1->nchans != a2->nchans){
194 			fprint(2, "jpg: vidmerge: nchans different in img %d\n", i);
195 			return nil;
196 		}
197 		if (a1->fields != a2->fields){
198 			fprint(2, "jpg: vidmerge: fields different in img %d\n", i);
199 			return nil;
200 		}
201 		ao->r.max.y += Dy(ao->r);
202 		ao->chanlen += ao->chanlen;
203 		if (ao->chanlen != Dx(ao->r)*Dy(ao->r)){
204 			fprint(2, "jpg: vidmerge: chanlen wrong %d != %d*%d\n",
205 				ao->chanlen, Dx(ao->r), Dy(ao->r));
206 			return nil;
207 		}
208 		row = Dx(a1->r);
209 		for (c = 0; c < ao->nchans; c++) {
210 			uchar *po, *p1, *p2;
211 
212 			ao->chans[c] = malloc(ao->chanlen);
213 			po = ao->chans[c];
214 			p1 = a1->chans[c];
215 			p2 = a2->chans[c];
216 			for (col = 0; col < Dy(a1->r); col++) {
217 				memcpy(po, p1, row);
218 				po += row, p1 += row;
219 				memcpy(po, p2, row);
220 				po += row, p2 += row;
221 			}
222 			free(a1->chans[c]);
223 			free(a2->chans[c]);
224 		}
225 		if(a2->cmap != nil)
226 			free(a2->cmap);
227 		free(a1);
228 		free(a2);
229 	}	
230 	if (aa2[i] != nil)
231 		fprint(2, "jpg: vidmerge: unequal lengths\n");
232 	free(aa1);
233 	free(aa2);
234 	return aao;
235 }
236 
237 char*
238 show(int fd, char *name, int outc)
239 {
240 	Rawimage **array, *r, *c;
241 	static int inited;
242 	Image *i;
243 	int j, ch, outchan;
244 	Biobuf b;
245 	char buf[32];
246 
247 	if(Binit(&b, fd, OREAD) < 0)
248 		return nil;
249 	outchan = outc;
250 rpt:	array = Breadjpg(&b, colorspace);
251 	if(array == nil || array[0]==nil){
252 		fprint(2, "jpg: decode %s failed: %r\n", name);
253 		return "decode";
254 	}
255 	if (fflag) {
256 		Rawimage **a;
257 
258 		a = Breadjpg(&b, colorspace);
259 		if(a == nil || a[0]==nil){
260 			fprint(2, "jpg: decode %s-2 failed: %r\n", name);
261 			return "decode";
262 		}
263 		array = vidmerge(a, array);
264 	} else
265 		Bterm(&b);
266 
267 	r = array[0];
268 	c = nil;
269 	if(jflag)
270 		goto Return;
271 	if(!dflag && !inited){
272 		if(initdraw(0, 0, 0) < 0){
273 			fprint(2, "jpg: initdraw failed: %r\n");
274 			return "initdraw";
275 		}
276 		if(Fflag == 0)
277 			einit(Ekeyboard|Emouse);
278 		if(defaultcolor && screen->depth>8 && outchan==CMAP8)
279 			outchan = RGB24;
280 		inited++;
281 	}
282 	if(outchan == CMAP8)
283 		c = torgbv(r, !eflag);
284 	else{
285 		if(outchan==GREY8 || (r->chandesc==CY && threeflag==0)){
286 			c = totruecolor(r, CY);
287 			outchan = GREY8;
288 		}else
289 			c = totruecolor(r, CRGB24);
290 	}
291 	if(c == nil){
292 		fprint(2, "jpg: conversion of %s failed: %r\n", name);
293 		return "torgbv";
294 	}
295 	if(!dflag){
296 		if(c->chandesc == CY)
297 			i = allocimage(display, c->r, GREY8, 0, 0);
298 		else
299 			i = allocimage(display, c->r, outchan, 0, 0);
300 		if(i == nil){
301 			fprint(2, "jpg: allocimage %s failed: %r\n", name);
302 			return "allocimage";
303 		}
304 		if(loadimage(i, i->r, c->chans[0], c->chanlen) < 0){
305 			fprint(2, "jpg: loadimage %s failed: %r\n", name);
306 			return "loadimage";
307 		}
308 		image = i;
309 		eresized(0);
310 		if (Fflag) {
311 			freeimage(i);
312 			for(j=0; j<r->nchans; j++)
313 				free(r->chans[j]);
314 			free(r->cmap);
315 			free(r);
316 			free(array);
317 			goto rpt;
318 		}
319 		if((ch=ekbd())=='q' || ch==0x7F || ch==0x04)
320 			exits(nil);
321 		draw(screen, screen->clipr, display->white, nil, ZP);
322 		image = nil;
323 		freeimage(i);
324 	}
325 	if(nineflag){
326 		chantostr(buf, outchan);
327 		print("%11s %11d %11d %11d %11d ", buf,
328 			c->r.min.x, c->r.min.y, c->r.max.x, c->r.max.y);
329 		if(write(1, c->chans[0], c->chanlen) != c->chanlen){
330 			fprint(2, "jpg: %s: write error %r\n", name);
331 			return "write";
332 		}
333 	}else if(cflag){
334 		if(writerawimage(1, c) < 0){
335 			fprint(2, "jpg: %s: write error: %r\n", name);
336 			return "write";
337 		}
338 	}
339     Return:
340 	for(j=0; j<r->nchans; j++)
341 		free(r->chans[j]);
342 	free(r->cmap);
343 	free(r);
344 	free(array);
345 	if(c){
346 		free(c->chans[0]);
347 		free(c);
348 	}
349 	if (Fflag) goto rpt;
350 	return nil;
351 }