zstd/lib/zstd_opt.h

1245 lines
57 KiB
C
Raw Normal View History

2016-02-10 05:26:30 -08:00
/*
ZSTD Optimal mode
Copyright (C) 2016, Przemyslaw Skibinski, 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
*/
/* Note : this file is intended to be included within zstd_compress.c */
/*- Local types -*/
typedef struct {
2016-02-09 11:47:39 -08:00
U32 off;
U32 len;
U32 back;
2016-02-03 10:08:48 -08:00
} ZSTD_match_t;
2016-01-31 02:25:48 -08:00
2016-02-10 05:26:30 -08:00
typedef struct {
2016-02-09 11:47:39 -08:00
U32 price;
U32 off;
U32 mlen;
U32 litlen;
U32 rep;
U32 rep2;
2016-02-10 04:37:52 -08:00
} ZSTD_optimal_t;
2016-01-31 02:25:48 -08:00
2016-02-10 05:50:22 -08:00
2016-02-10 05:26:30 -08:00
/*- Constants -*/
2016-02-19 02:05:25 -08:00
#define ZSTD_OPT_NUM (1<<12)
#define ZSTD_FREQ_START 1
#define ZSTD_FREQ_STEP 1
#define ZSTD_FREQ_DIV 6
2016-02-05 00:07:22 -08:00
2016-02-10 05:26:30 -08:00
/*- Debug -*/
2016-02-19 02:05:25 -08:00
#if defined(ZSTD_OPT_DEBUG) && ZSTD_OPT_DEBUG>=9
2016-02-10 05:50:22 -08:00
#define ZSTD_LOG_PARSER(...) printf(__VA_ARGS__)
#define ZSTD_LOG_ENCODE(...) printf(__VA_ARGS__)
#define ZSTD_LOG_TRY_PRICE(...) printf(__VA_ARGS__)
2016-02-02 06:41:16 -08:00
#else
2016-02-10 05:50:22 -08:00
#define ZSTD_LOG_PARSER(...)
#define ZSTD_LOG_ENCODE(...)
#define ZSTD_LOG_TRY_PRICE(...)
2016-02-02 06:41:16 -08:00
#endif
2016-01-31 02:25:48 -08:00
2016-02-19 02:05:25 -08:00
#if MINMATCH == 3
#define MEM_readMINMATCH(ptr) ((U32)(MEM_read32(ptr)<<8))
#else
#define MEM_readMINMATCH(ptr) (U32)(MEM_read32(ptr))
#endif
FORCE_INLINE U32 ZSTD_getLiteralPrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals)
2016-02-03 12:26:18 -08:00
{
2016-02-19 02:59:44 -08:00
U32 price, u;
2016-02-03 12:26:18 -08:00
2016-02-10 05:26:30 -08:00
/* literals */
price = litLength * ZSTD_highbit(seqStorePtr->litSum);
for (u=0; u < litLength; u++)
price -= ZSTD_highbit(seqStorePtr->litFreq[literals[u]]);
2016-02-10 05:26:30 -08:00
/* literal Length */
2016-02-19 02:05:25 -08:00
price += ((litLength >= MaxLL)<<3) + ((litLength >= 255+MaxLL)<<4) + ((litLength>=(1<<15))<<3);
2016-02-10 05:26:30 -08:00
if (litLength >= MaxLL) litLength = MaxLL;
2016-02-19 02:59:44 -08:00
price += ZSTD_highbit(seqStorePtr->litLengthSum) - ZSTD_highbit(seqStorePtr->litLengthFreq[litLength]);
2016-02-10 05:26:30 -08:00
return price;
2016-02-03 12:26:18 -08:00
}
2016-02-03 10:53:29 -08:00
2016-02-19 02:05:25 -08:00
FORCE_INLINE U32 ZSTD_getPrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals, U32 offset, U32 matchLength)
2016-02-04 01:04:33 -08:00
{
2016-02-10 04:37:52 -08:00
/* offset */
BYTE offCode = offset ? (BYTE)ZSTD_highbit(offset) + 1 : 0;
2016-02-19 02:05:25 -08:00
U32 price = offCode + ZSTD_highbit(seqStorePtr->offCodeSum) - ZSTD_highbit(seqStorePtr->offCodeFreq[offCode]);
2016-02-03 12:26:18 -08:00
/* match Length */
2016-02-19 02:05:25 -08:00
matchLength -= MINMATCH;
price += ((matchLength >= MaxML)<<3) + ((matchLength >= 255+MaxML)<<4) + ((matchLength>=(1<<15))<<3);
2016-02-10 04:37:52 -08:00
if (matchLength >= MaxML) matchLength = MaxML;
2016-02-19 02:59:44 -08:00
price += ZSTD_highbit(seqStorePtr->matchLengthSum) - ZSTD_highbit(seqStorePtr->matchLengthFreq[matchLength]);
if (!litLength)
return price + 1; /* special case */
2016-02-03 12:26:18 -08:00
2016-02-19 02:59:44 -08:00
return price + ZSTD_getLiteralPrice(seqStorePtr, litLength, literals); //((seqStorePtr->litSum>>4) / seqStorePtr->litLengthSum) + (matchLength==0) + (litLength==0) + price;
2016-02-04 01:04:33 -08:00
}
2016-02-04 02:11:08 -08:00
2016-02-19 02:05:25 -08:00
MEM_STATIC void ZSTD_rescaleFreqs(seqStore_t* ssPtr)
2016-02-04 01:04:33 -08:00
{
2016-02-19 02:05:25 -08:00
unsigned u;
// printf("matchLengthSum=%d litLengthSum=%d litSum=%d offCodeSum=%d\n", ssPtr->matchLengthSum, ssPtr->litLengthSum, ssPtr->litSum, ssPtr->offCodeSum);
if (ssPtr->litLengthSum == 0) {
ssPtr->matchLengthSum = (1<<MLbits);
ssPtr->litLengthSum = (1<<LLbits);
ssPtr->litSum = (1<<Litbits);
ssPtr->offCodeSum = (1<<Offbits);
for (u=0; u<=MaxLit; u++)
ssPtr->litFreq[u] = 1;
for (u=0; u<=MaxLL; u++)
ssPtr->litLengthFreq[u] = 1;
for (u=0; u<=MaxML; u++)
ssPtr->matchLengthFreq[u] = 1;
for (u=0; u<=MaxOff; u++)
ssPtr->offCodeFreq[u] = 1;
} else {
ssPtr->matchLengthSum = 0;
ssPtr->litLengthSum = 0;
ssPtr->litSum = 0;
ssPtr->offCodeSum = 0;
for (u=0; u<=MaxLit; u++) {
ssPtr->litFreq[u] = ZSTD_FREQ_START + (ssPtr->litFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->litSum += ssPtr->litFreq[u];
}
for (u=0; u<=MaxLL; u++) {
ssPtr->litLengthFreq[u] = ZSTD_FREQ_START + (ssPtr->litLengthFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->litLengthSum += ssPtr->litLengthFreq[u];
}
for (u=0; u<=MaxML; u++) {
ssPtr->matchLengthFreq[u] = ZSTD_FREQ_START + (ssPtr->matchLengthFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->matchLengthSum += ssPtr->matchLengthFreq[u];
}
for (u=0; u<=MaxOff; u++) {
ssPtr->offCodeFreq[u] = ZSTD_FREQ_START + (ssPtr->offCodeFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->offCodeSum += ssPtr->offCodeFreq[u];
}
}
2016-01-31 02:25:48 -08:00
}
2016-02-03 12:26:18 -08:00
2016-02-09 11:47:39 -08:00
MEM_STATIC void ZSTD_updatePrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals, U32 offset, U32 matchLength)
2016-02-03 08:25:42 -08:00
{
2016-02-10 04:37:52 -08:00
U32 u;
2016-02-10 05:26:30 -08:00
2016-02-03 10:08:48 -08:00
/* literals */
2016-02-19 02:05:25 -08:00
seqStorePtr->litSum += litLength * ZSTD_FREQ_STEP;
2016-02-10 04:37:52 -08:00
for (u=0; u < litLength; u++)
2016-02-19 02:05:25 -08:00
seqStorePtr->litFreq[literals[u]] += ZSTD_FREQ_STEP;
2016-02-10 04:37:52 -08:00
2016-02-03 08:25:42 -08:00
/* literal Length */
2016-02-19 02:05:25 -08:00
seqStorePtr->litLengthSum += ZSTD_FREQ_STEP;
2016-02-03 12:26:18 -08:00
if (litLength >= MaxLL)
2016-02-19 02:05:25 -08:00
seqStorePtr->litLengthFreq[MaxLL] += ZSTD_FREQ_STEP;
2016-02-10 04:37:52 -08:00
else
2016-02-19 02:05:25 -08:00
seqStorePtr->litLengthFreq[litLength] += ZSTD_FREQ_STEP;
2016-02-03 08:25:42 -08:00
/* match offset */
2016-02-19 02:05:25 -08:00
seqStorePtr->offCodeSum += ZSTD_FREQ_STEP;
BYTE offCode = offset ? (BYTE)ZSTD_highbit(offset) + 1 : 0;
seqStorePtr->offCodeFreq[offCode] += ZSTD_FREQ_STEP;
2016-02-03 08:25:42 -08:00
/* match Length */
2016-02-19 02:05:25 -08:00
seqStorePtr->matchLengthSum += ZSTD_FREQ_STEP;
2016-02-03 12:26:18 -08:00
if (matchLength >= MaxML)
2016-02-19 02:05:25 -08:00
seqStorePtr->matchLengthFreq[MaxML] += ZSTD_FREQ_STEP;
2016-02-10 04:37:52 -08:00
else
2016-02-19 02:05:25 -08:00
seqStorePtr->matchLengthFreq[matchLength] += ZSTD_FREQ_STEP;
2016-02-03 08:25:42 -08:00
}
2016-02-04 04:01:56 -08:00
#define SET_PRICE(pos, mlen_, offset_, litlen_, price_) \
2016-01-31 02:25:48 -08:00
{ \
while (last_pos < pos) { opt[last_pos+1].price = 1<<30; last_pos++; } \
2016-02-04 04:01:56 -08:00
opt[pos].mlen = mlen_; \
opt[pos].off = offset_; \
opt[pos].litlen = litlen_; \
opt[pos].price = price_; \
2016-02-03 10:08:48 -08:00
ZSTD_LOG_PARSER("%d: SET price[%d/%d]=%d litlen=%d len=%d off=%d\n", (int)(inr-base), (int)pos, (int)last_pos, opt[pos].price, opt[pos].litlen, opt[pos].mlen, opt[pos].off); \
2016-01-31 02:25:48 -08:00
}
2016-02-10 05:26:30 -08:00
/*-*************************************
* Binary Tree search
***************************************/
2016-02-19 02:05:25 -08:00
/* Update hashTable3 up to ip (excluded)
Assumption : always within prefix (ie. not within extDict) */
#if MINMATCH == 3
static U32 ZSTD_insertAndFindFirstIndexHash3 (ZSTD_CCtx* zc, const BYTE* ip)
{
U32* const hashTable3 = zc->hashTable3;
const U32 hashLog3 = zc->params.hashLog3;
const BYTE* const base = zc->base;
const U32 target = (U32)(ip - base);
U32 idx = zc->nextToUpdate3;
while(idx < target) {
hashTable3[ZSTD_hash3Ptr(base+idx, hashLog3)] = idx;
idx++;
}
zc->nextToUpdate3 = target;
return hashTable3[ZSTD_hash3Ptr(ip, hashLog3)];
}
#endif
static U32 ZSTD_insertBtAndGetAllMatches (
2016-01-31 02:25:48 -08:00
ZSTD_CCtx* zc,
2016-02-19 02:05:25 -08:00
const BYTE* const ip, const BYTE* const iLimit,
2016-01-31 02:25:48 -08:00
U32 nbCompares, const U32 mls,
2016-02-19 02:05:25 -08:00
U32 extDict, ZSTD_match_t* matches)
2016-01-31 02:25:48 -08:00
{
2016-02-06 06:18:26 -08:00
const BYTE* const base = zc->base;
const U32 current = (U32)(ip-base);
2016-01-31 02:25:48 -08:00
const U32 hashLog = zc->params.hashLog;
const size_t h = ZSTD_hashPtr(ip, hashLog, mls);
2016-02-06 06:18:26 -08:00
U32* const hashTable = zc->hashTable;
U32 matchIndex = hashTable[h];
2016-01-31 02:25:48 -08:00
U32* const bt = zc->contentTable;
const U32 btLog = zc->params.contentLog - 1;
2016-02-09 11:47:39 -08:00
const U32 btMask= (1U << btLog) - 1;
2016-01-31 02:25:48 -08:00
size_t commonLengthSmaller=0, commonLengthLarger=0;
const BYTE* const dictBase = zc->dictBase;
const U32 dictLimit = zc->dictLimit;
const BYTE* const dictEnd = dictBase + dictLimit;
const BYTE* const prefixStart = base + dictLimit;
const U32 btLow = btMask >= current ? 0 : current - btMask;
const U32 windowLow = zc->lowLimit;
U32* smallerPtr = bt + 2*(current&btMask);
U32* largerPtr = bt + 2*(current&btMask) + 1;
U32 matchEndIdx = current+8;
U32 dummy32; /* to be nullified at the end */
2016-02-09 11:47:39 -08:00
U32 mnum = 0;
2016-02-10 04:37:52 -08:00
2016-02-19 02:05:25 -08:00
size_t bestLength = MINMATCH-1;
2016-01-31 02:25:48 -08:00
hashTable[h] = current; /* Update Hash Table */
2016-02-19 02:05:25 -08:00
#if MINMATCH == 3
/* HC3 match finder */
U32 matchIndex3 = ZSTD_insertAndFindFirstIndexHash3 (zc, ip);
if (matchIndex3>windowLow) {
const BYTE* match;
size_t currentMl=0;
if ((!extDict) || matchIndex3 >= dictLimit) {
match = base + matchIndex3;
if (match[bestLength] == ip[bestLength]) currentMl = ZSTD_count(ip, match, iLimit); // faster
// if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) currentMl = ZSTD_count(ip+MINMATCH, match+MINMATCH, iLimit)+MINMATCH; // stronger
} else {
match = dictBase + matchIndex3;
if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) /* assumption : matchIndex3 <= dictLimit-4 (by table construction) */
currentMl = ZSTD_count_2segments(ip+MINMATCH, match+MINMATCH, iLimit, dictEnd, prefixStart) + MINMATCH;
}
/* save best solution */
if (currentMl > bestLength) {
bestLength = currentMl;
matches[mnum].off = current - matchIndex3;
matches[mnum].len = (U32)currentMl;
matches[mnum].back = 0;
mnum++;
if (currentMl > ZSTD_OPT_NUM) return mnum;
if (ip+currentMl == iLimit) return mnum; /* best possible, and avoid read overflow*/
}
}
#endif
2016-01-31 02:25:48 -08:00
while (nbCompares-- && (matchIndex > windowLow)) {
U32* nextPtr = bt + 2*(matchIndex & btMask);
size_t matchLength = MIN(commonLengthSmaller, commonLengthLarger); /* guaranteed minimum nb of common bytes */
const BYTE* match;
if ((!extDict) || (matchIndex+matchLength >= dictLimit)) {
match = base + matchIndex;
2016-02-19 02:05:25 -08:00
if (match[matchLength] == ip[matchLength]) {
#if ZSTD_OPT_DEBUG >= 5
if (memcmp(match, ip, matchLength) != 0)
printf("%d: ERROR: matchLength=%d ZSTD_count=%d\n", current, (int)matchLength, (int)ZSTD_count(ip, match, ip+matchLength));
#endif
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iLimit) +1;
}
2016-01-31 02:25:48 -08:00
} else {
match = dictBase + matchIndex;
2016-02-19 02:05:25 -08:00
matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iLimit, dictEnd, prefixStart);
2016-01-31 02:25:48 -08:00
if (matchIndex+matchLength >= dictLimit)
2016-02-03 04:30:34 -08:00
match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
2016-01-31 02:25:48 -08:00
}
if (matchLength > bestLength) {
2016-02-10 04:37:52 -08:00
if (matchLength > matchEndIdx - matchIndex) matchEndIdx = matchIndex + (U32)matchLength;
bestLength = matchLength;
matches[mnum].off = current - matchIndex;
matches[mnum].len = (U32)matchLength;
matches[mnum].back = 0;
mnum++;
if (matchLength > ZSTD_OPT_NUM) break;
2016-02-19 02:05:25 -08:00
if (ip+matchLength == iLimit) /* equal : no way to know if inf or sup */
2016-01-31 02:25:48 -08:00
break; /* drop, to guarantee consistency (miss a little bit of compression) */
}
if (match[matchLength] < ip[matchLength]) {
/* match is smaller than current */
*smallerPtr = matchIndex; /* update smaller idx */
commonLengthSmaller = matchLength; /* all smaller will now have at least this guaranteed common length */
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) */
} else {
/* match is larger than current */
*largerPtr = matchIndex;
commonLengthLarger = matchLength;
if (matchIndex <= btLow) { largerPtr=&dummy32; break; } /* beyond tree size, stop the search */
largerPtr = nextPtr;
matchIndex = nextPtr[0];
2016-02-10 04:37:52 -08:00
} }
2016-01-31 02:25:48 -08:00
*smallerPtr = *largerPtr = 0;
zc->nextToUpdate = (matchEndIdx > current + 8) ? matchEndIdx - 8 : current+1;
return mnum;
}
/** Tree updater, providing best match */
static U32 ZSTD_BtGetAllMatches (
2016-01-31 02:25:48 -08:00
ZSTD_CCtx* zc,
const BYTE* const ip, const BYTE* const iLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 mls, ZSTD_match_t* matches)
2016-01-31 02:25:48 -08:00
{
if (ip < zc->base + zc->nextToUpdate) return 0; /* skipped area */
ZSTD_updateTree(zc, ip, iLimit, maxNbAttempts, mls);
2016-02-19 02:05:25 -08:00
return ZSTD_insertBtAndGetAllMatches(zc, ip, iLimit, maxNbAttempts, mls, 0, matches);
2016-01-31 02:25:48 -08:00
}
static U32 ZSTD_BtGetAllMatches_selectMLS (
2016-01-31 02:25:48 -08:00
ZSTD_CCtx* zc, /* Index table will be updated */
2016-02-04 11:53:54 -08:00
const BYTE* ip, const BYTE* const iLowLimit, const BYTE* const iHighLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
2016-01-31 02:25:48 -08:00
{
2016-02-10 04:37:52 -08:00
(void)iLowLimit; /* unused */
2016-01-31 02:25:48 -08:00
switch(matchLengthSearch)
{
default :
2016-02-19 02:05:25 -08:00
case 4 : return ZSTD_BtGetAllMatches(zc, ip, iHighLimit, maxNbAttempts, 4, matches);
case 5 : return ZSTD_BtGetAllMatches(zc, ip, iHighLimit, maxNbAttempts, 5, matches);
case 6 : return ZSTD_BtGetAllMatches(zc, ip, iHighLimit, maxNbAttempts, 6, matches);
2016-01-31 02:25:48 -08:00
}
}
/** Tree updater, providing best match */
static U32 ZSTD_BtGetAllMatches_extDict (
ZSTD_CCtx* zc,
const BYTE* const ip, const BYTE* const iLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 mls, ZSTD_match_t* matches)
{
if (ip < zc->base + zc->nextToUpdate) return 0; /* skipped area */
ZSTD_updateTree_extDict(zc, ip, iLimit, maxNbAttempts, mls);
2016-02-19 02:05:25 -08:00
return ZSTD_insertBtAndGetAllMatches(zc, ip, iLimit, maxNbAttempts, mls, 1, matches);
}
static U32 ZSTD_BtGetAllMatches_selectMLS_extDict (
ZSTD_CCtx* zc, /* Index table will be updated */
const BYTE* ip, const BYTE* const iLowLimit, const BYTE* const iHighLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
{
(void)iLowLimit;
switch(matchLengthSearch)
{
default :
2016-02-19 02:05:25 -08:00
case 4 : return ZSTD_BtGetAllMatches_extDict(zc, ip, iHighLimit, maxNbAttempts, 4, matches);
case 5 : return ZSTD_BtGetAllMatches_extDict(zc, ip, iHighLimit, maxNbAttempts, 5, matches);
case 6 : return ZSTD_BtGetAllMatches_extDict(zc, ip, iHighLimit, maxNbAttempts, 6, matches);
}
}
/* ***********************
* Hash Chain
*************************/
2016-01-31 02:25:48 -08:00
FORCE_INLINE /* inlining is important to hardwire a hot branch (template emulation) */
2016-02-09 11:47:39 -08:00
U32 ZSTD_HcGetAllMatches_generic (
2016-01-31 02:25:48 -08:00
ZSTD_CCtx* zc, /* Index table will be updated */
2016-02-04 11:53:54 -08:00
const BYTE* const ip, const BYTE* const iLowLimit, const BYTE* const iHighLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 mls, const U32 extDict, ZSTD_match_t* matches)
2016-01-31 02:25:48 -08:00
{
U32* const chainTable = zc->contentTable;
2016-02-09 11:47:39 -08:00
const U32 chainSize = (1U << zc->params.contentLog);
2016-01-31 02:25:48 -08:00
const U32 chainMask = chainSize-1;
const BYTE* const base = zc->base;
const BYTE* const dictBase = zc->dictBase;
const U32 dictLimit = zc->dictLimit;
const BYTE* const prefixStart = base + dictLimit;
const BYTE* const dictEnd = dictBase + dictLimit;
const BYTE* const dictStart = dictBase + zc->lowLimit;
2016-01-31 02:25:48 -08:00
const U32 lowLimit = zc->lowLimit;
const U32 current = (U32)(ip-base);
const U32 minChain = current > chainSize ? current - chainSize : 0;
U32 matchIndex;
2016-02-09 11:47:39 -08:00
U32 mnum = 0;
2016-01-31 02:25:48 -08:00
const BYTE* match;
2016-02-09 11:47:39 -08:00
U32 nbAttempts=maxNbAttempts;
2016-02-19 02:05:25 -08:00
size_t minml=MINMATCH-1;
2016-01-31 02:25:48 -08:00
2016-02-19 02:05:25 -08:00
/* HC4 match finder with update */
2016-01-31 02:25:48 -08:00
matchIndex = ZSTD_insertAndFindFirstIndex (zc, ip, mls);
2016-02-19 02:05:25 -08:00
#if MINMATCH == 3
/* HC3 match finder */
U32 matchIndex3 = zc->hashTable3[ZSTD_hash3Ptr(ip, zc->params.hashLog3)];
if (matchIndex3>lowLimit) {
size_t currentMl=0;
if ((!extDict) || matchIndex3 >= dictLimit) {
match = base + matchIndex3;
if (match[minml] == ip[minml]) currentMl = ZSTD_count(ip, match, iHighLimit); // faster
//if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) currentMl = ZSTD_count(ip+MINMATCH, match+MINMATCH, iHighLimit)+MINMATCH; // stronger
} else {
match = dictBase + matchIndex3;
if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) /* assumption : matchIndex3 <= dictLimit-4 (by table construction) */
currentMl = ZSTD_count_2segments(ip+MINMATCH, match+MINMATCH, iHighLimit, dictEnd, prefixStart) + MINMATCH;
}
/* save best solution */
if (currentMl > minml) {
minml = currentMl;
matches[mnum].off = current - matchIndex3;
matches[mnum].len = (U32)currentMl;
matches[mnum].back = 0;
mnum++;
if (currentMl > ZSTD_OPT_NUM) return mnum;
if (ip+currentMl == iHighLimit) return mnum; /* best possible, and avoid read overflow*/
}
}
#endif
2016-01-31 02:25:48 -08:00
while ((matchIndex>lowLimit) && (nbAttempts)) {
size_t currentMl=0;
2016-02-09 14:26:30 -08:00
int back = 0;
2016-01-31 02:25:48 -08:00
nbAttempts--;
if ((!extDict) || matchIndex >= dictLimit) {
match = base + matchIndex;
2016-02-10 08:18:24 -08:00
if (match[minml] == ip[minml]) currentMl = ZSTD_count(ip, match, iHighLimit); if (currentMl>0) { // faster
2016-02-19 02:05:25 -08:00
//if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) { currentMl = ZSTD_count(ip+MINMATCH, match+MINMATCH, iHighLimit)+MINMATCH; // stronger
while ((match+back > prefixStart) && (ip+back > iLowLimit) && (ip[back-1] == match[back-1])) back--;
currentMl += (U32)(-back);
}
2016-01-31 02:25:48 -08:00
} else {
match = dictBase + matchIndex;
2016-02-19 02:05:25 -08:00
if (MEM_readMINMATCH(match) == MEM_readMINMATCH(ip)) { /* assumption : matchIndex <= dictLimit-4 (by table construction) */
2016-02-04 11:53:54 -08:00
currentMl = ZSTD_count_2segments(ip+MINMATCH, match+MINMATCH, iHighLimit, dictEnd, prefixStart) + MINMATCH;
2016-02-19 02:05:25 -08:00
while ((match+back > dictStart) && (ip+back > iLowLimit) && (ip[back-1] == match[back-1])) back--; /* backward match extension */
currentMl += (U32)(-back);
} }
2016-01-31 02:25:48 -08:00
/* save best solution */
2016-02-10 04:37:52 -08:00
if (currentMl > minml) {
minml = currentMl;
2016-01-31 02:25:48 -08:00
matches[mnum].off = current - matchIndex;
2016-02-09 11:47:39 -08:00
matches[mnum].len = (U32)currentMl;
2016-02-19 02:05:25 -08:00
matches[mnum].back = (U32)(-back);
2016-01-31 02:25:48 -08:00
mnum++;
2016-02-03 10:08:48 -08:00
if (currentMl > ZSTD_OPT_NUM) break;
2016-02-10 04:37:52 -08:00
if (ip+currentMl == iHighLimit) break; /* best possible, and avoid read overflow*/
2016-01-31 02:25:48 -08:00
}
if (matchIndex <= minChain) break;
matchIndex = NEXT_IN_CHAIN(matchIndex, chainMask);
}
return mnum;
}
static U32 ZSTD_HcGetAllMatches_selectMLS (
2016-01-31 02:25:48 -08:00
ZSTD_CCtx* zc,
2016-02-04 11:53:54 -08:00
const BYTE* ip, const BYTE* const iLowLimit, const BYTE* const iHighLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
2016-01-31 02:25:48 -08:00
{
switch(matchLengthSearch)
{
default :
2016-02-19 02:05:25 -08:00
case 4 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 4, 0, matches);
case 5 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 5, 0, matches);
case 6 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 6, 0, matches);
2016-01-31 02:25:48 -08:00
}
}
static U32 ZSTD_HcGetAllMatches_selectMLS_extDict (
ZSTD_CCtx* zc,
const BYTE* ip, const BYTE* const iLowLimit, const BYTE* const iHighLimit,
2016-02-19 02:05:25 -08:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
{
switch(matchLengthSearch)
{
default :
2016-02-19 02:05:25 -08:00
case 4 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 4, 1, matches);
case 5 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 5, 1, matches);
case 6 : return ZSTD_HcGetAllMatches_generic(zc, ip, iLowLimit, iHighLimit, maxNbAttempts, 6, 1, matches);
}
}
2016-01-31 02:25:48 -08:00
2016-01-31 02:57:01 -08:00
2016-02-10 05:26:30 -08:00
/*-*******************************
2016-01-31 02:57:01 -08:00
* Optimal parser
*********************************/
FORCE_INLINE
void ZSTD_compressBlock_opt_generic(ZSTD_CCtx* ctx,
2016-02-12 13:04:49 -08:00
const void* src, size_t srcSize,
const U32 searchMethod, const U32 depth)
2016-01-31 02:57:01 -08: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;
const BYTE* const base = ctx->base + ctx->dictLimit;
2016-02-09 11:47:39 -08:00
U32 rep_2=REPCODE_STARTVALUE, rep_1=REPCODE_STARTVALUE;
const U32 maxSearches = 1U << ctx->params.searchLog;
2016-01-31 02:57:01 -08:00
const U32 mls = ctx->params.searchLength;
2016-02-09 11:47:39 -08:00
typedef U32 (*getAllMatches_f)(ZSTD_CCtx* zc, const BYTE* ip, const BYTE* iLowLimit, const BYTE* iHighLimit,
2016-02-19 02:05:25 -08:00
U32 maxNbAttempts, U32 matchLengthSearch, ZSTD_match_t* matches);
2016-01-31 02:57:01 -08:00
getAllMatches_f getAllMatches = searchMethod ? ZSTD_BtGetAllMatches_selectMLS : ZSTD_HcGetAllMatches_selectMLS;
2016-02-03 10:08:48 -08:00
ZSTD_optimal_t opt[ZSTD_OPT_NUM+4];
ZSTD_match_t matches[ZSTD_OPT_NUM+1];
2016-02-10 04:37:52 -08:00
const BYTE* inr;
2016-02-09 11:47:39 -08:00
U32 skip_num, cur, cur2, match_num, last_pos, litlen, price;
2016-01-31 03:17:16 -08:00
const U32 sufficient_len = ctx->params.targetLength;
2016-02-10 04:37:52 -08:00
const U32 faster_get_matches = (ctx->params.strategy == ZSTD_opt);
2016-01-31 02:57:01 -08:00
/* init */
2016-02-19 02:05:25 -08:00
ctx->nextToUpdate3 = ctx->nextToUpdate;
2016-01-31 02:57:01 -08:00
ZSTD_resetSeqStore(seqStorePtr);
2016-02-19 02:05:25 -08:00
ZSTD_rescaleFreqs(seqStorePtr);
2016-01-31 02:57:01 -08:00
if ((ip-base) < REPCODE_STARTVALUE) ip = base + REPCODE_STARTVALUE;
/* Match Loop */
while (ip < ilimit) {
2016-02-10 04:37:52 -08:00
U32 u;
2016-02-09 11:47:39 -08:00
U32 mlen=0;
U32 best_mlen=0;
U32 best_off=0;
2016-02-03 10:08:48 -08:00
memset(opt, 0, sizeof(ZSTD_optimal_t));
2016-01-31 03:17:16 -08:00
last_pos = 0;
inr = ip;
2016-02-09 11:47:39 -08:00
opt[0].litlen = (U32)(ip - anchor);
2016-02-04 12:45:26 -08:00
2016-01-31 02:57:01 -08:00
/* check repCode */
2016-02-19 02:05:25 -08:00
if (MEM_readMINMATCH(ip+1) == MEM_readMINMATCH(ip+1 - rep_1)) {
2016-01-31 02:57:01 -08:00
/* repcode : we take it */
2016-02-09 11:47:39 -08:00
mlen = (U32)ZSTD_count(ip+1+MINMATCH, ip+1+MINMATCH-rep_1, iend) + MINMATCH;
2016-02-10 04:37:52 -08:00
2016-02-03 10:08:48 -08:00
ZSTD_LOG_PARSER("%d: start try REP rep=%d mlen=%d\n", (int)(ip-base), (int)rep_1, (int)mlen);
if (depth==0 || mlen > sufficient_len || mlen >= ZSTD_OPT_NUM) {
2016-02-03 04:30:34 -08:00
ip+=1; best_mlen = mlen; best_off = 0; cur = 0; last_pos = 1;
2016-01-31 03:17:16 -08:00
goto _storeSequence;
}
2016-02-04 07:02:05 -08:00
litlen = opt[0].litlen + 1;
do {
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, 0, mlen);
2016-02-03 04:30:34 -08:00
if (mlen + 1 > last_pos || price < opt[mlen + 1].price)
SET_PRICE(mlen + 1, mlen, 0, litlen, price); /* note : macro modifies last_pos */
2016-01-31 03:17:16 -08:00
mlen--;
} while (mlen >= MINMATCH);
2016-01-31 02:57:01 -08:00
}
2016-02-03 04:30:34 -08:00
2016-02-10 19:17:50 -08:00
if (faster_get_matches && last_pos)
2016-01-31 05:28:26 -08:00
match_num = 0;
2016-02-10 19:17:50 -08:00
else
2016-02-19 02:05:25 -08:00
match_num = getAllMatches(ctx, ip, ip, iend, maxSearches, mls, matches); /* first search (depth 0) */
2016-01-31 05:28:26 -08:00
2016-02-10 19:17:50 -08:00
ZSTD_LOG_PARSER("%d: match_num=%d last_pos=%d\n", (int)(ip-base), match_num, last_pos);
if (!last_pos && !match_num) { ip++; continue; }
2016-01-31 05:28:26 -08:00
2016-02-10 19:17:50 -08:00
opt[0].rep = rep_1;
opt[0].rep2 = rep_2;
opt[0].mlen = 1;
2016-02-04 06:25:37 -08:00
2016-02-10 19:17:50 -08:00
if (match_num && matches[match_num-1].len > sufficient_len) {
2016-01-31 05:28:26 -08:00
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
cur = 0;
last_pos = 1;
goto _storeSequence;
2016-02-10 19:17:50 -08:00
}
2016-01-31 05:28:26 -08:00
2016-02-19 02:05:25 -08:00
best_mlen = (last_pos) ? last_pos : MINMATCH;
2016-01-31 05:28:26 -08:00
// set prices using matches at position = 0
2016-02-10 04:37:52 -08:00
for (u = 0; u < match_num; u++) {
mlen = (u>0) ? matches[u-1].len+1 : best_mlen;
best_mlen = (matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM;
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: start Found mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(ip-base), matches[u].len, matches[u].off, (int)best_mlen, (int)last_pos);
2016-02-04 07:02:05 -08:00
litlen = opt[0].litlen;
while (mlen <= best_mlen) {
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, matches[u].off, mlen);
2016-01-31 05:28:26 -08:00
if (mlen > last_pos || price < opt[mlen].price)
2016-02-10 04:37:52 -08:00
SET_PRICE(mlen, mlen, matches[u].off, litlen, price);
2016-01-31 05:28:26 -08:00
mlen++;
} }
2016-01-31 02:57:01 -08:00
if (last_pos < MINMATCH) { ip++; continue; }
2016-02-02 06:41:16 -08:00
/* check further positions */
2016-02-10 19:17:50 -08:00
for (skip_num = 0, cur = 1; cur <= last_pos; cur++) {
2016-02-10 04:37:52 -08:00
size_t cur_rep;
2016-01-31 06:17:50 -08:00
inr = ip + cur;
2016-02-10 04:37:52 -08:00
if (opt[cur-1].mlen == 1) {
2016-01-31 06:17:50 -08:00
litlen = opt[cur-1].litlen + 1;
2016-02-10 04:37:52 -08:00
if (cur > litlen) {
2016-02-03 10:53:29 -08:00
price = opt[cur - litlen].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-litlen);
2016-02-03 10:08:48 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY1 opt[%d].price=%d price=%d cur=%d litlen=%d\n", (int)(inr-base), cur - litlen, opt[cur - litlen].price, price, cur, litlen);
2016-02-10 04:37:52 -08:00
} else
2016-02-04 06:52:35 -08:00
price = ZSTD_getLiteralPrice(seqStorePtr, litlen, anchor);
2016-02-10 04:37:52 -08:00
} else {
2016-01-31 06:17:50 -08:00
litlen = 1;
2016-02-10 04:37:52 -08:00
price = opt[cur - 1].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1);
2016-02-04 05:55:23 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY3 price=%d cur=%d litlen=%d litonly=%d\n", (int)(inr-base), price, cur, litlen, (int)ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1));
2016-01-31 06:17:50 -08:00
}
2016-02-10 04:37:52 -08:00
2016-02-03 10:08:48 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY4 price=%d opt[%d].price=%d\n", (int)(inr-base), price, cur, opt[cur].price);
2016-01-31 06:17:50 -08:00
if (cur > last_pos || price <= opt[cur].price) // || ((price == opt[cur].price) && (opt[cur-1].mlen == 1) && (cur != litlen)))
2016-02-04 11:53:54 -08:00
SET_PRICE(cur, 1, 0, litlen, price);
2016-01-31 02:57:01 -08:00
2016-02-04 05:55:23 -08:00
if (cur == last_pos) break;
2016-02-10 19:17:50 -08:00
if (inr > ilimit) /* last match must start at a minimum distance of 8 from oend */
continue;
2016-02-04 06:28:55 -08:00
2016-02-04 06:25:37 -08:00
mlen = opt[cur].mlen;
2016-02-10 04:37:52 -08:00
if (opt[cur-mlen].off) {
2016-02-04 06:25:37 -08:00
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].off;
2016-02-04 06:28:55 -08:00
ZSTD_LOG_PARSER("%d: COPYREP1 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} else {
if (cur!=mlen && opt[cur-mlen].litlen == 0) {
2016-02-04 06:25:37 -08:00
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].rep2;
2016-02-04 11:53:54 -08:00
ZSTD_LOG_PARSER("%d: COPYREP2 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} else {
2016-02-04 06:25:37 -08:00
opt[cur].rep2 = opt[cur-mlen].rep2;
opt[cur].rep = opt[cur-mlen].rep;
2016-02-04 11:53:54 -08:00
ZSTD_LOG_PARSER("%d: COPYREP3 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} }
2016-02-03 04:30:34 -08:00
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: CURRENT price[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep, opt[cur].rep2);
2016-02-04 06:28:55 -08:00
2016-02-06 06:18:26 -08:00
best_mlen = 0;
2016-02-04 08:48:33 -08:00
if (!opt[cur].off && opt[cur].mlen != 1) {
2016-02-06 06:18:26 -08:00
cur_rep = opt[cur].rep2;
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: try REP2 rep2=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-10 04:37:52 -08:00
} else {
2016-02-06 06:18:26 -08:00
cur_rep = opt[cur].rep;
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: try REP1 rep=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-04 08:48:33 -08:00
}
2016-02-19 02:05:25 -08:00
if (MEM_readMINMATCH(inr) == MEM_readMINMATCH(inr - cur_rep)) { // check rep
2016-02-10 19:17:50 -08:00
mlen = (U32)ZSTD_count(inr+MINMATCH, inr+MINMATCH - cur_rep, iend) + MINMATCH;
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d rep=%d opt[%d].off=%d\n", (int)(inr-base), mlen, 0, opt[cur].rep, cur, opt[cur].off);
2016-01-31 06:17:50 -08:00
2016-02-10 19:17:50 -08:00
if (mlen > sufficient_len || cur + mlen >= ZSTD_OPT_NUM) {
best_mlen = mlen;
best_off = 0;
ZSTD_LOG_PARSER("%d: REP sufficient_len=%d best_mlen=%d best_off=%d last_pos=%d\n", (int)(inr-base), sufficient_len, best_mlen, best_off, last_pos);
last_pos = cur + 1;
goto _storeSequence;
2016-01-31 06:17:50 -08:00
}
2016-02-10 04:37:52 -08:00
if (opt[cur].mlen == 1) {
2016-01-31 06:17:50 -08:00
litlen = opt[cur].litlen;
2016-02-10 04:37:52 -08:00
if (cur > litlen) {
2016-02-19 02:05:25 -08:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, 0, mlen);
2016-02-03 10:08:48 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY5 opt[%d].price=%d price=%d cur=%d litlen=%d\n", (int)(inr-base), cur - litlen, opt[cur - litlen].price, price, cur, litlen);
2016-02-10 04:37:52 -08:00
} else
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, 0, mlen);
2016-02-10 04:37:52 -08:00
} else {
2016-01-31 06:17:50 -08:00
litlen = 0;
2016-02-19 02:05:25 -08:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen);
ZSTD_LOG_TRY_PRICE("%d: TRY7 price=%d cur=%d litlen=0 getprice=%d\n", (int)(inr-base), price, cur, (int)ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen));
2016-01-31 06:17:50 -08:00
}
best_mlen = mlen;
if (faster_get_matches) skip_num = best_mlen;
2016-02-19 02:05:25 -08:00
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, 0, price, litlen);
2016-01-31 06:17:50 -08:00
2016-02-10 04:37:52 -08:00
do {
if (cur + mlen > last_pos || price <= opt[cur + mlen].price)
2016-01-31 06:17:50 -08:00
SET_PRICE(cur + mlen, mlen, 0, litlen, price);
mlen--;
2016-02-10 04:37:52 -08:00
} while (mlen >= MINMATCH);
2016-01-31 06:17:50 -08:00
}
2016-02-03 04:30:34 -08:00
2016-02-10 04:37:52 -08:00
if (faster_get_matches && skip_num > 0) { skip_num--; continue; }
2016-01-31 06:17:50 -08:00
2016-02-19 02:05:25 -08:00
match_num = getAllMatches(ctx, inr, ip, iend, maxSearches, mls, matches);
2016-02-03 10:08:48 -08:00
ZSTD_LOG_PARSER("%d: ZSTD_GetAllMatches match_num=%d\n", (int)(inr-base), match_num);
2016-01-31 06:17:50 -08:00
2016-02-10 04:37:52 -08:00
if (match_num > 0 && matches[match_num-1].len > sufficient_len) {
2016-01-31 06:17:50 -08:00
cur -= matches[match_num-1].back;
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
last_pos = cur + 1;
2016-01-31 06:32:01 -08:00
goto _storeSequence;
2016-01-31 06:17:50 -08:00
}
2016-02-19 02:05:25 -08:00
best_mlen = (best_mlen > MINMATCH) ? best_mlen : MINMATCH;
2016-02-10 04:37:52 -08:00
/* set prices using matches at position = cur */
for (u = 0; u < match_num; u++) {
mlen = (u>0) ? matches[u-1].len+1 : best_mlen;
cur2 = cur - matches[u].back;
best_mlen = (cur2 + matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM - cur2;
2016-02-02 06:41:16 -08:00
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: Found1 cur=%d cur2=%d mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(inr-base), cur, cur2, matches[u].len, matches[u].off, best_mlen, last_pos);
2016-02-10 04:37:52 -08:00
if (mlen < matches[u].back + 1)
mlen = matches[u].back + 1;
2016-01-31 06:17:50 -08:00
2016-02-10 04:37:52 -08:00
while (mlen <= best_mlen) {
if (opt[cur2].mlen == 1) {
2016-01-31 06:17:50 -08:00
litlen = opt[cur2].litlen;
2016-02-04 05:55:23 -08:00
if (cur2 > litlen)
2016-02-19 02:05:25 -08:00
price = opt[cur2 - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, ip+cur2-litlen, matches[u].off, mlen);
2016-01-31 06:17:50 -08:00
else
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, matches[u].off, mlen);
2016-02-10 04:37:52 -08:00
} else {
2016-01-31 06:17:50 -08:00
litlen = 0;
2016-02-19 02:05:25 -08:00
price = opt[cur2].price + ZSTD_getPrice(seqStorePtr, 0, NULL, matches[u].off, mlen);
2016-01-31 06:17:50 -08:00
}
2016-02-19 02:05:25 -08:00
ZSTD_LOG_PARSER("%d: Found2 pred=%d mlen=%d best_mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), matches[u].back, mlen, best_mlen, matches[u].off, price, litlen);
2016-02-04 11:53:54 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY8 price=%d opt[%d].price=%d\n", (int)(inr-base), price, cur2 + mlen, opt[cur2 + mlen].price);
2016-02-03 07:54:23 -08:00
if (cur2 + mlen > last_pos || (price < opt[cur2 + mlen].price))
2016-02-10 04:37:52 -08:00
SET_PRICE(cur2 + mlen, mlen, matches[u].off, litlen, price);
2016-01-31 06:17:50 -08:00
mlen++;
2016-02-10 19:17:50 -08:00
} } } // for (skip_num = 0, cur = 1; cur <= last_pos; cur++)
2016-01-31 02:57:01 -08:00
2016-01-31 05:28:26 -08:00
best_mlen = opt[last_pos].mlen;
best_off = opt[last_pos].off;
2016-01-31 06:17:50 -08:00
cur = last_pos - best_mlen;
2016-02-10 04:37:52 -08:00
// printf("%d: start=%d best_mlen=%d best_off=%d cur=%d\n", (int)(ip - base), (int)(start - ip), (int)best_mlen, (int)best_off, cur);
2016-01-31 03:36:41 -08:00
2016-01-31 02:57:01 -08:00
/* store sequence */
_storeSequence: /* cur, last_pos, best_mlen, best_off have to be set */
2016-02-10 04:37:52 -08:00
for (u = 1; u <= last_pos; u++)
ZSTD_LOG_PARSER("%d: price[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep, opt[u].rep2);
ZSTD_LOG_PARSER("%d: cur=%d/%d best_mlen=%d best_off=%d rep=%d\n", (int)(ip-base+cur), (int)cur, (int)last_pos, (int)best_mlen, (int)best_off, opt[cur].rep);
2016-01-31 03:36:41 -08:00
opt[0].mlen = 1;
2016-02-09 11:47:39 -08:00
U32 offset;
2016-02-04 11:53:54 -08:00
2016-02-10 04:37:52 -08:00
while (1) {
2016-01-31 03:36:41 -08:00
mlen = opt[cur].mlen;
2016-02-04 11:53:54 -08:00
ZSTD_LOG_PARSER("%d: cur=%d mlen=%d\n", (int)(ip-base), cur, mlen);
2016-01-31 03:36:41 -08:00
offset = opt[cur].off;
2016-02-10 04:37:52 -08:00
opt[cur].mlen = best_mlen;
2016-01-31 03:36:41 -08:00
opt[cur].off = best_off;
best_mlen = mlen;
2016-02-10 04:37:52 -08:00
best_off = offset;
if (mlen > cur) break;
2016-01-31 03:36:41 -08:00
cur -= mlen;
}
2016-02-10 04:37:52 -08:00
for (u = 0; u <= last_pos;) {
ZSTD_LOG_PARSER("%d: price2[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep, opt[u].rep2);
u += opt[u].mlen;
2016-01-31 03:36:41 -08:00
}
2016-02-10 19:17:50 -08:00
for (cur=0; cur < last_pos; ) {
2016-02-04 11:53:54 -08:00
ZSTD_LOG_PARSER("%d: price3[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+cur), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep, opt[cur].rep2);
2016-01-31 03:36:41 -08:00
mlen = opt[cur].mlen;
if (mlen == 1) { ip++; cur++; continue; }
offset = opt[cur].off;
cur += mlen;
2016-02-09 11:47:39 -08:00
U32 litLength = (U32)(ip - anchor);
2016-02-19 02:05:25 -08:00
ZSTD_LOG_ENCODE("%d/%d: ENCODE literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep_1, (int)rep_2);
2016-02-04 08:48:33 -08:00
2016-02-10 19:17:50 -08:00
if (offset) {
2016-02-03 03:53:07 -08:00
rep_2 = rep_1;
rep_1 = offset;
2016-02-10 19:17:50 -08:00
} else {
if (litLength == 0) {
2016-02-03 03:53:07 -08:00
best_off = rep_2;
rep_2 = rep_1;
rep_1 = best_off;
2016-02-10 19:17:50 -08:00
} }
2016-02-03 03:53:07 -08:00
2016-02-19 02:05:25 -08:00
// ZSTD_LOG_ENCODE("%d/%d: ENCODE2 literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep_1, (int)rep_2);
2016-02-10 04:37:52 -08:00
2016-02-05 00:07:22 -08:00
#if ZSTD_OPT_DEBUG >= 5
2016-02-19 02:05:25 -08:00
U32 ml2;
2016-01-31 05:28:26 -08:00
if (offset)
2016-02-19 02:05:25 -08:00
ml2 = (U32)ZSTD_count(ip, ip-offset, iend);
2016-02-02 15:04:14 -08:00
else
2016-02-19 02:05:25 -08:00
ml2 = (U32)ZSTD_count(ip, ip-rep_1, iend);
if (offset == 0 || offset >= 8)
2016-02-06 06:18:26 -08:00
if (ml2 < mlen && ml2 < MINMATCH) {
printf("%d: ERROR iend=%d mlen=%d offset=%d ml2=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset, (int)ml2); exit(0); }
if (ip < anchor) {
printf("%d: ERROR ip < anchor iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (ip - offset < ctx->base) {
printf("%d: ERROR ip - offset < base iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if ((int)offset >= (1 << ctx->params.windowLog)) {
printf("%d: offset >= (1 << params.windowLog) iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (mlen < MINMATCH) {
printf("%d: ERROR mlen < MINMATCH iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (ip + mlen > iend) {
printf("%d: ERROR ip + mlen >= iend iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
2016-01-31 05:28:26 -08:00
#endif
2016-02-03 09:47:30 -08:00
ZSTD_updatePrice(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
2016-01-31 03:36:41 -08:00
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
2016-01-31 06:17:50 -08:00
anchor = ip = ip + mlen;
} /* for (cur=0; cur < last_pos; ) */
2016-01-31 02:57:01 -08:00
2016-02-10 19:17:50 -08:00
/* check immediate repcode */
2016-02-02 10:47:09 -08:00
while ( (anchor <= ilimit)
2016-02-19 02:05:25 -08:00
&& (MEM_readMINMATCH(anchor) == MEM_readMINMATCH(anchor - rep_2)) ) {
2016-01-31 02:57:01 -08:00
/* store sequence */
2016-02-09 11:47:39 -08:00
best_mlen = (U32)ZSTD_count(anchor+MINMATCH, anchor+MINMATCH-rep_2, iend);
2016-01-31 03:36:41 -08:00
best_off = rep_2;
2016-01-31 03:17:16 -08:00
rep_2 = rep_1;
2016-01-31 03:36:41 -08:00
rep_1 = best_off;
2016-02-03 10:08:48 -08:00
ZSTD_LOG_ENCODE("%d/%d: ENCODE REP literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(anchor-base), (int)(iend-base), (int)(0), (int)best_mlen, (int)(0), (int)rep_1, (int)rep_2);
ZSTD_updatePrice(seqStorePtr, 0, anchor, 0, best_mlen);
2016-01-31 03:17:16 -08:00
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, best_mlen);
2016-02-02 10:47:09 -08:00
anchor += best_mlen+MINMATCH;
continue; /* faster when present ... (?) */
}
if (anchor > ip) ip = anchor;
}
2016-01-31 02:57:01 -08:00
2016-02-10 19:17:50 -08:00
{ /* Last Literals */
size_t lastLLSize = iend - anchor;
ZSTD_LOG_ENCODE("%d: lastLLSize literals=%u\n", (int)(ip-base), (U32)lastLLSize);
2016-01-31 02:57:01 -08:00
memcpy(seqStorePtr->lit, anchor, lastLLSize);
seqStorePtr->lit += lastLLSize;
}
}
FORCE_INLINE
void ZSTD_compressBlock_opt_extDict_generic(ZSTD_CCtx* ctx,
const void* src, size_t srcSize,
const U32 searchMethod, const U32 depth)
{
seqStore_t* seqStorePtr = &(ctx->seqStore);
const BYTE* const istart = (const BYTE*)src;
const BYTE* ip = istart;
const BYTE* anchor = istart;
const BYTE* const iend = istart + srcSize;
const BYTE* const ilimit = iend - 8;
const BYTE* const base = ctx->base;
const U32 dictLimit = ctx->dictLimit;
const BYTE* const prefixStart = base + dictLimit;
const BYTE* const dictBase = ctx->dictBase;
const BYTE* const dictEnd = dictBase + dictLimit;
U32 rep_2=REPCODE_STARTVALUE, rep_1=REPCODE_STARTVALUE;
const U32 maxSearches = 1U << ctx->params.searchLog;
const U32 mls = ctx->params.searchLength;
typedef U32 (*getAllMatches_f)(ZSTD_CCtx* zc, const BYTE* ip, const BYTE* iLowLimit, const BYTE* iHighLimit,
2016-02-19 02:05:25 -08:00
U32 maxNbAttempts, U32 matchLengthSearch, ZSTD_match_t* matches);
getAllMatches_f getAllMatches = searchMethod ? ZSTD_BtGetAllMatches_selectMLS_extDict : ZSTD_HcGetAllMatches_selectMLS_extDict;
ZSTD_optimal_t opt[ZSTD_OPT_NUM+4];
ZSTD_match_t matches[ZSTD_OPT_NUM+1];
2016-02-10 04:37:52 -08:00
const BYTE* inr;
U32 skip_num, cur, cur2, match_num, last_pos, litlen, price;
const U32 sufficient_len = ctx->params.targetLength;
2016-02-10 04:37:52 -08:00
const U32 faster_get_matches = (ctx->params.strategy == ZSTD_opt);
/* init */
2016-02-19 02:05:25 -08:00
ctx->nextToUpdate3 = ctx->nextToUpdate;
ZSTD_resetSeqStore(seqStorePtr);
2016-02-19 02:05:25 -08:00
ZSTD_rescaleFreqs(seqStorePtr);
if ((ip - prefixStart) < REPCODE_STARTVALUE) ip += REPCODE_STARTVALUE;
/* Match Loop */
while (ip < ilimit) {
2016-02-10 19:17:50 -08:00
U32 u, offset, best_off=0;
U32 mlen=0, best_mlen=0;
U32 current = (U32)(ip-base);
memset(opt, 0, sizeof(ZSTD_optimal_t));
last_pos = 0;
inr = ip;
opt[0].litlen = (U32)(ip - anchor);
/* check repCode */
{
const U32 repIndex = (U32)(current+1 - rep_1);
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
const BYTE* const repMatch = repBase + repIndex;
2016-02-10 05:26:30 -08:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
2016-02-19 02:05:25 -08:00
&& (MEM_readMINMATCH(ip+1) == MEM_readMINMATCH(repMatch)) ) {
/* repcode detected we should take it */
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
mlen = (U32)ZSTD_count_2segments(ip+1+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: start try REP rep=%d mlen=%d\n", (int)(ip-base), (int)rep_1, (int)mlen);
if (depth==0 || mlen > sufficient_len || mlen >= ZSTD_OPT_NUM) {
ip+=1; best_mlen = mlen; best_off = 0; cur = 0; last_pos = 1;
goto _storeSequence;
}
litlen = opt[0].litlen + 1;
2016-02-10 04:37:52 -08:00
do {
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, 0, mlen);
if (mlen + 1 > last_pos || price < opt[mlen + 1].price)
SET_PRICE(mlen + 1, mlen, 0, litlen, price);
mlen--;
2016-02-10 04:37:52 -08:00
} while (mlen >= MINMATCH);
} }
best_mlen = (last_pos) ? last_pos : MINMATCH;
2016-02-10 04:37:52 -08:00
if (faster_get_matches && last_pos)
match_num = 0;
else
2016-02-19 02:05:25 -08:00
match_num = getAllMatches(ctx, ip, ip, iend, maxSearches, mls, matches); /* first search (depth 0) */
ZSTD_LOG_PARSER("%d: match_num=%d last_pos=%d\n", (int)(ip-base), match_num, last_pos);
if (!last_pos && !match_num) { ip++; continue; }
opt[0].rep = rep_1;
opt[0].rep2 = rep_2;
opt[0].mlen = 1;
2016-02-10 04:37:52 -08:00
if (match_num && matches[match_num-1].len > sufficient_len) {
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
cur = 0;
last_pos = 1;
goto _storeSequence;
}
2016-02-10 05:26:30 -08:00
// set prices using matches at position = 0
for (u = 0; u < match_num; u++) {
mlen = (u>0) ? matches[u-1].len+1 : best_mlen;
best_mlen = (matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM;
ZSTD_LOG_PARSER("%d: start Found mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(ip-base), matches[u].len, matches[u].off, (int)best_mlen, (int)last_pos);
litlen = opt[0].litlen;
while (mlen <= best_mlen) {
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, matches[u].off, mlen);
if (mlen > last_pos || price < opt[mlen].price)
2016-02-10 04:37:52 -08:00
SET_PRICE(mlen, mlen, matches[u].off, litlen, price);
mlen++;
2016-02-10 05:26:30 -08:00
} }
2016-02-10 04:37:52 -08:00
if (last_pos < MINMATCH) {
// ip += ((ip-anchor) >> g_searchStrength) + 1; /* jump faster over incompressible sections */
ip++; continue;
}
2016-02-10 19:17:50 -08:00
/* check further positions */
2016-02-10 05:26:30 -08:00
for (skip_num = 0, cur = 1; cur <= last_pos; cur++) {
2016-02-10 04:37:52 -08:00
size_t cur_rep;
inr = ip + cur;
2016-02-10 04:37:52 -08:00
if (opt[cur-1].mlen == 1) {
litlen = opt[cur-1].litlen + 1;
2016-02-10 04:37:52 -08:00
if (cur > litlen) {
price = opt[cur - litlen].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-litlen);
ZSTD_LOG_TRY_PRICE("%d: TRY1 opt[%d].price=%d price=%d cur=%d litlen=%d\n", (int)(inr-base), cur - litlen, opt[cur - litlen].price, price, cur, litlen);
2016-02-10 04:37:52 -08:00
} else
price = ZSTD_getLiteralPrice(seqStorePtr, litlen, anchor);
2016-02-10 04:37:52 -08:00
} else {
litlen = 1;
2016-02-10 04:37:52 -08:00
price = opt[cur - 1].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1);
ZSTD_LOG_TRY_PRICE("%d: TRY3 price=%d cur=%d litlen=%d litonly=%d\n", (int)(inr-base), price, cur, litlen, (int)ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1));
}
2016-02-10 04:37:52 -08:00
ZSTD_LOG_TRY_PRICE("%d: TRY4 price=%d opt[%d].price=%d\n", (int)(inr-base), price, cur, opt[cur].price);
if (cur > last_pos || price <= opt[cur].price) // || ((price == opt[cur].price) && (opt[cur-1].mlen == 1) && (cur != litlen)))
SET_PRICE(cur, 1, 0, litlen, price);
if (cur == last_pos) break;
if (inr > ilimit) // last match must start at a minimum distance of 8 from oend
continue;
mlen = opt[cur].mlen;
2016-02-10 04:37:52 -08:00
if (opt[cur-mlen].off) {
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].off;
ZSTD_LOG_PARSER("%d: COPYREP1 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} else {
if (cur!=mlen && opt[cur-mlen].litlen == 0) {
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].rep2;
ZSTD_LOG_PARSER("%d: COPYREP2 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} else {
opt[cur].rep2 = opt[cur-mlen].rep2;
opt[cur].rep = opt[cur-mlen].rep;
ZSTD_LOG_PARSER("%d: COPYREP3 cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 04:37:52 -08:00
} }
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: CURRENT price[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep, opt[cur].rep2);
best_mlen = 0;
if (!opt[cur].off && opt[cur].mlen != 1) {
cur_rep = opt[cur].rep2;
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: try REP2 rep2=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-10 05:26:30 -08:00
} else {
cur_rep = opt[cur].rep;
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: try REP1 rep=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
}
const U32 repIndex = (U32)(current+cur - cur_rep);
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
const BYTE* const repMatch = repBase + repIndex;
2016-02-10 19:17:50 -08:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
2016-02-19 02:05:25 -08:00
&&(MEM_readMINMATCH(inr) == MEM_readMINMATCH(repMatch)) ) {
/* repcode detected */
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
mlen = (U32)ZSTD_count_2segments(inr+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d rep=%d opt[%d].off=%d\n", (int)(inr-base), mlen, 0, opt[cur].rep, cur, opt[cur].off);
2016-02-10 05:26:30 -08:00
if (mlen > sufficient_len || cur + mlen >= ZSTD_OPT_NUM) {
best_mlen = mlen;
best_off = 0;
ZSTD_LOG_PARSER("%d: REP sufficient_len=%d best_mlen=%d best_off=%d last_pos=%d\n", (int)(inr-base), sufficient_len, best_mlen, best_off, last_pos);
last_pos = cur + 1;
goto _storeSequence;
2016-02-10 19:17:50 -08:00
}
2016-02-10 05:26:30 -08:00
if (opt[cur].mlen == 1) {
litlen = opt[cur].litlen;
2016-02-10 05:26:30 -08:00
if (cur > litlen) {
2016-02-19 02:05:25 -08:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, 0, mlen);
ZSTD_LOG_TRY_PRICE("%d: TRY5 opt[%d].price=%d price=%d cur=%d litlen=%d\n", (int)(inr-base), cur - litlen, opt[cur - litlen].price, price, cur, litlen);
2016-02-10 05:26:30 -08:00
} else
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, 0, mlen);
2016-02-10 05:26:30 -08:00
} else {
litlen = 0;
2016-02-19 02:05:25 -08:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen);
ZSTD_LOG_TRY_PRICE("%d: TRY7 price=%d cur=%d litlen=0 getprice=%d\n", (int)(inr-base), price, cur, (int)ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen));
}
best_mlen = mlen;
2016-02-10 19:17:50 -08:00
if (faster_get_matches) skip_num = best_mlen;
2016-02-19 02:05:25 -08:00
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, 0, price, litlen);
2016-02-10 05:26:30 -08:00
do {
if (cur + mlen > last_pos || price <= opt[cur + mlen].price) // || ((price == opt[cur + mlen].price) && (opt[cur].mlen == 1) && (cur != litlen))) // at equal price prefer REP instead of MATCH
SET_PRICE(cur + mlen, mlen, 0, litlen, price);
mlen--;
2016-02-10 05:26:30 -08:00
} while (mlen >= MINMATCH);
}
2016-02-10 05:26:30 -08:00
if (faster_get_matches && skip_num > 0) { skip_num--; continue; }
2016-02-10 04:37:52 -08:00
best_mlen = (best_mlen > MINMATCH) ? best_mlen : MINMATCH;
2016-02-19 02:05:25 -08:00
match_num = getAllMatches(ctx, inr, ip, iend, maxSearches, mls, matches);
ZSTD_LOG_PARSER("%d: ZSTD_GetAllMatches match_num=%d\n", (int)(inr-base), match_num);
2016-02-10 04:37:52 -08:00
if (match_num > 0 && matches[match_num-1].len > sufficient_len) {
cur -= matches[match_num-1].back;
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
last_pos = cur + 1;
goto _storeSequence;
}
// set prices using matches at position = cur
2016-02-10 04:37:52 -08:00
for (u = 0; u < match_num; u++) {
mlen = (u>0) ? matches[u-1].len+1 : best_mlen;
cur2 = cur - matches[u].back;
best_mlen = (cur2 + matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM - cur2;
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: Found1 cur=%d cur2=%d mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(inr-base), cur, cur2, matches[u].len, matches[u].off, best_mlen, last_pos);
if (mlen < matches[u].back + 1)
mlen = matches[u].back + 1;
2016-02-10 04:37:52 -08:00
while (mlen <= best_mlen) {
if (opt[cur2].mlen == 1) {
litlen = opt[cur2].litlen;
if (cur2 > litlen)
2016-02-19 02:05:25 -08:00
price = opt[cur2 - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, ip+cur2-litlen, matches[u].off, mlen);
else
2016-02-19 02:05:25 -08:00
price = ZSTD_getPrice(seqStorePtr, litlen, anchor, matches[u].off, mlen);
2016-02-10 04:37:52 -08:00
} else {
litlen = 0;
2016-02-19 02:05:25 -08:00
price = opt[cur2].price + ZSTD_getPrice(seqStorePtr, 0, NULL, matches[u].off, mlen);
}
2016-02-19 02:05:25 -08:00
ZSTD_LOG_PARSER("%d: Found2 pred=%d mlen=%d best_mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), matches[u].back, mlen, best_mlen, matches[u].off, price, litlen);
ZSTD_LOG_TRY_PRICE("%d: TRY8 price=%d opt[%d].price=%d\n", (int)(inr-base), price, cur2 + mlen, opt[cur2 + mlen].price);
if (cur2 + mlen > last_pos || (price < opt[cur2 + mlen].price))
2016-02-10 04:37:52 -08:00
SET_PRICE(cur2 + mlen, mlen, matches[u].off, litlen, price);
mlen++;
2016-02-10 19:17:50 -08:00
} } } // for (skip_num = 0, cur = 1; cur <= last_pos; cur++)
best_mlen = opt[last_pos].mlen;
best_off = opt[last_pos].off;
cur = last_pos - best_mlen;
2016-02-10 04:37:52 -08:00
// printf("%d: start=%d best_mlen=%d best_off=%d cur=%d\n", (int)(ip - base), (int)(start - ip), (int)best_mlen, (int)best_off, cur);
/* store sequence */
_storeSequence: // cur, last_pos, best_mlen, best_off have to be set
2016-02-10 04:37:52 -08:00
for (u = 1; u <= last_pos; u++)
2016-02-10 05:50:22 -08:00
ZSTD_LOG_PARSER("%d: price[%u/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep, opt[u].rep2);
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: cur=%d/%d best_mlen=%d best_off=%d rep=%d\n", (int)(ip-base+cur), (int)cur, (int)last_pos, (int)best_mlen, (int)best_off, opt[cur].rep);
opt[0].mlen = 1;
2016-02-10 04:37:52 -08:00
while (1) {
mlen = opt[cur].mlen;
ZSTD_LOG_PARSER("%d: cur=%d mlen=%d\n", (int)(ip-base), cur, mlen);
offset = opt[cur].off;
2016-02-10 04:37:52 -08:00
opt[cur].mlen = best_mlen;
opt[cur].off = best_off;
best_mlen = mlen;
2016-02-10 04:37:52 -08:00
best_off = offset;
2016-02-10 19:17:50 -08:00
if (mlen > cur) break;
cur -= mlen;
}
2016-02-10 04:37:52 -08:00
for (u = 0; u <= last_pos; ) {
2016-02-10 04:37:52 -08:00
ZSTD_LOG_PARSER("%d: price2[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep, opt[u].rep2);
u += opt[u].mlen;
}
for (cur=0; cur < last_pos; ) {
2016-02-10 04:37:52 -08:00
U32 litLength;
ZSTD_LOG_PARSER("%d: price3[%d/%d]=%d off=%d mlen=%d litlen=%d rep=%d rep2=%d\n", (int)(ip-base+cur), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep, opt[cur].rep2);
mlen = opt[cur].mlen;
if (mlen == 1) { ip++; cur++; continue; }
offset = opt[cur].off;
cur += mlen;
2016-02-10 04:37:52 -08:00
litLength = (U32)(ip - anchor);
ZSTD_LOG_ENCODE("%d/%d: ENCODE1 literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep_1, (int)rep_2);
2016-02-10 04:37:52 -08:00
if (offset) {
rep_2 = rep_1;
rep_1 = offset;
2016-02-10 04:37:52 -08:00
} else {
if (litLength == 0) {
best_off = rep_2;
rep_2 = rep_1;
rep_1 = best_off;
2016-02-10 04:37:52 -08:00
} }
ZSTD_LOG_ENCODE("%d/%d: ENCODE2 literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep_1, (int)rep_2);
2016-02-10 04:37:52 -08:00
#if ZSTD_OPT_DEBUG >= 5
2016-02-19 02:05:25 -08:00
U32 ml2;
if (offset)
2016-02-19 02:05:25 -08:00
ml2 = (U32)ZSTD_count(ip, ip-offset, iend);
else
2016-02-19 02:05:25 -08:00
ml2 = (U32)ZSTD_count(ip, ip-rep_1, iend);
if (ml2 < mlen && ml2 < MINMATCH) {
printf("%d: ERROR iend=%d mlen=%d offset=%d ml2=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset, (int)ml2); exit(0); }
if (ip < anchor) {
printf("%d: ERROR ip < anchor iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (ip - offset < ctx->base) {
printf("%d: ERROR ip - offset < base iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if ((int)offset >= (1 << ctx->params.windowLog)) {
printf("%d: offset >= (1 << params.windowLog) iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (mlen < MINMATCH) {
printf("%d: ERROR mlen < MINMATCH iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (ip + mlen > iend) {
printf("%d: ERROR ip + mlen >= iend iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
#endif
ZSTD_updatePrice(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
anchor = ip = ip + mlen;
}
2016-02-10 04:37:52 -08:00
/* check immediate repcode */
while (anchor <= ilimit) {
const U32 repIndex = (U32)((anchor-base) - rep_2);
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
const BYTE* const repMatch = repBase + repIndex;
2016-02-10 04:37:52 -08:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
2016-02-19 02:05:25 -08:00
&& (MEM_readMINMATCH(anchor) == MEM_readMINMATCH(repMatch)) ) {
2016-02-10 04:37:52 -08:00
/* repcode detected, let's take it */
const BYTE* const repEnd = repIndex < dictLimit ? dictEnd : iend;
mlen = (U32)ZSTD_count_2segments(anchor+MINMATCH, repMatch+MINMATCH, iend, repEnd, prefixStart) + MINMATCH;
offset = rep_2; rep_2 = rep_1; rep_1 = offset; /* swap offset history */
ZSTD_LOG_ENCODE("%d/%d: ENCODE REP literals=%d mlen=%d off=%d rep1=%d rep2=%d\n", (int)(anchor-base), (int)(iend-base), (int)(0), (int)best_mlen, (int)(0), (int)rep_1, (int)rep_2);
ZSTD_updatePrice(seqStorePtr, 0, anchor, 0, mlen-MINMATCH);
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, mlen-MINMATCH);
anchor += mlen;
continue; /* faster when present ... (?) */
}
break;
}
if (anchor > ip) ip = anchor;
}
2016-02-10 19:17:50 -08:00
{ /* Last Literals */
size_t lastLLSize = iend - anchor;
ZSTD_LOG_ENCODE("%d: lastLLSize literals=%u\n", (int)(ip-base), (U32)(lastLLSize));
memcpy(seqStorePtr->lit, anchor, lastLLSize);
seqStorePtr->lit += lastLLSize;
}
}