/* * Copyright (C)2009-2014, 2017 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\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\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; if(!alloc || doyuv) { 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; if(!alloc || doyuv) { tjFree(dstBuf); dstBuf=NULL; } } } } printf("Done. \n"); bailout: if(srcBuf) free(srcBuf); if(dstBuf) tjFree(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; else if(!strcasecmp(argv[i], "-noyuvpad")) pad=1; else if(!strcasecmp(argv[i], "-alloc")) alloc=1; else 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; }