/* * Copyright (C)2009-2014, 2017-2019, 2022-2023 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 */ #ifdef _MSC_VER #define _CRT_SECURE_NO_DEPRECATE #endif #include #include #include #include #include #include "tjutil.h" #include "turbojpeg.h" #include "md5/md5.h" #include "cmyk.h" #ifdef _WIN32 #include #define random() rand() #else #include #endif static void usage(char *progName) { printf("\nUSAGE: %s [options]\n\n", progName); printf("Options:\n"); printf("-yuv = test YUV encoding/compression/decompression/decoding\n"); printf("-noyuvpad = do not pad each row in each Y, U, and V plane to the nearest\n"); printf(" multiple of 4 bytes\n"); printf("-alloc = test automatic JPEG buffer allocation\n"); printf("-bmp = test packed-pixel image I/O\n"); exit(1); } #define THROW_TJ() { \ printf("TurboJPEG ERROR:\n%s\n", tjGetErrorStr()); \ BAILOUT() \ } #define TRY_TJ(f) { if ((f) == -1) THROW_TJ(); } #define THROW(m) { printf("ERROR: %s\n", m); BAILOUT() } #define THROW_MD5(filename, md5sum, ref) { \ printf("\n%s has an MD5 sum of %s.\n Should be %s.\n", filename, md5sum, \ ref); \ 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 _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, yuvAlign = 4; int exitStatus = 0; #define BAILOUT() { exitStatus = -1; goto bailout; } static 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; \ } \ } static 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 = tjAlphaOffset[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_GRAY) roffset = goffset = boffset = 0; 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))) static 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, yuvAlign), uvpitch = PAD(cw, yuvAlign); 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) { 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; } static 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); } static 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, yuvAlign, h, subsamp); tjscalingfactor sf = { 1, 1 }; tjhandle handle2 = tjInitCompress(); if (!handle2) THROW_TJ(); if ((yuvBuf = (unsigned char *)malloc(yuvSize)) == NULL) THROW("Memory allocation failure"); memset(yuvBuf, 0, yuvSize); printf("%s %s -> YUV %s ... ", pfStr, buStrLong, subNameLong[subsamp]); TRY_TJ(tjEncodeYUV3(handle2, srcBuf, w, 0, h, pf, yuvBuf, yuvAlign, 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); TRY_TJ(tjCompressFromYUV(handle, yuvBuf, w, yuvAlign, h, subsamp, dstBuf, dstSize, jpegQual, flags)); } else { printf("%s %s -> %s Q%d ... ", pfStr, buStrLong, subNameLong[subsamp], jpegQual); TRY_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: free(yuvBuf); free(srcBuf); } static 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; TRY_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, yuvAlign, scaledHeight, subsamp); tjhandle handle2 = tjInitDecompress(); if (!handle2) THROW_TJ(); 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("... "); /* We pass scaledWidth + 1 and scaledHeight + 1 to validate that tjDecompressToYUV2() generates the largest possible scaled image that fits within the desired dimensions, as documented. */ TRY_TJ(tjDecompressToYUV2(handle, jpegBuf, jpegSize, yuvBuf, scaledWidth + 1, yuvAlign, scaledHeight + 1, 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 "); TRY_TJ(tjDecodeYUV(handle2, yuvBuf, yuvAlign, 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("... "); /* We pass scaledWidth + 1 and scaledHeight + 1 to validate that tjDecompress2() generates the largest possible scaled image that fits within the desired dimensions, as documented. */ TRY_TJ(tjDecompress2(handle, jpegBuf, jpegSize, dstBuf, scaledWidth + 1, 0, scaledHeight + 1, pf, flags)); } if (checkBuf(dstBuf, scaledWidth, scaledHeight, pf, subsamp, sf, flags)) printf("Passed."); else printf("FAILED!"); printf("\n"); bailout: free(yuvBuf); free(dstBuf); } static 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) THROW_TJ(); 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; } static 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) THROW_TJ(); 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); tjFree(dstBuf); } #if SIZEOF_SIZE_T == 8 #define CHECKSIZE(function) { \ if ((unsigned long long)size < (unsigned long long)0xFFFFFFFF) \ THROW(#function " overflow"); \ } #else #define CHECKSIZE(function) { \ if (size != (unsigned long)(-1) || \ !strcmp(tjGetErrorStr2(NULL), "No error")) \ THROW(#function " overflow"); \ } #endif #define CHECKSIZEINT(function) { \ if (intsize != -1 || !strcmp(tjGetErrorStr2(NULL), "No error")) \ THROW(#function " overflow"); \ } static void overflowTest(void) { /* Ensure that the various buffer size functions don't overflow */ unsigned long size; int intsize; size = tjBufSize(26755, 26755, TJSAMP_444); CHECKSIZE(tjBufSize()); size = TJBUFSIZE(26755, 26755); CHECKSIZE(TJBUFSIZE()); size = tjBufSizeYUV2(37838, 1, 37838, TJSAMP_444); CHECKSIZE(tjBufSizeYUV2()); size = tjBufSizeYUV2(37837, 3, 37837, TJSAMP_444); CHECKSIZE(tjBufSizeYUV2()); size = tjBufSizeYUV2(37837, -1, 37837, TJSAMP_444); CHECKSIZE(tjBufSizeYUV2()); size = TJBUFSIZEYUV(37838, 37838, TJSAMP_444); CHECKSIZE(TJBUFSIZEYUV()); size = tjBufSizeYUV(37838, 37838, TJSAMP_444); CHECKSIZE(tjBufSizeYUV()); size = tjPlaneSizeYUV(0, 65536, 0, 65536, TJSAMP_444); CHECKSIZE(tjPlaneSizeYUV()); intsize = tjPlaneWidth(0, INT_MAX, TJSAMP_420); CHECKSIZEINT(tjPlaneWidth()); intsize = tjPlaneHeight(0, INT_MAX, TJSAMP_420); CHECKSIZEINT(tjPlaneHeight()); bailout: return; } static 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) THROW_TJ(); 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, yuvAlign, 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) { TRY_TJ(tjEncodeYUV3(handle, srcBuf, w, 0, h, TJPF_BGRX, dstBuf, yuvAlign, subsamp, 0)); } else { TRY_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, yuvAlign, 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) { TRY_TJ(tjEncodeYUV3(handle, srcBuf, h, 0, w, TJPF_BGRX, dstBuf, yuvAlign, subsamp, 0)); } else { TRY_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: free(srcBuf); tjFree(dstBuf); if (handle) tjDestroy(handle); } static void initBitmap(unsigned char *buf, int width, int pitch, int height, int pf, int flags) { int roffset = tjRedOffset[pf]; int goffset = tjGreenOffset[pf]; int boffset = tjBlueOffset[pf]; int ps = tjPixelSize[pf]; int i, j; for (j = 0; j < height; j++) { int row = (flags & TJFLAG_BOTTOMUP) ? height - j - 1 : j; for (i = 0; i < width; i++) { unsigned char r = (i * 256 / width) % 256; unsigned char g = (j * 256 / height) % 256; unsigned char b = (j * 256 / height + i * 256 / width) % 256; memset(&buf[row * pitch + i * ps], 0, ps); if (pf == TJPF_GRAY) buf[row * pitch + i * ps] = b; else if (pf == TJPF_CMYK) rgb_to_cmyk(r, g, b, &buf[row * pitch + i * ps + 0], &buf[row * pitch + i * ps + 1], &buf[row * pitch + i * ps + 2], &buf[row * pitch + i * ps + 3]); else { buf[row * pitch + i * ps + roffset] = r; buf[row * pitch + i * ps + goffset] = g; buf[row * pitch + i * ps + boffset] = b; } } } } static int cmpBitmap(unsigned char *buf, int width, int pitch, int height, int pf, int flags, int gray2rgb) { int roffset = tjRedOffset[pf]; int goffset = tjGreenOffset[pf]; int boffset = tjBlueOffset[pf]; int aoffset = tjAlphaOffset[pf]; int ps = tjPixelSize[pf]; int i, j; for (j = 0; j < height; j++) { int row = (flags & TJFLAG_BOTTOMUP) ? height - j - 1 : j; for (i = 0; i < width; i++) { unsigned char r = (i * 256 / width) % 256; unsigned char g = (j * 256 / height) % 256; unsigned char b = (j * 256 / height + i * 256 / width) % 256; if (pf == TJPF_GRAY) { if (buf[row * pitch + i * ps] != b) return 0; } else if (pf == TJPF_CMYK) { unsigned char rf, gf, bf; cmyk_to_rgb(buf[row * pitch + i * ps + 0], buf[row * pitch + i * ps + 1], buf[row * pitch + i * ps + 2], buf[row * pitch + i * ps + 3], &rf, &gf, &bf); if (gray2rgb) { if (rf != b || gf != b || bf != b) return 0; } else if (rf != r || gf != g || bf != b) return 0; } else { if (gray2rgb) { if (buf[row * pitch + i * ps + roffset] != b || buf[row * pitch + i * ps + goffset] != b || buf[row * pitch + i * ps + boffset] != b) return 0; } else if (buf[row * pitch + i * ps + roffset] != r || buf[row * pitch + i * ps + goffset] != g || buf[row * pitch + i * ps + boffset] != b) return 0; if (aoffset >= 0 && buf[row * pitch + i * ps + aoffset] != 0xFF) return 0; } } } return 1; } static int doBmpTest(const char *ext, int width, int align, int height, int pf, int flags) { char filename[80], *md5sum, md5buf[65]; int ps = tjPixelSize[pf], pitch = PAD(width * ps, align), loadWidth = 0, loadHeight = 0, retval = 0, pixelFormat = pf; unsigned char *buf = NULL; char *md5ref; if (pf == TJPF_GRAY) { md5ref = !strcasecmp(ext, "ppm") ? "112c682e82ce5de1cca089e20d60000b" : "51976530acf75f02beddf5d21149101d"; } else { md5ref = !strcasecmp(ext, "ppm") ? "c0c9f772b464d1896326883a5c79c545" : "6d659071b9bfcdee2def22cb58ddadca"; } if ((buf = (unsigned char *)tjAlloc(pitch * height)) == NULL) THROW("Could not allocate memory"); initBitmap(buf, width, pitch, height, pf, flags); SNPRINTF(filename, 80, "test_bmp_%s_%d_%s.%s", pixFormatStr[pf], align, (flags & TJFLAG_BOTTOMUP) ? "bu" : "td", ext); TRY_TJ(tjSaveImage(filename, buf, width, pitch, height, pf, flags)); md5sum = MD5File(filename, md5buf); if (strcasecmp(md5sum, md5ref)) THROW_MD5(filename, md5sum, md5ref); tjFree(buf); buf = NULL; if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf, flags)) == NULL) THROW_TJ(); if (width != loadWidth || height != loadHeight) { printf("\n Image dimensions of %s are bogus\n", filename); retval = -1; goto bailout; } if (!cmpBitmap(buf, width, pitch, height, pf, flags, 0)) { printf("\n Pixel data in %s is bogus\n", filename); retval = -1; goto bailout; } if (pf == TJPF_GRAY) { tjFree(buf); buf = NULL; pf = TJPF_XBGR; if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf, flags)) == NULL) THROW_TJ(); pitch = PAD(width * tjPixelSize[pf], align); if (!cmpBitmap(buf, width, pitch, height, pf, flags, 1)) { printf("\n Converting %s to RGB failed\n", filename); retval = -1; goto bailout; } tjFree(buf); buf = NULL; pf = TJPF_CMYK; if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pf, flags)) == NULL) THROW_TJ(); pitch = PAD(width * tjPixelSize[pf], align); if (!cmpBitmap(buf, width, pitch, height, pf, flags, 1)) { printf("\n Converting %s to CMYK failed\n", filename); retval = -1; goto bailout; } } /* Verify that tjLoadImage() returns the proper "preferred" pixel format for the file type. */ tjFree(buf); buf = NULL; pf = pixelFormat; pixelFormat = TJPF_UNKNOWN; if ((buf = tjLoadImage(filename, &loadWidth, align, &loadHeight, &pixelFormat, flags)) == NULL) THROW_TJ(); if ((pf == TJPF_GRAY && pixelFormat != TJPF_GRAY) || (pf != TJPF_GRAY && !strcasecmp(ext, "bmp") && pixelFormat != TJPF_BGR) || (pf != TJPF_GRAY && !strcasecmp(ext, "ppm") && pixelFormat != TJPF_RGB)) { printf("\n tjLoadImage() returned unexpected pixel format: %s\n", pixFormatStr[pixelFormat]); retval = -1; } unlink(filename); bailout: tjFree(buf); if (exitStatus < 0) return exitStatus; return retval; } static int bmpTest(void) { int align, width = 35, height = 39, format; for (align = 1; align <= 8; align *= 2) { for (format = 0; format < TJ_NUMPF; format++) { printf("%s Top-Down BMP (row alignment = %d bytes) ... ", pixFormatStr[format], align); if (doBmpTest("bmp", width, align, height, format, 0) == -1) return -1; printf("OK.\n"); printf("%s Top-Down PPM (row alignment = %d bytes) ... ", pixFormatStr[format], align); if (doBmpTest("ppm", width, align, height, format, TJFLAG_BOTTOMUP) == -1) return -1; printf("OK.\n"); printf("%s Bottom-Up BMP (row alignment = %d bytes) ... ", pixFormatStr[format], align); if (doBmpTest("bmp", width, align, height, format, 0) == -1) return -1; printf("OK.\n"); printf("%s Bottom-Up PPM (row alignment = %d bytes) ... ", pixFormatStr[format], align); if (doBmpTest("ppm", width, align, height, format, TJFLAG_BOTTOMUP) == -1) return -1; printf("OK.\n"); } } return 0; } #ifdef _WIN32 #define setenv(envvar, value, dummy) _putenv_s(envvar, value) #endif int main(int argc, char *argv[]) { setenv("TJ_REVERT", "1", 1); 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")) yuvAlign = 1; else if (!strcasecmp(argv[i], "-alloc")) alloc = 1; else if (!strcasecmp(argv[i], "-bmp")) return bmpTest(); else usage(argv[0]); } } if (alloc) printf("Testing automatic buffer allocation\n"); if (doYUV) num4bf = 4; overflowTest(); 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; }