/* ************************************************************************** */ | |
/* * For conditions of distribution and use, * */ | |
/* * see copyright notice in libmng.h * */ | |
/* ************************************************************************** */ | |
/* * * */ | |
/* * project : libmng * */ | |
/* * file : libmng_zlib.c copyright (c) 2000-2004 G.Juyn * */ | |
/* * version : 1.0.9 * */ | |
/* * * */ | |
/* * purpose : ZLIB library interface (implementation) * */ | |
/* * * */ | |
/* * author : G.Juyn * */ | |
/* * * */ | |
/* * comment : implementation of the ZLIB library interface * */ | |
/* * * */ | |
/* * changes : 0.5.1 - 05/08/2000 - G.Juyn * */ | |
/* * - changed strict-ANSI stuff * */ | |
/* * 0.5.1 - 05/11/2000 - G.Juyn * */ | |
/* * - filled the deflatedata routine * */ | |
/* * 0.5.1 - 05/12/2000 - G.Juyn * */ | |
/* * - changed trace to macro for callback error-reporting * */ | |
/* * * */ | |
/* * 0.5.2 - 05/20/2000 - G.Juyn * */ | |
/* * - fixed for JNG alpha handling * */ | |
/* * 0.5.2 - 05/24/2000 - G.Juyn * */ | |
/* * - moved init of default zlib parms from here to * */ | |
/* * "mng_hlapi.c" * */ | |
/* * * */ | |
/* * 0.5.3 - 06/16/2000 - G.Juyn * */ | |
/* * - changed progressive-display processing * */ | |
/* * * */ | |
/* * 0.9.2 - 08/05/2000 - G.Juyn * */ | |
/* * - changed file-prefixes * */ | |
/* * * */ | |
/* * 0.9.3 - 08/08/2000 - G.Juyn * */ | |
/* * - fixed compiler-warnings from Mozilla * */ | |
/* * 0.9.3 - 09/07/2000 - G.Juyn * */ | |
/* * - added support for new filter_types * */ | |
/* * * */ | |
/* * 1.0.5 - 08/07/2002 - G.Juyn * */ | |
/* * - added test-option for PNG filter method 193 (=no filter) * */ | |
/* * 1.0.5 - 08/19/2002 - G.Juyn * */ | |
/* * - B597134 - libmng pollutes the linker namespace * */ | |
/* * 1.0.5 - 09/19/2002 - G.Juyn * */ | |
/* * - added warning for too much IDAT data * */ | |
/* * * */ | |
/* * 1.0.6 - 07/07/2003 - G.R-P * */ | |
/* * - added MNG_NO_16BIT_SUPPORT support * */ | |
/* * * */ | |
/* * 1.0.9 - 10/09/2004 - G.R-P * */ | |
/* * - added MNG_NO_1_2_4BIT_SUPPORT support * */ | |
/* * * */ | |
/* ************************************************************************** */ | |
#include "libmng.h" | |
#include "libmng_data.h" | |
#include "libmng_error.h" | |
#include "libmng_trace.h" | |
#ifdef __BORLANDC__ | |
#pragma hdrstop | |
#endif | |
#include "libmng_memory.h" | |
#include "libmng_pixels.h" | |
#include "libmng_filter.h" | |
#include "libmng_zlib.h" | |
#if defined(__BORLANDC__) && defined(MNG_STRICT_ANSI) | |
#pragma option -A /* force ANSI-C */ | |
#endif | |
/* ************************************************************************** */ | |
#ifdef MNG_INCLUDE_ZLIB | |
/* ************************************************************************** */ | |
voidpf mngzlib_alloc (voidpf pData, | |
uInt iCount, | |
uInt iSize) | |
{ | |
voidpf pPtr; /* temporary space */ | |
#ifdef MNG_INTERNAL_MEMMNGMT | |
pPtr = calloc (iCount, iSize); /* local allocation */ | |
#else | |
if (((mng_datap)pData)->fMemalloc) /* callback function set ? */ | |
pPtr = ((mng_datap)pData)->fMemalloc (iCount * iSize); | |
else | |
pPtr = Z_NULL; /* can't allocate! */ | |
#endif | |
return pPtr; /* return the result */ | |
} | |
/* ************************************************************************** */ | |
void mngzlib_free (voidpf pData, | |
voidpf pAddress) | |
{ | |
#ifdef MNG_INTERNAL_MEMMNGMT | |
free (pAddress); /* free locally */ | |
#else | |
if (((mng_datap)pData)->fMemfree) /* callback set? */ | |
((mng_datap)pData)->fMemfree (pAddress, 1); | |
#endif | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_initialize (mng_datap pData) | |
{ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INITIALIZE, MNG_LC_START); | |
#endif | |
#ifdef MNG_INTERNAL_MEMMNGMT | |
pData->sZlib.zalloc = Z_NULL; /* let zlib figure out memory management */ | |
pData->sZlib.zfree = Z_NULL; | |
pData->sZlib.opaque = Z_NULL; | |
#else /* use user-provided callbacks */ | |
pData->sZlib.zalloc = mngzlib_alloc; | |
pData->sZlib.zfree = mngzlib_free; | |
pData->sZlib.opaque = (voidpf)pData; | |
#endif | |
pData->bInflating = MNG_FALSE; /* not performing any action yet */ | |
pData->bDeflating = MNG_FALSE; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INITIALIZE, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_cleanup (mng_datap pData) | |
{ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_CLEANUP, MNG_LC_START); | |
#endif | |
if (pData->bInflating) /* force zlib cleanup */ | |
mngzlib_inflatefree (pData); | |
if (pData->bDeflating) | |
mngzlib_deflatefree (pData); | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_CLEANUP, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_inflateinit (mng_datap pData) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEINIT, MNG_LC_START); | |
#endif | |
/* initialize zlib structures and such */ | |
iZrslt = inflateInit (&pData->sZlib); | |
if (iZrslt != Z_OK) /* on error bail out */ | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
pData->bInflating = MNG_TRUE; /* really inflating something now */ | |
pData->sZlib.next_out = 0; /* force JIT initialization */ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEINIT, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
#ifdef MNG_SUPPORT_DISPLAY | |
mng_retcode mngzlib_inflaterows (mng_datap pData, | |
mng_uint32 iInlen, | |
mng_uint8p pIndata) | |
{ | |
int iZrslt; | |
mng_retcode iRslt; | |
mng_ptr pSwap; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEROWS, MNG_LC_START); | |
#endif | |
pData->sZlib.next_in = pIndata; /* let zlib know where to get stuff */ | |
pData->sZlib.avail_in = (uInt)iInlen; | |
if (pData->sZlib.next_out == 0) /* initialize output variables ? */ | |
{ /* let zlib know where to store stuff */ | |
pData->sZlib.next_out = pData->pWorkrow; | |
pData->sZlib.avail_out = (uInt)(pData->iRowsize + pData->iPixelofs); | |
#ifdef MNG_NO_1_2_4BIT_SUPPORT | |
if (pData->iPNGdepth < 8) | |
pData->sZlib.avail_out = (uInt)((pData->iPNGdepth*pData->iRowsize + 7)/8 | |
+ pData->iPixelofs); | |
#endif | |
#ifdef MNG_NO_16BIT_SUPPORT | |
if (pData->iPNGdepth > 8) | |
pData->sZlib.avail_out = (uInt)(2*pData->iRowsize + pData->iPixelofs); | |
#endif | |
} | |
do | |
{ /* now inflate a row */ | |
iZrslt = inflate (&pData->sZlib, Z_SYNC_FLUSH); | |
/* produced a full row ? */ | |
if (((iZrslt == Z_OK) || (iZrslt == Z_STREAM_END)) && | |
(pData->sZlib.avail_out == 0)) | |
{ /* image not completed yet ? */ | |
if (pData->iRow < (mng_int32)pData->iDataheight) | |
{ | |
#ifdef MNG_NO_1_2_4BIT_SUPPORT | |
if (pData->iPNGdepth == 1) | |
{ | |
/* Inflate Workrow to 8-bit */ | |
mng_int32 iX; | |
mng_uint8p pSrc = pData->pWorkrow+1; | |
mng_uint8p pDest = pSrc + pData->iRowsize - (pData->iRowsize+7)/8; | |
for (iX = ((pData->iRowsize+7)/8) ; iX > 0 ; iX--) | |
*pDest++ = *pSrc++; | |
pDest = pData->pWorkrow+1; | |
pSrc = pDest + pData->iRowsize - (pData->iRowsize+7)/8; | |
for (iX = pData->iRowsize; ;) | |
{ | |
*pDest++ = (((*pSrc)>>7)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>6)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>5)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>4)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>3)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>2)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>1)&1); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc) )&1); | |
if (iX-- <= 0) | |
break; | |
pSrc++; | |
} | |
} | |
else if (pData->iPNGdepth == 2) | |
{ | |
/* Inflate Workrow to 8-bit */ | |
mng_int32 iX; | |
mng_uint8p pSrc = pData->pWorkrow+1; | |
mng_uint8p pDest = pSrc + pData->iRowsize - (2*pData->iRowsize+7)/8; | |
for (iX = ((2*pData->iRowsize+7)/8) ; iX > 0 ; iX--) | |
*pDest++ = *pSrc++; | |
pDest = pData->pWorkrow+1; | |
pSrc = pDest + pData->iRowsize - (2*pData->iRowsize+7)/8; | |
for (iX = pData->iRowsize; ;) | |
{ | |
*pDest++ = (((*pSrc)>>6)&3); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>4)&3); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc)>>2)&3); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc) )&3); | |
if (iX-- <= 0) | |
break; | |
pSrc++; | |
} | |
} | |
else if (pData->iPNGdepth == 4) | |
{ | |
/* Inflate Workrow to 8-bit */ | |
mng_int32 iX; | |
mng_uint8p pSrc = pData->pWorkrow+1; | |
mng_uint8p pDest = pSrc + pData->iRowsize - (4*pData->iRowsize+7)/8; | |
for (iX = ((4*pData->iRowsize+7)/8) ; iX > 0 ; iX--) | |
*pDest++ = *pSrc++; | |
pDest = pData->pWorkrow+1; | |
pSrc = pDest + pData->iRowsize - (4*pData->iRowsize+7)/8; | |
for (iX = pData->iRowsize; ;) | |
{ | |
*pDest++ = (((*pSrc)>>4)&0x0f); | |
if (iX-- <= 0) | |
break; | |
*pDest++ = (((*pSrc) )&0x0f); | |
if (iX-- <= 0) | |
break; | |
pSrc++; | |
} | |
} | |
if (pData->iPNGdepth < 8 && pData->iColortype == 0) | |
{ | |
/* Expand samples to 8-bit by LBR */ | |
mng_int32 iX; | |
mng_uint8p pSrc = pData->pWorkrow+1; | |
mng_uint8 multiplier[]={0,255,85,0,17,0,0,0,1}; | |
for (iX = pData->iRowsize; iX > 0; iX--) | |
*pSrc++ *= multiplier[pData->iPNGdepth]; | |
} | |
#endif | |
#ifdef MNG_NO_16BIT_SUPPORT | |
if (pData->iPNGdepth > 8) | |
{ | |
/* Reduce Workrow to 8-bit */ | |
mng_int32 iX; | |
mng_uint8p pSrc = pData->pWorkrow+1; | |
mng_uint8p pDest = pSrc; | |
for (iX = pData->iRowsize; iX > 0; iX--) | |
{ | |
*pDest = *pSrc; | |
pDest++; | |
pSrc+=2; | |
} | |
} | |
#endif | |
#ifdef FILTER192 /* has leveling info ? */ | |
if (pData->iFilterofs == MNG_FILTER_DIFFERING) | |
iRslt = init_rowdiffering (pData); | |
else | |
#endif | |
iRslt = MNG_NOERROR; | |
/* filter the row if necessary */ | |
if ((!iRslt) && (pData->iFilterofs < pData->iPixelofs ) && | |
(*(pData->pWorkrow + pData->iFilterofs)) ) | |
iRslt = mng_filter_a_row (pData); | |
else | |
iRslt = MNG_NOERROR; | |
/* additional leveling/differing ? */ | |
if ((!iRslt) && (pData->fDifferrow)) | |
{ | |
iRslt = ((mng_differrow)pData->fDifferrow) (pData); | |
pSwap = pData->pWorkrow; | |
pData->pWorkrow = pData->pPrevrow; | |
pData->pPrevrow = pSwap; /* make sure we're processing the right data */ | |
} | |
if (!iRslt) | |
{ | |
#ifdef MNG_INCLUDE_JNG | |
if (pData->bHasJHDR) /* is JNG alpha-channel ? */ | |
{ /* just store in object ? */ | |
if ((!iRslt) && (pData->fStorerow)) | |
iRslt = ((mng_storerow)pData->fStorerow) (pData); | |
} | |
else | |
#endif /* MNG_INCLUDE_JNG */ | |
{ /* process this row */ | |
if ((!iRslt) && (pData->fProcessrow)) | |
iRslt = ((mng_processrow)pData->fProcessrow) (pData); | |
/* store in object ? */ | |
if ((!iRslt) && (pData->fStorerow)) | |
iRslt = ((mng_storerow)pData->fStorerow) (pData); | |
/* color correction ? */ | |
if ((!iRslt) && (pData->fCorrectrow)) | |
iRslt = ((mng_correctrow)pData->fCorrectrow) (pData); | |
/* slap onto canvas ? */ | |
if ((!iRslt) && (pData->fDisplayrow)) | |
{ | |
iRslt = ((mng_displayrow)pData->fDisplayrow) (pData); | |
if (!iRslt) /* check progressive display refresh */ | |
iRslt = mng_display_progressive_check (pData); | |
} | |
} | |
} | |
if (iRslt) /* on error bail out */ | |
MNG_ERROR (pData, iRslt); | |
if (!pData->fDifferrow) /* swap row-pointers */ | |
{ | |
pSwap = pData->pWorkrow; | |
pData->pWorkrow = pData->pPrevrow; | |
pData->pPrevrow = pSwap; /* so prev points to the processed row! */ | |
} | |
iRslt = mng_next_row (pData); /* adjust variables for next row */ | |
if (iRslt) /* on error bail out */ | |
MNG_ERROR (pData, iRslt); | |
} | |
/* let zlib know where to store next output */ | |
pData->sZlib.next_out = pData->pWorkrow; | |
pData->sZlib.avail_out = (uInt)(pData->iRowsize + pData->iPixelofs); | |
#ifdef MNG_NO_1_2_4BIT_SUPPORT | |
if (pData->iPNGdepth < 8) | |
pData->sZlib.avail_out = (uInt)((pData->iPNGdepth*pData->iRowsize + 7)/8 | |
+ pData->iPixelofs); | |
#endif | |
#ifdef MNG_NO_16BIT_SUPPORT | |
if (pData->iPNGdepth > 8) | |
pData->sZlib.avail_out = (uInt)(2*pData->iRowsize + pData->iPixelofs); | |
#endif | |
} | |
} /* until some error or EOI | |
or all pixels received */ | |
while ( (iZrslt == Z_OK) && (pData->sZlib.avail_in > 0) && | |
( (pData->iRow < (mng_int32)pData->iDataheight) || | |
( (pData->iPass >= 0) && (pData->iPass < 7) ) ) ); | |
/* on error bail out */ | |
if ((iZrslt != Z_OK) && (iZrslt != Z_STREAM_END)) | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
/* too much data ? */ | |
if ((iZrslt == Z_OK) && (pData->sZlib.avail_in > 0)) | |
MNG_WARNING (pData, MNG_TOOMUCHIDAT); | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEROWS, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; | |
} | |
#endif /* MNG_SUPPORT_DISPLAY */ | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_inflatedata (mng_datap pData, | |
mng_uint32 iInlen, | |
mng_uint8p pIndata) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEDATA, MNG_LC_START); | |
#endif | |
/* let zlib know where to get stuff */ | |
pData->sZlib.next_in = pIndata; | |
pData->sZlib.avail_in = (uInt)iInlen; | |
/* now inflate the data in one go! */ | |
iZrslt = inflate (&pData->sZlib, Z_FINISH); | |
/* not enough room in output-buffer ? */ | |
if ((iZrslt == Z_BUF_ERROR) || (pData->sZlib.avail_in > 0)) | |
return MNG_BUFOVERFLOW; | |
/* on error bail out */ | |
if ((iZrslt != Z_OK) && (iZrslt != Z_STREAM_END)) | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEDATA, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_inflatefree (mng_datap pData) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEFREE, MNG_LC_START); | |
#endif | |
pData->bInflating = MNG_FALSE; /* stopped it */ | |
iZrslt = inflateEnd (&pData->sZlib); /* let zlib cleanup its own stuff */ | |
if (iZrslt != Z_OK) /* on error bail out */ | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_INFLATEFREE, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_deflateinit (mng_datap pData) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEINIT, MNG_LC_START); | |
#endif | |
/* initialize zlib structures and such */ | |
iZrslt = deflateInit2 (&pData->sZlib, pData->iZlevel, pData->iZmethod, | |
pData->iZwindowbits, pData->iZmemlevel, | |
pData->iZstrategy); | |
if (iZrslt != Z_OK) /* on error bail out */ | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
pData->bDeflating = MNG_TRUE; /* really deflating something now */ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEINIT, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_deflaterows (mng_datap pData, | |
mng_uint32 iInlen, | |
mng_uint8p pIndata) | |
{ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEROWS, MNG_LC_START); | |
#endif | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEROWS, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_deflatedata (mng_datap pData, | |
mng_uint32 iInlen, | |
mng_uint8p pIndata) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEDATA, MNG_LC_START); | |
#endif | |
pData->sZlib.next_in = pIndata; /* let zlib know where to get stuff */ | |
pData->sZlib.avail_in = (uInt)iInlen; | |
/* now deflate the data in one go! */ | |
iZrslt = deflate (&pData->sZlib, Z_FINISH); | |
/* not enough room in output-buffer ? */ | |
if ((iZrslt == Z_BUF_ERROR) || (pData->sZlib.avail_in > 0)) | |
return MNG_BUFOVERFLOW; | |
/* on error bail out */ | |
if ((iZrslt != Z_OK) && (iZrslt != Z_STREAM_END)) | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEDATA, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; | |
} | |
/* ************************************************************************** */ | |
mng_retcode mngzlib_deflatefree (mng_datap pData) | |
{ | |
int iZrslt; | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEFREE, MNG_LC_START); | |
#endif | |
iZrslt = deflateEnd (&pData->sZlib); /* let zlib cleanup its own stuff */ | |
if (iZrslt != Z_OK) /* on error bail out */ | |
MNG_ERRORZ (pData, (mng_uint32)iZrslt); | |
pData->bDeflating = MNG_FALSE; /* stopped it */ | |
#ifdef MNG_SUPPORT_TRACE | |
MNG_TRACE (pData, MNG_FN_ZLIB_DEFLATEFREE, MNG_LC_END); | |
#endif | |
return MNG_NOERROR; /* done */ | |
} | |
/* ************************************************************************** */ | |
#endif /* MNG_INCLUDE_ZLIB */ | |
/* ************************************************************************** */ | |
/* * end of file * */ | |
/* ************************************************************************** */ | |