You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
461 lines
12 KiB
461 lines
12 KiB
/* |
|
* Copyright (C)2009-2012 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 |
|
|
|
|
|
#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" |
|
}; |
|
const char *subName[TJ_NUMSAMP]={"444", "422", "420", "GRAY", "440"}; |
|
|
|
const char *pixFormatStr[TJ_NUMPF]= |
|
{ |
|
"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "Grayscale", |
|
"RGBA", "BGRA", "ABGR", "ARGB" |
|
}; |
|
|
|
const int alphaOffset[TJ_NUMPF] = {-1, -1, -1, -1, -1, -1, -1, 3, 3, 0, 0}; |
|
|
|
const int _3byteFormats[]={TJPF_RGB, TJPF_BGR}; |
|
const int _4byteFormats[]={TJPF_RGBX, TJPF_BGRX, TJPF_XBGR, TJPF_XRGB}; |
|
const int _onlyGray[]={TJPF_GRAY}; |
|
const int _onlyRGB[]={TJPF_RGB}; |
|
|
|
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; |
|
|
|
memset(buf, 0, w*h*ps); |
|
if(pf==TJPF_GRAY) |
|
{ |
|
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 |
|
{ |
|
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; |
|
|
|
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) |
|
{ |
|
printf("\n"); |
|
for(row=0; row<h; row++) |
|
{ |
|
for(col=0; col<w; col++) |
|
{ |
|
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; |
|
} |
|
|
|
|
|
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; |
|
double t; |
|
|
|
printf("%s %s -> %s Q%d ... ", pixFormatStr[pf], |
|
(flags&TJFLAG_BOTTOMUP)? "Bottom-Up":"Top-Down ", subNameLong[subsamp], |
|
jpegQual); |
|
|
|
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); |
|
|
|
t=gettime(); |
|
*dstSize=tjBufSize(w, h, subsamp); |
|
_tj(tjCompress2(handle, srcBuf, w, 0, h, pf, dstBuf, dstSize, subsamp, |
|
jpegQual, flags)); |
|
t=gettime()-t; |
|
|
|
snprintf(tempStr, 1024, "%s_enc_%s_%s_%s_Q%d.jpg", basename, |
|
pixFormatStr[pf], (flags&TJFLAG_BOTTOMUP)? "BU":"TD", subName[subsamp], |
|
jpegQual); |
|
writeJPEG(*dstBuf, *dstSize, tempStr); |
|
printf("Done."); |
|
printf(" %f ms\n Result in %s\n", t*1000., tempStr); |
|
|
|
bailout: |
|
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; |
|
int _hdrw=0, _hdrh=0, _hdrsubsamp=-1; double t; |
|
int scaledWidth=TJSCALED(w, sf); |
|
int scaledHeight=TJSCALED(h, sf); |
|
unsigned long dstSize=0; |
|
|
|
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(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); |
|
|
|
t=gettime(); |
|
_tj(tjDecompress2(handle, jpegBuf, jpegSize, dstBuf, scaledWidth, 0, |
|
scaledHeight, pf, flags)); |
|
t=gettime()-t; |
|
|
|
if(checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, flags)) |
|
printf("Passed."); |
|
else printf("FAILED!"); |
|
printf(" %f ms\n", t*1000.); |
|
|
|
bailout: |
|
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), sf1={1, 1}; |
|
if(!sf || !n) _throwtj(); |
|
|
|
if((subsamp==TJSAMP_444 || subsamp==TJSAMP_GRAY)) |
|
{ |
|
for(i=0; i<n; i++) |
|
_decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp, |
|
flags, sf[i]); |
|
} |
|
else |
|
_decompTest(handle, jpegBuf, jpegSize, w, h, pf, basename, subsamp, flags, |
|
sf1); |
|
|
|
bailout: |
|
printf("\n"); |
|
} |
|
|
|
|
|
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; |
|
|
|
size=tjBufSize(w, h, subsamp); |
|
if((dstBuf=(unsigned char *)malloc(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) |
|
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) |
|
decompTest(dhandle, dstBuf, size, w, h, pf+(TJPF_RGBA-TJPF_RGBX), |
|
basename, subsamp, flags); |
|
} |
|
} |
|
|
|
bailout: |
|
if(chandle) tjDestroy(chandle); |
|
if(dhandle) tjDestroy(dhandle); |
|
|
|
if(dstBuf) free(dstBuf); |
|
} |
|
|
|
|
|
void bufSizeTest(void) |
|
{ |
|
int w, h, i, subsamp; |
|
unsigned char *srcBuf=NULL, *jpegBuf=NULL; |
|
tjhandle handle=NULL; |
|
unsigned long jpegSize=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((jpegBuf=(unsigned char *)malloc(tjBufSize(w, h, subsamp))) |
|
==NULL) |
|
_throw("Memory allocation failure"); |
|
jpegSize=tjBufSize(w, h, subsamp); |
|
|
|
for(i=0; i<w*h*4; i++) |
|
{ |
|
if(random()<RAND_MAX/2) srcBuf[i]=0; |
|
else srcBuf[i]=255; |
|
} |
|
|
|
_tj(tjCompress2(handle, srcBuf, w, 0, h, TJPF_BGRX, &jpegBuf, |
|
&jpegSize, subsamp, 100, 0)); |
|
free(srcBuf); srcBuf=NULL; |
|
free(jpegBuf); jpegBuf=NULL; |
|
|
|
if((srcBuf=(unsigned char *)malloc(h*w*4))==NULL) |
|
_throw("Memory allocation failure"); |
|
if((jpegBuf=(unsigned char *)malloc(tjBufSize(h, w, subsamp))) |
|
==NULL) |
|
_throw("Memory allocation failure"); |
|
jpegSize=tjBufSize(h, w, subsamp); |
|
|
|
for(i=0; i<h*w*4; i++) |
|
{ |
|
if(random()<RAND_MAX/2) srcBuf[i]=0; |
|
else srcBuf[i]=255; |
|
} |
|
|
|
_tj(tjCompress2(handle, srcBuf, h, 0, w, TJPF_BGRX, &jpegBuf, |
|
&jpegSize, subsamp, 100, 0)); |
|
free(srcBuf); srcBuf=NULL; |
|
free(jpegBuf); jpegBuf=NULL; |
|
} |
|
} |
|
} |
|
printf("Done. \n"); |
|
|
|
bailout: |
|
if(srcBuf) free(srcBuf); |
|
if(jpegBuf) free(jpegBuf); |
|
if(handle) tjDestroy(handle); |
|
} |
|
|
|
|
|
int main(int argc, char *argv[]) |
|
{ |
|
#ifdef _WIN32 |
|
srand((unsigned int)time(NULL)); |
|
#endif |
|
doTest(35, 39, _3byteFormats, 2, TJSAMP_444, "test"); |
|
doTest(39, 41, _4byteFormats, 4, TJSAMP_444, "test"); |
|
doTest(41, 35, _3byteFormats, 2, TJSAMP_422, "test"); |
|
doTest(35, 39, _4byteFormats, 4, TJSAMP_422, "test"); |
|
doTest(39, 41, _3byteFormats, 2, TJSAMP_420, "test"); |
|
doTest(41, 35, _4byteFormats, 4, TJSAMP_420, "test"); |
|
doTest(35, 39, _3byteFormats, 2, TJSAMP_440, "test"); |
|
doTest(39, 41, _4byteFormats, 4, TJSAMP_440, "test"); |
|
doTest(35, 39, _onlyGray, 1, TJSAMP_GRAY, "test"); |
|
doTest(39, 41, _3byteFormats, 2, TJSAMP_GRAY, "test"); |
|
doTest(41, 35, _4byteFormats, 4, TJSAMP_GRAY, "test"); |
|
bufSizeTest(); |
|
|
|
return exitStatus; |
|
}
|
|
|