Added to_byte() helper function to fix issue #329

This commit is contained in:
Neil MacIntosh 2016-09-13 12:10:42 -07:00 committed by GitHub
commit 2d74fd5c84
2 changed files with 57 additions and 18 deletions

View File

@ -106,6 +106,35 @@ constexpr IntegerType to_integer(byte b) noexcept
return {b}; return {b};
} }
template<bool E, typename T>
constexpr byte to_byte_impl(T t) noexcept
{
static_assert(
E,
"gsl::to_byte(t) must be provided an unsigned char, otherwise data loss may occur. "
"If you are calling to_byte with an integer contant use: gsl::to_byte<t>() version."
);
return static_cast<byte>(t);
}
template<>
constexpr byte to_byte_impl<true, unsigned char>(unsigned char t) noexcept
{
return byte(t);
}
template<typename T>
constexpr byte to_byte(T t) noexcept
{
return to_byte_impl<std::is_same<T, unsigned char>::value, T>(t);
}
template <int I>
constexpr byte to_byte() noexcept
{
static_assert(I >= 0 && I <= 255, "gsl::byte only has 8 bits of storage, values must be in range 0-255");
return static_cast<byte>(I);
}
} // namespace gsl } // namespace gsl
#ifdef _MSC_VER #ifdef _MSC_VER

View File

@ -44,6 +44,16 @@ SUITE(byte_tests)
CHECK(static_cast<unsigned char>(b) == 12); CHECK(static_cast<unsigned char>(b) == 12);
} }
{
byte b = to_byte<12>();
CHECK(static_cast<unsigned char>(b) == 12);
}
{
unsigned char uc = 12;
byte b = to_byte(uc);
CHECK(static_cast<unsigned char>(b) == 12);
}
// waiting for C++17 enum class direct initializer support // waiting for C++17 enum class direct initializer support
//{ //{
// byte b { 14 }; // byte b { 14 };
@ -53,38 +63,38 @@ SUITE(byte_tests)
TEST(bitwise_operations) TEST(bitwise_operations)
{ {
byte b = byte(0xFF); byte b = to_byte<0xFF>();
byte a = byte(0x00); byte a = to_byte<0x00>();
CHECK((b | a) == byte(0xFF)); CHECK((b | a) == to_byte<0xFF>());
CHECK(a == byte(0x00)); CHECK(a == to_byte<0x00>());
a |= b; a |= b;
CHECK(a == byte(0xFF)); CHECK(a == to_byte<0xFF>());
a = byte(0x01); a = to_byte<0x01>();
CHECK((b & a) == byte(0x01)); CHECK((b & a) == to_byte<0x01>());
a &= b; a &= b;
CHECK(a == byte(0x01)); CHECK(a == to_byte<0x01>());
CHECK((b ^ a) == byte(0xFE)); CHECK((b ^ a) == to_byte<0xFE>());
CHECK(a == byte(0x01)); CHECK(a == to_byte<0x01>());
a ^= b; a ^= b;
CHECK(a == byte(0xFE)); CHECK(a == to_byte<0xFE>());
a = byte(0x01); a = to_byte<0x01>();
CHECK(~a == byte(0xFE)); CHECK(~a == to_byte<0xFE>());
a = byte(0xFF); a = to_byte<0xFF>();
CHECK((a << 4) == byte(0xF0)); CHECK((a << 4) == to_byte<0xF0>());
CHECK((a >> 4) == byte(0x0F)); CHECK((a >> 4) == to_byte<0x0F>());
a <<= 4; a <<= 4;
CHECK(a == byte(0xF0)); CHECK(a == to_byte<0xF0>());
a >>= 4; a >>= 4;
CHECK(a == byte(0x0F)); CHECK(a == to_byte<0x0F>());
} }
} }