lzms_decompress.c: make 'lens' and 'decode_table' share memory

[wimlib] / src / lzms_decompress.c

diff --git a/src/lzms_decompress.c b/src/lzms_decompress.c
index 7f82803c9339b61d06da48756b45767ae12fa30d..b0d0670ea4a0a8a82d39ae2543d2c3b9789549f3 100644 (file)
--- a/src/lzms_decompress.c
+++ b/src/lzms_decompress.c
@@ -307,7 +307,6 @@ struct lzms_huffman_rebuild_info {
        unsigned num_syms;
        unsigned rebuild_freq;
        u32 *codewords;
-       u8 *lens;
        u32 *freqs;
        u16 *decode_table;
        unsigned table_bits;
@@ -386,7 +385,6 @@ struct lzms_decompressor {
        struct lzms_huffman_rebuild_info delta_power_rebuild_info;
 
        u32 codewords[LZMS_MAX_NUM_SYMS];
-       u8 lens[LZMS_MAX_NUM_SYMS];
 
        }; // struct
 
@@ -430,9 +428,7 @@ lzms_ensure_bits(struct lzms_input_bitstream *is, unsigned num_bits)
 static inline bitbuf_t
 lzms_peek_bits(struct lzms_input_bitstream *is, unsigned num_bits)
 {
-       if (unlikely(num_bits == 0))
-               return 0;
-       return is->bitbuf >> (sizeof(is->bitbuf) * 8 - num_bits);
+       return (is->bitbuf >> 1) >> (sizeof(is->bitbuf) * 8 - num_bits - 1);
 }
 
 /* Remove @num_bits bits from the bitbuffer variable.  */
@@ -584,13 +580,13 @@ lzms_build_huffman_code(struct lzms_huffman_rebuild_info *rebuild_info)
        make_canonical_huffman_code(rebuild_info->num_syms,
                                    LZMS_MAX_CODEWORD_LENGTH,
                                    rebuild_info->freqs,
-                                   rebuild_info->lens,
+                                   (u8 *)rebuild_info->decode_table,
                                    rebuild_info->codewords);
 
        make_huffman_decode_table(rebuild_info->decode_table,
                                  rebuild_info->num_syms,
                                  rebuild_info->table_bits,
-                                 rebuild_info->lens,
+                                 (u8 *)rebuild_info->decode_table,
                                  LZMS_MAX_CODEWORD_LENGTH);
 
        rebuild_info->num_syms_until_rebuild = rebuild_info->rebuild_freq;
@@ -599,13 +595,12 @@ lzms_build_huffman_code(struct lzms_huffman_rebuild_info *rebuild_info)
 static void
 lzms_init_huffman_code(struct lzms_huffman_rebuild_info *rebuild_info,
                       unsigned num_syms, unsigned rebuild_freq,
-                      u32 *codewords, u8 *lens, u32 *freqs,
+                      u32 *codewords, u32 *freqs,
                       u16 *decode_table, unsigned table_bits)
 {
        rebuild_info->num_syms = num_syms;
        rebuild_info->rebuild_freq = rebuild_freq;
        rebuild_info->codewords = codewords;
-       rebuild_info->lens = lens;
        rebuild_info->freqs = freqs;
        rebuild_info->decode_table = decode_table;
        rebuild_info->table_bits = table_bits;
@@ -936,7 +931,6 @@ lzms_init_decompressor(struct lzms_decompressor *d, const void *in,
                               LZMS_NUM_LITERAL_SYMS,
                               LZMS_LITERAL_CODE_REBUILD_FREQ,
                               d->codewords,
-                              d->lens,
                               d->literal_freqs,
                               d->literal_decode_table,
                               LZMS_LITERAL_TABLEBITS);
@@ -945,7 +939,6 @@ lzms_init_decompressor(struct lzms_decompressor *d, const void *in,
                               num_offset_slots,
                               LZMS_LZ_OFFSET_CODE_REBUILD_FREQ,
                               d->codewords,
-                              d->lens,
                               d->lz_offset_freqs,
                               d->lz_offset_decode_table,
                               LZMS_LZ_OFFSET_TABLEBITS);
@@ -954,7 +947,6 @@ lzms_init_decompressor(struct lzms_decompressor *d, const void *in,
                               LZMS_NUM_LENGTH_SYMS,
                               LZMS_LENGTH_CODE_REBUILD_FREQ,
                               d->codewords,
-                              d->lens,
                               d->length_freqs,
                               d->length_decode_table,
                               LZMS_LENGTH_TABLEBITS);
@@ -963,7 +955,6 @@ lzms_init_decompressor(struct lzms_decompressor *d, const void *in,
                               num_offset_slots,
                               LZMS_DELTA_OFFSET_CODE_REBUILD_FREQ,
                               d->codewords,
-                              d->lens,
                               d->delta_offset_freqs,
                               d->delta_offset_decode_table,
                               LZMS_DELTA_OFFSET_TABLEBITS);
@@ -972,7 +963,6 @@ lzms_init_decompressor(struct lzms_decompressor *d, const void *in,
                               LZMS_NUM_DELTA_POWER_SYMS,
                               LZMS_DELTA_POWER_CODE_REBUILD_FREQ,
                               d->codewords,
-                              d->lens,
                               d->delta_power_freqs,
                               d->delta_power_decode_table,
                               LZMS_DELTA_POWER_TABLEBITS);