bits.hpp

// Copyright Take Vos 2020-2022.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at https://www.boost.org/LICENSE_1_0.txt)
 
#pragma once
 
#include "utility/module.hpp"
#include <span>
#include <cstddef>
 
namespace hi::inline v1 {
 
[[nodiscard]] inline bool get_bit(std::span<std::byte const> buffer, std::size_t &index) noexcept
{
    hilet byte_index = index >> 3;
    hilet bit_index = index & 7;
    ++index;
 
    hi_axiom(byte_index < buffer.size());
    return to_bool(std::bit_cast<uint8_t>(buffer[byte_index] >> bit_index) & 1);
}
 
[[nodiscard]] inline std::size_t get_bits(std::span<std::byte const> buffer, std::size_t &index, std::size_t length) noexcept
{
    hi_axiom(length <= sizeof(std::size_t) * CHAR_BIT);
 
    auto value = 0;
 
    auto todo = length;
    auto done = 0_uz;
    while (todo) {
        hilet byte_index = index >> 3;
        hilet bit_index = index & 7;
        hi_axiom(byte_index < buffer.size());
 
        hilet available_bits = 8 - bit_index;
        hilet nr_bits = std::min(available_bits, todo);
 
        hilet mask = (1 << nr_bits) - 1;
 
        hilet tmp = static_cast<int>(buffer[byte_index] >> bit_index) & mask;
        value |= tmp << done;
 
        todo -= nr_bits;
        done += nr_bits;
        index += nr_bits;
    }
 
    return value;
}
 
} // namespace hi::inline v1