c/libjpeg-turbo
1
0
Fork 0
mirror of https://github.com/libjpeg-turbo/libjpeg-turbo synced 2025-03-28 21:13:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
libjpeg-turbo/src/tjtran.c

310 lines
12 KiB
C
Raw Blame History

/*
* Copyright (C)2011-2012, 2014-2015, 2017, 2019, 2021-2024
* 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 demonstrates how to use the TurboJPEG C API to approximate the
* functionality of the IJG's jpegtran program. jpegtran features that are not
* covered:
*
* - Adding restart markers to the output image
* - Scan scripts
* - Expanding the input image when cropping
* - Wiping a region of the input image
* - Dropping another JPEG image into the input image
* - Copying only comment markers or ICC profile markers
* - Embedding an ICC color management profile
* - Progress reporting
* - Debug output
*/
#ifdef _MSC_VER
#define _CRT_SECURE_NO_DEPRECATE
#endif
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <limits.h>
#if !defined(_MSC_VER) || _MSC_VER > 1600
#include <stdint.h>
#endif
#include <turbojpeg.h>
#ifdef _WIN32
#define strncasecmp strnicmp
#endif
#ifndef max
#define max(a, b) ((a) > (b) ? (a) : (b))
#endif
#define MATCH_ARG(arg, string, minChars) \
!strncasecmp(arg, string, max(strlen(arg), minChars))
#define IS_CROPPED(cr) (cr.x != 0 || cr.y != 0 || cr.w != 0 || cr.h != 0)
#define THROW(action, message) { \
printf("ERROR in line %d while %s:\n%s\n", __LINE__, action, message); \
retval = -1; goto bailout; \
}
#define THROW_TJ(action) { \
int errorCode = tj3GetErrorCode(tjInstance); \
printf("%s in line %d while %s:\n%s\n", \
errorCode == TJERR_WARNING ? "WARNING" : "ERROR", __LINE__, action, \
tj3GetErrorStr(tjInstance)); \
if (errorCode == TJERR_FATAL || stopOnWarning == 1) { \
retval = -1; goto bailout; \
} \
}
#define THROW_UNIX(action) THROW(action, strerror(errno))
static void usage(char *programName)
{
printf("\nUSAGE: %s [options] <JPEG input image> <JPEG output image>\n\n",
programName);
printf("This program reads the DCT coefficients from the lossy JPEG input image,\n");
printf("optionally transforms them, and writes them to a lossy JPEG output image.\n\n");
printf("OPTIONS (CAN BE ABBREVBIATED)\n");
printf("-----------------------------\n");
printf("-arithmetic\n");
printf(" Use arithmetic entropy coding in the output image instead of Huffman\n");
printf(" entropy coding (can be combined with -progressive)\n");
printf("-copy all\n");
printf(" Copy all extra markers (including comments, JFIF thumbnails, Exif data, and\n");
printf(" ICC profile data) from the input image to the output image [default]\n");
printf("-copy none\n");
printf(" Copy no extra markers from the input image to the output image\n");
printf("-crop WxH+X+Y\n");
printf(" Include only the specified region of the input image. (W, H, X, and Y are\n");
printf(" the width, height, left boundary, and upper boundary of the region, all\n");
printf(" specified relative to the transformed image dimensions.) If necessary, X\n");
printf(" and Y will be shifted up and left to the nearest iMCU boundary, and W and H\n");
printf(" will be increased accordingly.\n");
printf("-flip {horizontal|vertical}, -rotate {90|180|270}, -transpose, -transverse\n");
printf(" Perform the specified lossless transform operation (these options are\n");
printf(" mutually exclusive)\n");
printf("-grayscale\n");
printf(" Create a grayscale output image from a full-color input image\n");
printf("-maxmemory N\n");
printf(" Memory limit (in megabytes) for intermediate buffers used with progressive\n");
printf(" JPEG compression, Huffman table optimization, and lossless transformation\n");
printf(" [default = no limit]\n");
printf("-maxscans N\n");
printf(" Refuse to transform progressive JPEG images that have more than N scans\n");
printf("-optimize\n");
printf(" Use Huffman table optimization in the output image\n");
printf("-perfect\n");
printf(" Abort if the requested transform operation is imperfect (non-reversible.)\n");
printf(" '-flip horizontal', '-rotate 180', '-rotate 270', and '-transverse' are\n");
printf(" imperfect if the image width is not evenly divisible by the iMCU width.\n");
printf(" '-flip vertical', '-rotate 90', '-rotate 180', and '-transverse' are\n");
printf(" imperfect if the image height is not evenly divisible by the iMCU height.\n");
printf("-progressive\n");
printf(" Create a progressive output image instead of a single-scan output image\n");
printf(" (can be combined with -arithmetic; implies -optimize unless -arithmetic is\n");
printf(" also specified)\n");
printf("-strict\n");
printf(" Treat all warnings as fatal; abort immediately if incomplete or corrupt\n");
printf(" data is encountered in the input image, rather than trying to salvage the\n");
printf(" rest of the image\n");
printf("-trim\n");
printf(" If necessary, trim the partial iMCUs at the right or bottom edge of the\n");
printf(" image to make the requested transform perfect\n\n");
exit(1);
}
int main(int argc, char **argv)
{
int i, retval = 0;
int arithmetic = 0, maxMemory = -1, maxScans = -1, optimize = -1,
progressive = 0, stopOnWarning = -1, subsamp;
tjtransform xform;
tjhandle tjInstance = NULL;
FILE *jpegFile = NULL;
long size = 0;
size_t srcSize, dstSize;
unsigned char *srcBuf = NULL, *dstBuf = NULL;
memset(&xform, 0, sizeof(tjtransform));
for (i = 1; i < argc; i++) {
if (MATCH_ARG(argv[i], "-arithmetic", 2))
arithmetic = 1;
else if (MATCH_ARG(argv[i], "-crop", 3) && i < argc - 1) {
char tempc = -1;
if (sscanf(argv[++i], "%d%c%d+%d+%d", &xform.r.w, &tempc, &xform.r.h,
&xform.r.x, &xform.r.y) != 5 || xform.r.w < 1 ||
(tempc != 'x' && tempc != 'X') || xform.r.h < 1 || xform.r.x < 0 ||
xform.r.y < 0)
usage(argv[0]);
xform.options |= TJXOPT_CROP;
} else if (MATCH_ARG(argv[i], "-copy", 2) && i < argc - 1) {
i++;
if (MATCH_ARG(argv[i], "none", 1))
xform.options |= TJXOPT_COPYNONE;
else if (!MATCH_ARG(argv[i], "all", 1))
usage(argv[0]);
} else if (MATCH_ARG(argv[i], "-flip", 2) && i < argc - 1) {
i++;
if (MATCH_ARG(argv[i], "horizontal", 1))
xform.op = TJXOP_HFLIP;
else if (MATCH_ARG(argv[i], "vertical", 1))
xform.op = TJXOP_VFLIP;
else
usage(argv[0]);
} else if (MATCH_ARG(argv[i], "-grayscale", 2) ||
MATCH_ARG(argv[i], "-greyscale", 2))
xform.options |= TJXOPT_GRAY;
else if (MATCH_ARG(argv[i], "-maxscans", 5) && i < argc - 1) {
int tempi = atoi(argv[++i]);
if (tempi < 0) usage(argv[0]);
maxScans = tempi;
} else if (MATCH_ARG(argv[i], "-maxmemory", 2) && i < argc - 1) {
int tempi = atoi(argv[++i]);
if (tempi < 0) usage(argv[0]);
maxMemory = tempi;
} else if (MATCH_ARG(argv[i], "-optimize", 2) ||
MATCH_ARG(argv[i], "-optimise", 2))
optimize = 1;
else if (MATCH_ARG(argv[i], "-perfect", 3))
xform.options |= TJXOPT_PERFECT;
else if (MATCH_ARG(argv[i], "-progressive", 2))
progressive = 1;
else if (MATCH_ARG(argv[i], "-rotate", 2) && i < argc - 1) {
i++;
if (MATCH_ARG(argv[i], "90", 2))
xform.op = TJXOP_ROT90;
else if (MATCH_ARG(argv[i], "180", 3))
xform.op = TJXOP_ROT180;
else if (MATCH_ARG(argv[i], "270", 3))
xform.op = TJXOP_ROT270;
else
usage(argv[0]);
} else if (MATCH_ARG(argv[i], "-strict", 2))
stopOnWarning = 1;
else if (MATCH_ARG(argv[i], "-transverse", 7))
xform.op = TJXOP_TRANSVERSE;
else if (MATCH_ARG(argv[i], "-trim", 4))
xform.options |= TJXOPT_TRIM;
else if (MATCH_ARG(argv[i], "-transpose", 2))
xform.op = TJXOP_TRANSPOSE;
else break;
}
if (i != argc - 2)
usage(argv[0]);
if ((tjInstance = tj3Init(TJINIT_TRANSFORM)) == NULL)
THROW_TJ("creating TurboJPEG instance");
if (stopOnWarning >= 0 &&
tj3Set(tjInstance, TJPARAM_STOPONWARNING, stopOnWarning) < 0)
THROW_TJ("setting TJPARAM_STOPONWARNING");
if (optimize >= 0 && tj3Set(tjInstance, TJPARAM_OPTIMIZE, optimize) < 0)
THROW_TJ("setting TJPARAM_OPTIMIZE");
if (maxScans >= 0 && tj3Set(tjInstance, TJPARAM_SCANLIMIT, maxScans) < 0)
THROW_TJ("setting TJPARAM_SCANLIMIT");
if (maxMemory >= 0 && tj3Set(tjInstance, TJPARAM_MAXMEMORY, maxMemory) < 0)
THROW_TJ("setting TJPARAM_MAXMEMORY");
if ((jpegFile = fopen(argv[i++], "rb")) == NULL)
THROW_UNIX("opening input file");
if (fseek(jpegFile, 0, SEEK_END) < 0 || ((size = ftell(jpegFile)) < 0) ||
fseek(jpegFile, 0, SEEK_SET) < 0)
THROW_UNIX("determining input file size");
if (size == 0)
THROW("determining input file size", "Input file contains no data");
srcSize = size;
if ((srcBuf = tj3Alloc(srcSize)) == NULL)
THROW_UNIX("allocating JPEG buffer");
if (fread(srcBuf, srcSize, 1, jpegFile) < 1)
THROW_UNIX("reading input file");
fclose(jpegFile); jpegFile = NULL;
if (tj3DecompressHeader(tjInstance, srcBuf, srcSize) < 0)
THROW_TJ("reading JPEG header");
subsamp = tj3Get(tjInstance, TJPARAM_SUBSAMP);
if (xform.options & TJXOPT_GRAY)
subsamp = TJSAMP_GRAY;
if (tj3Set(tjInstance, TJPARAM_PROGRESSIVE, progressive) < 0)
THROW_TJ("setting TJPARAM_PROGRESSIVE");
if (tj3Set(tjInstance, TJPARAM_ARITHMETIC, arithmetic) < 0)
THROW_TJ("setting TJPARAM_ARITHMETIC");
if (IS_CROPPED(xform.r)) {
int xAdjust, yAdjust;
if (subsamp == TJSAMP_UNKNOWN)
THROW("adjusting cropping region",
"Could not determine subsampling level of input image");
if (xform.op == TJXOP_TRANSPOSE || xform.op == TJXOP_TRANSVERSE ||
xform.op == TJXOP_ROT90 || xform.op == TJXOP_ROT270) {
xAdjust = xform.r.x % tjMCUHeight[subsamp];
yAdjust = xform.r.y % tjMCUWidth[subsamp];
} else {
xAdjust = xform.r.x % tjMCUWidth[subsamp];
yAdjust = xform.r.y % tjMCUHeight[subsamp];
}
xform.r.x -= xAdjust;
xform.r.w += xAdjust;
xform.r.y -= yAdjust;
xform.r.h += yAdjust;
}
if (tj3Transform(tjInstance, srcBuf, srcSize, 1, &dstBuf, &dstSize,
&xform) < 0)
THROW_TJ("transforming input image");
tj3Free(srcBuf); srcBuf = NULL;
if ((jpegFile = fopen(argv[i], "wb")) == NULL)
THROW_UNIX("opening output file");
if (fwrite(dstBuf, dstSize, 1, jpegFile) < 1)
THROW_UNIX("writing output file");
bailout:
tj3Destroy(tjInstance);
tj3Free(srcBuf);
if (jpegFile) fclose(jpegFile);
tj3Free(dstBuf);
return retval;
}
Reference in New Issue View Git Blame Copy Permalink