diff options
Diffstat (limited to 'src/common/alignment.h')
| -rw-r--r-- | src/common/alignment.h | 29 |
1 files changed, 27 insertions, 2 deletions
diff --git a/src/common/alignment.h b/src/common/alignment.h index fa715d497..0057052af 100644 --- a/src/common/alignment.h +++ b/src/common/alignment.h | |||
| @@ -3,6 +3,7 @@ | |||
| 3 | 3 | ||
| 4 | #pragma once | 4 | #pragma once |
| 5 | 5 | ||
| 6 | #include <bit> | ||
| 6 | #include <cstddef> | 7 | #include <cstddef> |
| 7 | #include <new> | 8 | #include <new> |
| 8 | #include <type_traits> | 9 | #include <type_traits> |
| @@ -10,7 +11,7 @@ | |||
| 10 | namespace Common { | 11 | namespace Common { |
| 11 | 12 | ||
| 12 | template <typename T> | 13 | template <typename T> |
| 13 | requires std::is_unsigned_v<T> | 14 | requires std::is_integral_v<T> |
| 14 | [[nodiscard]] constexpr T AlignUp(T value, size_t size) { | 15 | [[nodiscard]] constexpr T AlignUp(T value, size_t size) { |
| 15 | auto mod{static_cast<T>(value % size)}; | 16 | auto mod{static_cast<T>(value % size)}; |
| 16 | value -= mod; | 17 | value -= mod; |
| @@ -24,7 +25,7 @@ template <typename T> | |||
| 24 | } | 25 | } |
| 25 | 26 | ||
| 26 | template <typename T> | 27 | template <typename T> |
| 27 | requires std::is_unsigned_v<T> | 28 | requires std::is_integral_v<T> |
| 28 | [[nodiscard]] constexpr T AlignDown(T value, size_t size) { | 29 | [[nodiscard]] constexpr T AlignDown(T value, size_t size) { |
| 29 | return static_cast<T>(value - value % size); | 30 | return static_cast<T>(value - value % size); |
| 30 | } | 31 | } |
| @@ -55,6 +56,30 @@ template <typename T, typename U> | |||
| 55 | return (x + (y - 1)) / y; | 56 | return (x + (y - 1)) / y; |
| 56 | } | 57 | } |
| 57 | 58 | ||
| 59 | template <typename T> | ||
| 60 | requires std::is_integral_v<T> | ||
| 61 | [[nodiscard]] constexpr T LeastSignificantOneBit(T x) { | ||
| 62 | return x & ~(x - 1); | ||
| 63 | } | ||
| 64 | |||
| 65 | template <typename T> | ||
| 66 | requires std::is_integral_v<T> | ||
| 67 | [[nodiscard]] constexpr T ResetLeastSignificantOneBit(T x) { | ||
| 68 | return x & (x - 1); | ||
| 69 | } | ||
| 70 | |||
| 71 | template <typename T> | ||
| 72 | requires std::is_integral_v<T> | ||
| 73 | [[nodiscard]] constexpr bool IsPowerOfTwo(T x) { | ||
| 74 | return x > 0 && ResetLeastSignificantOneBit(x) == 0; | ||
| 75 | } | ||
| 76 | |||
| 77 | template <typename T> | ||
| 78 | requires std::is_integral_v<T> | ||
| 79 | [[nodiscard]] constexpr T FloorPowerOfTwo(T x) { | ||
| 80 | return T{1} << (sizeof(T) * 8 - std::countl_zero(x) - 1); | ||
| 81 | } | ||
| 82 | |||
| 58 | template <typename T, size_t Align = 16> | 83 | template <typename T, size_t Align = 16> |
| 59 | class AlignmentAllocator { | 84 | class AlignmentAllocator { |
| 60 | public: | 85 | public: |