2015-10-22 14:31:46 +00:00
|
|
|
/*
|
|
|
|
ZSTD HC - High Compression Mode of Zstandard
|
|
|
|
Copyright (C) 2015, Yann Collet.
|
|
|
|
|
|
|
|
BSD 2-Clause License (http://www.opensource.org/licenses/bsd-license.php)
|
|
|
|
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
|
|
modification, are permitted provided that the following conditions are
|
|
|
|
met:
|
|
|
|
|
|
|
|
* Redistributions of source code must retain the above copyright
|
|
|
|
notice, this list of conditions and the following disclaimer.
|
|
|
|
* Redistributions in binary form must reproduce the above
|
|
|
|
copyright notice, this list of conditions and the following disclaimer
|
|
|
|
in the documentation and/or other materials provided with the
|
|
|
|
distribution.
|
|
|
|
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
|
|
OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
|
|
You can contact the author at :
|
|
|
|
- Zstd source repository : https://www.zstd.net
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
2015-10-30 14:49:48 +00:00
|
|
|
/* *******************************************************
|
|
|
|
* Compiler specifics
|
|
|
|
*********************************************************/
|
|
|
|
#ifdef _MSC_VER /* Visual Studio */
|
|
|
|
# define FORCE_INLINE static __forceinline
|
|
|
|
# include <intrin.h> /* For Visual 2005 */
|
|
|
|
# pragma warning(disable : 4127) /* disable: C4127: conditional expression is constant */
|
|
|
|
# pragma warning(disable : 4324) /* disable: C4324: padded structure */
|
|
|
|
#else
|
|
|
|
# define GCC_VERSION (__GNUC__ * 100 + __GNUC_MINOR__)
|
|
|
|
# ifdef __GNUC__
|
|
|
|
# define FORCE_INLINE static inline __attribute__((always_inline))
|
|
|
|
# else
|
|
|
|
# define FORCE_INLINE static inline
|
|
|
|
# endif
|
|
|
|
#endif
|
|
|
|
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
/* *************************************
|
|
|
|
* Includes
|
|
|
|
***************************************/
|
|
|
|
#include <stdlib.h> /* malloc */
|
|
|
|
#include <string.h> /* memset */
|
2015-11-11 20:38:21 +00:00
|
|
|
#include "mem.h"
|
|
|
|
#include "fse_static.h"
|
|
|
|
#include "huff0.h"
|
2015-10-22 14:31:46 +00:00
|
|
|
#include "zstd_static.h"
|
2015-10-29 16:15:14 +00:00
|
|
|
#include "zstd_internal.h"
|
2015-10-22 14:31:46 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* *************************************
|
2015-11-11 20:38:21 +00:00
|
|
|
* Constants
|
2015-10-22 14:31:46 +00:00
|
|
|
***************************************/
|
2015-11-11 20:38:21 +00:00
|
|
|
static const U32 g_searchStrength = 8;
|
2015-10-22 14:31:46 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* *************************************
|
2015-11-11 20:38:21 +00:00
|
|
|
* Sequence storage
|
|
|
|
***************************************/
|
|
|
|
typedef struct {
|
|
|
|
void* buffer;
|
|
|
|
U32* offsetStart;
|
|
|
|
U32* offset;
|
|
|
|
BYTE* offCodeStart;
|
|
|
|
BYTE* offCode;
|
|
|
|
BYTE* litStart;
|
|
|
|
BYTE* lit;
|
|
|
|
BYTE* litLengthStart;
|
|
|
|
BYTE* litLength;
|
|
|
|
BYTE* matchLengthStart;
|
|
|
|
BYTE* matchLength;
|
|
|
|
BYTE* dumpsStart;
|
|
|
|
BYTE* dumps;
|
|
|
|
} seqStore_t;
|
|
|
|
|
|
|
|
static void ZSTD_resetSeqStore(seqStore_t* ssPtr)
|
|
|
|
{
|
|
|
|
ssPtr->offset = ssPtr->offsetStart;
|
|
|
|
ssPtr->lit = ssPtr->litStart;
|
|
|
|
ssPtr->litLength = ssPtr->litLengthStart;
|
|
|
|
ssPtr->matchLength = ssPtr->matchLengthStart;
|
|
|
|
ssPtr->dumps = ssPtr->dumpsStart;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* *************************************
|
|
|
|
* Context memory management
|
2015-10-22 14:31:46 +00:00
|
|
|
***************************************/
|
|
|
|
#define WORKPLACESIZE (BLOCKSIZE*3)
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
struct ZSTD_CCtx_s
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
const BYTE* nextSrc; /* next block here to continue on current prefix */
|
2015-10-22 15:55:40 +00:00
|
|
|
const BYTE* base; /* All regular indexes relative to this position */
|
|
|
|
const BYTE* dictBase; /* extDict indexes relative to this position */
|
2015-10-22 14:31:46 +00:00
|
|
|
U32 dictLimit; /* below that point, need extDict */
|
2015-10-22 15:55:40 +00:00
|
|
|
U32 lowLimit; /* below that point, no more data */
|
2015-10-22 14:31:46 +00:00
|
|
|
U32 nextToUpdate; /* index from which to continue dictionary update */
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_parameters params;
|
2015-10-29 17:41:45 +00:00
|
|
|
void* workSpace;
|
|
|
|
size_t workSpaceSize;
|
|
|
|
|
|
|
|
seqStore_t seqStore; /* sequences storage ptrs */
|
2015-10-25 13:06:35 +00:00
|
|
|
U32* hashTable;
|
2015-11-05 16:32:18 +00:00
|
|
|
U32* contentTable;
|
2015-10-22 14:31:46 +00:00
|
|
|
};
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_CCtx* ZSTD_createCCtx(void)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
return (ZSTD_CCtx*) calloc(1, sizeof(ZSTD_CCtx));
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_freeCCtx(ZSTD_CCtx* cctx)
|
2015-10-25 13:06:35 +00:00
|
|
|
{
|
2015-10-29 17:41:45 +00:00
|
|
|
free(cctx->workSpace);
|
2015-10-25 13:06:35 +00:00
|
|
|
free(cctx);
|
|
|
|
return 0;
|
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
|
2015-11-11 20:38:21 +00:00
|
|
|
static unsigned ZSTD_highbit(U32 val);
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
/** ZSTD_validateParams
|
2015-11-04 11:05:27 +00:00
|
|
|
correct params value to remain within authorized range
|
|
|
|
optimize for srcSize if srcSize > 0 */
|
2015-11-11 12:43:58 +00:00
|
|
|
void ZSTD_validateParams(ZSTD_parameters* params, U64 srcSizeHint)
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
const U32 btPlus = (params->strategy == ZSTD_btlazy2);
|
2015-11-04 11:05:27 +00:00
|
|
|
|
|
|
|
/* validate params */
|
2015-11-11 12:43:58 +00:00
|
|
|
if (params->windowLog > ZSTD_WINDOWLOG_MAX) params->windowLog = ZSTD_WINDOWLOG_MAX;
|
|
|
|
if (params->windowLog < ZSTD_WINDOWLOG_MIN) params->windowLog = ZSTD_WINDOWLOG_MIN;
|
2015-11-04 11:05:27 +00:00
|
|
|
|
|
|
|
/* correct params, to use less memory */
|
2015-11-11 12:43:58 +00:00
|
|
|
if ((srcSizeHint > 0) && (srcSizeHint < (1<<ZSTD_WINDOWLOG_MAX)))
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-09 16:42:17 +00:00
|
|
|
U32 srcLog = ZSTD_highbit((U32)srcSizeHint-1) + 1;
|
2015-11-04 11:05:27 +00:00
|
|
|
if (params->windowLog > srcLog) params->windowLog = srcLog;
|
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
if (params->contentLog > params->windowLog+btPlus) params->contentLog = params->windowLog+btPlus; /* <= ZSTD_CONTENTLOG_MAX */
|
|
|
|
if (params->contentLog < ZSTD_CONTENTLOG_MIN) params->contentLog = ZSTD_CONTENTLOG_MIN;
|
|
|
|
if (params->hashLog > ZSTD_HASHLOG_MAX) params->hashLog = ZSTD_HASHLOG_MAX;
|
|
|
|
if (params->hashLog < ZSTD_HASHLOG_MIN) params->hashLog = ZSTD_HASHLOG_MIN;
|
|
|
|
if (params->searchLog > ZSTD_SEARCHLOG_MAX) params->searchLog = ZSTD_SEARCHLOG_MAX;
|
|
|
|
if (params->searchLog < ZSTD_SEARCHLOG_MIN) params->searchLog = ZSTD_SEARCHLOG_MIN;
|
|
|
|
if (params->searchLength> ZSTD_SEARCHLENGTH_MAX) params->searchLength = ZSTD_SEARCHLENGTH_MAX;
|
|
|
|
if (params->searchLength< ZSTD_SEARCHLENGTH_MIN) params->searchLength = ZSTD_SEARCHLENGTH_MIN;
|
|
|
|
if ((U32)params->strategy>(U32)ZSTD_btlazy2) params->strategy = ZSTD_btlazy2;
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_resetCCtx_advanced (ZSTD_CCtx* zc,
|
2015-11-13 10:27:46 +00:00
|
|
|
ZSTD_parameters params,
|
|
|
|
U64 srcSizeHint)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_validateParams(¶ms, srcSizeHint);
|
2015-10-29 15:49:43 +00:00
|
|
|
|
|
|
|
/* reserve table memory */
|
2015-10-25 13:06:35 +00:00
|
|
|
{
|
2015-11-20 11:03:53 +00:00
|
|
|
const U32 contentLog = (params.strategy == ZSTD_fast) ? 1 : params.contentLog;
|
2015-11-05 16:32:18 +00:00
|
|
|
const size_t tableSpace = ((1 << contentLog) + (1 << params.hashLog)) * sizeof(U32);
|
2015-10-29 17:41:45 +00:00
|
|
|
const size_t neededSpace = tableSpace + WORKPLACESIZE;
|
|
|
|
if (zc->workSpaceSize < neededSpace)
|
2015-10-29 15:49:43 +00:00
|
|
|
{
|
2015-10-29 17:41:45 +00:00
|
|
|
free(zc->workSpace);
|
|
|
|
zc->workSpaceSize = neededSpace;
|
|
|
|
zc->workSpace = malloc(neededSpace);
|
2015-10-30 05:40:22 +00:00
|
|
|
if (zc->workSpace == NULL) return ERROR(memory_allocation);
|
2015-10-29 15:49:43 +00:00
|
|
|
}
|
2015-11-06 09:52:17 +00:00
|
|
|
memset(zc->workSpace, 0, tableSpace );
|
|
|
|
zc->hashTable = (U32*)(zc->workSpace);
|
2015-11-05 16:32:18 +00:00
|
|
|
zc->contentTable = zc->hashTable + ((size_t)1 << params.hashLog);
|
|
|
|
zc->seqStore.buffer = (void*) (zc->contentTable + ((size_t)1 << contentLog));
|
2015-10-25 13:06:35 +00:00
|
|
|
}
|
|
|
|
|
2015-11-07 00:13:31 +00:00
|
|
|
zc->nextToUpdate = 1;
|
2015-11-13 10:27:46 +00:00
|
|
|
zc->nextSrc = NULL;
|
2015-10-29 15:49:43 +00:00
|
|
|
zc->base = NULL;
|
|
|
|
zc->dictBase = NULL;
|
|
|
|
zc->dictLimit = 0;
|
|
|
|
zc->lowLimit = 0;
|
2015-10-25 13:06:35 +00:00
|
|
|
zc->params = params;
|
2015-10-22 14:31:46 +00:00
|
|
|
zc->seqStore.offsetStart = (U32*) (zc->seqStore.buffer);
|
|
|
|
zc->seqStore.offCodeStart = (BYTE*) (zc->seqStore.offsetStart + (BLOCKSIZE>>2));
|
|
|
|
zc->seqStore.litStart = zc->seqStore.offCodeStart + (BLOCKSIZE>>2);
|
|
|
|
zc->seqStore.litLengthStart = zc->seqStore.litStart + BLOCKSIZE;
|
|
|
|
zc->seqStore.matchLengthStart = zc->seqStore.litLengthStart + (BLOCKSIZE>>2);
|
|
|
|
zc->seqStore.dumpsStart = zc->seqStore.matchLengthStart + (BLOCKSIZE>>2);
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2015-10-30 05:40:22 +00:00
|
|
|
return 0;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-10-25 13:06:35 +00:00
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
static void ZSTD_reduceIndex (ZSTD_CCtx* zc,
|
|
|
|
const U32 reducerValue)
|
|
|
|
{
|
|
|
|
const U32 contentLog = zc->params.strategy == ZSTD_fast ? 1 : zc->params.contentLog;
|
|
|
|
const U32 tableSpaceU32 = (1 << contentLog) + (1 << zc->params.hashLog);
|
|
|
|
U32* table32 = zc->hashTable;
|
|
|
|
U32 index;
|
|
|
|
|
|
|
|
for (index=0 ; index < tableSpaceU32 ; index++)
|
|
|
|
{
|
|
|
|
if (table32[index] < reducerValue) table32[index] = 0;
|
|
|
|
else table32[index] -= reducerValue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 20:38:21 +00:00
|
|
|
/* *******************************************************
|
|
|
|
* Block entropic compression
|
|
|
|
*********************************************************/
|
|
|
|
size_t ZSTD_compressBound(size_t srcSize) /* maximum compressed size */
|
|
|
|
{
|
|
|
|
return FSE_compressBound(srcSize) + 12;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
size_t ZSTD_noCompressBlock (void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
|
|
|
|
|
|
|
if (srcSize + ZSTD_blockHeaderSize > maxDstSize) return ERROR(dstSize_tooSmall);
|
|
|
|
memcpy(ostart + ZSTD_blockHeaderSize, src, srcSize);
|
|
|
|
|
|
|
|
/* Build header */
|
|
|
|
ostart[0] = (BYTE)(srcSize>>16);
|
|
|
|
ostart[1] = (BYTE)(srcSize>>8);
|
|
|
|
ostart[2] = (BYTE) srcSize;
|
|
|
|
ostart[0] += (BYTE)(bt_raw<<6); /* is a raw (uncompressed) block */
|
|
|
|
|
|
|
|
return ZSTD_blockHeaderSize+srcSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
static size_t ZSTD_noCompressLiterals (void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
|
|
|
|
|
|
|
if (srcSize + 3 > maxDstSize) return ERROR(dstSize_tooSmall);
|
|
|
|
|
|
|
|
MEM_writeLE32(dst, ((U32)srcSize << 2) | IS_RAW);
|
|
|
|
memcpy(ostart + 3, src, srcSize);
|
|
|
|
return srcSize + 3;
|
|
|
|
}
|
|
|
|
|
|
|
|
static size_t ZSTD_compressRleLiteralsBlock (void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
|
|
|
|
|
|
|
(void)maxDstSize;
|
|
|
|
MEM_writeLE32(dst, ((U32)srcSize << 2) | IS_RLE); /* note : maxDstSize > litHeaderSize > 4 */
|
|
|
|
ostart[3] = *(const BYTE*)src;
|
|
|
|
return 4;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t ZSTD_minGain(size_t srcSize) { return (srcSize >> 6) + 1; }
|
|
|
|
|
|
|
|
static size_t ZSTD_compressLiterals (void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
const size_t minGain = ZSTD_minGain(srcSize);
|
|
|
|
BYTE* const ostart = (BYTE*)dst;
|
|
|
|
size_t hsize;
|
|
|
|
static const size_t litHeaderSize = 5;
|
|
|
|
|
|
|
|
if (maxDstSize < litHeaderSize+1) return ERROR(dstSize_tooSmall); /* not enough space for compression */
|
|
|
|
|
|
|
|
hsize = HUF_compress(ostart+litHeaderSize, maxDstSize-litHeaderSize, src, srcSize);
|
|
|
|
|
|
|
|
if ((hsize==0) || (hsize >= srcSize - minGain)) return ZSTD_noCompressLiterals(dst, maxDstSize, src, srcSize);
|
|
|
|
if (hsize==1) return ZSTD_compressRleLiteralsBlock(dst, maxDstSize, src, srcSize);
|
|
|
|
|
|
|
|
/* Build header */
|
|
|
|
{
|
|
|
|
ostart[0] = (BYTE)(srcSize << 2); /* is a block, is compressed */
|
|
|
|
ostart[1] = (BYTE)(srcSize >> 6);
|
|
|
|
ostart[2] = (BYTE)(srcSize >>14);
|
|
|
|
ostart[2] += (BYTE)(hsize << 5);
|
|
|
|
ostart[3] = (BYTE)(hsize >> 3);
|
|
|
|
ostart[4] = (BYTE)(hsize >>11);
|
|
|
|
}
|
|
|
|
|
|
|
|
return hsize+litHeaderSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
#define LITERAL_NOENTROPY 63 /* cheap heuristic */
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t ZSTD_compressSequences(void* dst, size_t maxDstSize,
|
2015-11-11 20:38:21 +00:00
|
|
|
const seqStore_t* seqStorePtr,
|
|
|
|
size_t srcSize)
|
|
|
|
{
|
|
|
|
U32 count[MaxSeq+1];
|
|
|
|
S16 norm[MaxSeq+1];
|
|
|
|
size_t mostFrequent;
|
|
|
|
U32 max = 255;
|
|
|
|
U32 tableLog = 11;
|
|
|
|
U32 CTable_LitLength [FSE_CTABLE_SIZE_U32(LLFSELog, MaxLL )];
|
|
|
|
U32 CTable_OffsetBits [FSE_CTABLE_SIZE_U32(OffFSELog,MaxOff)];
|
|
|
|
U32 CTable_MatchLength[FSE_CTABLE_SIZE_U32(MLFSELog, MaxML )];
|
|
|
|
U32 LLtype, Offtype, MLtype; /* compressed, raw or rle */
|
|
|
|
const BYTE* const op_lit_start = seqStorePtr->litStart;
|
|
|
|
const BYTE* const llTable = seqStorePtr->litLengthStart;
|
|
|
|
const BYTE* const llPtr = seqStorePtr->litLength;
|
|
|
|
const BYTE* const mlTable = seqStorePtr->matchLengthStart;
|
|
|
|
const U32* const offsetTable = seqStorePtr->offsetStart;
|
|
|
|
BYTE* const offCodeTable = seqStorePtr->offCodeStart;
|
2015-11-23 12:34:21 +00:00
|
|
|
BYTE* const ostart = (BYTE*)dst;
|
|
|
|
BYTE* op = ostart;
|
|
|
|
BYTE* const oend = ostart + maxDstSize;
|
2015-11-11 20:38:21 +00:00
|
|
|
const size_t nbSeq = llPtr - llTable;
|
|
|
|
const size_t minGain = ZSTD_minGain(srcSize);
|
|
|
|
const size_t maxCSize = srcSize - minGain;
|
|
|
|
BYTE* seqHead;
|
|
|
|
|
|
|
|
|
|
|
|
/* Compress literals */
|
|
|
|
{
|
|
|
|
size_t cSize;
|
|
|
|
size_t litSize = seqStorePtr->lit - op_lit_start;
|
|
|
|
|
|
|
|
if (litSize <= LITERAL_NOENTROPY)
|
|
|
|
cSize = ZSTD_noCompressLiterals(op, maxDstSize, op_lit_start, litSize);
|
|
|
|
else
|
|
|
|
cSize = ZSTD_compressLiterals(op, maxDstSize, op_lit_start, litSize);
|
|
|
|
if (ZSTD_isError(cSize)) return cSize;
|
|
|
|
op += cSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Sequences Header */
|
|
|
|
if ((oend-op) < MIN_SEQUENCES_SIZE)
|
|
|
|
return ERROR(dstSize_tooSmall);
|
|
|
|
MEM_writeLE16(op, (U16)nbSeq); op+=2;
|
|
|
|
seqHead = op;
|
|
|
|
|
|
|
|
/* dumps : contains too large lengths */
|
|
|
|
{
|
|
|
|
size_t dumpsLength = seqStorePtr->dumps - seqStorePtr->dumpsStart;
|
|
|
|
if (dumpsLength < 512)
|
|
|
|
{
|
|
|
|
op[0] = (BYTE)(dumpsLength >> 8);
|
|
|
|
op[1] = (BYTE)(dumpsLength);
|
|
|
|
op += 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
op[0] = 2;
|
|
|
|
op[1] = (BYTE)(dumpsLength>>8);
|
|
|
|
op[2] = (BYTE)(dumpsLength);
|
|
|
|
op += 3;
|
|
|
|
}
|
|
|
|
if ((size_t)(oend-op) < dumpsLength+6) return ERROR(dstSize_tooSmall);
|
|
|
|
memcpy(op, seqStorePtr->dumpsStart, dumpsLength);
|
|
|
|
op += dumpsLength;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* CTable for Literal Lengths */
|
|
|
|
max = MaxLL;
|
|
|
|
mostFrequent = FSE_countFast(count, &max, seqStorePtr->litLengthStart, nbSeq);
|
|
|
|
if ((mostFrequent == nbSeq) && (nbSeq > 2))
|
|
|
|
{
|
|
|
|
*op++ = *(seqStorePtr->litLengthStart);
|
|
|
|
FSE_buildCTable_rle(CTable_LitLength, (BYTE)max);
|
|
|
|
LLtype = bt_rle;
|
|
|
|
}
|
|
|
|
else if ((nbSeq < 64) || (mostFrequent < (nbSeq >> (LLbits-1))))
|
|
|
|
{
|
|
|
|
FSE_buildCTable_raw(CTable_LitLength, LLbits);
|
|
|
|
LLtype = bt_raw;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
size_t NCountSize;
|
|
|
|
tableLog = FSE_optimalTableLog(LLFSELog, nbSeq, max);
|
|
|
|
FSE_normalizeCount(norm, tableLog, count, nbSeq, max);
|
|
|
|
NCountSize = FSE_writeNCount(op, oend-op, norm, max, tableLog); /* overflow protected */
|
|
|
|
if (FSE_isError(NCountSize)) return ERROR(GENERIC);
|
|
|
|
op += NCountSize;
|
|
|
|
FSE_buildCTable(CTable_LitLength, norm, max, tableLog);
|
|
|
|
LLtype = bt_compressed;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* CTable for Offsets codes */
|
|
|
|
{
|
|
|
|
/* create Offset codes */
|
|
|
|
size_t i;
|
|
|
|
max = MaxOff;
|
|
|
|
for (i=0; i<nbSeq; i++)
|
|
|
|
{
|
|
|
|
offCodeTable[i] = (BYTE)ZSTD_highbit(offsetTable[i]) + 1;
|
|
|
|
if (offsetTable[i]==0) offCodeTable[i]=0;
|
|
|
|
}
|
|
|
|
mostFrequent = FSE_countFast(count, &max, offCodeTable, nbSeq);
|
|
|
|
}
|
|
|
|
if ((mostFrequent == nbSeq) && (nbSeq > 2))
|
|
|
|
{
|
|
|
|
*op++ = *offCodeTable;
|
|
|
|
FSE_buildCTable_rle(CTable_OffsetBits, (BYTE)max);
|
|
|
|
Offtype = bt_rle;
|
|
|
|
}
|
|
|
|
else if ((nbSeq < 64) || (mostFrequent < (nbSeq >> (Offbits-1))))
|
|
|
|
{
|
|
|
|
FSE_buildCTable_raw(CTable_OffsetBits, Offbits);
|
|
|
|
Offtype = bt_raw;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
size_t NCountSize;
|
|
|
|
tableLog = FSE_optimalTableLog(OffFSELog, nbSeq, max);
|
|
|
|
FSE_normalizeCount(norm, tableLog, count, nbSeq, max);
|
|
|
|
NCountSize = FSE_writeNCount(op, oend-op, norm, max, tableLog); /* overflow protected */
|
|
|
|
if (FSE_isError(NCountSize)) return ERROR(GENERIC);
|
|
|
|
op += NCountSize;
|
|
|
|
FSE_buildCTable(CTable_OffsetBits, norm, max, tableLog);
|
|
|
|
Offtype = bt_compressed;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* CTable for MatchLengths */
|
|
|
|
max = MaxML;
|
|
|
|
mostFrequent = FSE_countFast(count, &max, seqStorePtr->matchLengthStart, nbSeq);
|
|
|
|
if ((mostFrequent == nbSeq) && (nbSeq > 2))
|
|
|
|
{
|
|
|
|
*op++ = *seqStorePtr->matchLengthStart;
|
|
|
|
FSE_buildCTable_rle(CTable_MatchLength, (BYTE)max);
|
|
|
|
MLtype = bt_rle;
|
|
|
|
}
|
|
|
|
else if ((nbSeq < 64) || (mostFrequent < (nbSeq >> (MLbits-1))))
|
|
|
|
{
|
|
|
|
FSE_buildCTable_raw(CTable_MatchLength, MLbits);
|
|
|
|
MLtype = bt_raw;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
size_t NCountSize;
|
|
|
|
tableLog = FSE_optimalTableLog(MLFSELog, nbSeq, max);
|
|
|
|
FSE_normalizeCount(norm, tableLog, count, nbSeq, max);
|
|
|
|
NCountSize = FSE_writeNCount(op, oend-op, norm, max, tableLog); /* overflow protected */
|
|
|
|
if (FSE_isError(NCountSize)) return ERROR(GENERIC);
|
|
|
|
op += NCountSize;
|
|
|
|
FSE_buildCTable(CTable_MatchLength, norm, max, tableLog);
|
|
|
|
MLtype = bt_compressed;
|
|
|
|
}
|
|
|
|
|
|
|
|
seqHead[0] += (BYTE)((LLtype<<6) + (Offtype<<4) + (MLtype<<2));
|
|
|
|
|
|
|
|
/* Encoding Sequences */
|
|
|
|
{
|
|
|
|
size_t streamSize, errorCode;
|
|
|
|
BIT_CStream_t blockStream;
|
|
|
|
FSE_CState_t stateMatchLength;
|
|
|
|
FSE_CState_t stateOffsetBits;
|
|
|
|
FSE_CState_t stateLitLength;
|
|
|
|
int i;
|
|
|
|
|
|
|
|
errorCode = BIT_initCStream(&blockStream, op, oend-op);
|
|
|
|
if (ERR_isError(errorCode)) return ERROR(dstSize_tooSmall); /* not enough space remaining */
|
|
|
|
FSE_initCState(&stateMatchLength, CTable_MatchLength);
|
|
|
|
FSE_initCState(&stateOffsetBits, CTable_OffsetBits);
|
|
|
|
FSE_initCState(&stateLitLength, CTable_LitLength);
|
|
|
|
|
|
|
|
for (i=(int)nbSeq-1; i>=0; i--)
|
|
|
|
{
|
|
|
|
BYTE matchLength = mlTable[i];
|
|
|
|
U32 offset = offsetTable[i];
|
|
|
|
BYTE offCode = offCodeTable[i]; /* 32b*/ /* 64b*/
|
|
|
|
U32 nbBits = (offCode-1) * (!!offCode);
|
|
|
|
BYTE litLength = llTable[i]; /* (7)*/ /* (7)*/
|
|
|
|
FSE_encodeSymbol(&blockStream, &stateMatchLength, matchLength); /* 17 */ /* 17 */
|
2015-11-20 11:03:53 +00:00
|
|
|
if (MEM_32bits()) BIT_flushBits(&blockStream); /* 7 */
|
2015-11-11 20:38:21 +00:00
|
|
|
BIT_addBits(&blockStream, offset, nbBits); /* 32 */ /* 42 */
|
2015-11-20 11:03:53 +00:00
|
|
|
if (MEM_32bits()) BIT_flushBits(&blockStream); /* 7 */
|
2015-11-11 20:38:21 +00:00
|
|
|
FSE_encodeSymbol(&blockStream, &stateOffsetBits, offCode); /* 16 */ /* 51 */
|
|
|
|
FSE_encodeSymbol(&blockStream, &stateLitLength, litLength); /* 26 */ /* 61 */
|
|
|
|
BIT_flushBits(&blockStream); /* 7 */ /* 7 */
|
|
|
|
}
|
|
|
|
|
|
|
|
FSE_flushCState(&blockStream, &stateMatchLength);
|
|
|
|
FSE_flushCState(&blockStream, &stateOffsetBits);
|
|
|
|
FSE_flushCState(&blockStream, &stateLitLength);
|
|
|
|
|
|
|
|
streamSize = BIT_closeCStream(&blockStream);
|
|
|
|
if (streamSize==0) return ERROR(dstSize_tooSmall); /* not enough space */
|
|
|
|
op += streamSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* check compressibility */
|
2015-11-23 12:34:21 +00:00
|
|
|
if ((size_t)(op-ostart) >= maxCSize) return 0;
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
return op - ostart;
|
2015-11-11 20:38:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** ZSTD_storeSeq
|
|
|
|
Store a sequence (literal length, literals, offset code and match length) into seqStore_t
|
|
|
|
@offsetCode : distance to match, or 0 == repCode
|
|
|
|
@matchCode : matchLength - MINMATCH
|
|
|
|
*/
|
|
|
|
MEM_STATIC void ZSTD_storeSeq(seqStore_t* seqStorePtr, size_t litLength, const BYTE* literals, size_t offsetCode, size_t matchCode)
|
|
|
|
{
|
|
|
|
#if 0
|
|
|
|
static const BYTE* g_start = NULL;
|
|
|
|
if (g_start==NULL) g_start = literals;
|
|
|
|
if (literals - g_start == 8695)
|
|
|
|
printf("pos %6u : %3u literals & match %3u bytes at distance %6u \n",
|
|
|
|
(U32)(literals - g_start), (U32)litLength, (U32)matchCode+4, (U32)offsetCode);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
/* copy Literals */
|
|
|
|
ZSTD_wildcopy(seqStorePtr->lit, literals, litLength);
|
|
|
|
seqStorePtr->lit += litLength;
|
|
|
|
|
|
|
|
/* literal Length */
|
|
|
|
if (litLength >= MaxLL)
|
|
|
|
{
|
|
|
|
*(seqStorePtr->litLength++) = MaxLL;
|
|
|
|
if (litLength<255 + MaxLL)
|
|
|
|
*(seqStorePtr->dumps++) = (BYTE)(litLength - MaxLL);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*(seqStorePtr->dumps++) = 255;
|
|
|
|
MEM_writeLE32(seqStorePtr->dumps, (U32)litLength); seqStorePtr->dumps += 3;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else *(seqStorePtr->litLength++) = (BYTE)litLength;
|
|
|
|
|
|
|
|
/* match offset */
|
|
|
|
*(seqStorePtr->offset++) = (U32)offsetCode;
|
|
|
|
|
|
|
|
/* match Length */
|
|
|
|
if (matchCode >= MaxML)
|
|
|
|
{
|
|
|
|
*(seqStorePtr->matchLength++) = MaxML;
|
|
|
|
if (matchCode < 255+MaxML)
|
|
|
|
*(seqStorePtr->dumps++) = (BYTE)(matchCode - MaxML);
|
|
|
|
else
|
|
|
|
{
|
|
|
|
*(seqStorePtr->dumps++) = 255;
|
|
|
|
MEM_writeLE32(seqStorePtr->dumps, (U32)matchCode); seqStorePtr->dumps += 3;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else *(seqStorePtr->matchLength++) = (BYTE)matchCode;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* *************************************
|
|
|
|
* Match length counter
|
|
|
|
***************************************/
|
|
|
|
static size_t ZSTD_read_ARCH(const void* p) { size_t r; memcpy(&r, p, sizeof(r)); return r; }
|
|
|
|
|
|
|
|
static unsigned ZSTD_highbit(U32 val)
|
|
|
|
{
|
|
|
|
# if defined(_MSC_VER) /* Visual */
|
|
|
|
unsigned long r=0;
|
|
|
|
_BitScanReverse(&r, val);
|
|
|
|
return (unsigned)r;
|
|
|
|
# elif defined(__GNUC__) && (__GNUC__ >= 3) /* GCC Intrinsic */
|
|
|
|
return 31 - __builtin_clz(val);
|
|
|
|
# else /* Software version */
|
|
|
|
static const int DeBruijnClz[32] = { 0, 9, 1, 10, 13, 21, 2, 29, 11, 14, 16, 18, 22, 25, 3, 30, 8, 12, 20, 28, 15, 17, 24, 7, 19, 27, 23, 6, 26, 5, 4, 31 };
|
|
|
|
U32 v = val;
|
|
|
|
int r;
|
|
|
|
v |= v >> 1;
|
|
|
|
v |= v >> 2;
|
|
|
|
v |= v >> 4;
|
|
|
|
v |= v >> 8;
|
|
|
|
v |= v >> 16;
|
|
|
|
r = DeBruijnClz[(U32)(v * 0x07C4ACDDU) >> 27];
|
|
|
|
return r;
|
|
|
|
# endif
|
|
|
|
}
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
static unsigned ZSTD_NbCommonBytes (register size_t val)
|
2015-11-11 20:38:21 +00:00
|
|
|
{
|
|
|
|
if (MEM_isLittleEndian())
|
|
|
|
{
|
|
|
|
if (MEM_64bits())
|
|
|
|
{
|
|
|
|
# if defined(_MSC_VER) && defined(_WIN64)
|
|
|
|
unsigned long r = 0;
|
|
|
|
_BitScanForward64( &r, (U64)val );
|
|
|
|
return (int)(r>>3);
|
|
|
|
# elif defined(__GNUC__) && (__GNUC__ >= 3)
|
|
|
|
return (__builtin_ctzll((U64)val) >> 3);
|
|
|
|
# else
|
|
|
|
static const int DeBruijnBytePos[64] = { 0, 0, 0, 0, 0, 1, 1, 2, 0, 3, 1, 3, 1, 4, 2, 7, 0, 2, 3, 6, 1, 5, 3, 5, 1, 3, 4, 4, 2, 5, 6, 7, 7, 0, 1, 2, 3, 3, 4, 6, 2, 6, 5, 5, 3, 4, 5, 6, 7, 1, 2, 4, 6, 4, 4, 5, 7, 2, 6, 5, 7, 6, 7, 7 };
|
|
|
|
return DeBruijnBytePos[((U64)((val & -(long long)val) * 0x0218A392CDABBD3FULL)) >> 58];
|
|
|
|
# endif
|
|
|
|
}
|
|
|
|
else /* 32 bits */
|
|
|
|
{
|
|
|
|
# if defined(_MSC_VER)
|
|
|
|
unsigned long r=0;
|
|
|
|
_BitScanForward( &r, (U32)val );
|
|
|
|
return (int)(r>>3);
|
|
|
|
# elif defined(__GNUC__) && (__GNUC__ >= 3)
|
|
|
|
return (__builtin_ctz((U32)val) >> 3);
|
|
|
|
# else
|
|
|
|
static const int DeBruijnBytePos[32] = { 0, 0, 3, 0, 3, 1, 3, 0, 3, 2, 2, 1, 3, 2, 0, 1, 3, 3, 1, 2, 2, 2, 2, 0, 3, 1, 2, 0, 1, 0, 1, 1 };
|
|
|
|
return DeBruijnBytePos[((U32)((val & -(S32)val) * 0x077CB531U)) >> 27];
|
|
|
|
# endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
else /* Big Endian CPU */
|
|
|
|
{
|
|
|
|
if (MEM_32bits())
|
|
|
|
{
|
|
|
|
# if defined(_MSC_VER) && defined(_WIN64)
|
|
|
|
unsigned long r = 0;
|
|
|
|
_BitScanReverse64( &r, val );
|
|
|
|
return (unsigned)(r>>3);
|
|
|
|
# elif defined(__GNUC__) && (__GNUC__ >= 3)
|
|
|
|
return (__builtin_clzll(val) >> 3);
|
|
|
|
# else
|
|
|
|
unsigned r;
|
|
|
|
const unsigned n32 = sizeof(size_t)*4; /* calculate this way due to compiler complaining in 32-bits mode */
|
|
|
|
if (!(val>>n32)) { r=4; } else { r=0; val>>=n32; }
|
|
|
|
if (!(val>>16)) { r+=2; val>>=8; } else { val>>=24; }
|
|
|
|
r += (!val);
|
|
|
|
return r;
|
|
|
|
# endif
|
|
|
|
}
|
|
|
|
else /* 32 bits */
|
|
|
|
{
|
|
|
|
# if defined(_MSC_VER)
|
|
|
|
unsigned long r = 0;
|
|
|
|
_BitScanReverse( &r, (unsigned long)val );
|
|
|
|
return (unsigned)(r>>3);
|
|
|
|
# elif defined(__GNUC__) && (__GNUC__ >= 3)
|
|
|
|
return (__builtin_clz((U32)val) >> 3);
|
|
|
|
# else
|
|
|
|
unsigned r;
|
|
|
|
if (!(val>>16)) { r=2; val>>=8; } else { r=0; val>>=24; }
|
|
|
|
r += (!val);
|
|
|
|
return r;
|
|
|
|
# endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
static size_t ZSTD_count(const BYTE* pIn, const BYTE* pMatch, const BYTE* pInLimit)
|
2015-11-11 20:38:21 +00:00
|
|
|
{
|
|
|
|
const BYTE* const pStart = pIn;
|
|
|
|
|
|
|
|
while ((pIn<pInLimit-(sizeof(size_t)-1)))
|
|
|
|
{
|
|
|
|
size_t diff = ZSTD_read_ARCH(pMatch) ^ ZSTD_read_ARCH(pIn);
|
|
|
|
if (!diff) { pIn+=sizeof(size_t); pMatch+=sizeof(size_t); continue; }
|
|
|
|
pIn += ZSTD_NbCommonBytes(diff);
|
|
|
|
return (size_t)(pIn - pStart);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (MEM_64bits()) if ((pIn<(pInLimit-3)) && (MEM_read32(pMatch) == MEM_read32(pIn))) { pIn+=4; pMatch+=4; }
|
|
|
|
if ((pIn<(pInLimit-1)) && (MEM_read16(pMatch) == MEM_read16(pIn))) { pIn+=2; pMatch+=2; }
|
|
|
|
if ((pIn<pInLimit) && (*pMatch == *pIn)) pIn++;
|
|
|
|
return (size_t)(pIn - pStart);
|
|
|
|
}
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
/** ZSTD_count_2segments
|
|
|
|
* can count match length with ip & match in potentially 2 different segments.
|
|
|
|
* convention : on reaching mEnd, match count continue starting from iStart
|
|
|
|
*/
|
|
|
|
static size_t ZSTD_count_2segments(const BYTE* ip, const BYTE* match, const BYTE* iEnd, const BYTE* mEnd, const BYTE* iStart)
|
|
|
|
{
|
|
|
|
size_t matchLength;
|
|
|
|
const BYTE* vEnd = ip + (mEnd - match);
|
|
|
|
if (vEnd > iEnd) vEnd = iEnd;
|
|
|
|
matchLength = ZSTD_count(ip, match, vEnd);
|
|
|
|
if (match + matchLength == mEnd)
|
|
|
|
matchLength += ZSTD_count(ip+matchLength, iStart, iEnd);
|
|
|
|
return matchLength;
|
|
|
|
}
|
|
|
|
|
2015-11-11 20:38:21 +00:00
|
|
|
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
/* *************************************
|
2015-11-11 20:38:21 +00:00
|
|
|
* Hashes
|
2015-10-22 14:31:46 +00:00
|
|
|
***************************************/
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2015-10-30 14:49:48 +00:00
|
|
|
static const U32 prime4bytes = 2654435761U;
|
2015-11-11 12:43:58 +00:00
|
|
|
static U32 ZSTD_hash4(U32 u, U32 h) { return (u * prime4bytes) >> (32-h) ; }
|
|
|
|
static size_t ZSTD_hash4Ptr(const void* ptr, U32 h) { return ZSTD_hash4(MEM_read32(ptr), h); }
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2015-10-30 14:49:48 +00:00
|
|
|
static const U64 prime5bytes = 889523592379ULL;
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_hash5(U64 u, U32 h) { return (size_t)((u * prime5bytes) << (64-40) >> (64-h)) ; }
|
|
|
|
static size_t ZSTD_hash5Ptr(const void* p, U32 h) { return ZSTD_hash5(MEM_read64(p), h); }
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2015-10-30 14:49:48 +00:00
|
|
|
static const U64 prime6bytes = 227718039650203ULL;
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_hash6(U64 u, U32 h) { return (size_t)((u * prime6bytes) << (64-48) >> (64-h)) ; }
|
|
|
|
static size_t ZSTD_hash6Ptr(const void* p, U32 h) { return ZSTD_hash6(MEM_read64(p), h); }
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-11 20:38:21 +00:00
|
|
|
static const U64 prime7bytes = 58295818150454627ULL;
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_hash7(U64 u, U32 h) { return (size_t)((u * prime7bytes) << (64-56) >> (64-h)) ; }
|
|
|
|
static size_t ZSTD_hash7Ptr(const void* p, U32 h) { return ZSTD_hash7(MEM_read64(p), h); }
|
2015-11-05 16:32:18 +00:00
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_hashPtr(const void* p, U32 hBits, U32 mls)
|
2015-10-30 14:49:48 +00:00
|
|
|
{
|
|
|
|
switch(mls)
|
|
|
|
{
|
|
|
|
default:
|
2015-11-11 12:43:58 +00:00
|
|
|
case 4: return ZSTD_hash4Ptr(p, hBits);
|
|
|
|
case 5: return ZSTD_hash5Ptr(p, hBits);
|
|
|
|
case 6: return ZSTD_hash6Ptr(p, hBits);
|
|
|
|
case 7: return ZSTD_hash7Ptr(p, hBits);
|
2015-10-30 14:49:48 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-05 16:32:18 +00:00
|
|
|
/* *************************************
|
|
|
|
* Fast Scan
|
|
|
|
***************************************/
|
|
|
|
|
|
|
|
FORCE_INLINE
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_fast_generic(ZSTD_CCtx* ctx,
|
2015-11-13 10:27:46 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize,
|
|
|
|
const U32 mls)
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
|
|
|
U32* hashTable = ctx->hashTable;
|
|
|
|
const U32 hBits = ctx->params.hashLog;
|
|
|
|
seqStore_t* seqStorePtr = &(ctx->seqStore);
|
|
|
|
const BYTE* const base = ctx->base;
|
|
|
|
const BYTE* const istart = (const BYTE*)src;
|
2015-11-06 09:52:17 +00:00
|
|
|
const BYTE* ip = istart;
|
2015-11-05 16:32:18 +00:00
|
|
|
const BYTE* anchor = istart;
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* const lowest = base + ctx->dictLimit;
|
2015-11-05 16:32:18 +00:00
|
|
|
const BYTE* const iend = istart + srcSize;
|
|
|
|
const BYTE* const ilimit = iend - 8;
|
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
size_t offset_2=REPCODE_STARTVALUE, offset_1=REPCODE_STARTVALUE;
|
2015-11-05 16:32:18 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* init */
|
2015-11-20 11:03:53 +00:00
|
|
|
ZSTD_resetSeqStore(seqStorePtr);
|
2015-11-05 16:32:18 +00:00
|
|
|
if (ip == base)
|
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
hashTable[ZSTD_hashPtr(base+1, hBits, mls)] = 1;
|
|
|
|
hashTable[ZSTD_hashPtr(base+2, hBits, mls)] = 2;
|
|
|
|
hashTable[ZSTD_hashPtr(base+3, hBits, mls)] = 3;
|
2015-11-05 16:32:18 +00:00
|
|
|
ip = base+4;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Main Search Loop */
|
2015-11-20 11:03:53 +00:00
|
|
|
while (ip < ilimit) /* < instead of <=, because repcode check at (ip+1) */
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t mlCode;
|
2015-11-20 11:03:53 +00:00
|
|
|
size_t offset;
|
2015-11-11 12:43:58 +00:00
|
|
|
const size_t h = ZSTD_hashPtr(ip, hBits, mls);
|
2015-11-05 16:32:18 +00:00
|
|
|
const BYTE* match = base + hashTable[h];
|
|
|
|
hashTable[h] = (U32)(ip-base);
|
|
|
|
|
2015-11-20 11:03:53 +00:00
|
|
|
if (MEM_read32(ip+1-offset_1) == MEM_read32(ip+1))
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
2015-11-23 12:34:21 +00:00
|
|
|
mlCode = ZSTD_count(ip+1+MINMATCH, ip+1+MINMATCH-offset_1, iend);
|
2015-11-20 11:46:08 +00:00
|
|
|
ip++;
|
|
|
|
offset = 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if ( (match < lowest) ||
|
|
|
|
(MEM_read32(match) != MEM_read32(ip)) )
|
|
|
|
{
|
|
|
|
ip += ((ip-anchor) >> g_searchStrength) + 1;
|
|
|
|
continue;
|
|
|
|
}
|
2015-11-23 12:34:21 +00:00
|
|
|
mlCode = ZSTD_count(ip+MINMATCH, match+MINMATCH, iend);
|
2015-11-20 11:46:08 +00:00
|
|
|
offset = ip-match;
|
2015-11-23 12:34:21 +00:00
|
|
|
while ((ip>anchor) && (match>lowest) && (ip[-1] == match[-1])) { ip--; match--; mlCode++; } /* catch up */
|
2015-11-20 11:46:08 +00:00
|
|
|
offset_2 = offset_1;
|
|
|
|
offset_1 = offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* match found */
|
2015-11-23 12:34:21 +00:00
|
|
|
ZSTD_storeSeq(seqStorePtr, ip-anchor, anchor, offset, mlCode);
|
|
|
|
hashTable[ZSTD_hashPtr(ip+2, hBits, mls)] = (U32)(ip+2-base); /* can't use current : ip may have changed */
|
|
|
|
ip += mlCode + MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
anchor = ip;
|
|
|
|
|
|
|
|
if (ip <= ilimit)
|
|
|
|
{
|
|
|
|
/* Fill Table */
|
|
|
|
hashTable[ZSTD_hashPtr(ip-2, hBits, mls)] = (U32)(ip-2-base);
|
|
|
|
/* check immediate repcode */
|
|
|
|
while ( (ip <= ilimit)
|
|
|
|
&& (MEM_read32(ip) == MEM_read32(ip - offset_2)) )
|
|
|
|
{
|
|
|
|
/* store sequence */
|
|
|
|
size_t ml = ZSTD_count(ip+MINMATCH, ip+MINMATCH-offset_2, iend);
|
|
|
|
size_t tmpOff = offset_2; offset_2 = offset_1; offset_1 = tmpOff; /* swap offset_2 <=> offset_1 */
|
|
|
|
hashTable[ZSTD_hashPtr(ip, hBits, mls)] = (U32)(ip-base);
|
|
|
|
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, ml);
|
|
|
|
ip += ml+MINMATCH;
|
|
|
|
anchor = ip;
|
|
|
|
continue; /* faster when present ... (?) */
|
|
|
|
}
|
|
|
|
}
|
2015-11-05 16:32:18 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Last Literals */
|
|
|
|
{
|
|
|
|
size_t lastLLSize = iend - anchor;
|
|
|
|
memcpy(seqStorePtr->lit, anchor, lastLLSize);
|
|
|
|
seqStorePtr->lit += lastLLSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Finale compression stage */
|
2015-11-23 12:34:21 +00:00
|
|
|
return ZSTD_compressSequences(dst, maxDstSize,
|
2015-11-05 16:32:18 +00:00
|
|
|
seqStorePtr, srcSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_fast(ZSTD_CCtx* ctx,
|
2015-11-05 16:32:18 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
switch(mls)
|
|
|
|
{
|
|
|
|
default:
|
|
|
|
case 4 :
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_compressBlock_fast_generic(ctx, dst, maxDstSize, src, srcSize, 4);
|
2015-11-05 16:32:18 +00:00
|
|
|
case 5 :
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_compressBlock_fast_generic(ctx, dst, maxDstSize, src, srcSize, 5);
|
2015-11-05 16:32:18 +00:00
|
|
|
case 6 :
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_compressBlock_fast_generic(ctx, dst, maxDstSize, src, srcSize, 6);
|
2015-11-05 16:32:18 +00:00
|
|
|
case 7 :
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_compressBlock_fast_generic(ctx, dst, maxDstSize, src, srcSize, 7);
|
2015-11-05 16:32:18 +00:00
|
|
|
}
|
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
|
|
|
|
2015-11-13 14:08:43 +00:00
|
|
|
//FORCE_INLINE
|
2015-11-13 10:27:46 +00:00
|
|
|
size_t ZSTD_compressBlock_fast_extDict_generic(ZSTD_CCtx* ctx,
|
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize,
|
|
|
|
const U32 mls)
|
|
|
|
{
|
|
|
|
U32* hashTable = ctx->hashTable;
|
|
|
|
const U32 hBits = ctx->params.hashLog;
|
|
|
|
seqStore_t* seqStorePtr = &(ctx->seqStore);
|
|
|
|
const BYTE* const base = ctx->base;
|
|
|
|
const BYTE* const dictBase = ctx->dictBase;
|
|
|
|
const BYTE* const istart = (const BYTE*)src;
|
|
|
|
const BYTE* ip = istart;
|
|
|
|
const BYTE* anchor = istart;
|
|
|
|
const U32 lowLimit = ctx->lowLimit;
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* const dictStart = dictBase + lowLimit;
|
2015-11-13 10:27:46 +00:00
|
|
|
const U32 dictLimit = ctx->dictLimit;
|
2015-11-20 11:03:53 +00:00
|
|
|
const BYTE* const lowPrefixPtr = base + dictLimit;
|
|
|
|
const BYTE* const dictEnd = dictBase + dictLimit;
|
2015-11-13 10:27:46 +00:00
|
|
|
const BYTE* const iend = istart + srcSize;
|
|
|
|
const BYTE* const ilimit = iend - 8;
|
|
|
|
|
2015-11-17 13:26:54 +00:00
|
|
|
U32 offset_2=REPCODE_STARTVALUE, offset_1=REPCODE_STARTVALUE;
|
2015-11-13 10:27:46 +00:00
|
|
|
|
|
|
|
|
|
|
|
/* init */
|
|
|
|
ZSTD_resetSeqStore(seqStorePtr);
|
|
|
|
{
|
|
|
|
/* skip first 4 positions to avoid read overflow during repcode match check */
|
|
|
|
hashTable[ZSTD_hashPtr(ip+0, hBits, mls)] = (U32)(ip-base+0);
|
|
|
|
hashTable[ZSTD_hashPtr(ip+1, hBits, mls)] = (U32)(ip-base+1);
|
|
|
|
hashTable[ZSTD_hashPtr(ip+2, hBits, mls)] = (U32)(ip-base+2);
|
|
|
|
hashTable[ZSTD_hashPtr(ip+3, hBits, mls)] = (U32)(ip-base+3);
|
|
|
|
ip += 4;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Main Search Loop */
|
2015-11-20 11:03:53 +00:00
|
|
|
while (ip < ilimit) /* < instead of <=, because (ip+1) */
|
2015-11-13 10:27:46 +00:00
|
|
|
{
|
|
|
|
const size_t h = ZSTD_hashPtr(ip, hBits, mls);
|
2015-11-20 11:03:53 +00:00
|
|
|
const U32 matchIndex = hashTable[h];
|
2015-11-13 10:27:46 +00:00
|
|
|
const BYTE* matchBase = matchIndex < dictLimit ? dictBase : base;
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* match = matchBase + matchIndex;
|
2015-11-13 10:27:46 +00:00
|
|
|
const U32 current = (U32)(ip-base);
|
2015-11-20 11:03:53 +00:00
|
|
|
const U32 repIndex = current + 1 - offset_1;
|
2015-11-20 11:46:08 +00:00
|
|
|
const BYTE* repBase = repIndex < dictLimit ? dictBase : base;
|
2015-11-13 10:27:46 +00:00
|
|
|
const BYTE* repMatch = repBase + repIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t mlCode;
|
2015-11-20 11:03:53 +00:00
|
|
|
U32 offset;
|
2015-11-13 10:27:46 +00:00
|
|
|
hashTable[h] = current; /* update hash table */
|
|
|
|
|
|
|
|
if ( ((repIndex <= dictLimit-4) || (repIndex >= dictLimit))
|
2015-11-20 11:03:53 +00:00
|
|
|
&& (MEM_read32(repMatch) == MEM_read32(ip+1)) )
|
2015-11-13 10:27:46 +00:00
|
|
|
{
|
2015-11-20 11:46:08 +00:00
|
|
|
const BYTE* repMatchEnd = repIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
mlCode = ZSTD_count_2segments(ip+1+MINMATCH, repMatch+MINMATCH, iend, repMatchEnd, lowPrefixPtr);
|
2015-11-20 11:03:53 +00:00
|
|
|
ip++;
|
2015-11-20 11:46:08 +00:00
|
|
|
offset = 0;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
if ( (matchIndex < lowLimit) ||
|
|
|
|
(MEM_read32(match) != MEM_read32(ip)) )
|
|
|
|
{ ip += ((ip-anchor) >> g_searchStrength) + 1; continue; }
|
|
|
|
{
|
|
|
|
const BYTE* matchEnd = matchIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* lowMatchPtr = matchIndex < dictLimit ? dictStart : lowPrefixPtr;
|
|
|
|
mlCode = ZSTD_count_2segments(ip+MINMATCH, match+MINMATCH, iend, matchEnd, lowPrefixPtr);
|
|
|
|
while ((ip>anchor) && (match>lowMatchPtr) && (ip[-1] == match[-1])) { ip--; match--; mlCode++; } /* catch up */
|
2015-11-20 11:46:08 +00:00
|
|
|
offset = current - matchIndex;
|
|
|
|
offset_2 = offset_1;
|
|
|
|
offset_1 = offset;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
/* found a match : store it */
|
|
|
|
ZSTD_storeSeq(seqStorePtr, ip-anchor, anchor, offset, mlCode);
|
|
|
|
hashTable[ZSTD_hashPtr(ip+2, hBits, mls)] = (U32)(ip+2-base); /* can't use current : ip may have changed */
|
|
|
|
ip += mlCode + MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
anchor = ip;
|
|
|
|
|
|
|
|
if (ip <= ilimit)
|
|
|
|
{
|
2015-11-23 12:34:21 +00:00
|
|
|
/* Fill Table */
|
2015-11-20 11:46:08 +00:00
|
|
|
hashTable[ZSTD_hashPtr(ip-2, hBits, mls)] = (U32)(ip-2-base);
|
|
|
|
/* check immediate repcode */
|
|
|
|
while (ip <= ilimit)
|
|
|
|
{
|
|
|
|
U32 current2 = (U32)(ip-base);
|
|
|
|
const U32 repIndex2 = current2 - offset_2;
|
|
|
|
const BYTE* repMatch2 = repIndex2 < dictLimit ? dictBase + repIndex2 : base + repIndex2;
|
2015-11-20 11:03:53 +00:00
|
|
|
if ( ((repIndex2 <= dictLimit-4) || (repIndex2 >= dictLimit))
|
|
|
|
&& (MEM_read32(repMatch2) == MEM_read32(ip)) )
|
2015-11-20 11:46:08 +00:00
|
|
|
{
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* const repEnd2 = repIndex2 < dictLimit ? dictEnd : iend;
|
|
|
|
size_t repLength2 = ZSTD_count_2segments(ip+MINMATCH, repMatch2+MINMATCH, iend, repEnd2, lowPrefixPtr);
|
|
|
|
U32 tmpOffset = offset_2; offset_2 = offset_1; offset_1 = tmpOffset; /* swap offset_2 <=> offset_1 */
|
|
|
|
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, repLength2);
|
|
|
|
hashTable[ZSTD_hashPtr(ip, hBits, mls)] = current2;
|
|
|
|
ip += repLength2+MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
anchor = ip;
|
|
|
|
continue;
|
|
|
|
}
|
2015-11-20 11:03:53 +00:00
|
|
|
break;
|
2015-11-20 11:46:08 +00:00
|
|
|
}
|
|
|
|
}
|
2015-11-13 10:27:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Last Literals */
|
|
|
|
{
|
|
|
|
size_t lastLLSize = iend - anchor;
|
|
|
|
memcpy(seqStorePtr->lit, anchor, lastLLSize);
|
|
|
|
seqStorePtr->lit += lastLLSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Finale compression stage */
|
2015-11-23 12:34:21 +00:00
|
|
|
return ZSTD_compressSequences(dst, maxDstSize,
|
2015-11-13 10:27:46 +00:00
|
|
|
seqStorePtr, srcSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
size_t ZSTD_compressBlock_fast_extDict(ZSTD_CCtx* ctx,
|
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
switch(mls)
|
|
|
|
{
|
|
|
|
default:
|
|
|
|
case 4 :
|
|
|
|
return ZSTD_compressBlock_fast_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 4);
|
|
|
|
case 5 :
|
|
|
|
return ZSTD_compressBlock_fast_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 5);
|
|
|
|
case 6 :
|
|
|
|
return ZSTD_compressBlock_fast_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 6);
|
|
|
|
case 7 :
|
|
|
|
return ZSTD_compressBlock_fast_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 7);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
/* *************************************
|
2015-11-04 02:52:54 +00:00
|
|
|
* Binary Tree search
|
2015-10-22 14:31:46 +00:00
|
|
|
***************************************/
|
2015-11-11 12:43:58 +00:00
|
|
|
/** ZSTD_insertBt1 : add one ptr to tree
|
2015-11-04 02:52:54 +00:00
|
|
|
@ip : assumed <= iend-8 */
|
2015-11-11 12:43:58 +00:00
|
|
|
static U32 ZSTD_insertBt1(ZSTD_CCtx* zc, const BYTE* const ip, const U32 mls, const BYTE* const iend, U32 nbCompares)
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
U32* const hashTable = zc->hashTable;
|
|
|
|
const U32 hashLog = zc->params.hashLog;
|
2015-11-11 12:43:58 +00:00
|
|
|
const size_t h = ZSTD_hashPtr(ip, hashLog, mls);
|
2015-11-05 16:32:18 +00:00
|
|
|
U32* const bt = zc->contentTable;
|
|
|
|
const U32 btLog = zc->params.contentLog - 1;
|
2015-11-04 02:52:54 +00:00
|
|
|
const U32 btMask= (1 << btLog) - 1;
|
|
|
|
U32 matchIndex = hashTable[h];
|
|
|
|
size_t commonLengthSmaller=0, commonLengthLarger=0;
|
|
|
|
const BYTE* const base = zc->base;
|
2015-11-07 00:13:31 +00:00
|
|
|
const BYTE* match = base + matchIndex;
|
|
|
|
U32 current = (U32)(ip-base);
|
2015-11-08 14:08:03 +00:00
|
|
|
const U32 btLow = btMask >= current ? 0 : current - btMask;
|
2015-11-04 02:52:54 +00:00
|
|
|
U32* smallerPtr = bt + 2*(current&btMask);
|
|
|
|
U32* largerPtr = bt + 2*(current&btMask) + 1;
|
2015-11-04 11:05:27 +00:00
|
|
|
U32 dummy32; /* to be nullified at the end */
|
2015-11-04 02:52:54 +00:00
|
|
|
const U32 windowSize = 1 << zc->params.windowLog;
|
|
|
|
const U32 windowLow = windowSize >= current ? 0 : current - windowSize;
|
|
|
|
|
2015-11-08 14:08:03 +00:00
|
|
|
if ((current-matchIndex == 1) /* RLE */
|
2015-11-08 14:49:20 +00:00
|
|
|
&& MEM_read64(match) == MEM_read64(ip))
|
2015-11-07 00:13:31 +00:00
|
|
|
{
|
2015-11-08 14:08:03 +00:00
|
|
|
size_t rleLength = ZSTD_count(ip+sizeof(size_t), match+sizeof(size_t), iend) + sizeof(size_t);
|
|
|
|
return (U32)(rleLength - mls);
|
2015-11-07 00:13:31 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
hashTable[h] = (U32)(ip - base); /* Update Hash Table */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
while (nbCompares-- && (matchIndex > windowLow))
|
|
|
|
{
|
|
|
|
U32* nextPtr = bt + 2*(matchIndex & btMask);
|
|
|
|
size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
|
|
|
|
|
2015-11-07 00:13:31 +00:00
|
|
|
match = base + matchIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
if (match[matchLength] == ip[matchLength])
|
|
|
|
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iend) +1;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
if (ip+matchLength == iend) /* equal : no way to know if inf or sup */
|
2015-11-07 00:13:31 +00:00
|
|
|
break; /* just drop , to guarantee consistency (miss a bit of compression; if someone knows better, please tell) */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
if (match[matchLength] < ip[matchLength])
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-04 02:52:54 +00:00
|
|
|
/* match is smaller than current */
|
|
|
|
*smallerPtr = matchIndex; /* update smaller idx */
|
|
|
|
commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
|
2015-11-07 00:13:31 +00:00
|
|
|
if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
2015-11-04 02:52:54 +00:00
|
|
|
smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
|
2015-11-07 00:13:31 +00:00
|
|
|
matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
else
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-04 02:52:54 +00:00
|
|
|
/* match is larger than current */
|
|
|
|
*largerPtr = matchIndex;
|
|
|
|
commonLengthLarger = matchLength;
|
2015-11-07 00:13:31 +00:00
|
|
|
if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
2015-11-04 02:52:54 +00:00
|
|
|
largerPtr = nextPtr;
|
2015-11-07 00:13:31 +00:00
|
|
|
matchIndex = nextPtr[0];
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
*smallerPtr = *largerPtr = 0;
|
2015-11-08 14:08:03 +00:00
|
|
|
return 1;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_insertBtAndFindBestMatch (
|
|
|
|
ZSTD_CCtx* zc,
|
2015-11-04 02:52:54 +00:00
|
|
|
const BYTE* const ip, const BYTE* const iend,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
U32 nbCompares, const U32 mls)
|
|
|
|
{
|
|
|
|
U32* const hashTable = zc->hashTable;
|
|
|
|
const U32 hashLog = zc->params.hashLog;
|
2015-11-11 12:43:58 +00:00
|
|
|
const size_t h = ZSTD_hashPtr(ip, hashLog, mls);
|
2015-11-05 16:32:18 +00:00
|
|
|
U32* const bt = zc->contentTable;
|
|
|
|
const U32 btLog = zc->params.contentLog - 1;
|
2015-11-04 02:52:54 +00:00
|
|
|
const U32 btMask= (1 << btLog) - 1;
|
|
|
|
U32 matchIndex = hashTable[h];
|
|
|
|
size_t commonLengthSmaller=0, commonLengthLarger=0;
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 current = (U32)(ip-base);
|
|
|
|
const U32 btLow = btMask >= current ? 0 : current - btMask;
|
|
|
|
const U32 windowSize = 1 << zc->params.windowLog;
|
|
|
|
const U32 windowLow = windowSize >= current ? 0 : current - windowSize;
|
|
|
|
U32* smallerPtr = bt + 2*(current&btMask);
|
|
|
|
U32* largerPtr = bt + 2*(current&btMask) + 1;
|
2015-11-04 12:57:24 +00:00
|
|
|
size_t bestLength = 0;
|
2015-11-04 11:05:27 +00:00
|
|
|
U32 dummy32; /* to be nullified at the end */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
hashTable[h] = (U32)(ip-base); /* Update Hash Table */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
while (nbCompares-- && (matchIndex > windowLow))
|
|
|
|
{
|
|
|
|
U32* nextPtr = bt + 2*(matchIndex & btMask);
|
|
|
|
const BYTE* match = base + matchIndex;
|
|
|
|
size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
if (match[matchLength] == ip[matchLength])
|
|
|
|
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iend) +1;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
if (matchLength > bestLength)
|
|
|
|
{
|
2015-11-04 16:41:20 +00:00
|
|
|
if ( (4*(int)(matchLength-bestLength)) > (int)(ZSTD_highbit(current-matchIndex+1) - ZSTD_highbit((U32)offsetPtr[0]+1)) )
|
2015-11-04 12:57:24 +00:00
|
|
|
bestLength = matchLength, *offsetPtr = current - matchIndex;
|
2015-11-04 11:05:27 +00:00
|
|
|
if (ip+matchLength == iend) /* equal : no way to know if inf or sup */
|
2015-11-23 12:34:21 +00:00
|
|
|
break; /* drop, to guarantee consistency (miss a little bit of compression) */
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (match[matchLength] < ip[matchLength])
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-04 02:52:54 +00:00
|
|
|
/* match is smaller than current */
|
|
|
|
*smallerPtr = matchIndex; /* update smaller idx */
|
|
|
|
commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
|
2015-11-23 12:34:21 +00:00
|
|
|
if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
2015-11-04 11:05:27 +00:00
|
|
|
smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
|
2015-11-23 12:34:21 +00:00
|
|
|
matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
else
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-04 02:52:54 +00:00
|
|
|
/* match is larger than current */
|
|
|
|
*largerPtr = matchIndex;
|
|
|
|
commonLengthLarger = matchLength;
|
2015-11-23 12:34:21 +00:00
|
|
|
if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
2015-11-04 02:52:54 +00:00
|
|
|
largerPtr = nextPtr;
|
2015-11-23 12:34:21 +00:00
|
|
|
matchIndex = nextPtr[0];
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
*smallerPtr = *largerPtr = 0;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
zc->nextToUpdate = current+1; /* current has been inserted */
|
|
|
|
return bestLength;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
static const BYTE* ZSTD_updateTree(ZSTD_CCtx* zc, const BYTE* const ip, const BYTE* const iend, const U32 nbCompares, const U32 mls)
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 target = (U32)(ip - base);
|
|
|
|
U32 idx = zc->nextToUpdate;
|
2015-11-04 11:05:27 +00:00
|
|
|
//size_t dummy;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-11-07 00:13:31 +00:00
|
|
|
for( ; idx < target ; )
|
2015-11-11 12:43:58 +00:00
|
|
|
idx += ZSTD_insertBt1(zc, base+idx, mls, iend, nbCompares);
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-11-07 00:13:31 +00:00
|
|
|
zc->nextToUpdate = idx;
|
|
|
|
return base + idx;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Tree updater, providing best match */
|
|
|
|
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_BtFindBestMatch (
|
|
|
|
ZSTD_CCtx* zc,
|
2015-11-04 02:52:54 +00:00
|
|
|
const BYTE* const ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 mls)
|
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
const BYTE* nextToUpdate = ZSTD_updateTree(zc, ip, iLimit, maxNbAttempts, mls);
|
2015-11-07 00:13:31 +00:00
|
|
|
if (nextToUpdate > ip)
|
|
|
|
{
|
|
|
|
/* RLE data */
|
|
|
|
*offsetPtr = 1;
|
|
|
|
return ZSTD_count(ip, ip-1, iLimit);
|
|
|
|
}
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_insertBtAndFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, mls);
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
FORCE_INLINE size_t ZSTD_BtFindBestMatch_selectMLS (
|
|
|
|
ZSTD_CCtx* zc, /* Index table will be updated */
|
2015-11-04 02:52:54 +00:00
|
|
|
const BYTE* ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
|
|
|
{
|
|
|
|
switch(matchLengthSearch)
|
|
|
|
{
|
|
|
|
default :
|
2015-11-11 12:43:58 +00:00
|
|
|
case 4 : return ZSTD_BtFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4);
|
|
|
|
case 5 : return ZSTD_BtFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5);
|
|
|
|
case 6 : return ZSTD_BtFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6);
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-05 14:00:24 +00:00
|
|
|
/* ***********************
|
|
|
|
* Hash Chain
|
|
|
|
*************************/
|
|
|
|
|
2015-11-05 16:32:18 +00:00
|
|
|
#define NEXT_IN_CHAIN(d, mask) chainTable[(d) & mask]
|
|
|
|
|
2015-11-05 14:00:24 +00:00
|
|
|
/* Update chains up to ip (excluded) */
|
2015-11-11 12:43:58 +00:00
|
|
|
static U32 ZSTD_insertAndFindFirstIndex (ZSTD_CCtx* zc, const BYTE* ip, U32 mls)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
|
|
|
U32* const hashTable = zc->hashTable;
|
|
|
|
const U32 hashLog = zc->params.hashLog;
|
2015-11-05 16:32:18 +00:00
|
|
|
U32* const chainTable = zc->contentTable;
|
|
|
|
const U32 chainMask = (1 << zc->params.contentLog) - 1;
|
2015-11-05 14:00:24 +00:00
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 target = (U32)(ip - base);
|
|
|
|
U32 idx = zc->nextToUpdate;
|
|
|
|
|
|
|
|
while(idx < target)
|
|
|
|
{
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t h = ZSTD_hashPtr(base+idx, hashLog, mls);
|
2015-11-05 14:00:24 +00:00
|
|
|
NEXT_IN_CHAIN(idx, chainMask) = hashTable[h];
|
|
|
|
hashTable[h] = idx;
|
|
|
|
idx++;
|
|
|
|
}
|
|
|
|
|
|
|
|
zc->nextToUpdate = target;
|
2015-11-11 12:43:58 +00:00
|
|
|
return hashTable[ZSTD_hashPtr(ip, hashLog, mls)];
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_HcFindBestMatch (
|
|
|
|
ZSTD_CCtx* zc, /* Index table will be updated */
|
2015-11-05 14:00:24 +00:00
|
|
|
const BYTE* const ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
2015-11-22 12:24:05 +00:00
|
|
|
{
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 current = (U32)(ip-base);
|
|
|
|
U32* const chainTable = zc->contentTable;
|
|
|
|
const U32 chainSize = (1 << zc->params.contentLog);
|
|
|
|
const U32 minChain = current > chainSize ? current - chainSize : 0;
|
|
|
|
const U32 chainMask = chainSize-1;
|
2015-11-23 12:34:21 +00:00
|
|
|
const U32 lowLimit = zc->dictLimit; /* should be equal to zc->lowLimit, but safer use this one if they are not since dictLimit >= lowLimit */
|
2015-11-22 12:24:05 +00:00
|
|
|
U32 matchIndex;
|
|
|
|
const BYTE* match;
|
|
|
|
int nbAttempts=maxNbAttempts;
|
|
|
|
size_t ml=0;
|
|
|
|
|
|
|
|
/* HC4 match finder */
|
|
|
|
matchIndex = ZSTD_insertAndFindFirstIndex (zc, ip, matchLengthSearch);
|
|
|
|
|
|
|
|
while ((matchIndex>lowLimit) && (nbAttempts))
|
|
|
|
{
|
|
|
|
nbAttempts--;
|
|
|
|
match = base + matchIndex;
|
|
|
|
if (match[ml] == ip[ml]) /* potentially better */
|
|
|
|
{
|
|
|
|
const size_t mlt = ZSTD_count(ip, match, iLimit);
|
|
|
|
if (mlt > ml)
|
2015-11-23 12:34:21 +00:00
|
|
|
//if ( (4*(int)(mlt-ml)) > (int)(ZSTD_highbit(current-matchIndex+1) - ZSTD_highbit((U32)(*offsetPtr) + 1)) ) /* stronger but slower */
|
2015-11-22 12:24:05 +00:00
|
|
|
{
|
|
|
|
ml = mlt; *offsetPtr = ip-match;
|
2015-11-23 12:34:21 +00:00
|
|
|
if (ip+mlt >= iLimit) break; /* max ml possible; avoid risk of reading too far when testing ip[ml] */
|
2015-11-22 12:24:05 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (matchIndex <= minChain) break;
|
|
|
|
matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
|
|
|
|
}
|
|
|
|
|
|
|
|
return ml;
|
|
|
|
}
|
|
|
|
|
|
|
|
FORCE_INLINE size_t ZSTD_HcFindBestMatch_selectMLS (
|
|
|
|
ZSTD_CCtx* zc, /* Index table will be updated */
|
|
|
|
const BYTE* ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
|
|
|
{
|
|
|
|
switch(matchLengthSearch)
|
|
|
|
{
|
|
|
|
default :
|
|
|
|
case 4 : return ZSTD_HcFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4);
|
|
|
|
case 5 : return ZSTD_HcFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5);
|
|
|
|
case 6 : return ZSTD_HcFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
|
|
|
|
size_t ZSTD_HcFindBestMatch_extDict (
|
|
|
|
ZSTD_CCtx* zc, /* Index table will be updated */
|
|
|
|
const BYTE* const ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
2015-11-05 16:32:18 +00:00
|
|
|
U32* const chainTable = zc->contentTable;
|
|
|
|
const U32 chainSize = (1 << zc->params.contentLog);
|
2015-11-05 14:00:24 +00:00
|
|
|
const U32 chainMask = chainSize-1;
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const BYTE* const dictBase = zc->dictBase;
|
|
|
|
const U32 dictLimit = zc->dictLimit;
|
2015-11-23 12:34:21 +00:00
|
|
|
const BYTE* const prefixStart = base + dictLimit;
|
|
|
|
const BYTE* const dictEnd = dictBase + dictLimit;
|
2015-11-13 10:27:46 +00:00
|
|
|
const U32 lowLimit = zc->lowLimit;
|
2015-11-22 01:53:43 +00:00
|
|
|
const U32 current = (U32)(ip-base);
|
|
|
|
const U32 minChain = current > chainSize ? current - chainSize : 0;
|
2015-11-05 14:00:24 +00:00
|
|
|
U32 matchIndex;
|
|
|
|
const BYTE* match;
|
|
|
|
int nbAttempts=maxNbAttempts;
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t ml=MINMATCH-1;
|
2015-11-05 14:00:24 +00:00
|
|
|
|
|
|
|
/* HC4 match finder */
|
2015-11-11 12:43:58 +00:00
|
|
|
matchIndex = ZSTD_insertAndFindFirstIndex (zc, ip, matchLengthSearch);
|
2015-11-05 14:00:24 +00:00
|
|
|
|
|
|
|
while ((matchIndex>lowLimit) && (nbAttempts))
|
|
|
|
{
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t currentMl=0;
|
2015-11-05 14:00:24 +00:00
|
|
|
nbAttempts--;
|
|
|
|
if (matchIndex >= dictLimit)
|
|
|
|
{
|
|
|
|
match = base + matchIndex;
|
2015-11-09 02:19:33 +00:00
|
|
|
if (match[ml] == ip[ml]) /* potentially better */
|
2015-11-23 12:34:21 +00:00
|
|
|
currentMl = ZSTD_count(ip, match, iLimit);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
match = dictBase + matchIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
if (MEM_read32(match) == MEM_read32(ip)) /* assumption : matchIndex <= dictLimit-4 (by table construction) */
|
|
|
|
currentMl = ZSTD_count_2segments(ip+MINMATCH, match+MINMATCH, iLimit, dictEnd, prefixStart) + MINMATCH;
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
/* save best solution */
|
|
|
|
if (currentMl > ml) { ml = currentMl; *offsetPtr = current - matchIndex; }
|
|
|
|
|
2015-11-22 01:53:43 +00:00
|
|
|
if (matchIndex <= minChain) break;
|
2015-11-05 14:00:24 +00:00
|
|
|
matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
|
|
|
|
}
|
|
|
|
|
|
|
|
return ml;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-22 12:24:05 +00:00
|
|
|
FORCE_INLINE size_t ZSTD_HcFindBestMatch_extDict_selectMLS (
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_CCtx* zc, /* Index table will be updated */
|
2015-11-05 14:00:24 +00:00
|
|
|
const BYTE* ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
|
|
|
{
|
|
|
|
switch(matchLengthSearch)
|
|
|
|
{
|
|
|
|
default :
|
2015-11-22 12:24:05 +00:00
|
|
|
case 4 : return ZSTD_HcFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4);
|
|
|
|
case 5 : return ZSTD_HcFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5);
|
|
|
|
case 6 : return ZSTD_HcFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-22 12:24:05 +00:00
|
|
|
/* *******************************
|
2015-11-22 01:53:43 +00:00
|
|
|
* Common parser - lazy strategy
|
2015-11-22 12:24:05 +00:00
|
|
|
*********************************/
|
2015-11-05 14:00:24 +00:00
|
|
|
FORCE_INLINE
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_lazy_generic(ZSTD_CCtx* ctx,
|
2015-11-19 16:28:35 +00:00
|
|
|
void* dst, size_t maxDstSize, const void* src, size_t srcSize,
|
2015-11-22 01:42:28 +00:00
|
|
|
const U32 searchMethod, const U32 depth)
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
seqStore_t* seqStorePtr = &(ctx->seqStore);
|
|
|
|
const BYTE* const istart = (const BYTE*)src;
|
|
|
|
const BYTE* ip = istart;
|
|
|
|
const BYTE* anchor = istart;
|
|
|
|
const BYTE* const iend = istart + srcSize;
|
|
|
|
const BYTE* const ilimit = iend - 8;
|
|
|
|
|
|
|
|
size_t offset_2=REPCODE_STARTVALUE, offset_1=REPCODE_STARTVALUE;
|
|
|
|
const U32 maxSearches = 1 << ctx->params.searchLog;
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
typedef size_t (*searchMax_f)(ZSTD_CCtx* zc, const BYTE* ip, const BYTE* iLimit,
|
2015-11-05 14:00:24 +00:00
|
|
|
size_t* offsetPtr,
|
|
|
|
U32 maxNbAttempts, U32 matchLengthSearch);
|
2015-11-11 12:43:58 +00:00
|
|
|
searchMax_f searchMax = searchMethod ? ZSTD_BtFindBestMatch_selectMLS : ZSTD_HcFindBestMatch_selectMLS;
|
2015-11-05 14:00:24 +00:00
|
|
|
|
2015-11-04 02:52:54 +00:00
|
|
|
/* init */
|
|
|
|
ZSTD_resetSeqStore(seqStorePtr);
|
|
|
|
if (((ip-ctx->base) - ctx->dictLimit) < REPCODE_STARTVALUE) ip += REPCODE_STARTVALUE;
|
|
|
|
|
|
|
|
/* Match Loop */
|
2015-11-21 14:27:35 +00:00
|
|
|
while (ip < ilimit)
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
2015-11-21 14:27:35 +00:00
|
|
|
size_t matchLength=0;
|
|
|
|
size_t offset=0;
|
|
|
|
const BYTE* start=ip+1;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-11-20 11:03:53 +00:00
|
|
|
/* check repCode */
|
2015-11-21 14:27:35 +00:00
|
|
|
if (MEM_read32(ip+1) == MEM_read32(ip+1 - offset_1))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
2015-11-20 11:03:53 +00:00
|
|
|
/* repcode : we take it */
|
2015-11-21 14:27:35 +00:00
|
|
|
matchLength = ZSTD_count(ip+1+MINMATCH, ip+1+MINMATCH-offset_1, iend) + MINMATCH;
|
2015-11-22 01:42:28 +00:00
|
|
|
if (depth==0) goto _storeSequence;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
2015-11-06 14:40:14 +00:00
|
|
|
{
|
2015-11-22 01:53:43 +00:00
|
|
|
/* first search (depth 0) */
|
2015-11-21 14:27:35 +00:00
|
|
|
size_t offsetFound = 99999999;
|
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offsetFound, maxSearches, mls);
|
|
|
|
if (ml2 > matchLength)
|
2015-11-22 01:53:43 +00:00
|
|
|
matchLength = ml2, start = ip, offset=offsetFound;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (matchLength < MINMATCH)
|
|
|
|
{
|
|
|
|
ip += ((ip-anchor) >> g_searchStrength) + 1; /* jump faster over incompressible sections */
|
|
|
|
continue;
|
2015-11-21 14:27:35 +00:00
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
/* let's try to find a better solution */
|
2015-11-22 01:42:28 +00:00
|
|
|
if (depth>=1)
|
|
|
|
while (ip<ilimit)
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
ip ++;
|
2015-11-21 14:27:35 +00:00
|
|
|
if ((offset) && (MEM_read32(ip) == MEM_read32(ip - offset_1)))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
2015-11-22 01:42:28 +00:00
|
|
|
size_t mlRep = ZSTD_count(ip+MINMATCH, ip+MINMATCH-offset_1, iend) + MINMATCH;
|
|
|
|
int gain2 = (int)(mlRep * 3);
|
2015-11-04 02:52:54 +00:00
|
|
|
int gain1 = (int)(matchLength*3 - ZSTD_highbit((U32)offset+1) + 1);
|
2015-11-22 01:42:28 +00:00
|
|
|
if ((mlRep >= MINMATCH) && (gain2 > gain1))
|
|
|
|
matchLength = mlRep, offset = 0, start = ip;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
{
|
|
|
|
size_t offset2=999999;
|
2015-11-05 14:00:24 +00:00
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offset2, maxSearches, mls);
|
2015-11-22 01:42:28 +00:00
|
|
|
int gain2 = (int)(ml2*4 - ZSTD_highbit((U32)offset2+1)); /* raw approx */
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 4);
|
2015-11-09 02:19:33 +00:00
|
|
|
if ((ml2 >= MINMATCH) && (gain2 > gain1))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
2015-11-06 17:44:54 +00:00
|
|
|
matchLength = ml2, offset = offset2, start = ip;
|
2015-11-04 02:52:54 +00:00
|
|
|
continue; /* search a better one */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* let's find an even better one */
|
2015-11-22 01:42:28 +00:00
|
|
|
if ((depth==2) && (ip<ilimit))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
ip ++;
|
2015-11-21 14:27:35 +00:00
|
|
|
if ((offset) && (MEM_read32(ip) == MEM_read32(ip - offset_1)))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
size_t ml2 = ZSTD_count(ip+MINMATCH, ip+MINMATCH-offset_1, iend) + MINMATCH;
|
|
|
|
int gain2 = (int)(ml2 * 4);
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 1);
|
2015-11-09 02:19:33 +00:00
|
|
|
if ((ml2 >= MINMATCH) && (gain2 > gain1))
|
2015-11-04 02:52:54 +00:00
|
|
|
matchLength = ml2, offset = 0, start = ip;
|
|
|
|
}
|
|
|
|
{
|
|
|
|
size_t offset2=999999;
|
2015-11-05 14:00:24 +00:00
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offset2, maxSearches, mls);
|
2015-11-06 17:44:54 +00:00
|
|
|
int gain2 = (int)(ml2*4 - ZSTD_highbit((U32)offset2+1)); /* raw approx */
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 7);
|
2015-11-09 02:19:33 +00:00
|
|
|
if ((ml2 >= MINMATCH) && (gain2 > gain1))
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
2015-11-06 17:44:54 +00:00
|
|
|
matchLength = ml2, offset = offset2, start = ip;
|
|
|
|
continue;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break; /* nothing found : store previous solution */
|
|
|
|
}
|
|
|
|
|
2015-11-09 02:19:33 +00:00
|
|
|
/* catch up */
|
2015-11-06 17:44:54 +00:00
|
|
|
if (offset)
|
2015-11-09 02:19:33 +00:00
|
|
|
{
|
2015-11-22 01:53:43 +00:00
|
|
|
while ((start>anchor) && (start>ctx->base+offset) && (start[-1] == start[-1-offset])) /* only search for offset within prefix */
|
2015-11-09 02:19:33 +00:00
|
|
|
{ start--; matchLength++; }
|
2015-11-20 11:03:53 +00:00
|
|
|
offset_2 = offset_1; offset_1 = offset;
|
2015-11-09 02:19:33 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* store sequence */
|
2015-11-21 14:27:35 +00:00
|
|
|
_storeSequence:
|
2015-11-04 02:52:54 +00:00
|
|
|
{
|
|
|
|
size_t litLength = start - anchor;
|
|
|
|
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, matchLength-MINMATCH);
|
2015-11-09 02:19:33 +00:00
|
|
|
anchor = ip = start + matchLength;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
2015-11-20 11:03:53 +00:00
|
|
|
/* check immediate repcode */
|
|
|
|
while ( (ip <= ilimit)
|
|
|
|
&& (MEM_read32(ip) == MEM_read32(ip - offset_2)) )
|
|
|
|
{
|
|
|
|
/* store sequence */
|
|
|
|
matchLength = ZSTD_count(ip+MINMATCH, ip+MINMATCH-offset_2, iend);
|
|
|
|
offset = offset_2;
|
|
|
|
offset_2 = offset_1;
|
|
|
|
offset_1 = offset;
|
|
|
|
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, matchLength);
|
|
|
|
ip += matchLength+MINMATCH;
|
|
|
|
anchor = ip;
|
|
|
|
continue; /* faster when present ... (?) */
|
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Last Literals */
|
|
|
|
{
|
|
|
|
size_t lastLLSize = iend - anchor;
|
|
|
|
memcpy(seqStorePtr->lit, anchor, lastLLSize);
|
|
|
|
seqStorePtr->lit += lastLLSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Final compression stage */
|
2015-11-23 12:34:21 +00:00
|
|
|
return ZSTD_compressSequences(dst, maxDstSize,
|
2015-11-04 02:52:54 +00:00
|
|
|
seqStorePtr, srcSize);
|
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_btlazy2(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-21 14:27:35 +00:00
|
|
|
return ZSTD_compressBlock_lazy_generic(ctx, dst, maxDstSize, src, srcSize, 1, 2);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_lazy2(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
2015-11-21 14:27:35 +00:00
|
|
|
return ZSTD_compressBlock_lazy_generic(ctx, dst, maxDstSize, src, srcSize, 0, 2);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_lazy(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
2015-11-21 14:27:35 +00:00
|
|
|
return ZSTD_compressBlock_lazy_generic(ctx, dst, maxDstSize, src, srcSize, 0, 1);
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock_greedy(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-10-31 11:57:14 +00:00
|
|
|
{
|
2015-11-21 14:27:35 +00:00
|
|
|
return ZSTD_compressBlock_lazy_generic(ctx, dst, maxDstSize, src, srcSize, 0, 0);
|
2015-10-31 11:57:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-22 01:53:43 +00:00
|
|
|
FORCE_INLINE
|
|
|
|
size_t ZSTD_compressBlock_lazy_extDict_generic(ZSTD_CCtx* ctx,
|
|
|
|
void* dst, size_t maxDstSize, const void* src, size_t srcSize,
|
|
|
|
const U32 searchMethod, const U32 depth)
|
|
|
|
{
|
|
|
|
seqStore_t* seqStorePtr = &(ctx->seqStore);
|
|
|
|
const BYTE* const istart = (const BYTE*)src;
|
|
|
|
const BYTE* ip = istart;
|
|
|
|
const BYTE* anchor = istart;
|
|
|
|
const BYTE* const iend = istart + srcSize;
|
|
|
|
const BYTE* const ilimit = iend - 8;
|
|
|
|
const BYTE* const base = ctx->base;
|
|
|
|
const U32 dictLimit = ctx->dictLimit;
|
|
|
|
const BYTE* const prefixStart = base + dictLimit;
|
|
|
|
const BYTE* const dictBase = ctx->dictBase;
|
|
|
|
const BYTE* const dictEnd = dictBase + dictLimit;
|
|
|
|
|
|
|
|
size_t offset_2=REPCODE_STARTVALUE, offset_1=REPCODE_STARTVALUE;
|
|
|
|
const U32 maxSearches = 1 << ctx->params.searchLog;
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
|
|
|
|
typedef size_t (*searchMax_f)(ZSTD_CCtx* zc, const BYTE* ip, const BYTE* iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
U32 maxNbAttempts, U32 matchLengthSearch);
|
2015-11-22 12:24:05 +00:00
|
|
|
searchMax_f searchMax = searchMethod ? ZSTD_BtFindBestMatch_selectMLS : ZSTD_HcFindBestMatch_extDict_selectMLS;
|
2015-11-22 01:53:43 +00:00
|
|
|
|
|
|
|
/* init */
|
|
|
|
ZSTD_resetSeqStore(seqStorePtr);
|
|
|
|
if (((ip-base) - dictLimit) < REPCODE_STARTVALUE) ip += REPCODE_STARTVALUE;
|
|
|
|
|
|
|
|
/* Match Loop */
|
|
|
|
while (ip < ilimit)
|
|
|
|
{
|
|
|
|
size_t matchLength=0;
|
|
|
|
size_t offset=0;
|
|
|
|
const BYTE* start=ip+1;
|
|
|
|
U32 current = (U32)(ip-base);
|
|
|
|
|
|
|
|
/* check repCode */
|
|
|
|
{
|
|
|
|
const U32 repIndex = (U32)(current+1 - offset_1);
|
|
|
|
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
|
|
|
|
const BYTE* const repMatch = repBase + repIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
if ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
|
2015-11-22 13:40:51 +00:00
|
|
|
if (MEM_read32(ip+1) == MEM_read32(repMatch))
|
2015-11-22 01:53:43 +00:00
|
|
|
{
|
|
|
|
/* repcode detected we should take it */
|
|
|
|
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
matchLength = ZSTD_count_2segments(ip+1+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
|
2015-11-22 01:53:43 +00:00
|
|
|
if (depth==0) goto _storeSequence;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
|
|
|
/* first search (depth 0) */
|
|
|
|
size_t offsetFound = 99999999;
|
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offsetFound, maxSearches, mls);
|
|
|
|
if (ml2 > matchLength)
|
|
|
|
matchLength = ml2, start = ip, offset=offsetFound;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (matchLength < MINMATCH)
|
|
|
|
{
|
|
|
|
ip += ((ip-anchor) >> g_searchStrength) + 1; /* jump faster over incompressible sections */
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* let's try to find a better solution */
|
|
|
|
if (depth>=1)
|
|
|
|
while (ip<ilimit)
|
|
|
|
{
|
|
|
|
ip ++;
|
|
|
|
current++;
|
|
|
|
/* check repCode */
|
|
|
|
if (offset)
|
|
|
|
{
|
|
|
|
const U32 repIndex = (U32)(current - offset_1);
|
|
|
|
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
|
|
|
|
const BYTE* const repMatch = repBase + repIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
if ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
|
2015-11-22 13:40:51 +00:00
|
|
|
if (MEM_read32(ip) == MEM_read32(repMatch))
|
2015-11-22 01:53:43 +00:00
|
|
|
{
|
|
|
|
/* repcode detected */
|
|
|
|
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t repLength = ZSTD_count_2segments(ip+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
|
|
|
|
int gain2 = (int)(repLength * 3);
|
|
|
|
int gain1 = (int)(matchLength*3 - ZSTD_highbit((U32)offset+1) + 1);
|
|
|
|
if ((repLength >= MINMATCH) && (gain2 > gain1))
|
|
|
|
matchLength = repLength, offset = 0, start = ip;
|
2015-11-22 01:53:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* search match, depth 1 */
|
|
|
|
{
|
|
|
|
size_t offset2=999999;
|
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offset2, maxSearches, mls);
|
|
|
|
int gain2 = (int)(ml2*4 - ZSTD_highbit((U32)offset2+1)); /* raw approx */
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 4);
|
|
|
|
if ((ml2 >= MINMATCH) && (gain2 > gain1))
|
|
|
|
{
|
|
|
|
matchLength = ml2, offset = offset2, start = ip;
|
|
|
|
continue; /* search a better one */
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* let's find an even better one */
|
|
|
|
if ((depth==2) && (ip<ilimit))
|
|
|
|
{
|
|
|
|
ip ++;
|
|
|
|
current++;
|
|
|
|
/* check repCode */
|
|
|
|
if (offset)
|
|
|
|
{
|
|
|
|
const U32 repIndex = (U32)(current - offset_1);
|
|
|
|
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
|
|
|
|
const BYTE* const repMatch = repBase + repIndex;
|
2015-11-23 12:34:21 +00:00
|
|
|
if ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
|
2015-11-22 13:40:51 +00:00
|
|
|
if (MEM_read32(ip) == MEM_read32(repMatch))
|
2015-11-22 01:53:43 +00:00
|
|
|
{
|
|
|
|
/* repcode detected */
|
|
|
|
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
size_t repLength = ZSTD_count_2segments(ip+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
|
|
|
|
int gain2 = (int)(repLength * 4);
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 1);
|
|
|
|
if ((repLength >= MINMATCH) && (gain2 > gain1))
|
|
|
|
matchLength = repLength, offset = 0, start = ip;
|
2015-11-22 01:53:43 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* search match, depth 2 */
|
|
|
|
{
|
|
|
|
size_t offset2=999999;
|
|
|
|
size_t ml2 = searchMax(ctx, ip, iend, &offset2, maxSearches, mls);
|
|
|
|
int gain2 = (int)(ml2*4 - ZSTD_highbit((U32)offset2+1)); /* raw approx */
|
|
|
|
int gain1 = (int)(matchLength*4 - ZSTD_highbit((U32)offset+1) + 7);
|
|
|
|
if ((ml2 >= MINMATCH) && (gain2 > gain1))
|
|
|
|
{
|
|
|
|
matchLength = ml2, offset = offset2, start = ip;
|
|
|
|
continue;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break; /* nothing found : store previous solution */
|
|
|
|
}
|
|
|
|
|
|
|
|
/* catch up */
|
|
|
|
if (offset)
|
|
|
|
{
|
|
|
|
while ((start>anchor) && (start>prefixStart+offset) && (start[-1] == start[-1-offset])) /* only search for offset within prefix */
|
|
|
|
{ start--; matchLength++; }
|
|
|
|
offset_2 = offset_1; offset_1 = offset;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* store sequence */
|
|
|
|
_storeSequence:
|
|
|
|
{
|
|
|
|
size_t litLength = start - anchor;
|
|
|
|
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, matchLength-MINMATCH);
|
|
|
|
anchor = ip = start + matchLength;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* check immediate repcode */
|
|
|
|
while (ip <= ilimit)
|
|
|
|
{
|
|
|
|
const U32 repIndex = (U32)((ip-base) - offset_2);
|
|
|
|
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
|
|
|
|
const BYTE* const repMatch = repBase + repIndex;
|
2015-11-22 13:40:51 +00:00
|
|
|
if (MEM_read32(ip) == MEM_read32(repMatch))
|
2015-11-22 01:53:43 +00:00
|
|
|
{
|
|
|
|
/* repcode detected we should take it */
|
|
|
|
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
|
2015-11-23 12:34:21 +00:00
|
|
|
matchLength = ZSTD_count_2segments(ip+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
|
|
|
|
offset = offset_2; offset_2 = offset_1; offset_1 = offset; /* swap offset history */
|
2015-11-22 01:53:43 +00:00
|
|
|
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, matchLength-MINMATCH);
|
|
|
|
ip += matchLength;
|
|
|
|
anchor = ip;
|
|
|
|
continue; /* faster when present ... (?) */
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Last Literals */
|
|
|
|
{
|
|
|
|
size_t lastLLSize = iend - anchor;
|
|
|
|
memcpy(seqStorePtr->lit, anchor, lastLLSize);
|
|
|
|
seqStorePtr->lit += lastLLSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Final compression stage */
|
2015-11-23 12:34:21 +00:00
|
|
|
return ZSTD_compressSequences(dst, maxDstSize,
|
2015-11-22 01:53:43 +00:00
|
|
|
seqStorePtr, srcSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t ZSTD_compressBlock_greedy_extDict(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
return ZSTD_compressBlock_lazy_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 0, 0);
|
|
|
|
}
|
|
|
|
|
2015-11-22 02:12:28 +00:00
|
|
|
size_t ZSTD_compressBlock_lazy_extDict(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
return ZSTD_compressBlock_lazy_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 0, 1);
|
|
|
|
}
|
2015-11-22 01:53:43 +00:00
|
|
|
|
2015-11-22 11:22:04 +00:00
|
|
|
size_t ZSTD_compressBlock_lazy2_extDict(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
return ZSTD_compressBlock_lazy_extDict_generic(ctx, dst, maxDstSize, src, srcSize, 0, 2);
|
|
|
|
}
|
|
|
|
|
2015-11-23 12:34:21 +00:00
|
|
|
static size_t ZSTD_compressError(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
(void)ctx; (void)dst; (void)maxDstSize; (void)src; (void)srcSize;
|
|
|
|
return ERROR(mode_unsupported);
|
|
|
|
}
|
|
|
|
|
2015-11-21 14:27:35 +00:00
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
typedef size_t (*ZSTD_blockCompressor) (ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize);
|
2015-11-04 11:05:27 +00:00
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
static ZSTD_blockCompressor ZSTD_selectBlockCompressor(ZSTD_strategy strat, int extDict)
|
2015-10-31 11:57:14 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
if (extDict)
|
2015-11-03 08:49:30 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
switch(strat)
|
|
|
|
{
|
|
|
|
default :
|
|
|
|
case ZSTD_fast:
|
|
|
|
return ZSTD_compressBlock_fast_extDict;
|
|
|
|
case ZSTD_greedy:
|
2015-11-22 01:53:43 +00:00
|
|
|
return ZSTD_compressBlock_greedy_extDict;
|
2015-11-13 10:27:46 +00:00
|
|
|
case ZSTD_lazy:
|
2015-11-22 02:12:28 +00:00
|
|
|
return ZSTD_compressBlock_lazy_extDict;
|
2015-11-13 10:27:46 +00:00
|
|
|
case ZSTD_lazy2:
|
2015-11-22 11:22:04 +00:00
|
|
|
return ZSTD_compressBlock_lazy2_extDict;
|
2015-11-13 10:27:46 +00:00
|
|
|
case ZSTD_btlazy2:
|
2015-11-23 12:34:21 +00:00
|
|
|
return ZSTD_compressError;
|
2015-11-13 10:27:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
switch(strat)
|
|
|
|
{
|
|
|
|
default :
|
|
|
|
case ZSTD_fast:
|
|
|
|
return ZSTD_compressBlock_fast;
|
|
|
|
case ZSTD_greedy:
|
|
|
|
return ZSTD_compressBlock_greedy;
|
|
|
|
case ZSTD_lazy:
|
|
|
|
return ZSTD_compressBlock_lazy;
|
|
|
|
case ZSTD_lazy2:
|
|
|
|
return ZSTD_compressBlock_lazy2;
|
|
|
|
case ZSTD_btlazy2:
|
|
|
|
return ZSTD_compressBlock_btlazy2;
|
|
|
|
}
|
2015-11-03 08:49:30 +00:00
|
|
|
}
|
2015-10-31 11:57:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBlock(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
ZSTD_blockCompressor blockCompressor = ZSTD_selectBlockCompressor(ctx->params.strategy, ctx->lowLimit < ctx->dictLimit);
|
2015-11-19 16:46:29 +00:00
|
|
|
if (srcSize < MIN_CBLOCK_SIZE+3) return 0; /* don't even attempt compression below a certain srcSize */
|
2015-11-04 11:05:27 +00:00
|
|
|
return blockCompressor(ctx, dst, maxDstSize, src, srcSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
static size_t ZSTD_compress_generic (ZSTD_CCtx* ctxPtr,
|
2015-10-22 14:31:46 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
2015-11-04 17:19:39 +00:00
|
|
|
size_t blockSize = BLOCKSIZE;
|
2015-10-22 14:31:46 +00:00
|
|
|
size_t remaining = srcSize;
|
|
|
|
const BYTE* ip = (const BYTE*)src;
|
|
|
|
BYTE* const ostart = (BYTE*)dst;
|
|
|
|
BYTE* op = ostart;
|
2015-11-13 10:27:46 +00:00
|
|
|
const U32 maxDist = 1 << ctxPtr->params.windowLog;
|
|
|
|
//const ZSTD_blockCompressor blockCompressor = ZSTD_selectBlockCompressor(ctxPtr->params.strategy, ctxPtr->lowLimit < ctxPtr->dictLimit);
|
2015-11-01 11:40:22 +00:00
|
|
|
|
2015-11-04 17:19:39 +00:00
|
|
|
while (remaining)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-04 17:19:39 +00:00
|
|
|
size_t cSize;
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-04 22:36:36 +00:00
|
|
|
if (maxDstSize < 3 + MIN_CBLOCK_SIZE) return ERROR(dstSize_tooSmall); /* not enough space to store compressed block */
|
2015-11-04 17:19:39 +00:00
|
|
|
if (remaining < blockSize) blockSize = remaining;
|
2015-11-13 10:27:46 +00:00
|
|
|
|
|
|
|
if ((U32)(ip+blockSize - (ctxPtr->base + ctxPtr->lowLimit)) > maxDist)
|
|
|
|
/* respect windowLog contract */
|
2015-11-23 12:34:21 +00:00
|
|
|
ctxPtr->dictLimit = ctxPtr->lowLimit = (U32)(ip+blockSize - ctxPtr->base) - maxDist;
|
2015-11-13 10:27:46 +00:00
|
|
|
|
|
|
|
//cSize = blockCompressor(ctxPtr, op+3, maxDstSize-3, ip, blockSize);
|
|
|
|
cSize = ZSTD_compressBlock(ctxPtr, op+3, maxDstSize-3, ip, blockSize);
|
2015-10-22 14:31:46 +00:00
|
|
|
if (ZSTD_isError(cSize)) return cSize;
|
|
|
|
|
|
|
|
if (cSize == 0)
|
|
|
|
{
|
2015-11-04 17:19:39 +00:00
|
|
|
cSize = ZSTD_noCompressBlock(op, maxDstSize, ip, blockSize); /* block is not compressible */
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
op[0] = (BYTE)(cSize>>16);
|
|
|
|
op[1] = (BYTE)(cSize>>8);
|
|
|
|
op[2] = (BYTE)cSize;
|
2015-11-19 16:28:35 +00:00
|
|
|
op[0] += (BYTE)(bt_compressed << 6); /* is a compressed block */
|
2015-10-22 14:31:46 +00:00
|
|
|
cSize += 3;
|
|
|
|
}
|
|
|
|
|
2015-11-04 17:19:39 +00:00
|
|
|
remaining -= blockSize;
|
|
|
|
maxDstSize -= cSize;
|
|
|
|
ip += blockSize;
|
2015-10-22 14:31:46 +00:00
|
|
|
op += cSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
return op-ostart;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressContinue (ZSTD_CCtx* ctxPtr,
|
2015-10-29 15:49:43 +00:00
|
|
|
void* dst, size_t dstSize,
|
|
|
|
const void* src, size_t srcSize)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-10-29 15:49:43 +00:00
|
|
|
const BYTE* const ip = (const BYTE*) src;
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
/* preemptive overflow correction */
|
|
|
|
if (ctxPtr->lowLimit > (1<<30) )
|
|
|
|
{
|
|
|
|
U32 correction = ctxPtr->lowLimit;
|
|
|
|
ZSTD_reduceIndex(ctxPtr, correction);
|
|
|
|
ctxPtr->base += correction;
|
|
|
|
ctxPtr->dictBase += correction;
|
|
|
|
ctxPtr->lowLimit -= correction;
|
|
|
|
ctxPtr->dictLimit -= correction;
|
|
|
|
if (ctxPtr->nextToUpdate < correction) ctxPtr->nextToUpdate = 0;
|
|
|
|
else ctxPtr->nextToUpdate -= correction;
|
|
|
|
}
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
/* Check if blocks follow each other */
|
2015-11-13 10:27:46 +00:00
|
|
|
if (src != ctxPtr->nextSrc)
|
|
|
|
{
|
|
|
|
/* not contiguous */
|
|
|
|
ctxPtr->lowLimit = ctxPtr->dictLimit;
|
|
|
|
ctxPtr->dictLimit = (U32)(ctxPtr->nextSrc - ctxPtr->base);
|
|
|
|
ctxPtr->dictBase = ctxPtr->base;
|
|
|
|
ctxPtr->base += ip - ctxPtr->nextSrc;
|
2015-11-22 13:40:51 +00:00
|
|
|
ctxPtr->nextToUpdate = ctxPtr->dictLimit;
|
2015-11-13 10:27:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* input-dictionary overlap */
|
|
|
|
if ((ip+srcSize > ctxPtr->dictBase + ctxPtr->lowLimit) && (ip < ctxPtr->dictBase + ctxPtr->dictLimit))
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
ctxPtr->lowLimit = (U32)(ip + srcSize - ctxPtr->dictBase);
|
|
|
|
if (ctxPtr->lowLimit > ctxPtr->dictLimit) ctxPtr->lowLimit = ctxPtr->dictLimit;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
ctxPtr->nextSrc = ip + srcSize;
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
return ZSTD_compress_generic (ctxPtr, dst, dstSize, src, srcSize);
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBegin_advanced(ZSTD_CCtx* ctx,
|
2015-11-13 10:27:46 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const ZSTD_parameters params,
|
|
|
|
const U64 srcSizeHint)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-10-30 05:40:22 +00:00
|
|
|
size_t errorCode;
|
2015-10-22 14:31:46 +00:00
|
|
|
if (maxDstSize < 4) return ERROR(dstSize_tooSmall);
|
2015-11-11 12:43:58 +00:00
|
|
|
errorCode = ZSTD_resetCCtx_advanced(ctx, params, srcSizeHint);
|
2015-10-30 05:40:22 +00:00
|
|
|
if (ZSTD_isError(errorCode)) return errorCode;
|
2015-11-13 10:27:46 +00:00
|
|
|
|
2015-10-29 15:49:43 +00:00
|
|
|
MEM_writeLE32(dst, ZSTD_magicNumber); /* Write Header */
|
2015-10-22 14:31:46 +00:00
|
|
|
return 4;
|
|
|
|
}
|
|
|
|
|
2015-10-25 13:06:35 +00:00
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBegin(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, int compressionLevel, U64 srcSizeHint)
|
2015-10-25 13:06:35 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
int tableID = ((srcSizeHint-1) > 128 KB); /* intentional underflow for srcSizeHint == 0 */
|
2015-10-29 15:49:43 +00:00
|
|
|
if (compressionLevel<=0) compressionLevel = 1;
|
2015-11-11 12:43:58 +00:00
|
|
|
if (compressionLevel > ZSTD_MAX_CLEVEL) compressionLevel = ZSTD_MAX_CLEVEL;
|
|
|
|
return ZSTD_compressBegin_advanced(ctx, dst, maxDstSize, ZSTD_defaultParameters[tableID][compressionLevel], srcSizeHint);
|
2015-10-25 13:06:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressEnd(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize)
|
2015-10-29 15:49:43 +00:00
|
|
|
{
|
|
|
|
BYTE* op = (BYTE*)dst;
|
|
|
|
|
|
|
|
/* Sanity check */
|
|
|
|
(void)ctx;
|
|
|
|
if (maxDstSize < 3) return ERROR(dstSize_tooSmall);
|
|
|
|
|
|
|
|
/* End of frame */
|
|
|
|
op[0] = (BYTE)(bt_end << 6);
|
|
|
|
op[1] = 0;
|
|
|
|
op[2] = 0;
|
|
|
|
|
|
|
|
return 3;
|
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compress_advanced (ZSTD_CCtx* ctx,
|
2015-10-25 13:06:35 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize,
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_parameters params)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
|
|
|
BYTE* const ostart = (BYTE*)dst;
|
|
|
|
BYTE* op = ostart;
|
2015-10-30 05:40:22 +00:00
|
|
|
size_t oSize;
|
2015-10-22 14:31:46 +00:00
|
|
|
|
|
|
|
/* Header */
|
2015-11-11 12:43:58 +00:00
|
|
|
oSize = ZSTD_compressBegin_advanced(ctx, dst, maxDstSize, params, srcSize);
|
2015-10-22 14:31:46 +00:00
|
|
|
if(ZSTD_isError(oSize)) return oSize;
|
|
|
|
op += oSize;
|
|
|
|
maxDstSize -= oSize;
|
|
|
|
|
|
|
|
/* body (compression) */
|
2015-10-29 16:15:14 +00:00
|
|
|
ctx->base = (const BYTE*)src;
|
2015-11-11 12:43:58 +00:00
|
|
|
oSize = ZSTD_compress_generic (ctx, op, maxDstSize, src, srcSize);
|
2015-10-22 14:31:46 +00:00
|
|
|
if(ZSTD_isError(oSize)) return oSize;
|
|
|
|
op += oSize;
|
|
|
|
maxDstSize -= oSize;
|
|
|
|
|
|
|
|
/* Close frame */
|
2015-11-11 12:43:58 +00:00
|
|
|
oSize = ZSTD_compressEnd(ctx, op, maxDstSize);
|
2015-10-22 14:31:46 +00:00
|
|
|
if(ZSTD_isError(oSize)) return oSize;
|
|
|
|
op += oSize;
|
|
|
|
|
|
|
|
return (op - ostart);
|
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressCCtx (ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize, int compressionLevel)
|
2015-10-25 13:06:35 +00:00
|
|
|
{
|
2015-11-09 15:38:17 +00:00
|
|
|
const int tableID = (srcSize > 128 KB);
|
2015-11-11 12:43:58 +00:00
|
|
|
if (compressionLevel < 1) compressionLevel = 1;
|
|
|
|
if (compressionLevel > ZSTD_MAX_CLEVEL) compressionLevel = ZSTD_MAX_CLEVEL;
|
|
|
|
return ZSTD_compress_advanced(ctx, dst, maxDstSize, src, srcSize, ZSTD_defaultParameters[tableID][compressionLevel]);
|
2015-10-25 13:06:35 +00:00
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compress(void* dst, size_t maxDstSize, const void* src, size_t srcSize, int compressionLevel)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-10-29 21:02:40 +00:00
|
|
|
size_t result;
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_CCtx ctxBody;
|
2015-10-29 17:41:45 +00:00
|
|
|
memset(&ctxBody, 0, sizeof(ctxBody));
|
2015-11-11 12:43:58 +00:00
|
|
|
result = ZSTD_compressCCtx(&ctxBody, dst, maxDstSize, src, srcSize, compressionLevel);
|
2015-11-13 10:27:46 +00:00
|
|
|
free(ctxBody.workSpace); /* can't free ctxBody, since it's on stack; take care of heap content */
|
2015-10-29 21:02:40 +00:00
|
|
|
return result;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|