bits_test.cc 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573
  1. // Copyright 2020 The Abseil Authors
  2. //
  3. // Licensed under the Apache License, Version 2.0 (the "License");
  4. // you may not use this file except in compliance with the License.
  5. // You may obtain a copy of the License at
  6. //
  7. // https://www.apache.org/licenses/LICENSE-2.0
  8. //
  9. // Unless required by applicable law or agreed to in writing, software
  10. // distributed under the License is distributed on an "AS IS" BASIS,
  11. // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. // See the License for the specific language governing permissions and
  13. // limitations under the License.
  14. #include "absl/numeric/bits.h"
  15. #include <limits>
  16. #include "gmock/gmock.h"
  17. #include "gtest/gtest.h"
  18. #include "absl/random/random.h"
  19. namespace absl {
  20. ABSL_NAMESPACE_BEGIN
  21. namespace {
  22. TEST(Rotate, Left) {
  23. static_assert(rotl(uint8_t{0x12}, 0) == uint8_t{0x12}, "");
  24. static_assert(rotl(uint16_t{0x1234}, 0) == uint16_t{0x1234}, "");
  25. static_assert(rotl(uint32_t{0x12345678UL}, 0) == uint32_t{0x12345678UL}, "");
  26. static_assert(rotl(uint64_t{0x12345678ABCDEF01ULL}, 0) ==
  27. uint64_t{0x12345678ABCDEF01ULL},
  28. "");
  29. EXPECT_EQ(rotl(uint8_t{0x12}, 0), uint8_t{0x12});
  30. EXPECT_EQ(rotl(uint16_t{0x1234}, 0), uint16_t{0x1234});
  31. EXPECT_EQ(rotl(uint32_t{0x12345678UL}, 0), uint32_t{0x12345678UL});
  32. EXPECT_EQ(rotl(uint64_t{0x12345678ABCDEF01ULL}, 0),
  33. uint64_t{0x12345678ABCDEF01ULL});
  34. EXPECT_EQ(rotl(uint8_t{0x12}, 8), uint8_t{0x12});
  35. EXPECT_EQ(rotl(uint16_t{0x1234}, 16), uint16_t{0x1234});
  36. EXPECT_EQ(rotl(uint32_t{0x12345678UL}, 32), uint32_t{0x12345678UL});
  37. EXPECT_EQ(rotl(uint64_t{0x12345678ABCDEF01ULL}, 64),
  38. uint64_t{0x12345678ABCDEF01ULL});
  39. EXPECT_EQ(rotl(uint8_t{0x12}, -8), uint8_t{0x12});
  40. EXPECT_EQ(rotl(uint16_t{0x1234}, -16), uint16_t{0x1234});
  41. EXPECT_EQ(rotl(uint32_t{0x12345678UL}, -32), uint32_t{0x12345678UL});
  42. EXPECT_EQ(rotl(uint64_t{0x12345678ABCDEF01ULL}, -64),
  43. uint64_t{0x12345678ABCDEF01ULL});
  44. EXPECT_EQ(rotl(uint8_t{0x12}, 4), uint8_t{0x21});
  45. EXPECT_EQ(rotl(uint16_t{0x1234}, 4), uint16_t{0x2341});
  46. EXPECT_EQ(rotl(uint32_t{0x12345678UL}, 4), uint32_t{0x23456781UL});
  47. EXPECT_EQ(rotl(uint64_t{0x12345678ABCDEF01ULL}, 4),
  48. uint64_t{0x2345678ABCDEF011ULL});
  49. EXPECT_EQ(rotl(uint8_t{0x12}, -4), uint8_t{0x21});
  50. EXPECT_EQ(rotl(uint16_t{0x1234}, -4), uint16_t{0x4123});
  51. EXPECT_EQ(rotl(uint32_t{0x12345678UL}, -4), uint32_t{0x81234567UL});
  52. EXPECT_EQ(rotl(uint64_t{0x12345678ABCDEF01ULL}, -4),
  53. uint64_t{0x112345678ABCDEF0ULL});
  54. }
  55. TEST(Rotate, Right) {
  56. static_assert(rotr(uint8_t{0x12}, 0) == uint8_t{0x12}, "");
  57. static_assert(rotr(uint16_t{0x1234}, 0) == uint16_t{0x1234}, "");
  58. static_assert(rotr(uint32_t{0x12345678UL}, 0) == uint32_t{0x12345678UL}, "");
  59. static_assert(rotr(uint64_t{0x12345678ABCDEF01ULL}, 0) ==
  60. uint64_t{0x12345678ABCDEF01ULL},
  61. "");
  62. EXPECT_EQ(rotr(uint8_t{0x12}, 0), uint8_t{0x12});
  63. EXPECT_EQ(rotr(uint16_t{0x1234}, 0), uint16_t{0x1234});
  64. EXPECT_EQ(rotr(uint32_t{0x12345678UL}, 0), uint32_t{0x12345678UL});
  65. EXPECT_EQ(rotr(uint64_t{0x12345678ABCDEF01ULL}, 0),
  66. uint64_t{0x12345678ABCDEF01ULL});
  67. EXPECT_EQ(rotr(uint8_t{0x12}, 8), uint8_t{0x12});
  68. EXPECT_EQ(rotr(uint16_t{0x1234}, 16), uint16_t{0x1234});
  69. EXPECT_EQ(rotr(uint32_t{0x12345678UL}, 32), uint32_t{0x12345678UL});
  70. EXPECT_EQ(rotr(uint64_t{0x12345678ABCDEF01ULL}, 64),
  71. uint64_t{0x12345678ABCDEF01ULL});
  72. EXPECT_EQ(rotr(uint8_t{0x12}, -8), uint8_t{0x12});
  73. EXPECT_EQ(rotr(uint16_t{0x1234}, -16), uint16_t{0x1234});
  74. EXPECT_EQ(rotr(uint32_t{0x12345678UL}, -32), uint32_t{0x12345678UL});
  75. EXPECT_EQ(rotr(uint64_t{0x12345678ABCDEF01ULL}, -64),
  76. uint64_t{0x12345678ABCDEF01ULL});
  77. EXPECT_EQ(rotr(uint8_t{0x12}, 4), uint8_t{0x21});
  78. EXPECT_EQ(rotr(uint16_t{0x1234}, 4), uint16_t{0x4123});
  79. EXPECT_EQ(rotr(uint32_t{0x12345678UL}, 4), uint32_t{0x81234567UL});
  80. EXPECT_EQ(rotr(uint64_t{0x12345678ABCDEF01ULL}, 4),
  81. uint64_t{0x112345678ABCDEF0ULL});
  82. EXPECT_EQ(rotr(uint8_t{0x12}, -4), uint8_t{0x21});
  83. EXPECT_EQ(rotr(uint16_t{0x1234}, -4), uint16_t{0x2341});
  84. EXPECT_EQ(rotr(uint32_t{0x12345678UL}, -4), uint32_t{0x23456781UL});
  85. EXPECT_EQ(rotr(uint64_t{0x12345678ABCDEF01ULL}, -4),
  86. uint64_t{0x2345678ABCDEF011ULL});
  87. }
  88. TEST(Rotate, Symmetry) {
  89. // rotr(x, s) is equivalent to rotl(x, -s)
  90. absl::BitGen rng;
  91. constexpr int kTrials = 100;
  92. for (int i = 0; i < kTrials; ++i) {
  93. uint8_t value = absl::Uniform(rng, std::numeric_limits<uint8_t>::min(),
  94. std::numeric_limits<uint8_t>::max());
  95. int shift = absl::Uniform(rng, -2 * std::numeric_limits<uint8_t>::digits,
  96. 2 * std::numeric_limits<uint8_t>::digits);
  97. EXPECT_EQ(rotl(value, shift), rotr(value, -shift));
  98. }
  99. for (int i = 0; i < kTrials; ++i) {
  100. uint16_t value = absl::Uniform(rng, std::numeric_limits<uint16_t>::min(),
  101. std::numeric_limits<uint16_t>::max());
  102. int shift = absl::Uniform(rng, -2 * std::numeric_limits<uint16_t>::digits,
  103. 2 * std::numeric_limits<uint16_t>::digits);
  104. EXPECT_EQ(rotl(value, shift), rotr(value, -shift));
  105. }
  106. for (int i = 0; i < kTrials; ++i) {
  107. uint32_t value = absl::Uniform(rng, std::numeric_limits<uint32_t>::min(),
  108. std::numeric_limits<uint32_t>::max());
  109. int shift = absl::Uniform(rng, -2 * std::numeric_limits<uint32_t>::digits,
  110. 2 * std::numeric_limits<uint32_t>::digits);
  111. EXPECT_EQ(rotl(value, shift), rotr(value, -shift));
  112. }
  113. for (int i = 0; i < kTrials; ++i) {
  114. uint64_t value = absl::Uniform(rng, std::numeric_limits<uint64_t>::min(),
  115. std::numeric_limits<uint64_t>::max());
  116. int shift = absl::Uniform(rng, -2 * std::numeric_limits<uint64_t>::digits,
  117. 2 * std::numeric_limits<uint64_t>::digits);
  118. EXPECT_EQ(rotl(value, shift), rotr(value, -shift));
  119. }
  120. }
  121. TEST(Counting, LeadingZeroes) {
  122. #if ABSL_INTERNAL_HAS_CONSTEXPR_CLZ
  123. static_assert(countl_zero(uint8_t{}) == 8, "");
  124. static_assert(countl_zero(static_cast<uint8_t>(-1)) == 0, "");
  125. static_assert(countl_zero(uint16_t{}) == 16, "");
  126. static_assert(countl_zero(static_cast<uint16_t>(-1)) == 0, "");
  127. static_assert(countl_zero(uint32_t{}) == 32, "");
  128. static_assert(countl_zero(~uint32_t{}) == 0, "");
  129. static_assert(countl_zero(uint64_t{}) == 64, "");
  130. static_assert(countl_zero(~uint64_t{}) == 0, "");
  131. #endif
  132. EXPECT_EQ(countl_zero(uint8_t{}), 8);
  133. EXPECT_EQ(countl_zero(static_cast<uint8_t>(-1)), 0);
  134. EXPECT_EQ(countl_zero(uint16_t{}), 16);
  135. EXPECT_EQ(countl_zero(static_cast<uint16_t>(-1)), 0);
  136. EXPECT_EQ(countl_zero(uint32_t{}), 32);
  137. EXPECT_EQ(countl_zero(~uint32_t{}), 0);
  138. EXPECT_EQ(countl_zero(uint64_t{}), 64);
  139. EXPECT_EQ(countl_zero(~uint64_t{}), 0);
  140. for (int i = 0; i < 8; i++) {
  141. EXPECT_EQ(countl_zero(static_cast<uint8_t>(1u << i)), 7 - i);
  142. }
  143. for (int i = 0; i < 16; i++) {
  144. EXPECT_EQ(countl_zero(static_cast<uint16_t>(1u << i)), 15 - i);
  145. }
  146. for (int i = 0; i < 32; i++) {
  147. EXPECT_EQ(countl_zero(uint32_t{1} << i), 31 - i);
  148. }
  149. for (int i = 0; i < 64; i++) {
  150. EXPECT_EQ(countl_zero(uint64_t{1} << i), 63 - i);
  151. }
  152. }
  153. TEST(Counting, LeadingOnes) {
  154. #if ABSL_INTERNAL_HAS_CONSTEXPR_CLZ
  155. static_assert(countl_one(uint8_t{}) == 0, "");
  156. static_assert(countl_one(static_cast<uint8_t>(-1)) == 8, "");
  157. static_assert(countl_one(uint16_t{}) == 0, "");
  158. static_assert(countl_one(static_cast<uint16_t>(-1)) == 16, "");
  159. static_assert(countl_one(uint32_t{}) == 0, "");
  160. static_assert(countl_one(~uint32_t{}) == 32, "");
  161. static_assert(countl_one(uint64_t{}) == 0, "");
  162. static_assert(countl_one(~uint64_t{}) == 64, "");
  163. #endif
  164. EXPECT_EQ(countl_one(uint8_t{}), 0);
  165. EXPECT_EQ(countl_one(static_cast<uint8_t>(-1)), 8);
  166. EXPECT_EQ(countl_one(uint16_t{}), 0);
  167. EXPECT_EQ(countl_one(static_cast<uint16_t>(-1)), 16);
  168. EXPECT_EQ(countl_one(uint32_t{}), 0);
  169. EXPECT_EQ(countl_one(~uint32_t{}), 32);
  170. EXPECT_EQ(countl_one(uint64_t{}), 0);
  171. EXPECT_EQ(countl_one(~uint64_t{}), 64);
  172. }
  173. TEST(Counting, TrailingZeroes) {
  174. #if ABSL_INTERNAL_HAS_CONSTEXPR_CTZ
  175. static_assert(countr_zero(uint8_t{}) == 8, "");
  176. static_assert(countr_zero(static_cast<uint8_t>(-1)) == 0, "");
  177. static_assert(countr_zero(uint16_t{}) == 16, "");
  178. static_assert(countr_zero(static_cast<uint16_t>(-1)) == 0, "");
  179. static_assert(countr_zero(uint32_t{}) == 32, "");
  180. static_assert(countr_zero(~uint32_t{}) == 0, "");
  181. static_assert(countr_zero(uint64_t{}) == 64, "");
  182. static_assert(countr_zero(~uint64_t{}) == 0, "");
  183. #endif
  184. EXPECT_EQ(countr_zero(uint8_t{}), 8);
  185. EXPECT_EQ(countr_zero(static_cast<uint8_t>(-1)), 0);
  186. EXPECT_EQ(countr_zero(uint16_t{}), 16);
  187. EXPECT_EQ(countr_zero(static_cast<uint16_t>(-1)), 0);
  188. EXPECT_EQ(countr_zero(uint32_t{}), 32);
  189. EXPECT_EQ(countr_zero(~uint32_t{}), 0);
  190. EXPECT_EQ(countr_zero(uint64_t{}), 64);
  191. EXPECT_EQ(countr_zero(~uint64_t{}), 0);
  192. }
  193. TEST(Counting, TrailingOnes) {
  194. #if ABSL_INTERNAL_HAS_CONSTEXPR_CTZ
  195. static_assert(countr_one(uint8_t{}) == 0, "");
  196. static_assert(countr_one(static_cast<uint8_t>(-1)) == 8, "");
  197. static_assert(countr_one(uint16_t{}) == 0, "");
  198. static_assert(countr_one(static_cast<uint16_t>(-1)) == 16, "");
  199. static_assert(countr_one(uint32_t{}) == 0, "");
  200. static_assert(countr_one(~uint32_t{}) == 32, "");
  201. static_assert(countr_one(uint64_t{}) == 0, "");
  202. static_assert(countr_one(~uint64_t{}) == 64, "");
  203. #endif
  204. EXPECT_EQ(countr_one(uint8_t{}), 0);
  205. EXPECT_EQ(countr_one(static_cast<uint8_t>(-1)), 8);
  206. EXPECT_EQ(countr_one(uint16_t{}), 0);
  207. EXPECT_EQ(countr_one(static_cast<uint16_t>(-1)), 16);
  208. EXPECT_EQ(countr_one(uint32_t{}), 0);
  209. EXPECT_EQ(countr_one(~uint32_t{}), 32);
  210. EXPECT_EQ(countr_one(uint64_t{}), 0);
  211. EXPECT_EQ(countr_one(~uint64_t{}), 64);
  212. }
  213. TEST(Counting, Popcount) {
  214. #if ABSL_INTERNAL_HAS_CONSTEXPR_POPCOUNT
  215. static_assert(popcount(uint8_t{}) == 0, "");
  216. static_assert(popcount(uint8_t{1}) == 1, "");
  217. static_assert(popcount(static_cast<uint8_t>(-1)) == 8, "");
  218. static_assert(popcount(uint16_t{}) == 0, "");
  219. static_assert(popcount(uint16_t{1}) == 1, "");
  220. static_assert(popcount(static_cast<uint16_t>(-1)) == 16, "");
  221. static_assert(popcount(uint32_t{}) == 0, "");
  222. static_assert(popcount(uint32_t{1}) == 1, "");
  223. static_assert(popcount(~uint32_t{}) == 32, "");
  224. static_assert(popcount(uint64_t{}) == 0, "");
  225. static_assert(popcount(uint64_t{1}) == 1, "");
  226. static_assert(popcount(~uint64_t{}) == 64, "");
  227. #endif // ABSL_INTERNAL_HAS_CONSTEXPR_POPCOUNT
  228. EXPECT_EQ(popcount(uint8_t{}), 0);
  229. EXPECT_EQ(popcount(uint8_t{1}), 1);
  230. EXPECT_EQ(popcount(static_cast<uint8_t>(-1)), 8);
  231. EXPECT_EQ(popcount(uint16_t{}), 0);
  232. EXPECT_EQ(popcount(uint16_t{1}), 1);
  233. EXPECT_EQ(popcount(static_cast<uint16_t>(-1)), 16);
  234. EXPECT_EQ(popcount(uint32_t{}), 0);
  235. EXPECT_EQ(popcount(uint32_t{1}), 1);
  236. EXPECT_EQ(popcount(~uint32_t{}), 32);
  237. EXPECT_EQ(popcount(uint64_t{}), 0);
  238. EXPECT_EQ(popcount(uint64_t{1}), 1);
  239. EXPECT_EQ(popcount(~uint64_t{}), 64);
  240. for (int i = 0; i < 8; i++) {
  241. EXPECT_EQ(popcount(static_cast<uint8_t>(uint8_t{1} << i)), 1);
  242. EXPECT_EQ(popcount(static_cast<uint8_t>(static_cast<uint8_t>(-1) ^
  243. (uint8_t{1} << i))),
  244. 7);
  245. }
  246. for (int i = 0; i < 16; i++) {
  247. EXPECT_EQ(popcount(static_cast<uint16_t>(uint16_t{1} << i)), 1);
  248. EXPECT_EQ(popcount(static_cast<uint16_t>(static_cast<uint16_t>(-1) ^
  249. (uint16_t{1} << i))),
  250. 15);
  251. }
  252. for (int i = 0; i < 32; i++) {
  253. EXPECT_EQ(popcount(uint32_t{1} << i), 1);
  254. EXPECT_EQ(popcount(static_cast<uint32_t>(-1) ^ (uint32_t{1} << i)), 31);
  255. }
  256. for (int i = 0; i < 64; i++) {
  257. EXPECT_EQ(popcount(uint64_t{1} << i), 1);
  258. EXPECT_EQ(popcount(static_cast<uint64_t>(-1) ^ (uint64_t{1} << i)), 63);
  259. }
  260. }
  261. template <typename T>
  262. struct PopcountInput {
  263. T value = 0;
  264. int expected = 0;
  265. };
  266. template <typename T>
  267. PopcountInput<T> GeneratePopcountInput(absl::BitGen& gen) {
  268. PopcountInput<T> ret;
  269. for (int i = 0; i < std::numeric_limits<T>::digits; i++) {
  270. bool coin = absl::Bernoulli(gen, 0.2);
  271. if (coin) {
  272. ret.value |= T{1} << i;
  273. ret.expected++;
  274. }
  275. }
  276. return ret;
  277. }
  278. TEST(Counting, PopcountFuzz) {
  279. absl::BitGen rng;
  280. constexpr int kTrials = 100;
  281. for (int i = 0; i < kTrials; ++i) {
  282. auto input = GeneratePopcountInput<uint8_t>(rng);
  283. EXPECT_EQ(popcount(input.value), input.expected);
  284. }
  285. for (int i = 0; i < kTrials; ++i) {
  286. auto input = GeneratePopcountInput<uint16_t>(rng);
  287. EXPECT_EQ(popcount(input.value), input.expected);
  288. }
  289. for (int i = 0; i < kTrials; ++i) {
  290. auto input = GeneratePopcountInput<uint32_t>(rng);
  291. EXPECT_EQ(popcount(input.value), input.expected);
  292. }
  293. for (int i = 0; i < kTrials; ++i) {
  294. auto input = GeneratePopcountInput<uint64_t>(rng);
  295. EXPECT_EQ(popcount(input.value), input.expected);
  296. }
  297. }
  298. TEST(IntegralPowersOfTwo, SingleBit) {
  299. EXPECT_FALSE(has_single_bit(uint8_t{}));
  300. EXPECT_FALSE(has_single_bit(static_cast<uint8_t>(-1)));
  301. EXPECT_FALSE(has_single_bit(uint16_t{}));
  302. EXPECT_FALSE(has_single_bit(static_cast<uint16_t>(-1)));
  303. EXPECT_FALSE(has_single_bit(uint32_t{}));
  304. EXPECT_FALSE(has_single_bit(~uint32_t{}));
  305. EXPECT_FALSE(has_single_bit(uint64_t{}));
  306. EXPECT_FALSE(has_single_bit(~uint64_t{}));
  307. static_assert(!has_single_bit(0u), "");
  308. static_assert(has_single_bit(1u), "");
  309. static_assert(has_single_bit(2u), "");
  310. static_assert(!has_single_bit(3u), "");
  311. static_assert(has_single_bit(4u), "");
  312. static_assert(!has_single_bit(1337u), "");
  313. static_assert(has_single_bit(65536u), "");
  314. static_assert(has_single_bit(uint32_t{1} << 30), "");
  315. static_assert(has_single_bit(uint64_t{1} << 42), "");
  316. EXPECT_FALSE(has_single_bit(0u));
  317. EXPECT_TRUE(has_single_bit(1u));
  318. EXPECT_TRUE(has_single_bit(2u));
  319. EXPECT_FALSE(has_single_bit(3u));
  320. EXPECT_TRUE(has_single_bit(4u));
  321. EXPECT_FALSE(has_single_bit(1337u));
  322. EXPECT_TRUE(has_single_bit(65536u));
  323. EXPECT_TRUE(has_single_bit(uint32_t{1} << 30));
  324. EXPECT_TRUE(has_single_bit(uint64_t{1} << 42));
  325. EXPECT_TRUE(has_single_bit(
  326. static_cast<uint8_t>(std::numeric_limits<uint8_t>::max() / 2 + 1)));
  327. EXPECT_TRUE(has_single_bit(
  328. static_cast<uint16_t>(std::numeric_limits<uint16_t>::max() / 2 + 1)));
  329. EXPECT_TRUE(has_single_bit(
  330. static_cast<uint32_t>(std::numeric_limits<uint32_t>::max() / 2 + 1)));
  331. EXPECT_TRUE(has_single_bit(
  332. static_cast<uint64_t>(std::numeric_limits<uint64_t>::max() / 2 + 1)));
  333. }
  334. template <typename T, T arg, T = bit_ceil(arg)>
  335. bool IsBitCeilConstantExpression(int) {
  336. return true;
  337. }
  338. template <typename T, T arg>
  339. bool IsBitCeilConstantExpression(char) {
  340. return false;
  341. }
  342. TEST(IntegralPowersOfTwo, Ceiling) {
  343. #if ABSL_INTERNAL_HAS_CONSTEXPR_CLZ
  344. static_assert(bit_ceil(0u) == 1, "");
  345. static_assert(bit_ceil(1u) == 1, "");
  346. static_assert(bit_ceil(2u) == 2, "");
  347. static_assert(bit_ceil(3u) == 4, "");
  348. static_assert(bit_ceil(4u) == 4, "");
  349. static_assert(bit_ceil(1337u) == 2048, "");
  350. static_assert(bit_ceil(65536u) == 65536, "");
  351. static_assert(bit_ceil(65536u - 1337u) == 65536, "");
  352. static_assert(bit_ceil(uint32_t{0x80000000}) == uint32_t{0x80000000}, "");
  353. static_assert(bit_ceil(uint64_t{0x40000000000}) == uint64_t{0x40000000000},
  354. "");
  355. static_assert(
  356. bit_ceil(uint64_t{0x8000000000000000}) == uint64_t{0x8000000000000000},
  357. "");
  358. EXPECT_TRUE((IsBitCeilConstantExpression<uint8_t, uint8_t{0x0}>(0)));
  359. EXPECT_TRUE((IsBitCeilConstantExpression<uint8_t, uint8_t{0x80}>(0)));
  360. EXPECT_FALSE((IsBitCeilConstantExpression<uint8_t, uint8_t{0x81}>(0)));
  361. EXPECT_FALSE((IsBitCeilConstantExpression<uint8_t, uint8_t{0xff}>(0)));
  362. EXPECT_TRUE((IsBitCeilConstantExpression<uint16_t, uint16_t{0x0}>(0)));
  363. EXPECT_TRUE((IsBitCeilConstantExpression<uint16_t, uint16_t{0x8000}>(0)));
  364. EXPECT_FALSE((IsBitCeilConstantExpression<uint16_t, uint16_t{0x8001}>(0)));
  365. EXPECT_FALSE((IsBitCeilConstantExpression<uint16_t, uint16_t{0xffff}>(0)));
  366. EXPECT_TRUE((IsBitCeilConstantExpression<uint32_t, uint32_t{0x0}>(0)));
  367. EXPECT_TRUE((IsBitCeilConstantExpression<uint32_t, uint32_t{0x80000000}>(0)));
  368. EXPECT_FALSE(
  369. (IsBitCeilConstantExpression<uint32_t, uint32_t{0x80000001}>(0)));
  370. EXPECT_FALSE(
  371. (IsBitCeilConstantExpression<uint32_t, uint32_t{0xffffffff}>(0)));
  372. EXPECT_TRUE((IsBitCeilConstantExpression<uint64_t, uint64_t{0x0}>(0)));
  373. EXPECT_TRUE(
  374. (IsBitCeilConstantExpression<uint64_t, uint64_t{0x8000000000000000}>(0)));
  375. EXPECT_FALSE(
  376. (IsBitCeilConstantExpression<uint64_t, uint64_t{0x8000000000000001}>(0)));
  377. EXPECT_FALSE(
  378. (IsBitCeilConstantExpression<uint64_t, uint64_t{0xffffffffffffffff}>(0)));
  379. #endif
  380. EXPECT_EQ(bit_ceil(0u), 1);
  381. EXPECT_EQ(bit_ceil(1u), 1);
  382. EXPECT_EQ(bit_ceil(2u), 2);
  383. EXPECT_EQ(bit_ceil(3u), 4);
  384. EXPECT_EQ(bit_ceil(4u), 4);
  385. EXPECT_EQ(bit_ceil(1337u), 2048);
  386. EXPECT_EQ(bit_ceil(65536u), 65536);
  387. EXPECT_EQ(bit_ceil(65536u - 1337u), 65536);
  388. EXPECT_EQ(bit_ceil(uint64_t{0x40000000000}), uint64_t{0x40000000000});
  389. }
  390. TEST(IntegralPowersOfTwo, Floor) {
  391. #if ABSL_INTERNAL_HAS_CONSTEXPR_CLZ
  392. static_assert(bit_floor(0u) == 0, "");
  393. static_assert(bit_floor(1u) == 1, "");
  394. static_assert(bit_floor(2u) == 2, "");
  395. static_assert(bit_floor(3u) == 2, "");
  396. static_assert(bit_floor(4u) == 4, "");
  397. static_assert(bit_floor(1337u) == 1024, "");
  398. static_assert(bit_floor(65536u) == 65536, "");
  399. static_assert(bit_floor(65536u - 1337u) == 32768, "");
  400. static_assert(bit_floor(uint64_t{0x40000000000}) == uint64_t{0x40000000000},
  401. "");
  402. #endif
  403. EXPECT_EQ(bit_floor(0u), 0);
  404. EXPECT_EQ(bit_floor(1u), 1);
  405. EXPECT_EQ(bit_floor(2u), 2);
  406. EXPECT_EQ(bit_floor(3u), 2);
  407. EXPECT_EQ(bit_floor(4u), 4);
  408. EXPECT_EQ(bit_floor(1337u), 1024);
  409. EXPECT_EQ(bit_floor(65536u), 65536);
  410. EXPECT_EQ(bit_floor(65536u - 1337u), 32768);
  411. EXPECT_EQ(bit_floor(uint64_t{0x40000000000}), uint64_t{0x40000000000});
  412. for (int i = 0; i < 8; i++) {
  413. uint8_t input = uint8_t{1} << i;
  414. EXPECT_EQ(bit_floor(input), input);
  415. if (i > 0) {
  416. EXPECT_EQ(bit_floor(static_cast<uint8_t>(input + 1)), input);
  417. }
  418. }
  419. for (int i = 0; i < 16; i++) {
  420. uint16_t input = uint16_t{1} << i;
  421. EXPECT_EQ(bit_floor(input), input);
  422. if (i > 0) {
  423. EXPECT_EQ(bit_floor(static_cast<uint16_t>(input + 1)), input);
  424. }
  425. }
  426. for (int i = 0; i < 32; i++) {
  427. uint32_t input = uint32_t{1} << i;
  428. EXPECT_EQ(bit_floor(input), input);
  429. if (i > 0) {
  430. EXPECT_EQ(bit_floor(input + 1), input);
  431. }
  432. }
  433. for (int i = 0; i < 64; i++) {
  434. uint64_t input = uint64_t{1} << i;
  435. EXPECT_EQ(bit_floor(input), input);
  436. if (i > 0) {
  437. EXPECT_EQ(bit_floor(input + 1), input);
  438. }
  439. }
  440. }
  441. TEST(IntegralPowersOfTwo, Width) {
  442. #if ABSL_INTERNAL_HAS_CONSTEXPR_CLZ
  443. static_assert(bit_width(uint8_t{}) == 0, "");
  444. static_assert(bit_width(uint8_t{1}) == 1, "");
  445. static_assert(bit_width(uint8_t{3}) == 2, "");
  446. static_assert(bit_width(static_cast<uint8_t>(-1)) == 8, "");
  447. static_assert(bit_width(uint16_t{}) == 0, "");
  448. static_assert(bit_width(uint16_t{1}) == 1, "");
  449. static_assert(bit_width(uint16_t{3}) == 2, "");
  450. static_assert(bit_width(static_cast<uint16_t>(-1)) == 16, "");
  451. static_assert(bit_width(uint32_t{}) == 0, "");
  452. static_assert(bit_width(uint32_t{1}) == 1, "");
  453. static_assert(bit_width(uint32_t{3}) == 2, "");
  454. static_assert(bit_width(~uint32_t{}) == 32, "");
  455. static_assert(bit_width(uint64_t{}) == 0, "");
  456. static_assert(bit_width(uint64_t{1}) == 1, "");
  457. static_assert(bit_width(uint64_t{3}) == 2, "");
  458. static_assert(bit_width(~uint64_t{}) == 64, "");
  459. #endif
  460. EXPECT_EQ(bit_width(uint8_t{}), 0);
  461. EXPECT_EQ(bit_width(uint8_t{1}), 1);
  462. EXPECT_EQ(bit_width(uint8_t{3}), 2);
  463. EXPECT_EQ(bit_width(static_cast<uint8_t>(-1)), 8);
  464. EXPECT_EQ(bit_width(uint16_t{}), 0);
  465. EXPECT_EQ(bit_width(uint16_t{1}), 1);
  466. EXPECT_EQ(bit_width(uint16_t{3}), 2);
  467. EXPECT_EQ(bit_width(static_cast<uint16_t>(-1)), 16);
  468. EXPECT_EQ(bit_width(uint32_t{}), 0);
  469. EXPECT_EQ(bit_width(uint32_t{1}), 1);
  470. EXPECT_EQ(bit_width(uint32_t{3}), 2);
  471. EXPECT_EQ(bit_width(~uint32_t{}), 32);
  472. EXPECT_EQ(bit_width(uint64_t{}), 0);
  473. EXPECT_EQ(bit_width(uint64_t{1}), 1);
  474. EXPECT_EQ(bit_width(uint64_t{3}), 2);
  475. EXPECT_EQ(bit_width(~uint64_t{}), 64);
  476. for (int i = 0; i < 8; i++) {
  477. EXPECT_EQ(bit_width(static_cast<uint8_t>(uint8_t{1} << i)), i + 1);
  478. }
  479. for (int i = 0; i < 16; i++) {
  480. EXPECT_EQ(bit_width(static_cast<uint16_t>(uint16_t{1} << i)), i + 1);
  481. }
  482. for (int i = 0; i < 32; i++) {
  483. EXPECT_EQ(bit_width(uint32_t{1} << i), i + 1);
  484. }
  485. for (int i = 0; i < 64; i++) {
  486. EXPECT_EQ(bit_width(uint64_t{1} << i), i + 1);
  487. }
  488. }
  489. // On GCC and Clang, anticiapte that implementations will be constexpr
  490. #if defined(__GNUC__)
  491. static_assert(ABSL_INTERNAL_HAS_CONSTEXPR_POPCOUNT,
  492. "popcount should be constexpr");
  493. static_assert(ABSL_INTERNAL_HAS_CONSTEXPR_CLZ, "clz should be constexpr");
  494. static_assert(ABSL_INTERNAL_HAS_CONSTEXPR_CTZ, "ctz should be constexpr");
  495. #endif
  496. } // namespace
  497. ABSL_NAMESPACE_END
  498. } // namespace absl