1*9880d681SAndroid Build Coastguard Worker //===-- APInt.cpp - Implement APInt class ---------------------------------===//
2*9880d681SAndroid Build Coastguard Worker //
3*9880d681SAndroid Build Coastguard Worker // The LLVM Compiler Infrastructure
4*9880d681SAndroid Build Coastguard Worker //
5*9880d681SAndroid Build Coastguard Worker // This file is distributed under the University of Illinois Open Source
6*9880d681SAndroid Build Coastguard Worker // License. See LICENSE.TXT for details.
7*9880d681SAndroid Build Coastguard Worker //
8*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
9*9880d681SAndroid Build Coastguard Worker //
10*9880d681SAndroid Build Coastguard Worker // This file implements a class to represent arbitrary precision integer
11*9880d681SAndroid Build Coastguard Worker // constant values and provide a variety of arithmetic operations on them.
12*9880d681SAndroid Build Coastguard Worker //
13*9880d681SAndroid Build Coastguard Worker //===----------------------------------------------------------------------===//
14*9880d681SAndroid Build Coastguard Worker
15*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/APInt.h"
16*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/ArrayRef.h"
17*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/FoldingSet.h"
18*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/Hashing.h"
19*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/SmallString.h"
20*9880d681SAndroid Build Coastguard Worker #include "llvm/ADT/StringRef.h"
21*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/Debug.h"
22*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/ErrorHandling.h"
23*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/MathExtras.h"
24*9880d681SAndroid Build Coastguard Worker #include "llvm/Support/raw_ostream.h"
25*9880d681SAndroid Build Coastguard Worker #include <cmath>
26*9880d681SAndroid Build Coastguard Worker #include <cstdlib>
27*9880d681SAndroid Build Coastguard Worker #include <cstring>
28*9880d681SAndroid Build Coastguard Worker #include <limits>
29*9880d681SAndroid Build Coastguard Worker using namespace llvm;
30*9880d681SAndroid Build Coastguard Worker
31*9880d681SAndroid Build Coastguard Worker #define DEBUG_TYPE "apint"
32*9880d681SAndroid Build Coastguard Worker
33*9880d681SAndroid Build Coastguard Worker /// A utility function for allocating memory, checking for allocation failures,
34*9880d681SAndroid Build Coastguard Worker /// and ensuring the contents are zeroed.
getClearedMemory(unsigned numWords)35*9880d681SAndroid Build Coastguard Worker inline static uint64_t* getClearedMemory(unsigned numWords) {
36*9880d681SAndroid Build Coastguard Worker uint64_t * result = new uint64_t[numWords];
37*9880d681SAndroid Build Coastguard Worker assert(result && "APInt memory allocation fails!");
38*9880d681SAndroid Build Coastguard Worker memset(result, 0, numWords * sizeof(uint64_t));
39*9880d681SAndroid Build Coastguard Worker return result;
40*9880d681SAndroid Build Coastguard Worker }
41*9880d681SAndroid Build Coastguard Worker
42*9880d681SAndroid Build Coastguard Worker /// A utility function for allocating memory and checking for allocation
43*9880d681SAndroid Build Coastguard Worker /// failure. The content is not zeroed.
getMemory(unsigned numWords)44*9880d681SAndroid Build Coastguard Worker inline static uint64_t* getMemory(unsigned numWords) {
45*9880d681SAndroid Build Coastguard Worker uint64_t * result = new uint64_t[numWords];
46*9880d681SAndroid Build Coastguard Worker assert(result && "APInt memory allocation fails!");
47*9880d681SAndroid Build Coastguard Worker return result;
48*9880d681SAndroid Build Coastguard Worker }
49*9880d681SAndroid Build Coastguard Worker
50*9880d681SAndroid Build Coastguard Worker /// A utility function that converts a character to a digit.
getDigit(char cdigit,uint8_t radix)51*9880d681SAndroid Build Coastguard Worker inline static unsigned getDigit(char cdigit, uint8_t radix) {
52*9880d681SAndroid Build Coastguard Worker unsigned r;
53*9880d681SAndroid Build Coastguard Worker
54*9880d681SAndroid Build Coastguard Worker if (radix == 16 || radix == 36) {
55*9880d681SAndroid Build Coastguard Worker r = cdigit - '0';
56*9880d681SAndroid Build Coastguard Worker if (r <= 9)
57*9880d681SAndroid Build Coastguard Worker return r;
58*9880d681SAndroid Build Coastguard Worker
59*9880d681SAndroid Build Coastguard Worker r = cdigit - 'A';
60*9880d681SAndroid Build Coastguard Worker if (r <= radix - 11U)
61*9880d681SAndroid Build Coastguard Worker return r + 10;
62*9880d681SAndroid Build Coastguard Worker
63*9880d681SAndroid Build Coastguard Worker r = cdigit - 'a';
64*9880d681SAndroid Build Coastguard Worker if (r <= radix - 11U)
65*9880d681SAndroid Build Coastguard Worker return r + 10;
66*9880d681SAndroid Build Coastguard Worker
67*9880d681SAndroid Build Coastguard Worker radix = 10;
68*9880d681SAndroid Build Coastguard Worker }
69*9880d681SAndroid Build Coastguard Worker
70*9880d681SAndroid Build Coastguard Worker r = cdigit - '0';
71*9880d681SAndroid Build Coastguard Worker if (r < radix)
72*9880d681SAndroid Build Coastguard Worker return r;
73*9880d681SAndroid Build Coastguard Worker
74*9880d681SAndroid Build Coastguard Worker return -1U;
75*9880d681SAndroid Build Coastguard Worker }
76*9880d681SAndroid Build Coastguard Worker
77*9880d681SAndroid Build Coastguard Worker
initSlowCase(uint64_t val,bool isSigned)78*9880d681SAndroid Build Coastguard Worker void APInt::initSlowCase(uint64_t val, bool isSigned) {
79*9880d681SAndroid Build Coastguard Worker pVal = getClearedMemory(getNumWords());
80*9880d681SAndroid Build Coastguard Worker pVal[0] = val;
81*9880d681SAndroid Build Coastguard Worker if (isSigned && int64_t(val) < 0)
82*9880d681SAndroid Build Coastguard Worker for (unsigned i = 1; i < getNumWords(); ++i)
83*9880d681SAndroid Build Coastguard Worker pVal[i] = -1ULL;
84*9880d681SAndroid Build Coastguard Worker }
85*9880d681SAndroid Build Coastguard Worker
initSlowCase(const APInt & that)86*9880d681SAndroid Build Coastguard Worker void APInt::initSlowCase(const APInt& that) {
87*9880d681SAndroid Build Coastguard Worker pVal = getMemory(getNumWords());
88*9880d681SAndroid Build Coastguard Worker memcpy(pVal, that.pVal, getNumWords() * APINT_WORD_SIZE);
89*9880d681SAndroid Build Coastguard Worker }
90*9880d681SAndroid Build Coastguard Worker
initFromArray(ArrayRef<uint64_t> bigVal)91*9880d681SAndroid Build Coastguard Worker void APInt::initFromArray(ArrayRef<uint64_t> bigVal) {
92*9880d681SAndroid Build Coastguard Worker assert(BitWidth && "Bitwidth too small");
93*9880d681SAndroid Build Coastguard Worker assert(bigVal.data() && "Null pointer detected!");
94*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
95*9880d681SAndroid Build Coastguard Worker VAL = bigVal[0];
96*9880d681SAndroid Build Coastguard Worker else {
97*9880d681SAndroid Build Coastguard Worker // Get memory, cleared to 0
98*9880d681SAndroid Build Coastguard Worker pVal = getClearedMemory(getNumWords());
99*9880d681SAndroid Build Coastguard Worker // Calculate the number of words to copy
100*9880d681SAndroid Build Coastguard Worker unsigned words = std::min<unsigned>(bigVal.size(), getNumWords());
101*9880d681SAndroid Build Coastguard Worker // Copy the words from bigVal to pVal
102*9880d681SAndroid Build Coastguard Worker memcpy(pVal, bigVal.data(), words * APINT_WORD_SIZE);
103*9880d681SAndroid Build Coastguard Worker }
104*9880d681SAndroid Build Coastguard Worker // Make sure unused high bits are cleared
105*9880d681SAndroid Build Coastguard Worker clearUnusedBits();
106*9880d681SAndroid Build Coastguard Worker }
107*9880d681SAndroid Build Coastguard Worker
APInt(unsigned numBits,ArrayRef<uint64_t> bigVal)108*9880d681SAndroid Build Coastguard Worker APInt::APInt(unsigned numBits, ArrayRef<uint64_t> bigVal)
109*9880d681SAndroid Build Coastguard Worker : BitWidth(numBits), VAL(0) {
110*9880d681SAndroid Build Coastguard Worker initFromArray(bigVal);
111*9880d681SAndroid Build Coastguard Worker }
112*9880d681SAndroid Build Coastguard Worker
APInt(unsigned numBits,unsigned numWords,const uint64_t bigVal[])113*9880d681SAndroid Build Coastguard Worker APInt::APInt(unsigned numBits, unsigned numWords, const uint64_t bigVal[])
114*9880d681SAndroid Build Coastguard Worker : BitWidth(numBits), VAL(0) {
115*9880d681SAndroid Build Coastguard Worker initFromArray(makeArrayRef(bigVal, numWords));
116*9880d681SAndroid Build Coastguard Worker }
117*9880d681SAndroid Build Coastguard Worker
APInt(unsigned numbits,StringRef Str,uint8_t radix)118*9880d681SAndroid Build Coastguard Worker APInt::APInt(unsigned numbits, StringRef Str, uint8_t radix)
119*9880d681SAndroid Build Coastguard Worker : BitWidth(numbits), VAL(0) {
120*9880d681SAndroid Build Coastguard Worker assert(BitWidth && "Bitwidth too small");
121*9880d681SAndroid Build Coastguard Worker fromString(numbits, Str, radix);
122*9880d681SAndroid Build Coastguard Worker }
123*9880d681SAndroid Build Coastguard Worker
AssignSlowCase(const APInt & RHS)124*9880d681SAndroid Build Coastguard Worker APInt& APInt::AssignSlowCase(const APInt& RHS) {
125*9880d681SAndroid Build Coastguard Worker // Don't do anything for X = X
126*9880d681SAndroid Build Coastguard Worker if (this == &RHS)
127*9880d681SAndroid Build Coastguard Worker return *this;
128*9880d681SAndroid Build Coastguard Worker
129*9880d681SAndroid Build Coastguard Worker if (BitWidth == RHS.getBitWidth()) {
130*9880d681SAndroid Build Coastguard Worker // assume same bit-width single-word case is already handled
131*9880d681SAndroid Build Coastguard Worker assert(!isSingleWord());
132*9880d681SAndroid Build Coastguard Worker memcpy(pVal, RHS.pVal, getNumWords() * APINT_WORD_SIZE);
133*9880d681SAndroid Build Coastguard Worker return *this;
134*9880d681SAndroid Build Coastguard Worker }
135*9880d681SAndroid Build Coastguard Worker
136*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
137*9880d681SAndroid Build Coastguard Worker // assume case where both are single words is already handled
138*9880d681SAndroid Build Coastguard Worker assert(!RHS.isSingleWord());
139*9880d681SAndroid Build Coastguard Worker VAL = 0;
140*9880d681SAndroid Build Coastguard Worker pVal = getMemory(RHS.getNumWords());
141*9880d681SAndroid Build Coastguard Worker memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
142*9880d681SAndroid Build Coastguard Worker } else if (getNumWords() == RHS.getNumWords())
143*9880d681SAndroid Build Coastguard Worker memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
144*9880d681SAndroid Build Coastguard Worker else if (RHS.isSingleWord()) {
145*9880d681SAndroid Build Coastguard Worker delete [] pVal;
146*9880d681SAndroid Build Coastguard Worker VAL = RHS.VAL;
147*9880d681SAndroid Build Coastguard Worker } else {
148*9880d681SAndroid Build Coastguard Worker delete [] pVal;
149*9880d681SAndroid Build Coastguard Worker pVal = getMemory(RHS.getNumWords());
150*9880d681SAndroid Build Coastguard Worker memcpy(pVal, RHS.pVal, RHS.getNumWords() * APINT_WORD_SIZE);
151*9880d681SAndroid Build Coastguard Worker }
152*9880d681SAndroid Build Coastguard Worker BitWidth = RHS.BitWidth;
153*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
154*9880d681SAndroid Build Coastguard Worker }
155*9880d681SAndroid Build Coastguard Worker
operator =(uint64_t RHS)156*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator=(uint64_t RHS) {
157*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
158*9880d681SAndroid Build Coastguard Worker VAL = RHS;
159*9880d681SAndroid Build Coastguard Worker else {
160*9880d681SAndroid Build Coastguard Worker pVal[0] = RHS;
161*9880d681SAndroid Build Coastguard Worker memset(pVal+1, 0, (getNumWords() - 1) * APINT_WORD_SIZE);
162*9880d681SAndroid Build Coastguard Worker }
163*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
164*9880d681SAndroid Build Coastguard Worker }
165*9880d681SAndroid Build Coastguard Worker
166*9880d681SAndroid Build Coastguard Worker /// This method 'profiles' an APInt for use with FoldingSet.
Profile(FoldingSetNodeID & ID) const167*9880d681SAndroid Build Coastguard Worker void APInt::Profile(FoldingSetNodeID& ID) const {
168*9880d681SAndroid Build Coastguard Worker ID.AddInteger(BitWidth);
169*9880d681SAndroid Build Coastguard Worker
170*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
171*9880d681SAndroid Build Coastguard Worker ID.AddInteger(VAL);
172*9880d681SAndroid Build Coastguard Worker return;
173*9880d681SAndroid Build Coastguard Worker }
174*9880d681SAndroid Build Coastguard Worker
175*9880d681SAndroid Build Coastguard Worker unsigned NumWords = getNumWords();
176*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < NumWords; ++i)
177*9880d681SAndroid Build Coastguard Worker ID.AddInteger(pVal[i]);
178*9880d681SAndroid Build Coastguard Worker }
179*9880d681SAndroid Build Coastguard Worker
180*9880d681SAndroid Build Coastguard Worker /// This function adds a single "digit" integer, y, to the multiple
181*9880d681SAndroid Build Coastguard Worker /// "digit" integer array, x[]. x[] is modified to reflect the addition and
182*9880d681SAndroid Build Coastguard Worker /// 1 is returned if there is a carry out, otherwise 0 is returned.
183*9880d681SAndroid Build Coastguard Worker /// @returns the carry of the addition.
add_1(uint64_t dest[],uint64_t x[],unsigned len,uint64_t y)184*9880d681SAndroid Build Coastguard Worker static bool add_1(uint64_t dest[], uint64_t x[], unsigned len, uint64_t y) {
185*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < len; ++i) {
186*9880d681SAndroid Build Coastguard Worker dest[i] = y + x[i];
187*9880d681SAndroid Build Coastguard Worker if (dest[i] < y)
188*9880d681SAndroid Build Coastguard Worker y = 1; // Carry one to next digit.
189*9880d681SAndroid Build Coastguard Worker else {
190*9880d681SAndroid Build Coastguard Worker y = 0; // No need to carry so exit early
191*9880d681SAndroid Build Coastguard Worker break;
192*9880d681SAndroid Build Coastguard Worker }
193*9880d681SAndroid Build Coastguard Worker }
194*9880d681SAndroid Build Coastguard Worker return y;
195*9880d681SAndroid Build Coastguard Worker }
196*9880d681SAndroid Build Coastguard Worker
197*9880d681SAndroid Build Coastguard Worker /// @brief Prefix increment operator. Increments the APInt by one.
operator ++()198*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator++() {
199*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
200*9880d681SAndroid Build Coastguard Worker ++VAL;
201*9880d681SAndroid Build Coastguard Worker else
202*9880d681SAndroid Build Coastguard Worker add_1(pVal, pVal, getNumWords(), 1);
203*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
204*9880d681SAndroid Build Coastguard Worker }
205*9880d681SAndroid Build Coastguard Worker
206*9880d681SAndroid Build Coastguard Worker /// This function subtracts a single "digit" (64-bit word), y, from
207*9880d681SAndroid Build Coastguard Worker /// the multi-digit integer array, x[], propagating the borrowed 1 value until
208*9880d681SAndroid Build Coastguard Worker /// no further borrowing is neeeded or it runs out of "digits" in x. The result
209*9880d681SAndroid Build Coastguard Worker /// is 1 if "borrowing" exhausted the digits in x, or 0 if x was not exhausted.
210*9880d681SAndroid Build Coastguard Worker /// In other words, if y > x then this function returns 1, otherwise 0.
211*9880d681SAndroid Build Coastguard Worker /// @returns the borrow out of the subtraction
sub_1(uint64_t x[],unsigned len,uint64_t y)212*9880d681SAndroid Build Coastguard Worker static bool sub_1(uint64_t x[], unsigned len, uint64_t y) {
213*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < len; ++i) {
214*9880d681SAndroid Build Coastguard Worker uint64_t X = x[i];
215*9880d681SAndroid Build Coastguard Worker x[i] -= y;
216*9880d681SAndroid Build Coastguard Worker if (y > X)
217*9880d681SAndroid Build Coastguard Worker y = 1; // We have to "borrow 1" from next "digit"
218*9880d681SAndroid Build Coastguard Worker else {
219*9880d681SAndroid Build Coastguard Worker y = 0; // No need to borrow
220*9880d681SAndroid Build Coastguard Worker break; // Remaining digits are unchanged so exit early
221*9880d681SAndroid Build Coastguard Worker }
222*9880d681SAndroid Build Coastguard Worker }
223*9880d681SAndroid Build Coastguard Worker return bool(y);
224*9880d681SAndroid Build Coastguard Worker }
225*9880d681SAndroid Build Coastguard Worker
226*9880d681SAndroid Build Coastguard Worker /// @brief Prefix decrement operator. Decrements the APInt by one.
operator --()227*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator--() {
228*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
229*9880d681SAndroid Build Coastguard Worker --VAL;
230*9880d681SAndroid Build Coastguard Worker else
231*9880d681SAndroid Build Coastguard Worker sub_1(pVal, getNumWords(), 1);
232*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
233*9880d681SAndroid Build Coastguard Worker }
234*9880d681SAndroid Build Coastguard Worker
235*9880d681SAndroid Build Coastguard Worker /// This function adds the integer array x to the integer array Y and
236*9880d681SAndroid Build Coastguard Worker /// places the result in dest.
237*9880d681SAndroid Build Coastguard Worker /// @returns the carry out from the addition
238*9880d681SAndroid Build Coastguard Worker /// @brief General addition of 64-bit integer arrays
add(uint64_t * dest,const uint64_t * x,const uint64_t * y,unsigned len)239*9880d681SAndroid Build Coastguard Worker static bool add(uint64_t *dest, const uint64_t *x, const uint64_t *y,
240*9880d681SAndroid Build Coastguard Worker unsigned len) {
241*9880d681SAndroid Build Coastguard Worker bool carry = false;
242*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i< len; ++i) {
243*9880d681SAndroid Build Coastguard Worker uint64_t limit = std::min(x[i],y[i]); // must come first in case dest == x
244*9880d681SAndroid Build Coastguard Worker dest[i] = x[i] + y[i] + carry;
245*9880d681SAndroid Build Coastguard Worker carry = dest[i] < limit || (carry && dest[i] == limit);
246*9880d681SAndroid Build Coastguard Worker }
247*9880d681SAndroid Build Coastguard Worker return carry;
248*9880d681SAndroid Build Coastguard Worker }
249*9880d681SAndroid Build Coastguard Worker
250*9880d681SAndroid Build Coastguard Worker /// Adds the RHS APint to this APInt.
251*9880d681SAndroid Build Coastguard Worker /// @returns this, after addition of RHS.
252*9880d681SAndroid Build Coastguard Worker /// @brief Addition assignment operator.
operator +=(const APInt & RHS)253*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator+=(const APInt& RHS) {
254*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
255*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
256*9880d681SAndroid Build Coastguard Worker VAL += RHS.VAL;
257*9880d681SAndroid Build Coastguard Worker else {
258*9880d681SAndroid Build Coastguard Worker add(pVal, pVal, RHS.pVal, getNumWords());
259*9880d681SAndroid Build Coastguard Worker }
260*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
261*9880d681SAndroid Build Coastguard Worker }
262*9880d681SAndroid Build Coastguard Worker
263*9880d681SAndroid Build Coastguard Worker /// Subtracts the integer array y from the integer array x
264*9880d681SAndroid Build Coastguard Worker /// @returns returns the borrow out.
265*9880d681SAndroid Build Coastguard Worker /// @brief Generalized subtraction of 64-bit integer arrays.
sub(uint64_t * dest,const uint64_t * x,const uint64_t * y,unsigned len)266*9880d681SAndroid Build Coastguard Worker static bool sub(uint64_t *dest, const uint64_t *x, const uint64_t *y,
267*9880d681SAndroid Build Coastguard Worker unsigned len) {
268*9880d681SAndroid Build Coastguard Worker bool borrow = false;
269*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < len; ++i) {
270*9880d681SAndroid Build Coastguard Worker uint64_t x_tmp = borrow ? x[i] - 1 : x[i];
271*9880d681SAndroid Build Coastguard Worker borrow = y[i] > x_tmp || (borrow && x[i] == 0);
272*9880d681SAndroid Build Coastguard Worker dest[i] = x_tmp - y[i];
273*9880d681SAndroid Build Coastguard Worker }
274*9880d681SAndroid Build Coastguard Worker return borrow;
275*9880d681SAndroid Build Coastguard Worker }
276*9880d681SAndroid Build Coastguard Worker
277*9880d681SAndroid Build Coastguard Worker /// Subtracts the RHS APInt from this APInt
278*9880d681SAndroid Build Coastguard Worker /// @returns this, after subtraction
279*9880d681SAndroid Build Coastguard Worker /// @brief Subtraction assignment operator.
operator -=(const APInt & RHS)280*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator-=(const APInt& RHS) {
281*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
282*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
283*9880d681SAndroid Build Coastguard Worker VAL -= RHS.VAL;
284*9880d681SAndroid Build Coastguard Worker else
285*9880d681SAndroid Build Coastguard Worker sub(pVal, pVal, RHS.pVal, getNumWords());
286*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
287*9880d681SAndroid Build Coastguard Worker }
288*9880d681SAndroid Build Coastguard Worker
289*9880d681SAndroid Build Coastguard Worker /// Multiplies an integer array, x, by a uint64_t integer and places the result
290*9880d681SAndroid Build Coastguard Worker /// into dest.
291*9880d681SAndroid Build Coastguard Worker /// @returns the carry out of the multiplication.
292*9880d681SAndroid Build Coastguard Worker /// @brief Multiply a multi-digit APInt by a single digit (64-bit) integer.
mul_1(uint64_t dest[],uint64_t x[],unsigned len,uint64_t y)293*9880d681SAndroid Build Coastguard Worker static uint64_t mul_1(uint64_t dest[], uint64_t x[], unsigned len, uint64_t y) {
294*9880d681SAndroid Build Coastguard Worker // Split y into high 32-bit part (hy) and low 32-bit part (ly)
295*9880d681SAndroid Build Coastguard Worker uint64_t ly = y & 0xffffffffULL, hy = y >> 32;
296*9880d681SAndroid Build Coastguard Worker uint64_t carry = 0;
297*9880d681SAndroid Build Coastguard Worker
298*9880d681SAndroid Build Coastguard Worker // For each digit of x.
299*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < len; ++i) {
300*9880d681SAndroid Build Coastguard Worker // Split x into high and low words
301*9880d681SAndroid Build Coastguard Worker uint64_t lx = x[i] & 0xffffffffULL;
302*9880d681SAndroid Build Coastguard Worker uint64_t hx = x[i] >> 32;
303*9880d681SAndroid Build Coastguard Worker // hasCarry - A flag to indicate if there is a carry to the next digit.
304*9880d681SAndroid Build Coastguard Worker // hasCarry == 0, no carry
305*9880d681SAndroid Build Coastguard Worker // hasCarry == 1, has carry
306*9880d681SAndroid Build Coastguard Worker // hasCarry == 2, no carry and the calculation result == 0.
307*9880d681SAndroid Build Coastguard Worker uint8_t hasCarry = 0;
308*9880d681SAndroid Build Coastguard Worker dest[i] = carry + lx * ly;
309*9880d681SAndroid Build Coastguard Worker // Determine if the add above introduces carry.
310*9880d681SAndroid Build Coastguard Worker hasCarry = (dest[i] < carry) ? 1 : 0;
311*9880d681SAndroid Build Coastguard Worker carry = hx * ly + (dest[i] >> 32) + (hasCarry ? (1ULL << 32) : 0);
312*9880d681SAndroid Build Coastguard Worker // The upper limit of carry can be (2^32 - 1)(2^32 - 1) +
313*9880d681SAndroid Build Coastguard Worker // (2^32 - 1) + 2^32 = 2^64.
314*9880d681SAndroid Build Coastguard Worker hasCarry = (!carry && hasCarry) ? 1 : (!carry ? 2 : 0);
315*9880d681SAndroid Build Coastguard Worker
316*9880d681SAndroid Build Coastguard Worker carry += (lx * hy) & 0xffffffffULL;
317*9880d681SAndroid Build Coastguard Worker dest[i] = (carry << 32) | (dest[i] & 0xffffffffULL);
318*9880d681SAndroid Build Coastguard Worker carry = (((!carry && hasCarry != 2) || hasCarry == 1) ? (1ULL << 32) : 0) +
319*9880d681SAndroid Build Coastguard Worker (carry >> 32) + ((lx * hy) >> 32) + hx * hy;
320*9880d681SAndroid Build Coastguard Worker }
321*9880d681SAndroid Build Coastguard Worker return carry;
322*9880d681SAndroid Build Coastguard Worker }
323*9880d681SAndroid Build Coastguard Worker
324*9880d681SAndroid Build Coastguard Worker /// Multiplies integer array x by integer array y and stores the result into
325*9880d681SAndroid Build Coastguard Worker /// the integer array dest. Note that dest's size must be >= xlen + ylen.
326*9880d681SAndroid Build Coastguard Worker /// @brief Generalized multiplicate of integer arrays.
mul(uint64_t dest[],uint64_t x[],unsigned xlen,uint64_t y[],unsigned ylen)327*9880d681SAndroid Build Coastguard Worker static void mul(uint64_t dest[], uint64_t x[], unsigned xlen, uint64_t y[],
328*9880d681SAndroid Build Coastguard Worker unsigned ylen) {
329*9880d681SAndroid Build Coastguard Worker dest[xlen] = mul_1(dest, x, xlen, y[0]);
330*9880d681SAndroid Build Coastguard Worker for (unsigned i = 1; i < ylen; ++i) {
331*9880d681SAndroid Build Coastguard Worker uint64_t ly = y[i] & 0xffffffffULL, hy = y[i] >> 32;
332*9880d681SAndroid Build Coastguard Worker uint64_t carry = 0, lx = 0, hx = 0;
333*9880d681SAndroid Build Coastguard Worker for (unsigned j = 0; j < xlen; ++j) {
334*9880d681SAndroid Build Coastguard Worker lx = x[j] & 0xffffffffULL;
335*9880d681SAndroid Build Coastguard Worker hx = x[j] >> 32;
336*9880d681SAndroid Build Coastguard Worker // hasCarry - A flag to indicate if has carry.
337*9880d681SAndroid Build Coastguard Worker // hasCarry == 0, no carry
338*9880d681SAndroid Build Coastguard Worker // hasCarry == 1, has carry
339*9880d681SAndroid Build Coastguard Worker // hasCarry == 2, no carry and the calculation result == 0.
340*9880d681SAndroid Build Coastguard Worker uint8_t hasCarry = 0;
341*9880d681SAndroid Build Coastguard Worker uint64_t resul = carry + lx * ly;
342*9880d681SAndroid Build Coastguard Worker hasCarry = (resul < carry) ? 1 : 0;
343*9880d681SAndroid Build Coastguard Worker carry = (hasCarry ? (1ULL << 32) : 0) + hx * ly + (resul >> 32);
344*9880d681SAndroid Build Coastguard Worker hasCarry = (!carry && hasCarry) ? 1 : (!carry ? 2 : 0);
345*9880d681SAndroid Build Coastguard Worker
346*9880d681SAndroid Build Coastguard Worker carry += (lx * hy) & 0xffffffffULL;
347*9880d681SAndroid Build Coastguard Worker resul = (carry << 32) | (resul & 0xffffffffULL);
348*9880d681SAndroid Build Coastguard Worker dest[i+j] += resul;
349*9880d681SAndroid Build Coastguard Worker carry = (((!carry && hasCarry != 2) || hasCarry == 1) ? (1ULL << 32) : 0)+
350*9880d681SAndroid Build Coastguard Worker (carry >> 32) + (dest[i+j] < resul ? 1 : 0) +
351*9880d681SAndroid Build Coastguard Worker ((lx * hy) >> 32) + hx * hy;
352*9880d681SAndroid Build Coastguard Worker }
353*9880d681SAndroid Build Coastguard Worker dest[i+xlen] = carry;
354*9880d681SAndroid Build Coastguard Worker }
355*9880d681SAndroid Build Coastguard Worker }
356*9880d681SAndroid Build Coastguard Worker
operator *=(const APInt & RHS)357*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator*=(const APInt& RHS) {
358*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
359*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
360*9880d681SAndroid Build Coastguard Worker VAL *= RHS.VAL;
361*9880d681SAndroid Build Coastguard Worker clearUnusedBits();
362*9880d681SAndroid Build Coastguard Worker return *this;
363*9880d681SAndroid Build Coastguard Worker }
364*9880d681SAndroid Build Coastguard Worker
365*9880d681SAndroid Build Coastguard Worker // Get some bit facts about LHS and check for zero
366*9880d681SAndroid Build Coastguard Worker unsigned lhsBits = getActiveBits();
367*9880d681SAndroid Build Coastguard Worker unsigned lhsWords = !lhsBits ? 0 : whichWord(lhsBits - 1) + 1;
368*9880d681SAndroid Build Coastguard Worker if (!lhsWords)
369*9880d681SAndroid Build Coastguard Worker // 0 * X ===> 0
370*9880d681SAndroid Build Coastguard Worker return *this;
371*9880d681SAndroid Build Coastguard Worker
372*9880d681SAndroid Build Coastguard Worker // Get some bit facts about RHS and check for zero
373*9880d681SAndroid Build Coastguard Worker unsigned rhsBits = RHS.getActiveBits();
374*9880d681SAndroid Build Coastguard Worker unsigned rhsWords = !rhsBits ? 0 : whichWord(rhsBits - 1) + 1;
375*9880d681SAndroid Build Coastguard Worker if (!rhsWords) {
376*9880d681SAndroid Build Coastguard Worker // X * 0 ===> 0
377*9880d681SAndroid Build Coastguard Worker clearAllBits();
378*9880d681SAndroid Build Coastguard Worker return *this;
379*9880d681SAndroid Build Coastguard Worker }
380*9880d681SAndroid Build Coastguard Worker
381*9880d681SAndroid Build Coastguard Worker // Allocate space for the result
382*9880d681SAndroid Build Coastguard Worker unsigned destWords = rhsWords + lhsWords;
383*9880d681SAndroid Build Coastguard Worker uint64_t *dest = getMemory(destWords);
384*9880d681SAndroid Build Coastguard Worker
385*9880d681SAndroid Build Coastguard Worker // Perform the long multiply
386*9880d681SAndroid Build Coastguard Worker mul(dest, pVal, lhsWords, RHS.pVal, rhsWords);
387*9880d681SAndroid Build Coastguard Worker
388*9880d681SAndroid Build Coastguard Worker // Copy result back into *this
389*9880d681SAndroid Build Coastguard Worker clearAllBits();
390*9880d681SAndroid Build Coastguard Worker unsigned wordsToCopy = destWords >= getNumWords() ? getNumWords() : destWords;
391*9880d681SAndroid Build Coastguard Worker memcpy(pVal, dest, wordsToCopy * APINT_WORD_SIZE);
392*9880d681SAndroid Build Coastguard Worker clearUnusedBits();
393*9880d681SAndroid Build Coastguard Worker
394*9880d681SAndroid Build Coastguard Worker // delete dest array and return
395*9880d681SAndroid Build Coastguard Worker delete[] dest;
396*9880d681SAndroid Build Coastguard Worker return *this;
397*9880d681SAndroid Build Coastguard Worker }
398*9880d681SAndroid Build Coastguard Worker
operator &=(const APInt & RHS)399*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator&=(const APInt& RHS) {
400*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
401*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
402*9880d681SAndroid Build Coastguard Worker VAL &= RHS.VAL;
403*9880d681SAndroid Build Coastguard Worker return *this;
404*9880d681SAndroid Build Coastguard Worker }
405*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
406*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
407*9880d681SAndroid Build Coastguard Worker pVal[i] &= RHS.pVal[i];
408*9880d681SAndroid Build Coastguard Worker return *this;
409*9880d681SAndroid Build Coastguard Worker }
410*9880d681SAndroid Build Coastguard Worker
operator |=(const APInt & RHS)411*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator|=(const APInt& RHS) {
412*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
413*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
414*9880d681SAndroid Build Coastguard Worker VAL |= RHS.VAL;
415*9880d681SAndroid Build Coastguard Worker return *this;
416*9880d681SAndroid Build Coastguard Worker }
417*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
418*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
419*9880d681SAndroid Build Coastguard Worker pVal[i] |= RHS.pVal[i];
420*9880d681SAndroid Build Coastguard Worker return *this;
421*9880d681SAndroid Build Coastguard Worker }
422*9880d681SAndroid Build Coastguard Worker
operator ^=(const APInt & RHS)423*9880d681SAndroid Build Coastguard Worker APInt& APInt::operator^=(const APInt& RHS) {
424*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
425*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
426*9880d681SAndroid Build Coastguard Worker VAL ^= RHS.VAL;
427*9880d681SAndroid Build Coastguard Worker this->clearUnusedBits();
428*9880d681SAndroid Build Coastguard Worker return *this;
429*9880d681SAndroid Build Coastguard Worker }
430*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
431*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
432*9880d681SAndroid Build Coastguard Worker pVal[i] ^= RHS.pVal[i];
433*9880d681SAndroid Build Coastguard Worker return clearUnusedBits();
434*9880d681SAndroid Build Coastguard Worker }
435*9880d681SAndroid Build Coastguard Worker
AndSlowCase(const APInt & RHS) const436*9880d681SAndroid Build Coastguard Worker APInt APInt::AndSlowCase(const APInt& RHS) const {
437*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
438*9880d681SAndroid Build Coastguard Worker uint64_t* val = getMemory(numWords);
439*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
440*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i] & RHS.pVal[i];
441*9880d681SAndroid Build Coastguard Worker return APInt(val, getBitWidth());
442*9880d681SAndroid Build Coastguard Worker }
443*9880d681SAndroid Build Coastguard Worker
OrSlowCase(const APInt & RHS) const444*9880d681SAndroid Build Coastguard Worker APInt APInt::OrSlowCase(const APInt& RHS) const {
445*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
446*9880d681SAndroid Build Coastguard Worker uint64_t *val = getMemory(numWords);
447*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
448*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i] | RHS.pVal[i];
449*9880d681SAndroid Build Coastguard Worker return APInt(val, getBitWidth());
450*9880d681SAndroid Build Coastguard Worker }
451*9880d681SAndroid Build Coastguard Worker
XorSlowCase(const APInt & RHS) const452*9880d681SAndroid Build Coastguard Worker APInt APInt::XorSlowCase(const APInt& RHS) const {
453*9880d681SAndroid Build Coastguard Worker unsigned numWords = getNumWords();
454*9880d681SAndroid Build Coastguard Worker uint64_t *val = getMemory(numWords);
455*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < numWords; ++i)
456*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i] ^ RHS.pVal[i];
457*9880d681SAndroid Build Coastguard Worker
458*9880d681SAndroid Build Coastguard Worker APInt Result(val, getBitWidth());
459*9880d681SAndroid Build Coastguard Worker // 0^0==1 so clear the high bits in case they got set.
460*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
461*9880d681SAndroid Build Coastguard Worker return Result;
462*9880d681SAndroid Build Coastguard Worker }
463*9880d681SAndroid Build Coastguard Worker
operator *(const APInt & RHS) const464*9880d681SAndroid Build Coastguard Worker APInt APInt::operator*(const APInt& RHS) const {
465*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
466*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
467*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL * RHS.VAL);
468*9880d681SAndroid Build Coastguard Worker APInt Result(*this);
469*9880d681SAndroid Build Coastguard Worker Result *= RHS;
470*9880d681SAndroid Build Coastguard Worker return Result;
471*9880d681SAndroid Build Coastguard Worker }
472*9880d681SAndroid Build Coastguard Worker
operator +(const APInt & RHS) const473*9880d681SAndroid Build Coastguard Worker APInt APInt::operator+(const APInt& RHS) const {
474*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
475*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
476*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL + RHS.VAL);
477*9880d681SAndroid Build Coastguard Worker APInt Result(BitWidth, 0);
478*9880d681SAndroid Build Coastguard Worker add(Result.pVal, this->pVal, RHS.pVal, getNumWords());
479*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
480*9880d681SAndroid Build Coastguard Worker return Result;
481*9880d681SAndroid Build Coastguard Worker }
482*9880d681SAndroid Build Coastguard Worker
operator +(uint64_t RHS) const483*9880d681SAndroid Build Coastguard Worker APInt APInt::operator+(uint64_t RHS) const {
484*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
485*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL + RHS);
486*9880d681SAndroid Build Coastguard Worker APInt Result(*this);
487*9880d681SAndroid Build Coastguard Worker add_1(Result.pVal, Result.pVal, getNumWords(), RHS);
488*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
489*9880d681SAndroid Build Coastguard Worker return Result;
490*9880d681SAndroid Build Coastguard Worker }
491*9880d681SAndroid Build Coastguard Worker
operator -(const APInt & RHS) const492*9880d681SAndroid Build Coastguard Worker APInt APInt::operator-(const APInt& RHS) const {
493*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
494*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
495*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL - RHS.VAL);
496*9880d681SAndroid Build Coastguard Worker APInt Result(BitWidth, 0);
497*9880d681SAndroid Build Coastguard Worker sub(Result.pVal, this->pVal, RHS.pVal, getNumWords());
498*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
499*9880d681SAndroid Build Coastguard Worker return Result;
500*9880d681SAndroid Build Coastguard Worker }
501*9880d681SAndroid Build Coastguard Worker
operator -(uint64_t RHS) const502*9880d681SAndroid Build Coastguard Worker APInt APInt::operator-(uint64_t RHS) const {
503*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
504*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL - RHS);
505*9880d681SAndroid Build Coastguard Worker APInt Result(*this);
506*9880d681SAndroid Build Coastguard Worker sub_1(Result.pVal, getNumWords(), RHS);
507*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
508*9880d681SAndroid Build Coastguard Worker return Result;
509*9880d681SAndroid Build Coastguard Worker }
510*9880d681SAndroid Build Coastguard Worker
EqualSlowCase(const APInt & RHS) const511*9880d681SAndroid Build Coastguard Worker bool APInt::EqualSlowCase(const APInt& RHS) const {
512*9880d681SAndroid Build Coastguard Worker return std::equal(pVal, pVal + getNumWords(), RHS.pVal);
513*9880d681SAndroid Build Coastguard Worker }
514*9880d681SAndroid Build Coastguard Worker
EqualSlowCase(uint64_t Val) const515*9880d681SAndroid Build Coastguard Worker bool APInt::EqualSlowCase(uint64_t Val) const {
516*9880d681SAndroid Build Coastguard Worker unsigned n = getActiveBits();
517*9880d681SAndroid Build Coastguard Worker if (n <= APINT_BITS_PER_WORD)
518*9880d681SAndroid Build Coastguard Worker return pVal[0] == Val;
519*9880d681SAndroid Build Coastguard Worker else
520*9880d681SAndroid Build Coastguard Worker return false;
521*9880d681SAndroid Build Coastguard Worker }
522*9880d681SAndroid Build Coastguard Worker
ult(const APInt & RHS) const523*9880d681SAndroid Build Coastguard Worker bool APInt::ult(const APInt& RHS) const {
524*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be same for comparison");
525*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
526*9880d681SAndroid Build Coastguard Worker return VAL < RHS.VAL;
527*9880d681SAndroid Build Coastguard Worker
528*9880d681SAndroid Build Coastguard Worker // Get active bit length of both operands
529*9880d681SAndroid Build Coastguard Worker unsigned n1 = getActiveBits();
530*9880d681SAndroid Build Coastguard Worker unsigned n2 = RHS.getActiveBits();
531*9880d681SAndroid Build Coastguard Worker
532*9880d681SAndroid Build Coastguard Worker // If magnitude of LHS is less than RHS, return true.
533*9880d681SAndroid Build Coastguard Worker if (n1 < n2)
534*9880d681SAndroid Build Coastguard Worker return true;
535*9880d681SAndroid Build Coastguard Worker
536*9880d681SAndroid Build Coastguard Worker // If magnitude of RHS is greather than LHS, return false.
537*9880d681SAndroid Build Coastguard Worker if (n2 < n1)
538*9880d681SAndroid Build Coastguard Worker return false;
539*9880d681SAndroid Build Coastguard Worker
540*9880d681SAndroid Build Coastguard Worker // If they bot fit in a word, just compare the low order word
541*9880d681SAndroid Build Coastguard Worker if (n1 <= APINT_BITS_PER_WORD && n2 <= APINT_BITS_PER_WORD)
542*9880d681SAndroid Build Coastguard Worker return pVal[0] < RHS.pVal[0];
543*9880d681SAndroid Build Coastguard Worker
544*9880d681SAndroid Build Coastguard Worker // Otherwise, compare all words
545*9880d681SAndroid Build Coastguard Worker unsigned topWord = whichWord(std::max(n1,n2)-1);
546*9880d681SAndroid Build Coastguard Worker for (int i = topWord; i >= 0; --i) {
547*9880d681SAndroid Build Coastguard Worker if (pVal[i] > RHS.pVal[i])
548*9880d681SAndroid Build Coastguard Worker return false;
549*9880d681SAndroid Build Coastguard Worker if (pVal[i] < RHS.pVal[i])
550*9880d681SAndroid Build Coastguard Worker return true;
551*9880d681SAndroid Build Coastguard Worker }
552*9880d681SAndroid Build Coastguard Worker return false;
553*9880d681SAndroid Build Coastguard Worker }
554*9880d681SAndroid Build Coastguard Worker
slt(const APInt & RHS) const555*9880d681SAndroid Build Coastguard Worker bool APInt::slt(const APInt& RHS) const {
556*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be same for comparison");
557*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
558*9880d681SAndroid Build Coastguard Worker int64_t lhsSext = SignExtend64(VAL, BitWidth);
559*9880d681SAndroid Build Coastguard Worker int64_t rhsSext = SignExtend64(RHS.VAL, BitWidth);
560*9880d681SAndroid Build Coastguard Worker return lhsSext < rhsSext;
561*9880d681SAndroid Build Coastguard Worker }
562*9880d681SAndroid Build Coastguard Worker
563*9880d681SAndroid Build Coastguard Worker bool lhsNeg = isNegative();
564*9880d681SAndroid Build Coastguard Worker bool rhsNeg = RHS.isNegative();
565*9880d681SAndroid Build Coastguard Worker
566*9880d681SAndroid Build Coastguard Worker // If the sign bits don't match, then (LHS < RHS) if LHS is negative
567*9880d681SAndroid Build Coastguard Worker if (lhsNeg != rhsNeg)
568*9880d681SAndroid Build Coastguard Worker return lhsNeg;
569*9880d681SAndroid Build Coastguard Worker
570*9880d681SAndroid Build Coastguard Worker // Otherwise we can just use an unsigned comparision, because even negative
571*9880d681SAndroid Build Coastguard Worker // numbers compare correctly this way if both have the same signed-ness.
572*9880d681SAndroid Build Coastguard Worker return ult(RHS);
573*9880d681SAndroid Build Coastguard Worker }
574*9880d681SAndroid Build Coastguard Worker
setBit(unsigned bitPosition)575*9880d681SAndroid Build Coastguard Worker void APInt::setBit(unsigned bitPosition) {
576*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
577*9880d681SAndroid Build Coastguard Worker VAL |= maskBit(bitPosition);
578*9880d681SAndroid Build Coastguard Worker else
579*9880d681SAndroid Build Coastguard Worker pVal[whichWord(bitPosition)] |= maskBit(bitPosition);
580*9880d681SAndroid Build Coastguard Worker }
581*9880d681SAndroid Build Coastguard Worker
582*9880d681SAndroid Build Coastguard Worker /// Set the given bit to 0 whose position is given as "bitPosition".
583*9880d681SAndroid Build Coastguard Worker /// @brief Set a given bit to 0.
clearBit(unsigned bitPosition)584*9880d681SAndroid Build Coastguard Worker void APInt::clearBit(unsigned bitPosition) {
585*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
586*9880d681SAndroid Build Coastguard Worker VAL &= ~maskBit(bitPosition);
587*9880d681SAndroid Build Coastguard Worker else
588*9880d681SAndroid Build Coastguard Worker pVal[whichWord(bitPosition)] &= ~maskBit(bitPosition);
589*9880d681SAndroid Build Coastguard Worker }
590*9880d681SAndroid Build Coastguard Worker
591*9880d681SAndroid Build Coastguard Worker /// @brief Toggle every bit to its opposite value.
592*9880d681SAndroid Build Coastguard Worker
593*9880d681SAndroid Build Coastguard Worker /// Toggle a given bit to its opposite value whose position is given
594*9880d681SAndroid Build Coastguard Worker /// as "bitPosition".
595*9880d681SAndroid Build Coastguard Worker /// @brief Toggles a given bit to its opposite value.
flipBit(unsigned bitPosition)596*9880d681SAndroid Build Coastguard Worker void APInt::flipBit(unsigned bitPosition) {
597*9880d681SAndroid Build Coastguard Worker assert(bitPosition < BitWidth && "Out of the bit-width range!");
598*9880d681SAndroid Build Coastguard Worker if ((*this)[bitPosition]) clearBit(bitPosition);
599*9880d681SAndroid Build Coastguard Worker else setBit(bitPosition);
600*9880d681SAndroid Build Coastguard Worker }
601*9880d681SAndroid Build Coastguard Worker
getBitsNeeded(StringRef str,uint8_t radix)602*9880d681SAndroid Build Coastguard Worker unsigned APInt::getBitsNeeded(StringRef str, uint8_t radix) {
603*9880d681SAndroid Build Coastguard Worker assert(!str.empty() && "Invalid string length");
604*9880d681SAndroid Build Coastguard Worker assert((radix == 10 || radix == 8 || radix == 16 || radix == 2 ||
605*9880d681SAndroid Build Coastguard Worker radix == 36) &&
606*9880d681SAndroid Build Coastguard Worker "Radix should be 2, 8, 10, 16, or 36!");
607*9880d681SAndroid Build Coastguard Worker
608*9880d681SAndroid Build Coastguard Worker size_t slen = str.size();
609*9880d681SAndroid Build Coastguard Worker
610*9880d681SAndroid Build Coastguard Worker // Each computation below needs to know if it's negative.
611*9880d681SAndroid Build Coastguard Worker StringRef::iterator p = str.begin();
612*9880d681SAndroid Build Coastguard Worker unsigned isNegative = *p == '-';
613*9880d681SAndroid Build Coastguard Worker if (*p == '-' || *p == '+') {
614*9880d681SAndroid Build Coastguard Worker p++;
615*9880d681SAndroid Build Coastguard Worker slen--;
616*9880d681SAndroid Build Coastguard Worker assert(slen && "String is only a sign, needs a value.");
617*9880d681SAndroid Build Coastguard Worker }
618*9880d681SAndroid Build Coastguard Worker
619*9880d681SAndroid Build Coastguard Worker // For radixes of power-of-two values, the bits required is accurately and
620*9880d681SAndroid Build Coastguard Worker // easily computed
621*9880d681SAndroid Build Coastguard Worker if (radix == 2)
622*9880d681SAndroid Build Coastguard Worker return slen + isNegative;
623*9880d681SAndroid Build Coastguard Worker if (radix == 8)
624*9880d681SAndroid Build Coastguard Worker return slen * 3 + isNegative;
625*9880d681SAndroid Build Coastguard Worker if (radix == 16)
626*9880d681SAndroid Build Coastguard Worker return slen * 4 + isNegative;
627*9880d681SAndroid Build Coastguard Worker
628*9880d681SAndroid Build Coastguard Worker // FIXME: base 36
629*9880d681SAndroid Build Coastguard Worker
630*9880d681SAndroid Build Coastguard Worker // This is grossly inefficient but accurate. We could probably do something
631*9880d681SAndroid Build Coastguard Worker // with a computation of roughly slen*64/20 and then adjust by the value of
632*9880d681SAndroid Build Coastguard Worker // the first few digits. But, I'm not sure how accurate that could be.
633*9880d681SAndroid Build Coastguard Worker
634*9880d681SAndroid Build Coastguard Worker // Compute a sufficient number of bits that is always large enough but might
635*9880d681SAndroid Build Coastguard Worker // be too large. This avoids the assertion in the constructor. This
636*9880d681SAndroid Build Coastguard Worker // calculation doesn't work appropriately for the numbers 0-9, so just use 4
637*9880d681SAndroid Build Coastguard Worker // bits in that case.
638*9880d681SAndroid Build Coastguard Worker unsigned sufficient
639*9880d681SAndroid Build Coastguard Worker = radix == 10? (slen == 1 ? 4 : slen * 64/18)
640*9880d681SAndroid Build Coastguard Worker : (slen == 1 ? 7 : slen * 16/3);
641*9880d681SAndroid Build Coastguard Worker
642*9880d681SAndroid Build Coastguard Worker // Convert to the actual binary value.
643*9880d681SAndroid Build Coastguard Worker APInt tmp(sufficient, StringRef(p, slen), radix);
644*9880d681SAndroid Build Coastguard Worker
645*9880d681SAndroid Build Coastguard Worker // Compute how many bits are required. If the log is infinite, assume we need
646*9880d681SAndroid Build Coastguard Worker // just bit.
647*9880d681SAndroid Build Coastguard Worker unsigned log = tmp.logBase2();
648*9880d681SAndroid Build Coastguard Worker if (log == (unsigned)-1) {
649*9880d681SAndroid Build Coastguard Worker return isNegative + 1;
650*9880d681SAndroid Build Coastguard Worker } else {
651*9880d681SAndroid Build Coastguard Worker return isNegative + log + 1;
652*9880d681SAndroid Build Coastguard Worker }
653*9880d681SAndroid Build Coastguard Worker }
654*9880d681SAndroid Build Coastguard Worker
hash_value(const APInt & Arg)655*9880d681SAndroid Build Coastguard Worker hash_code llvm::hash_value(const APInt &Arg) {
656*9880d681SAndroid Build Coastguard Worker if (Arg.isSingleWord())
657*9880d681SAndroid Build Coastguard Worker return hash_combine(Arg.VAL);
658*9880d681SAndroid Build Coastguard Worker
659*9880d681SAndroid Build Coastguard Worker return hash_combine_range(Arg.pVal, Arg.pVal + Arg.getNumWords());
660*9880d681SAndroid Build Coastguard Worker }
661*9880d681SAndroid Build Coastguard Worker
isSplat(unsigned SplatSizeInBits) const662*9880d681SAndroid Build Coastguard Worker bool APInt::isSplat(unsigned SplatSizeInBits) const {
663*9880d681SAndroid Build Coastguard Worker assert(getBitWidth() % SplatSizeInBits == 0 &&
664*9880d681SAndroid Build Coastguard Worker "SplatSizeInBits must divide width!");
665*9880d681SAndroid Build Coastguard Worker // We can check that all parts of an integer are equal by making use of a
666*9880d681SAndroid Build Coastguard Worker // little trick: rotate and check if it's still the same value.
667*9880d681SAndroid Build Coastguard Worker return *this == rotl(SplatSizeInBits);
668*9880d681SAndroid Build Coastguard Worker }
669*9880d681SAndroid Build Coastguard Worker
670*9880d681SAndroid Build Coastguard Worker /// This function returns the high "numBits" bits of this APInt.
getHiBits(unsigned numBits) const671*9880d681SAndroid Build Coastguard Worker APInt APInt::getHiBits(unsigned numBits) const {
672*9880d681SAndroid Build Coastguard Worker return APIntOps::lshr(*this, BitWidth - numBits);
673*9880d681SAndroid Build Coastguard Worker }
674*9880d681SAndroid Build Coastguard Worker
675*9880d681SAndroid Build Coastguard Worker /// This function returns the low "numBits" bits of this APInt.
getLoBits(unsigned numBits) const676*9880d681SAndroid Build Coastguard Worker APInt APInt::getLoBits(unsigned numBits) const {
677*9880d681SAndroid Build Coastguard Worker return APIntOps::lshr(APIntOps::shl(*this, BitWidth - numBits),
678*9880d681SAndroid Build Coastguard Worker BitWidth - numBits);
679*9880d681SAndroid Build Coastguard Worker }
680*9880d681SAndroid Build Coastguard Worker
countLeadingZerosSlowCase() const681*9880d681SAndroid Build Coastguard Worker unsigned APInt::countLeadingZerosSlowCase() const {
682*9880d681SAndroid Build Coastguard Worker unsigned Count = 0;
683*9880d681SAndroid Build Coastguard Worker for (int i = getNumWords()-1; i >= 0; --i) {
684*9880d681SAndroid Build Coastguard Worker integerPart V = pVal[i];
685*9880d681SAndroid Build Coastguard Worker if (V == 0)
686*9880d681SAndroid Build Coastguard Worker Count += APINT_BITS_PER_WORD;
687*9880d681SAndroid Build Coastguard Worker else {
688*9880d681SAndroid Build Coastguard Worker Count += llvm::countLeadingZeros(V);
689*9880d681SAndroid Build Coastguard Worker break;
690*9880d681SAndroid Build Coastguard Worker }
691*9880d681SAndroid Build Coastguard Worker }
692*9880d681SAndroid Build Coastguard Worker // Adjust for unused bits in the most significant word (they are zero).
693*9880d681SAndroid Build Coastguard Worker unsigned Mod = BitWidth % APINT_BITS_PER_WORD;
694*9880d681SAndroid Build Coastguard Worker Count -= Mod > 0 ? APINT_BITS_PER_WORD - Mod : 0;
695*9880d681SAndroid Build Coastguard Worker return Count;
696*9880d681SAndroid Build Coastguard Worker }
697*9880d681SAndroid Build Coastguard Worker
countLeadingOnes() const698*9880d681SAndroid Build Coastguard Worker unsigned APInt::countLeadingOnes() const {
699*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
700*9880d681SAndroid Build Coastguard Worker return llvm::countLeadingOnes(VAL << (APINT_BITS_PER_WORD - BitWidth));
701*9880d681SAndroid Build Coastguard Worker
702*9880d681SAndroid Build Coastguard Worker unsigned highWordBits = BitWidth % APINT_BITS_PER_WORD;
703*9880d681SAndroid Build Coastguard Worker unsigned shift;
704*9880d681SAndroid Build Coastguard Worker if (!highWordBits) {
705*9880d681SAndroid Build Coastguard Worker highWordBits = APINT_BITS_PER_WORD;
706*9880d681SAndroid Build Coastguard Worker shift = 0;
707*9880d681SAndroid Build Coastguard Worker } else {
708*9880d681SAndroid Build Coastguard Worker shift = APINT_BITS_PER_WORD - highWordBits;
709*9880d681SAndroid Build Coastguard Worker }
710*9880d681SAndroid Build Coastguard Worker int i = getNumWords() - 1;
711*9880d681SAndroid Build Coastguard Worker unsigned Count = llvm::countLeadingOnes(pVal[i] << shift);
712*9880d681SAndroid Build Coastguard Worker if (Count == highWordBits) {
713*9880d681SAndroid Build Coastguard Worker for (i--; i >= 0; --i) {
714*9880d681SAndroid Build Coastguard Worker if (pVal[i] == -1ULL)
715*9880d681SAndroid Build Coastguard Worker Count += APINT_BITS_PER_WORD;
716*9880d681SAndroid Build Coastguard Worker else {
717*9880d681SAndroid Build Coastguard Worker Count += llvm::countLeadingOnes(pVal[i]);
718*9880d681SAndroid Build Coastguard Worker break;
719*9880d681SAndroid Build Coastguard Worker }
720*9880d681SAndroid Build Coastguard Worker }
721*9880d681SAndroid Build Coastguard Worker }
722*9880d681SAndroid Build Coastguard Worker return Count;
723*9880d681SAndroid Build Coastguard Worker }
724*9880d681SAndroid Build Coastguard Worker
countTrailingZeros() const725*9880d681SAndroid Build Coastguard Worker unsigned APInt::countTrailingZeros() const {
726*9880d681SAndroid Build Coastguard Worker if (isSingleWord())
727*9880d681SAndroid Build Coastguard Worker return std::min(unsigned(llvm::countTrailingZeros(VAL)), BitWidth);
728*9880d681SAndroid Build Coastguard Worker unsigned Count = 0;
729*9880d681SAndroid Build Coastguard Worker unsigned i = 0;
730*9880d681SAndroid Build Coastguard Worker for (; i < getNumWords() && pVal[i] == 0; ++i)
731*9880d681SAndroid Build Coastguard Worker Count += APINT_BITS_PER_WORD;
732*9880d681SAndroid Build Coastguard Worker if (i < getNumWords())
733*9880d681SAndroid Build Coastguard Worker Count += llvm::countTrailingZeros(pVal[i]);
734*9880d681SAndroid Build Coastguard Worker return std::min(Count, BitWidth);
735*9880d681SAndroid Build Coastguard Worker }
736*9880d681SAndroid Build Coastguard Worker
countTrailingOnesSlowCase() const737*9880d681SAndroid Build Coastguard Worker unsigned APInt::countTrailingOnesSlowCase() const {
738*9880d681SAndroid Build Coastguard Worker unsigned Count = 0;
739*9880d681SAndroid Build Coastguard Worker unsigned i = 0;
740*9880d681SAndroid Build Coastguard Worker for (; i < getNumWords() && pVal[i] == -1ULL; ++i)
741*9880d681SAndroid Build Coastguard Worker Count += APINT_BITS_PER_WORD;
742*9880d681SAndroid Build Coastguard Worker if (i < getNumWords())
743*9880d681SAndroid Build Coastguard Worker Count += llvm::countTrailingOnes(pVal[i]);
744*9880d681SAndroid Build Coastguard Worker return std::min(Count, BitWidth);
745*9880d681SAndroid Build Coastguard Worker }
746*9880d681SAndroid Build Coastguard Worker
countPopulationSlowCase() const747*9880d681SAndroid Build Coastguard Worker unsigned APInt::countPopulationSlowCase() const {
748*9880d681SAndroid Build Coastguard Worker unsigned Count = 0;
749*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < getNumWords(); ++i)
750*9880d681SAndroid Build Coastguard Worker Count += llvm::countPopulation(pVal[i]);
751*9880d681SAndroid Build Coastguard Worker return Count;
752*9880d681SAndroid Build Coastguard Worker }
753*9880d681SAndroid Build Coastguard Worker
754*9880d681SAndroid Build Coastguard Worker /// Perform a logical right-shift from Src to Dst, which must be equal or
755*9880d681SAndroid Build Coastguard Worker /// non-overlapping, of Words words, by Shift, which must be less than 64.
lshrNear(uint64_t * Dst,uint64_t * Src,unsigned Words,unsigned Shift)756*9880d681SAndroid Build Coastguard Worker static void lshrNear(uint64_t *Dst, uint64_t *Src, unsigned Words,
757*9880d681SAndroid Build Coastguard Worker unsigned Shift) {
758*9880d681SAndroid Build Coastguard Worker uint64_t Carry = 0;
759*9880d681SAndroid Build Coastguard Worker for (int I = Words - 1; I >= 0; --I) {
760*9880d681SAndroid Build Coastguard Worker uint64_t Tmp = Src[I];
761*9880d681SAndroid Build Coastguard Worker Dst[I] = (Tmp >> Shift) | Carry;
762*9880d681SAndroid Build Coastguard Worker Carry = Tmp << (64 - Shift);
763*9880d681SAndroid Build Coastguard Worker }
764*9880d681SAndroid Build Coastguard Worker }
765*9880d681SAndroid Build Coastguard Worker
byteSwap() const766*9880d681SAndroid Build Coastguard Worker APInt APInt::byteSwap() const {
767*9880d681SAndroid Build Coastguard Worker assert(BitWidth >= 16 && BitWidth % 16 == 0 && "Cannot byteswap!");
768*9880d681SAndroid Build Coastguard Worker if (BitWidth == 16)
769*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, ByteSwap_16(uint16_t(VAL)));
770*9880d681SAndroid Build Coastguard Worker if (BitWidth == 32)
771*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, ByteSwap_32(unsigned(VAL)));
772*9880d681SAndroid Build Coastguard Worker if (BitWidth == 48) {
773*9880d681SAndroid Build Coastguard Worker unsigned Tmp1 = unsigned(VAL >> 16);
774*9880d681SAndroid Build Coastguard Worker Tmp1 = ByteSwap_32(Tmp1);
775*9880d681SAndroid Build Coastguard Worker uint16_t Tmp2 = uint16_t(VAL);
776*9880d681SAndroid Build Coastguard Worker Tmp2 = ByteSwap_16(Tmp2);
777*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, (uint64_t(Tmp2) << 32) | Tmp1);
778*9880d681SAndroid Build Coastguard Worker }
779*9880d681SAndroid Build Coastguard Worker if (BitWidth == 64)
780*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, ByteSwap_64(VAL));
781*9880d681SAndroid Build Coastguard Worker
782*9880d681SAndroid Build Coastguard Worker APInt Result(getNumWords() * APINT_BITS_PER_WORD, 0);
783*9880d681SAndroid Build Coastguard Worker for (unsigned I = 0, N = getNumWords(); I != N; ++I)
784*9880d681SAndroid Build Coastguard Worker Result.pVal[I] = ByteSwap_64(pVal[N - I - 1]);
785*9880d681SAndroid Build Coastguard Worker if (Result.BitWidth != BitWidth) {
786*9880d681SAndroid Build Coastguard Worker lshrNear(Result.pVal, Result.pVal, getNumWords(),
787*9880d681SAndroid Build Coastguard Worker Result.BitWidth - BitWidth);
788*9880d681SAndroid Build Coastguard Worker Result.BitWidth = BitWidth;
789*9880d681SAndroid Build Coastguard Worker }
790*9880d681SAndroid Build Coastguard Worker return Result;
791*9880d681SAndroid Build Coastguard Worker }
792*9880d681SAndroid Build Coastguard Worker
reverseBits() const793*9880d681SAndroid Build Coastguard Worker APInt APInt::reverseBits() const {
794*9880d681SAndroid Build Coastguard Worker switch (BitWidth) {
795*9880d681SAndroid Build Coastguard Worker case 64:
796*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, llvm::reverseBits<uint64_t>(VAL));
797*9880d681SAndroid Build Coastguard Worker case 32:
798*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, llvm::reverseBits<uint32_t>(VAL));
799*9880d681SAndroid Build Coastguard Worker case 16:
800*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, llvm::reverseBits<uint16_t>(VAL));
801*9880d681SAndroid Build Coastguard Worker case 8:
802*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, llvm::reverseBits<uint8_t>(VAL));
803*9880d681SAndroid Build Coastguard Worker default:
804*9880d681SAndroid Build Coastguard Worker break;
805*9880d681SAndroid Build Coastguard Worker }
806*9880d681SAndroid Build Coastguard Worker
807*9880d681SAndroid Build Coastguard Worker APInt Val(*this);
808*9880d681SAndroid Build Coastguard Worker APInt Reversed(*this);
809*9880d681SAndroid Build Coastguard Worker int S = BitWidth - 1;
810*9880d681SAndroid Build Coastguard Worker
811*9880d681SAndroid Build Coastguard Worker const APInt One(BitWidth, 1);
812*9880d681SAndroid Build Coastguard Worker
813*9880d681SAndroid Build Coastguard Worker for ((Val = Val.lshr(1)); Val != 0; (Val = Val.lshr(1))) {
814*9880d681SAndroid Build Coastguard Worker Reversed <<= 1;
815*9880d681SAndroid Build Coastguard Worker Reversed |= (Val & One);
816*9880d681SAndroid Build Coastguard Worker --S;
817*9880d681SAndroid Build Coastguard Worker }
818*9880d681SAndroid Build Coastguard Worker
819*9880d681SAndroid Build Coastguard Worker Reversed <<= S;
820*9880d681SAndroid Build Coastguard Worker return Reversed;
821*9880d681SAndroid Build Coastguard Worker }
822*9880d681SAndroid Build Coastguard Worker
GreatestCommonDivisor(const APInt & API1,const APInt & API2)823*9880d681SAndroid Build Coastguard Worker APInt llvm::APIntOps::GreatestCommonDivisor(const APInt& API1,
824*9880d681SAndroid Build Coastguard Worker const APInt& API2) {
825*9880d681SAndroid Build Coastguard Worker APInt A = API1, B = API2;
826*9880d681SAndroid Build Coastguard Worker while (!!B) {
827*9880d681SAndroid Build Coastguard Worker APInt T = B;
828*9880d681SAndroid Build Coastguard Worker B = APIntOps::urem(A, B);
829*9880d681SAndroid Build Coastguard Worker A = T;
830*9880d681SAndroid Build Coastguard Worker }
831*9880d681SAndroid Build Coastguard Worker return A;
832*9880d681SAndroid Build Coastguard Worker }
833*9880d681SAndroid Build Coastguard Worker
RoundDoubleToAPInt(double Double,unsigned width)834*9880d681SAndroid Build Coastguard Worker APInt llvm::APIntOps::RoundDoubleToAPInt(double Double, unsigned width) {
835*9880d681SAndroid Build Coastguard Worker union {
836*9880d681SAndroid Build Coastguard Worker double D;
837*9880d681SAndroid Build Coastguard Worker uint64_t I;
838*9880d681SAndroid Build Coastguard Worker } T;
839*9880d681SAndroid Build Coastguard Worker T.D = Double;
840*9880d681SAndroid Build Coastguard Worker
841*9880d681SAndroid Build Coastguard Worker // Get the sign bit from the highest order bit
842*9880d681SAndroid Build Coastguard Worker bool isNeg = T.I >> 63;
843*9880d681SAndroid Build Coastguard Worker
844*9880d681SAndroid Build Coastguard Worker // Get the 11-bit exponent and adjust for the 1023 bit bias
845*9880d681SAndroid Build Coastguard Worker int64_t exp = ((T.I >> 52) & 0x7ff) - 1023;
846*9880d681SAndroid Build Coastguard Worker
847*9880d681SAndroid Build Coastguard Worker // If the exponent is negative, the value is < 0 so just return 0.
848*9880d681SAndroid Build Coastguard Worker if (exp < 0)
849*9880d681SAndroid Build Coastguard Worker return APInt(width, 0u);
850*9880d681SAndroid Build Coastguard Worker
851*9880d681SAndroid Build Coastguard Worker // Extract the mantissa by clearing the top 12 bits (sign + exponent).
852*9880d681SAndroid Build Coastguard Worker uint64_t mantissa = (T.I & (~0ULL >> 12)) | 1ULL << 52;
853*9880d681SAndroid Build Coastguard Worker
854*9880d681SAndroid Build Coastguard Worker // If the exponent doesn't shift all bits out of the mantissa
855*9880d681SAndroid Build Coastguard Worker if (exp < 52)
856*9880d681SAndroid Build Coastguard Worker return isNeg ? -APInt(width, mantissa >> (52 - exp)) :
857*9880d681SAndroid Build Coastguard Worker APInt(width, mantissa >> (52 - exp));
858*9880d681SAndroid Build Coastguard Worker
859*9880d681SAndroid Build Coastguard Worker // If the client didn't provide enough bits for us to shift the mantissa into
860*9880d681SAndroid Build Coastguard Worker // then the result is undefined, just return 0
861*9880d681SAndroid Build Coastguard Worker if (width <= exp - 52)
862*9880d681SAndroid Build Coastguard Worker return APInt(width, 0);
863*9880d681SAndroid Build Coastguard Worker
864*9880d681SAndroid Build Coastguard Worker // Otherwise, we have to shift the mantissa bits up to the right location
865*9880d681SAndroid Build Coastguard Worker APInt Tmp(width, mantissa);
866*9880d681SAndroid Build Coastguard Worker Tmp = Tmp.shl((unsigned)exp - 52);
867*9880d681SAndroid Build Coastguard Worker return isNeg ? -Tmp : Tmp;
868*9880d681SAndroid Build Coastguard Worker }
869*9880d681SAndroid Build Coastguard Worker
870*9880d681SAndroid Build Coastguard Worker /// This function converts this APInt to a double.
871*9880d681SAndroid Build Coastguard Worker /// The layout for double is as following (IEEE Standard 754):
872*9880d681SAndroid Build Coastguard Worker /// --------------------------------------
873*9880d681SAndroid Build Coastguard Worker /// | Sign Exponent Fraction Bias |
874*9880d681SAndroid Build Coastguard Worker /// |-------------------------------------- |
875*9880d681SAndroid Build Coastguard Worker /// | 1[63] 11[62-52] 52[51-00] 1023 |
876*9880d681SAndroid Build Coastguard Worker /// --------------------------------------
roundToDouble(bool isSigned) const877*9880d681SAndroid Build Coastguard Worker double APInt::roundToDouble(bool isSigned) const {
878*9880d681SAndroid Build Coastguard Worker
879*9880d681SAndroid Build Coastguard Worker // Handle the simple case where the value is contained in one uint64_t.
880*9880d681SAndroid Build Coastguard Worker // It is wrong to optimize getWord(0) to VAL; there might be more than one word.
881*9880d681SAndroid Build Coastguard Worker if (isSingleWord() || getActiveBits() <= APINT_BITS_PER_WORD) {
882*9880d681SAndroid Build Coastguard Worker if (isSigned) {
883*9880d681SAndroid Build Coastguard Worker int64_t sext = SignExtend64(getWord(0), BitWidth);
884*9880d681SAndroid Build Coastguard Worker return double(sext);
885*9880d681SAndroid Build Coastguard Worker } else
886*9880d681SAndroid Build Coastguard Worker return double(getWord(0));
887*9880d681SAndroid Build Coastguard Worker }
888*9880d681SAndroid Build Coastguard Worker
889*9880d681SAndroid Build Coastguard Worker // Determine if the value is negative.
890*9880d681SAndroid Build Coastguard Worker bool isNeg = isSigned ? (*this)[BitWidth-1] : false;
891*9880d681SAndroid Build Coastguard Worker
892*9880d681SAndroid Build Coastguard Worker // Construct the absolute value if we're negative.
893*9880d681SAndroid Build Coastguard Worker APInt Tmp(isNeg ? -(*this) : (*this));
894*9880d681SAndroid Build Coastguard Worker
895*9880d681SAndroid Build Coastguard Worker // Figure out how many bits we're using.
896*9880d681SAndroid Build Coastguard Worker unsigned n = Tmp.getActiveBits();
897*9880d681SAndroid Build Coastguard Worker
898*9880d681SAndroid Build Coastguard Worker // The exponent (without bias normalization) is just the number of bits
899*9880d681SAndroid Build Coastguard Worker // we are using. Note that the sign bit is gone since we constructed the
900*9880d681SAndroid Build Coastguard Worker // absolute value.
901*9880d681SAndroid Build Coastguard Worker uint64_t exp = n;
902*9880d681SAndroid Build Coastguard Worker
903*9880d681SAndroid Build Coastguard Worker // Return infinity for exponent overflow
904*9880d681SAndroid Build Coastguard Worker if (exp > 1023) {
905*9880d681SAndroid Build Coastguard Worker if (!isSigned || !isNeg)
906*9880d681SAndroid Build Coastguard Worker return std::numeric_limits<double>::infinity();
907*9880d681SAndroid Build Coastguard Worker else
908*9880d681SAndroid Build Coastguard Worker return -std::numeric_limits<double>::infinity();
909*9880d681SAndroid Build Coastguard Worker }
910*9880d681SAndroid Build Coastguard Worker exp += 1023; // Increment for 1023 bias
911*9880d681SAndroid Build Coastguard Worker
912*9880d681SAndroid Build Coastguard Worker // Number of bits in mantissa is 52. To obtain the mantissa value, we must
913*9880d681SAndroid Build Coastguard Worker // extract the high 52 bits from the correct words in pVal.
914*9880d681SAndroid Build Coastguard Worker uint64_t mantissa;
915*9880d681SAndroid Build Coastguard Worker unsigned hiWord = whichWord(n-1);
916*9880d681SAndroid Build Coastguard Worker if (hiWord == 0) {
917*9880d681SAndroid Build Coastguard Worker mantissa = Tmp.pVal[0];
918*9880d681SAndroid Build Coastguard Worker if (n > 52)
919*9880d681SAndroid Build Coastguard Worker mantissa >>= n - 52; // shift down, we want the top 52 bits.
920*9880d681SAndroid Build Coastguard Worker } else {
921*9880d681SAndroid Build Coastguard Worker assert(hiWord > 0 && "huh?");
922*9880d681SAndroid Build Coastguard Worker uint64_t hibits = Tmp.pVal[hiWord] << (52 - n % APINT_BITS_PER_WORD);
923*9880d681SAndroid Build Coastguard Worker uint64_t lobits = Tmp.pVal[hiWord-1] >> (11 + n % APINT_BITS_PER_WORD);
924*9880d681SAndroid Build Coastguard Worker mantissa = hibits | lobits;
925*9880d681SAndroid Build Coastguard Worker }
926*9880d681SAndroid Build Coastguard Worker
927*9880d681SAndroid Build Coastguard Worker // The leading bit of mantissa is implicit, so get rid of it.
928*9880d681SAndroid Build Coastguard Worker uint64_t sign = isNeg ? (1ULL << (APINT_BITS_PER_WORD - 1)) : 0;
929*9880d681SAndroid Build Coastguard Worker union {
930*9880d681SAndroid Build Coastguard Worker double D;
931*9880d681SAndroid Build Coastguard Worker uint64_t I;
932*9880d681SAndroid Build Coastguard Worker } T;
933*9880d681SAndroid Build Coastguard Worker T.I = sign | (exp << 52) | mantissa;
934*9880d681SAndroid Build Coastguard Worker return T.D;
935*9880d681SAndroid Build Coastguard Worker }
936*9880d681SAndroid Build Coastguard Worker
937*9880d681SAndroid Build Coastguard Worker // Truncate to new width.
trunc(unsigned width) const938*9880d681SAndroid Build Coastguard Worker APInt APInt::trunc(unsigned width) const {
939*9880d681SAndroid Build Coastguard Worker assert(width < BitWidth && "Invalid APInt Truncate request");
940*9880d681SAndroid Build Coastguard Worker assert(width && "Can't truncate to 0 bits");
941*9880d681SAndroid Build Coastguard Worker
942*9880d681SAndroid Build Coastguard Worker if (width <= APINT_BITS_PER_WORD)
943*9880d681SAndroid Build Coastguard Worker return APInt(width, getRawData()[0]);
944*9880d681SAndroid Build Coastguard Worker
945*9880d681SAndroid Build Coastguard Worker APInt Result(getMemory(getNumWords(width)), width);
946*9880d681SAndroid Build Coastguard Worker
947*9880d681SAndroid Build Coastguard Worker // Copy full words.
948*9880d681SAndroid Build Coastguard Worker unsigned i;
949*9880d681SAndroid Build Coastguard Worker for (i = 0; i != width / APINT_BITS_PER_WORD; i++)
950*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = pVal[i];
951*9880d681SAndroid Build Coastguard Worker
952*9880d681SAndroid Build Coastguard Worker // Truncate and copy any partial word.
953*9880d681SAndroid Build Coastguard Worker unsigned bits = (0 - width) % APINT_BITS_PER_WORD;
954*9880d681SAndroid Build Coastguard Worker if (bits != 0)
955*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = pVal[i] << bits >> bits;
956*9880d681SAndroid Build Coastguard Worker
957*9880d681SAndroid Build Coastguard Worker return Result;
958*9880d681SAndroid Build Coastguard Worker }
959*9880d681SAndroid Build Coastguard Worker
960*9880d681SAndroid Build Coastguard Worker // Sign extend to a new width.
sext(unsigned width) const961*9880d681SAndroid Build Coastguard Worker APInt APInt::sext(unsigned width) const {
962*9880d681SAndroid Build Coastguard Worker assert(width > BitWidth && "Invalid APInt SignExtend request");
963*9880d681SAndroid Build Coastguard Worker
964*9880d681SAndroid Build Coastguard Worker if (width <= APINT_BITS_PER_WORD) {
965*9880d681SAndroid Build Coastguard Worker uint64_t val = VAL << (APINT_BITS_PER_WORD - BitWidth);
966*9880d681SAndroid Build Coastguard Worker val = (int64_t)val >> (width - BitWidth);
967*9880d681SAndroid Build Coastguard Worker return APInt(width, val >> (APINT_BITS_PER_WORD - width));
968*9880d681SAndroid Build Coastguard Worker }
969*9880d681SAndroid Build Coastguard Worker
970*9880d681SAndroid Build Coastguard Worker APInt Result(getMemory(getNumWords(width)), width);
971*9880d681SAndroid Build Coastguard Worker
972*9880d681SAndroid Build Coastguard Worker // Copy full words.
973*9880d681SAndroid Build Coastguard Worker unsigned i;
974*9880d681SAndroid Build Coastguard Worker uint64_t word = 0;
975*9880d681SAndroid Build Coastguard Worker for (i = 0; i != BitWidth / APINT_BITS_PER_WORD; i++) {
976*9880d681SAndroid Build Coastguard Worker word = getRawData()[i];
977*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = word;
978*9880d681SAndroid Build Coastguard Worker }
979*9880d681SAndroid Build Coastguard Worker
980*9880d681SAndroid Build Coastguard Worker // Read and sign-extend any partial word.
981*9880d681SAndroid Build Coastguard Worker unsigned bits = (0 - BitWidth) % APINT_BITS_PER_WORD;
982*9880d681SAndroid Build Coastguard Worker if (bits != 0)
983*9880d681SAndroid Build Coastguard Worker word = (int64_t)getRawData()[i] << bits >> bits;
984*9880d681SAndroid Build Coastguard Worker else
985*9880d681SAndroid Build Coastguard Worker word = (int64_t)word >> (APINT_BITS_PER_WORD - 1);
986*9880d681SAndroid Build Coastguard Worker
987*9880d681SAndroid Build Coastguard Worker // Write remaining full words.
988*9880d681SAndroid Build Coastguard Worker for (; i != width / APINT_BITS_PER_WORD; i++) {
989*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = word;
990*9880d681SAndroid Build Coastguard Worker word = (int64_t)word >> (APINT_BITS_PER_WORD - 1);
991*9880d681SAndroid Build Coastguard Worker }
992*9880d681SAndroid Build Coastguard Worker
993*9880d681SAndroid Build Coastguard Worker // Write any partial word.
994*9880d681SAndroid Build Coastguard Worker bits = (0 - width) % APINT_BITS_PER_WORD;
995*9880d681SAndroid Build Coastguard Worker if (bits != 0)
996*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = word << bits >> bits;
997*9880d681SAndroid Build Coastguard Worker
998*9880d681SAndroid Build Coastguard Worker return Result;
999*9880d681SAndroid Build Coastguard Worker }
1000*9880d681SAndroid Build Coastguard Worker
1001*9880d681SAndroid Build Coastguard Worker // Zero extend to a new width.
zext(unsigned width) const1002*9880d681SAndroid Build Coastguard Worker APInt APInt::zext(unsigned width) const {
1003*9880d681SAndroid Build Coastguard Worker assert(width > BitWidth && "Invalid APInt ZeroExtend request");
1004*9880d681SAndroid Build Coastguard Worker
1005*9880d681SAndroid Build Coastguard Worker if (width <= APINT_BITS_PER_WORD)
1006*9880d681SAndroid Build Coastguard Worker return APInt(width, VAL);
1007*9880d681SAndroid Build Coastguard Worker
1008*9880d681SAndroid Build Coastguard Worker APInt Result(getMemory(getNumWords(width)), width);
1009*9880d681SAndroid Build Coastguard Worker
1010*9880d681SAndroid Build Coastguard Worker // Copy words.
1011*9880d681SAndroid Build Coastguard Worker unsigned i;
1012*9880d681SAndroid Build Coastguard Worker for (i = 0; i != getNumWords(); i++)
1013*9880d681SAndroid Build Coastguard Worker Result.pVal[i] = getRawData()[i];
1014*9880d681SAndroid Build Coastguard Worker
1015*9880d681SAndroid Build Coastguard Worker // Zero remaining words.
1016*9880d681SAndroid Build Coastguard Worker memset(&Result.pVal[i], 0, (Result.getNumWords() - i) * APINT_WORD_SIZE);
1017*9880d681SAndroid Build Coastguard Worker
1018*9880d681SAndroid Build Coastguard Worker return Result;
1019*9880d681SAndroid Build Coastguard Worker }
1020*9880d681SAndroid Build Coastguard Worker
zextOrTrunc(unsigned width) const1021*9880d681SAndroid Build Coastguard Worker APInt APInt::zextOrTrunc(unsigned width) const {
1022*9880d681SAndroid Build Coastguard Worker if (BitWidth < width)
1023*9880d681SAndroid Build Coastguard Worker return zext(width);
1024*9880d681SAndroid Build Coastguard Worker if (BitWidth > width)
1025*9880d681SAndroid Build Coastguard Worker return trunc(width);
1026*9880d681SAndroid Build Coastguard Worker return *this;
1027*9880d681SAndroid Build Coastguard Worker }
1028*9880d681SAndroid Build Coastguard Worker
sextOrTrunc(unsigned width) const1029*9880d681SAndroid Build Coastguard Worker APInt APInt::sextOrTrunc(unsigned width) const {
1030*9880d681SAndroid Build Coastguard Worker if (BitWidth < width)
1031*9880d681SAndroid Build Coastguard Worker return sext(width);
1032*9880d681SAndroid Build Coastguard Worker if (BitWidth > width)
1033*9880d681SAndroid Build Coastguard Worker return trunc(width);
1034*9880d681SAndroid Build Coastguard Worker return *this;
1035*9880d681SAndroid Build Coastguard Worker }
1036*9880d681SAndroid Build Coastguard Worker
zextOrSelf(unsigned width) const1037*9880d681SAndroid Build Coastguard Worker APInt APInt::zextOrSelf(unsigned width) const {
1038*9880d681SAndroid Build Coastguard Worker if (BitWidth < width)
1039*9880d681SAndroid Build Coastguard Worker return zext(width);
1040*9880d681SAndroid Build Coastguard Worker return *this;
1041*9880d681SAndroid Build Coastguard Worker }
1042*9880d681SAndroid Build Coastguard Worker
sextOrSelf(unsigned width) const1043*9880d681SAndroid Build Coastguard Worker APInt APInt::sextOrSelf(unsigned width) const {
1044*9880d681SAndroid Build Coastguard Worker if (BitWidth < width)
1045*9880d681SAndroid Build Coastguard Worker return sext(width);
1046*9880d681SAndroid Build Coastguard Worker return *this;
1047*9880d681SAndroid Build Coastguard Worker }
1048*9880d681SAndroid Build Coastguard Worker
1049*9880d681SAndroid Build Coastguard Worker /// Arithmetic right-shift this APInt by shiftAmt.
1050*9880d681SAndroid Build Coastguard Worker /// @brief Arithmetic right-shift function.
ashr(const APInt & shiftAmt) const1051*9880d681SAndroid Build Coastguard Worker APInt APInt::ashr(const APInt &shiftAmt) const {
1052*9880d681SAndroid Build Coastguard Worker return ashr((unsigned)shiftAmt.getLimitedValue(BitWidth));
1053*9880d681SAndroid Build Coastguard Worker }
1054*9880d681SAndroid Build Coastguard Worker
1055*9880d681SAndroid Build Coastguard Worker /// Arithmetic right-shift this APInt by shiftAmt.
1056*9880d681SAndroid Build Coastguard Worker /// @brief Arithmetic right-shift function.
ashr(unsigned shiftAmt) const1057*9880d681SAndroid Build Coastguard Worker APInt APInt::ashr(unsigned shiftAmt) const {
1058*9880d681SAndroid Build Coastguard Worker assert(shiftAmt <= BitWidth && "Invalid shift amount");
1059*9880d681SAndroid Build Coastguard Worker // Handle a degenerate case
1060*9880d681SAndroid Build Coastguard Worker if (shiftAmt == 0)
1061*9880d681SAndroid Build Coastguard Worker return *this;
1062*9880d681SAndroid Build Coastguard Worker
1063*9880d681SAndroid Build Coastguard Worker // Handle single word shifts with built-in ashr
1064*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
1065*9880d681SAndroid Build Coastguard Worker if (shiftAmt == BitWidth)
1066*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0); // undefined
1067*9880d681SAndroid Build Coastguard Worker else {
1068*9880d681SAndroid Build Coastguard Worker unsigned SignBit = APINT_BITS_PER_WORD - BitWidth;
1069*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth,
1070*9880d681SAndroid Build Coastguard Worker (((int64_t(VAL) << SignBit) >> SignBit) >> shiftAmt));
1071*9880d681SAndroid Build Coastguard Worker }
1072*9880d681SAndroid Build Coastguard Worker }
1073*9880d681SAndroid Build Coastguard Worker
1074*9880d681SAndroid Build Coastguard Worker // If all the bits were shifted out, the result is, technically, undefined.
1075*9880d681SAndroid Build Coastguard Worker // We return -1 if it was negative, 0 otherwise. We check this early to avoid
1076*9880d681SAndroid Build Coastguard Worker // issues in the algorithm below.
1077*9880d681SAndroid Build Coastguard Worker if (shiftAmt == BitWidth) {
1078*9880d681SAndroid Build Coastguard Worker if (isNegative())
1079*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, -1ULL, true);
1080*9880d681SAndroid Build Coastguard Worker else
1081*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1082*9880d681SAndroid Build Coastguard Worker }
1083*9880d681SAndroid Build Coastguard Worker
1084*9880d681SAndroid Build Coastguard Worker // Create some space for the result.
1085*9880d681SAndroid Build Coastguard Worker uint64_t * val = new uint64_t[getNumWords()];
1086*9880d681SAndroid Build Coastguard Worker
1087*9880d681SAndroid Build Coastguard Worker // Compute some values needed by the following shift algorithms
1088*9880d681SAndroid Build Coastguard Worker unsigned wordShift = shiftAmt % APINT_BITS_PER_WORD; // bits to shift per word
1089*9880d681SAndroid Build Coastguard Worker unsigned offset = shiftAmt / APINT_BITS_PER_WORD; // word offset for shift
1090*9880d681SAndroid Build Coastguard Worker unsigned breakWord = getNumWords() - 1 - offset; // last word affected
1091*9880d681SAndroid Build Coastguard Worker unsigned bitsInWord = whichBit(BitWidth); // how many bits in last word?
1092*9880d681SAndroid Build Coastguard Worker if (bitsInWord == 0)
1093*9880d681SAndroid Build Coastguard Worker bitsInWord = APINT_BITS_PER_WORD;
1094*9880d681SAndroid Build Coastguard Worker
1095*9880d681SAndroid Build Coastguard Worker // If we are shifting whole words, just move whole words
1096*9880d681SAndroid Build Coastguard Worker if (wordShift == 0) {
1097*9880d681SAndroid Build Coastguard Worker // Move the words containing significant bits
1098*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i <= breakWord; ++i)
1099*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i+offset]; // move whole word
1100*9880d681SAndroid Build Coastguard Worker
1101*9880d681SAndroid Build Coastguard Worker // Adjust the top significant word for sign bit fill, if negative
1102*9880d681SAndroid Build Coastguard Worker if (isNegative())
1103*9880d681SAndroid Build Coastguard Worker if (bitsInWord < APINT_BITS_PER_WORD)
1104*9880d681SAndroid Build Coastguard Worker val[breakWord] |= ~0ULL << bitsInWord; // set high bits
1105*9880d681SAndroid Build Coastguard Worker } else {
1106*9880d681SAndroid Build Coastguard Worker // Shift the low order words
1107*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < breakWord; ++i) {
1108*9880d681SAndroid Build Coastguard Worker // This combines the shifted corresponding word with the low bits from
1109*9880d681SAndroid Build Coastguard Worker // the next word (shifted into this word's high bits).
1110*9880d681SAndroid Build Coastguard Worker val[i] = (pVal[i+offset] >> wordShift) |
1111*9880d681SAndroid Build Coastguard Worker (pVal[i+offset+1] << (APINT_BITS_PER_WORD - wordShift));
1112*9880d681SAndroid Build Coastguard Worker }
1113*9880d681SAndroid Build Coastguard Worker
1114*9880d681SAndroid Build Coastguard Worker // Shift the break word. In this case there are no bits from the next word
1115*9880d681SAndroid Build Coastguard Worker // to include in this word.
1116*9880d681SAndroid Build Coastguard Worker val[breakWord] = pVal[breakWord+offset] >> wordShift;
1117*9880d681SAndroid Build Coastguard Worker
1118*9880d681SAndroid Build Coastguard Worker // Deal with sign extension in the break word, and possibly the word before
1119*9880d681SAndroid Build Coastguard Worker // it.
1120*9880d681SAndroid Build Coastguard Worker if (isNegative()) {
1121*9880d681SAndroid Build Coastguard Worker if (wordShift > bitsInWord) {
1122*9880d681SAndroid Build Coastguard Worker if (breakWord > 0)
1123*9880d681SAndroid Build Coastguard Worker val[breakWord-1] |=
1124*9880d681SAndroid Build Coastguard Worker ~0ULL << (APINT_BITS_PER_WORD - (wordShift - bitsInWord));
1125*9880d681SAndroid Build Coastguard Worker val[breakWord] |= ~0ULL;
1126*9880d681SAndroid Build Coastguard Worker } else
1127*9880d681SAndroid Build Coastguard Worker val[breakWord] |= (~0ULL << (bitsInWord - wordShift));
1128*9880d681SAndroid Build Coastguard Worker }
1129*9880d681SAndroid Build Coastguard Worker }
1130*9880d681SAndroid Build Coastguard Worker
1131*9880d681SAndroid Build Coastguard Worker // Remaining words are 0 or -1, just assign them.
1132*9880d681SAndroid Build Coastguard Worker uint64_t fillValue = (isNegative() ? -1ULL : 0);
1133*9880d681SAndroid Build Coastguard Worker for (unsigned i = breakWord+1; i < getNumWords(); ++i)
1134*9880d681SAndroid Build Coastguard Worker val[i] = fillValue;
1135*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1136*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1137*9880d681SAndroid Build Coastguard Worker return Result;
1138*9880d681SAndroid Build Coastguard Worker }
1139*9880d681SAndroid Build Coastguard Worker
1140*9880d681SAndroid Build Coastguard Worker /// Logical right-shift this APInt by shiftAmt.
1141*9880d681SAndroid Build Coastguard Worker /// @brief Logical right-shift function.
lshr(const APInt & shiftAmt) const1142*9880d681SAndroid Build Coastguard Worker APInt APInt::lshr(const APInt &shiftAmt) const {
1143*9880d681SAndroid Build Coastguard Worker return lshr((unsigned)shiftAmt.getLimitedValue(BitWidth));
1144*9880d681SAndroid Build Coastguard Worker }
1145*9880d681SAndroid Build Coastguard Worker
1146*9880d681SAndroid Build Coastguard Worker /// Logical right-shift this APInt by shiftAmt.
1147*9880d681SAndroid Build Coastguard Worker /// @brief Logical right-shift function.
lshr(unsigned shiftAmt) const1148*9880d681SAndroid Build Coastguard Worker APInt APInt::lshr(unsigned shiftAmt) const {
1149*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
1150*9880d681SAndroid Build Coastguard Worker if (shiftAmt >= BitWidth)
1151*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1152*9880d681SAndroid Build Coastguard Worker else
1153*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, this->VAL >> shiftAmt);
1154*9880d681SAndroid Build Coastguard Worker }
1155*9880d681SAndroid Build Coastguard Worker
1156*9880d681SAndroid Build Coastguard Worker // If all the bits were shifted out, the result is 0. This avoids issues
1157*9880d681SAndroid Build Coastguard Worker // with shifting by the size of the integer type, which produces undefined
1158*9880d681SAndroid Build Coastguard Worker // results. We define these "undefined results" to always be 0.
1159*9880d681SAndroid Build Coastguard Worker if (shiftAmt >= BitWidth)
1160*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1161*9880d681SAndroid Build Coastguard Worker
1162*9880d681SAndroid Build Coastguard Worker // If none of the bits are shifted out, the result is *this. This avoids
1163*9880d681SAndroid Build Coastguard Worker // issues with shifting by the size of the integer type, which produces
1164*9880d681SAndroid Build Coastguard Worker // undefined results in the code below. This is also an optimization.
1165*9880d681SAndroid Build Coastguard Worker if (shiftAmt == 0)
1166*9880d681SAndroid Build Coastguard Worker return *this;
1167*9880d681SAndroid Build Coastguard Worker
1168*9880d681SAndroid Build Coastguard Worker // Create some space for the result.
1169*9880d681SAndroid Build Coastguard Worker uint64_t * val = new uint64_t[getNumWords()];
1170*9880d681SAndroid Build Coastguard Worker
1171*9880d681SAndroid Build Coastguard Worker // If we are shifting less than a word, compute the shift with a simple carry
1172*9880d681SAndroid Build Coastguard Worker if (shiftAmt < APINT_BITS_PER_WORD) {
1173*9880d681SAndroid Build Coastguard Worker lshrNear(val, pVal, getNumWords(), shiftAmt);
1174*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1175*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1176*9880d681SAndroid Build Coastguard Worker return Result;
1177*9880d681SAndroid Build Coastguard Worker }
1178*9880d681SAndroid Build Coastguard Worker
1179*9880d681SAndroid Build Coastguard Worker // Compute some values needed by the remaining shift algorithms
1180*9880d681SAndroid Build Coastguard Worker unsigned wordShift = shiftAmt % APINT_BITS_PER_WORD;
1181*9880d681SAndroid Build Coastguard Worker unsigned offset = shiftAmt / APINT_BITS_PER_WORD;
1182*9880d681SAndroid Build Coastguard Worker
1183*9880d681SAndroid Build Coastguard Worker // If we are shifting whole words, just move whole words
1184*9880d681SAndroid Build Coastguard Worker if (wordShift == 0) {
1185*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < getNumWords() - offset; ++i)
1186*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i+offset];
1187*9880d681SAndroid Build Coastguard Worker for (unsigned i = getNumWords()-offset; i < getNumWords(); i++)
1188*9880d681SAndroid Build Coastguard Worker val[i] = 0;
1189*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1190*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1191*9880d681SAndroid Build Coastguard Worker return Result;
1192*9880d681SAndroid Build Coastguard Worker }
1193*9880d681SAndroid Build Coastguard Worker
1194*9880d681SAndroid Build Coastguard Worker // Shift the low order words
1195*9880d681SAndroid Build Coastguard Worker unsigned breakWord = getNumWords() - offset -1;
1196*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < breakWord; ++i)
1197*9880d681SAndroid Build Coastguard Worker val[i] = (pVal[i+offset] >> wordShift) |
1198*9880d681SAndroid Build Coastguard Worker (pVal[i+offset+1] << (APINT_BITS_PER_WORD - wordShift));
1199*9880d681SAndroid Build Coastguard Worker // Shift the break word.
1200*9880d681SAndroid Build Coastguard Worker val[breakWord] = pVal[breakWord+offset] >> wordShift;
1201*9880d681SAndroid Build Coastguard Worker
1202*9880d681SAndroid Build Coastguard Worker // Remaining words are 0
1203*9880d681SAndroid Build Coastguard Worker for (unsigned i = breakWord+1; i < getNumWords(); ++i)
1204*9880d681SAndroid Build Coastguard Worker val[i] = 0;
1205*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1206*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1207*9880d681SAndroid Build Coastguard Worker return Result;
1208*9880d681SAndroid Build Coastguard Worker }
1209*9880d681SAndroid Build Coastguard Worker
1210*9880d681SAndroid Build Coastguard Worker /// Left-shift this APInt by shiftAmt.
1211*9880d681SAndroid Build Coastguard Worker /// @brief Left-shift function.
shl(const APInt & shiftAmt) const1212*9880d681SAndroid Build Coastguard Worker APInt APInt::shl(const APInt &shiftAmt) const {
1213*9880d681SAndroid Build Coastguard Worker // It's undefined behavior in C to shift by BitWidth or greater.
1214*9880d681SAndroid Build Coastguard Worker return shl((unsigned)shiftAmt.getLimitedValue(BitWidth));
1215*9880d681SAndroid Build Coastguard Worker }
1216*9880d681SAndroid Build Coastguard Worker
shlSlowCase(unsigned shiftAmt) const1217*9880d681SAndroid Build Coastguard Worker APInt APInt::shlSlowCase(unsigned shiftAmt) const {
1218*9880d681SAndroid Build Coastguard Worker // If all the bits were shifted out, the result is 0. This avoids issues
1219*9880d681SAndroid Build Coastguard Worker // with shifting by the size of the integer type, which produces undefined
1220*9880d681SAndroid Build Coastguard Worker // results. We define these "undefined results" to always be 0.
1221*9880d681SAndroid Build Coastguard Worker if (shiftAmt == BitWidth)
1222*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1223*9880d681SAndroid Build Coastguard Worker
1224*9880d681SAndroid Build Coastguard Worker // If none of the bits are shifted out, the result is *this. This avoids a
1225*9880d681SAndroid Build Coastguard Worker // lshr by the words size in the loop below which can produce incorrect
1226*9880d681SAndroid Build Coastguard Worker // results. It also avoids the expensive computation below for a common case.
1227*9880d681SAndroid Build Coastguard Worker if (shiftAmt == 0)
1228*9880d681SAndroid Build Coastguard Worker return *this;
1229*9880d681SAndroid Build Coastguard Worker
1230*9880d681SAndroid Build Coastguard Worker // Create some space for the result.
1231*9880d681SAndroid Build Coastguard Worker uint64_t * val = new uint64_t[getNumWords()];
1232*9880d681SAndroid Build Coastguard Worker
1233*9880d681SAndroid Build Coastguard Worker // If we are shifting less than a word, do it the easy way
1234*9880d681SAndroid Build Coastguard Worker if (shiftAmt < APINT_BITS_PER_WORD) {
1235*9880d681SAndroid Build Coastguard Worker uint64_t carry = 0;
1236*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < getNumWords(); i++) {
1237*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i] << shiftAmt | carry;
1238*9880d681SAndroid Build Coastguard Worker carry = pVal[i] >> (APINT_BITS_PER_WORD - shiftAmt);
1239*9880d681SAndroid Build Coastguard Worker }
1240*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1241*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1242*9880d681SAndroid Build Coastguard Worker return Result;
1243*9880d681SAndroid Build Coastguard Worker }
1244*9880d681SAndroid Build Coastguard Worker
1245*9880d681SAndroid Build Coastguard Worker // Compute some values needed by the remaining shift algorithms
1246*9880d681SAndroid Build Coastguard Worker unsigned wordShift = shiftAmt % APINT_BITS_PER_WORD;
1247*9880d681SAndroid Build Coastguard Worker unsigned offset = shiftAmt / APINT_BITS_PER_WORD;
1248*9880d681SAndroid Build Coastguard Worker
1249*9880d681SAndroid Build Coastguard Worker // If we are shifting whole words, just move whole words
1250*9880d681SAndroid Build Coastguard Worker if (wordShift == 0) {
1251*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < offset; i++)
1252*9880d681SAndroid Build Coastguard Worker val[i] = 0;
1253*9880d681SAndroid Build Coastguard Worker for (unsigned i = offset; i < getNumWords(); i++)
1254*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i-offset];
1255*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1256*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1257*9880d681SAndroid Build Coastguard Worker return Result;
1258*9880d681SAndroid Build Coastguard Worker }
1259*9880d681SAndroid Build Coastguard Worker
1260*9880d681SAndroid Build Coastguard Worker // Copy whole words from this to Result.
1261*9880d681SAndroid Build Coastguard Worker unsigned i = getNumWords() - 1;
1262*9880d681SAndroid Build Coastguard Worker for (; i > offset; --i)
1263*9880d681SAndroid Build Coastguard Worker val[i] = pVal[i-offset] << wordShift |
1264*9880d681SAndroid Build Coastguard Worker pVal[i-offset-1] >> (APINT_BITS_PER_WORD - wordShift);
1265*9880d681SAndroid Build Coastguard Worker val[offset] = pVal[0] << wordShift;
1266*9880d681SAndroid Build Coastguard Worker for (i = 0; i < offset; ++i)
1267*9880d681SAndroid Build Coastguard Worker val[i] = 0;
1268*9880d681SAndroid Build Coastguard Worker APInt Result(val, BitWidth);
1269*9880d681SAndroid Build Coastguard Worker Result.clearUnusedBits();
1270*9880d681SAndroid Build Coastguard Worker return Result;
1271*9880d681SAndroid Build Coastguard Worker }
1272*9880d681SAndroid Build Coastguard Worker
rotl(const APInt & rotateAmt) const1273*9880d681SAndroid Build Coastguard Worker APInt APInt::rotl(const APInt &rotateAmt) const {
1274*9880d681SAndroid Build Coastguard Worker return rotl((unsigned)rotateAmt.getLimitedValue(BitWidth));
1275*9880d681SAndroid Build Coastguard Worker }
1276*9880d681SAndroid Build Coastguard Worker
rotl(unsigned rotateAmt) const1277*9880d681SAndroid Build Coastguard Worker APInt APInt::rotl(unsigned rotateAmt) const {
1278*9880d681SAndroid Build Coastguard Worker rotateAmt %= BitWidth;
1279*9880d681SAndroid Build Coastguard Worker if (rotateAmt == 0)
1280*9880d681SAndroid Build Coastguard Worker return *this;
1281*9880d681SAndroid Build Coastguard Worker return shl(rotateAmt) | lshr(BitWidth - rotateAmt);
1282*9880d681SAndroid Build Coastguard Worker }
1283*9880d681SAndroid Build Coastguard Worker
rotr(const APInt & rotateAmt) const1284*9880d681SAndroid Build Coastguard Worker APInt APInt::rotr(const APInt &rotateAmt) const {
1285*9880d681SAndroid Build Coastguard Worker return rotr((unsigned)rotateAmt.getLimitedValue(BitWidth));
1286*9880d681SAndroid Build Coastguard Worker }
1287*9880d681SAndroid Build Coastguard Worker
rotr(unsigned rotateAmt) const1288*9880d681SAndroid Build Coastguard Worker APInt APInt::rotr(unsigned rotateAmt) const {
1289*9880d681SAndroid Build Coastguard Worker rotateAmt %= BitWidth;
1290*9880d681SAndroid Build Coastguard Worker if (rotateAmt == 0)
1291*9880d681SAndroid Build Coastguard Worker return *this;
1292*9880d681SAndroid Build Coastguard Worker return lshr(rotateAmt) | shl(BitWidth - rotateAmt);
1293*9880d681SAndroid Build Coastguard Worker }
1294*9880d681SAndroid Build Coastguard Worker
1295*9880d681SAndroid Build Coastguard Worker // Square Root - this method computes and returns the square root of "this".
1296*9880d681SAndroid Build Coastguard Worker // Three mechanisms are used for computation. For small values (<= 5 bits),
1297*9880d681SAndroid Build Coastguard Worker // a table lookup is done. This gets some performance for common cases. For
1298*9880d681SAndroid Build Coastguard Worker // values using less than 52 bits, the value is converted to double and then
1299*9880d681SAndroid Build Coastguard Worker // the libc sqrt function is called. The result is rounded and then converted
1300*9880d681SAndroid Build Coastguard Worker // back to a uint64_t which is then used to construct the result. Finally,
1301*9880d681SAndroid Build Coastguard Worker // the Babylonian method for computing square roots is used.
sqrt() const1302*9880d681SAndroid Build Coastguard Worker APInt APInt::sqrt() const {
1303*9880d681SAndroid Build Coastguard Worker
1304*9880d681SAndroid Build Coastguard Worker // Determine the magnitude of the value.
1305*9880d681SAndroid Build Coastguard Worker unsigned magnitude = getActiveBits();
1306*9880d681SAndroid Build Coastguard Worker
1307*9880d681SAndroid Build Coastguard Worker // Use a fast table for some small values. This also gets rid of some
1308*9880d681SAndroid Build Coastguard Worker // rounding errors in libc sqrt for small values.
1309*9880d681SAndroid Build Coastguard Worker if (magnitude <= 5) {
1310*9880d681SAndroid Build Coastguard Worker static const uint8_t results[32] = {
1311*9880d681SAndroid Build Coastguard Worker /* 0 */ 0,
1312*9880d681SAndroid Build Coastguard Worker /* 1- 2 */ 1, 1,
1313*9880d681SAndroid Build Coastguard Worker /* 3- 6 */ 2, 2, 2, 2,
1314*9880d681SAndroid Build Coastguard Worker /* 7-12 */ 3, 3, 3, 3, 3, 3,
1315*9880d681SAndroid Build Coastguard Worker /* 13-20 */ 4, 4, 4, 4, 4, 4, 4, 4,
1316*9880d681SAndroid Build Coastguard Worker /* 21-30 */ 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1317*9880d681SAndroid Build Coastguard Worker /* 31 */ 6
1318*9880d681SAndroid Build Coastguard Worker };
1319*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, results[ (isSingleWord() ? VAL : pVal[0]) ]);
1320*9880d681SAndroid Build Coastguard Worker }
1321*9880d681SAndroid Build Coastguard Worker
1322*9880d681SAndroid Build Coastguard Worker // If the magnitude of the value fits in less than 52 bits (the precision of
1323*9880d681SAndroid Build Coastguard Worker // an IEEE double precision floating point value), then we can use the
1324*9880d681SAndroid Build Coastguard Worker // libc sqrt function which will probably use a hardware sqrt computation.
1325*9880d681SAndroid Build Coastguard Worker // This should be faster than the algorithm below.
1326*9880d681SAndroid Build Coastguard Worker if (magnitude < 52) {
1327*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth,
1328*9880d681SAndroid Build Coastguard Worker uint64_t(::round(::sqrt(double(isSingleWord()?VAL:pVal[0])))));
1329*9880d681SAndroid Build Coastguard Worker }
1330*9880d681SAndroid Build Coastguard Worker
1331*9880d681SAndroid Build Coastguard Worker // Okay, all the short cuts are exhausted. We must compute it. The following
1332*9880d681SAndroid Build Coastguard Worker // is a classical Babylonian method for computing the square root. This code
1333*9880d681SAndroid Build Coastguard Worker // was adapted to APInt from a wikipedia article on such computations.
1334*9880d681SAndroid Build Coastguard Worker // See http://www.wikipedia.org/ and go to the page named
1335*9880d681SAndroid Build Coastguard Worker // Calculate_an_integer_square_root.
1336*9880d681SAndroid Build Coastguard Worker unsigned nbits = BitWidth, i = 4;
1337*9880d681SAndroid Build Coastguard Worker APInt testy(BitWidth, 16);
1338*9880d681SAndroid Build Coastguard Worker APInt x_old(BitWidth, 1);
1339*9880d681SAndroid Build Coastguard Worker APInt x_new(BitWidth, 0);
1340*9880d681SAndroid Build Coastguard Worker APInt two(BitWidth, 2);
1341*9880d681SAndroid Build Coastguard Worker
1342*9880d681SAndroid Build Coastguard Worker // Select a good starting value using binary logarithms.
1343*9880d681SAndroid Build Coastguard Worker for (;; i += 2, testy = testy.shl(2))
1344*9880d681SAndroid Build Coastguard Worker if (i >= nbits || this->ule(testy)) {
1345*9880d681SAndroid Build Coastguard Worker x_old = x_old.shl(i / 2);
1346*9880d681SAndroid Build Coastguard Worker break;
1347*9880d681SAndroid Build Coastguard Worker }
1348*9880d681SAndroid Build Coastguard Worker
1349*9880d681SAndroid Build Coastguard Worker // Use the Babylonian method to arrive at the integer square root:
1350*9880d681SAndroid Build Coastguard Worker for (;;) {
1351*9880d681SAndroid Build Coastguard Worker x_new = (this->udiv(x_old) + x_old).udiv(two);
1352*9880d681SAndroid Build Coastguard Worker if (x_old.ule(x_new))
1353*9880d681SAndroid Build Coastguard Worker break;
1354*9880d681SAndroid Build Coastguard Worker x_old = x_new;
1355*9880d681SAndroid Build Coastguard Worker }
1356*9880d681SAndroid Build Coastguard Worker
1357*9880d681SAndroid Build Coastguard Worker // Make sure we return the closest approximation
1358*9880d681SAndroid Build Coastguard Worker // NOTE: The rounding calculation below is correct. It will produce an
1359*9880d681SAndroid Build Coastguard Worker // off-by-one discrepancy with results from pari/gp. That discrepancy has been
1360*9880d681SAndroid Build Coastguard Worker // determined to be a rounding issue with pari/gp as it begins to use a
1361*9880d681SAndroid Build Coastguard Worker // floating point representation after 192 bits. There are no discrepancies
1362*9880d681SAndroid Build Coastguard Worker // between this algorithm and pari/gp for bit widths < 192 bits.
1363*9880d681SAndroid Build Coastguard Worker APInt square(x_old * x_old);
1364*9880d681SAndroid Build Coastguard Worker APInt nextSquare((x_old + 1) * (x_old +1));
1365*9880d681SAndroid Build Coastguard Worker if (this->ult(square))
1366*9880d681SAndroid Build Coastguard Worker return x_old;
1367*9880d681SAndroid Build Coastguard Worker assert(this->ule(nextSquare) && "Error in APInt::sqrt computation");
1368*9880d681SAndroid Build Coastguard Worker APInt midpoint((nextSquare - square).udiv(two));
1369*9880d681SAndroid Build Coastguard Worker APInt offset(*this - square);
1370*9880d681SAndroid Build Coastguard Worker if (offset.ult(midpoint))
1371*9880d681SAndroid Build Coastguard Worker return x_old;
1372*9880d681SAndroid Build Coastguard Worker return x_old + 1;
1373*9880d681SAndroid Build Coastguard Worker }
1374*9880d681SAndroid Build Coastguard Worker
1375*9880d681SAndroid Build Coastguard Worker /// Computes the multiplicative inverse of this APInt for a given modulo. The
1376*9880d681SAndroid Build Coastguard Worker /// iterative extended Euclidean algorithm is used to solve for this value,
1377*9880d681SAndroid Build Coastguard Worker /// however we simplify it to speed up calculating only the inverse, and take
1378*9880d681SAndroid Build Coastguard Worker /// advantage of div+rem calculations. We also use some tricks to avoid copying
1379*9880d681SAndroid Build Coastguard Worker /// (potentially large) APInts around.
multiplicativeInverse(const APInt & modulo) const1380*9880d681SAndroid Build Coastguard Worker APInt APInt::multiplicativeInverse(const APInt& modulo) const {
1381*9880d681SAndroid Build Coastguard Worker assert(ult(modulo) && "This APInt must be smaller than the modulo");
1382*9880d681SAndroid Build Coastguard Worker
1383*9880d681SAndroid Build Coastguard Worker // Using the properties listed at the following web page (accessed 06/21/08):
1384*9880d681SAndroid Build Coastguard Worker // http://www.numbertheory.org/php/euclid.html
1385*9880d681SAndroid Build Coastguard Worker // (especially the properties numbered 3, 4 and 9) it can be proved that
1386*9880d681SAndroid Build Coastguard Worker // BitWidth bits suffice for all the computations in the algorithm implemented
1387*9880d681SAndroid Build Coastguard Worker // below. More precisely, this number of bits suffice if the multiplicative
1388*9880d681SAndroid Build Coastguard Worker // inverse exists, but may not suffice for the general extended Euclidean
1389*9880d681SAndroid Build Coastguard Worker // algorithm.
1390*9880d681SAndroid Build Coastguard Worker
1391*9880d681SAndroid Build Coastguard Worker APInt r[2] = { modulo, *this };
1392*9880d681SAndroid Build Coastguard Worker APInt t[2] = { APInt(BitWidth, 0), APInt(BitWidth, 1) };
1393*9880d681SAndroid Build Coastguard Worker APInt q(BitWidth, 0);
1394*9880d681SAndroid Build Coastguard Worker
1395*9880d681SAndroid Build Coastguard Worker unsigned i;
1396*9880d681SAndroid Build Coastguard Worker for (i = 0; r[i^1] != 0; i ^= 1) {
1397*9880d681SAndroid Build Coastguard Worker // An overview of the math without the confusing bit-flipping:
1398*9880d681SAndroid Build Coastguard Worker // q = r[i-2] / r[i-1]
1399*9880d681SAndroid Build Coastguard Worker // r[i] = r[i-2] % r[i-1]
1400*9880d681SAndroid Build Coastguard Worker // t[i] = t[i-2] - t[i-1] * q
1401*9880d681SAndroid Build Coastguard Worker udivrem(r[i], r[i^1], q, r[i]);
1402*9880d681SAndroid Build Coastguard Worker t[i] -= t[i^1] * q;
1403*9880d681SAndroid Build Coastguard Worker }
1404*9880d681SAndroid Build Coastguard Worker
1405*9880d681SAndroid Build Coastguard Worker // If this APInt and the modulo are not coprime, there is no multiplicative
1406*9880d681SAndroid Build Coastguard Worker // inverse, so return 0. We check this by looking at the next-to-last
1407*9880d681SAndroid Build Coastguard Worker // remainder, which is the gcd(*this,modulo) as calculated by the Euclidean
1408*9880d681SAndroid Build Coastguard Worker // algorithm.
1409*9880d681SAndroid Build Coastguard Worker if (r[i] != 1)
1410*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1411*9880d681SAndroid Build Coastguard Worker
1412*9880d681SAndroid Build Coastguard Worker // The next-to-last t is the multiplicative inverse. However, we are
1413*9880d681SAndroid Build Coastguard Worker // interested in a positive inverse. Calcuate a positive one from a negative
1414*9880d681SAndroid Build Coastguard Worker // one if necessary. A simple addition of the modulo suffices because
1415*9880d681SAndroid Build Coastguard Worker // abs(t[i]) is known to be less than *this/2 (see the link above).
1416*9880d681SAndroid Build Coastguard Worker return t[i].isNegative() ? t[i] + modulo : t[i];
1417*9880d681SAndroid Build Coastguard Worker }
1418*9880d681SAndroid Build Coastguard Worker
1419*9880d681SAndroid Build Coastguard Worker /// Calculate the magic numbers required to implement a signed integer division
1420*9880d681SAndroid Build Coastguard Worker /// by a constant as a sequence of multiplies, adds and shifts. Requires that
1421*9880d681SAndroid Build Coastguard Worker /// the divisor not be 0, 1, or -1. Taken from "Hacker's Delight", Henry S.
1422*9880d681SAndroid Build Coastguard Worker /// Warren, Jr., chapter 10.
magic() const1423*9880d681SAndroid Build Coastguard Worker APInt::ms APInt::magic() const {
1424*9880d681SAndroid Build Coastguard Worker const APInt& d = *this;
1425*9880d681SAndroid Build Coastguard Worker unsigned p;
1426*9880d681SAndroid Build Coastguard Worker APInt ad, anc, delta, q1, r1, q2, r2, t;
1427*9880d681SAndroid Build Coastguard Worker APInt signedMin = APInt::getSignedMinValue(d.getBitWidth());
1428*9880d681SAndroid Build Coastguard Worker struct ms mag;
1429*9880d681SAndroid Build Coastguard Worker
1430*9880d681SAndroid Build Coastguard Worker ad = d.abs();
1431*9880d681SAndroid Build Coastguard Worker t = signedMin + (d.lshr(d.getBitWidth() - 1));
1432*9880d681SAndroid Build Coastguard Worker anc = t - 1 - t.urem(ad); // absolute value of nc
1433*9880d681SAndroid Build Coastguard Worker p = d.getBitWidth() - 1; // initialize p
1434*9880d681SAndroid Build Coastguard Worker q1 = signedMin.udiv(anc); // initialize q1 = 2p/abs(nc)
1435*9880d681SAndroid Build Coastguard Worker r1 = signedMin - q1*anc; // initialize r1 = rem(2p,abs(nc))
1436*9880d681SAndroid Build Coastguard Worker q2 = signedMin.udiv(ad); // initialize q2 = 2p/abs(d)
1437*9880d681SAndroid Build Coastguard Worker r2 = signedMin - q2*ad; // initialize r2 = rem(2p,abs(d))
1438*9880d681SAndroid Build Coastguard Worker do {
1439*9880d681SAndroid Build Coastguard Worker p = p + 1;
1440*9880d681SAndroid Build Coastguard Worker q1 = q1<<1; // update q1 = 2p/abs(nc)
1441*9880d681SAndroid Build Coastguard Worker r1 = r1<<1; // update r1 = rem(2p/abs(nc))
1442*9880d681SAndroid Build Coastguard Worker if (r1.uge(anc)) { // must be unsigned comparison
1443*9880d681SAndroid Build Coastguard Worker q1 = q1 + 1;
1444*9880d681SAndroid Build Coastguard Worker r1 = r1 - anc;
1445*9880d681SAndroid Build Coastguard Worker }
1446*9880d681SAndroid Build Coastguard Worker q2 = q2<<1; // update q2 = 2p/abs(d)
1447*9880d681SAndroid Build Coastguard Worker r2 = r2<<1; // update r2 = rem(2p/abs(d))
1448*9880d681SAndroid Build Coastguard Worker if (r2.uge(ad)) { // must be unsigned comparison
1449*9880d681SAndroid Build Coastguard Worker q2 = q2 + 1;
1450*9880d681SAndroid Build Coastguard Worker r2 = r2 - ad;
1451*9880d681SAndroid Build Coastguard Worker }
1452*9880d681SAndroid Build Coastguard Worker delta = ad - r2;
1453*9880d681SAndroid Build Coastguard Worker } while (q1.ult(delta) || (q1 == delta && r1 == 0));
1454*9880d681SAndroid Build Coastguard Worker
1455*9880d681SAndroid Build Coastguard Worker mag.m = q2 + 1;
1456*9880d681SAndroid Build Coastguard Worker if (d.isNegative()) mag.m = -mag.m; // resulting magic number
1457*9880d681SAndroid Build Coastguard Worker mag.s = p - d.getBitWidth(); // resulting shift
1458*9880d681SAndroid Build Coastguard Worker return mag;
1459*9880d681SAndroid Build Coastguard Worker }
1460*9880d681SAndroid Build Coastguard Worker
1461*9880d681SAndroid Build Coastguard Worker /// Calculate the magic numbers required to implement an unsigned integer
1462*9880d681SAndroid Build Coastguard Worker /// division by a constant as a sequence of multiplies, adds and shifts.
1463*9880d681SAndroid Build Coastguard Worker /// Requires that the divisor not be 0. Taken from "Hacker's Delight", Henry
1464*9880d681SAndroid Build Coastguard Worker /// S. Warren, Jr., chapter 10.
1465*9880d681SAndroid Build Coastguard Worker /// LeadingZeros can be used to simplify the calculation if the upper bits
1466*9880d681SAndroid Build Coastguard Worker /// of the divided value are known zero.
magicu(unsigned LeadingZeros) const1467*9880d681SAndroid Build Coastguard Worker APInt::mu APInt::magicu(unsigned LeadingZeros) const {
1468*9880d681SAndroid Build Coastguard Worker const APInt& d = *this;
1469*9880d681SAndroid Build Coastguard Worker unsigned p;
1470*9880d681SAndroid Build Coastguard Worker APInt nc, delta, q1, r1, q2, r2;
1471*9880d681SAndroid Build Coastguard Worker struct mu magu;
1472*9880d681SAndroid Build Coastguard Worker magu.a = 0; // initialize "add" indicator
1473*9880d681SAndroid Build Coastguard Worker APInt allOnes = APInt::getAllOnesValue(d.getBitWidth()).lshr(LeadingZeros);
1474*9880d681SAndroid Build Coastguard Worker APInt signedMin = APInt::getSignedMinValue(d.getBitWidth());
1475*9880d681SAndroid Build Coastguard Worker APInt signedMax = APInt::getSignedMaxValue(d.getBitWidth());
1476*9880d681SAndroid Build Coastguard Worker
1477*9880d681SAndroid Build Coastguard Worker nc = allOnes - (allOnes - d).urem(d);
1478*9880d681SAndroid Build Coastguard Worker p = d.getBitWidth() - 1; // initialize p
1479*9880d681SAndroid Build Coastguard Worker q1 = signedMin.udiv(nc); // initialize q1 = 2p/nc
1480*9880d681SAndroid Build Coastguard Worker r1 = signedMin - q1*nc; // initialize r1 = rem(2p,nc)
1481*9880d681SAndroid Build Coastguard Worker q2 = signedMax.udiv(d); // initialize q2 = (2p-1)/d
1482*9880d681SAndroid Build Coastguard Worker r2 = signedMax - q2*d; // initialize r2 = rem((2p-1),d)
1483*9880d681SAndroid Build Coastguard Worker do {
1484*9880d681SAndroid Build Coastguard Worker p = p + 1;
1485*9880d681SAndroid Build Coastguard Worker if (r1.uge(nc - r1)) {
1486*9880d681SAndroid Build Coastguard Worker q1 = q1 + q1 + 1; // update q1
1487*9880d681SAndroid Build Coastguard Worker r1 = r1 + r1 - nc; // update r1
1488*9880d681SAndroid Build Coastguard Worker }
1489*9880d681SAndroid Build Coastguard Worker else {
1490*9880d681SAndroid Build Coastguard Worker q1 = q1+q1; // update q1
1491*9880d681SAndroid Build Coastguard Worker r1 = r1+r1; // update r1
1492*9880d681SAndroid Build Coastguard Worker }
1493*9880d681SAndroid Build Coastguard Worker if ((r2 + 1).uge(d - r2)) {
1494*9880d681SAndroid Build Coastguard Worker if (q2.uge(signedMax)) magu.a = 1;
1495*9880d681SAndroid Build Coastguard Worker q2 = q2+q2 + 1; // update q2
1496*9880d681SAndroid Build Coastguard Worker r2 = r2+r2 + 1 - d; // update r2
1497*9880d681SAndroid Build Coastguard Worker }
1498*9880d681SAndroid Build Coastguard Worker else {
1499*9880d681SAndroid Build Coastguard Worker if (q2.uge(signedMin)) magu.a = 1;
1500*9880d681SAndroid Build Coastguard Worker q2 = q2+q2; // update q2
1501*9880d681SAndroid Build Coastguard Worker r2 = r2+r2 + 1; // update r2
1502*9880d681SAndroid Build Coastguard Worker }
1503*9880d681SAndroid Build Coastguard Worker delta = d - 1 - r2;
1504*9880d681SAndroid Build Coastguard Worker } while (p < d.getBitWidth()*2 &&
1505*9880d681SAndroid Build Coastguard Worker (q1.ult(delta) || (q1 == delta && r1 == 0)));
1506*9880d681SAndroid Build Coastguard Worker magu.m = q2 + 1; // resulting magic number
1507*9880d681SAndroid Build Coastguard Worker magu.s = p - d.getBitWidth(); // resulting shift
1508*9880d681SAndroid Build Coastguard Worker return magu;
1509*9880d681SAndroid Build Coastguard Worker }
1510*9880d681SAndroid Build Coastguard Worker
1511*9880d681SAndroid Build Coastguard Worker /// Implementation of Knuth's Algorithm D (Division of nonnegative integers)
1512*9880d681SAndroid Build Coastguard Worker /// from "Art of Computer Programming, Volume 2", section 4.3.1, p. 272. The
1513*9880d681SAndroid Build Coastguard Worker /// variables here have the same names as in the algorithm. Comments explain
1514*9880d681SAndroid Build Coastguard Worker /// the algorithm and any deviation from it.
KnuthDiv(unsigned * u,unsigned * v,unsigned * q,unsigned * r,unsigned m,unsigned n)1515*9880d681SAndroid Build Coastguard Worker static void KnuthDiv(unsigned *u, unsigned *v, unsigned *q, unsigned* r,
1516*9880d681SAndroid Build Coastguard Worker unsigned m, unsigned n) {
1517*9880d681SAndroid Build Coastguard Worker assert(u && "Must provide dividend");
1518*9880d681SAndroid Build Coastguard Worker assert(v && "Must provide divisor");
1519*9880d681SAndroid Build Coastguard Worker assert(q && "Must provide quotient");
1520*9880d681SAndroid Build Coastguard Worker assert(u != v && u != q && v != q && "Must use different memory");
1521*9880d681SAndroid Build Coastguard Worker assert(n>1 && "n must be > 1");
1522*9880d681SAndroid Build Coastguard Worker
1523*9880d681SAndroid Build Coastguard Worker // b denotes the base of the number system. In our case b is 2^32.
1524*9880d681SAndroid Build Coastguard Worker LLVM_CONSTEXPR uint64_t b = uint64_t(1) << 32;
1525*9880d681SAndroid Build Coastguard Worker
1526*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: m=" << m << " n=" << n << '\n');
1527*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: original:");
1528*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = m+n; i >=0; i--) dbgs() << " " << u[i]);
1529*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << " by");
1530*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = n; i >0; i--) dbgs() << " " << v[i-1]);
1531*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1532*9880d681SAndroid Build Coastguard Worker // D1. [Normalize.] Set d = b / (v[n-1] + 1) and multiply all the digits of
1533*9880d681SAndroid Build Coastguard Worker // u and v by d. Note that we have taken Knuth's advice here to use a power
1534*9880d681SAndroid Build Coastguard Worker // of 2 value for d such that d * v[n-1] >= b/2 (b is the base). A power of
1535*9880d681SAndroid Build Coastguard Worker // 2 allows us to shift instead of multiply and it is easy to determine the
1536*9880d681SAndroid Build Coastguard Worker // shift amount from the leading zeros. We are basically normalizing the u
1537*9880d681SAndroid Build Coastguard Worker // and v so that its high bits are shifted to the top of v's range without
1538*9880d681SAndroid Build Coastguard Worker // overflow. Note that this can require an extra word in u so that u must
1539*9880d681SAndroid Build Coastguard Worker // be of length m+n+1.
1540*9880d681SAndroid Build Coastguard Worker unsigned shift = countLeadingZeros(v[n-1]);
1541*9880d681SAndroid Build Coastguard Worker unsigned v_carry = 0;
1542*9880d681SAndroid Build Coastguard Worker unsigned u_carry = 0;
1543*9880d681SAndroid Build Coastguard Worker if (shift) {
1544*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < m+n; ++i) {
1545*9880d681SAndroid Build Coastguard Worker unsigned u_tmp = u[i] >> (32 - shift);
1546*9880d681SAndroid Build Coastguard Worker u[i] = (u[i] << shift) | u_carry;
1547*9880d681SAndroid Build Coastguard Worker u_carry = u_tmp;
1548*9880d681SAndroid Build Coastguard Worker }
1549*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < n; ++i) {
1550*9880d681SAndroid Build Coastguard Worker unsigned v_tmp = v[i] >> (32 - shift);
1551*9880d681SAndroid Build Coastguard Worker v[i] = (v[i] << shift) | v_carry;
1552*9880d681SAndroid Build Coastguard Worker v_carry = v_tmp;
1553*9880d681SAndroid Build Coastguard Worker }
1554*9880d681SAndroid Build Coastguard Worker }
1555*9880d681SAndroid Build Coastguard Worker u[m+n] = u_carry;
1556*9880d681SAndroid Build Coastguard Worker
1557*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: normal:");
1558*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = m+n; i >=0; i--) dbgs() << " " << u[i]);
1559*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << " by");
1560*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = n; i >0; i--) dbgs() << " " << v[i-1]);
1561*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1562*9880d681SAndroid Build Coastguard Worker
1563*9880d681SAndroid Build Coastguard Worker // D2. [Initialize j.] Set j to m. This is the loop counter over the places.
1564*9880d681SAndroid Build Coastguard Worker int j = m;
1565*9880d681SAndroid Build Coastguard Worker do {
1566*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: quotient digit #" << j << '\n');
1567*9880d681SAndroid Build Coastguard Worker // D3. [Calculate q'.].
1568*9880d681SAndroid Build Coastguard Worker // Set qp = (u[j+n]*b + u[j+n-1]) / v[n-1]. (qp=qprime=q')
1569*9880d681SAndroid Build Coastguard Worker // Set rp = (u[j+n]*b + u[j+n-1]) % v[n-1]. (rp=rprime=r')
1570*9880d681SAndroid Build Coastguard Worker // Now test if qp == b or qp*v[n-2] > b*rp + u[j+n-2]; if so, decrease
1571*9880d681SAndroid Build Coastguard Worker // qp by 1, inrease rp by v[n-1], and repeat this test if rp < b. The test
1572*9880d681SAndroid Build Coastguard Worker // on v[n-2] determines at high speed most of the cases in which the trial
1573*9880d681SAndroid Build Coastguard Worker // value qp is one too large, and it eliminates all cases where qp is two
1574*9880d681SAndroid Build Coastguard Worker // too large.
1575*9880d681SAndroid Build Coastguard Worker uint64_t dividend = ((uint64_t(u[j+n]) << 32) + u[j+n-1]);
1576*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: dividend == " << dividend << '\n');
1577*9880d681SAndroid Build Coastguard Worker uint64_t qp = dividend / v[n-1];
1578*9880d681SAndroid Build Coastguard Worker uint64_t rp = dividend % v[n-1];
1579*9880d681SAndroid Build Coastguard Worker if (qp == b || qp*v[n-2] > b*rp + u[j+n-2]) {
1580*9880d681SAndroid Build Coastguard Worker qp--;
1581*9880d681SAndroid Build Coastguard Worker rp += v[n-1];
1582*9880d681SAndroid Build Coastguard Worker if (rp < b && (qp == b || qp*v[n-2] > b*rp + u[j+n-2]))
1583*9880d681SAndroid Build Coastguard Worker qp--;
1584*9880d681SAndroid Build Coastguard Worker }
1585*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: qp == " << qp << ", rp == " << rp << '\n');
1586*9880d681SAndroid Build Coastguard Worker
1587*9880d681SAndroid Build Coastguard Worker // D4. [Multiply and subtract.] Replace (u[j+n]u[j+n-1]...u[j]) with
1588*9880d681SAndroid Build Coastguard Worker // (u[j+n]u[j+n-1]..u[j]) - qp * (v[n-1]...v[1]v[0]). This computation
1589*9880d681SAndroid Build Coastguard Worker // consists of a simple multiplication by a one-place number, combined with
1590*9880d681SAndroid Build Coastguard Worker // a subtraction.
1591*9880d681SAndroid Build Coastguard Worker // The digits (u[j+n]...u[j]) should be kept positive; if the result of
1592*9880d681SAndroid Build Coastguard Worker // this step is actually negative, (u[j+n]...u[j]) should be left as the
1593*9880d681SAndroid Build Coastguard Worker // true value plus b**(n+1), namely as the b's complement of
1594*9880d681SAndroid Build Coastguard Worker // the true value, and a "borrow" to the left should be remembered.
1595*9880d681SAndroid Build Coastguard Worker int64_t borrow = 0;
1596*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < n; ++i) {
1597*9880d681SAndroid Build Coastguard Worker uint64_t p = uint64_t(qp) * uint64_t(v[i]);
1598*9880d681SAndroid Build Coastguard Worker int64_t subres = int64_t(u[j+i]) - borrow - (unsigned)p;
1599*9880d681SAndroid Build Coastguard Worker u[j+i] = (unsigned)subres;
1600*9880d681SAndroid Build Coastguard Worker borrow = (p >> 32) - (subres >> 32);
1601*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: u[j+i] = " << u[j+i]
1602*9880d681SAndroid Build Coastguard Worker << ", borrow = " << borrow << '\n');
1603*9880d681SAndroid Build Coastguard Worker }
1604*9880d681SAndroid Build Coastguard Worker bool isNeg = u[j+n] < borrow;
1605*9880d681SAndroid Build Coastguard Worker u[j+n] -= (unsigned)borrow;
1606*9880d681SAndroid Build Coastguard Worker
1607*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: after subtraction:");
1608*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = m+n; i >=0; i--) dbgs() << " " << u[i]);
1609*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1610*9880d681SAndroid Build Coastguard Worker
1611*9880d681SAndroid Build Coastguard Worker // D5. [Test remainder.] Set q[j] = qp. If the result of step D4 was
1612*9880d681SAndroid Build Coastguard Worker // negative, go to step D6; otherwise go on to step D7.
1613*9880d681SAndroid Build Coastguard Worker q[j] = (unsigned)qp;
1614*9880d681SAndroid Build Coastguard Worker if (isNeg) {
1615*9880d681SAndroid Build Coastguard Worker // D6. [Add back]. The probability that this step is necessary is very
1616*9880d681SAndroid Build Coastguard Worker // small, on the order of only 2/b. Make sure that test data accounts for
1617*9880d681SAndroid Build Coastguard Worker // this possibility. Decrease q[j] by 1
1618*9880d681SAndroid Build Coastguard Worker q[j]--;
1619*9880d681SAndroid Build Coastguard Worker // and add (0v[n-1]...v[1]v[0]) to (u[j+n]u[j+n-1]...u[j+1]u[j]).
1620*9880d681SAndroid Build Coastguard Worker // A carry will occur to the left of u[j+n], and it should be ignored
1621*9880d681SAndroid Build Coastguard Worker // since it cancels with the borrow that occurred in D4.
1622*9880d681SAndroid Build Coastguard Worker bool carry = false;
1623*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < n; i++) {
1624*9880d681SAndroid Build Coastguard Worker unsigned limit = std::min(u[j+i],v[i]);
1625*9880d681SAndroid Build Coastguard Worker u[j+i] += v[i] + carry;
1626*9880d681SAndroid Build Coastguard Worker carry = u[j+i] < limit || (carry && u[j+i] == limit);
1627*9880d681SAndroid Build Coastguard Worker }
1628*9880d681SAndroid Build Coastguard Worker u[j+n] += carry;
1629*9880d681SAndroid Build Coastguard Worker }
1630*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: after correction:");
1631*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = m+n; i >=0; i--) dbgs() << " " << u[i]);
1632*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "\nKnuthDiv: digit result = " << q[j] << '\n');
1633*9880d681SAndroid Build Coastguard Worker
1634*9880d681SAndroid Build Coastguard Worker // D7. [Loop on j.] Decrease j by one. Now if j >= 0, go back to D3.
1635*9880d681SAndroid Build Coastguard Worker } while (--j >= 0);
1636*9880d681SAndroid Build Coastguard Worker
1637*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: quotient:");
1638*9880d681SAndroid Build Coastguard Worker DEBUG(for (int i = m; i >=0; i--) dbgs() <<" " << q[i]);
1639*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1640*9880d681SAndroid Build Coastguard Worker
1641*9880d681SAndroid Build Coastguard Worker // D8. [Unnormalize]. Now q[...] is the desired quotient, and the desired
1642*9880d681SAndroid Build Coastguard Worker // remainder may be obtained by dividing u[...] by d. If r is non-null we
1643*9880d681SAndroid Build Coastguard Worker // compute the remainder (urem uses this).
1644*9880d681SAndroid Build Coastguard Worker if (r) {
1645*9880d681SAndroid Build Coastguard Worker // The value d is expressed by the "shift" value above since we avoided
1646*9880d681SAndroid Build Coastguard Worker // multiplication by d by using a shift left. So, all we have to do is
1647*9880d681SAndroid Build Coastguard Worker // shift right here. In order to mak
1648*9880d681SAndroid Build Coastguard Worker if (shift) {
1649*9880d681SAndroid Build Coastguard Worker unsigned carry = 0;
1650*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << "KnuthDiv: remainder:");
1651*9880d681SAndroid Build Coastguard Worker for (int i = n-1; i >= 0; i--) {
1652*9880d681SAndroid Build Coastguard Worker r[i] = (u[i] >> shift) | carry;
1653*9880d681SAndroid Build Coastguard Worker carry = u[i] << (32 - shift);
1654*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << " " << r[i]);
1655*9880d681SAndroid Build Coastguard Worker }
1656*9880d681SAndroid Build Coastguard Worker } else {
1657*9880d681SAndroid Build Coastguard Worker for (int i = n-1; i >= 0; i--) {
1658*9880d681SAndroid Build Coastguard Worker r[i] = u[i];
1659*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << " " << r[i]);
1660*9880d681SAndroid Build Coastguard Worker }
1661*9880d681SAndroid Build Coastguard Worker }
1662*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1663*9880d681SAndroid Build Coastguard Worker }
1664*9880d681SAndroid Build Coastguard Worker DEBUG(dbgs() << '\n');
1665*9880d681SAndroid Build Coastguard Worker }
1666*9880d681SAndroid Build Coastguard Worker
divide(const APInt & LHS,unsigned lhsWords,const APInt & RHS,unsigned rhsWords,APInt * Quotient,APInt * Remainder)1667*9880d681SAndroid Build Coastguard Worker void APInt::divide(const APInt &LHS, unsigned lhsWords, const APInt &RHS,
1668*9880d681SAndroid Build Coastguard Worker unsigned rhsWords, APInt *Quotient, APInt *Remainder) {
1669*9880d681SAndroid Build Coastguard Worker assert(lhsWords >= rhsWords && "Fractional result");
1670*9880d681SAndroid Build Coastguard Worker
1671*9880d681SAndroid Build Coastguard Worker // First, compose the values into an array of 32-bit words instead of
1672*9880d681SAndroid Build Coastguard Worker // 64-bit words. This is a necessity of both the "short division" algorithm
1673*9880d681SAndroid Build Coastguard Worker // and the Knuth "classical algorithm" which requires there to be native
1674*9880d681SAndroid Build Coastguard Worker // operations for +, -, and * on an m bit value with an m*2 bit result. We
1675*9880d681SAndroid Build Coastguard Worker // can't use 64-bit operands here because we don't have native results of
1676*9880d681SAndroid Build Coastguard Worker // 128-bits. Furthermore, casting the 64-bit values to 32-bit values won't
1677*9880d681SAndroid Build Coastguard Worker // work on large-endian machines.
1678*9880d681SAndroid Build Coastguard Worker uint64_t mask = ~0ull >> (sizeof(unsigned)*CHAR_BIT);
1679*9880d681SAndroid Build Coastguard Worker unsigned n = rhsWords * 2;
1680*9880d681SAndroid Build Coastguard Worker unsigned m = (lhsWords * 2) - n;
1681*9880d681SAndroid Build Coastguard Worker
1682*9880d681SAndroid Build Coastguard Worker // Allocate space for the temporary values we need either on the stack, if
1683*9880d681SAndroid Build Coastguard Worker // it will fit, or on the heap if it won't.
1684*9880d681SAndroid Build Coastguard Worker unsigned SPACE[128];
1685*9880d681SAndroid Build Coastguard Worker unsigned *U = nullptr;
1686*9880d681SAndroid Build Coastguard Worker unsigned *V = nullptr;
1687*9880d681SAndroid Build Coastguard Worker unsigned *Q = nullptr;
1688*9880d681SAndroid Build Coastguard Worker unsigned *R = nullptr;
1689*9880d681SAndroid Build Coastguard Worker if ((Remainder?4:3)*n+2*m+1 <= 128) {
1690*9880d681SAndroid Build Coastguard Worker U = &SPACE[0];
1691*9880d681SAndroid Build Coastguard Worker V = &SPACE[m+n+1];
1692*9880d681SAndroid Build Coastguard Worker Q = &SPACE[(m+n+1) + n];
1693*9880d681SAndroid Build Coastguard Worker if (Remainder)
1694*9880d681SAndroid Build Coastguard Worker R = &SPACE[(m+n+1) + n + (m+n)];
1695*9880d681SAndroid Build Coastguard Worker } else {
1696*9880d681SAndroid Build Coastguard Worker U = new unsigned[m + n + 1];
1697*9880d681SAndroid Build Coastguard Worker V = new unsigned[n];
1698*9880d681SAndroid Build Coastguard Worker Q = new unsigned[m+n];
1699*9880d681SAndroid Build Coastguard Worker if (Remainder)
1700*9880d681SAndroid Build Coastguard Worker R = new unsigned[n];
1701*9880d681SAndroid Build Coastguard Worker }
1702*9880d681SAndroid Build Coastguard Worker
1703*9880d681SAndroid Build Coastguard Worker // Initialize the dividend
1704*9880d681SAndroid Build Coastguard Worker memset(U, 0, (m+n+1)*sizeof(unsigned));
1705*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < lhsWords; ++i) {
1706*9880d681SAndroid Build Coastguard Worker uint64_t tmp = (LHS.getNumWords() == 1 ? LHS.VAL : LHS.pVal[i]);
1707*9880d681SAndroid Build Coastguard Worker U[i * 2] = (unsigned)(tmp & mask);
1708*9880d681SAndroid Build Coastguard Worker U[i * 2 + 1] = (unsigned)(tmp >> (sizeof(unsigned)*CHAR_BIT));
1709*9880d681SAndroid Build Coastguard Worker }
1710*9880d681SAndroid Build Coastguard Worker U[m+n] = 0; // this extra word is for "spill" in the Knuth algorithm.
1711*9880d681SAndroid Build Coastguard Worker
1712*9880d681SAndroid Build Coastguard Worker // Initialize the divisor
1713*9880d681SAndroid Build Coastguard Worker memset(V, 0, (n)*sizeof(unsigned));
1714*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < rhsWords; ++i) {
1715*9880d681SAndroid Build Coastguard Worker uint64_t tmp = (RHS.getNumWords() == 1 ? RHS.VAL : RHS.pVal[i]);
1716*9880d681SAndroid Build Coastguard Worker V[i * 2] = (unsigned)(tmp & mask);
1717*9880d681SAndroid Build Coastguard Worker V[i * 2 + 1] = (unsigned)(tmp >> (sizeof(unsigned)*CHAR_BIT));
1718*9880d681SAndroid Build Coastguard Worker }
1719*9880d681SAndroid Build Coastguard Worker
1720*9880d681SAndroid Build Coastguard Worker // initialize the quotient and remainder
1721*9880d681SAndroid Build Coastguard Worker memset(Q, 0, (m+n) * sizeof(unsigned));
1722*9880d681SAndroid Build Coastguard Worker if (Remainder)
1723*9880d681SAndroid Build Coastguard Worker memset(R, 0, n * sizeof(unsigned));
1724*9880d681SAndroid Build Coastguard Worker
1725*9880d681SAndroid Build Coastguard Worker // Now, adjust m and n for the Knuth division. n is the number of words in
1726*9880d681SAndroid Build Coastguard Worker // the divisor. m is the number of words by which the dividend exceeds the
1727*9880d681SAndroid Build Coastguard Worker // divisor (i.e. m+n is the length of the dividend). These sizes must not
1728*9880d681SAndroid Build Coastguard Worker // contain any zero words or the Knuth algorithm fails.
1729*9880d681SAndroid Build Coastguard Worker for (unsigned i = n; i > 0 && V[i-1] == 0; i--) {
1730*9880d681SAndroid Build Coastguard Worker n--;
1731*9880d681SAndroid Build Coastguard Worker m++;
1732*9880d681SAndroid Build Coastguard Worker }
1733*9880d681SAndroid Build Coastguard Worker for (unsigned i = m+n; i > 0 && U[i-1] == 0; i--)
1734*9880d681SAndroid Build Coastguard Worker m--;
1735*9880d681SAndroid Build Coastguard Worker
1736*9880d681SAndroid Build Coastguard Worker // If we're left with only a single word for the divisor, Knuth doesn't work
1737*9880d681SAndroid Build Coastguard Worker // so we implement the short division algorithm here. This is much simpler
1738*9880d681SAndroid Build Coastguard Worker // and faster because we are certain that we can divide a 64-bit quantity
1739*9880d681SAndroid Build Coastguard Worker // by a 32-bit quantity at hardware speed and short division is simply a
1740*9880d681SAndroid Build Coastguard Worker // series of such operations. This is just like doing short division but we
1741*9880d681SAndroid Build Coastguard Worker // are using base 2^32 instead of base 10.
1742*9880d681SAndroid Build Coastguard Worker assert(n != 0 && "Divide by zero?");
1743*9880d681SAndroid Build Coastguard Worker if (n == 1) {
1744*9880d681SAndroid Build Coastguard Worker unsigned divisor = V[0];
1745*9880d681SAndroid Build Coastguard Worker unsigned remainder = 0;
1746*9880d681SAndroid Build Coastguard Worker for (int i = m+n-1; i >= 0; i--) {
1747*9880d681SAndroid Build Coastguard Worker uint64_t partial_dividend = uint64_t(remainder) << 32 | U[i];
1748*9880d681SAndroid Build Coastguard Worker if (partial_dividend == 0) {
1749*9880d681SAndroid Build Coastguard Worker Q[i] = 0;
1750*9880d681SAndroid Build Coastguard Worker remainder = 0;
1751*9880d681SAndroid Build Coastguard Worker } else if (partial_dividend < divisor) {
1752*9880d681SAndroid Build Coastguard Worker Q[i] = 0;
1753*9880d681SAndroid Build Coastguard Worker remainder = (unsigned)partial_dividend;
1754*9880d681SAndroid Build Coastguard Worker } else if (partial_dividend == divisor) {
1755*9880d681SAndroid Build Coastguard Worker Q[i] = 1;
1756*9880d681SAndroid Build Coastguard Worker remainder = 0;
1757*9880d681SAndroid Build Coastguard Worker } else {
1758*9880d681SAndroid Build Coastguard Worker Q[i] = (unsigned)(partial_dividend / divisor);
1759*9880d681SAndroid Build Coastguard Worker remainder = (unsigned)(partial_dividend - (Q[i] * divisor));
1760*9880d681SAndroid Build Coastguard Worker }
1761*9880d681SAndroid Build Coastguard Worker }
1762*9880d681SAndroid Build Coastguard Worker if (R)
1763*9880d681SAndroid Build Coastguard Worker R[0] = remainder;
1764*9880d681SAndroid Build Coastguard Worker } else {
1765*9880d681SAndroid Build Coastguard Worker // Now we're ready to invoke the Knuth classical divide algorithm. In this
1766*9880d681SAndroid Build Coastguard Worker // case n > 1.
1767*9880d681SAndroid Build Coastguard Worker KnuthDiv(U, V, Q, R, m, n);
1768*9880d681SAndroid Build Coastguard Worker }
1769*9880d681SAndroid Build Coastguard Worker
1770*9880d681SAndroid Build Coastguard Worker // If the caller wants the quotient
1771*9880d681SAndroid Build Coastguard Worker if (Quotient) {
1772*9880d681SAndroid Build Coastguard Worker // Set up the Quotient value's memory.
1773*9880d681SAndroid Build Coastguard Worker if (Quotient->BitWidth != LHS.BitWidth) {
1774*9880d681SAndroid Build Coastguard Worker if (Quotient->isSingleWord())
1775*9880d681SAndroid Build Coastguard Worker Quotient->VAL = 0;
1776*9880d681SAndroid Build Coastguard Worker else
1777*9880d681SAndroid Build Coastguard Worker delete [] Quotient->pVal;
1778*9880d681SAndroid Build Coastguard Worker Quotient->BitWidth = LHS.BitWidth;
1779*9880d681SAndroid Build Coastguard Worker if (!Quotient->isSingleWord())
1780*9880d681SAndroid Build Coastguard Worker Quotient->pVal = getClearedMemory(Quotient->getNumWords());
1781*9880d681SAndroid Build Coastguard Worker } else
1782*9880d681SAndroid Build Coastguard Worker Quotient->clearAllBits();
1783*9880d681SAndroid Build Coastguard Worker
1784*9880d681SAndroid Build Coastguard Worker // The quotient is in Q. Reconstitute the quotient into Quotient's low
1785*9880d681SAndroid Build Coastguard Worker // order words.
1786*9880d681SAndroid Build Coastguard Worker // This case is currently dead as all users of divide() handle trivial cases
1787*9880d681SAndroid Build Coastguard Worker // earlier.
1788*9880d681SAndroid Build Coastguard Worker if (lhsWords == 1) {
1789*9880d681SAndroid Build Coastguard Worker uint64_t tmp =
1790*9880d681SAndroid Build Coastguard Worker uint64_t(Q[0]) | (uint64_t(Q[1]) << (APINT_BITS_PER_WORD / 2));
1791*9880d681SAndroid Build Coastguard Worker if (Quotient->isSingleWord())
1792*9880d681SAndroid Build Coastguard Worker Quotient->VAL = tmp;
1793*9880d681SAndroid Build Coastguard Worker else
1794*9880d681SAndroid Build Coastguard Worker Quotient->pVal[0] = tmp;
1795*9880d681SAndroid Build Coastguard Worker } else {
1796*9880d681SAndroid Build Coastguard Worker assert(!Quotient->isSingleWord() && "Quotient APInt not large enough");
1797*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < lhsWords; ++i)
1798*9880d681SAndroid Build Coastguard Worker Quotient->pVal[i] =
1799*9880d681SAndroid Build Coastguard Worker uint64_t(Q[i*2]) | (uint64_t(Q[i*2+1]) << (APINT_BITS_PER_WORD / 2));
1800*9880d681SAndroid Build Coastguard Worker }
1801*9880d681SAndroid Build Coastguard Worker }
1802*9880d681SAndroid Build Coastguard Worker
1803*9880d681SAndroid Build Coastguard Worker // If the caller wants the remainder
1804*9880d681SAndroid Build Coastguard Worker if (Remainder) {
1805*9880d681SAndroid Build Coastguard Worker // Set up the Remainder value's memory.
1806*9880d681SAndroid Build Coastguard Worker if (Remainder->BitWidth != RHS.BitWidth) {
1807*9880d681SAndroid Build Coastguard Worker if (Remainder->isSingleWord())
1808*9880d681SAndroid Build Coastguard Worker Remainder->VAL = 0;
1809*9880d681SAndroid Build Coastguard Worker else
1810*9880d681SAndroid Build Coastguard Worker delete [] Remainder->pVal;
1811*9880d681SAndroid Build Coastguard Worker Remainder->BitWidth = RHS.BitWidth;
1812*9880d681SAndroid Build Coastguard Worker if (!Remainder->isSingleWord())
1813*9880d681SAndroid Build Coastguard Worker Remainder->pVal = getClearedMemory(Remainder->getNumWords());
1814*9880d681SAndroid Build Coastguard Worker } else
1815*9880d681SAndroid Build Coastguard Worker Remainder->clearAllBits();
1816*9880d681SAndroid Build Coastguard Worker
1817*9880d681SAndroid Build Coastguard Worker // The remainder is in R. Reconstitute the remainder into Remainder's low
1818*9880d681SAndroid Build Coastguard Worker // order words.
1819*9880d681SAndroid Build Coastguard Worker if (rhsWords == 1) {
1820*9880d681SAndroid Build Coastguard Worker uint64_t tmp =
1821*9880d681SAndroid Build Coastguard Worker uint64_t(R[0]) | (uint64_t(R[1]) << (APINT_BITS_PER_WORD / 2));
1822*9880d681SAndroid Build Coastguard Worker if (Remainder->isSingleWord())
1823*9880d681SAndroid Build Coastguard Worker Remainder->VAL = tmp;
1824*9880d681SAndroid Build Coastguard Worker else
1825*9880d681SAndroid Build Coastguard Worker Remainder->pVal[0] = tmp;
1826*9880d681SAndroid Build Coastguard Worker } else {
1827*9880d681SAndroid Build Coastguard Worker assert(!Remainder->isSingleWord() && "Remainder APInt not large enough");
1828*9880d681SAndroid Build Coastguard Worker for (unsigned i = 0; i < rhsWords; ++i)
1829*9880d681SAndroid Build Coastguard Worker Remainder->pVal[i] =
1830*9880d681SAndroid Build Coastguard Worker uint64_t(R[i*2]) | (uint64_t(R[i*2+1]) << (APINT_BITS_PER_WORD / 2));
1831*9880d681SAndroid Build Coastguard Worker }
1832*9880d681SAndroid Build Coastguard Worker }
1833*9880d681SAndroid Build Coastguard Worker
1834*9880d681SAndroid Build Coastguard Worker // Clean up the memory we allocated.
1835*9880d681SAndroid Build Coastguard Worker if (U != &SPACE[0]) {
1836*9880d681SAndroid Build Coastguard Worker delete [] U;
1837*9880d681SAndroid Build Coastguard Worker delete [] V;
1838*9880d681SAndroid Build Coastguard Worker delete [] Q;
1839*9880d681SAndroid Build Coastguard Worker delete [] R;
1840*9880d681SAndroid Build Coastguard Worker }
1841*9880d681SAndroid Build Coastguard Worker }
1842*9880d681SAndroid Build Coastguard Worker
udiv(const APInt & RHS) const1843*9880d681SAndroid Build Coastguard Worker APInt APInt::udiv(const APInt& RHS) const {
1844*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
1845*9880d681SAndroid Build Coastguard Worker
1846*9880d681SAndroid Build Coastguard Worker // First, deal with the easy case
1847*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
1848*9880d681SAndroid Build Coastguard Worker assert(RHS.VAL != 0 && "Divide by zero?");
1849*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL / RHS.VAL);
1850*9880d681SAndroid Build Coastguard Worker }
1851*9880d681SAndroid Build Coastguard Worker
1852*9880d681SAndroid Build Coastguard Worker // Get some facts about the LHS and RHS number of bits and words
1853*9880d681SAndroid Build Coastguard Worker unsigned rhsBits = RHS.getActiveBits();
1854*9880d681SAndroid Build Coastguard Worker unsigned rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
1855*9880d681SAndroid Build Coastguard Worker assert(rhsWords && "Divided by zero???");
1856*9880d681SAndroid Build Coastguard Worker unsigned lhsBits = this->getActiveBits();
1857*9880d681SAndroid Build Coastguard Worker unsigned lhsWords = !lhsBits ? 0 : (APInt::whichWord(lhsBits - 1) + 1);
1858*9880d681SAndroid Build Coastguard Worker
1859*9880d681SAndroid Build Coastguard Worker // Deal with some degenerate cases
1860*9880d681SAndroid Build Coastguard Worker if (!lhsWords)
1861*9880d681SAndroid Build Coastguard Worker // 0 / X ===> 0
1862*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1863*9880d681SAndroid Build Coastguard Worker else if (lhsWords < rhsWords || this->ult(RHS)) {
1864*9880d681SAndroid Build Coastguard Worker // X / Y ===> 0, iff X < Y
1865*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1866*9880d681SAndroid Build Coastguard Worker } else if (*this == RHS) {
1867*9880d681SAndroid Build Coastguard Worker // X / X ===> 1
1868*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 1);
1869*9880d681SAndroid Build Coastguard Worker } else if (lhsWords == 1 && rhsWords == 1) {
1870*9880d681SAndroid Build Coastguard Worker // All high words are zero, just use native divide
1871*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, this->pVal[0] / RHS.pVal[0]);
1872*9880d681SAndroid Build Coastguard Worker }
1873*9880d681SAndroid Build Coastguard Worker
1874*9880d681SAndroid Build Coastguard Worker // We have to compute it the hard way. Invoke the Knuth divide algorithm.
1875*9880d681SAndroid Build Coastguard Worker APInt Quotient(1,0); // to hold result.
1876*9880d681SAndroid Build Coastguard Worker divide(*this, lhsWords, RHS, rhsWords, &Quotient, nullptr);
1877*9880d681SAndroid Build Coastguard Worker return Quotient;
1878*9880d681SAndroid Build Coastguard Worker }
1879*9880d681SAndroid Build Coastguard Worker
sdiv(const APInt & RHS) const1880*9880d681SAndroid Build Coastguard Worker APInt APInt::sdiv(const APInt &RHS) const {
1881*9880d681SAndroid Build Coastguard Worker if (isNegative()) {
1882*9880d681SAndroid Build Coastguard Worker if (RHS.isNegative())
1883*9880d681SAndroid Build Coastguard Worker return (-(*this)).udiv(-RHS);
1884*9880d681SAndroid Build Coastguard Worker return -((-(*this)).udiv(RHS));
1885*9880d681SAndroid Build Coastguard Worker }
1886*9880d681SAndroid Build Coastguard Worker if (RHS.isNegative())
1887*9880d681SAndroid Build Coastguard Worker return -(this->udiv(-RHS));
1888*9880d681SAndroid Build Coastguard Worker return this->udiv(RHS);
1889*9880d681SAndroid Build Coastguard Worker }
1890*9880d681SAndroid Build Coastguard Worker
urem(const APInt & RHS) const1891*9880d681SAndroid Build Coastguard Worker APInt APInt::urem(const APInt& RHS) const {
1892*9880d681SAndroid Build Coastguard Worker assert(BitWidth == RHS.BitWidth && "Bit widths must be the same");
1893*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
1894*9880d681SAndroid Build Coastguard Worker assert(RHS.VAL != 0 && "Remainder by zero?");
1895*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, VAL % RHS.VAL);
1896*9880d681SAndroid Build Coastguard Worker }
1897*9880d681SAndroid Build Coastguard Worker
1898*9880d681SAndroid Build Coastguard Worker // Get some facts about the LHS
1899*9880d681SAndroid Build Coastguard Worker unsigned lhsBits = getActiveBits();
1900*9880d681SAndroid Build Coastguard Worker unsigned lhsWords = !lhsBits ? 0 : (whichWord(lhsBits - 1) + 1);
1901*9880d681SAndroid Build Coastguard Worker
1902*9880d681SAndroid Build Coastguard Worker // Get some facts about the RHS
1903*9880d681SAndroid Build Coastguard Worker unsigned rhsBits = RHS.getActiveBits();
1904*9880d681SAndroid Build Coastguard Worker unsigned rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
1905*9880d681SAndroid Build Coastguard Worker assert(rhsWords && "Performing remainder operation by zero ???");
1906*9880d681SAndroid Build Coastguard Worker
1907*9880d681SAndroid Build Coastguard Worker // Check the degenerate cases
1908*9880d681SAndroid Build Coastguard Worker if (lhsWords == 0) {
1909*9880d681SAndroid Build Coastguard Worker // 0 % Y ===> 0
1910*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1911*9880d681SAndroid Build Coastguard Worker } else if (lhsWords < rhsWords || this->ult(RHS)) {
1912*9880d681SAndroid Build Coastguard Worker // X % Y ===> X, iff X < Y
1913*9880d681SAndroid Build Coastguard Worker return *this;
1914*9880d681SAndroid Build Coastguard Worker } else if (*this == RHS) {
1915*9880d681SAndroid Build Coastguard Worker // X % X == 0;
1916*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
1917*9880d681SAndroid Build Coastguard Worker } else if (lhsWords == 1) {
1918*9880d681SAndroid Build Coastguard Worker // All high words are zero, just use native remainder
1919*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, pVal[0] % RHS.pVal[0]);
1920*9880d681SAndroid Build Coastguard Worker }
1921*9880d681SAndroid Build Coastguard Worker
1922*9880d681SAndroid Build Coastguard Worker // We have to compute it the hard way. Invoke the Knuth divide algorithm.
1923*9880d681SAndroid Build Coastguard Worker APInt Remainder(1,0);
1924*9880d681SAndroid Build Coastguard Worker divide(*this, lhsWords, RHS, rhsWords, nullptr, &Remainder);
1925*9880d681SAndroid Build Coastguard Worker return Remainder;
1926*9880d681SAndroid Build Coastguard Worker }
1927*9880d681SAndroid Build Coastguard Worker
srem(const APInt & RHS) const1928*9880d681SAndroid Build Coastguard Worker APInt APInt::srem(const APInt &RHS) const {
1929*9880d681SAndroid Build Coastguard Worker if (isNegative()) {
1930*9880d681SAndroid Build Coastguard Worker if (RHS.isNegative())
1931*9880d681SAndroid Build Coastguard Worker return -((-(*this)).urem(-RHS));
1932*9880d681SAndroid Build Coastguard Worker return -((-(*this)).urem(RHS));
1933*9880d681SAndroid Build Coastguard Worker }
1934*9880d681SAndroid Build Coastguard Worker if (RHS.isNegative())
1935*9880d681SAndroid Build Coastguard Worker return this->urem(-RHS);
1936*9880d681SAndroid Build Coastguard Worker return this->urem(RHS);
1937*9880d681SAndroid Build Coastguard Worker }
1938*9880d681SAndroid Build Coastguard Worker
udivrem(const APInt & LHS,const APInt & RHS,APInt & Quotient,APInt & Remainder)1939*9880d681SAndroid Build Coastguard Worker void APInt::udivrem(const APInt &LHS, const APInt &RHS,
1940*9880d681SAndroid Build Coastguard Worker APInt &Quotient, APInt &Remainder) {
1941*9880d681SAndroid Build Coastguard Worker assert(LHS.BitWidth == RHS.BitWidth && "Bit widths must be the same");
1942*9880d681SAndroid Build Coastguard Worker
1943*9880d681SAndroid Build Coastguard Worker // First, deal with the easy case
1944*9880d681SAndroid Build Coastguard Worker if (LHS.isSingleWord()) {
1945*9880d681SAndroid Build Coastguard Worker assert(RHS.VAL != 0 && "Divide by zero?");
1946*9880d681SAndroid Build Coastguard Worker uint64_t QuotVal = LHS.VAL / RHS.VAL;
1947*9880d681SAndroid Build Coastguard Worker uint64_t RemVal = LHS.VAL % RHS.VAL;
1948*9880d681SAndroid Build Coastguard Worker Quotient = APInt(LHS.BitWidth, QuotVal);
1949*9880d681SAndroid Build Coastguard Worker Remainder = APInt(LHS.BitWidth, RemVal);
1950*9880d681SAndroid Build Coastguard Worker return;
1951*9880d681SAndroid Build Coastguard Worker }
1952*9880d681SAndroid Build Coastguard Worker
1953*9880d681SAndroid Build Coastguard Worker // Get some size facts about the dividend and divisor
1954*9880d681SAndroid Build Coastguard Worker unsigned lhsBits = LHS.getActiveBits();
1955*9880d681SAndroid Build Coastguard Worker unsigned lhsWords = !lhsBits ? 0 : (APInt::whichWord(lhsBits - 1) + 1);
1956*9880d681SAndroid Build Coastguard Worker unsigned rhsBits = RHS.getActiveBits();
1957*9880d681SAndroid Build Coastguard Worker unsigned rhsWords = !rhsBits ? 0 : (APInt::whichWord(rhsBits - 1) + 1);
1958*9880d681SAndroid Build Coastguard Worker
1959*9880d681SAndroid Build Coastguard Worker // Check the degenerate cases
1960*9880d681SAndroid Build Coastguard Worker if (lhsWords == 0) {
1961*9880d681SAndroid Build Coastguard Worker Quotient = 0; // 0 / Y ===> 0
1962*9880d681SAndroid Build Coastguard Worker Remainder = 0; // 0 % Y ===> 0
1963*9880d681SAndroid Build Coastguard Worker return;
1964*9880d681SAndroid Build Coastguard Worker }
1965*9880d681SAndroid Build Coastguard Worker
1966*9880d681SAndroid Build Coastguard Worker if (lhsWords < rhsWords || LHS.ult(RHS)) {
1967*9880d681SAndroid Build Coastguard Worker Remainder = LHS; // X % Y ===> X, iff X < Y
1968*9880d681SAndroid Build Coastguard Worker Quotient = 0; // X / Y ===> 0, iff X < Y
1969*9880d681SAndroid Build Coastguard Worker return;
1970*9880d681SAndroid Build Coastguard Worker }
1971*9880d681SAndroid Build Coastguard Worker
1972*9880d681SAndroid Build Coastguard Worker if (LHS == RHS) {
1973*9880d681SAndroid Build Coastguard Worker Quotient = 1; // X / X ===> 1
1974*9880d681SAndroid Build Coastguard Worker Remainder = 0; // X % X ===> 0;
1975*9880d681SAndroid Build Coastguard Worker return;
1976*9880d681SAndroid Build Coastguard Worker }
1977*9880d681SAndroid Build Coastguard Worker
1978*9880d681SAndroid Build Coastguard Worker if (lhsWords == 1 && rhsWords == 1) {
1979*9880d681SAndroid Build Coastguard Worker // There is only one word to consider so use the native versions.
1980*9880d681SAndroid Build Coastguard Worker uint64_t lhsValue = LHS.isSingleWord() ? LHS.VAL : LHS.pVal[0];
1981*9880d681SAndroid Build Coastguard Worker uint64_t rhsValue = RHS.isSingleWord() ? RHS.VAL : RHS.pVal[0];
1982*9880d681SAndroid Build Coastguard Worker Quotient = APInt(LHS.getBitWidth(), lhsValue / rhsValue);
1983*9880d681SAndroid Build Coastguard Worker Remainder = APInt(LHS.getBitWidth(), lhsValue % rhsValue);
1984*9880d681SAndroid Build Coastguard Worker return;
1985*9880d681SAndroid Build Coastguard Worker }
1986*9880d681SAndroid Build Coastguard Worker
1987*9880d681SAndroid Build Coastguard Worker // Okay, lets do it the long way
1988*9880d681SAndroid Build Coastguard Worker divide(LHS, lhsWords, RHS, rhsWords, &Quotient, &Remainder);
1989*9880d681SAndroid Build Coastguard Worker }
1990*9880d681SAndroid Build Coastguard Worker
sdivrem(const APInt & LHS,const APInt & RHS,APInt & Quotient,APInt & Remainder)1991*9880d681SAndroid Build Coastguard Worker void APInt::sdivrem(const APInt &LHS, const APInt &RHS,
1992*9880d681SAndroid Build Coastguard Worker APInt &Quotient, APInt &Remainder) {
1993*9880d681SAndroid Build Coastguard Worker if (LHS.isNegative()) {
1994*9880d681SAndroid Build Coastguard Worker if (RHS.isNegative())
1995*9880d681SAndroid Build Coastguard Worker APInt::udivrem(-LHS, -RHS, Quotient, Remainder);
1996*9880d681SAndroid Build Coastguard Worker else {
1997*9880d681SAndroid Build Coastguard Worker APInt::udivrem(-LHS, RHS, Quotient, Remainder);
1998*9880d681SAndroid Build Coastguard Worker Quotient = -Quotient;
1999*9880d681SAndroid Build Coastguard Worker }
2000*9880d681SAndroid Build Coastguard Worker Remainder = -Remainder;
2001*9880d681SAndroid Build Coastguard Worker } else if (RHS.isNegative()) {
2002*9880d681SAndroid Build Coastguard Worker APInt::udivrem(LHS, -RHS, Quotient, Remainder);
2003*9880d681SAndroid Build Coastguard Worker Quotient = -Quotient;
2004*9880d681SAndroid Build Coastguard Worker } else {
2005*9880d681SAndroid Build Coastguard Worker APInt::udivrem(LHS, RHS, Quotient, Remainder);
2006*9880d681SAndroid Build Coastguard Worker }
2007*9880d681SAndroid Build Coastguard Worker }
2008*9880d681SAndroid Build Coastguard Worker
sadd_ov(const APInt & RHS,bool & Overflow) const2009*9880d681SAndroid Build Coastguard Worker APInt APInt::sadd_ov(const APInt &RHS, bool &Overflow) const {
2010*9880d681SAndroid Build Coastguard Worker APInt Res = *this+RHS;
2011*9880d681SAndroid Build Coastguard Worker Overflow = isNonNegative() == RHS.isNonNegative() &&
2012*9880d681SAndroid Build Coastguard Worker Res.isNonNegative() != isNonNegative();
2013*9880d681SAndroid Build Coastguard Worker return Res;
2014*9880d681SAndroid Build Coastguard Worker }
2015*9880d681SAndroid Build Coastguard Worker
uadd_ov(const APInt & RHS,bool & Overflow) const2016*9880d681SAndroid Build Coastguard Worker APInt APInt::uadd_ov(const APInt &RHS, bool &Overflow) const {
2017*9880d681SAndroid Build Coastguard Worker APInt Res = *this+RHS;
2018*9880d681SAndroid Build Coastguard Worker Overflow = Res.ult(RHS);
2019*9880d681SAndroid Build Coastguard Worker return Res;
2020*9880d681SAndroid Build Coastguard Worker }
2021*9880d681SAndroid Build Coastguard Worker
ssub_ov(const APInt & RHS,bool & Overflow) const2022*9880d681SAndroid Build Coastguard Worker APInt APInt::ssub_ov(const APInt &RHS, bool &Overflow) const {
2023*9880d681SAndroid Build Coastguard Worker APInt Res = *this - RHS;
2024*9880d681SAndroid Build Coastguard Worker Overflow = isNonNegative() != RHS.isNonNegative() &&
2025*9880d681SAndroid Build Coastguard Worker Res.isNonNegative() != isNonNegative();
2026*9880d681SAndroid Build Coastguard Worker return Res;
2027*9880d681SAndroid Build Coastguard Worker }
2028*9880d681SAndroid Build Coastguard Worker
usub_ov(const APInt & RHS,bool & Overflow) const2029*9880d681SAndroid Build Coastguard Worker APInt APInt::usub_ov(const APInt &RHS, bool &Overflow) const {
2030*9880d681SAndroid Build Coastguard Worker APInt Res = *this-RHS;
2031*9880d681SAndroid Build Coastguard Worker Overflow = Res.ugt(*this);
2032*9880d681SAndroid Build Coastguard Worker return Res;
2033*9880d681SAndroid Build Coastguard Worker }
2034*9880d681SAndroid Build Coastguard Worker
sdiv_ov(const APInt & RHS,bool & Overflow) const2035*9880d681SAndroid Build Coastguard Worker APInt APInt::sdiv_ov(const APInt &RHS, bool &Overflow) const {
2036*9880d681SAndroid Build Coastguard Worker // MININT/-1 --> overflow.
2037*9880d681SAndroid Build Coastguard Worker Overflow = isMinSignedValue() && RHS.isAllOnesValue();
2038*9880d681SAndroid Build Coastguard Worker return sdiv(RHS);
2039*9880d681SAndroid Build Coastguard Worker }
2040*9880d681SAndroid Build Coastguard Worker
smul_ov(const APInt & RHS,bool & Overflow) const2041*9880d681SAndroid Build Coastguard Worker APInt APInt::smul_ov(const APInt &RHS, bool &Overflow) const {
2042*9880d681SAndroid Build Coastguard Worker APInt Res = *this * RHS;
2043*9880d681SAndroid Build Coastguard Worker
2044*9880d681SAndroid Build Coastguard Worker if (*this != 0 && RHS != 0)
2045*9880d681SAndroid Build Coastguard Worker Overflow = Res.sdiv(RHS) != *this || Res.sdiv(*this) != RHS;
2046*9880d681SAndroid Build Coastguard Worker else
2047*9880d681SAndroid Build Coastguard Worker Overflow = false;
2048*9880d681SAndroid Build Coastguard Worker return Res;
2049*9880d681SAndroid Build Coastguard Worker }
2050*9880d681SAndroid Build Coastguard Worker
umul_ov(const APInt & RHS,bool & Overflow) const2051*9880d681SAndroid Build Coastguard Worker APInt APInt::umul_ov(const APInt &RHS, bool &Overflow) const {
2052*9880d681SAndroid Build Coastguard Worker APInt Res = *this * RHS;
2053*9880d681SAndroid Build Coastguard Worker
2054*9880d681SAndroid Build Coastguard Worker if (*this != 0 && RHS != 0)
2055*9880d681SAndroid Build Coastguard Worker Overflow = Res.udiv(RHS) != *this || Res.udiv(*this) != RHS;
2056*9880d681SAndroid Build Coastguard Worker else
2057*9880d681SAndroid Build Coastguard Worker Overflow = false;
2058*9880d681SAndroid Build Coastguard Worker return Res;
2059*9880d681SAndroid Build Coastguard Worker }
2060*9880d681SAndroid Build Coastguard Worker
sshl_ov(const APInt & ShAmt,bool & Overflow) const2061*9880d681SAndroid Build Coastguard Worker APInt APInt::sshl_ov(const APInt &ShAmt, bool &Overflow) const {
2062*9880d681SAndroid Build Coastguard Worker Overflow = ShAmt.uge(getBitWidth());
2063*9880d681SAndroid Build Coastguard Worker if (Overflow)
2064*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
2065*9880d681SAndroid Build Coastguard Worker
2066*9880d681SAndroid Build Coastguard Worker if (isNonNegative()) // Don't allow sign change.
2067*9880d681SAndroid Build Coastguard Worker Overflow = ShAmt.uge(countLeadingZeros());
2068*9880d681SAndroid Build Coastguard Worker else
2069*9880d681SAndroid Build Coastguard Worker Overflow = ShAmt.uge(countLeadingOnes());
2070*9880d681SAndroid Build Coastguard Worker
2071*9880d681SAndroid Build Coastguard Worker return *this << ShAmt;
2072*9880d681SAndroid Build Coastguard Worker }
2073*9880d681SAndroid Build Coastguard Worker
ushl_ov(const APInt & ShAmt,bool & Overflow) const2074*9880d681SAndroid Build Coastguard Worker APInt APInt::ushl_ov(const APInt &ShAmt, bool &Overflow) const {
2075*9880d681SAndroid Build Coastguard Worker Overflow = ShAmt.uge(getBitWidth());
2076*9880d681SAndroid Build Coastguard Worker if (Overflow)
2077*9880d681SAndroid Build Coastguard Worker return APInt(BitWidth, 0);
2078*9880d681SAndroid Build Coastguard Worker
2079*9880d681SAndroid Build Coastguard Worker Overflow = ShAmt.ugt(countLeadingZeros());
2080*9880d681SAndroid Build Coastguard Worker
2081*9880d681SAndroid Build Coastguard Worker return *this << ShAmt;
2082*9880d681SAndroid Build Coastguard Worker }
2083*9880d681SAndroid Build Coastguard Worker
2084*9880d681SAndroid Build Coastguard Worker
2085*9880d681SAndroid Build Coastguard Worker
2086*9880d681SAndroid Build Coastguard Worker
fromString(unsigned numbits,StringRef str,uint8_t radix)2087*9880d681SAndroid Build Coastguard Worker void APInt::fromString(unsigned numbits, StringRef str, uint8_t radix) {
2088*9880d681SAndroid Build Coastguard Worker // Check our assumptions here
2089*9880d681SAndroid Build Coastguard Worker assert(!str.empty() && "Invalid string length");
2090*9880d681SAndroid Build Coastguard Worker assert((radix == 10 || radix == 8 || radix == 16 || radix == 2 ||
2091*9880d681SAndroid Build Coastguard Worker radix == 36) &&
2092*9880d681SAndroid Build Coastguard Worker "Radix should be 2, 8, 10, 16, or 36!");
2093*9880d681SAndroid Build Coastguard Worker
2094*9880d681SAndroid Build Coastguard Worker StringRef::iterator p = str.begin();
2095*9880d681SAndroid Build Coastguard Worker size_t slen = str.size();
2096*9880d681SAndroid Build Coastguard Worker bool isNeg = *p == '-';
2097*9880d681SAndroid Build Coastguard Worker if (*p == '-' || *p == '+') {
2098*9880d681SAndroid Build Coastguard Worker p++;
2099*9880d681SAndroid Build Coastguard Worker slen--;
2100*9880d681SAndroid Build Coastguard Worker assert(slen && "String is only a sign, needs a value.");
2101*9880d681SAndroid Build Coastguard Worker }
2102*9880d681SAndroid Build Coastguard Worker assert((slen <= numbits || radix != 2) && "Insufficient bit width");
2103*9880d681SAndroid Build Coastguard Worker assert(((slen-1)*3 <= numbits || radix != 8) && "Insufficient bit width");
2104*9880d681SAndroid Build Coastguard Worker assert(((slen-1)*4 <= numbits || radix != 16) && "Insufficient bit width");
2105*9880d681SAndroid Build Coastguard Worker assert((((slen-1)*64)/22 <= numbits || radix != 10) &&
2106*9880d681SAndroid Build Coastguard Worker "Insufficient bit width");
2107*9880d681SAndroid Build Coastguard Worker
2108*9880d681SAndroid Build Coastguard Worker // Allocate memory
2109*9880d681SAndroid Build Coastguard Worker if (!isSingleWord())
2110*9880d681SAndroid Build Coastguard Worker pVal = getClearedMemory(getNumWords());
2111*9880d681SAndroid Build Coastguard Worker
2112*9880d681SAndroid Build Coastguard Worker // Figure out if we can shift instead of multiply
2113*9880d681SAndroid Build Coastguard Worker unsigned shift = (radix == 16 ? 4 : radix == 8 ? 3 : radix == 2 ? 1 : 0);
2114*9880d681SAndroid Build Coastguard Worker
2115*9880d681SAndroid Build Coastguard Worker // Set up an APInt for the digit to add outside the loop so we don't
2116*9880d681SAndroid Build Coastguard Worker // constantly construct/destruct it.
2117*9880d681SAndroid Build Coastguard Worker APInt apdigit(getBitWidth(), 0);
2118*9880d681SAndroid Build Coastguard Worker APInt apradix(getBitWidth(), radix);
2119*9880d681SAndroid Build Coastguard Worker
2120*9880d681SAndroid Build Coastguard Worker // Enter digit traversal loop
2121*9880d681SAndroid Build Coastguard Worker for (StringRef::iterator e = str.end(); p != e; ++p) {
2122*9880d681SAndroid Build Coastguard Worker unsigned digit = getDigit(*p, radix);
2123*9880d681SAndroid Build Coastguard Worker assert(digit < radix && "Invalid character in digit string");
2124*9880d681SAndroid Build Coastguard Worker
2125*9880d681SAndroid Build Coastguard Worker // Shift or multiply the value by the radix
2126*9880d681SAndroid Build Coastguard Worker if (slen > 1) {
2127*9880d681SAndroid Build Coastguard Worker if (shift)
2128*9880d681SAndroid Build Coastguard Worker *this <<= shift;
2129*9880d681SAndroid Build Coastguard Worker else
2130*9880d681SAndroid Build Coastguard Worker *this *= apradix;
2131*9880d681SAndroid Build Coastguard Worker }
2132*9880d681SAndroid Build Coastguard Worker
2133*9880d681SAndroid Build Coastguard Worker // Add in the digit we just interpreted
2134*9880d681SAndroid Build Coastguard Worker if (apdigit.isSingleWord())
2135*9880d681SAndroid Build Coastguard Worker apdigit.VAL = digit;
2136*9880d681SAndroid Build Coastguard Worker else
2137*9880d681SAndroid Build Coastguard Worker apdigit.pVal[0] = digit;
2138*9880d681SAndroid Build Coastguard Worker *this += apdigit;
2139*9880d681SAndroid Build Coastguard Worker }
2140*9880d681SAndroid Build Coastguard Worker // If its negative, put it in two's complement form
2141*9880d681SAndroid Build Coastguard Worker if (isNeg) {
2142*9880d681SAndroid Build Coastguard Worker --(*this);
2143*9880d681SAndroid Build Coastguard Worker this->flipAllBits();
2144*9880d681SAndroid Build Coastguard Worker }
2145*9880d681SAndroid Build Coastguard Worker }
2146*9880d681SAndroid Build Coastguard Worker
toString(SmallVectorImpl<char> & Str,unsigned Radix,bool Signed,bool formatAsCLiteral) const2147*9880d681SAndroid Build Coastguard Worker void APInt::toString(SmallVectorImpl<char> &Str, unsigned Radix,
2148*9880d681SAndroid Build Coastguard Worker bool Signed, bool formatAsCLiteral) const {
2149*9880d681SAndroid Build Coastguard Worker assert((Radix == 10 || Radix == 8 || Radix == 16 || Radix == 2 ||
2150*9880d681SAndroid Build Coastguard Worker Radix == 36) &&
2151*9880d681SAndroid Build Coastguard Worker "Radix should be 2, 8, 10, 16, or 36!");
2152*9880d681SAndroid Build Coastguard Worker
2153*9880d681SAndroid Build Coastguard Worker const char *Prefix = "";
2154*9880d681SAndroid Build Coastguard Worker if (formatAsCLiteral) {
2155*9880d681SAndroid Build Coastguard Worker switch (Radix) {
2156*9880d681SAndroid Build Coastguard Worker case 2:
2157*9880d681SAndroid Build Coastguard Worker // Binary literals are a non-standard extension added in gcc 4.3:
2158*9880d681SAndroid Build Coastguard Worker // http://gcc.gnu.org/onlinedocs/gcc-4.3.0/gcc/Binary-constants.html
2159*9880d681SAndroid Build Coastguard Worker Prefix = "0b";
2160*9880d681SAndroid Build Coastguard Worker break;
2161*9880d681SAndroid Build Coastguard Worker case 8:
2162*9880d681SAndroid Build Coastguard Worker Prefix = "0";
2163*9880d681SAndroid Build Coastguard Worker break;
2164*9880d681SAndroid Build Coastguard Worker case 10:
2165*9880d681SAndroid Build Coastguard Worker break; // No prefix
2166*9880d681SAndroid Build Coastguard Worker case 16:
2167*9880d681SAndroid Build Coastguard Worker Prefix = "0x";
2168*9880d681SAndroid Build Coastguard Worker break;
2169*9880d681SAndroid Build Coastguard Worker default:
2170*9880d681SAndroid Build Coastguard Worker llvm_unreachable("Invalid radix!");
2171*9880d681SAndroid Build Coastguard Worker }
2172*9880d681SAndroid Build Coastguard Worker }
2173*9880d681SAndroid Build Coastguard Worker
2174*9880d681SAndroid Build Coastguard Worker // First, check for a zero value and just short circuit the logic below.
2175*9880d681SAndroid Build Coastguard Worker if (*this == 0) {
2176*9880d681SAndroid Build Coastguard Worker while (*Prefix) {
2177*9880d681SAndroid Build Coastguard Worker Str.push_back(*Prefix);
2178*9880d681SAndroid Build Coastguard Worker ++Prefix;
2179*9880d681SAndroid Build Coastguard Worker };
2180*9880d681SAndroid Build Coastguard Worker Str.push_back('0');
2181*9880d681SAndroid Build Coastguard Worker return;
2182*9880d681SAndroid Build Coastguard Worker }
2183*9880d681SAndroid Build Coastguard Worker
2184*9880d681SAndroid Build Coastguard Worker static const char Digits[] = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ";
2185*9880d681SAndroid Build Coastguard Worker
2186*9880d681SAndroid Build Coastguard Worker if (isSingleWord()) {
2187*9880d681SAndroid Build Coastguard Worker char Buffer[65];
2188*9880d681SAndroid Build Coastguard Worker char *BufPtr = Buffer+65;
2189*9880d681SAndroid Build Coastguard Worker
2190*9880d681SAndroid Build Coastguard Worker uint64_t N;
2191*9880d681SAndroid Build Coastguard Worker if (!Signed) {
2192*9880d681SAndroid Build Coastguard Worker N = getZExtValue();
2193*9880d681SAndroid Build Coastguard Worker } else {
2194*9880d681SAndroid Build Coastguard Worker int64_t I = getSExtValue();
2195*9880d681SAndroid Build Coastguard Worker if (I >= 0) {
2196*9880d681SAndroid Build Coastguard Worker N = I;
2197*9880d681SAndroid Build Coastguard Worker } else {
2198*9880d681SAndroid Build Coastguard Worker Str.push_back('-');
2199*9880d681SAndroid Build Coastguard Worker N = -(uint64_t)I;
2200*9880d681SAndroid Build Coastguard Worker }
2201*9880d681SAndroid Build Coastguard Worker }
2202*9880d681SAndroid Build Coastguard Worker
2203*9880d681SAndroid Build Coastguard Worker while (*Prefix) {
2204*9880d681SAndroid Build Coastguard Worker Str.push_back(*Prefix);
2205*9880d681SAndroid Build Coastguard Worker ++Prefix;
2206*9880d681SAndroid Build Coastguard Worker };
2207*9880d681SAndroid Build Coastguard Worker
2208*9880d681SAndroid Build Coastguard Worker while (N) {
2209*9880d681SAndroid Build Coastguard Worker *--BufPtr = Digits[N % Radix];
2210*9880d681SAndroid Build Coastguard Worker N /= Radix;
2211*9880d681SAndroid Build Coastguard Worker }
2212*9880d681SAndroid Build Coastguard Worker Str.append(BufPtr, Buffer+65);
2213*9880d681SAndroid Build Coastguard Worker return;
2214*9880d681SAndroid Build Coastguard Worker }
2215*9880d681SAndroid Build Coastguard Worker
2216*9880d681SAndroid Build Coastguard Worker APInt Tmp(*this);
2217*9880d681SAndroid Build Coastguard Worker
2218*9880d681SAndroid Build Coastguard Worker if (Signed && isNegative()) {
2219*9880d681SAndroid Build Coastguard Worker // They want to print the signed version and it is a negative value
2220*9880d681SAndroid Build Coastguard Worker // Flip the bits and add one to turn it into the equivalent positive
2221*9880d681SAndroid Build Coastguard Worker // value and put a '-' in the result.
2222*9880d681SAndroid Build Coastguard Worker Tmp.flipAllBits();
2223*9880d681SAndroid Build Coastguard Worker ++Tmp;
2224*9880d681SAndroid Build Coastguard Worker Str.push_back('-');
2225*9880d681SAndroid Build Coastguard Worker }
2226*9880d681SAndroid Build Coastguard Worker
2227*9880d681SAndroid Build Coastguard Worker while (*Prefix) {
2228*9880d681SAndroid Build Coastguard Worker Str.push_back(*Prefix);
2229*9880d681SAndroid Build Coastguard Worker ++Prefix;
2230*9880d681SAndroid Build Coastguard Worker };
2231*9880d681SAndroid Build Coastguard Worker
2232*9880d681SAndroid Build Coastguard Worker // We insert the digits backward, then reverse them to get the right order.
2233*9880d681SAndroid Build Coastguard Worker unsigned StartDig = Str.size();
2234*9880d681SAndroid Build Coastguard Worker
2235*9880d681SAndroid Build Coastguard Worker // For the 2, 8 and 16 bit cases, we can just shift instead of divide
2236*9880d681SAndroid Build Coastguard Worker // because the number of bits per digit (1, 3 and 4 respectively) divides
2237*9880d681SAndroid Build Coastguard Worker // equaly. We just shift until the value is zero.
2238*9880d681SAndroid Build Coastguard Worker if (Radix == 2 || Radix == 8 || Radix == 16) {
2239*9880d681SAndroid Build Coastguard Worker // Just shift tmp right for each digit width until it becomes zero
2240*9880d681SAndroid Build Coastguard Worker unsigned ShiftAmt = (Radix == 16 ? 4 : (Radix == 8 ? 3 : 1));
2241*9880d681SAndroid Build Coastguard Worker unsigned MaskAmt = Radix - 1;
2242*9880d681SAndroid Build Coastguard Worker
2243*9880d681SAndroid Build Coastguard Worker while (Tmp != 0) {
2244*9880d681SAndroid Build Coastguard Worker unsigned Digit = unsigned(Tmp.getRawData()[0]) & MaskAmt;
2245*9880d681SAndroid Build Coastguard Worker Str.push_back(Digits[Digit]);
2246*9880d681SAndroid Build Coastguard Worker Tmp = Tmp.lshr(ShiftAmt);
2247*9880d681SAndroid Build Coastguard Worker }
2248*9880d681SAndroid Build Coastguard Worker } else {
2249*9880d681SAndroid Build Coastguard Worker APInt divisor(Radix == 10? 4 : 8, Radix);
2250*9880d681SAndroid Build Coastguard Worker while (Tmp != 0) {
2251*9880d681SAndroid Build Coastguard Worker APInt APdigit(1, 0);
2252*9880d681SAndroid Build Coastguard Worker APInt tmp2(Tmp.getBitWidth(), 0);
2253*9880d681SAndroid Build Coastguard Worker divide(Tmp, Tmp.getNumWords(), divisor, divisor.getNumWords(), &tmp2,
2254*9880d681SAndroid Build Coastguard Worker &APdigit);
2255*9880d681SAndroid Build Coastguard Worker unsigned Digit = (unsigned)APdigit.getZExtValue();
2256*9880d681SAndroid Build Coastguard Worker assert(Digit < Radix && "divide failed");
2257*9880d681SAndroid Build Coastguard Worker Str.push_back(Digits[Digit]);
2258*9880d681SAndroid Build Coastguard Worker Tmp = tmp2;
2259*9880d681SAndroid Build Coastguard Worker }
2260*9880d681SAndroid Build Coastguard Worker }
2261*9880d681SAndroid Build Coastguard Worker
2262*9880d681SAndroid Build Coastguard Worker // Reverse the digits before returning.
2263*9880d681SAndroid Build Coastguard Worker std::reverse(Str.begin()+StartDig, Str.end());
2264*9880d681SAndroid Build Coastguard Worker }
2265*9880d681SAndroid Build Coastguard Worker
2266*9880d681SAndroid Build Coastguard Worker /// Returns the APInt as a std::string. Note that this is an inefficient method.
2267*9880d681SAndroid Build Coastguard Worker /// It is better to pass in a SmallVector/SmallString to the methods above.
toString(unsigned Radix=10,bool Signed=true) const2268*9880d681SAndroid Build Coastguard Worker std::string APInt::toString(unsigned Radix = 10, bool Signed = true) const {
2269*9880d681SAndroid Build Coastguard Worker SmallString<40> S;
2270*9880d681SAndroid Build Coastguard Worker toString(S, Radix, Signed, /* formatAsCLiteral = */false);
2271*9880d681SAndroid Build Coastguard Worker return S.str();
2272*9880d681SAndroid Build Coastguard Worker }
2273*9880d681SAndroid Build Coastguard Worker
2274*9880d681SAndroid Build Coastguard Worker
dump() const2275*9880d681SAndroid Build Coastguard Worker LLVM_DUMP_METHOD void APInt::dump() const {
2276*9880d681SAndroid Build Coastguard Worker SmallString<40> S, U;
2277*9880d681SAndroid Build Coastguard Worker this->toStringUnsigned(U);
2278*9880d681SAndroid Build Coastguard Worker this->toStringSigned(S);
2279*9880d681SAndroid Build Coastguard Worker dbgs() << "APInt(" << BitWidth << "b, "
2280*9880d681SAndroid Build Coastguard Worker << U << "u " << S << "s)";
2281*9880d681SAndroid Build Coastguard Worker }
2282*9880d681SAndroid Build Coastguard Worker
print(raw_ostream & OS,bool isSigned) const2283*9880d681SAndroid Build Coastguard Worker void APInt::print(raw_ostream &OS, bool isSigned) const {
2284*9880d681SAndroid Build Coastguard Worker SmallString<40> S;
2285*9880d681SAndroid Build Coastguard Worker this->toString(S, 10, isSigned, /* formatAsCLiteral = */false);
2286*9880d681SAndroid Build Coastguard Worker OS << S;
2287*9880d681SAndroid Build Coastguard Worker }
2288*9880d681SAndroid Build Coastguard Worker
2289*9880d681SAndroid Build Coastguard Worker // This implements a variety of operations on a representation of
2290*9880d681SAndroid Build Coastguard Worker // arbitrary precision, two's-complement, bignum integer values.
2291*9880d681SAndroid Build Coastguard Worker
2292*9880d681SAndroid Build Coastguard Worker // Assumed by lowHalf, highHalf, partMSB and partLSB. A fairly safe
2293*9880d681SAndroid Build Coastguard Worker // and unrestricting assumption.
2294*9880d681SAndroid Build Coastguard Worker static_assert(integerPartWidth % 2 == 0, "Part width must be divisible by 2!");
2295*9880d681SAndroid Build Coastguard Worker
2296*9880d681SAndroid Build Coastguard Worker /* Some handy functions local to this file. */
2297*9880d681SAndroid Build Coastguard Worker namespace {
2298*9880d681SAndroid Build Coastguard Worker
2299*9880d681SAndroid Build Coastguard Worker /* Returns the integer part with the least significant BITS set.
2300*9880d681SAndroid Build Coastguard Worker BITS cannot be zero. */
2301*9880d681SAndroid Build Coastguard Worker static inline integerPart
lowBitMask(unsigned int bits)2302*9880d681SAndroid Build Coastguard Worker lowBitMask(unsigned int bits)
2303*9880d681SAndroid Build Coastguard Worker {
2304*9880d681SAndroid Build Coastguard Worker assert(bits != 0 && bits <= integerPartWidth);
2305*9880d681SAndroid Build Coastguard Worker
2306*9880d681SAndroid Build Coastguard Worker return ~(integerPart) 0 >> (integerPartWidth - bits);
2307*9880d681SAndroid Build Coastguard Worker }
2308*9880d681SAndroid Build Coastguard Worker
2309*9880d681SAndroid Build Coastguard Worker /* Returns the value of the lower half of PART. */
2310*9880d681SAndroid Build Coastguard Worker static inline integerPart
lowHalf(integerPart part)2311*9880d681SAndroid Build Coastguard Worker lowHalf(integerPart part)
2312*9880d681SAndroid Build Coastguard Worker {
2313*9880d681SAndroid Build Coastguard Worker return part & lowBitMask(integerPartWidth / 2);
2314*9880d681SAndroid Build Coastguard Worker }
2315*9880d681SAndroid Build Coastguard Worker
2316*9880d681SAndroid Build Coastguard Worker /* Returns the value of the upper half of PART. */
2317*9880d681SAndroid Build Coastguard Worker static inline integerPart
highHalf(integerPart part)2318*9880d681SAndroid Build Coastguard Worker highHalf(integerPart part)
2319*9880d681SAndroid Build Coastguard Worker {
2320*9880d681SAndroid Build Coastguard Worker return part >> (integerPartWidth / 2);
2321*9880d681SAndroid Build Coastguard Worker }
2322*9880d681SAndroid Build Coastguard Worker
2323*9880d681SAndroid Build Coastguard Worker /* Returns the bit number of the most significant set bit of a part.
2324*9880d681SAndroid Build Coastguard Worker If the input number has no bits set -1U is returned. */
2325*9880d681SAndroid Build Coastguard Worker static unsigned int
partMSB(integerPart value)2326*9880d681SAndroid Build Coastguard Worker partMSB(integerPart value)
2327*9880d681SAndroid Build Coastguard Worker {
2328*9880d681SAndroid Build Coastguard Worker return findLastSet(value, ZB_Max);
2329*9880d681SAndroid Build Coastguard Worker }
2330*9880d681SAndroid Build Coastguard Worker
2331*9880d681SAndroid Build Coastguard Worker /* Returns the bit number of the least significant set bit of a
2332*9880d681SAndroid Build Coastguard Worker part. If the input number has no bits set -1U is returned. */
2333*9880d681SAndroid Build Coastguard Worker static unsigned int
partLSB(integerPart value)2334*9880d681SAndroid Build Coastguard Worker partLSB(integerPart value)
2335*9880d681SAndroid Build Coastguard Worker {
2336*9880d681SAndroid Build Coastguard Worker return findFirstSet(value, ZB_Max);
2337*9880d681SAndroid Build Coastguard Worker }
2338*9880d681SAndroid Build Coastguard Worker }
2339*9880d681SAndroid Build Coastguard Worker
2340*9880d681SAndroid Build Coastguard Worker /* Sets the least significant part of a bignum to the input value, and
2341*9880d681SAndroid Build Coastguard Worker zeroes out higher parts. */
2342*9880d681SAndroid Build Coastguard Worker void
tcSet(integerPart * dst,integerPart part,unsigned int parts)2343*9880d681SAndroid Build Coastguard Worker APInt::tcSet(integerPart *dst, integerPart part, unsigned int parts)
2344*9880d681SAndroid Build Coastguard Worker {
2345*9880d681SAndroid Build Coastguard Worker unsigned int i;
2346*9880d681SAndroid Build Coastguard Worker
2347*9880d681SAndroid Build Coastguard Worker assert(parts > 0);
2348*9880d681SAndroid Build Coastguard Worker
2349*9880d681SAndroid Build Coastguard Worker dst[0] = part;
2350*9880d681SAndroid Build Coastguard Worker for (i = 1; i < parts; i++)
2351*9880d681SAndroid Build Coastguard Worker dst[i] = 0;
2352*9880d681SAndroid Build Coastguard Worker }
2353*9880d681SAndroid Build Coastguard Worker
2354*9880d681SAndroid Build Coastguard Worker /* Assign one bignum to another. */
2355*9880d681SAndroid Build Coastguard Worker void
tcAssign(integerPart * dst,const integerPart * src,unsigned int parts)2356*9880d681SAndroid Build Coastguard Worker APInt::tcAssign(integerPart *dst, const integerPart *src, unsigned int parts)
2357*9880d681SAndroid Build Coastguard Worker {
2358*9880d681SAndroid Build Coastguard Worker unsigned int i;
2359*9880d681SAndroid Build Coastguard Worker
2360*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2361*9880d681SAndroid Build Coastguard Worker dst[i] = src[i];
2362*9880d681SAndroid Build Coastguard Worker }
2363*9880d681SAndroid Build Coastguard Worker
2364*9880d681SAndroid Build Coastguard Worker /* Returns true if a bignum is zero, false otherwise. */
2365*9880d681SAndroid Build Coastguard Worker bool
tcIsZero(const integerPart * src,unsigned int parts)2366*9880d681SAndroid Build Coastguard Worker APInt::tcIsZero(const integerPart *src, unsigned int parts)
2367*9880d681SAndroid Build Coastguard Worker {
2368*9880d681SAndroid Build Coastguard Worker unsigned int i;
2369*9880d681SAndroid Build Coastguard Worker
2370*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2371*9880d681SAndroid Build Coastguard Worker if (src[i])
2372*9880d681SAndroid Build Coastguard Worker return false;
2373*9880d681SAndroid Build Coastguard Worker
2374*9880d681SAndroid Build Coastguard Worker return true;
2375*9880d681SAndroid Build Coastguard Worker }
2376*9880d681SAndroid Build Coastguard Worker
2377*9880d681SAndroid Build Coastguard Worker /* Extract the given bit of a bignum; returns 0 or 1. */
2378*9880d681SAndroid Build Coastguard Worker int
tcExtractBit(const integerPart * parts,unsigned int bit)2379*9880d681SAndroid Build Coastguard Worker APInt::tcExtractBit(const integerPart *parts, unsigned int bit)
2380*9880d681SAndroid Build Coastguard Worker {
2381*9880d681SAndroid Build Coastguard Worker return (parts[bit / integerPartWidth] &
2382*9880d681SAndroid Build Coastguard Worker ((integerPart) 1 << bit % integerPartWidth)) != 0;
2383*9880d681SAndroid Build Coastguard Worker }
2384*9880d681SAndroid Build Coastguard Worker
2385*9880d681SAndroid Build Coastguard Worker /* Set the given bit of a bignum. */
2386*9880d681SAndroid Build Coastguard Worker void
tcSetBit(integerPart * parts,unsigned int bit)2387*9880d681SAndroid Build Coastguard Worker APInt::tcSetBit(integerPart *parts, unsigned int bit)
2388*9880d681SAndroid Build Coastguard Worker {
2389*9880d681SAndroid Build Coastguard Worker parts[bit / integerPartWidth] |= (integerPart) 1 << (bit % integerPartWidth);
2390*9880d681SAndroid Build Coastguard Worker }
2391*9880d681SAndroid Build Coastguard Worker
2392*9880d681SAndroid Build Coastguard Worker /* Clears the given bit of a bignum. */
2393*9880d681SAndroid Build Coastguard Worker void
tcClearBit(integerPart * parts,unsigned int bit)2394*9880d681SAndroid Build Coastguard Worker APInt::tcClearBit(integerPart *parts, unsigned int bit)
2395*9880d681SAndroid Build Coastguard Worker {
2396*9880d681SAndroid Build Coastguard Worker parts[bit / integerPartWidth] &=
2397*9880d681SAndroid Build Coastguard Worker ~((integerPart) 1 << (bit % integerPartWidth));
2398*9880d681SAndroid Build Coastguard Worker }
2399*9880d681SAndroid Build Coastguard Worker
2400*9880d681SAndroid Build Coastguard Worker /* Returns the bit number of the least significant set bit of a
2401*9880d681SAndroid Build Coastguard Worker number. If the input number has no bits set -1U is returned. */
2402*9880d681SAndroid Build Coastguard Worker unsigned int
tcLSB(const integerPart * parts,unsigned int n)2403*9880d681SAndroid Build Coastguard Worker APInt::tcLSB(const integerPart *parts, unsigned int n)
2404*9880d681SAndroid Build Coastguard Worker {
2405*9880d681SAndroid Build Coastguard Worker unsigned int i, lsb;
2406*9880d681SAndroid Build Coastguard Worker
2407*9880d681SAndroid Build Coastguard Worker for (i = 0; i < n; i++) {
2408*9880d681SAndroid Build Coastguard Worker if (parts[i] != 0) {
2409*9880d681SAndroid Build Coastguard Worker lsb = partLSB(parts[i]);
2410*9880d681SAndroid Build Coastguard Worker
2411*9880d681SAndroid Build Coastguard Worker return lsb + i * integerPartWidth;
2412*9880d681SAndroid Build Coastguard Worker }
2413*9880d681SAndroid Build Coastguard Worker }
2414*9880d681SAndroid Build Coastguard Worker
2415*9880d681SAndroid Build Coastguard Worker return -1U;
2416*9880d681SAndroid Build Coastguard Worker }
2417*9880d681SAndroid Build Coastguard Worker
2418*9880d681SAndroid Build Coastguard Worker /* Returns the bit number of the most significant set bit of a number.
2419*9880d681SAndroid Build Coastguard Worker If the input number has no bits set -1U is returned. */
2420*9880d681SAndroid Build Coastguard Worker unsigned int
tcMSB(const integerPart * parts,unsigned int n)2421*9880d681SAndroid Build Coastguard Worker APInt::tcMSB(const integerPart *parts, unsigned int n)
2422*9880d681SAndroid Build Coastguard Worker {
2423*9880d681SAndroid Build Coastguard Worker unsigned int msb;
2424*9880d681SAndroid Build Coastguard Worker
2425*9880d681SAndroid Build Coastguard Worker do {
2426*9880d681SAndroid Build Coastguard Worker --n;
2427*9880d681SAndroid Build Coastguard Worker
2428*9880d681SAndroid Build Coastguard Worker if (parts[n] != 0) {
2429*9880d681SAndroid Build Coastguard Worker msb = partMSB(parts[n]);
2430*9880d681SAndroid Build Coastguard Worker
2431*9880d681SAndroid Build Coastguard Worker return msb + n * integerPartWidth;
2432*9880d681SAndroid Build Coastguard Worker }
2433*9880d681SAndroid Build Coastguard Worker } while (n);
2434*9880d681SAndroid Build Coastguard Worker
2435*9880d681SAndroid Build Coastguard Worker return -1U;
2436*9880d681SAndroid Build Coastguard Worker }
2437*9880d681SAndroid Build Coastguard Worker
2438*9880d681SAndroid Build Coastguard Worker /* Copy the bit vector of width srcBITS from SRC, starting at bit
2439*9880d681SAndroid Build Coastguard Worker srcLSB, to DST, of dstCOUNT parts, such that the bit srcLSB becomes
2440*9880d681SAndroid Build Coastguard Worker the least significant bit of DST. All high bits above srcBITS in
2441*9880d681SAndroid Build Coastguard Worker DST are zero-filled. */
2442*9880d681SAndroid Build Coastguard Worker void
tcExtract(integerPart * dst,unsigned int dstCount,const integerPart * src,unsigned int srcBits,unsigned int srcLSB)2443*9880d681SAndroid Build Coastguard Worker APInt::tcExtract(integerPart *dst, unsigned int dstCount,const integerPart *src,
2444*9880d681SAndroid Build Coastguard Worker unsigned int srcBits, unsigned int srcLSB)
2445*9880d681SAndroid Build Coastguard Worker {
2446*9880d681SAndroid Build Coastguard Worker unsigned int firstSrcPart, dstParts, shift, n;
2447*9880d681SAndroid Build Coastguard Worker
2448*9880d681SAndroid Build Coastguard Worker dstParts = (srcBits + integerPartWidth - 1) / integerPartWidth;
2449*9880d681SAndroid Build Coastguard Worker assert(dstParts <= dstCount);
2450*9880d681SAndroid Build Coastguard Worker
2451*9880d681SAndroid Build Coastguard Worker firstSrcPart = srcLSB / integerPartWidth;
2452*9880d681SAndroid Build Coastguard Worker tcAssign (dst, src + firstSrcPart, dstParts);
2453*9880d681SAndroid Build Coastguard Worker
2454*9880d681SAndroid Build Coastguard Worker shift = srcLSB % integerPartWidth;
2455*9880d681SAndroid Build Coastguard Worker tcShiftRight (dst, dstParts, shift);
2456*9880d681SAndroid Build Coastguard Worker
2457*9880d681SAndroid Build Coastguard Worker /* We now have (dstParts * integerPartWidth - shift) bits from SRC
2458*9880d681SAndroid Build Coastguard Worker in DST. If this is less that srcBits, append the rest, else
2459*9880d681SAndroid Build Coastguard Worker clear the high bits. */
2460*9880d681SAndroid Build Coastguard Worker n = dstParts * integerPartWidth - shift;
2461*9880d681SAndroid Build Coastguard Worker if (n < srcBits) {
2462*9880d681SAndroid Build Coastguard Worker integerPart mask = lowBitMask (srcBits - n);
2463*9880d681SAndroid Build Coastguard Worker dst[dstParts - 1] |= ((src[firstSrcPart + dstParts] & mask)
2464*9880d681SAndroid Build Coastguard Worker << n % integerPartWidth);
2465*9880d681SAndroid Build Coastguard Worker } else if (n > srcBits) {
2466*9880d681SAndroid Build Coastguard Worker if (srcBits % integerPartWidth)
2467*9880d681SAndroid Build Coastguard Worker dst[dstParts - 1] &= lowBitMask (srcBits % integerPartWidth);
2468*9880d681SAndroid Build Coastguard Worker }
2469*9880d681SAndroid Build Coastguard Worker
2470*9880d681SAndroid Build Coastguard Worker /* Clear high parts. */
2471*9880d681SAndroid Build Coastguard Worker while (dstParts < dstCount)
2472*9880d681SAndroid Build Coastguard Worker dst[dstParts++] = 0;
2473*9880d681SAndroid Build Coastguard Worker }
2474*9880d681SAndroid Build Coastguard Worker
2475*9880d681SAndroid Build Coastguard Worker /* DST += RHS + C where C is zero or one. Returns the carry flag. */
2476*9880d681SAndroid Build Coastguard Worker integerPart
tcAdd(integerPart * dst,const integerPart * rhs,integerPart c,unsigned int parts)2477*9880d681SAndroid Build Coastguard Worker APInt::tcAdd(integerPart *dst, const integerPart *rhs,
2478*9880d681SAndroid Build Coastguard Worker integerPart c, unsigned int parts)
2479*9880d681SAndroid Build Coastguard Worker {
2480*9880d681SAndroid Build Coastguard Worker unsigned int i;
2481*9880d681SAndroid Build Coastguard Worker
2482*9880d681SAndroid Build Coastguard Worker assert(c <= 1);
2483*9880d681SAndroid Build Coastguard Worker
2484*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++) {
2485*9880d681SAndroid Build Coastguard Worker integerPart l;
2486*9880d681SAndroid Build Coastguard Worker
2487*9880d681SAndroid Build Coastguard Worker l = dst[i];
2488*9880d681SAndroid Build Coastguard Worker if (c) {
2489*9880d681SAndroid Build Coastguard Worker dst[i] += rhs[i] + 1;
2490*9880d681SAndroid Build Coastguard Worker c = (dst[i] <= l);
2491*9880d681SAndroid Build Coastguard Worker } else {
2492*9880d681SAndroid Build Coastguard Worker dst[i] += rhs[i];
2493*9880d681SAndroid Build Coastguard Worker c = (dst[i] < l);
2494*9880d681SAndroid Build Coastguard Worker }
2495*9880d681SAndroid Build Coastguard Worker }
2496*9880d681SAndroid Build Coastguard Worker
2497*9880d681SAndroid Build Coastguard Worker return c;
2498*9880d681SAndroid Build Coastguard Worker }
2499*9880d681SAndroid Build Coastguard Worker
2500*9880d681SAndroid Build Coastguard Worker /* DST -= RHS + C where C is zero or one. Returns the carry flag. */
2501*9880d681SAndroid Build Coastguard Worker integerPart
tcSubtract(integerPart * dst,const integerPart * rhs,integerPart c,unsigned int parts)2502*9880d681SAndroid Build Coastguard Worker APInt::tcSubtract(integerPart *dst, const integerPart *rhs,
2503*9880d681SAndroid Build Coastguard Worker integerPart c, unsigned int parts)
2504*9880d681SAndroid Build Coastguard Worker {
2505*9880d681SAndroid Build Coastguard Worker unsigned int i;
2506*9880d681SAndroid Build Coastguard Worker
2507*9880d681SAndroid Build Coastguard Worker assert(c <= 1);
2508*9880d681SAndroid Build Coastguard Worker
2509*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++) {
2510*9880d681SAndroid Build Coastguard Worker integerPart l;
2511*9880d681SAndroid Build Coastguard Worker
2512*9880d681SAndroid Build Coastguard Worker l = dst[i];
2513*9880d681SAndroid Build Coastguard Worker if (c) {
2514*9880d681SAndroid Build Coastguard Worker dst[i] -= rhs[i] + 1;
2515*9880d681SAndroid Build Coastguard Worker c = (dst[i] >= l);
2516*9880d681SAndroid Build Coastguard Worker } else {
2517*9880d681SAndroid Build Coastguard Worker dst[i] -= rhs[i];
2518*9880d681SAndroid Build Coastguard Worker c = (dst[i] > l);
2519*9880d681SAndroid Build Coastguard Worker }
2520*9880d681SAndroid Build Coastguard Worker }
2521*9880d681SAndroid Build Coastguard Worker
2522*9880d681SAndroid Build Coastguard Worker return c;
2523*9880d681SAndroid Build Coastguard Worker }
2524*9880d681SAndroid Build Coastguard Worker
2525*9880d681SAndroid Build Coastguard Worker /* Negate a bignum in-place. */
2526*9880d681SAndroid Build Coastguard Worker void
tcNegate(integerPart * dst,unsigned int parts)2527*9880d681SAndroid Build Coastguard Worker APInt::tcNegate(integerPart *dst, unsigned int parts)
2528*9880d681SAndroid Build Coastguard Worker {
2529*9880d681SAndroid Build Coastguard Worker tcComplement(dst, parts);
2530*9880d681SAndroid Build Coastguard Worker tcIncrement(dst, parts);
2531*9880d681SAndroid Build Coastguard Worker }
2532*9880d681SAndroid Build Coastguard Worker
2533*9880d681SAndroid Build Coastguard Worker /* DST += SRC * MULTIPLIER + CARRY if add is true
2534*9880d681SAndroid Build Coastguard Worker DST = SRC * MULTIPLIER + CARRY if add is false
2535*9880d681SAndroid Build Coastguard Worker
2536*9880d681SAndroid Build Coastguard Worker Requires 0 <= DSTPARTS <= SRCPARTS + 1. If DST overlaps SRC
2537*9880d681SAndroid Build Coastguard Worker they must start at the same point, i.e. DST == SRC.
2538*9880d681SAndroid Build Coastguard Worker
2539*9880d681SAndroid Build Coastguard Worker If DSTPARTS == SRCPARTS + 1 no overflow occurs and zero is
2540*9880d681SAndroid Build Coastguard Worker returned. Otherwise DST is filled with the least significant
2541*9880d681SAndroid Build Coastguard Worker DSTPARTS parts of the result, and if all of the omitted higher
2542*9880d681SAndroid Build Coastguard Worker parts were zero return zero, otherwise overflow occurred and
2543*9880d681SAndroid Build Coastguard Worker return one. */
2544*9880d681SAndroid Build Coastguard Worker int
tcMultiplyPart(integerPart * dst,const integerPart * src,integerPart multiplier,integerPart carry,unsigned int srcParts,unsigned int dstParts,bool add)2545*9880d681SAndroid Build Coastguard Worker APInt::tcMultiplyPart(integerPart *dst, const integerPart *src,
2546*9880d681SAndroid Build Coastguard Worker integerPart multiplier, integerPart carry,
2547*9880d681SAndroid Build Coastguard Worker unsigned int srcParts, unsigned int dstParts,
2548*9880d681SAndroid Build Coastguard Worker bool add)
2549*9880d681SAndroid Build Coastguard Worker {
2550*9880d681SAndroid Build Coastguard Worker unsigned int i, n;
2551*9880d681SAndroid Build Coastguard Worker
2552*9880d681SAndroid Build Coastguard Worker /* Otherwise our writes of DST kill our later reads of SRC. */
2553*9880d681SAndroid Build Coastguard Worker assert(dst <= src || dst >= src + srcParts);
2554*9880d681SAndroid Build Coastguard Worker assert(dstParts <= srcParts + 1);
2555*9880d681SAndroid Build Coastguard Worker
2556*9880d681SAndroid Build Coastguard Worker /* N loops; minimum of dstParts and srcParts. */
2557*9880d681SAndroid Build Coastguard Worker n = dstParts < srcParts ? dstParts: srcParts;
2558*9880d681SAndroid Build Coastguard Worker
2559*9880d681SAndroid Build Coastguard Worker for (i = 0; i < n; i++) {
2560*9880d681SAndroid Build Coastguard Worker integerPart low, mid, high, srcPart;
2561*9880d681SAndroid Build Coastguard Worker
2562*9880d681SAndroid Build Coastguard Worker /* [ LOW, HIGH ] = MULTIPLIER * SRC[i] + DST[i] + CARRY.
2563*9880d681SAndroid Build Coastguard Worker
2564*9880d681SAndroid Build Coastguard Worker This cannot overflow, because
2565*9880d681SAndroid Build Coastguard Worker
2566*9880d681SAndroid Build Coastguard Worker (n - 1) * (n - 1) + 2 (n - 1) = (n - 1) * (n + 1)
2567*9880d681SAndroid Build Coastguard Worker
2568*9880d681SAndroid Build Coastguard Worker which is less than n^2. */
2569*9880d681SAndroid Build Coastguard Worker
2570*9880d681SAndroid Build Coastguard Worker srcPart = src[i];
2571*9880d681SAndroid Build Coastguard Worker
2572*9880d681SAndroid Build Coastguard Worker if (multiplier == 0 || srcPart == 0) {
2573*9880d681SAndroid Build Coastguard Worker low = carry;
2574*9880d681SAndroid Build Coastguard Worker high = 0;
2575*9880d681SAndroid Build Coastguard Worker } else {
2576*9880d681SAndroid Build Coastguard Worker low = lowHalf(srcPart) * lowHalf(multiplier);
2577*9880d681SAndroid Build Coastguard Worker high = highHalf(srcPart) * highHalf(multiplier);
2578*9880d681SAndroid Build Coastguard Worker
2579*9880d681SAndroid Build Coastguard Worker mid = lowHalf(srcPart) * highHalf(multiplier);
2580*9880d681SAndroid Build Coastguard Worker high += highHalf(mid);
2581*9880d681SAndroid Build Coastguard Worker mid <<= integerPartWidth / 2;
2582*9880d681SAndroid Build Coastguard Worker if (low + mid < low)
2583*9880d681SAndroid Build Coastguard Worker high++;
2584*9880d681SAndroid Build Coastguard Worker low += mid;
2585*9880d681SAndroid Build Coastguard Worker
2586*9880d681SAndroid Build Coastguard Worker mid = highHalf(srcPart) * lowHalf(multiplier);
2587*9880d681SAndroid Build Coastguard Worker high += highHalf(mid);
2588*9880d681SAndroid Build Coastguard Worker mid <<= integerPartWidth / 2;
2589*9880d681SAndroid Build Coastguard Worker if (low + mid < low)
2590*9880d681SAndroid Build Coastguard Worker high++;
2591*9880d681SAndroid Build Coastguard Worker low += mid;
2592*9880d681SAndroid Build Coastguard Worker
2593*9880d681SAndroid Build Coastguard Worker /* Now add carry. */
2594*9880d681SAndroid Build Coastguard Worker if (low + carry < low)
2595*9880d681SAndroid Build Coastguard Worker high++;
2596*9880d681SAndroid Build Coastguard Worker low += carry;
2597*9880d681SAndroid Build Coastguard Worker }
2598*9880d681SAndroid Build Coastguard Worker
2599*9880d681SAndroid Build Coastguard Worker if (add) {
2600*9880d681SAndroid Build Coastguard Worker /* And now DST[i], and store the new low part there. */
2601*9880d681SAndroid Build Coastguard Worker if (low + dst[i] < low)
2602*9880d681SAndroid Build Coastguard Worker high++;
2603*9880d681SAndroid Build Coastguard Worker dst[i] += low;
2604*9880d681SAndroid Build Coastguard Worker } else
2605*9880d681SAndroid Build Coastguard Worker dst[i] = low;
2606*9880d681SAndroid Build Coastguard Worker
2607*9880d681SAndroid Build Coastguard Worker carry = high;
2608*9880d681SAndroid Build Coastguard Worker }
2609*9880d681SAndroid Build Coastguard Worker
2610*9880d681SAndroid Build Coastguard Worker if (i < dstParts) {
2611*9880d681SAndroid Build Coastguard Worker /* Full multiplication, there is no overflow. */
2612*9880d681SAndroid Build Coastguard Worker assert(i + 1 == dstParts);
2613*9880d681SAndroid Build Coastguard Worker dst[i] = carry;
2614*9880d681SAndroid Build Coastguard Worker return 0;
2615*9880d681SAndroid Build Coastguard Worker } else {
2616*9880d681SAndroid Build Coastguard Worker /* We overflowed if there is carry. */
2617*9880d681SAndroid Build Coastguard Worker if (carry)
2618*9880d681SAndroid Build Coastguard Worker return 1;
2619*9880d681SAndroid Build Coastguard Worker
2620*9880d681SAndroid Build Coastguard Worker /* We would overflow if any significant unwritten parts would be
2621*9880d681SAndroid Build Coastguard Worker non-zero. This is true if any remaining src parts are non-zero
2622*9880d681SAndroid Build Coastguard Worker and the multiplier is non-zero. */
2623*9880d681SAndroid Build Coastguard Worker if (multiplier)
2624*9880d681SAndroid Build Coastguard Worker for (; i < srcParts; i++)
2625*9880d681SAndroid Build Coastguard Worker if (src[i])
2626*9880d681SAndroid Build Coastguard Worker return 1;
2627*9880d681SAndroid Build Coastguard Worker
2628*9880d681SAndroid Build Coastguard Worker /* We fitted in the narrow destination. */
2629*9880d681SAndroid Build Coastguard Worker return 0;
2630*9880d681SAndroid Build Coastguard Worker }
2631*9880d681SAndroid Build Coastguard Worker }
2632*9880d681SAndroid Build Coastguard Worker
2633*9880d681SAndroid Build Coastguard Worker /* DST = LHS * RHS, where DST has the same width as the operands and
2634*9880d681SAndroid Build Coastguard Worker is filled with the least significant parts of the result. Returns
2635*9880d681SAndroid Build Coastguard Worker one if overflow occurred, otherwise zero. DST must be disjoint
2636*9880d681SAndroid Build Coastguard Worker from both operands. */
2637*9880d681SAndroid Build Coastguard Worker int
tcMultiply(integerPart * dst,const integerPart * lhs,const integerPart * rhs,unsigned int parts)2638*9880d681SAndroid Build Coastguard Worker APInt::tcMultiply(integerPart *dst, const integerPart *lhs,
2639*9880d681SAndroid Build Coastguard Worker const integerPart *rhs, unsigned int parts)
2640*9880d681SAndroid Build Coastguard Worker {
2641*9880d681SAndroid Build Coastguard Worker unsigned int i;
2642*9880d681SAndroid Build Coastguard Worker int overflow;
2643*9880d681SAndroid Build Coastguard Worker
2644*9880d681SAndroid Build Coastguard Worker assert(dst != lhs && dst != rhs);
2645*9880d681SAndroid Build Coastguard Worker
2646*9880d681SAndroid Build Coastguard Worker overflow = 0;
2647*9880d681SAndroid Build Coastguard Worker tcSet(dst, 0, parts);
2648*9880d681SAndroid Build Coastguard Worker
2649*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2650*9880d681SAndroid Build Coastguard Worker overflow |= tcMultiplyPart(&dst[i], lhs, rhs[i], 0, parts,
2651*9880d681SAndroid Build Coastguard Worker parts - i, true);
2652*9880d681SAndroid Build Coastguard Worker
2653*9880d681SAndroid Build Coastguard Worker return overflow;
2654*9880d681SAndroid Build Coastguard Worker }
2655*9880d681SAndroid Build Coastguard Worker
2656*9880d681SAndroid Build Coastguard Worker /* DST = LHS * RHS, where DST has width the sum of the widths of the
2657*9880d681SAndroid Build Coastguard Worker operands. No overflow occurs. DST must be disjoint from both
2658*9880d681SAndroid Build Coastguard Worker operands. Returns the number of parts required to hold the
2659*9880d681SAndroid Build Coastguard Worker result. */
2660*9880d681SAndroid Build Coastguard Worker unsigned int
tcFullMultiply(integerPart * dst,const integerPart * lhs,const integerPart * rhs,unsigned int lhsParts,unsigned int rhsParts)2661*9880d681SAndroid Build Coastguard Worker APInt::tcFullMultiply(integerPart *dst, const integerPart *lhs,
2662*9880d681SAndroid Build Coastguard Worker const integerPart *rhs, unsigned int lhsParts,
2663*9880d681SAndroid Build Coastguard Worker unsigned int rhsParts)
2664*9880d681SAndroid Build Coastguard Worker {
2665*9880d681SAndroid Build Coastguard Worker /* Put the narrower number on the LHS for less loops below. */
2666*9880d681SAndroid Build Coastguard Worker if (lhsParts > rhsParts) {
2667*9880d681SAndroid Build Coastguard Worker return tcFullMultiply (dst, rhs, lhs, rhsParts, lhsParts);
2668*9880d681SAndroid Build Coastguard Worker } else {
2669*9880d681SAndroid Build Coastguard Worker unsigned int n;
2670*9880d681SAndroid Build Coastguard Worker
2671*9880d681SAndroid Build Coastguard Worker assert(dst != lhs && dst != rhs);
2672*9880d681SAndroid Build Coastguard Worker
2673*9880d681SAndroid Build Coastguard Worker tcSet(dst, 0, rhsParts);
2674*9880d681SAndroid Build Coastguard Worker
2675*9880d681SAndroid Build Coastguard Worker for (n = 0; n < lhsParts; n++)
2676*9880d681SAndroid Build Coastguard Worker tcMultiplyPart(&dst[n], rhs, lhs[n], 0, rhsParts, rhsParts + 1, true);
2677*9880d681SAndroid Build Coastguard Worker
2678*9880d681SAndroid Build Coastguard Worker n = lhsParts + rhsParts;
2679*9880d681SAndroid Build Coastguard Worker
2680*9880d681SAndroid Build Coastguard Worker return n - (dst[n - 1] == 0);
2681*9880d681SAndroid Build Coastguard Worker }
2682*9880d681SAndroid Build Coastguard Worker }
2683*9880d681SAndroid Build Coastguard Worker
2684*9880d681SAndroid Build Coastguard Worker /* If RHS is zero LHS and REMAINDER are left unchanged, return one.
2685*9880d681SAndroid Build Coastguard Worker Otherwise set LHS to LHS / RHS with the fractional part discarded,
2686*9880d681SAndroid Build Coastguard Worker set REMAINDER to the remainder, return zero. i.e.
2687*9880d681SAndroid Build Coastguard Worker
2688*9880d681SAndroid Build Coastguard Worker OLD_LHS = RHS * LHS + REMAINDER
2689*9880d681SAndroid Build Coastguard Worker
2690*9880d681SAndroid Build Coastguard Worker SCRATCH is a bignum of the same size as the operands and result for
2691*9880d681SAndroid Build Coastguard Worker use by the routine; its contents need not be initialized and are
2692*9880d681SAndroid Build Coastguard Worker destroyed. LHS, REMAINDER and SCRATCH must be distinct.
2693*9880d681SAndroid Build Coastguard Worker */
2694*9880d681SAndroid Build Coastguard Worker int
tcDivide(integerPart * lhs,const integerPart * rhs,integerPart * remainder,integerPart * srhs,unsigned int parts)2695*9880d681SAndroid Build Coastguard Worker APInt::tcDivide(integerPart *lhs, const integerPart *rhs,
2696*9880d681SAndroid Build Coastguard Worker integerPart *remainder, integerPart *srhs,
2697*9880d681SAndroid Build Coastguard Worker unsigned int parts)
2698*9880d681SAndroid Build Coastguard Worker {
2699*9880d681SAndroid Build Coastguard Worker unsigned int n, shiftCount;
2700*9880d681SAndroid Build Coastguard Worker integerPart mask;
2701*9880d681SAndroid Build Coastguard Worker
2702*9880d681SAndroid Build Coastguard Worker assert(lhs != remainder && lhs != srhs && remainder != srhs);
2703*9880d681SAndroid Build Coastguard Worker
2704*9880d681SAndroid Build Coastguard Worker shiftCount = tcMSB(rhs, parts) + 1;
2705*9880d681SAndroid Build Coastguard Worker if (shiftCount == 0)
2706*9880d681SAndroid Build Coastguard Worker return true;
2707*9880d681SAndroid Build Coastguard Worker
2708*9880d681SAndroid Build Coastguard Worker shiftCount = parts * integerPartWidth - shiftCount;
2709*9880d681SAndroid Build Coastguard Worker n = shiftCount / integerPartWidth;
2710*9880d681SAndroid Build Coastguard Worker mask = (integerPart) 1 << (shiftCount % integerPartWidth);
2711*9880d681SAndroid Build Coastguard Worker
2712*9880d681SAndroid Build Coastguard Worker tcAssign(srhs, rhs, parts);
2713*9880d681SAndroid Build Coastguard Worker tcShiftLeft(srhs, parts, shiftCount);
2714*9880d681SAndroid Build Coastguard Worker tcAssign(remainder, lhs, parts);
2715*9880d681SAndroid Build Coastguard Worker tcSet(lhs, 0, parts);
2716*9880d681SAndroid Build Coastguard Worker
2717*9880d681SAndroid Build Coastguard Worker /* Loop, subtracting SRHS if REMAINDER is greater and adding that to
2718*9880d681SAndroid Build Coastguard Worker the total. */
2719*9880d681SAndroid Build Coastguard Worker for (;;) {
2720*9880d681SAndroid Build Coastguard Worker int compare;
2721*9880d681SAndroid Build Coastguard Worker
2722*9880d681SAndroid Build Coastguard Worker compare = tcCompare(remainder, srhs, parts);
2723*9880d681SAndroid Build Coastguard Worker if (compare >= 0) {
2724*9880d681SAndroid Build Coastguard Worker tcSubtract(remainder, srhs, 0, parts);
2725*9880d681SAndroid Build Coastguard Worker lhs[n] |= mask;
2726*9880d681SAndroid Build Coastguard Worker }
2727*9880d681SAndroid Build Coastguard Worker
2728*9880d681SAndroid Build Coastguard Worker if (shiftCount == 0)
2729*9880d681SAndroid Build Coastguard Worker break;
2730*9880d681SAndroid Build Coastguard Worker shiftCount--;
2731*9880d681SAndroid Build Coastguard Worker tcShiftRight(srhs, parts, 1);
2732*9880d681SAndroid Build Coastguard Worker if ((mask >>= 1) == 0) {
2733*9880d681SAndroid Build Coastguard Worker mask = (integerPart) 1 << (integerPartWidth - 1);
2734*9880d681SAndroid Build Coastguard Worker n--;
2735*9880d681SAndroid Build Coastguard Worker }
2736*9880d681SAndroid Build Coastguard Worker }
2737*9880d681SAndroid Build Coastguard Worker
2738*9880d681SAndroid Build Coastguard Worker return false;
2739*9880d681SAndroid Build Coastguard Worker }
2740*9880d681SAndroid Build Coastguard Worker
2741*9880d681SAndroid Build Coastguard Worker /* Shift a bignum left COUNT bits in-place. Shifted in bits are zero.
2742*9880d681SAndroid Build Coastguard Worker There are no restrictions on COUNT. */
2743*9880d681SAndroid Build Coastguard Worker void
tcShiftLeft(integerPart * dst,unsigned int parts,unsigned int count)2744*9880d681SAndroid Build Coastguard Worker APInt::tcShiftLeft(integerPart *dst, unsigned int parts, unsigned int count)
2745*9880d681SAndroid Build Coastguard Worker {
2746*9880d681SAndroid Build Coastguard Worker if (count) {
2747*9880d681SAndroid Build Coastguard Worker unsigned int jump, shift;
2748*9880d681SAndroid Build Coastguard Worker
2749*9880d681SAndroid Build Coastguard Worker /* Jump is the inter-part jump; shift is is intra-part shift. */
2750*9880d681SAndroid Build Coastguard Worker jump = count / integerPartWidth;
2751*9880d681SAndroid Build Coastguard Worker shift = count % integerPartWidth;
2752*9880d681SAndroid Build Coastguard Worker
2753*9880d681SAndroid Build Coastguard Worker while (parts > jump) {
2754*9880d681SAndroid Build Coastguard Worker integerPart part;
2755*9880d681SAndroid Build Coastguard Worker
2756*9880d681SAndroid Build Coastguard Worker parts--;
2757*9880d681SAndroid Build Coastguard Worker
2758*9880d681SAndroid Build Coastguard Worker /* dst[i] comes from the two parts src[i - jump] and, if we have
2759*9880d681SAndroid Build Coastguard Worker an intra-part shift, src[i - jump - 1]. */
2760*9880d681SAndroid Build Coastguard Worker part = dst[parts - jump];
2761*9880d681SAndroid Build Coastguard Worker if (shift) {
2762*9880d681SAndroid Build Coastguard Worker part <<= shift;
2763*9880d681SAndroid Build Coastguard Worker if (parts >= jump + 1)
2764*9880d681SAndroid Build Coastguard Worker part |= dst[parts - jump - 1] >> (integerPartWidth - shift);
2765*9880d681SAndroid Build Coastguard Worker }
2766*9880d681SAndroid Build Coastguard Worker
2767*9880d681SAndroid Build Coastguard Worker dst[parts] = part;
2768*9880d681SAndroid Build Coastguard Worker }
2769*9880d681SAndroid Build Coastguard Worker
2770*9880d681SAndroid Build Coastguard Worker while (parts > 0)
2771*9880d681SAndroid Build Coastguard Worker dst[--parts] = 0;
2772*9880d681SAndroid Build Coastguard Worker }
2773*9880d681SAndroid Build Coastguard Worker }
2774*9880d681SAndroid Build Coastguard Worker
2775*9880d681SAndroid Build Coastguard Worker /* Shift a bignum right COUNT bits in-place. Shifted in bits are
2776*9880d681SAndroid Build Coastguard Worker zero. There are no restrictions on COUNT. */
2777*9880d681SAndroid Build Coastguard Worker void
tcShiftRight(integerPart * dst,unsigned int parts,unsigned int count)2778*9880d681SAndroid Build Coastguard Worker APInt::tcShiftRight(integerPart *dst, unsigned int parts, unsigned int count)
2779*9880d681SAndroid Build Coastguard Worker {
2780*9880d681SAndroid Build Coastguard Worker if (count) {
2781*9880d681SAndroid Build Coastguard Worker unsigned int i, jump, shift;
2782*9880d681SAndroid Build Coastguard Worker
2783*9880d681SAndroid Build Coastguard Worker /* Jump is the inter-part jump; shift is is intra-part shift. */
2784*9880d681SAndroid Build Coastguard Worker jump = count / integerPartWidth;
2785*9880d681SAndroid Build Coastguard Worker shift = count % integerPartWidth;
2786*9880d681SAndroid Build Coastguard Worker
2787*9880d681SAndroid Build Coastguard Worker /* Perform the shift. This leaves the most significant COUNT bits
2788*9880d681SAndroid Build Coastguard Worker of the result at zero. */
2789*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++) {
2790*9880d681SAndroid Build Coastguard Worker integerPart part;
2791*9880d681SAndroid Build Coastguard Worker
2792*9880d681SAndroid Build Coastguard Worker if (i + jump >= parts) {
2793*9880d681SAndroid Build Coastguard Worker part = 0;
2794*9880d681SAndroid Build Coastguard Worker } else {
2795*9880d681SAndroid Build Coastguard Worker part = dst[i + jump];
2796*9880d681SAndroid Build Coastguard Worker if (shift) {
2797*9880d681SAndroid Build Coastguard Worker part >>= shift;
2798*9880d681SAndroid Build Coastguard Worker if (i + jump + 1 < parts)
2799*9880d681SAndroid Build Coastguard Worker part |= dst[i + jump + 1] << (integerPartWidth - shift);
2800*9880d681SAndroid Build Coastguard Worker }
2801*9880d681SAndroid Build Coastguard Worker }
2802*9880d681SAndroid Build Coastguard Worker
2803*9880d681SAndroid Build Coastguard Worker dst[i] = part;
2804*9880d681SAndroid Build Coastguard Worker }
2805*9880d681SAndroid Build Coastguard Worker }
2806*9880d681SAndroid Build Coastguard Worker }
2807*9880d681SAndroid Build Coastguard Worker
2808*9880d681SAndroid Build Coastguard Worker /* Bitwise and of two bignums. */
2809*9880d681SAndroid Build Coastguard Worker void
tcAnd(integerPart * dst,const integerPart * rhs,unsigned int parts)2810*9880d681SAndroid Build Coastguard Worker APInt::tcAnd(integerPart *dst, const integerPart *rhs, unsigned int parts)
2811*9880d681SAndroid Build Coastguard Worker {
2812*9880d681SAndroid Build Coastguard Worker unsigned int i;
2813*9880d681SAndroid Build Coastguard Worker
2814*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2815*9880d681SAndroid Build Coastguard Worker dst[i] &= rhs[i];
2816*9880d681SAndroid Build Coastguard Worker }
2817*9880d681SAndroid Build Coastguard Worker
2818*9880d681SAndroid Build Coastguard Worker /* Bitwise inclusive or of two bignums. */
2819*9880d681SAndroid Build Coastguard Worker void
tcOr(integerPart * dst,const integerPart * rhs,unsigned int parts)2820*9880d681SAndroid Build Coastguard Worker APInt::tcOr(integerPart *dst, const integerPart *rhs, unsigned int parts)
2821*9880d681SAndroid Build Coastguard Worker {
2822*9880d681SAndroid Build Coastguard Worker unsigned int i;
2823*9880d681SAndroid Build Coastguard Worker
2824*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2825*9880d681SAndroid Build Coastguard Worker dst[i] |= rhs[i];
2826*9880d681SAndroid Build Coastguard Worker }
2827*9880d681SAndroid Build Coastguard Worker
2828*9880d681SAndroid Build Coastguard Worker /* Bitwise exclusive or of two bignums. */
2829*9880d681SAndroid Build Coastguard Worker void
tcXor(integerPart * dst,const integerPart * rhs,unsigned int parts)2830*9880d681SAndroid Build Coastguard Worker APInt::tcXor(integerPart *dst, const integerPart *rhs, unsigned int parts)
2831*9880d681SAndroid Build Coastguard Worker {
2832*9880d681SAndroid Build Coastguard Worker unsigned int i;
2833*9880d681SAndroid Build Coastguard Worker
2834*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2835*9880d681SAndroid Build Coastguard Worker dst[i] ^= rhs[i];
2836*9880d681SAndroid Build Coastguard Worker }
2837*9880d681SAndroid Build Coastguard Worker
2838*9880d681SAndroid Build Coastguard Worker /* Complement a bignum in-place. */
2839*9880d681SAndroid Build Coastguard Worker void
tcComplement(integerPart * dst,unsigned int parts)2840*9880d681SAndroid Build Coastguard Worker APInt::tcComplement(integerPart *dst, unsigned int parts)
2841*9880d681SAndroid Build Coastguard Worker {
2842*9880d681SAndroid Build Coastguard Worker unsigned int i;
2843*9880d681SAndroid Build Coastguard Worker
2844*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2845*9880d681SAndroid Build Coastguard Worker dst[i] = ~dst[i];
2846*9880d681SAndroid Build Coastguard Worker }
2847*9880d681SAndroid Build Coastguard Worker
2848*9880d681SAndroid Build Coastguard Worker /* Comparison (unsigned) of two bignums. */
2849*9880d681SAndroid Build Coastguard Worker int
tcCompare(const integerPart * lhs,const integerPart * rhs,unsigned int parts)2850*9880d681SAndroid Build Coastguard Worker APInt::tcCompare(const integerPart *lhs, const integerPart *rhs,
2851*9880d681SAndroid Build Coastguard Worker unsigned int parts)
2852*9880d681SAndroid Build Coastguard Worker {
2853*9880d681SAndroid Build Coastguard Worker while (parts) {
2854*9880d681SAndroid Build Coastguard Worker parts--;
2855*9880d681SAndroid Build Coastguard Worker if (lhs[parts] == rhs[parts])
2856*9880d681SAndroid Build Coastguard Worker continue;
2857*9880d681SAndroid Build Coastguard Worker
2858*9880d681SAndroid Build Coastguard Worker if (lhs[parts] > rhs[parts])
2859*9880d681SAndroid Build Coastguard Worker return 1;
2860*9880d681SAndroid Build Coastguard Worker else
2861*9880d681SAndroid Build Coastguard Worker return -1;
2862*9880d681SAndroid Build Coastguard Worker }
2863*9880d681SAndroid Build Coastguard Worker
2864*9880d681SAndroid Build Coastguard Worker return 0;
2865*9880d681SAndroid Build Coastguard Worker }
2866*9880d681SAndroid Build Coastguard Worker
2867*9880d681SAndroid Build Coastguard Worker /* Increment a bignum in-place, return the carry flag. */
2868*9880d681SAndroid Build Coastguard Worker integerPart
tcIncrement(integerPart * dst,unsigned int parts)2869*9880d681SAndroid Build Coastguard Worker APInt::tcIncrement(integerPart *dst, unsigned int parts)
2870*9880d681SAndroid Build Coastguard Worker {
2871*9880d681SAndroid Build Coastguard Worker unsigned int i;
2872*9880d681SAndroid Build Coastguard Worker
2873*9880d681SAndroid Build Coastguard Worker for (i = 0; i < parts; i++)
2874*9880d681SAndroid Build Coastguard Worker if (++dst[i] != 0)
2875*9880d681SAndroid Build Coastguard Worker break;
2876*9880d681SAndroid Build Coastguard Worker
2877*9880d681SAndroid Build Coastguard Worker return i == parts;
2878*9880d681SAndroid Build Coastguard Worker }
2879*9880d681SAndroid Build Coastguard Worker
2880*9880d681SAndroid Build Coastguard Worker /* Decrement a bignum in-place, return the borrow flag. */
2881*9880d681SAndroid Build Coastguard Worker integerPart
tcDecrement(integerPart * dst,unsigned int parts)2882*9880d681SAndroid Build Coastguard Worker APInt::tcDecrement(integerPart *dst, unsigned int parts) {
2883*9880d681SAndroid Build Coastguard Worker for (unsigned int i = 0; i < parts; i++) {
2884*9880d681SAndroid Build Coastguard Worker // If the current word is non-zero, then the decrement has no effect on the
2885*9880d681SAndroid Build Coastguard Worker // higher-order words of the integer and no borrow can occur. Exit early.
2886*9880d681SAndroid Build Coastguard Worker if (dst[i]--)
2887*9880d681SAndroid Build Coastguard Worker return 0;
2888*9880d681SAndroid Build Coastguard Worker }
2889*9880d681SAndroid Build Coastguard Worker // If every word was zero, then there is a borrow.
2890*9880d681SAndroid Build Coastguard Worker return 1;
2891*9880d681SAndroid Build Coastguard Worker }
2892*9880d681SAndroid Build Coastguard Worker
2893*9880d681SAndroid Build Coastguard Worker
2894*9880d681SAndroid Build Coastguard Worker /* Set the least significant BITS bits of a bignum, clear the
2895*9880d681SAndroid Build Coastguard Worker rest. */
2896*9880d681SAndroid Build Coastguard Worker void
tcSetLeastSignificantBits(integerPart * dst,unsigned int parts,unsigned int bits)2897*9880d681SAndroid Build Coastguard Worker APInt::tcSetLeastSignificantBits(integerPart *dst, unsigned int parts,
2898*9880d681SAndroid Build Coastguard Worker unsigned int bits)
2899*9880d681SAndroid Build Coastguard Worker {
2900*9880d681SAndroid Build Coastguard Worker unsigned int i;
2901*9880d681SAndroid Build Coastguard Worker
2902*9880d681SAndroid Build Coastguard Worker i = 0;
2903*9880d681SAndroid Build Coastguard Worker while (bits > integerPartWidth) {
2904*9880d681SAndroid Build Coastguard Worker dst[i++] = ~(integerPart) 0;
2905*9880d681SAndroid Build Coastguard Worker bits -= integerPartWidth;
2906*9880d681SAndroid Build Coastguard Worker }
2907*9880d681SAndroid Build Coastguard Worker
2908*9880d681SAndroid Build Coastguard Worker if (bits)
2909*9880d681SAndroid Build Coastguard Worker dst[i++] = ~(integerPart) 0 >> (integerPartWidth - bits);
2910*9880d681SAndroid Build Coastguard Worker
2911*9880d681SAndroid Build Coastguard Worker while (i < parts)
2912*9880d681SAndroid Build Coastguard Worker dst[i++] = 0;
2913*9880d681SAndroid Build Coastguard Worker }
2914