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 <ostream> 11 #include "llvm/Support/raw_ostream.h" 12 #include "gtest/gtest.h" 13 #include "llvm/ADT/APInt.h" 14 #include "llvm/ADT/SmallString.h" 15 16 using namespace llvm; 17 18 namespace { 19 20 // Support APInt output to an std::ostream. 21 inline std::ostream &operator<<(std::ostream &OS, const APInt &Value) { 22 raw_os_ostream RawOS(OS); 23 RawOS << Value; 24 return OS; 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_EQ(true, Shl[0]); 32 EXPECT_EQ(false, Shl[1]); 33 } 34 35 TEST(APIntTest, i128_NegativeCount) { 36 APInt Minus3(128, static_cast<uint64_t>(-3), true); 37 EXPECT_EQ(126u, Minus3.countLeadingOnes()); 38 EXPECT_EQ(-3, Minus3.getSExtValue()); 39 40 APInt Minus1(128, static_cast<uint64_t>(-1), true); 41 EXPECT_EQ(0u, Minus1.countLeadingZeros()); 42 EXPECT_EQ(128u, Minus1.countLeadingOnes()); 43 EXPECT_EQ(128u, Minus1.getActiveBits()); 44 EXPECT_EQ(0u, Minus1.countTrailingZeros()); 45 EXPECT_EQ(128u, Minus1.countTrailingOnes()); 46 EXPECT_EQ(128u, Minus1.countPopulation()); 47 EXPECT_EQ(-1, Minus1.getSExtValue()); 48 } 49 50 TEST(APIntTest, i33_Count) { 51 APInt i33minus2(33, static_cast<uint64_t>(-2), true); 52 EXPECT_EQ(0u, i33minus2.countLeadingZeros()); 53 EXPECT_EQ(32u, i33minus2.countLeadingOnes()); 54 EXPECT_EQ(33u, i33minus2.getActiveBits()); 55 EXPECT_EQ(1u, i33minus2.countTrailingZeros()); 56 EXPECT_EQ(32u, i33minus2.countPopulation()); 57 EXPECT_EQ(-2, i33minus2.getSExtValue()); 58 EXPECT_EQ(((uint64_t)-2)&((1ull<<33) -1), i33minus2.getZExtValue()); 59 } 60 61 TEST(APIntTest, i65_Count) { 62 APInt i65minus(65, 0, true); 63 i65minus.set(64); 64 EXPECT_EQ(0u, i65minus.countLeadingZeros()); 65 EXPECT_EQ(1u, i65minus.countLeadingOnes()); 66 EXPECT_EQ(65u, i65minus.getActiveBits()); 67 EXPECT_EQ(64u, i65minus.countTrailingZeros()); 68 EXPECT_EQ(1u, i65minus.countPopulation()); 69 } 70 71 TEST(APIntTest, i128_PositiveCount) { 72 APInt u128max = APInt::getAllOnesValue(128); 73 EXPECT_EQ(128u, u128max.countLeadingOnes()); 74 EXPECT_EQ(0u, u128max.countLeadingZeros()); 75 EXPECT_EQ(128u, u128max.getActiveBits()); 76 EXPECT_EQ(0u, u128max.countTrailingZeros()); 77 EXPECT_EQ(128u, u128max.countTrailingOnes()); 78 EXPECT_EQ(128u, u128max.countPopulation()); 79 80 APInt u64max(128, static_cast<uint64_t>(-1), false); 81 EXPECT_EQ(64u, u64max.countLeadingZeros()); 82 EXPECT_EQ(0u, u64max.countLeadingOnes()); 83 EXPECT_EQ(64u, u64max.getActiveBits()); 84 EXPECT_EQ(0u, u64max.countTrailingZeros()); 85 EXPECT_EQ(64u, u64max.countTrailingOnes()); 86 EXPECT_EQ(64u, u64max.countPopulation()); 87 EXPECT_EQ((uint64_t)~0ull, u64max.getZExtValue()); 88 89 APInt zero(128, 0, true); 90 EXPECT_EQ(128u, zero.countLeadingZeros()); 91 EXPECT_EQ(0u, zero.countLeadingOnes()); 92 EXPECT_EQ(0u, zero.getActiveBits()); 93 EXPECT_EQ(128u, zero.countTrailingZeros()); 94 EXPECT_EQ(0u, zero.countTrailingOnes()); 95 EXPECT_EQ(0u, zero.countPopulation()); 96 EXPECT_EQ(0u, zero.getSExtValue()); 97 EXPECT_EQ(0u, zero.getZExtValue()); 98 99 APInt one(128, 1, true); 100 EXPECT_EQ(127u, one.countLeadingZeros()); 101 EXPECT_EQ(0u, one.countLeadingOnes()); 102 EXPECT_EQ(1u, one.getActiveBits()); 103 EXPECT_EQ(0u, one.countTrailingZeros()); 104 EXPECT_EQ(1u, one.countTrailingOnes()); 105 EXPECT_EQ(1u, one.countPopulation()); 106 EXPECT_EQ(1, one.getSExtValue()); 107 EXPECT_EQ(1u, one.getZExtValue()); 108 } 109 110 TEST(APIntTest, i1) { 111 const APInt neg_two(1, static_cast<uint64_t>(-2), true); 112 const APInt neg_one(1, static_cast<uint64_t>(-1), true); 113 const APInt zero(1, 0); 114 const APInt one(1, 1); 115 const APInt two(1, 2); 116 117 EXPECT_EQ(0, neg_two.getSExtValue()); 118 EXPECT_EQ(-1, neg_one.getSExtValue()); 119 EXPECT_EQ(1u, neg_one.getZExtValue()); 120 EXPECT_EQ(0u, zero.getZExtValue()); 121 EXPECT_EQ(-1, one.getSExtValue()); 122 EXPECT_EQ(1u, one.getZExtValue()); 123 EXPECT_EQ(0u, two.getZExtValue()); 124 EXPECT_EQ(0, two.getSExtValue()); 125 126 // Basic equalities for 1-bit values. 127 EXPECT_EQ(zero, two); 128 EXPECT_EQ(zero, neg_two); 129 EXPECT_EQ(one, neg_one); 130 EXPECT_EQ(two, neg_two); 131 132 // Additions. 133 EXPECT_EQ(two, one + one); 134 EXPECT_EQ(zero, neg_one + one); 135 EXPECT_EQ(neg_two, neg_one + neg_one); 136 137 // Subtractions. 138 EXPECT_EQ(neg_two, neg_one - one); 139 EXPECT_EQ(two, one - neg_one); 140 EXPECT_EQ(zero, one - one); 141 142 // Shifts. 143 EXPECT_EQ(zero, one << one); 144 EXPECT_EQ(one, one << zero); 145 EXPECT_EQ(zero, one.shl(1)); 146 EXPECT_EQ(one, one.shl(0)); 147 EXPECT_EQ(zero, one.lshr(1)); 148 EXPECT_EQ(zero, one.ashr(1)); 149 150 // Multiplies. 151 EXPECT_EQ(neg_one, neg_one * one); 152 EXPECT_EQ(neg_one, one * neg_one); 153 EXPECT_EQ(one, neg_one * neg_one); 154 EXPECT_EQ(one, one * one); 155 156 // Divides. 157 EXPECT_EQ(neg_one, one.sdiv(neg_one)); 158 EXPECT_EQ(neg_one, neg_one.sdiv(one)); 159 EXPECT_EQ(one, neg_one.sdiv(neg_one)); 160 EXPECT_EQ(one, one.sdiv(one)); 161 162 EXPECT_EQ(neg_one, one.udiv(neg_one)); 163 EXPECT_EQ(neg_one, neg_one.udiv(one)); 164 EXPECT_EQ(one, neg_one.udiv(neg_one)); 165 EXPECT_EQ(one, one.udiv(one)); 166 167 // Remainders. 168 EXPECT_EQ(zero, neg_one.srem(one)); 169 EXPECT_EQ(zero, neg_one.urem(one)); 170 EXPECT_EQ(zero, one.srem(neg_one)); 171 } 172 173 TEST(APIntTest, fromString) { 174 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 2)); 175 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 2)); 176 EXPECT_EQ(APInt(32, 2), APInt(32, "10", 2)); 177 EXPECT_EQ(APInt(32, 3), APInt(32, "11", 2)); 178 EXPECT_EQ(APInt(32, 4), APInt(32, "100", 2)); 179 180 EXPECT_EQ(APInt(32, 0), APInt(32, "+0", 2)); 181 EXPECT_EQ(APInt(32, 1), APInt(32, "+1", 2)); 182 EXPECT_EQ(APInt(32, 2), APInt(32, "+10", 2)); 183 EXPECT_EQ(APInt(32, 3), APInt(32, "+11", 2)); 184 EXPECT_EQ(APInt(32, 4), APInt(32, "+100", 2)); 185 186 EXPECT_EQ(APInt(32, -0), APInt(32, "-0", 2)); 187 EXPECT_EQ(APInt(32, -1), APInt(32, "-1", 2)); 188 EXPECT_EQ(APInt(32, -2), APInt(32, "-10", 2)); 189 EXPECT_EQ(APInt(32, -3), APInt(32, "-11", 2)); 190 EXPECT_EQ(APInt(32, -4), APInt(32, "-100", 2)); 191 192 193 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 8)); 194 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 8)); 195 EXPECT_EQ(APInt(32, 7), APInt(32, "7", 8)); 196 EXPECT_EQ(APInt(32, 8), APInt(32, "10", 8)); 197 EXPECT_EQ(APInt(32, 15), APInt(32, "17", 8)); 198 EXPECT_EQ(APInt(32, 16), APInt(32, "20", 8)); 199 200 EXPECT_EQ(APInt(32, +0), APInt(32, "+0", 8)); 201 EXPECT_EQ(APInt(32, +1), APInt(32, "+1", 8)); 202 EXPECT_EQ(APInt(32, +7), APInt(32, "+7", 8)); 203 EXPECT_EQ(APInt(32, +8), APInt(32, "+10", 8)); 204 EXPECT_EQ(APInt(32, +15), APInt(32, "+17", 8)); 205 EXPECT_EQ(APInt(32, +16), APInt(32, "+20", 8)); 206 207 EXPECT_EQ(APInt(32, -0), APInt(32, "-0", 8)); 208 EXPECT_EQ(APInt(32, -1), APInt(32, "-1", 8)); 209 EXPECT_EQ(APInt(32, -7), APInt(32, "-7", 8)); 210 EXPECT_EQ(APInt(32, -8), APInt(32, "-10", 8)); 211 EXPECT_EQ(APInt(32, -15), APInt(32, "-17", 8)); 212 EXPECT_EQ(APInt(32, -16), APInt(32, "-20", 8)); 213 214 215 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 10)); 216 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 10)); 217 EXPECT_EQ(APInt(32, 9), APInt(32, "9", 10)); 218 EXPECT_EQ(APInt(32, 10), APInt(32, "10", 10)); 219 EXPECT_EQ(APInt(32, 19), APInt(32, "19", 10)); 220 EXPECT_EQ(APInt(32, 20), APInt(32, "20", 10)); 221 222 EXPECT_EQ(APInt(32, -0), APInt(32, "-0", 10)); 223 EXPECT_EQ(APInt(32, -1), APInt(32, "-1", 10)); 224 EXPECT_EQ(APInt(32, -9), APInt(32, "-9", 10)); 225 EXPECT_EQ(APInt(32, -10), APInt(32, "-10", 10)); 226 EXPECT_EQ(APInt(32, -19), APInt(32, "-19", 10)); 227 EXPECT_EQ(APInt(32, -20), APInt(32, "-20", 10)); 228 229 230 EXPECT_EQ(APInt(32, 0), APInt(32, "0", 16)); 231 EXPECT_EQ(APInt(32, 1), APInt(32, "1", 16)); 232 EXPECT_EQ(APInt(32, 15), APInt(32, "F", 16)); 233 EXPECT_EQ(APInt(32, 16), APInt(32, "10", 16)); 234 EXPECT_EQ(APInt(32, 31), APInt(32, "1F", 16)); 235 EXPECT_EQ(APInt(32, 32), APInt(32, "20", 16)); 236 237 EXPECT_EQ(APInt(32, -0), APInt(32, "-0", 16)); 238 EXPECT_EQ(APInt(32, -1), APInt(32, "-1", 16)); 239 EXPECT_EQ(APInt(32, -15), APInt(32, "-F", 16)); 240 EXPECT_EQ(APInt(32, -16), APInt(32, "-10", 16)); 241 EXPECT_EQ(APInt(32, -31), APInt(32, "-1F", 16)); 242 EXPECT_EQ(APInt(32, -32), APInt(32, "-20", 16)); 243 } 244 245 TEST(APIntTest, StringBitsNeeded2) { 246 EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 2)); 247 EXPECT_EQ(1U, APInt::getBitsNeeded( "1", 2)); 248 EXPECT_EQ(2U, APInt::getBitsNeeded( "10", 2)); 249 EXPECT_EQ(2U, APInt::getBitsNeeded( "11", 2)); 250 EXPECT_EQ(3U, APInt::getBitsNeeded("100", 2)); 251 252 EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 2)); 253 EXPECT_EQ(1U, APInt::getBitsNeeded( "+1", 2)); 254 EXPECT_EQ(2U, APInt::getBitsNeeded( "+10", 2)); 255 EXPECT_EQ(2U, APInt::getBitsNeeded( "+11", 2)); 256 EXPECT_EQ(3U, APInt::getBitsNeeded("+100", 2)); 257 258 EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 2)); 259 EXPECT_EQ(2U, APInt::getBitsNeeded( "-1", 2)); 260 EXPECT_EQ(3U, APInt::getBitsNeeded( "-10", 2)); 261 EXPECT_EQ(3U, APInt::getBitsNeeded( "-11", 2)); 262 EXPECT_EQ(4U, APInt::getBitsNeeded("-100", 2)); 263 } 264 265 TEST(APIntTest, StringBitsNeeded8) { 266 EXPECT_EQ(3U, APInt::getBitsNeeded( "0", 8)); 267 EXPECT_EQ(3U, APInt::getBitsNeeded( "7", 8)); 268 EXPECT_EQ(6U, APInt::getBitsNeeded("10", 8)); 269 EXPECT_EQ(6U, APInt::getBitsNeeded("17", 8)); 270 EXPECT_EQ(6U, APInt::getBitsNeeded("20", 8)); 271 272 EXPECT_EQ(3U, APInt::getBitsNeeded( "+0", 8)); 273 EXPECT_EQ(3U, APInt::getBitsNeeded( "+7", 8)); 274 EXPECT_EQ(6U, APInt::getBitsNeeded("+10", 8)); 275 EXPECT_EQ(6U, APInt::getBitsNeeded("+17", 8)); 276 EXPECT_EQ(6U, APInt::getBitsNeeded("+20", 8)); 277 278 EXPECT_EQ(4U, APInt::getBitsNeeded( "-0", 8)); 279 EXPECT_EQ(4U, APInt::getBitsNeeded( "-7", 8)); 280 EXPECT_EQ(7U, APInt::getBitsNeeded("-10", 8)); 281 EXPECT_EQ(7U, APInt::getBitsNeeded("-17", 8)); 282 EXPECT_EQ(7U, APInt::getBitsNeeded("-20", 8)); 283 } 284 285 TEST(APIntTest, StringBitsNeeded10) { 286 EXPECT_EQ(1U, APInt::getBitsNeeded( "0", 10)); 287 EXPECT_EQ(2U, APInt::getBitsNeeded( "3", 10)); 288 EXPECT_EQ(4U, APInt::getBitsNeeded( "9", 10)); 289 EXPECT_EQ(4U, APInt::getBitsNeeded("10", 10)); 290 EXPECT_EQ(5U, APInt::getBitsNeeded("19", 10)); 291 EXPECT_EQ(5U, APInt::getBitsNeeded("20", 10)); 292 293 EXPECT_EQ(1U, APInt::getBitsNeeded( "+0", 10)); 294 EXPECT_EQ(4U, APInt::getBitsNeeded( "+9", 10)); 295 EXPECT_EQ(4U, APInt::getBitsNeeded("+10", 10)); 296 EXPECT_EQ(5U, APInt::getBitsNeeded("+19", 10)); 297 EXPECT_EQ(5U, APInt::getBitsNeeded("+20", 10)); 298 299 EXPECT_EQ(2U, APInt::getBitsNeeded( "-0", 10)); 300 EXPECT_EQ(5U, APInt::getBitsNeeded( "-9", 10)); 301 EXPECT_EQ(5U, APInt::getBitsNeeded("-10", 10)); 302 EXPECT_EQ(6U, APInt::getBitsNeeded("-19", 10)); 303 EXPECT_EQ(6U, APInt::getBitsNeeded("-20", 10)); 304 } 305 306 TEST(APIntTest, StringBitsNeeded16) { 307 EXPECT_EQ(4U, APInt::getBitsNeeded( "0", 16)); 308 EXPECT_EQ(4U, APInt::getBitsNeeded( "F", 16)); 309 EXPECT_EQ(8U, APInt::getBitsNeeded("10", 16)); 310 EXPECT_EQ(8U, APInt::getBitsNeeded("1F", 16)); 311 EXPECT_EQ(8U, APInt::getBitsNeeded("20", 16)); 312 313 EXPECT_EQ(4U, APInt::getBitsNeeded( "+0", 16)); 314 EXPECT_EQ(4U, APInt::getBitsNeeded( "+F", 16)); 315 EXPECT_EQ(8U, APInt::getBitsNeeded("+10", 16)); 316 EXPECT_EQ(8U, APInt::getBitsNeeded("+1F", 16)); 317 EXPECT_EQ(8U, APInt::getBitsNeeded("+20", 16)); 318 319 EXPECT_EQ(5U, APInt::getBitsNeeded( "-0", 16)); 320 EXPECT_EQ(5U, APInt::getBitsNeeded( "-F", 16)); 321 EXPECT_EQ(9U, APInt::getBitsNeeded("-10", 16)); 322 EXPECT_EQ(9U, APInt::getBitsNeeded("-1F", 16)); 323 EXPECT_EQ(9U, APInt::getBitsNeeded("-20", 16)); 324 } 325 326 327 #ifdef GTEST_HAS_DEATH_TEST 328 TEST(APIntTest, StringDeath) { 329 EXPECT_DEATH(APInt(0, "", 0), "Bitwidth too small"); 330 EXPECT_DEATH(APInt(32, "", 0), "Invalid string length"); 331 EXPECT_DEATH(APInt(32, "0", 0), "Radix should be 2, 8, 10, or 16!"); 332 EXPECT_DEATH(APInt(32, "", 10), "Invalid string length"); 333 EXPECT_DEATH(APInt(32, "-", 10), "String is only a sign, needs a value."); 334 EXPECT_DEATH(APInt(1, "1234", 10), "Insufficient bit width"); 335 EXPECT_DEATH(APInt(32, "\0", 10), "Invalid string length"); 336 EXPECT_DEATH(APInt(32, StringRef("1\02", 3), 10), "Invalid character in digit string"); 337 EXPECT_DEATH(APInt(32, "1L", 10), "Invalid character in digit string"); 338 } 339 #endif 340 341 } 342