lzms_decompress.c: move first queue update into rep case

[wimlib] / src / lzms_decompress.c

diff --git a/src/lzms_decompress.c b/src/lzms_decompress.c
index 7f82803c9339b61d06da48756b45767ae12fa30d..0bce40871eee8e6de081649fd75c12e58332e3e5 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;
@@ -742,23 +737,23 @@ lzms_decode_items(struct lzms_decompressor * const restrict d,
                        u32 offset;
                        u32 length;
 
-                       if (d->pending_lz_offset != 0 &&
-                           out_next != d->lz_offset_still_pending)
-                       {
-                               BUILD_BUG_ON(LZMS_NUM_LZ_REPS != 3);
-                               d->recent_lz_offsets[3] = d->recent_lz_offsets[2];
-                               d->recent_lz_offsets[2] = d->recent_lz_offsets[1];
-                               d->recent_lz_offsets[1] = d->recent_lz_offsets[0];
-                               d->recent_lz_offsets[0] = d->pending_lz_offset;
-                               d->pending_lz_offset = 0;
-                       }
-
                        if (!lzms_decode_lz_bit(d)) {
                                /* Explicit offset  */
                                offset = lzms_decode_lz_offset(d);
                        } else {
                                /* Repeat offset  */
 
+                               if (d->pending_lz_offset != 0 &&
+                                   out_next != d->lz_offset_still_pending)
+                               {
+                                       BUILD_BUG_ON(LZMS_NUM_LZ_REPS != 3);
+                                       d->recent_lz_offsets[3] = d->recent_lz_offsets[2];
+                                       d->recent_lz_offsets[2] = d->recent_lz_offsets[1];
+                                       d->recent_lz_offsets[1] = d->recent_lz_offsets[0];
+                                       d->recent_lz_offsets[0] = d->pending_lz_offset;
+                                       d->pending_lz_offset = 0;
+                               }
+
                                BUILD_BUG_ON(LZMS_NUM_LZ_REPS != 3);
                                if (!lzms_decode_lz_rep_bit(d, 0)) {
                                        offset = d->recent_lz_offsets[0];
@@ -807,17 +802,6 @@ lzms_decode_items(struct lzms_decompressor * const restrict d,
                        const u8 *matchptr;
                        u32 length;
 
-                       if (d->pending_delta_pair != 0 &&
-                           out_next != d->delta_pair_still_pending)
-                       {
-                               BUILD_BUG_ON(LZMS_NUM_DELTA_REPS != 3);
-                               d->recent_delta_pairs[3] = d->recent_delta_pairs[2];
-                               d->recent_delta_pairs[2] = d->recent_delta_pairs[1];
-                               d->recent_delta_pairs[1] = d->recent_delta_pairs[0];
-                               d->recent_delta_pairs[0] = d->pending_delta_pair;
-                               d->pending_delta_pair = 0;
-                       }
-
                        if (!lzms_decode_delta_bit(d)) {
                                /* Explicit offset  */
                                power = lzms_decode_delta_power(d);
@@ -826,6 +810,17 @@ lzms_decode_items(struct lzms_decompressor * const restrict d,
                                /* Repeat offset  */
                                u64 val;
 
+                               if (d->pending_delta_pair != 0 &&
+                                   out_next != d->delta_pair_still_pending)
+                               {
+                                       BUILD_BUG_ON(LZMS_NUM_DELTA_REPS != 3);
+                                       d->recent_delta_pairs[3] = d->recent_delta_pairs[2];
+                                       d->recent_delta_pairs[2] = d->recent_delta_pairs[1];
+                                       d->recent_delta_pairs[1] = d->recent_delta_pairs[0];
+                                       d->recent_delta_pairs[0] = d->pending_delta_pair;
+                                       d->pending_delta_pair = 0;
+                               }
+
                                BUILD_BUG_ON(LZMS_NUM_DELTA_REPS != 3);
                                if (!lzms_decode_delta_rep_bit(d, 0)) {
                                        val = d->recent_delta_pairs[0];
@@ -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);