unittests/ADT/APIntTest.cpp

   1 //===- llvm/unittest/ADT/APInt.cpp - APInt unit tests ---------------------===//
   2 //
   3 //                     The LLVM Compiler Infrastructure
   4 //
   5 // This file is distributed under the University of Illinois Open Source
   6 // License. See LICENSE.TXT for details.
   7 //
   8 //===----------------------------------------------------------------------===//
   9
  10 #include "llvm/ADT/APInt.h"
  11 #include "llvm/ADT/ArrayRef.h"
  12 #include "llvm/ADT/SmallString.h"
  13 #include "gtest/gtest.h"
  14 #include <array>
  15
  16 using namespace llvm;
  17
  18 namespace {
  19
  20 TEST(APIntTest, ValueInit) {
  21   APInt Zero = APInt();
  22   EXPECT_TRUE(!Zero);
  23   EXPECT_TRUE(!Zero.zext(64));
  24   EXPECT_TRUE(!Zero.sext(64));
  25 }
  26
  27 // Test that APInt shift left works when bitwidth > 64 and shiftamt == 0
  28 TEST(APIntTest, ShiftLeftByZero) {
  29   APInt One = APInt::getNullValue(65) + 1;
  30   APInt Shl = One.shl(0);
  31   EXPECT_TRUE(Shl[0]);
  32   EXPECT_FALSE(Shl[1]);
  33 }
  34
  35 TEST(APIntTest, i64_ArithmeticRightShiftNegative) {
  36   const APInt neg_one(64, static_cast<uint64_t>(-1), true);
  37   EXPECT_EQ(neg_one, neg_one.ashr(7));
  38 }
  39
  40 TEST(APIntTest, i128_NegativeCount) {
  41   APInt Minus3(128, static_cast<uint64_t>(-3), true);
  42   EXPECT_EQ(126u, Minus3.countLeadingOnes());
  43   EXPECT_EQ(-3, Minus3.getSExtValue());
  44
  45   APInt Minus1(128, static_cast<uint64_t>(-1), true);
  46   EXPECT_EQ(0u, Minus1.countLeadingZeros());
  47   EXPECT_EQ(128u, Minus1.countLeadingOnes());
  48   EXPECT_EQ(128u, Minus1.getActiveBits());
  49   EXPECT_EQ(0u, Minus1.countTrailingZeros());
  50   EXPECT_EQ(128u, Minus1.countTrailingOnes());
  51   EXPECT_EQ(128u, Minus1.countPopulation());
  52   EXPECT_EQ(-1, Minus1.getSExtValue());
  53 }
  54
  55 TEST(APIntTest, i33_Count) {
  56   APInt i33minus2(33, static_cast<uint64_t>(-2), true);
  57   EXPECT_EQ(0u, i33minus2.countLeadingZeros());
  58   EXPECT_EQ(32u, i33minus2.countLeadingOnes());
  59   EXPECT_EQ(33u, i33minus2.getActiveBits());
  60   EXPECT_EQ(1u, i33minus2.countTrailingZeros());
  61   EXPECT_EQ(32u, i33minus2.countPopulation());
  62   EXPECT_EQ(-2, i33minus2.getSExtValue());
  63   EXPECT_EQ(((uint64_t)-2)&((1ull<<33) -1), i33minus2.getZExtValue());
  64 }
  65
  66 TEST(APIntTest, i61_Count) {
  67   APInt i61(61, 1 << 15);
  68   EXPECT_EQ(45u, i61.countLeadingZeros());
  69   EXPECT_EQ(0u, i61.countLeadingOnes());
  70   EXPECT_EQ(16u, i61.getActiveBits());
  71   EXPECT_EQ(15u, i61.countTrailingZeros());
  72   EXPECT_EQ(1u, i61.countPopulation());
  73   EXPECT_EQ(static_cast<int64_t>(1 << 15), i61.getSExtValue());
  74   EXPECT_EQ(static_cast<uint64_t>(1 << 15), i61.getZExtValue());
  75
  76   i61.setBits(8, 19);
  77   EXPECT_EQ(42u, i61.countLeadingZeros());
  78   EXPECT_EQ(0u, i61.countLeadingOnes());
  79   EXPECT_EQ(19u, i61.getActiveBits());
  80   EXPECT_EQ(8u, i61.countTrailingZeros());
  81   EXPECT_EQ(11u, i61.countPopulation());
  82   EXPECT_EQ(static_cast<int64_t>((1 << 19) - (1 << 8)), i61.getSExtValue());
  83   EXPECT_EQ(static_cast<uint64_t>((1 << 19) - (1 << 8)), i61.getZExtValue());
  84 }
  85
  86 TEST(APIntTest, i65_Count) {
  87   APInt i65(65, 0, true);
  88   EXPECT_EQ(65u, i65.countLeadingZeros());
  89   EXPECT_EQ(0u, i65.countLeadingOnes());
  90   EXPECT_EQ(0u, i65.getActiveBits());
  91   EXPECT_EQ(1u, i65.getActiveWords());
  92   EXPECT_EQ(65u, i65.countTrailingZeros());
  93   EXPECT_EQ(0u, i65.countPopulation());
  94
  95   APInt i65minus(65, 0, true);
  96   i65minus.setBit(64);
  97   EXPECT_EQ(0u, i65minus.countLeadingZeros());
  98   EXPECT_EQ(1u, i65minus.countLeadingOnes());
  99   EXPECT_EQ(65u, i65minus.getActiveBits());
 100   EXPECT_EQ(64u, i65minus.countTrailingZeros());
 101   EXPECT_EQ(1u, i65minus.countPopulation());
 102 }
 103
 104 TEST(APIntTest, i128_PositiveCount) {
 105   APInt u128max = APInt::getAllOnesValue(128);
 106   EXPECT_EQ(128u, u128max.countLeadingOnes());
 107   EXPECT_EQ(0u, u128max.countLeadingZeros());
 108   EXPECT_EQ(128u, u128max.getActiveBits());
 109   EXPECT_EQ(0u, u128max.countTrailingZeros());
 110   EXPECT_EQ(128u, u128max.countTrailingOnes());
 111   EXPECT_EQ(128u, u128max.countPopulation());
 112
 113   APInt u64max(128, static_cast<uint64_t>(-1), false);
 114   EXPECT_EQ(64u, u64max.countLeadingZeros());
 115   EXPECT_EQ(0u, u64max.countLeadingOnes());
 116   EXPECT_EQ(64u, u64max.getActiveBits());
 117   EXPECT_EQ(0u, u64max.countTrailingZeros());
 118   EXPECT_EQ(64u, u64max.countTrailingOnes());
 119   EXPECT_EQ(64u, u64max.countPopulation());
 120   EXPECT_EQ((uint64_t)~0ull, u64max.getZExtValue());
 121
 122   APInt zero(128, 0, true);
 123   EXPECT_EQ(128u, zero.countLeadingZeros());
 124   EXPECT_EQ(0u, zero.countLeadingOnes());
 125   EXPECT_EQ(0u, zero.getActiveBits());
 126   EXPECT_EQ(128u, zero.countTrailingZeros());
 127   EXPECT_EQ(0u, zero.countTrailingOnes());
 128   EXPECT_EQ(0u, zero.countPopulation());
 129   EXPECT_EQ(0u, zero.getSExtValue());
 130   EXPECT_EQ(0u, zero.getZExtValue());
 131
 132   APInt one(128, 1, true);
 133   EXPECT_EQ(127u, one.countLeadingZeros());
 134   EXPECT_EQ(0u, one.countLeadingOnes());
 135   EXPECT_EQ(1u, one.getActiveBits());
 136   EXPECT_EQ(0u, one.countTrailingZeros());
 137   EXPECT_EQ(1u, one.countTrailingOnes());
 138   EXPECT_EQ(1u, one.countPopulation());
 139   EXPECT_EQ(1, one.getSExtValue());
 140   EXPECT_EQ(1u, one.getZExtValue());
 141
 142   APInt s128(128, 2, true);
 143   EXPECT_EQ(126u, s128.countLeadingZeros());
 144   EXPECT_EQ(0u, s128.countLeadingOnes());
 145   EXPECT_EQ(2u, s128.getActiveBits());
 146   EXPECT_EQ(1u, s128.countTrailingZeros());
 147   EXPECT_EQ(0u, s128.countTrailingOnes());
 148   EXPECT_EQ(1u, s128.countPopulation());
 149   EXPECT_EQ(2, s128.getSExtValue());
 150   EXPECT_EQ(2u, s128.getZExtValue());
 151
 152   // NOP Test
 153   s128.setBits(42, 42);
 154   EXPECT_EQ(126u, s128.countLeadingZeros());
 155   EXPECT_EQ(0u, s128.countLeadingOnes());
 156   EXPECT_EQ(2u, s128.getActiveBits());
 157   EXPECT_EQ(1u, s128.countTrailingZeros());
 158   EXPECT_EQ(0u, s128.countTrailingOnes());
 159   EXPECT_EQ(1u, s128.countPopulation());
 160   EXPECT_EQ(2, s128.getSExtValue());
 161   EXPECT_EQ(2u, s128.getZExtValue());
 162
 163   s128.setBits(3, 32);
 164   EXPECT_EQ(96u, s128.countLeadingZeros());
 165   EXPECT_EQ(0u, s128.countLeadingOnes());
 166   EXPECT_EQ(32u, s128.getActiveBits());
 167   EXPECT_EQ(33u, s128.getMinSignedBits());
 168   EXPECT_EQ(1u, s128.countTrailingZeros());
 169   EXPECT_EQ(0u, s128.countTrailingOnes());
 170   EXPECT_EQ(30u, s128.countPopulation());
 171   EXPECT_EQ(static_cast<uint32_t>((~0u << 3) | 2), s128.getZExtValue());
 172
 173   s128.setBits(62, 128);
 174   EXPECT_EQ(0u, s128.countLeadingZeros());
 175   EXPECT_EQ(66u, s128.countLeadingOnes());
 176   EXPECT_EQ(128u, s128.getActiveBits());
 177   EXPECT_EQ(63u, s128.getMinSignedBits());
 178   EXPECT_EQ(1u, s128.countTrailingZeros());
 179   EXPECT_EQ(0u, s128.countTrailingOnes());
 180   EXPECT_EQ(96u, s128.countPopulation());
 181   EXPECT_EQ(static_cast<int64_t>((3ull << 62) |
 182                                  static_cast<uint32_t>((~0u << 3) | 2)),
 183             s128.getSExtValue());
 184 }
 185
 186 TEST(APIntTest, i256) {
 187   APInt s256(256, 15, true);
 188   EXPECT_EQ(252u, s256.countLeadingZeros());
 189   EXPECT_EQ(0u, s256.countLeadingOnes());
 190   EXPECT_EQ(4u, s256.getActiveBits());
 191   EXPECT_EQ(0u, s256.countTrailingZeros());
 192   EXPECT_EQ(4u, s256.countTrailingOnes());
 193   EXPECT_EQ(4u, s256.countPopulation());
 194   EXPECT_EQ(15, s256.getSExtValue());
 195   EXPECT_EQ(15u, s256.getZExtValue());
 196
 197   s256.setBits(62, 66);
 198   EXPECT_EQ(190u, s256.countLeadingZeros());
 199   EXPECT_EQ(0u, s256.countLeadingOnes());
 200   EXPECT_EQ(66u, s256.getActiveBits());
 201   EXPECT_EQ(67u, s256.getMinSignedBits());
 202   EXPECT_EQ(0u, s256.countTrailingZeros());
 203   EXPECT_EQ(4u, s256.countTrailingOnes());
 204   EXPECT_EQ(8u, s256.countPopulation());
 205
 206   s256.setBits(60, 256);
 207   EXPECT_EQ(0u, s256.countLeadingZeros());
 208   EXPECT_EQ(196u, s256.countLeadingOnes());
 209   EXPECT_EQ(256u, s256.getActiveBits());
 210   EXPECT_EQ(61u, s256.getMinSignedBits());
 211   EXPECT_EQ(0u, s256.countTrailingZeros());
 212   EXPECT_EQ(4u, s256.countTrailingOnes());
 213   EXPECT_EQ(200u, s256.countPopulation());
 214   EXPECT_EQ(static_cast<int64_t>((~0ull << 60) | 15), s256.getSExtValue());
 215 }
 216
 217 TEST(APIntTest, i1) {
 218   const APInt neg_two(1, static_cast<uint64_t>(-2), true);
 219   const APInt neg_one(1, static_cast<uint64_t>(-1), true);
 220   const APInt zero(1, 0);
 221   const APInt one(1, 1);
 222   const APInt two(1, 2);
 223
 224   EXPECT_EQ(0, neg_two.getSExtValue());
 225   EXPECT_EQ(-1, neg_one.getSExtValue());
 226   EXPECT_EQ(1u, neg_one.getZExtValue());
 227   EXPECT_EQ(0u, zero.getZExtValue());
 228   EXPECT_EQ(-1, one.getSExtValue());
 229   EXPECT_EQ(1u, one.getZExtValue());
 230   EXPECT_EQ(0u, two.getZExtValue());
 231   EXPECT_EQ(0, two.getSExtValue());
 232
 233   // Basic equalities for 1-bit values.
 234   EXPECT_EQ(zero, two);
 235   EXPECT_EQ(zero, neg_two);
 236   EXPECT_EQ(one, neg_one);
 237   EXPECT_EQ(two, neg_two);
 238
 239   // Min/max signed values.
 240   EXPECT_TRUE(zero.isMaxSignedValue());
 241   EXPECT_FALSE(one.isMaxSignedValue());
 242   EXPECT_FALSE(zero.isMinSignedValue());
 243   EXPECT_TRUE(one.isMinSignedValue());
 244
 245   // Additions.
 246   EXPECT_EQ(two, one + one);
 247   EXPECT_EQ(zero, neg_one + one);
 248   EXPECT_EQ(neg_two, neg_one + neg_one);
 249
 250   // Subtractions.
 251   EXPECT_EQ(neg_two, neg_one - one);
 252   EXPECT_EQ(two, one - neg_one);
 253   EXPECT_EQ(zero, one - one);
 254
 255   // And
 256   EXPECT_EQ(zero, zero & zero);
 257   EXPECT_EQ(zero, one & zero);
 258   EXPECT_EQ(zero, zero & one);
 259   EXPECT_EQ(one, one & one);
 260   EXPECT_EQ(zero, zero & zero);
 261   EXPECT_EQ(zero, neg_one & zero);
 262   EXPECT_EQ(zero, zero & neg_one);
 263   EXPECT_EQ(neg_one, neg_one & neg_one);
 264
 265   // Or
 266   EXPECT_EQ(zero, zero | zero);
 267   EXPECT_EQ(one, one | zero);
 268   EXPECT_EQ(one, zero | one);
 269   EXPECT_EQ(one, one | one);
 270   EXPECT_EQ(zero, zero | zero);
 271   EXPECT_EQ(neg_one, neg_one | zero);
 272   EXPECT_EQ(neg_one, zero | neg_one);
 273   EXPECT_EQ(neg_one, neg_one | neg_one);
 274
 275   // Xor
 276   EXPECT_EQ(zero, zero ^ zero);
 277   EXPECT_EQ(one, one ^ zero);
 278   EXPECT_EQ(one, zero ^ one);
 279   EXPECT_EQ(zero, one ^ one);
 280   EXPECT_EQ(zero, zero ^ zero);
 281   EXPECT_EQ(neg_one, neg_one ^ zero);
 282   EXPECT_EQ(neg_one, zero ^ neg_one);
 283   EXPECT_EQ(zero, neg_one ^ neg_one);
 284
 285   // Shifts.
 286   EXPECT_EQ(zero, one << one);
 287   EXPECT_EQ(one, one << zero);
 288   EXPECT_EQ(zero, one.shl(1));
 289   EXPECT_EQ(one, one.shl(0));
 290   EXPECT_EQ(zero, one.lshr(1));
 291   EXPECT_EQ(one, one.ashr(1));
 292
 293   // Rotates.
 294   EXPECT_EQ(one, one.rotl(0));
 295   EXPECT_EQ(one, one.rotl(1));
 296   EXPECT_EQ(one, one.rotr(0));
 297   EXPECT_EQ(one, one.rotr(1));
 298
 299   // Multiplies.
 300   EXPECT_EQ(neg_one, neg_one * one);
 301   EXPECT_EQ(neg_one, one * neg_one);
 302   EXPECT_EQ(one, neg_one * neg_one);
 303   EXPECT_EQ(one, one * one);
 304
 305   // Divides.
 306   EXPECT_EQ(neg_one, one.sdiv(neg_one));
 307   EXPECT_EQ(neg_one, neg_one.sdiv(one));
 308   EXPECT_EQ(one, neg_one.sdiv(neg_one));
 309   EXPECT_EQ(one, one.sdiv(one));
 310
 311   EXPECT_EQ(neg_one, one.udiv(neg_one));
 312   EXPECT_EQ(neg_one, neg_one.udiv(one));
 313   EXPECT_EQ(one, neg_one.udiv(neg_one));
 314   EXPECT_EQ(one, one.udiv(one));
 315
 316   // Remainders.
 317   EXPECT_EQ(zero, neg_one.srem(one));
 318   EXPECT_EQ(zero, neg_one.urem(one));
 319   EXPECT_EQ(zero, one.srem(neg_one));
 320
 321   // sdivrem
 322   {
 323   APInt q(8, 0);
 324   APInt r(8, 0);
 325   APInt one(8, 1);
 326   APInt two(8, 2);
 327   APInt nine(8, 9);
 328   APInt four(8, 4);
 329
 330   EXPECT_EQ(nine.srem(two), one);
 331   EXPECT_EQ(nine.srem(-two), one);
 332   EXPECT_EQ((-nine).srem(two), -one);
 333   EXPECT_EQ((-nine).srem(-two), -one);
 334
 335   APInt::sdivrem(nine, two, q, r);
 336   EXPECT_EQ(four, q);
 337   EXPECT_EQ(one, r);
 338   APInt::sdivrem(-nine, two, q, r);
 339   EXPECT_EQ(-four, q);
 340   EXPECT_EQ(-one, r);
 341   APInt::sdivrem(nine, -two, q, r);
 342   EXPECT_EQ(-four, q);
 343   EXPECT_EQ(one, r);
 344   APInt::sdivrem(-nine, -two, q, r);
 345   EXPECT_EQ(four, q);
 346   EXPECT_EQ(-one, r);
 347   }
 348 }
 349
 350 TEST(APIntTest, compare) {
 351   std::array<APInt, 5> testVals{{
 352     APInt{16, 2},
 353     APInt{16, 1},
 354     APInt{16, 0},
 355     APInt{16, (uint64_t)-1, true},
 356     APInt{16, (uint64_t)-2, true},
 357   }};
 358
 359   for (auto &arg1 : testVals)
 360     for (auto &arg2 : testVals) {
 361       auto uv1 = arg1.getZExtValue();
 362       auto uv2 = arg2.getZExtValue();
 363       auto sv1 = arg1.getSExtValue();
 364       auto sv2 = arg2.getSExtValue();
 365
 366       EXPECT_EQ(uv1 <  uv2, arg1.ult(arg2));
 367       EXPECT_EQ(uv1 <= uv2, arg1.ule(arg2));
 368       EXPECT_EQ(uv1 >  uv2, arg1.ugt(arg2));
 369       EXPECT_EQ(uv1 >= uv2, arg1.uge(arg2));
 370
 371       EXPECT_EQ(sv1 <  sv2, arg1.slt(arg2));
 372       EXPECT_EQ(sv1 <= sv2, arg1.sle(arg2));
 373       EXPECT_EQ(sv1 >  sv2, arg1.sgt(arg2));
 374       EXPECT_EQ(sv1 >= sv2, arg1.sge(arg2));
 375
 376       EXPECT_EQ(uv1 <  uv2, arg1.ult(uv2));
 377       EXPECT_EQ(uv1 <= uv2, arg1.ule(uv2));
 378       EXPECT_EQ(uv1 >  uv2, arg1.ugt(uv2));
 379       EXPECT_EQ(uv1 >= uv2, arg1.uge(uv2));
 380
 381       EXPECT_EQ(sv1 <  sv2, arg1.slt(sv2));
 382       EXPECT_EQ(sv1 <= sv2, arg1.sle(sv2));
 383       EXPECT_EQ(sv1 >  sv2, arg1.sgt(sv2));
 384       EXPECT_EQ(sv1 >= sv2, arg1.sge(sv2));
 385     }
 386 }
 387
 388 TEST(APIntTest, compareWithRawIntegers) {
 389   EXPECT_TRUE(!APInt(8, 1).uge(256));
 390   EXPECT_TRUE(!APInt(8, 1).ugt(256));
 391   EXPECT_TRUE( APInt(8, 1).ule(256));
 392   EXPECT_TRUE( APInt(8, 1).ult(256));
 393   EXPECT_TRUE(!APInt(8, 1).sge(256));
 394   EXPECT_TRUE(!APInt(8, 1).sgt(256));
 395   EXPECT_TRUE( APInt(8, 1).sle(256));
 396   EXPECT_TRUE( APInt(8, 1).slt(256));
 397   EXPECT_TRUE(!(APInt(8, 0) == 256));
 398   EXPECT_TRUE(  APInt(8, 0) != 256);
 399   EXPECT_TRUE(!(APInt(8, 1) == 256));
 400   EXPECT_TRUE(  APInt(8, 1) != 256);
 401
 402   auto uint64max = UINT64_MAX;
 403   auto int64max  = INT64_MAX;
 404   auto int64min  = INT64_MIN;
 405
 406   auto u64 = APInt{128, uint64max};
 407   auto s64 = APInt{128, static_cast<uint64_t>(int64max), true};
 408   auto big = u64 + 1;
 409
 410   EXPECT_TRUE( u64.uge(uint64max));
 411   EXPECT_TRUE(!u64.ugt(uint64max));
 412   EXPECT_TRUE( u64.ule(uint64max));
 413   EXPECT_TRUE(!u64.ult(uint64max));
 414   EXPECT_TRUE( u64.sge(int64max));
 415   EXPECT_TRUE( u64.sgt(int64max));
 416   EXPECT_TRUE(!u64.sle(int64max));
 417   EXPECT_TRUE(!u64.slt(int64max));
 418   EXPECT_TRUE( u64.sge(int64min));
 419   EXPECT_TRUE( u64.sgt(int64min));
 420   EXPECT_TRUE(!u64.sle(int64min));
 421   EXPECT_TRUE(!u64.slt(int64min));
 422
 423   EXPECT_TRUE(u64 == uint64max);
 424   EXPECT_TRUE(u64 != int64max);
 425   EXPECT_TRUE(u64 != int64min);
 426
 427   EXPECT_TRUE(!s64.uge(uint64max));
 428   EXPECT_TRUE(!s64.ugt(uint64max));
 429   EXPECT_TRUE( s64.ule(uint64max));
 430   EXPECT_TRUE( s64.ult(uint64max));
 431   EXPECT_TRUE( s64.sge(int64max));
 432   EXPECT_TRUE(!s64.sgt(int64max));
 433   EXPECT_TRUE( s64.sle(int64max));
 434   EXPECT_TRUE(!s64.slt(int64max));
 435   EXPECT_TRUE( s64.sge(int64min));
 436   EXPECT_TRUE( s64.sgt(int64min));
 437   EXPECT_TRUE(!s64.sle(int64min));
 438   EXPECT_TRUE(!s64.slt(int64min));
 439
 440   EXPECT_TRUE(s64 != uint64max);
 441   EXPECT_TRUE(s64 == int64max);
 442   EXPECT_TRUE(s64 != int64min);
 443
 444   EXPECT_TRUE( big.uge(uint64max));
 445   EXPECT_TRUE( big.ugt(uint64max));
 446   EXPECT_TRUE(!big.ule(uint64max));
 447   EXPECT_TRUE(!big.ult(uint64max));
 448   EXPECT_TRUE( big.sge(int64max));
 449   EXPECT_TRUE( big.sgt(int64max));
 450   EXPECT_TRUE(!big.sle(int64max));
 451   EXPECT_TRUE(!big.slt(int64max));
 452   EXPECT_TRUE( big.sge(int64min));
 453   EXPECT_TRUE( big.sgt(int64min));
 454   EXPECT_TRUE(!big.sle(int64min));
 455   EXPECT_TRUE(!big.slt(int64min));
 456
 457   EXPECT_TRUE(big != uint64max);
 458   EXPECT_TRUE(big != int64max);
 459   EXPECT_TRUE(big != int64min);
 460 }
 461
 462 TEST(APIntTest, compareWithInt64Min) {
 463   int64_t edge = INT64_MIN;
 464   int64_t edgeP1 = edge + 1;
 465   int64_t edgeM1 = INT64_MAX;
 466   auto a = APInt{64, static_cast<uint64_t>(edge), true};
 467
 468   EXPECT_TRUE(!a.slt(edge));
 469   EXPECT_TRUE( a.sle(edge));
 470   EXPECT_TRUE(!a.sgt(edge));
 471   EXPECT_TRUE( a.sge(edge));
 472   EXPECT_TRUE( a.slt(edgeP1));
 473   EXPECT_TRUE( a.sle(edgeP1));
 474   EXPECT_TRUE(!a.sgt(edgeP1));
 475   EXPECT_TRUE(!a.sge(edgeP1));
 476   EXPECT_TRUE( a.slt(edgeM1));
 477   EXPECT_TRUE( a.sle(edgeM1));
 478   EXPECT_TRUE(!a.sgt(edgeM1));
 479   EXPECT_TRUE(!a.sge(edgeM1));
 480 }
 481
 482 TEST(APIntTest, compareWithHalfInt64Max) {
 483   uint64_t edge = 0x4000000000000000;
 484   uint64_t edgeP1 = edge + 1;
 485   uint64_t edgeM1 = edge - 1;
 486   auto a = APInt{64, edge};
 487
 488   EXPECT_TRUE(!a.ult(edge));
 489   EXPECT_TRUE( a.ule(edge));
 490   EXPECT_TRUE(!a.ugt(edge));
 491   EXPECT_TRUE( a.uge(edge));
 492   EXPECT_TRUE( a.ult(edgeP1));
 493   EXPECT_TRUE( a.ule(edgeP1));
 494   EXPECT_TRUE(!a.ugt(edgeP1));
 495   EXPECT_TRUE(!a.uge(edgeP1));
 496   EXPECT_TRUE(!a.ult(edgeM1));
 497   EXPECT_TRUE(!a.ule(edgeM1));
 498   EXPECT_TRUE( a.ugt(edgeM1));
 499   EXPECT_TRUE( a.uge(edgeM1));
 500
 501   EXPECT_TRUE(!a.slt(edge));
 502   EXPECT_TRUE( a.sle(edge));
 503   EXPECT_TRUE(!a.sgt(edge));
 504   EXPECT_TRUE( a.sge(edge));
 505   EXPECT_TRUE( a.slt(edgeP1));
 506   EXPECT_TRUE( a.sle(edgeP1));
 507   EXPECT_TRUE(!a.sgt(edgeP1));
 508   EXPECT_TRUE(!a.sge(edgeP1));
 509   EXPECT_TRUE(!a.slt(edgeM1));
 510   EXPECT_TRUE(!a.sle(edgeM1));
 511   EXPECT_TRUE( a.sgt(edgeM1));
 512   EXPECT_TRUE( a.sge(edgeM1));
 513 }
 514
 515 TEST(APIntTest, compareLargeIntegers) {
 516   // Make sure all the combinations of signed comparisons work with big ints.
 517   auto One = APInt{128, static_cast<uint64_t>(1), true};
 518   auto Two = APInt{128, static_cast<uint64_t>(2), true};
 519   auto MinusOne = APInt{128, static_cast<uint64_t>(-1), true};
 520   auto MinusTwo = APInt{128, static_cast<uint64_t>(-2), true};
 521
 522   EXPECT_TRUE(!One.slt(One));
 523   EXPECT_TRUE(!Two.slt(One));
 524   EXPECT_TRUE(MinusOne.slt(One));
 525   EXPECT_TRUE(MinusTwo.slt(One));
 526
 527   EXPECT_TRUE(One.slt(Two));
 528   EXPECT_TRUE(!Two.slt(Two));
 529   EXPECT_TRUE(MinusOne.slt(Two));
 530   EXPECT_TRUE(MinusTwo.slt(Two));
 531
 532   EXPECT_TRUE(!One.slt(MinusOne));
 533   EXPECT_TRUE(!Two.slt(MinusOne));
 534   EXPECT_TRUE(!MinusOne.slt(MinusOne));
 535   EXPECT_TRUE(MinusTwo.slt(MinusOne));
 536
 537   EXPECT_TRUE(!One.slt(MinusTwo));
 538   EXPECT_TRUE(!Two.slt(MinusTwo));
 539   EXPECT_TRUE(!MinusOne.slt(MinusTwo));
 540   EXPECT_TRUE(!MinusTwo.slt(MinusTwo));
 541 }
 542
 543 TEST(APIntTest, binaryOpsWithRawIntegers) {
 544   // Single word check.
 545   uint64_t E1 = 0x2CA7F46BF6569915ULL;
 546   APInt A1(64, E1);
 547
 548   EXPECT_EQ(A1 & E1, E1);
 549   EXPECT_EQ(A1 & 0, 0);
 550   EXPECT_EQ(A1 & 1, 1);
 551   EXPECT_EQ(A1 & 5, 5);
 552   EXPECT_EQ(A1 & UINT64_MAX, E1);
 553
 554   EXPECT_EQ(A1 | E1, E1);
 555   EXPECT_EQ(A1 | 0, E1);
 556   EXPECT_EQ(A1 | 1, E1);
 557   EXPECT_EQ(A1 | 2, E1 | 2);
 558   EXPECT_EQ(A1 | UINT64_MAX, UINT64_MAX);
 559
 560   EXPECT_EQ(A1 ^ E1, 0);
 561   EXPECT_EQ(A1 ^ 0, E1);
 562   EXPECT_EQ(A1 ^ 1, E1 ^ 1);
 563   EXPECT_EQ(A1 ^ 7, E1 ^ 7);
 564   EXPECT_EQ(A1 ^ UINT64_MAX, ~E1);
 565
 566   // Multiword check.
 567   uint64_t N = 0xEB6EB136591CBA21ULL;
 568   APInt::WordType E2[4] = {
 569     N,
 570     0x7B9358BD6A33F10AULL,
 571     0x7E7FFA5EADD8846ULL,
 572     0x305F341CA00B613DULL
 573   };
 574   APInt A2(APInt::APINT_BITS_PER_WORD*4, E2);
 575
 576   EXPECT_EQ(A2 & N, N);
 577   EXPECT_EQ(A2 & 0, 0);
 578   EXPECT_EQ(A2 & 1, 1);
 579   EXPECT_EQ(A2 & 5, 1);
 580   EXPECT_EQ(A2 & UINT64_MAX, N);
 581
 582   EXPECT_EQ(A2 | N, A2);
 583   EXPECT_EQ(A2 | 0, A2);
 584   EXPECT_EQ(A2 | 1, A2);
 585   EXPECT_EQ(A2 | 2, A2 + 2);
 586   EXPECT_EQ(A2 | UINT64_MAX, A2 - N + UINT64_MAX);
 587
 588   EXPECT_EQ(A2 ^ N, A2 - N);
 589   EXPECT_EQ(A2 ^ 0, A2);
 590   EXPECT_EQ(A2 ^ 1, A2 - 1);
 591   EXPECT_EQ(A2 ^ 7, A2 + 5);
 592   EXPECT_EQ(A2 ^ UINT64_MAX, A2 - N + ~N);
 593 }
 594
 595 TEST(APIntTest, rvalue_arithmetic) {
 596   // Test all combinations of lvalue/rvalue lhs/rhs of add/sub
 597
 598   // Lamdba to return an APInt by value, but also provide the raw value of the
 599   // allocated data.
 600   auto getRValue = [](const char *HexString, uint64_t const *&RawData) {
 601     APInt V(129, HexString, 16);
 602     RawData = V.getRawData();
 603     return V;
 604   };
 605
 606   APInt One(129, "1", 16);
 607   APInt Two(129, "2", 16);
 608   APInt Three(129, "3", 16);
 609   APInt MinusOne = -One;
 610
 611   const uint64_t *RawDataL = nullptr;
 612   const uint64_t *RawDataR = nullptr;
 613
 614   {
 615     // 1 + 1 = 2
 616     APInt AddLL = One + One;
 617     EXPECT_EQ(AddLL, Two);
 618
 619     APInt AddLR = One + getRValue("1", RawDataR);
 620     EXPECT_EQ(AddLR, Two);
 621     EXPECT_EQ(AddLR.getRawData(), RawDataR);
 622
 623     APInt AddRL = getRValue("1", RawDataL) + One;
 624     EXPECT_EQ(AddRL, Two);
 625     EXPECT_EQ(AddRL.getRawData(), RawDataL);
 626
 627     APInt AddRR = getRValue("1", RawDataL) + getRValue("1", RawDataR);
 628     EXPECT_EQ(AddRR, Two);
 629     EXPECT_EQ(AddRR.getRawData(), RawDataR);
 630
 631     // LValue's and constants
 632     APInt AddLK = One + 1;
 633     EXPECT_EQ(AddLK, Two);
 634
 635     APInt AddKL = 1 + One;
 636     EXPECT_EQ(AddKL, Two);
 637
 638     // RValue's and constants
 639     APInt AddRK = getRValue("1", RawDataL) + 1;
 640     EXPECT_EQ(AddRK, Two);
 641     EXPECT_EQ(AddRK.getRawData(), RawDataL);
 642
 643     APInt AddKR = 1 + getRValue("1", RawDataR);
 644     EXPECT_EQ(AddKR, Two);
 645     EXPECT_EQ(AddKR.getRawData(), RawDataR);
 646   }
 647
 648   {
 649     // 0x0,FFFF...FFFF + 0x2 = 0x100...0001
 650     APInt AllOnes(129, "0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", 16);
 651     APInt HighOneLowOne(129, "100000000000000000000000000000001", 16);
 652
 653     APInt AddLL = AllOnes + Two;
 654     EXPECT_EQ(AddLL, HighOneLowOne);
 655
 656     APInt AddLR = AllOnes + getRValue("2", RawDataR);
 657     EXPECT_EQ(AddLR, HighOneLowOne);
 658     EXPECT_EQ(AddLR.getRawData(), RawDataR);
 659
 660     APInt AddRL = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) + Two;
 661     EXPECT_EQ(AddRL, HighOneLowOne);
 662     EXPECT_EQ(AddRL.getRawData(), RawDataL);
 663
 664     APInt AddRR = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) +
 665                   getRValue("2", RawDataR);
 666     EXPECT_EQ(AddRR, HighOneLowOne);
 667     EXPECT_EQ(AddRR.getRawData(), RawDataR);
 668
 669     // LValue's and constants
 670     APInt AddLK = AllOnes + 2;
 671     EXPECT_EQ(AddLK, HighOneLowOne);
 672
 673     APInt AddKL = 2 + AllOnes;
 674     EXPECT_EQ(AddKL, HighOneLowOne);
 675
 676     // RValue's and constants
 677     APInt AddRK = getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataL) + 2;
 678     EXPECT_EQ(AddRK, HighOneLowOne);
 679     EXPECT_EQ(AddRK.getRawData(), RawDataL);
 680
 681     APInt AddKR = 2 + getRValue("FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR);
 682     EXPECT_EQ(AddKR, HighOneLowOne);
 683     EXPECT_EQ(AddKR.getRawData(), RawDataR);
 684   }
 685
 686   {
 687     // 2 - 1 = 1
 688     APInt SubLL = Two - One;
 689     EXPECT_EQ(SubLL, One);
 690
 691     APInt SubLR = Two - getRValue("1", RawDataR);
 692     EXPECT_EQ(SubLR, One);
 693     EXPECT_EQ(SubLR.getRawData(), RawDataR);
 694
 695     APInt SubRL = getRValue("2", RawDataL) - One;
 696     EXPECT_EQ(SubRL, One);
 697     EXPECT_EQ(SubRL.getRawData(), RawDataL);
 698
 699     APInt SubRR = getRValue("2", RawDataL) - getRValue("1", RawDataR);
 700     EXPECT_EQ(SubRR, One);
 701     EXPECT_EQ(SubRR.getRawData(), RawDataR);
 702
 703     // LValue's and constants
 704     APInt SubLK = Two - 1;
 705     EXPECT_EQ(SubLK, One);
 706
 707     APInt SubKL = 2 - One;
 708     EXPECT_EQ(SubKL, One);
 709
 710     // RValue's and constants
 711     APInt SubRK = getRValue("2", RawDataL) - 1;
 712     EXPECT_EQ(SubRK, One);
 713     EXPECT_EQ(SubRK.getRawData(), RawDataL);
 714
 715     APInt SubKR = 2 - getRValue("1", RawDataR);
 716     EXPECT_EQ(SubKR, One);
 717     EXPECT_EQ(SubKR.getRawData(), RawDataR);
 718   }
 719
 720   {
 721     // 0x100...0001 - 0x0,FFFF...FFFF = 0x2
 722     APInt AllOnes(129, "0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", 16);
 723     APInt HighOneLowOne(129, "100000000000000000000000000000001", 16);
 724
 725     APInt SubLL = HighOneLowOne - AllOnes;
 726     EXPECT_EQ(SubLL, Two);
 727
 728     APInt SubLR = HighOneLowOne -
 729                   getRValue("0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR);
 730     EXPECT_EQ(SubLR, Two);
 731     EXPECT_EQ(SubLR.getRawData(), RawDataR);
 732
 733     APInt SubRL = getRValue("100000000000000000000000000000001", RawDataL) -
 734                   AllOnes;
 735     EXPECT_EQ(SubRL, Two);
 736     EXPECT_EQ(SubRL.getRawData(), RawDataL);
 737
 738     APInt SubRR = getRValue("100000000000000000000000000000001", RawDataL) -
 739                   getRValue("0FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR);
 740     EXPECT_EQ(SubRR, Two);
 741     EXPECT_EQ(SubRR.getRawData(), RawDataR);
 742
 743     // LValue's and constants
 744     // 0x100...0001 - 0x2 = 0x0,FFFF...FFFF
 745     APInt SubLK = HighOneLowOne - 2;
 746     EXPECT_EQ(SubLK, AllOnes);
 747
 748     // 2 - (-1) = 3
 749     APInt SubKL = 2 - MinusOne;
 750     EXPECT_EQ(SubKL, Three);
 751
 752     // RValue's and constants
 753     // 0x100...0001 - 0x2 = 0x0,FFFF...FFFF
 754     APInt SubRK = getRValue("100000000000000000000000000000001", RawDataL) - 2;
 755     EXPECT_EQ(SubRK, AllOnes);
 756     EXPECT_EQ(SubRK.getRawData(), RawDataL);
 757
 758     APInt SubKR = 2 - getRValue("1FFFFFFFFFFFFFFFFFFFFFFFFFFFFFFFF", RawDataR);
 759     EXPECT_EQ(SubKR, Three);
 760     EXPECT_EQ(SubKR.getRawData(), RawDataR);
 761   }
 762 }
 763
 764 TEST(APIntTest, rvalue_bitwise) {
 765   // Test all combinations of lvalue/rvalue lhs/rhs of and/or/xor
 766
 767   // Lamdba to return an APInt by value, but also provide the raw value of the
 768   // allocated data.
 769   auto getRValue = [](const char *HexString, uint64_t const *&RawData) {
 770     APInt V(129, HexString, 16);
 771     RawData = V.getRawData();
 772     return V;
 773   };
 774
 775   APInt Ten(129, "A", 16);
 776   APInt Twelve(129, "C", 16);
 777
 778   const uint64_t *RawDataL = nullptr;
 779   const uint64_t *RawDataR = nullptr;
 780
 781   {
 782     // 12 & 10 = 8
 783     APInt AndLL = Ten & Twelve;
 784     EXPECT_EQ(AndLL, 0x8);
 785
 786     APInt AndLR = Ten & getRValue("C", RawDataR);
 787     EXPECT_EQ(AndLR, 0x8);
 788     EXPECT_EQ(AndLR.getRawData(), RawDataR);
 789
 790     APInt AndRL = getRValue("A", RawDataL) & Twelve;
 791     EXPECT_EQ(AndRL, 0x8);
 792     EXPECT_EQ(AndRL.getRawData(), RawDataL);
 793
 794     APInt AndRR = getRValue("A", RawDataL) & getRValue("C", RawDataR);
 795     EXPECT_EQ(AndRR, 0x8);
 796     EXPECT_EQ(AndRR.getRawData(), RawDataR);
 797
 798     // LValue's and constants
 799     APInt AndLK = Ten & 0xc;
 800     EXPECT_EQ(AndLK, 0x8);
 801
 802     APInt AndKL = 0xa & Twelve;
 803     EXPECT_EQ(AndKL, 0x8);
 804
 805     // RValue's and constants
 806     APInt AndRK = getRValue("A", RawDataL) & 0xc;
 807     EXPECT_EQ(AndRK, 0x8);
 808     EXPECT_EQ(AndRK.getRawData(), RawDataL);
 809
 810     APInt AndKR = 0xa & getRValue("C", RawDataR);
 811     EXPECT_EQ(AndKR, 0x8);
 812     EXPECT_EQ(AndKR.getRawData(), RawDataR);
 813   }
 814
 815   {
 816     // 12 | 10 = 14
 817     APInt OrLL = Ten | Twelve;
 818     EXPECT_EQ(OrLL, 0xe);
 819
 820     APInt OrLR = Ten | getRValue("C", RawDataR);
 821     EXPECT_EQ(OrLR, 0xe);
 822     EXPECT_EQ(OrLR.getRawData(), RawDataR);
 823
 824     APInt OrRL = getRValue("A", RawDataL) | Twelve;
 825     EXPECT_EQ(OrRL, 0xe);
 826     EXPECT_EQ(OrRL.getRawData(), RawDataL);
 827
 828     APInt OrRR = getRValue("A", RawDataL) | getRValue("C", RawDataR);
 829     EXPECT_EQ(OrRR, 0xe);
 830     EXPECT_EQ(OrRR.getRawData(), RawDataR);
 831
 832     // LValue's and constants
 833     APInt OrLK = Ten | 0xc;
 834     EXPECT_EQ(OrLK, 0xe);
 835
 836     APInt OrKL = 0xa | Twelve;
 837     EXPECT_EQ(OrKL, 0xe);
 838
 839     // RValue's and constants
 840     APInt OrRK = getRValue("A", RawDataL) | 0xc;
 841     EXPECT_EQ(OrRK, 0xe);
 842     EXPECT_EQ(OrRK.getRawData(), RawDataL);
 843
 844     APInt OrKR = 0xa | getRValue("C", RawDataR);
 845     EXPECT_EQ(OrKR, 0xe);
 846     EXPECT_EQ(OrKR.getRawData(), RawDataR);
 847   }
 848
 849   {
 850     // 12 ^ 10 = 6
 851     APInt XorLL = Ten ^ Twelve;
 852     EXPECT_EQ(XorLL, 0x6);
 853
 854     APInt XorLR = Ten ^ getRValue("C", RawDataR);
 855     EXPECT_EQ(XorLR, 0x6);
 856     EXPECT_EQ(XorLR.getRawData(), RawDataR);
 857
 858     APInt XorRL = getRValue("A", RawDataL) ^ Twelve;
 859     EXPECT_EQ(XorRL, 0x6);
 860     EXPECT_EQ(XorRL.getRawData(), RawDataL);
 861
 862     APInt XorRR = getRValue("A", RawDataL) ^ getRValue("C", RawDataR);
 863     EXPECT_EQ(XorRR, 0x6);
 864     EXPECT_EQ(XorRR.getRawData(), RawDataR);
 865
 866     // LValue's and constants
 867     APInt XorLK = Ten ^ 0xc;
 868     EXPECT_EQ(XorLK, 0x6);
 869
 870     APInt XorKL = 0xa ^ Twelve;
 871     EXPECT_EQ(XorKL, 0x6);
 872
 873     // RValue's and constants
 874     APInt XorRK = getRValue("A", RawDataL) ^ 0xc;
 875     EXPECT_EQ(XorRK, 0x6);
 876     EXPECT_EQ(XorRK.getRawData(), RawDataL);
 877
 878     APInt XorKR = 0xa ^ getRValue("C", RawDataR);
 879     EXPECT_EQ(XorKR, 0x6);
 880     EXPECT_EQ(XorKR.getRawData(), RawDataR);
 881   }
 882 }
 883
 884 TEST(APIntTest, rvalue_invert) {
 885   // Lamdba to return an APInt by value, but also provide the raw value of the
 886   // allocated data.
 887   auto getRValue = [](const char *HexString, uint64_t const *&RawData) {
 888     APInt V(129, HexString, 16);
 889     RawData = V.getRawData();
 890     return V;
 891   };
 892
 893   APInt One(129, 1);
 894   APInt NegativeTwo(129, -2ULL, true);
 895
 896   const uint64_t *RawData = nullptr;
 897
 898   {
 899     // ~1 = -2
 900     APInt NegL = ~One;
 901     EXPECT_EQ(NegL, NegativeTwo);
 902
 903     APInt NegR = ~getRValue("1", RawData);
 904     EXPECT_EQ(NegR, NegativeTwo);
 905     EXPECT_EQ(NegR.getRawData(), RawData);
 906   }
 907 }
 908
 909 // Tests different div/rem varaints using scheme (a * b + c) / a
 910 void testDiv(APInt a, APInt b, APInt c) {
 911   ASSERT_TRUE(a.uge(b)); // Must: a >= b
 912   ASSERT_TRUE(a.ugt(c)); // Must: a > c
 913
 914   auto p = a * b + c;
 915
 916   auto q = p.udiv(a);
 917   auto r = p.urem(a);
 918   EXPECT_EQ(b, q);
 919   EXPECT_EQ(c, r);
 920   APInt::udivrem(p, a, q, r);
 921   EXPECT_EQ(b, q);
 922   EXPECT_EQ(c, r);
 923   q = p.sdiv(a);
 924   r = p.srem(a);
 925   EXPECT_EQ(b, q);
 926   EXPECT_EQ(c, r);
 927   APInt::sdivrem(p, a, q, r);
 928   EXPECT_EQ(b, q);
 929   EXPECT_EQ(c, r);
 930
 931   if (b.ugt(c)) { // Test also symmetric case
 932     q = p.udiv(b);
 933     r = p.urem(b);
 934     EXPECT_EQ(a, q);
 935     EXPECT_EQ(c, r);
 936     APInt::udivrem(p, b, q, r);
 937     EXPECT_EQ(a, q);
 938     EXPECT_EQ(c, r);
 939     q = p.sdiv(b);
 940     r = p.srem(b);
 941     EXPECT_EQ(a, q);
 942     EXPECT_EQ(c, r);
 943     APInt::sdivrem(p, b, q, r);
 944     EXPECT_EQ(a, q);
 945     EXPECT_EQ(c, r);
 946   }
 947 }
 948
 949 TEST(APIntTest, divrem_big1) {
 950   // Tests KnuthDiv rare step D6
 951   testDiv({256, "1ffffffffffffffff", 16},
 952           {256, "1ffffffffffffffff", 16},
 953           {256, 0});
 954 }
 955
 956 TEST(APIntTest, divrem_big2) {
 957   // Tests KnuthDiv rare step D6
 958   testDiv({1024,                       "112233ceff"
 959                  "cecece000000ffffffffffffffffffff"
 960                  "ffffffffffffffffffffffffffffffff"
 961                  "ffffffffffffffffffffffffffffffff"
 962                  "ffffffffffffffffffffffffffffff33", 16},
 963           {1024,           "111111ffffffffffffffff"
 964                  "ffffffffffffffffffffffffffffffff"
 965                  "fffffffffffffffffffffffffffffccf"
 966                  "ffffffffffffffffffffffffffffff00", 16},
 967           {1024, 7919});
 968 }
 969
 970 TEST(APIntTest, divrem_big3) {
 971   // Tests KnuthDiv case without shift
 972   testDiv({256, "80000001ffffffffffffffff", 16},
 973           {256, "ffffffffffffff0000000", 16},
 974           {256, 4219});
 975 }
 976
 977 TEST(APIntTest, divrem_big4) {
 978   // Tests heap allocation in divide() enfoced by huge numbers
 979   testDiv(APInt{4096, 5}.shl(2001),
 980           APInt{4096, 1}.shl(2000),
 981           APInt{4096, 4219*13});
 982 }
 983
 984 TEST(APIntTest, divrem_big5) {
 985   // Tests one word divisor case of divide()
 986   testDiv(APInt{1024, 19}.shl(811),
 987           APInt{1024, 4356013}, // one word
 988           APInt{1024, 1});
 989 }
 990
 991 TEST(APIntTest, divrem_big6) {
 992   // Tests some rare "borrow" cases in D4 step
 993   testDiv(APInt{512, "ffffffffffffffff00000000000000000000000001", 16},
 994           APInt{512, "10000000000000001000000000000001", 16},
 995           APInt{512, "10000000000000000000000000000000", 16});
 996 }
 997
 998 TEST(APIntTest, divrem_big7) {
 999   // Yet another test for KnuthDiv rare step D6.
1000   testDiv({224, "800000008000000200000005", 16},
1001           {224, "fffffffd", 16},
1002           {224, "80000000800000010000000f", 16});
1003 }
1004
1005 void testDiv(APInt a, uint64_t b, APInt c) {
1006   auto p = a * b + c;
1007
1008   APInt q;
1009   uint64_t r;
1010   // Unsigned division will only work if our original number wasn't negative.
1011   if (!a.isNegative()) {
1012     q = p.udiv(b);
1013     r = p.urem(b);
1014     EXPECT_EQ(a, q);
1015     EXPECT_EQ(c, r);
1016     APInt::udivrem(p, b, q, r);
1017     EXPECT_EQ(a, q);
1018     EXPECT_EQ(c, r);
1019   }
1020   q = p.sdiv(b);
1021   r = p.srem(b);
1022   EXPECT_EQ(a, q);
1023   if (c.isNegative())
1024     EXPECT_EQ(-c, -r); // Need to negate so the uint64_t compare will work.
1025   else
1026     EXPECT_EQ(c, r);
1027   int64_t sr;
1028   APInt::sdivrem(p, b, q, sr);
1029   EXPECT_EQ(a, q);
1030   if (c.isNegative())
1031     EXPECT_EQ(-c, -sr); // Need to negate so the uint64_t compare will work.
1032   else
1033     EXPECT_EQ(c, sr);
1034 }
1035
1036 TEST(APIntTest, divremuint) {
1037   // Single word APInt
1038   testDiv(APInt{64, 9},
1039           2,
1040           APInt{64, 1});
1041
1042   // Single word negative APInt
1043   testDiv(-APInt{64, 9},
1044           2,
1045           -APInt{64, 1});
1046
1047   // Multiword dividend with only one significant word.
1048   testDiv(APInt{256, 9},
1049           2,
1050           APInt{256, 1});
1051
1052   // Negative dividend.
1053   testDiv(-APInt{256, 9},
1054           2,
1055           -APInt{256, 1});
1056
1057   // Multiword dividend
1058   testDiv(APInt{1024, 19}.shl(811),
1059           4356013, // one word
1060           APInt{1024, 1});
1061 }
1062
1063 TEST(APIntTest, fromString) {
1064   EXPECT_EQ(APInt(32, 0), APInt(32,   "0", 2));
1065   EXPECT_EQ(APInt(32, 1), APInt(32,   "1", 2));
1066   EXPECT_EQ(APInt(32, 2), APInt(32,  "10", 2));
1067   EXPECT_EQ(APInt(32, 3), APInt(32,  "11", 2));
1068   EXPECT_EQ(APInt(32, 4), APInt(32, "100", 2));
1069
1070   EXPECT_EQ(APInt(32, 0), APInt(32,   "+0", 2));
1071   EXPECT_EQ(APInt(32, 1), APInt(32,   "+1", 2));
1072   EXPECT_EQ(APInt(32, 2), APInt(32,  "+10", 2));
1073   EXPECT_EQ(APInt(32, 3), APInt(32,  "+11", 2));
1074   EXPECT_EQ(APInt(32, 4), APInt(32, "+100", 2));
1075
1076   EXPECT_EQ(APInt(32, uint64_t(-0LL)), APInt(32,   "-0", 2));
1077   EXPECT_EQ(APInt(32, uint64_t(-1LL)), APInt(32,   "-1", 2));
1078   EXPECT_EQ(APInt(32, uint64_t(-2LL)), APInt(32,  "-10", 2));
1079   EXPECT_EQ(APInt(32, uint64_t(-3LL)), APInt(32,  "-11", 2));
1080   EXPECT_EQ(APInt(32, uint64_t(-4LL)), APInt(32, "-100", 2));
1081
1082   EXPECT_EQ(APInt(32,  0), APInt(32,  "0",  8));
1083   EXPECT_EQ(APInt(32,  1), APInt(32,  "1",  8));
1084   EXPECT_EQ(APInt(32,  7), APInt(32,  "7",  8));
1085   EXPECT_EQ(APInt(32,  8), APInt(32,  "10", 8));
1086   EXPECT_EQ(APInt(32, 15), APInt(32,  "17", 8));
1087   EXPECT_EQ(APInt(32, 16), APInt(32,  "20", 8));
1088
1089   EXPECT_EQ(APInt(32,  +0), APInt(32,  "+0",  8));
1090   EXPECT_EQ(APInt(32,  +1), APInt(32,  "+1",  8));
1091   EXPECT_EQ(APInt(32,  +7), APInt(32,  "+7",  8));
1092   EXPECT_EQ(APInt(32,  +8), APInt(32,  "+10", 8));
1093   EXPECT_EQ(APInt(32, +15), APInt(32,  "+17", 8));
1094   EXPECT_EQ(APInt(32, +16), APInt(32,  "+20", 8));
1095
1096   EXPECT_EQ(APInt(32,  uint64_t(-0LL)), APInt(32,  "-0",  8));
1097   EXPECT_EQ(APInt(32,  uint64_t(-1LL)), APInt(32,  "-1",  8));
1098   EXPECT_EQ(APInt(32,  uint64_t(-7LL)), APInt(32,  "-7",  8));
1099   EXPECT_EQ(APInt(32,  uint64_t(-8LL)), APInt(32,  "-10", 8));
1100   EXPECT_EQ(APInt(32, uint64_t(-15LL)), APInt(32,  "-17", 8));
1101   EXPECT_EQ(APInt(32, uint64_t(-16LL)), APInt(32,  "-20", 8));
1102
1103   EXPECT_EQ(APInt(32,  0), APInt(32,  "0", 10));
1104   EXPECT_EQ(APInt(32,  1), APInt(32,  "1", 10));
1105   EXPECT_EQ(APInt(32,  9), APInt(32,  "9", 10));
1106   EXPECT_EQ(APInt(32, 10), APInt(32, "10", 10));
1107   EXPECT_EQ(APInt(32, 19), APInt(32, "19", 10));
1108   EXPECT_EQ(APInt(32, 20), APInt(32, "20", 10));
1109
1110   EXPECT_EQ(APInt(32,  uint64_t(-0LL)), APInt(32,  "-0", 10));
1111   EXPECT_EQ(APInt(32,  uint64_t(-1LL)), APInt(32,  "-1", 10));
1112   EXPECT_EQ(APInt(32,  uint64_t(-9LL)), APInt(32,  "-9", 10));
1113   EXPECT_EQ(APInt(32, uint64_t(-10LL)), APInt(32, "-10", 10));
1114   EXPECT_EQ(APInt(32, uint64_t(-19LL)), APInt(32, "-19", 10));
1115   EXPECT_EQ(APInt(32, uint64_t(-20LL)), APInt(32, "-20", 10));
1116
1117   EXPECT_EQ(APInt(32,  0), APInt(32,  "0", 16));
1118   EXPECT_EQ(APInt(32,  1), APInt(32,  "1", 16));
1119   EXPECT_EQ(APInt(32, 15), APInt(32,  "F", 16));
1120   EXPECT_EQ(APInt(32, 16), APInt(32, "10", 16));
1121   EXPECT_EQ(APInt(32, 31), APInt(32, "1F", 16));
1122   EXPECT_EQ(APInt(32, 32), APInt(32, "20", 16));
1123
1124   EXPECT_EQ(APInt(32,  uint64_t(-0LL)), APInt(32,  "-0", 16));
1125   EXPECT_EQ(APInt(32,  uint64_t(-1LL)), APInt(32,  "-1", 16));
1126   EXPECT_EQ(APInt(32, uint64_t(-15LL)), APInt(32,  "-F", 16));
1127   EXPECT_EQ(APInt(32, uint64_t(-16LL)), APInt(32, "-10", 16));
1128   EXPECT_EQ(APInt(32, uint64_t(-31LL)), APInt(32, "-1F", 16));
1129   EXPECT_EQ(APInt(32, uint64_t(-32LL)), APInt(32, "-20", 16));
1130
1131   EXPECT_EQ(APInt(32,  0), APInt(32,  "0", 36));
1132   EXPECT_EQ(APInt(32,  1), APInt(32,  "1", 36));
1133   EXPECT_EQ(APInt(32, 35), APInt(32,  "Z", 36));
1134   EXPECT_EQ(APInt(32, 36), APInt(32, "10", 36));
1135   EXPECT_EQ(APInt(32, 71), APInt(32, "1Z", 36));
1136   EXPECT_EQ(APInt(32, 72), APInt(32, "20", 36));
1137
1138   EXPECT_EQ(APInt(32,  uint64_t(-0LL)), APInt(32,  "-0", 36));
1139   EXPECT_EQ(APInt(32,  uint64_t(-1LL)), APInt(32,  "-1", 36));
1140   EXPECT_EQ(APInt(32, uint64_t(-35LL)), APInt(32,  "-Z", 36));
1141   EXPECT_EQ(APInt(32, uint64_t(-36LL)), APInt(32, "-10", 36));
1142   EXPECT_EQ(APInt(32, uint64_t(-71LL)), APInt(32, "-1Z", 36));
1143   EXPECT_EQ(APInt(32, uint64_t(-72LL)), APInt(32, "-20", 36));
1144 }
1145
1146 TEST(APIntTest, FromArray) {
1147   EXPECT_EQ(APInt(32, uint64_t(1)), APInt(32, ArrayRef<uint64_t>(1)));
1148 }
1149
1150 TEST(APIntTest, StringBitsNeeded2) {
1151   EXPECT_EQ(1U, APInt::getBitsNeeded(  "0", 2));
1152   EXPECT_EQ(1U, APInt::getBitsNeeded(  "1", 2));
1153   EXPECT_EQ(2U, APInt::getBitsNeeded( "10", 2));
1154   EXPECT_EQ(2U, APInt::getBitsNeeded( "11", 2));
1155   EXPECT_EQ(3U, APInt::getBitsNeeded("100", 2));
1156
1157   EXPECT_EQ(1U, APInt::getBitsNeeded(  "+0", 2));
1158   EXPECT_EQ(1U, APInt::getBitsNeeded(  "+1", 2));
1159   EXPECT_EQ(2U, APInt::getBitsNeeded( "+10", 2));
1160   EXPECT_EQ(2U, APInt::getBitsNeeded( "+11", 2));
1161   EXPECT_EQ(3U, APInt::getBitsNeeded("+100", 2));
1162
1163   EXPECT_EQ(2U, APInt::getBitsNeeded(  "-0", 2));
1164   EXPECT_EQ(2U, APInt::getBitsNeeded(  "-1", 2));
1165   EXPECT_EQ(3U, APInt::getBitsNeeded( "-10", 2));
1166   EXPECT_EQ(3U, APInt::getBitsNeeded( "-11", 2));
1167   EXPECT_EQ(4U, APInt::getBitsNeeded("-100", 2));
1168 }
1169
1170 TEST(APIntTest, StringBitsNeeded8) {
1171   EXPECT_EQ(3U, APInt::getBitsNeeded( "0", 8));
1172   EXPECT_EQ(3U, APInt::getBitsNeeded( "7", 8));
1173   EXPECT_EQ(6U, APInt::getBitsNeeded("10", 8));
1174   EXPECT_EQ(6U, APInt::getBitsNeeded("17", 8));
1175   EXPECT_EQ(6U, APInt::getBitsNeeded("20", 8));
1176
1177   EXPECT_EQ(3U, APInt::getBitsNeeded( "+0", 8));
1178   EXPECT_EQ(3U, APInt::getBitsNeeded( "+7", 8));
1179   EXPECT_EQ(6U, APInt::getBitsNeeded("+10", 8));
1180   EXPECT_EQ(6U, APInt::getBitsNeeded("+17", 8));
1181   EXPECT_EQ(6U, APInt::getBitsNeeded("+20", 8));
1182
1183   EXPECT_EQ(4U, APInt::getBitsNeeded( "-0", 8));
1184   EXPECT_EQ(4U, APInt::getBitsNeeded( "-7", 8));
1185   EXPECT_EQ(7U, APInt::getBitsNeeded("-10", 8));
1186   EXPECT_EQ(7U, APInt::getBitsNeeded("-17", 8));
1187   EXPECT_EQ(7U, APInt::getBitsNeeded("-20", 8));
1188 }
1189
1190 TEST(APIntTest, StringBitsNeeded10) {
1191   EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 10));
1192   EXPECT_EQ(2U, APInt::getBitsNeeded( "3", 10));
1193   EXPECT_EQ(4U, APInt::getBitsNeeded( "9", 10));
1194   EXPECT_EQ(4U, APInt::getBitsNeeded("10", 10));
1195   EXPECT_EQ(5U, APInt::getBitsNeeded("19", 10));
1196   EXPECT_EQ(5U, APInt::getBitsNeeded("20", 10));
1197
1198   EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 10));
1199   EXPECT_EQ(4U, APInt::getBitsNeeded( "+9", 10));
1200   EXPECT_EQ(4U, APInt::getBitsNeeded("+10", 10));
1201   EXPECT_EQ(5U, APInt::getBitsNeeded("+19", 10));
1202   EXPECT_EQ(5U, APInt::getBitsNeeded("+20", 10));
1203
1204   EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 10));
1205   EXPECT_EQ(5U, APInt::getBitsNeeded( "-9", 10));
1206   EXPECT_EQ(5U, APInt::getBitsNeeded("-10", 10));
1207   EXPECT_EQ(6U, APInt::getBitsNeeded("-19", 10));
1208   EXPECT_EQ(6U, APInt::getBitsNeeded("-20", 10));
1209 }
1210
1211 TEST(APIntTest, StringBitsNeeded16) {
1212   EXPECT_EQ(4U, APInt::getBitsNeeded( "0", 16));
1213   EXPECT_EQ(4U, APInt::getBitsNeeded( "F", 16));
1214   EXPECT_EQ(8U, APInt::getBitsNeeded("10", 16));
1215   EXPECT_EQ(8U, APInt::getBitsNeeded("1F", 16));
1216   EXPECT_EQ(8U, APInt::getBitsNeeded("20", 16));
1217
1218   EXPECT_EQ(4U, APInt::getBitsNeeded( "+0", 16));
1219   EXPECT_EQ(4U, APInt::getBitsNeeded( "+F", 16));
1220   EXPECT_EQ(8U, APInt::getBitsNeeded("+10", 16));
1221   EXPECT_EQ(8U, APInt::getBitsNeeded("+1F", 16));
1222   EXPECT_EQ(8U, APInt::getBitsNeeded("+20", 16));
1223
1224   EXPECT_EQ(5U, APInt::getBitsNeeded( "-0", 16));
1225   EXPECT_EQ(5U, APInt::getBitsNeeded( "-F", 16));
1226   EXPECT_EQ(9U, APInt::getBitsNeeded("-10", 16));
1227   EXPECT_EQ(9U, APInt::getBitsNeeded("-1F", 16));
1228   EXPECT_EQ(9U, APInt::getBitsNeeded("-20", 16));
1229 }
1230
1231 TEST(APIntTest, toString) {
1232   SmallString<16> S;
1233   bool isSigned;
1234
1235   APInt(8, 0).toString(S, 2, true, true);
1236   EXPECT_EQ(S.str().str(), "0b0");
1237   S.clear();
1238   APInt(8, 0).toString(S, 8, true, true);
1239   EXPECT_EQ(S.str().str(), "00");
1240   S.clear();
1241   APInt(8, 0).toString(S, 10, true, true);
1242   EXPECT_EQ(S.str().str(), "0");
1243   S.clear();
1244   APInt(8, 0).toString(S, 16, true, true);
1245   EXPECT_EQ(S.str().str(), "0x0");
1246   S.clear();
1247   APInt(8, 0).toString(S, 36, true, false);
1248   EXPECT_EQ(S.str().str(), "0");
1249   S.clear();
1250
1251   isSigned = false;
1252   APInt(8, 255, isSigned).toString(S, 2, isSigned, true);
1253   EXPECT_EQ(S.str().str(), "0b11111111");
1254   S.clear();
1255   APInt(8, 255, isSigned).toString(S, 8, isSigned, true);
1256   EXPECT_EQ(S.str().str(), "0377");
1257   S.clear();
1258   APInt(8, 255, isSigned).toString(S, 10, isSigned, true);
1259   EXPECT_EQ(S.str().str(), "255");
1260   S.clear();
1261   APInt(8, 255, isSigned).toString(S, 16, isSigned, true);
1262   EXPECT_EQ(S.str().str(), "0xFF");
1263   S.clear();
1264   APInt(8, 255, isSigned).toString(S, 36, isSigned, false);
1265   EXPECT_EQ(S.str().str(), "73");
1266   S.clear();
1267
1268   isSigned = true;
1269   APInt(8, 255, isSigned).toString(S, 2, isSigned, true);
1270   EXPECT_EQ(S.str().str(), "-0b1");
1271   S.clear();
1272   APInt(8, 255, isSigned).toString(S, 8, isSigned, true);
1273   EXPECT_EQ(S.str().str(), "-01");
1274   S.clear();
1275   APInt(8, 255, isSigned).toString(S, 10, isSigned, true);
1276   EXPECT_EQ(S.str().str(), "-1");
1277   S.clear();
1278   APInt(8, 255, isSigned).toString(S, 16, isSigned, true);
1279   EXPECT_EQ(S.str().str(), "-0x1");
1280   S.clear();
1281   APInt(8, 255, isSigned).toString(S, 36, isSigned, false);
1282   EXPECT_EQ(S.str().str(), "-1");
1283   S.clear();
1284 }
1285
1286 TEST(APIntTest, Log2) {
1287   EXPECT_EQ(APInt(15, 7).logBase2(), 2U);
1288   EXPECT_EQ(APInt(15, 7).ceilLogBase2(), 3U);
1289   EXPECT_EQ(APInt(15, 7).exactLogBase2(), -1);
1290   EXPECT_EQ(APInt(15, 8).logBase2(), 3U);
1291   EXPECT_EQ(APInt(15, 8).ceilLogBase2(), 3U);
1292   EXPECT_EQ(APInt(15, 8).exactLogBase2(), 3);
1293   EXPECT_EQ(APInt(15, 9).logBase2(), 3U);
1294   EXPECT_EQ(APInt(15, 9).ceilLogBase2(), 4U);
1295   EXPECT_EQ(APInt(15, 9).exactLogBase2(), -1);
1296 }
1297
1298 TEST(APIntTest, magic) {
1299   EXPECT_EQ(APInt(32, 3).magic().m, APInt(32, "55555556", 16));
1300   EXPECT_EQ(APInt(32, 3).magic().s, 0U);
1301   EXPECT_EQ(APInt(32, 5).magic().m, APInt(32, "66666667", 16));
1302   EXPECT_EQ(APInt(32, 5).magic().s, 1U);
1303   EXPECT_EQ(APInt(32, 7).magic().m, APInt(32, "92492493", 16));
1304   EXPECT_EQ(APInt(32, 7).magic().s, 2U);
1305 }
1306
1307 TEST(APIntTest, magicu) {
1308   EXPECT_EQ(APInt(32, 3).magicu().m, APInt(32, "AAAAAAAB", 16));
1309   EXPECT_EQ(APInt(32, 3).magicu().s, 1U);
1310   EXPECT_EQ(APInt(32, 5).magicu().m, APInt(32, "CCCCCCCD", 16));
1311   EXPECT_EQ(APInt(32, 5).magicu().s, 2U);
1312   EXPECT_EQ(APInt(32, 7).magicu().m, APInt(32, "24924925", 16));
1313   EXPECT_EQ(APInt(32, 7).magicu().s, 3U);
1314   EXPECT_EQ(APInt(64, 25).magicu(1).m, APInt(64, "A3D70A3D70A3D70B", 16));
1315   EXPECT_EQ(APInt(64, 25).magicu(1).s, 4U);
1316 }
1317
1318 #ifdef GTEST_HAS_DEATH_TEST
1319 #ifndef NDEBUG
1320 TEST(APIntTest, StringDeath) {
1321   EXPECT_DEATH(APInt(0, "", 0), "Bitwidth too small");
1322   EXPECT_DEATH(APInt(32, "", 0), "Invalid string length");
1323   EXPECT_DEATH(APInt(32, "0", 0), "Radix should be 2, 8, 10, 16, or 36!");
1324   EXPECT_DEATH(APInt(32, "", 10), "Invalid string length");
1325   EXPECT_DEATH(APInt(32, "-", 10), "String is only a sign, needs a value.");
1326   EXPECT_DEATH(APInt(1, "1234", 10), "Insufficient bit width");
1327   EXPECT_DEATH(APInt(32, "\0", 10), "Invalid string length");
1328   EXPECT_DEATH(APInt(32, StringRef("1\02", 3), 10), "Invalid character in digit string");
1329   EXPECT_DEATH(APInt(32, "1L", 10), "Invalid character in digit string");
1330 }
1331 #endif
1332 #endif
1333
1334 TEST(APIntTest, mul_clear) {
1335   APInt ValA(65, -1ULL);
1336   APInt ValB(65, 4);
1337   APInt ValC(65, 0);
1338   ValC = ValA * ValB;
1339   ValA *= ValB;
1340   EXPECT_EQ(ValA.toString(10, false), ValC.toString(10, false));
1341 }
1342
1343 TEST(APIntTest, Rotate) {
1344   EXPECT_EQ(APInt(8, 1),  APInt(8, 1).rotl(0));
1345   EXPECT_EQ(APInt(8, 2),  APInt(8, 1).rotl(1));
1346   EXPECT_EQ(APInt(8, 4),  APInt(8, 1).rotl(2));
1347   EXPECT_EQ(APInt(8, 16), APInt(8, 1).rotl(4));
1348   EXPECT_EQ(APInt(8, 1),  APInt(8, 1).rotl(8));
1349
1350   EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotl(0));
1351   EXPECT_EQ(APInt(8, 32), APInt(8, 16).rotl(1));
1352   EXPECT_EQ(APInt(8, 64), APInt(8, 16).rotl(2));
1353   EXPECT_EQ(APInt(8, 1),  APInt(8, 16).rotl(4));
1354   EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotl(8));
1355
1356   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(33));
1357   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(32, 33)));
1358
1359   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(33));
1360   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(32, 33)));
1361   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(33, 33)));
1362   EXPECT_EQ(APInt(32, (1 << 8)), APInt(32, 1).rotl(APInt(32, 40)));
1363   EXPECT_EQ(APInt(32, (1 << 30)), APInt(32, 1).rotl(APInt(31, 30)));
1364   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotl(APInt(31, 31)));
1365
1366   EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotl(APInt(1, 0)));
1367   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(1, 1)));
1368
1369   EXPECT_EQ(APInt(32, 16), APInt(32, 1).rotl(APInt(3, 4)));
1370
1371   EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotl(APInt(64, 64)));
1372   EXPECT_EQ(APInt(32, 2), APInt(32, 1).rotl(APInt(64, 65)));
1373
1374   EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(7, 3)));
1375   EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(7, 10)));
1376   EXPECT_EQ(APInt(7, 24), APInt(7, 3).rotl(APInt(5, 10)));
1377   EXPECT_EQ(APInt(7, 6), APInt(7, 3).rotl(APInt(12, 120)));
1378
1379   EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotr(0));
1380   EXPECT_EQ(APInt(8, 8),  APInt(8, 16).rotr(1));
1381   EXPECT_EQ(APInt(8, 4),  APInt(8, 16).rotr(2));
1382   EXPECT_EQ(APInt(8, 1),  APInt(8, 16).rotr(4));
1383   EXPECT_EQ(APInt(8, 16), APInt(8, 16).rotr(8));
1384
1385   EXPECT_EQ(APInt(8, 1),   APInt(8, 1).rotr(0));
1386   EXPECT_EQ(APInt(8, 128), APInt(8, 1).rotr(1));
1387   EXPECT_EQ(APInt(8, 64),  APInt(8, 1).rotr(2));
1388   EXPECT_EQ(APInt(8, 16),  APInt(8, 1).rotr(4));
1389   EXPECT_EQ(APInt(8, 1),   APInt(8, 1).rotr(8));
1390
1391   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(33));
1392   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(32, 33)));
1393
1394   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(33));
1395   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(32, 33)));
1396   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(33, 33)));
1397   EXPECT_EQ(APInt(32, (1 << 24)), APInt(32, 1).rotr(APInt(32, 40)));
1398
1399   EXPECT_EQ(APInt(32, (1 << 2)), APInt(32, 1).rotr(APInt(31, 30)));
1400   EXPECT_EQ(APInt(32, (1 << 1)), APInt(32, 1).rotr(APInt(31, 31)));
1401
1402   EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotr(APInt(1, 0)));
1403   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(1, 1)));
1404
1405   EXPECT_EQ(APInt(32, (1 << 28)), APInt(32, 1).rotr(APInt(3, 4)));
1406
1407   EXPECT_EQ(APInt(32, 1), APInt(32, 1).rotr(APInt(64, 64)));
1408   EXPECT_EQ(APInt(32, (1 << 31)), APInt(32, 1).rotr(APInt(64, 65)));
1409
1410   EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(7, 3)));
1411   EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(7, 10)));
1412   EXPECT_EQ(APInt(7, 48), APInt(7, 3).rotr(APInt(5, 10)));
1413   EXPECT_EQ(APInt(7, 65), APInt(7, 3).rotr(APInt(12, 120)));
1414
1415   APInt Big(256, "00004000800000000000000000003fff8000000000000003", 16);
1416   APInt Rot(256, "3fff80000000000000030000000000000000000040008000", 16);
1417   EXPECT_EQ(Rot, Big.rotr(144));
1418
1419   EXPECT_EQ(APInt(32, 8), APInt(32, 1).rotl(Big));
1420   EXPECT_EQ(APInt(32, (1 << 29)), APInt(32, 1).rotr(Big));
1421 }
1422
1423 TEST(APIntTest, Splat) {
1424   APInt ValA(8, 0x01);
1425   EXPECT_EQ(ValA, APInt::getSplat(8, ValA));
1426   EXPECT_EQ(APInt(64, 0x0101010101010101ULL), APInt::getSplat(64, ValA));
1427
1428   APInt ValB(3, 5);
1429   EXPECT_EQ(APInt(4, 0xD), APInt::getSplat(4, ValB));
1430   EXPECT_EQ(APInt(15, 0xDB6D), APInt::getSplat(15, ValB));
1431 }
1432
1433 TEST(APIntTest, tcDecrement) {
1434   // Test single word decrement.
1435
1436   // No out borrow.
1437   {
1438     APInt::WordType singleWord = ~APInt::WordType(0) << (APInt::APINT_BITS_PER_WORD - 1);
1439     APInt::WordType carry = APInt::tcDecrement(&singleWord, 1);
1440     EXPECT_EQ(carry, APInt::WordType(0));
1441     EXPECT_EQ(singleWord, ~APInt::WordType(0) >> 1);
1442   }
1443
1444   // With out borrow.
1445   {
1446     APInt::WordType singleWord = 0;
1447     APInt::WordType carry = APInt::tcDecrement(&singleWord, 1);
1448     EXPECT_EQ(carry, APInt::WordType(1));
1449     EXPECT_EQ(singleWord, ~APInt::WordType(0));
1450   }
1451
1452   // Test multiword decrement.
1453
1454   // No across word borrow, no out borrow.
1455   {
1456     APInt::WordType test[4] = {0x1, 0x1, 0x1, 0x1};
1457     APInt::WordType expected[4] = {0x0, 0x1, 0x1, 0x1};
1458     APInt::tcDecrement(test, 4);
1459     EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0);
1460   }
1461
1462   // 1 across word borrow, no out borrow.
1463   {
1464     APInt::WordType test[4] = {0x0, 0xF, 0x1, 0x1};
1465     APInt::WordType expected[4] = {~APInt::WordType(0), 0xE, 0x1, 0x1};
1466     APInt::WordType carry = APInt::tcDecrement(test, 4);
1467     EXPECT_EQ(carry, APInt::WordType(0));
1468     EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0);
1469   }
1470
1471   // 2 across word borrow, no out borrow.
1472   {
1473     APInt::WordType test[4] = {0x0, 0x0, 0xC, 0x1};
1474     APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), 0xB, 0x1};
1475     APInt::WordType carry = APInt::tcDecrement(test, 4);
1476     EXPECT_EQ(carry, APInt::WordType(0));
1477     EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0);
1478   }
1479
1480   // 3 across word borrow, no out borrow.
1481   {
1482     APInt::WordType test[4] = {0x0, 0x0, 0x0, 0x1};
1483     APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0), 0x0};
1484     APInt::WordType carry = APInt::tcDecrement(test, 4);
1485     EXPECT_EQ(carry, APInt::WordType(0));
1486     EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0);
1487   }
1488
1489   // 3 across word borrow, with out borrow.
1490   {
1491     APInt::WordType test[4] = {0x0, 0x0, 0x0, 0x0};
1492     APInt::WordType expected[4] = {~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0), ~APInt::WordType(0)};
1493     APInt::WordType carry = APInt::tcDecrement(test, 4);
1494     EXPECT_EQ(carry, APInt::WordType(1));
1495     EXPECT_EQ(APInt::tcCompare(test, expected, 4), 0);
1496   }
1497 }
1498
1499 TEST(APIntTest, arrayAccess) {
1500   // Single word check.
1501   uint64_t E1 = 0x2CA7F46BF6569915ULL;
1502   APInt A1(64, E1);
1503   for (unsigned i = 0, e = 64; i < e; ++i) {
1504     EXPECT_EQ(bool(E1 & (1ULL << i)),
1505               A1[i]);
1506   }
1507
1508   // Multiword check.
1509   APInt::WordType E2[4] = {
1510     0xEB6EB136591CBA21ULL,
1511     0x7B9358BD6A33F10AULL,
1512     0x7E7FFA5EADD8846ULL,
1513     0x305F341CA00B613DULL
1514   };
1515   APInt A2(APInt::APINT_BITS_PER_WORD*4, E2);
1516   for (unsigned i = 0; i < 4; ++i) {
1517     for (unsigned j = 0; j < APInt::APINT_BITS_PER_WORD; ++j) {
1518       EXPECT_EQ(bool(E2[i] & (1ULL << j)),
1519                 A2[i*APInt::APINT_BITS_PER_WORD + j]);
1520     }
1521   }
1522 }
1523
1524 TEST(APIntTest, LargeAPIntConstruction) {
1525   // Check that we can properly construct very large APInt. It is very
1526   // unlikely that people will ever do this, but it is a legal input,
1527   // so we should not crash on it.
1528   APInt A9(UINT32_MAX, 0);
1529   EXPECT_FALSE(A9.getBoolValue());
1530 }
1531
1532 TEST(APIntTest, nearestLogBase2) {
1533   // Single word check.
1534
1535   // Test round up.
1536   uint64_t I1 = 0x1800001;
1537   APInt A1(64, I1);
1538   EXPECT_EQ(A1.nearestLogBase2(), A1.ceilLogBase2());
1539
1540   // Test round down.
1541   uint64_t I2 = 0x1000011;
1542   APInt A2(64, I2);
1543   EXPECT_EQ(A2.nearestLogBase2(), A2.logBase2());
1544
1545   // Test ties round up.
1546   uint64_t I3 = 0x1800000;
1547   APInt A3(64, I3);
1548   EXPECT_EQ(A3.nearestLogBase2(), A3.ceilLogBase2());
1549
1550   // Multiple word check.
1551
1552   // Test round up.
1553   APInt::WordType I4[4] = {0x0, 0xF, 0x18, 0x0};
1554   APInt A4(APInt::APINT_BITS_PER_WORD*4, I4);
1555   EXPECT_EQ(A4.nearestLogBase2(), A4.ceilLogBase2());
1556
1557   // Test round down.
1558   APInt::WordType I5[4] = {0x0, 0xF, 0x10, 0x0};
1559   APInt A5(APInt::APINT_BITS_PER_WORD*4, I5);
1560   EXPECT_EQ(A5.nearestLogBase2(), A5.logBase2());
1561
1562   // Test ties round up.
1563   uint64_t I6[4] = {0x0, 0x0, 0x0, 0x18};
1564   APInt A6(APInt::APINT_BITS_PER_WORD*4, I6);
1565   EXPECT_EQ(A6.nearestLogBase2(), A6.ceilLogBase2());
1566
1567   // Test BitWidth == 1 special cases.
1568   APInt A7(1, 1);
1569   EXPECT_EQ(A7.nearestLogBase2(), 0ULL);
1570   APInt A8(1, 0);
1571   EXPECT_EQ(A8.nearestLogBase2(), UINT32_MAX);
1572
1573   // Test the zero case when we have a bit width large enough such
1574   // that the bit width is larger than UINT32_MAX-1.
1575   APInt A9(UINT32_MAX, 0);
1576   EXPECT_EQ(A9.nearestLogBase2(), UINT32_MAX);
1577 }
1578
1579 TEST(APIntTest, IsSplat) {
1580   APInt A(32, 0x01010101);
1581   EXPECT_FALSE(A.isSplat(1));
1582   EXPECT_FALSE(A.isSplat(2));
1583   EXPECT_FALSE(A.isSplat(4));
1584   EXPECT_TRUE(A.isSplat(8));
1585   EXPECT_TRUE(A.isSplat(16));
1586   EXPECT_TRUE(A.isSplat(32));
1587
1588   APInt B(24, 0xAAAAAA);
1589   EXPECT_FALSE(B.isSplat(1));
1590   EXPECT_TRUE(B.isSplat(2));
1591   EXPECT_TRUE(B.isSplat(4));
1592   EXPECT_TRUE(B.isSplat(8));
1593   EXPECT_TRUE(B.isSplat(24));
1594
1595   APInt C(24, 0xABAAAB);
1596   EXPECT_FALSE(C.isSplat(1));
1597   EXPECT_FALSE(C.isSplat(2));
1598   EXPECT_FALSE(C.isSplat(4));
1599   EXPECT_FALSE(C.isSplat(8));
1600   EXPECT_TRUE(C.isSplat(24));
1601
1602   APInt D(32, 0xABBAABBA);
1603   EXPECT_FALSE(D.isSplat(1));
1604   EXPECT_FALSE(D.isSplat(2));
1605   EXPECT_FALSE(D.isSplat(4));
1606   EXPECT_FALSE(D.isSplat(8));
1607   EXPECT_TRUE(D.isSplat(16));
1608   EXPECT_TRUE(D.isSplat(32));
1609
1610   APInt E(32, 0);
1611   EXPECT_TRUE(E.isSplat(1));
1612   EXPECT_TRUE(E.isSplat(2));
1613   EXPECT_TRUE(E.isSplat(4));
1614   EXPECT_TRUE(E.isSplat(8));
1615   EXPECT_TRUE(E.isSplat(16));
1616   EXPECT_TRUE(E.isSplat(32));
1617 }
1618
1619 TEST(APIntTest, isMask) {
1620   EXPECT_FALSE(APInt(32, 0x01010101).isMask());
1621   EXPECT_FALSE(APInt(32, 0xf0000000).isMask());
1622   EXPECT_FALSE(APInt(32, 0xffff0000).isMask());
1623   EXPECT_FALSE(APInt(32, 0xff << 1).isMask());
1624
1625   for (int N : { 1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256 }) {
1626     EXPECT_FALSE(APInt(N, 0).isMask());
1627
1628     APInt One(N, 1);
1629     for (int I = 1; I <= N; ++I) {
1630       APInt MaskVal = One.shl(I) - 1;
1631       EXPECT_TRUE(MaskVal.isMask());
1632       EXPECT_TRUE(MaskVal.isMask(I));
1633     }
1634   }
1635 }
1636
1637 TEST(APIntTest, isShiftedMask) {
1638   EXPECT_FALSE(APInt(32, 0x01010101).isShiftedMask());
1639   EXPECT_TRUE(APInt(32, 0xf0000000).isShiftedMask());
1640   EXPECT_TRUE(APInt(32, 0xffff0000).isShiftedMask());
1641   EXPECT_TRUE(APInt(32, 0xff << 1).isShiftedMask());
1642
1643   for (int N : { 1, 2, 3, 4, 7, 8, 16, 32, 64, 127, 128, 129, 256 }) {
1644     EXPECT_FALSE(APInt(N, 0).isShiftedMask());
1645
1646     APInt One(N, 1);
1647     for (int I = 1; I < N; ++I) {
1648       APInt MaskVal = One.shl(I) - 1;
1649       EXPECT_TRUE(MaskVal.isShiftedMask());
1650     }
1651     for (int I = 1; I < N - 1; ++I) {
1652       APInt MaskVal = One.shl(I);
1653       EXPECT_TRUE(MaskVal.isShiftedMask());
1654     }
1655     for (int I = 1; I < N; ++I) {
1656       APInt MaskVal = APInt::getHighBitsSet(N, I);
1657       EXPECT_TRUE(MaskVal.isShiftedMask());
1658     }
1659   }
1660 }
1661
1662 TEST(APIntTest, reverseBits) {
1663   EXPECT_EQ(1, APInt(1, 1).reverseBits());
1664   EXPECT_EQ(0, APInt(1, 0).reverseBits());
1665
1666   EXPECT_EQ(3, APInt(2, 3).reverseBits());
1667   EXPECT_EQ(3, APInt(2, 3).reverseBits());
1668
1669   EXPECT_EQ(0xb, APInt(4, 0xd).reverseBits());
1670   EXPECT_EQ(0xd, APInt(4, 0xb).reverseBits());
1671   EXPECT_EQ(0xf, APInt(4, 0xf).reverseBits());
1672
1673   EXPECT_EQ(0x30, APInt(7, 0x6).reverseBits());
1674   EXPECT_EQ(0x5a, APInt(7, 0x2d).reverseBits());
1675
1676   EXPECT_EQ(0x0f, APInt(8, 0xf0).reverseBits());
1677   EXPECT_EQ(0xf0, APInt(8, 0x0f).reverseBits());
1678
1679   EXPECT_EQ(0x0f0f, APInt(16, 0xf0f0).reverseBits());
1680   EXPECT_EQ(0xf0f0, APInt(16, 0x0f0f).reverseBits());
1681
1682   EXPECT_EQ(0x0f0f0f0f, APInt(32, 0xf0f0f0f0).reverseBits());
1683   EXPECT_EQ(0xf0f0f0f0, APInt(32, 0x0f0f0f0f).reverseBits());
1684
1685   EXPECT_EQ(0x402880a0 >> 1, APInt(31, 0x05011402).reverseBits());
1686
1687   EXPECT_EQ(0x0f0f0f0f, APInt(32, 0xf0f0f0f0).reverseBits());
1688   EXPECT_EQ(0xf0f0f0f0, APInt(32, 0x0f0f0f0f).reverseBits());
1689
1690   EXPECT_EQ(0x0f0f0f0f0f0f0f0f, APInt(64, 0xf0f0f0f0f0f0f0f0).reverseBits());
1691   EXPECT_EQ(0xf0f0f0f0f0f0f0f0, APInt(64, 0x0f0f0f0f0f0f0f0f).reverseBits());
1692
1693   for (unsigned N : { 1, 8, 16, 24, 31, 32, 33,
1694                       63, 64, 65, 127, 128, 257, 1024 }) {
1695     for (unsigned I = 0; I < N; ++I) {
1696       APInt X = APInt::getOneBitSet(N, I);
1697       APInt Y = APInt::getOneBitSet(N, N - (I + 1));
1698       EXPECT_EQ(Y, X.reverseBits());
1699       EXPECT_EQ(X, Y.reverseBits());
1700     }
1701   }
1702 }
1703
1704 TEST(APIntTest, insertBits) {
1705   APInt iSrc(31, 0x00123456);
1706
1707   // Direct copy.
1708   APInt i31(31, 0x76543210ull);
1709   i31.insertBits(iSrc, 0);
1710   EXPECT_EQ(static_cast<int64_t>(0x00123456ull), i31.getSExtValue());
1711
1712   // Single word src/dst insertion.
1713   APInt i63(63, 0x01234567FFFFFFFFull);
1714   i63.insertBits(iSrc, 4);
1715   EXPECT_EQ(static_cast<int64_t>(0x012345600123456Full), i63.getSExtValue());
1716
1717   // Insert single word src into one word of dst.
1718   APInt i120(120, UINT64_MAX, true);
1719   i120.insertBits(iSrc, 8);
1720   EXPECT_EQ(static_cast<int64_t>(0xFFFFFF80123456FFull), i120.getSExtValue());
1721
1722   // Insert single word src into two words of dst.
1723   APInt i127(127, UINT64_MAX, true);
1724   i127.insertBits(iSrc, 48);
1725   EXPECT_EQ(i127.extractBits(64, 0).getZExtValue(), 0x3456FFFFFFFFFFFFull);
1726   EXPECT_EQ(i127.extractBits(63, 64).getZExtValue(), 0x7FFFFFFFFFFF8012ull);
1727
1728   // Insert on word boundaries.
1729   APInt i128(128, 0);
1730   i128.insertBits(APInt(64, UINT64_MAX, true), 0);
1731   i128.insertBits(APInt(64, UINT64_MAX, true), 64);
1732   EXPECT_EQ(-1, i128.getSExtValue());
1733
1734   APInt i256(256, UINT64_MAX, true);
1735   i256.insertBits(APInt(65, 0), 0);
1736   i256.insertBits(APInt(69, 0), 64);
1737   i256.insertBits(APInt(128, 0), 128);
1738   EXPECT_EQ(0u, i256.getSExtValue());
1739
1740   APInt i257(257, 0);
1741   i257.insertBits(APInt(96, UINT64_MAX, true), 64);
1742   EXPECT_EQ(i257.extractBits(64, 0).getZExtValue(), 0x0000000000000000ull);
1743   EXPECT_EQ(i257.extractBits(64, 64).getZExtValue(), 0xFFFFFFFFFFFFFFFFull);
1744   EXPECT_EQ(i257.extractBits(64, 128).getZExtValue(), 0x00000000FFFFFFFFull);
1745   EXPECT_EQ(i257.extractBits(65, 192).getZExtValue(), 0x0000000000000000ull);
1746
1747   // General insertion.
1748   APInt i260(260, UINT64_MAX, true);
1749   i260.insertBits(APInt(129, 1ull << 48), 15);
1750   EXPECT_EQ(i260.extractBits(64, 0).getZExtValue(), 0x8000000000007FFFull);
1751   EXPECT_EQ(i260.extractBits(64, 64).getZExtValue(), 0x0000000000000000ull);
1752   EXPECT_EQ(i260.extractBits(64, 128).getZExtValue(), 0xFFFFFFFFFFFF0000ull);
1753   EXPECT_EQ(i260.extractBits(64, 192).getZExtValue(), 0xFFFFFFFFFFFFFFFFull);
1754   EXPECT_EQ(i260.extractBits(4, 256).getZExtValue(), 0x000000000000000Full);
1755 }
1756
1757 TEST(APIntTest, extractBits) {
1758   APInt i32(32, 0x1234567);
1759   EXPECT_EQ(0x3456, i32.extractBits(16, 4));
1760
1761   APInt i257(257, 0xFFFFFFFFFF0000FFull, true);
1762   EXPECT_EQ(0xFFu, i257.extractBits(16, 0));
1763   EXPECT_EQ((0xFFu >> 1), i257.extractBits(16, 1));
1764   EXPECT_EQ(-1, i257.extractBits(32, 64).getSExtValue());
1765   EXPECT_EQ(-1, i257.extractBits(128, 128).getSExtValue());
1766   EXPECT_EQ(-1, i257.extractBits(66, 191).getSExtValue());
1767   EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full),
1768             i257.extractBits(128, 1).getSExtValue());
1769   EXPECT_EQ(static_cast<int64_t>(0xFFFFFFFFFF80007Full),
1770             i257.extractBits(129, 1).getSExtValue());
1771
1772   EXPECT_EQ(APInt(48, 0),
1773             APInt(144, "281474976710655", 10).extractBits(48, 48));
1774   EXPECT_EQ(APInt(48, 0x0000ffffffffffffull),
1775             APInt(144, "281474976710655", 10).extractBits(48, 0));
1776   EXPECT_EQ(APInt(48, 0x00007fffffffffffull),
1777             APInt(144, "281474976710655", 10).extractBits(48, 1));
1778 }
1779
1780 TEST(APIntTest, getLowBitsSet) {
1781   APInt i128lo64 = APInt::getLowBitsSet(128, 64);
1782   EXPECT_EQ(0u, i128lo64.countLeadingOnes());
1783   EXPECT_EQ(64u, i128lo64.countLeadingZeros());
1784   EXPECT_EQ(64u, i128lo64.getActiveBits());
1785   EXPECT_EQ(0u, i128lo64.countTrailingZeros());
1786   EXPECT_EQ(64u, i128lo64.countTrailingOnes());
1787   EXPECT_EQ(64u, i128lo64.countPopulation());
1788 }
1789
1790 TEST(APIntTest, getBitsSet) {
1791   APInt i64hi1lo1 = APInt::getBitsSet(64, 1, 63);
1792   EXPECT_EQ(0u, i64hi1lo1.countLeadingOnes());
1793   EXPECT_EQ(1u, i64hi1lo1.countLeadingZeros());
1794   EXPECT_EQ(63u, i64hi1lo1.getActiveBits());
1795   EXPECT_EQ(1u, i64hi1lo1.countTrailingZeros());
1796   EXPECT_EQ(0u, i64hi1lo1.countTrailingOnes());
1797   EXPECT_EQ(62u, i64hi1lo1.countPopulation());
1798
1799   APInt i127hi1lo1 = APInt::getBitsSet(127, 1, 126);
1800   EXPECT_EQ(0u, i127hi1lo1.countLeadingOnes());
1801   EXPECT_EQ(1u, i127hi1lo1.countLeadingZeros());
1802   EXPECT_EQ(126u, i127hi1lo1.getActiveBits());
1803   EXPECT_EQ(1u, i127hi1lo1.countTrailingZeros());
1804   EXPECT_EQ(0u, i127hi1lo1.countTrailingOnes());
1805   EXPECT_EQ(125u, i127hi1lo1.countPopulation());
1806 }
1807
1808 TEST(APIntTest, getHighBitsSet) {
1809   APInt i64hi32 = APInt::getHighBitsSet(64, 32);
1810   EXPECT_EQ(32u, i64hi32.countLeadingOnes());
1811   EXPECT_EQ(0u, i64hi32.countLeadingZeros());
1812   EXPECT_EQ(64u, i64hi32.getActiveBits());
1813   EXPECT_EQ(32u, i64hi32.countTrailingZeros());
1814   EXPECT_EQ(0u, i64hi32.countTrailingOnes());
1815   EXPECT_EQ(32u, i64hi32.countPopulation());
1816 }
1817
1818 TEST(APIntTest, getBitsSetFrom) {
1819   APInt i64hi31 = APInt::getBitsSetFrom(64, 33);
1820   EXPECT_EQ(31u, i64hi31.countLeadingOnes());
1821   EXPECT_EQ(0u, i64hi31.countLeadingZeros());
1822   EXPECT_EQ(64u, i64hi31.getActiveBits());
1823   EXPECT_EQ(33u, i64hi31.countTrailingZeros());
1824   EXPECT_EQ(0u, i64hi31.countTrailingOnes());
1825   EXPECT_EQ(31u, i64hi31.countPopulation());
1826 }
1827
1828 TEST(APIntTest, setLowBits) {
1829   APInt i64lo32(64, 0);
1830   i64lo32.setLowBits(32);
1831   EXPECT_EQ(0u, i64lo32.countLeadingOnes());
1832   EXPECT_EQ(32u, i64lo32.countLeadingZeros());
1833   EXPECT_EQ(32u, i64lo32.getActiveBits());
1834   EXPECT_EQ(0u, i64lo32.countTrailingZeros());
1835   EXPECT_EQ(32u, i64lo32.countTrailingOnes());
1836   EXPECT_EQ(32u, i64lo32.countPopulation());
1837
1838   APInt i128lo64(128, 0);
1839   i128lo64.setLowBits(64);
1840   EXPECT_EQ(0u, i128lo64.countLeadingOnes());
1841   EXPECT_EQ(64u, i128lo64.countLeadingZeros());
1842   EXPECT_EQ(64u, i128lo64.getActiveBits());
1843   EXPECT_EQ(0u, i128lo64.countTrailingZeros());
1844   EXPECT_EQ(64u, i128lo64.countTrailingOnes());
1845   EXPECT_EQ(64u, i128lo64.countPopulation());
1846
1847   APInt i128lo24(128, 0);
1848   i128lo24.setLowBits(24);
1849   EXPECT_EQ(0u, i128lo24.countLeadingOnes());
1850   EXPECT_EQ(104u, i128lo24.countLeadingZeros());
1851   EXPECT_EQ(24u, i128lo24.getActiveBits());
1852   EXPECT_EQ(0u, i128lo24.countTrailingZeros());
1853   EXPECT_EQ(24u, i128lo24.countTrailingOnes());
1854   EXPECT_EQ(24u, i128lo24.countPopulation());
1855
1856   APInt i128lo104(128, 0);
1857   i128lo104.setLowBits(104);
1858   EXPECT_EQ(0u, i128lo104.countLeadingOnes());
1859   EXPECT_EQ(24u, i128lo104.countLeadingZeros());
1860   EXPECT_EQ(104u, i128lo104.getActiveBits());
1861   EXPECT_EQ(0u, i128lo104.countTrailingZeros());
1862   EXPECT_EQ(104u, i128lo104.countTrailingOnes());
1863   EXPECT_EQ(104u, i128lo104.countPopulation());
1864
1865   APInt i128lo0(128, 0);
1866   i128lo0.setLowBits(0);
1867   EXPECT_EQ(0u, i128lo0.countLeadingOnes());
1868   EXPECT_EQ(128u, i128lo0.countLeadingZeros());
1869   EXPECT_EQ(0u, i128lo0.getActiveBits());
1870   EXPECT_EQ(128u, i128lo0.countTrailingZeros());
1871   EXPECT_EQ(0u, i128lo0.countTrailingOnes());
1872   EXPECT_EQ(0u, i128lo0.countPopulation());
1873
1874   APInt i80lo79(80, 0);
1875   i80lo79.setLowBits(79);
1876   EXPECT_EQ(0u, i80lo79.countLeadingOnes());
1877   EXPECT_EQ(1u, i80lo79.countLeadingZeros());
1878   EXPECT_EQ(79u, i80lo79.getActiveBits());
1879   EXPECT_EQ(0u, i80lo79.countTrailingZeros());
1880   EXPECT_EQ(79u, i80lo79.countTrailingOnes());
1881   EXPECT_EQ(79u, i80lo79.countPopulation());
1882 }
1883
1884 TEST(APIntTest, setHighBits) {
1885   APInt i64hi32(64, 0);
1886   i64hi32.setHighBits(32);
1887   EXPECT_EQ(32u, i64hi32.countLeadingOnes());
1888   EXPECT_EQ(0u, i64hi32.countLeadingZeros());
1889   EXPECT_EQ(64u, i64hi32.getActiveBits());
1890   EXPECT_EQ(32u, i64hi32.countTrailingZeros());
1891   EXPECT_EQ(0u, i64hi32.countTrailingOnes());
1892   EXPECT_EQ(32u, i64hi32.countPopulation());
1893
1894   APInt i128hi64(128, 0);
1895   i128hi64.setHighBits(64);
1896   EXPECT_EQ(64u, i128hi64.countLeadingOnes());
1897   EXPECT_EQ(0u, i128hi64.countLeadingZeros());
1898   EXPECT_EQ(128u, i128hi64.getActiveBits());
1899   EXPECT_EQ(64u, i128hi64.countTrailingZeros());
1900   EXPECT_EQ(0u, i128hi64.countTrailingOnes());
1901   EXPECT_EQ(64u, i128hi64.countPopulation());
1902
1903   APInt i128hi24(128, 0);
1904   i128hi24.setHighBits(24);
1905   EXPECT_EQ(24u, i128hi24.countLeadingOnes());
1906   EXPECT_EQ(0u, i128hi24.countLeadingZeros());
1907   EXPECT_EQ(128u, i128hi24.getActiveBits());
1908   EXPECT_EQ(104u, i128hi24.countTrailingZeros());
1909   EXPECT_EQ(0u, i128hi24.countTrailingOnes());
1910   EXPECT_EQ(24u, i128hi24.countPopulation());
1911
1912   APInt i128hi104(128, 0);
1913   i128hi104.setHighBits(104);
1914   EXPECT_EQ(104u, i128hi104.countLeadingOnes());
1915   EXPECT_EQ(0u, i128hi104.countLeadingZeros());
1916   EXPECT_EQ(128u, i128hi104.getActiveBits());
1917   EXPECT_EQ(24u, i128hi104.countTrailingZeros());
1918   EXPECT_EQ(0u, i128hi104.countTrailingOnes());
1919   EXPECT_EQ(104u, i128hi104.countPopulation());
1920
1921   APInt i128hi0(128, 0);
1922   i128hi0.setHighBits(0);
1923   EXPECT_EQ(0u, i128hi0.countLeadingOnes());
1924   EXPECT_EQ(128u, i128hi0.countLeadingZeros());
1925   EXPECT_EQ(0u, i128hi0.getActiveBits());
1926   EXPECT_EQ(128u, i128hi0.countTrailingZeros());
1927   EXPECT_EQ(0u, i128hi0.countTrailingOnes());
1928   EXPECT_EQ(0u, i128hi0.countPopulation());
1929
1930   APInt i80hi1(80, 0);
1931   i80hi1.setHighBits(1);
1932   EXPECT_EQ(1u, i80hi1.countLeadingOnes());
1933   EXPECT_EQ(0u, i80hi1.countLeadingZeros());
1934   EXPECT_EQ(80u, i80hi1.getActiveBits());
1935   EXPECT_EQ(79u, i80hi1.countTrailingZeros());
1936   EXPECT_EQ(0u, i80hi1.countTrailingOnes());
1937   EXPECT_EQ(1u, i80hi1.countPopulation());
1938
1939   APInt i32hi16(32, 0);
1940   i32hi16.setHighBits(16);
1941   EXPECT_EQ(16u, i32hi16.countLeadingOnes());
1942   EXPECT_EQ(0u, i32hi16.countLeadingZeros());
1943   EXPECT_EQ(32u, i32hi16.getActiveBits());
1944   EXPECT_EQ(16u, i32hi16.countTrailingZeros());
1945   EXPECT_EQ(0u, i32hi16.countTrailingOnes());
1946   EXPECT_EQ(16u, i32hi16.countPopulation());
1947 }
1948
1949 TEST(APIntTest, setBitsFrom) {
1950   APInt i64from63(64, 0);
1951   i64from63.setBitsFrom(63);
1952   EXPECT_EQ(1u, i64from63.countLeadingOnes());
1953   EXPECT_EQ(0u, i64from63.countLeadingZeros());
1954   EXPECT_EQ(64u, i64from63.getActiveBits());
1955   EXPECT_EQ(63u, i64from63.countTrailingZeros());
1956   EXPECT_EQ(0u, i64from63.countTrailingOnes());
1957   EXPECT_EQ(1u, i64from63.countPopulation());
1958 }
1959
1960 TEST(APIntTest, setAllBits) {
1961   APInt i32(32, 0);
1962   i32.setAllBits();
1963   EXPECT_EQ(32u, i32.countLeadingOnes());
1964   EXPECT_EQ(0u, i32.countLeadingZeros());
1965   EXPECT_EQ(32u, i32.getActiveBits());
1966   EXPECT_EQ(0u, i32.countTrailingZeros());
1967   EXPECT_EQ(32u, i32.countTrailingOnes());
1968   EXPECT_EQ(32u, i32.countPopulation());
1969
1970   APInt i64(64, 0);
1971   i64.setAllBits();
1972   EXPECT_EQ(64u, i64.countLeadingOnes());
1973   EXPECT_EQ(0u, i64.countLeadingZeros());
1974   EXPECT_EQ(64u, i64.getActiveBits());
1975   EXPECT_EQ(0u, i64.countTrailingZeros());
1976   EXPECT_EQ(64u, i64.countTrailingOnes());
1977   EXPECT_EQ(64u, i64.countPopulation());
1978
1979   APInt i96(96, 0);
1980   i96.setAllBits();
1981   EXPECT_EQ(96u, i96.countLeadingOnes());
1982   EXPECT_EQ(0u, i96.countLeadingZeros());
1983   EXPECT_EQ(96u, i96.getActiveBits());
1984   EXPECT_EQ(0u, i96.countTrailingZeros());
1985   EXPECT_EQ(96u, i96.countTrailingOnes());
1986   EXPECT_EQ(96u, i96.countPopulation());
1987
1988   APInt i128(128, 0);
1989   i128.setAllBits();
1990   EXPECT_EQ(128u, i128.countLeadingOnes());
1991   EXPECT_EQ(0u, i128.countLeadingZeros());
1992   EXPECT_EQ(128u, i128.getActiveBits());
1993   EXPECT_EQ(0u, i128.countTrailingZeros());
1994   EXPECT_EQ(128u, i128.countTrailingOnes());
1995   EXPECT_EQ(128u, i128.countPopulation());
1996 }
1997
1998 TEST(APIntTest, getLoBits) {
1999   APInt i32(32, 0xfa);
2000   i32.setHighBits(1);
2001   EXPECT_EQ(0xa, i32.getLoBits(4));
2002   APInt i128(128, 0xfa);
2003   i128.setHighBits(1);
2004   EXPECT_EQ(0xa, i128.getLoBits(4));
2005 }
2006
2007 TEST(APIntTest, getHiBits) {
2008   APInt i32(32, 0xfa);
2009   i32.setHighBits(2);
2010   EXPECT_EQ(0xc, i32.getHiBits(4));
2011   APInt i128(128, 0xfa);
2012   i128.setHighBits(2);
2013   EXPECT_EQ(0xc, i128.getHiBits(4));
2014 }
2015
2016 TEST(APIntTest, GCD) {
2017   using APIntOps::GreatestCommonDivisor;
2018
2019   for (unsigned Bits : {1, 2, 32, 63, 64, 65}) {
2020     // Test some corner cases near zero.
2021     APInt Zero(Bits, 0), One(Bits, 1);
2022     EXPECT_EQ(GreatestCommonDivisor(Zero, Zero), Zero);
2023     EXPECT_EQ(GreatestCommonDivisor(Zero, One), One);
2024     EXPECT_EQ(GreatestCommonDivisor(One, Zero), One);
2025     EXPECT_EQ(GreatestCommonDivisor(One, One), One);
2026
2027     if (Bits > 1) {
2028       APInt Two(Bits, 2);
2029       EXPECT_EQ(GreatestCommonDivisor(Zero, Two), Two);
2030       EXPECT_EQ(GreatestCommonDivisor(One, Two), One);
2031       EXPECT_EQ(GreatestCommonDivisor(Two, Two), Two);
2032
2033       // Test some corner cases near the highest representable value.
2034       APInt Max(Bits, 0);
2035       Max.setAllBits();
2036       EXPECT_EQ(GreatestCommonDivisor(Zero, Max), Max);
2037       EXPECT_EQ(GreatestCommonDivisor(One, Max), One);
2038       EXPECT_EQ(GreatestCommonDivisor(Two, Max), One);
2039       EXPECT_EQ(GreatestCommonDivisor(Max, Max), Max);
2040
2041       APInt MaxOver2 = Max.udiv(Two);
2042       EXPECT_EQ(GreatestCommonDivisor(MaxOver2, Max), One);
2043       // Max - 1 == Max / 2 * 2, because Max is odd.
2044       EXPECT_EQ(GreatestCommonDivisor(MaxOver2, Max - 1), MaxOver2);
2045     }
2046   }
2047
2048   // Compute the 20th Mersenne prime.
2049   const unsigned BitWidth = 4450;
2050   APInt HugePrime = APInt::getLowBitsSet(BitWidth, 4423);
2051
2052   // 9931 and 123456 are coprime.
2053   APInt A = HugePrime * APInt(BitWidth, 9931);
2054   APInt B = HugePrime * APInt(BitWidth, 123456);
2055   APInt C = GreatestCommonDivisor(A, B);
2056   EXPECT_EQ(C, HugePrime);
2057 }
2058
2059 TEST(APIntTest, LogicalRightShift) {
2060   APInt i256(APInt::getHighBitsSet(256, 2));
2061
2062   i256.lshrInPlace(1);
2063   EXPECT_EQ(1U, i256.countLeadingZeros());
2064   EXPECT_EQ(253U, i256.countTrailingZeros());
2065   EXPECT_EQ(2U, i256.countPopulation());
2066
2067   i256.lshrInPlace(62);
2068   EXPECT_EQ(63U, i256.countLeadingZeros());
2069   EXPECT_EQ(191U, i256.countTrailingZeros());
2070   EXPECT_EQ(2U, i256.countPopulation());
2071
2072   i256.lshrInPlace(65);
2073   EXPECT_EQ(128U, i256.countLeadingZeros());
2074   EXPECT_EQ(126U, i256.countTrailingZeros());
2075   EXPECT_EQ(2U, i256.countPopulation());
2076
2077   i256.lshrInPlace(64);
2078   EXPECT_EQ(192U, i256.countLeadingZeros());
2079   EXPECT_EQ(62U, i256.countTrailingZeros());
2080   EXPECT_EQ(2U, i256.countPopulation());
2081
2082   i256.lshrInPlace(63);
2083   EXPECT_EQ(255U, i256.countLeadingZeros());
2084   EXPECT_EQ(0U, i256.countTrailingZeros());
2085   EXPECT_EQ(1U, i256.countPopulation());
2086
2087   // Ensure we handle large shifts of multi-word.
2088   const APInt neg_one(128, static_cast<uint64_t>(-1), true);
2089   EXPECT_EQ(0, neg_one.lshr(128));
2090 }
2091
2092 TEST(APIntTest, ArithmeticRightShift) {
2093   APInt i72(APInt::getHighBitsSet(72, 1));
2094   i72.ashrInPlace(46);
2095   EXPECT_EQ(47U, i72.countLeadingOnes());
2096   EXPECT_EQ(25U, i72.countTrailingZeros());
2097   EXPECT_EQ(47U, i72.countPopulation());
2098
2099   i72 = APInt::getHighBitsSet(72, 1);
2100   i72.ashrInPlace(64);
2101   EXPECT_EQ(65U, i72.countLeadingOnes());
2102   EXPECT_EQ(7U, i72.countTrailingZeros());
2103   EXPECT_EQ(65U, i72.countPopulation());
2104
2105   APInt i128(APInt::getHighBitsSet(128, 1));
2106   i128.ashrInPlace(64);
2107   EXPECT_EQ(65U, i128.countLeadingOnes());
2108   EXPECT_EQ(63U, i128.countTrailingZeros());
2109   EXPECT_EQ(65U, i128.countPopulation());
2110
2111   // Ensure we handle large shifts of multi-word.
2112   const APInt signmin32(APInt::getSignedMinValue(32));
2113   EXPECT_TRUE(signmin32.ashr(32).isAllOnesValue());
2114
2115   // Ensure we handle large shifts of multi-word.
2116   const APInt umax32(APInt::getSignedMaxValue(32));
2117   EXPECT_EQ(0, umax32.ashr(32));
2118
2119   // Ensure we handle large shifts of multi-word.
2120   const APInt signmin128(APInt::getSignedMinValue(128));
2121   EXPECT_TRUE(signmin128.ashr(128).isAllOnesValue());
2122
2123   // Ensure we handle large shifts of multi-word.
2124   const APInt umax128(APInt::getSignedMaxValue(128));
2125   EXPECT_EQ(0, umax128.ashr(128));
2126 }
2127
2128 TEST(APIntTest, LeftShift) {
2129   APInt i256(APInt::getLowBitsSet(256, 2));
2130
2131   i256 <<= 1;
2132   EXPECT_EQ(253U, i256.countLeadingZeros());
2133   EXPECT_EQ(1U, i256.countTrailingZeros());
2134   EXPECT_EQ(2U, i256.countPopulation());
2135
2136   i256 <<= 62;
2137   EXPECT_EQ(191U, i256.countLeadingZeros());
2138   EXPECT_EQ(63U, i256.countTrailingZeros());
2139   EXPECT_EQ(2U, i256.countPopulation());
2140
2141   i256 <<= 65;
2142   EXPECT_EQ(126U, i256.countLeadingZeros());
2143   EXPECT_EQ(128U, i256.countTrailingZeros());
2144   EXPECT_EQ(2U, i256.countPopulation());
2145
2146   i256 <<= 64;
2147   EXPECT_EQ(62U, i256.countLeadingZeros());
2148   EXPECT_EQ(192U, i256.countTrailingZeros());
2149   EXPECT_EQ(2U, i256.countPopulation());
2150
2151   i256 <<= 63;
2152   EXPECT_EQ(0U, i256.countLeadingZeros());
2153   EXPECT_EQ(255U, i256.countTrailingZeros());
2154   EXPECT_EQ(1U, i256.countPopulation());
2155
2156   // Ensure we handle large shifts of multi-word.
2157   const APInt neg_one(128, static_cast<uint64_t>(-1), true);
2158   EXPECT_EQ(0, neg_one.shl(128));
2159 }
2160
2161 TEST(APIntTest, isSubsetOf) {
2162   APInt i32_1(32, 1);
2163   APInt i32_2(32, 2);
2164   APInt i32_3(32, 3);
2165   EXPECT_FALSE(i32_3.isSubsetOf(i32_1));
2166   EXPECT_TRUE(i32_1.isSubsetOf(i32_3));
2167   EXPECT_FALSE(i32_2.isSubsetOf(i32_1));
2168   EXPECT_FALSE(i32_1.isSubsetOf(i32_2));
2169   EXPECT_TRUE(i32_3.isSubsetOf(i32_3));
2170
2171   APInt i128_1(128, 1);
2172   APInt i128_2(128, 2);
2173   APInt i128_3(128, 3);
2174   EXPECT_FALSE(i128_3.isSubsetOf(i128_1));
2175   EXPECT_TRUE(i128_1.isSubsetOf(i128_3));
2176   EXPECT_FALSE(i128_2.isSubsetOf(i128_1));
2177   EXPECT_FALSE(i128_1.isSubsetOf(i128_2));
2178   EXPECT_TRUE(i128_3.isSubsetOf(i128_3));
2179
2180   i128_1 <<= 64;
2181   i128_2 <<= 64;
2182   i128_3 <<= 64;
2183   EXPECT_FALSE(i128_3.isSubsetOf(i128_1));
2184   EXPECT_TRUE(i128_1.isSubsetOf(i128_3));
2185   EXPECT_FALSE(i128_2.isSubsetOf(i128_1));
2186   EXPECT_FALSE(i128_1.isSubsetOf(i128_2));
2187   EXPECT_TRUE(i128_3.isSubsetOf(i128_3));
2188 }
2189
2190 TEST(APIntTest, sext) {
2191   EXPECT_EQ(0, APInt(1, 0).sext(64));
2192   EXPECT_EQ(~uint64_t(0), APInt(1, 1).sext(64));
2193
2194   APInt i32_max(APInt::getSignedMaxValue(32).sext(63));
2195   EXPECT_EQ(32U, i32_max.countLeadingZeros());
2196   EXPECT_EQ(0U, i32_max.countTrailingZeros());
2197   EXPECT_EQ(31U, i32_max.countPopulation());
2198
2199   APInt i32_min(APInt::getSignedMinValue(32).sext(63));
2200   EXPECT_EQ(32U, i32_min.countLeadingOnes());
2201   EXPECT_EQ(31U, i32_min.countTrailingZeros());
2202   EXPECT_EQ(32U, i32_min.countPopulation());
2203
2204   APInt i32_neg1(APInt(32, ~uint64_t(0)).sext(63));
2205   EXPECT_EQ(63U, i32_neg1.countLeadingOnes());
2206   EXPECT_EQ(0U, i32_neg1.countTrailingZeros());
2207   EXPECT_EQ(63U, i32_neg1.countPopulation());
2208 }
2209
2210 TEST(APIntTest, multiply) {
2211   APInt i64(64, 1234);
2212
2213   EXPECT_EQ(7006652, i64 * 5678);
2214   EXPECT_EQ(7006652, 5678 * i64);
2215
2216   APInt i128 = APInt::getOneBitSet(128, 64);
2217   APInt i128_1234(128, 1234);
2218   i128_1234 <<= 64;
2219   EXPECT_EQ(i128_1234, i128 * 1234);
2220   EXPECT_EQ(i128_1234, 1234 * i128);
2221
2222   APInt i96 = APInt::getOneBitSet(96, 64);
2223   i96 *= ~0ULL;
2224   EXPECT_EQ(32U, i96.countLeadingOnes());
2225   EXPECT_EQ(32U, i96.countPopulation());
2226   EXPECT_EQ(64U, i96.countTrailingZeros());
2227 }
2228
2229 } // end anonymous namespace