This crate is a Rust port of Daniel Lemire's simdcomp C library.
It makes it possible to compress/decompress :
- sequence of small integers
- sequences of increasing integers
⭐ It is fast. Expect > 4 billions integers per seconds.
bitpacking
compiles on stable rust but require rust > 1.27 to compile.
Just add to your Cargo.toml
:
bitpacking = "0.5"
For some bitpacking flavor and for some platform, the bitpacking crate may benefit from some specific simd instruction set.
In this case, it will always ship an alternative scalar implementation and will fall back to the scalar implementation at runtime.
In other words, your do not need to configure anything. Your program will run correctly, and at the fastest speed available for your CPU.
Traditional compression schemes like LZ4 are not really suited to address this problem efficiently. Instead, there are different families of solutions to this problem.
One of the most straightforward and efficient ones is bitpacking
:
- Integers are first grouped into blocks of constant size (e.g.
128
when using the SSE2 implementation). - If not available implicitly, compute the minimum number of bits
b
that makes it possible to represent all the integers. In other words, the smallestb
such that all integers in the block are stricly smaller than 2b. - The bitpacked representation is then some variation of the concatenation of the integers restricted to their least significant
b
-bits.
For instance, assuming a block of 4
, when encoding 4, 9, 3, 2
. Assuming that the highest value in the block is 9, b = 4
. All values will then be encoded over 4 bits as follows.
original number | binary representation |
---|---|
4 | 0100 |
9 | 1001 |
3 | 0011 |
2 | 0010 |
... | ... |
As a result, each integer of this block will only require 4 bits.
BitPacker1x
, BitPacker4x
, and BitPacker8x
produce different formats,
and are incompatible one with another.
BitPacker4x
and BitPacker8x
are designed specifically to leverage SSE3
and AVX2
instructions respectively.
It will safely fall back at runtime to a scalar implementation of these format if these instruction sets are not available on the running CPU.
👌 I recommend using BitPacker4x
if you are in doubt.
BitPacker1x
is what you would expect from a bitpacker.
The integer representation over b
bits are simply concatenated one
after the other. One block must contain 32 integers
.
BitPacker4x
bits ordering works in layers of 4 integers. This gives an opportunity
to leverage SSE3
instructions to encode and decode the stream.
One block must contain 128 integers
.
BitPacker8x
bits ordering works in layers of 8 integers. This gives an opportunity
to leverage AVX2
instructions to encode and decode the stream.
One block must contain 256 integers
.
use bitpacking::{BitPacker4x, BitPacker};
fn main() {
let my_data: Vec<u32> = vec![7, 7, 7, 7, 11, 10, 15, 13, 6, 5, 3, 14, 5, 7,
15, 12, 1, 10, 8, 10, 12, 14, 13, 1, 10, 1, 1, 10, 4, 15, 12,
1, 2, 0, 8, 5, 14, 5, 2, 4, 1, 6, 14, 13, 5, 10, 10, 1, 6, 4,
1, 12, 1, 1, 5, 15, 15, 2, 8, 6, 4, 3, 10, 8, 8, 9, 2, 6, 10,
5, 7, 9, 0, 13, 15, 5, 13, 10, 0, 2, 10, 14, 5, 9, 12, 8, 5, 10,
8, 8, 10, 5, 13, 8, 11, 14, 7, 14, 4, 2, 9, 12, 14, 5, 15, 12, 0,
12, 13, 3, 13, 5, 4, 15, 9, 8, 9, 3, 3, 3, 1, 12, 0, 6, 11, 11, 12, 4];
// Detects if `SSE3` is available on the current computed
// and uses the best available implementation accordingly.
let bitpacker = BitPacker4x::new();
// Computes the number of bits used for each integer in the blocks.
// my_data is assumed to have a len of 128 for `BitPacker4x`.
let num_bits: u8 = bitpacker.num_bits(&my_data);
assert_eq!(num_bits, 4);
// The compressed array will take exactly `num_bits * BitPacker4x::BLOCK_LEN / 8`.
// But it is ok to have an output with a different len as long as it is larger
// than this.
let mut compressed = vec![0u8; 4 * BitPacker4x::BLOCK_LEN];
// Compress returns the len.
let compressed_len = bitpacker.compress(&my_data, &mut compressed[..], num_bits);
assert_eq!((num_bits as usize) * BitPacker4x::BLOCK_LEN / 8, compressed_len);
// Decompressing
let mut decompressed = vec![0u32; BitPacker4x::BLOCK_LEN];
bitpacker.decompress(&compressed[..compressed_len], &mut decompressed[..], num_bits);
assert_eq!(&my_data, &decompressed);
}
The following benchmarks have been run on one thread on my laptop's CPU: Intel(R) Core(TM) i5-8250U CPU @ 1.60GHz.
You can get accurate figures on your hardware by running the following command.
cargo bench
operation | throughput |
---|---|
compress | 1.4 billions int/s |
compress_delta | 1.0 billions int/s |
decompress | 1.8 billions int/s |
decompress_delta | 1.4 billions int/s |
operation | throughput |
---|---|
compress | 5.3 billions int/s |
compress_delta | 2.8 billions int/s |
decompress | 5.5 billions int/s |
decompress_delta | 5 billions int/s |
operation | throughput |
---|---|
compress | 7 billions int/s |
compress_delta | 600 millions int/s |
decompress | 6.5 billions int/s |
decompress_delta | 5.6 billions int/s |
- stream vbyte A Stream-VByte implementation
- mayda Another crate implementation the same algorithms.