/******************************************************************************

 * $Id: tiffset.c,v 1.19 2017-10-01 17:38:12 erouault Exp $

 *

 * Project:  libtiff tools

 * Purpose:  Mainline for setting metadata in existing TIFF files.

 * Author:   Frank Warmerdam, warmerdam@pobox.com

 *

 ******************************************************************************

 * Copyright (c) 2000, Frank Warmerdam

 *

 * 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 <stdio.h>

#include <string.h>

#include <stdlib.h>

#include "tiffio.h"

static char* usageMsg[] = {

"usage: tiffset [options] filename",

"where options are:",

" -s <tagname> [count] <value>...   set the tag value",

" -u <tagname> to unset the tag",

" -d <dirno> set the directory",

" -sd <diroff> set the subdirectory",

" -sf <tagname> <filename>  read the tag value from file (for ASCII tags only)",

NULL

};

static void

usage(void)

{

	int i;

	for (i = 0; usageMsg[i]; i++)

		fprintf(stderr, "%s\n", usageMsg[i]);

	exit(-1);

}

static const TIFFField *

GetField(TIFF *tiff, const char *tagname)

{

    const TIFFField *fip;

    if( atoi(tagname) > 0 )

        fip = TIFFFieldWithTag(tiff, (ttag_t)atoi(tagname));

    else

        fip = TIFFFieldWithName(tiff, tagname);

    if (!fip) {

        fprintf( stderr, "Field name \"%s\" is not recognised.\n", tagname );

        return (TIFFField *)NULL;

    }

    return fip;

}

int

main(int argc, char* argv[])

{

    TIFF *tiff;

    int  arg_index;

    if (argc < 2)

        usage();

    tiff = TIFFOpen(argv[argc-1], "r+");

    if (tiff == NULL)

        return 2;

    for( arg_index = 1; arg_index < argc-1; arg_index++ ) {

	if (strcmp(argv[arg_index],"-d") == 0 && arg_index < argc-2) {

	    arg_index++;

	    if( TIFFSetDirectory(tiff, atoi(argv[arg_index]) ) != 1 )

            {

               fprintf( stderr, "Failed to set directory=%s\n", argv[arg_index] );

               return 6;

            }

	    arg_index++;

	}

	if (strcmp(argv[arg_index],"-sd") == 0 && arg_index < argc-2) {

	    arg_index++;

	    if( TIFFSetSubDirectory(tiff, atoi(argv[arg_index]) ) != 1 )

            {

               fprintf( stderr, "Failed to set sub directory=%s\n", argv[arg_index] );

               return 7;

            }

	    arg_index++;

	}

    /* Add unset option to tiffset -- Zach Baker (niquil@niquil.net) 11/14/2012 */ 

    if (strcmp(argv[arg_index],"-u") == 0 && arg_index < argc-2) {

            const TIFFField *fip;

            const char *tagname;

            arg_index++;

            tagname = argv[arg_index];

            fip = GetField(tiff, tagname);

            if (!fip)

                return 3;

            if (TIFFUnsetField(tiff, TIFFFieldTag(fip)) != 1)

            {

                    fprintf(stderr, "Failed to unset %s\n", TIFFFieldName(fip));

            }

            arg_index++;

    } else if (strcmp(argv[arg_index],"-s") == 0 && arg_index < argc-3) {

            const TIFFField *fip;

            const char *tagname;

            arg_index++;

            tagname = argv[arg_index];

            fip = GetField(tiff, tagname);

            if (!fip)

                return 3;

            arg_index++;

            if (TIFFFieldDataType(fip) == TIFF_ASCII) {

                if (TIFFSetField(tiff, TIFFFieldTag(fip), argv[arg_index]) != 1)

                    fprintf( stderr, "Failed to set %s=%s\n",

                             TIFFFieldName(fip), argv[arg_index] );

            } else if (TIFFFieldWriteCount(fip) > 0

		       || TIFFFieldWriteCount(fip) == TIFF_VARIABLE) {

                int     ret = 1;

                short   wc;

                if (TIFFFieldWriteCount(fip) == TIFF_VARIABLE)

                        wc = atoi(argv[arg_index++]);

                else

                        wc = TIFFFieldWriteCount(fip);

                if (argc - arg_index < wc) {

                    fprintf( stderr,

                             "Number of tag values is not enough. "

                             "Expected %d values for %s tag, got %d\n",

                             wc, TIFFFieldName(fip), argc - arg_index);

                    return 4;

                }

                if (wc > 1 || TIFFFieldWriteCount(fip) == TIFF_VARIABLE) {

                        int     i, size;

                        void    *array;

                        switch (TIFFFieldDataType(fip)) {

                                /*

                                 * XXX: We can't use TIFFDataWidth()

                                 * to determine the space needed to store

                                 * the value. For TIFF_RATIONAL values

                                 * TIFFDataWidth() returns 8, but we use 4-byte

                                 * float to represent rationals.

                                 */

                                case TIFF_BYTE:

                                case TIFF_ASCII:

                                case TIFF_SBYTE:

                                case TIFF_UNDEFINED:

				default:

                                    size = 1;

                                    break;

                                case TIFF_SHORT:

                                case TIFF_SSHORT:

                                    size = 2;

                                    break;

                                case TIFF_LONG:

                                case TIFF_SLONG:

                                case TIFF_FLOAT:

                                case TIFF_IFD:

                                case TIFF_RATIONAL:

                                case TIFF_SRATIONAL:

                                    size = 4;

                                    break;

                                case TIFF_DOUBLE:

                                    size = 8;

                                    break;

                        }

                        array = _TIFFmalloc(wc * size);

                        if (!array) {

                                fprintf(stderr, "No space for %s tag\n",

                                        tagname);

                                return 4;

                        }

                        switch (TIFFFieldDataType(fip)) {

                            case TIFF_BYTE:

                                for (i = 0; i < wc; i++)

                                    ((uint8 *)array)[i] = atoi(argv[arg_index+i]);

                                break;

                            case TIFF_SHORT:

                                for (i = 0; i < wc; i++)

                                    ((uint16 *)array)[i] = atoi(argv[arg_index+i]);

                                break;

                            case TIFF_SBYTE:

                                for (i = 0; i < wc; i++)

                                    ((int8 *)array)[i] = atoi(argv[arg_index+i]);

                                break;

                            case TIFF_SSHORT:

                                for (i = 0; i < wc; i++)

                                    ((int16 *)array)[i] = atoi(argv[arg_index+i]);

                                break;

                            case TIFF_LONG:

                                for (i = 0; i < wc; i++)

                                    ((uint32 *)array)[i] = atol(argv[arg_index+i]);

                                break;

                            case TIFF_SLONG:

                            case TIFF_IFD:

                                for (i = 0; i < wc; i++)

                                    ((uint32 *)array)[i] = atol(argv[arg_index+i]);

                                break;

                            case TIFF_DOUBLE:

                                for (i = 0; i < wc; i++)

                                    ((double *)array)[i] = atof(argv[arg_index+i]);

                                break;

                            case TIFF_RATIONAL:

                            case TIFF_SRATIONAL:

                            case TIFF_FLOAT:

                                for (i = 0; i < wc; i++)

                                    ((float *)array)[i] = (float)atof(argv[arg_index+i]);

                                break;

                            default:

                                break;

                        }

                        if (TIFFFieldPassCount(fip)) {

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   wc, array);

                        } else if (TIFFFieldTag(fip) == TIFFTAG_PAGENUMBER

				   || TIFFFieldTag(fip) == TIFFTAG_HALFTONEHINTS

				   || TIFFFieldTag(fip) == TIFFTAG_YCBCRSUBSAMPLING

				   || TIFFFieldTag(fip) == TIFFTAG_DOTRANGE) {

       				if (TIFFFieldDataType(fip) == TIFF_BYTE) {

					ret = TIFFSetField(tiff, TIFFFieldTag(fip),

						((uint8 *)array)[0], ((uint8 *)array)[1]);

				} else if (TIFFFieldDataType(fip) == TIFF_SHORT) {

					ret = TIFFSetField(tiff, TIFFFieldTag(fip),

						((uint16 *)array)[0], ((uint16 *)array)[1]);

				}

			} else {

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   array);

                        }

                        _TIFFfree(array);

                } else {

                        switch (TIFFFieldDataType(fip)) {

                            case TIFF_BYTE:

                            case TIFF_SHORT:

                            case TIFF_SBYTE:

                            case TIFF_SSHORT:

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   atoi(argv[arg_index++]));

                                break;

                            case TIFF_LONG:

                            case TIFF_SLONG:

                            case TIFF_IFD:

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   atol(argv[arg_index++]));

                                break;

                            case TIFF_DOUBLE:

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   atof(argv[arg_index++]));

                                break;

                            case TIFF_RATIONAL:

                            case TIFF_SRATIONAL:

                            case TIFF_FLOAT:

                                ret = TIFFSetField(tiff, TIFFFieldTag(fip),

                                                   (float)atof(argv[arg_index++]));

                                break;

                            default:

                                break;

                        }

                }

                if (ret != 1)

                    fprintf(stderr, "Failed to set %s\n", TIFFFieldName(fip));

                arg_index += wc;

            }

        } else if (strcmp(argv[arg_index],"-sf") == 0 && arg_index < argc-3) {

            FILE    *fp;

            const TIFFField *fip;

            char    *text;

            size_t  len;

            arg_index++;

            fip = GetField(tiff, argv[arg_index]);

            if (!fip)

                return 3;

            if (TIFFFieldDataType(fip) != TIFF_ASCII) {

                fprintf( stderr,

                         "Only ASCII tags can be set from file. "

                         "%s is not ASCII tag.\n", TIFFFieldName(fip) );

                return 5;

            }

            arg_index++;

            fp = fopen( argv[arg_index], "rt" );

            if(fp == NULL) {

                perror( argv[arg_index] );

                continue;

            }

            text = (char *) malloc(1000000);

            len = fread( text, 1, 999999, fp );

            text[len] = '\0';

            fclose( fp );

            if(TIFFSetField( tiff, TIFFFieldTag(fip), text ) != 1) {

                fprintf(stderr, "Failed to set %s from file %s\n", 

                        TIFFFieldName(fip), argv[arg_index]);

            }

            _TIFFfree( text );

            arg_index++;

        } else {

            fprintf(stderr, "Unrecognised option: %s\n",

                    argv[arg_index]);

            usage();

        }

    }

    TIFFRewriteDirectory(tiff);

    TIFFClose(tiff);

    return 0;

}

/* vim: set ts=8 sts=8 sw=8 noet: */

/*

 * Local Variables:

 * mode: c

 * c-basic-offset: 8

 * fill-column: 78

 * End:

 */