/*
 * 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 background = 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:B: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 'B':
                background = atoi(optarg) & 0xFF;
                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)
        {
            /* do alpha compositing */
            const int src_alpha = src[3];
            const int background_contribution = background * (0xFF - src_alpha);
            *(dst++) = (*(src)*src_alpha + background_contribution) / 0xFF;
            src++;
            *(dst++) = (*(src)*src_alpha + background_contribution) / 0xFF;
            src++;
            *(dst++) = (*(src)*src_alpha + background_contribution) / 0xFF;
            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"
    " -t rows/strip (same as -r)\n"
    " -b (progress by block rather than as a whole image)\n"
    " -n don't emit alpha component.\n"
    " -B use this value as the background when doing alpha compositing\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);
}
