123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936 |
- /*
- * Copyright (C)2009-2014, 2017-2019, 2022 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 <stdio.h>
- #include <stdlib.h>
- #include <string.h>
- #include <errno.h>
- #include "tjutil.h"
- #include "turbojpeg.h"
- #include "md5/md5.h"
- #include "cmyk.h"
- #ifdef _WIN32
- #include <time.h>
- #define random() rand()
- #else
- #include <unistd.h>
- #endif
- static 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");
- printf("-bmp = tjLoadImage()/tjSaveImage() unit test\n\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, pad = 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, 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) {
- 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, pad, 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, 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);
- TRY_TJ(tjCompressFromYUV(handle, yuvBuf, w, pad, 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, pad, 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("... ");
- TRY_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 ");
- TRY_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("... ");
- TRY_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:
- 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
- static void overflowTest(void)
- {
- /* Ensure that the various buffer size functions don't overflow */
- unsigned long size;
- size = tjBufSize(26755, 26755, TJSAMP_444);
- CHECKSIZE(tjBufSize());
- size = TJBUFSIZE(26755, 26755);
- CHECKSIZE(TJBUFSIZE());
- size = tjBufSizeYUV2(37838, 1, 37838, 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());
- 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, 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) {
- TRY_TJ(tjEncodeYUV3(handle, srcBuf, w, 0, h, TJPF_BGRX, dstBuf, pad,
- 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, 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) {
- TRY_TJ(tjEncodeYUV3(handle, srcBuf, h, 0, w, TJPF_BGRX, dstBuf, pad,
- 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;
- }
- 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 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;
- }
|