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 */
|
|
|
|
#else
|
|
|
|
# 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"
|
2016-01-25 03:10:46 +00:00
|
|
|
#include "huff0_static.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
|
|
|
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
/* *************************************
|
|
|
|
* Helper functions
|
|
|
|
***************************************/
|
|
|
|
unsigned ZSTD_maxCLevel(void) { return ZSTD_MAX_CLEVEL; }
|
|
|
|
|
|
|
|
size_t ZSTD_compressBound(size_t srcSize) { return FSE_compressBound(srcSize) + 12; }
|
|
|
|
|
|
|
|
|
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
|
|
|
***************************************/
|
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 */
|
2016-01-07 14:35:18 +00:00
|
|
|
U32 stage;
|
2015-11-11 12:43:58 +00:00
|
|
|
ZSTD_parameters params;
|
2015-10-29 17:41:45 +00:00
|
|
|
void* workSpace;
|
|
|
|
size_t workSpaceSize;
|
2015-12-02 13:00:45 +00:00
|
|
|
size_t blockSize;
|
2016-01-07 14:35:18 +00:00
|
|
|
size_t hbSize;
|
2016-01-08 16:27:50 +00:00
|
|
|
char headerBuffer[ZSTD_frameHeaderSize_max];
|
2016-01-07 14:35:18 +00:00
|
|
|
|
2015-10-29 17:41:45 +00:00
|
|
|
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;
|
2016-01-26 02:14:20 +00:00
|
|
|
HUF_CElt* hufTable;
|
|
|
|
U32 flagHufTable;
|
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-25 13:42:45 +00:00
|
|
|
void ZSTD_validateParams(ZSTD_parameters* params)
|
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-28 15:03:22 +00:00
|
|
|
if (MEM_32bits()) if (params->windowLog > 25) params->windowLog = 25; /* 32 bits mode cannot flush > 24 bits */
|
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-25 13:42:45 +00:00
|
|
|
if ((params->srcSize > 0) && (params->srcSize < (1<<ZSTD_WINDOWLOG_MAX)))
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-25 13:42:45 +00:00
|
|
|
U32 srcLog = ZSTD_highbit((U32)(params->srcSize)-1) + 1;
|
2015-11-04 11:05:27 +00:00
|
|
|
if (params->windowLog > srcLog) params->windowLog = srcLog;
|
|
|
|
}
|
|
|
|
|
2015-11-28 15:03:22 +00:00
|
|
|
if (params->windowLog < ZSTD_WINDOWLOG_ABSOLUTEMIN) params->windowLog = ZSTD_WINDOWLOG_ABSOLUTEMIN; /* required for frame header */
|
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-25 13:42:45 +00:00
|
|
|
ZSTD_parameters params)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-12-02 13:00:45 +00:00
|
|
|
/* note : params considered validated here */
|
|
|
|
const size_t blockSize = MIN(BLOCKSIZE, (size_t)1 << params.windowLog);
|
|
|
|
|
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);
|
2016-01-26 02:14:20 +00:00
|
|
|
const size_t neededSpace = tableSpace + (256*sizeof(U32)) + (3*blockSize);
|
2015-10-29 17:41:45 +00:00
|
|
|
if (zc->workSpaceSize < neededSpace)
|
2015-10-29 15:49:43 +00:00
|
|
|
{
|
2015-10-29 17:41:45 +00:00
|
|
|
free(zc->workSpace);
|
|
|
|
zc->workSpace = malloc(neededSpace);
|
2015-10-30 05:40:22 +00:00
|
|
|
if (zc->workSpace == NULL) return ERROR(memory_allocation);
|
2016-01-26 02:14:20 +00:00
|
|
|
zc->workSpaceSize = neededSpace;
|
2015-10-29 15:49:43 +00:00
|
|
|
}
|
2016-01-26 02:14:20 +00:00
|
|
|
memset(zc->workSpace, 0, tableSpace ); /* reset only tables */
|
2015-11-06 09:52:17 +00:00
|
|
|
zc->hashTable = (U32*)(zc->workSpace);
|
2015-11-05 16:32:18 +00:00
|
|
|
zc->contentTable = zc->hashTable + ((size_t)1 << params.hashLog);
|
2016-01-26 02:14:20 +00:00
|
|
|
zc->seqStore.buffer = zc->contentTable + ((size_t)1 << contentLog);
|
|
|
|
zc->hufTable = (HUF_CElt*)zc->seqStore.buffer;
|
|
|
|
zc->flagHufTable = 0;
|
|
|
|
zc->seqStore.buffer = (U32*)(zc->seqStore.buffer) + 256;
|
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-12-02 13:00:45 +00:00
|
|
|
zc->blockSize = blockSize;
|
2015-10-22 14:31:46 +00:00
|
|
|
zc->seqStore.offsetStart = (U32*) (zc->seqStore.buffer);
|
2015-12-02 13:00:45 +00:00
|
|
|
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);
|
2016-01-07 14:35:18 +00:00
|
|
|
zc->hbSize = 0;
|
2016-01-08 16:27:50 +00:00
|
|
|
zc->stage = 0;
|
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
|
|
|
|
2016-01-26 14:58:49 +00:00
|
|
|
/*! ZSTD_copyCCtx
|
|
|
|
* Duplicate an existing context @srcCCtx into another one @dstCCtx.
|
|
|
|
* Only works during stage 0 (i.e. before first call to ZSTD_compressContinue())
|
|
|
|
* @return : 0, or an error code */
|
|
|
|
size_t ZSTD_copyCCtx(ZSTD_CCtx* dstCCtx, const ZSTD_CCtx* srcCCtx)
|
|
|
|
{
|
|
|
|
const U32 contentLog = (srcCCtx->params.strategy == ZSTD_fast) ? 1 : srcCCtx->params.contentLog;
|
|
|
|
const size_t tableSpace = ((1 << contentLog) + (1 << srcCCtx->params.hashLog)) * sizeof(U32);
|
|
|
|
|
|
|
|
if (srcCCtx->stage!=0) return ERROR(stage_wrong);
|
|
|
|
|
|
|
|
ZSTD_resetCCtx_advanced(dstCCtx, srcCCtx->params);
|
|
|
|
|
|
|
|
/* copy tables */
|
|
|
|
memcpy(dstCCtx->hashTable, srcCCtx->hashTable, tableSpace);
|
|
|
|
|
|
|
|
/* copy frame header */
|
|
|
|
dstCCtx->hbSize = srcCCtx->hbSize;
|
|
|
|
memcpy(dstCCtx->headerBuffer , srcCCtx->headerBuffer, srcCCtx->hbSize);
|
|
|
|
|
|
|
|
/* copy dictionary pointers */
|
|
|
|
dstCCtx->nextToUpdate= srcCCtx->nextToUpdate;
|
|
|
|
dstCCtx->nextSrc = srcCCtx->nextSrc;
|
|
|
|
dstCCtx->base = srcCCtx->base;
|
|
|
|
dstCCtx->dictBase = srcCCtx->dictBase;
|
|
|
|
dstCCtx->dictLimit = srcCCtx->dictLimit;
|
|
|
|
dstCCtx->lowLimit = srcCCtx->lowLimit;
|
|
|
|
|
|
|
|
dstCCtx->flagHufTable = srcCCtx->flagHufTable;
|
|
|
|
if (dstCCtx->flagHufTable)
|
|
|
|
memcpy(dstCCtx->hufTable, srcCCtx->hufTable, 256*4);
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-25 13:42:45 +00:00
|
|
|
/** ZSTD_reduceIndex
|
|
|
|
* rescale indexes to avoid future overflow (indexes are U32) */
|
2015-11-13 10:27:46 +00:00
|
|
|
static void ZSTD_reduceIndex (ZSTD_CCtx* zc,
|
|
|
|
const U32 reducerValue)
|
|
|
|
{
|
2015-11-25 13:42:45 +00:00
|
|
|
const U32 contentLog = (zc->params.strategy == ZSTD_fast) ? 1 : zc->params.contentLog;
|
2015-11-13 10:27:46 +00:00
|
|
|
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
|
|
|
|
*********************************************************/
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
/* Block format description
|
|
|
|
|
|
|
|
Block = Literal Section - Sequences Section
|
|
|
|
Prerequisite : size of (compressed) block, maximum size of regenerated data
|
|
|
|
|
|
|
|
1) Literal Section
|
|
|
|
|
|
|
|
1.1) Header : 1-5 bytes
|
|
|
|
flags: 2 bits
|
|
|
|
00 compressed by Huff0
|
|
|
|
01 unused
|
|
|
|
10 is Raw (uncompressed)
|
|
|
|
11 is Rle
|
|
|
|
Note : using 01 => Huff0 with precomputed table ?
|
|
|
|
Note : delta map ? => compressed ?
|
|
|
|
|
|
|
|
1.1.1) Huff0-compressed literal block : 3-5 bytes
|
2016-01-25 03:10:46 +00:00
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10) => single stream
|
2016-01-23 18:28:41 +00:00
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10)
|
2016-01-25 03:22:03 +00:00
|
|
|
srcSize < 16KB => 4 bytes (2-2-14-14)
|
2016-01-23 18:28:41 +00:00
|
|
|
else => 5 bytes (2-2-18-18)
|
|
|
|
big endian convention
|
|
|
|
|
|
|
|
1.1.2) Raw (uncompressed) literal block header : 1-3 bytes
|
|
|
|
size : 5 bits: (IS_RAW<<6) + (0<<4) + size
|
|
|
|
12 bits: (IS_RAW<<6) + (2<<4) + (size>>8)
|
|
|
|
size&255
|
|
|
|
20 bits: (IS_RAW<<6) + (3<<4) + (size>>16)
|
|
|
|
size>>8&255
|
|
|
|
size&255
|
|
|
|
|
|
|
|
1.1.3) Rle (repeated single byte) literal block header : 1-3 bytes
|
|
|
|
size : 5 bits: (IS_RLE<<6) + (0<<4) + size
|
|
|
|
12 bits: (IS_RLE<<6) + (2<<4) + (size>>8)
|
|
|
|
size&255
|
|
|
|
20 bits: (IS_RLE<<6) + (3<<4) + (size>>16)
|
|
|
|
size>>8&255
|
|
|
|
size&255
|
|
|
|
|
2016-01-25 03:22:03 +00:00
|
|
|
1.1.4) Huff0-compressed literal block, using precomputed CTables : 3-5 bytes
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10) => single stream
|
|
|
|
srcSize < 1 KB => 3 bytes (2-2-10-10)
|
|
|
|
srcSize < 16KB => 4 bytes (2-2-14-14)
|
|
|
|
else => 5 bytes (2-2-18-18)
|
|
|
|
big endian convention
|
|
|
|
|
|
|
|
1- CTable available (stored into workspace ?)
|
2016-01-26 02:14:20 +00:00
|
|
|
2- Small input (fast heuristic ? Full comparison ? depend on clevel ?)
|
2016-01-25 03:22:03 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
|
|
|
|
1.2) Literal block content
|
|
|
|
|
|
|
|
1.2.1) Huff0 block, using sizes from header
|
|
|
|
See Huff0 format
|
|
|
|
|
|
|
|
1.2.2) Raw content
|
|
|
|
|
|
|
|
1.2.3) single byte
|
|
|
|
|
|
|
|
1.2.4) _usingDTable variant ?
|
|
|
|
|
|
|
|
|
|
|
|
2) Sequences section
|
|
|
|
TO DO
|
|
|
|
*/
|
2015-11-11 20:38:21 +00:00
|
|
|
|
|
|
|
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;
|
2016-01-23 18:28:41 +00:00
|
|
|
const U32 flSize = 1 + (srcSize>31) + (srcSize>4095);
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
if (srcSize + flSize > maxDstSize) return ERROR(dstSize_tooSmall);
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
switch(flSize)
|
|
|
|
{
|
|
|
|
case 1: /* 2 - 1 - 5 */
|
|
|
|
ostart[0] = (BYTE)((IS_RAW<<6) + (0<<5) + srcSize);
|
|
|
|
break;
|
|
|
|
case 2: /* 2 - 2 - 12 */
|
|
|
|
ostart[0] = (BYTE)((IS_RAW<<6) + (2<<4) + (srcSize >> 8));
|
|
|
|
ostart[1] = (BYTE)srcSize;
|
|
|
|
break;
|
|
|
|
default: /*note : should not be necessary : flSize is within {1,2,3} */
|
|
|
|
case 3: /* 2 - 2 - 20 */
|
|
|
|
ostart[0] = (BYTE)((IS_RAW<<6) + (3<<4) + (srcSize >> 16));
|
|
|
|
ostart[1] = (BYTE)(srcSize>>8);
|
|
|
|
ostart[2] = (BYTE)srcSize;
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
memcpy(ostart + flSize, src, srcSize);
|
|
|
|
return srcSize + flSize;
|
2015-11-11 20:38:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
static size_t ZSTD_compressRleLiteralsBlock (void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
BYTE* const ostart = (BYTE* const)dst;
|
2016-01-23 18:28:41 +00:00
|
|
|
U32 flSize = 1 + (srcSize>31) + (srcSize>4095);
|
|
|
|
|
|
|
|
(void)maxDstSize; /* maxDstSize guaranteed to be >=4, hence large enough ? */
|
|
|
|
|
|
|
|
switch(flSize)
|
|
|
|
{
|
|
|
|
case 1: /* 2 - 1 - 5 */
|
|
|
|
ostart[0] = (BYTE)((IS_RLE<<6) + (0<<5) + srcSize);
|
|
|
|
break;
|
|
|
|
case 2: /* 2 - 2 - 12 */
|
|
|
|
ostart[0] = (BYTE)((IS_RLE<<6) + (2<<4) + (srcSize >> 8));
|
|
|
|
ostart[1] = (BYTE)srcSize;
|
|
|
|
break;
|
|
|
|
default: /*note : should not be necessary : flSize is necessary within {1,2,3} */
|
|
|
|
case 3: /* 2 - 2 - 20 */
|
|
|
|
ostart[0] = (BYTE)((IS_RLE<<6) + (3<<4) + (srcSize >> 16));
|
|
|
|
ostart[1] = (BYTE)(srcSize>>8);
|
|
|
|
ostart[2] = (BYTE)srcSize;
|
|
|
|
break;
|
|
|
|
}
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
ostart[flSize] = *(const BYTE*)src;
|
|
|
|
return flSize+1;
|
2015-11-11 20:38:21 +00:00
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
size_t ZSTD_minGain(size_t srcSize) { return (srcSize >> 6) + 2; }
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
static size_t ZSTD_compressLiterals (ZSTD_CCtx* zc,
|
|
|
|
void* dst, size_t maxDstSize,
|
2015-11-11 20:38:21 +00:00
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
const size_t minGain = ZSTD_minGain(srcSize);
|
|
|
|
BYTE* const ostart = (BYTE*)dst;
|
2016-01-26 02:14:20 +00:00
|
|
|
const size_t lhSize = 3 + (srcSize >= 1 KB) + (srcSize >= 16 KB);
|
2016-01-25 03:10:46 +00:00
|
|
|
U32 singleStream = srcSize < 256;
|
2016-01-26 02:14:20 +00:00
|
|
|
U32 hType = IS_HUF;
|
2016-01-23 18:28:41 +00:00
|
|
|
size_t clitSize;
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
if (maxDstSize < 4) return ERROR(dstSize_tooSmall); /* not enough space for compression */
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
if (zc->flagHufTable && (lhSize==3))
|
|
|
|
{
|
|
|
|
hType = IS_PCH;
|
|
|
|
singleStream = 1;
|
|
|
|
clitSize = HUF_compress1X_usingCTable(ostart+lhSize, maxDstSize-lhSize, src, srcSize, zc->hufTable);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
clitSize = singleStream ? HUF_compress1X(ostart+lhSize, maxDstSize-lhSize, src, srcSize, 255, 12)
|
|
|
|
: HUF_compress2 (ostart+lhSize, maxDstSize-lhSize, src, srcSize, 255, 12);
|
|
|
|
}
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
if ((clitSize==0) || (clitSize >= srcSize - minGain)) return ZSTD_noCompressLiterals(dst, maxDstSize, src, srcSize);
|
|
|
|
if (clitSize==1) return ZSTD_compressRleLiteralsBlock(dst, maxDstSize, src, srcSize);
|
2015-11-11 20:38:21 +00:00
|
|
|
|
|
|
|
/* Build header */
|
2016-01-23 18:28:41 +00:00
|
|
|
switch(lhSize)
|
2015-11-11 20:38:21 +00:00
|
|
|
{
|
2016-01-23 18:28:41 +00:00
|
|
|
case 3: /* 2 - 2 - 10 - 10 */
|
2016-01-26 02:14:20 +00:00
|
|
|
ostart[0] = (BYTE)((srcSize>>6) + (singleStream << 4) + (hType<<6));
|
2016-01-23 18:28:41 +00:00
|
|
|
ostart[1] = (BYTE)((srcSize<<2) + (clitSize>>8));
|
|
|
|
ostart[2] = (BYTE)(clitSize);
|
|
|
|
break;
|
|
|
|
case 4: /* 2 - 2 - 14 - 14 */
|
2016-01-26 02:14:20 +00:00
|
|
|
ostart[0] = (BYTE)((srcSize>>10) + (2<<4) + (hType<<6));
|
2016-01-23 18:28:41 +00:00
|
|
|
ostart[1] = (BYTE)(srcSize>> 2);
|
|
|
|
ostart[2] = (BYTE)((srcSize<<6) + (clitSize>>8));
|
|
|
|
ostart[3] = (BYTE)(clitSize);
|
|
|
|
break;
|
|
|
|
default: /* should not be necessary, lhSize is {3,4,5} */
|
|
|
|
case 5: /* 2 - 2 - 18 - 18 */
|
2016-01-26 02:14:20 +00:00
|
|
|
ostart[0] = (BYTE)((srcSize>>14) + (3<<4) + (hType<<6));
|
2016-01-23 18:28:41 +00:00
|
|
|
ostart[1] = (BYTE)(srcSize>>6);
|
|
|
|
ostart[2] = (BYTE)((srcSize<<2) + (clitSize>>16));
|
|
|
|
ostart[3] = (BYTE)(clitSize>>8);
|
|
|
|
ostart[4] = (BYTE)(clitSize);
|
|
|
|
break;
|
2015-11-11 20:38:21 +00:00
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
return lhSize+clitSize;
|
2015-11-11 20:38:21 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-25 03:10:46 +00:00
|
|
|
#define LITERAL_NOENTROPY 63 /* don't even attempt to compress literals below this threshold (cheap heuristic) */
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
size_t ZSTD_compressSequences(ZSTD_CCtx* zc,
|
|
|
|
void* dst, size_t maxDstSize,
|
2015-11-11 20:38:21 +00:00
|
|
|
size_t srcSize)
|
|
|
|
{
|
2016-01-26 02:14:20 +00:00
|
|
|
const seqStore_t* seqStorePtr = &(zc->seqStore);
|
2015-11-11 20:38:21 +00:00
|
|
|
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;
|
2016-01-26 02:14:20 +00:00
|
|
|
const size_t minLitSize = zc->flagHufTable ? 6 : LITERAL_NOENTROPY;
|
2015-11-11 20:38:21 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
if (litSize <= minLitSize)
|
2015-11-11 20:38:21 +00:00
|
|
|
cSize = ZSTD_noCompressLiterals(op, maxDstSize, op_lit_start, litSize);
|
|
|
|
else
|
2016-01-26 02:14:20 +00:00
|
|
|
cSize = ZSTD_compressLiterals(zc, op, maxDstSize, op_lit_start, litSize);
|
2015-11-11 20:38:21 +00:00
|
|
|
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-12-04 16:16:37 +00:00
|
|
|
BIT_addBits(&blockStream, offset, nbBits); /* 31 */ /* 42 */ /* 24 bits max in 32-bits mode */
|
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 );
|
2015-12-09 08:05:22 +00:00
|
|
|
return (unsigned)(r>>3);
|
2015-11-11 20:38:21 +00:00
|
|
|
# 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 );
|
2015-12-09 08:05:22 +00:00
|
|
|
return (unsigned)(r>>3);
|
2015-11-11 20:38:21 +00:00
|
|
|
# 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 */
|
|
|
|
{
|
2015-12-01 19:58:24 +00:00
|
|
|
if (MEM_64bits())
|
2015-11-11 20:38:21 +00:00
|
|
|
{
|
|
|
|
# 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
|
|
|
|
***************************************/
|
|
|
|
|
2015-12-04 16:16:37 +00:00
|
|
|
#define FILLHASHSTEP 3
|
|
|
|
static void ZSTD_fillHashTable (ZSTD_CCtx* zc, const void* end, const U32 mls)
|
|
|
|
{
|
|
|
|
U32* const hashTable = zc->hashTable;
|
|
|
|
const U32 hBits = zc->params.hashLog;
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const BYTE* ip = base + zc->nextToUpdate;
|
|
|
|
const BYTE* const iend = (const BYTE*) end;
|
|
|
|
|
|
|
|
while(ip <= iend)
|
|
|
|
{
|
|
|
|
hashTable[ZSTD_hashPtr(ip, hBits, mls)] = (U32)(ip - base);
|
|
|
|
ip += FILLHASHSTEP;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-05 16:32:18 +00:00
|
|
|
FORCE_INLINE
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_fast_generic(ZSTD_CCtx* zc,
|
2015-11-13 10:27:46 +00:00
|
|
|
const void* src, size_t srcSize,
|
|
|
|
const U32 mls)
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
2015-12-04 16:16:37 +00:00
|
|
|
U32* const hashTable = zc->hashTable;
|
2015-11-24 13:06:07 +00:00
|
|
|
const U32 hBits = zc->params.hashLog;
|
|
|
|
seqStore_t* seqStorePtr = &(zc->seqStore);
|
|
|
|
const BYTE* const base = zc->base;
|
2015-11-05 16:32:18 +00:00
|
|
|
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-12-29 13:29:08 +00:00
|
|
|
const U32 lowIndex = zc->dictLimit;
|
|
|
|
const BYTE* const lowest = base + lowIndex;
|
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-12-05 08:23:53 +00:00
|
|
|
if (ip < lowest+4)
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
2015-12-29 13:29:08 +00:00
|
|
|
hashTable[ZSTD_hashPtr(lowest+1, hBits, mls)] = lowIndex+1;
|
|
|
|
hashTable[ZSTD_hashPtr(lowest+2, hBits, mls)] = lowIndex+2;
|
|
|
|
hashTable[ZSTD_hashPtr(lowest+3, hBits, mls)] = lowIndex+3;
|
2015-12-05 08:23:53 +00:00
|
|
|
ip = lowest+4;
|
2015-11-05 16:32:18 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/* 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-12-29 13:29:08 +00:00
|
|
|
const U32 matchIndex = hashTable[h];
|
|
|
|
const BYTE* match = base + matchIndex;
|
2016-01-02 00:16:28 +00:00
|
|
|
const U32 current = (U32)(ip-base);
|
|
|
|
hashTable[h] = current; /* update hash table */
|
2015-11-05 16:32:18 +00:00
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
if (MEM_read32(ip+1-offset_1) == MEM_read32(ip+1)) /* note : by construction, offset_1 <= current */
|
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
|
|
|
|
{
|
2015-12-29 13:29:08 +00:00
|
|
|
if ( (matchIndex <= lowIndex) ||
|
2015-11-20 11:46:08 +00:00
|
|
|
(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-26 10:43:00 +00:00
|
|
|
ZSTD_storeSeq(seqStorePtr, ip-anchor, anchor, offset, mlCode);
|
2015-11-23 12:34:21 +00:00
|
|
|
ip += mlCode + MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
anchor = ip;
|
|
|
|
|
|
|
|
if (ip <= ilimit)
|
|
|
|
{
|
|
|
|
/* Fill Table */
|
2016-01-07 14:35:18 +00:00
|
|
|
hashTable[ZSTD_hashPtr(base+current+2, hBits, mls)] = current+2; /* here because current+2 could be > iend-8 */
|
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)
|
|
|
|
&& (MEM_read32(ip) == MEM_read32(ip - offset_2)) )
|
|
|
|
{
|
|
|
|
/* store sequence */
|
2015-11-23 13:23:47 +00:00
|
|
|
size_t rlCode = ZSTD_count(ip+MINMATCH, ip+MINMATCH-offset_2, iend);
|
2015-11-20 11:46:08 +00:00
|
|
|
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);
|
2015-11-23 13:23:47 +00:00
|
|
|
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, rlCode);
|
|
|
|
ip += rlCode+MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_fast(ZSTD_CCtx* ctx,
|
|
|
|
const void* src, size_t srcSize)
|
2015-11-05 16:32:18 +00:00
|
|
|
{
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
switch(mls)
|
|
|
|
{
|
|
|
|
default:
|
|
|
|
case 4 :
|
2016-01-23 18:28:41 +00:00
|
|
|
ZSTD_compressBlock_fast_generic(ctx, src, srcSize, 4); return;
|
2015-11-05 16:32:18 +00:00
|
|
|
case 5 :
|
2016-01-23 18:28:41 +00:00
|
|
|
ZSTD_compressBlock_fast_generic(ctx, src, srcSize, 5); return;
|
2015-11-05 16:32:18 +00:00
|
|
|
case 6 :
|
2016-01-23 18:28:41 +00:00
|
|
|
ZSTD_compressBlock_fast_generic(ctx, src, srcSize, 6); return;
|
2015-11-05 16:32:18 +00:00
|
|
|
case 7 :
|
2016-01-23 18:28:41 +00:00
|
|
|
ZSTD_compressBlock_fast_generic(ctx, src, srcSize, 7); return;
|
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
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_fast_extDict_generic(ZSTD_CCtx* ctx,
|
|
|
|
const void* src, size_t srcSize,
|
|
|
|
const U32 mls)
|
2015-11-13 10:27:46 +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 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-26 10:43:00 +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);
|
|
|
|
ip += mlCode + MINMATCH;
|
2015-11-20 11:46:08 +00:00
|
|
|
anchor = ip;
|
|
|
|
|
|
|
|
if (ip <= ilimit)
|
|
|
|
{
|
2015-11-26 10:43:00 +00:00
|
|
|
/* Fill Table */
|
2015-11-23 15:17:21 +00:00
|
|
|
hashTable[ZSTD_hashPtr(base+current+2, hBits, mls)] = current+2;
|
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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_fast_extDict(ZSTD_CCtx* ctx,
|
2015-11-13 10:27:46 +00:00
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
const U32 mls = ctx->params.searchLength;
|
|
|
|
switch(mls)
|
|
|
|
{
|
|
|
|
default:
|
|
|
|
case 4 :
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_fast_extDict_generic(ctx, src, srcSize, 4); return;
|
2015-11-13 10:27:46 +00:00
|
|
|
case 5 :
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_fast_extDict_generic(ctx, src, srcSize, 5); return;
|
2015-11-13 10:27:46 +00:00
|
|
|
case 6 :
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_fast_extDict_generic(ctx, src, srcSize, 6); return;
|
2015-11-13 10:27:46 +00:00
|
|
|
case 7 :
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_fast_extDict_generic(ctx, src, srcSize, 7); return;
|
2015-11-13 10:27:46 +00:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
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
|
|
|
***************************************/
|
2016-01-01 06:29:39 +00:00
|
|
|
/** ZSTD_insertBt1 : add one or multiple positions to tree
|
2015-11-26 10:43:00 +00:00
|
|
|
* @ip : assumed <= iend-8
|
2015-11-23 13:23:47 +00:00
|
|
|
* @return : nb of positions added */
|
2016-01-01 06:29:39 +00:00
|
|
|
static U32 ZSTD_insertBt1(ZSTD_CCtx* zc, const BYTE* const ip, const U32 mls, const BYTE* const iend, U32 nbCompares,
|
|
|
|
U32 extDict)
|
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;
|
2016-01-01 06:29:39 +00:00
|
|
|
const BYTE* const dictBase = zc->dictBase;
|
|
|
|
const U32 dictLimit = zc->dictLimit;
|
|
|
|
const BYTE* const dictEnd = dictBase + dictLimit;
|
|
|
|
const BYTE* const prefixStart = base + dictLimit;
|
2015-11-07 00:13:31 +00:00
|
|
|
const BYTE* match = base + matchIndex;
|
2015-12-11 09:44:07 +00:00
|
|
|
const 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);
|
2016-01-16 23:12:55 +00:00
|
|
|
U32* largerPtr = smallerPtr + 1;
|
2015-11-04 11:05:27 +00:00
|
|
|
U32 dummy32; /* to be nullified at the end */
|
2015-11-23 14:29:15 +00:00
|
|
|
const U32 windowLow = zc->lowLimit;
|
2015-12-31 18:08:44 +00:00
|
|
|
U32 matchEndIdx = current+8;
|
2016-01-21 10:57:45 +00:00
|
|
|
U32 predictedSmall = *(bt + 2*((current-1)&btMask) + 0);
|
|
|
|
U32 predictedLarge = *(bt + 2*((current-1)&btMask) + 1);
|
|
|
|
predictedSmall += (predictedSmall>0);
|
|
|
|
predictedLarge += (predictedLarge>0);
|
2015-11-07 00:13:31 +00:00
|
|
|
|
2015-12-11 09:44:07 +00:00
|
|
|
hashTable[h] = current; /* Update Hash Table */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
while (nbCompares-- && (matchIndex > windowLow))
|
|
|
|
{
|
|
|
|
U32* nextPtr = bt + 2*(matchIndex & btMask);
|
2016-01-16 23:12:55 +00:00
|
|
|
const U32* predictPtr = bt + 2*((matchIndex-1) & btMask); /* written this way, as bt is a roll buffer */
|
2015-11-04 02:52:54 +00:00
|
|
|
size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
|
|
|
|
|
2016-01-16 23:12:55 +00:00
|
|
|
if (matchIndex == predictedSmall)
|
|
|
|
{ /* no need to check length, result known */
|
|
|
|
*smallerPtr = matchIndex;
|
|
|
|
if (matchIndex <= btLow) { smallerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
|
|
|
smallerPtr = nextPtr+1; /* new "smaller" => larger of match */
|
|
|
|
matchIndex = nextPtr[1]; /* new matchIndex larger than previous (closer to current) */
|
2016-01-21 10:57:45 +00:00
|
|
|
predictedSmall = predictPtr[1] + (predictPtr[1]>0);
|
2016-01-16 23:12:55 +00:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (matchIndex == predictedLarge)
|
|
|
|
{
|
|
|
|
*largerPtr = matchIndex;
|
|
|
|
if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
|
|
|
|
largerPtr = nextPtr;
|
|
|
|
matchIndex = nextPtr[0];
|
2016-01-21 10:57:45 +00:00
|
|
|
predictedLarge = predictPtr[0] + (predictPtr[0]>0);
|
2016-01-16 23:12:55 +00:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
|
2016-01-01 06:29:39 +00:00
|
|
|
if ((!extDict) || (matchIndex+matchLength >= dictLimit))
|
|
|
|
{
|
|
|
|
match = base + matchIndex;
|
|
|
|
if (match[matchLength] == ip[matchLength])
|
|
|
|
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iend) +1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
match = dictBase + matchIndex;
|
|
|
|
matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
|
|
|
|
if (matchIndex+matchLength >= dictLimit)
|
|
|
|
match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
|
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2015-12-29 21:26:09 +00:00
|
|
|
if (matchLength > matchEndIdx - matchIndex)
|
2015-12-29 22:40:02 +00:00
|
|
|
matchEndIdx = matchIndex + (U32)matchLength;
|
2015-12-29 21:26:09 +00:00
|
|
|
|
2015-11-04 11:05:27 +00:00
|
|
|
if (ip+matchLength == iend) /* equal : no way to know if inf or sup */
|
2016-01-01 06:29:39 +00:00
|
|
|
break; /* drop , to guarantee consistency ; miss a bit of compression, but other solutions can corrupt the tree */
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2016-01-01 06:29:39 +00:00
|
|
|
if (match[matchLength] < ip[matchLength]) /* necessarily within correct buffer */
|
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-12-31 18:08:44 +00:00
|
|
|
return (matchEndIdx > current + 8) ? matchEndIdx - current - 8 : 1;
|
2015-11-04 02:52:54 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-01 06:47:58 +00:00
|
|
|
static void ZSTD_updateTree(ZSTD_CCtx* zc, const BYTE* const ip, const BYTE* const iend, const U32 nbCompares, const U32 mls)
|
|
|
|
{
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 target = (U32)(ip - base);
|
|
|
|
U32 idx = zc->nextToUpdate;
|
|
|
|
|
|
|
|
for( ; idx < target ; )
|
|
|
|
idx += ZSTD_insertBt1(zc, base+idx, mls, iend, nbCompares, 0);
|
|
|
|
}
|
|
|
|
|
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,
|
2016-01-01 06:47:58 +00:00
|
|
|
U32 nbCompares, const U32 mls,
|
|
|
|
U32 extDict)
|
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;
|
2016-01-01 06:47:58 +00:00
|
|
|
const BYTE* const dictBase = zc->dictBase;
|
|
|
|
const U32 dictLimit = zc->dictLimit;
|
|
|
|
const BYTE* const dictEnd = dictBase + dictLimit;
|
|
|
|
const BYTE* const prefixStart = base + dictLimit;
|
2015-11-04 02:52:54 +00:00
|
|
|
const U32 current = (U32)(ip-base);
|
|
|
|
const U32 btLow = btMask >= current ? 0 : current - btMask;
|
2015-11-23 14:29:15 +00:00
|
|
|
const U32 windowLow = zc->lowLimit;
|
2015-11-04 02:52:54 +00:00
|
|
|
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-12-31 18:08:44 +00:00
|
|
|
U32 matchEndIdx = current+8;
|
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-12-11 09:44:07 +00:00
|
|
|
hashTable[h] = current; /* 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 */
|
2016-01-01 06:47:58 +00:00
|
|
|
const BYTE* match;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
2016-01-01 06:47:58 +00:00
|
|
|
if ((!extDict) || (matchIndex+matchLength >= dictLimit))
|
|
|
|
{
|
|
|
|
match = base + matchIndex;
|
|
|
|
if (match[matchLength] == ip[matchLength])
|
|
|
|
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iend) +1;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
match = dictBase + matchIndex;
|
|
|
|
matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iend, dictEnd, prefixStart);
|
|
|
|
if (matchIndex+matchLength >= dictLimit)
|
|
|
|
match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
|
|
|
|
}
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
if (matchLength > bestLength)
|
|
|
|
{
|
2015-12-29 21:26:09 +00:00
|
|
|
if (matchLength > matchEndIdx - matchIndex)
|
2015-12-29 22:40:02 +00:00
|
|
|
matchEndIdx = matchIndex + (U32)matchLength;
|
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
|
|
|
|
2015-12-31 18:08:44 +00:00
|
|
|
zc->nextToUpdate = (matchEndIdx > current + 8) ? matchEndIdx - 8 : current+1;
|
2015-11-04 02:52:54 +00:00
|
|
|
return bestLength;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** 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-12-29 21:26:09 +00:00
|
|
|
if (ip < zc->base + zc->nextToUpdate) return 0; /* skipped area */
|
|
|
|
ZSTD_updateTree(zc, ip, iLimit, maxNbAttempts, mls);
|
2016-01-01 06:47:58 +00:00
|
|
|
return ZSTD_insertBtAndFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, mls, 0);
|
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-12-29 21:26:09 +00:00
|
|
|
static void ZSTD_updateTree_extDict(ZSTD_CCtx* zc, const BYTE* const ip, const BYTE* const iend, const U32 nbCompares, const U32 mls)
|
2015-11-23 14:29:15 +00:00
|
|
|
{
|
|
|
|
const BYTE* const base = zc->base;
|
|
|
|
const U32 target = (U32)(ip - base);
|
|
|
|
U32 idx = zc->nextToUpdate;
|
|
|
|
|
|
|
|
for( ; idx < target ; )
|
2016-01-01 06:29:39 +00:00
|
|
|
idx += ZSTD_insertBt1(zc, base+idx, mls, iend, nbCompares, 1);
|
2015-11-23 14:29:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/** Tree updater, providing best match */
|
|
|
|
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
|
|
|
|
size_t ZSTD_BtFindBestMatch_extDict (
|
|
|
|
ZSTD_CCtx* zc,
|
|
|
|
const BYTE* const ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 mls)
|
|
|
|
{
|
2015-12-29 21:26:09 +00:00
|
|
|
if (ip < zc->base + zc->nextToUpdate) return 0; /* skipped area */
|
|
|
|
ZSTD_updateTree_extDict(zc, ip, iLimit, maxNbAttempts, mls);
|
2016-01-01 06:47:58 +00:00
|
|
|
return ZSTD_insertBtAndFindBestMatch(zc, ip, iLimit, offsetPtr, maxNbAttempts, mls, 1);
|
2015-11-23 14:29:15 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
FORCE_INLINE size_t ZSTD_BtFindBestMatch_selectMLS_extDict (
|
|
|
|
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_BtFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4);
|
|
|
|
case 5 : return ZSTD_BtFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5);
|
|
|
|
case 6 : return ZSTD_BtFindBestMatch_extDict(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
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-26 10:43:00 +00:00
|
|
|
/* Update chains up to ip (excluded)
|
2015-11-23 13:23:47 +00:00
|
|
|
Assumption : always within prefix (ie. not within extDict) */
|
|
|
|
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-23 13:37:59 +00:00
|
|
|
size_t ZSTD_HcFindBestMatch_generic (
|
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* const ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
2015-11-23 13:37:59 +00:00
|
|
|
const U32 maxNbAttempts, const U32 mls, const U32 extDict)
|
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-23 13:37:59 +00:00
|
|
|
matchIndex = ZSTD_insertAndFindFirstIndex (zc, ip, mls);
|
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--;
|
2015-11-23 13:37:59 +00:00
|
|
|
if ((!extDict) || matchIndex >= dictLimit)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
|
|
|
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 */
|
2015-11-23 15:17:21 +00:00
|
|
|
if (currentMl > ml) { ml = currentMl; *offsetPtr = current - matchIndex; if (ip+currentMl == iLimit) break; /* best possible, and avoid read overflow*/ }
|
2015-11-23 12:34:21 +00:00
|
|
|
|
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-23 13:37:59 +00:00
|
|
|
FORCE_INLINE size_t ZSTD_HcFindBestMatch_selectMLS (
|
|
|
|
ZSTD_CCtx* zc,
|
|
|
|
const BYTE* ip, const BYTE* const iLimit,
|
|
|
|
size_t* offsetPtr,
|
|
|
|
const U32 maxNbAttempts, const U32 matchLengthSearch)
|
|
|
|
{
|
|
|
|
switch(matchLengthSearch)
|
|
|
|
{
|
|
|
|
default :
|
|
|
|
case 4 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4, 0);
|
|
|
|
case 5 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5, 0);
|
|
|
|
case 6 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6, 0);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-22 12:24:05 +00:00
|
|
|
FORCE_INLINE size_t ZSTD_HcFindBestMatch_extDict_selectMLS (
|
2015-11-23 13:37:59 +00:00
|
|
|
ZSTD_CCtx* zc,
|
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-23 13:37:59 +00:00
|
|
|
case 4 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 4, 1);
|
|
|
|
case 5 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 5, 1);
|
|
|
|
case 6 : return ZSTD_HcFindBestMatch_generic(zc, ip, iLimit, offsetPtr, maxNbAttempts, 6, 1);
|
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
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_lazy_generic(ZSTD_CCtx* ctx,
|
|
|
|
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;
|
2015-12-05 08:23:53 +00:00
|
|
|
const BYTE* const base = ctx->base + ctx->dictLimit;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
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);
|
2015-12-05 08:23:53 +00:00
|
|
|
if ((ip-base) < REPCODE_STARTVALUE) ip = base + REPCODE_STARTVALUE;
|
2015-11-04 02:52:54 +00:00
|
|
|
|
|
|
|
/* 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-23 15:17:21 +00:00
|
|
|
/* 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;
|
|
|
|
}
|
|
|
|
|
2015-12-05 08:23:53 +00:00
|
|
|
if (matchLength < MINMATCH)
|
2015-11-23 15:17:21 +00:00
|
|
|
{
|
|
|
|
ip += ((ip-anchor) >> g_searchStrength) + 1; /* jump faster over incompressible sections */
|
|
|
|
continue;
|
|
|
|
}
|
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-12-05 08:23:53 +00:00
|
|
|
while ((start>anchor) && (start>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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_btlazy2(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_generic(ctx, src, srcSize, 1, 2);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_lazy2(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_generic(ctx, src, srcSize, 0, 2);
|
2015-11-05 14:00:24 +00:00
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_lazy(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-05 14:00:24 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_generic(ctx, src, srcSize, 0, 1);
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_greedy(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-10-31 11:57:14 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_generic(ctx, src, srcSize, 0, 0);
|
2015-10-31 11:57:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-11-22 01:53:43 +00:00
|
|
|
FORCE_INLINE
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_lazy_extDict_generic(ZSTD_CCtx* ctx,
|
|
|
|
const void* src, size_t srcSize,
|
2015-11-22 01:53:43 +00:00
|
|
|
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;
|
2015-11-24 13:06:07 +00:00
|
|
|
const BYTE* const dictStart = dictBase + ctx->lowLimit;
|
2015-11-22 01:53:43 +00:00
|
|
|
|
|
|
|
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-23 14:29:15 +00:00
|
|
|
searchMax_f searchMax = searchMethod ? ZSTD_BtFindBestMatch_selectMLS_extDict : ZSTD_HcFindBestMatch_extDict_selectMLS;
|
2015-11-22 01:53:43 +00:00
|
|
|
|
|
|
|
/* init */
|
|
|
|
ZSTD_resetSeqStore(seqStorePtr);
|
2015-12-11 09:44:07 +00:00
|
|
|
if ((ip - prefixStart) < REPCODE_STARTVALUE) ip += REPCODE_STARTVALUE;
|
2015-11-22 01:53:43 +00:00
|
|
|
|
|
|
|
/* 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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
{
|
2015-11-23 15:17:21 +00:00
|
|
|
/* 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;
|
|
|
|
}
|
2015-11-22 01:53:43 +00:00
|
|
|
|
|
|
|
/* 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)
|
|
|
|
{
|
2015-11-24 13:06:07 +00:00
|
|
|
U32 matchIndex = (U32)((start-base) - offset);
|
|
|
|
const BYTE* match = (matchIndex < dictLimit) ? dictBase + matchIndex : base + matchIndex;
|
|
|
|
const BYTE* const mStart = (matchIndex < dictLimit) ? dictStart : prefixStart;
|
|
|
|
while ((start>anchor) && (match>mStart) && (start[-1] == match[-1])) { start--; match--; matchLength++; } /* catch up */
|
2015-11-22 01:53:43 +00:00
|
|
|
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-23 15:17: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 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;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
void ZSTD_compressBlock_greedy_extDict(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-22 01:53:43 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_extDict_generic(ctx, src, srcSize, 0, 0);
|
2015-11-22 01:53:43 +00:00
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_lazy_extDict(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-22 02:12:28 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_extDict_generic(ctx, src, srcSize, 0, 1);
|
2015-11-22 02:12:28 +00:00
|
|
|
}
|
2015-11-22 01:53:43 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_lazy2_extDict(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-22 11:22:04 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_extDict_generic(ctx, src, srcSize, 0, 2);
|
2015-11-22 11:22:04 +00:00
|
|
|
}
|
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
static void ZSTD_compressBlock_btlazy2_extDict(ZSTD_CCtx* ctx, const void* src, size_t srcSize)
|
2015-11-23 12:34:21 +00:00
|
|
|
{
|
2016-01-25 03:22:03 +00:00
|
|
|
ZSTD_compressBlock_lazy_extDict_generic(ctx, src, srcSize, 1, 2);
|
2015-11-23 12:34:21 +00:00
|
|
|
}
|
|
|
|
|
2015-11-21 14:27:35 +00:00
|
|
|
|
2016-01-23 18:28:41 +00:00
|
|
|
typedef void (*ZSTD_blockCompressor) (ZSTD_CCtx* ctx, const void* src, size_t srcSize);
|
2015-11-04 11:05:27 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
static ZSTD_blockCompressor ZSTD_selectBlockCompressor(ZSTD_strategy strat, int extDict)
|
2015-10-31 11:57:14 +00:00
|
|
|
{
|
2015-11-29 01:38:09 +00:00
|
|
|
static const ZSTD_blockCompressor blockCompressor[2][5] = {
|
|
|
|
{ ZSTD_compressBlock_fast, ZSTD_compressBlock_greedy, ZSTD_compressBlock_lazy,ZSTD_compressBlock_lazy2, ZSTD_compressBlock_btlazy2 },
|
|
|
|
{ ZSTD_compressBlock_fast_extDict, ZSTD_compressBlock_greedy_extDict, ZSTD_compressBlock_lazy_extDict,ZSTD_compressBlock_lazy2_extDict, ZSTD_compressBlock_btlazy2_extDict }
|
|
|
|
};
|
|
|
|
|
|
|
|
return blockCompressor[extDict][(U32)strat];
|
2015-10-31 11:57:14 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
static size_t ZSTD_compressBlock_internal(ZSTD_CCtx* zc, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
2015-11-04 11:05:27 +00:00
|
|
|
{
|
2015-11-23 14:29:15 +00:00
|
|
|
ZSTD_blockCompressor blockCompressor = ZSTD_selectBlockCompressor(zc->params.strategy, zc->lowLimit < zc->dictLimit);
|
2016-01-09 01:10:40 +00:00
|
|
|
if (srcSize < MIN_CBLOCK_SIZE+3) return 0; /* don't even attempt compression below a certain srcSize */
|
2016-01-23 18:28:41 +00:00
|
|
|
blockCompressor(zc, src, srcSize);
|
2016-01-26 02:14:20 +00:00
|
|
|
return ZSTD_compressSequences(zc, dst, maxDstSize, srcSize);
|
2015-11-04 11:05:27 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
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-12-02 13:00:45 +00:00
|
|
|
size_t blockSize = ctxPtr->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;
|
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)
|
2015-11-24 13:06:07 +00:00
|
|
|
{
|
2015-11-13 10:27:46 +00:00
|
|
|
/* respect windowLog contract */
|
2015-11-24 13:06:07 +00:00
|
|
|
ctxPtr->lowLimit = (U32)(ip+blockSize - ctxPtr->base) - maxDist;
|
|
|
|
if (ctxPtr->dictLimit < ctxPtr->lowLimit) ctxPtr->dictLimit = ctxPtr->lowLimit;
|
|
|
|
}
|
2015-11-13 10:27:46 +00:00
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
cSize = ZSTD_compressBlock_internal(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 */
|
2016-01-09 01:10:40 +00:00
|
|
|
if (ZSTD_isError(cSize)) return cSize;
|
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;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
static size_t ZSTD_compressContinue_internal (ZSTD_CCtx* zc,
|
|
|
|
void* dst, size_t dstSize,
|
|
|
|
const void* src, size_t srcSize,
|
|
|
|
U32 frame)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-10-29 15:49:43 +00:00
|
|
|
const BYTE* const ip = (const BYTE*) src;
|
2016-01-07 14:35:18 +00:00
|
|
|
size_t hbSize = 0;
|
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
if (frame && (zc->stage==0))
|
2016-01-07 14:35:18 +00:00
|
|
|
{
|
|
|
|
hbSize = zc->hbSize;
|
|
|
|
if (dstSize <= hbSize) return ERROR(dstSize_tooSmall);
|
|
|
|
zc->stage = 1;
|
|
|
|
memcpy(dst, zc->headerBuffer, hbSize);
|
|
|
|
dstSize -= hbSize;
|
|
|
|
dst = (char*)dst + hbSize;
|
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2015-12-04 16:16:37 +00:00
|
|
|
/* Check if blocks follow each other */
|
|
|
|
if (src != zc->nextSrc)
|
|
|
|
{
|
|
|
|
/* not contiguous */
|
|
|
|
size_t delta = zc->nextSrc - ip;
|
|
|
|
zc->lowLimit = zc->dictLimit;
|
|
|
|
zc->dictLimit = (U32)(zc->nextSrc - zc->base);
|
|
|
|
zc->dictBase = zc->base;
|
|
|
|
zc->base -= delta;
|
|
|
|
zc->nextToUpdate = zc->dictLimit;
|
2015-12-05 08:23:53 +00:00
|
|
|
if (zc->dictLimit - zc->lowLimit < 8) zc->lowLimit = zc->dictLimit; /* too small extDict */
|
2015-12-04 16:16:37 +00:00
|
|
|
}
|
|
|
|
|
2015-11-13 10:27:46 +00:00
|
|
|
/* preemptive overflow correction */
|
2015-12-11 09:44:07 +00:00
|
|
|
if (zc->lowLimit > (1<<30))
|
2015-11-13 10:27:46 +00:00
|
|
|
{
|
2015-12-11 09:44:07 +00:00
|
|
|
U32 btplus = (zc->params.strategy == ZSTD_btlazy2);
|
|
|
|
U32 contentMask = (1 << (zc->params.contentLog - btplus)) - 1;
|
|
|
|
U32 newLowLimit = zc->lowLimit & contentMask; /* preserve position % contentSize */
|
|
|
|
U32 correction = zc->lowLimit - newLowLimit;
|
2015-11-24 13:06:07 +00:00
|
|
|
ZSTD_reduceIndex(zc, correction);
|
|
|
|
zc->base += correction;
|
|
|
|
zc->dictBase += correction;
|
2015-12-11 09:44:07 +00:00
|
|
|
zc->lowLimit = newLowLimit;
|
2015-11-24 13:06:07 +00:00
|
|
|
zc->dictLimit -= correction;
|
|
|
|
if (zc->nextToUpdate < correction) zc->nextToUpdate = 0;
|
|
|
|
else zc->nextToUpdate -= correction;
|
2015-11-13 10:27:46 +00:00
|
|
|
}
|
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
/* if input and dictionary overlap : reduce dictionary (presumed modified by input) */
|
2015-11-24 13:06:07 +00:00
|
|
|
if ((ip+srcSize > zc->dictBase + zc->lowLimit) && (ip < zc->dictBase + zc->dictLimit))
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-11-24 13:06:07 +00:00
|
|
|
zc->lowLimit = (U32)(ip + srcSize - zc->dictBase);
|
|
|
|
if (zc->lowLimit > zc->dictLimit) zc->lowLimit = zc->dictLimit;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-11-24 13:06:07 +00:00
|
|
|
zc->nextSrc = ip + srcSize;
|
2016-01-07 14:35:18 +00:00
|
|
|
{
|
2016-01-09 00:08:23 +00:00
|
|
|
size_t cSize;
|
|
|
|
if (frame) cSize = ZSTD_compress_generic (zc, dst, dstSize, src, srcSize);
|
|
|
|
else cSize = ZSTD_compressBlock_internal (zc, dst, dstSize, src, srcSize);
|
2016-01-07 14:35:18 +00:00
|
|
|
if (ZSTD_isError(cSize)) return cSize;
|
|
|
|
return cSize + hbSize;
|
|
|
|
}
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2016-01-09 00:08:23 +00:00
|
|
|
|
|
|
|
size_t ZSTD_compressContinue (ZSTD_CCtx* zc,
|
|
|
|
void* dst, size_t dstSize,
|
|
|
|
const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
return ZSTD_compressContinue_internal(zc, dst, dstSize, src, srcSize, 1);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
size_t ZSTD_compressBlock(ZSTD_CCtx* zc, void* dst, size_t maxDstSize, const void* src, size_t srcSize)
|
|
|
|
{
|
|
|
|
if (srcSize > BLOCKSIZE) return ERROR(srcSize_wrong);
|
|
|
|
return ZSTD_compressContinue_internal(zc, dst, maxDstSize, src, srcSize, 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
static size_t ZSTD_loadDictionaryContent(ZSTD_CCtx* zc, const void* src, size_t srcSize)
|
2015-12-04 16:16:37 +00:00
|
|
|
{
|
|
|
|
const BYTE* const ip = (const BYTE*) src;
|
|
|
|
const BYTE* const iend = ip + srcSize;
|
|
|
|
|
|
|
|
/* input becomes current prefix */
|
|
|
|
zc->lowLimit = zc->dictLimit;
|
|
|
|
zc->dictLimit = (U32)(zc->nextSrc - zc->base);
|
|
|
|
zc->dictBase = zc->base;
|
|
|
|
zc->base += ip - zc->nextSrc;
|
|
|
|
zc->nextToUpdate = zc->dictLimit;
|
|
|
|
|
|
|
|
zc->nextSrc = iend;
|
|
|
|
if (srcSize <= 8) return 0;
|
|
|
|
|
|
|
|
switch(zc->params.strategy)
|
|
|
|
{
|
|
|
|
case ZSTD_fast:
|
|
|
|
ZSTD_fillHashTable (zc, iend-8, zc->params.searchLength);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ZSTD_greedy:
|
|
|
|
case ZSTD_lazy:
|
|
|
|
case ZSTD_lazy2:
|
|
|
|
ZSTD_insertAndFindFirstIndex (zc, iend-8, zc->params.searchLength);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case ZSTD_btlazy2:
|
|
|
|
ZSTD_updateTree(zc, iend-8, iend, 1 << zc->params.searchLog, zc->params.searchLength);
|
2015-12-29 22:40:02 +00:00
|
|
|
zc->nextToUpdate = (U32)(iend - zc->base);
|
2015-12-04 16:16:37 +00:00
|
|
|
break;
|
|
|
|
|
|
|
|
default:
|
|
|
|
return ERROR(GENERIC); /* strategy doesn't exist; impossible */
|
|
|
|
}
|
|
|
|
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
/* Dictionary format :
|
|
|
|
Magic == ZSTD_DICT_MAGIC (4 bytes)
|
|
|
|
Huff0 CTable (256 * 4 bytes) => to be changed to read from writeCTable
|
|
|
|
Dictionary content
|
|
|
|
*/
|
|
|
|
/*! ZSTD_loadDictEntropyStats
|
|
|
|
@return : size read from dictionary */
|
|
|
|
static size_t ZSTD_loadDictEntropyStats(ZSTD_CCtx* zc, const void* dict, size_t dictSize)
|
|
|
|
{
|
|
|
|
/* note : magic number already checked */
|
|
|
|
const size_t hufHeaderSize = HUF_readCTable(zc->hufTable, 255, dict, dictSize);
|
|
|
|
if (HUF_isError(hufHeaderSize)) return ERROR(dictionary_corrupted);
|
|
|
|
zc->flagHufTable = 1;
|
|
|
|
return hufHeaderSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
size_t ZSTD_compress_insertDictionary(ZSTD_CCtx* zc, const void* dict, size_t dictSize)
|
|
|
|
{
|
2016-01-26 14:58:49 +00:00
|
|
|
if (dict && dictSize)
|
|
|
|
{
|
|
|
|
U32 magic = MEM_readLE32(dict);
|
|
|
|
size_t eSize;
|
|
|
|
if (magic != ZSTD_DICT_MAGIC)
|
|
|
|
return ZSTD_loadDictionaryContent(zc, dict, dictSize);
|
|
|
|
|
|
|
|
eSize = ZSTD_loadDictEntropyStats(zc, (const char*)dict+4, dictSize-4) + 4;
|
|
|
|
if (ZSTD_isError(eSize)) return eSize;
|
|
|
|
return ZSTD_loadDictionaryContent(zc, (const char*)dict+eSize, dictSize-eSize);
|
|
|
|
}
|
2016-01-07 14:35:18 +00:00
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-12-04 16:16:37 +00:00
|
|
|
/*! ZSTD_compressBegin_advanced
|
2016-01-07 14:35:18 +00:00
|
|
|
* @return : 0, or an error code */
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compressBegin_advanced(ZSTD_CCtx* ctx,
|
2015-11-25 13:42:45 +00:00
|
|
|
ZSTD_parameters params)
|
2015-10-22 14:31:46 +00:00
|
|
|
{
|
2015-10-30 05:40:22 +00:00
|
|
|
size_t errorCode;
|
2015-11-25 13:42:45 +00:00
|
|
|
|
|
|
|
ZSTD_validateParams(¶ms);
|
|
|
|
|
|
|
|
errorCode = ZSTD_resetCCtx_advanced(ctx, params);
|
2015-10-30 05:40:22 +00:00
|
|
|
if (ZSTD_isError(errorCode)) return errorCode;
|
2015-11-13 10:27:46 +00:00
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
MEM_writeLE32(ctx->headerBuffer, ZSTD_MAGICNUMBER); /* Write Header */
|
|
|
|
((BYTE*)ctx->headerBuffer)[4] = (BYTE)(params.windowLog - ZSTD_WINDOWLOG_ABSOLUTEMIN);
|
|
|
|
ctx->hbSize = ZSTD_frameHeaderSize_min;
|
|
|
|
ctx->stage = 0;
|
|
|
|
|
|
|
|
return 0;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
|
|
|
|
2015-10-25 13:06:35 +00:00
|
|
|
|
2015-11-25 13:42:45 +00:00
|
|
|
/** ZSTD_getParams
|
|
|
|
* return ZSTD_parameters structure for a selected compression level and srcSize.
|
|
|
|
* srcSizeHint value is optional, select 0 if not known */
|
|
|
|
ZSTD_parameters ZSTD_getParams(int compressionLevel, U64 srcSizeHint)
|
2015-10-25 13:06:35 +00:00
|
|
|
{
|
2015-11-25 13:42:45 +00:00
|
|
|
ZSTD_parameters result;
|
2015-12-09 08:05:22 +00:00
|
|
|
int tableID = ((srcSizeHint-1) <= 256 KB) + ((srcSizeHint-1) <= 128 KB) + ((srcSizeHint-1) <= 16 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;
|
2015-11-25 13:42:45 +00:00
|
|
|
result = ZSTD_defaultParameters[tableID][compressionLevel];
|
|
|
|
result.srcSize = srcSizeHint;
|
|
|
|
return result;
|
|
|
|
}
|
|
|
|
|
2016-01-26 02:14:20 +00:00
|
|
|
/* to do
|
|
|
|
size_t ZSTD_compressBegin_usingDict(ZSTD_CCtx* cctx, const void* dict,size_t dictSize, int compressionLevel)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}*/
|
2015-11-25 13:42:45 +00:00
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
size_t ZSTD_compressBegin(ZSTD_CCtx* ctx, int compressionLevel)
|
2015-11-25 13:42:45 +00:00
|
|
|
{
|
2016-01-07 14:35:18 +00:00
|
|
|
return ZSTD_compressBegin_advanced(ctx, ZSTD_getParams(compressionLevel, 0));
|
2015-10-25 13:06:35 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2015-12-18 00:26:48 +00:00
|
|
|
/*! ZSTD_compressEnd
|
2015-11-25 13:42:45 +00:00
|
|
|
* Write frame epilogue
|
|
|
|
* @return : nb of bytes written into dst (or an error code) */
|
2016-01-07 14:35:18 +00:00
|
|
|
size_t ZSTD_compressEnd(ZSTD_CCtx* zc, void* dst, size_t maxDstSize)
|
2015-10-29 15:49:43 +00:00
|
|
|
{
|
|
|
|
BYTE* op = (BYTE*)dst;
|
2016-01-07 14:35:18 +00:00
|
|
|
size_t hbSize = 0;
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
/* empty frame */
|
|
|
|
if (zc->stage==0)
|
|
|
|
{
|
|
|
|
hbSize = zc->hbSize;
|
|
|
|
if (maxDstSize <= hbSize) return ERROR(dstSize_tooSmall);
|
|
|
|
zc->stage = 1;
|
|
|
|
memcpy(dst, zc->headerBuffer, hbSize);
|
|
|
|
maxDstSize -= hbSize;
|
|
|
|
op += hbSize;
|
|
|
|
}
|
2015-10-29 15:49:43 +00:00
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
/* frame epilogue */
|
|
|
|
if (maxDstSize < 3) return ERROR(dstSize_tooSmall);
|
2015-10-29 15:49:43 +00:00
|
|
|
op[0] = (BYTE)(bt_end << 6);
|
|
|
|
op[1] = 0;
|
|
|
|
op[2] = 0;
|
|
|
|
|
2016-01-07 14:35:18 +00:00
|
|
|
return 3+hbSize;
|
2015-10-29 15:49:43 +00:00
|
|
|
}
|
|
|
|
|
2015-11-11 12:43:58 +00:00
|
|
|
size_t ZSTD_compress_advanced (ZSTD_CCtx* ctx,
|
2015-11-25 13:42:45 +00:00
|
|
|
void* dst, size_t maxDstSize,
|
|
|
|
const void* src, size_t srcSize,
|
2015-12-18 00:26:48 +00:00
|
|
|
const void* dict,size_t dictSize,
|
2015-11-25 13:42:45 +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 */
|
2016-01-07 14:35:18 +00:00
|
|
|
oSize = ZSTD_compressBegin_advanced(ctx, params);
|
2015-10-22 14:31:46 +00:00
|
|
|
if(ZSTD_isError(oSize)) return oSize;
|
|
|
|
|
2015-12-18 00:26:48 +00:00
|
|
|
/* dictionary */
|
|
|
|
if (dict)
|
|
|
|
{
|
|
|
|
oSize = ZSTD_compress_insertDictionary(ctx, dict, dictSize);
|
|
|
|
if (ZSTD_isError(oSize)) return oSize;
|
|
|
|
}
|
|
|
|
|
2015-10-22 14:31:46 +00:00
|
|
|
/* body (compression) */
|
2015-12-18 00:26:48 +00:00
|
|
|
oSize = ZSTD_compressContinue (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-12-18 00:26:48 +00:00
|
|
|
size_t ZSTD_compress_usingDict(ZSTD_CCtx* ctx, void* dst, size_t maxDstSize, const void* src, size_t srcSize, const void* dict, size_t dictSize, int compressionLevel)
|
|
|
|
{
|
|
|
|
return ZSTD_compress_advanced(ctx, dst, maxDstSize, src, srcSize, dict, dictSize, ZSTD_getParams(compressionLevel, srcSize+dictSize));
|
|
|
|
}
|
|
|
|
|
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-12-18 00:26:48 +00:00
|
|
|
return ZSTD_compress_advanced(ctx, dst, maxDstSize, src, srcSize, NULL, 0, ZSTD_getParams(compressionLevel, srcSize));
|
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);
|
2016-01-07 14:35:18 +00:00
|
|
|
free(ctxBody.workSpace); /* can't free ctxBody, since it's on stack; just free heap content */
|
2015-10-29 21:02:40 +00:00
|
|
|
return result;
|
2015-10-22 14:31:46 +00:00
|
|
|
}
|
2015-12-17 22:50:15 +00:00
|
|
|
|