LZMS: share 'struct lzms_probabilities' between compressor and decompressor

[wimlib] / src / lzms_common.c

diff --git a/src/lzms_common.c b/src/lzms_common.c
index b23a2785fff907dad62f92a749cda041957b7b9c..9d9d7b2ae88bb5d4c116dd82981349c5a51f5733 100644 (file)
--- a/src/lzms_common.c
+++ b/src/lzms_common.c
@@ -351,9 +351,13 @@ lzms_get_num_offset_slots(size_t uncompressed_size)
 }
 
 void
-lzms_init_probability_entries(struct lzms_probability_entry *entries, size_t count)
+lzms_init_probabilities(struct lzms_probabilites *probs)
 {
-       for (size_t i = 0; i < count; i++) {
+       struct lzms_probability_entry *entries =
+               (struct lzms_probability_entry *)probs;
+       size_t num_entries = sizeof(struct lzms_probabilites) /
+                            sizeof(struct lzms_probability_entry);
+       for (size_t i = 0; i < num_entries; i++) {
                entries[i].num_recent_zero_bits = LZMS_INITIAL_PROBABILITY;
                entries[i].recent_bits = LZMS_INITIAL_RECENT_BITS;
        }