| /* |
| lz4opt.h - Optimal Mode of LZ4 |
| Copyright (C) 2015-2017, Przemyslaw Skibinski <inikep@gmail.com> |
| Note : this file is intended to be included within lz4hc.c |
| |
| 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 : |
| - LZ4 source repository : https://github.com/lz4/lz4 |
| - LZ4 public forum : https://groups.google.com/forum/#!forum/lz4c |
| */ |
| |
| #define LZ4_OPT_NUM (1<<12) |
| |
| |
| typedef struct { |
| int off; |
| int len; |
| } LZ4HC_match_t; |
| |
| typedef struct { |
| int price; |
| int off; |
| int mlen; |
| int litlen; |
| } LZ4HC_optimal_t; |
| |
| |
| /* price in bytes */ |
| LZ4_FORCE_INLINE int LZ4HC_literalsPrice(int const litlen) |
| { |
| int price = litlen; |
| if (litlen >= (int)RUN_MASK) |
| price += 1 + (litlen-RUN_MASK)/255; |
| return price; |
| } |
| |
| |
| /* requires mlen >= MINMATCH */ |
| LZ4_FORCE_INLINE int LZ4HC_sequencePrice(int litlen, int mlen) |
| { |
| int price = 2 + 1; /* 16-bit offset + token */ |
| |
| price += LZ4HC_literalsPrice(litlen); |
| |
| if (mlen >= (int)(ML_MASK+MINMATCH)) |
| price += 1 + (mlen-(ML_MASK+MINMATCH))/255; |
| |
| return price; |
| } |
| |
| |
| /*-************************************* |
| * Binary Tree search |
| ***************************************/ |
| LZ4_FORCE_INLINE int LZ4HC_BinTree_InsertAndGetAllMatches ( |
| LZ4HC_CCtx_internal* ctx, |
| const BYTE* const ip, |
| const BYTE* const iHighLimit, |
| size_t best_mlen, |
| LZ4HC_match_t* matches, |
| int* matchNum) |
| { |
| U16* const chainTable = ctx->chainTable; |
| U32* const HashTable = ctx->hashTable; |
| const BYTE* const base = ctx->base; |
| const U32 dictLimit = ctx->dictLimit; |
| const U32 current = (U32)(ip - base); |
| const U32 lowLimit = (ctx->lowLimit + MAX_DISTANCE > current) ? ctx->lowLimit : current - (MAX_DISTANCE - 1); |
| const BYTE* const dictBase = ctx->dictBase; |
| const BYTE* match; |
| int nbAttempts = ctx->searchNum; |
| int mnum = 0; |
| U16 *ptr0, *ptr1, delta0, delta1; |
| U32 matchIndex; |
| size_t matchLength = 0; |
| U32* HashPos; |
| |
| if (ip + MINMATCH > iHighLimit) return 1; |
| |
| /* HC4 match finder */ |
| HashPos = &HashTable[LZ4HC_hashPtr(ip)]; |
| matchIndex = *HashPos; |
| *HashPos = current; |
| |
| ptr0 = &DELTANEXTMAXD(current*2+1); |
| ptr1 = &DELTANEXTMAXD(current*2); |
| delta0 = delta1 = (U16)(current - matchIndex); |
| |
| while ((matchIndex < current) && (matchIndex>=lowLimit) && (nbAttempts)) { |
| nbAttempts--; |
| if (matchIndex >= dictLimit) { |
| match = base + matchIndex; |
| matchLength = LZ4_count(ip, match, iHighLimit); |
| } else { |
| const BYTE* vLimit = ip + (dictLimit - matchIndex); |
| match = dictBase + matchIndex; |
| if (vLimit > iHighLimit) vLimit = iHighLimit; |
| matchLength = LZ4_count(ip, match, vLimit); |
| if ((ip+matchLength == vLimit) && (vLimit < iHighLimit)) |
| matchLength += LZ4_count(ip+matchLength, base+dictLimit, iHighLimit); |
| if (matchIndex+matchLength >= dictLimit) |
| match = base + matchIndex; /* to prepare for next usage of match[matchLength] */ |
| } |
| |
| if (matchLength > best_mlen) { |
| best_mlen = matchLength; |
| if (matches) { |
| if (matchIndex >= dictLimit) |
| matches[mnum].off = (int)(ip - match); |
| else |
| matches[mnum].off = (int)(ip - (base + matchIndex)); /* virtual matchpos */ |
| matches[mnum].len = (int)matchLength; |
| mnum++; |
| } |
| if (best_mlen > LZ4_OPT_NUM) break; |
| } |
| |
| if (ip+matchLength >= iHighLimit) /* equal : no way to know if inf or sup */ |
| break; /* drop , to guarantee consistency ; miss a bit of compression, but other solutions can corrupt the tree */ |
| |
| DEBUGLOG(6, "ip :%016llX", (U64)ip); |
| DEBUGLOG(6, "match:%016llX", (U64)match); |
| if (*(ip+matchLength) < *(match+matchLength)) { |
| *ptr0 = delta0; |
| ptr0 = &DELTANEXTMAXD(matchIndex*2); |
| if (*ptr0 == (U16)-1) break; |
| delta0 = *ptr0; |
| delta1 += delta0; |
| matchIndex -= delta0; |
| } else { |
| *ptr1 = delta1; |
| ptr1 = &DELTANEXTMAXD(matchIndex*2+1); |
| if (*ptr1 == (U16)-1) break; |
| delta1 = *ptr1; |
| delta0 += delta1; |
| matchIndex -= delta1; |
| } |
| } |
| |
| *ptr0 = (U16)-1; |
| *ptr1 = (U16)-1; |
| if (matchNum) *matchNum = mnum; |
| /* if (best_mlen > 8) return best_mlen-8; */ |
| if (!matchNum) return 1; |
| return 1; |
| } |
| |
| |
| LZ4_FORCE_INLINE void LZ4HC_updateBinTree(LZ4HC_CCtx_internal* ctx, const BYTE* const ip, const BYTE* const iHighLimit) |
| { |
| const BYTE* const base = ctx->base; |
| const U32 target = (U32)(ip - base); |
| U32 idx = ctx->nextToUpdate; |
| while(idx < target) |
| idx += LZ4HC_BinTree_InsertAndGetAllMatches(ctx, base+idx, iHighLimit, 8, NULL, NULL); |
| } |
| |
| |
| /** Tree updater, providing best match */ |
| LZ4_FORCE_INLINE int LZ4HC_BinTree_GetAllMatches ( |
| LZ4HC_CCtx_internal* ctx, |
| const BYTE* const ip, const BYTE* const iHighLimit, |
| size_t best_mlen, LZ4HC_match_t* matches, const int fullUpdate) |
| { |
| int mnum = 0; |
| if (ip < ctx->base + ctx->nextToUpdate) return 0; /* skipped area */ |
| if (fullUpdate) LZ4HC_updateBinTree(ctx, ip, iHighLimit); |
| best_mlen = LZ4HC_BinTree_InsertAndGetAllMatches(ctx, ip, iHighLimit, best_mlen, matches, &mnum); |
| ctx->nextToUpdate = (U32)(ip - ctx->base + best_mlen); |
| return mnum; |
| } |
| |
| |
| LZ4_FORCE_INLINE |
| int LZ4HC_FindLongerMatch(LZ4HC_CCtx_internal* const ctx, /* Index table will be updated */ |
| const BYTE* const ip, const BYTE* const iHighLimit, |
| int longest, |
| const BYTE** matchpos, |
| const int maxNbAttempts) |
| { |
| const BYTE* uselessPtr = ip; |
| return LZ4HC_InsertAndGetWiderMatch(ctx, ip, ip, iHighLimit, longest, matchpos, &uselessPtr, maxNbAttempts); |
| } |
| |
| LZ4_FORCE_INLINE int LZ4HC_HashChain_GetAllMatches ( |
| LZ4HC_CCtx_internal* ctx, |
| const BYTE* const ip, const BYTE* const iHighLimit, |
| size_t best_mlen, LZ4HC_match_t* matches, const int fullUpdate) |
| { |
| const BYTE* matchPtr = NULL; |
| int matchLength = LZ4HC_FindLongerMatch(ctx, ip, iHighLimit, (int)best_mlen, &matchPtr, ctx->searchNum); |
| if ((size_t)matchLength <= best_mlen) return 0; |
| assert(matches != NULL); |
| matches[0].len = matchLength; |
| matches[0].off = (int)(ip-matchPtr); |
| (void)fullUpdate; |
| return 1; |
| } |
| |
| |
| static int LZ4HC_compress_optimal ( |
| LZ4HC_CCtx_internal* ctx, |
| const char* const source, |
| char* dst, |
| int inputSize, |
| int dstCapacity, |
| limitedOutput_directive limit, |
| size_t sufficient_len, |
| const int fullUpdate |
| ) |
| { |
| LZ4HC_optimal_t opt[LZ4_OPT_NUM + 1]; /* this uses a bit too much stack memory to my taste ... */ |
| LZ4HC_match_t matches[LZ4_OPT_NUM + 1]; |
| |
| const BYTE* ip = (const BYTE*) source; |
| const BYTE* anchor = ip; |
| const BYTE* const iend = ip + inputSize; |
| const BYTE* const mflimit = iend - MFLIMIT; |
| const BYTE* const matchlimit = (iend - LASTLITERALS); |
| BYTE* op = (BYTE*) dst; |
| BYTE* const oend = op + dstCapacity; |
| |
| /* init */ |
| DEBUGLOG(5, "LZ4HC_compress_optimal"); |
| if (sufficient_len >= LZ4_OPT_NUM) sufficient_len = LZ4_OPT_NUM-1; |
| ctx->end += inputSize; |
| ip++; |
| |
| /* Main Loop */ |
| assert(ip - anchor < LZ4_MAX_INPUT_SIZE); |
| while (ip < mflimit) { |
| int const llen = (int)(ip - anchor); |
| int best_mlen, best_off; |
| int cur, last_match_pos = 0; |
| |
| //int const nb_matches_initial = LZ4HC_BinTree_GetAllMatches(ctx, ip, matchlimit, MINMATCH-1, matches, fullUpdate); |
| int const nb_matches_initial = LZ4HC_HashChain_GetAllMatches(ctx, ip, matchlimit, MINMATCH-1, matches, fullUpdate); |
| if (!nb_matches_initial) { ip++; continue; } |
| |
| if ((size_t)matches[nb_matches_initial-1].len > sufficient_len) { |
| /* good enough solution : immediate encoding */ |
| int const firstML = matches[nb_matches_initial-1].len; |
| const BYTE* const matchPos = ip - matches[nb_matches_initial-1].off; |
| if ( LZ4HC_encodeSequence(&ip, &op, &anchor, firstML, matchPos, limit, oend) ) /* updates ip, op and anchor */ |
| return 0; /* error */ |
| continue; |
| } |
| |
| /* set prices for first positions (literals) */ |
| { int rPos; |
| for (rPos = 0 ; rPos < MINMATCH ; rPos++) { |
| int const cost = LZ4HC_literalsPrice(llen + rPos); |
| opt[rPos].mlen = 1; |
| opt[rPos].off = 0; |
| opt[rPos].litlen = llen + rPos; |
| opt[rPos].price = cost; |
| DEBUGLOG(7, "rPos:%3u => cost:%3i (litlen=%i)", |
| (U32)rPos, cost, opt[rPos].litlen); |
| } } |
| /* set prices using matches found for rPos = 0 */ |
| { int matchNb; |
| for (matchNb = 0; matchNb < nb_matches_initial; matchNb++) { |
| int mlen = (matchNb>0) ? matches[matchNb-1].len+1 : MINMATCH; |
| int const matchML = matches[matchNb].len; /* necessarily < sufficient_len < LZ4_OPT_NUM */ |
| int const offset = matches[matchNb].off; |
| assert(matchML < LZ4_OPT_NUM); |
| for ( ; mlen <= matchML ; mlen++) { |
| int const cost = LZ4HC_sequencePrice(llen, mlen); |
| opt[mlen].mlen = mlen; |
| opt[mlen].off = offset; |
| opt[mlen].litlen = llen; |
| opt[mlen].price = cost; |
| } } } |
| last_match_pos = matches[nb_matches_initial-1].len; |
| |
| /* check further positions */ |
| for (cur = 1; cur <= last_match_pos; cur++) { |
| const BYTE* const curPtr = ip + cur; |
| int nb_matches; |
| |
| /* establish baseline price for cur as a literal. |
| * fixes unused positions (between 2 matches) |
| * and inefficient match stack */ |
| { int price; |
| int litlen; |
| if (opt[cur-1].mlen == 1) { |
| /* no match at previous position */ |
| litlen = opt[cur-1].litlen + 1; |
| price = opt[cur-1].price - LZ4HC_literalsPrice(litlen-1) + LZ4HC_literalsPrice(litlen); |
| } else { |
| litlen = 1; |
| price = opt[cur - 1].price + LZ4HC_literalsPrice(1); |
| } |
| if (price < opt[cur].price) { |
| opt[cur].mlen = 1; |
| opt[cur].off = 0; |
| opt[cur].litlen = litlen; |
| opt[cur].price = price; |
| } |
| } |
| |
| if (cur == last_match_pos || curPtr >= mflimit) break; |
| |
| //nb_matches = LZ4HC_BinTree_GetAllMatches(ctx, curPtr, matchlimit, MINMATCH-1, matches, fullUpdate); |
| nb_matches = LZ4HC_HashChain_GetAllMatches(ctx, curPtr, matchlimit, MINMATCH-1, matches, fullUpdate); |
| //nb_matches = LZ4HC_HashChain_GetAllMatches(ctx, curPtr, matchlimit, last_match_pos - cur + 1, matches, fullUpdate); /* only works if last_match_pos is really the last match pos */ |
| if ((nb_matches > 0) && (size_t)matches[nb_matches-1].len > sufficient_len) { |
| /* immediate encoding */ |
| best_mlen = matches[nb_matches-1].len; |
| best_off = matches[nb_matches-1].off; |
| last_match_pos = cur + 1; |
| goto encode; |
| } |
| |
| /* set prices using matches at position = cur */ |
| { int matchNb; |
| for (matchNb = 0; matchNb < nb_matches; matchNb++) { |
| int const matchML = (cur + matches[matchNb].len < LZ4_OPT_NUM) ? |
| matches[matchNb].len : (int)(LZ4_OPT_NUM - cur); |
| int ml = (matchNb>0) ? matches[matchNb-1].len+1 : MINMATCH; |
| |
| for ( ; ml <= matchML ; ml++) { |
| int const pos = cur + ml; |
| int const offset = matches[matchNb].off; |
| int price; |
| int ll; |
| if (opt[cur].mlen == 1) { |
| ll = opt[cur].litlen; |
| price = ((cur > ll) ? opt[cur - ll].price : 0) + LZ4HC_sequencePrice(ll, ml); |
| } else { |
| ll = 0; |
| price = opt[cur].price + LZ4HC_sequencePrice(0, ml); |
| } |
| |
| if (pos > last_match_pos || price < opt[pos].price) { |
| while (last_match_pos < pos) opt[++last_match_pos].price = 1<<30; |
| opt[pos].mlen = ml; |
| opt[pos].off = offset; |
| opt[pos].litlen = ll; |
| opt[pos].price = price; |
| } } } } |
| |
| } /* for (cur = 1; cur <= last_match_pos; cur++) */ |
| |
| best_mlen = opt[last_match_pos].mlen; |
| best_off = opt[last_match_pos].off; |
| cur = last_match_pos - best_mlen; |
| |
| encode: /* cur, last_match_pos, best_mlen, best_off must be set */ |
| assert(cur < LZ4_OPT_NUM); |
| assert(last_match_pos >= 1); /* == 1 when only one candidate */ |
| { int candidate_pos = cur; |
| int selected_matchLength = best_mlen; |
| int selected_offset = best_off; |
| while (1) { /* from end to beginning */ |
| int const next_matchLength = opt[candidate_pos].mlen; /* can be 1, means literal */ |
| int const next_offset = opt[candidate_pos].off; |
| opt[candidate_pos].mlen = selected_matchLength; |
| opt[candidate_pos].off = selected_offset; |
| selected_matchLength = next_matchLength; |
| selected_offset = next_offset; |
| if (next_matchLength > candidate_pos) break; /* last match elected, first match to encode */ |
| assert(next_matchLength > 0); /* can be 1, means literal */ |
| candidate_pos -= next_matchLength; |
| } } |
| |
| /* encode all recorded sequences in order */ |
| { int rPos = 0; /* relative position (to ip) */ |
| while (rPos < last_match_pos) { |
| int const ml = opt[rPos].mlen; |
| int const offset = opt[rPos].off; |
| if (ml == 1) { ip++; rPos++; continue; } /* literal; note: can end up with several literals, in which case, skip them */ |
| rPos += ml; |
| assert(ml >= MINMATCH); |
| assert((offset >= 1) && (offset <=65535)); |
| if ( LZ4HC_encodeSequence(&ip, &op, &anchor, ml, ip - offset, limit, oend) ) /* updates ip, op and anchor */ |
| return 0; /* error */ |
| } } |
| } /* while (ip < mflimit) */ |
| |
| /* Encode Last Literals */ |
| { int lastRun = (int)(iend - anchor); |
| if ( (limit) |
| && (((char*)op - dst) + lastRun + 1 + ((lastRun+255-RUN_MASK)/255) > (U32)dstCapacity)) |
| return 0; /* Check output limit */ |
| if (lastRun >= (int)RUN_MASK) { |
| *op++=(RUN_MASK<<ML_BITS); |
| lastRun-=RUN_MASK; |
| for (; lastRun > 254 ; lastRun-=255) *op++ = 255; |
| *op++ = (BYTE) lastRun; |
| } else *op++ = (BYTE)(lastRun<<ML_BITS); |
| memcpy(op, anchor, iend - anchor); |
| op += iend-anchor; |
| } |
| |
| /* End */ |
| return (int) ((char*)op-dst); |
| } |