vortex-buffer 0.54.0

A byte buffer implementation for Vortex
Documentation 
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
33
34
35
36
37
38
39
40
41
42

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

//! Packed bitmaps that can be used to store boolean values.
//!
//! This module provides a wrapper on top of the `Buffer` type to store mutable and immutable
//! bitsets. The bitsets are stored in little-endian order, meaning that the least significant bit
//! of the first byte is the first bit in the bitset.
#[cfg(feature = "arrow")]
mod arrow;
mod buf;
mod buf_mut;
mod macros;
mod ops;

pub use arrow_buffer::bit_chunk_iterator::{
    BitChunkIterator, BitChunks, UnalignedBitChunk, UnalignedBitChunkIterator,
};
pub use arrow_buffer::bit_iterator::{BitIndexIterator, BitIterator, BitSliceIterator};
pub use buf::*;
pub use buf_mut::*;

/// Get the bit value at `index` out of `buf` without bounds checking
///
/// # Safety
/// `index` must be between 0 and length of `buf`
#[inline(always)]
unsafe fn get_bit_unchecked(buf: *const u8, index: usize) -> bool {
    (unsafe { *buf.add(index / 8) } & (1 << (index % 8))) != 0
}

/// Set the bit value at `index` in `buf` without bounds checking
#[inline(always)]
unsafe fn set_bit_unchecked(buf: *mut u8, index: usize) {
    unsafe { *buf.add(index / 8) |= 1 << (index % 8) };
}

/// Unset the bit value at `index` in `buf` without bounds checking
#[inline(always)]
unsafe fn unset_bit_unchecked(buf: *mut u8, index: usize) {
    unsafe { *buf.add(index / 8) &= !(1 << (index % 8)) };
}