Atmosphere/libraries/libvapours/include/vapours/util/util_endian.hpp

157 lines
5.5 KiB
C++
Raw Normal View History

2020-02-24 01:34:30 +00:00
/*
* Copyright (c) 2018-2020 Atmosphère-NX
*
* This program is free software; you can redistribute it and/or modify it
* under the terms and conditions of the GNU General Public License,
* version 2, as published by the Free Software Foundation.
*
* This program is distributed in the hope it will be useful, but WITHOUT
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
* more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#pragma once
#include <vapours/common.hpp>
#include <vapours/assert.hpp>
namespace ams::util {
constexpr bool IsLittleEndian() {
return std::endian::native == std::endian::little;
2020-02-24 01:34:30 +00:00
}
constexpr bool IsBigEndian() {
return std::endian::native == std::endian::big;
2020-02-24 01:34:30 +00:00
}
static_assert(IsLittleEndian() ^ IsBigEndian());
template<typename U> requires std::unsigned_integral<U>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE U SwapBytes(const U u) {
static_assert(BITSIZEOF(u8) == 8);
constexpr U ByteMask = 0xFFu;
if constexpr (std::is_same<U, u64>::value) {
return ((u & (ByteMask << 56)) >> 56) |
((u & (ByteMask << 48)) >> 40) |
((u & (ByteMask << 40)) >> 24) |
((u & (ByteMask << 32)) >> 8) |
((u & (ByteMask << 24)) << 8) |
((u & (ByteMask << 16)) << 24) |
((u & (ByteMask << 8)) << 40) |
((u & (ByteMask << 0)) << 56);
} else if constexpr (std::is_same<U, u32>::value) {
return ((u & (ByteMask << 24)) >> 24) |
((u & (ByteMask << 16)) >> 8) |
((u & (ByteMask << 8)) << 8) |
((u & (ByteMask << 0)) << 24);
} else if constexpr (std::is_same<U, u16>::value) {
return ((u & (ByteMask << 8)) >> 8) |
((u & (ByteMask << 0)) << 8);
} else if constexpr (std::is_same<U, u8>::value) {
2020-02-25 04:14:29 +00:00
AMS_UNUSED(ByteMask);
2020-02-24 01:34:30 +00:00
return u;
} else {
static_assert(!std::is_same<U, U>::value);
}
}
2020-02-25 03:29:00 +00:00
constexpr ALWAYS_INLINE u64 SwapBytes48(const u64 u) {
using U = u64;
static_assert(BITSIZEOF(u8) == 8);
constexpr U ByteMask = 0xFFu;
AMS_ASSERT((u & UINT64_C(0xFFFF000000000000)) == 0);
return ((u & (ByteMask << 40)) >> 40) |
((u & (ByteMask << 32)) >> 24) |
((u & (ByteMask << 24)) >> 8) |
((u & (ByteMask << 16)) << 8) |
((u & (ByteMask << 8)) << 24) |
((u & (ByteMask << 0)) << 40);
}
template<typename T> requires std::integral<T>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE void SwapBytes(T *ptr) {
using U = typename std::make_unsigned<T>::type;
*ptr = static_cast<T>(SwapBytes<U>(static_cast<U>(*ptr)));
2020-02-24 01:34:30 +00:00
}
template<typename T> requires std::integral<T>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE T ConvertToBigEndian(const T val) {
2020-02-25 03:29:00 +00:00
using U = typename std::make_unsigned<T>::type;
2020-02-24 01:34:30 +00:00
if constexpr (IsBigEndian()) {
2020-02-25 03:29:00 +00:00
return static_cast<T>(static_cast<U>(val));
2020-02-24 01:34:30 +00:00
} else {
static_assert(IsLittleEndian());
return static_cast<T>(SwapBytes<U>(static_cast<U>(val)));
2020-02-24 01:34:30 +00:00
}
}
template<typename T> requires std::integral<T>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE T ConvertToLittleEndian(const T val) {
2020-02-25 03:29:00 +00:00
using U = typename std::make_unsigned<T>::type;
if constexpr (IsBigEndian()) {
return static_cast<T>(SwapBytes<U>(static_cast<U>(val)));
2020-02-25 03:29:00 +00:00
} else {
static_assert(IsLittleEndian());
return static_cast<T>(static_cast<U>(val));
}
}
template<typename T> requires std::integral<T>
2020-02-25 03:29:00 +00:00
constexpr ALWAYS_INLINE T ConvertToBigEndian48(const T val) {
using U = typename std::make_unsigned<T>::type;
static_assert(sizeof(T) == sizeof(u64));
if constexpr (IsBigEndian()) {
AMS_ASSERT((static_cast<U>(val) & UINT64_C(0xFFFF000000000000)) == 0);
return static_cast<T>(static_cast<U>(val));
} else {
static_assert(IsLittleEndian());
return static_cast<T>(SwapBytes48(static_cast<U>(val)));
}
}
template<typename T> requires std::integral<T>
2020-02-25 03:29:00 +00:00
constexpr ALWAYS_INLINE T ConvertToLittleEndian48(const T val) {
using U = typename std::make_unsigned<T>::type;
static_assert(sizeof(T) == sizeof(u64));
2020-02-24 01:34:30 +00:00
if constexpr (IsBigEndian()) {
2020-02-25 03:29:00 +00:00
return static_cast<T>(SwapBytes48(static_cast<U>(val)));
2020-02-24 01:34:30 +00:00
} else {
static_assert(IsLittleEndian());
2020-02-25 03:29:00 +00:00
AMS_ASSERT((static_cast<U>(val) & UINT64_C(0xFFFF000000000000)) == 0);
return static_cast<T>(static_cast<U>(val));
2020-02-24 01:34:30 +00:00
}
}
template<typename T> requires std::integral<T>
constexpr ALWAYS_INLINE T LoadBigEndian(const T *ptr) {
return ConvertToBigEndian<T>(*ptr);
2020-02-24 01:34:30 +00:00
}
template<typename T> requires std::integral<T>
constexpr ALWAYS_INLINE T LoadLittleEndian(const T *ptr) {
return ConvertToLittleEndian<T>(*ptr);
2020-02-24 01:34:30 +00:00
}
template<typename T> requires std::integral<T>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE void StoreBigEndian(T *ptr, T val) {
*ptr = ConvertToBigEndian<T>(val);
2020-02-24 01:34:30 +00:00
}
template<typename T> requires std::integral<T>
2020-02-24 01:34:30 +00:00
constexpr ALWAYS_INLINE void StoreLittleEndian(T *ptr, T val) {
*ptr = ConvertToLittleEndian<T>(val);
2020-02-24 01:34:30 +00:00
}
}