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: */