You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

51 lines
1.3 KiB
C++

// SPDX-License-Identifier: BSD-2-Clause
// Copyright CM4all GmbH
// author: Max Kellermann <mk@cm4all.com>
#pragma once
#include <concepts>
template<std::unsigned_integral T>
constexpr T
RoundUpToPowerOfTwo(T value) noexcept
{
if (value <= 0)
return 1;
--value;
for (unsigned bits = 1; bits < sizeof(T) * 8; bits <<= 1)
value |= value >> bits;
++value;
return value;
}
static_assert(RoundUpToPowerOfTwo(0U) == 1U);
static_assert(RoundUpToPowerOfTwo(2U) == 2U);
static_assert(RoundUpToPowerOfTwo(3U) == 4U);
static_assert(RoundUpToPowerOfTwo(4U) == 4U);
static_assert(RoundUpToPowerOfTwo(5U) == 8U);
static_assert(RoundUpToPowerOfTwo(0x7fffU) == 0x8000U);
static_assert(RoundUpToPowerOfTwo(0x7ffffU) == 0x80000U);
static_assert(RoundUpToPowerOfTwo(0x1000000000000000ULL) == 0x1000000000000000ULL);
static_assert(RoundUpToPowerOfTwo(0x1fffffffffffffffULL) == 0x2000000000000000ULL);
static_assert(RoundUpToPowerOfTwo(0x7fffffffffffffffULL) == 0x8000000000000000ULL);
static_assert(RoundUpToPowerOfTwo(0x8000000000000000ULL) == 0x8000000000000000ULL);
template<std::unsigned_integral T>
constexpr T
RoundUpToPowerOfTwo(T value, T power_of_two) noexcept
{
return ((value - 1) | (power_of_two - 1)) + 1;
}
template<std::unsigned_integral T>
constexpr T
RoundDownToPowerOfTwo(T value, T power_of_two) noexcept
{
return value & ~(power_of_two - 1);
}