MathExtrasTest.cpp 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598
  1. //===- unittests/Support/MathExtrasTest.cpp - math utils tests ------------===//
  2. //
  3. // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
  4. // See https://llvm.org/LICENSE.txt for license information.
  5. // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
  6. //
  7. //===----------------------------------------------------------------------===//
  8. #include "llvm/Support/MathExtras.h"
  9. #include "gtest/gtest.h"
  10. using namespace llvm;
  11. namespace {
  12. TEST(MathExtras, countTrailingZeros) {
  13. uint8_t Z8 = 0;
  14. uint16_t Z16 = 0;
  15. uint32_t Z32 = 0;
  16. uint64_t Z64 = 0;
  17. EXPECT_EQ(8u, countTrailingZeros(Z8));
  18. EXPECT_EQ(16u, countTrailingZeros(Z16));
  19. EXPECT_EQ(32u, countTrailingZeros(Z32));
  20. EXPECT_EQ(64u, countTrailingZeros(Z64));
  21. uint8_t NZ8 = 42;
  22. uint16_t NZ16 = 42;
  23. uint32_t NZ32 = 42;
  24. uint64_t NZ64 = 42;
  25. EXPECT_EQ(1u, countTrailingZeros(NZ8));
  26. EXPECT_EQ(1u, countTrailingZeros(NZ16));
  27. EXPECT_EQ(1u, countTrailingZeros(NZ32));
  28. EXPECT_EQ(1u, countTrailingZeros(NZ64));
  29. }
  30. TEST(MathExtras, countLeadingZeros) {
  31. uint8_t Z8 = 0;
  32. uint16_t Z16 = 0;
  33. uint32_t Z32 = 0;
  34. uint64_t Z64 = 0;
  35. EXPECT_EQ(8u, countLeadingZeros(Z8));
  36. EXPECT_EQ(16u, countLeadingZeros(Z16));
  37. EXPECT_EQ(32u, countLeadingZeros(Z32));
  38. EXPECT_EQ(64u, countLeadingZeros(Z64));
  39. uint8_t NZ8 = 42;
  40. uint16_t NZ16 = 42;
  41. uint32_t NZ32 = 42;
  42. uint64_t NZ64 = 42;
  43. EXPECT_EQ(2u, countLeadingZeros(NZ8));
  44. EXPECT_EQ(10u, countLeadingZeros(NZ16));
  45. EXPECT_EQ(26u, countLeadingZeros(NZ32));
  46. EXPECT_EQ(58u, countLeadingZeros(NZ64));
  47. EXPECT_EQ(8u, countLeadingZeros(0x00F000FFu));
  48. EXPECT_EQ(8u, countLeadingZeros(0x00F12345u));
  49. for (unsigned i = 0; i <= 30; ++i) {
  50. EXPECT_EQ(31 - i, countLeadingZeros(1u << i));
  51. }
  52. EXPECT_EQ(8u, countLeadingZeros(0x00F1234500F12345ULL));
  53. EXPECT_EQ(1u, countLeadingZeros(1ULL << 62));
  54. for (unsigned i = 0; i <= 62; ++i) {
  55. EXPECT_EQ(63 - i, countLeadingZeros(1ULL << i));
  56. }
  57. }
  58. TEST(MathExtras, onesMask) {
  59. EXPECT_EQ(0U, maskLeadingOnes<uint8_t>(0));
  60. EXPECT_EQ(0U, maskTrailingOnes<uint8_t>(0));
  61. EXPECT_EQ(0U, maskLeadingOnes<uint16_t>(0));
  62. EXPECT_EQ(0U, maskTrailingOnes<uint16_t>(0));
  63. EXPECT_EQ(0U, maskLeadingOnes<uint32_t>(0));
  64. EXPECT_EQ(0U, maskTrailingOnes<uint32_t>(0));
  65. EXPECT_EQ(0U, maskLeadingOnes<uint64_t>(0));
  66. EXPECT_EQ(0U, maskTrailingOnes<uint64_t>(0));
  67. EXPECT_EQ(0x00000003U, maskTrailingOnes<uint32_t>(2U));
  68. EXPECT_EQ(0xC0000000U, maskLeadingOnes<uint32_t>(2U));
  69. EXPECT_EQ(0x000007FFU, maskTrailingOnes<uint32_t>(11U));
  70. EXPECT_EQ(0xFFE00000U, maskLeadingOnes<uint32_t>(11U));
  71. EXPECT_EQ(0xFFFFFFFFU, maskTrailingOnes<uint32_t>(32U));
  72. EXPECT_EQ(0xFFFFFFFFU, maskLeadingOnes<uint32_t>(32U));
  73. EXPECT_EQ(0xFFFFFFFFFFFFFFFFULL, maskTrailingOnes<uint64_t>(64U));
  74. EXPECT_EQ(0xFFFFFFFFFFFFFFFFULL, maskLeadingOnes<uint64_t>(64U));
  75. EXPECT_EQ(0x0000FFFFFFFFFFFFULL, maskTrailingOnes<uint64_t>(48U));
  76. EXPECT_EQ(0xFFFFFFFFFFFF0000ULL, maskLeadingOnes<uint64_t>(48U));
  77. }
  78. TEST(MathExtras, findFirstSet) {
  79. uint8_t Z8 = 0;
  80. uint16_t Z16 = 0;
  81. uint32_t Z32 = 0;
  82. uint64_t Z64 = 0;
  83. EXPECT_EQ(0xFFULL, findFirstSet(Z8));
  84. EXPECT_EQ(0xFFFFULL, findFirstSet(Z16));
  85. EXPECT_EQ(0xFFFFFFFFULL, findFirstSet(Z32));
  86. EXPECT_EQ(0xFFFFFFFFFFFFFFFFULL, findFirstSet(Z64));
  87. uint8_t NZ8 = 42;
  88. uint16_t NZ16 = 42;
  89. uint32_t NZ32 = 42;
  90. uint64_t NZ64 = 42;
  91. EXPECT_EQ(1u, findFirstSet(NZ8));
  92. EXPECT_EQ(1u, findFirstSet(NZ16));
  93. EXPECT_EQ(1u, findFirstSet(NZ32));
  94. EXPECT_EQ(1u, findFirstSet(NZ64));
  95. }
  96. TEST(MathExtras, findLastSet) {
  97. uint8_t Z8 = 0;
  98. uint16_t Z16 = 0;
  99. uint32_t Z32 = 0;
  100. uint64_t Z64 = 0;
  101. EXPECT_EQ(0xFFULL, findLastSet(Z8));
  102. EXPECT_EQ(0xFFFFULL, findLastSet(Z16));
  103. EXPECT_EQ(0xFFFFFFFFULL, findLastSet(Z32));
  104. EXPECT_EQ(0xFFFFFFFFFFFFFFFFULL, findLastSet(Z64));
  105. uint8_t NZ8 = 42;
  106. uint16_t NZ16 = 42;
  107. uint32_t NZ32 = 42;
  108. uint64_t NZ64 = 42;
  109. EXPECT_EQ(5u, findLastSet(NZ8));
  110. EXPECT_EQ(5u, findLastSet(NZ16));
  111. EXPECT_EQ(5u, findLastSet(NZ32));
  112. EXPECT_EQ(5u, findLastSet(NZ64));
  113. }
  114. TEST(MathExtras, isIntN) {
  115. EXPECT_TRUE(isIntN(16, 32767));
  116. EXPECT_FALSE(isIntN(16, 32768));
  117. }
  118. TEST(MathExtras, isUIntN) {
  119. EXPECT_TRUE(isUIntN(16, 65535));
  120. EXPECT_FALSE(isUIntN(16, 65536));
  121. EXPECT_TRUE(isUIntN(1, 0));
  122. EXPECT_TRUE(isUIntN(6, 63));
  123. }
  124. TEST(MathExtras, maxIntN) {
  125. EXPECT_EQ(32767, maxIntN(16));
  126. EXPECT_EQ(2147483647, maxIntN(32));
  127. EXPECT_EQ(std::numeric_limits<int32_t>::max(), maxIntN(32));
  128. EXPECT_EQ(std::numeric_limits<int64_t>::max(), maxIntN(64));
  129. }
  130. TEST(MathExtras, minIntN) {
  131. EXPECT_EQ(-32768LL, minIntN(16));
  132. EXPECT_EQ(-64LL, minIntN(7));
  133. EXPECT_EQ(std::numeric_limits<int32_t>::min(), minIntN(32));
  134. EXPECT_EQ(std::numeric_limits<int64_t>::min(), minIntN(64));
  135. }
  136. TEST(MathExtras, maxUIntN) {
  137. EXPECT_EQ(0xffffULL, maxUIntN(16));
  138. EXPECT_EQ(0xffffffffULL, maxUIntN(32));
  139. EXPECT_EQ(0xffffffffffffffffULL, maxUIntN(64));
  140. EXPECT_EQ(1ULL, maxUIntN(1));
  141. EXPECT_EQ(0x0fULL, maxUIntN(4));
  142. }
  143. TEST(MathExtras, reverseBits) {
  144. uint8_t NZ8 = 42;
  145. uint16_t NZ16 = 42;
  146. uint32_t NZ32 = 42;
  147. uint64_t NZ64 = 42;
  148. EXPECT_EQ(0x54ULL, reverseBits(NZ8));
  149. EXPECT_EQ(0x5400ULL, reverseBits(NZ16));
  150. EXPECT_EQ(0x54000000ULL, reverseBits(NZ32));
  151. EXPECT_EQ(0x5400000000000000ULL, reverseBits(NZ64));
  152. }
  153. TEST(MathExtras, isPowerOf2_32) {
  154. EXPECT_FALSE(isPowerOf2_32(0));
  155. EXPECT_TRUE(isPowerOf2_32(1 << 6));
  156. EXPECT_TRUE(isPowerOf2_32(1 << 12));
  157. EXPECT_FALSE(isPowerOf2_32((1 << 19) + 3));
  158. EXPECT_FALSE(isPowerOf2_32(0xABCDEF0));
  159. }
  160. TEST(MathExtras, isPowerOf2_64) {
  161. EXPECT_FALSE(isPowerOf2_64(0));
  162. EXPECT_TRUE(isPowerOf2_64(1LL << 46));
  163. EXPECT_TRUE(isPowerOf2_64(1LL << 12));
  164. EXPECT_FALSE(isPowerOf2_64((1LL << 53) + 3));
  165. EXPECT_FALSE(isPowerOf2_64(0xABCDEF0ABCDEF0LL));
  166. }
  167. TEST(MathExtras, PowerOf2Ceil) {
  168. EXPECT_EQ(0U, PowerOf2Ceil(0U));
  169. EXPECT_EQ(8U, PowerOf2Ceil(8U));
  170. EXPECT_EQ(8U, PowerOf2Ceil(7U));
  171. }
  172. TEST(MathExtras, PowerOf2Floor) {
  173. EXPECT_EQ(0U, PowerOf2Floor(0U));
  174. EXPECT_EQ(8U, PowerOf2Floor(8U));
  175. EXPECT_EQ(4U, PowerOf2Floor(7U));
  176. }
  177. TEST(MathExtras, ByteSwap_32) {
  178. EXPECT_EQ(0x44332211u, ByteSwap_32(0x11223344));
  179. EXPECT_EQ(0xDDCCBBAAu, ByteSwap_32(0xAABBCCDD));
  180. }
  181. TEST(MathExtras, ByteSwap_64) {
  182. EXPECT_EQ(0x8877665544332211ULL, ByteSwap_64(0x1122334455667788LL));
  183. EXPECT_EQ(0x1100FFEEDDCCBBAAULL, ByteSwap_64(0xAABBCCDDEEFF0011LL));
  184. }
  185. TEST(MathExtras, countLeadingOnes) {
  186. for (int i = 30; i >= 0; --i) {
  187. // Start with all ones and unset some bit.
  188. EXPECT_EQ(31u - i, countLeadingOnes(0xFFFFFFFF ^ (1 << i)));
  189. }
  190. for (int i = 62; i >= 0; --i) {
  191. // Start with all ones and unset some bit.
  192. EXPECT_EQ(63u - i, countLeadingOnes(0xFFFFFFFFFFFFFFFFULL ^ (1LL << i)));
  193. }
  194. for (int i = 30; i >= 0; --i) {
  195. // Start with all ones and unset some bit.
  196. EXPECT_EQ(31u - i, countLeadingOnes(0xFFFFFFFF ^ (1 << i)));
  197. }
  198. }
  199. TEST(MathExtras, FloatBits) {
  200. static const float kValue = 5632.34f;
  201. EXPECT_FLOAT_EQ(kValue, BitsToFloat(FloatToBits(kValue)));
  202. }
  203. TEST(MathExtras, DoubleBits) {
  204. static const double kValue = 87987234.983498;
  205. EXPECT_DOUBLE_EQ(kValue, BitsToDouble(DoubleToBits(kValue)));
  206. }
  207. TEST(MathExtras, MinAlign) {
  208. EXPECT_EQ(1u, MinAlign(2, 3));
  209. EXPECT_EQ(2u, MinAlign(2, 4));
  210. EXPECT_EQ(1u, MinAlign(17, 64));
  211. EXPECT_EQ(256u, MinAlign(256, 512));
  212. }
  213. TEST(MathExtras, NextPowerOf2) {
  214. EXPECT_EQ(4u, NextPowerOf2(3));
  215. EXPECT_EQ(16u, NextPowerOf2(15));
  216. EXPECT_EQ(256u, NextPowerOf2(128));
  217. }
  218. TEST(MathExtras, alignTo) {
  219. EXPECT_EQ(8u, alignTo(5, 8));
  220. EXPECT_EQ(24u, alignTo(17, 8));
  221. EXPECT_EQ(0u, alignTo(~0LL, 8));
  222. EXPECT_EQ(7u, alignTo(5, 8, 7));
  223. EXPECT_EQ(17u, alignTo(17, 8, 1));
  224. EXPECT_EQ(3u, alignTo(~0LL, 8, 3));
  225. EXPECT_EQ(552u, alignTo(321, 255, 42));
  226. }
  227. template<typename T>
  228. void SaturatingAddTestHelper()
  229. {
  230. const T Max = std::numeric_limits<T>::max();
  231. bool ResultOverflowed;
  232. EXPECT_EQ(T(3), SaturatingAdd(T(1), T(2)));
  233. EXPECT_EQ(T(3), SaturatingAdd(T(1), T(2), &ResultOverflowed));
  234. EXPECT_FALSE(ResultOverflowed);
  235. EXPECT_EQ(Max, SaturatingAdd(Max, T(1)));
  236. EXPECT_EQ(Max, SaturatingAdd(Max, T(1), &ResultOverflowed));
  237. EXPECT_TRUE(ResultOverflowed);
  238. EXPECT_EQ(Max, SaturatingAdd(T(1), T(Max - 1)));
  239. EXPECT_EQ(Max, SaturatingAdd(T(1), T(Max - 1), &ResultOverflowed));
  240. EXPECT_FALSE(ResultOverflowed);
  241. EXPECT_EQ(Max, SaturatingAdd(T(1), Max));
  242. EXPECT_EQ(Max, SaturatingAdd(T(1), Max, &ResultOverflowed));
  243. EXPECT_TRUE(ResultOverflowed);
  244. EXPECT_EQ(Max, SaturatingAdd(Max, Max));
  245. EXPECT_EQ(Max, SaturatingAdd(Max, Max, &ResultOverflowed));
  246. EXPECT_TRUE(ResultOverflowed);
  247. }
  248. TEST(MathExtras, SaturatingAdd) {
  249. SaturatingAddTestHelper<uint8_t>();
  250. SaturatingAddTestHelper<uint16_t>();
  251. SaturatingAddTestHelper<uint32_t>();
  252. SaturatingAddTestHelper<uint64_t>();
  253. }
  254. template<typename T>
  255. void SaturatingMultiplyTestHelper()
  256. {
  257. const T Max = std::numeric_limits<T>::max();
  258. bool ResultOverflowed;
  259. // Test basic multiplication.
  260. EXPECT_EQ(T(6), SaturatingMultiply(T(2), T(3)));
  261. EXPECT_EQ(T(6), SaturatingMultiply(T(2), T(3), &ResultOverflowed));
  262. EXPECT_FALSE(ResultOverflowed);
  263. EXPECT_EQ(T(6), SaturatingMultiply(T(3), T(2)));
  264. EXPECT_EQ(T(6), SaturatingMultiply(T(3), T(2), &ResultOverflowed));
  265. EXPECT_FALSE(ResultOverflowed);
  266. // Test multiplication by zero.
  267. EXPECT_EQ(T(0), SaturatingMultiply(T(0), T(0)));
  268. EXPECT_EQ(T(0), SaturatingMultiply(T(0), T(0), &ResultOverflowed));
  269. EXPECT_FALSE(ResultOverflowed);
  270. EXPECT_EQ(T(0), SaturatingMultiply(T(1), T(0)));
  271. EXPECT_EQ(T(0), SaturatingMultiply(T(1), T(0), &ResultOverflowed));
  272. EXPECT_FALSE(ResultOverflowed);
  273. EXPECT_EQ(T(0), SaturatingMultiply(T(0), T(1)));
  274. EXPECT_EQ(T(0), SaturatingMultiply(T(0), T(1), &ResultOverflowed));
  275. EXPECT_FALSE(ResultOverflowed);
  276. EXPECT_EQ(T(0), SaturatingMultiply(Max, T(0)));
  277. EXPECT_EQ(T(0), SaturatingMultiply(Max, T(0), &ResultOverflowed));
  278. EXPECT_FALSE(ResultOverflowed);
  279. EXPECT_EQ(T(0), SaturatingMultiply(T(0), Max));
  280. EXPECT_EQ(T(0), SaturatingMultiply(T(0), Max, &ResultOverflowed));
  281. EXPECT_FALSE(ResultOverflowed);
  282. // Test multiplication by maximum value.
  283. EXPECT_EQ(Max, SaturatingMultiply(Max, T(2)));
  284. EXPECT_EQ(Max, SaturatingMultiply(Max, T(2), &ResultOverflowed));
  285. EXPECT_TRUE(ResultOverflowed);
  286. EXPECT_EQ(Max, SaturatingMultiply(T(2), Max));
  287. EXPECT_EQ(Max, SaturatingMultiply(T(2), Max, &ResultOverflowed));
  288. EXPECT_TRUE(ResultOverflowed);
  289. EXPECT_EQ(Max, SaturatingMultiply(Max, Max));
  290. EXPECT_EQ(Max, SaturatingMultiply(Max, Max, &ResultOverflowed));
  291. EXPECT_TRUE(ResultOverflowed);
  292. // Test interesting boundary conditions for algorithm -
  293. // ((1 << A) - 1) * ((1 << B) + K) for K in [-1, 0, 1]
  294. // and A + B == std::numeric_limits<T>::digits.
  295. // We expect overflow iff A > B and K = 1.
  296. const int Digits = std::numeric_limits<T>::digits;
  297. for (int A = 1, B = Digits - 1; B >= 1; ++A, --B) {
  298. for (int K = -1; K <= 1; ++K) {
  299. T X = (T(1) << A) - T(1);
  300. T Y = (T(1) << B) + K;
  301. bool OverflowExpected = A > B && K == 1;
  302. if(OverflowExpected) {
  303. EXPECT_EQ(Max, SaturatingMultiply(X, Y));
  304. EXPECT_EQ(Max, SaturatingMultiply(X, Y, &ResultOverflowed));
  305. EXPECT_TRUE(ResultOverflowed);
  306. } else {
  307. EXPECT_EQ(X * Y, SaturatingMultiply(X, Y));
  308. EXPECT_EQ(X * Y, SaturatingMultiply(X, Y, &ResultOverflowed));
  309. EXPECT_FALSE(ResultOverflowed);
  310. }
  311. }
  312. }
  313. }
  314. TEST(MathExtras, SaturatingMultiply) {
  315. SaturatingMultiplyTestHelper<uint8_t>();
  316. SaturatingMultiplyTestHelper<uint16_t>();
  317. SaturatingMultiplyTestHelper<uint32_t>();
  318. SaturatingMultiplyTestHelper<uint64_t>();
  319. }
  320. template<typename T>
  321. void SaturatingMultiplyAddTestHelper()
  322. {
  323. const T Max = std::numeric_limits<T>::max();
  324. bool ResultOverflowed;
  325. // Test basic multiply-add.
  326. EXPECT_EQ(T(16), SaturatingMultiplyAdd(T(2), T(3), T(10)));
  327. EXPECT_EQ(T(16), SaturatingMultiplyAdd(T(2), T(3), T(10), &ResultOverflowed));
  328. EXPECT_FALSE(ResultOverflowed);
  329. // Test multiply overflows, add doesn't overflow
  330. EXPECT_EQ(Max, SaturatingMultiplyAdd(Max, Max, T(0), &ResultOverflowed));
  331. EXPECT_TRUE(ResultOverflowed);
  332. // Test multiply doesn't overflow, add overflows
  333. EXPECT_EQ(Max, SaturatingMultiplyAdd(T(1), T(1), Max, &ResultOverflowed));
  334. EXPECT_TRUE(ResultOverflowed);
  335. // Test multiply-add with Max as operand
  336. EXPECT_EQ(Max, SaturatingMultiplyAdd(T(1), T(1), Max, &ResultOverflowed));
  337. EXPECT_TRUE(ResultOverflowed);
  338. EXPECT_EQ(Max, SaturatingMultiplyAdd(T(1), Max, T(1), &ResultOverflowed));
  339. EXPECT_TRUE(ResultOverflowed);
  340. EXPECT_EQ(Max, SaturatingMultiplyAdd(Max, Max, T(1), &ResultOverflowed));
  341. EXPECT_TRUE(ResultOverflowed);
  342. EXPECT_EQ(Max, SaturatingMultiplyAdd(Max, Max, Max, &ResultOverflowed));
  343. EXPECT_TRUE(ResultOverflowed);
  344. // Test multiply-add with 0 as operand
  345. EXPECT_EQ(T(1), SaturatingMultiplyAdd(T(1), T(1), T(0), &ResultOverflowed));
  346. EXPECT_FALSE(ResultOverflowed);
  347. EXPECT_EQ(T(1), SaturatingMultiplyAdd(T(1), T(0), T(1), &ResultOverflowed));
  348. EXPECT_FALSE(ResultOverflowed);
  349. EXPECT_EQ(T(1), SaturatingMultiplyAdd(T(0), T(0), T(1), &ResultOverflowed));
  350. EXPECT_FALSE(ResultOverflowed);
  351. EXPECT_EQ(T(0), SaturatingMultiplyAdd(T(0), T(0), T(0), &ResultOverflowed));
  352. EXPECT_FALSE(ResultOverflowed);
  353. }
  354. TEST(MathExtras, SaturatingMultiplyAdd) {
  355. SaturatingMultiplyAddTestHelper<uint8_t>();
  356. SaturatingMultiplyAddTestHelper<uint16_t>();
  357. SaturatingMultiplyAddTestHelper<uint32_t>();
  358. SaturatingMultiplyAddTestHelper<uint64_t>();
  359. }
  360. TEST(MathExtras, IsShiftedUInt) {
  361. EXPECT_TRUE((isShiftedUInt<1, 0>(0)));
  362. EXPECT_TRUE((isShiftedUInt<1, 0>(1)));
  363. EXPECT_FALSE((isShiftedUInt<1, 0>(2)));
  364. EXPECT_FALSE((isShiftedUInt<1, 0>(3)));
  365. EXPECT_FALSE((isShiftedUInt<1, 0>(0x8000000000000000)));
  366. EXPECT_TRUE((isShiftedUInt<1, 63>(0x8000000000000000)));
  367. EXPECT_TRUE((isShiftedUInt<2, 62>(0xC000000000000000)));
  368. EXPECT_FALSE((isShiftedUInt<2, 62>(0xE000000000000000)));
  369. // 0x201 is ten bits long and has a 1 in the MSB and LSB.
  370. EXPECT_TRUE((isShiftedUInt<10, 5>(uint64_t(0x201) << 5)));
  371. EXPECT_FALSE((isShiftedUInt<10, 5>(uint64_t(0x201) << 4)));
  372. EXPECT_FALSE((isShiftedUInt<10, 5>(uint64_t(0x201) << 6)));
  373. }
  374. TEST(MathExtras, IsShiftedInt) {
  375. EXPECT_TRUE((isShiftedInt<1, 0>(0)));
  376. EXPECT_TRUE((isShiftedInt<1, 0>(-1)));
  377. EXPECT_FALSE((isShiftedInt<1, 0>(2)));
  378. EXPECT_FALSE((isShiftedInt<1, 0>(3)));
  379. EXPECT_FALSE((isShiftedInt<1, 0>(0x8000000000000000)));
  380. EXPECT_TRUE((isShiftedInt<1, 63>(0x8000000000000000)));
  381. EXPECT_TRUE((isShiftedInt<2, 62>(0xC000000000000000)));
  382. EXPECT_FALSE((isShiftedInt<2, 62>(0xE000000000000000)));
  383. // 0x201 is ten bits long and has a 1 in the MSB and LSB.
  384. EXPECT_TRUE((isShiftedInt<11, 5>(int64_t(0x201) << 5)));
  385. EXPECT_FALSE((isShiftedInt<11, 5>(int64_t(0x201) << 3)));
  386. EXPECT_FALSE((isShiftedInt<11, 5>(int64_t(0x201) << 6)));
  387. EXPECT_TRUE((isShiftedInt<11, 5>(-(int64_t(0x201) << 5))));
  388. EXPECT_FALSE((isShiftedInt<11, 5>(-(int64_t(0x201) << 3))));
  389. EXPECT_FALSE((isShiftedInt<11, 5>(-(int64_t(0x201) << 6))));
  390. EXPECT_TRUE((isShiftedInt<6, 10>(-(int64_t(1) << 15))));
  391. EXPECT_FALSE((isShiftedInt<6, 10>(int64_t(1) << 15)));
  392. }
  393. template <typename T>
  394. class OverflowTest : public ::testing::Test { };
  395. using OverflowTestTypes = ::testing::Types<signed char, short, int, long,
  396. long long>;
  397. TYPED_TEST_CASE(OverflowTest, OverflowTestTypes);
  398. TYPED_TEST(OverflowTest, AddNoOverflow) {
  399. TypeParam Result;
  400. EXPECT_FALSE(AddOverflow<TypeParam>(1, 2, Result));
  401. EXPECT_EQ(Result, TypeParam(3));
  402. }
  403. TYPED_TEST(OverflowTest, AddOverflowToNegative) {
  404. TypeParam Result;
  405. auto MaxValue = std::numeric_limits<TypeParam>::max();
  406. EXPECT_TRUE(AddOverflow<TypeParam>(MaxValue, MaxValue, Result));
  407. EXPECT_EQ(Result, TypeParam(-2));
  408. }
  409. TYPED_TEST(OverflowTest, AddOverflowToMin) {
  410. TypeParam Result;
  411. auto MaxValue = std::numeric_limits<TypeParam>::max();
  412. EXPECT_TRUE(AddOverflow<TypeParam>(MaxValue, TypeParam(1), Result));
  413. EXPECT_EQ(Result, std::numeric_limits<TypeParam>::min());
  414. }
  415. TYPED_TEST(OverflowTest, AddOverflowToZero) {
  416. TypeParam Result;
  417. auto MinValue = std::numeric_limits<TypeParam>::min();
  418. EXPECT_TRUE(AddOverflow<TypeParam>(MinValue, MinValue, Result));
  419. EXPECT_EQ(Result, TypeParam(0));
  420. }
  421. TYPED_TEST(OverflowTest, AddOverflowToMax) {
  422. TypeParam Result;
  423. auto MinValue = std::numeric_limits<TypeParam>::min();
  424. EXPECT_TRUE(AddOverflow<TypeParam>(MinValue, TypeParam(-1), Result));
  425. EXPECT_EQ(Result, std::numeric_limits<TypeParam>::max());
  426. }
  427. TYPED_TEST(OverflowTest, SubNoOverflow) {
  428. TypeParam Result;
  429. EXPECT_FALSE(SubOverflow<TypeParam>(1, 2, Result));
  430. EXPECT_EQ(Result, TypeParam(-1));
  431. }
  432. TYPED_TEST(OverflowTest, SubOverflowToMax) {
  433. TypeParam Result;
  434. auto MinValue = std::numeric_limits<TypeParam>::min();
  435. EXPECT_TRUE(SubOverflow<TypeParam>(0, MinValue, Result));
  436. EXPECT_EQ(Result, MinValue);
  437. }
  438. TYPED_TEST(OverflowTest, SubOverflowToMin) {
  439. TypeParam Result;
  440. auto MinValue = std::numeric_limits<TypeParam>::min();
  441. EXPECT_TRUE(SubOverflow<TypeParam>(0, MinValue, Result));
  442. EXPECT_EQ(Result, MinValue);
  443. }
  444. TYPED_TEST(OverflowTest, SubOverflowToNegative) {
  445. TypeParam Result;
  446. auto MaxValue = std::numeric_limits<TypeParam>::max();
  447. auto MinValue = std::numeric_limits<TypeParam>::min();
  448. EXPECT_TRUE(SubOverflow<TypeParam>(MaxValue, MinValue, Result));
  449. EXPECT_EQ(Result, TypeParam(-1));
  450. }
  451. TYPED_TEST(OverflowTest, SubOverflowToPositive) {
  452. TypeParam Result;
  453. auto MaxValue = std::numeric_limits<TypeParam>::max();
  454. auto MinValue = std::numeric_limits<TypeParam>::min();
  455. EXPECT_TRUE(SubOverflow<TypeParam>(MinValue, MaxValue, Result));
  456. EXPECT_EQ(Result, TypeParam(1));
  457. }
  458. TYPED_TEST(OverflowTest, MulNoOverflow) {
  459. TypeParam Result;
  460. EXPECT_FALSE(MulOverflow<TypeParam>(1, 2, Result));
  461. EXPECT_EQ(Result, 2);
  462. EXPECT_FALSE(MulOverflow<TypeParam>(-1, 3, Result));
  463. EXPECT_EQ(Result, -3);
  464. EXPECT_FALSE(MulOverflow<TypeParam>(4, -2, Result));
  465. EXPECT_EQ(Result, -8);
  466. EXPECT_FALSE(MulOverflow<TypeParam>(-6, -5, Result));
  467. EXPECT_EQ(Result, 30);
  468. }
  469. TYPED_TEST(OverflowTest, MulNoOverflowToMax) {
  470. TypeParam Result;
  471. auto MaxValue = std::numeric_limits<TypeParam>::max();
  472. auto MinValue = std::numeric_limits<TypeParam>::min();
  473. EXPECT_FALSE(MulOverflow<TypeParam>(MinValue + 1, -1, Result));
  474. EXPECT_EQ(Result, MaxValue);
  475. }
  476. TYPED_TEST(OverflowTest, MulOverflowToMin) {
  477. TypeParam Result;
  478. auto MinValue = std::numeric_limits<TypeParam>::min();
  479. EXPECT_TRUE(MulOverflow<TypeParam>(MinValue, -1, Result));
  480. EXPECT_EQ(Result, MinValue);
  481. }
  482. TYPED_TEST(OverflowTest, MulOverflowMax) {
  483. TypeParam Result;
  484. auto MinValue = std::numeric_limits<TypeParam>::min();
  485. auto MaxValue = std::numeric_limits<TypeParam>::max();
  486. EXPECT_TRUE(MulOverflow<TypeParam>(MinValue, MinValue, Result));
  487. EXPECT_EQ(Result, 0);
  488. EXPECT_TRUE(MulOverflow<TypeParam>(MaxValue, MaxValue, Result));
  489. EXPECT_EQ(Result, 1);
  490. }
  491. TYPED_TEST(OverflowTest, MulResultZero) {
  492. TypeParam Result;
  493. EXPECT_FALSE(MulOverflow<TypeParam>(4, 0, Result));
  494. EXPECT_EQ(Result, TypeParam(0));
  495. EXPECT_FALSE(MulOverflow<TypeParam>(-5, 0, Result));
  496. EXPECT_EQ(Result, TypeParam(0));
  497. EXPECT_FALSE(MulOverflow<TypeParam>(0, 5, Result));
  498. EXPECT_EQ(Result, TypeParam(0));
  499. EXPECT_FALSE(MulOverflow<TypeParam>(0, -5, Result));
  500. EXPECT_EQ(Result, TypeParam(0));
  501. }
  502. } // namespace