diff --git a/gsl/gsl_byte b/gsl/gsl_byte index 5134032..09592eb 100644 --- a/gsl/gsl_byte +++ b/gsl/gsl_byte @@ -106,6 +106,35 @@ constexpr IntegerType to_integer(byte b) noexcept return {b}; } +template +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() version." + ); + return static_cast(t); +} +template<> +constexpr byte to_byte_impl(unsigned char t) noexcept +{ + return byte(t); +} + +template +constexpr byte to_byte(T t) noexcept +{ + return to_byte_impl::value, T>(t); +} + +template +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(I); +} + } // namespace gsl #ifdef _MSC_VER diff --git a/tests/byte_tests.cpp b/tests/byte_tests.cpp index f2f8026..59ff0cd 100644 --- a/tests/byte_tests.cpp +++ b/tests/byte_tests.cpp @@ -43,6 +43,16 @@ SUITE(byte_tests) byte b = byte(12); CHECK(static_cast(b) == 12); } + + { + byte b = to_byte<12>(); + CHECK(static_cast(b) == 12); + } + { + unsigned char uc = 12; + byte b = to_byte(uc); + CHECK(static_cast(b) == 12); + } // waiting for C++17 enum class direct initializer support //{ @@ -53,38 +63,38 @@ SUITE(byte_tests) TEST(bitwise_operations) { - byte b = byte(0xFF); + byte b = to_byte<0xFF>(); - byte a = byte(0x00); - CHECK((b | a) == byte(0xFF)); - CHECK(a == byte(0x00)); + byte a = to_byte<0x00>(); + CHECK((b | a) == to_byte<0xFF>()); + CHECK(a == to_byte<0x00>()); a |= b; - CHECK(a == byte(0xFF)); + CHECK(a == to_byte<0xFF>()); - a = byte(0x01); - CHECK((b & a) == byte(0x01)); + a = to_byte<0x01>(); + CHECK((b & a) == to_byte<0x01>()); 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; - CHECK(a == byte(0xFE)); + CHECK(a == to_byte<0xFE>()); - a = byte(0x01); - CHECK(~a == byte(0xFE)); + a = to_byte<0x01>(); + CHECK(~a == to_byte<0xFE>()); - a = byte(0xFF); - CHECK((a << 4) == byte(0xF0)); - CHECK((a >> 4) == byte(0x0F)); + a = to_byte<0xFF>(); + CHECK((a << 4) == to_byte<0xF0>()); + CHECK((a >> 4) == to_byte<0x0F>()); a <<= 4; - CHECK(a == byte(0xF0)); + CHECK(a == to_byte<0xF0>()); a >>= 4; - CHECK(a == byte(0x0F)); + CHECK(a == to_byte<0x0F>()); } }