This is rgb2ycbcr.c in view mode; [Download] [Up]
#ifndef lint
static char rcsid[] = "/mode/users/src/master/tiff/tools/rgb2ycbcr.c,v 1.1.1.1 1994/04/01 17:15:33 fedor Exp";
#endif
/*
* Copyright (c) 1991, 1992 Sam Leffler
* Copyright (c) 1991, 1992 Silicon Graphics, Inc.
*
* Permission to use, copy, modify, distribute, and sell this software and
* its documentation for any purpose is hereby granted without fee, provided
* that (i) the above copyright notices and this permission notice appear in
* all copies of the software and related documentation, and (ii) the names of
* Sam Leffler and Silicon Graphics may not be used in any advertising or
* publicity relating to the software without the specific, prior written
* permission of Sam Leffler and Silicon Graphics.
*
* THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
* EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
* WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
*
* IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
* ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
* OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
* WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
* LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
* OF THIS SOFTWARE.
*/
#include <stdio.h>
#include "tiffio.h"
#include <string.h>
typedef unsigned char u_char;
typedef unsigned short u_short;
typedef unsigned int u_int;
typedef unsigned long u_long;
#define streq(a,b) (strcmp(a,b) == 0)
#define CopyField(tag, v) \
if (TIFFGetField(in, tag, &v)) TIFFSetField(out, tag, v)
#define howmany(x, y) (((x)+((y)-1))/(y))
#define roundup(x, y) (howmany(x,y)*((u_int)(y)))
#define LumaRed ycbcrCoeffs[0]
#define LumaGreen ycbcrCoeffs[1]
#define LumaBlue ycbcrCoeffs[2]
u_short compression = COMPRESSION_LZW;
long rowsperstrip = -1;
u_short horizSubSampling = 2; /* YCbCr horizontal subsampling */
u_short vertSubSampling = 2; /* YCbCr vertical subsampling */
float ycbcrCoeffs[3] = { .299, .587, .114 };
/* default coding range is CCIR Rec 601-1 with no headroom/footroom */
float refBlackWhite[6] = { 0., 255., 128., 255., 128., 255. };
static int tiffcvt();
static void usage();
static void setupLumaTables();
main(argc, argv)
char *argv[];
{
TIFF *in, *out;
int c;
extern int optind;
extern char *optarg;
while ((c = getopt(argc, argv, "c:h:r:v:z")) != -1)
switch (c) {
case 'c':
if (streq(optarg, "none"))
compression = COMPRESSION_NONE;
else if (streq(optarg, "packbits"))
compression = COMPRESSION_PACKBITS;
else if (streq(optarg, "lzw"))
compression = COMPRESSION_LZW;
else
usage();
break;
case 'h':
horizSubSampling = atoi(optarg);
break;
case 'v':
vertSubSampling = atoi(optarg);
break;
case 'r':
rowsperstrip = atoi(optarg);
break;
case 'z': /* CCIR Rec 601-1 w/ headroom/footroom */
refBlackWhite[0] = 16.;
refBlackWhite[1] = 235.;
refBlackWhite[2] = 128.;
refBlackWhite[3] = 240.;
refBlackWhite[4] = 128.;
refBlackWhite[5] = 240.;
break;
case '?':
usage();
/*NOTREACHED*/
}
if (argc - optind < 2)
usage();
out = TIFFOpen(argv[argc-1], "w");
if (out == NULL)
exit(-2);
setupLumaTables();
for (; optind < argc-1; optind++) {
in = TIFFOpen(argv[optind], "r");
if (in != NULL) {
do {
if (!tiffcvt(in, out) ||
!TIFFWriteDirectory(out))
goto bad;
} while (TIFFReadDirectory(in));
(void) TIFFClose(in);
}
}
(void) TIFFClose(out);
exit(0);
bad:
unlink(TIFFFileName(out));
exit(1);
}
float *lumaRed;
float *lumaGreen;
float *lumaBlue;
float D1, D2;
int Yzero;
static float*
setupLuma(c)
float c;
{
float *v = (float *)malloc(256 * sizeof (float));
int i;
for (i = 0; i < 256; i++)
v[i] = c * i;
return (v);
}
static unsigned
V2Code(f, RB, RW, CR)
float f;
float RB, RW;
int CR;
{
unsigned int c = (unsigned int)((((f)*(RW-RB)/CR)+RB)+.5);
return (c > 255 ? 255 : c);
}
static void
setupLumaTables()
{
lumaRed = setupLuma(LumaRed);
lumaGreen = setupLuma(LumaGreen);
lumaBlue = setupLuma(LumaBlue);
D1 = 1./(2 - 2*LumaBlue);
D2 = 1./(2 - 2*LumaRed);
Yzero = V2Code(0, refBlackWhite[0], refBlackWhite[1], 255);
}
static void
cvtClump(op, raster, ch, cw, w)
u_char *op;
u_long *raster;
u_int ch, cw;
u_long w;
{
float Y, Cb = 0, Cr = 0;
int j, k;
/*
* Convert ch-by-cw block of RGB
* to YCbCr and sample accordingly.
*/
for (k = 0; k < ch; k++) {
for (j = 0; j < cw; j++) {
u_long RGB = (raster - k*w)[j];
Y = lumaRed[TIFFGetR(RGB)] +
lumaGreen[TIFFGetG(RGB)] +
lumaBlue[TIFFGetB(RGB)];
/* accumulate chrominance */
Cb += (TIFFGetB(RGB) - Y) * D1;
Cr += (TIFFGetR(RGB) - Y) * D2;
/* emit luminence */
*op++ = V2Code(Y,
refBlackWhite[0], refBlackWhite[1], 255);
}
for (; j < horizSubSampling; j++)
*op++ = Yzero;
}
for (; k < vertSubSampling; k++) {
for (j = 0; j < horizSubSampling; j++)
*op++ = Yzero;
}
/* emit sampled chrominance values */
*op++ = V2Code(Cb / (ch*cw), refBlackWhite[2], refBlackWhite[3], 127);
*op++ = V2Code(Cr / (ch*cw), refBlackWhite[4], refBlackWhite[5], 127);
}
#undef LumaRed
#undef LumaGreen
#undef LumaBlue
#undef V2Code
/*
* Convert a strip of RGB data to YCbCr and
* sample to generate the output data.
*/
static
cvtStrip(op, raster, nrows, width)
u_char *op;
u_long *raster;
u_long nrows, width;
{
long x, y;
int clumpSize = vertSubSampling * horizSubSampling + 2;
u_long *tp;
for (; nrows >= vertSubSampling; nrows -= vertSubSampling) {
tp = raster;
for (x = width; x >= horizSubSampling; x -= horizSubSampling) {
cvtClump(op, tp,
vertSubSampling, horizSubSampling, width);
op += clumpSize;
tp += horizSubSampling;
}
if (x > 0) {
cvtClump(op, tp, vertSubSampling, x, width);
op += clumpSize;
}
raster -= vertSubSampling*width;
}
if (nrows > 0) {
tp = raster;
for (x = width; x >= horizSubSampling; x -= horizSubSampling) {
cvtClump(op, tp, nrows, horizSubSampling, width);
op += clumpSize;
tp += horizSubSampling;
}
if (x > 0)
cvtClump(op, tp, nrows, x, width);
}
}
static int
cvtRaster(tif, raster, width, height)
TIFF *tif;
u_long *raster;
u_long width, height;
{
long y;
u_int strip = 0;
u_int cc;
u_char *buf;
u_long rwidth = roundup(width, horizSubSampling);
u_long rheight = roundup(height, vertSubSampling);
u_long nrows = (rowsperstrip > rheight ? rheight : rowsperstrip);
cc = nrows*rwidth +
2*((nrows*rwidth) / (horizSubSampling*vertSubSampling));
buf = (u_char *)malloc(cc);
for (y = height; y > 0; y -= nrows) {
u_long nr = (y > nrows ? nrows : y);
cvtStrip(buf, raster + (y-1)*width, nr, width, cc);
if (nr % vertSubSampling) {
u_int acc;
/*
* Short strip, zero data after image.
*/
nr = roundup(nr, vertSubSampling);
acc = nr*rwidth +
2*((nr*rwidth)/(horizSubSampling*vertSubSampling));
bzero(buf+acc, cc - acc);
}
if (!TIFFWriteEncodedStrip(tif, strip++, buf, cc)) {
free(buf);
return (0);
}
}
free(buf);
return (1);
}
static int
tiffcvt(in, out)
TIFF *in, *out;
{
u_long width, height; /* image width & height */
u_long *raster; /* retrieve RGBA image */
short shortv;
float floatv;
char *stringv;
u_long longv;
TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
raster = (u_long *)malloc(width * height * sizeof (u_long));
if (raster == 0) {
TIFFError(TIFFFileName(in), "No space for raster buffer");
return (0);
}
if (!TIFFReadRGBAImage(in, width, height, raster, 0)) {
free(raster);
return (0);
}
CopyField(TIFFTAG_SUBFILETYPE, longv);
TIFFSetField(out, TIFFTAG_IMAGEWIDTH, width);
TIFFSetField(out, TIFFTAG_IMAGELENGTH, height);
TIFFSetField(out, TIFFTAG_BITSPERSAMPLE, 8);
TIFFSetField(out, TIFFTAG_COMPRESSION, compression);
TIFFSetField(out, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_YCBCR);
CopyField(TIFFTAG_FILLORDER, shortv);
TIFFSetField(out, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, 3);
CopyField(TIFFTAG_XRESOLUTION, floatv);
CopyField(TIFFTAG_YRESOLUTION, floatv);
CopyField(TIFFTAG_RESOLUTIONUNIT, shortv);
TIFFSetField(out, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
if (rowsperstrip <= 0)
rowsperstrip = (8*1024)/TIFFScanlineSize(out);
rowsperstrip = roundup(rowsperstrip, vertSubSampling);
TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip);
{ char buf[2048];
char *cp = strrchr(TIFFFileName(in), '/');
if (!cp)
cp = TIFFFileName(in);
else
cp++;
sprintf(buf, "YCbCr conversion of %s", cp);
TIFFSetField(out, TIFFTAG_IMAGEDESCRIPTION, buf);
}
TIFFSetField(out, TIFFTAG_SOFTWARE, TIFFVersion);
CopyField(TIFFTAG_DOCUMENTNAME, stringv);
TIFFSetField(out, TIFFTAG_REFERENCEBLACKWHITE, refBlackWhite);
TIFFSetField(out, TIFFTAG_YCBCRSUBSAMPLING,
horizSubSampling, vertSubSampling);
TIFFSetField(out, TIFFTAG_YCBCRPOSITIONING, YCBCRPOSITION_CENTERED);
TIFFSetField(out, TIFFTAG_YCBCRCOEFFICIENTS, ycbcrCoeffs);
return (cvtRaster(out, raster, width, height));
}
static char* usageMsg[] = {
"usage: rgb2ycbcr [-c comp] [-r rows] [-h N] [-v N] input... output\n",
"where comp is one of the following compression algorithms:\n",
" lzw\t\tLempel-Ziv & Welch encoding\n",
" packbits\tPackBits encoding\n",
" none\t\tno compression\n",
"and the other options are:\n",
" -r\trows/strip\n",
" -h\thorizontal sampling factor (1,2,4)\n",
" -v\tvertical sampling factor (1,2,4)\n",
NULL
};
static void
usage()
{
int i;
for (i = 0; usageMsg[i]; i++)
fprintf(stderr, "%s", usageMsg[i]);
exit(-1);
}
These are the contents of the former NiCE NeXT User Group NeXTSTEP/OpenStep software archive, currently hosted by Netfuture.ch.