blob: d5ce5035acaa80a92113e7bcb74157e1b08a399a [file] [log] [blame]
Zhou Shengdac63782007-02-06 03:00:16 +00001//===-- APInt.cpp - Implement APInt class ---------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattnerf3ebc3f2007-12-29 20:36:04 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Zhou Shengdac63782007-02-06 03:00:16 +00007//
8//===----------------------------------------------------------------------===//
9//
Reid Spencera41e93b2007-02-25 19:32:03 +000010// This file implements a class to represent arbitrary precision integer
11// constant values and provide a variety of arithmetic operations on them.
Zhou Shengdac63782007-02-06 03:00:16 +000012//
13//===----------------------------------------------------------------------===//
14
Reid Spencera5e0d202007-02-24 03:58:46 +000015#define DEBUG_TYPE "apint"
Zhou Shengdac63782007-02-06 03:00:16 +000016#include "llvm/ADT/APInt.h"
Ted Kremenek5c75d542008-01-19 04:23:33 +000017#include "llvm/ADT/FoldingSet.h"
Reid Spencera5e0d202007-02-24 03:58:46 +000018#include "llvm/Support/Debug.h"
Zhou Shengdac63782007-02-06 03:00:16 +000019#include "llvm/Support/MathExtras.h"
Jeff Cohenb622c112007-03-05 00:00:42 +000020#include <math.h>
Jeff Cohene06855e2007-03-20 20:42:36 +000021#include <limits>
Zhou Sheng3e8022d2007-02-07 06:14:53 +000022#include <cstring>
Zhou Shengdac63782007-02-06 03:00:16 +000023#include <cstdlib>
Reid Spencer1ba83352007-02-21 03:55:44 +000024#include <iomanip>
Reid Spencer1ba83352007-02-21 03:55:44 +000025
Zhou Shengdac63782007-02-06 03:00:16 +000026using namespace llvm;
27
Reid Spencer0b1df1d2007-12-11 06:53:58 +000028/// This enumeration just provides for internal constants used in this
29/// translation unit.
30enum {
31 MIN_INT_BITS = 1, ///< Minimum number of bits that can be specified
32 ///< Note that this must remain synchronized with IntegerType::MIN_INT_BITS
33 MAX_INT_BITS = (1<<23)-1 ///< Maximum number of bits that can be specified
34 ///< Note that this must remain synchronized with IntegerType::MAX_INT_BITS
35};
36
Reid Spencera41e93b2007-02-25 19:32:03 +000037/// A utility function for allocating memory, checking for allocation failures,
38/// and ensuring the contents are zeroed.
Reid Spencera856b6e2007-02-18 18:38:44 +000039inline static uint64_t* getClearedMemory(uint32_t numWords) {
40 uint64_t * result = new uint64_t[numWords];
41 assert(result && "APInt memory allocation fails!");
42 memset(result, 0, numWords * sizeof(uint64_t));
43 return result;
Zhou Sheng94b623a2007-02-06 06:04:53 +000044}
45
Reid Spencera41e93b2007-02-25 19:32:03 +000046/// A utility function for allocating memory and checking for allocation
47/// failure. The content is not zeroed.
Reid Spencera856b6e2007-02-18 18:38:44 +000048inline static uint64_t* getMemory(uint32_t numWords) {
49 uint64_t * result = new uint64_t[numWords];
50 assert(result && "APInt memory allocation fails!");
51 return result;
52}
53
Reid Spencerb1ec2e82007-03-19 21:19:02 +000054APInt::APInt(uint32_t numBits, uint64_t val, bool isSigned)
Reid Spencer6fae35a2007-03-19 20:37:47 +000055 : BitWidth(numBits), VAL(0) {
Reid Spencer0b1df1d2007-12-11 06:53:58 +000056 assert(BitWidth >= MIN_INT_BITS && "bitwidth too small");
57 assert(BitWidth <= MAX_INT_BITS && "bitwidth too large");
Reid Spencera41e93b2007-02-25 19:32:03 +000058 if (isSingleWord())
59 VAL = val;
Zhou Shengdac63782007-02-06 03:00:16 +000060 else {
Reid Spencera856b6e2007-02-18 18:38:44 +000061 pVal = getClearedMemory(getNumWords());
Zhou Shengdac63782007-02-06 03:00:16 +000062 pVal[0] = val;
Reid Spencer6fae35a2007-03-19 20:37:47 +000063 if (isSigned && int64_t(val) < 0)
64 for (unsigned i = 1; i < getNumWords(); ++i)
65 pVal[i] = -1ULL;
Zhou Shengdac63782007-02-06 03:00:16 +000066 }
Reid Spencera41e93b2007-02-25 19:32:03 +000067 clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +000068}
69
Dale Johannesen42305122007-09-21 22:09:37 +000070APInt::APInt(uint32_t numBits, uint32_t numWords, const uint64_t bigVal[])
Reid Spencer1ba83352007-02-21 03:55:44 +000071 : BitWidth(numBits), VAL(0) {
Reid Spencer0b1df1d2007-12-11 06:53:58 +000072 assert(BitWidth >= MIN_INT_BITS && "bitwidth too small");
73 assert(BitWidth <= MAX_INT_BITS && "bitwidth too large");
Zhou Shengdac63782007-02-06 03:00:16 +000074 assert(bigVal && "Null pointer detected!");
75 if (isSingleWord())
Reid Spencerdf6cf5a2007-02-24 10:01:42 +000076 VAL = bigVal[0];
Zhou Shengdac63782007-02-06 03:00:16 +000077 else {
Reid Spencerdf6cf5a2007-02-24 10:01:42 +000078 // Get memory, cleared to 0
79 pVal = getClearedMemory(getNumWords());
80 // Calculate the number of words to copy
81 uint32_t words = std::min<uint32_t>(numWords, getNumWords());
82 // Copy the words from bigVal to pVal
83 memcpy(pVal, bigVal, words * APINT_WORD_SIZE);
Zhou Shengdac63782007-02-06 03:00:16 +000084 }
Reid Spencerdf6cf5a2007-02-24 10:01:42 +000085 // Make sure unused high bits are cleared
86 clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +000087}
88
Reid Spencera856b6e2007-02-18 18:38:44 +000089APInt::APInt(uint32_t numbits, const char StrStart[], uint32_t slen,
Reid Spencerfb77b2b2007-02-20 08:51:03 +000090 uint8_t radix)
Reid Spencer1ba83352007-02-21 03:55:44 +000091 : BitWidth(numbits), VAL(0) {
Reid Spencer0b1df1d2007-12-11 06:53:58 +000092 assert(BitWidth >= MIN_INT_BITS && "bitwidth too small");
93 assert(BitWidth <= MAX_INT_BITS && "bitwidth too large");
Reid Spencer1d072122007-02-16 22:36:51 +000094 fromString(numbits, StrStart, slen, radix);
Zhou Sheng3e8022d2007-02-07 06:14:53 +000095}
96
Reid Spencerfb77b2b2007-02-20 08:51:03 +000097APInt::APInt(uint32_t numbits, const std::string& Val, uint8_t radix)
Reid Spencer1ba83352007-02-21 03:55:44 +000098 : BitWidth(numbits), VAL(0) {
Reid Spencer0b1df1d2007-12-11 06:53:58 +000099 assert(BitWidth >= MIN_INT_BITS && "bitwidth too small");
100 assert(BitWidth <= MAX_INT_BITS && "bitwidth too large");
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000101 assert(!Val.empty() && "String empty?");
Reid Spencer1d072122007-02-16 22:36:51 +0000102 fromString(numbits, Val.c_str(), Val.size(), radix);
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000103}
104
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000105APInt::APInt(const APInt& that)
Reid Spencer1ba83352007-02-21 03:55:44 +0000106 : BitWidth(that.BitWidth), VAL(0) {
Reid Spencer0b1df1d2007-12-11 06:53:58 +0000107 assert(BitWidth >= MIN_INT_BITS && "bitwidth too small");
108 assert(BitWidth <= MAX_INT_BITS && "bitwidth too large");
Reid Spencera856b6e2007-02-18 18:38:44 +0000109 if (isSingleWord())
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000110 VAL = that.VAL;
Zhou Shengdac63782007-02-06 03:00:16 +0000111 else {
Reid Spencera856b6e2007-02-18 18:38:44 +0000112 pVal = getMemory(getNumWords());
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000113 memcpy(pVal, that.pVal, getNumWords() * APINT_WORD_SIZE);
Zhou Shengdac63782007-02-06 03:00:16 +0000114 }
115}
116
117APInt::~APInt() {
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000118 if (!isSingleWord() && pVal)
Reid Spencer7c16cd22007-02-26 23:38:21 +0000119 delete [] pVal;
Zhou Shengdac63782007-02-06 03:00:16 +0000120}
121
Zhou Shengdac63782007-02-06 03:00:16 +0000122APInt& APInt::operator=(const APInt& RHS) {
Reid Spencer7c16cd22007-02-26 23:38:21 +0000123 // Don't do anything for X = X
124 if (this == &RHS)
125 return *this;
126
127 // If the bitwidths are the same, we can avoid mucking with memory
128 if (BitWidth == RHS.getBitWidth()) {
129 if (isSingleWord())
130 VAL = RHS.VAL;
131 else
132 memcpy(pVal, RHS.pVal, getNumWords() * APINT_WORD_SIZE);
133 return *this;
134 }
135
136 if (isSingleWord())
137 if (RHS.isSingleWord())
138 VAL = RHS.VAL;
139 else {
140 VAL = 0;
141 pVal = getMemory(RHS.getNumWords());
142 memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
143 }
144 else if (getNumWords() == RHS.getNumWords())
145 memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
146 else if (RHS.isSingleWord()) {
147 delete [] pVal;
Reid Spencera856b6e2007-02-18 18:38:44 +0000148 VAL = RHS.VAL;
Reid Spencer7c16cd22007-02-26 23:38:21 +0000149 } else {
150 delete [] pVal;
151 pVal = getMemory(RHS.getNumWords());
152 memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
153 }
154 BitWidth = RHS.BitWidth;
155 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000156}
157
Zhou Shengdac63782007-02-06 03:00:16 +0000158APInt& APInt::operator=(uint64_t RHS) {
Reid Spencer1d072122007-02-16 22:36:51 +0000159 if (isSingleWord())
160 VAL = RHS;
Zhou Shengdac63782007-02-06 03:00:16 +0000161 else {
162 pVal[0] = RHS;
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000163 memset(pVal+1, 0, (getNumWords() - 1) * APINT_WORD_SIZE);
Zhou Shengdac63782007-02-06 03:00:16 +0000164 }
Reid Spencer7c16cd22007-02-26 23:38:21 +0000165 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000166}
167
Ted Kremenek5c75d542008-01-19 04:23:33 +0000168/// Profile - This method 'profiles' an APInt for use with FoldingSet.
169void APInt::Profile(FoldingSetNodeID& ID) const {
Ted Kremenek901540f2008-02-19 20:50:41 +0000170 ID.AddInteger(BitWidth);
171
Ted Kremenek5c75d542008-01-19 04:23:33 +0000172 if (isSingleWord()) {
173 ID.AddInteger(VAL);
174 return;
175 }
176
177 uint32_t NumWords = getNumWords();
178 for (unsigned i = 0; i < NumWords; ++i)
179 ID.AddInteger(pVal[i]);
180}
181
Reid Spencera856b6e2007-02-18 18:38:44 +0000182/// add_1 - This function adds a single "digit" integer, y, to the multiple
183/// "digit" integer array, x[]. x[] is modified to reflect the addition and
184/// 1 is returned if there is a carry out, otherwise 0 is returned.
Reid Spencer100502d2007-02-17 03:16:00 +0000185/// @returns the carry of the addition.
Reid Spencera41e93b2007-02-25 19:32:03 +0000186static bool add_1(uint64_t dest[], uint64_t x[], uint32_t len, uint64_t y) {
Reid Spencera856b6e2007-02-18 18:38:44 +0000187 for (uint32_t i = 0; i < len; ++i) {
Reid Spenceree0a6852007-02-18 06:39:42 +0000188 dest[i] = y + x[i];
189 if (dest[i] < y)
Reid Spencerdf6cf5a2007-02-24 10:01:42 +0000190 y = 1; // Carry one to next digit.
Reid Spenceree0a6852007-02-18 06:39:42 +0000191 else {
Reid Spencerdf6cf5a2007-02-24 10:01:42 +0000192 y = 0; // No need to carry so exit early
Reid Spenceree0a6852007-02-18 06:39:42 +0000193 break;
194 }
Reid Spencer100502d2007-02-17 03:16:00 +0000195 }
Reid Spenceree0a6852007-02-18 06:39:42 +0000196 return y;
Reid Spencer100502d2007-02-17 03:16:00 +0000197}
198
Zhou Shengdac63782007-02-06 03:00:16 +0000199/// @brief Prefix increment operator. Increments the APInt by one.
200APInt& APInt::operator++() {
Reid Spencer1d072122007-02-16 22:36:51 +0000201 if (isSingleWord())
202 ++VAL;
Zhou Shengdac63782007-02-06 03:00:16 +0000203 else
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000204 add_1(pVal, pVal, getNumWords(), 1);
Reid Spencera41e93b2007-02-25 19:32:03 +0000205 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000206}
207
Reid Spencera856b6e2007-02-18 18:38:44 +0000208/// sub_1 - This function subtracts a single "digit" (64-bit word), y, from
209/// the multi-digit integer array, x[], propagating the borrowed 1 value until
210/// no further borrowing is neeeded or it runs out of "digits" in x. The result
211/// is 1 if "borrowing" exhausted the digits in x, or 0 if x was not exhausted.
212/// In other words, if y > x then this function returns 1, otherwise 0.
Reid Spencera41e93b2007-02-25 19:32:03 +0000213/// @returns the borrow out of the subtraction
214static bool sub_1(uint64_t x[], uint32_t len, uint64_t y) {
Reid Spencera856b6e2007-02-18 18:38:44 +0000215 for (uint32_t i = 0; i < len; ++i) {
Reid Spencer100502d2007-02-17 03:16:00 +0000216 uint64_t X = x[i];
Reid Spenceree0a6852007-02-18 06:39:42 +0000217 x[i] -= y;
218 if (y > X)
Reid Spencera856b6e2007-02-18 18:38:44 +0000219 y = 1; // We have to "borrow 1" from next "digit"
Reid Spencer100502d2007-02-17 03:16:00 +0000220 else {
Reid Spencera856b6e2007-02-18 18:38:44 +0000221 y = 0; // No need to borrow
222 break; // Remaining digits are unchanged so exit early
Reid Spencer100502d2007-02-17 03:16:00 +0000223 }
224 }
Reid Spencera41e93b2007-02-25 19:32:03 +0000225 return bool(y);
Reid Spencer100502d2007-02-17 03:16:00 +0000226}
227
Zhou Shengdac63782007-02-06 03:00:16 +0000228/// @brief Prefix decrement operator. Decrements the APInt by one.
229APInt& APInt::operator--() {
Reid Spencera856b6e2007-02-18 18:38:44 +0000230 if (isSingleWord())
231 --VAL;
Zhou Shengdac63782007-02-06 03:00:16 +0000232 else
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000233 sub_1(pVal, getNumWords(), 1);
Reid Spencera41e93b2007-02-25 19:32:03 +0000234 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000235}
236
Reid Spencera41e93b2007-02-25 19:32:03 +0000237/// add - This function adds the integer array x to the integer array Y and
238/// places the result in dest.
239/// @returns the carry out from the addition
240/// @brief General addition of 64-bit integer arrays
Reid Spencera5e0d202007-02-24 03:58:46 +0000241static bool add(uint64_t *dest, const uint64_t *x, const uint64_t *y,
242 uint32_t len) {
243 bool carry = false;
Reid Spencera856b6e2007-02-18 18:38:44 +0000244 for (uint32_t i = 0; i< len; ++i) {
Reid Spencercb292e42007-02-23 01:57:13 +0000245 uint64_t limit = std::min(x[i],y[i]); // must come first in case dest == x
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000246 dest[i] = x[i] + y[i] + carry;
Reid Spencerdb2abec2007-02-21 05:44:56 +0000247 carry = dest[i] < limit || (carry && dest[i] == limit);
Reid Spencer100502d2007-02-17 03:16:00 +0000248 }
249 return carry;
250}
251
Reid Spencera41e93b2007-02-25 19:32:03 +0000252/// Adds the RHS APint to this APInt.
253/// @returns this, after addition of RHS.
254/// @brief Addition assignment operator.
Zhou Shengdac63782007-02-06 03:00:16 +0000255APInt& APInt::operator+=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000256 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000257 if (isSingleWord())
258 VAL += RHS.VAL;
Zhou Shengdac63782007-02-06 03:00:16 +0000259 else {
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000260 add(pVal, pVal, RHS.pVal, getNumWords());
Zhou Shengdac63782007-02-06 03:00:16 +0000261 }
Reid Spencera41e93b2007-02-25 19:32:03 +0000262 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000263}
264
Reid Spencera41e93b2007-02-25 19:32:03 +0000265/// Subtracts the integer array y from the integer array x
266/// @returns returns the borrow out.
267/// @brief Generalized subtraction of 64-bit integer arrays.
Reid Spencera5e0d202007-02-24 03:58:46 +0000268static bool sub(uint64_t *dest, const uint64_t *x, const uint64_t *y,
269 uint32_t len) {
Reid Spencer1ba83352007-02-21 03:55:44 +0000270 bool borrow = false;
Reid Spencera856b6e2007-02-18 18:38:44 +0000271 for (uint32_t i = 0; i < len; ++i) {
Reid Spencer1ba83352007-02-21 03:55:44 +0000272 uint64_t x_tmp = borrow ? x[i] - 1 : x[i];
273 borrow = y[i] > x_tmp || (borrow && x[i] == 0);
274 dest[i] = x_tmp - y[i];
Reid Spencer100502d2007-02-17 03:16:00 +0000275 }
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000276 return borrow;
Reid Spencer100502d2007-02-17 03:16:00 +0000277}
278
Reid Spencera41e93b2007-02-25 19:32:03 +0000279/// Subtracts the RHS APInt from this APInt
280/// @returns this, after subtraction
281/// @brief Subtraction assignment operator.
Zhou Shengdac63782007-02-06 03:00:16 +0000282APInt& APInt::operator-=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000283 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Zhou Shengdac63782007-02-06 03:00:16 +0000284 if (isSingleWord())
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000285 VAL -= RHS.VAL;
286 else
287 sub(pVal, pVal, RHS.pVal, getNumWords());
Reid Spencera41e93b2007-02-25 19:32:03 +0000288 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000289}
290
Reid Spencera41e93b2007-02-25 19:32:03 +0000291/// Multiplies an integer array, x by a a uint64_t integer and places the result
292/// into dest.
293/// @returns the carry out of the multiplication.
294/// @brief Multiply a multi-digit APInt by a single digit (64-bit) integer.
Reid Spencerdf6cf5a2007-02-24 10:01:42 +0000295static uint64_t mul_1(uint64_t dest[], uint64_t x[], uint32_t len, uint64_t y) {
296 // Split y into high 32-bit part (hy) and low 32-bit part (ly)
Reid Spencer100502d2007-02-17 03:16:00 +0000297 uint64_t ly = y & 0xffffffffULL, hy = y >> 32;
Reid Spencera41e93b2007-02-25 19:32:03 +0000298 uint64_t carry = 0;
299
300 // For each digit of x.
Reid Spencera856b6e2007-02-18 18:38:44 +0000301 for (uint32_t i = 0; i < len; ++i) {
Reid Spencera41e93b2007-02-25 19:32:03 +0000302 // Split x into high and low words
303 uint64_t lx = x[i] & 0xffffffffULL;
304 uint64_t hx = x[i] >> 32;
305 // hasCarry - A flag to indicate if there is a carry to the next digit.
Reid Spencer100502d2007-02-17 03:16:00 +0000306 // hasCarry == 0, no carry
307 // hasCarry == 1, has carry
308 // hasCarry == 2, no carry and the calculation result == 0.
309 uint8_t hasCarry = 0;
310 dest[i] = carry + lx * ly;
311 // Determine if the add above introduces carry.
312 hasCarry = (dest[i] < carry) ? 1 : 0;
313 carry = hx * ly + (dest[i] >> 32) + (hasCarry ? (1ULL << 32) : 0);
314 // The upper limit of carry can be (2^32 - 1)(2^32 - 1) +
315 // (2^32 - 1) + 2^32 = 2^64.
316 hasCarry = (!carry && hasCarry) ? 1 : (!carry ? 2 : 0);
317
318 carry += (lx * hy) & 0xffffffffULL;
319 dest[i] = (carry << 32) | (dest[i] & 0xffffffffULL);
320 carry = (((!carry && hasCarry != 2) || hasCarry == 1) ? (1ULL << 32) : 0) +
321 (carry >> 32) + ((lx * hy) >> 32) + hx * hy;
322 }
Reid Spencer100502d2007-02-17 03:16:00 +0000323 return carry;
324}
325
Reid Spencera41e93b2007-02-25 19:32:03 +0000326/// Multiplies integer array x by integer array y and stores the result into
327/// the integer array dest. Note that dest's size must be >= xlen + ylen.
328/// @brief Generalized multiplicate of integer arrays.
Reid Spencerdf6cf5a2007-02-24 10:01:42 +0000329static void mul(uint64_t dest[], uint64_t x[], uint32_t xlen, uint64_t y[],
330 uint32_t ylen) {
Reid Spencer100502d2007-02-17 03:16:00 +0000331 dest[xlen] = mul_1(dest, x, xlen, y[0]);
Reid Spencera856b6e2007-02-18 18:38:44 +0000332 for (uint32_t i = 1; i < ylen; ++i) {
Reid Spencer100502d2007-02-17 03:16:00 +0000333 uint64_t ly = y[i] & 0xffffffffULL, hy = y[i] >> 32;
Reid Spencer58a6a432007-02-21 08:21:52 +0000334 uint64_t carry = 0, lx = 0, hx = 0;
Reid Spencera856b6e2007-02-18 18:38:44 +0000335 for (uint32_t j = 0; j < xlen; ++j) {
Reid Spencer100502d2007-02-17 03:16:00 +0000336 lx = x[j] & 0xffffffffULL;
337 hx = x[j] >> 32;
338 // hasCarry - A flag to indicate if has carry.
339 // hasCarry == 0, no carry
340 // hasCarry == 1, has carry
341 // hasCarry == 2, no carry and the calculation result == 0.
342 uint8_t hasCarry = 0;
343 uint64_t resul = carry + lx * ly;
344 hasCarry = (resul < carry) ? 1 : 0;
345 carry = (hasCarry ? (1ULL << 32) : 0) + hx * ly + (resul >> 32);
346 hasCarry = (!carry && hasCarry) ? 1 : (!carry ? 2 : 0);
347
348 carry += (lx * hy) & 0xffffffffULL;
349 resul = (carry << 32) | (resul & 0xffffffffULL);
350 dest[i+j] += resul;
351 carry = (((!carry && hasCarry != 2) || hasCarry == 1) ? (1ULL << 32) : 0)+
352 (carry >> 32) + (dest[i+j] < resul ? 1 : 0) +
353 ((lx * hy) >> 32) + hx * hy;
354 }
355 dest[i+xlen] = carry;
356 }
357}
358
Zhou Shengdac63782007-02-06 03:00:16 +0000359APInt& APInt::operator*=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000360 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencer58a6a432007-02-21 08:21:52 +0000361 if (isSingleWord()) {
Reid Spencer4bb430c2007-02-20 20:42:10 +0000362 VAL *= RHS.VAL;
Reid Spencer58a6a432007-02-21 08:21:52 +0000363 clearUnusedBits();
364 return *this;
Zhou Shengdac63782007-02-06 03:00:16 +0000365 }
Reid Spencer58a6a432007-02-21 08:21:52 +0000366
367 // Get some bit facts about LHS and check for zero
368 uint32_t lhsBits = getActiveBits();
369 uint32_t lhsWords = !lhsBits ? 0 : whichWord(lhsBits - 1) + 1;
370 if (!lhsWords)
371 // 0 * X ===> 0
372 return *this;
373
374 // Get some bit facts about RHS and check for zero
375 uint32_t rhsBits = RHS.getActiveBits();
376 uint32_t rhsWords = !rhsBits ? 0 : whichWord(rhsBits - 1) + 1;
377 if (!rhsWords) {
378 // X * 0 ===> 0
379 clear();
380 return *this;
381 }
382
383 // Allocate space for the result
384 uint32_t destWords = rhsWords + lhsWords;
385 uint64_t *dest = getMemory(destWords);
386
387 // Perform the long multiply
388 mul(dest, pVal, lhsWords, RHS.pVal, rhsWords);
389
390 // Copy result back into *this
391 clear();
392 uint32_t wordsToCopy = destWords >= getNumWords() ? getNumWords() : destWords;
393 memcpy(pVal, dest, wordsToCopy * APINT_WORD_SIZE);
394
395 // delete dest array and return
396 delete[] dest;
Zhou Shengdac63782007-02-06 03:00:16 +0000397 return *this;
398}
399
Zhou Shengdac63782007-02-06 03:00:16 +0000400APInt& APInt::operator&=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000401 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Zhou Shengdac63782007-02-06 03:00:16 +0000402 if (isSingleWord()) {
Reid Spencera856b6e2007-02-18 18:38:44 +0000403 VAL &= RHS.VAL;
404 return *this;
Zhou Shengdac63782007-02-06 03:00:16 +0000405 }
Reid Spencera856b6e2007-02-18 18:38:44 +0000406 uint32_t numWords = getNumWords();
407 for (uint32_t i = 0; i < numWords; ++i)
408 pVal[i] &= RHS.pVal[i];
Zhou Shengdac63782007-02-06 03:00:16 +0000409 return *this;
410}
411
Zhou Shengdac63782007-02-06 03:00:16 +0000412APInt& APInt::operator|=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000413 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Zhou Shengdac63782007-02-06 03:00:16 +0000414 if (isSingleWord()) {
Reid Spencera856b6e2007-02-18 18:38:44 +0000415 VAL |= RHS.VAL;
416 return *this;
Zhou Shengdac63782007-02-06 03:00:16 +0000417 }
Reid Spencera856b6e2007-02-18 18:38:44 +0000418 uint32_t numWords = getNumWords();
419 for (uint32_t i = 0; i < numWords; ++i)
420 pVal[i] |= RHS.pVal[i];
Zhou Shengdac63782007-02-06 03:00:16 +0000421 return *this;
422}
423
Zhou Shengdac63782007-02-06 03:00:16 +0000424APInt& APInt::operator^=(const APInt& RHS) {
Reid Spencera32372d12007-02-17 00:18:01 +0000425 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Zhou Shengdac63782007-02-06 03:00:16 +0000426 if (isSingleWord()) {
Reid Spenceree0a6852007-02-18 06:39:42 +0000427 VAL ^= RHS.VAL;
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000428 this->clearUnusedBits();
Reid Spenceree0a6852007-02-18 06:39:42 +0000429 return *this;
430 }
Reid Spencera856b6e2007-02-18 18:38:44 +0000431 uint32_t numWords = getNumWords();
432 for (uint32_t i = 0; i < numWords; ++i)
433 pVal[i] ^= RHS.pVal[i];
Reid Spencera41e93b2007-02-25 19:32:03 +0000434 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000435}
436
Zhou Shengdac63782007-02-06 03:00:16 +0000437APInt APInt::operator&(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000438 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera856b6e2007-02-18 18:38:44 +0000439 if (isSingleWord())
440 return APInt(getBitWidth(), VAL & RHS.VAL);
441
Reid Spencera856b6e2007-02-18 18:38:44 +0000442 uint32_t numWords = getNumWords();
Reid Spencera41e93b2007-02-25 19:32:03 +0000443 uint64_t* val = getMemory(numWords);
Reid Spencera856b6e2007-02-18 18:38:44 +0000444 for (uint32_t i = 0; i < numWords; ++i)
Reid Spencera41e93b2007-02-25 19:32:03 +0000445 val[i] = pVal[i] & RHS.pVal[i];
446 return APInt(val, getBitWidth());
Zhou Shengdac63782007-02-06 03:00:16 +0000447}
448
Zhou Shengdac63782007-02-06 03:00:16 +0000449APInt APInt::operator|(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000450 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera856b6e2007-02-18 18:38:44 +0000451 if (isSingleWord())
452 return APInt(getBitWidth(), VAL | RHS.VAL);
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000453
Reid Spencera856b6e2007-02-18 18:38:44 +0000454 uint32_t numWords = getNumWords();
Reid Spencera41e93b2007-02-25 19:32:03 +0000455 uint64_t *val = getMemory(numWords);
Reid Spencera856b6e2007-02-18 18:38:44 +0000456 for (uint32_t i = 0; i < numWords; ++i)
Reid Spencera41e93b2007-02-25 19:32:03 +0000457 val[i] = pVal[i] | RHS.pVal[i];
458 return APInt(val, getBitWidth());
Zhou Shengdac63782007-02-06 03:00:16 +0000459}
460
Zhou Shengdac63782007-02-06 03:00:16 +0000461APInt APInt::operator^(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000462 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera41e93b2007-02-25 19:32:03 +0000463 if (isSingleWord())
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000464 return APInt(BitWidth, VAL ^ RHS.VAL);
Reid Spencera41e93b2007-02-25 19:32:03 +0000465
Reid Spencera856b6e2007-02-18 18:38:44 +0000466 uint32_t numWords = getNumWords();
Reid Spencera41e93b2007-02-25 19:32:03 +0000467 uint64_t *val = getMemory(numWords);
Reid Spencera856b6e2007-02-18 18:38:44 +0000468 for (uint32_t i = 0; i < numWords; ++i)
Reid Spencera41e93b2007-02-25 19:32:03 +0000469 val[i] = pVal[i] ^ RHS.pVal[i];
470
471 // 0^0==1 so clear the high bits in case they got set.
472 return APInt(val, getBitWidth()).clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000473}
474
Zhou Shengdac63782007-02-06 03:00:16 +0000475bool APInt::operator !() const {
476 if (isSingleWord())
477 return !VAL;
Reid Spencera856b6e2007-02-18 18:38:44 +0000478
479 for (uint32_t i = 0; i < getNumWords(); ++i)
480 if (pVal[i])
481 return false;
Zhou Shengdac63782007-02-06 03:00:16 +0000482 return true;
483}
484
Zhou Shengdac63782007-02-06 03:00:16 +0000485APInt APInt::operator*(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000486 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera41e93b2007-02-25 19:32:03 +0000487 if (isSingleWord())
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000488 return APInt(BitWidth, VAL * RHS.VAL);
Reid Spencer4bb430c2007-02-20 20:42:10 +0000489 APInt Result(*this);
490 Result *= RHS;
Reid Spencera41e93b2007-02-25 19:32:03 +0000491 return Result.clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000492}
493
Zhou Shengdac63782007-02-06 03:00:16 +0000494APInt APInt::operator+(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000495 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera41e93b2007-02-25 19:32:03 +0000496 if (isSingleWord())
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000497 return APInt(BitWidth, VAL + RHS.VAL);
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000498 APInt Result(BitWidth, 0);
499 add(Result.pVal, this->pVal, RHS.pVal, getNumWords());
Reid Spencera41e93b2007-02-25 19:32:03 +0000500 return Result.clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000501}
502
Zhou Shengdac63782007-02-06 03:00:16 +0000503APInt APInt::operator-(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +0000504 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencera41e93b2007-02-25 19:32:03 +0000505 if (isSingleWord())
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000506 return APInt(BitWidth, VAL - RHS.VAL);
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000507 APInt Result(BitWidth, 0);
508 sub(Result.pVal, this->pVal, RHS.pVal, getNumWords());
Reid Spencera41e93b2007-02-25 19:32:03 +0000509 return Result.clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000510}
511
Reid Spencera856b6e2007-02-18 18:38:44 +0000512bool APInt::operator[](uint32_t bitPosition) const {
Reid Spencera41e93b2007-02-25 19:32:03 +0000513 return (maskBit(bitPosition) &
514 (isSingleWord() ? VAL : pVal[whichWord(bitPosition)])) != 0;
Zhou Shengdac63782007-02-06 03:00:16 +0000515}
516
Zhou Shengdac63782007-02-06 03:00:16 +0000517bool APInt::operator==(const APInt& RHS) const {
Reid Spencer7c16cd22007-02-26 23:38:21 +0000518 assert(BitWidth == RHS.BitWidth && "Comparison requires equal bit widths");
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000519 if (isSingleWord())
520 return VAL == RHS.VAL;
521
Reid Spencera41e93b2007-02-25 19:32:03 +0000522 // Get some facts about the number of bits used in the two operands.
Reid Spencera856b6e2007-02-18 18:38:44 +0000523 uint32_t n1 = getActiveBits();
524 uint32_t n2 = RHS.getActiveBits();
Reid Spencera41e93b2007-02-25 19:32:03 +0000525
526 // If the number of bits isn't the same, they aren't equal
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000527 if (n1 != n2)
528 return false;
529
Reid Spencera41e93b2007-02-25 19:32:03 +0000530 // If the number of bits fits in a word, we only need to compare the low word.
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000531 if (n1 <= APINT_BITS_PER_WORD)
532 return pVal[0] == RHS.pVal[0];
533
Reid Spencera41e93b2007-02-25 19:32:03 +0000534 // Otherwise, compare everything
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000535 for (int i = whichWord(n1 - 1); i >= 0; --i)
536 if (pVal[i] != RHS.pVal[i])
537 return false;
Zhou Shengdac63782007-02-06 03:00:16 +0000538 return true;
539}
540
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000541bool APInt::operator==(uint64_t Val) const {
542 if (isSingleWord())
543 return VAL == Val;
Reid Spencer7a6a8d52007-02-20 23:40:25 +0000544
545 uint32_t n = getActiveBits();
546 if (n <= APINT_BITS_PER_WORD)
547 return pVal[0] == Val;
548 else
549 return false;
Zhou Shengdac63782007-02-06 03:00:16 +0000550}
551
Reid Spencer1d072122007-02-16 22:36:51 +0000552bool APInt::ult(const APInt& RHS) const {
553 assert(BitWidth == RHS.BitWidth && "Bit widths must be same for comparison");
554 if (isSingleWord())
555 return VAL < RHS.VAL;
Reid Spencera41e93b2007-02-25 19:32:03 +0000556
557 // Get active bit length of both operands
558 uint32_t n1 = getActiveBits();
559 uint32_t n2 = RHS.getActiveBits();
560
561 // If magnitude of LHS is less than RHS, return true.
562 if (n1 < n2)
563 return true;
564
565 // If magnitude of RHS is greather than LHS, return false.
566 if (n2 < n1)
567 return false;
568
569 // If they bot fit in a word, just compare the low order word
570 if (n1 <= APINT_BITS_PER_WORD && n2 <= APINT_BITS_PER_WORD)
571 return pVal[0] < RHS.pVal[0];
572
573 // Otherwise, compare all words
Reid Spencer54abdcf2007-02-27 18:23:40 +0000574 uint32_t topWord = whichWord(std::max(n1,n2)-1);
575 for (int i = topWord; i >= 0; --i) {
Reid Spencera41e93b2007-02-25 19:32:03 +0000576 if (pVal[i] > RHS.pVal[i])
Reid Spencer1d072122007-02-16 22:36:51 +0000577 return false;
Reid Spencera41e93b2007-02-25 19:32:03 +0000578 if (pVal[i] < RHS.pVal[i])
579 return true;
Zhou Shengdac63782007-02-06 03:00:16 +0000580 }
581 return false;
582}
583
Reid Spencer1d072122007-02-16 22:36:51 +0000584bool APInt::slt(const APInt& RHS) const {
585 assert(BitWidth == RHS.BitWidth && "Bit widths must be same for comparison");
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000586 if (isSingleWord()) {
587 int64_t lhsSext = (int64_t(VAL) << (64-BitWidth)) >> (64-BitWidth);
588 int64_t rhsSext = (int64_t(RHS.VAL) << (64-BitWidth)) >> (64-BitWidth);
589 return lhsSext < rhsSext;
Reid Spencer1d072122007-02-16 22:36:51 +0000590 }
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000591
592 APInt lhs(*this);
Reid Spencer54abdcf2007-02-27 18:23:40 +0000593 APInt rhs(RHS);
594 bool lhsNeg = isNegative();
595 bool rhsNeg = rhs.isNegative();
596 if (lhsNeg) {
597 // Sign bit is set so perform two's complement to make it positive
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000598 lhs.flip();
599 lhs++;
600 }
Reid Spencer54abdcf2007-02-27 18:23:40 +0000601 if (rhsNeg) {
602 // Sign bit is set so perform two's complement to make it positive
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000603 rhs.flip();
604 rhs++;
605 }
Reid Spencera41e93b2007-02-25 19:32:03 +0000606
607 // Now we have unsigned values to compare so do the comparison if necessary
608 // based on the negativeness of the values.
Reid Spencer54abdcf2007-02-27 18:23:40 +0000609 if (lhsNeg)
610 if (rhsNeg)
611 return lhs.ugt(rhs);
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000612 else
613 return true;
Reid Spencer54abdcf2007-02-27 18:23:40 +0000614 else if (rhsNeg)
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000615 return false;
616 else
617 return lhs.ult(rhs);
Zhou Shengdac63782007-02-06 03:00:16 +0000618}
619
Reid Spencera856b6e2007-02-18 18:38:44 +0000620APInt& APInt::set(uint32_t bitPosition) {
Reid Spencera41e93b2007-02-25 19:32:03 +0000621 if (isSingleWord())
622 VAL |= maskBit(bitPosition);
623 else
624 pVal[whichWord(bitPosition)] |= maskBit(bitPosition);
Zhou Shengdac63782007-02-06 03:00:16 +0000625 return *this;
626}
627
Zhou Shengdac63782007-02-06 03:00:16 +0000628APInt& APInt::set() {
Reid Spencera41e93b2007-02-25 19:32:03 +0000629 if (isSingleWord()) {
630 VAL = -1ULL;
631 return clearUnusedBits();
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000632 }
Reid Spencera41e93b2007-02-25 19:32:03 +0000633
634 // Set all the bits in all the words.
Zhou Sheng0458c1d2007-03-21 04:34:37 +0000635 for (uint32_t i = 0; i < getNumWords(); ++i)
Reid Spencera41e93b2007-02-25 19:32:03 +0000636 pVal[i] = -1ULL;
637 // Clear the unused ones
638 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000639}
640
641/// Set the given bit to 0 whose position is given as "bitPosition".
642/// @brief Set a given bit to 0.
Reid Spencera856b6e2007-02-18 18:38:44 +0000643APInt& APInt::clear(uint32_t bitPosition) {
644 if (isSingleWord())
645 VAL &= ~maskBit(bitPosition);
646 else
647 pVal[whichWord(bitPosition)] &= ~maskBit(bitPosition);
Zhou Shengdac63782007-02-06 03:00:16 +0000648 return *this;
649}
650
651/// @brief Set every bit to 0.
652APInt& APInt::clear() {
Reid Spencera856b6e2007-02-18 18:38:44 +0000653 if (isSingleWord())
654 VAL = 0;
Zhou Sheng3e8022d2007-02-07 06:14:53 +0000655 else
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000656 memset(pVal, 0, getNumWords() * APINT_WORD_SIZE);
Zhou Shengdac63782007-02-06 03:00:16 +0000657 return *this;
658}
659
Zhou Shengdac63782007-02-06 03:00:16 +0000660/// @brief Bitwise NOT operator. Performs a bitwise logical NOT operation on
661/// this APInt.
662APInt APInt::operator~() const {
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000663 APInt Result(*this);
664 Result.flip();
665 return Result;
Zhou Shengdac63782007-02-06 03:00:16 +0000666}
667
668/// @brief Toggle every bit to its opposite value.
669APInt& APInt::flip() {
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000670 if (isSingleWord()) {
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000671 VAL ^= -1ULL;
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000672 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000673 }
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000674 for (uint32_t i = 0; i < getNumWords(); ++i)
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000675 pVal[i] ^= -1ULL;
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000676 return clearUnusedBits();
Zhou Shengdac63782007-02-06 03:00:16 +0000677}
678
679/// Toggle a given bit to its opposite value whose position is given
680/// as "bitPosition".
681/// @brief Toggles a given bit to its opposite value.
Reid Spencera856b6e2007-02-18 18:38:44 +0000682APInt& APInt::flip(uint32_t bitPosition) {
Reid Spencer1d072122007-02-16 22:36:51 +0000683 assert(bitPosition < BitWidth && "Out of the bit-width range!");
Zhou Shengdac63782007-02-06 03:00:16 +0000684 if ((*this)[bitPosition]) clear(bitPosition);
685 else set(bitPosition);
686 return *this;
687}
688
Reid Spencer9329e7b2007-04-13 19:19:07 +0000689uint32_t APInt::getBitsNeeded(const char* str, uint32_t slen, uint8_t radix) {
690 assert(str != 0 && "Invalid value string");
691 assert(slen > 0 && "Invalid string length");
692
693 // Each computation below needs to know if its negative
694 uint32_t isNegative = str[0] == '-';
695 if (isNegative) {
696 slen--;
697 str++;
698 }
699 // For radixes of power-of-two values, the bits required is accurately and
700 // easily computed
701 if (radix == 2)
702 return slen + isNegative;
703 if (radix == 8)
704 return slen * 3 + isNegative;
705 if (radix == 16)
706 return slen * 4 + isNegative;
707
708 // Otherwise it must be radix == 10, the hard case
709 assert(radix == 10 && "Invalid radix");
710
711 // This is grossly inefficient but accurate. We could probably do something
712 // with a computation of roughly slen*64/20 and then adjust by the value of
713 // the first few digits. But, I'm not sure how accurate that could be.
714
715 // Compute a sufficient number of bits that is always large enough but might
716 // be too large. This avoids the assertion in the constructor.
717 uint32_t sufficient = slen*64/18;
718
719 // Convert to the actual binary value.
720 APInt tmp(sufficient, str, slen, radix);
721
722 // Compute how many bits are required.
Reid Spencer67378b22007-04-14 00:00:10 +0000723 return isNegative + tmp.logBase2() + 1;
Reid Spencer9329e7b2007-04-13 19:19:07 +0000724}
725
Reid Spencerb2bc9852007-02-26 21:02:27 +0000726uint64_t APInt::getHashValue() const {
Reid Spencer7c16cd22007-02-26 23:38:21 +0000727 // Put the bit width into the low order bits.
728 uint64_t hash = BitWidth;
Reid Spencerb2bc9852007-02-26 21:02:27 +0000729
730 // Add the sum of the words to the hash.
731 if (isSingleWord())
Reid Spencer7c16cd22007-02-26 23:38:21 +0000732 hash += VAL << 6; // clear separation of up to 64 bits
Reid Spencerb2bc9852007-02-26 21:02:27 +0000733 else
734 for (uint32_t i = 0; i < getNumWords(); ++i)
Reid Spencer7c16cd22007-02-26 23:38:21 +0000735 hash += pVal[i] << 6; // clear sepration of up to 64 bits
Reid Spencerb2bc9852007-02-26 21:02:27 +0000736 return hash;
737}
738
Zhou Shengdac63782007-02-06 03:00:16 +0000739/// HiBits - This function returns the high "numBits" bits of this APInt.
Reid Spencera856b6e2007-02-18 18:38:44 +0000740APInt APInt::getHiBits(uint32_t numBits) const {
Reid Spencer1d072122007-02-16 22:36:51 +0000741 return APIntOps::lshr(*this, BitWidth - numBits);
Zhou Shengdac63782007-02-06 03:00:16 +0000742}
743
744/// LoBits - This function returns the low "numBits" bits of this APInt.
Reid Spencera856b6e2007-02-18 18:38:44 +0000745APInt APInt::getLoBits(uint32_t numBits) const {
Reid Spencer1d072122007-02-16 22:36:51 +0000746 return APIntOps::lshr(APIntOps::shl(*this, BitWidth - numBits),
747 BitWidth - numBits);
Zhou Shengdac63782007-02-06 03:00:16 +0000748}
749
Reid Spencer1d072122007-02-16 22:36:51 +0000750bool APInt::isPowerOf2() const {
751 return (!!*this) && !(*this & (*this - APInt(BitWidth,1)));
752}
753
Reid Spencera856b6e2007-02-18 18:38:44 +0000754uint32_t APInt::countLeadingZeros() const {
Reid Spencera856b6e2007-02-18 18:38:44 +0000755 uint32_t Count = 0;
Reid Spencer74cf82e2007-02-21 00:29:48 +0000756 if (isSingleWord())
757 Count = CountLeadingZeros_64(VAL);
758 else {
759 for (uint32_t i = getNumWords(); i > 0u; --i) {
760 if (pVal[i-1] == 0)
761 Count += APINT_BITS_PER_WORD;
762 else {
763 Count += CountLeadingZeros_64(pVal[i-1]);
764 break;
765 }
766 }
Zhou Shengdac63782007-02-06 03:00:16 +0000767 }
Reid Spencere4ce71d2007-02-22 00:22:00 +0000768 uint32_t remainder = BitWidth % APINT_BITS_PER_WORD;
769 if (remainder)
770 Count -= APINT_BITS_PER_WORD - remainder;
Chris Lattner893fe3b2007-11-23 22:42:31 +0000771 return std::min(Count, BitWidth);
Zhou Shengdac63782007-02-06 03:00:16 +0000772}
773
Reid Spencer31acef52007-02-27 21:59:26 +0000774static uint32_t countLeadingOnes_64(uint64_t V, uint32_t skip) {
775 uint32_t Count = 0;
776 if (skip)
777 V <<= skip;
778 while (V && (V & (1ULL << 63))) {
779 Count++;
780 V <<= 1;
781 }
782 return Count;
783}
784
785uint32_t APInt::countLeadingOnes() const {
786 if (isSingleWord())
787 return countLeadingOnes_64(VAL, APINT_BITS_PER_WORD - BitWidth);
788
789 uint32_t highWordBits = BitWidth % APINT_BITS_PER_WORD;
790 uint32_t shift = (highWordBits == 0 ? 0 : APINT_BITS_PER_WORD - highWordBits);
791 int i = getNumWords() - 1;
792 uint32_t Count = countLeadingOnes_64(pVal[i], shift);
793 if (Count == highWordBits) {
794 for (i--; i >= 0; --i) {
795 if (pVal[i] == -1ULL)
796 Count += APINT_BITS_PER_WORD;
797 else {
798 Count += countLeadingOnes_64(pVal[i], 0);
799 break;
800 }
801 }
802 }
803 return Count;
804}
805
Reid Spencera856b6e2007-02-18 18:38:44 +0000806uint32_t APInt::countTrailingZeros() const {
Zhou Shengdac63782007-02-06 03:00:16 +0000807 if (isSingleWord())
Anton Korobeynikovee6af402007-12-24 11:16:47 +0000808 return std::min(uint32_t(CountTrailingZeros_64(VAL)), BitWidth);
Reid Spenceraa8dcfe2007-02-26 07:44:38 +0000809 uint32_t Count = 0;
810 uint32_t i = 0;
811 for (; i < getNumWords() && pVal[i] == 0; ++i)
812 Count += APINT_BITS_PER_WORD;
813 if (i < getNumWords())
814 Count += CountTrailingZeros_64(pVal[i]);
Chris Lattnerc2c4c742007-11-23 22:36:25 +0000815 return std::min(Count, BitWidth);
Zhou Shengdac63782007-02-06 03:00:16 +0000816}
817
Dan Gohman8b4fa9d2008-02-13 21:11:05 +0000818uint32_t APInt::countTrailingOnes() const {
819 if (isSingleWord())
820 return std::min(uint32_t(CountTrailingOnes_64(VAL)), BitWidth);
821 uint32_t Count = 0;
822 uint32_t i = 0;
Dan Gohmanc354ebd2008-02-14 22:38:45 +0000823 for (; i < getNumWords() && pVal[i] == -1ULL; ++i)
Dan Gohman8b4fa9d2008-02-13 21:11:05 +0000824 Count += APINT_BITS_PER_WORD;
825 if (i < getNumWords())
826 Count += CountTrailingOnes_64(pVal[i]);
827 return std::min(Count, BitWidth);
828}
829
Reid Spencera856b6e2007-02-18 18:38:44 +0000830uint32_t APInt::countPopulation() const {
Zhou Shengdac63782007-02-06 03:00:16 +0000831 if (isSingleWord())
832 return CountPopulation_64(VAL);
Reid Spencera856b6e2007-02-18 18:38:44 +0000833 uint32_t Count = 0;
834 for (uint32_t i = 0; i < getNumWords(); ++i)
Zhou Shengdac63782007-02-06 03:00:16 +0000835 Count += CountPopulation_64(pVal[i]);
836 return Count;
837}
838
Reid Spencer1d072122007-02-16 22:36:51 +0000839APInt APInt::byteSwap() const {
840 assert(BitWidth >= 16 && BitWidth % 16 == 0 && "Cannot byteswap!");
841 if (BitWidth == 16)
Jeff Cohene06855e2007-03-20 20:42:36 +0000842 return APInt(BitWidth, ByteSwap_16(uint16_t(VAL)));
Reid Spencer1d072122007-02-16 22:36:51 +0000843 else if (BitWidth == 32)
Jeff Cohene06855e2007-03-20 20:42:36 +0000844 return APInt(BitWidth, ByteSwap_32(uint32_t(VAL)));
Reid Spencer1d072122007-02-16 22:36:51 +0000845 else if (BitWidth == 48) {
Jeff Cohene06855e2007-03-20 20:42:36 +0000846 uint32_t Tmp1 = uint32_t(VAL >> 16);
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000847 Tmp1 = ByteSwap_32(Tmp1);
Jeff Cohene06855e2007-03-20 20:42:36 +0000848 uint16_t Tmp2 = uint16_t(VAL);
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000849 Tmp2 = ByteSwap_16(Tmp2);
Jeff Cohene06855e2007-03-20 20:42:36 +0000850 return APInt(BitWidth, (uint64_t(Tmp2) << 32) | Tmp1);
Reid Spencer1d072122007-02-16 22:36:51 +0000851 } else if (BitWidth == 64)
Reid Spencera32372d12007-02-17 00:18:01 +0000852 return APInt(BitWidth, ByteSwap_64(VAL));
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000853 else {
Reid Spencera32372d12007-02-17 00:18:01 +0000854 APInt Result(BitWidth, 0);
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000855 char *pByte = (char*)Result.pVal;
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000856 for (uint32_t i = 0; i < BitWidth / APINT_WORD_SIZE / 2; ++i) {
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000857 char Tmp = pByte[i];
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000858 pByte[i] = pByte[BitWidth / APINT_WORD_SIZE - 1 - i];
859 pByte[BitWidth / APINT_WORD_SIZE - i - 1] = Tmp;
Zhou Shengcfa2ac02007-02-15 06:36:31 +0000860 }
861 return Result;
862 }
Zhou Shengdac63782007-02-06 03:00:16 +0000863}
864
Zhou Shengfbf61ea2007-02-08 14:35:19 +0000865APInt llvm::APIntOps::GreatestCommonDivisor(const APInt& API1,
866 const APInt& API2) {
Zhou Shengdac63782007-02-06 03:00:16 +0000867 APInt A = API1, B = API2;
868 while (!!B) {
869 APInt T = B;
Reid Spencer1d072122007-02-16 22:36:51 +0000870 B = APIntOps::urem(A, B);
Zhou Shengdac63782007-02-06 03:00:16 +0000871 A = T;
872 }
873 return A;
874}
Chris Lattner28cbd1d2007-02-06 05:38:37 +0000875
Reid Spencer54abdcf2007-02-27 18:23:40 +0000876APInt llvm::APIntOps::RoundDoubleToAPInt(double Double, uint32_t width) {
Zhou Shengd707d632007-02-12 20:02:55 +0000877 union {
878 double D;
879 uint64_t I;
880 } T;
881 T.D = Double;
Reid Spencer974551a2007-02-27 01:28:10 +0000882
883 // Get the sign bit from the highest order bit
Zhou Shengd707d632007-02-12 20:02:55 +0000884 bool isNeg = T.I >> 63;
Reid Spencer974551a2007-02-27 01:28:10 +0000885
886 // Get the 11-bit exponent and adjust for the 1023 bit bias
Zhou Shengd707d632007-02-12 20:02:55 +0000887 int64_t exp = ((T.I >> 52) & 0x7ff) - 1023;
Reid Spencer974551a2007-02-27 01:28:10 +0000888
889 // If the exponent is negative, the value is < 0 so just return 0.
Zhou Shengd707d632007-02-12 20:02:55 +0000890 if (exp < 0)
Reid Spencer66d0d572007-02-28 01:30:08 +0000891 return APInt(width, 0u);
Reid Spencer974551a2007-02-27 01:28:10 +0000892
893 // Extract the mantissa by clearing the top 12 bits (sign + exponent).
894 uint64_t mantissa = (T.I & (~0ULL >> 12)) | 1ULL << 52;
895
896 // If the exponent doesn't shift all bits out of the mantissa
Zhou Shengd707d632007-02-12 20:02:55 +0000897 if (exp < 52)
Reid Spencer54abdcf2007-02-27 18:23:40 +0000898 return isNeg ? -APInt(width, mantissa >> (52 - exp)) :
899 APInt(width, mantissa >> (52 - exp));
900
901 // If the client didn't provide enough bits for us to shift the mantissa into
902 // then the result is undefined, just return 0
903 if (width <= exp - 52)
904 return APInt(width, 0);
Reid Spencer974551a2007-02-27 01:28:10 +0000905
906 // Otherwise, we have to shift the mantissa bits up to the right location
Reid Spencer54abdcf2007-02-27 18:23:40 +0000907 APInt Tmp(width, mantissa);
Reid Spencer1d072122007-02-16 22:36:51 +0000908 Tmp = Tmp.shl(exp - 52);
Zhou Shengd707d632007-02-12 20:02:55 +0000909 return isNeg ? -Tmp : Tmp;
910}
911
Reid Spencer51535252007-02-13 22:41:58 +0000912/// RoundToDouble - This function convert this APInt to a double.
Zhou Shengd707d632007-02-12 20:02:55 +0000913/// The layout for double is as following (IEEE Standard 754):
914/// --------------------------------------
915/// | Sign Exponent Fraction Bias |
916/// |-------------------------------------- |
917/// | 1[63] 11[62-52] 52[51-00] 1023 |
918/// --------------------------------------
Reid Spencer1d072122007-02-16 22:36:51 +0000919double APInt::roundToDouble(bool isSigned) const {
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000920
921 // Handle the simple case where the value is contained in one uint64_t.
Reid Spencerbe4ddf62007-02-18 20:09:41 +0000922 if (isSingleWord() || getActiveBits() <= APINT_BITS_PER_WORD) {
923 if (isSigned) {
924 int64_t sext = (int64_t(VAL) << (64-BitWidth)) >> (64-BitWidth);
925 return double(sext);
926 } else
927 return double(VAL);
928 }
929
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000930 // Determine if the value is negative.
Reid Spencer1d072122007-02-16 22:36:51 +0000931 bool isNeg = isSigned ? (*this)[BitWidth-1] : false;
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000932
933 // Construct the absolute value if we're negative.
Zhou Shengd707d632007-02-12 20:02:55 +0000934 APInt Tmp(isNeg ? -(*this) : (*this));
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000935
936 // Figure out how many bits we're using.
Reid Spencera856b6e2007-02-18 18:38:44 +0000937 uint32_t n = Tmp.getActiveBits();
Zhou Shengd707d632007-02-12 20:02:55 +0000938
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000939 // The exponent (without bias normalization) is just the number of bits
940 // we are using. Note that the sign bit is gone since we constructed the
941 // absolute value.
942 uint64_t exp = n;
Zhou Shengd707d632007-02-12 20:02:55 +0000943
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000944 // Return infinity for exponent overflow
945 if (exp > 1023) {
946 if (!isSigned || !isNeg)
Jeff Cohene06855e2007-03-20 20:42:36 +0000947 return std::numeric_limits<double>::infinity();
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000948 else
Jeff Cohene06855e2007-03-20 20:42:36 +0000949 return -std::numeric_limits<double>::infinity();
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000950 }
951 exp += 1023; // Increment for 1023 bias
952
953 // Number of bits in mantissa is 52. To obtain the mantissa value, we must
954 // extract the high 52 bits from the correct words in pVal.
Zhou Shengd707d632007-02-12 20:02:55 +0000955 uint64_t mantissa;
Reid Spencerfb77b2b2007-02-20 08:51:03 +0000956 unsigned hiWord = whichWord(n-1);
957 if (hiWord == 0) {
958 mantissa = Tmp.pVal[0];
959 if (n > 52)
960 mantissa >>= n - 52; // shift down, we want the top 52 bits.
961 } else {
962 assert(hiWord > 0 && "huh?");
963 uint64_t hibits = Tmp.pVal[hiWord] << (52 - n % APINT_BITS_PER_WORD);
964 uint64_t lobits = Tmp.pVal[hiWord-1] >> (11 + n % APINT_BITS_PER_WORD);
965 mantissa = hibits | lobits;
966 }
967
Zhou Shengd707d632007-02-12 20:02:55 +0000968 // The leading bit of mantissa is implicit, so get rid of it.
Reid Spencerfbd48a52007-02-18 00:44:22 +0000969 uint64_t sign = isNeg ? (1ULL << (APINT_BITS_PER_WORD - 1)) : 0;
Zhou Shengd707d632007-02-12 20:02:55 +0000970 union {
971 double D;
972 uint64_t I;
973 } T;
974 T.I = sign | (exp << 52) | mantissa;
975 return T.D;
976}
977
Reid Spencer1d072122007-02-16 22:36:51 +0000978// Truncate to new width.
Reid Spencer91d3b3f2007-02-28 17:34:32 +0000979APInt &APInt::trunc(uint32_t width) {
Reid Spencer1d072122007-02-16 22:36:51 +0000980 assert(width < BitWidth && "Invalid APInt Truncate request");
Reid Spencer0b1df1d2007-12-11 06:53:58 +0000981 assert(width >= MIN_INT_BITS && "Can't truncate to 0 bits");
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000982 uint32_t wordsBefore = getNumWords();
983 BitWidth = width;
984 uint32_t wordsAfter = getNumWords();
985 if (wordsBefore != wordsAfter) {
986 if (wordsAfter == 1) {
987 uint64_t *tmp = pVal;
988 VAL = pVal[0];
Reid Spencer7c16cd22007-02-26 23:38:21 +0000989 delete [] tmp;
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000990 } else {
991 uint64_t *newVal = getClearedMemory(wordsAfter);
992 for (uint32_t i = 0; i < wordsAfter; ++i)
993 newVal[i] = pVal[i];
Reid Spencer7c16cd22007-02-26 23:38:21 +0000994 delete [] pVal;
Reid Spencerb6b5cc32007-02-25 23:44:53 +0000995 pVal = newVal;
996 }
997 }
Reid Spencer91d3b3f2007-02-28 17:34:32 +0000998 return clearUnusedBits();
Reid Spencer1d072122007-02-16 22:36:51 +0000999}
1000
1001// Sign extend to a new width.
Reid Spencer91d3b3f2007-02-28 17:34:32 +00001002APInt &APInt::sext(uint32_t width) {
Reid Spencer1d072122007-02-16 22:36:51 +00001003 assert(width > BitWidth && "Invalid APInt SignExtend request");
Reid Spencer0b1df1d2007-12-11 06:53:58 +00001004 assert(width <= MAX_INT_BITS && "Too many bits");
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001005 // If the sign bit isn't set, this is the same as zext.
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001006 if (!isNegative()) {
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001007 zext(width);
Reid Spencer91d3b3f2007-02-28 17:34:32 +00001008 return *this;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001009 }
1010
1011 // The sign bit is set. First, get some facts
1012 uint32_t wordsBefore = getNumWords();
1013 uint32_t wordBits = BitWidth % APINT_BITS_PER_WORD;
1014 BitWidth = width;
1015 uint32_t wordsAfter = getNumWords();
1016
1017 // Mask the high order word appropriately
1018 if (wordsBefore == wordsAfter) {
1019 uint32_t newWordBits = width % APINT_BITS_PER_WORD;
1020 // The extension is contained to the wordsBefore-1th word.
Reid Spencerc442c842007-03-02 01:19:42 +00001021 uint64_t mask = ~0ULL;
1022 if (newWordBits)
1023 mask >>= APINT_BITS_PER_WORD - newWordBits;
1024 mask <<= wordBits;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001025 if (wordsBefore == 1)
1026 VAL |= mask;
1027 else
1028 pVal[wordsBefore-1] |= mask;
Reid Spencer1b8dfcba2007-03-01 23:30:25 +00001029 return clearUnusedBits();
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001030 }
1031
Reid Spencerfb55b7b2007-02-25 23:54:00 +00001032 uint64_t mask = wordBits == 0 ? 0 : ~0ULL << wordBits;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001033 uint64_t *newVal = getMemory(wordsAfter);
1034 if (wordsBefore == 1)
1035 newVal[0] = VAL | mask;
1036 else {
1037 for (uint32_t i = 0; i < wordsBefore; ++i)
1038 newVal[i] = pVal[i];
1039 newVal[wordsBefore-1] |= mask;
1040 }
1041 for (uint32_t i = wordsBefore; i < wordsAfter; i++)
1042 newVal[i] = -1ULL;
1043 if (wordsBefore != 1)
Reid Spencer7c16cd22007-02-26 23:38:21 +00001044 delete [] pVal;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001045 pVal = newVal;
Reid Spencer91d3b3f2007-02-28 17:34:32 +00001046 return clearUnusedBits();
Reid Spencer1d072122007-02-16 22:36:51 +00001047}
1048
1049// Zero extend to a new width.
Reid Spencer91d3b3f2007-02-28 17:34:32 +00001050APInt &APInt::zext(uint32_t width) {
Reid Spencer1d072122007-02-16 22:36:51 +00001051 assert(width > BitWidth && "Invalid APInt ZeroExtend request");
Reid Spencer0b1df1d2007-12-11 06:53:58 +00001052 assert(width <= MAX_INT_BITS && "Too many bits");
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001053 uint32_t wordsBefore = getNumWords();
1054 BitWidth = width;
1055 uint32_t wordsAfter = getNumWords();
1056 if (wordsBefore != wordsAfter) {
1057 uint64_t *newVal = getClearedMemory(wordsAfter);
1058 if (wordsBefore == 1)
1059 newVal[0] = VAL;
1060 else
1061 for (uint32_t i = 0; i < wordsBefore; ++i)
1062 newVal[i] = pVal[i];
1063 if (wordsBefore != 1)
Reid Spencer7c16cd22007-02-26 23:38:21 +00001064 delete [] pVal;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001065 pVal = newVal;
1066 }
Reid Spencer91d3b3f2007-02-28 17:34:32 +00001067 return *this;
Reid Spencer1d072122007-02-16 22:36:51 +00001068}
1069
Reid Spencer742d1702007-03-01 17:15:32 +00001070APInt &APInt::zextOrTrunc(uint32_t width) {
1071 if (BitWidth < width)
1072 return zext(width);
1073 if (BitWidth > width)
1074 return trunc(width);
1075 return *this;
1076}
1077
1078APInt &APInt::sextOrTrunc(uint32_t width) {
1079 if (BitWidth < width)
1080 return sext(width);
1081 if (BitWidth > width)
1082 return trunc(width);
1083 return *this;
1084}
1085
Zhou Shenge93db8f2007-02-09 07:48:24 +00001086/// Arithmetic right-shift this APInt by shiftAmt.
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001087/// @brief Arithmetic right-shift function.
Dan Gohman105c1d42008-02-29 01:40:47 +00001088APInt APInt::ashr(const APInt &shiftAmt) const {
1089 return ashr(shiftAmt.getLimitedValue(BitWidth));
1090}
1091
1092/// Arithmetic right-shift this APInt by shiftAmt.
1093/// @brief Arithmetic right-shift function.
Reid Spencera856b6e2007-02-18 18:38:44 +00001094APInt APInt::ashr(uint32_t shiftAmt) const {
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001095 assert(shiftAmt <= BitWidth && "Invalid shift amount");
Reid Spencer1825dd02007-03-02 22:39:11 +00001096 // Handle a degenerate case
1097 if (shiftAmt == 0)
1098 return *this;
1099
1100 // Handle single word shifts with built-in ashr
Reid Spencer522ca7c2007-02-25 01:56:07 +00001101 if (isSingleWord()) {
1102 if (shiftAmt == BitWidth)
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001103 return APInt(BitWidth, 0); // undefined
1104 else {
1105 uint32_t SignBit = APINT_BITS_PER_WORD - BitWidth;
Reid Spencer522ca7c2007-02-25 01:56:07 +00001106 return APInt(BitWidth,
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001107 (((int64_t(VAL) << SignBit) >> SignBit) >> shiftAmt));
1108 }
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001109 }
Reid Spencer522ca7c2007-02-25 01:56:07 +00001110
Reid Spencer1825dd02007-03-02 22:39:11 +00001111 // If all the bits were shifted out, the result is, technically, undefined.
1112 // We return -1 if it was negative, 0 otherwise. We check this early to avoid
1113 // issues in the algorithm below.
Chris Lattnerdad2d092007-05-03 18:15:36 +00001114 if (shiftAmt == BitWidth) {
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001115 if (isNegative())
1116 return APInt(BitWidth, -1ULL);
Reid Spencera41e93b2007-02-25 19:32:03 +00001117 else
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001118 return APInt(BitWidth, 0);
Chris Lattnerdad2d092007-05-03 18:15:36 +00001119 }
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001120
1121 // Create some space for the result.
1122 uint64_t * val = new uint64_t[getNumWords()];
1123
Reid Spencer1825dd02007-03-02 22:39:11 +00001124 // Compute some values needed by the following shift algorithms
1125 uint32_t wordShift = shiftAmt % APINT_BITS_PER_WORD; // bits to shift per word
1126 uint32_t offset = shiftAmt / APINT_BITS_PER_WORD; // word offset for shift
1127 uint32_t breakWord = getNumWords() - 1 - offset; // last word affected
1128 uint32_t bitsInWord = whichBit(BitWidth); // how many bits in last word?
1129 if (bitsInWord == 0)
1130 bitsInWord = APINT_BITS_PER_WORD;
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001131
1132 // If we are shifting whole words, just move whole words
1133 if (wordShift == 0) {
Reid Spencer1825dd02007-03-02 22:39:11 +00001134 // Move the words containing significant bits
1135 for (uint32_t i = 0; i <= breakWord; ++i)
1136 val[i] = pVal[i+offset]; // move whole word
1137
1138 // Adjust the top significant word for sign bit fill, if negative
1139 if (isNegative())
1140 if (bitsInWord < APINT_BITS_PER_WORD)
1141 val[breakWord] |= ~0ULL << bitsInWord; // set high bits
1142 } else {
1143 // Shift the low order words
1144 for (uint32_t i = 0; i < breakWord; ++i) {
1145 // This combines the shifted corresponding word with the low bits from
1146 // the next word (shifted into this word's high bits).
1147 val[i] = (pVal[i+offset] >> wordShift) |
1148 (pVal[i+offset+1] << (APINT_BITS_PER_WORD - wordShift));
1149 }
1150
1151 // Shift the break word. In this case there are no bits from the next word
1152 // to include in this word.
1153 val[breakWord] = pVal[breakWord+offset] >> wordShift;
1154
1155 // Deal with sign extenstion in the break word, and possibly the word before
1156 // it.
Chris Lattnerdad2d092007-05-03 18:15:36 +00001157 if (isNegative()) {
Reid Spencer1825dd02007-03-02 22:39:11 +00001158 if (wordShift > bitsInWord) {
1159 if (breakWord > 0)
1160 val[breakWord-1] |=
1161 ~0ULL << (APINT_BITS_PER_WORD - (wordShift - bitsInWord));
1162 val[breakWord] |= ~0ULL;
1163 } else
1164 val[breakWord] |= (~0ULL << (bitsInWord - wordShift));
Chris Lattnerdad2d092007-05-03 18:15:36 +00001165 }
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001166 }
1167
Reid Spencer1825dd02007-03-02 22:39:11 +00001168 // Remaining words are 0 or -1, just assign them.
1169 uint64_t fillValue = (isNegative() ? -1ULL : 0);
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001170 for (uint32_t i = breakWord+1; i < getNumWords(); ++i)
Reid Spencer1825dd02007-03-02 22:39:11 +00001171 val[i] = fillValue;
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001172 return APInt(val, BitWidth).clearUnusedBits();
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001173}
1174
Zhou Shenge93db8f2007-02-09 07:48:24 +00001175/// Logical right-shift this APInt by shiftAmt.
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001176/// @brief Logical right-shift function.
Dan Gohman105c1d42008-02-29 01:40:47 +00001177APInt APInt::lshr(const APInt &shiftAmt) const {
1178 return ashr(shiftAmt.getLimitedValue(BitWidth));
1179}
1180
1181/// Logical right-shift this APInt by shiftAmt.
1182/// @brief Logical right-shift function.
Reid Spencera856b6e2007-02-18 18:38:44 +00001183APInt APInt::lshr(uint32_t shiftAmt) const {
Chris Lattnerdad2d092007-05-03 18:15:36 +00001184 if (isSingleWord()) {
Reid Spencer522ca7c2007-02-25 01:56:07 +00001185 if (shiftAmt == BitWidth)
1186 return APInt(BitWidth, 0);
1187 else
1188 return APInt(BitWidth, this->VAL >> shiftAmt);
Chris Lattnerdad2d092007-05-03 18:15:36 +00001189 }
Reid Spencer522ca7c2007-02-25 01:56:07 +00001190
Reid Spencer44eef162007-02-26 01:19:48 +00001191 // If all the bits were shifted out, the result is 0. This avoids issues
1192 // with shifting by the size of the integer type, which produces undefined
1193 // results. We define these "undefined results" to always be 0.
1194 if (shiftAmt == BitWidth)
1195 return APInt(BitWidth, 0);
1196
Reid Spencerfffdf102007-05-17 06:26:29 +00001197 // If none of the bits are shifted out, the result is *this. This avoids
1198 // issues with shifting byt he size of the integer type, which produces
1199 // undefined results in the code below. This is also an optimization.
1200 if (shiftAmt == 0)
1201 return *this;
1202
Reid Spencer44eef162007-02-26 01:19:48 +00001203 // Create some space for the result.
1204 uint64_t * val = new uint64_t[getNumWords()];
1205
1206 // If we are shifting less than a word, compute the shift with a simple carry
1207 if (shiftAmt < APINT_BITS_PER_WORD) {
1208 uint64_t carry = 0;
1209 for (int i = getNumWords()-1; i >= 0; --i) {
Reid Spencerd99feaf2007-03-01 05:39:56 +00001210 val[i] = (pVal[i] >> shiftAmt) | carry;
Reid Spencer44eef162007-02-26 01:19:48 +00001211 carry = pVal[i] << (APINT_BITS_PER_WORD - shiftAmt);
1212 }
1213 return APInt(val, BitWidth).clearUnusedBits();
Reid Spencera41e93b2007-02-25 19:32:03 +00001214 }
1215
Reid Spencer44eef162007-02-26 01:19:48 +00001216 // Compute some values needed by the remaining shift algorithms
1217 uint32_t wordShift = shiftAmt % APINT_BITS_PER_WORD;
1218 uint32_t offset = shiftAmt / APINT_BITS_PER_WORD;
1219
1220 // If we are shifting whole words, just move whole words
1221 if (wordShift == 0) {
1222 for (uint32_t i = 0; i < getNumWords() - offset; ++i)
1223 val[i] = pVal[i+offset];
1224 for (uint32_t i = getNumWords()-offset; i < getNumWords(); i++)
1225 val[i] = 0;
1226 return APInt(val,BitWidth).clearUnusedBits();
1227 }
1228
1229 // Shift the low order words
1230 uint32_t breakWord = getNumWords() - offset -1;
1231 for (uint32_t i = 0; i < breakWord; ++i)
Reid Spencerd99feaf2007-03-01 05:39:56 +00001232 val[i] = (pVal[i+offset] >> wordShift) |
1233 (pVal[i+offset+1] << (APINT_BITS_PER_WORD - wordShift));
Reid Spencer44eef162007-02-26 01:19:48 +00001234 // Shift the break word.
1235 val[breakWord] = pVal[breakWord+offset] >> wordShift;
1236
1237 // Remaining words are 0
1238 for (uint32_t i = breakWord+1; i < getNumWords(); ++i)
1239 val[i] = 0;
1240 return APInt(val, BitWidth).clearUnusedBits();
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001241}
1242
Zhou Shenge93db8f2007-02-09 07:48:24 +00001243/// Left-shift this APInt by shiftAmt.
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001244/// @brief Left-shift function.
Dan Gohman105c1d42008-02-29 01:40:47 +00001245APInt APInt::shl(const APInt &shiftAmt) const {
1246 // It's undefined behavior in C to shift by BitWidth or greater, but
1247 return shl(shiftAmt.getLimitedValue(BitWidth));
1248}
1249
1250/// Left-shift this APInt by shiftAmt.
1251/// @brief Left-shift function.
Reid Spencera856b6e2007-02-18 18:38:44 +00001252APInt APInt::shl(uint32_t shiftAmt) const {
Reid Spencer632ebdf2007-02-24 20:19:37 +00001253 assert(shiftAmt <= BitWidth && "Invalid shift amount");
Reid Spencera5c84d92007-02-25 00:56:44 +00001254 if (isSingleWord()) {
Reid Spencer632ebdf2007-02-24 20:19:37 +00001255 if (shiftAmt == BitWidth)
Reid Spencera5c84d92007-02-25 00:56:44 +00001256 return APInt(BitWidth, 0); // avoid undefined shift results
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001257 return APInt(BitWidth, VAL << shiftAmt);
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001258 }
Reid Spencer632ebdf2007-02-24 20:19:37 +00001259
Reid Spencera5c84d92007-02-25 00:56:44 +00001260 // If all the bits were shifted out, the result is 0. This avoids issues
1261 // with shifting by the size of the integer type, which produces undefined
1262 // results. We define these "undefined results" to always be 0.
1263 if (shiftAmt == BitWidth)
1264 return APInt(BitWidth, 0);
1265
Reid Spencer81ee0202007-05-12 18:01:57 +00001266 // If none of the bits are shifted out, the result is *this. This avoids a
1267 // lshr by the words size in the loop below which can produce incorrect
1268 // results. It also avoids the expensive computation below for a common case.
1269 if (shiftAmt == 0)
1270 return *this;
1271
Reid Spencera5c84d92007-02-25 00:56:44 +00001272 // Create some space for the result.
1273 uint64_t * val = new uint64_t[getNumWords()];
1274
1275 // If we are shifting less than a word, do it the easy way
1276 if (shiftAmt < APINT_BITS_PER_WORD) {
1277 uint64_t carry = 0;
Reid Spencera5c84d92007-02-25 00:56:44 +00001278 for (uint32_t i = 0; i < getNumWords(); i++) {
1279 val[i] = pVal[i] << shiftAmt | carry;
1280 carry = pVal[i] >> (APINT_BITS_PER_WORD - shiftAmt);
1281 }
Reid Spencera41e93b2007-02-25 19:32:03 +00001282 return APInt(val, BitWidth).clearUnusedBits();
Reid Spencer632ebdf2007-02-24 20:19:37 +00001283 }
1284
Reid Spencera5c84d92007-02-25 00:56:44 +00001285 // Compute some values needed by the remaining shift algorithms
1286 uint32_t wordShift = shiftAmt % APINT_BITS_PER_WORD;
1287 uint32_t offset = shiftAmt / APINT_BITS_PER_WORD;
1288
1289 // If we are shifting whole words, just move whole words
1290 if (wordShift == 0) {
1291 for (uint32_t i = 0; i < offset; i++)
1292 val[i] = 0;
1293 for (uint32_t i = offset; i < getNumWords(); i++)
1294 val[i] = pVal[i-offset];
Reid Spencera41e93b2007-02-25 19:32:03 +00001295 return APInt(val,BitWidth).clearUnusedBits();
Reid Spencer632ebdf2007-02-24 20:19:37 +00001296 }
Reid Spencera5c84d92007-02-25 00:56:44 +00001297
1298 // Copy whole words from this to Result.
1299 uint32_t i = getNumWords() - 1;
1300 for (; i > offset; --i)
1301 val[i] = pVal[i-offset] << wordShift |
1302 pVal[i-offset-1] >> (APINT_BITS_PER_WORD - wordShift);
Reid Spencerab0e08a2007-02-25 01:08:58 +00001303 val[offset] = pVal[0] << wordShift;
Reid Spencera5c84d92007-02-25 00:56:44 +00001304 for (i = 0; i < offset; ++i)
1305 val[i] = 0;
Reid Spencera41e93b2007-02-25 19:32:03 +00001306 return APInt(val, BitWidth).clearUnusedBits();
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001307}
1308
Dan Gohman105c1d42008-02-29 01:40:47 +00001309APInt APInt::rotl(const APInt &rotateAmt) const {
1310 return rotl(rotateAmt.getLimitedValue(BitWidth));
1311}
1312
Reid Spencer4c50b522007-05-13 23:44:59 +00001313APInt APInt::rotl(uint32_t rotateAmt) const {
Reid Spencer98ed7db2007-05-14 00:15:28 +00001314 if (rotateAmt == 0)
1315 return *this;
Reid Spencer4c50b522007-05-13 23:44:59 +00001316 // Don't get too fancy, just use existing shift/or facilities
1317 APInt hi(*this);
1318 APInt lo(*this);
1319 hi.shl(rotateAmt);
1320 lo.lshr(BitWidth - rotateAmt);
1321 return hi | lo;
1322}
1323
Dan Gohman105c1d42008-02-29 01:40:47 +00001324APInt APInt::rotr(const APInt &rotateAmt) const {
1325 return rotr(rotateAmt.getLimitedValue(BitWidth));
1326}
1327
Reid Spencer4c50b522007-05-13 23:44:59 +00001328APInt APInt::rotr(uint32_t rotateAmt) const {
Reid Spencer98ed7db2007-05-14 00:15:28 +00001329 if (rotateAmt == 0)
1330 return *this;
Reid Spencer4c50b522007-05-13 23:44:59 +00001331 // Don't get too fancy, just use existing shift/or facilities
1332 APInt hi(*this);
1333 APInt lo(*this);
1334 lo.lshr(rotateAmt);
1335 hi.shl(BitWidth - rotateAmt);
1336 return hi | lo;
1337}
Reid Spencerd99feaf2007-03-01 05:39:56 +00001338
1339// Square Root - this method computes and returns the square root of "this".
1340// Three mechanisms are used for computation. For small values (<= 5 bits),
1341// a table lookup is done. This gets some performance for common cases. For
1342// values using less than 52 bits, the value is converted to double and then
1343// the libc sqrt function is called. The result is rounded and then converted
1344// back to a uint64_t which is then used to construct the result. Finally,
1345// the Babylonian method for computing square roots is used.
1346APInt APInt::sqrt() const {
1347
1348 // Determine the magnitude of the value.
1349 uint32_t magnitude = getActiveBits();
1350
1351 // Use a fast table for some small values. This also gets rid of some
1352 // rounding errors in libc sqrt for small values.
1353 if (magnitude <= 5) {
Reid Spencer2f6ad4d2007-03-01 17:47:31 +00001354 static const uint8_t results[32] = {
Reid Spencerc8841d22007-03-01 06:23:32 +00001355 /* 0 */ 0,
1356 /* 1- 2 */ 1, 1,
1357 /* 3- 6 */ 2, 2, 2, 2,
1358 /* 7-12 */ 3, 3, 3, 3, 3, 3,
1359 /* 13-20 */ 4, 4, 4, 4, 4, 4, 4, 4,
1360 /* 21-30 */ 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1361 /* 31 */ 6
1362 };
1363 return APInt(BitWidth, results[ (isSingleWord() ? VAL : pVal[0]) ]);
Reid Spencerd99feaf2007-03-01 05:39:56 +00001364 }
1365
1366 // If the magnitude of the value fits in less than 52 bits (the precision of
1367 // an IEEE double precision floating point value), then we can use the
1368 // libc sqrt function which will probably use a hardware sqrt computation.
1369 // This should be faster than the algorithm below.
Jeff Cohenb622c112007-03-05 00:00:42 +00001370 if (magnitude < 52) {
1371#ifdef _MSC_VER
1372 // Amazingly, VC++ doesn't have round().
1373 return APInt(BitWidth,
1374 uint64_t(::sqrt(double(isSingleWord()?VAL:pVal[0]))) + 0.5);
1375#else
Reid Spencerd99feaf2007-03-01 05:39:56 +00001376 return APInt(BitWidth,
1377 uint64_t(::round(::sqrt(double(isSingleWord()?VAL:pVal[0])))));
Jeff Cohenb622c112007-03-05 00:00:42 +00001378#endif
1379 }
Reid Spencerd99feaf2007-03-01 05:39:56 +00001380
1381 // Okay, all the short cuts are exhausted. We must compute it. The following
1382 // is a classical Babylonian method for computing the square root. This code
1383 // was adapted to APINt from a wikipedia article on such computations.
1384 // See http://www.wikipedia.org/ and go to the page named
1385 // Calculate_an_integer_square_root.
1386 uint32_t nbits = BitWidth, i = 4;
1387 APInt testy(BitWidth, 16);
1388 APInt x_old(BitWidth, 1);
1389 APInt x_new(BitWidth, 0);
1390 APInt two(BitWidth, 2);
1391
1392 // Select a good starting value using binary logarithms.
1393 for (;; i += 2, testy = testy.shl(2))
1394 if (i >= nbits || this->ule(testy)) {
1395 x_old = x_old.shl(i / 2);
1396 break;
1397 }
1398
1399 // Use the Babylonian method to arrive at the integer square root:
1400 for (;;) {
1401 x_new = (this->udiv(x_old) + x_old).udiv(two);
1402 if (x_old.ule(x_new))
1403 break;
1404 x_old = x_new;
1405 }
1406
1407 // Make sure we return the closest approximation
Reid Spencercf817562007-03-02 04:21:55 +00001408 // NOTE: The rounding calculation below is correct. It will produce an
1409 // off-by-one discrepancy with results from pari/gp. That discrepancy has been
1410 // determined to be a rounding issue with pari/gp as it begins to use a
1411 // floating point representation after 192 bits. There are no discrepancies
1412 // between this algorithm and pari/gp for bit widths < 192 bits.
Reid Spencerd99feaf2007-03-01 05:39:56 +00001413 APInt square(x_old * x_old);
1414 APInt nextSquare((x_old + 1) * (x_old +1));
1415 if (this->ult(square))
1416 return x_old;
Reid Spencercf817562007-03-02 04:21:55 +00001417 else if (this->ule(nextSquare)) {
1418 APInt midpoint((nextSquare - square).udiv(two));
1419 APInt offset(*this - square);
1420 if (offset.ult(midpoint))
Reid Spencerd99feaf2007-03-01 05:39:56 +00001421 return x_old;
Reid Spencercf817562007-03-02 04:21:55 +00001422 else
1423 return x_old + 1;
1424 } else
Reid Spencerd99feaf2007-03-01 05:39:56 +00001425 assert(0 && "Error in APInt::sqrt computation");
1426 return x_old + 1;
1427}
1428
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001429/// Implementation of Knuth's Algorithm D (Division of nonnegative integers)
1430/// from "Art of Computer Programming, Volume 2", section 4.3.1, p. 272. The
1431/// variables here have the same names as in the algorithm. Comments explain
1432/// the algorithm and any deviation from it.
1433static void KnuthDiv(uint32_t *u, uint32_t *v, uint32_t *q, uint32_t* r,
1434 uint32_t m, uint32_t n) {
1435 assert(u && "Must provide dividend");
1436 assert(v && "Must provide divisor");
1437 assert(q && "Must provide quotient");
Reid Spencera5e0d202007-02-24 03:58:46 +00001438 assert(u != v && u != q && v != q && "Must us different memory");
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001439 assert(n>1 && "n must be > 1");
1440
1441 // Knuth uses the value b as the base of the number system. In our case b
1442 // is 2^31 so we just set it to -1u.
1443 uint64_t b = uint64_t(1) << 32;
1444
Reid Spencera5e0d202007-02-24 03:58:46 +00001445 DEBUG(cerr << "KnuthDiv: m=" << m << " n=" << n << '\n');
1446 DEBUG(cerr << "KnuthDiv: original:");
1447 DEBUG(for (int i = m+n; i >=0; i--) cerr << " " << std::setbase(16) << u[i]);
1448 DEBUG(cerr << " by");
1449 DEBUG(for (int i = n; i >0; i--) cerr << " " << std::setbase(16) << v[i-1]);
1450 DEBUG(cerr << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001451 // D1. [Normalize.] Set d = b / (v[n-1] + 1) and multiply all the digits of
1452 // u and v by d. Note that we have taken Knuth's advice here to use a power
1453 // of 2 value for d such that d * v[n-1] >= b/2 (b is the base). A power of
1454 // 2 allows us to shift instead of multiply and it is easy to determine the
1455 // shift amount from the leading zeros. We are basically normalizing the u
1456 // and v so that its high bits are shifted to the top of v's range without
1457 // overflow. Note that this can require an extra word in u so that u must
1458 // be of length m+n+1.
1459 uint32_t shift = CountLeadingZeros_32(v[n-1]);
1460 uint32_t v_carry = 0;
1461 uint32_t u_carry = 0;
1462 if (shift) {
1463 for (uint32_t i = 0; i < m+n; ++i) {
1464 uint32_t u_tmp = u[i] >> (32 - shift);
1465 u[i] = (u[i] << shift) | u_carry;
1466 u_carry = u_tmp;
Reid Spencer100502d2007-02-17 03:16:00 +00001467 }
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001468 for (uint32_t i = 0; i < n; ++i) {
1469 uint32_t v_tmp = v[i] >> (32 - shift);
1470 v[i] = (v[i] << shift) | v_carry;
1471 v_carry = v_tmp;
1472 }
1473 }
1474 u[m+n] = u_carry;
Reid Spencera5e0d202007-02-24 03:58:46 +00001475 DEBUG(cerr << "KnuthDiv: normal:");
1476 DEBUG(for (int i = m+n; i >=0; i--) cerr << " " << std::setbase(16) << u[i]);
1477 DEBUG(cerr << " by");
1478 DEBUG(for (int i = n; i >0; i--) cerr << " " << std::setbase(16) << v[i-1]);
1479 DEBUG(cerr << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001480
1481 // D2. [Initialize j.] Set j to m. This is the loop counter over the places.
1482 int j = m;
1483 do {
Reid Spencera5e0d202007-02-24 03:58:46 +00001484 DEBUG(cerr << "KnuthDiv: quotient digit #" << j << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001485 // D3. [Calculate q'.].
1486 // Set qp = (u[j+n]*b + u[j+n-1]) / v[n-1]. (qp=qprime=q')
1487 // Set rp = (u[j+n]*b + u[j+n-1]) % v[n-1]. (rp=rprime=r')
1488 // Now test if qp == b or qp*v[n-2] > b*rp + u[j+n-2]; if so, decrease
1489 // qp by 1, inrease rp by v[n-1], and repeat this test if rp < b. The test
1490 // on v[n-2] determines at high speed most of the cases in which the trial
1491 // value qp is one too large, and it eliminates all cases where qp is two
1492 // too large.
Reid Spencercb292e42007-02-23 01:57:13 +00001493 uint64_t dividend = ((uint64_t(u[j+n]) << 32) + u[j+n-1]);
Reid Spencera5e0d202007-02-24 03:58:46 +00001494 DEBUG(cerr << "KnuthDiv: dividend == " << dividend << '\n');
Reid Spencercb292e42007-02-23 01:57:13 +00001495 uint64_t qp = dividend / v[n-1];
1496 uint64_t rp = dividend % v[n-1];
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001497 if (qp == b || qp*v[n-2] > b*rp + u[j+n-2]) {
1498 qp--;
1499 rp += v[n-1];
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001500 if (rp < b && (qp == b || qp*v[n-2] > b*rp + u[j+n-2]))
Reid Spencera5e0d202007-02-24 03:58:46 +00001501 qp--;
Reid Spencercb292e42007-02-23 01:57:13 +00001502 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001503 DEBUG(cerr << "KnuthDiv: qp == " << qp << ", rp == " << rp << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001504
Reid Spencercb292e42007-02-23 01:57:13 +00001505 // D4. [Multiply and subtract.] Replace (u[j+n]u[j+n-1]...u[j]) with
1506 // (u[j+n]u[j+n-1]..u[j]) - qp * (v[n-1]...v[1]v[0]). This computation
1507 // consists of a simple multiplication by a one-place number, combined with
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001508 // a subtraction.
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001509 bool isNeg = false;
Reid Spencercb292e42007-02-23 01:57:13 +00001510 for (uint32_t i = 0; i < n; ++i) {
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001511 uint64_t u_tmp = uint64_t(u[j+i]) | (uint64_t(u[j+i+1]) << 32);
Reid Spencera5e0d202007-02-24 03:58:46 +00001512 uint64_t subtrahend = uint64_t(qp) * uint64_t(v[i]);
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001513 bool borrow = subtrahend > u_tmp;
Reid Spencera5e0d202007-02-24 03:58:46 +00001514 DEBUG(cerr << "KnuthDiv: u_tmp == " << u_tmp
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001515 << ", subtrahend == " << subtrahend
1516 << ", borrow = " << borrow << '\n');
Reid Spencera5e0d202007-02-24 03:58:46 +00001517
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001518 uint64_t result = u_tmp - subtrahend;
1519 uint32_t k = j + i;
1520 u[k++] = result & (b-1); // subtract low word
1521 u[k++] = result >> 32; // subtract high word
1522 while (borrow && k <= m+n) { // deal with borrow to the left
1523 borrow = u[k] == 0;
1524 u[k]--;
1525 k++;
1526 }
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001527 isNeg |= borrow;
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001528 DEBUG(cerr << "KnuthDiv: u[j+i] == " << u[j+i] << ", u[j+i+1] == " <<
1529 u[j+i+1] << '\n');
Reid Spencera5e0d202007-02-24 03:58:46 +00001530 }
1531 DEBUG(cerr << "KnuthDiv: after subtraction:");
1532 DEBUG(for (int i = m+n; i >=0; i--) cerr << " " << u[i]);
1533 DEBUG(cerr << '\n');
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001534 // The digits (u[j+n]...u[j]) should be kept positive; if the result of
1535 // this step is actually negative, (u[j+n]...u[j]) should be left as the
1536 // true value plus b**(n+1), namely as the b's complement of
Reid Spencercb292e42007-02-23 01:57:13 +00001537 // the true value, and a "borrow" to the left should be remembered.
1538 //
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001539 if (isNeg) {
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00001540 bool carry = true; // true because b's complement is "complement + 1"
1541 for (uint32_t i = 0; i <= m+n; ++i) {
1542 u[i] = ~u[i] + carry; // b's complement
1543 carry = carry && u[i] == 0;
Reid Spencera5e0d202007-02-24 03:58:46 +00001544 }
Reid Spencercb292e42007-02-23 01:57:13 +00001545 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001546 DEBUG(cerr << "KnuthDiv: after complement:");
1547 DEBUG(for (int i = m+n; i >=0; i--) cerr << " " << u[i]);
1548 DEBUG(cerr << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001549
1550 // D5. [Test remainder.] Set q[j] = qp. If the result of step D4 was
1551 // negative, go to step D6; otherwise go on to step D7.
1552 q[j] = qp;
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001553 if (isNeg) {
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001554 // D6. [Add back]. The probability that this step is necessary is very
1555 // small, on the order of only 2/b. Make sure that test data accounts for
Reid Spencercb292e42007-02-23 01:57:13 +00001556 // this possibility. Decrease q[j] by 1
1557 q[j]--;
1558 // and add (0v[n-1]...v[1]v[0]) to (u[j+n]u[j+n-1]...u[j+1]u[j]).
1559 // A carry will occur to the left of u[j+n], and it should be ignored
1560 // since it cancels with the borrow that occurred in D4.
1561 bool carry = false;
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001562 for (uint32_t i = 0; i < n; i++) {
Reid Spencera5e0d202007-02-24 03:58:46 +00001563 uint32_t limit = std::min(u[j+i],v[i]);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001564 u[j+i] += v[i] + carry;
Reid Spencera5e0d202007-02-24 03:58:46 +00001565 carry = u[j+i] < limit || (carry && u[j+i] == limit);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001566 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001567 u[j+n] += carry;
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001568 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001569 DEBUG(cerr << "KnuthDiv: after correction:");
1570 DEBUG(for (int i = m+n; i >=0; i--) cerr <<" " << u[i]);
1571 DEBUG(cerr << "\nKnuthDiv: digit result = " << q[j] << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001572
Reid Spencercb292e42007-02-23 01:57:13 +00001573 // D7. [Loop on j.] Decrease j by one. Now if j >= 0, go back to D3.
1574 } while (--j >= 0);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001575
Reid Spencera5e0d202007-02-24 03:58:46 +00001576 DEBUG(cerr << "KnuthDiv: quotient:");
1577 DEBUG(for (int i = m; i >=0; i--) cerr <<" " << q[i]);
1578 DEBUG(cerr << '\n');
1579
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001580 // D8. [Unnormalize]. Now q[...] is the desired quotient, and the desired
1581 // remainder may be obtained by dividing u[...] by d. If r is non-null we
1582 // compute the remainder (urem uses this).
1583 if (r) {
1584 // The value d is expressed by the "shift" value above since we avoided
1585 // multiplication by d by using a shift left. So, all we have to do is
1586 // shift right here. In order to mak
Reid Spencer468ad9112007-02-24 20:38:01 +00001587 if (shift) {
1588 uint32_t carry = 0;
1589 DEBUG(cerr << "KnuthDiv: remainder:");
1590 for (int i = n-1; i >= 0; i--) {
1591 r[i] = (u[i] >> shift) | carry;
1592 carry = u[i] << (32 - shift);
1593 DEBUG(cerr << " " << r[i]);
1594 }
1595 } else {
1596 for (int i = n-1; i >= 0; i--) {
1597 r[i] = u[i];
1598 DEBUG(cerr << " " << r[i]);
1599 }
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001600 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001601 DEBUG(cerr << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001602 }
Reid Spencera5e0d202007-02-24 03:58:46 +00001603 DEBUG(cerr << std::setbase(10) << '\n');
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001604}
1605
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001606void APInt::divide(const APInt LHS, uint32_t lhsWords,
1607 const APInt &RHS, uint32_t rhsWords,
1608 APInt *Quotient, APInt *Remainder)
1609{
1610 assert(lhsWords >= rhsWords && "Fractional result");
1611
1612 // First, compose the values into an array of 32-bit words instead of
1613 // 64-bit words. This is a necessity of both the "short division" algorithm
1614 // and the the Knuth "classical algorithm" which requires there to be native
1615 // operations for +, -, and * on an m bit value with an m*2 bit result. We
1616 // can't use 64-bit operands here because we don't have native results of
1617 // 128-bits. Furthremore, casting the 64-bit values to 32-bit values won't
1618 // work on large-endian machines.
1619 uint64_t mask = ~0ull >> (sizeof(uint32_t)*8);
1620 uint32_t n = rhsWords * 2;
1621 uint32_t m = (lhsWords * 2) - n;
Reid Spencer522ca7c2007-02-25 01:56:07 +00001622
1623 // Allocate space for the temporary values we need either on the stack, if
1624 // it will fit, or on the heap if it won't.
1625 uint32_t SPACE[128];
1626 uint32_t *U = 0;
1627 uint32_t *V = 0;
1628 uint32_t *Q = 0;
1629 uint32_t *R = 0;
1630 if ((Remainder?4:3)*n+2*m+1 <= 128) {
1631 U = &SPACE[0];
1632 V = &SPACE[m+n+1];
1633 Q = &SPACE[(m+n+1) + n];
1634 if (Remainder)
1635 R = &SPACE[(m+n+1) + n + (m+n)];
1636 } else {
1637 U = new uint32_t[m + n + 1];
1638 V = new uint32_t[n];
1639 Q = new uint32_t[m+n];
1640 if (Remainder)
1641 R = new uint32_t[n];
1642 }
1643
1644 // Initialize the dividend
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001645 memset(U, 0, (m+n+1)*sizeof(uint32_t));
1646 for (unsigned i = 0; i < lhsWords; ++i) {
Reid Spencer867b4062007-02-22 00:58:45 +00001647 uint64_t tmp = (LHS.getNumWords() == 1 ? LHS.VAL : LHS.pVal[i]);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001648 U[i * 2] = tmp & mask;
1649 U[i * 2 + 1] = tmp >> (sizeof(uint32_t)*8);
1650 }
1651 U[m+n] = 0; // this extra word is for "spill" in the Knuth algorithm.
1652
Reid Spencer522ca7c2007-02-25 01:56:07 +00001653 // Initialize the divisor
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001654 memset(V, 0, (n)*sizeof(uint32_t));
1655 for (unsigned i = 0; i < rhsWords; ++i) {
Reid Spencer867b4062007-02-22 00:58:45 +00001656 uint64_t tmp = (RHS.getNumWords() == 1 ? RHS.VAL : RHS.pVal[i]);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001657 V[i * 2] = tmp & mask;
1658 V[i * 2 + 1] = tmp >> (sizeof(uint32_t)*8);
1659 }
1660
Reid Spencer522ca7c2007-02-25 01:56:07 +00001661 // initialize the quotient and remainder
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001662 memset(Q, 0, (m+n) * sizeof(uint32_t));
Reid Spencer522ca7c2007-02-25 01:56:07 +00001663 if (Remainder)
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001664 memset(R, 0, n * sizeof(uint32_t));
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001665
1666 // Now, adjust m and n for the Knuth division. n is the number of words in
1667 // the divisor. m is the number of words by which the dividend exceeds the
1668 // divisor (i.e. m+n is the length of the dividend). These sizes must not
1669 // contain any zero words or the Knuth algorithm fails.
1670 for (unsigned i = n; i > 0 && V[i-1] == 0; i--) {
1671 n--;
1672 m++;
1673 }
1674 for (unsigned i = m+n; i > 0 && U[i-1] == 0; i--)
1675 m--;
1676
1677 // If we're left with only a single word for the divisor, Knuth doesn't work
1678 // so we implement the short division algorithm here. This is much simpler
1679 // and faster because we are certain that we can divide a 64-bit quantity
1680 // by a 32-bit quantity at hardware speed and short division is simply a
1681 // series of such operations. This is just like doing short division but we
1682 // are using base 2^32 instead of base 10.
1683 assert(n != 0 && "Divide by zero?");
1684 if (n == 1) {
1685 uint32_t divisor = V[0];
1686 uint32_t remainder = 0;
1687 for (int i = m+n-1; i >= 0; i--) {
1688 uint64_t partial_dividend = uint64_t(remainder) << 32 | U[i];
1689 if (partial_dividend == 0) {
1690 Q[i] = 0;
1691 remainder = 0;
1692 } else if (partial_dividend < divisor) {
1693 Q[i] = 0;
1694 remainder = partial_dividend;
1695 } else if (partial_dividend == divisor) {
1696 Q[i] = 1;
1697 remainder = 0;
1698 } else {
1699 Q[i] = partial_dividend / divisor;
1700 remainder = partial_dividend - (Q[i] * divisor);
1701 }
1702 }
1703 if (R)
1704 R[0] = remainder;
1705 } else {
1706 // Now we're ready to invoke the Knuth classical divide algorithm. In this
1707 // case n > 1.
1708 KnuthDiv(U, V, Q, R, m, n);
1709 }
1710
1711 // If the caller wants the quotient
1712 if (Quotient) {
1713 // Set up the Quotient value's memory.
1714 if (Quotient->BitWidth != LHS.BitWidth) {
1715 if (Quotient->isSingleWord())
1716 Quotient->VAL = 0;
1717 else
Reid Spencer7c16cd22007-02-26 23:38:21 +00001718 delete [] Quotient->pVal;
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001719 Quotient->BitWidth = LHS.BitWidth;
1720 if (!Quotient->isSingleWord())
Reid Spencer58a6a432007-02-21 08:21:52 +00001721 Quotient->pVal = getClearedMemory(Quotient->getNumWords());
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001722 } else
1723 Quotient->clear();
1724
1725 // The quotient is in Q. Reconstitute the quotient into Quotient's low
1726 // order words.
1727 if (lhsWords == 1) {
1728 uint64_t tmp =
1729 uint64_t(Q[0]) | (uint64_t(Q[1]) << (APINT_BITS_PER_WORD / 2));
1730 if (Quotient->isSingleWord())
1731 Quotient->VAL = tmp;
1732 else
1733 Quotient->pVal[0] = tmp;
1734 } else {
1735 assert(!Quotient->isSingleWord() && "Quotient APInt not large enough");
1736 for (unsigned i = 0; i < lhsWords; ++i)
1737 Quotient->pVal[i] =
1738 uint64_t(Q[i*2]) | (uint64_t(Q[i*2+1]) << (APINT_BITS_PER_WORD / 2));
1739 }
1740 }
1741
1742 // If the caller wants the remainder
1743 if (Remainder) {
1744 // Set up the Remainder value's memory.
1745 if (Remainder->BitWidth != RHS.BitWidth) {
1746 if (Remainder->isSingleWord())
1747 Remainder->VAL = 0;
1748 else
Reid Spencer7c16cd22007-02-26 23:38:21 +00001749 delete [] Remainder->pVal;
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001750 Remainder->BitWidth = RHS.BitWidth;
1751 if (!Remainder->isSingleWord())
Reid Spencer58a6a432007-02-21 08:21:52 +00001752 Remainder->pVal = getClearedMemory(Remainder->getNumWords());
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001753 } else
1754 Remainder->clear();
1755
1756 // The remainder is in R. Reconstitute the remainder into Remainder's low
1757 // order words.
1758 if (rhsWords == 1) {
1759 uint64_t tmp =
1760 uint64_t(R[0]) | (uint64_t(R[1]) << (APINT_BITS_PER_WORD / 2));
1761 if (Remainder->isSingleWord())
1762 Remainder->VAL = tmp;
1763 else
1764 Remainder->pVal[0] = tmp;
1765 } else {
1766 assert(!Remainder->isSingleWord() && "Remainder APInt not large enough");
1767 for (unsigned i = 0; i < rhsWords; ++i)
1768 Remainder->pVal[i] =
1769 uint64_t(R[i*2]) | (uint64_t(R[i*2+1]) << (APINT_BITS_PER_WORD / 2));
1770 }
1771 }
1772
1773 // Clean up the memory we allocated.
Reid Spencer522ca7c2007-02-25 01:56:07 +00001774 if (U != &SPACE[0]) {
1775 delete [] U;
1776 delete [] V;
1777 delete [] Q;
1778 delete [] R;
1779 }
Reid Spencer100502d2007-02-17 03:16:00 +00001780}
1781
Reid Spencer1d072122007-02-16 22:36:51 +00001782APInt APInt::udiv(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +00001783 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencer39867762007-02-17 02:07:07 +00001784
1785 // First, deal with the easy case
1786 if (isSingleWord()) {
1787 assert(RHS.VAL != 0 && "Divide by zero?");
1788 return APInt(BitWidth, VAL / RHS.VAL);
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001789 }
Reid Spencer39867762007-02-17 02:07:07 +00001790
Reid Spencer39867762007-02-17 02:07:07 +00001791 // Get some facts about the LHS and RHS number of bits and words
Reid Spencera856b6e2007-02-18 18:38:44 +00001792 uint32_t rhsBits = RHS.getActiveBits();
1793 uint32_t rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
Reid Spencer39867762007-02-17 02:07:07 +00001794 assert(rhsWords && "Divided by zero???");
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001795 uint32_t lhsBits = this->getActiveBits();
Reid Spencera856b6e2007-02-18 18:38:44 +00001796 uint32_t lhsWords = !lhsBits ? 0 : (APInt::whichWord(lhsBits - 1) + 1);
Reid Spencer39867762007-02-17 02:07:07 +00001797
1798 // Deal with some degenerate cases
1799 if (!lhsWords)
Reid Spencer58a6a432007-02-21 08:21:52 +00001800 // 0 / X ===> 0
1801 return APInt(BitWidth, 0);
1802 else if (lhsWords < rhsWords || this->ult(RHS)) {
1803 // X / Y ===> 0, iff X < Y
1804 return APInt(BitWidth, 0);
1805 } else if (*this == RHS) {
1806 // X / X ===> 1
1807 return APInt(BitWidth, 1);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001808 } else if (lhsWords == 1 && rhsWords == 1) {
Reid Spencer39867762007-02-17 02:07:07 +00001809 // All high words are zero, just use native divide
Reid Spencer58a6a432007-02-21 08:21:52 +00001810 return APInt(BitWidth, this->pVal[0] / RHS.pVal[0]);
Reid Spencer39867762007-02-17 02:07:07 +00001811 }
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001812
1813 // We have to compute it the hard way. Invoke the Knuth divide algorithm.
1814 APInt Quotient(1,0); // to hold result.
1815 divide(*this, lhsWords, RHS, rhsWords, &Quotient, 0);
1816 return Quotient;
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001817}
1818
Reid Spencer1d072122007-02-16 22:36:51 +00001819APInt APInt::urem(const APInt& RHS) const {
Reid Spencera32372d12007-02-17 00:18:01 +00001820 assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
Reid Spencer39867762007-02-17 02:07:07 +00001821 if (isSingleWord()) {
1822 assert(RHS.VAL != 0 && "Remainder by zero?");
1823 return APInt(BitWidth, VAL % RHS.VAL);
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001824 }
Reid Spencer39867762007-02-17 02:07:07 +00001825
Reid Spencer58a6a432007-02-21 08:21:52 +00001826 // Get some facts about the LHS
1827 uint32_t lhsBits = getActiveBits();
1828 uint32_t lhsWords = !lhsBits ? 0 : (whichWord(lhsBits - 1) + 1);
Reid Spencer39867762007-02-17 02:07:07 +00001829
1830 // Get some facts about the RHS
Reid Spencera856b6e2007-02-18 18:38:44 +00001831 uint32_t rhsBits = RHS.getActiveBits();
1832 uint32_t rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
Reid Spencer39867762007-02-17 02:07:07 +00001833 assert(rhsWords && "Performing remainder operation by zero ???");
1834
Reid Spencer39867762007-02-17 02:07:07 +00001835 // Check the degenerate cases
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001836 if (lhsWords == 0) {
Reid Spencer58a6a432007-02-21 08:21:52 +00001837 // 0 % Y ===> 0
1838 return APInt(BitWidth, 0);
1839 } else if (lhsWords < rhsWords || this->ult(RHS)) {
1840 // X % Y ===> X, iff X < Y
1841 return *this;
1842 } else if (*this == RHS) {
Reid Spencer39867762007-02-17 02:07:07 +00001843 // X % X == 0;
Reid Spencer58a6a432007-02-21 08:21:52 +00001844 return APInt(BitWidth, 0);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001845 } else if (lhsWords == 1) {
Reid Spencer39867762007-02-17 02:07:07 +00001846 // All high words are zero, just use native remainder
Reid Spencer58a6a432007-02-21 08:21:52 +00001847 return APInt(BitWidth, pVal[0] % RHS.pVal[0]);
Reid Spencer39867762007-02-17 02:07:07 +00001848 }
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001849
Reid Spencer4c50b522007-05-13 23:44:59 +00001850 // We have to compute it the hard way. Invoke the Knuth divide algorithm.
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001851 APInt Remainder(1,0);
1852 divide(*this, lhsWords, RHS, rhsWords, 0, &Remainder);
1853 return Remainder;
Zhou Shengfbf61ea2007-02-08 14:35:19 +00001854}
Reid Spencer100502d2007-02-17 03:16:00 +00001855
Reid Spencer4c50b522007-05-13 23:44:59 +00001856void APInt::udivrem(const APInt &LHS, const APInt &RHS,
1857 APInt &Quotient, APInt &Remainder) {
1858 // Get some size facts about the dividend and divisor
1859 uint32_t lhsBits = LHS.getActiveBits();
1860 uint32_t lhsWords = !lhsBits ? 0 : (APInt::whichWord(lhsBits - 1) + 1);
1861 uint32_t rhsBits = RHS.getActiveBits();
1862 uint32_t rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
1863
1864 // Check the degenerate cases
1865 if (lhsWords == 0) {
1866 Quotient = 0; // 0 / Y ===> 0
1867 Remainder = 0; // 0 % Y ===> 0
1868 return;
1869 }
1870
1871 if (lhsWords < rhsWords || LHS.ult(RHS)) {
1872 Quotient = 0; // X / Y ===> 0, iff X < Y
1873 Remainder = LHS; // X % Y ===> X, iff X < Y
1874 return;
1875 }
1876
1877 if (LHS == RHS) {
1878 Quotient = 1; // X / X ===> 1
1879 Remainder = 0; // X % X ===> 0;
1880 return;
1881 }
1882
1883 if (lhsWords == 1 && rhsWords == 1) {
1884 // There is only one word to consider so use the native versions.
1885 if (LHS.isSingleWord()) {
1886 Quotient = APInt(LHS.getBitWidth(), LHS.VAL / RHS.VAL);
1887 Remainder = APInt(LHS.getBitWidth(), LHS.VAL % RHS.VAL);
1888 } else {
1889 Quotient = APInt(LHS.getBitWidth(), LHS.pVal[0] / RHS.pVal[0]);
1890 Remainder = APInt(LHS.getBitWidth(), LHS.pVal[0] % RHS.pVal[0]);
1891 }
1892 return;
1893 }
1894
1895 // Okay, lets do it the long way
1896 divide(LHS, lhsWords, RHS, rhsWords, &Quotient, &Remainder);
1897}
1898
Reid Spencer1ba83352007-02-21 03:55:44 +00001899void APInt::fromString(uint32_t numbits, const char *str, uint32_t slen,
Reid Spencer100502d2007-02-17 03:16:00 +00001900 uint8_t radix) {
Reid Spencer1ba83352007-02-21 03:55:44 +00001901 // Check our assumptions here
Reid Spencer100502d2007-02-17 03:16:00 +00001902 assert((radix == 10 || radix == 8 || radix == 16 || radix == 2) &&
1903 "Radix should be 2, 8, 10, or 16!");
Reid Spencer1ba83352007-02-21 03:55:44 +00001904 assert(str && "String is null?");
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001905 bool isNeg = str[0] == '-';
1906 if (isNeg)
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001907 str++, slen--;
Chris Lattnerdad2d092007-05-03 18:15:36 +00001908 assert((slen <= numbits || radix != 2) && "Insufficient bit width");
1909 assert((slen*3 <= numbits || radix != 8) && "Insufficient bit width");
1910 assert((slen*4 <= numbits || radix != 16) && "Insufficient bit width");
1911 assert(((slen*64)/22 <= numbits || radix != 10) && "Insufficient bit width");
Reid Spencer1ba83352007-02-21 03:55:44 +00001912
1913 // Allocate memory
1914 if (!isSingleWord())
1915 pVal = getClearedMemory(getNumWords());
1916
1917 // Figure out if we can shift instead of multiply
1918 uint32_t shift = (radix == 16 ? 4 : radix == 8 ? 3 : radix == 2 ? 1 : 0);
1919
1920 // Set up an APInt for the digit to add outside the loop so we don't
1921 // constantly construct/destruct it.
1922 APInt apdigit(getBitWidth(), 0);
1923 APInt apradix(getBitWidth(), radix);
1924
1925 // Enter digit traversal loop
1926 for (unsigned i = 0; i < slen; i++) {
1927 // Get a digit
1928 uint32_t digit = 0;
1929 char cdigit = str[i];
Reid Spencera93c9812007-05-16 19:18:22 +00001930 if (radix == 16) {
1931 if (!isxdigit(cdigit))
1932 assert(0 && "Invalid hex digit in string");
1933 if (isdigit(cdigit))
1934 digit = cdigit - '0';
1935 else if (cdigit >= 'a')
Reid Spencer1ba83352007-02-21 03:55:44 +00001936 digit = cdigit - 'a' + 10;
1937 else if (cdigit >= 'A')
1938 digit = cdigit - 'A' + 10;
1939 else
Reid Spencera93c9812007-05-16 19:18:22 +00001940 assert(0 && "huh? we shouldn't get here");
1941 } else if (isdigit(cdigit)) {
1942 digit = cdigit - '0';
1943 } else {
Reid Spencer1ba83352007-02-21 03:55:44 +00001944 assert(0 && "Invalid character in digit string");
Reid Spencera93c9812007-05-16 19:18:22 +00001945 }
Reid Spencer1ba83352007-02-21 03:55:44 +00001946
Reid Spencera93c9812007-05-16 19:18:22 +00001947 // Shift or multiply the value by the radix
Reid Spencer1ba83352007-02-21 03:55:44 +00001948 if (shift)
Reid Spencera93c9812007-05-16 19:18:22 +00001949 *this <<= shift;
Reid Spencer1ba83352007-02-21 03:55:44 +00001950 else
1951 *this *= apradix;
1952
1953 // Add in the digit we just interpreted
Reid Spencer632ebdf2007-02-24 20:19:37 +00001954 if (apdigit.isSingleWord())
1955 apdigit.VAL = digit;
1956 else
1957 apdigit.pVal[0] = digit;
Reid Spencer1ba83352007-02-21 03:55:44 +00001958 *this += apdigit;
Reid Spencer100502d2007-02-17 03:16:00 +00001959 }
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001960 // If its negative, put it in two's complement form
Reid Spenceraa8dcfe2007-02-26 07:44:38 +00001961 if (isNeg) {
1962 (*this)--;
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001963 this->flip();
Reid Spencerb6b5cc32007-02-25 23:44:53 +00001964 }
Reid Spencer100502d2007-02-17 03:16:00 +00001965}
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001966
Reid Spencerfb77b2b2007-02-20 08:51:03 +00001967std::string APInt::toString(uint8_t radix, bool wantSigned) const {
1968 assert((radix == 10 || radix == 8 || radix == 16 || radix == 2) &&
1969 "Radix should be 2, 8, 10, or 16!");
1970 static const char *digits[] = {
1971 "0","1","2","3","4","5","6","7","8","9","A","B","C","D","E","F"
1972 };
1973 std::string result;
1974 uint32_t bits_used = getActiveBits();
1975 if (isSingleWord()) {
1976 char buf[65];
1977 const char *format = (radix == 10 ? (wantSigned ? "%lld" : "%llu") :
1978 (radix == 16 ? "%llX" : (radix == 8 ? "%llo" : 0)));
1979 if (format) {
1980 if (wantSigned) {
1981 int64_t sextVal = (int64_t(VAL) << (APINT_BITS_PER_WORD-BitWidth)) >>
1982 (APINT_BITS_PER_WORD-BitWidth);
1983 sprintf(buf, format, sextVal);
1984 } else
1985 sprintf(buf, format, VAL);
1986 } else {
1987 memset(buf, 0, 65);
1988 uint64_t v = VAL;
1989 while (bits_used) {
1990 uint32_t bit = v & 1;
1991 bits_used--;
1992 buf[bits_used] = digits[bit][0];
1993 v >>=1;
1994 }
1995 }
1996 result = buf;
1997 return result;
1998 }
1999
2000 if (radix != 10) {
Reid Spencer5c138132007-05-17 19:23:02 +00002001 // For the 2, 8 and 16 bit cases, we can just shift instead of divide
2002 // because the number of bits per digit (1,3 and 4 respectively) divides
2003 // equaly. We just shift until there value is zero.
2004
2005 // First, check for a zero value and just short circuit the logic below.
2006 if (*this == 0)
2007 result = "0";
2008 else {
2009 APInt tmp(*this);
2010 size_t insert_at = 0;
2011 if (wantSigned && this->isNegative()) {
2012 // They want to print the signed version and it is a negative value
2013 // Flip the bits and add one to turn it into the equivalent positive
2014 // value and put a '-' in the result.
2015 tmp.flip();
2016 tmp++;
2017 result = "-";
2018 insert_at = 1;
2019 }
2020 // Just shift tmp right for each digit width until it becomes zero
2021 uint32_t shift = (radix == 16 ? 4 : (radix == 8 ? 3 : 1));
2022 uint64_t mask = radix - 1;
2023 APInt zero(tmp.getBitWidth(), 0);
2024 while (tmp.ne(zero)) {
Reid Spencerb6af1aa2007-05-19 00:29:55 +00002025 unsigned digit = (tmp.isSingleWord() ? tmp.VAL : tmp.pVal[0]) & mask;
Reid Spencer5c138132007-05-17 19:23:02 +00002026 result.insert(insert_at, digits[digit]);
Reid Spencerb6af1aa2007-05-19 00:29:55 +00002027 tmp = tmp.lshr(shift);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00002028 }
2029 }
2030 return result;
2031 }
2032
2033 APInt tmp(*this);
2034 APInt divisor(4, radix);
2035 APInt zero(tmp.getBitWidth(), 0);
2036 size_t insert_at = 0;
2037 if (wantSigned && tmp[BitWidth-1]) {
2038 // They want to print the signed version and it is a negative value
2039 // Flip the bits and add one to turn it into the equivalent positive
2040 // value and put a '-' in the result.
2041 tmp.flip();
2042 tmp++;
2043 result = "-";
2044 insert_at = 1;
2045 }
Reid Spencer74cf82e2007-02-21 00:29:48 +00002046 if (tmp == APInt(tmp.getBitWidth(), 0))
Reid Spencerfb77b2b2007-02-20 08:51:03 +00002047 result = "0";
2048 else while (tmp.ne(zero)) {
2049 APInt APdigit(1,0);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00002050 APInt tmp2(tmp.getBitWidth(), 0);
Reid Spencer1ba83352007-02-21 03:55:44 +00002051 divide(tmp, tmp.getNumWords(), divisor, divisor.getNumWords(), &tmp2,
2052 &APdigit);
Reid Spencerb2bc9852007-02-26 21:02:27 +00002053 uint32_t digit = APdigit.getZExtValue();
Reid Spencer1ba83352007-02-21 03:55:44 +00002054 assert(digit < radix && "divide failed");
2055 result.insert(insert_at,digits[digit]);
Reid Spencerfb77b2b2007-02-20 08:51:03 +00002056 tmp = tmp2;
2057 }
2058
2059 return result;
2060}
2061
Reid Spencer1ba83352007-02-21 03:55:44 +00002062void APInt::dump() const
2063{
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00002064 cerr << "APInt(" << BitWidth << ")=" << std::setbase(16);
Reid Spencer1ba83352007-02-21 03:55:44 +00002065 if (isSingleWord())
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00002066 cerr << VAL;
Reid Spencer1ba83352007-02-21 03:55:44 +00002067 else for (unsigned i = getNumWords(); i > 0; i--) {
Reid Spencerdf6cf5a2007-02-24 10:01:42 +00002068 cerr << pVal[i-1] << " ";
Reid Spencer1ba83352007-02-21 03:55:44 +00002069 }
Chris Lattnerb0f158c2007-08-23 05:15:32 +00002070 cerr << " U(" << this->toStringUnsigned(10) << ") S("
Dale Johannesen98d3a082007-09-14 22:26:36 +00002071 << this->toStringSigned(10) << ")" << std::setbase(10);
Reid Spencer1ba83352007-02-21 03:55:44 +00002072}
Chris Lattner6b695682007-08-16 15:56:55 +00002073
2074// This implements a variety of operations on a representation of
2075// arbitrary precision, two's-complement, bignum integer values.
2076
2077/* Assumed by lowHalf, highHalf, partMSB and partLSB. A fairly safe
2078 and unrestricting assumption. */
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002079COMPILE_TIME_ASSERT(integerPartWidth % 2 == 0);
Chris Lattner6b695682007-08-16 15:56:55 +00002080
2081/* Some handy functions local to this file. */
2082namespace {
2083
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002084 /* Returns the integer part with the least significant BITS set.
2085 BITS cannot be zero. */
2086 inline integerPart
2087 lowBitMask(unsigned int bits)
2088 {
2089 assert (bits != 0 && bits <= integerPartWidth);
2090
2091 return ~(integerPart) 0 >> (integerPartWidth - bits);
2092 }
2093
Neil Boothc8b650a2007-10-06 00:43:45 +00002094 /* Returns the value of the lower half of PART. */
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002095 inline integerPart
2096 lowHalf(integerPart part)
2097 {
2098 return part & lowBitMask(integerPartWidth / 2);
2099 }
2100
Neil Boothc8b650a2007-10-06 00:43:45 +00002101 /* Returns the value of the upper half of PART. */
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002102 inline integerPart
2103 highHalf(integerPart part)
2104 {
2105 return part >> (integerPartWidth / 2);
2106 }
2107
Neil Boothc8b650a2007-10-06 00:43:45 +00002108 /* Returns the bit number of the most significant set bit of a part.
2109 If the input number has no bits set -1U is returned. */
Chris Lattner6b695682007-08-16 15:56:55 +00002110 unsigned int
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002111 partMSB(integerPart value)
Chris Lattner6b695682007-08-16 15:56:55 +00002112 {
2113 unsigned int n, msb;
2114
2115 if (value == 0)
2116 return -1U;
2117
2118 n = integerPartWidth / 2;
2119
2120 msb = 0;
2121 do {
2122 if (value >> n) {
2123 value >>= n;
2124 msb += n;
2125 }
2126
2127 n >>= 1;
2128 } while (n);
2129
2130 return msb;
2131 }
2132
Neil Boothc8b650a2007-10-06 00:43:45 +00002133 /* Returns the bit number of the least significant set bit of a
2134 part. If the input number has no bits set -1U is returned. */
Chris Lattner6b695682007-08-16 15:56:55 +00002135 unsigned int
2136 partLSB(integerPart value)
2137 {
2138 unsigned int n, lsb;
2139
2140 if (value == 0)
2141 return -1U;
2142
2143 lsb = integerPartWidth - 1;
2144 n = integerPartWidth / 2;
2145
2146 do {
2147 if (value << n) {
2148 value <<= n;
2149 lsb -= n;
2150 }
2151
2152 n >>= 1;
2153 } while (n);
2154
2155 return lsb;
2156 }
2157}
2158
2159/* Sets the least significant part of a bignum to the input value, and
2160 zeroes out higher parts. */
2161void
2162APInt::tcSet(integerPart *dst, integerPart part, unsigned int parts)
2163{
2164 unsigned int i;
2165
Neil Boothb6182162007-10-08 13:47:12 +00002166 assert (parts > 0);
2167
Chris Lattner6b695682007-08-16 15:56:55 +00002168 dst[0] = part;
2169 for(i = 1; i < parts; i++)
2170 dst[i] = 0;
2171}
2172
2173/* Assign one bignum to another. */
2174void
2175APInt::tcAssign(integerPart *dst, const integerPart *src, unsigned int parts)
2176{
2177 unsigned int i;
2178
2179 for(i = 0; i < parts; i++)
2180 dst[i] = src[i];
2181}
2182
2183/* Returns true if a bignum is zero, false otherwise. */
2184bool
2185APInt::tcIsZero(const integerPart *src, unsigned int parts)
2186{
2187 unsigned int i;
2188
2189 for(i = 0; i < parts; i++)
2190 if (src[i])
2191 return false;
2192
2193 return true;
2194}
2195
2196/* Extract the given bit of a bignum; returns 0 or 1. */
2197int
2198APInt::tcExtractBit(const integerPart *parts, unsigned int bit)
2199{
2200 return(parts[bit / integerPartWidth]
2201 & ((integerPart) 1 << bit % integerPartWidth)) != 0;
2202}
2203
2204/* Set the given bit of a bignum. */
2205void
2206APInt::tcSetBit(integerPart *parts, unsigned int bit)
2207{
2208 parts[bit / integerPartWidth] |= (integerPart) 1 << (bit % integerPartWidth);
2209}
2210
Neil Boothc8b650a2007-10-06 00:43:45 +00002211/* Returns the bit number of the least significant set bit of a
2212 number. If the input number has no bits set -1U is returned. */
Chris Lattner6b695682007-08-16 15:56:55 +00002213unsigned int
2214APInt::tcLSB(const integerPart *parts, unsigned int n)
2215{
2216 unsigned int i, lsb;
2217
2218 for(i = 0; i < n; i++) {
2219 if (parts[i] != 0) {
2220 lsb = partLSB(parts[i]);
2221
2222 return lsb + i * integerPartWidth;
2223 }
2224 }
2225
2226 return -1U;
2227}
2228
Neil Boothc8b650a2007-10-06 00:43:45 +00002229/* Returns the bit number of the most significant set bit of a number.
2230 If the input number has no bits set -1U is returned. */
Chris Lattner6b695682007-08-16 15:56:55 +00002231unsigned int
2232APInt::tcMSB(const integerPart *parts, unsigned int n)
2233{
2234 unsigned int msb;
2235
2236 do {
2237 --n;
2238
2239 if (parts[n] != 0) {
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002240 msb = partMSB(parts[n]);
Chris Lattner6b695682007-08-16 15:56:55 +00002241
2242 return msb + n * integerPartWidth;
2243 }
2244 } while (n);
2245
2246 return -1U;
2247}
2248
Neil Boothb6182162007-10-08 13:47:12 +00002249/* Copy the bit vector of width srcBITS from SRC, starting at bit
2250 srcLSB, to DST, of dstCOUNT parts, such that the bit srcLSB becomes
2251 the least significant bit of DST. All high bits above srcBITS in
2252 DST are zero-filled. */
2253void
2254APInt::tcExtract(integerPart *dst, unsigned int dstCount, const integerPart *src,
2255 unsigned int srcBits, unsigned int srcLSB)
2256{
2257 unsigned int firstSrcPart, dstParts, shift, n;
2258
2259 dstParts = (srcBits + integerPartWidth - 1) / integerPartWidth;
2260 assert (dstParts <= dstCount);
2261
2262 firstSrcPart = srcLSB / integerPartWidth;
2263 tcAssign (dst, src + firstSrcPart, dstParts);
2264
2265 shift = srcLSB % integerPartWidth;
2266 tcShiftRight (dst, dstParts, shift);
2267
2268 /* We now have (dstParts * integerPartWidth - shift) bits from SRC
2269 in DST. If this is less that srcBits, append the rest, else
2270 clear the high bits. */
2271 n = dstParts * integerPartWidth - shift;
2272 if (n < srcBits) {
2273 integerPart mask = lowBitMask (srcBits - n);
2274 dst[dstParts - 1] |= ((src[firstSrcPart + dstParts] & mask)
2275 << n % integerPartWidth);
2276 } else if (n > srcBits) {
Neil Booth7e74b172007-10-12 15:31:31 +00002277 if (srcBits % integerPartWidth)
2278 dst[dstParts - 1] &= lowBitMask (srcBits % integerPartWidth);
Neil Boothb6182162007-10-08 13:47:12 +00002279 }
2280
2281 /* Clear high parts. */
2282 while (dstParts < dstCount)
2283 dst[dstParts++] = 0;
2284}
2285
Chris Lattner6b695682007-08-16 15:56:55 +00002286/* DST += RHS + C where C is zero or one. Returns the carry flag. */
2287integerPart
2288APInt::tcAdd(integerPart *dst, const integerPart *rhs,
2289 integerPart c, unsigned int parts)
2290{
2291 unsigned int i;
2292
2293 assert(c <= 1);
2294
2295 for(i = 0; i < parts; i++) {
2296 integerPart l;
2297
2298 l = dst[i];
2299 if (c) {
2300 dst[i] += rhs[i] + 1;
2301 c = (dst[i] <= l);
2302 } else {
2303 dst[i] += rhs[i];
2304 c = (dst[i] < l);
2305 }
2306 }
2307
2308 return c;
2309}
2310
2311/* DST -= RHS + C where C is zero or one. Returns the carry flag. */
2312integerPart
2313APInt::tcSubtract(integerPart *dst, const integerPart *rhs,
2314 integerPart c, unsigned int parts)
2315{
2316 unsigned int i;
2317
2318 assert(c <= 1);
2319
2320 for(i = 0; i < parts; i++) {
2321 integerPart l;
2322
2323 l = dst[i];
2324 if (c) {
2325 dst[i] -= rhs[i] + 1;
2326 c = (dst[i] >= l);
2327 } else {
2328 dst[i] -= rhs[i];
2329 c = (dst[i] > l);
2330 }
2331 }
2332
2333 return c;
2334}
2335
2336/* Negate a bignum in-place. */
2337void
2338APInt::tcNegate(integerPart *dst, unsigned int parts)
2339{
2340 tcComplement(dst, parts);
2341 tcIncrement(dst, parts);
2342}
2343
Neil Boothc8b650a2007-10-06 00:43:45 +00002344/* DST += SRC * MULTIPLIER + CARRY if add is true
2345 DST = SRC * MULTIPLIER + CARRY if add is false
Chris Lattner6b695682007-08-16 15:56:55 +00002346
2347 Requires 0 <= DSTPARTS <= SRCPARTS + 1. If DST overlaps SRC
2348 they must start at the same point, i.e. DST == SRC.
2349
2350 If DSTPARTS == SRCPARTS + 1 no overflow occurs and zero is
2351 returned. Otherwise DST is filled with the least significant
2352 DSTPARTS parts of the result, and if all of the omitted higher
2353 parts were zero return zero, otherwise overflow occurred and
2354 return one. */
2355int
2356APInt::tcMultiplyPart(integerPart *dst, const integerPart *src,
2357 integerPart multiplier, integerPart carry,
2358 unsigned int srcParts, unsigned int dstParts,
2359 bool add)
2360{
2361 unsigned int i, n;
2362
2363 /* Otherwise our writes of DST kill our later reads of SRC. */
2364 assert(dst <= src || dst >= src + srcParts);
2365 assert(dstParts <= srcParts + 1);
2366
2367 /* N loops; minimum of dstParts and srcParts. */
2368 n = dstParts < srcParts ? dstParts: srcParts;
2369
2370 for(i = 0; i < n; i++) {
2371 integerPart low, mid, high, srcPart;
2372
2373 /* [ LOW, HIGH ] = MULTIPLIER * SRC[i] + DST[i] + CARRY.
2374
2375 This cannot overflow, because
2376
2377 (n - 1) * (n - 1) + 2 (n - 1) = (n - 1) * (n + 1)
2378
2379 which is less than n^2. */
2380
2381 srcPart = src[i];
2382
2383 if (multiplier == 0 || srcPart == 0) {
2384 low = carry;
2385 high = 0;
2386 } else {
2387 low = lowHalf(srcPart) * lowHalf(multiplier);
2388 high = highHalf(srcPart) * highHalf(multiplier);
2389
2390 mid = lowHalf(srcPart) * highHalf(multiplier);
2391 high += highHalf(mid);
2392 mid <<= integerPartWidth / 2;
2393 if (low + mid < low)
2394 high++;
2395 low += mid;
2396
2397 mid = highHalf(srcPart) * lowHalf(multiplier);
2398 high += highHalf(mid);
2399 mid <<= integerPartWidth / 2;
2400 if (low + mid < low)
2401 high++;
2402 low += mid;
2403
2404 /* Now add carry. */
2405 if (low + carry < low)
2406 high++;
2407 low += carry;
2408 }
2409
2410 if (add) {
2411 /* And now DST[i], and store the new low part there. */
2412 if (low + dst[i] < low)
2413 high++;
2414 dst[i] += low;
2415 } else
2416 dst[i] = low;
2417
2418 carry = high;
2419 }
2420
2421 if (i < dstParts) {
2422 /* Full multiplication, there is no overflow. */
2423 assert(i + 1 == dstParts);
2424 dst[i] = carry;
2425 return 0;
2426 } else {
2427 /* We overflowed if there is carry. */
2428 if (carry)
2429 return 1;
2430
2431 /* We would overflow if any significant unwritten parts would be
2432 non-zero. This is true if any remaining src parts are non-zero
2433 and the multiplier is non-zero. */
2434 if (multiplier)
2435 for(; i < srcParts; i++)
2436 if (src[i])
2437 return 1;
2438
2439 /* We fitted in the narrow destination. */
2440 return 0;
2441 }
2442}
2443
2444/* DST = LHS * RHS, where DST has the same width as the operands and
2445 is filled with the least significant parts of the result. Returns
2446 one if overflow occurred, otherwise zero. DST must be disjoint
2447 from both operands. */
2448int
2449APInt::tcMultiply(integerPart *dst, const integerPart *lhs,
2450 const integerPart *rhs, unsigned int parts)
2451{
2452 unsigned int i;
2453 int overflow;
2454
2455 assert(dst != lhs && dst != rhs);
2456
2457 overflow = 0;
2458 tcSet(dst, 0, parts);
2459
2460 for(i = 0; i < parts; i++)
2461 overflow |= tcMultiplyPart(&dst[i], lhs, rhs[i], 0, parts,
2462 parts - i, true);
2463
2464 return overflow;
2465}
2466
Neil Booth0ea72a92007-10-06 00:24:48 +00002467/* DST = LHS * RHS, where DST has width the sum of the widths of the
2468 operands. No overflow occurs. DST must be disjoint from both
2469 operands. Returns the number of parts required to hold the
2470 result. */
2471unsigned int
Chris Lattner6b695682007-08-16 15:56:55 +00002472APInt::tcFullMultiply(integerPart *dst, const integerPart *lhs,
Neil Booth0ea72a92007-10-06 00:24:48 +00002473 const integerPart *rhs, unsigned int lhsParts,
2474 unsigned int rhsParts)
Chris Lattner6b695682007-08-16 15:56:55 +00002475{
Neil Booth0ea72a92007-10-06 00:24:48 +00002476 /* Put the narrower number on the LHS for less loops below. */
2477 if (lhsParts > rhsParts) {
2478 return tcFullMultiply (dst, rhs, lhs, rhsParts, lhsParts);
2479 } else {
2480 unsigned int n;
Chris Lattner6b695682007-08-16 15:56:55 +00002481
Neil Booth0ea72a92007-10-06 00:24:48 +00002482 assert(dst != lhs && dst != rhs);
Chris Lattner6b695682007-08-16 15:56:55 +00002483
Neil Booth0ea72a92007-10-06 00:24:48 +00002484 tcSet(dst, 0, rhsParts);
Chris Lattner6b695682007-08-16 15:56:55 +00002485
Neil Booth0ea72a92007-10-06 00:24:48 +00002486 for(n = 0; n < lhsParts; n++)
2487 tcMultiplyPart(&dst[n], rhs, lhs[n], 0, rhsParts, rhsParts + 1, true);
Chris Lattner6b695682007-08-16 15:56:55 +00002488
Neil Booth0ea72a92007-10-06 00:24:48 +00002489 n = lhsParts + rhsParts;
2490
2491 return n - (dst[n - 1] == 0);
2492 }
Chris Lattner6b695682007-08-16 15:56:55 +00002493}
2494
2495/* If RHS is zero LHS and REMAINDER are left unchanged, return one.
2496 Otherwise set LHS to LHS / RHS with the fractional part discarded,
2497 set REMAINDER to the remainder, return zero. i.e.
2498
2499 OLD_LHS = RHS * LHS + REMAINDER
2500
2501 SCRATCH is a bignum of the same size as the operands and result for
2502 use by the routine; its contents need not be initialized and are
2503 destroyed. LHS, REMAINDER and SCRATCH must be distinct.
2504*/
2505int
2506APInt::tcDivide(integerPart *lhs, const integerPart *rhs,
2507 integerPart *remainder, integerPart *srhs,
2508 unsigned int parts)
2509{
2510 unsigned int n, shiftCount;
2511 integerPart mask;
2512
2513 assert(lhs != remainder && lhs != srhs && remainder != srhs);
2514
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002515 shiftCount = tcMSB(rhs, parts) + 1;
2516 if (shiftCount == 0)
Chris Lattner6b695682007-08-16 15:56:55 +00002517 return true;
2518
Chris Lattnerfe02c1f2007-08-20 22:49:32 +00002519 shiftCount = parts * integerPartWidth - shiftCount;
Chris Lattner6b695682007-08-16 15:56:55 +00002520 n = shiftCount / integerPartWidth;
2521 mask = (integerPart) 1 << (shiftCount % integerPartWidth);
2522
2523 tcAssign(srhs, rhs, parts);
2524 tcShiftLeft(srhs, parts, shiftCount);
2525 tcAssign(remainder, lhs, parts);
2526 tcSet(lhs, 0, parts);
2527
2528 /* Loop, subtracting SRHS if REMAINDER is greater and adding that to
2529 the total. */
2530 for(;;) {
2531 int compare;
2532
2533 compare = tcCompare(remainder, srhs, parts);
2534 if (compare >= 0) {
2535 tcSubtract(remainder, srhs, 0, parts);
2536 lhs[n] |= mask;
2537 }
2538
2539 if (shiftCount == 0)
2540 break;
2541 shiftCount--;
2542 tcShiftRight(srhs, parts, 1);
2543 if ((mask >>= 1) == 0)
2544 mask = (integerPart) 1 << (integerPartWidth - 1), n--;
2545 }
2546
2547 return false;
2548}
2549
2550/* Shift a bignum left COUNT bits in-place. Shifted in bits are zero.
2551 There are no restrictions on COUNT. */
2552void
2553APInt::tcShiftLeft(integerPart *dst, unsigned int parts, unsigned int count)
2554{
Neil Boothb6182162007-10-08 13:47:12 +00002555 if (count) {
2556 unsigned int jump, shift;
Chris Lattner6b695682007-08-16 15:56:55 +00002557
Neil Boothb6182162007-10-08 13:47:12 +00002558 /* Jump is the inter-part jump; shift is is intra-part shift. */
2559 jump = count / integerPartWidth;
2560 shift = count % integerPartWidth;
Chris Lattner6b695682007-08-16 15:56:55 +00002561
Neil Boothb6182162007-10-08 13:47:12 +00002562 while (parts > jump) {
2563 integerPart part;
Chris Lattner6b695682007-08-16 15:56:55 +00002564
Neil Boothb6182162007-10-08 13:47:12 +00002565 parts--;
Chris Lattner6b695682007-08-16 15:56:55 +00002566
Neil Boothb6182162007-10-08 13:47:12 +00002567 /* dst[i] comes from the two parts src[i - jump] and, if we have
2568 an intra-part shift, src[i - jump - 1]. */
2569 part = dst[parts - jump];
2570 if (shift) {
2571 part <<= shift;
Chris Lattner6b695682007-08-16 15:56:55 +00002572 if (parts >= jump + 1)
2573 part |= dst[parts - jump - 1] >> (integerPartWidth - shift);
2574 }
2575
Neil Boothb6182162007-10-08 13:47:12 +00002576 dst[parts] = part;
2577 }
Chris Lattner6b695682007-08-16 15:56:55 +00002578
Neil Boothb6182162007-10-08 13:47:12 +00002579 while (parts > 0)
2580 dst[--parts] = 0;
2581 }
Chris Lattner6b695682007-08-16 15:56:55 +00002582}
2583
2584/* Shift a bignum right COUNT bits in-place. Shifted in bits are
2585 zero. There are no restrictions on COUNT. */
2586void
2587APInt::tcShiftRight(integerPart *dst, unsigned int parts, unsigned int count)
2588{
Neil Boothb6182162007-10-08 13:47:12 +00002589 if (count) {
2590 unsigned int i, jump, shift;
Chris Lattner6b695682007-08-16 15:56:55 +00002591
Neil Boothb6182162007-10-08 13:47:12 +00002592 /* Jump is the inter-part jump; shift is is intra-part shift. */
2593 jump = count / integerPartWidth;
2594 shift = count % integerPartWidth;
Chris Lattner6b695682007-08-16 15:56:55 +00002595
Neil Boothb6182162007-10-08 13:47:12 +00002596 /* Perform the shift. This leaves the most significant COUNT bits
2597 of the result at zero. */
2598 for(i = 0; i < parts; i++) {
2599 integerPart part;
Chris Lattner6b695682007-08-16 15:56:55 +00002600
Neil Boothb6182162007-10-08 13:47:12 +00002601 if (i + jump >= parts) {
2602 part = 0;
2603 } else {
2604 part = dst[i + jump];
2605 if (shift) {
2606 part >>= shift;
2607 if (i + jump + 1 < parts)
2608 part |= dst[i + jump + 1] << (integerPartWidth - shift);
2609 }
Chris Lattner6b695682007-08-16 15:56:55 +00002610 }
Chris Lattner6b695682007-08-16 15:56:55 +00002611
Neil Boothb6182162007-10-08 13:47:12 +00002612 dst[i] = part;
2613 }
Chris Lattner6b695682007-08-16 15:56:55 +00002614 }
2615}
2616
2617/* Bitwise and of two bignums. */
2618void
2619APInt::tcAnd(integerPart *dst, const integerPart *rhs, unsigned int parts)
2620{
2621 unsigned int i;
2622
2623 for(i = 0; i < parts; i++)
2624 dst[i] &= rhs[i];
2625}
2626
2627/* Bitwise inclusive or of two bignums. */
2628void
2629APInt::tcOr(integerPart *dst, const integerPart *rhs, unsigned int parts)
2630{
2631 unsigned int i;
2632
2633 for(i = 0; i < parts; i++)
2634 dst[i] |= rhs[i];
2635}
2636
2637/* Bitwise exclusive or of two bignums. */
2638void
2639APInt::tcXor(integerPart *dst, const integerPart *rhs, unsigned int parts)
2640{
2641 unsigned int i;
2642
2643 for(i = 0; i < parts; i++)
2644 dst[i] ^= rhs[i];
2645}
2646
2647/* Complement a bignum in-place. */
2648void
2649APInt::tcComplement(integerPart *dst, unsigned int parts)
2650{
2651 unsigned int i;
2652
2653 for(i = 0; i < parts; i++)
2654 dst[i] = ~dst[i];
2655}
2656
2657/* Comparison (unsigned) of two bignums. */
2658int
2659APInt::tcCompare(const integerPart *lhs, const integerPart *rhs,
2660 unsigned int parts)
2661{
2662 while (parts) {
2663 parts--;
2664 if (lhs[parts] == rhs[parts])
2665 continue;
2666
2667 if (lhs[parts] > rhs[parts])
2668 return 1;
2669 else
2670 return -1;
2671 }
2672
2673 return 0;
2674}
2675
2676/* Increment a bignum in-place, return the carry flag. */
2677integerPart
2678APInt::tcIncrement(integerPart *dst, unsigned int parts)
2679{
2680 unsigned int i;
2681
2682 for(i = 0; i < parts; i++)
2683 if (++dst[i] != 0)
2684 break;
2685
2686 return i == parts;
2687}
2688
2689/* Set the least significant BITS bits of a bignum, clear the
2690 rest. */
2691void
2692APInt::tcSetLeastSignificantBits(integerPart *dst, unsigned int parts,
2693 unsigned int bits)
2694{
2695 unsigned int i;
2696
2697 i = 0;
2698 while (bits > integerPartWidth) {
2699 dst[i++] = ~(integerPart) 0;
2700 bits -= integerPartWidth;
2701 }
2702
2703 if (bits)
2704 dst[i++] = ~(integerPart) 0 >> (integerPartWidth - bits);
2705
2706 while (i < parts)
2707 dst[i++] = 0;
2708}