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