Skip to content

Commit 8fc6c85

Browse files
committed
fix review
1 parent b2ed0c3 commit 8fc6c85

File tree

2 files changed

+37
-69
lines changed

2 files changed

+37
-69
lines changed

src/iceberg/util/endian.h

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ namespace iceberg {
3030

3131
/// \brief Concept for values that can be converted to/from another endian format.
3232
template <typename T>
33-
concept EndianConvertible = std::is_arithmetic_v<T> && !std::same_as<T, bool>;
33+
concept EndianConvertible = std::is_arithmetic_v<T>;
3434

3535
/// \brief Byte-swap a value. For floating-point types, only support 32-bit and 64-bit
3636
/// floats.
@@ -41,7 +41,9 @@ constexpr T ByteSwap(T value) {
4141
} else if constexpr (std::is_integral_v<T>) {
4242
return std::byteswap(value);
4343
} else if constexpr (std::is_floating_point_v<T>) {
44-
if constexpr (sizeof(T) == sizeof(uint32_t)) {
44+
if constexpr (sizeof(T) == sizeof(uint16_t)) {
45+
return std::bit_cast<T>(std::byteswap(std::bit_cast<uint16_t>(value)));
46+
} else if constexpr (sizeof(T) == sizeof(uint32_t)) {
4547
return std::bit_cast<T>(std::byteswap(std::bit_cast<uint32_t>(value)));
4648
} else if constexpr (sizeof(T) == sizeof(uint64_t)) {
4749
return std::bit_cast<T>(std::byteswap(std::bit_cast<uint64_t>(value)));
@@ -54,7 +56,7 @@ constexpr T ByteSwap(T value) {
5456
/// \brief Convert a value to little-endian format.
5557
template <EndianConvertible T>
5658
constexpr T ToLittleEndian(T value) {
57-
if constexpr (std::endian::native == std::endian::little || sizeof(T) <= 1) {
59+
if constexpr (std::endian::native == std::endian::little) {
5860
return value;
5961
} else {
6062
return ByteSwap(value);
@@ -64,7 +66,7 @@ constexpr T ToLittleEndian(T value) {
6466
/// \brief Convert a value from little-endian format.
6567
template <EndianConvertible T>
6668
constexpr T FromLittleEndian(T value) {
67-
if constexpr (std::endian::native == std::endian::little || sizeof(T) <= 1) {
69+
if constexpr (std::endian::native == std::endian::little) {
6870
return value;
6971
} else {
7072
return ByteSwap(value);
@@ -74,7 +76,7 @@ constexpr T FromLittleEndian(T value) {
7476
/// \brief Convert a value to big-endian format.
7577
template <EndianConvertible T>
7678
constexpr T ToBigEndian(T value) {
77-
if constexpr (std::endian::native == std::endian::big || sizeof(T) <= 1) {
79+
if constexpr (std::endian::native == std::endian::big) {
7880
return value;
7981
} else {
8082
return ByteSwap(value);
@@ -84,7 +86,7 @@ constexpr T ToBigEndian(T value) {
8486
/// \brief Convert a value from big-endian format.
8587
template <EndianConvertible T>
8688
constexpr T FromBigEndian(T value) {
87-
if constexpr (std::endian::native == std::endian::big || sizeof(T) <= 1) {
89+
if constexpr (std::endian::native == std::endian::big) {
8890
return value;
8991
} else {
9092
return ByteSwap(value);

test/endian_test.cc

Lines changed: 29 additions & 63 deletions
Original file line numberDiff line numberDiff line change
@@ -27,76 +27,42 @@
2727

2828
namespace iceberg {
2929

30-
// test round trip preserves value
30+
#define EXPECT_ROUNDTRIP(value) \
31+
do { \
32+
EXPECT_EQ(FromLittleEndian(ToLittleEndian(value)), value); \
33+
EXPECT_EQ(FromBigEndian(ToBigEndian(value)), value); \
34+
} while (false)
35+
3136
TEST(EndianTest, RoundTripPreservesValue) {
32-
EXPECT_EQ(FromLittleEndian(ToLittleEndian<uint16_t>(0x1234)), 0x1234);
33-
EXPECT_EQ(FromBigEndian(ToBigEndian<uint32_t>(0xDEADBEEF)), 0xDEADBEEF);
34-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(std::numeric_limits<uint64_t>::max())),
35-
std::numeric_limits<uint64_t>::max());
36-
EXPECT_EQ(FromBigEndian(ToBigEndian<uint32_t>(0)), 0);
37-
38-
EXPECT_EQ(FromBigEndian(ToBigEndian<int16_t>(-1)), -1);
39-
EXPECT_EQ(FromLittleEndian(ToLittleEndian<int32_t>(-0x12345678)), -0x12345678);
40-
EXPECT_EQ(FromBigEndian(ToBigEndian(std::numeric_limits<int64_t>::min())),
41-
std::numeric_limits<int64_t>::min());
42-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(std::numeric_limits<int16_t>::max())),
43-
std::numeric_limits<int16_t>::max());
44-
45-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(3.14f)), 3.14f);
46-
EXPECT_EQ(FromBigEndian(ToBigEndian(2.718281828459045)), 2.718281828459045);
47-
48-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(std::numeric_limits<float>::infinity())),
49-
std::numeric_limits<float>::infinity());
50-
EXPECT_EQ(FromBigEndian(ToBigEndian(-std::numeric_limits<float>::infinity())),
51-
-std::numeric_limits<float>::infinity());
37+
EXPECT_ROUNDTRIP(static_cast<uint16_t>(0x1234));
38+
EXPECT_ROUNDTRIP(static_cast<uint32_t>(0xDEADBEEF));
39+
EXPECT_ROUNDTRIP(std::numeric_limits<uint64_t>::max());
40+
EXPECT_ROUNDTRIP(static_cast<uint32_t>(0));
41+
42+
EXPECT_ROUNDTRIP(static_cast<int16_t>(-1));
43+
EXPECT_ROUNDTRIP(static_cast<int32_t>(-0x12345678));
44+
EXPECT_ROUNDTRIP(std::numeric_limits<int64_t>::min());
45+
EXPECT_ROUNDTRIP(std::numeric_limits<int16_t>::max());
46+
47+
EXPECT_ROUNDTRIP(3.14f);
48+
EXPECT_ROUNDTRIP(2.718281828459045);
49+
EXPECT_ROUNDTRIP(0.0f);
50+
EXPECT_ROUNDTRIP(-0.0f);
51+
EXPECT_ROUNDTRIP(0.0);
52+
EXPECT_ROUNDTRIP(-0.0);
53+
54+
EXPECT_ROUNDTRIP(std::numeric_limits<float>::infinity());
55+
EXPECT_ROUNDTRIP(-std::numeric_limits<float>::infinity());
56+
EXPECT_ROUNDTRIP(std::numeric_limits<double>::infinity());
57+
EXPECT_ROUNDTRIP(-std::numeric_limits<double>::infinity());
58+
5259
EXPECT_TRUE(std::isnan(
5360
FromLittleEndian(ToLittleEndian(std::numeric_limits<float>::quiet_NaN()))));
54-
EXPECT_EQ(FromBigEndian(ToBigEndian(0.0f)), 0.0f);
55-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(-0.0f)), -0.0f);
56-
57-
EXPECT_EQ(FromBigEndian(ToBigEndian(std::numeric_limits<double>::infinity())),
58-
std::numeric_limits<double>::infinity());
59-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(-std::numeric_limits<double>::infinity())),
60-
-std::numeric_limits<double>::infinity());
6161
EXPECT_TRUE(
6262
std::isnan(FromBigEndian(ToBigEndian(std::numeric_limits<double>::quiet_NaN()))));
63-
EXPECT_EQ(FromLittleEndian(ToLittleEndian(0.0)), 0.0);
64-
EXPECT_EQ(FromBigEndian(ToBigEndian(-0.0)), -0.0);
65-
}
66-
67-
// test constexpr evaluation
68-
TEST(EndianTest, ConstexprEvaluation) {
69-
static_assert(FromBigEndian(ToBigEndian<uint16_t>(0x1234)) == 0x1234);
70-
static_assert(FromLittleEndian(ToLittleEndian<uint32_t>(0x12345678)) == 0x12345678);
71-
static_assert(FromBigEndian(ToBigEndian<int64_t>(-1)) == -1);
72-
73-
static_assert(ToBigEndian<uint8_t>(0xFF) == 0xFF);
74-
static_assert(FromLittleEndian<int8_t>(-1) == -1);
75-
76-
static_assert(FromLittleEndian(ToLittleEndian(3.14f)) == 3.14f);
77-
static_assert(FromBigEndian(ToBigEndian(2.71)) == 2.71);
78-
}
79-
80-
// test platform dependent behavior
81-
TEST(EndianTest, PlatformDependentBehavior) {
82-
uint32_t test_value = 0x12345678;
83-
84-
if constexpr (std::endian::native == std::endian::little) {
85-
EXPECT_EQ(ToLittleEndian(test_value), test_value);
86-
EXPECT_EQ(FromLittleEndian(test_value), test_value);
87-
EXPECT_NE(ToBigEndian(test_value), test_value);
88-
} else if constexpr (std::endian::native == std::endian::big) {
89-
EXPECT_EQ(ToBigEndian(test_value), test_value);
90-
EXPECT_EQ(FromBigEndian(test_value), test_value);
91-
EXPECT_NE(ToLittleEndian(test_value), test_value);
92-
}
93-
94-
EXPECT_EQ(ToLittleEndian<uint8_t>(0xAB), 0xAB);
95-
EXPECT_EQ(ToBigEndian<uint8_t>(0xAB), 0xAB);
9663
}
9764

98-
// test specific byte pattern validation
99-
TEST(EndianTest, SpecificBytePatternValidation) {
65+
TEST(EndianTest, ByteWiseValidation) {
10066
uint32_t original_int = 0x12345678;
10167
uint32_t little_endian_int = ToLittleEndian(original_int);
10268
uint32_t big_endian_int = ToBigEndian(original_int);

0 commit comments

Comments
 (0)