src/tiff/tools/raw2tiff.c

/* $Id$
 *
 * Project:  libtiff tools
 * Purpose:  Convert raw byte sequences in TIFF images
 * Author:   Andrey Kiselev, dron@ak4719.spb.edu
 *
 ******************************************************************************
 * Copyright (c) 2002, Andrey Kiselev <dron@ak4719.spb.edu>
 *
 * 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 "tif_config.h"

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <math.h>
#include <ctype.h>

#ifdef HAVE_UNISTD_H
# include <unistd.h>
#endif

#if HAVE_FCNTL_H
# include <fcntl.h>
#endif

#if HAVE_SYS_TYPES_H
# include <sys/types.h>
#endif

#if HAVE_IO_H
# include <io.h>
#endif

#include "tiffio.h"

#ifndef HAVE_GETOPT
extern int getopt(int, char**, char*);
#endif

#ifndef O_BINARY
# define O_BINARY 0
#endif

typedef enum {
	PIXEL,
	BAND
} InterleavingType;

static	uint16 compression = (uint16) -1;
static	int jpegcolormode = JPEGCOLORMODE_RGB;
static	int quality = 75;		/* JPEG quality */
static	uint16 predictor = 0;

static void swapBytesInScanline(void *, uint32, TIFFDataType);
static int guessSize(int, TIFFDataType, off_t, uint32, int,
		     uint32 *, uint32 *);
static double correlation(void *, void *, uint32, TIFFDataType);
static void usage(void);
static	int processCompressOptions(char*);

int
main(int argc, char* argv[])
{
	uint32	width = 0, length = 0, linebytes, bufsize;
	uint32	nbands = 1;		    /* number of bands in input image*/
	off_t	hdr_size = 0;		    /* size of the header to skip */
	TIFFDataType dtype = TIFF_BYTE;
	int16	depth = 1;		    /* bytes per pixel in input image */
	int	swab = 0;		    /* byte swapping flag */
	InterleavingType interleaving = 0;  /* interleaving type flag */
	uint32  rowsperstrip = (uint32) -1;
	uint16	photometric = PHOTOMETRIC_MINISBLACK;
	uint16	config = PLANARCONFIG_CONTIG;
	uint16	fillorder = FILLORDER_LSB2MSB;
	int	fd;
	char	*outfilename = NULL;
	TIFF	*out;

	uint32 row, col, band;
	int	c;
	unsigned char *buf = NULL, *buf1 = NULL;
	extern int optind;
	extern char* optarg;

	while ((c = getopt(argc, argv, "c:r:H:w:l:b:d:LMp:si:o:h")) != -1) {
		switch (c) {
		case 'c':		/* compression scheme */
			if (!processCompressOptions(optarg))
				usage();
			break;
		case 'r':		/* rows/strip */
			rowsperstrip = atoi(optarg);
			break;
		case 'H':		/* size of input image file header */
			hdr_size = atoi(optarg);
			break;
		case 'w':		/* input image width */
			width = atoi(optarg);
			break;
		case 'l':		/* input image length */
			length = atoi(optarg);
			break;
		case 'b':		/* number of bands in input image */
			nbands = atoi(optarg);
			break;
		case 'd':		/* type of samples in input image */
			if (strncmp(optarg, "byte", 4) == 0)
				dtype = TIFF_BYTE;
			else if (strncmp(optarg, "short", 5) == 0)
				dtype = TIFF_SHORT;
			else if  (strncmp(optarg, "long", 4) == 0)
				dtype = TIFF_LONG;
			else if  (strncmp(optarg, "sbyte", 5) == 0)
				dtype = TIFF_SBYTE;
			else if  (strncmp(optarg, "sshort", 6) == 0)
				dtype = TIFF_SSHORT;
			else if  (strncmp(optarg, "slong", 5) == 0)
				dtype = TIFF_SLONG;
			else if  (strncmp(optarg, "float", 5) == 0)
				dtype = TIFF_FLOAT;
			else if  (strncmp(optarg, "double", 6) == 0)
				dtype = TIFF_DOUBLE;
			else
				dtype = TIFF_BYTE;
			depth = TIFFDataWidth(dtype);
			break;
		case 'L':		/* input has lsb-to-msb fillorder */
			fillorder = FILLORDER_LSB2MSB;
			break;
		case 'M':		/* input has msb-to-lsb fillorder */
			fillorder = FILLORDER_MSB2LSB;
			break;
		case 'p':		/* photometric interpretation */
			if (strncmp(optarg, "miniswhite", 10) == 0)
				photometric = PHOTOMETRIC_MINISWHITE;
			else if (strncmp(optarg, "minisblack", 10) == 0)
				photometric = PHOTOMETRIC_MINISBLACK;
			else if (strncmp(optarg, "rgb", 3) == 0)
				photometric = PHOTOMETRIC_RGB;
			else if (strncmp(optarg, "cmyk", 4) == 0)
				photometric = PHOTOMETRIC_SEPARATED;
			else if (strncmp(optarg, "ycbcr", 5) == 0)
				photometric = PHOTOMETRIC_YCBCR;
			else if (strncmp(optarg, "cielab", 6) == 0)
				photometric = PHOTOMETRIC_CIELAB;
			else if (strncmp(optarg, "icclab", 6) == 0)
				photometric = PHOTOMETRIC_ICCLAB;
			else if (strncmp(optarg, "itulab", 6) == 0)
				photometric = PHOTOMETRIC_ITULAB;
			else
				photometric = PHOTOMETRIC_MINISBLACK;
			break;
		case 's':		/* do we need to swap bytes? */
			swab = 1;
			break;
		case 'i':		/* type of interleaving */
			if (strncmp(optarg, "pixel", 4) == 0)
				interleaving = PIXEL;
			else if  (strncmp(optarg, "band", 6) == 0)
				interleaving = BAND;
			else
				interleaving = 0;
			break;
		case 'o':
			outfilename = optarg;
			break;
		case 'h':
			usage();
		default:
			break;
		}
        }

        if (argc - optind < 2)
		usage();

        fd = open(argv[optind], O_RDONLY|O_BINARY, 0);
	if (fd < 0) {
		fprintf(stderr, "%s: %s: Cannot open input file.\n",
			argv[0], argv[optind]);
		return (-1);
	}

	if (guessSize(fd, dtype, hdr_size, nbands, swab, &width, &length) < 0)
		return 1;

	if (outfilename == NULL)
		outfilename = argv[optind+1];
	out = TIFFOpen(outfilename, "w");
	if (out == NULL) {
		fprintf(stderr, "%s: %s: Cannot open file for output.\n",
			argv[0], outfilename);
		return (-1);
	}
	TIFFSetField(out, TIFFTAG_IMAGEWIDTH, width);
	TIFFSetField(out, TIFFTAG_IMAGELENGTH, length);
	TIFFSetField(out, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
	TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, nbands);
	TIFFSetField(out, TIFFTAG_BITSPERSAMPLE, depth * 8);
	TIFFSetField(out, TIFFTAG_FILLORDER, fillorder);
	TIFFSetField(out, TIFFTAG_PLANARCONFIG, config);
	TIFFSetField(out, TIFFTAG_PHOTOMETRIC, photometric);
	switch (dtype) {
	case TIFF_BYTE:
	case TIFF_SHORT:
	case TIFF_LONG:
		TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_UINT);
		break;
	case TIFF_SBYTE:
	case TIFF_SSHORT:
	case TIFF_SLONG:
		TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_INT);
		break;
	case TIFF_FLOAT:
	case TIFF_DOUBLE:
		TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_IEEEFP);
		break;
	default:
		TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_VOID);
		break;
	}
	if (compression == (uint16) -1)
		compression = COMPRESSION_PACKBITS;
	TIFFSetField(out, TIFFTAG_COMPRESSION, compression);
	switch (compression) {
	case COMPRESSION_JPEG:
		if (photometric == PHOTOMETRIC_RGB
		    && jpegcolormode == JPEGCOLORMODE_RGB)
			photometric = PHOTOMETRIC_YCBCR;
		TIFFSetField(out, TIFFTAG_JPEGQUALITY, quality);
		TIFFSetField(out, TIFFTAG_JPEGCOLORMODE, jpegcolormode);
		break;
	case COMPRESSION_LZW:
	case COMPRESSION_DEFLATE:
		if (predictor != 0)
			TIFFSetField(out, TIFFTAG_PREDICTOR, predictor);
		break;
	}
	switch(interleaving) {
	case BAND:				/* band interleaved data */
		linebytes = width * depth;
		buf = (unsigned char *)_TIFFmalloc(linebytes);
		break;
	case PIXEL:				/* pixel interleaved data */
	default:
		linebytes = width * nbands * depth;
		break;
	}
	bufsize = width * nbands * depth;
	buf1 = (unsigned char *)_TIFFmalloc(bufsize);

	rowsperstrip = TIFFDefaultStripSize(out, rowsperstrip);
	if (rowsperstrip > length) {
		rowsperstrip = length;
	}
	TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip );

	lseek(fd, hdr_size, SEEK_SET);		/* Skip the file header */
	for (row = 0; row < length; row++) {
		switch(interleaving) {
		case BAND:			/* band interleaved data */
			for (band = 0; band < nbands; band++) {
				lseek(fd,
				      hdr_size + (length*band+row)*linebytes,
				      SEEK_SET);
				if (read(fd, buf, linebytes) < 0) {
					fprintf(stderr,
					"%s: %s: scanline %lu: Read error.\n",
					argv[0], argv[optind],
					(unsigned long) row);
				break;
				}
				if (swab)	/* Swap bytes if needed */
					swapBytesInScanline(buf, width, dtype);
				for (col = 0; col < width; col++)
					memcpy(buf1 + (col*nbands+band)*depth,
					       buf + col * depth, depth);
			}
			break;
		case PIXEL:			/* pixel interleaved data */
		default:
			if (read(fd, buf1, bufsize) < 0) {
				fprintf(stderr,
					"%s: %s: scanline %lu: Read error.\n",
					argv[0], argv[optind],
					(unsigned long) row);
				break;
			}
			if (swab)		/* Swap bytes if needed */
				swapBytesInScanline(buf1, width, dtype);
			break;
		}

		if (TIFFWriteScanline(out, buf1, row, 0) < 0) {
			fprintf(stderr,	"%s: %s: scanline %lu: Write error.\n",
				argv[0], outfilename, (unsigned long) row);
			break;
		}
	}
	if (buf)
		_TIFFfree(buf);
	if (buf1)
		_TIFFfree(buf1);
	TIFFClose(out);
	return (0);
}

static void
swapBytesInScanline(void *buf, uint32 width, TIFFDataType dtype)
{
	switch (dtype) {
		case TIFF_SHORT:
		case TIFF_SSHORT:
			TIFFSwabArrayOfShort((uint16*)buf,
                                             (unsigned long)width);
			break;
		case TIFF_LONG:
		case TIFF_SLONG:
			TIFFSwabArrayOfLong((uint32*)buf,
                                            (unsigned long)width);
			break;
		/* case TIFF_FLOAT: */	/* FIXME */
		case TIFF_DOUBLE:
			TIFFSwabArrayOfDouble((double*)buf,
                                              (unsigned long)width);
			break;
		default:
			break;
	}
}

static int
guessSize(int fd, TIFFDataType dtype, off_t hdr_size, uint32 nbands,
	  int swab, uint32 *width, uint32 *length)
{
	const float longt = 40.0;    /* maximum possible height/width ratio */
	char	    *buf1, *buf2;
	struct stat filestat;
	uint32	    w, h, scanlinesize, imagesize;
	uint32	    depth = TIFFDataWidth(dtype);
	float	    cor_coef = 0, tmp;

	fstat(fd, &filestat);

	if (filestat.st_size < hdr_size) {
		fprintf(stderr, "Too large header size specified.\n");
		return -1;
	}

	imagesize = (filestat.st_size - hdr_size) / nbands / depth;

	if (*width != 0 && *length == 0) {
		fprintf(stderr,	"Image height is not specified.\n");

		*length = imagesize / *width;

		fprintf(stderr, "Height is guessed as %lu.\n",
			(unsigned long)*length);

		return 1;
	} else if (*width == 0 && *length != 0) {
		fprintf(stderr, "Image width is not specified.\n");

		*width = imagesize / *length;

		fprintf(stderr,	"Width is guessed as %lu.\n",
			(unsigned long)*width);

		return 1;
	} else if (*width == 0 && *length == 0) {
		fprintf(stderr,	"Image width and height are not specified.\n");

		for (w = (uint32) sqrt(imagesize / longt);
		     w < sqrt(imagesize * longt);
		     w++) {
			if (imagesize % w == 0) {
				scanlinesize = w * depth;
				buf1 = _TIFFmalloc(scanlinesize);
				buf2 = _TIFFmalloc(scanlinesize);
				h = imagesize / w;
				lseek(fd, hdr_size + (int)(h/2)*scanlinesize,
				      SEEK_SET);
				read(fd, buf1, scanlinesize);
				read(fd, buf2, scanlinesize);
				if (swab) {
					swapBytesInScanline(buf1, w, dtype);
					swapBytesInScanline(buf2, w, dtype);
				}
				tmp = (float) fabs(correlation(buf1, buf2,
							       w, dtype));
				if (tmp > cor_coef) {
					cor_coef = tmp;
					*width = w, *length = h;
				}

				_TIFFfree(buf1);
				_TIFFfree(buf2);
			}
		}

		fprintf(stderr,
			"Width is guessed as %lu, height is guessed as %lu.\n",
			(unsigned long)*width, (unsigned long)*length);

		return 1;
	} else {
		if (filestat.st_size<(off_t)(hdr_size+(*width)*(*length)*nbands*depth)) {
			fprintf(stderr, "Input file too small.\n");
		return -1;
		}
	}

	return 1;
}

/* Calculate correlation coefficient between two numeric vectors */
static double
correlation(void *buf1, void *buf2, uint32 n_elem, TIFFDataType dtype)
{
	double	X, Y, M1 = 0.0, M2 = 0.0, D1 = 0.0, D2 = 0.0, K = 0.0;
	uint32	i;

	switch (dtype) {
		case TIFF_BYTE:
		default:
                        for (i = 0; i < n_elem; i++) {
				X = ((unsigned char *)buf1)[i];
				Y = ((unsigned char *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_SBYTE:
                        for (i = 0; i < n_elem; i++) {
				X = ((signed char *)buf1)[i];
				Y = ((signed char *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_SHORT:
                        for (i = 0; i < n_elem; i++) {
				X = ((uint16 *)buf1)[i];
				Y = ((uint16 *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_SSHORT:
                        for (i = 0; i < n_elem; i++) {
				X = ((int16 *)buf1)[i];
				Y = ((int16 *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_LONG:
                        for (i = 0; i < n_elem; i++) {
				X = ((uint32 *)buf1)[i];
				Y = ((uint32 *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_SLONG:
                        for (i = 0; i < n_elem; i++) {
				X = ((int32 *)buf1)[i];
				Y = ((int32 *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_FLOAT:
                        for (i = 0; i < n_elem; i++) {
				X = ((float *)buf1)[i];
				Y = ((float *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
		case TIFF_DOUBLE:
                        for (i = 0; i < n_elem; i++) {
				X = ((double *)buf1)[i];
				Y = ((double *)buf2)[i];
				M1 += X, M2 += Y;
				D1 += X * X, D2 += Y * Y;
				K += X * Y;
                        }
			break;
	}

	M1 /= n_elem;
	M2 /= n_elem;
	D1 -= M1 * M1 * n_elem;
	D2 -= M2 * M2 * n_elem;
	K = (K - M1 * M2 * n_elem) / sqrt(D1 * D2);

	return K;
}

static int
processCompressOptions(char* opt)
{
	if (strcmp(opt, "none") == 0)
		compression = COMPRESSION_NONE;
	else if (strcmp(opt, "packbits") == 0)
		compression = COMPRESSION_PACKBITS;
	else if (strncmp(opt, "jpeg", 4) == 0) {
		char* cp = strchr(opt, ':');

                compression = COMPRESSION_JPEG;
                while( cp )
                {
                    if (isdigit((int)cp[1]))
			quality = atoi(cp+1);
                    else if (cp[1] == 'r' )
			jpegcolormode = JPEGCOLORMODE_RAW;
                    else
                        usage();

                    cp = strchr(cp+1,':');
                }
	} else if (strncmp(opt, "lzw", 3) == 0) {
		char* cp = strchr(opt, ':');
		if (cp)
			predictor = atoi(cp+1);
		compression = COMPRESSION_LZW;
	} else if (strncmp(opt, "zip", 3) == 0) {
		char* cp = strchr(opt, ':');
		if (cp)
			predictor = atoi(cp+1);
		compression = COMPRESSION_DEFLATE;
	} else
		return (0);
	return (1);
}

static char* stuff[] = {
"raw2tiff --- tool for converting raw byte sequences in TIFF images",
"usage: raw2tiff [options] input.raw output.tif",
"where options are:",
" -L		input data has LSB2MSB bit order (default)",
" -M		input data has MSB2LSB bit order",
" -r #		make each strip have no more than # rows",
" -H #		size of input image file header in bytes (0 by default)",
" -w #		width of input image in pixels",
" -l #		length of input image in lines",
" -b #		number of bands in input image (1 by default)",
"",
" -d data_type	type of samples in input image",
"where data_type may be:",
" byte		8-bit unsigned integer (default)",
" short		16-bit unsigned integer",
" long		32-bit unsigned integer",
" sbyte		8-bit signed integer",
" sshort		16-bit signed integer",
" slong		32-bit signed integer",
" float		32-bit IEEE floating point",
" double		64-bit IEEE floating point",
"",
" -p photo	photometric interpretation (color space) of the input image",
"where photo may be:",
" miniswhite	white color represented with 0 value",
" minisblack	black color represented with 0 value (default)",
" rgb		image has RGB color model",
" cmyk		image has CMYK (separated) color model",
" ycbcr		image has YCbCr color model",
" cielab		image has CIE L*a*b color model",
" icclab		image has ICC L*a*b color model",
" itulab		image has ITU L*a*b color model",
"",
" -s		swap bytes fetched from input file",
"",
" -i config	type of samples interleaving in input image",
"where config may be:",
" pixel		pixel interleaved data (default)",
" band		band interleaved data",
"",
" -c lzw[:opts]	compress output with Lempel-Ziv & Welch encoding",
" -c zip[:opts]	compress output with deflate encoding",
" -c jpeg[:opts]	compress output with JPEG encoding",
" -c packbits	compress output with packbits encoding",
" -c none	use no compression algorithm on output",
"",
"JPEG options:",
" #		set compression quality level (0-100, default 75)",
" r		output color image as RGB rather than YCbCr",
"For example, -c jpeg:r:50 to get JPEG-encoded RGB data with 50% comp. quality",
"",
"LZW and deflate options:",
" #		set predictor value",
"For example, -c lzw:2 to get LZW-encoded data with horizontal differencing",
" -o out.tif	write output to out.tif",
" -h		this help message",
NULL
};

static void
usage(void)
{
	char buf[BUFSIZ];
	int i;

	setbuf(stderr, buf);
        fprintf(stderr, "%s\n\n", TIFFGetVersion());
	for (i = 0; stuff[i] != NULL; i++)
		fprintf(stderr, "%s\n", stuff[i]);
	exit(-1);
}

/* vim: set ts=8 sts=8 sw=8 noet: */
Commit	Line	Data
	1	/* $Id$
	2	*
	3	* Project: libtiff tools
	4	* Purpose: Convert raw byte sequences in TIFF images
	5	* Author: Andrey Kiselev, dron@ak4719.spb.edu
	6	*
	7	******************************************************************************
	8	* Copyright (c) 2002, Andrey Kiselev <dron@ak4719.spb.edu>
	9	*
	10	* Permission to use, copy, modify, distribute, and sell this software and
	11	* its documentation for any purpose is hereby granted without fee, provided
	12	* that (i) the above copyright notices and this permission notice appear in
	13	* all copies of the software and related documentation, and (ii) the names of
	14	* Sam Leffler and Silicon Graphics may not be used in any advertising or
	15	* publicity relating to the software without the specific, prior written
	16	* permission of Sam Leffler and Silicon Graphics.
	17	*
	18	* THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
	19	* EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
	20	* WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
	21	*
	22	* IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
	23	* ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
	24	* OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
	25	* WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
	26	* LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
	27	* OF THIS SOFTWARE.
	28	*/
	29
	30	#include "tif_config.h"
	31
	32	#include <stdio.h>
	33	#include <stdlib.h>
	34	#include <string.h>
	35	#include <sys/stat.h>
	36	#include <sys/types.h>
	37	#include <math.h>
	38	#include <ctype.h>
	39
	40	#ifdef HAVE_UNISTD_H
	41	# include <unistd.h>
	42	#endif
	43
	44	#if HAVE_FCNTL_H
	45	# include <fcntl.h>
	46	#endif
	47
	48	#if HAVE_SYS_TYPES_H
	49	# include <sys/types.h>
	50	#endif
	51
	52	#if HAVE_IO_H
	53	# include <io.h>
	54	#endif
	55
	56	#include "tiffio.h"
	57
	58	#ifndef HAVE_GETOPT
	59	extern int getopt(int, char*, char);
	60	#endif
	61
	62	#ifndef O_BINARY
	63	# define O_BINARY 0
	64	#endif
	65
	66	typedef enum {
	67	PIXEL,
	68	BAND
	69	} InterleavingType;
	70
	71	static uint16 compression = (uint16) -1;
	72	static int jpegcolormode = JPEGCOLORMODE_RGB;
	73	static int quality = 75; /* JPEG quality */
	74	static uint16 predictor = 0;
	75
	76	static void swapBytesInScanline(void *, uint32, TIFFDataType);
	77	static int guessSize(int, TIFFDataType, off_t, uint32, int,
	78	uint32 , uint32 );
	79	static double correlation(void , void , uint32, TIFFDataType);
	80	static void usage(void);
	81	static int processCompressOptions(char*);
	82
	83	int
	84	main(int argc, char* argv[])
	85	{
	86	uint32 width = 0, length = 0, linebytes, bufsize;
	87	uint32 nbands = 1; /* number of bands in input image*/
	88	off_t hdr_size = 0; /* size of the header to skip */
	89	TIFFDataType dtype = TIFF_BYTE;
	90	int16 depth = 1; /* bytes per pixel in input image */
	91	int swab = 0; /* byte swapping flag */
	92	InterleavingType interleaving = 0; /* interleaving type flag */
	93	uint32 rowsperstrip = (uint32) -1;
	94	uint16 photometric = PHOTOMETRIC_MINISBLACK;
	95	uint16 config = PLANARCONFIG_CONTIG;
	96	uint16 fillorder = FILLORDER_LSB2MSB;
	97	int fd;
	98	char *outfilename = NULL;
	99	TIFF *out;
	100
	101	uint32 row, col, band;
	102	int c;
	103	unsigned char buf = NULL, buf1 = NULL;
	104	extern int optind;
	105	extern char* optarg;
	106
	107	while ((c = getopt(argc, argv, "c:r:H:w:l:b:d:LMp:si:o:h")) != -1) {
	108	switch (c) {
	109	case 'c': /* compression scheme */
	110	if (!processCompressOptions(optarg))
	111	usage();
	112	break;
	113	case 'r': /* rows/strip */
	114	rowsperstrip = atoi(optarg);
	115	break;
	116	case 'H': /* size of input image file header */
	117	hdr_size = atoi(optarg);
	118	break;
	119	case 'w': /* input image width */
	120	width = atoi(optarg);
	121	break;
	122	case 'l': /* input image length */
	123	length = atoi(optarg);
	124	break;
	125	case 'b': /* number of bands in input image */
	126	nbands = atoi(optarg);
	127	break;
	128	case 'd': /* type of samples in input image */
	129	if (strncmp(optarg, "byte", 4) == 0)
	130	dtype = TIFF_BYTE;
	131	else if (strncmp(optarg, "short", 5) == 0)
	132	dtype = TIFF_SHORT;
	133	else if (strncmp(optarg, "long", 4) == 0)
	134	dtype = TIFF_LONG;
	135	else if (strncmp(optarg, "sbyte", 5) == 0)
	136	dtype = TIFF_SBYTE;
	137	else if (strncmp(optarg, "sshort", 6) == 0)
	138	dtype = TIFF_SSHORT;
	139	else if (strncmp(optarg, "slong", 5) == 0)
	140	dtype = TIFF_SLONG;
	141	else if (strncmp(optarg, "float", 5) == 0)
	142	dtype = TIFF_FLOAT;
	143	else if (strncmp(optarg, "double", 6) == 0)
	144	dtype = TIFF_DOUBLE;
	145	else
	146	dtype = TIFF_BYTE;
	147	depth = TIFFDataWidth(dtype);
	148	break;
	149	case 'L': /* input has lsb-to-msb fillorder */
	150	fillorder = FILLORDER_LSB2MSB;
	151	break;
	152	case 'M': /* input has msb-to-lsb fillorder */
	153	fillorder = FILLORDER_MSB2LSB;
	154	break;
	155	case 'p': /* photometric interpretation */
	156	if (strncmp(optarg, "miniswhite", 10) == 0)
	157	photometric = PHOTOMETRIC_MINISWHITE;
	158	else if (strncmp(optarg, "minisblack", 10) == 0)
	159	photometric = PHOTOMETRIC_MINISBLACK;
	160	else if (strncmp(optarg, "rgb", 3) == 0)
	161	photometric = PHOTOMETRIC_RGB;
	162	else if (strncmp(optarg, "cmyk", 4) == 0)
	163	photometric = PHOTOMETRIC_SEPARATED;
	164	else if (strncmp(optarg, "ycbcr", 5) == 0)
	165	photometric = PHOTOMETRIC_YCBCR;
	166	else if (strncmp(optarg, "cielab", 6) == 0)
	167	photometric = PHOTOMETRIC_CIELAB;
	168	else if (strncmp(optarg, "icclab", 6) == 0)
	169	photometric = PHOTOMETRIC_ICCLAB;
	170	else if (strncmp(optarg, "itulab", 6) == 0)
	171	photometric = PHOTOMETRIC_ITULAB;
	172	else
	173	photometric = PHOTOMETRIC_MINISBLACK;
	174	break;
	175	case 's': /* do we need to swap bytes? */
	176	swab = 1;
	177	break;
	178	case 'i': /* type of interleaving */
	179	if (strncmp(optarg, "pixel", 4) == 0)
	180	interleaving = PIXEL;
	181	else if (strncmp(optarg, "band", 6) == 0)
	182	interleaving = BAND;
	183	else
	184	interleaving = 0;
	185	break;
	186	case 'o':
	187	outfilename = optarg;
	188	break;
	189	case 'h':
	190	usage();
	191	default:
	192	break;
	193	}
	194	}
	195
	196	if (argc - optind < 2)
	197	usage();
	198
	199	fd = open(argv[optind], O_RDONLY\|O_BINARY, 0);
	200	if (fd < 0) {
	201	fprintf(stderr, "%s: %s: Cannot open input file.\n",
	202	argv[0], argv[optind]);
	203	return (-1);
	204	}
	205
	206	if (guessSize(fd, dtype, hdr_size, nbands, swab, &width, &length) < 0)
	207	return 1;
	208
	209	if (outfilename == NULL)
	210	outfilename = argv[optind+1];
	211	out = TIFFOpen(outfilename, "w");
	212	if (out == NULL) {
	213	fprintf(stderr, "%s: %s: Cannot open file for output.\n",
	214	argv[0], outfilename);
	215	return (-1);
	216	}
	217	TIFFSetField(out, TIFFTAG_IMAGEWIDTH, width);
	218	TIFFSetField(out, TIFFTAG_IMAGELENGTH, length);
	219	TIFFSetField(out, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
	220	TIFFSetField(out, TIFFTAG_SAMPLESPERPIXEL, nbands);
	221	TIFFSetField(out, TIFFTAG_BITSPERSAMPLE, depth * 8);
	222	TIFFSetField(out, TIFFTAG_FILLORDER, fillorder);
	223	TIFFSetField(out, TIFFTAG_PLANARCONFIG, config);
	224	TIFFSetField(out, TIFFTAG_PHOTOMETRIC, photometric);
	225	switch (dtype) {
	226	case TIFF_BYTE:
	227	case TIFF_SHORT:
	228	case TIFF_LONG:
	229	TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_UINT);
	230	break;
	231	case TIFF_SBYTE:
	232	case TIFF_SSHORT:
	233	case TIFF_SLONG:
	234	TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_INT);
	235	break;
	236	case TIFF_FLOAT:
	237	case TIFF_DOUBLE:
	238	TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_IEEEFP);
	239	break;
	240	default:
	241	TIFFSetField(out, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_VOID);
	242	break;
	243	}
	244	if (compression == (uint16) -1)
	245	compression = COMPRESSION_PACKBITS;
	246	TIFFSetField(out, TIFFTAG_COMPRESSION, compression);
	247	switch (compression) {
	248	case COMPRESSION_JPEG:
	249	if (photometric == PHOTOMETRIC_RGB
	250	&& jpegcolormode == JPEGCOLORMODE_RGB)
	251	photometric = PHOTOMETRIC_YCBCR;
	252	TIFFSetField(out, TIFFTAG_JPEGQUALITY, quality);
	253	TIFFSetField(out, TIFFTAG_JPEGCOLORMODE, jpegcolormode);
	254	break;
	255	case COMPRESSION_LZW:
	256	case COMPRESSION_DEFLATE:
	257	if (predictor != 0)
	258	TIFFSetField(out, TIFFTAG_PREDICTOR, predictor);
	259	break;
	260	}
	261	switch(interleaving) {
	262	case BAND: /* band interleaved data */
	263	linebytes = width * depth;
	264	buf = (unsigned char *)_TIFFmalloc(linebytes);
	265	break;
	266	case PIXEL: /* pixel interleaved data */
	267	default:
	268	linebytes = width * nbands * depth;
	269	break;
	270	}
	271	bufsize = width * nbands * depth;
	272	buf1 = (unsigned char *)_TIFFmalloc(bufsize);
	273
	274	rowsperstrip = TIFFDefaultStripSize(out, rowsperstrip);
	275	if (rowsperstrip > length) {
	276	rowsperstrip = length;
	277	}
	278	TIFFSetField(out, TIFFTAG_ROWSPERSTRIP, rowsperstrip );
	279
	280	lseek(fd, hdr_size, SEEK_SET); /* Skip the file header */
	281	for (row = 0; row < length; row++) {
	282	switch(interleaving) {
	283	case BAND: /* band interleaved data */
	284	for (band = 0; band < nbands; band++) {
	285	lseek(fd,
	286	hdr_size + (lengthband+row)linebytes,
	287	SEEK_SET);
	288	if (read(fd, buf, linebytes) < 0) {
	289	fprintf(stderr,
	290	"%s: %s: scanline %lu: Read error.\n",
	291	argv[0], argv[optind],
	292	(unsigned long) row);
	293	break;
	294	}
	295	if (swab) /* Swap bytes if needed */
	296	swapBytesInScanline(buf, width, dtype);
	297	for (col = 0; col < width; col++)
	298	memcpy(buf1 + (colnbands+band)depth,
	299	buf + col * depth, depth);
	300	}
	301	break;
	302	case PIXEL: /* pixel interleaved data */
	303	default:
	304	if (read(fd, buf1, bufsize) < 0) {
	305	fprintf(stderr,
	306	"%s: %s: scanline %lu: Read error.\n",
	307	argv[0], argv[optind],
	308	(unsigned long) row);
	309	break;
	310	}
	311	if (swab) /* Swap bytes if needed */
	312	swapBytesInScanline(buf1, width, dtype);
	313	break;
	314	}
	315
	316	if (TIFFWriteScanline(out, buf1, row, 0) < 0) {
	317	fprintf(stderr, "%s: %s: scanline %lu: Write error.\n",
	318	argv[0], outfilename, (unsigned long) row);
	319	break;
	320	}
	321	}
	322	if (buf)
	323	_TIFFfree(buf);
	324	if (buf1)
	325	_TIFFfree(buf1);
	326	TIFFClose(out);
	327	return (0);
	328	}
	329
	330	static void
	331	swapBytesInScanline(void *buf, uint32 width, TIFFDataType dtype)
	332	{
	333	switch (dtype) {
	334	case TIFF_SHORT:
	335	case TIFF_SSHORT:
	336	TIFFSwabArrayOfShort((uint16*)buf,
	337	(unsigned long)width);
	338	break;
	339	case TIFF_LONG:
	340	case TIFF_SLONG:
	341	TIFFSwabArrayOfLong((uint32*)buf,
	342	(unsigned long)width);
	343	break;
	344	/* case TIFF_FLOAT: / / FIXME */
	345	case TIFF_DOUBLE:
	346	TIFFSwabArrayOfDouble((double*)buf,
	347	(unsigned long)width);
	348	break;
	349	default:
	350	break;
	351	}
	352	}
	353
	354	static int
	355	guessSize(int fd, TIFFDataType dtype, off_t hdr_size, uint32 nbands,
	356	int swab, uint32 width, uint32 length)
	357	{
	358	const float longt = 40.0; /* maximum possible height/width ratio */
	359	char buf1, buf2;
	360	struct stat filestat;
	361	uint32 w, h, scanlinesize, imagesize;
	362	uint32 depth = TIFFDataWidth(dtype);
	363	float cor_coef = 0, tmp;
	364
	365	fstat(fd, &filestat);
	366
	367	if (filestat.st_size < hdr_size) {
	368	fprintf(stderr, "Too large header size specified.\n");
	369	return -1;
	370	}
	371
	372	imagesize = (filestat.st_size - hdr_size) / nbands / depth;
	373
	374	if (width != 0 && length == 0) {
	375	fprintf(stderr, "Image height is not specified.\n");
	376
	377	length = imagesize / width;
	378
	379	fprintf(stderr, "Height is guessed as %lu.\n",
	380	(unsigned long)*length);
	381
	382	return 1;
	383	} else if (width == 0 && length != 0) {
	384	fprintf(stderr, "Image width is not specified.\n");
	385
	386	width = imagesize / length;
	387
	388	fprintf(stderr, "Width is guessed as %lu.\n",
	389	(unsigned long)*width);
	390
	391	return 1;
	392	} else if (width == 0 && length == 0) {
	393	fprintf(stderr, "Image width and height are not specified.\n");
	394
	395	for (w = (uint32) sqrt(imagesize / longt);
	396	w < sqrt(imagesize * longt);
	397	w++) {
	398	if (imagesize % w == 0) {
	399	scanlinesize = w * depth;
	400	buf1 = _TIFFmalloc(scanlinesize);
	401	buf2 = _TIFFmalloc(scanlinesize);
	402	h = imagesize / w;
	403	lseek(fd, hdr_size + (int)(h/2)*scanlinesize,
	404	SEEK_SET);
	405	read(fd, buf1, scanlinesize);
	406	read(fd, buf2, scanlinesize);
	407	if (swab) {
	408	swapBytesInScanline(buf1, w, dtype);
	409	swapBytesInScanline(buf2, w, dtype);
	410	}
	411	tmp = (float) fabs(correlation(buf1, buf2,
	412	w, dtype));
	413	if (tmp > cor_coef) {
	414	cor_coef = tmp;
	415	width = w, length = h;
	416	}
	417
	418	_TIFFfree(buf1);
	419	_TIFFfree(buf2);
	420	}
	421	}
	422
	423	fprintf(stderr,
	424	"Width is guessed as %lu, height is guessed as %lu.\n",
	425	(unsigned long)width, (unsigned long)length);
	426
	427	return 1;
	428	} else {
	429	if (filestat.st_size<(off_t)(hdr_size+(width)(length)nbands*depth)) {
	430	fprintf(stderr, "Input file too small.\n");
	431	return -1;
	432	}
	433	}
	434
	435	return 1;
	436	}
	437
	438	/* Calculate correlation coefficient between two numeric vectors */
	439	static double
	440	correlation(void buf1, void buf2, uint32 n_elem, TIFFDataType dtype)
	441	{
	442	double X, Y, M1 = 0.0, M2 = 0.0, D1 = 0.0, D2 = 0.0, K = 0.0;
	443	uint32 i;
	444
	445	switch (dtype) {
	446	case TIFF_BYTE:
	447	default:
	448	for (i = 0; i < n_elem; i++) {
	449	X = ((unsigned char *)buf1)[i];
	450	Y = ((unsigned char *)buf2)[i];
	451	M1 += X, M2 += Y;
	452	D1 += X * X, D2 += Y * Y;
	453	K += X * Y;
	454	}
	455	break;
	456	case TIFF_SBYTE:
	457	for (i = 0; i < n_elem; i++) {
	458	X = ((signed char *)buf1)[i];
	459	Y = ((signed char *)buf2)[i];
	460	M1 += X, M2 += Y;
	461	D1 += X * X, D2 += Y * Y;
	462	K += X * Y;
	463	}
	464	break;
	465	case TIFF_SHORT:
	466	for (i = 0; i < n_elem; i++) {
	467	X = ((uint16 *)buf1)[i];
	468	Y = ((uint16 *)buf2)[i];
	469	M1 += X, M2 += Y;
	470	D1 += X * X, D2 += Y * Y;
	471	K += X * Y;
	472	}
	473	break;
	474	case TIFF_SSHORT:
	475	for (i = 0; i < n_elem; i++) {
	476	X = ((int16 *)buf1)[i];
	477	Y = ((int16 *)buf2)[i];
	478	M1 += X, M2 += Y;
	479	D1 += X * X, D2 += Y * Y;
	480	K += X * Y;
	481	}
	482	break;
	483	case TIFF_LONG:
	484	for (i = 0; i < n_elem; i++) {
	485	X = ((uint32 *)buf1)[i];
	486	Y = ((uint32 *)buf2)[i];
	487	M1 += X, M2 += Y;
	488	D1 += X * X, D2 += Y * Y;
	489	K += X * Y;
	490	}
	491	break;
	492	case TIFF_SLONG:
	493	for (i = 0; i < n_elem; i++) {
	494	X = ((int32 *)buf1)[i];
	495	Y = ((int32 *)buf2)[i];
	496	M1 += X, M2 += Y;
	497	D1 += X * X, D2 += Y * Y;
	498	K += X * Y;
	499	}
	500	break;
	501	case TIFF_FLOAT:
	502	for (i = 0; i < n_elem; i++) {
	503	X = ((float *)buf1)[i];
	504	Y = ((float *)buf2)[i];
	505	M1 += X, M2 += Y;
	506	D1 += X * X, D2 += Y * Y;
	507	K += X * Y;
	508	}
	509	break;
	510	case TIFF_DOUBLE:
	511	for (i = 0; i < n_elem; i++) {
	512	X = ((double *)buf1)[i];
	513	Y = ((double *)buf2)[i];
	514	M1 += X, M2 += Y;
	515	D1 += X * X, D2 += Y * Y;
	516	K += X * Y;
	517	}
	518	break;
	519	}
	520
	521	M1 /= n_elem;
	522	M2 /= n_elem;
	523	D1 -= M1 * M1 * n_elem;
	524	D2 -= M2 * M2 * n_elem;
	525	K = (K - M1 * M2 * n_elem) / sqrt(D1 * D2);
	526
	527	return K;
	528	}
	529
	530	static int
	531	processCompressOptions(char* opt)
	532	{
	533	if (strcmp(opt, "none") == 0)
	534	compression = COMPRESSION_NONE;
	535	else if (strcmp(opt, "packbits") == 0)
	536	compression = COMPRESSION_PACKBITS;
	537	else if (strncmp(opt, "jpeg", 4) == 0) {
	538	char* cp = strchr(opt, ':');
	539
	540	compression = COMPRESSION_JPEG;
	541	while( cp )
	542	{
	543	if (isdigit((int)cp[1]))
	544	quality = atoi(cp+1);
	545	else if (cp[1] == 'r' )
	546	jpegcolormode = JPEGCOLORMODE_RAW;
	547	else
	548	usage();
	549
	550	cp = strchr(cp+1,':');
	551	}
	552	} else if (strncmp(opt, "lzw", 3) == 0) {
	553	char* cp = strchr(opt, ':');
	554	if (cp)
	555	predictor = atoi(cp+1);
	556	compression = COMPRESSION_LZW;
	557	} else if (strncmp(opt, "zip", 3) == 0) {
	558	char* cp = strchr(opt, ':');
	559	if (cp)
	560	predictor = atoi(cp+1);
	561	compression = COMPRESSION_DEFLATE;
	562	} else
	563	return (0);
	564	return (1);
	565	}
	566
	567	static char* stuff[] = {
	568	"raw2tiff --- tool for converting raw byte sequences in TIFF images",
	569	"usage: raw2tiff [options] input.raw output.tif",
	570	"where options are:",
	571	" -L input data has LSB2MSB bit order (default)",
	572	" -M input data has MSB2LSB bit order",
	573	" -r # make each strip have no more than # rows",
	574	" -H # size of input image file header in bytes (0 by default)",
	575	" -w # width of input image in pixels",
	576	" -l # length of input image in lines",
	577	" -b # number of bands in input image (1 by default)",
	578	"",
	579	" -d data_type type of samples in input image",
	580	"where data_type may be:",
	581	" byte 8-bit unsigned integer (default)",
	582	" short 16-bit unsigned integer",
	583	" long 32-bit unsigned integer",
	584	" sbyte 8-bit signed integer",
	585	" sshort 16-bit signed integer",
	586	" slong 32-bit signed integer",
	587	" float 32-bit IEEE floating point",
	588	" double 64-bit IEEE floating point",
	589	"",
	590	" -p photo photometric interpretation (color space) of the input image",
	591	"where photo may be:",
	592	" miniswhite white color represented with 0 value",
	593	" minisblack black color represented with 0 value (default)",
	594	" rgb image has RGB color model",
	595	" cmyk image has CMYK (separated) color model",
	596	" ycbcr image has YCbCr color model",
	597	" cielab image has CIE Lab color model",
	598	" icclab image has ICC Lab color model",
	599	" itulab image has ITU Lab color model",
	600	"",
	601	" -s swap bytes fetched from input file",
	602	"",
	603	" -i config type of samples interleaving in input image",
	604	"where config may be:",
	605	" pixel pixel interleaved data (default)",
	606	" band band interleaved data",
	607	"",
	608	" -c lzw[:opts] compress output with Lempel-Ziv & Welch encoding",
	609	" -c zip[:opts] compress output with deflate encoding",
	610	" -c jpeg[:opts] compress output with JPEG encoding",
	611	" -c packbits compress output with packbits encoding",
	612	" -c none use no compression algorithm on output",
	613	"",
	614	"JPEG options:",
	615	" # set compression quality level (0-100, default 75)",
	616	" r output color image as RGB rather than YCbCr",
	617	"For example, -c jpeg:r:50 to get JPEG-encoded RGB data with 50% comp. quality",
	618	"",
	619	"LZW and deflate options:",
	620	" # set predictor value",
	621	"For example, -c lzw:2 to get LZW-encoded data with horizontal differencing",
	622	" -o out.tif write output to out.tif",
	623	" -h this help message",
	624	NULL
	625	};
	626
	627	static void
	628	usage(void)
	629	{
	630	char buf[BUFSIZ];
	631	int i;
	632
	633	setbuf(stderr, buf);
	634	fprintf(stderr, "%s\n\n", TIFFGetVersion());
	635	for (i = 0; stuff[i] != NULL; i++)
	636	fprintf(stderr, "%s\n", stuff[i]);
	637	exit(-1);
	638	}
	639
	640	/* vim: set ts=8 sts=8 sw=8 noet: */