mirror of
https://github.com/google/brotli.git
synced 2024-11-25 13:00:06 +00:00
fe754f3459
PiperOrigin-RevId: 638686412
756 lines
24 KiB
C
756 lines
24 KiB
C
/* Copyright 2010 Google Inc. All Rights Reserved.
|
|
|
|
Distributed under MIT license.
|
|
See file LICENSE for detail or copy at https://opensource.org/licenses/MIT
|
|
*/
|
|
|
|
/* A (forgetful) hash table to the data seen by the compressor, to
|
|
help create backward references to previous data. */
|
|
|
|
#ifndef BROTLI_ENC_HASH_H_
|
|
#define BROTLI_ENC_HASH_H_
|
|
|
|
#include <stdlib.h> /* exit */
|
|
#include <string.h> /* memcmp, memset */
|
|
|
|
#include <brotli/types.h>
|
|
|
|
#include "../common/constants.h"
|
|
#include "../common/dictionary.h"
|
|
#include "../common/platform.h"
|
|
#include "compound_dictionary.h"
|
|
#include "encoder_dict.h"
|
|
#include "fast_log.h"
|
|
#include "find_match_length.h"
|
|
#include "matching_tag_mask.h"
|
|
#include "memory.h"
|
|
#include "quality.h"
|
|
#include "static_dict.h"
|
|
|
|
#if defined(__cplusplus) || defined(c_plusplus)
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct {
|
|
/**
|
|
* Dynamically allocated areas; regular hasher uses one or two allocations;
|
|
* "composite" hasher uses up to 4 allocations.
|
|
*/
|
|
void* extra[4];
|
|
|
|
/**
|
|
* False before the fisrt invocation of HasherSetup (where "extra" memory)
|
|
* is allocated.
|
|
*/
|
|
BROTLI_BOOL is_setup_;
|
|
|
|
size_t dict_num_lookups;
|
|
size_t dict_num_matches;
|
|
|
|
BrotliHasherParams params;
|
|
|
|
/**
|
|
* False if hasher needs to be "prepared" before use (before the first
|
|
* invocation of HasherSetup or after HasherReset). "preparation" is hasher
|
|
* data initialization (using input ringbuffer).
|
|
*/
|
|
BROTLI_BOOL is_prepared_;
|
|
} HasherCommon;
|
|
|
|
#define score_t size_t
|
|
|
|
static const uint32_t kCutoffTransformsCount = 10;
|
|
/* 0, 12, 27, 23, 42, 63, 56, 48, 59, 64 */
|
|
/* 0+0, 4+8, 8+19, 12+11, 16+26, 20+43, 24+32, 28+20, 32+27, 36+28 */
|
|
static const uint64_t kCutoffTransforms =
|
|
BROTLI_MAKE_UINT64_T(0x071B520A, 0xDA2D3200);
|
|
|
|
typedef struct HasherSearchResult {
|
|
size_t len;
|
|
size_t distance;
|
|
score_t score;
|
|
int len_code_delta; /* == len_code - len */
|
|
} HasherSearchResult;
|
|
|
|
/* kHashMul32 multiplier has these properties:
|
|
* The multiplier must be odd. Otherwise we may lose the highest bit.
|
|
* No long streaks of ones or zeros.
|
|
* There is no effort to ensure that it is a prime, the oddity is enough
|
|
for this use.
|
|
* The number has been tuned heuristically against compression benchmarks. */
|
|
static const uint32_t kHashMul32 = 0x1E35A7BD;
|
|
static const uint64_t kHashMul64 =
|
|
BROTLI_MAKE_UINT64_T(0x1FE35A7Bu, 0xD3579BD3u);
|
|
|
|
static BROTLI_INLINE uint32_t Hash14(const uint8_t* data) {
|
|
uint32_t h = BROTLI_UNALIGNED_LOAD32LE(data) * kHashMul32;
|
|
/* The higher bits contain more mixture from the multiplication,
|
|
so we take our results from there. */
|
|
return h >> (32 - 14);
|
|
}
|
|
|
|
static BROTLI_INLINE void PrepareDistanceCache(
|
|
int* BROTLI_RESTRICT distance_cache, const int num_distances) {
|
|
if (num_distances > 4) {
|
|
int last_distance = distance_cache[0];
|
|
distance_cache[4] = last_distance - 1;
|
|
distance_cache[5] = last_distance + 1;
|
|
distance_cache[6] = last_distance - 2;
|
|
distance_cache[7] = last_distance + 2;
|
|
distance_cache[8] = last_distance - 3;
|
|
distance_cache[9] = last_distance + 3;
|
|
if (num_distances > 10) {
|
|
int next_last_distance = distance_cache[1];
|
|
distance_cache[10] = next_last_distance - 1;
|
|
distance_cache[11] = next_last_distance + 1;
|
|
distance_cache[12] = next_last_distance - 2;
|
|
distance_cache[13] = next_last_distance + 2;
|
|
distance_cache[14] = next_last_distance - 3;
|
|
distance_cache[15] = next_last_distance + 3;
|
|
}
|
|
}
|
|
}
|
|
|
|
#define BROTLI_LITERAL_BYTE_SCORE 135
|
|
#define BROTLI_DISTANCE_BIT_PENALTY 30
|
|
/* Score must be positive after applying maximal penalty. */
|
|
#define BROTLI_SCORE_BASE (BROTLI_DISTANCE_BIT_PENALTY * 8 * sizeof(size_t))
|
|
|
|
/* Usually, we always choose the longest backward reference. This function
|
|
allows for the exception of that rule.
|
|
|
|
If we choose a backward reference that is further away, it will
|
|
usually be coded with more bits. We approximate this by assuming
|
|
log2(distance). If the distance can be expressed in terms of the
|
|
last four distances, we use some heuristic constants to estimate
|
|
the bits cost. For the first up to four literals we use the bit
|
|
cost of the literals from the literal cost model, after that we
|
|
use the average bit cost of the cost model.
|
|
|
|
This function is used to sometimes discard a longer backward reference
|
|
when it is not much longer and the bit cost for encoding it is more
|
|
than the saved literals.
|
|
|
|
backward_reference_offset MUST be positive. */
|
|
static BROTLI_INLINE score_t BackwardReferenceScore(
|
|
size_t copy_length, size_t backward_reference_offset) {
|
|
return BROTLI_SCORE_BASE + BROTLI_LITERAL_BYTE_SCORE * (score_t)copy_length -
|
|
BROTLI_DISTANCE_BIT_PENALTY * Log2FloorNonZero(backward_reference_offset);
|
|
}
|
|
|
|
static BROTLI_INLINE score_t BackwardReferenceScoreUsingLastDistance(
|
|
size_t copy_length) {
|
|
return BROTLI_LITERAL_BYTE_SCORE * (score_t)copy_length +
|
|
BROTLI_SCORE_BASE + 15;
|
|
}
|
|
|
|
static BROTLI_INLINE score_t BackwardReferencePenaltyUsingLastDistance(
|
|
size_t distance_short_code) {
|
|
return (score_t)39 + ((0x1CA10 >> (distance_short_code & 0xE)) & 0xE);
|
|
}
|
|
|
|
static BROTLI_INLINE BROTLI_BOOL TestStaticDictionaryItem(
|
|
const BrotliEncoderDictionary* dictionary, size_t len, size_t word_idx,
|
|
const uint8_t* data, size_t max_length, size_t max_backward,
|
|
size_t max_distance, HasherSearchResult* out) {
|
|
size_t offset;
|
|
size_t matchlen;
|
|
size_t backward;
|
|
score_t score;
|
|
offset = dictionary->words->offsets_by_length[len] + len * word_idx;
|
|
if (len > max_length) {
|
|
return BROTLI_FALSE;
|
|
}
|
|
|
|
matchlen =
|
|
FindMatchLengthWithLimit(data, &dictionary->words->data[offset], len);
|
|
if (matchlen + dictionary->cutoffTransformsCount <= len || matchlen == 0) {
|
|
return BROTLI_FALSE;
|
|
}
|
|
{
|
|
size_t cut = len - matchlen;
|
|
size_t transform_id = (cut << 2) +
|
|
(size_t)((dictionary->cutoffTransforms >> (cut * 6)) & 0x3F);
|
|
backward = max_backward + 1 + word_idx +
|
|
(transform_id << dictionary->words->size_bits_by_length[len]);
|
|
}
|
|
if (backward > max_distance) {
|
|
return BROTLI_FALSE;
|
|
}
|
|
score = BackwardReferenceScore(matchlen, backward);
|
|
if (score < out->score) {
|
|
return BROTLI_FALSE;
|
|
}
|
|
out->len = matchlen;
|
|
out->len_code_delta = (int)len - (int)matchlen;
|
|
out->distance = backward;
|
|
out->score = score;
|
|
return BROTLI_TRUE;
|
|
}
|
|
|
|
static BROTLI_INLINE void SearchInStaticDictionary(
|
|
const BrotliEncoderDictionary* dictionary,
|
|
HasherCommon* common, const uint8_t* data, size_t max_length,
|
|
size_t max_backward, size_t max_distance,
|
|
HasherSearchResult* out, BROTLI_BOOL shallow) {
|
|
size_t key;
|
|
size_t i;
|
|
if (common->dict_num_matches < (common->dict_num_lookups >> 7)) {
|
|
return;
|
|
}
|
|
key = Hash14(data) << 1;
|
|
for (i = 0; i < (shallow ? 1u : 2u); ++i, ++key) {
|
|
common->dict_num_lookups++;
|
|
if (dictionary->hash_table_lengths[key] != 0) {
|
|
BROTLI_BOOL item_matches = TestStaticDictionaryItem(
|
|
dictionary, dictionary->hash_table_lengths[key],
|
|
dictionary->hash_table_words[key], data,
|
|
max_length, max_backward, max_distance, out);
|
|
if (item_matches) {
|
|
common->dict_num_matches++;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
typedef struct BackwardMatch {
|
|
uint32_t distance;
|
|
uint32_t length_and_code;
|
|
} BackwardMatch;
|
|
|
|
static BROTLI_INLINE void InitBackwardMatch(BackwardMatch* self,
|
|
size_t dist, size_t len) {
|
|
self->distance = (uint32_t)dist;
|
|
self->length_and_code = (uint32_t)(len << 5);
|
|
}
|
|
|
|
static BROTLI_INLINE void InitDictionaryBackwardMatch(BackwardMatch* self,
|
|
size_t dist, size_t len, size_t len_code) {
|
|
self->distance = (uint32_t)dist;
|
|
self->length_and_code =
|
|
(uint32_t)((len << 5) | (len == len_code ? 0 : len_code));
|
|
}
|
|
|
|
static BROTLI_INLINE size_t BackwardMatchLength(const BackwardMatch* self) {
|
|
return self->length_and_code >> 5;
|
|
}
|
|
|
|
static BROTLI_INLINE size_t BackwardMatchLengthCode(const BackwardMatch* self) {
|
|
size_t code = self->length_and_code & 31;
|
|
return code ? code : BackwardMatchLength(self);
|
|
}
|
|
|
|
#define EXPAND_CAT(a, b) CAT(a, b)
|
|
#define CAT(a, b) a ## b
|
|
#define FN(X) EXPAND_CAT(X, HASHER())
|
|
|
|
#define HASHER() H10
|
|
#define BUCKET_BITS 17
|
|
#define MAX_TREE_SEARCH_DEPTH 64
|
|
#define MAX_TREE_COMP_LENGTH 128
|
|
#include "hash_to_binary_tree_inc.h" /* NOLINT(build/include) */
|
|
#undef MAX_TREE_SEARCH_DEPTH
|
|
#undef MAX_TREE_COMP_LENGTH
|
|
#undef BUCKET_BITS
|
|
#undef HASHER
|
|
/* MAX_NUM_MATCHES == 64 + MAX_TREE_SEARCH_DEPTH */
|
|
#define MAX_NUM_MATCHES_H10 128
|
|
|
|
/* For BUCKET_SWEEP_BITS == 0, enabling the dictionary lookup makes compression
|
|
a little faster (0.5% - 1%) and it compresses 0.15% better on small text
|
|
and HTML inputs. */
|
|
|
|
#define HASHER() H2
|
|
#define BUCKET_BITS 16
|
|
#define BUCKET_SWEEP_BITS 0
|
|
#define HASH_LEN 5
|
|
#define USE_DICTIONARY 1
|
|
#include "hash_longest_match_quickly_inc.h" /* NOLINT(build/include) */
|
|
#undef BUCKET_SWEEP_BITS
|
|
#undef USE_DICTIONARY
|
|
#undef HASHER
|
|
|
|
#define HASHER() H3
|
|
#define BUCKET_SWEEP_BITS 1
|
|
#define USE_DICTIONARY 0
|
|
#include "hash_longest_match_quickly_inc.h" /* NOLINT(build/include) */
|
|
#undef USE_DICTIONARY
|
|
#undef BUCKET_SWEEP_BITS
|
|
#undef BUCKET_BITS
|
|
#undef HASHER
|
|
|
|
#define HASHER() H4
|
|
#define BUCKET_BITS 17
|
|
#define BUCKET_SWEEP_BITS 2
|
|
#define USE_DICTIONARY 1
|
|
#include "hash_longest_match_quickly_inc.h" /* NOLINT(build/include) */
|
|
#undef USE_DICTIONARY
|
|
#undef HASH_LEN
|
|
#undef BUCKET_SWEEP_BITS
|
|
#undef BUCKET_BITS
|
|
#undef HASHER
|
|
|
|
#define HASHER() H5
|
|
#include "hash_longest_match_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
|
|
#define HASHER() H6
|
|
#include "hash_longest_match64_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
|
|
#if defined(BROTLI_MAX_SIMD_QUALITY)
|
|
#define HASHER() H58
|
|
#include "hash_longest_match_simd_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
|
|
#define HASHER() H68
|
|
#include "hash_longest_match64_simd_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
#endif
|
|
|
|
#define BUCKET_BITS 15
|
|
|
|
#define NUM_LAST_DISTANCES_TO_CHECK 4
|
|
#define NUM_BANKS 1
|
|
#define BANK_BITS 16
|
|
#define HASHER() H40
|
|
#include "hash_forgetful_chain_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
#undef NUM_LAST_DISTANCES_TO_CHECK
|
|
|
|
#define NUM_LAST_DISTANCES_TO_CHECK 10
|
|
#define HASHER() H41
|
|
#include "hash_forgetful_chain_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
#undef NUM_LAST_DISTANCES_TO_CHECK
|
|
#undef NUM_BANKS
|
|
#undef BANK_BITS
|
|
|
|
#define NUM_LAST_DISTANCES_TO_CHECK 16
|
|
#define NUM_BANKS 512
|
|
#define BANK_BITS 9
|
|
#define HASHER() H42
|
|
#include "hash_forgetful_chain_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER
|
|
#undef NUM_LAST_DISTANCES_TO_CHECK
|
|
#undef NUM_BANKS
|
|
#undef BANK_BITS
|
|
|
|
#undef BUCKET_BITS
|
|
|
|
#define HASHER() H54
|
|
#define BUCKET_BITS 20
|
|
#define BUCKET_SWEEP_BITS 2
|
|
#define HASH_LEN 7
|
|
#define USE_DICTIONARY 0
|
|
#include "hash_longest_match_quickly_inc.h" /* NOLINT(build/include) */
|
|
#undef USE_DICTIONARY
|
|
#undef HASH_LEN
|
|
#undef BUCKET_SWEEP_BITS
|
|
#undef BUCKET_BITS
|
|
#undef HASHER
|
|
|
|
/* fast large window hashers */
|
|
|
|
#define HASHER() HROLLING_FAST
|
|
#define CHUNKLEN 32
|
|
#define JUMP 4
|
|
#define NUMBUCKETS 16777216
|
|
#define MASK ((NUMBUCKETS * 64) - 1)
|
|
#include "hash_rolling_inc.h" /* NOLINT(build/include) */
|
|
#undef JUMP
|
|
#undef HASHER
|
|
|
|
|
|
#define HASHER() HROLLING
|
|
#define JUMP 1
|
|
#include "hash_rolling_inc.h" /* NOLINT(build/include) */
|
|
#undef MASK
|
|
#undef NUMBUCKETS
|
|
#undef JUMP
|
|
#undef CHUNKLEN
|
|
#undef HASHER
|
|
|
|
#define HASHER() H35
|
|
#define HASHER_A H3
|
|
#define HASHER_B HROLLING_FAST
|
|
#include "hash_composite_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER_A
|
|
#undef HASHER_B
|
|
#undef HASHER
|
|
|
|
#define HASHER() H55
|
|
#define HASHER_A H54
|
|
#define HASHER_B HROLLING_FAST
|
|
#include "hash_composite_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER_A
|
|
#undef HASHER_B
|
|
#undef HASHER
|
|
|
|
#define HASHER() H65
|
|
#define HASHER_A H6
|
|
#define HASHER_B HROLLING
|
|
#include "hash_composite_inc.h" /* NOLINT(build/include) */
|
|
#undef HASHER_A
|
|
#undef HASHER_B
|
|
#undef HASHER
|
|
|
|
#undef FN
|
|
#undef CAT
|
|
#undef EXPAND_CAT
|
|
|
|
#if defined(BROTLI_MAX_SIMD_QUALITY)
|
|
#define FOR_SIMPLE_HASHERS(H) \
|
|
H(2) H(3) H(4) H(5) H(6) H(40) H(41) H(42) H(54) H(58) H(68)
|
|
#else
|
|
#define FOR_SIMPLE_HASHERS(H) \
|
|
H(2) H(3) H(4) H(5) H(6) H(40) H(41) H(42) H(54)
|
|
#endif
|
|
#define FOR_COMPOSITE_HASHERS(H) H(35) H(55) H(65)
|
|
#define FOR_GENERIC_HASHERS(H) FOR_SIMPLE_HASHERS(H) FOR_COMPOSITE_HASHERS(H)
|
|
#define FOR_ALL_HASHERS(H) FOR_GENERIC_HASHERS(H) H(10)
|
|
|
|
typedef struct {
|
|
HasherCommon common;
|
|
|
|
union {
|
|
#define MEMBER_(N) \
|
|
H ## N _H ## N;
|
|
FOR_ALL_HASHERS(MEMBER_)
|
|
#undef MEMBER_
|
|
} privat;
|
|
} Hasher;
|
|
|
|
/* MUST be invoked before any other method. */
|
|
static BROTLI_INLINE void HasherInit(Hasher* hasher) {
|
|
hasher->common.is_setup_ = BROTLI_FALSE;
|
|
hasher->common.extra[0] = NULL;
|
|
hasher->common.extra[1] = NULL;
|
|
hasher->common.extra[2] = NULL;
|
|
hasher->common.extra[3] = NULL;
|
|
}
|
|
|
|
static BROTLI_INLINE void DestroyHasher(MemoryManager* m, Hasher* hasher) {
|
|
if (hasher->common.extra[0] != NULL) BROTLI_FREE(m, hasher->common.extra[0]);
|
|
if (hasher->common.extra[1] != NULL) BROTLI_FREE(m, hasher->common.extra[1]);
|
|
if (hasher->common.extra[2] != NULL) BROTLI_FREE(m, hasher->common.extra[2]);
|
|
if (hasher->common.extra[3] != NULL) BROTLI_FREE(m, hasher->common.extra[3]);
|
|
}
|
|
|
|
static BROTLI_INLINE void HasherReset(Hasher* hasher) {
|
|
hasher->common.is_prepared_ = BROTLI_FALSE;
|
|
}
|
|
|
|
static BROTLI_INLINE void HasherSize(const BrotliEncoderParams* params,
|
|
BROTLI_BOOL one_shot, const size_t input_size, size_t* alloc_size) {
|
|
switch (params->hasher.type) {
|
|
#define SIZE_(N) \
|
|
case N: \
|
|
HashMemAllocInBytesH ## N(params, one_shot, input_size, alloc_size); \
|
|
break;
|
|
FOR_ALL_HASHERS(SIZE_)
|
|
#undef SIZE_
|
|
default:
|
|
break;
|
|
}
|
|
}
|
|
|
|
static BROTLI_INLINE void HasherSetup(MemoryManager* m, Hasher* hasher,
|
|
BrotliEncoderParams* params, const uint8_t* data, size_t position,
|
|
size_t input_size, BROTLI_BOOL is_last) {
|
|
BROTLI_BOOL one_shot = (position == 0 && is_last);
|
|
if (!hasher->common.is_setup_) {
|
|
size_t alloc_size[4] = {0};
|
|
size_t i;
|
|
ChooseHasher(params, ¶ms->hasher);
|
|
hasher->common.params = params->hasher;
|
|
hasher->common.dict_num_lookups = 0;
|
|
hasher->common.dict_num_matches = 0;
|
|
HasherSize(params, one_shot, input_size, alloc_size);
|
|
for (i = 0; i < 4; ++i) {
|
|
if (alloc_size[i] == 0) continue;
|
|
hasher->common.extra[i] = BROTLI_ALLOC(m, uint8_t, alloc_size[i]);
|
|
if (BROTLI_IS_OOM(m) || BROTLI_IS_NULL(hasher->common.extra[i])) return;
|
|
}
|
|
switch (hasher->common.params.type) {
|
|
#define INITIALIZE_(N) \
|
|
case N: \
|
|
InitializeH ## N(&hasher->common, \
|
|
&hasher->privat._H ## N, params); \
|
|
break;
|
|
FOR_ALL_HASHERS(INITIALIZE_);
|
|
#undef INITIALIZE_
|
|
default:
|
|
break;
|
|
}
|
|
HasherReset(hasher);
|
|
hasher->common.is_setup_ = BROTLI_TRUE;
|
|
}
|
|
|
|
if (!hasher->common.is_prepared_) {
|
|
switch (hasher->common.params.type) {
|
|
#define PREPARE_(N) \
|
|
case N: \
|
|
PrepareH ## N( \
|
|
&hasher->privat._H ## N, \
|
|
one_shot, input_size, data); \
|
|
break;
|
|
FOR_ALL_HASHERS(PREPARE_)
|
|
#undef PREPARE_
|
|
default: break;
|
|
}
|
|
hasher->common.is_prepared_ = BROTLI_TRUE;
|
|
}
|
|
}
|
|
|
|
static BROTLI_INLINE void InitOrStitchToPreviousBlock(
|
|
MemoryManager* m, Hasher* hasher, const uint8_t* data, size_t mask,
|
|
BrotliEncoderParams* params, size_t position, size_t input_size,
|
|
BROTLI_BOOL is_last) {
|
|
HasherSetup(m, hasher, params, data, position, input_size, is_last);
|
|
if (BROTLI_IS_OOM(m)) return;
|
|
switch (hasher->common.params.type) {
|
|
#define INIT_(N) \
|
|
case N: \
|
|
StitchToPreviousBlockH ## N( \
|
|
&hasher->privat._H ## N, \
|
|
input_size, position, data, mask); \
|
|
break;
|
|
FOR_ALL_HASHERS(INIT_)
|
|
#undef INIT_
|
|
default: break;
|
|
}
|
|
}
|
|
|
|
/* NB: when seamless dictionary-ring-buffer copies are implemented, don't forget
|
|
to add proper guards for non-zero-BROTLI_PARAM_STREAM_OFFSET. */
|
|
static BROTLI_INLINE void FindCompoundDictionaryMatch(
|
|
const PreparedDictionary* self, const uint8_t* BROTLI_RESTRICT data,
|
|
const size_t ring_buffer_mask, const int* BROTLI_RESTRICT distance_cache,
|
|
const size_t cur_ix, const size_t max_length, const size_t distance_offset,
|
|
const size_t max_distance, HasherSearchResult* BROTLI_RESTRICT out) {
|
|
const uint32_t source_size = self->source_size;
|
|
const size_t boundary = distance_offset - source_size;
|
|
const uint32_t hash_bits = self->hash_bits;
|
|
const uint32_t bucket_bits = self->bucket_bits;
|
|
const uint32_t slot_bits = self->slot_bits;
|
|
|
|
const uint32_t hash_shift = 64u - bucket_bits;
|
|
const uint32_t slot_mask = (~((uint32_t)0U)) >> (32 - slot_bits);
|
|
const uint64_t hash_mask = (~((uint64_t)0U)) >> (64 - hash_bits);
|
|
|
|
const uint32_t* slot_offsets = (uint32_t*)(&self[1]);
|
|
const uint16_t* heads = (uint16_t*)(&slot_offsets[1u << slot_bits]);
|
|
const uint32_t* items = (uint32_t*)(&heads[1u << bucket_bits]);
|
|
const uint8_t* source = NULL;
|
|
|
|
const size_t cur_ix_masked = cur_ix & ring_buffer_mask;
|
|
score_t best_score = out->score;
|
|
size_t best_len = out->len;
|
|
size_t i;
|
|
const uint64_t h =
|
|
(BROTLI_UNALIGNED_LOAD64LE(&data[cur_ix_masked]) & hash_mask) *
|
|
kPreparedDictionaryHashMul64Long;
|
|
const uint32_t key = (uint32_t)(h >> hash_shift);
|
|
const uint32_t slot = key & slot_mask;
|
|
const uint32_t head = heads[key];
|
|
const uint32_t* BROTLI_RESTRICT chain = &items[slot_offsets[slot] + head];
|
|
uint32_t item = (head == 0xFFFF) ? 1 : 0;
|
|
|
|
const void* tail = (void*)&items[self->num_items];
|
|
if (self->magic == kPreparedDictionaryMagic) {
|
|
source = (const uint8_t*)tail;
|
|
} else {
|
|
/* kLeanPreparedDictionaryMagic */
|
|
source = (const uint8_t*)BROTLI_UNALIGNED_LOAD_PTR((const uint8_t**)tail);
|
|
}
|
|
|
|
BROTLI_DCHECK(cur_ix_masked + max_length <= ring_buffer_mask);
|
|
|
|
for (i = 0; i < 4; ++i) {
|
|
const size_t distance = (size_t)distance_cache[i];
|
|
size_t offset;
|
|
size_t limit;
|
|
size_t len;
|
|
if (distance <= boundary || distance > distance_offset) continue;
|
|
offset = distance_offset - distance;
|
|
limit = source_size - offset;
|
|
limit = limit > max_length ? max_length : limit;
|
|
len = FindMatchLengthWithLimit(&source[offset], &data[cur_ix_masked],
|
|
limit);
|
|
if (len >= 2) {
|
|
score_t score = BackwardReferenceScoreUsingLastDistance(len);
|
|
if (best_score < score) {
|
|
if (i != 0) score -= BackwardReferencePenaltyUsingLastDistance(i);
|
|
if (best_score < score) {
|
|
best_score = score;
|
|
if (len > best_len) best_len = len;
|
|
out->len = len;
|
|
out->len_code_delta = 0;
|
|
out->distance = distance;
|
|
out->score = best_score;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
/* we require matches of len >4, so increase best_len to 3, so we can compare
|
|
* 4 bytes all the time. */
|
|
if (best_len < 3) {
|
|
best_len = 3;
|
|
}
|
|
while (item == 0) {
|
|
size_t offset;
|
|
size_t distance;
|
|
size_t limit;
|
|
item = *chain;
|
|
chain++;
|
|
offset = item & 0x7FFFFFFF;
|
|
item &= 0x80000000;
|
|
distance = distance_offset - offset;
|
|
limit = source_size - offset;
|
|
limit = (limit > max_length) ? max_length : limit;
|
|
if (distance > max_distance) continue;
|
|
if (cur_ix_masked + best_len > ring_buffer_mask || best_len >= limit ||
|
|
/* compare 4 bytes ending at best_len + 1 */
|
|
BrotliUnalignedRead32(&data[cur_ix_masked + best_len - 3]) !=
|
|
BrotliUnalignedRead32(&source[offset + best_len - 3])) {
|
|
continue;
|
|
}
|
|
{
|
|
const size_t len = FindMatchLengthWithLimit(&source[offset],
|
|
&data[cur_ix_masked],
|
|
limit);
|
|
if (len >= 4) {
|
|
score_t score = BackwardReferenceScore(len, distance);
|
|
if (best_score < score) {
|
|
best_score = score;
|
|
best_len = len;
|
|
out->len = best_len;
|
|
out->len_code_delta = 0;
|
|
out->distance = distance;
|
|
out->score = best_score;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* NB: when seamless dictionary-ring-buffer copies are implemented, don't forget
|
|
to add proper guards for non-zero-BROTLI_PARAM_STREAM_OFFSET. */
|
|
static BROTLI_INLINE size_t FindAllCompoundDictionaryMatches(
|
|
const PreparedDictionary* self, const uint8_t* BROTLI_RESTRICT data,
|
|
const size_t ring_buffer_mask, const size_t cur_ix, const size_t min_length,
|
|
const size_t max_length, const size_t distance_offset,
|
|
const size_t max_distance, BackwardMatch* matches, size_t match_limit) {
|
|
const uint32_t source_size = self->source_size;
|
|
const uint32_t hash_bits = self->hash_bits;
|
|
const uint32_t bucket_bits = self->bucket_bits;
|
|
const uint32_t slot_bits = self->slot_bits;
|
|
|
|
const uint32_t hash_shift = 64u - bucket_bits;
|
|
const uint32_t slot_mask = (~((uint32_t)0U)) >> (32 - slot_bits);
|
|
const uint64_t hash_mask = (~((uint64_t)0U)) >> (64 - hash_bits);
|
|
|
|
const uint32_t* slot_offsets = (uint32_t*)(&self[1]);
|
|
const uint16_t* heads = (uint16_t*)(&slot_offsets[1u << slot_bits]);
|
|
const uint32_t* items = (uint32_t*)(&heads[1u << bucket_bits]);
|
|
const uint8_t* source = NULL;
|
|
|
|
const size_t cur_ix_masked = cur_ix & ring_buffer_mask;
|
|
size_t best_len = min_length;
|
|
const uint64_t h =
|
|
(BROTLI_UNALIGNED_LOAD64LE(&data[cur_ix_masked]) & hash_mask) *
|
|
kPreparedDictionaryHashMul64Long;
|
|
const uint32_t key = (uint32_t)(h >> hash_shift);
|
|
const uint32_t slot = key & slot_mask;
|
|
const uint32_t head = heads[key];
|
|
const uint32_t* BROTLI_RESTRICT chain = &items[slot_offsets[slot] + head];
|
|
uint32_t item = (head == 0xFFFF) ? 1 : 0;
|
|
size_t found = 0;
|
|
|
|
const void* tail = (void*)&items[self->num_items];
|
|
if (self->magic == kPreparedDictionaryMagic) {
|
|
source = (const uint8_t*)tail;
|
|
} else {
|
|
/* kLeanPreparedDictionaryMagic */
|
|
source = (const uint8_t*)BROTLI_UNALIGNED_LOAD_PTR((const uint8_t**)tail);
|
|
}
|
|
|
|
BROTLI_DCHECK(cur_ix_masked + max_length <= ring_buffer_mask);
|
|
|
|
while (item == 0) {
|
|
size_t offset;
|
|
size_t distance;
|
|
size_t limit;
|
|
size_t len;
|
|
item = *chain;
|
|
chain++;
|
|
offset = item & 0x7FFFFFFF;
|
|
item &= 0x80000000;
|
|
distance = distance_offset - offset;
|
|
limit = source_size - offset;
|
|
limit = (limit > max_length) ? max_length : limit;
|
|
if (distance > max_distance) continue;
|
|
if (cur_ix_masked + best_len > ring_buffer_mask ||
|
|
best_len >= limit ||
|
|
data[cur_ix_masked + best_len] != source[offset + best_len]) {
|
|
continue;
|
|
}
|
|
len = FindMatchLengthWithLimit(
|
|
&source[offset], &data[cur_ix_masked], limit);
|
|
if (len > best_len) {
|
|
best_len = len;
|
|
InitBackwardMatch(matches++, distance, len);
|
|
found++;
|
|
if (found == match_limit) break;
|
|
}
|
|
}
|
|
return found;
|
|
}
|
|
|
|
static BROTLI_INLINE void LookupCompoundDictionaryMatch(
|
|
const CompoundDictionary* addon, const uint8_t* BROTLI_RESTRICT data,
|
|
const size_t ring_buffer_mask, const int* BROTLI_RESTRICT distance_cache,
|
|
const size_t cur_ix, const size_t max_length,
|
|
const size_t max_ring_buffer_distance, const size_t max_distance,
|
|
HasherSearchResult* sr) {
|
|
size_t base_offset = max_ring_buffer_distance + 1 + addon->total_size - 1;
|
|
size_t d;
|
|
for (d = 0; d < addon->num_chunks; ++d) {
|
|
/* Only one prepared dictionary type is currently supported. */
|
|
FindCompoundDictionaryMatch(
|
|
(const PreparedDictionary*)addon->chunks[d], data, ring_buffer_mask,
|
|
distance_cache, cur_ix, max_length,
|
|
base_offset - addon->chunk_offsets[d], max_distance, sr);
|
|
}
|
|
}
|
|
|
|
static BROTLI_INLINE size_t LookupAllCompoundDictionaryMatches(
|
|
const CompoundDictionary* addon, const uint8_t* BROTLI_RESTRICT data,
|
|
const size_t ring_buffer_mask, const size_t cur_ix, size_t min_length,
|
|
const size_t max_length, const size_t max_ring_buffer_distance,
|
|
const size_t max_distance, BackwardMatch* matches,
|
|
size_t match_limit) {
|
|
size_t base_offset = max_ring_buffer_distance + 1 + addon->total_size - 1;
|
|
size_t d;
|
|
size_t total_found = 0;
|
|
for (d = 0; d < addon->num_chunks; ++d) {
|
|
/* Only one prepared dictionary type is currently supported. */
|
|
total_found += FindAllCompoundDictionaryMatches(
|
|
(const PreparedDictionary*)addon->chunks[d], data, ring_buffer_mask,
|
|
cur_ix, min_length, max_length, base_offset - addon->chunk_offsets[d],
|
|
max_distance, matches + total_found, match_limit - total_found);
|
|
if (total_found == match_limit) break;
|
|
if (total_found > 0) {
|
|
min_length = BackwardMatchLength(&matches[total_found - 1]);
|
|
}
|
|
}
|
|
return total_found;
|
|
}
|
|
|
|
#if defined(__cplusplus) || defined(c_plusplus)
|
|
} /* extern "C" */
|
|
#endif
|
|
|
|
#endif /* BROTLI_ENC_HASH_H_ */
|