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		nineflag = 0;

 12 28994509 2004-04-21 devnull int		threeflag = 0;

 13 28994509 2004-04-21 devnull int		output = 0;

 14 28994509 2004-04-21 devnull ulong	outchan = CMAP8;

 15 28994509 2004-04-21 devnull int		defaultcolor = 1;

 16 28994509 2004-04-21 devnull Image	*image;

 17 28994509 2004-04-21 devnull 

 18 28994509 2004-04-21 devnull enum{

 19 28994509 2004-04-21 devnull 	Border	= 2,

 20 28994509 2004-04-21 devnull 	Edge		= 5

 21 28994509 2004-04-21 devnull };

 22 28994509 2004-04-21 devnull 

 23 28994509 2004-04-21 devnull char	*show(int, char*);

 24 28994509 2004-04-21 devnull 

 25 28994509 2004-04-21 devnull Rawimage** readyuv(int fd, int colorspace);

 26 28994509 2004-04-21 devnull 

 27 28994509 2004-04-21 devnull void

 28 28994509 2004-04-21 devnull eresized(int new)

 29 28994509 2004-04-21 devnull {

 30 28994509 2004-04-21 devnull 	Rectangle r;

 31 28994509 2004-04-21 devnull 

 32 28994509 2004-04-21 devnull 	if(new && getwindow(display, Refnone) < 0){

 33 28994509 2004-04-21 devnull 		fprint(2, "yuv: can't reattach to window\n");

 34 28994509 2004-04-21 devnull 		exits("resize");

 35 28994509 2004-04-21 devnull 	}

 36 28994509 2004-04-21 devnull 	if(image == nil)

 37 28994509 2004-04-21 devnull 		return;

 38 28994509 2004-04-21 devnull 	r = insetrect(screen->clipr, Edge+Border);

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

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

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

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

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

 44 28994509 2004-04-21 devnull }

 45 28994509 2004-04-21 devnull 

 46 28994509 2004-04-21 devnull void

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

 48 be36ff68 2004-04-29 devnull {

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

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

 51 be36ff68 2004-04-29 devnull }

 52 be36ff68 2004-04-29 devnull 

 53 be36ff68 2004-04-29 devnull void

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

 55 28994509 2004-04-21 devnull {

 56 28994509 2004-04-21 devnull 	int fd, i;

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

 58 28994509 2004-04-21 devnull 

 59 28994509 2004-04-21 devnull 	ARGBEGIN{

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

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

 62 be36ff68 2004-04-29 devnull 		break;

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

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

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

 66 28994509 2004-04-21 devnull 	case 't':		/* produce encoded, compressed, true-color 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 		defaultcolor = 0;

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

 72 28994509 2004-04-21 devnull 		break;

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

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

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

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

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

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

 79 28994509 2004-04-21 devnull 		break;

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

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

 82 28994509 2004-04-21 devnull 		break;

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

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

 85 28994509 2004-04-21 devnull 		break;

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

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

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

 89 28994509 2004-04-21 devnull 		break;

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

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

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

 93 28994509 2004-04-21 devnull 		break;

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

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

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

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

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

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

100 28994509 2004-04-21 devnull 		break;

101 28994509 2004-04-21 devnull 	default:

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

103 28994509 2004-04-21 devnull 	}ARGEND;

104 28994509 2004-04-21 devnull 

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

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

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

108 28994509 2004-04-21 devnull 	else{

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

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

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

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

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

114 28994509 2004-04-21 devnull 			}else{

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

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

117 28994509 2004-04-21 devnull 			}

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

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

120 28994509 2004-04-21 devnull 				break;

121 28994509 2004-04-21 devnull 			}

122 28994509 2004-04-21 devnull 		}

123 28994509 2004-04-21 devnull 	}

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

125 28994509 2004-04-21 devnull }

126 28994509 2004-04-21 devnull 

127 28994509 2004-04-21 devnull int

128 28994509 2004-04-21 devnull init(void)

129 28994509 2004-04-21 devnull {

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

131 28994509 2004-04-21 devnull 

132 28994509 2004-04-21 devnull 	if(inited == 0){

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

134 28994509 2004-04-21 devnull 			fprint(2, "yuv: initdraw failed: %r");

135 28994509 2004-04-21 devnull 			return -1;

136 28994509 2004-04-21 devnull 		}

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

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

139 28994509 2004-04-21 devnull 	}

140 28994509 2004-04-21 devnull 	return 1;

141 28994509 2004-04-21 devnull }

142 28994509 2004-04-21 devnull 

143 28994509 2004-04-21 devnull char*

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

145 28994509 2004-04-21 devnull {

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

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

148 28994509 2004-04-21 devnull 	int j, ch;

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

150 28994509 2004-04-21 devnull 

151 28994509 2004-04-21 devnull 	array = readyuv(fd, CYCbCr);

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

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

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

155 28994509 2004-04-21 devnull 	}

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

157 28994509 2004-04-21 devnull 		if(init() < 0)

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

159 28994509 2004-04-21 devnull 		if(defaultcolor && screen->depth>8)

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

161 28994509 2004-04-21 devnull 	}

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

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

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

165 28994509 2004-04-21 devnull 	else{

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

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

168 28994509 2004-04-21 devnull 		else

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

170 28994509 2004-04-21 devnull 	}

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

172 28994509 2004-04-21 devnull 		fprint(2, "yuv: converting %s to local format failed: %r\n", name);

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

174 28994509 2004-04-21 devnull 	}

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

176 28994509 2004-04-21 devnull 		if(r->chandesc == CY)

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

178 28994509 2004-04-21 devnull 		else

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

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

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

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

183 28994509 2004-04-21 devnull 		}

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

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

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

187 28994509 2004-04-21 devnull 		}

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

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

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

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

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

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

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

195 28994509 2004-04-21 devnull 	}

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

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

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

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

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

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

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

203 28994509 2004-04-21 devnull 		}

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

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

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

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

208 28994509 2004-04-21 devnull 		}

209 28994509 2004-04-21 devnull 	}

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

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

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

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

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

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

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

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

218 28994509 2004-04-21 devnull 	}

219 28994509 2004-04-21 devnull 	return nil;

220 28994509 2004-04-21 devnull }