zstd/lib/zstd_opt.h

1091 lines
49 KiB
C
Raw Normal View History

2016-02-10 13:26:30 +00: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 */
2016-02-03 20:26:18 +00:00
2016-02-03 18:53:29 +00:00
2016-03-04 18:31:57 +00:00
#define ZSTD_FREQ_DIV 5
/*-*************************************
* Price functions for optimal parser
***************************************/
FORCE_INLINE void ZSTD_setLog2Prices(seqStore_t* ssPtr)
{
ssPtr->log2matchLengthSum = ZSTD_highbit(ssPtr->matchLengthSum+1);
ssPtr->log2litLengthSum = ZSTD_highbit(ssPtr->litLengthSum+1);
ssPtr->log2litSum = ZSTD_highbit(ssPtr->litSum+1);
ssPtr->log2offCodeSum = ZSTD_highbit(ssPtr->offCodeSum+1);
ssPtr->factor = 1 + ((ssPtr->litSum>>5) / ssPtr->litLengthSum) + ((ssPtr->litSum<<1) / (ssPtr->litSum + ssPtr->matchSum));
}
2016-03-04 18:31:57 +00:00
MEM_STATIC void ZSTD_rescaleFreqs(seqStore_t* ssPtr)
{
unsigned u;
if (ssPtr->litLengthSum == 0) {
ssPtr->litSum = (2<<Litbits);
ssPtr->litLengthSum = MaxLL+1;
ssPtr->matchLengthSum = (1<<MLbits);
ssPtr->offCodeSum = (1<<Offbits);
ssPtr->matchSum = (2<<Litbits);
2016-03-04 18:31:57 +00:00
for (u=0; u<=MaxLit; u++)
ssPtr->litFreq[u] = 2;
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->offCodeSum = 0;
ssPtr->matchSum = 0;
ssPtr->litSum = 0;
for (u=0; u<=MaxLit; u++) {
ssPtr->litFreq[u] = 1 + (ssPtr->litFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->litSum += ssPtr->litFreq[u];
}
for (u=0; u<=MaxLL; u++) {
ssPtr->litLengthFreq[u] = 1 + (ssPtr->litLengthFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->litLengthSum += ssPtr->litLengthFreq[u];
}
for (u=0; u<=MaxML; u++) {
ssPtr->matchLengthFreq[u] = 1 + (ssPtr->matchLengthFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->matchLengthSum += ssPtr->matchLengthFreq[u];
ssPtr->matchSum += ssPtr->matchLengthFreq[u] * (u + 3);
}
for (u=0; u<=MaxOff; u++) {
ssPtr->offCodeFreq[u] = 1 + (ssPtr->offCodeFreq[u]>>ZSTD_FREQ_DIV);
ssPtr->offCodeSum += ssPtr->offCodeFreq[u];
}
}
ZSTD_setLog2Prices(ssPtr);
2016-03-04 18:31:57 +00:00
}
FORCE_INLINE U32 ZSTD_getLiteralPrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals)
{
U32 price, u;
if (litLength == 0)
return seqStorePtr->log2litLengthSum - ZSTD_highbit(seqStorePtr->litLengthFreq[0]+1);
2016-03-04 18:31:57 +00:00
/* literals */
price = litLength * seqStorePtr->log2litSum;
2016-03-04 18:31:57 +00:00
for (u=0; u < litLength; u++)
price -= ZSTD_highbit(seqStorePtr->litFreq[literals[u]]+1);
/* literal Length */
price += ((litLength >= MaxLL)<<3) + ((litLength >= 255+MaxLL)<<4) + ((litLength>=(1<<15))<<3);
if (litLength >= MaxLL) litLength = MaxLL;
price += seqStorePtr->log2litLengthSum - ZSTD_highbit(seqStorePtr->litLengthFreq[litLength]+1);
2016-03-04 18:31:57 +00:00
return price;
}
FORCE_INLINE U32 ZSTD_getPrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals, U32 offset, U32 matchLength)
2016-02-04 09:04:33 +00:00
{
2016-02-10 12:37:52 +00:00
/* offset */
2016-02-24 17:09:36 +00:00
BYTE offCode = offset ? (BYTE)ZSTD_highbit(offset+1) + 1 : 0;
U32 price = (offCode-1) + (!offCode) + seqStorePtr->log2offCodeSum - ZSTD_highbit(seqStorePtr->offCodeFreq[offCode]+1);
2016-02-03 20:26:18 +00:00
/* match Length */
2016-02-19 10:05:25 +00:00
price += ((matchLength >= MaxML)<<3) + ((matchLength >= 255+MaxML)<<4) + ((matchLength>=(1<<15))<<3);
2016-02-10 12:37:52 +00:00
if (matchLength >= MaxML) matchLength = MaxML;
price += ZSTD_getLiteralPrice(seqStorePtr, litLength, literals) + seqStorePtr->log2matchLengthSum - ZSTD_highbit(seqStorePtr->matchLengthFreq[matchLength]+1);
2016-02-19 10:59:44 +00:00
2016-03-04 18:52:23 +00:00
#if ZSTD_OPT_DEBUG == 3
2016-03-01 13:14:35 +00:00
switch (seqStorePtr->priceFunc) {
2016-02-24 17:09:36 +00:00
default:
2016-02-23 15:25:04 +00:00
case 0:
2016-03-04 18:52:23 +00:00
return 1 + price + ((seqStorePtr->litSum>>5) / seqStorePtr->litLengthSum) + ((seqStorePtr->litSum<<1) / (seqStorePtr->litSum + seqStorePtr->matchSum));
2016-02-23 15:25:04 +00:00
case 1:
2016-02-24 17:09:36 +00:00
return 1 + price;
2016-02-23 15:25:04 +00:00
}
2016-02-25 15:47:57 +00:00
#else
return price + seqStorePtr->factor;
2016-02-25 15:47:57 +00:00
#endif
2016-02-04 09:04:33 +00:00
}
2016-02-04 10:11:08 +00:00
2016-03-04 18:31:57 +00:00
MEM_STATIC void ZSTD_updatePrice(seqStore_t* seqStorePtr, U32 litLength, const BYTE* literals, U32 offset, U32 matchLength)
{
U32 u;
/* literals */
seqStorePtr->litSum += litLength;
for (u=0; u < litLength; u++)
seqStorePtr->litFreq[literals[u]]++;
/* literal Length */
seqStorePtr->litLengthSum++;
if (litLength >= MaxLL)
seqStorePtr->litLengthFreq[MaxLL]++;
else
seqStorePtr->litLengthFreq[litLength]++;
/* match offset */
seqStorePtr->offCodeSum++;
BYTE offCode = offset ? (BYTE)ZSTD_highbit(offset+1) + 1 : 0;
seqStorePtr->offCodeFreq[offCode]++;
/* match Length */
seqStorePtr->matchLengthSum++;
if (matchLength >= MaxML)
seqStorePtr->matchLengthFreq[MaxML]++;
else
seqStorePtr->matchLengthFreq[matchLength]++;
ZSTD_setLog2Prices(seqStorePtr);
2016-03-04 18:31:57 +00:00
}
#define SET_PRICE(pos, mlen_, offset_, litlen_, price_) \
{ \
while (last_pos < pos) { opt[last_pos+1].price = 1<<30; last_pos++; } \
opt[pos].mlen = mlen_; \
opt[pos].off = offset_; \
opt[pos].litlen = litlen_; \
opt[pos].price = price_; \
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-02-10 13:26:30 +00:00
/*-*************************************
* Binary Tree search
***************************************/
2016-03-04 18:31:57 +00:00
/* Update hashTable3 up to ip (excluded)
Assumption : always within prefix (ie. not within extDict) */
static U32 ZSTD_insertAndFindFirstIndexHash3 (ZSTD_CCtx* zc, const BYTE* ip)
{
U32* const hashTable3 = zc->hashTable3;
const BYTE* const base = zc->base;
const U32 target = (U32)(ip - base);
U32 idx = zc->nextToUpdate3;
while(idx < target) {
2016-03-10 14:17:24 +00:00
hashTable3[ZSTD_hash3Ptr(base+idx, HASHLOG3)] = idx;
2016-03-04 18:31:57 +00:00
idx++;
}
zc->nextToUpdate3 = target;
2016-03-10 14:17:24 +00:00
return hashTable3[ZSTD_hash3Ptr(ip, HASHLOG3)];
2016-03-04 18:31:57 +00:00
}
static U32 ZSTD_insertBtAndGetAllMatches (
2016-01-31 10:25:48 +00:00
ZSTD_CCtx* zc,
2016-02-19 10:05:25 +00:00
const BYTE* const ip, const BYTE* const iLimit,
2016-01-31 10:25:48 +00:00
U32 nbCompares, const U32 mls,
2016-02-19 10:05:25 +00:00
U32 extDict, ZSTD_match_t* matches)
2016-01-31 10:25:48 +00:00
{
2016-02-06 14:18:26 +00:00
const BYTE* const base = zc->base;
const U32 current = (U32)(ip-base);
2016-01-31 10:25:48 +00:00
const U32 hashLog = zc->params.hashLog;
const size_t h = ZSTD_hashPtr(ip, hashLog, mls);
2016-02-06 14:18:26 +00:00
U32* const hashTable = zc->hashTable;
U32 matchIndex = hashTable[h];
2016-01-31 10:25:48 +00:00
U32* const bt = zc->contentTable;
const U32 btLog = zc->params.contentLog - 1;
2016-02-09 19:47:39 +00:00
const U32 btMask= (1U << btLog) - 1;
2016-01-31 10:25:48 +00: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 19:47:39 +00:00
U32 mnum = 0;
2016-02-10 12:37:52 +00:00
const U32 minMatch = (mls == 3) ? 3 : 4;
size_t bestLength = minMatch-1;
2016-01-31 10:25:48 +00:00
if (minMatch == 3) { /* HC3 match finder */
U32 matchIndex3 = ZSTD_insertAndFindFirstIndexHash3 (zc, ip);
if (matchIndex3>windowLow && (current - matchIndex3 < (1<<18))) {
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);
} else {
match = dictBase + matchIndex3;
if (MEM_readMINMATCH(match, minMatch) == MEM_readMINMATCH(ip, minMatch)) /* assumption : matchIndex3 <= dictLimit-4 (by table construction) */
currentMl = ZSTD_count_2segments(ip+minMatch, match+minMatch, iLimit, dictEnd, prefixStart) + minMatch;
}
2016-02-19 10:05:25 +00:00
/* save best solution */
if (currentMl > bestLength) {
bestLength = currentMl;
matches[mnum].off = current - matchIndex3;
matches[mnum].len = (U32)currentMl;
mnum++;
2016-03-07 15:14:58 +00:00
if (currentMl > ZSTD_OPT_NUM) goto update;
if (ip+currentMl == iLimit) goto update; /* best possible, and avoid read overflow*/
}
2016-02-19 10:05:25 +00:00
}
}
hashTable[h] = current; /* Update Hash Table */
2016-01-31 10:25:48 +00: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 10:05:25 +00:00
if (match[matchLength] == ip[matchLength]) {
#if ZSTD_OPT_DEBUG >= 5
2016-02-29 21:07:40 +00:00
size_t ml;
if (matchIndex < dictLimit)
ml = ZSTD_count_2segments(ip, dictBase + matchIndex, iLimit, dictEnd, prefixStart);
else
ml = ZSTD_count(ip, match, ip+matchLength);
if (ml < matchLength)
printf("%d: ERROR_NOEXT: offset=%d matchLength=%d matchIndex=%d dictLimit=%d ml=%d\n", current, (int)(current - matchIndex), (int)matchLength, (int)matchIndex, (int)dictLimit, (int)ml), exit(0);
2016-02-19 10:05:25 +00:00
#endif
matchLength += ZSTD_count(ip+matchLength+1, match+matchLength+1, iLimit) +1;
}
2016-01-31 10:25:48 +00:00
} else {
match = dictBase + matchIndex;
2016-02-29 21:07:40 +00:00
#if ZSTD_OPT_DEBUG >= 5
if (memcmp(match, ip, matchLength) != 0)
printf("%d: ERROR_EXT: matchLength=%d ZSTD_count=%d\n", current, (int)matchLength, (int)ZSTD_count_2segments(ip+matchLength, match+matchLength, iLimit, dictEnd, prefixStart)), exit(0);
#endif
2016-02-19 10:05:25 +00:00
matchLength += ZSTD_count_2segments(ip+matchLength, match+matchLength, iLimit, dictEnd, prefixStart);
2016-02-29 21:07:40 +00:00
ZSTD_LOG_PARSER("%d: ZSTD_INSERTBTANDGETALLMATCHES=%d offset=%d dictBase=%p dictEnd=%p prefixStart=%p ip=%p match=%p\n", (int)current, (int)matchLength, (int)(current - matchIndex), dictBase, dictEnd, prefixStart, ip, match);
2016-01-31 10:25:48 +00:00
if (matchIndex+matchLength >= dictLimit)
2016-02-03 12:30:34 +00:00
match = base + matchIndex; /* to prepare for next usage of match[matchLength] */
2016-01-31 10:25:48 +00:00
}
if (matchLength > bestLength) {
2016-02-10 12:37:52 +00:00
if (matchLength > matchEndIdx - matchIndex) matchEndIdx = matchIndex + (U32)matchLength;
bestLength = matchLength;
matches[mnum].off = current - matchIndex;
matches[mnum].len = (U32)matchLength;
mnum++;
if (matchLength > ZSTD_OPT_NUM) break;
2016-02-19 10:05:25 +00:00
if (ip+matchLength == iLimit) /* equal : no way to know if inf or sup */
2016-01-31 10:25:48 +00: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 12:37:52 +00:00
} }
2016-01-31 10:25:48 +00:00
*smallerPtr = *largerPtr = 0;
2016-03-07 15:14:58 +00:00
update:
2016-01-31 10:25:48 +00:00
zc->nextToUpdate = (matchEndIdx > current + 8) ? matchEndIdx - 8 : current+1;
return mnum;
}
/** Tree updater, providing best match */
2016-03-04 18:31:57 +00:00
static U32 ZSTD_BtGetAllMatches (
2016-01-31 10:25:48 +00:00
ZSTD_CCtx* zc,
const BYTE* const ip, const BYTE* const iLimit,
2016-02-19 10:05:25 +00:00
const U32 maxNbAttempts, const U32 mls, ZSTD_match_t* matches)
2016-01-31 10:25:48 +00:00
{
if (ip < zc->base + zc->nextToUpdate) return 0; /* skipped area */
ZSTD_updateTree(zc, ip, iLimit, maxNbAttempts, mls);
2016-03-04 18:31:57 +00:00
return ZSTD_insertBtAndGetAllMatches(zc, ip, iLimit, maxNbAttempts, mls, 0, matches);
2016-01-31 10:25:48 +00:00
}
2016-03-04 18:31:57 +00:00
static U32 ZSTD_BtGetAllMatches_selectMLS (
2016-01-31 10:25:48 +00:00
ZSTD_CCtx* zc, /* Index table will be updated */
const BYTE* ip, const BYTE* const iHighLimit,
2016-02-19 10:05:25 +00:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
2016-01-31 10:25:48 +00:00
{
switch(matchLengthSearch)
{
case 3 : return ZSTD_BtGetAllMatches(zc, ip, iHighLimit, maxNbAttempts, 3, matches);
2016-03-07 15:14:58 +00:00
default :
2016-03-04 18:31:57 +00: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 10:25:48 +00:00
}
}
/** Tree updater, providing best match */
2016-03-04 18:31:57 +00:00
static U32 ZSTD_BtGetAllMatches_extDict (
ZSTD_CCtx* zc,
const BYTE* const ip, const BYTE* const iLimit,
2016-02-19 10:05:25 +00: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-03-04 18:31:57 +00:00
return ZSTD_insertBtAndGetAllMatches(zc, ip, iLimit, maxNbAttempts, mls, 1, matches);
}
2016-03-04 18:31:57 +00:00
static U32 ZSTD_BtGetAllMatches_selectMLS_extDict (
ZSTD_CCtx* zc, /* Index table will be updated */
const BYTE* ip, const BYTE* const iHighLimit,
2016-02-19 10:05:25 +00:00
const U32 maxNbAttempts, const U32 matchLengthSearch, ZSTD_match_t* matches)
{
switch(matchLengthSearch)
{
case 3 : return ZSTD_BtGetAllMatches_extDict(zc, ip, iHighLimit, maxNbAttempts, 3, matches);
2016-03-07 15:14:58 +00:00
default :
2016-03-04 18:31:57 +00: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);
}
}
2016-02-10 13:26:30 +00:00
/*-*******************************
2016-01-31 10:57:01 +00:00
* Optimal parser
*********************************/
FORCE_INLINE
2016-03-04 18:31:57 +00:00
void ZSTD_compressBlock_opt_generic(ZSTD_CCtx* ctx,
2016-02-12 21:04:49 +00:00
const void* src, size_t srcSize,
2016-02-22 09:06:17 +00:00
const U32 depth)
2016-01-31 10:57:01 +00:00
{
seqStore_t* seqStorePtr = &(ctx->seqStore);
const BYTE* const istart = (const BYTE*)src;
const BYTE* ip = istart;
const BYTE* anchor = istart;
2016-02-19 17:24:14 +00:00
const BYTE* litstart;
2016-01-31 10:57:01 +00:00
const BYTE* const iend = istart + srcSize;
const BYTE* const ilimit = iend - 8;
2016-02-29 21:07:40 +00:00
const BYTE* const base = ctx->base;
const BYTE* const prefixStart = base + ctx->dictLimit;
2016-02-09 19:47:39 +00:00
U32 rep_2=REPCODE_STARTVALUE, rep_1=REPCODE_STARTVALUE;
const U32 maxSearches = 1U << ctx->params.searchLog;
const U32 sufficient_len = ctx->params.targetLength;
const U32 mls = ctx->params.searchLength;
const U32 minMatch = (ctx->params.searchLength == 3) ? 3 : 4;
2016-01-31 10:57:01 +00:00
2016-03-02 14:56:24 +00:00
ZSTD_optimal_t* opt = seqStorePtr->priceTable;
ZSTD_match_t* matches = seqStorePtr->matchTable;
2016-02-10 12:37:52 +00:00
const BYTE* inr;
2016-02-22 09:11:39 +00:00
U32 cur, match_num, last_pos, litlen, price;
2016-01-31 11:17:16 +00:00
2016-01-31 10:57:01 +00:00
/* init */
2016-02-19 10:05:25 +00:00
ctx->nextToUpdate3 = ctx->nextToUpdate;
2016-01-31 10:57:01 +00:00
ZSTD_resetSeqStore(seqStorePtr);
2016-02-19 10:05:25 +00:00
ZSTD_rescaleFreqs(seqStorePtr);
2016-02-29 21:07:40 +00:00
if ((ip-prefixStart) < REPCODE_STARTVALUE) ip = prefixStart + REPCODE_STARTVALUE;
2016-01-31 10:57:01 +00:00
2016-02-29 21:07:40 +00:00
ZSTD_LOG_BLOCK("%d: COMPBLOCK_OPT_GENERIC srcSz=%d maxSrch=%d mls=%d sufLen=%d\n", (int)(ip-base), (int)srcSize, maxSearches, mls, sufficient_len);
2016-02-24 13:40:30 +00:00
2016-01-31 10:57:01 +00:00
/* Match Loop */
while (ip < ilimit) {
2016-02-10 12:37:52 +00:00
U32 u;
2016-02-09 19:47:39 +00:00
U32 mlen=0;
U32 best_mlen=0;
U32 best_off=0;
2016-02-03 18:08:48 +00:00
memset(opt, 0, sizeof(ZSTD_optimal_t));
2016-01-31 11:17:16 +00:00
last_pos = 0;
inr = ip;
2016-02-19 17:24:14 +00:00
litstart = ((U32)(ip - anchor) > 128) ? ip - 128 : anchor;
opt[0].litlen = (U32)(ip - litstart);
2016-02-04 20:45:26 +00:00
2016-01-31 10:57:01 +00:00
/* check repCode */
if (MEM_readMINMATCH(ip+1, minMatch) == MEM_readMINMATCH(ip+1 - rep_1, minMatch)) {
2016-01-31 10:57:01 +00:00
/* repcode : we take it */
mlen = (U32)ZSTD_count(ip+1+minMatch, ip+1+minMatch-rep_1, iend) + minMatch;
2016-02-10 12:37:52 +00:00
2016-02-03 18:08:48 +00: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 12:30:34 +00:00
ip+=1; best_mlen = mlen; best_off = 0; cur = 0; last_pos = 1;
2016-01-31 11:17:16 +00:00
goto _storeSequence;
}
2016-02-04 15:02:05 +00:00
litlen = opt[0].litlen + 1;
do {
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, 0, mlen - minMatch);
2016-02-03 12:30:34 +00: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 11:17:16 +00:00
mlen--;
} while (mlen >= minMatch);
2016-01-31 10:57:01 +00:00
}
2016-02-03 12:30:34 +00:00
match_num = ZSTD_BtGetAllMatches_selectMLS(ctx, ip, iend, maxSearches, mls, matches); /* first search (depth 0) */
2016-01-31 13:28:26 +00:00
2016-02-11 03:17:50 +00: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 13:28:26 +00:00
2016-02-11 03:17:50 +00:00
opt[0].rep = rep_1;
opt[0].rep2 = rep_2;
opt[0].mlen = 1;
2016-02-04 14:25:37 +00:00
2016-02-11 03:17:50 +00:00
if (match_num && matches[match_num-1].len > sufficient_len) {
2016-01-31 13:28:26 +00:00
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
cur = 0;
last_pos = 1;
goto _storeSequence;
2016-02-11 03:17:50 +00:00
}
2016-01-31 13:28:26 +00:00
best_mlen = (last_pos) ? last_pos : minMatch;
2016-02-19 10:05:25 +00:00
2016-01-31 13:28:26 +00:00
// set prices using matches at position = 0
2016-02-10 12:37:52 +00: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 13:50:22 +00: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 15:02:05 +00:00
litlen = opt[0].litlen;
while (mlen <= best_mlen) {
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, matches[u].off, mlen - minMatch);
2016-01-31 13:28:26 +00:00
if (mlen > last_pos || price < opt[mlen].price)
2016-02-10 12:37:52 +00:00
SET_PRICE(mlen, mlen, matches[u].off, litlen, price);
2016-01-31 13:28:26 +00:00
mlen++;
} }
2016-01-31 10:57:01 +00:00
if (last_pos < minMatch) { ip++; continue; }
2016-02-02 14:41:16 +00:00
/* check further positions */
2016-02-22 09:06:17 +00:00
for (cur = 1; cur <= last_pos; cur++) {
2016-02-10 12:37:52 +00:00
size_t cur_rep;
2016-01-31 14:17:50 +00:00
inr = ip + cur;
2016-02-10 12:37:52 +00:00
if (opt[cur-1].mlen == 1) {
2016-01-31 14:17:50 +00:00
litlen = opt[cur-1].litlen + 1;
2016-02-10 12:37:52 +00:00
if (cur > litlen) {
2016-02-03 18:53:29 +00:00
price = opt[cur - litlen].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-litlen);
2016-02-10 12:37:52 +00:00
} else
2016-02-19 17:24:14 +00:00
price = ZSTD_getLiteralPrice(seqStorePtr, litlen, litstart);
2016-02-10 12:37:52 +00:00
} else {
2016-01-31 14:17:50 +00:00
litlen = 1;
2016-02-10 12:37:52 +00:00
price = opt[cur - 1].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1);
2016-01-31 14:17:50 +00:00
}
2016-02-10 12:37:52 +00:00
2016-01-31 14:17:50 +00:00
if (cur > last_pos || price <= opt[cur].price) // || ((price == opt[cur].price) && (opt[cur-1].mlen == 1) && (cur != litlen)))
2016-02-04 19:53:54 +00:00
SET_PRICE(cur, 1, 0, litlen, price);
2016-01-31 10:57:01 +00:00
2016-02-04 13:55:23 +00:00
if (cur == last_pos) break;
2016-02-11 03:17:50 +00:00
if (inr > ilimit) /* last match must start at a minimum distance of 8 from oend */
continue;
2016-02-04 14:28:55 +00:00
mlen = opt[cur].mlen;
2016-02-10 12:37:52 +00:00
if (opt[cur].off) {
2016-02-04 14:25:37 +00:00
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur].off;
ZSTD_LOG_ENCODE("%d: COPYREP_OFF cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
} else {
if (cur!=mlen && opt[cur].litlen == 0) {
2016-02-04 14:25:37 +00:00
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].rep2;
ZSTD_LOG_ENCODE("%d: COPYREP_SWI cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 12:37:52 +00:00
} else {
2016-02-04 14:25:37 +00:00
opt[cur].rep2 = opt[cur-mlen].rep2;
opt[cur].rep = opt[cur-mlen].rep;
ZSTD_LOG_ENCODE("%d: COPYREP_NOR cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
} }
2016-02-03 12:30:34 +00:00
ZSTD_LOG_PARSER("%d: CURRENT_NoExt 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 14:28:55 +00:00
2016-02-06 14:18:26 +00:00
best_mlen = 0;
if (opt[cur].mlen != 1) {
2016-02-06 14:18:26 +00:00
cur_rep = opt[cur].rep2;
ZSTD_LOG_PARSER("%d: tryNoExt REP2 rep2=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-10 12:37:52 +00:00
} else {
2016-02-06 14:18:26 +00:00
cur_rep = opt[cur].rep;
ZSTD_LOG_PARSER("%d: tryNoExt REP1 rep=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-04 16:48:33 +00:00
}
if (MEM_readMINMATCH(inr, minMatch) == MEM_readMINMATCH(inr - cur_rep, minMatch)) { // check rep
mlen = (U32)ZSTD_count(inr+minMatch, inr+minMatch - cur_rep, iend) + minMatch;
2016-02-11 03:17:50 +00:00
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 14:17:50 +00:00
2016-02-11 03:17:50 +00: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 14:17:50 +00:00
}
2016-02-10 12:37:52 +00:00
if (opt[cur].mlen == 1) {
2016-01-31 14:17:50 +00:00
litlen = opt[cur].litlen;
2016-02-10 12:37:52 +00:00
if (cur > litlen) {
2016-03-04 18:31:57 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, 0, mlen - minMatch);
2016-02-10 12:37:52 +00:00
} else
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, 0, mlen - minMatch);
2016-02-10 12:37:52 +00:00
} else {
2016-01-31 14:17:50 +00:00
litlen = 0;
2016-03-04 18:31:57 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen - minMatch);
2016-01-31 14:17:50 +00:00
}
best_mlen = mlen;
2016-02-22 09:06:17 +00: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 14:17:50 +00:00
2016-02-10 12:37:52 +00:00
do {
if (cur + mlen > last_pos || price <= opt[cur + mlen].price)
2016-01-31 14:17:50 +00:00
SET_PRICE(cur + mlen, mlen, 0, litlen, price);
mlen--;
} while (mlen >= minMatch);
2016-01-31 14:17:50 +00:00
}
2016-02-03 12:30:34 +00:00
match_num = ZSTD_BtGetAllMatches_selectMLS(ctx, inr, iend, maxSearches, mls, matches);
2016-02-03 18:08:48 +00:00
ZSTD_LOG_PARSER("%d: ZSTD_GetAllMatches match_num=%d\n", (int)(inr-base), match_num);
2016-01-31 14:17:50 +00:00
2016-02-10 12:37:52 +00:00
if (match_num > 0 && matches[match_num-1].len > sufficient_len) {
2016-01-31 14:17:50 +00:00
best_mlen = matches[match_num-1].len;
best_off = matches[match_num-1].off;
last_pos = cur + 1;
2016-01-31 14:32:01 +00:00
goto _storeSequence;
2016-01-31 14:17:50 +00:00
}
best_mlen = (best_mlen > minMatch) ? best_mlen : minMatch;
2016-02-19 10:05:25 +00:00
2016-02-10 12:37:52 +00: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;
2016-02-22 09:11:39 +00:00
best_mlen = (cur + matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM - cur;
2016-02-02 14:41:16 +00:00
2016-02-29 21:07:40 +00:00
// ZSTD_LOG_PARSER("%d: Found1 cur=%d mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(inr-base), cur, matches[u].len, matches[u].off, best_mlen, last_pos);
2016-01-31 14:17:50 +00:00
2016-02-10 12:37:52 +00:00
while (mlen <= best_mlen) {
2016-02-22 09:11:39 +00:00
if (opt[cur].mlen == 1) {
litlen = opt[cur].litlen;
if (cur > litlen)
2016-03-04 18:31:57 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, ip+cur-litlen, matches[u].off, mlen - minMatch);
2016-01-31 14:17:50 +00:00
else
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, matches[u].off, mlen - minMatch);
2016-02-10 12:37:52 +00:00
} else {
2016-01-31 14:17:50 +00:00
litlen = 0;
2016-03-04 18:31:57 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, matches[u].off, mlen - minMatch);
2016-01-31 14:17:50 +00:00
}
2016-02-29 21:07:40 +00:00
// ZSTD_LOG_PARSER("%d: Found2 mlen=%d best_mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, best_mlen, matches[u].off, price, litlen);
2016-02-03 15:54:23 +00:00
2016-02-22 09:11:39 +00:00
if (cur + mlen > last_pos || (price < opt[cur + mlen].price))
SET_PRICE(cur + mlen, mlen, matches[u].off, litlen, price);
2016-01-31 14:17:50 +00:00
mlen++;
2016-02-22 09:06:17 +00:00
} } } // for (cur = 1; cur <= last_pos; cur++)
2016-01-31 10:57:01 +00:00
2016-01-31 13:28:26 +00:00
best_mlen = opt[last_pos].mlen;
best_off = opt[last_pos].off;
2016-01-31 14:17:50 +00:00
cur = last_pos - best_mlen;
2016-01-31 11:36:41 +00:00
2016-01-31 10:57:01 +00:00
/* store sequence */
_storeSequence: /* cur, last_pos, best_mlen, best_off have to be set */
2016-02-10 12:37:52 +00: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 11:36:41 +00:00
opt[0].mlen = 1;
2016-02-09 19:47:39 +00:00
U32 offset;
2016-02-04 19:53:54 +00:00
2016-02-10 12:37:52 +00:00
while (1) {
2016-01-31 11:36:41 +00:00
mlen = opt[cur].mlen;
offset = opt[cur].off;
2016-02-10 12:37:52 +00:00
opt[cur].mlen = best_mlen;
2016-01-31 11:36:41 +00:00
opt[cur].off = best_off;
best_mlen = mlen;
2016-02-10 12:37:52 +00:00
best_off = offset;
if (mlen > cur) break;
2016-01-31 11:36:41 +00:00
cur -= mlen;
}
2016-02-10 12:37:52 +00: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 11:36:41 +00:00
}
2016-02-11 03:17:50 +00:00
for (cur=0; cur < last_pos; ) {
2016-02-04 19:53:54 +00: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 11:36:41 +00:00
mlen = opt[cur].mlen;
if (mlen == 1) { ip++; cur++; continue; }
offset = opt[cur].off;
cur += mlen;
2016-02-09 19:47:39 +00:00
U32 litLength = (U32)(ip - anchor);
2016-02-19 10:05:25 +00: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 16:48:33 +00:00
2016-02-11 03:17:50 +00:00
if (offset) {
2016-02-03 11:53:07 +00:00
rep_2 = rep_1;
rep_1 = offset;
2016-02-11 03:17:50 +00:00
} else {
if (litLength == 0) {
2016-02-03 11:53:07 +00:00
best_off = rep_2;
rep_2 = rep_1;
rep_1 = best_off;
2016-02-11 03:17:50 +00:00
} }
2016-02-03 11:53:07 +00:00
2016-02-19 10:05:25 +00: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 12:37:52 +00:00
2016-02-05 08:07:22 +00:00
#if ZSTD_OPT_DEBUG >= 5
2016-02-19 10:05:25 +00:00
U32 ml2;
2016-01-31 13:28:26 +00:00
if (offset)
2016-02-19 10:05:25 +00:00
ml2 = (U32)ZSTD_count(ip, ip-offset, iend);
2016-02-02 23:04:14 +00:00
else
2016-02-19 10:05:25 +00:00
ml2 = (U32)ZSTD_count(ip, ip-rep_1, iend);
if ((offset >= 8) && (ml2 < mlen || ml2 < minMatch)) {
printf("%d: ERROR_NoExt iend=%d mlen=%d offset=%d ml2=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset, (int)ml2); exit(0); }
2016-02-06 14:18:26 +00:00
if (ip < anchor) {
2016-02-29 21:07:40 +00:00
printf("%d: ERROR_NoExt ip < anchor iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
2016-02-06 14:18:26 +00:00
if (ip + mlen > iend) {
2016-02-29 21:07:40 +00:00
printf("%d: ERROR_NoExt 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 13:28:26 +00:00
#endif
ZSTD_updatePrice(seqStorePtr, litLength, anchor, offset, mlen-minMatch);
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, mlen-minMatch);
2016-01-31 14:17:50 +00:00
anchor = ip = ip + mlen;
} /* for (cur=0; cur < last_pos; ) */
2016-01-31 10:57:01 +00:00
2016-02-11 03:17:50 +00:00
/* check immediate repcode */
2016-03-01 13:14:35 +00:00
while ((anchor >= prefixStart + rep_2) && (anchor <= ilimit)
&& (MEM_readMINMATCH(anchor, minMatch) == MEM_readMINMATCH(anchor - rep_2, minMatch)) ) {
2016-01-31 10:57:01 +00:00
/* store sequence */
best_mlen = (U32)ZSTD_count(anchor+minMatch, anchor+minMatch-rep_2, iend);
2016-01-31 11:36:41 +00:00
best_off = rep_2;
2016-01-31 11:17:16 +00:00
rep_2 = rep_1;
2016-01-31 11:36:41 +00:00
rep_1 = best_off;
2016-02-03 18:08:48 +00: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 11:17:16 +00:00
ZSTD_storeSeq(seqStorePtr, 0, anchor, 0, best_mlen);
anchor += best_mlen+minMatch;
continue; /* faster when present ... (?) */
}
if (anchor > ip) ip = anchor;
}
2016-01-31 10:57:01 +00:00
2016-02-11 03:17:50 +00:00
{ /* Last Literals */
size_t lastLLSize = iend - anchor;
ZSTD_LOG_ENCODE("%d: lastLLSize literals=%u\n", (int)(ip-base), (U32)lastLLSize);
2016-01-31 10:57:01 +00:00
memcpy(seqStorePtr->lit, anchor, lastLLSize);
seqStorePtr->lit += lastLLSize;
}
}
FORCE_INLINE
2016-03-04 18:31:57 +00:00
void ZSTD_compressBlock_opt_extDict_generic(ZSTD_CCtx* ctx,
const void* src, size_t srcSize,
2016-02-22 09:06:17 +00:00
const U32 depth)
{
seqStore_t* seqStorePtr = &(ctx->seqStore);
const BYTE* const istart = (const BYTE*)src;
const BYTE* ip = istart;
const BYTE* anchor = istart;
2016-02-19 17:24:14 +00:00
const BYTE* litstart;
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;
2016-03-01 13:14:35 +00:00
const U32 lowLimit = ctx->lowLimit;
U32 rep_2=REPCODE_STARTVALUE, rep_1=REPCODE_STARTVALUE;
const U32 maxSearches = 1U << ctx->params.searchLog;
const U32 sufficient_len = ctx->params.targetLength;
const U32 mls = ctx->params.searchLength;
const U32 minMatch = (ctx->params.searchLength == 3) ? 3 : 4;
2016-03-02 14:56:24 +00:00
ZSTD_optimal_t* opt = seqStorePtr->priceTable;
ZSTD_match_t* matches = seqStorePtr->matchTable;
2016-02-10 12:37:52 +00:00
const BYTE* inr;
2016-02-22 09:11:39 +00:00
U32 cur, match_num, last_pos, litlen, price;
/* init */
2016-02-19 10:05:25 +00:00
ctx->nextToUpdate3 = ctx->nextToUpdate;
ZSTD_resetSeqStore(seqStorePtr);
2016-02-19 10:05:25 +00:00
ZSTD_rescaleFreqs(seqStorePtr);
if ((ip - prefixStart) < REPCODE_STARTVALUE) ip += REPCODE_STARTVALUE;
2016-02-29 21:07:40 +00:00
ZSTD_LOG_BLOCK("%d: COMPBLOCK_OPT_EXTDICT srcSz=%d maxSrch=%d mls=%d sufLen=%d\n", (int)(ip-base), (int)srcSize, maxSearches, mls, sufficient_len);
/* Match Loop */
while (ip < ilimit) {
2016-02-11 03:17:50 +00: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;
2016-02-19 17:24:14 +00:00
litstart = ((U32)(ip - anchor) > 128) ? ip - 128 : anchor;
opt[0].litlen = (U32)(ip - litstart);
/* 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 13:26:30 +00:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
&& (MEM_readMINMATCH(ip+1, minMatch) == MEM_readMINMATCH(repMatch, minMatch)) ) {
/* 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 12:37:52 +00: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 12:37:52 +00:00
do {
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, 0, mlen - minMatch);
if (mlen + 1 > last_pos || price < opt[mlen + 1].price)
SET_PRICE(mlen + 1, mlen, 0, litlen, price);
mlen--;
} while (mlen >= minMatch);
2016-02-10 12:37:52 +00:00
} }
best_mlen = (last_pos) ? last_pos : minMatch;
2016-02-10 12:37:52 +00:00
match_num = ZSTD_BtGetAllMatches_selectMLS_extDict(ctx, 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 12:37:52 +00: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 13:26:30 +00: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-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, matches[u].off, mlen - minMatch);
if (mlen > last_pos || price < opt[mlen].price)
2016-02-10 12:37:52 +00:00
SET_PRICE(mlen, mlen, matches[u].off, litlen, price);
mlen++;
2016-02-10 13:26:30 +00:00
} }
if (last_pos < minMatch) {
2016-02-10 12:37:52 +00:00
// ip += ((ip-anchor) >> g_searchStrength) + 1; /* jump faster over incompressible sections */
ip++; continue;
}
2016-02-11 03:17:50 +00:00
/* check further positions */
2016-02-22 09:06:17 +00:00
for (cur = 1; cur <= last_pos; cur++) {
2016-02-10 12:37:52 +00:00
size_t cur_rep;
inr = ip + cur;
2016-02-10 12:37:52 +00:00
if (opt[cur-1].mlen == 1) {
litlen = opt[cur-1].litlen + 1;
2016-02-10 12:37:52 +00:00
if (cur > litlen) {
price = opt[cur - litlen].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-litlen);
2016-02-10 12:37:52 +00:00
} else
2016-02-19 17:24:14 +00:00
price = ZSTD_getLiteralPrice(seqStorePtr, litlen, litstart);
2016-02-10 12:37:52 +00:00
} else {
litlen = 1;
2016-02-10 12:37:52 +00:00
price = opt[cur - 1].price + ZSTD_getLiteralPrice(seqStorePtr, litlen, inr-1);
}
2016-02-10 12:37:52 +00:00
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 12:37:52 +00:00
if (opt[cur].off) {
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur].off;
ZSTD_LOG_ENCODE("%d: COPYREP_OFF cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
} else {
if (cur!=mlen && opt[cur].litlen == 0) {
opt[cur].rep2 = opt[cur-mlen].rep;
opt[cur].rep = opt[cur-mlen].rep2;
ZSTD_LOG_ENCODE("%d: COPYREP_SWI cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
2016-02-10 12:37:52 +00:00
} else {
opt[cur].rep2 = opt[cur-mlen].rep2;
opt[cur].rep = opt[cur-mlen].rep;
ZSTD_LOG_ENCODE("%d: COPYREP_NOR cur=%d mlen=%d rep=%d rep2=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep, opt[cur].rep2);
} }
ZSTD_LOG_PARSER("%d: CURRENT_Ext 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].mlen != 1) {
cur_rep = opt[cur].rep2;
ZSTD_LOG_PARSER("%d: tryExt REP2 rep2=%u mlen=%u\n", (int)(inr-base), (U32)cur_rep, mlen);
2016-02-10 13:26:30 +00:00
} else {
cur_rep = opt[cur].rep;
ZSTD_LOG_PARSER("%d: tryExt 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-11 03:17:50 +00:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
&& (MEM_readMINMATCH(inr, minMatch) == MEM_readMINMATCH(repMatch, minMatch)) ) {
/* 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 13:26:30 +00: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-11 03:17:50 +00:00
}
2016-02-10 13:26:30 +00:00
if (opt[cur].mlen == 1) {
litlen = opt[cur].litlen;
2016-02-10 13:26:30 +00:00
if (cur > litlen) {
2016-03-04 18:31:57 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, 0, mlen - minMatch);
2016-02-10 13:26:30 +00:00
} else
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, 0, mlen - minMatch);
2016-02-10 13:26:30 +00:00
} else {
litlen = 0;
2016-03-04 18:31:57 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, 0, mlen - minMatch);
}
best_mlen = mlen;
2016-02-19 10:05:25 +00: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 13:26:30 +00: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--;
} while (mlen >= minMatch);
}
best_mlen = (best_mlen > minMatch) ? best_mlen : minMatch;
match_num = ZSTD_BtGetAllMatches_selectMLS_extDict(ctx, inr, iend, maxSearches, mls, matches);
ZSTD_LOG_PARSER("%d: ZSTD_GetAllMatches match_num=%d\n", (int)(inr-base), match_num);
2016-02-10 12:37:52 +00:00
if (match_num > 0 && matches[match_num-1].len > sufficient_len) {
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 12:37:52 +00:00
for (u = 0; u < match_num; u++) {
mlen = (u>0) ? matches[u-1].len+1 : best_mlen;
2016-02-22 09:11:39 +00:00
best_mlen = (cur + matches[u].len < ZSTD_OPT_NUM) ? matches[u].len : ZSTD_OPT_NUM - cur;
2016-02-29 21:07:40 +00:00
// ZSTD_LOG_PARSER("%d: Found1 cur=%d mlen=%d off=%d best_mlen=%d last_pos=%d\n", (int)(inr-base), cur, matches[u].len, matches[u].off, best_mlen, last_pos);
2016-02-10 12:37:52 +00:00
while (mlen <= best_mlen) {
2016-02-22 09:11:39 +00:00
if (opt[cur].mlen == 1) {
litlen = opt[cur].litlen;
if (cur > litlen)
2016-03-04 18:31:57 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, ip+cur-litlen, matches[u].off, mlen - minMatch);
else
2016-03-04 18:31:57 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, matches[u].off, mlen - minMatch);
2016-02-10 12:37:52 +00:00
} else {
litlen = 0;
2016-03-04 18:31:57 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, matches[u].off, mlen - minMatch);
}
2016-02-29 21:07:40 +00:00
// ZSTD_LOG_PARSER("%d: Found2 mlen=%d best_mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, best_mlen, matches[u].off, price, litlen);
2016-02-22 09:11:39 +00:00
if (cur + mlen > last_pos || (price < opt[cur + mlen].price))
SET_PRICE(cur + mlen, mlen, matches[u].off, litlen, price);
mlen++;
2016-02-22 09:06:17 +00:00
} } } // for (cur = 1; cur <= last_pos; cur++)
best_mlen = opt[last_pos].mlen;
best_off = opt[last_pos].off;
cur = last_pos - best_mlen;
/* store sequence */
_storeSequence: // cur, last_pos, best_mlen, best_off have to be set
2016-02-10 12:37:52 +00:00
for (u = 1; u <= last_pos; u++)
2016-02-10 13:50:22 +00: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 12:37:52 +00: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 12:37:52 +00:00
while (1) {
mlen = opt[cur].mlen;
offset = opt[cur].off;
2016-02-10 12:37:52 +00:00
opt[cur].mlen = best_mlen;
opt[cur].off = best_off;
best_mlen = mlen;
2016-02-10 12:37:52 +00:00
best_off = offset;
2016-02-11 03:17:50 +00:00
if (mlen > cur) break;
cur -= mlen;
}
2016-02-10 12:37:52 +00:00
for (u = 0; u <= last_pos; ) {
2016-02-10 12:37:52 +00: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 12:37:52 +00: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 12:37:52 +00: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 12:37:52 +00:00
if (offset) {
rep_2 = rep_1;
rep_1 = offset;
2016-02-10 12:37:52 +00:00
} else {
if (litLength == 0) {
best_off = rep_2;
rep_2 = rep_1;
rep_1 = best_off;
2016-02-10 12:37:52 +00: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 12:37:52 +00:00
#if ZSTD_OPT_DEBUG >= 5
2016-02-19 10:05:25 +00:00
U32 ml2;
2016-03-01 13:14:35 +00:00
if (offset) {
if (offset > (size_t)(ip - prefixStart)) {
2016-02-29 21:07:40 +00:00
const BYTE* match = dictEnd - (offset - (ip - prefixStart));
ml2 = ZSTD_count_2segments(ip, match, iend, dictEnd, prefixStart);
ZSTD_LOG_PARSER("%d: ZSTD_count_2segments=%d offset=%d dictBase=%p dictEnd=%p prefixStart=%p ip=%p match=%p\n", (int)current, (int)ml2, (int)offset, dictBase, dictEnd, prefixStart, ip, match);
}
2016-03-01 13:14:35 +00:00
else ml2 = (U32)ZSTD_count(ip, ip-offset, iend);
2016-02-29 21:07:40 +00:00
}
2016-03-01 13:14:35 +00:00
else ml2 = (U32)ZSTD_count(ip, ip-rep_1, iend);
if ((offset >= 8) && (ml2 < mlen || ml2 < minMatch)) {
2016-02-29 21:07:40 +00:00
printf("%d: ERROR_Ext 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) {
2016-02-29 21:07:40 +00:00
printf("%d: ERROR_Ext ip < anchor iend=%d mlen=%d offset=%d\n", (int)(ip - base), (int)(iend - ip), (int)mlen, (int)offset); exit(0); }
if (ip + mlen > iend) {
2016-02-29 21:07:40 +00:00
printf("%d: ERROR_Ext 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-03-02 14:56:24 +00:00
#if 0
2016-02-10 12:37:52 +00:00
/* check immediate repcode */
2016-03-02 14:56:24 +00:00
while ((anchor >= base + lowLimit + rep_2) && (anchor <= ilimit)) {
if ((anchor - rep_2) >= prefixStart) {
if (MEM_readMINMATCH(anchor, minMatch) == MEM_readMINMATCH(anchor - rep_2, minMatch))
mlen = (U32)ZSTD_count(anchor+minMatch, anchor - rep_2 + minMatch, iend) + minMatch;
2016-03-02 14:56:24 +00:00
else
break;
} else {
const BYTE* repMatch = dictBase + ((anchor-base) - rep_2);
if ((repMatch + minMatch <= dictEnd) && (MEM_readMINMATCH(anchor, minMatch) == MEM_readMINMATCH(repMatch, minMatch)))
mlen = (U32)ZSTD_count_2segments(anchor+minMatch, repMatch+minMatch, iend, dictEnd, prefixStart) + minMatch;
2016-03-02 14:56:24 +00:00
else
break;
}
2016-03-02 14:56:24 +00:00
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);
2016-03-02 14:56:24 +00:00
anchor += mlen;
}
#else
/* check immediate repcode */
/* minimal correctness condition = while ((anchor >= prefixStart + REPCODE_STARTVALUE) && (anchor <= ilimit)) { */
2016-03-01 13:14:35 +00:00
while ((anchor >= base + lowLimit + rep_2) && (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 12:37:52 +00:00
if ( ((U32)((dictLimit-1) - repIndex) >= 3) /* intentional overflow */
&& (MEM_readMINMATCH(anchor, minMatch) == MEM_readMINMATCH(repMatch, minMatch)) ) {
2016-02-10 12:37:52 +00: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;
}
2016-03-02 14:56:24 +00:00
#endif
if (anchor > ip) ip = anchor;
}
2016-02-11 03:17:50 +00: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;
}
}