|
- function HuffmanCode(bits, value) {
- this.bits = bits; /* number of bits used for this symbol */
- this.value = value; /* symbol value or table offset */
- }
-
- exports.HuffmanCode = HuffmanCode;
-
- var MAX_LENGTH = 15;
-
- /* Returns reverse(reverse(key, len) + 1, len), where reverse(key, len) is the
- bit-wise reversal of the len least significant bits of key. */
- function GetNextKey(key, len) {
- var step = 1 << (len - 1);
- while (key & step) {
- step >>= 1;
- }
- return (key & (step - 1)) + step;
- }
-
- /* Stores code in table[0], table[step], table[2*step], ..., table[end] */
- /* Assumes that end is an integer multiple of step */
- function ReplicateValue(table, i, step, end, code) {
- do {
- end -= step;
- table[i + end] = new HuffmanCode(code.bits, code.value);
- } while (end > 0);
- }
-
- /* Returns the table width of the next 2nd level table. count is the histogram
- of bit lengths for the remaining symbols, len is the code length of the next
- processed symbol */
- function NextTableBitSize(count, len, root_bits) {
- var left = 1 << (len - root_bits);
- while (len < MAX_LENGTH) {
- left -= count[len];
- if (left <= 0) break;
- ++len;
- left <<= 1;
- }
- return len - root_bits;
- }
-
- exports.BrotliBuildHuffmanTable = function(root_table, table, root_bits, code_lengths, code_lengths_size) {
- var start_table = table;
- var code; /* current table entry */
- var len; /* current code length */
- var symbol; /* symbol index in original or sorted table */
- var key; /* reversed prefix code */
- var step; /* step size to replicate values in current table */
- var low; /* low bits for current root entry */
- var mask; /* mask for low bits */
- var table_bits; /* key length of current table */
- var table_size; /* size of current table */
- var total_size; /* sum of root table size and 2nd level table sizes */
- var sorted; /* symbols sorted by code length */
- var count = new Int32Array(MAX_LENGTH + 1); /* number of codes of each length */
- var offset = new Int32Array(MAX_LENGTH + 1); /* offsets in sorted table for each length */
-
- sorted = new Int32Array(code_lengths_size);
-
- /* build histogram of code lengths */
- for (symbol = 0; symbol < code_lengths_size; symbol++) {
- count[code_lengths[symbol]]++;
- }
-
- /* generate offsets into sorted symbol table by code length */
- offset[1] = 0;
- for (len = 1; len < MAX_LENGTH; len++) {
- offset[len + 1] = offset[len] + count[len];
- }
-
- /* sort symbols by length, by symbol order within each length */
- for (symbol = 0; symbol < code_lengths_size; symbol++) {
- if (code_lengths[symbol] !== 0) {
- sorted[offset[code_lengths[symbol]]++] = symbol;
- }
- }
-
- table_bits = root_bits;
- table_size = 1 << table_bits;
- total_size = table_size;
-
- /* special case code with only one value */
- if (offset[MAX_LENGTH] === 1) {
- for (key = 0; key < total_size; ++key) {
- root_table[table + key] = new HuffmanCode(0, sorted[0] & 0xffff);
- }
-
- return total_size;
- }
-
- /* fill in root table */
- key = 0;
- symbol = 0;
- for (len = 1, step = 2; len <= root_bits; ++len, step <<= 1) {
- for (; count[len] > 0; --count[len]) {
- code = new HuffmanCode(len & 0xff, sorted[symbol++] & 0xffff);
- ReplicateValue(root_table, table + key, step, table_size, code);
- key = GetNextKey(key, len);
- }
- }
-
- /* fill in 2nd level tables and add pointers to root table */
- mask = total_size - 1;
- low = -1;
- for (len = root_bits + 1, step = 2; len <= MAX_LENGTH; ++len, step <<= 1) {
- for (; count[len] > 0; --count[len]) {
- if ((key & mask) !== low) {
- table += table_size;
- table_bits = NextTableBitSize(count, len, root_bits);
- table_size = 1 << table_bits;
- total_size += table_size;
- low = key & mask;
- root_table[start_table + low] = new HuffmanCode((table_bits + root_bits) & 0xff, ((table - start_table) - low) & 0xffff);
- }
- code = new HuffmanCode((len - root_bits) & 0xff, sorted[symbol++] & 0xffff);
- ReplicateValue(root_table, table + (key >> root_bits), step, table_size, code);
- key = GetNextKey(key, len);
- }
- }
-
- return total_size;
- }
|