Files
mozjpeg/tjbench.c
DRC fc01f4673b TurboJPEG 3 API overhaul
(ChangeLog update forthcoming)

- Prefix all function names with "tj3" and remove version suffixes from
  function names.  (Future API overhauls will increment the prefix to
  "tj4", etc., thus retaining backward API/ABI compatibility without
  versioning each individual function.)

- Replace stateless boolean flags (including TJ*FLAG_ARITHMETIC and
  TJ*FLAG_LOSSLESS, which were never released) with stateful integer
  parameters, the value of which persists between function calls.
  * Use parameters for the JPEG quality and subsampling as well, in
    order to eliminate the awkwardness of specifying function arguments
    that weren't relevant for lossless compression.
  * tj3DecompressHeader() now stores all relevant information about the
    JPEG image, including the width, height, subsampling type, entropy
    coding type, etc. in parameters rather than returning that
    information in its arguments.
  * TJ*FLAG_LIMITSCANS has been reimplemented as an integer parameter
    (TJ*PARAM_SCANLIMIT) that allows the number of scans to be
    specified.

- Use the const keyword for all pointer arguments to unmodified
  buffers, as well as for both dimensions of 2D pointers.  Addresses
  #395.

- Use size_t rather than unsigned long to represent buffer sizes, since
  unsigned long is a 32-bit type on Windows.  Addresses #24.

- Return 0 from all buffer size functions if an error occurs, rather
  than awkwardly trying to return -1 in an unsigned data type.

- Implement 12-bit and 16-bit data precision using dedicated
  compression, decompression, and image I/O functions/methods.
  * Suffix the names of all data-precision-specific functions with 8,
    12, or 16.
  * Because the YUV functions are intended to be used for video, they
    are currently only implemented with 8-bit data precision, but they
    can be expanded to 12-bit data precision in the future, if
    necessary.
  * Extend TJUnitTest and TJBench to test 12-bit and 16-bit data
    precision, using a new -precision option.
  * Add appropriate regression tests for all of the above to the 'test'
    target.
  * Extend tjbenchtest to test 12-bit and 16-bit data precision, and
    add separate 'tjtest12' and 'tjtest16' targets.
  * BufferedImage I/O in the Java API is currently limited to 8-bit
    data precision, since the BufferedImage class does not
    straightforwardly support higher data precisions.
  * Extend the PPM reader to convert 12-bit and 16-bit PBMPLUS files
    to grayscale or CMYK pixels, as it already does for 8-bit files.

- Properly accommodate lossless JPEG using dedicated parameters
  (TJ*PARAM_LOSSLESS, TJ*PARAM_LOSSLESSPSV, and TJ*PARAM_LOSSLESSPT),
  rather than using a flag and awkwardly repurposing the JPEG quality.
  Update TJBench to properly reflect whether a JPEG image is lossless.

- Re-organize the TJBench usage screen.

- Update the Java docs using Java 11, to improve the formatting and
  eliminate HTML frames.

- Use the accurate integer DCT algorithm by default for both
  compression and decompression, since the "fast" algorithm is a legacy
  feature, it does not pass the ISO compliance tests, and it is not
  actually faster on modern x86 CPUs.
  * Remove the -accuratedct option from TJBench and TJExample.

- Re-implement the 'tjtest' target using a CMake script that enables
  the appropriate tests, depending on the data precision and whether or
  not the Java API is part of the build.

- Consolidate the C and Java versions of tjbenchtest into one script.

- Consolidate the C and Java versions of tjexampletest into one script.

- Combine all initialization functions into a single function
  (tj3Init()) that accepts an integer parameter specifying the
  subsystems to initialize.

- Enable decompression scaling explicitly, using a new function/method
  (tj3SetScalingFactor()/TJDecompressor.setScalingFactor()), rather
  than implicitly using awkward "desired width"/"desired height"
  parameters.

- Introduce a new macro/constant (TJUNSCALED/TJ.UNSCALED) that maps to
  a scaling factor of 1/1.

- Implement partial image decompression, using a new function/method
  (tj3SetCroppingRegion()/TJDecompressor.setCroppingRegion()) and
  TJBench option (-crop).  Extend tjbenchtest to test the new feature.
  Addresses #1.

- Allow the JPEG colorspace to be specified explicitly when
  compressing, using a new parameter (TJ*PARAM_COLORSPACE).  This
  allows JPEG images with the RGB and CMYK colorspaces to be created.

- Remove the error/difference image feature from TJBench.  Identical
  images to the ones that TJBench created can be generated using
  ImageMagick with
  'magick composite <original_image> <output_image> -compose difference <diff_image>'

- Handle JPEG images with unknown subsampling types.  TJ*PARAM_SUBSAMP
  is set to TJ*SAMP_UNKNOWN (== -1) for such images, but they can still
  be decompressed fully into packed-pixel images or losslessly
  transformed (with the exception of lossless cropping.)  They cannot
  be partially decompressed or decompressed into planar YUV images.
  Note also that TJBench, due to its lack of support for imperfect
  transforms, requires that the subsampling type be known when
  rotating, flipping, or transversely transposing an image.  Addresses
  #436

- The Java version of TJBench now has identical functionality to the C
  version.  This was accomplished by (somewhat hackishly) calling the
  TurboJPEG C image I/O functions through JNI and copying the pixels
  between the C heap and the Java heap.

- Add parameters (TJ*PARAM_RESTARTROWS and TJ*PARAM_RESTARTBLOCKS) and
  a TJBench option (-restart) to allow the restart marker interval to
  be specified when compressing.  Eliminate the undocumented TJ_RESTART
  environment variable.

- Add a parameter (TJ*PARAM_OPTIMIZE), a transform option
  (TJ*OPT_OPTIMIZE), and a TJBench option (-optimize) to allow
  optimized baseline Huffman coding to be specified when compressing.
  Eliminate the undocumented TJ_OPTIMIZE environment variable.

- Add parameters (TJ*PARAM_XDENSITY, TJ*PARAM_DENSITY, and
  TJ*DENSITYUNITS) to allow the pixel density to be specified when
  compressing or saving a Windows BMP image and to be queried when
  decompressing or loading a Windows BMP image.  Addresses #77.

- Refactor the fuzz targets to use the new API.
  * Extend decompression coverage to 12-bit and 16-bit data precision.
  * Replace the awkward cjpeg12 and cjpeg16 targets with proper
    TurboJPEG-based compress12, compress12-lossless, and
    compress16-lossless targets

- Fix innocuous UBSan warnings uncovered by the new fuzzers.

- Implement previous versions of the TurboJPEG API by wrapping the new
  functions (tested by running the 2.1.x versions of TJBench, via
  tjbenchtest, and TJUnitTest against the new implementation.)
  * Remove all JNI functions for deprecated Java methods and implement
    the deprecated methods using pure Java wrappers.  It should be
    understood that backward API compatibility in Java applies only to
    the Java classes and that one cannot mix and match a JAR file from
    one version of libjpeg-turbo with a JNI library from another
    version.

- tj3Destroy() now silently accepts a NULL handle.

- tj3Alloc() and tj3Free() now return/accept void pointers, as malloc()
  and free() do.

- The image I/O functions now accept a TurboJPEG instance handle, which
  is used to transmit/receive parameters and to receive error
  information.

Closes #517
2023-01-25 19:09:34 -06:00

1284 lines
49 KiB
C

/*
* Copyright (C)2009-2019, 2021-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.
*/
#ifdef _MSC_VER
#define _CRT_SECURE_NO_DEPRECATE
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <ctype.h>
#include <math.h>
#include <errno.h>
#include <limits.h>
#include <cdjpeg.h>
#include "./tjutil.h"
#include "./turbojpeg.h"
#define THROW(op, err) { \
printf("ERROR in line %d while %s:\n%s\n", __LINE__, op, err); \
retval = -1; goto bailout; \
}
#define THROW_UNIX(m) THROW(m, strerror(errno))
char tjErrorStr[JMSG_LENGTH_MAX] = "\0", tjErrorMsg[JMSG_LENGTH_MAX] = "\0";
int tjErrorLine = -1, tjErrorCode = -1;
#define THROW_TJG(m) { \
printf("ERROR in line %d while %s:\n%s\n", __LINE__, m, \
tj3GetErrorStr(NULL)); \
retval = -1; goto bailout; \
}
#define THROW_TJ(m) { \
int _tjErrorCode = tj3GetErrorCode(handle); \
char *_tjErrorStr = tj3GetErrorStr(handle); \
\
if (!tj3Get(handle, TJPARAM_STOPONWARNING) && \
_tjErrorCode == TJERR_WARNING) { \
if (strncmp(tjErrorStr, _tjErrorStr, JMSG_LENGTH_MAX) || \
strncmp(tjErrorMsg, m, JMSG_LENGTH_MAX) || \
tjErrorCode != _tjErrorCode || tjErrorLine != __LINE__) { \
strncpy(tjErrorStr, _tjErrorStr, JMSG_LENGTH_MAX); \
tjErrorStr[JMSG_LENGTH_MAX - 1] = '\0'; \
strncpy(tjErrorMsg, m, JMSG_LENGTH_MAX); \
tjErrorMsg[JMSG_LENGTH_MAX - 1] = '\0'; \
tjErrorCode = _tjErrorCode; \
tjErrorLine = __LINE__; \
printf("WARNING in line %d while %s:\n%s\n", __LINE__, m, _tjErrorStr); \
} \
} else { \
printf("%s in line %d while %s:\n%s\n", \
_tjErrorCode == TJERR_WARNING ? "WARNING" : "ERROR", __LINE__, m, \
_tjErrorStr); \
retval = -1; goto bailout; \
} \
}
#define IS_CROPPED(cr) (cr.x != 0 || cr.y != 0 || cr.w != 0 || cr.h != 0)
#define CROPPED_WIDTH(width) \
(IS_CROPPED(cr) ? (cr.w != 0 ? cr.w : TJSCALED(width, sf) - cr.x) : \
TJSCALED(width, sf))
#define CROPPED_HEIGHT(height) \
(IS_CROPPED(cr) ? (cr.h != 0 ? cr.h : TJSCALED(height, sf) - cr.y) : \
TJSCALED(height, sf))
int stopOnWarning = 0, bottomUp = 0, noRealloc = 1, fastUpsample = 0,
fastDCT = 0, optimize = 0, progressive = 0, limitScans = 0, arithmetic = 0,
lossless = 0, restartIntervalBlocks = 0, restartIntervalRows = 0;
int precision = 8, sampleSize, compOnly = 0, decompOnly = 0, doYUV = 0,
quiet = 0, doTile = 0, pf = TJPF_BGR, yuvAlign = 1, doWrite = 1;
char *ext = "ppm";
const char *pixFormatStr[TJ_NUMPF] = {
"RGB", "BGR", "RGBX", "BGRX", "XBGR", "XRGB", "GRAY", "", "", "", "", "CMYK"
};
const char *subNameLong[TJ_NUMSAMP] = {
"4:4:4", "4:2:2", "4:2:0", "GRAY", "4:4:0", "4:1:1"
};
const char *csName[TJ_NUMCS] = {
"RGB", "YCbCr", "GRAY", "CMYK", "YCCK"
};
const char *subName[TJ_NUMSAMP] = {
"444", "422", "420", "GRAY", "440", "411"
};
tjscalingfactor *scalingFactors = NULL, sf = { 1, 1 };
tjregion cr = { 0, 0, 0, 0 };
int nsf = 0, xformOp = TJXOP_NONE, xformOpt = 0;
int (*customFilter) (short *, tjregion, tjregion, int, int, tjtransform *);
double benchTime = 5.0, warmup = 1.0;
static char *formatName(int subsamp, int cs, char *buf)
{
if (quiet == 1) {
if (lossless)
SNPRINTF(buf, 80, "%-2d/LOSSLESS ", precision);
else if (subsamp == TJSAMP_UNKNOWN)
SNPRINTF(buf, 80, "%-2d/%-5s ", precision, csName[cs]);
else
SNPRINTF(buf, 80, "%-2d/%-5s/%-5s", precision, csName[cs],
subNameLong[subsamp]);
return buf;
} else {
if (lossless)
return (char *)"Lossless";
else if (subsamp == TJSAMP_UNKNOWN)
return (char *)csName[cs];
else {
SNPRINTF(buf, 80, "%s %s", csName[cs], subNameLong[subsamp]);
return buf;
}
}
}
static char *sigfig(double val, int figs, char *buf, int len)
{
char format[80];
int digitsAfterDecimal = figs - (int)ceil(log10(fabs(val)));
if (digitsAfterDecimal < 1)
SNPRINTF(format, 80, "%%.0f");
else
SNPRINTF(format, 80, "%%.%df", digitsAfterDecimal);
SNPRINTF(buf, len, format, val);
return buf;
}
/* Custom DCT filter which produces a negative of the image */
static int dummyDCTFilter(short *coeffs, tjregion arrayRegion,
tjregion planeRegion, int componentIndex,
int transformIndex, tjtransform *transform)
{
int i;
for (i = 0; i < arrayRegion.w * arrayRegion.h; i++)
coeffs[i] = -coeffs[i];
return 0;
}
/* Decompression test */
static int decomp(unsigned char **jpegBufs, size_t *jpegSizes, void *dstBuf,
int w, int h, int subsamp, int jpegQual, char *fileName,
int tilew, int tileh)
{
char tempStr[1024], sizeStr[24] = "\0", qualStr[16] = "\0";
FILE *file = NULL;
tjhandle handle = NULL;
int i, row, col, iter = 0, dstBufAlloc = 0, retval = 0;
double elapsed, elapsedDecode;
int ps = tjPixelSize[pf];
int scaledw, scaledh, pitch;
int ntilesw = (w + tilew - 1) / tilew, ntilesh = (h + tileh - 1) / tileh;
unsigned char *dstPtr, *dstPtr2, *yuvBuf = NULL;
if (lossless) sf = TJUNSCALED;
scaledw = TJSCALED(w, sf);
scaledh = TJSCALED(h, sf);
if (jpegQual > 0) {
SNPRINTF(qualStr, 16, "_%s%d", lossless ? "PSV" : "Q", jpegQual);
qualStr[15] = 0;
}
if ((handle = tj3Init(TJINIT_DECOMPRESS)) == NULL)
THROW_TJ("executing tj3Init()");
if (tj3Set(handle, TJPARAM_STOPONWARNING, stopOnWarning) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_BOTTOMUP, bottomUp) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_FASTUPSAMPLE, fastUpsample) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_FASTDCT, fastDCT) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_SCANLIMIT, limitScans ? 500 : 0) == -1)
THROW_TJ("executing tj3Set()");
if (IS_CROPPED(cr)) {
if (tj3DecompressHeader(handle, jpegBufs[0], jpegSizes[0]) == -1)
THROW_TJ("executing tj3DecompressHeader()");
}
if (tj3SetScalingFactor(handle, sf) == -1)
THROW_TJ("executing tj3SetScalingFactor()");
if (tj3SetCroppingRegion(handle, cr) == -1)
THROW_TJ("executing tj3SetCroppingRegion()");
if (IS_CROPPED(cr)) {
scaledw = cr.w ? cr.w : scaledw - cr.x;
scaledh = cr.h ? cr.h : scaledh - cr.y;
}
pitch = scaledw * ps;
if (dstBuf == NULL) {
if ((unsigned long long)pitch * (unsigned long long)scaledh *
(unsigned long long)sampleSize > (unsigned long long)((size_t)-1))
THROW("allocating destination buffer", "Image is too large");
if ((dstBuf = malloc((size_t)pitch * scaledh * sampleSize)) == NULL)
THROW_UNIX("allocating destination buffer");
dstBufAlloc = 1;
}
/* Set the destination buffer to gray so we know whether the decompressor
attempted to write to it */
if (precision == 8)
memset((unsigned char *)dstBuf, 127, (size_t)pitch * scaledh);
else if (precision == 12) {
for (i = 0; i < pitch * scaledh; i++)
((short *)dstBuf)[i] = (short)2047;
} else {
for (i = 0; i < pitch * scaledh; i++)
((unsigned short *)dstBuf)[i] = (unsigned short)32767;
}
if (doYUV) {
int width = doTile ? tilew : scaledw;
int height = doTile ? tileh : scaledh;
size_t yuvSize = tj3YUVBufSize(width, yuvAlign, height, subsamp);
if (yuvSize == 0)
THROW_TJ("allocating YUV buffer");
if ((yuvBuf = (unsigned char *)malloc(yuvSize)) == NULL)
THROW_UNIX("allocating YUV buffer");
memset(yuvBuf, 127, yuvSize);
}
/* Benchmark */
iter = -1;
elapsed = elapsedDecode = 0.;
while (1) {
int tile = 0;
double start = getTime();
for (row = 0, dstPtr = dstBuf; row < ntilesh;
row++, dstPtr += (size_t)pitch * tileh * sampleSize) {
for (col = 0, dstPtr2 = dstPtr; col < ntilesw;
col++, tile++, dstPtr2 += ps * tilew * sampleSize) {
int width = doTile ? min(tilew, w - col * tilew) : scaledw;
int height = doTile ? min(tileh, h - row * tileh) : scaledh;
if (doYUV) {
double startDecode;
if (tj3DecompressToYUV8(handle, jpegBufs[tile], jpegSizes[tile],
yuvBuf, yuvAlign) == -1)
THROW_TJ("executing tj3DecompressToYUV8()");
startDecode = getTime();
if (tj3DecodeYUV8(handle, yuvBuf, yuvAlign, dstPtr2, width, pitch,
height, pf) == -1)
THROW_TJ("executing tj3DecodeYUV8()");
if (iter >= 0) elapsedDecode += getTime() - startDecode;
} else {
if (precision == 8) {
if (tj3Decompress8(handle, jpegBufs[tile], jpegSizes[tile],
dstPtr2, pitch, pf) == -1)
THROW_TJ("executing tj3Decompress8()");
} else if (precision == 12) {
if (tj3Decompress12(handle, jpegBufs[tile], jpegSizes[tile],
(short *)dstPtr2, pitch, pf) == -1)
THROW_TJ("executing tj3Decompress12()");
} else {
if (tj3Decompress16(handle, jpegBufs[tile], jpegSizes[tile],
(unsigned short *)dstPtr2, pitch, pf) == -1)
THROW_TJ("executing tj3Decompress16()");
}
}
}
}
elapsed += getTime() - start;
if (iter >= 0) {
iter++;
if (elapsed >= benchTime) break;
} else if (elapsed >= warmup) {
iter = 0;
elapsed = elapsedDecode = 0.;
}
}
if (doYUV) elapsed -= elapsedDecode;
if (quiet) {
printf("%-6s%s",
sigfig((double)(w * h) / 1000000. * (double)iter / elapsed, 4,
tempStr, 1024),
quiet == 2 ? "\n" : " ");
if (doYUV)
printf("%s\n",
sigfig((double)(w * h) / 1000000. * (double)iter / elapsedDecode,
4, tempStr, 1024));
else if (quiet != 2) printf("\n");
} else {
printf("%s --> Frame rate: %f fps\n",
doYUV ? "Decomp to YUV" : "Decompress ", (double)iter / elapsed);
printf(" Throughput: %f Megapixels/sec\n",
(double)(w * h) / 1000000. * (double)iter / elapsed);
if (doYUV) {
printf("YUV Decode --> Frame rate: %f fps\n",
(double)iter / elapsedDecode);
printf(" Throughput: %f Megapixels/sec\n",
(double)(w * h) / 1000000. * (double)iter / elapsedDecode);
}
}
if (!doWrite) goto bailout;
if (sf.num != 1 || sf.denom != 1)
SNPRINTF(sizeStr, 24, "%d_%d", sf.num, sf.denom);
else if (tilew != w || tileh != h)
SNPRINTF(sizeStr, 24, "%dx%d", tilew, tileh);
else SNPRINTF(sizeStr, 24, "full");
if (decompOnly)
SNPRINTF(tempStr, 1024, "%s_%s.%s", fileName, sizeStr, ext);
else
SNPRINTF(tempStr, 1024, "%s_%s%s_%s.%s", fileName,
lossless ? "LOSSLS" : subName[subsamp], qualStr, sizeStr, ext);
if (precision == 8) {
if (tj3SaveImage8(handle, tempStr, (unsigned char *)dstBuf, scaledw, 0,
scaledh, pf) == -1)
THROW_TJG("saving output image");
} else if (precision == 12) {
if (tj3SaveImage12(handle, tempStr, (short *)dstBuf, scaledw, 0, scaledh,
pf) == -1)
THROW_TJG("saving output image");
} else {
if (tj3SaveImage16(handle, tempStr, (unsigned short *)dstBuf, scaledw, 0,
scaledh, pf) == -1)
THROW_TJG("saving output image");
}
bailout:
if (file) fclose(file);
tj3Destroy(handle);
if (dstBufAlloc) free(dstBuf);
free(yuvBuf);
return retval;
}
static int fullTest(tjhandle handle, void *srcBuf, int w, int h, int subsamp,
int jpegQual, char *fileName)
{
char tempStr[1024], tempStr2[80];
FILE *file = NULL;
unsigned char **jpegBufs = NULL, *yuvBuf = NULL, *srcPtr, *srcPtr2;
void *tmpBuf = NULL;
double start, elapsed, elapsedEncode;
int row, col, i, tilew = w, tileh = h, retval = 0;
int iter;
size_t totalJpegSize = 0, *jpegSizes = NULL, yuvSize = 0;
int ps = tjPixelSize[pf];
int ntilesw = 1, ntilesh = 1, pitch = w * ps;
const char *pfStr = pixFormatStr[pf];
if ((unsigned long long)pitch * (unsigned long long)h *
(unsigned long long)sampleSize > (unsigned long long)((size_t)-1))
THROW("allocating temporary image buffer", "Image is too large");
if ((tmpBuf = malloc((size_t)pitch * h * sampleSize)) == NULL)
THROW_UNIX("allocating temporary image buffer");
if (!quiet)
printf(">>>>> %s (%s) <--> %d-bit JPEG (%s %s%d) <<<<<\n", pfStr,
bottomUp ? "Bottom-up" : "Top-down", precision,
lossless ? "Lossless" : subNameLong[subsamp],
lossless ? "PSV" : "Q", jpegQual);
for (tilew = doTile ? 8 : w, tileh = doTile ? 8 : h; ;
tilew *= 2, tileh *= 2) {
if (tilew > w) tilew = w;
if (tileh > h) tileh = h;
ntilesw = (w + tilew - 1) / tilew;
ntilesh = (h + tileh - 1) / tileh;
if ((jpegBufs = (unsigned char **)malloc(sizeof(unsigned char *) *
ntilesw * ntilesh)) == NULL)
THROW_UNIX("allocating JPEG tile array");
memset(jpegBufs, 0, sizeof(unsigned char *) * ntilesw * ntilesh);
if ((jpegSizes = (size_t *)malloc(sizeof(size_t) * ntilesw *
ntilesh)) == NULL)
THROW_UNIX("allocating JPEG size array");
memset(jpegSizes, 0, sizeof(size_t) * ntilesw * ntilesh);
if (noRealloc) {
for (i = 0; i < ntilesw * ntilesh; i++) {
size_t jpegBufSize = tj3JPEGBufSize(tilew, tileh, subsamp);
if (jpegBufSize == 0)
THROW_TJ("getting buffer size");
if (jpegBufSize > (size_t)INT_MAX)
THROW("getting buffer size", "Image is too large");
if ((jpegBufs[i] = tj3Alloc(jpegBufSize)) == NULL)
THROW_UNIX("allocating JPEG tiles");
}
}
/* Compression test */
if (quiet == 1)
printf("%-4s(%s) %-2d/%-6s %-3d ", pfStr, bottomUp ? "BU" : "TD",
precision, lossless ? "LOSSLS" : subNameLong[subsamp], jpegQual);
if (precision == 8) {
for (i = 0; i < h; i++)
memcpy(&((unsigned char *)tmpBuf)[pitch * i],
&((unsigned char *)srcBuf)[w * ps * i], w * ps);
} else {
for (i = 0; i < h; i++)
memcpy(&((unsigned short *)tmpBuf)[pitch * i],
&((unsigned short *)srcBuf)[w * ps * i], w * ps * sampleSize);
}
if (tj3Set(handle, TJPARAM_NOREALLOC, noRealloc) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_SUBSAMP, subsamp) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_FASTDCT, fastDCT) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_OPTIMIZE, optimize) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_PROGRESSIVE, progressive) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_ARITHMETIC, arithmetic) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_LOSSLESS, lossless) == -1)
THROW_TJ("executing tj3Set()");
if (lossless) {
if (tj3Set(handle, TJPARAM_LOSSLESSPSV, jpegQual) == -1)
THROW_TJ("executing tj3Set()");
} else {
if (tj3Set(handle, TJPARAM_QUALITY, jpegQual) == -1)
THROW_TJ("executing tj3Set()");
}
if (tj3Set(handle, TJPARAM_RESTARTBLOCKS, restartIntervalBlocks) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_RESTARTROWS, restartIntervalRows) == -1)
THROW_TJ("executing tj3Set()");
if (doYUV) {
yuvSize = tj3YUVBufSize(tilew, yuvAlign, tileh, subsamp);
if (yuvSize == 0)
THROW_TJ("allocating YUV buffer");
if ((yuvBuf = (unsigned char *)malloc(yuvSize)) == NULL)
THROW_UNIX("allocating YUV buffer");
memset(yuvBuf, 127, yuvSize);
}
/* Benchmark */
iter = -1;
elapsed = elapsedEncode = 0.;
while (1) {
int tile = 0;
totalJpegSize = 0;
start = getTime();
for (row = 0, srcPtr = srcBuf; row < ntilesh;
row++, srcPtr += pitch * tileh * sampleSize) {
for (col = 0, srcPtr2 = srcPtr; col < ntilesw;
col++, tile++, srcPtr2 += ps * tilew * sampleSize) {
int width = min(tilew, w - col * tilew);
int height = min(tileh, h - row * tileh);
if (doYUV) {
double startEncode = getTime();
if (tj3EncodeYUV8(handle, srcPtr2, width, pitch, height, pf,
yuvBuf, yuvAlign) == -1)
THROW_TJ("executing tj3EncodeYUV8()");
if (iter >= 0) elapsedEncode += getTime() - startEncode;
if (tj3CompressFromYUV8(handle, yuvBuf, width, yuvAlign, height,
&jpegBufs[tile], &jpegSizes[tile]) == -1)
THROW_TJ("executing tj3CompressFromYUV8()");
} else {
if (precision == 8) {
if (tj3Compress8(handle, srcPtr2, width, pitch, height, pf,
&jpegBufs[tile], &jpegSizes[tile]) == -1)
THROW_TJ("executing tj3Compress8()");
} else if (precision == 12) {
if (tj3Compress12(handle, (short *)srcPtr2, width, pitch, height,
pf, &jpegBufs[tile], &jpegSizes[tile]) == -1)
THROW_TJ("executing tj3Compress12()");
} else {
if (tj3Compress16(handle, (unsigned short *)srcPtr2, width,
pitch, height, pf, &jpegBufs[tile],
&jpegSizes[tile]) == -1)
THROW_TJ("executing tj3Compress16()");
}
}
totalJpegSize += jpegSizes[tile];
}
}
elapsed += getTime() - start;
if (iter >= 0) {
iter++;
if (elapsed >= benchTime) break;
} else if (elapsed >= warmup) {
iter = 0;
elapsed = elapsedEncode = 0.;
}
}
if (doYUV) elapsed -= elapsedEncode;
if (quiet == 1) printf("%-5d %-5d ", tilew, tileh);
if (quiet) {
if (doYUV)
printf("%-6s%s",
sigfig((double)(w * h) / 1000000. *
(double)iter / elapsedEncode, 4, tempStr, 1024),
quiet == 2 ? "\n" : " ");
printf("%-6s%s",
sigfig((double)(w * h) / 1000000. * (double)iter / elapsed, 4,
tempStr, 1024),
quiet == 2 ? "\n" : " ");
printf("%-6s%s",
sigfig((double)(w * h * ps) / (double)totalJpegSize, 4, tempStr2,
80),
quiet == 2 ? "\n" : " ");
} else {
printf("\n%s size: %d x %d\n", doTile ? "Tile" : "Image", tilew, tileh);
if (doYUV) {
printf("Encode YUV --> Frame rate: %f fps\n",
(double)iter / elapsedEncode);
printf(" Output image size: %lu bytes\n",
(unsigned long)yuvSize);
printf(" Compression ratio: %f:1\n",
(double)(w * h * ps) / (double)yuvSize);
printf(" Throughput: %f Megapixels/sec\n",
(double)(w * h) / 1000000. * (double)iter / elapsedEncode);
printf(" Output bit stream: %f Megabits/sec\n",
(double)yuvSize * 8. / 1000000. * (double)iter / elapsedEncode);
}
printf("%s --> Frame rate: %f fps\n",
doYUV ? "Comp from YUV" : "Compress ",
(double)iter / elapsed);
printf(" Output image size: %lu bytes\n",
(unsigned long)totalJpegSize);
printf(" Compression ratio: %f:1\n",
(double)(w * h * ps) / (double)totalJpegSize);
printf(" Throughput: %f Megapixels/sec\n",
(double)(w * h) / 1000000. * (double)iter / elapsed);
printf(" Output bit stream: %f Megabits/sec\n",
(double)totalJpegSize * 8. / 1000000. * (double)iter / elapsed);
}
if (tilew == w && tileh == h && doWrite) {
SNPRINTF(tempStr, 1024, "%s_%s_%s%d.jpg", fileName,
lossless ? "LOSSLS" : subName[subsamp],
lossless ? "PSV" : "Q", jpegQual);
if ((file = fopen(tempStr, "wb")) == NULL)
THROW_UNIX("opening reference image");
if (fwrite(jpegBufs[0], jpegSizes[0], 1, file) != 1)
THROW_UNIX("writing reference image");
fclose(file); file = NULL;
if (!quiet) printf("Reference image written to %s\n", tempStr);
}
/* Decompression test */
if (!compOnly) {
if (decomp(jpegBufs, jpegSizes, tmpBuf, w, h, subsamp, jpegQual,
fileName, tilew, tileh) == -1)
goto bailout;
} else if (quiet == 1) printf("N/A\n");
for (i = 0; i < ntilesw * ntilesh; i++) {
tj3Free(jpegBufs[i]);
jpegBufs[i] = NULL;
}
free(jpegBufs); jpegBufs = NULL;
free(jpegSizes); jpegSizes = NULL;
if (doYUV) {
free(yuvBuf); yuvBuf = NULL;
}
if (tilew == w && tileh == h) break;
}
bailout:
if (file) fclose(file);
if (jpegBufs) {
for (i = 0; i < ntilesw * ntilesh; i++)
tj3Free(jpegBufs[i]);
}
free(jpegBufs);
free(yuvBuf);
free(jpegSizes);
free(tmpBuf);
return retval;
}
static int decompTest(char *fileName)
{
FILE *file = NULL;
tjhandle handle = NULL;
unsigned char **jpegBufs = NULL, *srcBuf = NULL;
size_t *jpegSizes = NULL, srcSize, totalJpegSize;
tjtransform *t = NULL;
double start, elapsed;
int ps = tjPixelSize[pf], tile, row, col, i, iter, retval = 0, decompsrc = 0;
char *temp = NULL, tempStr[80], tempStr2[80];
/* Original image */
int w = 0, h = 0, tilew, tileh, ntilesw = 1, ntilesh = 1, subsamp = -1,
cs = -1;
/* Transformed image */
int tw, th, ttilew, ttileh, tntilesw, tntilesh, tsubsamp;
if ((file = fopen(fileName, "rb")) == NULL)
THROW_UNIX("opening file");
if (fseek(file, 0, SEEK_END) < 0 ||
(srcSize = ftell(file)) == (size_t)-1)
THROW_UNIX("determining file size");
if ((srcBuf = (unsigned char *)malloc(srcSize)) == NULL)
THROW_UNIX("allocating memory");
if (fseek(file, 0, SEEK_SET) < 0)
THROW_UNIX("setting file position");
if (fread(srcBuf, srcSize, 1, file) < 1)
THROW_UNIX("reading JPEG data");
fclose(file); file = NULL;
temp = strrchr(fileName, '.');
if (temp != NULL) *temp = '\0';
if ((handle = tj3Init(TJINIT_TRANSFORM)) == NULL)
THROW_TJ("executing tj3Init()");
if (tj3Set(handle, TJPARAM_STOPONWARNING, stopOnWarning) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_BOTTOMUP, bottomUp) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_NOREALLOC, noRealloc) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_FASTUPSAMPLE, fastUpsample) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_FASTDCT, fastDCT) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_SCANLIMIT, limitScans ? 500 : 0) == -1)
THROW_TJ("executing tj3Set()");
if (tj3DecompressHeader(handle, srcBuf, srcSize) == -1)
THROW_TJ("executing tj3DecompressHeader()");
w = tj3Get(handle, TJPARAM_JPEGWIDTH);
h = tj3Get(handle, TJPARAM_JPEGHEIGHT);
subsamp = tj3Get(handle, TJPARAM_SUBSAMP);
precision = tj3Get(handle, TJPARAM_PRECISION);
if (tj3Get(handle, TJPARAM_PROGRESSIVE) == 1)
printf("JPEG image uses progressive entropy coding\n\n");
if (tj3Get(handle, TJPARAM_ARITHMETIC) == 1)
printf("JPEG image uses arithmetic entropy coding\n\n");
if (tj3Set(handle, TJPARAM_PROGRESSIVE, progressive) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_ARITHMETIC, arithmetic) == -1)
THROW_TJ("executing tj3Set()");
lossless = tj3Get(handle, TJPARAM_LOSSLESS);
sampleSize = (precision == 8 ? sizeof(unsigned char) : sizeof(short));
cs = tj3Get(handle, TJPARAM_COLORSPACE);
if (w < 1 || h < 1)
THROW("reading JPEG header", "Invalid image dimensions");
if (cs == TJCS_YCCK || cs == TJCS_CMYK) {
pf = TJPF_CMYK; ps = tjPixelSize[pf];
}
if (lossless) sf = TJUNSCALED;
if (tj3SetScalingFactor(handle, sf) == -1)
THROW_TJ("executing tj3SetScalingFactor()");
if (tj3SetCroppingRegion(handle, cr) == -1)
THROW_TJ("executing tj3SetCroppingRegion()");
if (quiet == 1) {
printf("All performance values in Mpixels/sec\n\n");
printf("Pixel JPEG %s %s Xform Comp Decomp ",
doTile ? "Tile " : "Image", doTile ? "Tile " : "Image");
if (doYUV) printf("Decode");
printf("\n");
printf("Format Format Width Height Perf Ratio Perf ");
if (doYUV) printf("Perf");
printf("\n\n");
} else if (!quiet)
printf(">>>>> %d-bit JPEG (%s) --> %s (%s) <<<<<\n", precision,
formatName(subsamp, cs, tempStr), pixFormatStr[pf],
bottomUp ? "Bottom-up" : "Top-down");
for (tilew = doTile ? 16 : w, tileh = doTile ? 16 : h; ;
tilew *= 2, tileh *= 2) {
if (tilew > w) tilew = w;
if (tileh > h) tileh = h;
ntilesw = (w + tilew - 1) / tilew;
ntilesh = (h + tileh - 1) / tileh;
if ((jpegBufs = (unsigned char **)malloc(sizeof(unsigned char *) *
ntilesw * ntilesh)) == NULL)
THROW_UNIX("allocating JPEG tile array");
memset(jpegBufs, 0, sizeof(unsigned char *) * ntilesw * ntilesh);
if ((jpegSizes = (size_t *)malloc(sizeof(size_t) * ntilesw *
ntilesh)) == NULL)
THROW_UNIX("allocating JPEG size array");
memset(jpegSizes, 0, sizeof(size_t) * ntilesw * ntilesh);
if (noRealloc &&
(doTile || xformOp != TJXOP_NONE || xformOpt != 0 || customFilter)) {
for (i = 0; i < ntilesw * ntilesh; i++) {
size_t jpegBufSize = tj3JPEGBufSize(tilew, tileh, subsamp);
if (jpegBufSize == 0)
THROW_TJ("getting buffer size");
if (jpegBufSize > (size_t)INT_MAX)
THROW("getting buffer size", "Image is too large");
if ((jpegBufs[i] = tj3Alloc(jpegBufSize)) == NULL)
THROW_UNIX("allocating JPEG tiles");
}
}
tw = w; th = h; ttilew = tilew; ttileh = tileh;
if (!quiet) {
printf("\n%s size: %d x %d", doTile ? "Tile" : "Image", ttilew, ttileh);
if (sf.num != 1 || sf.denom != 1 || IS_CROPPED(cr))
printf(" --> %d x %d", CROPPED_WIDTH(tw), CROPPED_HEIGHT(th));
printf("\n");
} else if (quiet == 1) {
printf("%-4s(%s) %-14s ", pixFormatStr[pf],
bottomUp ? "BU" : "TD", formatName(subsamp, cs, tempStr));
printf("%-5d %-5d ", CROPPED_WIDTH(tilew), CROPPED_HEIGHT(tileh));
}
tsubsamp = subsamp;
if (doTile || xformOp != TJXOP_NONE || xformOpt != 0 || customFilter) {
if ((t = (tjtransform *)malloc(sizeof(tjtransform) * ntilesw *
ntilesh)) == NULL)
THROW_UNIX("allocating image transform array");
if (xformOp == TJXOP_TRANSPOSE || xformOp == TJXOP_TRANSVERSE ||
xformOp == TJXOP_ROT90 || xformOp == TJXOP_ROT270) {
tw = h; th = w; ttilew = tileh; ttileh = tilew;
}
if (xformOp != TJXOP_NONE && xformOp != TJXOP_TRANSPOSE &&
subsamp == TJSAMP_UNKNOWN)
THROW("transforming",
"Could not determine subsampling level of JPEG image");
if (xformOpt & TJXOPT_GRAY) tsubsamp = TJSAMP_GRAY;
if (xformOp == TJXOP_HFLIP || xformOp == TJXOP_ROT180)
tw = tw - (tw % tjMCUWidth[tsubsamp]);
if (xformOp == TJXOP_VFLIP || xformOp == TJXOP_ROT180)
th = th - (th % tjMCUHeight[tsubsamp]);
if (xformOp == TJXOP_TRANSVERSE || xformOp == TJXOP_ROT90)
tw = tw - (tw % tjMCUHeight[tsubsamp]);
if (xformOp == TJXOP_TRANSVERSE || xformOp == TJXOP_ROT270)
th = th - (th % tjMCUWidth[tsubsamp]);
tntilesw = (tw + ttilew - 1) / ttilew;
tntilesh = (th + ttileh - 1) / ttileh;
if (xformOp == TJXOP_TRANSPOSE || xformOp == TJXOP_TRANSVERSE ||
xformOp == TJXOP_ROT90 || xformOp == TJXOP_ROT270) {
if (tsubsamp == TJSAMP_422) tsubsamp = TJSAMP_440;
else if (tsubsamp == TJSAMP_440) tsubsamp = TJSAMP_422;
}
for (row = 0, tile = 0; row < tntilesh; row++) {
for (col = 0; col < tntilesw; col++, tile++) {
t[tile].r.w = min(ttilew, tw - col * ttilew);
t[tile].r.h = min(ttileh, th - row * ttileh);
t[tile].r.x = col * ttilew;
t[tile].r.y = row * ttileh;
t[tile].op = xformOp;
t[tile].options = xformOpt | TJXOPT_TRIM;
t[tile].customFilter = customFilter;
if (t[tile].options & TJXOPT_NOOUTPUT && jpegBufs[tile]) {
tj3Free(jpegBufs[tile]); jpegBufs[tile] = NULL;
}
}
}
iter = -1;
elapsed = 0.;
while (1) {
start = getTime();
if (tj3Transform(handle, srcBuf, srcSize, tntilesw * tntilesh,
jpegBufs, jpegSizes, t) == -1)
THROW_TJ("executing tj3Transform()");
elapsed += getTime() - start;
if (iter >= 0) {
iter++;
if (elapsed >= benchTime) break;
} else if (elapsed >= warmup) {
iter = 0;
elapsed = 0.;
}
}
free(t); t = NULL;
for (tile = 0, totalJpegSize = 0; tile < tntilesw * tntilesh; tile++)
totalJpegSize += jpegSizes[tile];
if (quiet) {
printf("%-6s%s%-6s%s",
sigfig((double)(w * h) / 1000000. / elapsed, 4, tempStr, 80),
quiet == 2 ? "\n" : " ",
sigfig((double)(w * h * ps) / (double)totalJpegSize, 4,
tempStr2, 80),
quiet == 2 ? "\n" : " ");
} else {
printf("Transform --> Frame rate: %f fps\n",
1.0 / elapsed);
printf(" Output image size: %lu bytes\n",
(unsigned long)totalJpegSize);
printf(" Compression ratio: %f:1\n",
(double)(w * h * ps) / (double)totalJpegSize);
printf(" Throughput: %f Megapixels/sec\n",
(double)(w * h) / 1000000. / elapsed);
printf(" Output bit stream: %f Megabits/sec\n",
(double)totalJpegSize * 8. / 1000000. / elapsed);
}
} else {
if (quiet == 1) printf("N/A N/A ");
tj3Free(jpegBufs[0]);
jpegBufs[0] = NULL;
decompsrc = 1;
}
if (w == tilew) ttilew = tw;
if (h == tileh) ttileh = th;
if (!(xformOpt & TJXOPT_NOOUTPUT)) {
if (decomp(decompsrc ? &srcBuf : jpegBufs,
decompsrc ? &srcSize : jpegSizes, NULL, tw, th, tsubsamp, 0,
fileName, ttilew, ttileh) == -1)
goto bailout;
} else if (quiet == 1) printf("N/A\n");
for (i = 0; i < ntilesw * ntilesh; i++) {
tj3Free(jpegBufs[i]);
jpegBufs[i] = NULL;
}
free(jpegBufs); jpegBufs = NULL;
free(jpegSizes); jpegSizes = NULL;
if (tilew == w && tileh == h) break;
}
bailout:
if (file) fclose(file);
if (jpegBufs) {
for (i = 0; i < ntilesw * ntilesh; i++)
tj3Free(jpegBufs[i]);
}
free(jpegBufs);
free(jpegSizes);
free(srcBuf);
free(t);
tj3Destroy(handle);
return retval;
}
static void usage(char *progName)
{
int i;
printf("USAGE: %s\n", progName);
printf(" <Inputimage (BMP|PPM)> <Quality or PSV> [options]\n\n");
printf(" %s\n", progName);
printf(" <Inputimage (JPG)> [options]\n");
printf("\nGENERAL OPTIONS\n");
printf("---------------\n");
printf("-alloc = Dynamically allocate JPEG buffers\n");
printf("-benchtime T = Run each benchmark for at least T seconds [default = 5.0]\n");
printf("-bmp = Use Windows Bitmap format for output images [default = PPM]\n");
printf(" ** 8-bit data precision only **\n");
printf("-bottomup = Use bottom-up row order for packed-pixel source/destination buffers\n");
printf("-componly = Stop after running compression tests. Do not test decompression.\n");
printf("-lossless = Generate lossless JPEG images when compressing (implies\n");
printf(" -subsamp 444). PSV is the predictor selection value (1-7).\n");
printf("-nowrite = Do not write reference or output images (improves consistency of\n");
printf(" benchmark results)\n");
printf("-rgb, -bgr, -rgbx, -bgrx, -xbgr, -xrgb =\n");
printf(" Use the specified pixel format for packed-pixel source/destination buffers\n");
printf(" [default = BGR]\n");
printf("-cmyk = Indirectly test YCCK JPEG compression/decompression\n");
printf(" (use the CMYK pixel format for packed-pixel source/destination buffers)\n");
printf("-precision N = Use N-bit data precision when compressing [N is 8, 12, or 16;\n");
printf(" default = 8; if N is 16, then -lossless must also be specified]\n");
printf("-quiet = Output results in tabular rather than verbose format\n");
printf("-restart N = When compressing, add a restart marker every N MCU rows (lossy) or\n");
printf(" N sample rows (lossless) [default = 0 (no restart markers)]. Append 'B'\n");
printf(" to specify the restart marker interval in MCU blocks (lossy) or samples\n");
printf(" (lossless).\n");
printf("-stoponwarning = Immediately discontinue the current\n");
printf(" compression/decompression/transform operation if a warning (non-fatal\n");
printf(" error) occurs\n");
printf("-tile = Compress/transform the input image into separate JPEG tiles of varying\n");
printf(" sizes (useful for measuring JPEG overhead)\n");
printf("-warmup T = Run each benchmark for T seconds [default = 1.0] prior to starting\n");
printf(" the timer, in order to prime the caches and thus improve the consistency\n");
printf(" of the benchmark results\n");
printf("\nLOSSY JPEG OPTIONS\n");
printf("------------------\n");
printf("-arithmetic = Use arithmetic entropy coding in JPEG images generated by\n");
printf(" compression and transform operations (can be combined with -progressive)\n");
printf(" ** 8-bit data precision only **\n");
printf("-crop WxH+X+Y = Decompress only the specified region of the JPEG image, where W\n");
printf(" and H are the width and height of the region (0 = maximum possible width\n");
printf(" or height) and X and Y are the left and upper boundary of the region, all\n");
printf(" specified relative to the scaled image dimensions. X must be divible by\n");
printf(" the scaled MCU width.\n");
printf("-fastdct = Use the fastest DCT/IDCT algorithm available\n");
printf("-fastupsample = Use the fastest chrominance upsampling algorithm available\n");
printf("-optimize = Use optimized baseline entropy coding in JPEG images generated by\n");
printf(" compession and transform operations\n");
printf("-progressive = Use progressive entropy coding in JPEG images generated by\n");
printf(" compression and transform operations (implies -optimize; can be combined\n");
printf(" with -arithmetic)\n");
printf("-limitscans = Refuse to decompress or transform progressive JPEG images that\n");
printf(" have an unreasonably large number of scans\n");
printf("-scale M/N = When decompressing, scale the width/height of the JPEG image by a\n");
printf(" factor of M/N (M/N = ");
for (i = 0; i < nsf; i++) {
printf("%d/%d", scalingFactors[i].num, scalingFactors[i].denom);
if (nsf == 2 && i != nsf - 1) printf(" or ");
else if (nsf > 2) {
if (i != nsf - 1) printf(", ");
if (i == nsf - 2) printf("or ");
}
if (i % 8 == 0 && i != 0) printf("\n ");
}
printf(")\n");
printf("-subsamp S = When compressing, use the specified level of chrominance\n");
printf(" subsampling (S = 444, 422, 440, 420, 411, or GRAY) [default = test\n");
printf(" Grayscale, 4:2:0, 4:2:2, and 4:4:4 in sequence]\n");
printf("-hflip, -vflip, -transpose, -transverse, -rot90, -rot180, -rot270 =\n");
printf(" Perform the specified lossless transform operation on the input image\n");
printf(" prior to decompression (these operations are mutually exclusive)\n");
printf("-grayscale = Transform the input image into a grayscale JPEG image prior to\n");
printf(" decompression (can be combined with the other transform operations above)\n");
printf("-copynone = Do not copy any extra markers (including EXIF and ICC profile data)\n");
printf(" when transforming the input image\n");
printf("-yuv = Compress from/decompress to intermediate planar YUV images\n");
printf(" ** 8-bit data precision only **\n");
printf("-yuvpad N = The number of bytes by which each row in each plane of an\n");
printf(" intermediate YUV image is evenly divisible (N must be a power of 2)\n");
printf(" [default = 1]\n");
printf("\nNOTE: If the quality/PSV is specified as a range (e.g. 90-100 or 1-4), a\n");
printf("separate test will be performed for all values in the range.\n\n");
exit(1);
}
int main(int argc, char *argv[])
{
void *srcBuf = NULL;
int w = 0, h = 0, i, j, minQual = -1, maxQual = -1;
char *temp;
int minArg = 2, retval = 0, subsamp = -1;
tjhandle handle = NULL;
if ((scalingFactors = tj3GetScalingFactors(&nsf)) == NULL || nsf == 0)
THROW("executing tj3GetScalingFactors()", tj3GetErrorStr(NULL));
if (argc < minArg) usage(argv[0]);
temp = strrchr(argv[1], '.');
if (temp != NULL) {
if (!strcasecmp(temp, ".bmp")) ext = "bmp";
if (!strcasecmp(temp, ".jpg") || !strcasecmp(temp, ".jpeg"))
decompOnly = 1;
}
printf("\n");
if (!decompOnly) {
minArg = 3;
if (argc < minArg) usage(argv[0]);
minQual = atoi(argv[2]);
if ((temp = strchr(argv[2], '-')) != NULL && strlen(temp) > 1 &&
sscanf(&temp[1], "%d", &maxQual) == 1 && maxQual > minQual) {}
else maxQual = minQual;
}
if (argc > minArg) {
for (i = minArg; i < argc; i++) {
if (!strcasecmp(argv[i], "-tile")) {
doTile = 1; xformOpt |= TJXOPT_CROP;
} else if (!strcasecmp(argv[i], "-precision") && i < argc - 1) {
int tempi = atoi(argv[++i]);
if (tempi != 8 && tempi != 12 && tempi != 16)
usage(argv[0]);
precision = tempi;
} else if (!strcasecmp(argv[i], "-fastupsample")) {
printf("Using fastest upsampling algorithm\n\n");
fastUpsample = 1;
} else if (!strcasecmp(argv[i], "-fastdct")) {
printf("Using fastest DCT/IDCT algorithm\n\n");
fastDCT = 1;
} else if (!strcasecmp(argv[i], "-optimize")) {
printf("Using optimized baseline entropy coding\n\n");
optimize = 1;
xformOpt |= TJXOPT_OPTIMIZE;
} else if (!strcasecmp(argv[i], "-progressive")) {
printf("Using progressive entropy coding\n\n");
progressive = 1;
xformOpt |= TJXOPT_PROGRESSIVE;
} else if (!strcasecmp(argv[i], "-arithmetic")) {
printf("Using arithmetic entropy coding\n\n");
arithmetic = 1;
xformOpt |= TJXOPT_ARITHMETIC;
} else if (!strcasecmp(argv[i], "-lossless")) {
lossless = 1;
subsamp = TJSAMP_444;
} else if (!strcasecmp(argv[i], "-rgb"))
pf = TJPF_RGB;
else if (!strcasecmp(argv[i], "-rgbx"))
pf = TJPF_RGBX;
else if (!strcasecmp(argv[i], "-bgr"))
pf = TJPF_BGR;
else if (!strcasecmp(argv[i], "-bgrx"))
pf = TJPF_BGRX;
else if (!strcasecmp(argv[i], "-xbgr"))
pf = TJPF_XBGR;
else if (!strcasecmp(argv[i], "-xrgb"))
pf = TJPF_XRGB;
else if (!strcasecmp(argv[i], "-cmyk"))
pf = TJPF_CMYK;
else if (!strcasecmp(argv[i], "-bottomup"))
bottomUp = 1;
else if (!strcasecmp(argv[i], "-quiet"))
quiet = 1;
else if (!strcasecmp(argv[i], "-qq"))
quiet = 2;
else if (!strcasecmp(argv[i], "-scale") && i < argc - 1) {
int temp1 = 0, temp2 = 0, match = 0;
if (sscanf(argv[++i], "%d/%d", &temp1, &temp2) == 2) {
for (j = 0; j < nsf; j++) {
if (temp1 == scalingFactors[j].num &&
temp2 == scalingFactors[j].denom) {
sf = scalingFactors[j];
match = 1; break;
}
}
if (!match) usage(argv[0]);
} else usage(argv[0]);
} else if (!strcasecmp(argv[i], "-crop") && i < argc - 1) {
int temp1 = -1, temp2 = -1, temp3 = -1, temp4 = -1;
if (sscanf(argv[++i], "%dx%d+%d+%d", &temp1, &temp2, &temp3,
&temp4) == 4 && temp1 >= 0 && temp2 >= 0 && temp3 >= 0 &&
temp4 >= 0) {
cr.w = temp1; cr.h = temp2; cr.x = temp3; cr.y = temp4;
} else usage(argv[0]);
} else if (!strcasecmp(argv[i], "-hflip"))
xformOp = TJXOP_HFLIP;
else if (!strcasecmp(argv[i], "-vflip"))
xformOp = TJXOP_VFLIP;
else if (!strcasecmp(argv[i], "-transpose"))
xformOp = TJXOP_TRANSPOSE;
else if (!strcasecmp(argv[i], "-transverse"))
xformOp = TJXOP_TRANSVERSE;
else if (!strcasecmp(argv[i], "-rot90"))
xformOp = TJXOP_ROT90;
else if (!strcasecmp(argv[i], "-rot180"))
xformOp = TJXOP_ROT180;
else if (!strcasecmp(argv[i], "-rot270"))
xformOp = TJXOP_ROT270;
else if (!strcasecmp(argv[i], "-grayscale"))
xformOpt |= TJXOPT_GRAY;
else if (!strcasecmp(argv[i], "-custom"))
customFilter = dummyDCTFilter;
else if (!strcasecmp(argv[i], "-nooutput"))
xformOpt |= TJXOPT_NOOUTPUT;
else if (!strcasecmp(argv[i], "-copynone"))
xformOpt |= TJXOPT_COPYNONE;
else if (!strcasecmp(argv[i], "-benchtime") && i < argc - 1) {
double tempd = atof(argv[++i]);
if (tempd > 0.0) benchTime = tempd;
else usage(argv[0]);
} else if (!strcasecmp(argv[i], "-warmup") && i < argc - 1) {
double tempd = atof(argv[++i]);
if (tempd >= 0.0) warmup = tempd;
else usage(argv[0]);
printf("Warmup time = %.1f seconds\n\n", warmup);
} else if (!strcasecmp(argv[i], "-alloc"))
noRealloc = 0;
else if (!strcasecmp(argv[i], "-bmp"))
ext = "bmp";
else if (!strcasecmp(argv[i], "-yuv")) {
printf("Testing planar YUV encoding/decoding\n\n");
doYUV = 1;
} else if (!strcasecmp(argv[i], "-yuvpad") && i < argc - 1) {
int tempi = atoi(argv[++i]);
if (tempi >= 1 && (tempi & (tempi - 1)) == 0) yuvAlign = tempi;
else usage(argv[0]);
} else if (!strcasecmp(argv[i], "-subsamp") && i < argc - 1) {
i++;
if (toupper(argv[i][0]) == 'G') subsamp = TJSAMP_GRAY;
else {
int tempi = atoi(argv[i]);
switch (tempi) {
case 444: subsamp = TJSAMP_444; break;
case 422: subsamp = TJSAMP_422; break;
case 440: subsamp = TJSAMP_440; break;
case 420: subsamp = TJSAMP_420; break;
case 411: subsamp = TJSAMP_411; break;
default: usage(argv[0]);
}
}
} else if (!strcasecmp(argv[i], "-componly"))
compOnly = 1;
else if (!strcasecmp(argv[i], "-nowrite"))
doWrite = 0;
else if (!strcasecmp(argv[i], "-limitscans"))
limitScans = 1;
else if (!strcasecmp(argv[i], "-restart") && i < argc - 1) {
int tempi = -1, nscan; char tempc = 0;
if ((nscan = sscanf(argv[++i], "%d%c", &tempi, &tempc)) < 1 ||
tempi < 0 || tempi > 65535 ||
(nscan == 2 && tempc != 'B' && tempc != 'b'))
usage(argv[0]);
if (tempc == 'B' || tempc == 'b')
restartIntervalBlocks = tempi;
else
restartIntervalRows = tempi;
} else if (!strcasecmp(argv[i], "-stoponwarning"))
stopOnWarning = 1;
else usage(argv[0]);
}
}
if (precision == 16 && !lossless) {
printf("ERROR: -lossless must be specified along with -precision 16\n");
retval = -1; goto bailout;
}
if (precision != 8 && arithmetic) {
printf("ERROR: -arithmetic requires 8-bit data precision\n");
retval = -1; goto bailout;
}
if (precision != 8 && doYUV) {
printf("ERROR: -yuv requires 8-bit data precision\n");
retval = -1; goto bailout;
}
if (lossless && doYUV) {
printf("ERROR: -lossless and -yuv are incompatible\n");
retval = -1; goto bailout;
}
sampleSize = (precision == 8 ? sizeof(unsigned char) : sizeof(short));
if ((sf.num != 1 || sf.denom != 1) && doTile) {
printf("Disabling tiled compression/decompression tests, because those tests do not\n");
printf("work when scaled decompression is enabled.\n\n");
doTile = 0; xformOpt &= (~TJXOPT_CROP);
}
if (IS_CROPPED(cr)) {
if (!decompOnly) {
printf("ERROR: Partial image decompression can only be enabled for JPEG input images\n");
retval = -1; goto bailout;
}
if (doTile) {
printf("Disabling tiled compression/decompression tests, because those tests do not\n");
printf("work when partial image decompression is enabled.\n\n");
doTile = 0; xformOpt &= (~TJXOPT_CROP);
}
if (doYUV) {
printf("ERROR: -crop and -yuv are incompatible\n");
retval = -1; goto bailout;
}
}
if (!noRealloc && doTile) {
printf("Disabling tiled compression/decompression tests, because those tests do not\n");
printf("work when dynamic JPEG buffer allocation is enabled.\n\n");
doTile = 0; xformOpt &= (~TJXOPT_CROP);
}
if (!decompOnly) {
if ((handle = tj3Init(TJINIT_COMPRESS)) == NULL)
THROW_TJ("executing tj3Init()");
if (tj3Set(handle, TJPARAM_STOPONWARNING, stopOnWarning) == -1)
THROW_TJ("executing tj3Set()");
if (tj3Set(handle, TJPARAM_BOTTOMUP, bottomUp) == -1)
THROW_TJ("executing tj3Set()");
if (precision == 8) {
if ((srcBuf = tj3LoadImage8(handle, argv[1], &w, 1, &h, &pf)) == NULL)
THROW_TJG("loading input image");
} else if (precision == 12) {
if ((srcBuf = tj3LoadImage12(handle, argv[1], &w, 1, &h, &pf)) == NULL)
THROW_TJG("loading input image");
} else {
if ((srcBuf = tj3LoadImage16(handle, argv[1], &w, 1, &h, &pf)) == NULL)
THROW_TJG("loading input image");
}
temp = strrchr(argv[1], '.');
if (temp != NULL) *temp = '\0';
}
if (quiet == 1 && !decompOnly) {
printf("All performance values in Mpixels/sec\n\n");
printf("Pixel JPEG JPEG %s %s ",
doTile ? "Tile " : "Image", doTile ? "Tile " : "Image");
if (doYUV) printf("Encode ");
printf("Comp Comp Decomp ");
if (doYUV) printf("Decode");
printf("\n");
printf("Format Format %s Width Height ",
lossless ? "PSV " : "Qual");
if (doYUV) printf("Perf ");
printf("Perf Ratio Perf ");
if (doYUV) printf("Perf");
printf("\n\n");
}
if (decompOnly) {
decompTest(argv[1]);
printf("\n");
goto bailout;
}
if (lossless) {
if (minQual < 1 || minQual > 7 || maxQual < 1 || maxQual > 7) {
puts("ERROR: PSV must be between 1 and 7.");
exit(1);
}
} else {
if (minQual < 1 || minQual > 100 || maxQual < 1 || maxQual > 100) {
puts("ERROR: Quality must be between 1 and 100.");
exit(1);
}
}
if (subsamp >= 0 && subsamp < TJ_NUMSAMP) {
for (i = maxQual; i >= minQual; i--)
fullTest(handle, srcBuf, w, h, subsamp, i, argv[1]);
printf("\n");
} else {
if (pf != TJPF_CMYK) {
for (i = maxQual; i >= minQual; i--)
fullTest(handle, srcBuf, w, h, TJSAMP_GRAY, i, argv[1]);
printf("\n");
}
for (i = maxQual; i >= minQual; i--)
fullTest(handle, srcBuf, w, h, TJSAMP_420, i, argv[1]);
printf("\n");
for (i = maxQual; i >= minQual; i--)
fullTest(handle, srcBuf, w, h, TJSAMP_422, i, argv[1]);
printf("\n");
for (i = maxQual; i >= minQual; i--)
fullTest(handle, srcBuf, w, h, TJSAMP_444, i, argv[1]);
printf("\n");
}
bailout:
tj3Destroy(handle);
tj3Free(srcBuf);
return retval;
}