singe/thirdparty/SDL2_image/external/libtiff/tools/tiff2rgba.c
2023-10-23 19:38:18 -05:00

632 lines
18 KiB
C

/*
* Copyright (c) 1991-1997 Sam Leffler
* Copyright (c) 1991-1997 Silicon Graphics, Inc.
*
* Permission to use, copy, modify, distribute, and sell this software and
* its documentation for any purpose is hereby granted without fee, provided
* that (i) the above copyright notices and this permission notice appear in
* all copies of the software and related documentation, and (ii) the names of
* Sam Leffler and Silicon Graphics may not be used in any advertising or
* publicity relating to the software without the specific, prior written
* permission of Sam Leffler and Silicon Graphics.
*
* THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
* EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
* WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
*
* IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
* ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
* OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
* WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
* LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
* OF THIS SOFTWARE.
*/
#include "libport.h"
#include "tif_config.h"
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif
#include "tiffio.h"
#include "tiffiop.h"
#ifndef EXIT_SUCCESS
#define EXIT_SUCCESS 0
#endif
#ifndef EXIT_FAILURE
#define EXIT_FAILURE 1
#endif
#define streq(a, b) (strcmp(a, b) == 0)
#define CopyField(tag, v) \
if (TIFFGetField(in, tag, &v)) \
TIFFSetField(out, tag, v)
#ifndef howmany
#define howmany(x, y) (((x) + ((y)-1)) / (y))
#endif
#define roundup(x, y) (howmany(x, y) * ((uint32_t)(y)))
static uint16_t compression = COMPRESSION_PACKBITS;
static uint32_t rowsperstrip = (uint32_t)-1;
static int process_by_block = 0; /* default is whole image at once */
static int no_alpha = 0;
static int bigtiff_output = 0;
#define DEFAULT_MAX_MALLOC (256 * 1024 * 1024)
/* malloc size limit (in bytes)
* disabled when set to 0 */
static tmsize_t maxMalloc = DEFAULT_MAX_MALLOC;
static int tiffcvt(TIFF *in, TIFF *out);
static void usage(int code);
int main(int argc, char *argv[])
{
TIFF *in, *out;
int c;
#if !HAVE_DECL_OPTARG
extern int optind;
extern char *optarg;
#endif
while ((c = getopt(argc, argv, "c:r:t:bn8hM:")) != -1)
switch (c)
{
case 'M':
maxMalloc = (tmsize_t)strtoul(optarg, NULL, 0) << 20;
break;
case 'b':
process_by_block = 1;
break;
case 'c':
if (streq(optarg, "none"))
compression = COMPRESSION_NONE;
else if (streq(optarg, "packbits"))
compression = COMPRESSION_PACKBITS;
else if (streq(optarg, "lzw"))
compression = COMPRESSION_LZW;
else if (streq(optarg, "jpeg"))
compression = COMPRESSION_JPEG;
else if (streq(optarg, "zip"))
compression = COMPRESSION_ADOBE_DEFLATE;
else
usage(EXIT_FAILURE);
break;
case 'r':
rowsperstrip = atoi(optarg);
break;
case 't':
rowsperstrip = atoi(optarg);
break;
case 'n':
no_alpha = 1;
break;
case '8':
bigtiff_output = 1;
break;
case 'h':
usage(EXIT_SUCCESS);
/*NOTREACHED*/
break;
case '?':
usage(EXIT_FAILURE);
/*NOTREACHED*/
break;
}
if (argc - optind < 2)
usage(EXIT_FAILURE);
TIFFOpenOptions *opts = TIFFOpenOptionsAlloc();
if (opts == NULL)
{
return EXIT_FAILURE;
}
TIFFOpenOptionsSetMaxSingleMemAlloc(opts, maxMalloc);
out = TIFFOpenExt(argv[argc - 1], bigtiff_output ? "w8" : "w", opts);
if (out == NULL)
{
TIFFOpenOptionsFree(opts);
return (EXIT_FAILURE);
}
for (; optind < argc - 1; optind++)
{
in = TIFFOpenExt(argv[optind], "r", opts);
if (in != NULL)
{
do
{
if (!tiffcvt(in, out) || !TIFFWriteDirectory(out))
{
(void)TIFFClose(out);
(void)TIFFClose(in);
TIFFOpenOptionsFree(opts);
return (1);
}
} while (TIFFReadDirectory(in));
(void)TIFFClose(in);
}
}
TIFFOpenOptionsFree(opts);
(void)TIFFClose(out);
return (EXIT_SUCCESS);
}
static int cvt_by_tile(TIFF *in, TIFF *out)
{
uint32_t *raster; /* retrieve RGBA image */
uint32_t width, height; /* image width & height */
uint32_t tile_width, tile_height;
uint32_t row, col;
uint32_t *wrk_line;
int ok = 1;
uint32_t rastersize, wrk_linesize;
TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
if (!TIFFGetField(in, TIFFTAG_TILEWIDTH, &tile_width) ||
!TIFFGetField(in, TIFFTAG_TILELENGTH, &tile_height))
{
TIFFError(TIFFFileName(in), "Source image not tiled");
return (0);
}
TIFFSetField(out, TIFFTAG_TILEWIDTH, tile_width);
TIFFSetField(out, TIFFTAG_TILELENGTH, tile_height);
/*
* Allocate tile buffer
*/
rastersize = tile_width * tile_height * sizeof(uint32_t);
if (tile_width != (rastersize / tile_height) / sizeof(uint32_t))
{
TIFFError(TIFFFileName(in),
"Integer overflow when calculating raster buffer");
exit(EXIT_FAILURE);
}
raster = (uint32_t *)_TIFFmalloc(rastersize);
if (raster == 0)
{
TIFFError(TIFFFileName(in), "No space for raster buffer");
return (0);
}
/*
* Allocate a scanline buffer for swapping during the vertical
* mirroring pass.
*/
wrk_linesize = tile_width * sizeof(uint32_t);
if (tile_width != wrk_linesize / sizeof(uint32_t))
{
TIFFError(TIFFFileName(in),
"Integer overflow when calculating wrk_line buffer");
exit(EXIT_FAILURE);
}
wrk_line = (uint32_t *)_TIFFmalloc(wrk_linesize);
if (!wrk_line)
{
TIFFError(TIFFFileName(in), "No space for raster scanline buffer");
ok = 0;
}
/*
* Loop over the tiles.
*/
for (row = 0; ok && row < height; row += tile_height)
{
for (col = 0; ok && col < width; col += tile_width)
{
uint32_t i_row;
/* Read the tile into an RGBA array */
if (!TIFFReadRGBATile(in, col, row, raster))
{
ok = 0;
break;
}
/*
* XXX: raster array has 4-byte unsigned integer type, that is why
* we should rearrange it here.
*/
#if HOST_BIGENDIAN
TIFFSwabArrayOfLong(raster, tile_width * tile_height);
#endif
/*
* For some reason the TIFFReadRGBATile() function chooses the
* lower left corner as the origin. Vertically mirror scanlines.
*/
for (i_row = 0; i_row < tile_height / 2; i_row++)
{
uint32_t *top_line, *bottom_line;
top_line = raster + tile_width * i_row;
bottom_line = raster + tile_width * (tile_height - i_row - 1);
_TIFFmemcpy(wrk_line, top_line, 4 * tile_width);
_TIFFmemcpy(top_line, bottom_line, 4 * tile_width);
_TIFFmemcpy(bottom_line, wrk_line, 4 * tile_width);
}
/*
* Write out the result in a tile.
*/
if (TIFFWriteEncodedTile(out, TIFFComputeTile(out, col, row, 0, 0),
raster,
4 * tile_width * tile_height) == -1)
{
ok = 0;
break;
}
}
}
_TIFFfree(raster);
_TIFFfree(wrk_line);
return ok;
}
static int cvt_by_strip(TIFF *in, TIFF *out)
{
uint32_t *raster; /* retrieve RGBA image */
uint32_t width, height; /* image width & height */
uint32_t row;
uint32_t *wrk_line;
int ok = 1;
uint32_t rastersize, wrk_linesize;
TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
if (!TIFFGetField(in, TIFFTAG_ROWSPERSTRIP, &rowsperstrip))
{
TIFFError(TIFFFileName(in), "Source image not in strips");
return (0);
}
TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip);
/*
* Allocate strip buffer
*/
rastersize = width * rowsperstrip * sizeof(uint32_t);
if (width != (rastersize / rowsperstrip) / sizeof(uint32_t))
{
TIFFError(TIFFFileName(in),
"Integer overflow when calculating raster buffer");
exit(EXIT_FAILURE);
}
raster = (uint32_t *)_TIFFmalloc(rastersize);
if (raster == 0)
{
TIFFError(TIFFFileName(in), "No space for raster buffer");
return (0);
}
/*
* Allocate a scanline buffer for swapping during the vertical
* mirroring pass.
*/
wrk_linesize = width * sizeof(uint32_t);
if (width != wrk_linesize / sizeof(uint32_t))
{
TIFFError(TIFFFileName(in),
"Integer overflow when calculating wrk_line buffer");
exit(EXIT_FAILURE);
}
wrk_line = (uint32_t *)_TIFFmalloc(wrk_linesize);
if (!wrk_line)
{
TIFFError(TIFFFileName(in), "No space for raster scanline buffer");
ok = 0;
}
/*
* Loop over the strips.
*/
for (row = 0; ok && row < height; row += rowsperstrip)
{
int rows_to_write, i_row;
/* Read the strip into an RGBA array */
if (!TIFFReadRGBAStrip(in, row, raster))
{
ok = 0;
break;
}
/*
* XXX: raster array has 4-byte unsigned integer type, that is why
* we should rearrange it here.
*/
#if HOST_BIGENDIAN
TIFFSwabArrayOfLong(raster, width * rowsperstrip);
#endif
/*
* Figure out the number of scanlines actually in this strip.
*/
if (row + rowsperstrip > height)
rows_to_write = height - row;
else
rows_to_write = rowsperstrip;
/*
* For some reason the TIFFReadRGBAStrip() function chooses the
* lower left corner as the origin. Vertically mirror scanlines.
*/
for (i_row = 0; i_row < rows_to_write / 2; i_row++)
{
uint32_t *top_line, *bottom_line;
top_line = raster + width * i_row;
bottom_line = raster + width * (rows_to_write - i_row - 1);
_TIFFmemcpy(wrk_line, top_line, 4 * width);
_TIFFmemcpy(top_line, bottom_line, 4 * width);
_TIFFmemcpy(bottom_line, wrk_line, 4 * width);
}
/*
* Write out the result in a strip
*/
if (TIFFWriteEncodedStrip(out, row / rowsperstrip, raster,
4 * rows_to_write * width) == -1)
{
ok = 0;
break;
}
}
_TIFFfree(raster);
_TIFFfree(wrk_line);
return ok;
}
/*
* cvt_whole_image()
*
* read the whole image into one big RGBA buffer and then write out
* strips from that. This is using the traditional TIFFReadRGBAImage()
* API that we trust.
*/
static int cvt_whole_image(TIFF *in, TIFF *out)
{
uint32_t *raster; /* retrieve RGBA image */
uint32_t width, height; /* image width & height */
uint32_t row;
size_t pixel_count;
TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
pixel_count = width * height;
/* XXX: Check the integer overflow. */
if (!width || !height || pixel_count / width != height)
{
TIFFError(
TIFFFileName(in),
"Malformed input file; can't allocate buffer for raster of %" PRIu32
"x%" PRIu32 " size",
width, height);
return 0;
}
if (maxMalloc != 0 &&
(tmsize_t)pixel_count * (tmsize_t)sizeof(uint32_t) > maxMalloc)
{
TIFFError(TIFFFileName(in),
"Raster size %" TIFF_SIZE_FORMAT
" over memory limit (%" TIFF_SSIZE_FORMAT "), try -b option.",
pixel_count * sizeof(uint32_t), maxMalloc);
return 0;
}
rowsperstrip = TIFFDefaultStripSize(out, rowsperstrip);
TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip);
raster = (uint32_t *)_TIFFCheckMalloc(in, pixel_count, sizeof(uint32_t),
"raster buffer");
if (raster == 0)
{
TIFFError(TIFFFileName(in),
"Failed to allocate buffer (%" TIFF_SIZE_FORMAT
" elements of %" TIFF_SIZE_FORMAT " each)",
pixel_count, sizeof(uint32_t));
return (0);
}
/* Read the image in one chunk into an RGBA array */
if (!TIFFReadRGBAImageOriented(in, width, height, raster,
ORIENTATION_TOPLEFT, 0))
{
_TIFFfree(raster);
return (0);
}
/*
* XXX: raster array has 4-byte unsigned integer type, that is why
* we should rearrange it here.
*/
#if HOST_BIGENDIAN
TIFFSwabArrayOfLong(raster, width * height);
#endif
/*
* Do we want to strip away alpha components?
*/
if (no_alpha)
{
size_t count = pixel_count;
unsigned char *src, *dst;
src = dst = (unsigned char *)raster;
while (count > 0)
{
*(dst++) = *(src++);
*(dst++) = *(src++);
*(dst++) = *(src++);
src++;
count--;
}
}
/*
* Write out the result in strips
*/
for (row = 0; row < height; row += rowsperstrip)
{
unsigned char *raster_strip;
int rows_to_write;
int bytes_per_pixel;
if (no_alpha)
{
raster_strip = ((unsigned char *)raster) + 3 * row * width;
bytes_per_pixel = 3;
}
else
{
raster_strip = (unsigned char *)(raster + row * width);
bytes_per_pixel = 4;
}
if (row + rowsperstrip > height)
rows_to_write = height - row;
else
rows_to_write = rowsperstrip;
if (TIFFWriteEncodedStrip(out, row / rowsperstrip, raster_strip,
bytes_per_pixel * rows_to_write * width) ==
-1)
{
_TIFFfree(raster);
return 0;
}
}
_TIFFfree(raster);
return 1;
}
static int tiffcvt(TIFF *in, TIFF *out)
{
uint32_t width, height; /* image width & height */
uint16_t shortv;
float floatv;
char *stringv;
uint32_t longv;
uint16_t v[1];
TIFFGetField(in, TIFFTAG_IMAGEWIDTH, &width);
TIFFGetField(in, TIFFTAG_IMAGELENGTH, &height);
CopyField(TIFFTAG_SUBFILETYPE, longv);
TIFFSetField(out, TIFFTAG_IMAGEWIDTH, width);
TIFFSetField(out, TIFFTAG_IMAGELENGTH, height);
TIFFSetField(out, TIFFTAG_BITSPERSAMPLE, 8);
TIFFSetField(out, TIFFTAG_COMPRESSION, compression);
TIFFSetField(out, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_RGB);
CopyField(TIFFTAG_FILLORDER, shortv);
TIFFSetField(out, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
if (no_alpha)
TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, 3);
else
TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, 4);
if (!no_alpha)
{
v[0] = EXTRASAMPLE_ASSOCALPHA;
TIFFSetField(out, TIFFTAG_EXTRASAMPLES, 1, v);
}
CopyField(TIFFTAG_XRESOLUTION, floatv);
CopyField(TIFFTAG_YRESOLUTION, floatv);
CopyField(TIFFTAG_RESOLUTIONUNIT, shortv);
TIFFSetField(out, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
TIFFSetField(out, TIFFTAG_SOFTWARE, TIFFGetVersion());
CopyField(TIFFTAG_DOCUMENTNAME, stringv);
if (maxMalloc != 0 && TIFFStripSize(in) > maxMalloc)
{
TIFFError(TIFFFileName(in),
"Strip Size %" TIFF_SSIZE_FORMAT
" over memory limit (%" TIFF_SSIZE_FORMAT ")",
TIFFStripSize(in), maxMalloc);
return 0;
}
if (process_by_block && TIFFIsTiled(in))
return (cvt_by_tile(in, out));
else if (process_by_block)
return (cvt_by_strip(in, out));
else
return (cvt_whole_image(in, out));
}
static const char usage_info[] =
/* Help information format modified for the sake of consistency with the
other tiff tools */
/* "usage: tiff2rgba [-c comp] [-r rows] [-b] [-n] [-8] [-M size]
input... output" */
/* "where comp is one of the following compression algorithms:" */
"Convert a TIFF image to RGBA color space\n\n"
"usage: tiff2rgba [options] input output\n"
"where options are:\n"
#ifdef JPEG_SUPPORT
" -c jpeg JPEG encoding\n"
#endif
#ifdef ZIP_SUPPORT
" -c zip Zip/Deflate encoding\n"
#endif
#ifdef LZW_SUPPORT
" -c lzw Lempel-Ziv & Welch encoding\n"
#endif
#ifdef PACKBITS_SUPPORT
" -c packbits PackBits encoding\n"
#endif
#if defined(JPEG_SUPPORT) || defined(ZIP_SUPPORT) || defined(LZW_SUPPORT) || \
defined(PACKBITS_SUPPORT)
" -c none no compression\n"
#endif
"\n"
/* "and the other options are:\n" */
" -r rows/strip\n"
" -b (progress by block rather than as a whole image)\n"
" -n don't emit alpha component.\n"
" -8 write BigTIFF file instead of ClassicTIFF\n"
" -M set the memory allocation limit in MiB. 0 to disable limit\n";
static void usage(int code)
{
FILE *out = (code == EXIT_SUCCESS) ? stdout : stderr;
fprintf(out, "%s\n\n", TIFFGetVersion());
fprintf(out, "%s", usage_info);
exit(code);
}