zstd-ruby 1.4.1.0 → 1.5.0.0

Sign up to get free protection for your applications and to get access to all the features.
Files changed (96) hide show
  1. checksums.yaml +4 -4
  2. data/.github/dependabot.yml +8 -0
  3. data/.github/workflows/ruby.yml +35 -0
  4. data/README.md +2 -2
  5. data/ext/zstdruby/libzstd/BUCK +5 -7
  6. data/ext/zstdruby/libzstd/Makefile +304 -113
  7. data/ext/zstdruby/libzstd/README.md +83 -20
  8. data/ext/zstdruby/libzstd/common/bitstream.h +59 -51
  9. data/ext/zstdruby/libzstd/common/compiler.h +150 -8
  10. data/ext/zstdruby/libzstd/common/cpu.h +1 -3
  11. data/ext/zstdruby/libzstd/common/debug.c +11 -31
  12. data/ext/zstdruby/libzstd/common/debug.h +22 -49
  13. data/ext/zstdruby/libzstd/common/entropy_common.c +201 -75
  14. data/ext/zstdruby/libzstd/common/error_private.c +3 -1
  15. data/ext/zstdruby/libzstd/common/error_private.h +8 -4
  16. data/ext/zstdruby/libzstd/common/fse.h +50 -42
  17. data/ext/zstdruby/libzstd/common/fse_decompress.c +149 -55
  18. data/ext/zstdruby/libzstd/common/huf.h +43 -39
  19. data/ext/zstdruby/libzstd/common/mem.h +69 -25
  20. data/ext/zstdruby/libzstd/common/pool.c +30 -20
  21. data/ext/zstdruby/libzstd/common/pool.h +3 -3
  22. data/ext/zstdruby/libzstd/common/threading.c +51 -4
  23. data/ext/zstdruby/libzstd/common/threading.h +36 -4
  24. data/ext/zstdruby/libzstd/common/xxhash.c +40 -92
  25. data/ext/zstdruby/libzstd/common/xxhash.h +12 -32
  26. data/ext/zstdruby/libzstd/common/zstd_common.c +10 -10
  27. data/ext/zstdruby/libzstd/common/zstd_deps.h +111 -0
  28. data/ext/zstdruby/libzstd/common/zstd_internal.h +230 -111
  29. data/ext/zstdruby/libzstd/common/zstd_trace.h +154 -0
  30. data/ext/zstdruby/libzstd/compress/fse_compress.c +47 -63
  31. data/ext/zstdruby/libzstd/compress/hist.c +41 -63
  32. data/ext/zstdruby/libzstd/compress/hist.h +13 -33
  33. data/ext/zstdruby/libzstd/compress/huf_compress.c +332 -193
  34. data/ext/zstdruby/libzstd/compress/zstd_compress.c +3614 -1696
  35. data/ext/zstdruby/libzstd/compress/zstd_compress_internal.h +546 -86
  36. data/ext/zstdruby/libzstd/compress/zstd_compress_literals.c +158 -0
  37. data/ext/zstdruby/libzstd/compress/zstd_compress_literals.h +29 -0
  38. data/ext/zstdruby/libzstd/compress/zstd_compress_sequences.c +441 -0
  39. data/ext/zstdruby/libzstd/compress/zstd_compress_sequences.h +54 -0
  40. data/ext/zstdruby/libzstd/compress/zstd_compress_superblock.c +572 -0
  41. data/ext/zstdruby/libzstd/compress/zstd_compress_superblock.h +32 -0
  42. data/ext/zstdruby/libzstd/compress/zstd_cwksp.h +662 -0
  43. data/ext/zstdruby/libzstd/compress/zstd_double_fast.c +43 -41
  44. data/ext/zstdruby/libzstd/compress/zstd_double_fast.h +2 -2
  45. data/ext/zstdruby/libzstd/compress/zstd_fast.c +85 -80
  46. data/ext/zstdruby/libzstd/compress/zstd_fast.h +2 -2
  47. data/ext/zstdruby/libzstd/compress/zstd_lazy.c +1184 -111
  48. data/ext/zstdruby/libzstd/compress/zstd_lazy.h +59 -1
  49. data/ext/zstdruby/libzstd/compress/zstd_ldm.c +333 -208
  50. data/ext/zstdruby/libzstd/compress/zstd_ldm.h +15 -3
  51. data/ext/zstdruby/libzstd/compress/zstd_ldm_geartab.h +103 -0
  52. data/ext/zstdruby/libzstd/compress/zstd_opt.c +228 -129
  53. data/ext/zstdruby/libzstd/compress/zstd_opt.h +1 -1
  54. data/ext/zstdruby/libzstd/compress/zstdmt_compress.c +151 -440
  55. data/ext/zstdruby/libzstd/compress/zstdmt_compress.h +32 -114
  56. data/ext/zstdruby/libzstd/decompress/huf_decompress.c +395 -276
  57. data/ext/zstdruby/libzstd/decompress/zstd_ddict.c +20 -16
  58. data/ext/zstdruby/libzstd/decompress/zstd_ddict.h +3 -3
  59. data/ext/zstdruby/libzstd/decompress/zstd_decompress.c +630 -231
  60. data/ext/zstdruby/libzstd/decompress/zstd_decompress_block.c +606 -380
  61. data/ext/zstdruby/libzstd/decompress/zstd_decompress_block.h +8 -5
  62. data/ext/zstdruby/libzstd/decompress/zstd_decompress_internal.h +39 -9
  63. data/ext/zstdruby/libzstd/deprecated/zbuff.h +9 -8
  64. data/ext/zstdruby/libzstd/deprecated/zbuff_common.c +2 -2
  65. data/ext/zstdruby/libzstd/deprecated/zbuff_compress.c +1 -1
  66. data/ext/zstdruby/libzstd/deprecated/zbuff_decompress.c +1 -1
  67. data/ext/zstdruby/libzstd/dictBuilder/cover.c +55 -46
  68. data/ext/zstdruby/libzstd/dictBuilder/cover.h +20 -9
  69. data/ext/zstdruby/libzstd/dictBuilder/divsufsort.c +1 -1
  70. data/ext/zstdruby/libzstd/dictBuilder/fastcover.c +43 -31
  71. data/ext/zstdruby/libzstd/dictBuilder/zdict.c +53 -30
  72. data/ext/zstdruby/libzstd/dll/example/Makefile +2 -1
  73. data/ext/zstdruby/libzstd/dll/example/README.md +16 -22
  74. data/ext/zstdruby/libzstd/legacy/zstd_legacy.h +4 -4
  75. data/ext/zstdruby/libzstd/legacy/zstd_v01.c +24 -14
  76. data/ext/zstdruby/libzstd/legacy/zstd_v01.h +1 -1
  77. data/ext/zstdruby/libzstd/legacy/zstd_v02.c +17 -8
  78. data/ext/zstdruby/libzstd/legacy/zstd_v02.h +1 -1
  79. data/ext/zstdruby/libzstd/legacy/zstd_v03.c +17 -8
  80. data/ext/zstdruby/libzstd/legacy/zstd_v03.h +1 -1
  81. data/ext/zstdruby/libzstd/legacy/zstd_v04.c +25 -11
  82. data/ext/zstdruby/libzstd/legacy/zstd_v04.h +1 -1
  83. data/ext/zstdruby/libzstd/legacy/zstd_v05.c +43 -32
  84. data/ext/zstdruby/libzstd/legacy/zstd_v05.h +2 -2
  85. data/ext/zstdruby/libzstd/legacy/zstd_v06.c +27 -19
  86. data/ext/zstdruby/libzstd/legacy/zstd_v06.h +1 -1
  87. data/ext/zstdruby/libzstd/legacy/zstd_v07.c +32 -20
  88. data/ext/zstdruby/libzstd/legacy/zstd_v07.h +1 -1
  89. data/ext/zstdruby/libzstd/libzstd.pc.in +2 -1
  90. data/ext/zstdruby/libzstd/{dictBuilder/zdict.h → zdict.h} +201 -31
  91. data/ext/zstdruby/libzstd/zstd.h +740 -153
  92. data/ext/zstdruby/libzstd/{common/zstd_errors.h → zstd_errors.h} +3 -1
  93. data/lib/zstd-ruby/version.rb +1 -1
  94. data/zstd-ruby.gemspec +1 -1
  95. metadata +21 -10
  96. data/.travis.yml +0 -14
@@ -0,0 +1,158 @@
1
+ /*
2
+ * Copyright (c) Yann Collet, Facebook, Inc.
3
+ * All rights reserved.
4
+ *
5
+ * This source code is licensed under both the BSD-style license (found in the
6
+ * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7
+ * in the COPYING file in the root directory of this source tree).
8
+ * You may select, at your option, one of the above-listed licenses.
9
+ */
10
+
11
+ /*-*************************************
12
+ * Dependencies
13
+ ***************************************/
14
+ #include "zstd_compress_literals.h"
15
+
16
+ size_t ZSTD_noCompressLiterals (void* dst, size_t dstCapacity, const void* src, size_t srcSize)
17
+ {
18
+ BYTE* const ostart = (BYTE*)dst;
19
+ U32 const flSize = 1 + (srcSize>31) + (srcSize>4095);
20
+
21
+ RETURN_ERROR_IF(srcSize + flSize > dstCapacity, dstSize_tooSmall, "");
22
+
23
+ switch(flSize)
24
+ {
25
+ case 1: /* 2 - 1 - 5 */
26
+ ostart[0] = (BYTE)((U32)set_basic + (srcSize<<3));
27
+ break;
28
+ case 2: /* 2 - 2 - 12 */
29
+ MEM_writeLE16(ostart, (U16)((U32)set_basic + (1<<2) + (srcSize<<4)));
30
+ break;
31
+ case 3: /* 2 - 2 - 20 */
32
+ MEM_writeLE32(ostart, (U32)((U32)set_basic + (3<<2) + (srcSize<<4)));
33
+ break;
34
+ default: /* not necessary : flSize is {1,2,3} */
35
+ assert(0);
36
+ }
37
+
38
+ ZSTD_memcpy(ostart + flSize, src, srcSize);
39
+ DEBUGLOG(5, "Raw literals: %u -> %u", (U32)srcSize, (U32)(srcSize + flSize));
40
+ return srcSize + flSize;
41
+ }
42
+
43
+ size_t ZSTD_compressRleLiteralsBlock (void* dst, size_t dstCapacity, const void* src, size_t srcSize)
44
+ {
45
+ BYTE* const ostart = (BYTE*)dst;
46
+ U32 const flSize = 1 + (srcSize>31) + (srcSize>4095);
47
+
48
+ (void)dstCapacity; /* dstCapacity already guaranteed to be >=4, hence large enough */
49
+
50
+ switch(flSize)
51
+ {
52
+ case 1: /* 2 - 1 - 5 */
53
+ ostart[0] = (BYTE)((U32)set_rle + (srcSize<<3));
54
+ break;
55
+ case 2: /* 2 - 2 - 12 */
56
+ MEM_writeLE16(ostart, (U16)((U32)set_rle + (1<<2) + (srcSize<<4)));
57
+ break;
58
+ case 3: /* 2 - 2 - 20 */
59
+ MEM_writeLE32(ostart, (U32)((U32)set_rle + (3<<2) + (srcSize<<4)));
60
+ break;
61
+ default: /* not necessary : flSize is {1,2,3} */
62
+ assert(0);
63
+ }
64
+
65
+ ostart[flSize] = *(const BYTE*)src;
66
+ DEBUGLOG(5, "RLE literals: %u -> %u", (U32)srcSize, (U32)flSize + 1);
67
+ return flSize+1;
68
+ }
69
+
70
+ size_t ZSTD_compressLiterals (ZSTD_hufCTables_t const* prevHuf,
71
+ ZSTD_hufCTables_t* nextHuf,
72
+ ZSTD_strategy strategy, int disableLiteralCompression,
73
+ void* dst, size_t dstCapacity,
74
+ const void* src, size_t srcSize,
75
+ void* entropyWorkspace, size_t entropyWorkspaceSize,
76
+ const int bmi2)
77
+ {
78
+ size_t const minGain = ZSTD_minGain(srcSize, strategy);
79
+ size_t const lhSize = 3 + (srcSize >= 1 KB) + (srcSize >= 16 KB);
80
+ BYTE* const ostart = (BYTE*)dst;
81
+ U32 singleStream = srcSize < 256;
82
+ symbolEncodingType_e hType = set_compressed;
83
+ size_t cLitSize;
84
+
85
+ DEBUGLOG(5,"ZSTD_compressLiterals (disableLiteralCompression=%i srcSize=%u)",
86
+ disableLiteralCompression, (U32)srcSize);
87
+
88
+ /* Prepare nextEntropy assuming reusing the existing table */
89
+ ZSTD_memcpy(nextHuf, prevHuf, sizeof(*prevHuf));
90
+
91
+ if (disableLiteralCompression)
92
+ return ZSTD_noCompressLiterals(dst, dstCapacity, src, srcSize);
93
+
94
+ /* small ? don't even attempt compression (speed opt) */
95
+ # define COMPRESS_LITERALS_SIZE_MIN 63
96
+ { size_t const minLitSize = (prevHuf->repeatMode == HUF_repeat_valid) ? 6 : COMPRESS_LITERALS_SIZE_MIN;
97
+ if (srcSize <= minLitSize) return ZSTD_noCompressLiterals(dst, dstCapacity, src, srcSize);
98
+ }
99
+
100
+ RETURN_ERROR_IF(dstCapacity < lhSize+1, dstSize_tooSmall, "not enough space for compression");
101
+ { HUF_repeat repeat = prevHuf->repeatMode;
102
+ int const preferRepeat = strategy < ZSTD_lazy ? srcSize <= 1024 : 0;
103
+ if (repeat == HUF_repeat_valid && lhSize == 3) singleStream = 1;
104
+ cLitSize = singleStream ?
105
+ HUF_compress1X_repeat(
106
+ ostart+lhSize, dstCapacity-lhSize, src, srcSize,
107
+ HUF_SYMBOLVALUE_MAX, HUF_TABLELOG_DEFAULT, entropyWorkspace, entropyWorkspaceSize,
108
+ (HUF_CElt*)nextHuf->CTable, &repeat, preferRepeat, bmi2) :
109
+ HUF_compress4X_repeat(
110
+ ostart+lhSize, dstCapacity-lhSize, src, srcSize,
111
+ HUF_SYMBOLVALUE_MAX, HUF_TABLELOG_DEFAULT, entropyWorkspace, entropyWorkspaceSize,
112
+ (HUF_CElt*)nextHuf->CTable, &repeat, preferRepeat, bmi2);
113
+ if (repeat != HUF_repeat_none) {
114
+ /* reused the existing table */
115
+ DEBUGLOG(5, "Reusing previous huffman table");
116
+ hType = set_repeat;
117
+ }
118
+ }
119
+
120
+ if ((cLitSize==0) || (cLitSize >= srcSize - minGain) || ERR_isError(cLitSize)) {
121
+ ZSTD_memcpy(nextHuf, prevHuf, sizeof(*prevHuf));
122
+ return ZSTD_noCompressLiterals(dst, dstCapacity, src, srcSize);
123
+ }
124
+ if (cLitSize==1) {
125
+ ZSTD_memcpy(nextHuf, prevHuf, sizeof(*prevHuf));
126
+ return ZSTD_compressRleLiteralsBlock(dst, dstCapacity, src, srcSize);
127
+ }
128
+
129
+ if (hType == set_compressed) {
130
+ /* using a newly constructed table */
131
+ nextHuf->repeatMode = HUF_repeat_check;
132
+ }
133
+
134
+ /* Build header */
135
+ switch(lhSize)
136
+ {
137
+ case 3: /* 2 - 2 - 10 - 10 */
138
+ { U32 const lhc = hType + ((!singleStream) << 2) + ((U32)srcSize<<4) + ((U32)cLitSize<<14);
139
+ MEM_writeLE24(ostart, lhc);
140
+ break;
141
+ }
142
+ case 4: /* 2 - 2 - 14 - 14 */
143
+ { U32 const lhc = hType + (2 << 2) + ((U32)srcSize<<4) + ((U32)cLitSize<<18);
144
+ MEM_writeLE32(ostart, lhc);
145
+ break;
146
+ }
147
+ case 5: /* 2 - 2 - 18 - 18 */
148
+ { U32 const lhc = hType + (3 << 2) + ((U32)srcSize<<4) + ((U32)cLitSize<<22);
149
+ MEM_writeLE32(ostart, lhc);
150
+ ostart[4] = (BYTE)(cLitSize >> 10);
151
+ break;
152
+ }
153
+ default: /* not possible : lhSize is {3,4,5} */
154
+ assert(0);
155
+ }
156
+ DEBUGLOG(5, "Compressed literals: %u -> %u", (U32)srcSize, (U32)(lhSize+cLitSize));
157
+ return lhSize+cLitSize;
158
+ }
@@ -0,0 +1,29 @@
1
+ /*
2
+ * Copyright (c) Yann Collet, Facebook, Inc.
3
+ * All rights reserved.
4
+ *
5
+ * This source code is licensed under both the BSD-style license (found in the
6
+ * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7
+ * in the COPYING file in the root directory of this source tree).
8
+ * You may select, at your option, one of the above-listed licenses.
9
+ */
10
+
11
+ #ifndef ZSTD_COMPRESS_LITERALS_H
12
+ #define ZSTD_COMPRESS_LITERALS_H
13
+
14
+ #include "zstd_compress_internal.h" /* ZSTD_hufCTables_t, ZSTD_minGain() */
15
+
16
+
17
+ size_t ZSTD_noCompressLiterals (void* dst, size_t dstCapacity, const void* src, size_t srcSize);
18
+
19
+ size_t ZSTD_compressRleLiteralsBlock (void* dst, size_t dstCapacity, const void* src, size_t srcSize);
20
+
21
+ size_t ZSTD_compressLiterals (ZSTD_hufCTables_t const* prevHuf,
22
+ ZSTD_hufCTables_t* nextHuf,
23
+ ZSTD_strategy strategy, int disableLiteralCompression,
24
+ void* dst, size_t dstCapacity,
25
+ const void* src, size_t srcSize,
26
+ void* entropyWorkspace, size_t entropyWorkspaceSize,
27
+ const int bmi2);
28
+
29
+ #endif /* ZSTD_COMPRESS_LITERALS_H */
@@ -0,0 +1,441 @@
1
+ /*
2
+ * Copyright (c) Yann Collet, Facebook, Inc.
3
+ * All rights reserved.
4
+ *
5
+ * This source code is licensed under both the BSD-style license (found in the
6
+ * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7
+ * in the COPYING file in the root directory of this source tree).
8
+ * You may select, at your option, one of the above-listed licenses.
9
+ */
10
+
11
+ /*-*************************************
12
+ * Dependencies
13
+ ***************************************/
14
+ #include "zstd_compress_sequences.h"
15
+
16
+ /**
17
+ * -log2(x / 256) lookup table for x in [0, 256).
18
+ * If x == 0: Return 0
19
+ * Else: Return floor(-log2(x / 256) * 256)
20
+ */
21
+ static unsigned const kInverseProbabilityLog256[256] = {
22
+ 0, 2048, 1792, 1642, 1536, 1453, 1386, 1329, 1280, 1236, 1197, 1162,
23
+ 1130, 1100, 1073, 1047, 1024, 1001, 980, 960, 941, 923, 906, 889,
24
+ 874, 859, 844, 830, 817, 804, 791, 779, 768, 756, 745, 734,
25
+ 724, 714, 704, 694, 685, 676, 667, 658, 650, 642, 633, 626,
26
+ 618, 610, 603, 595, 588, 581, 574, 567, 561, 554, 548, 542,
27
+ 535, 529, 523, 517, 512, 506, 500, 495, 489, 484, 478, 473,
28
+ 468, 463, 458, 453, 448, 443, 438, 434, 429, 424, 420, 415,
29
+ 411, 407, 402, 398, 394, 390, 386, 382, 377, 373, 370, 366,
30
+ 362, 358, 354, 350, 347, 343, 339, 336, 332, 329, 325, 322,
31
+ 318, 315, 311, 308, 305, 302, 298, 295, 292, 289, 286, 282,
32
+ 279, 276, 273, 270, 267, 264, 261, 258, 256, 253, 250, 247,
33
+ 244, 241, 239, 236, 233, 230, 228, 225, 222, 220, 217, 215,
34
+ 212, 209, 207, 204, 202, 199, 197, 194, 192, 190, 187, 185,
35
+ 182, 180, 178, 175, 173, 171, 168, 166, 164, 162, 159, 157,
36
+ 155, 153, 151, 149, 146, 144, 142, 140, 138, 136, 134, 132,
37
+ 130, 128, 126, 123, 121, 119, 117, 115, 114, 112, 110, 108,
38
+ 106, 104, 102, 100, 98, 96, 94, 93, 91, 89, 87, 85,
39
+ 83, 82, 80, 78, 76, 74, 73, 71, 69, 67, 66, 64,
40
+ 62, 61, 59, 57, 55, 54, 52, 50, 49, 47, 46, 44,
41
+ 42, 41, 39, 37, 36, 34, 33, 31, 30, 28, 26, 25,
42
+ 23, 22, 20, 19, 17, 16, 14, 13, 11, 10, 8, 7,
43
+ 5, 4, 2, 1,
44
+ };
45
+
46
+ static unsigned ZSTD_getFSEMaxSymbolValue(FSE_CTable const* ctable) {
47
+ void const* ptr = ctable;
48
+ U16 const* u16ptr = (U16 const*)ptr;
49
+ U32 const maxSymbolValue = MEM_read16(u16ptr + 1);
50
+ return maxSymbolValue;
51
+ }
52
+
53
+ /**
54
+ * Returns true if we should use ncount=-1 else we should
55
+ * use ncount=1 for low probability symbols instead.
56
+ */
57
+ static unsigned ZSTD_useLowProbCount(size_t const nbSeq)
58
+ {
59
+ /* Heuristic: This should cover most blocks <= 16K and
60
+ * start to fade out after 16K to about 32K depending on
61
+ * comprssibility.
62
+ */
63
+ return nbSeq >= 2048;
64
+ }
65
+
66
+ /**
67
+ * Returns the cost in bytes of encoding the normalized count header.
68
+ * Returns an error if any of the helper functions return an error.
69
+ */
70
+ static size_t ZSTD_NCountCost(unsigned const* count, unsigned const max,
71
+ size_t const nbSeq, unsigned const FSELog)
72
+ {
73
+ BYTE wksp[FSE_NCOUNTBOUND];
74
+ S16 norm[MaxSeq + 1];
75
+ const U32 tableLog = FSE_optimalTableLog(FSELog, nbSeq, max);
76
+ FORWARD_IF_ERROR(FSE_normalizeCount(norm, tableLog, count, nbSeq, max, ZSTD_useLowProbCount(nbSeq)), "");
77
+ return FSE_writeNCount(wksp, sizeof(wksp), norm, max, tableLog);
78
+ }
79
+
80
+ /**
81
+ * Returns the cost in bits of encoding the distribution described by count
82
+ * using the entropy bound.
83
+ */
84
+ static size_t ZSTD_entropyCost(unsigned const* count, unsigned const max, size_t const total)
85
+ {
86
+ unsigned cost = 0;
87
+ unsigned s;
88
+
89
+ assert(total > 0);
90
+ for (s = 0; s <= max; ++s) {
91
+ unsigned norm = (unsigned)((256 * count[s]) / total);
92
+ if (count[s] != 0 && norm == 0)
93
+ norm = 1;
94
+ assert(count[s] < total);
95
+ cost += count[s] * kInverseProbabilityLog256[norm];
96
+ }
97
+ return cost >> 8;
98
+ }
99
+
100
+ /**
101
+ * Returns the cost in bits of encoding the distribution in count using ctable.
102
+ * Returns an error if ctable cannot represent all the symbols in count.
103
+ */
104
+ size_t ZSTD_fseBitCost(
105
+ FSE_CTable const* ctable,
106
+ unsigned const* count,
107
+ unsigned const max)
108
+ {
109
+ unsigned const kAccuracyLog = 8;
110
+ size_t cost = 0;
111
+ unsigned s;
112
+ FSE_CState_t cstate;
113
+ FSE_initCState(&cstate, ctable);
114
+ if (ZSTD_getFSEMaxSymbolValue(ctable) < max) {
115
+ DEBUGLOG(5, "Repeat FSE_CTable has maxSymbolValue %u < %u",
116
+ ZSTD_getFSEMaxSymbolValue(ctable), max);
117
+ return ERROR(GENERIC);
118
+ }
119
+ for (s = 0; s <= max; ++s) {
120
+ unsigned const tableLog = cstate.stateLog;
121
+ unsigned const badCost = (tableLog + 1) << kAccuracyLog;
122
+ unsigned const bitCost = FSE_bitCost(cstate.symbolTT, tableLog, s, kAccuracyLog);
123
+ if (count[s] == 0)
124
+ continue;
125
+ if (bitCost >= badCost) {
126
+ DEBUGLOG(5, "Repeat FSE_CTable has Prob[%u] == 0", s);
127
+ return ERROR(GENERIC);
128
+ }
129
+ cost += (size_t)count[s] * bitCost;
130
+ }
131
+ return cost >> kAccuracyLog;
132
+ }
133
+
134
+ /**
135
+ * Returns the cost in bits of encoding the distribution in count using the
136
+ * table described by norm. The max symbol support by norm is assumed >= max.
137
+ * norm must be valid for every symbol with non-zero probability in count.
138
+ */
139
+ size_t ZSTD_crossEntropyCost(short const* norm, unsigned accuracyLog,
140
+ unsigned const* count, unsigned const max)
141
+ {
142
+ unsigned const shift = 8 - accuracyLog;
143
+ size_t cost = 0;
144
+ unsigned s;
145
+ assert(accuracyLog <= 8);
146
+ for (s = 0; s <= max; ++s) {
147
+ unsigned const normAcc = (norm[s] != -1) ? (unsigned)norm[s] : 1;
148
+ unsigned const norm256 = normAcc << shift;
149
+ assert(norm256 > 0);
150
+ assert(norm256 < 256);
151
+ cost += count[s] * kInverseProbabilityLog256[norm256];
152
+ }
153
+ return cost >> 8;
154
+ }
155
+
156
+ symbolEncodingType_e
157
+ ZSTD_selectEncodingType(
158
+ FSE_repeat* repeatMode, unsigned const* count, unsigned const max,
159
+ size_t const mostFrequent, size_t nbSeq, unsigned const FSELog,
160
+ FSE_CTable const* prevCTable,
161
+ short const* defaultNorm, U32 defaultNormLog,
162
+ ZSTD_defaultPolicy_e const isDefaultAllowed,
163
+ ZSTD_strategy const strategy)
164
+ {
165
+ ZSTD_STATIC_ASSERT(ZSTD_defaultDisallowed == 0 && ZSTD_defaultAllowed != 0);
166
+ if (mostFrequent == nbSeq) {
167
+ *repeatMode = FSE_repeat_none;
168
+ if (isDefaultAllowed && nbSeq <= 2) {
169
+ /* Prefer set_basic over set_rle when there are 2 or less symbols,
170
+ * since RLE uses 1 byte, but set_basic uses 5-6 bits per symbol.
171
+ * If basic encoding isn't possible, always choose RLE.
172
+ */
173
+ DEBUGLOG(5, "Selected set_basic");
174
+ return set_basic;
175
+ }
176
+ DEBUGLOG(5, "Selected set_rle");
177
+ return set_rle;
178
+ }
179
+ if (strategy < ZSTD_lazy) {
180
+ if (isDefaultAllowed) {
181
+ size_t const staticFse_nbSeq_max = 1000;
182
+ size_t const mult = 10 - strategy;
183
+ size_t const baseLog = 3;
184
+ size_t const dynamicFse_nbSeq_min = (((size_t)1 << defaultNormLog) * mult) >> baseLog; /* 28-36 for offset, 56-72 for lengths */
185
+ assert(defaultNormLog >= 5 && defaultNormLog <= 6); /* xx_DEFAULTNORMLOG */
186
+ assert(mult <= 9 && mult >= 7);
187
+ if ( (*repeatMode == FSE_repeat_valid)
188
+ && (nbSeq < staticFse_nbSeq_max) ) {
189
+ DEBUGLOG(5, "Selected set_repeat");
190
+ return set_repeat;
191
+ }
192
+ if ( (nbSeq < dynamicFse_nbSeq_min)
193
+ || (mostFrequent < (nbSeq >> (defaultNormLog-1))) ) {
194
+ DEBUGLOG(5, "Selected set_basic");
195
+ /* The format allows default tables to be repeated, but it isn't useful.
196
+ * When using simple heuristics to select encoding type, we don't want
197
+ * to confuse these tables with dictionaries. When running more careful
198
+ * analysis, we don't need to waste time checking both repeating tables
199
+ * and default tables.
200
+ */
201
+ *repeatMode = FSE_repeat_none;
202
+ return set_basic;
203
+ }
204
+ }
205
+ } else {
206
+ size_t const basicCost = isDefaultAllowed ? ZSTD_crossEntropyCost(defaultNorm, defaultNormLog, count, max) : ERROR(GENERIC);
207
+ size_t const repeatCost = *repeatMode != FSE_repeat_none ? ZSTD_fseBitCost(prevCTable, count, max) : ERROR(GENERIC);
208
+ size_t const NCountCost = ZSTD_NCountCost(count, max, nbSeq, FSELog);
209
+ size_t const compressedCost = (NCountCost << 3) + ZSTD_entropyCost(count, max, nbSeq);
210
+
211
+ if (isDefaultAllowed) {
212
+ assert(!ZSTD_isError(basicCost));
213
+ assert(!(*repeatMode == FSE_repeat_valid && ZSTD_isError(repeatCost)));
214
+ }
215
+ assert(!ZSTD_isError(NCountCost));
216
+ assert(compressedCost < ERROR(maxCode));
217
+ DEBUGLOG(5, "Estimated bit costs: basic=%u\trepeat=%u\tcompressed=%u",
218
+ (unsigned)basicCost, (unsigned)repeatCost, (unsigned)compressedCost);
219
+ if (basicCost <= repeatCost && basicCost <= compressedCost) {
220
+ DEBUGLOG(5, "Selected set_basic");
221
+ assert(isDefaultAllowed);
222
+ *repeatMode = FSE_repeat_none;
223
+ return set_basic;
224
+ }
225
+ if (repeatCost <= compressedCost) {
226
+ DEBUGLOG(5, "Selected set_repeat");
227
+ assert(!ZSTD_isError(repeatCost));
228
+ return set_repeat;
229
+ }
230
+ assert(compressedCost < basicCost && compressedCost < repeatCost);
231
+ }
232
+ DEBUGLOG(5, "Selected set_compressed");
233
+ *repeatMode = FSE_repeat_check;
234
+ return set_compressed;
235
+ }
236
+
237
+ typedef struct {
238
+ S16 norm[MaxSeq + 1];
239
+ U32 wksp[FSE_BUILD_CTABLE_WORKSPACE_SIZE_U32(MaxSeq, MaxFSELog)];
240
+ } ZSTD_BuildCTableWksp;
241
+
242
+ size_t
243
+ ZSTD_buildCTable(void* dst, size_t dstCapacity,
244
+ FSE_CTable* nextCTable, U32 FSELog, symbolEncodingType_e type,
245
+ unsigned* count, U32 max,
246
+ const BYTE* codeTable, size_t nbSeq,
247
+ const S16* defaultNorm, U32 defaultNormLog, U32 defaultMax,
248
+ const FSE_CTable* prevCTable, size_t prevCTableSize,
249
+ void* entropyWorkspace, size_t entropyWorkspaceSize)
250
+ {
251
+ BYTE* op = (BYTE*)dst;
252
+ const BYTE* const oend = op + dstCapacity;
253
+ DEBUGLOG(6, "ZSTD_buildCTable (dstCapacity=%u)", (unsigned)dstCapacity);
254
+
255
+ switch (type) {
256
+ case set_rle:
257
+ FORWARD_IF_ERROR(FSE_buildCTable_rle(nextCTable, (BYTE)max), "");
258
+ RETURN_ERROR_IF(dstCapacity==0, dstSize_tooSmall, "not enough space");
259
+ *op = codeTable[0];
260
+ return 1;
261
+ case set_repeat:
262
+ ZSTD_memcpy(nextCTable, prevCTable, prevCTableSize);
263
+ return 0;
264
+ case set_basic:
265
+ FORWARD_IF_ERROR(FSE_buildCTable_wksp(nextCTable, defaultNorm, defaultMax, defaultNormLog, entropyWorkspace, entropyWorkspaceSize), ""); /* note : could be pre-calculated */
266
+ return 0;
267
+ case set_compressed: {
268
+ ZSTD_BuildCTableWksp* wksp = (ZSTD_BuildCTableWksp*)entropyWorkspace;
269
+ size_t nbSeq_1 = nbSeq;
270
+ const U32 tableLog = FSE_optimalTableLog(FSELog, nbSeq, max);
271
+ if (count[codeTable[nbSeq-1]] > 1) {
272
+ count[codeTable[nbSeq-1]]--;
273
+ nbSeq_1--;
274
+ }
275
+ assert(nbSeq_1 > 1);
276
+ assert(entropyWorkspaceSize >= sizeof(ZSTD_BuildCTableWksp));
277
+ (void)entropyWorkspaceSize;
278
+ FORWARD_IF_ERROR(FSE_normalizeCount(wksp->norm, tableLog, count, nbSeq_1, max, ZSTD_useLowProbCount(nbSeq_1)), "");
279
+ { size_t const NCountSize = FSE_writeNCount(op, oend - op, wksp->norm, max, tableLog); /* overflow protected */
280
+ FORWARD_IF_ERROR(NCountSize, "FSE_writeNCount failed");
281
+ FORWARD_IF_ERROR(FSE_buildCTable_wksp(nextCTable, wksp->norm, max, tableLog, wksp->wksp, sizeof(wksp->wksp)), "");
282
+ return NCountSize;
283
+ }
284
+ }
285
+ default: assert(0); RETURN_ERROR(GENERIC, "impossible to reach");
286
+ }
287
+ }
288
+
289
+ FORCE_INLINE_TEMPLATE size_t
290
+ ZSTD_encodeSequences_body(
291
+ void* dst, size_t dstCapacity,
292
+ FSE_CTable const* CTable_MatchLength, BYTE const* mlCodeTable,
293
+ FSE_CTable const* CTable_OffsetBits, BYTE const* ofCodeTable,
294
+ FSE_CTable const* CTable_LitLength, BYTE const* llCodeTable,
295
+ seqDef const* sequences, size_t nbSeq, int longOffsets)
296
+ {
297
+ BIT_CStream_t blockStream;
298
+ FSE_CState_t stateMatchLength;
299
+ FSE_CState_t stateOffsetBits;
300
+ FSE_CState_t stateLitLength;
301
+
302
+ RETURN_ERROR_IF(
303
+ ERR_isError(BIT_initCStream(&blockStream, dst, dstCapacity)),
304
+ dstSize_tooSmall, "not enough space remaining");
305
+ DEBUGLOG(6, "available space for bitstream : %i (dstCapacity=%u)",
306
+ (int)(blockStream.endPtr - blockStream.startPtr),
307
+ (unsigned)dstCapacity);
308
+
309
+ /* first symbols */
310
+ FSE_initCState2(&stateMatchLength, CTable_MatchLength, mlCodeTable[nbSeq-1]);
311
+ FSE_initCState2(&stateOffsetBits, CTable_OffsetBits, ofCodeTable[nbSeq-1]);
312
+ FSE_initCState2(&stateLitLength, CTable_LitLength, llCodeTable[nbSeq-1]);
313
+ BIT_addBits(&blockStream, sequences[nbSeq-1].litLength, LL_bits[llCodeTable[nbSeq-1]]);
314
+ if (MEM_32bits()) BIT_flushBits(&blockStream);
315
+ BIT_addBits(&blockStream, sequences[nbSeq-1].matchLength, ML_bits[mlCodeTable[nbSeq-1]]);
316
+ if (MEM_32bits()) BIT_flushBits(&blockStream);
317
+ if (longOffsets) {
318
+ U32 const ofBits = ofCodeTable[nbSeq-1];
319
+ unsigned const extraBits = ofBits - MIN(ofBits, STREAM_ACCUMULATOR_MIN-1);
320
+ if (extraBits) {
321
+ BIT_addBits(&blockStream, sequences[nbSeq-1].offset, extraBits);
322
+ BIT_flushBits(&blockStream);
323
+ }
324
+ BIT_addBits(&blockStream, sequences[nbSeq-1].offset >> extraBits,
325
+ ofBits - extraBits);
326
+ } else {
327
+ BIT_addBits(&blockStream, sequences[nbSeq-1].offset, ofCodeTable[nbSeq-1]);
328
+ }
329
+ BIT_flushBits(&blockStream);
330
+
331
+ { size_t n;
332
+ for (n=nbSeq-2 ; n<nbSeq ; n--) { /* intentional underflow */
333
+ BYTE const llCode = llCodeTable[n];
334
+ BYTE const ofCode = ofCodeTable[n];
335
+ BYTE const mlCode = mlCodeTable[n];
336
+ U32 const llBits = LL_bits[llCode];
337
+ U32 const ofBits = ofCode;
338
+ U32 const mlBits = ML_bits[mlCode];
339
+ DEBUGLOG(6, "encoding: litlen:%2u - matchlen:%2u - offCode:%7u",
340
+ (unsigned)sequences[n].litLength,
341
+ (unsigned)sequences[n].matchLength + MINMATCH,
342
+ (unsigned)sequences[n].offset);
343
+ /* 32b*/ /* 64b*/
344
+ /* (7)*/ /* (7)*/
345
+ FSE_encodeSymbol(&blockStream, &stateOffsetBits, ofCode); /* 15 */ /* 15 */
346
+ FSE_encodeSymbol(&blockStream, &stateMatchLength, mlCode); /* 24 */ /* 24 */
347
+ if (MEM_32bits()) BIT_flushBits(&blockStream); /* (7)*/
348
+ FSE_encodeSymbol(&blockStream, &stateLitLength, llCode); /* 16 */ /* 33 */
349
+ if (MEM_32bits() || (ofBits+mlBits+llBits >= 64-7-(LLFSELog+MLFSELog+OffFSELog)))
350
+ BIT_flushBits(&blockStream); /* (7)*/
351
+ BIT_addBits(&blockStream, sequences[n].litLength, llBits);
352
+ if (MEM_32bits() && ((llBits+mlBits)>24)) BIT_flushBits(&blockStream);
353
+ BIT_addBits(&blockStream, sequences[n].matchLength, mlBits);
354
+ if (MEM_32bits() || (ofBits+mlBits+llBits > 56)) BIT_flushBits(&blockStream);
355
+ if (longOffsets) {
356
+ unsigned const extraBits = ofBits - MIN(ofBits, STREAM_ACCUMULATOR_MIN-1);
357
+ if (extraBits) {
358
+ BIT_addBits(&blockStream, sequences[n].offset, extraBits);
359
+ BIT_flushBits(&blockStream); /* (7)*/
360
+ }
361
+ BIT_addBits(&blockStream, sequences[n].offset >> extraBits,
362
+ ofBits - extraBits); /* 31 */
363
+ } else {
364
+ BIT_addBits(&blockStream, sequences[n].offset, ofBits); /* 31 */
365
+ }
366
+ BIT_flushBits(&blockStream); /* (7)*/
367
+ DEBUGLOG(7, "remaining space : %i", (int)(blockStream.endPtr - blockStream.ptr));
368
+ } }
369
+
370
+ DEBUGLOG(6, "ZSTD_encodeSequences: flushing ML state with %u bits", stateMatchLength.stateLog);
371
+ FSE_flushCState(&blockStream, &stateMatchLength);
372
+ DEBUGLOG(6, "ZSTD_encodeSequences: flushing Off state with %u bits", stateOffsetBits.stateLog);
373
+ FSE_flushCState(&blockStream, &stateOffsetBits);
374
+ DEBUGLOG(6, "ZSTD_encodeSequences: flushing LL state with %u bits", stateLitLength.stateLog);
375
+ FSE_flushCState(&blockStream, &stateLitLength);
376
+
377
+ { size_t const streamSize = BIT_closeCStream(&blockStream);
378
+ RETURN_ERROR_IF(streamSize==0, dstSize_tooSmall, "not enough space");
379
+ return streamSize;
380
+ }
381
+ }
382
+
383
+ static size_t
384
+ ZSTD_encodeSequences_default(
385
+ void* dst, size_t dstCapacity,
386
+ FSE_CTable const* CTable_MatchLength, BYTE const* mlCodeTable,
387
+ FSE_CTable const* CTable_OffsetBits, BYTE const* ofCodeTable,
388
+ FSE_CTable const* CTable_LitLength, BYTE const* llCodeTable,
389
+ seqDef const* sequences, size_t nbSeq, int longOffsets)
390
+ {
391
+ return ZSTD_encodeSequences_body(dst, dstCapacity,
392
+ CTable_MatchLength, mlCodeTable,
393
+ CTable_OffsetBits, ofCodeTable,
394
+ CTable_LitLength, llCodeTable,
395
+ sequences, nbSeq, longOffsets);
396
+ }
397
+
398
+
399
+ #if DYNAMIC_BMI2
400
+
401
+ static TARGET_ATTRIBUTE("bmi2") size_t
402
+ ZSTD_encodeSequences_bmi2(
403
+ void* dst, size_t dstCapacity,
404
+ FSE_CTable const* CTable_MatchLength, BYTE const* mlCodeTable,
405
+ FSE_CTable const* CTable_OffsetBits, BYTE const* ofCodeTable,
406
+ FSE_CTable const* CTable_LitLength, BYTE const* llCodeTable,
407
+ seqDef const* sequences, size_t nbSeq, int longOffsets)
408
+ {
409
+ return ZSTD_encodeSequences_body(dst, dstCapacity,
410
+ CTable_MatchLength, mlCodeTable,
411
+ CTable_OffsetBits, ofCodeTable,
412
+ CTable_LitLength, llCodeTable,
413
+ sequences, nbSeq, longOffsets);
414
+ }
415
+
416
+ #endif
417
+
418
+ size_t ZSTD_encodeSequences(
419
+ void* dst, size_t dstCapacity,
420
+ FSE_CTable const* CTable_MatchLength, BYTE const* mlCodeTable,
421
+ FSE_CTable const* CTable_OffsetBits, BYTE const* ofCodeTable,
422
+ FSE_CTable const* CTable_LitLength, BYTE const* llCodeTable,
423
+ seqDef const* sequences, size_t nbSeq, int longOffsets, int bmi2)
424
+ {
425
+ DEBUGLOG(5, "ZSTD_encodeSequences: dstCapacity = %u", (unsigned)dstCapacity);
426
+ #if DYNAMIC_BMI2
427
+ if (bmi2) {
428
+ return ZSTD_encodeSequences_bmi2(dst, dstCapacity,
429
+ CTable_MatchLength, mlCodeTable,
430
+ CTable_OffsetBits, ofCodeTable,
431
+ CTable_LitLength, llCodeTable,
432
+ sequences, nbSeq, longOffsets);
433
+ }
434
+ #endif
435
+ (void)bmi2;
436
+ return ZSTD_encodeSequences_default(dst, dstCapacity,
437
+ CTable_MatchLength, mlCodeTable,
438
+ CTable_OffsetBits, ofCodeTable,
439
+ CTable_LitLength, llCodeTable,
440
+ sequences, nbSeq, longOffsets);
441
+ }