]> git.saurik.com Git - wxWidgets.git/blob - src/tiff/tools/rgb2ycbcr.c
Add missing WXK constants for the control keys
[wxWidgets.git] / src / tiff / tools / rgb2ycbcr.c
1 /* $Id$ */
2
3 /*
4 * Copyright (c) 1991-1997 Sam Leffler
5 * Copyright (c) 1991-1997 Silicon Graphics, Inc.
6 *
7 * Permission to use, copy, modify, distribute, and sell this software and
8 * its documentation for any purpose is hereby granted without fee, provided
9 * that (i) the above copyright notices and this permission notice appear in
10 * all copies of the software and related documentation, and (ii) the names of
11 * Sam Leffler and Silicon Graphics may not be used in any advertising or
12 * publicity relating to the software without the specific, prior written
13 * permission of Sam Leffler and Silicon Graphics.
14 *
15 * THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
16 * EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
17 * WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
18 *
19 * IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
20 * ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
21 * OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
22 * WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
23 * LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
24 * OF THIS SOFTWARE.
25 */
26
27 #include "tif_config.h"
28
29 #include <stdio.h>
30 #include <string.h>
31 #include <stdlib.h>
32
33 #ifdef HAVE_UNISTD_H
34 # include <unistd.h>
35 #endif
36
37 #include "tiffio.h"
38
39 #define streq(a,b) (strcmp(a,b) == 0)
40 #define CopyField(tag, v) \
41 if (TIFFGetField(in, tag, &v)) TIFFSetField(out, tag, v)
42
43 #ifndef howmany
44 #define howmany(x, y) (((x)+((y)-1))/(y))
45 #endif
46 #define roundup(x, y) (howmany(x,y)*((uint32)(y)))
47
48 #define LumaRed ycbcrCoeffs[0]
49 #define LumaGreen ycbcrCoeffs[1]
50 #define LumaBlue ycbcrCoeffs[2]
51
52 uint16 compression = COMPRESSION_PACKBITS;
53 uint32 rowsperstrip = (uint32) -1;
54
55 uint16 horizSubSampling = 2; /* YCbCr horizontal subsampling */
56 uint16 vertSubSampling = 2; /* YCbCr vertical subsampling */
57 float ycbcrCoeffs[3] = { .299F, .587F, .114F };
58 /* default coding range is CCIR Rec 601-1 with no headroom/footroom */
59 float refBlackWhite[6] = { 0.F, 255.F, 128.F, 255.F, 128.F, 255.F };
60
61 static int tiffcvt(TIFF* in, TIFF* out);
62 static void usage(int code);
63 static void setupLumaTables(void);
64
65 int
66 main(int argc, char* argv[])
67 {
68 TIFF *in, *out;
69 int c;
70 extern int optind;
71 extern char *optarg;
72
73 while ((c = getopt(argc, argv, "c:h:r:v:z")) != -1)
74 switch (c) {
75 case 'c':
76 if (streq(optarg, "none"))
77 compression = COMPRESSION_NONE;
78 else if (streq(optarg, "packbits"))
79 compression = COMPRESSION_PACKBITS;
80 else if (streq(optarg, "lzw"))
81 compression = COMPRESSION_LZW;
82 else if (streq(optarg, "jpeg"))
83 compression = COMPRESSION_JPEG;
84 else if (streq(optarg, "zip"))
85 compression = COMPRESSION_ADOBE_DEFLATE;
86 else
87 usage(-1);
88 break;
89 case 'h':
90 horizSubSampling = atoi(optarg);
91 break;
92 case 'v':
93 vertSubSampling = atoi(optarg);
94 break;
95 case 'r':
96 rowsperstrip = atoi(optarg);
97 break;
98 case 'z': /* CCIR Rec 601-1 w/ headroom/footroom */
99 refBlackWhite[0] = 16.;
100 refBlackWhite[1] = 235.;
101 refBlackWhite[2] = 128.;
102 refBlackWhite[3] = 240.;
103 refBlackWhite[4] = 128.;
104 refBlackWhite[5] = 240.;
105 break;
106 case '?':
107 usage(0);
108 /*NOTREACHED*/
109 }
110 if (argc - optind < 2)
111 usage(-1);
112 out = TIFFOpen(argv[argc-1], "w");
113 if (out == NULL)
114 return (-2);
115 setupLumaTables();
116 for (; optind < argc-1; optind++) {
117 in = TIFFOpen(argv[optind], "r");
118 if (in != NULL) {
119 do {
120 if (!tiffcvt(in, out) ||
121 !TIFFWriteDirectory(out)) {
122 (void) TIFFClose(out);
123 return (1);
124 }
125 } while (TIFFReadDirectory(in));
126 (void) TIFFClose(in);
127 }
128 }
129 (void) TIFFClose(out);
130 return (0);
131 }
132
133 float *lumaRed;
134 float *lumaGreen;
135 float *lumaBlue;
136 float D1, D2;
137 int Yzero;
138
139 static float*
140 setupLuma(float c)
141 {
142 float *v = (float *)_TIFFmalloc(256 * sizeof (float));
143 int i;
144 for (i = 0; i < 256; i++)
145 v[i] = c * i;
146 return (v);
147 }
148
149 static unsigned
150 V2Code(float f, float RB, float RW, int CR)
151 {
152 unsigned int c = (unsigned int)((((f)*(RW-RB)/CR)+RB)+.5);
153 return (c > 255 ? 255 : c);
154 }
155
156 static void
157 setupLumaTables(void)
158 {
159 lumaRed = setupLuma(LumaRed);
160 lumaGreen = setupLuma(LumaGreen);
161 lumaBlue = setupLuma(LumaBlue);
162 D1 = 1.F/(2.F - 2.F*LumaBlue);
163 D2 = 1.F/(2.F - 2.F*LumaRed);
164 Yzero = V2Code(0, refBlackWhite[0], refBlackWhite[1], 255);
165 }
166
167 static void
168 cvtClump(unsigned char* op, uint32* raster, uint32 ch, uint32 cw, uint32 w)
169 {
170 float Y, Cb = 0, Cr = 0;
171 uint32 j, k;
172 /*
173 * Convert ch-by-cw block of RGB
174 * to YCbCr and sample accordingly.
175 */
176 for (k = 0; k < ch; k++) {
177 for (j = 0; j < cw; j++) {
178 uint32 RGB = (raster - k*w)[j];
179 Y = lumaRed[TIFFGetR(RGB)] +
180 lumaGreen[TIFFGetG(RGB)] +
181 lumaBlue[TIFFGetB(RGB)];
182 /* accumulate chrominance */
183 Cb += (TIFFGetB(RGB) - Y) * D1;
184 Cr += (TIFFGetR(RGB) - Y) * D2;
185 /* emit luminence */
186 *op++ = V2Code(Y,
187 refBlackWhite[0], refBlackWhite[1], 255);
188 }
189 for (; j < horizSubSampling; j++)
190 *op++ = Yzero;
191 }
192 for (; k < vertSubSampling; k++) {
193 for (j = 0; j < horizSubSampling; j++)
194 *op++ = Yzero;
195 }
196 /* emit sampled chrominance values */
197 *op++ = V2Code(Cb / (ch*cw), refBlackWhite[2], refBlackWhite[3], 127);
198 *op++ = V2Code(Cr / (ch*cw), refBlackWhite[4], refBlackWhite[5], 127);
199 }
200 #undef LumaRed
201 #undef LumaGreen
202 #undef LumaBlue
203 #undef V2Code
204
205 /*
206 * Convert a strip of RGB data to YCbCr and
207 * sample to generate the output data.
208 */
209 static void
210 cvtStrip(unsigned char* op, uint32* raster, uint32 nrows, uint32 width)
211 {
212 uint32 x;
213 int clumpSize = vertSubSampling * horizSubSampling + 2;
214 uint32 *tp;
215
216 for (; nrows >= vertSubSampling; nrows -= vertSubSampling) {
217 tp = raster;
218 for (x = width; x >= horizSubSampling; x -= horizSubSampling) {
219 cvtClump(op, tp,
220 vertSubSampling, horizSubSampling, width);
221 op += clumpSize;
222 tp += horizSubSampling;
223 }
224 if (x > 0) {
225 cvtClump(op, tp, vertSubSampling, x, width);
226 op += clumpSize;
227 }
228 raster -= vertSubSampling*width;
229 }
230 if (nrows > 0) {
231 tp = raster;
232 for (x = width; x >= horizSubSampling; x -= horizSubSampling) {
233 cvtClump(op, tp, nrows, horizSubSampling, width);
234 op += clumpSize;
235 tp += horizSubSampling;
236 }
237 if (x > 0)
238 cvtClump(op, tp, nrows, x, width);
239 }
240 }
241
242 static int
243 cvtRaster(TIFF* tif, uint32* raster, uint32 width, uint32 height)
244 {
245 uint32 y;
246 tstrip_t strip = 0;
247 tsize_t cc, acc;
248 unsigned char* buf;
249 uint32 rwidth = roundup(width, horizSubSampling);
250 uint32 rheight = roundup(height, vertSubSampling);
251 uint32 nrows = (rowsperstrip > rheight ? rheight : rowsperstrip);
252 uint32 rnrows = roundup(nrows,vertSubSampling);
253
254 cc = rnrows*rwidth +
255 2*((rnrows*rwidth) / (horizSubSampling*vertSubSampling));
256 buf = (unsigned char*)_TIFFmalloc(cc);
257 for (y = height; (int32) y > 0; y -= nrows) {
258 uint32 nr = (y > nrows ? nrows : y);
259 cvtStrip(buf, raster + (y-1)*width, nr, width);
260 nr = roundup(nr, vertSubSampling);
261 acc = nr*rwidth +
262 2*((nr*rwidth)/(horizSubSampling*vertSubSampling));
263 if (!TIFFWriteEncodedStrip(tif, strip++, buf, acc)) {
264 _TIFFfree(buf);
265 return (0);
266 }
267 }
268 _TIFFfree(buf);
269 return (1);
270 }
271
272 static int
273 tiffcvt(TIFF* in, TIFF* out)
274 {
275 uint32 width, height; /* image width & height */
276 uint32* raster; /* retrieve RGBA image */
277 uint16 shortv;
278 float floatv;
279 char *stringv;
280 uint32 longv;
281
282 TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
283 TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
284 raster = (uint32*)_TIFFmalloc(width * height * sizeof (uint32));
285 if (raster == 0) {
286 TIFFError(TIFFFileName(in), "No space for raster buffer");
287 return (0);
288 }
289 if (!TIFFReadRGBAImage(in, width, height, raster, 0)) {
290 _TIFFfree(raster);
291 return (0);
292 }
293
294 CopyField(TIFFTAG_SUBFILETYPE, longv);
295 TIFFSetField(out, TIFFTAG_IMAGEWIDTH, width);
296 TIFFSetField(out, TIFFTAG_IMAGELENGTH, height);
297 TIFFSetField(out, TIFFTAG_BITSPERSAMPLE, 8);
298 TIFFSetField(out, TIFFTAG_COMPRESSION, compression);
299 TIFFSetField(out, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_YCBCR);
300 if (compression == COMPRESSION_JPEG)
301 TIFFSetField(out, TIFFTAG_JPEGCOLORMODE, JPEGCOLORMODE_RAW);
302 CopyField(TIFFTAG_FILLORDER, shortv);
303 TIFFSetField(out, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
304 TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, 3);
305 CopyField(TIFFTAG_XRESOLUTION, floatv);
306 CopyField(TIFFTAG_YRESOLUTION, floatv);
307 CopyField(TIFFTAG_RESOLUTIONUNIT, shortv);
308 TIFFSetField(out, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
309 { char buf[2048];
310 char *cp = strrchr(TIFFFileName(in), '/');
311 sprintf(buf, "YCbCr conversion of %s", cp ? cp+1 : TIFFFileName(in));
312 TIFFSetField(out, TIFFTAG_IMAGEDESCRIPTION, buf);
313 }
314 TIFFSetField(out, TIFFTAG_SOFTWARE, TIFFGetVersion());
315 CopyField(TIFFTAG_DOCUMENTNAME, stringv);
316
317 TIFFSetField(out, TIFFTAG_REFERENCEBLACKWHITE, refBlackWhite);
318 TIFFSetField(out, TIFFTAG_YCBCRSUBSAMPLING,
319 horizSubSampling, vertSubSampling);
320 TIFFSetField(out, TIFFTAG_YCBCRPOSITIONING, YCBCRPOSITION_CENTERED);
321 TIFFSetField(out, TIFFTAG_YCBCRCOEFFICIENTS, ycbcrCoeffs);
322 rowsperstrip = TIFFDefaultStripSize(out, rowsperstrip);
323 TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip);
324
325 return (cvtRaster(out, raster, width, height));
326 }
327
328 char* stuff[] = {
329 "usage: rgb2ycbcr [-c comp] [-r rows] [-h N] [-v N] input... output\n",
330 "where comp is one of the following compression algorithms:\n",
331 " jpeg\t\tJPEG encoding\n",
332 " lzw\t\tLempel-Ziv & Welch encoding\n",
333 " zip\t\tdeflate encoding\n",
334 " packbits\tPackBits encoding (default)\n",
335 " none\t\tno compression\n",
336 "and the other options are:\n",
337 " -r\trows/strip\n",
338 " -h\thorizontal sampling factor (1,2,4)\n",
339 " -v\tvertical sampling factor (1,2,4)\n",
340 NULL
341 };
342
343 static void
344 usage(int code)
345 {
346 char buf[BUFSIZ];
347 int i;
348
349 setbuf(stderr, buf);
350
351 fprintf(stderr, "%s\n\n", TIFFGetVersion());
352 for (i = 0; stuff[i] != NULL; i++)
353 fprintf(stderr, "%s\n", stuff[i]);
354 exit(code);
355 }
356
357 /* vim: set ts=8 sts=8 sw=8 noet: */