Blame

Date:
Message:
add -W to specify window size. various other little fixes.
Actions:
History | Blob | Raw File
  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 28994509 2004-04-21 devnull 	r = insetrect(screen->clipr, Edge+Border);

 41 28994509 2004-04-21 devnull 	r.max.x = r.min.x+Dx(image->r);

 42 28994509 2004-04-21 devnull 	r.max.y = r.min.y+Dy(image->r);

 43 28994509 2004-04-21 devnull 	border(screen, r, -Border, nil, ZP);

 44 28994509 2004-04-21 devnull 	draw(screen, r, image, nil, image->r.min);

 45 28994509 2004-04-21 devnull 	flushimage(display, 1);

 46 28994509 2004-04-21 devnull }

 47 28994509 2004-04-21 devnull 

 48 28994509 2004-04-21 devnull void

 49 be36ff68 2004-04-29 devnull usage(void)

 50 be36ff68 2004-04-29 devnull {

 51 be36ff68 2004-04-29 devnull 	fprint(2, "usage: jpg -39cdefFkJrtv -W winsize [file.jpg ...]\n");

 52 be36ff68 2004-04-29 devnull 	exits("usage");

 53 be36ff68 2004-04-29 devnull }

 54 be36ff68 2004-04-29 devnull 

 55 be36ff68 2004-04-29 devnull void

 56 28994509 2004-04-21 devnull main(int argc, char *argv[])

 57 28994509 2004-04-21 devnull {

 58 28994509 2004-04-21 devnull 	int fd, i, yflag;

 59 28994509 2004-04-21 devnull 	char *err;

 60 28994509 2004-04-21 devnull 	char buf[12+1];

 61 28994509 2004-04-21 devnull 

 62 28994509 2004-04-21 devnull 	yflag = 0;

 63 28994509 2004-04-21 devnull 	ARGBEGIN{

 64 be36ff68 2004-04-29 devnull 	case 'W':

 65 be36ff68 2004-04-29 devnull 		winsize = EARGF(usage());

 66 be36ff68 2004-04-29 devnull 		break;

 67 28994509 2004-04-21 devnull 	case 'c':		/* produce encoded, compressed, bitmap file; no display by default */

 68 28994509 2004-04-21 devnull 		cflag++;

 69 28994509 2004-04-21 devnull 		dflag++;

 70 28994509 2004-04-21 devnull 		output++;

 71 28994509 2004-04-21 devnull 		if(defaultcolor)

 72 28994509 2004-04-21 devnull 			outchan = CMAP8;

 73 28994509 2004-04-21 devnull 		break;

 74 28994509 2004-04-21 devnull 	case 'd':		/* suppress display of image */

 75 28994509 2004-04-21 devnull 		dflag++;

 76 28994509 2004-04-21 devnull 		break;

 77 28994509 2004-04-21 devnull 	case 'e':		/* disable floyd-steinberg error diffusion */

 78 28994509 2004-04-21 devnull 		eflag++;

 79 28994509 2004-04-21 devnull 		break;

 80 28994509 2004-04-21 devnull 	case 'F':

 81 28994509 2004-04-21 devnull 		Fflag++;	/* make a movie */

 82 28994509 2004-04-21 devnull 		fflag++;	/* merge two fields per image */

 83 28994509 2004-04-21 devnull 		break;

 84 28994509 2004-04-21 devnull 	case 'f':

 85 28994509 2004-04-21 devnull 		fflag++;	/* merge two fields per image */

 86 28994509 2004-04-21 devnull 		break;

 87 28994509 2004-04-21 devnull 	case 'J':		/* decode jpeg only; no display or remap (for debugging, etc.) */

 88 28994509 2004-04-21 devnull 		jflag++;

 89 28994509 2004-04-21 devnull 		break;

 90 28994509 2004-04-21 devnull 	case 'k':		/* force black and white */

 91 28994509 2004-04-21 devnull 		defaultcolor = 0;

 92 28994509 2004-04-21 devnull 		outchan = GREY8;

 93 28994509 2004-04-21 devnull 		break;

 94 28994509 2004-04-21 devnull 	case 'r':

 95 28994509 2004-04-21 devnull 		colorspace = CRGB;

 96 28994509 2004-04-21 devnull 		break;

 97 28994509 2004-04-21 devnull 	case '3':		/* produce encoded, compressed, three-color bitmap file; no display by default */

 98 28994509 2004-04-21 devnull 		threeflag++;

 99 28994509 2004-04-21 devnull 		/* fall through */

100 28994509 2004-04-21 devnull 	case 't':		/* produce encoded, compressed, true-color bitmap file; no display by default */

101 28994509 2004-04-21 devnull 		cflag++;

102 28994509 2004-04-21 devnull 		dflag++;

103 28994509 2004-04-21 devnull 		output++;

104 28994509 2004-04-21 devnull 		defaultcolor = 0;

105 28994509 2004-04-21 devnull 		outchan = RGB24;

106 28994509 2004-04-21 devnull 		break;

107 28994509 2004-04-21 devnull 	case 'v':		/* force RGBV */

108 28994509 2004-04-21 devnull 		defaultcolor = 0;

109 28994509 2004-04-21 devnull 		outchan = CMAP8;

110 28994509 2004-04-21 devnull 		break;

111 28994509 2004-04-21 devnull 	case 'y':	/* leave it in CYCbCr; for debugging only */

112 28994509 2004-04-21 devnull 		yflag = 1;

113 28994509 2004-04-21 devnull 		colorspace = CYCbCr;

114 28994509 2004-04-21 devnull 		break;

115 28994509 2004-04-21 devnull 	case '9':		/* produce plan 9, uncompressed, bitmap file; no display by default */

116 28994509 2004-04-21 devnull 		nineflag++;

117 28994509 2004-04-21 devnull 		dflag++;

118 28994509 2004-04-21 devnull 		output++;

119 28994509 2004-04-21 devnull 		if(defaultcolor)

120 28994509 2004-04-21 devnull 			outchan = CMAP8;

121 28994509 2004-04-21 devnull 		break;

122 28994509 2004-04-21 devnull 	default:

123 be36ff68 2004-04-29 devnull 		usage();

124 28994509 2004-04-21 devnull 	}ARGEND;

125 28994509 2004-04-21 devnull 

126 28994509 2004-04-21 devnull 	if(yflag==0 && dflag==0 && colorspace==CYCbCr){	/* see if we should convert right to RGB */

127 28994509 2004-04-21 devnull 		fd = open("/dev/screen", OREAD);

128 28994509 2004-04-21 devnull 		if(fd > 0){

129 28994509 2004-04-21 devnull 			buf[12] = '\0';

130 28994509 2004-04-21 devnull 			if(read(fd, buf, 12)==12 && chantodepth(strtochan(buf))>8)

131 28994509 2004-04-21 devnull 				colorspace = CRGB;

132 28994509 2004-04-21 devnull 			close(fd);

133 28994509 2004-04-21 devnull 		}

134 28994509 2004-04-21 devnull 	}

135 28994509 2004-04-21 devnull 

136 28994509 2004-04-21 devnull 	err = nil;

137 28994509 2004-04-21 devnull 	if(argc == 0)

138 28994509 2004-04-21 devnull 		err = show(0, "<stdin>", outchan);

139 28994509 2004-04-21 devnull 	else{

140 28994509 2004-04-21 devnull 		for(i=0; i<argc; i++){

141 28994509 2004-04-21 devnull 			fd = open(argv[i], OREAD);

142 28994509 2004-04-21 devnull 			if(fd < 0){

143 28994509 2004-04-21 devnull 				fprint(2, "jpg: can't open %s: %r\n", argv[i]);

144 28994509 2004-04-21 devnull 				err = "open";

145 28994509 2004-04-21 devnull 			}else{

146 28994509 2004-04-21 devnull 				err = show(fd, argv[i], outchan);

147 28994509 2004-04-21 devnull 				close(fd);

148 28994509 2004-04-21 devnull 			}

149 28994509 2004-04-21 devnull 			if((nineflag || cflag) && argc>1 && err==nil){

150 28994509 2004-04-21 devnull 				fprint(2, "jpg: exiting after one file\n");

151 28994509 2004-04-21 devnull 				break;

152 28994509 2004-04-21 devnull 			}

153 28994509 2004-04-21 devnull 		}

154 28994509 2004-04-21 devnull 	}

155 28994509 2004-04-21 devnull 	exits(err);

156 28994509 2004-04-21 devnull }

157 28994509 2004-04-21 devnull 

158 28994509 2004-04-21 devnull Rawimage**

159 28994509 2004-04-21 devnull vidmerge(Rawimage **aa1, Rawimage **aa2)

160 28994509 2004-04-21 devnull {

161 28994509 2004-04-21 devnull 	Rawimage **aao, *ao, *a1, *a2;

162 28994509 2004-04-21 devnull 	int i, c, row, col;

163 28994509 2004-04-21 devnull 

164 28994509 2004-04-21 devnull 	aao = nil;

165 28994509 2004-04-21 devnull 	for (i = 0; aa1[i]; i++) {

166 28994509 2004-04-21 devnull 

167 28994509 2004-04-21 devnull 		a1 = aa1[i];

168 28994509 2004-04-21 devnull 		a2 = aa2[i];

169 28994509 2004-04-21 devnull 		if (a2 == nil){

170 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: unequal lengths\n");

171 28994509 2004-04-21 devnull 			return nil;

172 28994509 2004-04-21 devnull 		}

173 28994509 2004-04-21 devnull 		aao = realloc(aao, (i+2)*sizeof(Rawimage *));

174 28994509 2004-04-21 devnull 		if (aao == nil){

175 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: realloc\n");

176 28994509 2004-04-21 devnull 			return nil;

177 28994509 2004-04-21 devnull 		}

178 28994509 2004-04-21 devnull 		aao[i+1] = nil;

179 28994509 2004-04-21 devnull 		ao = aao[i] = malloc(sizeof(Rawimage));

180 28994509 2004-04-21 devnull 		if (ao == nil){

181 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: realloc\n");

182 28994509 2004-04-21 devnull 			return nil;

183 28994509 2004-04-21 devnull 		}

184 28994509 2004-04-21 devnull 		memcpy(ao, a1, sizeof(Rawimage));

185 28994509 2004-04-21 devnull 		if (!eqrect(a1->r , a2->r)){

186 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: rects different in img %d\n", i);

187 28994509 2004-04-21 devnull 			return nil;

188 28994509 2004-04-21 devnull 		}

189 28994509 2004-04-21 devnull 		if (a1->cmaplen != a2->cmaplen){

190 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: cmaplen different in img %d\n", i);

191 28994509 2004-04-21 devnull 			return nil;

192 28994509 2004-04-21 devnull 		}

193 28994509 2004-04-21 devnull 		if (a1->nchans != a2->nchans){

194 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: nchans different in img %d\n", i);

195 28994509 2004-04-21 devnull 			return nil;

196 28994509 2004-04-21 devnull 		}

197 28994509 2004-04-21 devnull 		if (a1->fields != a2->fields){

198 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: fields different in img %d\n", i);

199 28994509 2004-04-21 devnull 			return nil;

200 28994509 2004-04-21 devnull 		}

201 28994509 2004-04-21 devnull 		ao->r.max.y += Dy(ao->r);

202 28994509 2004-04-21 devnull 		ao->chanlen += ao->chanlen;

203 28994509 2004-04-21 devnull 		if (ao->chanlen != Dx(ao->r)*Dy(ao->r)){

204 28994509 2004-04-21 devnull 			fprint(2, "jpg: vidmerge: chanlen wrong %d != %d*%d\n",

205 28994509 2004-04-21 devnull 				ao->chanlen, Dx(ao->r), Dy(ao->r));

206 28994509 2004-04-21 devnull 			return nil;

207 28994509 2004-04-21 devnull 		}

208 28994509 2004-04-21 devnull 		row = Dx(a1->r);

209 28994509 2004-04-21 devnull 		for (c = 0; c < ao->nchans; c++) {

210 28994509 2004-04-21 devnull 			uchar *po, *p1, *p2;

211 28994509 2004-04-21 devnull 

212 28994509 2004-04-21 devnull 			ao->chans[c] = malloc(ao->chanlen);

213 28994509 2004-04-21 devnull 			po = ao->chans[c];

214 28994509 2004-04-21 devnull 			p1 = a1->chans[c];

215 28994509 2004-04-21 devnull 			p2 = a2->chans[c];

216 28994509 2004-04-21 devnull 			for (col = 0; col < Dy(a1->r); col++) {

217 28994509 2004-04-21 devnull 				memcpy(po, p1, row);

218 28994509 2004-04-21 devnull 				po += row, p1 += row;

219 28994509 2004-04-21 devnull 				memcpy(po, p2, row);

220 28994509 2004-04-21 devnull 				po += row, p2 += row;

221 28994509 2004-04-21 devnull 			}

222 28994509 2004-04-21 devnull 			free(a1->chans[c]);

223 28994509 2004-04-21 devnull 			free(a2->chans[c]);

224 28994509 2004-04-21 devnull 		}

225 28994509 2004-04-21 devnull 		if(a2->cmap != nil)

226 28994509 2004-04-21 devnull 			free(a2->cmap);

227 28994509 2004-04-21 devnull 		free(a1);

228 28994509 2004-04-21 devnull 		free(a2);

229 28994509 2004-04-21 devnull 	}	

230 28994509 2004-04-21 devnull 	if (aa2[i] != nil)

231 28994509 2004-04-21 devnull 		fprint(2, "jpg: vidmerge: unequal lengths\n");

232 28994509 2004-04-21 devnull 	free(aa1);

233 28994509 2004-04-21 devnull 	free(aa2);

234 28994509 2004-04-21 devnull 	return aao;

235 28994509 2004-04-21 devnull }

236 28994509 2004-04-21 devnull 

237 28994509 2004-04-21 devnull char*

238 28994509 2004-04-21 devnull show(int fd, char *name, int outc)

239 28994509 2004-04-21 devnull {

240 28994509 2004-04-21 devnull 	Rawimage **array, *r, *c;

241 28994509 2004-04-21 devnull 	static int inited;

242 28994509 2004-04-21 devnull 	Image *i;

243 28994509 2004-04-21 devnull 	int j, ch, outchan;

244 28994509 2004-04-21 devnull 	Biobuf b;

245 28994509 2004-04-21 devnull 	char buf[32];

246 28994509 2004-04-21 devnull 

247 28994509 2004-04-21 devnull 	if(Binit(&b, fd, OREAD) < 0)

248 28994509 2004-04-21 devnull 		return nil;

249 28994509 2004-04-21 devnull 	outchan = outc;

250 28994509 2004-04-21 devnull rpt:	array = Breadjpg(&b, colorspace);

251 28994509 2004-04-21 devnull 	if(array == nil || array[0]==nil){

252 28994509 2004-04-21 devnull 		fprint(2, "jpg: decode %s failed: %r\n", name);

253 28994509 2004-04-21 devnull 		return "decode";

254 28994509 2004-04-21 devnull 	}

255 28994509 2004-04-21 devnull 	if (fflag) {

256 28994509 2004-04-21 devnull 		Rawimage **a;

257 28994509 2004-04-21 devnull 

258 28994509 2004-04-21 devnull 		a = Breadjpg(&b, colorspace);

259 28994509 2004-04-21 devnull 		if(a == nil || a[0]==nil){

260 28994509 2004-04-21 devnull 			fprint(2, "jpg: decode %s-2 failed: %r\n", name);

261 28994509 2004-04-21 devnull 			return "decode";

262 28994509 2004-04-21 devnull 		}

263 28994509 2004-04-21 devnull 		array = vidmerge(a, array);

264 28994509 2004-04-21 devnull 	} else

265 28994509 2004-04-21 devnull 		Bterm(&b);

266 28994509 2004-04-21 devnull 

267 28994509 2004-04-21 devnull 	r = array[0];

268 28994509 2004-04-21 devnull 	c = nil;

269 28994509 2004-04-21 devnull 	if(jflag)

270 28994509 2004-04-21 devnull 		goto Return;

271 28994509 2004-04-21 devnull 	if(!dflag && !inited){

272 28994509 2004-04-21 devnull 		if(initdraw(0, 0, 0) < 0){

273 28994509 2004-04-21 devnull 			fprint(2, "jpg: initdraw failed: %r\n");

274 28994509 2004-04-21 devnull 			return "initdraw";

275 28994509 2004-04-21 devnull 		}

276 28994509 2004-04-21 devnull 		if(Fflag == 0)

277 28994509 2004-04-21 devnull 			einit(Ekeyboard|Emouse);

278 28994509 2004-04-21 devnull 		if(defaultcolor && screen->depth>8 && outchan==CMAP8)

279 28994509 2004-04-21 devnull 			outchan = RGB24;

280 28994509 2004-04-21 devnull 		inited++;

281 28994509 2004-04-21 devnull 	}

282 28994509 2004-04-21 devnull 	if(outchan == CMAP8)

283 28994509 2004-04-21 devnull 		c = torgbv(r, !eflag);

284 28994509 2004-04-21 devnull 	else{

285 28994509 2004-04-21 devnull 		if(outchan==GREY8 || (r->chandesc==CY && threeflag==0)){

286 28994509 2004-04-21 devnull 			c = totruecolor(r, CY);

287 28994509 2004-04-21 devnull 			outchan = GREY8;

288 28994509 2004-04-21 devnull 		}else

289 28994509 2004-04-21 devnull 			c = totruecolor(r, CRGB24);

290 28994509 2004-04-21 devnull 	}

291 28994509 2004-04-21 devnull 	if(c == nil){

292 28994509 2004-04-21 devnull 		fprint(2, "jpg: conversion of %s failed: %r\n", name);

293 28994509 2004-04-21 devnull 		return "torgbv";

294 28994509 2004-04-21 devnull 	}

295 28994509 2004-04-21 devnull 	if(!dflag){

296 28994509 2004-04-21 devnull 		if(c->chandesc == CY)

297 28994509 2004-04-21 devnull 			i = allocimage(display, c->r, GREY8, 0, 0);

298 28994509 2004-04-21 devnull 		else

299 28994509 2004-04-21 devnull 			i = allocimage(display, c->r, outchan, 0, 0);

300 28994509 2004-04-21 devnull 		if(i == nil){

301 28994509 2004-04-21 devnull 			fprint(2, "jpg: allocimage %s failed: %r\n", name);

302 28994509 2004-04-21 devnull 			return "allocimage";

303 28994509 2004-04-21 devnull 		}

304 28994509 2004-04-21 devnull 		if(loadimage(i, i->r, c->chans[0], c->chanlen) < 0){

305 28994509 2004-04-21 devnull 			fprint(2, "jpg: loadimage %s failed: %r\n", name);

306 28994509 2004-04-21 devnull 			return "loadimage";

307 28994509 2004-04-21 devnull 		}

308 28994509 2004-04-21 devnull 		image = i;

309 28994509 2004-04-21 devnull 		eresized(0);

310 28994509 2004-04-21 devnull 		if (Fflag) {

311 28994509 2004-04-21 devnull 			freeimage(i);

312 28994509 2004-04-21 devnull 			for(j=0; j<r->nchans; j++)

313 28994509 2004-04-21 devnull 				free(r->chans[j]);

314 28994509 2004-04-21 devnull 			free(r->cmap);

315 28994509 2004-04-21 devnull 			free(r);

316 28994509 2004-04-21 devnull 			free(array);

317 28994509 2004-04-21 devnull 			goto rpt;

318 28994509 2004-04-21 devnull 		}

319 28994509 2004-04-21 devnull 		if((ch=ekbd())=='q' || ch==0x7F || ch==0x04)

320 28994509 2004-04-21 devnull 			exits(nil);

321 28994509 2004-04-21 devnull 		draw(screen, screen->clipr, display->white, nil, ZP);

322 28994509 2004-04-21 devnull 		image = nil;

323 28994509 2004-04-21 devnull 		freeimage(i);

324 28994509 2004-04-21 devnull 	}

325 28994509 2004-04-21 devnull 	if(nineflag){

326 28994509 2004-04-21 devnull 		chantostr(buf, outchan);

327 28994509 2004-04-21 devnull 		print("%11s %11d %11d %11d %11d ", buf,

328 28994509 2004-04-21 devnull 			c->r.min.x, c->r.min.y, c->r.max.x, c->r.max.y);

329 28994509 2004-04-21 devnull 		if(write(1, c->chans[0], c->chanlen) != c->chanlen){

330 28994509 2004-04-21 devnull 			fprint(2, "jpg: %s: write error %r\n", name);

331 28994509 2004-04-21 devnull 			return "write";

332 28994509 2004-04-21 devnull 		}

333 28994509 2004-04-21 devnull 	}else if(cflag){

334 28994509 2004-04-21 devnull 		if(writerawimage(1, c) < 0){

335 28994509 2004-04-21 devnull 			fprint(2, "jpg: %s: write error: %r\n", name);

336 28994509 2004-04-21 devnull 			return "write";

337 28994509 2004-04-21 devnull 		}

338 28994509 2004-04-21 devnull 	}

339 28994509 2004-04-21 devnull     Return:

340 28994509 2004-04-21 devnull 	for(j=0; j<r->nchans; j++)

341 28994509 2004-04-21 devnull 		free(r->chans[j]);

342 28994509 2004-04-21 devnull 	free(r->cmap);

343 28994509 2004-04-21 devnull 	free(r);

344 28994509 2004-04-21 devnull 	free(array);

345 28994509 2004-04-21 devnull 	if(c){

346 28994509 2004-04-21 devnull 		free(c->chans[0]);

347 28994509 2004-04-21 devnull 		free(c);

348 28994509 2004-04-21 devnull 	}

349 28994509 2004-04-21 devnull 	if (Fflag) goto rpt;

350 28994509 2004-04-21 devnull 	return nil;

351 28994509 2004-04-21 devnull }