1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
pub fn lengths_to_symbols(lengths: &[u32], max_bits: u32) -> Vec<u32> {
let mut bl_count = vec![0; (max_bits + 1) as usize];
let mut next_code = vec![0; (max_bits + 1) as usize];
let n = lengths.len();
let mut symbols = vec![0; n];
for &length in lengths.iter() {
assert!(length <= max_bits);
bl_count[length as usize] += 1;
}
let mut code = 0;
bl_count[0] = 0;
for bits in 1..(max_bits + 1) {
code = (code + bl_count[(bits - 1) as usize]) << 1;
next_code[bits as usize] = code;
}
for i in 0..n {
let len = lengths[i] as usize;
if len != 0 {
symbols[i] = next_code[len];
next_code[len] += 1;
}
}
symbols
}