[wxWidgets.git] / src / tiff / libtiff / tif_zip.c

/* $Id$ */

/*
 * Copyright (c) 1995-1997 Sam Leffler
 * Copyright (c) 1995-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 "tiffiop.h"
#ifdef ZIP_SUPPORT
/*
 * TIFF Library.
 *
 * ZIP (aka Deflate) Compression Support
 *
 * This file is simply an interface to the zlib library written by
 * Jean-loup Gailly and Mark Adler.  You must use version 1.0 or later
 * of the library: this code assumes the 1.0 API and also depends on
 * the ability to write the zlib header multiple times (one per strip)
 * which was not possible with versions prior to 0.95.  Note also that
 * older versions of this codec avoided this bug by supressing the header
 * entirely.  This means that files written with the old library cannot
 * be read; they should be converted to a different compression scheme
 * and then reconverted.
 *
 * The data format used by the zlib library is described in the files
 * zlib-3.1.doc, deflate-1.1.doc and gzip-4.1.doc, available in the
 * directory ftp://ftp.uu.net/pub/archiving/zip/doc.  The library was
 * last found at ftp://ftp.uu.net/pub/archiving/zip/zlib/zlib-0.99.tar.gz.
 */
#include "tif_predict.h"
#include "zlib.h"

#include <stdio.h>

/*
 * Sigh, ZLIB_VERSION is defined as a string so there's no
 * way to do a proper check here.  Instead we guess based
 * on the presence of #defines that were added between the
 * 0.95 and 1.0 distributions.
 */
#if !defined(Z_NO_COMPRESSION) || !defined(Z_DEFLATED)
#error "Antiquated ZLIB software; you must use version 1.0 or later"
#endif

/*
 * State block for each open TIFF
 * file using ZIP compression/decompression.
 */
typedef struct {
	TIFFPredictorState predict;
        z_stream        stream;
	int             zipquality;            /* compression level */
	int             state;                 /* state flags */
#define ZSTATE_INIT_DECODE 0x01
#define ZSTATE_INIT_ENCODE 0x02

	TIFFVGetMethod  vgetparent;            /* super-class method */
	TIFFVSetMethod  vsetparent;            /* super-class method */
} ZIPState;

#define ZState(tif)             ((ZIPState*) (tif)->tif_data)
#define DecoderState(tif)       ZState(tif)
#define EncoderState(tif)       ZState(tif)

static int ZIPEncode(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s);
static int ZIPDecode(TIFF* tif, uint8* op, tmsize_t occ, uint16 s);

static int
ZIPFixupTags(TIFF* tif)
{
	(void) tif;
	return (1);
}

static int
ZIPSetupDecode(TIFF* tif)
{
	static const char module[] = "ZIPSetupDecode";
	ZIPState* sp = DecoderState(tif);

	assert(sp != NULL);
        
        /* if we were last encoding, terminate this mode */
	if (sp->state & ZSTATE_INIT_ENCODE) {
	    deflateEnd(&sp->stream);
	    sp->state = 0;
	}

	if (inflateInit(&sp->stream) != Z_OK) {
		TIFFErrorExt(tif->tif_clientdata, module, "%s", sp->stream.msg);
		return (0);
	} else {
		sp->state |= ZSTATE_INIT_DECODE;
		return (1);
	}
}

/*
 * Setup state for decoding a strip.
 */
static int
ZIPPreDecode(TIFF* tif, uint16 s)
{
	static const char module[] = "ZIPPreDecode";
	ZIPState* sp = DecoderState(tif);

	(void) s;
	assert(sp != NULL);

	if( (sp->state & ZSTATE_INIT_DECODE) == 0 )
            tif->tif_setupdecode( tif );

	sp->stream.next_in = tif->tif_rawdata;
	assert(sizeof(sp->stream.avail_in)==4);  /* if this assert gets raised,
	    we need to simplify this code to reflect a ZLib that is likely updated
	    to deal with 8byte memory sizes, though this code will respond
	    apropriately even before we simplify it */
	sp->stream.avail_in = (uInt) tif->tif_rawcc;
	if ((tmsize_t)sp->stream.avail_in != tif->tif_rawcc)
	{
		TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
		return (0);
	}
	return (inflateReset(&sp->stream) == Z_OK);
}

static int
ZIPDecode(TIFF* tif, uint8* op, tmsize_t occ, uint16 s)
{
	static const char module[] = "ZIPDecode";
	ZIPState* sp = DecoderState(tif);

	(void) s;
	assert(sp != NULL);
	assert(sp->state == ZSTATE_INIT_DECODE);

        sp->stream.next_in = tif->tif_rawcp;
	sp->stream.avail_in = (uInt) tif->tif_rawcc;
        
	sp->stream.next_out = op;
	assert(sizeof(sp->stream.avail_out)==4);  /* if this assert gets raised,
	    we need to simplify this code to reflect a ZLib that is likely updated
	    to deal with 8byte memory sizes, though this code will respond
	    apropriately even before we simplify it */
	sp->stream.avail_out = (uInt) occ;
	if ((tmsize_t)sp->stream.avail_out != occ)
	{
		TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
		return (0);
	}
	do {
		int state = inflate(&sp->stream, Z_PARTIAL_FLUSH);
		if (state == Z_STREAM_END)
			break;
		if (state == Z_DATA_ERROR) {
			TIFFErrorExt(tif->tif_clientdata, module,
			    "Decoding error at scanline %lu, %s",
			    (unsigned long) tif->tif_row, sp->stream.msg);
			if (inflateSync(&sp->stream) != Z_OK)
				return (0);
			continue;
		}
		if (state != Z_OK) {
			TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
			    sp->stream.msg);
			return (0);
		}
	} while (sp->stream.avail_out > 0);
	if (sp->stream.avail_out != 0) {
		TIFFErrorExt(tif->tif_clientdata, module,
		    "Not enough data at scanline %lu (short " TIFF_UINT64_FORMAT " bytes)",
		    (unsigned long) tif->tif_row, (TIFF_UINT64_T) sp->stream.avail_out);
		return (0);
	}

        tif->tif_rawcp = sp->stream.next_in;
        tif->tif_rawcc = sp->stream.avail_in;

	return (1);
}

static int
ZIPSetupEncode(TIFF* tif)
{
	static const char module[] = "ZIPSetupEncode";
	ZIPState* sp = EncoderState(tif);

	assert(sp != NULL);
	if (sp->state & ZSTATE_INIT_DECODE) {
		inflateEnd(&sp->stream);
		sp->state = 0;
	}

	if (deflateInit(&sp->stream, sp->zipquality) != Z_OK) {
		TIFFErrorExt(tif->tif_clientdata, module, "%s", sp->stream.msg);
		return (0);
	} else {
		sp->state |= ZSTATE_INIT_ENCODE;
		return (1);
	}
}

/*
 * Reset encoding state at the start of a strip.
 */
static int
ZIPPreEncode(TIFF* tif, uint16 s)
{
	static const char module[] = "ZIPPreEncode";
	ZIPState *sp = EncoderState(tif);

	(void) s;
	assert(sp != NULL);
	if( sp->state != ZSTATE_INIT_ENCODE )
            tif->tif_setupencode( tif );

	sp->stream.next_out = tif->tif_rawdata;
	assert(sizeof(sp->stream.avail_out)==4);  /* if this assert gets raised,
	    we need to simplify this code to reflect a ZLib that is likely updated
	    to deal with 8byte memory sizes, though this code will respond
	    apropriately even before we simplify it */
	sp->stream.avail_out = tif->tif_rawdatasize;
	if ((tmsize_t)sp->stream.avail_out != tif->tif_rawdatasize)
	{
		TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
		return (0);
	}
	return (deflateReset(&sp->stream) == Z_OK);
}

/*
 * Encode a chunk of pixels.
 */
static int
ZIPEncode(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
{
	static const char module[] = "ZIPEncode";
	ZIPState *sp = EncoderState(tif);

	assert(sp != NULL);
	assert(sp->state == ZSTATE_INIT_ENCODE);

	(void) s;
	sp->stream.next_in = bp;
	assert(sizeof(sp->stream.avail_in)==4);  /* if this assert gets raised,
	    we need to simplify this code to reflect a ZLib that is likely updated
	    to deal with 8byte memory sizes, though this code will respond
	    apropriately even before we simplify it */
	sp->stream.avail_in = (uInt) cc;
	if ((tmsize_t)sp->stream.avail_in != cc)
	{
		TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
		return (0);
	}
	do {
		if (deflate(&sp->stream, Z_NO_FLUSH) != Z_OK) {
			TIFFErrorExt(tif->tif_clientdata, module, "Encoder error: %s",
			    sp->stream.msg);
			return (0);
		}
		if (sp->stream.avail_out == 0) {
			tif->tif_rawcc = tif->tif_rawdatasize;
			TIFFFlushData1(tif);
			sp->stream.next_out = tif->tif_rawdata;
			sp->stream.avail_out = (uInt) tif->tif_rawdatasize;  /* this is a safe typecast, as check is made already in ZIPPreEncode */
		}
	} while (sp->stream.avail_in > 0);
	return (1);
}

/*
 * Finish off an encoded strip by flushing the last
 * string and tacking on an End Of Information code.
 */
static int
ZIPPostEncode(TIFF* tif)
{
	static const char module[] = "ZIPPostEncode";
	ZIPState *sp = EncoderState(tif);
	int state;

	sp->stream.avail_in = 0;
	do {
		state = deflate(&sp->stream, Z_FINISH);
		switch (state) {
		case Z_STREAM_END:
		case Z_OK:
			if ((tmsize_t)sp->stream.avail_out != tif->tif_rawdatasize)
			{
				tif->tif_rawcc =  tif->tif_rawdatasize - sp->stream.avail_out;
				TIFFFlushData1(tif);
				sp->stream.next_out = tif->tif_rawdata;
				sp->stream.avail_out = (uInt) tif->tif_rawdatasize;  /* this is a safe typecast, as check is made already in ZIPPreEncode */
			}
			break;
		default:
			TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
			    sp->stream.msg);
			return (0);
		}
	} while (state != Z_STREAM_END);
	return (1);
}

static void
ZIPCleanup(TIFF* tif)
{
	ZIPState* sp = ZState(tif);

	assert(sp != 0);

	(void)TIFFPredictorCleanup(tif);

	tif->tif_tagmethods.vgetfield = sp->vgetparent;
	tif->tif_tagmethods.vsetfield = sp->vsetparent;

	if (sp->state & ZSTATE_INIT_ENCODE) {
		deflateEnd(&sp->stream);
		sp->state = 0;
	} else if( sp->state & ZSTATE_INIT_DECODE) {
		inflateEnd(&sp->stream);
		sp->state = 0;
	}
	_TIFFfree(sp);
	tif->tif_data = NULL;

	_TIFFSetDefaultCompressionState(tif);
}

static int
ZIPVSetField(TIFF* tif, uint32 tag, va_list ap)
{
	static const char module[] = "ZIPVSetField";
	ZIPState* sp = ZState(tif);

	switch (tag) {
	case TIFFTAG_ZIPQUALITY:
		sp->zipquality = (int) va_arg(ap, int);
		if ( sp->state&ZSTATE_INIT_ENCODE ) {
			if (deflateParams(&sp->stream,
			    sp->zipquality, Z_DEFAULT_STRATEGY) != Z_OK) {
				TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
				    sp->stream.msg);
				return (0);
			}
		}
		return (1);
	default:
		return (*sp->vsetparent)(tif, tag, ap);
	}
	/*NOTREACHED*/
}

static int
ZIPVGetField(TIFF* tif, uint32 tag, va_list ap)
{
	ZIPState* sp = ZState(tif);

	switch (tag) {
	case TIFFTAG_ZIPQUALITY:
		*va_arg(ap, int*) = sp->zipquality;
		break;
	default:
		return (*sp->vgetparent)(tif, tag, ap);
	}
	return (1);
}

static const TIFFField zipFields[] = {
    { TIFFTAG_ZIPQUALITY, 0, 0, TIFF_ANY, 0, TIFF_SETGET_INT, TIFF_SETGET_UNDEFINED, FIELD_PSEUDO, TRUE, FALSE, "", NULL },
};

int
TIFFInitZIP(TIFF* tif, int scheme)
{
	static const char module[] = "TIFFInitZIP";
	ZIPState* sp;

	assert( (scheme == COMPRESSION_DEFLATE)
		|| (scheme == COMPRESSION_ADOBE_DEFLATE));

	/*
	 * Merge codec-specific tag information.
	 */
	if (!_TIFFMergeFields(tif, zipFields, TIFFArrayCount(zipFields))) {
		TIFFErrorExt(tif->tif_clientdata, module,
			     "Merging Deflate codec-specific tags failed");
		return 0;
	}

	/*
	 * Allocate state block so tag methods have storage to record values.
	 */
	tif->tif_data = (uint8*) _TIFFmalloc(sizeof (ZIPState));
	if (tif->tif_data == NULL)
		goto bad;
	sp = ZState(tif);
	sp->stream.zalloc = NULL;
	sp->stream.zfree = NULL;
	sp->stream.opaque = NULL;
	sp->stream.data_type = Z_BINARY;

	/*
	 * Override parent get/set field methods.
	 */
	sp->vgetparent = tif->tif_tagmethods.vgetfield;
	tif->tif_tagmethods.vgetfield = ZIPVGetField; /* hook for codec tags */
	sp->vsetparent = tif->tif_tagmethods.vsetfield;
	tif->tif_tagmethods.vsetfield = ZIPVSetField; /* hook for codec tags */

	/* Default values for codec-specific fields */
	sp->zipquality = Z_DEFAULT_COMPRESSION;	/* default comp. level */
	sp->state = 0;

	/*
	 * Install codec methods.
	 */
	tif->tif_fixuptags = ZIPFixupTags; 
	tif->tif_setupdecode = ZIPSetupDecode;
	tif->tif_predecode = ZIPPreDecode;
	tif->tif_decoderow = ZIPDecode;
	tif->tif_decodestrip = ZIPDecode;
	tif->tif_decodetile = ZIPDecode;  
	tif->tif_setupencode = ZIPSetupEncode;
	tif->tif_preencode = ZIPPreEncode;
	tif->tif_postencode = ZIPPostEncode;
	tif->tif_encoderow = ZIPEncode;
	tif->tif_encodestrip = ZIPEncode;
	tif->tif_encodetile = ZIPEncode;
	tif->tif_cleanup = ZIPCleanup;
	/*
	 * Setup predictor setup.
	 */
	(void) TIFFPredictorInit(tif);
	return (1);
bad:
	TIFFErrorExt(tif->tif_clientdata, module,
		     "No space for ZIP state block");
	return (0);
}
#endif /* ZIP_SUPORT */

/* vim: set ts=8 sts=8 sw=8 noet: */
/*
 * Local Variables:
 * mode: c
 * c-basic-offset: 8
 * fill-column: 78
 * End:
 */
Commit	Line	Data
8414a40c VZ	1	/* $Id$ */
	2
	3	/*
	4	* Copyright (c) 1995-1997 Sam Leffler
	5	* Copyright (c) 1995-1997 Silicon Graphics, Inc.
	6	*
	7	* Permission to use, copy, modify, distribute, and sell this software and
	8	* its documentation for any purpose is hereby granted without fee, provided
	9	* that (i) the above copyright notices and this permission notice appear in
	10	* all copies of the software and related documentation, and (ii) the names of
	11	* Sam Leffler and Silicon Graphics may not be used in any advertising or
	12	* publicity relating to the software without the specific, prior written
	13	* permission of Sam Leffler and Silicon Graphics.
	14	*
	15	* THE SOFTWARE IS PROVIDED "AS-IS" AND WITHOUT WARRANTY OF ANY KIND,
	16	* EXPRESS, IMPLIED OR OTHERWISE, INCLUDING WITHOUT LIMITATION, ANY
	17	* WARRANTY OF MERCHANTABILITY OR FITNESS FOR A PARTICULAR PURPOSE.
	18	*
	19	* IN NO EVENT SHALL SAM LEFFLER OR SILICON GRAPHICS BE LIABLE FOR
	20	* ANY SPECIAL, INCIDENTAL, INDIRECT OR CONSEQUENTIAL DAMAGES OF ANY KIND,
	21	* OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
	22	* WHETHER OR NOT ADVISED OF THE POSSIBILITY OF DAMAGE, AND ON ANY THEORY OF
	23	* LIABILITY, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
	24	* OF THIS SOFTWARE.
	25	*/
	26
	27	#include "tiffiop.h"
	28	#ifdef ZIP_SUPPORT
	29	/*
	30	* TIFF Library.
	31	*
	32	* ZIP (aka Deflate) Compression Support
	33	*
	34	* This file is simply an interface to the zlib library written by
	35	* Jean-loup Gailly and Mark Adler. You must use version 1.0 or later
	36	* of the library: this code assumes the 1.0 API and also depends on
	37	* the ability to write the zlib header multiple times (one per strip)
	38	* which was not possible with versions prior to 0.95. Note also that
	39	* older versions of this codec avoided this bug by supressing the header
	40	* entirely. This means that files written with the old library cannot
	41	* be read; they should be converted to a different compression scheme
	42	* and then reconverted.
	43	*
	44	* The data format used by the zlib library is described in the files
	45	* zlib-3.1.doc, deflate-1.1.doc and gzip-4.1.doc, available in the
	46	* directory ftp://ftp.uu.net/pub/archiving/zip/doc. The library was
	47	* last found at ftp://ftp.uu.net/pub/archiving/zip/zlib/zlib-0.99.tar.gz.
	48	*/
	49	#include "tif_predict.h"
	50	#include "zlib.h"
	51
	52	#include <stdio.h>
	53
	54	/*
	55	* Sigh, ZLIB_VERSION is defined as a string so there's no
	56	* way to do a proper check here. Instead we guess based
	57	* on the presence of #defines that were added between the
	58	* 0.95 and 1.0 distributions.
	59	*/
	60	#if !defined(Z_NO_COMPRESSION) \|\| !defined(Z_DEFLATED)
	61	#error "Antiquated ZLIB software; you must use version 1.0 or later"
	62	#endif
	63
	64	/*
65	* State block for each open TIFF
66	* file using ZIP compression/decompression.
67	*/
80ed523f	68	typedef struct {
8414a40c	69	TIFFPredictorState predict;
80ed523f VZ	70	z_stream stream;
	71	int zipquality; /* compression level */
	72	int state; /* state flags */
	73	#define ZSTATE_INIT_DECODE 0x01
	74	#define ZSTATE_INIT_ENCODE 0x02
	75
	76	TIFFVGetMethod vgetparent; /* super-class method */
	77	TIFFVSetMethod vsetparent; /* super-class method */
8414a40c VZ	78	} ZIPState;
8414a40c VZ	79
80ed523f VZ	80	#define ZState(tif) ((ZIPState*) (tif)->tif_data)
	81	#define DecoderState(tif) ZState(tif)
	82	#define EncoderState(tif) ZState(tif)
	83
	84	static int ZIPEncode(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s);
	85	static int ZIPDecode(TIFF* tif, uint8* op, tmsize_t occ, uint16 s);
8414a40c	86
80ed523f VZ	87	static int
	88	ZIPFixupTags(TIFF* tif)
	89	{
	90	(void) tif;
	91	return (1);
	92	}
8414a40c VZ	93
	94	static int
	95	ZIPSetupDecode(TIFF* tif)
	96	{
8414a40c	97	static const char module[] = "ZIPSetupDecode";
80ed523f	98	ZIPState* sp = DecoderState(tif);
8414a40c VZ	99
8414a40c VZ	100	assert(sp != NULL);
80ed523f VZ	101
	102	/* if we were last encoding, terminate this mode */
	103	if (sp->state & ZSTATE_INIT_ENCODE) {
	104	deflateEnd(&sp->stream);
	105	sp->state = 0;
	106	}
	107
8414a40c	108	if (inflateInit(&sp->stream) != Z_OK) {
80ed523f	109	TIFFErrorExt(tif->tif_clientdata, module, "%s", sp->stream.msg);
8414a40c VZ	110	return (0);
8414a40c VZ	111	} else {
80ed523f	112	sp->state \|= ZSTATE_INIT_DECODE;
8414a40c VZ	113	return (1);
	114	}
	115	}
	116
	117	/*
	118	* Setup state for decoding a strip.
	119	*/
	120	static int
80ed523f	121	ZIPPreDecode(TIFF* tif, uint16 s)
8414a40c	122	{
80ed523f	123	static const char module[] = "ZIPPreDecode";
8414a40c VZ	124	ZIPState* sp = DecoderState(tif);
	125
	126	(void) s;
	127	assert(sp != NULL);
80ed523f VZ	128
	129	if( (sp->state & ZSTATE_INIT_DECODE) == 0 )
	130	tif->tif_setupdecode( tif );
	131
8414a40c	132	sp->stream.next_in = tif->tif_rawdata;
80ed523f VZ	133	assert(sizeof(sp->stream.avail_in)==4); /* if this assert gets raised,
	134	we need to simplify this code to reflect a ZLib that is likely updated
	135	to deal with 8byte memory sizes, though this code will respond
	136	apropriately even before we simplify it */
	137	sp->stream.avail_in = (uInt) tif->tif_rawcc;
	138	if ((tmsize_t)sp->stream.avail_in != tif->tif_rawcc)
	139	{
	140	TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
	141	return (0);
	142	}
8414a40c VZ	143	return (inflateReset(&sp->stream) == Z_OK);
	144	}
	145
	146	static int
80ed523f	147	ZIPDecode(TIFF* tif, uint8* op, tmsize_t occ, uint16 s)
8414a40c	148	{
8414a40c	149	static const char module[] = "ZIPDecode";
80ed523f	150	ZIPState* sp = DecoderState(tif);
8414a40c VZ	151
	152	(void) s;
	153	assert(sp != NULL);
80ed523f VZ	154	assert(sp->state == ZSTATE_INIT_DECODE);
	155
	156	sp->stream.next_in = tif->tif_rawcp;
	157	sp->stream.avail_in = (uInt) tif->tif_rawcc;
	158
8414a40c	159	sp->stream.next_out = op;
80ed523f VZ	160	assert(sizeof(sp->stream.avail_out)==4); /* if this assert gets raised,
	161	we need to simplify this code to reflect a ZLib that is likely updated
	162	to deal with 8byte memory sizes, though this code will respond
	163	apropriately even before we simplify it */
	164	sp->stream.avail_out = (uInt) occ;
	165	if ((tmsize_t)sp->stream.avail_out != occ)
	166	{
	167	TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
	168	return (0);
	169	}
8414a40c VZ	170	do {
	171	int state = inflate(&sp->stream, Z_PARTIAL_FLUSH);
	172	if (state == Z_STREAM_END)
	173	break;
	174	if (state == Z_DATA_ERROR) {
	175	TIFFErrorExt(tif->tif_clientdata, module,
80ed523f VZ	176	"Decoding error at scanline %lu, %s",
80ed523f VZ	177	(unsigned long) tif->tif_row, sp->stream.msg);
8414a40c VZ	178	if (inflateSync(&sp->stream) != Z_OK)
	179	return (0);
	180	continue;
	181	}
	182	if (state != Z_OK) {
80ed523f VZ	183	TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
80ed523f VZ	184	sp->stream.msg);
8414a40c VZ	185	return (0);
	186	}
	187	} while (sp->stream.avail_out > 0);
	188	if (sp->stream.avail_out != 0) {
	189	TIFFErrorExt(tif->tif_clientdata, module,
80ed523f VZ	190	"Not enough data at scanline %lu (short " TIFF_UINT64_FORMAT " bytes)",
80ed523f VZ	191	(unsigned long) tif->tif_row, (TIFF_UINT64_T) sp->stream.avail_out);
8414a40c VZ	192	return (0);
8414a40c VZ	193	}
80ed523f VZ	194
	195	tif->tif_rawcp = sp->stream.next_in;
	196	tif->tif_rawcc = sp->stream.avail_in;
	197
8414a40c VZ	198	return (1);
	199	}
	200
	201	static int
	202	ZIPSetupEncode(TIFF* tif)
	203	{
8414a40c	204	static const char module[] = "ZIPSetupEncode";
80ed523f	205	ZIPState* sp = EncoderState(tif);
8414a40c VZ	206
8414a40c VZ	207	assert(sp != NULL);
80ed523f VZ	208	if (sp->state & ZSTATE_INIT_DECODE) {
	209	inflateEnd(&sp->stream);
	210	sp->state = 0;
	211	}
	212
8414a40c	213	if (deflateInit(&sp->stream, sp->zipquality) != Z_OK) {
80ed523f	214	TIFFErrorExt(tif->tif_clientdata, module, "%s", sp->stream.msg);
8414a40c VZ	215	return (0);
8414a40c VZ	216	} else {
80ed523f	217	sp->state \|= ZSTATE_INIT_ENCODE;
8414a40c VZ	218	return (1);
	219	}
	220	}
	221
	222	/*
	223	* Reset encoding state at the start of a strip.
	224	*/
	225	static int
80ed523f	226	ZIPPreEncode(TIFF* tif, uint16 s)
8414a40c	227	{
80ed523f	228	static const char module[] = "ZIPPreEncode";
8414a40c VZ	229	ZIPState *sp = EncoderState(tif);
	230
	231	(void) s;
	232	assert(sp != NULL);
80ed523f VZ	233	if( sp->state != ZSTATE_INIT_ENCODE )
	234	tif->tif_setupencode( tif );
	235
8414a40c	236	sp->stream.next_out = tif->tif_rawdata;
80ed523f VZ	237	assert(sizeof(sp->stream.avail_out)==4); /* if this assert gets raised,
	238	we need to simplify this code to reflect a ZLib that is likely updated
	239	to deal with 8byte memory sizes, though this code will respond
	240	apropriately even before we simplify it */
8414a40c	241	sp->stream.avail_out = tif->tif_rawdatasize;
80ed523f VZ	242	if ((tmsize_t)sp->stream.avail_out != tif->tif_rawdatasize)
	243	{
	244	TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
	245	return (0);
	246	}
8414a40c VZ	247	return (deflateReset(&sp->stream) == Z_OK);
	248	}
	249
	250	/*
	251	* Encode a chunk of pixels.
	252	*/
	253	static int
80ed523f	254	ZIPEncode(TIFF* tif, uint8* bp, tmsize_t cc, uint16 s)
8414a40c	255	{
8414a40c	256	static const char module[] = "ZIPEncode";
80ed523f VZ	257	ZIPState *sp = EncoderState(tif);
	258
	259	assert(sp != NULL);
	260	assert(sp->state == ZSTATE_INIT_ENCODE);
8414a40c VZ	261
	262	(void) s;
	263	sp->stream.next_in = bp;
80ed523f VZ	264	assert(sizeof(sp->stream.avail_in)==4); /* if this assert gets raised,
	265	we need to simplify this code to reflect a ZLib that is likely updated
	266	to deal with 8byte memory sizes, though this code will respond
	267	apropriately even before we simplify it */
	268	sp->stream.avail_in = (uInt) cc;
	269	if ((tmsize_t)sp->stream.avail_in != cc)
	270	{
	271	TIFFErrorExt(tif->tif_clientdata, module, "ZLib cannot deal with buffers this size");
	272	return (0);
	273	}
8414a40c VZ	274	do {
8414a40c VZ	275	if (deflate(&sp->stream, Z_NO_FLUSH) != Z_OK) {
80ed523f VZ	276	TIFFErrorExt(tif->tif_clientdata, module, "Encoder error: %s",
80ed523f VZ	277	sp->stream.msg);
8414a40c VZ	278	return (0);
	279	}
	280	if (sp->stream.avail_out == 0) {
	281	tif->tif_rawcc = tif->tif_rawdatasize;
	282	TIFFFlushData1(tif);
	283	sp->stream.next_out = tif->tif_rawdata;
80ed523f	284	sp->stream.avail_out = (uInt) tif->tif_rawdatasize; /* this is a safe typecast, as check is made already in ZIPPreEncode */
8414a40c VZ	285	}
	286	} while (sp->stream.avail_in > 0);
	287	return (1);
	288	}
	289
	290	/*
	291	* Finish off an encoded strip by flushing the last
	292	* string and tacking on an End Of Information code.
	293	*/
	294	static int
	295	ZIPPostEncode(TIFF* tif)
	296	{
8414a40c	297	static const char module[] = "ZIPPostEncode";
80ed523f	298	ZIPState *sp = EncoderState(tif);
8414a40c VZ	299	int state;
	300
	301	sp->stream.avail_in = 0;
	302	do {
	303	state = deflate(&sp->stream, Z_FINISH);
	304	switch (state) {
	305	case Z_STREAM_END:
	306	case Z_OK:
80ed523f VZ	307	if ((tmsize_t)sp->stream.avail_out != tif->tif_rawdatasize)
	308	{
	309	tif->tif_rawcc = tif->tif_rawdatasize - sp->stream.avail_out;
	310	TIFFFlushData1(tif);
	311	sp->stream.next_out = tif->tif_rawdata;
	312	sp->stream.avail_out = (uInt) tif->tif_rawdatasize; /* this is a safe typecast, as check is made already in ZIPPreEncode */
	313	}
	314	break;
8414a40c	315	default:
80ed523f VZ	316	TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
	317	sp->stream.msg);
	318	return (0);
8414a40c VZ	319	}
	320	} while (state != Z_STREAM_END);
	321	return (1);
	322	}
	323
	324	static void
	325	ZIPCleanup(TIFF* tif)
	326	{
	327	ZIPState* sp = ZState(tif);
	328
	329	assert(sp != 0);
	330
	331	(void)TIFFPredictorCleanup(tif);
	332
	333	tif->tif_tagmethods.vgetfield = sp->vgetparent;
	334	tif->tif_tagmethods.vsetfield = sp->vsetparent;
	335
80ed523f VZ	336	if (sp->state & ZSTATE_INIT_ENCODE) {
	337	deflateEnd(&sp->stream);
	338	sp->state = 0;
	339	} else if( sp->state & ZSTATE_INIT_DECODE) {
	340	inflateEnd(&sp->stream);
	341	sp->state = 0;
8414a40c VZ	342	}
	343	_TIFFfree(sp);
	344	tif->tif_data = NULL;
	345
	346	_TIFFSetDefaultCompressionState(tif);
	347	}
	348
	349	static int
80ed523f	350	ZIPVSetField(TIFF* tif, uint32 tag, va_list ap)
8414a40c	351	{
8414a40c	352	static const char module[] = "ZIPVSetField";
80ed523f	353	ZIPState* sp = ZState(tif);
8414a40c VZ	354
	355	switch (tag) {
	356	case TIFFTAG_ZIPQUALITY:
80ed523f VZ	357	sp->zipquality = (int) va_arg(ap, int);
80ed523f VZ	358	if ( sp->state&ZSTATE_INIT_ENCODE ) {
8414a40c VZ	359	if (deflateParams(&sp->stream,
8414a40c VZ	360	sp->zipquality, Z_DEFAULT_STRATEGY) != Z_OK) {
80ed523f VZ	361	TIFFErrorExt(tif->tif_clientdata, module, "ZLib error: %s",
80ed523f VZ	362	sp->stream.msg);
8414a40c VZ	363	return (0);
	364	}
	365	}
	366	return (1);
	367	default:
	368	return (*sp->vsetparent)(tif, tag, ap);
	369	}
	370	/NOTREACHED/
	371	}
	372
	373	static int
80ed523f	374	ZIPVGetField(TIFF* tif, uint32 tag, va_list ap)
8414a40c VZ	375	{
	376	ZIPState* sp = ZState(tif);
	377
	378	switch (tag) {
	379	case TIFFTAG_ZIPQUALITY:
	380	va_arg(ap, int) = sp->zipquality;
	381	break;
	382	default:
	383	return (*sp->vgetparent)(tif, tag, ap);
	384	}
	385	return (1);
	386	}
	387
80ed523f VZ	388	static const TIFFField zipFields[] = {
80ed523f VZ	389	{ TIFFTAG_ZIPQUALITY, 0, 0, TIFF_ANY, 0, TIFF_SETGET_INT, TIFF_SETGET_UNDEFINED, FIELD_PSEUDO, TRUE, FALSE, "", NULL },
8414a40c VZ	390	};
	391
	392	int
	393	TIFFInitZIP(TIFF* tif, int scheme)
	394	{
80ed523f	395	static const char module[] = "TIFFInitZIP";
8414a40c VZ	396	ZIPState* sp;
	397
	398	assert( (scheme == COMPRESSION_DEFLATE)
	399	\|\| (scheme == COMPRESSION_ADOBE_DEFLATE));
	400
80ed523f VZ	401	/*
	402	* Merge codec-specific tag information.
	403	*/
	404	if (!_TIFFMergeFields(tif, zipFields, TIFFArrayCount(zipFields))) {
	405	TIFFErrorExt(tif->tif_clientdata, module,
	406	"Merging Deflate codec-specific tags failed");
	407	return 0;
	408	}
	409
8414a40c VZ	410	/*
	411	* Allocate state block so tag methods have storage to record values.
	412	*/
80ed523f	413	tif->tif_data = (uint8*) _TIFFmalloc(sizeof (ZIPState));
8414a40c VZ	414	if (tif->tif_data == NULL)
	415	goto bad;
	416	sp = ZState(tif);
	417	sp->stream.zalloc = NULL;
	418	sp->stream.zfree = NULL;
	419	sp->stream.opaque = NULL;
	420	sp->stream.data_type = Z_BINARY;
	421
	422	/*
80ed523f	423	* Override parent get/set field methods.
8414a40c	424	*/
8414a40c VZ	425	sp->vgetparent = tif->tif_tagmethods.vgetfield;
	426	tif->tif_tagmethods.vgetfield = ZIPVGetField; /* hook for codec tags */
	427	sp->vsetparent = tif->tif_tagmethods.vsetfield;
	428	tif->tif_tagmethods.vsetfield = ZIPVSetField; /* hook for codec tags */
	429
	430	/* Default values for codec-specific fields */
	431	sp->zipquality = Z_DEFAULT_COMPRESSION; /* default comp. level */
	432	sp->state = 0;
	433
	434	/*
	435	* Install codec methods.
	436	*/
80ed523f	437	tif->tif_fixuptags = ZIPFixupTags;
8414a40c VZ	438	tif->tif_setupdecode = ZIPSetupDecode;
	439	tif->tif_predecode = ZIPPreDecode;
	440	tif->tif_decoderow = ZIPDecode;
	441	tif->tif_decodestrip = ZIPDecode;
80ed523f	442	tif->tif_decodetile = ZIPDecode;
8414a40c VZ	443	tif->tif_setupencode = ZIPSetupEncode;
	444	tif->tif_preencode = ZIPPreEncode;
	445	tif->tif_postencode = ZIPPostEncode;
	446	tif->tif_encoderow = ZIPEncode;
	447	tif->tif_encodestrip = ZIPEncode;
	448	tif->tif_encodetile = ZIPEncode;
	449	tif->tif_cleanup = ZIPCleanup;
	450	/*
	451	* Setup predictor setup.
	452	*/
	453	(void) TIFFPredictorInit(tif);
	454	return (1);
	455	bad:
80ed523f	456	TIFFErrorExt(tif->tif_clientdata, module,
8414a40c VZ	457	"No space for ZIP state block");
	458	return (0);
	459	}
	460	#endif /* ZIP_SUPORT */
	461
	462	/* vim: set ts=8 sts=8 sw=8 noet: */
80ed523f VZ	463	/*
	464	* Local Variables:
	465	* mode: c
	466	* c-basic-offset: 8
	467	* fill-column: 78
	468	* End:
	469	*/