tjunittest.c - platform/external/libjpeg-turbo - Gitiles

 /*
  * Copyright (C)2009-2014 D. R. Commander.  All Rights Reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are met:
  *
  * - Redistributions of source code must retain the above copyright notice,
  *   this list of conditions and the following disclaimer.
  * - Redistributions in binary form must reproduce the above copyright notice,
  *   this list of conditions and the following disclaimer in the documentation
  *   and/or other materials provided with the distribution.
  * - Neither the name of the libjpeg-turbo Project nor the names of its
  *   contributors may be used to endorse or promote products derived from this
  *   software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS",
  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  * ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
  * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  * POSSIBILITY OF SUCH DAMAGE.
  */

 /*
  * This program tests the various code paths in the TurboJPEG C Wrapper
  */

 #include <stdio.h>
 #include <stdlib.h>
 #include <string.h>
 #include <errno.h>
 #include "./tjutil.h"
 #include "./turbojpeg.h"
 #ifdef _WIN32
  #include <time.h>
  #define random() rand()
 #endif


 void usage(char *progName)
 {
 	printf("\nUSAGE: %s [options]\n", progName);
 	printf("Options:\n");
 	printf("-yuv = test YUV encoding/decoding support\n");
 	printf("-noyuvpad = do not pad each line of each Y, U, and V plane to the nearest\n");
 	printf("            4-byte boundary\n");
 	printf("-alloc = test automatic buffer allocation\n");
 	exit(1);
 }


 #define _throwtj() {printf("TurboJPEG ERROR:\n%s\n", tjGetErrorStr());  \
 	bailout();}
 #define _tj(f) {if((f)==-1) _throwtj();}
 #define _throw(m) {printf("ERROR: %s\n", m);  bailout();}

 const char *subNameLong[TJ_NUMSAMP]=
 {
 	"4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1"
 };
 const char *subName[TJ_NUMSAMP]={"444", "422", "420", "GRAY", "440", "411"};

 const char *pixFormatStr[TJ_NUMPF]=
 {
 	"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale",
 	"RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
 };

 const int alphaOffset[TJ_NUMPF] = {-1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0, -1};

 const int _3byteFormats[]={TJPF_RGB, TJPF_BGR};
 const int _4byteFormats[]={TJPF_RGBX, TJPF_BGRX, TJPF_XBGR, TJPF_XRGB,
 	TJPF_CMYK};
 const int _onlyGray[]={TJPF_GRAY};
 const int _onlyRGB[]={TJPF_RGB};

 int doyuv=0, alloc=0, pad=4;

 int exitStatus=0;
 #define bailout() {exitStatus=-1;  goto bailout;}


 void initBuf(unsigned char *buf, int w, int h, int pf, int flags)
 {
 	int roffset=tjRedOffset[pf];
 	int goffset=tjGreenOffset[pf];
 	int boffset=tjBlueOffset[pf];
 	int ps=tjPixelSize[pf];
 	int index, row, col, halfway=16;

 	if(pf==TJPF_GRAY)
 	{
 		memset(buf, 0, w*h*ps);
 		for(row=0; row<h; row++)
 		{
 			for(col=0; col<w; col++)
 			{
 				if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
 				else index=row*w+col;
 				if(((row/8)+(col/8))%2==0) buf[index]=(row<halfway)? 255:0;
 				else buf[index]=(row<halfway)? 76:226;
 			}
 		}
 	}
 	else if(pf==TJPF_CMYK)
 	{
 		memset(buf, 255, w*h*ps);
 		for(row=0; row<h; row++)
 		{
 			for(col=0; col<w; col++)
 			{
 				if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
 				else index=row*w+col;
 				if(((row/8)+(col/8))%2==0)
 				{
 					if(row>=halfway) buf[index*ps+3]=0;
 				}
 				else
 				{
 					buf[index*ps+2]=0;
 					if(row<halfway) buf[index*ps+1]=0;
 				}
 			}
 		}
 	}
 	else
 	{
 		memset(buf, 0, w*h*ps);
 		for(row=0; row<h; row++)
 		{
 			for(col=0; col<w; col++)
 			{
 				if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
 				else index=row*w+col;
 				if(((row/8)+(col/8))%2==0)
 				{
 					if(row<halfway)
 					{
 						buf[index*ps+roffset]=255;
 						buf[index*ps+goffset]=255;
 						buf[index*ps+boffset]=255;
 					}
 				}
 				else
 				{
 					buf[index*ps+roffset]=255;
 					if(row>=halfway) buf[index*ps+goffset]=255;
 				}
 			}
 		}
 	}
 }


 #define checkval(v, cv) { \
 	if(v<cv-1 || v>cv+1) { \
 		printf("\nComp. %s at %d,%d should be %d, not %d\n",  \
 			#v, row, col, cv, v); \
 		retval=0;  exitStatus=-1;  goto bailout; \
 	}}

 #define checkval0(v) { \
 	if(v>1) { \
 		printf("\nComp. %s at %d,%d should be 0, not %d\n", #v, row, col, v); \
 		retval=0;  exitStatus=-1;  goto bailout; \
 	}}

 #define checkval255(v) { \
 	if(v<254) { \
 		printf("\nComp. %s at %d,%d should be 255, not %d\n", #v, row, col, v); \
 		retval=0;  exitStatus=-1;  goto bailout; \
 	}}


 int checkBuf(unsigned char *buf, int w, int h, int pf, int subsamp,
 	tjscalingfactor sf, int flags)
 {
 	int roffset=tjRedOffset[pf];
 	int goffset=tjGreenOffset[pf];
 	int boffset=tjBlueOffset[pf];
 	int aoffset=alphaOffset[pf];
 	int ps=tjPixelSize[pf];
 	int index, row, col, retval=1;
 	int halfway=16*sf.num/sf.denom;
 	int blocksize=8*sf.num/sf.denom;

 	if(pf==TJPF_CMYK)
 	{
 		for(row=0; row<h; row++)
 		{
 			for(col=0; col<w; col++)
 			{
 				unsigned char c, m, y, k;
 				if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
 				else index=row*w+col;
 				c=buf[index*ps];
 				m=buf[index*ps+1];
 				y=buf[index*ps+2];
 				k=buf[index*ps+3];
 				if(((row/blocksize)+(col/blocksize))%2==0)
 				{
 					checkval255(c);  checkval255(m);  checkval255(y);
 					if(row<halfway) checkval255(k)
 					else checkval0(k)
 				}
 				else
 				{
 					checkval255(c);  checkval0(y);  checkval255(k);
 					if(row<halfway) checkval0(m)
 					else checkval255(m)
 				}
 			}
 		}
 		return 1;
 	}

 	for(row=0; row<h; row++)
 	{
 		for(col=0; col<w; col++)
 		{
 			unsigned char r, g, b, a;
 			if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
 			else index=row*w+col;
 			r=buf[index*ps+roffset];
 			g=buf[index*ps+goffset];
 			b=buf[index*ps+boffset];
 			a=aoffset>=0? buf[index*ps+aoffset]:0xFF;
 			if(((row/blocksize)+(col/blocksize))%2==0)
 			{
 				if(row<halfway)
 				{
 					checkval255(r);  checkval255(g);  checkval255(b);
 				}
 				else
 				{
 					checkval0(r);  checkval0(g);  checkval0(b);
 				}
 			}
 			else
 			{
 				if(subsamp==TJSAMP_GRAY)
 				{
 					if(row<halfway)
 					{
 						checkval(r, 76);  checkval(g, 76);  checkval(b, 76);
 					}
 					else
 					{
 						checkval(r, 226);  checkval(g, 226);  checkval(b, 226);
 					}
 				}
 				else
 				{
 					if(row<halfway)
 					{
 						checkval255(r);  checkval0(g);  checkval0(b);
 					}
 					else
 					{
 						checkval255(r);  checkval255(g);  checkval0(b);
 					}
 				}
 			}
 			checkval255(a);
 		}
 	}

 	bailout:
 	if(retval==0)
 	{
 		for(row=0; row<h; row++)
 		{
 			for(col=0; col<w; col++)
 			{
 				if(pf==TJPF_CMYK)
 					printf("%.3d/%.3d/%.3d/%.3d ", buf[(row*w+col)*ps],
 						buf[(row*w+col)*ps+1], buf[(row*w+col)*ps+2],
 						buf[(row*w+col)*ps+3]);
 				else
 					printf("%.3d/%.3d/%.3d ", buf[(row*w+col)*ps+roffset],
 						buf[(row*w+col)*ps+goffset], buf[(row*w+col)*ps+boffset]);
 			}
 			printf("\n");
 		}
 	}
 	return retval;
 }


 #define PAD(v, p) ((v+(p)-1)&(~((p)-1)))

 int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
 	tjscalingfactor sf)
 {
 	int row, col;
 	int hsf=tjMCUWidth[subsamp]/8, vsf=tjMCUHeight[subsamp]/8;
 	int pw=PAD(w, hsf), ph=PAD(h, vsf);
 	int cw=pw/hsf, ch=ph/vsf;
 	int ypitch=PAD(pw, pad), uvpitch=PAD(cw, pad);
 	int retval=1;
 	int halfway=16*sf.num/sf.denom;
 	int blocksize=8*sf.num/sf.denom;

 	for(row=0; row<ph; row++)
 	{
 		for(col=0; col<pw; col++)
 		{
 			unsigned char y=buf[ypitch*row+col];
 			if(((row/blocksize)+(col/blocksize))%2==0)
 			{
 				if(row<halfway) checkval255(y)  else checkval0(y);
 			}
 			else
 			{
 				if(row<halfway) checkval(y, 76)  else checkval(y, 226);
 			}
 		}
 	}
 	if(subsamp!=TJSAMP_GRAY)
 	{
 		int halfway=16/vsf*sf.num/sf.denom;
 		for(row=0; row<ch; row++)
 		{
 			for(col=0; col<cw; col++)
 			{
 				unsigned char u=buf[ypitch*ph + (uvpitch*row+col)],
 					v=buf[ypitch*ph + uvpitch*ch + (uvpitch*row+col)];
 				if(((row*vsf/blocksize)+(col*hsf/blocksize))%2==0)
 				{
 					checkval(u, 128);  checkval(v, 128);
 				}
 				else
 				{
 					if(row<halfway)
 					{
 						checkval(u, 85);  checkval255(v);
 					}
 					else
 					{
 						checkval0(u);  checkval(v, 149);
 					}
 				}
 			}
 		}
 	}

 	bailout:
 	if(retval==0)
 	{
 		for(row=0; row<ph; row++)
 		{
 			for(col=0; col<pw; col++)
 				printf("%.3d ", buf[ypitch*row+col]);
 			printf("\n");
 		}
 		printf("\n");
 		for(row=0; row<ch; row++)
 		{
 			for(col=0; col<cw; col++)
 				printf("%.3d ", buf[ypitch*ph + (uvpitch*row+col)]);
 			printf("\n");
 		}
 		printf("\n");
 		for(row=0; row<ch; row++)
 		{
 			for(col=0; col<cw; col++)
 				printf("%.3d ", buf[ypitch*ph + uvpitch*ch + (uvpitch*row+col)]);
 			printf("\n");
 		}
 	}

 	return retval;
 }


 void writeJPEG(unsigned char *jpegBuf, unsigned long jpegSize, char *filename)
 {
 	FILE *file=fopen(filename, "wb");
 	if(!file || fwrite(jpegBuf, jpegSize, 1, file)!=1)
 	{
 		printf("ERROR: Could not write to %s.\n%s\n", filename, strerror(errno));
 		bailout();
 	}

 	bailout:
 	if(file) fclose(file);
 }


 void compTest(tjhandle handle, unsigned char **dstBuf,
 	unsigned long *dstSize, int w, int h, int pf, char *basename,
 	int subsamp, int jpegQual, int flags)
 {
 	char tempStr[1024];  unsigned char *srcBuf=NULL, *yuvBuf=NULL;
 	const char *pfStr=pixFormatStr[pf];
 	const char *buStrLong=(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ";
 	const char *buStr=(flags&TJFLAG_BOTTOMUP)? "BU":"TD";

 	if((srcBuf=(unsigned char *)malloc(w*h*tjPixelSize[pf]))==NULL)
 		_throw("Memory allocation failure");
 	initBuf(srcBuf, w, h, pf, flags);

 	if(*dstBuf && *dstSize>0) memset(*dstBuf, 0, *dstSize);


 	if(!alloc) flags|=TJFLAG_NOREALLOC;
 	if(doyuv)
 	{
 		unsigned long yuvSize=tjBufSizeYUV2(w, pad, h, subsamp);
 		tjscalingfactor sf={1, 1};
 		tjhandle handle2=tjInitCompress();
 		if(!handle2) _throwtj();

 		if((yuvBuf=(unsigned char *)malloc(yuvSize))==NULL)
 			_throw("Memory allocation failure");
 		memset(yuvBuf, 0, yuvSize);

 		printf("%s %s -> YUV %s ... ", pfStr, buStrLong, subNameLong[subsamp]);
 		_tj(tjEncodeYUV3(handle2, srcBuf, w, 0, h, pf, yuvBuf, pad, subsamp,
 			flags));
 		tjDestroy(handle2);
 		if(checkBufYUV(yuvBuf, w, h, subsamp, sf)) printf("Passed.\n");
 		else printf("FAILED!\n");

 		printf("YUV %s %s -> JPEG Q%d ... ", subNameLong[subsamp], buStrLong,
 			jpegQual);
 		_tj(tjCompressFromYUV(handle, yuvBuf, w, pad, h, subsamp, dstBuf,
 			dstSize, jpegQual, flags));
 	}
 	else
 	{
 		printf("%s %s -> %s Q%d ... ", pfStr, buStrLong, subNameLong[subsamp],
 			jpegQual);
 		_tj(tjCompress2(handle, srcBuf, w, 0, h, pf, dstBuf, dstSize, subsamp,
 			jpegQual, flags));
 	}

 	snprintf(tempStr, 1024, "%s_enc_%s_%s_%s_Q%d.jpg", basename, pfStr, buStr,
 		subName[subsamp], jpegQual);
 	writeJPEG(*dstBuf, *dstSize, tempStr);
 	printf("Done.\n  Result in %s\n", tempStr);

 	bailout:
 	if(yuvBuf) free(yuvBuf);
 	if(srcBuf) free(srcBuf);
 }


 void _decompTest(tjhandle handle, unsigned char *jpegBuf,
 	unsigned long jpegSize, int w, int h, int pf, char *basename, int subsamp,
 	int flags, tjscalingfactor sf)
 {
 	unsigned char *dstBuf=NULL, *yuvBuf=NULL;
 	int _hdrw=0, _hdrh=0, _hdrsubsamp=-1;
 	int scaledWidth=TJSCALED(w, sf);
 	int scaledHeight=TJSCALED(h, sf);
 	unsigned long dstSize=0;

 	_tj(tjDecompressHeader2(handle, jpegBuf, jpegSize, &_hdrw, &_hdrh,
 		&_hdrsubsamp));
 	if(_hdrw!=w || _hdrh!=h || _hdrsubsamp!=subsamp)
 		_throw("Incorrect JPEG header");

 	dstSize=scaledWidth*scaledHeight*tjPixelSize[pf];
 	if((dstBuf=(unsigned char *)malloc(dstSize))==NULL)
 		_throw("Memory allocation failure");
 	memset(dstBuf, 0, dstSize);

 	if(doyuv)
 	{
 		unsigned long yuvSize=tjBufSizeYUV2(scaledWidth, pad, scaledHeight,
 			subsamp);
 		tjhandle handle2=tjInitDecompress();
 		if(!handle2) _throwtj();

 		if((yuvBuf=(unsigned char *)malloc(yuvSize))==NULL)
 			_throw("Memory allocation failure");
 		memset(yuvBuf, 0, yuvSize);

 		printf("JPEG -> YUV %s ", subNameLong[subsamp]);
 		if(sf.num!=1 || sf.denom!=1)
 			printf("%d/%d ... ", sf.num, sf.denom);
 		else printf("... ");
 		_tj(tjDecompressToYUV2(handle, jpegBuf, jpegSize, yuvBuf, scaledWidth,
 			pad, scaledHeight, flags));
 		if(checkBufYUV(yuvBuf, scaledWidth, scaledHeight, subsamp, sf))
 			printf("Passed.\n");
 		else printf("FAILED!\n");

 		printf("YUV %s -> %s %s ... ", subNameLong[subsamp], pixFormatStr[pf],
 			(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ");
 		_tj(tjDecodeYUV(handle2, yuvBuf, pad, subsamp, dstBuf, scaledWidth, 0,
 			scaledHeight, pf, flags));
 		tjDestroy(handle2);
 	}
 	else
 	{
 		printf("JPEG -> %s %s ", pixFormatStr[pf],
 			(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ");
 		if(sf.num!=1 || sf.denom!=1)
 			printf("%d/%d ... ", sf.num, sf.denom);
 		else printf("... ");
 		_tj(tjDecompress2(handle, jpegBuf, jpegSize, dstBuf, scaledWidth, 0,
 			scaledHeight, pf, flags));
 	}

 	if(checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, flags))
 		printf("Passed.");
 	else printf("FAILED!");
 	printf("\n");

 	bailout:
 	if(yuvBuf) free(yuvBuf);
 	if(dstBuf) free(dstBuf);
 }


 void decompTest(tjhandle handle, unsigned char *jpegBuf,
 	unsigned long jpegSize, int w, int h, int pf, char *basename, int subsamp,
 	int flags)
 {
 	int i, n=0;
 	tjscalingfactor *sf=tjGetScalingFactors(&n);
 	if(!sf || !n) _throwtj();

 	for(i=0; i<n; i++)
 	{
 		if(subsamp==TJSAMP_444 || subsamp==TJSAMP_GRAY ||
 			(subsamp==TJSAMP_411 && sf[i].num==1 &&
 				(sf[i].denom==2 || sf[i].denom==1)) ||
 			(subsamp!=TJSAMP_411 && sf[i].num==1 &&
 				(sf[i].denom==4 || sf[i].denom==2 || sf[i].denom==1)))
 			_decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp,
 				flags, sf[i]);
 	}

 	bailout:
 	return;
 }


 void doTest(int w, int h, const int *formats, int nformats, int subsamp,
 	char *basename)
 {
 	tjhandle chandle=NULL, dhandle=NULL;
 	unsigned char *dstBuf=NULL;
 	unsigned long size=0;  int pfi, pf, i;

 	if(!alloc)
 		size=tjBufSize(w, h, subsamp);
 	if(size!=0)
 		if((dstBuf=(unsigned char *)tjAlloc(size))==NULL)
 			_throw("Memory allocation failure.");

 	if((chandle=tjInitCompress())==NULL || (dhandle=tjInitDecompress())==NULL)
 		_throwtj();

 	for(pfi=0; pfi<nformats; pfi++)
 	{
 		for(i=0; i<2; i++)
 		{
 			int flags=0;
 			if(subsamp==TJSAMP_422 || subsamp==TJSAMP_420 || subsamp==TJSAMP_440 ||
 				subsamp==TJSAMP_411)
 				flags|=TJFLAG_FASTUPSAMPLE;
 			if(i==1) flags|=TJFLAG_BOTTOMUP;
 			pf=formats[pfi];
 			compTest(chandle, &dstBuf, &size, w, h, pf, basename, subsamp, 100,
 				flags);
 			decompTest(dhandle, dstBuf, size, w, h, pf, basename, subsamp,
 				flags);
 			if(pf>=TJPF_RGBX && pf<=TJPF_XRGB)
 			{
 				printf("\n");
 				decompTest(dhandle, dstBuf, size, w, h, pf+(TJPF_RGBA-TJPF_RGBX),
 					basename, subsamp, flags);
 			}
 			printf("\n");
 		}
 	}
 	printf("--------------------\n\n");

 	bailout:
 	if(chandle) tjDestroy(chandle);
 	if(dhandle) tjDestroy(dhandle);

 	if(dstBuf) tjFree(dstBuf);
 }


 void bufSizeTest(void)
 {
 	int w, h, i, subsamp;
 	unsigned char *srcBuf=NULL, *dstBuf=NULL;
 	tjhandle handle=NULL;
 	unsigned long dstSize=0;

 	if((handle=tjInitCompress())==NULL) _throwtj();

 	printf("Buffer size regression test\n");
 	for(subsamp=0; subsamp<TJ_NUMSAMP; subsamp++)
 	{
 		for(w=1; w<48; w++)
 		{
 			int maxh=(w==1)? 2048:48;
 			for(h=1; h<maxh; h++)
 			{
 				if(h%100==0) printf("%.4d x %.4d\b\b\b\b\b\b\b\b\b\b\b", w, h);
 				if((srcBuf=(unsigned char *)malloc(w*h*4))==NULL)
 					_throw("Memory allocation failure");
 				if(!alloc || doyuv)
 				{
 					if(doyuv) dstSize=tjBufSizeYUV2(w, pad, h, subsamp);
 					else dstSize=tjBufSize(w, h, subsamp);
 					if((dstBuf=(unsigned char *)tjAlloc(dstSize))==NULL)
 						_throw("Memory allocation failure");
 				}

 				for(i=0; i<w*h*4; i++)
 				{
 					if(random()<RAND_MAX/2) srcBuf[i]=0;
 					else srcBuf[i]=255;
 				}

 				if(doyuv)
 				{
 					_tj(tjEncodeYUV3(handle, srcBuf, w, 0, h, TJPF_BGRX, dstBuf, pad,
 						subsamp, 0));
 				}
 				else
 				{
 					_tj(tjCompress2(handle, srcBuf, w, 0, h, TJPF_BGRX, &dstBuf,
 						&dstSize, subsamp, 100, alloc? 0:TJFLAG_NOREALLOC));
 				}
 				free(srcBuf);  srcBuf=NULL;
 				tjFree(dstBuf);  dstBuf=NULL;

 				if((srcBuf=(unsigned char *)malloc(h*w*4))==NULL)
 					_throw("Memory allocation failure");
 				if(!alloc || doyuv)
 				{
 					if(doyuv) dstSize=tjBufSizeYUV2(h, pad, w, subsamp);
 					else dstSize=tjBufSize(h, w, subsamp);
 					if((dstBuf=(unsigned char *)tjAlloc(dstSize))==NULL)
 						_throw("Memory allocation failure");
 				}

 				for(i=0; i<h*w*4; i++)
 				{
 					if(random()<RAND_MAX/2) srcBuf[i]=0;
 					else srcBuf[i]=255;
 				}

 				if(doyuv)
 				{
 					_tj(tjEncodeYUV3(handle, srcBuf, h, 0, w, TJPF_BGRX, dstBuf, pad,
 						subsamp, 0));
 				}
 				else
 				{
 					_tj(tjCompress2(handle, srcBuf, h, 0, w, TJPF_BGRX, &dstBuf,
 						&dstSize, subsamp, 100, alloc? 0:TJFLAG_NOREALLOC));
 				}
 				free(srcBuf);  srcBuf=NULL;
 				tjFree(dstBuf);  dstBuf=NULL;
 			}
 		}
 	}
 	printf("Done.      \n");

 	bailout:
 	if(srcBuf) free(srcBuf);
 	if(dstBuf) free(dstBuf);
 	if(handle) tjDestroy(handle);
 }


 int main(int argc, char *argv[])
 {
 	int i, num4bf=5;
 	#ifdef _WIN32
 	srand((unsigned int)time(NULL));
 	#endif
 	if(argc>1)
 	{
 		for(i=1; i<argc; i++)
 		{
 			if(!strcasecmp(argv[i], "-yuv")) doyuv=1;
 			if(!strcasecmp(argv[i], "-noyuvpad")) pad=1;
 			if(!strcasecmp(argv[i], "-alloc")) alloc=1;
 			if(!strncasecmp(argv[i], "-h", 2) || !strcasecmp(argv[i], "-?"))
 				usage(argv[0]);
 		}
 	}
 	if(alloc) printf("Testing automatic buffer allocation\n");
 	if(doyuv) num4bf=4;
 	doTest(35, 39, _3byteFormats, 2, TJSAMP_444, "test");
 	doTest(39, 41, _4byteFormats, num4bf, TJSAMP_444, "test");
 	doTest(41, 35, _3byteFormats, 2, TJSAMP_422, "test");
 	doTest(35, 39, _4byteFormats, num4bf, TJSAMP_422, "test");
 	doTest(39, 41, _3byteFormats, 2, TJSAMP_420, "test");
 	doTest(41, 35, _4byteFormats, num4bf, TJSAMP_420, "test");
 	doTest(35, 39, _3byteFormats, 2, TJSAMP_440, "test");
 	doTest(39, 41, _4byteFormats, num4bf, TJSAMP_440, "test");
 	doTest(41, 35, _3byteFormats, 2, TJSAMP_411, "test");
 	doTest(35, 39, _4byteFormats, num4bf, TJSAMP_411, "test");
 	doTest(39, 41, _onlyGray, 1, TJSAMP_GRAY, "test");
 	doTest(41, 35, _3byteFormats, 2, TJSAMP_GRAY, "test");
 	doTest(35, 39, _4byteFormats, 4, TJSAMP_GRAY, "test");
 	bufSizeTest();
 	if(doyuv)
 	{
 		printf("\n--------------------\n\n");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_444, "test_yuv0");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_422, "test_yuv0");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_420, "test_yuv0");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_440, "test_yuv0");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_411, "test_yuv0");
 		doTest(48, 48, _onlyRGB, 1, TJSAMP_GRAY, "test_yuv0");
 		doTest(48, 48, _onlyGray, 1, TJSAMP_GRAY, "test_yuv0");
 	}

 	return exitStatus;
 }
	/*
	* Copyright (C)2009-2014 D. R. Commander. All Rights Reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions are met:
	*
	* - Redistributions of source code must retain the above copyright notice,
	* this list of conditions and the following disclaimer.
	* - Redistributions in binary form must reproduce the above copyright notice,
	* this list of conditions and the following disclaimer in the documentation
	* and/or other materials provided with the distribution.
	* - Neither the name of the libjpeg-turbo Project nor the names of its
	* contributors may be used to endorse or promote products derived from this
	* software without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS",
	* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	* ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
	* LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
	* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
	* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
	* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
	* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
	* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
	* POSSIBILITY OF SUCH DAMAGE.
	*/

	/*
	* This program tests the various code paths in the TurboJPEG C Wrapper
	*/

	#include <stdio.h>
	#include <stdlib.h>
	#include <string.h>
	#include <errno.h>
	#include "./tjutil.h"
	#include "./turbojpeg.h"
	#ifdef _WIN32
	#include <time.h>
	#define random() rand()
	#endif


	void usage(char *progName)
	{
	printf("\nUSAGE: %s [options]\n", progName);
	printf("Options:\n");
	printf("-yuv = test YUV encoding/decoding support\n");
	printf("-noyuvpad = do not pad each line of each Y, U, and V plane to the nearest\n");
	printf(" 4-byte boundary\n");
	printf("-alloc = test automatic buffer allocation\n");
	exit(1);
	}


	#define _throwtj() {printf("TurboJPEG ERROR:\n%s\n", tjGetErrorStr()); \
	bailout();}
	#define _tj(f) {if((f)==-1) _throwtj();}
	#define _throw(m) {printf("ERROR: %s\n", m); bailout();}

	const char *subNameLong[TJ_NUMSAMP]=
	{
	"4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1"
	};
	const char *subName[TJ_NUMSAMP]={"444", "422", "420", "GRAY", "440", "411"};

	const char *pixFormatStr[TJ_NUMPF]=
	{
	"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale",
	"RGBA", "BGRA", "ABGR", "ARGB", "CMYK"
	};

	const int alphaOffset[TJ_NUMPF] = {-1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0, -1};

	const int _3byteFormats[]={TJPF_RGB, TJPF_BGR};
	const int _4byteFormats[]={TJPF_RGBX, TJPF_BGRX, TJPF_XBGR, TJPF_XRGB,
	TJPF_CMYK};
	const int _onlyGray[]={TJPF_GRAY};
	const int _onlyRGB[]={TJPF_RGB};

	int doyuv=0, alloc=0, pad=4;

	int exitStatus=0;
	#define bailout() {exitStatus=-1; goto bailout;}


	void initBuf(unsigned char *buf, int w, int h, int pf, int flags)
	{
	int roffset=tjRedOffset[pf];
	int goffset=tjGreenOffset[pf];
	int boffset=tjBlueOffset[pf];
	int ps=tjPixelSize[pf];
	int index, row, col, halfway=16;

	if(pf==TJPF_GRAY)
	{
	memset(buf, 0, whps);
	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
	else index=row*w+col;
	if(((row/8)+(col/8))%2==0) buf[index]=(row<halfway)? 255:0;
	else buf[index]=(row<halfway)? 76:226;
	}
	}
	}
	else if(pf==TJPF_CMYK)
	{
	memset(buf, 255, whps);
	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
	else index=row*w+col;
	if(((row/8)+(col/8))%2==0)
	{
	if(row>=halfway) buf[index*ps+3]=0;
	}
	else
	{
	buf[index*ps+2]=0;
	if(row<halfway) buf[index*ps+1]=0;
	}
	}
	}
	}
	else
	{
	memset(buf, 0, whps);
	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
	else index=row*w+col;
	if(((row/8)+(col/8))%2==0)
	{
	if(row<halfway)
	{
	buf[index*ps+roffset]=255;
	buf[index*ps+goffset]=255;
	buf[index*ps+boffset]=255;
	}
	}
	else
	{
	buf[index*ps+roffset]=255;
	if(row>=halfway) buf[index*ps+goffset]=255;
	}
	}
	}
	}
	}


	#define checkval(v, cv) { \
	if(v<cv-1 \|\| v>cv+1) { \
	printf("\nComp. %s at %d,%d should be %d, not %d\n", \
	#v, row, col, cv, v); \
	retval=0; exitStatus=-1; goto bailout; \
	}}

	#define checkval0(v) { \
	if(v>1) { \
	printf("\nComp. %s at %d,%d should be 0, not %d\n", #v, row, col, v); \
	retval=0; exitStatus=-1; goto bailout; \
	}}

	#define checkval255(v) { \
	if(v<254) { \
	printf("\nComp. %s at %d,%d should be 255, not %d\n", #v, row, col, v); \
	retval=0; exitStatus=-1; goto bailout; \
	}}


	int checkBuf(unsigned char *buf, int w, int h, int pf, int subsamp,
	tjscalingfactor sf, int flags)
	{
	int roffset=tjRedOffset[pf];
	int goffset=tjGreenOffset[pf];
	int boffset=tjBlueOffset[pf];
	int aoffset=alphaOffset[pf];
	int ps=tjPixelSize[pf];
	int index, row, col, retval=1;
	int halfway=16*sf.num/sf.denom;
	int blocksize=8*sf.num/sf.denom;

	if(pf==TJPF_CMYK)
	{
	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	unsigned char c, m, y, k;
	if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
	else index=row*w+col;
	c=buf[index*ps];
	m=buf[index*ps+1];
	y=buf[index*ps+2];
	k=buf[index*ps+3];
	if(((row/blocksize)+(col/blocksize))%2==0)
	{
	checkval255(c); checkval255(m); checkval255(y);
	if(row<halfway) checkval255(k)
	else checkval0(k)
	}
	else
	{
	checkval255(c); checkval0(y); checkval255(k);
	if(row<halfway) checkval0(m)
	else checkval255(m)
	}
	}
	}
	return 1;
	}

	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	unsigned char r, g, b, a;
	if(flags&TJFLAG_BOTTOMUP) index=(h-row-1)*w+col;
	else index=row*w+col;
	r=buf[index*ps+roffset];
	g=buf[index*ps+goffset];
	b=buf[index*ps+boffset];
	a=aoffset>=0? buf[index*ps+aoffset]:0xFF;
	if(((row/blocksize)+(col/blocksize))%2==0)
	{
	if(row<halfway)
	{
	checkval255(r); checkval255(g); checkval255(b);
	}
	else
	{
	checkval0(r); checkval0(g); checkval0(b);
	}
	}
	else
	{
	if(subsamp==TJSAMP_GRAY)
	{
	if(row<halfway)
	{
	checkval(r, 76); checkval(g, 76); checkval(b, 76);
	}
	else
	{
	checkval(r, 226); checkval(g, 226); checkval(b, 226);
	}
	}
	else
	{
	if(row<halfway)
	{
	checkval255(r); checkval0(g); checkval0(b);
	}
	else
	{
	checkval255(r); checkval255(g); checkval0(b);
	}
	}
	}
	checkval255(a);
	}
	}

	bailout:
	if(retval==0)
	{
	for(row=0; row<h; row++)
	{
	for(col=0; col<w; col++)
	{
	if(pf==TJPF_CMYK)
	printf("%.3d/%.3d/%.3d/%.3d ", buf[(roww+col)ps],
	buf[(roww+col)ps+1], buf[(roww+col)ps+2],
	buf[(roww+col)ps+3]);
	else
	printf("%.3d/%.3d/%.3d ", buf[(roww+col)ps+roffset],
	buf[(roww+col)ps+goffset], buf[(roww+col)ps+boffset]);
	}
	printf("\n");
	}
	}
	return retval;
	}


	#define PAD(v, p) ((v+(p)-1)&(~((p)-1)))

	int checkBufYUV(unsigned char *buf, int w, int h, int subsamp,
	tjscalingfactor sf)
	{
	int row, col;
	int hsf=tjMCUWidth[subsamp]/8, vsf=tjMCUHeight[subsamp]/8;
	int pw=PAD(w, hsf), ph=PAD(h, vsf);
	int cw=pw/hsf, ch=ph/vsf;
	int ypitch=PAD(pw, pad), uvpitch=PAD(cw, pad);
	int retval=1;
	int halfway=16*sf.num/sf.denom;
	int blocksize=8*sf.num/sf.denom;

	for(row=0; row<ph; row++)
	{
	for(col=0; col<pw; col++)
	{
	unsigned char y=buf[ypitch*row+col];
	if(((row/blocksize)+(col/blocksize))%2==0)
	{
	if(row<halfway) checkval255(y) else checkval0(y);
	}
	else
	{
	if(row<halfway) checkval(y, 76) else checkval(y, 226);
	}
	}
	}
	if(subsamp!=TJSAMP_GRAY)
	{
	int halfway=16/vsf*sf.num/sf.denom;
	for(row=0; row<ch; row++)
	{
	for(col=0; col<cw; col++)
	{
	unsigned char u=buf[ypitchph + (uvpitchrow+col)],
	v=buf[ypitchph + uvpitchch + (uvpitch*row+col)];
	if(((rowvsf/blocksize)+(colhsf/blocksize))%2==0)
	{
	checkval(u, 128); checkval(v, 128);
	}
	else
	{
	if(row<halfway)
	{
	checkval(u, 85); checkval255(v);
	}
	else
	{
	checkval0(u); checkval(v, 149);
	}
	}
	}
	}
	}

	bailout:
	if(retval==0)
	{
	for(row=0; row<ph; row++)
	{
	for(col=0; col<pw; col++)
	printf("%.3d ", buf[ypitch*row+col]);
	printf("\n");
	}
	printf("\n");
	for(row=0; row<ch; row++)
	{
	for(col=0; col<cw; col++)
	printf("%.3d ", buf[ypitchph + (uvpitchrow+col)]);
	printf("\n");
	}
	printf("\n");
	for(row=0; row<ch; row++)
	{
	for(col=0; col<cw; col++)
	printf("%.3d ", buf[ypitchph + uvpitchch + (uvpitch*row+col)]);
	printf("\n");
	}
	}

	return retval;
	}


	void writeJPEG(unsigned char jpegBuf, unsigned long jpegSize, char filename)
	{
	FILE *file=fopen(filename, "wb");
	if(!file \|\| fwrite(jpegBuf, jpegSize, 1, file)!=1)
	{
	printf("ERROR: Could not write to %s.\n%s\n", filename, strerror(errno));
	bailout();
	}

	bailout:
	if(file) fclose(file);
	}


	void compTest(tjhandle handle, unsigned char **dstBuf,
	unsigned long dstSize, int w, int h, int pf, char basename,
	int subsamp, int jpegQual, int flags)
	{
	char tempStr[1024]; unsigned char srcBuf=NULL, yuvBuf=NULL;
	const char *pfStr=pixFormatStr[pf];
	const char *buStrLong=(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ";
	const char *buStr=(flags&TJFLAG_BOTTOMUP)? "BU":"TD";

	if((srcBuf=(unsigned char )malloc(wh*tjPixelSize[pf]))==NULL)
	_throw("Memory allocation failure");
	initBuf(srcBuf, w, h, pf, flags);

	if(dstBuf && dstSize>0) memset(dstBuf, 0, dstSize);


	if(!alloc) flags\|=TJFLAG_NOREALLOC;
	if(doyuv)
	{
	unsigned long yuvSize=tjBufSizeYUV2(w, pad, h, subsamp);
	tjscalingfactor sf={1, 1};
	tjhandle handle2=tjInitCompress();
	if(!handle2) _throwtj();

	if((yuvBuf=(unsigned char *)malloc(yuvSize))==NULL)
	_throw("Memory allocation failure");
	memset(yuvBuf, 0, yuvSize);

	printf("%s %s -> YUV %s ... ", pfStr, buStrLong, subNameLong[subsamp]);
	_tj(tjEncodeYUV3(handle2, srcBuf, w, 0, h, pf, yuvBuf, pad, subsamp,
	flags));
	tjDestroy(handle2);
	if(checkBufYUV(yuvBuf, w, h, subsamp, sf)) printf("Passed.\n");
	else printf("FAILED!\n");

	printf("YUV %s %s -> JPEG Q%d ... ", subNameLong[subsamp], buStrLong,
	jpegQual);
	_tj(tjCompressFromYUV(handle, yuvBuf, w, pad, h, subsamp, dstBuf,
	dstSize, jpegQual, flags));
	}
	else
	{
	printf("%s %s -> %s Q%d ... ", pfStr, buStrLong, subNameLong[subsamp],
	jpegQual);
	_tj(tjCompress2(handle, srcBuf, w, 0, h, pf, dstBuf, dstSize, subsamp,
	jpegQual, flags));
	}

	snprintf(tempStr, 1024, "%s_enc_%s_%s_%s_Q%d.jpg", basename, pfStr, buStr,
	subName[subsamp], jpegQual);
	writeJPEG(dstBuf, dstSize, tempStr);
	printf("Done.\n Result in %s\n", tempStr);

	bailout:
	if(yuvBuf) free(yuvBuf);
	if(srcBuf) free(srcBuf);
	}


	void _decompTest(tjhandle handle, unsigned char *jpegBuf,
	unsigned long jpegSize, int w, int h, int pf, char *basename, int subsamp,
	int flags, tjscalingfactor sf)
	{
	unsigned char dstBuf=NULL, yuvBuf=NULL;
	int _hdrw=0, _hdrh=0, _hdrsubsamp=-1;
	int scaledWidth=TJSCALED(w, sf);
	int scaledHeight=TJSCALED(h, sf);
	unsigned long dstSize=0;

	_tj(tjDecompressHeader2(handle, jpegBuf, jpegSize, &_hdrw, &_hdrh,
	&_hdrsubsamp));
	if(_hdrw!=w \|\| _hdrh!=h \|\| _hdrsubsamp!=subsamp)
	_throw("Incorrect JPEG header");

	dstSize=scaledWidthscaledHeighttjPixelSize[pf];
	if((dstBuf=(unsigned char *)malloc(dstSize))==NULL)
	_throw("Memory allocation failure");
	memset(dstBuf, 0, dstSize);

	if(doyuv)
	{
	unsigned long yuvSize=tjBufSizeYUV2(scaledWidth, pad, scaledHeight,
	subsamp);
	tjhandle handle2=tjInitDecompress();
	if(!handle2) _throwtj();

	if((yuvBuf=(unsigned char *)malloc(yuvSize))==NULL)
	_throw("Memory allocation failure");
	memset(yuvBuf, 0, yuvSize);

	printf("JPEG -> YUV %s ", subNameLong[subsamp]);
	if(sf.num!=1 \|\| sf.denom!=1)
	printf("%d/%d ... ", sf.num, sf.denom);
	else printf("... ");
	_tj(tjDecompressToYUV2(handle, jpegBuf, jpegSize, yuvBuf, scaledWidth,
	pad, scaledHeight, flags));
	if(checkBufYUV(yuvBuf, scaledWidth, scaledHeight, subsamp, sf))
	printf("Passed.\n");
	else printf("FAILED!\n");

	printf("YUV %s -> %s %s ... ", subNameLong[subsamp], pixFormatStr[pf],
	(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ");
	_tj(tjDecodeYUV(handle2, yuvBuf, pad, subsamp, dstBuf, scaledWidth, 0,
	scaledHeight, pf, flags));
	tjDestroy(handle2);
	}
	else
	{
	printf("JPEG -> %s %s ", pixFormatStr[pf],
	(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ");
	if(sf.num!=1 \|\| sf.denom!=1)
	printf("%d/%d ... ", sf.num, sf.denom);
	else printf("... ");
	_tj(tjDecompress2(handle, jpegBuf, jpegSize, dstBuf, scaledWidth, 0,
	scaledHeight, pf, flags));
	}

	if(checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, flags))
	printf("Passed.");
	else printf("FAILED!");
	printf("\n");

	bailout:
	if(yuvBuf) free(yuvBuf);
	if(dstBuf) free(dstBuf);
	}


	void decompTest(tjhandle handle, unsigned char *jpegBuf,
	unsigned long jpegSize, int w, int h, int pf, char *basename, int subsamp,
	int flags)
	{
	int i, n=0;
	tjscalingfactor *sf=tjGetScalingFactors(&n);
	if(!sf \|\| !n) _throwtj();

	for(i=0; i<n; i++)
	{
	if(subsamp==TJSAMP_444 \|\| subsamp==TJSAMP_GRAY \|\|
	(subsamp==TJSAMP_411 && sf[i].num==1 &&
	(sf[i].denom==2 \|\| sf[i].denom==1)) \|\|
	(subsamp!=TJSAMP_411 && sf[i].num==1 &&
	(sf[i].denom==4 \|\| sf[i].denom==2 \|\| sf[i].denom==1)))
	_decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp,
	flags, sf[i]);
	}

	bailout:
	return;
	}


	void doTest(int w, int h, const int *formats, int nformats, int subsamp,
	char *basename)
	{
	tjhandle chandle=NULL, dhandle=NULL;
	unsigned char *dstBuf=NULL;
	unsigned long size=0; int pfi, pf, i;

	if(!alloc)
	size=tjBufSize(w, h, subsamp);
	if(size!=0)
	if((dstBuf=(unsigned char *)tjAlloc(size))==NULL)
	_throw("Memory allocation failure.");

	if((chandle=tjInitCompress())==NULL \|\| (dhandle=tjInitDecompress())==NULL)
	_throwtj();

	for(pfi=0; pfi<nformats; pfi++)
	{
	for(i=0; i<2; i++)
	{
	int flags=0;
	if(subsamp==TJSAMP_422 \|\| subsamp==TJSAMP_420 \|\| subsamp==TJSAMP_440 \|\|
	subsamp==TJSAMP_411)
	flags\|=TJFLAG_FASTUPSAMPLE;
	if(i==1) flags\|=TJFLAG_BOTTOMUP;
	pf=formats[pfi];
	compTest(chandle, &dstBuf, &size, w, h, pf, basename, subsamp, 100,
	flags);
	decompTest(dhandle, dstBuf, size, w, h, pf, basename, subsamp,
	flags);
	if(pf>=TJPF_RGBX && pf<=TJPF_XRGB)
	{
	printf("\n");
	decompTest(dhandle, dstBuf, size, w, h, pf+(TJPF_RGBA-TJPF_RGBX),
	basename, subsamp, flags);
	}
	printf("\n");
	}
	}
	printf("--------------------\n\n");

	bailout:
	if(chandle) tjDestroy(chandle);
	if(dhandle) tjDestroy(dhandle);

	if(dstBuf) tjFree(dstBuf);
	}


	void bufSizeTest(void)
	{
	int w, h, i, subsamp;
	unsigned char srcBuf=NULL, dstBuf=NULL;
	tjhandle handle=NULL;
	unsigned long dstSize=0;

	if((handle=tjInitCompress())==NULL) _throwtj();

	printf("Buffer size regression test\n");
	for(subsamp=0; subsamp<TJ_NUMSAMP; subsamp++)
	{
	for(w=1; w<48; w++)
	{
	int maxh=(w==1)? 2048:48;
	for(h=1; h<maxh; h++)
	{
	if(h%100==0) printf("%.4d x %.4d\b\b\b\b\b\b\b\b\b\b\b", w, h);
	if((srcBuf=(unsigned char )malloc(wh*4))==NULL)
	_throw("Memory allocation failure");
	if(!alloc \|\| doyuv)
	{
	if(doyuv) dstSize=tjBufSizeYUV2(w, pad, h, subsamp);
	else dstSize=tjBufSize(w, h, subsamp);
	if((dstBuf=(unsigned char *)tjAlloc(dstSize))==NULL)
	_throw("Memory allocation failure");
	}

	for(i=0; i<wh4; i++)
	{
	if(random()<RAND_MAX/2) srcBuf[i]=0;
	else srcBuf[i]=255;
	}

	if(doyuv)
	{
	_tj(tjEncodeYUV3(handle, srcBuf, w, 0, h, TJPF_BGRX, dstBuf, pad,
	subsamp, 0));
	}
	else
	{
	_tj(tjCompress2(handle, srcBuf, w, 0, h, TJPF_BGRX, &dstBuf,
	&dstSize, subsamp, 100, alloc? 0:TJFLAG_NOREALLOC));
	}
	free(srcBuf); srcBuf=NULL;
	tjFree(dstBuf); dstBuf=NULL;

	if((srcBuf=(unsigned char )malloc(hw*4))==NULL)
	_throw("Memory allocation failure");
	if(!alloc \|\| doyuv)
	{
	if(doyuv) dstSize=tjBufSizeYUV2(h, pad, w, subsamp);
	else dstSize=tjBufSize(h, w, subsamp);
	if((dstBuf=(unsigned char *)tjAlloc(dstSize))==NULL)
	_throw("Memory allocation failure");
	}

	for(i=0; i<hw4; i++)
	{
	if(random()<RAND_MAX/2) srcBuf[i]=0;
	else srcBuf[i]=255;
	}

	if(doyuv)
	{
	_tj(tjEncodeYUV3(handle, srcBuf, h, 0, w, TJPF_BGRX, dstBuf, pad,
	subsamp, 0));
	}
	else
	{
	_tj(tjCompress2(handle, srcBuf, h, 0, w, TJPF_BGRX, &dstBuf,
	&dstSize, subsamp, 100, alloc? 0:TJFLAG_NOREALLOC));
	}
	free(srcBuf); srcBuf=NULL;
	tjFree(dstBuf); dstBuf=NULL;
	}
	}
	}
	printf("Done. \n");

	bailout:
	if(srcBuf) free(srcBuf);
	if(dstBuf) free(dstBuf);
	if(handle) tjDestroy(handle);
	}


	int main(int argc, char *argv[])
	{
	int i, num4bf=5;
	#ifdef _WIN32
	srand((unsigned int)time(NULL));
	#endif
	if(argc>1)
	{
	for(i=1; i<argc; i++)
	{
	if(!strcasecmp(argv[i], "-yuv")) doyuv=1;
	if(!strcasecmp(argv[i], "-noyuvpad")) pad=1;
	if(!strcasecmp(argv[i], "-alloc")) alloc=1;
	if(!strncasecmp(argv[i], "-h", 2) \|\| !strcasecmp(argv[i], "-?"))
	usage(argv[0]);
	}
	}
	if(alloc) printf("Testing automatic buffer allocation\n");
	if(doyuv) num4bf=4;
	doTest(35, 39, _3byteFormats, 2, TJSAMP_444, "test");
	doTest(39, 41, _4byteFormats, num4bf, TJSAMP_444, "test");
	doTest(41, 35, _3byteFormats, 2, TJSAMP_422, "test");
	doTest(35, 39, _4byteFormats, num4bf, TJSAMP_422, "test");
	doTest(39, 41, _3byteFormats, 2, TJSAMP_420, "test");
	doTest(41, 35, _4byteFormats, num4bf, TJSAMP_420, "test");
	doTest(35, 39, _3byteFormats, 2, TJSAMP_440, "test");
	doTest(39, 41, _4byteFormats, num4bf, TJSAMP_440, "test");
	doTest(41, 35, _3byteFormats, 2, TJSAMP_411, "test");
	doTest(35, 39, _4byteFormats, num4bf, TJSAMP_411, "test");
	doTest(39, 41, _onlyGray, 1, TJSAMP_GRAY, "test");
	doTest(41, 35, _3byteFormats, 2, TJSAMP_GRAY, "test");
	doTest(35, 39, _4byteFormats, 4, TJSAMP_GRAY, "test");
	bufSizeTest();
	if(doyuv)
	{
	printf("\n--------------------\n\n");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_444, "test_yuv0");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_422, "test_yuv0");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_420, "test_yuv0");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_440, "test_yuv0");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_411, "test_yuv0");
	doTest(48, 48, _onlyRGB, 1, TJSAMP_GRAY, "test_yuv0");
	doTest(48, 48, _onlyGray, 1, TJSAMP_GRAY, "test_yuv0");
	}

	return exitStatus;
	}