Use std::endian for endianness test

Remove legacy IS_LE_MACHINE IS_BE_MACHINE macro.
This commit is contained in:
Nekotekina 2020-02-17 20:55:20 +03:00
parent 244e74ebe2
commit 6a1a0bf48d
3 changed files with 80 additions and 39 deletions

View File

@ -4,6 +4,12 @@
#include "util/endian.hpp" #include "util/endian.hpp"
#include <cstring> #include <cstring>
#if __has_include(<bit>)
#include <bit>
#else
#include <type_traits>
#endif
// 128-bit vector type and also se_storage<> storage type // 128-bit vector type and also se_storage<> storage type
union alignas(16) v128 union alignas(16) v128
{ {
@ -26,12 +32,10 @@ union alignas(16) v128
} }
}; };
#if IS_LE_MACHINE == 1
template <typename T, std::size_t N = 16 / sizeof(T)> template <typename T, std::size_t N = 16 / sizeof(T)>
using normal_array_t = masked_array_t<T, N, 0>; using normal_array_t = masked_array_t<T, N, std::endian::little == std::endian::native ? 0 : N - 1>;
template <typename T, std::size_t N = 16 / sizeof(T)> template <typename T, std::size_t N = 16 / sizeof(T)>
using reversed_array_t = masked_array_t<T, N, N - 1>; using reversed_array_t = masked_array_t<T, N, std::endian::little == std::endian::native ? N - 1 : 0>;
#endif
normal_array_t<u64> _u64; normal_array_t<u64> _u64;
normal_array_t<s64> _s64; normal_array_t<s64> _s64;
@ -114,17 +118,27 @@ union alignas(16) v128
// Index 0 returns the MSB and index 127 returns the LSB // Index 0 returns the MSB and index 127 returns the LSB
bit_element operator[](u32 index) bit_element operator[](u32 index)
{ {
#if IS_LE_MACHINE == 1 if constexpr (std::endian::little == std::endian::native)
{
return bit_element(m_data[1 - (index >> 6)], 0x8000000000000000ull >> (index & 0x3F)); return bit_element(m_data[1 - (index >> 6)], 0x8000000000000000ull >> (index & 0x3F));
#endif }
else
{
return bit_element(m_data[index >> 6], 0x8000000000000000ull >> (index & 0x3F));
}
} }
// Index 0 returns the MSB and index 127 returns the LSB // Index 0 returns the MSB and index 127 returns the LSB
bool operator[](u32 index) const bool operator[](u32 index) const
{ {
#if IS_LE_MACHINE == 1 if constexpr (std::endian::little == std::endian::native)
{
return (m_data[1 - (index >> 6)] & (0x8000000000000000ull >> (index & 0x3F))) != 0; return (m_data[1 - (index >> 6)] & (0x8000000000000000ull >> (index & 0x3F))) != 0;
#endif }
else
{
return (m_data[index >> 6] & (0x8000000000000000ull >> (index & 0x3F))) != 0;
}
} }
} _bit; } _bit;
@ -340,12 +354,10 @@ using stx::se_storage;
template <typename T, std::size_t Align = alignof(T)> template <typename T, std::size_t Align = alignof(T)>
using nse_t = se_t<T, false, Align>; using nse_t = se_t<T, false, Align>;
#if IS_LE_MACHINE == 1
template <typename T, std::size_t Align = alignof(T)> template <typename T, std::size_t Align = alignof(T)>
using be_t = se_t<T, true, Align>; using be_t = se_t<T, std::endian::little == std::endian::native, Align>;
template <typename T, std::size_t Align = alignof(T)> template <typename T, std::size_t Align = alignof(T)>
using le_t = se_t<T, false, Align>; using le_t = se_t<T, std::endian::big == std::endian::native, Align>;
#endif
// Type converter: converts native endianness arithmetic/enum types to appropriate se_t<> type // Type converter: converts native endianness arithmetic/enum types to appropriate se_t<> type
template <typename T, bool Se, typename = void> template <typename T, bool Se, typename = void>
@ -414,20 +426,16 @@ struct to_se<T[N], Se>
}; };
// BE/LE aliases for to_se<> // BE/LE aliases for to_se<>
#if IS_LE_MACHINE == 1
template <typename T> template <typename T>
using to_be_t = typename to_se<T, true>::type; using to_be_t = typename to_se<T, std::endian::little == std::endian::native>::type;
template <typename T> template <typename T>
using to_le_t = typename to_se<T, false>::type; using to_le_t = typename to_se<T, std::endian::big == std::endian::native>::type;
#endif
// BE/LE aliases for atomic_t // BE/LE aliases for atomic_t
#if IS_LE_MACHINE == 1
template <typename T> template <typename T>
using atomic_be_t = atomic_t<be_t<T>>; using atomic_be_t = atomic_t<be_t<T>>;
template <typename T> template <typename T>
using atomic_le_t = atomic_t<le_t<T>>; using atomic_le_t = atomic_t<le_t<T>>;
#endif
template <typename T, bool Se, std::size_t Align> template <typename T, bool Se, std::size_t Align>
struct fmt_unveil<se_t<T, Se, Align>, void> struct fmt_unveil<se_t<T, Se, Align>, void>

View File

@ -17,9 +17,9 @@
#include <limits> #include <limits>
#include <array> #include <array>
// Assume little-endian #if __has_include(<bit>)
#define IS_LE_MACHINE 1 #include <bit>
#define IS_BE_MACHINE 0 #endif
#ifndef __has_builtin #ifndef __has_builtin
#define __has_builtin(x) 0 #define __has_builtin(x) 0
@ -82,7 +82,7 @@
#define AUDIT(...) ((void)0) #define AUDIT(...) ((void)0)
#endif #endif
#if defined(__cpp_lib_bit_cast) && (__cpp_lib_bit_cast >= 201806L) #if __cpp_lib_bit_cast >= 201806L
#include <bit> #include <bit>
#else #else
namespace std namespace std
@ -530,31 +530,56 @@ constexpr u32 to_u8(char c)
return static_cast<u8>(c); return static_cast<u8>(c);
} }
// Convert 2-byte string to u16 value like reinterpret_cast does // Convert 1-2-byte string to u16 value like reinterpret_cast does
constexpr u16 operator""_u16(const char* s, std::size_t /*length*/) constexpr u16 operator""_u16(const char* s, std::size_t /*length*/)
{ {
return if constexpr (std::endian::little == std::endian::native)
#if IS_LE_MACHINE == 1 {
static_cast<u16>(to_u8(s[1]) << 8 | to_u8(s[0])); return static_cast<u16>(to_u8(s[1]) << 8 | to_u8(s[0]));
#endif }
else
{
return static_cast<u16>(to_u8(s[0]) << 8 | to_u8(s[1]));
}
} }
// Convert 4-byte string to u32 value like reinterpret_cast does // Convert 3-4-byte string to u32 value like reinterpret_cast does
constexpr u32 operator""_u32(const char* s, std::size_t /*length*/) constexpr u32 operator""_u32(const char* s, std::size_t /*length*/)
{ {
return if constexpr (std::endian::little == std::endian::native)
#if IS_LE_MACHINE == 1 {
to_u8(s[3]) << 24 | to_u8(s[2]) << 16 | to_u8(s[1]) << 8 | to_u8(s[0]); return to_u8(s[3]) << 24 | to_u8(s[2]) << 16 | to_u8(s[1]) << 8 | to_u8(s[0]);
#endif }
else
{
return to_u8(s[0]) << 24 | to_u8(s[1]) << 16 | to_u8(s[2]) << 8 | to_u8(s[3]);
}
} }
// Convert 8-byte string to u64 value like reinterpret_cast does // Convert 5-6-byte string to u64 value like reinterpret_cast does
constexpr u64 operator""_u48(const char* s, std::size_t /*length*/)
{
if constexpr (std::endian::little == std::endian::native)
{
return static_cast<u64>(to_u8(s[5]) << 8 | to_u8(s[4])) << 32 | to_u8(s[3]) << 24 | to_u8(s[2]) << 16 | to_u8(s[1]) << 8 | to_u8(s[0]);
}
else
{
return static_cast<u64>(to_u8(s[0]) << 8 | to_u8(s[1])) << 32 | to_u8(s[2]) << 24 | to_u8(s[3]) << 16 | to_u8(s[4]) << 8 | to_u8(s[5]);
}
}
// Convert 7-8-byte string to u64 value like reinterpret_cast does
constexpr u64 operator""_u64(const char* s, std::size_t /*length*/) constexpr u64 operator""_u64(const char* s, std::size_t /*length*/)
{ {
return if constexpr (std::endian::little == std::endian::native)
#if IS_LE_MACHINE == 1 {
static_cast<u64>(to_u8(s[7]) << 24 | to_u8(s[6]) << 16 | to_u8(s[5]) << 8 | to_u8(s[4])) << 32 | to_u8(s[3]) << 24 | to_u8(s[2]) << 16 | to_u8(s[1]) << 8 | to_u8(s[0]); return static_cast<u64>(to_u8(s[7]) << 24 | to_u8(s[6]) << 16 | to_u8(s[5]) << 8 | to_u8(s[4])) << 32 | to_u8(s[3]) << 24 | to_u8(s[2]) << 16 | to_u8(s[1]) << 8 | to_u8(s[0]);
#endif }
else
{
return static_cast<u64>(to_u8(s[0]) << 24 | to_u8(s[1]) << 16 | to_u8(s[2]) << 8 | to_u8(s[3])) << 32 | to_u8(s[4]) << 24 | to_u8(s[5]) << 16 | to_u8(s[6]) << 8 | to_u8(s[7]);
}
} }
namespace fmt namespace fmt

View File

@ -3,8 +3,16 @@
#include <cstdint> #include <cstdint>
#include "Utilities/types.h" #include "Utilities/types.h"
#if __has_include(<bit>)
#include <bit>
#else
#include <type_traits>
#endif
namespace stx namespace stx
{ {
static_assert(std::endian::native == std::endian::little || std::endian::native == std::endian::big);
template <typename T, std::size_t Align = alignof(T), std::size_t Size = sizeof(T)> template <typename T, std::size_t Align = alignof(T), std::size_t Size = sizeof(T)>
struct se_storage struct se_storage
{ {