zstd/lib/zstd_opt.h

1034 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 = MaxML+1;
ssPtr->offCodeSum = (MaxOff+1);
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 */
{ static const BYTE LL_Code[64] = { 0, 1, 2, 3, 4, 5, 6, 7,
8, 9, 10, 11, 12, 13, 14, 15,
16, 16, 17, 17, 18, 18, 19, 19,
20, 20, 20, 20, 21, 21, 21, 21,
22, 22, 22, 22, 22, 22, 22, 22,
23, 23, 23, 23, 23, 23, 23, 23,
24, 24, 24, 24, 24, 24, 24, 24,
24, 24, 24, 24, 24, 24, 24, 24 };
const BYTE LL_deltaCode = 19;
const BYTE llCode = (litLength>63) ? (BYTE)ZSTD_highbit(litLength) + LL_deltaCode : LL_Code[litLength];
price += LL_bits[llCode] + seqStorePtr->log2litLengthSum - ZSTD_highbit(seqStorePtr->litLengthFreq[llCode]+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 */
BYTE offCode = (BYTE)ZSTD_highbit(offset+1);
U32 price = offCode + seqStorePtr->log2offCodeSum - ZSTD_highbit(seqStorePtr->offCodeFreq[offCode]+1);
2016-02-03 20:26:18 +00:00
/* match Length */
{ static const BYTE ML_Code[128] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
32, 32, 33, 33, 34, 34, 35, 35, 36, 36, 36, 36, 37, 37, 37, 37,
38, 38, 38, 38, 38, 38, 38, 38, 39, 39, 39, 39, 39, 39, 39, 39,
40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40,
41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41,
42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42,
42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42 };
const BYTE ML_deltaCode = 36;
const BYTE mlCode = (matchLength>127) ? (BYTE)ZSTD_highbit(matchLength) + ML_deltaCode : ML_Code[matchLength];
price += ML_bits[mlCode] + seqStorePtr->log2matchLengthSum - ZSTD_highbit(seqStorePtr->matchLengthFreq[mlCode]+1);
2016-02-23 15:25:04 +00:00
}
return price + ZSTD_getLiteralPrice(seqStorePtr, litLength, literals) + seqStorePtr->factor;
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 */
{ static const BYTE LL_Code[64] = { 0, 1, 2, 3, 4, 5, 6, 7,
8, 9, 10, 11, 12, 13, 14, 15,
16, 16, 17, 17, 18, 18, 19, 19,
20, 20, 20, 20, 21, 21, 21, 21,
22, 22, 22, 22, 22, 22, 22, 22,
23, 23, 23, 23, 23, 23, 23, 23,
24, 24, 24, 24, 24, 24, 24, 24,
24, 24, 24, 24, 24, 24, 24, 24 };
const BYTE LL_deltaCode = 19;
const BYTE llCode = (litLength>63) ? (BYTE)ZSTD_highbit(litLength) + LL_deltaCode : LL_Code[litLength];
seqStorePtr->litLengthFreq[llCode]++;
seqStorePtr->litLengthSum++;
}
2016-03-04 18:31:57 +00:00
/* match offset */
seqStorePtr->offCodeSum++;
2016-04-04 13:43:45 +00:00
BYTE offCode = (BYTE)ZSTD_highbit(offset+1);
2016-03-04 18:31:57 +00:00
seqStorePtr->offCodeFreq[offCode]++;
/* match Length */
{ static const BYTE ML_Code[128] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15,
16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31,
32, 32, 33, 33, 34, 34, 35, 35, 36, 36, 36, 36, 37, 37, 37, 37,
38, 38, 38, 38, 38, 38, 38, 38, 39, 39, 39, 39, 39, 39, 39, 39,
40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40, 40,
41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41, 41,
42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42,
42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42, 42 };
const BYTE ML_deltaCode = 36;
const BYTE mlCode = (matchLength>127) ? (BYTE)ZSTD_highbit(matchLength) + ML_deltaCode : ML_Code[matchLength];
seqStorePtr->matchLengthFreq[mlCode]++;
seqStorePtr->matchLengthSum++;
}
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); \
}
/* Update hashTable3 up to ip (excluded)
Assumption : always within prefix (ie. not within extDict) */
FORCE_INLINE
U32 ZSTD_insertAndFindFirstIndexHash3 (ZSTD_CCtx* zc, const BYTE* ip)
{
U32* const hashTable3 = zc->hashTable3;
U32 const hashLog3 = zc->hashLog3;
const BYTE* const base = zc->base;
U32 idx = zc->nextToUpdate3;
const U32 target = zc->nextToUpdate3 = (U32)(ip - base);
const size_t hash3 = ZSTD_hash3Ptr(ip, hashLog3);
2016-04-08 13:25:32 +00:00
while(idx < target) {
hashTable3[ZSTD_hash3Ptr(base+idx, hashLog3)] = idx;
idx++;
}
return hashTable3[hash3];
}
2016-02-10 13:26:30 +00:00
/*-*************************************
* Binary Tree search
***************************************/
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);
const U32 hashLog = zc->params.cParams.hashLog;
2016-01-31 10:25:48 +00:00
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-04-04 11:49:18 +00:00
U32* const bt = zc->chainTable;
const U32 btLog = zc->params.cParams.chainLog - 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 */
2016-04-07 09:35:17 +00:00
U32 const 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;
2016-04-07 09:35:17 +00:00
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;
2016-03-18 15:10:35 +00:00
matches[mnum].off = ZSTD_REP_MOVE + 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;
2016-03-18 15:10:35 +00:00
matches[mnum].off = ZSTD_REP_MOVE + current - matchIndex;
2016-02-10 12:37:52 +00:00
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,
const void* src, size_t srcSize)
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;
const U32 maxSearches = 1U << ctx->params.cParams.searchLog;
const U32 sufficient_len = ctx->params.cParams.targetLength;
const U32 mls = ctx->params.cParams.searchLength;
const U32 minMatch = (ctx->params.cParams.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 */
U32 rep[ZSTD_REP_INIT];
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_INIT; i++) rep[i]=REPCODE_STARTVALUE; }
2016-03-16 10:56:59 +00:00
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 */
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_NUM; i++)
2016-03-21 18:48:54 +00:00
if (MEM_readMINMATCH(ip, minMatch) == MEM_readMINMATCH(ip - rep[i], minMatch)) {
2016-01-31 10:57:01 +00:00
/* repcode : we take it */
2016-03-21 18:48:54 +00:00
mlen = (U32)ZSTD_count(ip+minMatch, ip+minMatch-rep[i], iend) + minMatch;
2016-02-10 12:37:52 +00:00
ZSTD_LOG_PARSER("%d: start try REP rep[%d]=%d mlen=%d\n", (int)(ip-base), i, (int)rep[i], (int)mlen);
if (mlen > sufficient_len || mlen >= ZSTD_OPT_NUM) {
2016-03-21 18:48:54 +00:00
best_mlen = mlen; best_off = i; cur = 0; last_pos = 1;
2016-01-31 11:17:16 +00:00
goto _storeSequence;
}
best_off = (i<=1 && ip == anchor) ? 1-i : i;
2016-03-21 18:48:54 +00:00
litlen = opt[0].litlen;
do {
2016-04-06 07:46:01 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, best_off, mlen - MINMATCH);
2016-03-21 18:48:54 +00:00
if (mlen > last_pos || price < opt[mlen].price)
SET_PRICE(mlen, mlen, i, litlen, price); /* note : macro modifies last_pos */
2016-01-31 11:17:16 +00:00
mlen--;
} while (mlen >= minMatch);
2016-04-08 13:25:32 +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-04-08 13:25:32 +00:00
{ U32 i ; for (i=0; i<ZSTD_REP_INIT; i++) opt[0].rep[i] = rep[i]; }
2016-02-11 03:17:50 +00:00
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-04-06 07:46:01 +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-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-03-21 16:23:02 +00:00
if (opt[cur].off >= ZSTD_REP_NUM) {
2016-04-05 08:18:37 +00:00
opt[cur].rep[2] = opt[cur-mlen].rep[1];
opt[cur].rep[1] = opt[cur-mlen].rep[0];
opt[cur].rep[0] = opt[cur].off - ZSTD_REP_MOVE;
ZSTD_LOG_ENCODE("%d: COPYREP_OFF cur=%d mlen=%d rep[0]=%d rep[1]=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep[0], opt[cur].rep[1]);
} else {
2016-03-21 19:14:03 +00:00
opt[cur].rep[2] = (opt[cur].off > 1) ? opt[cur-mlen].rep[1] : opt[cur-mlen].rep[2];
opt[cur].rep[1] = (opt[cur].off > 0) ? opt[cur-mlen].rep[0] : opt[cur-mlen].rep[1];
opt[cur].rep[0] = opt[cur-mlen].rep[opt[cur].off];
ZSTD_LOG_ENCODE("%d: COPYREP_NOR cur=%d mlen=%d rep[0]=%d rep[1]=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep[0], opt[cur].rep[1]);
2016-03-18 15:10:35 +00:00
}
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[0]=%d rep[1]=%d\n", (int)(inr-base), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep[0], opt[cur].rep[1]);
2016-02-06 14:18:26 +00:00
best_mlen = 0;
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_NUM; i++)
2016-04-06 07:46:01 +00:00
if (MEM_readMINMATCH(inr, minMatch) == MEM_readMINMATCH(inr - opt[cur].rep[i], minMatch)) { /* check rep */
2016-03-18 15:10:35 +00:00
mlen = (U32)ZSTD_count(inr+minMatch, inr+minMatch - opt[cur].rep[i], iend) + minMatch;
2016-03-21 16:23:02 +00:00
ZSTD_LOG_PARSER("%d: Found REP %d/%d mlen=%d off=%d rep=%d opt[%d].off=%d\n", (int)(inr-base), i, ZSTD_REP_NUM, mlen, i, opt[cur].rep[i], cur, opt[cur].off);
2016-02-11 03:17:50 +00:00
if (mlen > sufficient_len || cur + mlen >= ZSTD_OPT_NUM) {
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);
best_mlen = mlen; best_off = i; last_pos = cur + 1;
2016-02-11 03:17:50 +00:00
goto _storeSequence;
2016-01-31 14:17:50 +00:00
}
best_off = (i<=1 && opt[cur].mlen != 1) ? 1-i : i;
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-04-06 07:46:01 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, best_off, mlen - MINMATCH);
2016-02-10 12:37:52 +00:00
} else
2016-04-06 07:46:01 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, best_off, mlen - MINMATCH);
2016-02-10 12:37:52 +00:00
} else {
2016-01-31 14:17:50 +00:00
litlen = 0;
2016-04-06 07:46:01 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, best_off, mlen - MINMATCH);
2016-01-31 14:17:50 +00:00
}
best_mlen = mlen;
2016-03-21 16:23:02 +00:00
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, best_off, 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-03-21 18:02:16 +00:00
SET_PRICE(cur + mlen, mlen, i, litlen, price);
2016-01-31 14:17:50 +00:00
mlen--;
} while (mlen >= minMatch);
2016-04-08 13:25:32 +00:00
} }
2016-02-03 12:30:34 +00:00
2016-03-21 16:23:02 +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-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-04-06 07:46:01 +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-04-06 07:46:01 +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-04-06 07:46:01 +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-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[0]=%d rep[1]=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep[0], opt[u].rep[1]);
ZSTD_LOG_PARSER("%d: cur=%d/%d best_mlen=%d best_off=%d rep[0]=%d\n", (int)(ip-base+cur), (int)cur, (int)last_pos, (int)best_mlen, (int)best_off, opt[cur].rep[0]);
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[0]=%d rep[1]=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep[0], opt[u].rep[1]);
2016-02-10 12:37:52 +00:00
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; ) {
ZSTD_LOG_PARSER("%d: price3[%d/%d]=%d off=%d mlen=%d litlen=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base+cur), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep[0], opt[cur].rep[1]);
2016-01-31 11:36:41 +00:00
mlen = opt[cur].mlen;
if (mlen == 1) { ip++; cur++; continue; }
offset = opt[cur].off;
cur += mlen;
U32 const litLength = (U32)(ip - anchor);
// ZSTD_LOG_ENCODE("%d/%d: ENCODE literals=%d mlen=%d off=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep[0], (int)rep[1]);
2016-02-04 16:48:33 +00:00
2016-03-18 15:10:35 +00:00
if (offset >= ZSTD_REP_NUM) {
2016-04-05 08:18:37 +00:00
rep[2] = rep[1];
rep[1] = rep[0];
rep[0] = offset - ZSTD_REP_MOVE;
2016-02-11 03:17:50 +00:00
} else {
2016-03-21 19:14:03 +00:00
if (offset != 0) {
best_off = rep[offset];
2016-04-05 15:11:56 +00:00
if (offset != 1) rep[2] = rep[1];
2016-03-21 19:14:03 +00:00
rep[1] = rep[0];
rep[0] = best_off;
2016-03-18 15:10:35 +00:00
}
2016-03-21 19:14:03 +00:00
if (litLength == 0 && offset<=1) offset = 1-offset;
}
2016-02-03 11:53:07 +00:00
ZSTD_LOG_ENCODE("%d/%d: ENCODE literals=%d mlen=%d off=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep[0], (int)rep[1]);
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;
if (offset >= ZSTD_REP_NUM)
ml2 = (U32)ZSTD_count(ip, ip-(offset-ZSTD_REP_MOVE), iend);
2016-02-02 23:04:14 +00:00
else
2016-03-16 10:56:59 +00:00
ml2 = (U32)ZSTD_count(ip, ip-rep[0], 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
2016-04-06 07:46:01 +00:00
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
{ /* 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)
{
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;
const U32 maxSearches = 1U << ctx->params.cParams.searchLog;
const U32 sufficient_len = ctx->params.cParams.targetLength;
const U32 mls = ctx->params.cParams.searchLength;
const U32 minMatch = (ctx->params.cParams.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 */
U32 rep[ZSTD_REP_INIT];
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_INIT; i++) rep[i]=REPCODE_STARTVALUE; }
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) {
U32 u;
U32 mlen=0;
U32 best_mlen=0;
U32 best_off=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 */
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_NUM; i++) {
const U32 repIndex = (U32)(current - rep[i]);
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, 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+minMatch, repMatch+minMatch, iend, repEnd, prefixStart) + minMatch;
2016-02-10 12:37:52 +00:00
ZSTD_LOG_PARSER("%d: start try REP rep[%d]=%d mlen=%d\n", (int)(ip-base), i, (int)rep[i], (int)mlen);
if (mlen > sufficient_len || mlen >= ZSTD_OPT_NUM) {
best_mlen = mlen; best_off = i; cur = 0; last_pos = 1;
goto _storeSequence;
}
best_off = (i<=1 && ip == anchor) ? 1-i : i;
litlen = opt[0].litlen;
2016-02-10 12:37:52 +00:00
do {
2016-04-06 07:46:01 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, best_off, mlen - MINMATCH);
if (mlen > last_pos || price < opt[mlen].price)
SET_PRICE(mlen, mlen, i, litlen, price); /* note : macro modifies last_pos */
mlen--;
} while (mlen >= minMatch);
2016-04-08 13:25:32 +00:00
} } }
2016-04-05 19:30:03 +00:00
match_num = ZSTD_BtGetAllMatches_selectMLS_extDict(ctx, ip, iend, maxSearches, mls, matches); /* first search (depth 0) */
2016-04-05 19:30:03 +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-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_INIT; i++) opt[0].rep[i] = rep[i]; }
2016-04-05 19:30:03 +00:00
opt[0].mlen = 1;
2016-04-05 19:30:03 +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-04-05 19:30:03 +00:00
}
best_mlen = (last_pos) ? last_pos : minMatch;
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-04-06 07:46:01 +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++) {
inr = ip + cur;
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);
} 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;
if (opt[cur].off >= ZSTD_REP_NUM) {
opt[cur].rep[2] = opt[cur-mlen].rep[1];
opt[cur].rep[1] = opt[cur-mlen].rep[0];
opt[cur].rep[0] = opt[cur].off - ZSTD_REP_MOVE;
ZSTD_LOG_ENCODE("%d: COPYREP_OFF cur=%d mlen=%d rep[0]=%d rep[1]=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep[0], opt[cur].rep[1]);
} else {
opt[cur].rep[2] = (opt[cur].off > 1) ? opt[cur-mlen].rep[1] : opt[cur-mlen].rep[2];
opt[cur].rep[1] = (opt[cur].off > 0) ? opt[cur-mlen].rep[0] : opt[cur-mlen].rep[1];
opt[cur].rep[0] = opt[cur-mlen].rep[opt[cur].off];
ZSTD_LOG_ENCODE("%d: COPYREP_NOR cur=%d mlen=%d rep[0]=%d rep[1]=%d\n", (int)(inr-base), cur, mlen, opt[cur].rep[0], opt[cur].rep[1]);
}
ZSTD_LOG_PARSER("%d: CURRENT_Ext price[%d/%d]=%d off=%d mlen=%d litlen=%d rep[0]=%d rep[1]=%d\n", (int)(inr-base), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep[0], opt[cur].rep[1]);
best_mlen = 0;
2016-04-08 13:25:32 +00:00
{ U32 i; for (i=0; i<ZSTD_REP_NUM; i++) {
const U32 repIndex = (U32)(current+cur - opt[cur].rep[i]);
const BYTE* const repBase = repIndex < dictLimit ? dictBase : base;
const BYTE* const repMatch = repBase + repIndex;
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 %d/%d mlen=%d off=%d rep=%d opt[%d].off=%d\n", (int)(inr-base), i, ZSTD_REP_NUM, mlen, i, opt[cur].rep[i], cur, opt[cur].off);
if (mlen > sufficient_len || cur + mlen >= ZSTD_OPT_NUM) {
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);
2016-04-05 19:30:03 +00:00
best_mlen = mlen; best_off = i; last_pos = cur + 1;
goto _storeSequence;
}
best_off = (i<=1 && opt[cur].mlen != 1) ? 1-i : i;
if (opt[cur].mlen == 1) {
litlen = opt[cur].litlen;
if (cur > litlen) {
2016-04-06 07:46:01 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, inr-litlen, best_off, mlen - MINMATCH);
} else
2016-04-06 07:46:01 +00:00
price = ZSTD_getPrice(seqStorePtr, litlen, litstart, best_off, mlen - MINMATCH);
} else {
litlen = 0;
2016-04-06 07:46:01 +00:00
price = opt[cur].price + ZSTD_getPrice(seqStorePtr, 0, NULL, best_off, mlen - MINMATCH);
}
best_mlen = mlen;
ZSTD_LOG_PARSER("%d: Found REP mlen=%d off=%d price=%d litlen=%d\n", (int)(inr-base), mlen, best_off, price, litlen);
do {
if (cur + mlen > last_pos || price <= opt[cur + mlen].price)
SET_PRICE(cur + mlen, mlen, i, litlen, price);
mlen--;
} while (mlen >= minMatch);
2016-04-08 13:25:32 +00:00
} } }
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;
}
2016-04-05 19:30:03 +00:00
best_mlen = (best_mlen > minMatch) ? best_mlen : minMatch;
/* 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-04-06 07:46:01 +00:00
price = opt[cur - litlen].price + ZSTD_getPrice(seqStorePtr, litlen, ip+cur-litlen, matches[u].off, mlen - MINMATCH);
else
2016-04-06 07:46:01 +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-04-06 07:46:01 +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++;
} } } /* 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++)
ZSTD_LOG_PARSER("%d: price[%u/%d]=%d off=%d mlen=%d litlen=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep[0], opt[u].rep[1]);
ZSTD_LOG_PARSER("%d: cur=%d/%d best_mlen=%d best_off=%d rep[0]=%d\n", (int)(ip-base+cur), (int)cur, (int)last_pos, (int)best_mlen, (int)best_off, opt[cur].rep[0]);
opt[0].mlen = 1;
U32 offset;
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; ) {
ZSTD_LOG_PARSER("%d: price2[%d/%d]=%d off=%d mlen=%d litlen=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base+u), u, last_pos, opt[u].price, opt[u].off, opt[u].mlen, opt[u].litlen, opt[u].rep[0], opt[u].rep[1]);
2016-02-10 12:37:52 +00:00
u += opt[u].mlen;
}
for (cur=0; cur < last_pos; ) {
ZSTD_LOG_PARSER("%d: price3[%d/%d]=%d off=%d mlen=%d litlen=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base+cur), cur, last_pos, opt[cur].price, opt[cur].off, opt[cur].mlen, opt[cur].litlen, opt[cur].rep[0], opt[cur].rep[1]);
mlen = opt[cur].mlen;
if (mlen == 1) { ip++; cur++; continue; }
offset = opt[cur].off;
cur += mlen;
U32 const litLength = (U32)(ip - anchor);
// ZSTD_LOG_ENCODE("%d/%d: ENCODE1 literals=%d mlen=%d off=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep[0], (int)rep[1]);
if (offset >= ZSTD_REP_NUM) {
rep[2] = rep[1];
rep[1] = rep[0];
rep[0] = offset - ZSTD_REP_MOVE;
2016-02-10 12:37:52 +00:00
} else {
if (offset != 0) {
best_off = rep[offset];
if (offset != 1) rep[2] = rep[1];
rep[1] = rep[0];
rep[0] = best_off;
}
if (litLength == 0 && offset<=1) offset = 1-offset;
}
ZSTD_LOG_ENCODE("%d/%d: ENCODE literals=%d mlen=%d off=%d rep[0]=%d rep[1]=%d\n", (int)(ip-base), (int)(iend-base), (int)(litLength), (int)mlen, (int)(offset), (int)rep[0], (int)rep[1]);
2016-02-10 12:37:52 +00:00
#if ZSTD_OPT_DEBUG >= 5
2016-02-19 10:05:25 +00:00
U32 ml2;
if (offset >= ZSTD_REP_NUM) {
best_off = offset - ZSTD_REP_MOVE;
if (best_off > (size_t)(ip - prefixStart)) {
const BYTE* match = dictEnd - (best_off - (ip - prefixStart));
2016-02-29 21:07:40 +00:00
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)best_off, dictBase, dictEnd, prefixStart, ip, match);
2016-02-29 21:07:40 +00:00
}
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
}
else ml2 = (U32)ZSTD_count(ip, ip-rep[0], 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
2016-04-06 07:46:01 +00:00
ZSTD_updatePrice(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
ZSTD_storeSeq(seqStorePtr, litLength, anchor, offset, mlen-MINMATCH);
anchor = ip = ip + mlen;
} } /* for (cur=0; cur < last_pos; ) */
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;
}
}