blob: 834b8a875213fc8a1e41ffebd2a8b74042137dbf [file] [log] [blame]
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001/***********************************************************
Guido van Rossum6610ad91995-01-04 19:07:38 +00002Copyright 1991-1995 by Stichting Mathematisch Centrum, Amsterdam,
3The Netherlands.
Guido van Rossumedcc38a1991-05-05 20:09:44 +00004
5 All Rights Reserved
6
Guido van Rossumd266eb41996-10-25 14:44:06 +00007Permission to use, copy, modify, and distribute this software and its
8documentation for any purpose and without fee is hereby granted,
Guido van Rossumedcc38a1991-05-05 20:09:44 +00009provided that the above copyright notice appear in all copies and that
Guido van Rossumd266eb41996-10-25 14:44:06 +000010both that copyright notice and this permission notice appear in
Guido van Rossumedcc38a1991-05-05 20:09:44 +000011supporting documentation, and that the names of Stichting Mathematisch
Guido van Rossumd266eb41996-10-25 14:44:06 +000012Centrum or CWI or Corporation for National Research Initiatives or
13CNRI not be used in advertising or publicity pertaining to
14distribution of the software without specific, written prior
15permission.
Guido van Rossumedcc38a1991-05-05 20:09:44 +000016
Guido van Rossumd266eb41996-10-25 14:44:06 +000017While CWI is the initial source for this software, a modified version
18is made available by the Corporation for National Research Initiatives
19(CNRI) at the Internet address ftp://ftp.python.org.
20
21STICHTING MATHEMATISCH CENTRUM AND CNRI DISCLAIM ALL WARRANTIES WITH
22REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES OF
23MERCHANTABILITY AND FITNESS, IN NO EVENT SHALL STICHTING MATHEMATISCH
24CENTRUM OR CNRI BE LIABLE FOR ANY SPECIAL, INDIRECT OR CONSEQUENTIAL
25DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
26PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER
27TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
28PERFORMANCE OF THIS SOFTWARE.
Guido van Rossumedcc38a1991-05-05 20:09:44 +000029
30******************************************************************/
31
32/* Long (arbitrary precision) integer object implementation */
33
Guido van Rossum23d6f0e1991-05-14 12:06:49 +000034/* XXX The functional organization of this file is terrible */
35
Guido van Rossumc0b618a1997-05-02 03:12:38 +000036#include "Python.h"
Guido van Rossumedcc38a1991-05-05 20:09:44 +000037#include "longintrepr.h"
Guido van Rossum687ec181995-03-04 22:43:47 +000038#include "mymath.h"
Guido van Rossumc0b618a1997-05-02 03:12:38 +000039
Guido van Rossumedcc38a1991-05-05 20:09:44 +000040#include <assert.h>
Guido van Rossumeb1fafc1994-08-29 12:47:19 +000041#include <ctype.h>
Guido van Rossumedcc38a1991-05-05 20:09:44 +000042
Guido van Rossume32e0141992-01-19 16:31:05 +000043#define ABS(x) ((x) < 0 ? -(x) : (x))
44
45/* Forward */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000046static PyLongObject *long_normalize Py_PROTO((PyLongObject *));
47static PyLongObject *mul1 Py_PROTO((PyLongObject *, wdigit));
48static PyLongObject *muladd1 Py_PROTO((PyLongObject *, wdigit, wdigit));
49static PyLongObject *divrem1 Py_PROTO((PyLongObject *, wdigit, digit *));
50static PyObject *long_format Py_PROTO((PyObject *aa, int base));
Guido van Rossume32e0141992-01-19 16:31:05 +000051
Guido van Rossum23d6f0e1991-05-14 12:06:49 +000052static int ticker; /* XXX Could be shared with ceval? */
53
Guido van Rossumc0b618a1997-05-02 03:12:38 +000054#define SIGCHECK(PyTryBlock) \
Guido van Rossum23d6f0e1991-05-14 12:06:49 +000055 if (--ticker < 0) { \
56 ticker = 100; \
Guido van Rossumc0b618a1997-05-02 03:12:38 +000057 if (PyErr_CheckSignals()) { PyTryBlock; } \
Guido van Rossum23d6f0e1991-05-14 12:06:49 +000058 }
59
Guido van Rossumedcc38a1991-05-05 20:09:44 +000060/* Normalize (remove leading zeros from) a long int object.
61 Doesn't attempt to free the storage--in most cases, due to the nature
62 of the algorithms used, this could save at most be one word anyway. */
63
Guido van Rossumc0b618a1997-05-02 03:12:38 +000064static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +000065long_normalize(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +000066 register PyLongObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +000067{
Guido van Rossum4c260ff1992-01-14 18:36:43 +000068 int j = ABS(v->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +000069 register int i = j;
70
71 while (i > 0 && v->ob_digit[i-1] == 0)
72 --i;
73 if (i != j)
Guido van Rossum4c260ff1992-01-14 18:36:43 +000074 v->ob_size = (v->ob_size < 0) ? -(i) : i;
Guido van Rossumedcc38a1991-05-05 20:09:44 +000075 return v;
76}
77
78/* Allocate a new long int object with size digits.
79 Return NULL and set exception if we run out of memory. */
80
Guido van Rossumc0b618a1997-05-02 03:12:38 +000081PyLongObject *
82_PyLong_New(size)
Guido van Rossumedcc38a1991-05-05 20:09:44 +000083 int size;
84{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000085 return PyObject_NEW_VAR(PyLongObject, &PyLong_Type, size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +000086}
87
88/* Create a new long int object from a C long int */
89
Guido van Rossumc0b618a1997-05-02 03:12:38 +000090PyObject *
91PyLong_FromLong(ival)
Guido van Rossumedcc38a1991-05-05 20:09:44 +000092 long ival;
93{
Guido van Rossum472c04f1996-12-05 21:57:21 +000094 /* Assume a C long fits in at most 5 'digits' */
95 /* Works on both 32- and 64-bit machines */
Guido van Rossumc0b618a1997-05-02 03:12:38 +000096 PyLongObject *v = _PyLong_New(5);
Guido van Rossumedcc38a1991-05-05 20:09:44 +000097 if (v != NULL) {
Guido van Rossum472c04f1996-12-05 21:57:21 +000098 unsigned long t = ival;
99 int i;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000100 if (ival < 0) {
Guido van Rossum472c04f1996-12-05 21:57:21 +0000101 t = -ival;
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000102 v->ob_size = -(v->ob_size);
Guido van Rossum472c04f1996-12-05 21:57:21 +0000103 }
104 for (i = 0; i < 5; i++) {
Guido van Rossum2095d241997-04-09 19:41:24 +0000105 v->ob_digit[i] = (digit) (t & MASK);
Guido van Rossum472c04f1996-12-05 21:57:21 +0000106 t >>= SHIFT;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000107 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000108 v = long_normalize(v);
109 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000110 return (PyObject *)v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000111}
112
Guido van Rossum53756b11997-01-03 17:14:46 +0000113/* Create a new long int object from a C unsigned long int */
114
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000115PyObject *
Guido van Rossum53756b11997-01-03 17:14:46 +0000116PyLong_FromUnsignedLong(ival)
117 unsigned long ival;
118{
119 /* Assume a C long fits in at most 5 'digits' */
120 /* Works on both 32- and 64-bit machines */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000121 PyLongObject *v = _PyLong_New(5);
Guido van Rossum53756b11997-01-03 17:14:46 +0000122 if (v != NULL) {
123 unsigned long t = ival;
124 int i;
125 for (i = 0; i < 5; i++) {
Guido van Rossum2095d241997-04-09 19:41:24 +0000126 v->ob_digit[i] = (digit) (t & MASK);
Guido van Rossum53756b11997-01-03 17:14:46 +0000127 t >>= SHIFT;
128 }
129 v = long_normalize(v);
130 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000131 return (PyObject *)v;
Guido van Rossum53756b11997-01-03 17:14:46 +0000132}
133
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000134/* Create a new long int object from a C double */
135
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000136PyObject *
Guido van Rossum687ec181995-03-04 22:43:47 +0000137#ifdef MPW
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000138PyLong_FromDouble(double dval)
Guido van Rossum687ec181995-03-04 22:43:47 +0000139#else
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000140PyLong_FromDouble(dval)
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000141 double dval;
Guido van Rossum687ec181995-03-04 22:43:47 +0000142#endif /* MPW */
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000143{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000144 PyLongObject *v;
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000145 double frac;
146 int i, ndig, expo, neg;
147 neg = 0;
148 if (dval < 0.0) {
149 neg = 1;
150 dval = -dval;
151 }
152 frac = frexp(dval, &expo); /* dval = frac*2**expo; 0.0 <= frac < 1.0 */
153 if (expo <= 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000154 return PyLong_FromLong(0L);
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000155 ndig = (expo-1) / SHIFT + 1; /* Number of 'digits' in result */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000156 v = _PyLong_New(ndig);
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000157 if (v == NULL)
158 return NULL;
159 frac = ldexp(frac, (expo-1) % SHIFT + 1);
160 for (i = ndig; --i >= 0; ) {
161 long bits = (long)frac;
Guido van Rossum2095d241997-04-09 19:41:24 +0000162 v->ob_digit[i] = (digit) bits;
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000163 frac = frac - (double)bits;
164 frac = ldexp(frac, SHIFT);
165 }
166 if (neg)
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000167 v->ob_size = -(v->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000168 return (PyObject *)v;
Guido van Rossum149e9ea1991-06-03 10:58:24 +0000169}
170
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000171/* Get a C long int from a long int object.
172 Returns -1 and sets an error condition if overflow occurs. */
173
174long
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000175PyLong_AsLong(vv)
176 PyObject *vv;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000177{
Guido van Rossumf7531811998-05-26 14:33:37 +0000178 /* This version by Tim Peters */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000179 register PyLongObject *v;
Guido van Rossumf7531811998-05-26 14:33:37 +0000180 unsigned long x, prev;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000181 int i, sign;
Guido van Rossumf7531811998-05-26 14:33:37 +0000182
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000183 if (vv == NULL || !PyLong_Check(vv)) {
184 PyErr_BadInternalCall();
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000185 return -1;
186 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000187 v = (PyLongObject *)vv;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000188 i = v->ob_size;
189 sign = 1;
190 x = 0;
191 if (i < 0) {
192 sign = -1;
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000193 i = -(i);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000194 }
195 while (--i >= 0) {
196 prev = x;
197 x = (x << SHIFT) + v->ob_digit[i];
Guido van Rossumf7531811998-05-26 14:33:37 +0000198 if ((x >> SHIFT) != prev)
199 goto overflow;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000200 }
Guido van Rossumf7531811998-05-26 14:33:37 +0000201 /* Haven't lost any bits, but if the sign bit is set we're in
202 * trouble *unless* this is the min negative number. So,
203 * trouble iff sign bit set && (positive || some bit set other
204 * than the sign bit).
205 */
206 if ((long)x < 0 && (sign > 0 || (x << 1) != 0))
207 goto overflow;
208 return (long)x * sign;
209
210 overflow:
211 PyErr_SetString(PyExc_OverflowError,
212 "long int too long to convert");
213 return -1;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000214}
215
Guido van Rossum53756b11997-01-03 17:14:46 +0000216/* Get a C long int from a long int object.
217 Returns -1 and sets an error condition if overflow occurs. */
218
219unsigned long
220PyLong_AsUnsignedLong(vv)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000221 PyObject *vv;
Guido van Rossum53756b11997-01-03 17:14:46 +0000222{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000223 register PyLongObject *v;
Guido van Rossum53756b11997-01-03 17:14:46 +0000224 unsigned long x, prev;
225 int i;
226
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000227 if (vv == NULL || !PyLong_Check(vv)) {
228 PyErr_BadInternalCall();
Guido van Rossum53756b11997-01-03 17:14:46 +0000229 return (unsigned long) -1;
230 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000231 v = (PyLongObject *)vv;
Guido van Rossum53756b11997-01-03 17:14:46 +0000232 i = v->ob_size;
233 x = 0;
234 if (i < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000235 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum53756b11997-01-03 17:14:46 +0000236 "can't convert negative value to unsigned long");
237 return (unsigned long) -1;
238 }
239 while (--i >= 0) {
240 prev = x;
241 x = (x << SHIFT) + v->ob_digit[i];
242 if ((x >> SHIFT) != prev) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000243 PyErr_SetString(PyExc_OverflowError,
Guido van Rossum53756b11997-01-03 17:14:46 +0000244 "long int too long to convert");
245 return (unsigned long) -1;
246 }
247 }
248 return x;
249}
250
Guido van Rossum09e6ad01997-02-14 22:54:21 +0000251/* Get a C double from a long int object. */
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000252
253double
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000254PyLong_AsDouble(vv)
255 PyObject *vv;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000256{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000257 register PyLongObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000258 double x;
259 double multiplier = (double) (1L << SHIFT);
260 int i, sign;
261
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000262 if (vv == NULL || !PyLong_Check(vv)) {
263 PyErr_BadInternalCall();
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000264 return -1;
265 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000266 v = (PyLongObject *)vv;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000267 i = v->ob_size;
268 sign = 1;
269 x = 0.0;
270 if (i < 0) {
271 sign = -1;
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000272 i = -(i);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000273 }
274 while (--i >= 0) {
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000275 x = x*multiplier + (double)v->ob_digit[i];
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000276 }
277 return x * sign;
278}
279
Guido van Rossum78694d91998-09-18 14:14:13 +0000280/* Create a new long (or int) object from a C pointer */
281
282PyObject *
283PyLong_FromVoidPtr(p)
284 void *p;
285{
286#if SIZEOF_VOID_P == SIZEOF_LONG
287 return PyInt_FromLong((long)p);
288#else
289 /* optimize null pointers */
290 if ( p == NULL )
291 return PyInt_FromLong(0);
292
293 /* we can assume that HAVE_LONG_LONG is true. if not, then the
294 configuration process should have bailed (having big pointers
295 without long longs seems non-sensical) */
296 return PyLong_FromLongLong((LONG_LONG)p);
297#endif /* SIZEOF_VOID_P == SIZEOF_LONG */
298}
299
300/* Get a C pointer from a long object (or an int object in some cases) */
301
302void *
303PyLong_AsVoidPtr(vv)
304 PyObject *vv;
305{
306 /* This function will allow int or long objects. If vv is neither,
307 then the PyLong_AsLong*() functions will raise the exception:
308 PyExc_SystemError, "bad argument to internal function"
309 */
310
311#if SIZEOF_VOID_P == SIZEOF_LONG
312 long x;
313
314 if ( PyInt_Check(vv) )
315 x = PyInt_AS_LONG(vv);
316 else
317 x = PyLong_AsLong(vv);
318#else
319 /* we can assume that HAVE_LONG_LONG is true. if not, then the
320 configuration process should have bailed (having big pointers
321 without long longs seems non-sensical) */
322 LONG_LONG x;
323
324 if ( PyInt_Check(vv) )
325 x = PyInt_AS_LONG(vv);
326 else
327 x = PyLong_AsLongLong(vv);
328#endif /* SIZEOF_VOID_P == SIZEOF_LONG */
329
330 if (x == -1 && PyErr_Occurred())
331 return NULL;
332 return (void *)x;
333}
334
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000335#ifdef HAVE_LONG_LONG
336/*
Guido van Rossum3293b071998-08-25 16:07:15 +0000337 * LONG_LONG support by Chris Herborth (chrish@qnx.com)
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000338 *
339 * For better or worse :-), I tried to follow the coding style already
340 * here.
341 */
342
Guido van Rossum3293b071998-08-25 16:07:15 +0000343/* Create a new long int object from a C LONG_LONG int */
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000344
345PyObject *
346PyLong_FromLongLong(ival)
Guido van Rossum3293b071998-08-25 16:07:15 +0000347 LONG_LONG ival;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000348{
349#if SIZEOF_LONG_LONG == SIZEOF_LONG
350 /* In case the compiler is faking it. */
351 return PyLong_FromLong( (long)ival );
352#else
Guido van Rossum3293b071998-08-25 16:07:15 +0000353 if( ival <= (LONG_LONG)LONG_MAX ) {
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000354 return PyLong_FromLong( (long)ival );
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000355 }
Guido van Rossum3293b071998-08-25 16:07:15 +0000356 else if( ival <= (unsigned LONG_LONG)ULONG_MAX ) {
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000357 return PyLong_FromUnsignedLong( (unsigned long)ival );
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000358 }
359 else {
Guido van Rossum3293b071998-08-25 16:07:15 +0000360 /* Assume a C LONG_LONG fits in at most 10 'digits'.
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000361 * Should be OK if we're assuming long fits in 5.
362 */
363 PyLongObject *v = _PyLong_New(10);
364
365 if (v != NULL) {
Guido van Rossum3293b071998-08-25 16:07:15 +0000366 unsigned LONG_LONG t = ival;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000367 int i;
368 if (ival < 0) {
369 t = -ival;
370 v->ob_size = -(v->ob_size);
371 }
372
373 for (i = 0; i < 10; i++) {
374 v->ob_digit[i] = (digit) (t & MASK);
375 t >>= SHIFT;
376 }
377
378 v = long_normalize(v);
379 }
380
381 return (PyObject *)v;
382 }
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000383#endif
384}
385
Guido van Rossum3293b071998-08-25 16:07:15 +0000386/* Create a new long int object from a C unsigned LONG_LONG int */
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000387PyObject *
388PyLong_FromUnsignedLongLong(ival)
Guido van Rossum3293b071998-08-25 16:07:15 +0000389 unsigned LONG_LONG ival;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000390{
391#if SIZEOF_LONG_LONG == SIZEOF_LONG
392 /* In case the compiler is faking it. */
393 return PyLong_FromUnsignedLong( (unsigned long)ival );
394#else
Guido van Rossum3293b071998-08-25 16:07:15 +0000395 if( ival <= (unsigned LONG_LONG)ULONG_MAX ) {
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000396 return PyLong_FromUnsignedLong( (unsigned long)ival );
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000397 }
398 else {
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000399 /* Assume a C long fits in at most 10 'digits'. */
400 PyLongObject *v = _PyLong_New(10);
401
402 if (v != NULL) {
Guido van Rossum3293b071998-08-25 16:07:15 +0000403 unsigned LONG_LONG t = ival;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000404 int i;
405 for (i = 0; i < 10; i++) {
406 v->ob_digit[i] = (digit) (t & MASK);
407 t >>= SHIFT;
408 }
409
410 v = long_normalize(v);
411 }
412
413 return (PyObject *)v;
414 }
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000415#endif
416}
417
Guido van Rossum3293b071998-08-25 16:07:15 +0000418/* Get a C LONG_LONG int from a long int object.
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000419 Returns -1 and sets an error condition if overflow occurs. */
420
Guido van Rossum3293b071998-08-25 16:07:15 +0000421LONG_LONG
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000422PyLong_AsLongLong(vv)
423 PyObject *vv;
424{
425#if SIZEOF_LONG_LONG == SIZEOF_LONG
426 /* In case the compiler is faking it. */
Guido van Rossum3293b071998-08-25 16:07:15 +0000427 return (LONG_LONG)PyLong_AsLong( vv );
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000428#else
429 register PyLongObject *v;
Guido van Rossum3293b071998-08-25 16:07:15 +0000430 LONG_LONG x, prev;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000431 int i, sign;
432
433 if (vv == NULL || !PyLong_Check(vv)) {
434 PyErr_BadInternalCall();
435 return -1;
436 }
437
438 v = (PyLongObject *)vv;
439 i = v->ob_size;
440 sign = 1;
441 x = 0;
442
443 if (i < 0) {
444 sign = -1;
445 i = -(i);
446 }
447
448 while (--i >= 0) {
449 prev = x;
450 x = (x << SHIFT) + v->ob_digit[i];
451 if ((x >> SHIFT) != prev) {
452 PyErr_SetString(PyExc_OverflowError,
453 "long int too long to convert");
454 return -1;
455 }
456 }
457
458 return x * sign;
459#endif
460}
461
Guido van Rossum3293b071998-08-25 16:07:15 +0000462unsigned LONG_LONG
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000463PyLong_AsUnsignedLongLong(vv)
464 PyObject *vv;
465{
466#if SIZEOF_LONG_LONG == 4
467 /* In case the compiler is faking it. */
Guido van Rossum3293b071998-08-25 16:07:15 +0000468 return (unsigned LONG_LONG)PyLong_AsUnsignedLong( vv );
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000469#else
470 register PyLongObject *v;
Guido van Rossum3293b071998-08-25 16:07:15 +0000471 unsigned LONG_LONG x, prev;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000472 int i;
473
474 if (vv == NULL || !PyLong_Check(vv)) {
475 PyErr_BadInternalCall();
Guido van Rossum3293b071998-08-25 16:07:15 +0000476 return (unsigned LONG_LONG) -1;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000477 }
478
479 v = (PyLongObject *)vv;
480 i = v->ob_size;
481 x = 0;
482
483 if (i < 0) {
484 PyErr_SetString(PyExc_OverflowError,
485 "can't convert negative value to unsigned long");
Guido van Rossum3293b071998-08-25 16:07:15 +0000486 return (unsigned LONG_LONG) -1;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000487 }
488
489 while (--i >= 0) {
490 prev = x;
491 x = (x << SHIFT) + v->ob_digit[i];
492 if ((x >> SHIFT) != prev) {
493 PyErr_SetString(PyExc_OverflowError,
494 "long int too long to convert");
Guido van Rossum3293b071998-08-25 16:07:15 +0000495 return (unsigned LONG_LONG) -1;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000496 }
497 }
498
499 return x;
500#endif
501}
502#endif /* HAVE_LONG_LONG */
503
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000504/* Multiply by a single digit, ignoring the sign. */
505
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000506static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000507mul1(a, n)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000508 PyLongObject *a;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000509 wdigit n;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000510{
511 return muladd1(a, n, (digit)0);
512}
513
514/* Multiply by a single digit and add a single digit, ignoring the sign. */
515
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000516static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000517muladd1(a, n, extra)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000518 PyLongObject *a;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000519 wdigit n;
520 wdigit extra;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000521{
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000522 int size_a = ABS(a->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000523 PyLongObject *z = _PyLong_New(size_a+1);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000524 twodigits carry = extra;
525 int i;
526
527 if (z == NULL)
528 return NULL;
529 for (i = 0; i < size_a; ++i) {
530 carry += (twodigits)a->ob_digit[i] * n;
Guido van Rossum2095d241997-04-09 19:41:24 +0000531 z->ob_digit[i] = (digit) (carry & MASK);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000532 carry >>= SHIFT;
533 }
Guido van Rossum2095d241997-04-09 19:41:24 +0000534 z->ob_digit[i] = (digit) carry;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000535 return long_normalize(z);
536}
537
538/* Divide a long integer by a digit, returning both the quotient
539 (as function result) and the remainder (through *prem).
540 The sign of a is ignored; n should not be zero. */
541
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000542static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000543divrem1(a, n, prem)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000544 PyLongObject *a;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000545 wdigit n;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000546 digit *prem;
547{
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000548 int size = ABS(a->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000549 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000550 int i;
551 twodigits rem = 0;
552
553 assert(n > 0 && n <= MASK);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000554 z = _PyLong_New(size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000555 if (z == NULL)
556 return NULL;
557 for (i = size; --i >= 0; ) {
558 rem = (rem << SHIFT) + a->ob_digit[i];
Guido van Rossum2095d241997-04-09 19:41:24 +0000559 z->ob_digit[i] = (digit) (rem/n);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000560 rem %= n;
561 }
Guido van Rossum2095d241997-04-09 19:41:24 +0000562 *prem = (digit) rem;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000563 return long_normalize(z);
564}
565
566/* Convert a long int object to a string, using a given conversion base.
Guido van Rossum3d3037d1991-10-24 14:55:57 +0000567 Return a string object.
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000568 If base is 8 or 16, add the proper prefix '0' or '0x'.
569 External linkage: used in bltinmodule.c by hex() and oct(). */
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000570
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000571static PyObject *
Guido van Rossume32e0141992-01-19 16:31:05 +0000572long_format(aa, base)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000573 PyObject *aa;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000574 int base;
575{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000576 register PyLongObject *a = (PyLongObject *)aa;
577 PyStringObject *str;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000578 int i;
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000579 int size_a = ABS(a->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000580 char *p;
581 int bits;
582 char sign = '\0';
Guido van Rossume32e0141992-01-19 16:31:05 +0000583
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000584 if (a == NULL || !PyLong_Check(a)) {
585 PyErr_BadInternalCall();
Guido van Rossume32e0141992-01-19 16:31:05 +0000586 return NULL;
587 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000588 assert(base >= 2 && base <= 36);
589
590 /* Compute a rough upper bound for the length of the string */
591 i = base;
592 bits = 0;
593 while (i > 1) {
594 ++bits;
595 i >>= 1;
596 }
Guido van Rossumc6913e71991-11-19 20:26:46 +0000597 i = 6 + (size_a*SHIFT + bits-1) / bits;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000598 str = (PyStringObject *) PyString_FromStringAndSize((char *)0, i);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000599 if (str == NULL)
600 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000601 p = PyString_AS_STRING(str) + i;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000602 *p = '\0';
Guido van Rossumc6913e71991-11-19 20:26:46 +0000603 *--p = 'L';
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000604 if (a->ob_size < 0)
605 sign = '-';
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000606
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000607 if (a->ob_size == 0) {
608 *--p = '0';
609 }
610 else if ((base & (base - 1)) == 0) {
611 /* JRH: special case for power-of-2 bases */
612 twodigits temp = a->ob_digit[0];
613 int bitsleft = SHIFT;
614 int rem;
615 int last = abs(a->ob_size);
616 int basebits = 1;
617 i = base;
618 while ((i >>= 1) > 1) ++basebits;
619
620 i = 0;
621 for (;;) {
622 while (bitsleft >= basebits) {
623 if ((temp == 0) && (i >= last - 1)) break;
624 rem = temp & (base - 1);
625 if (rem < 10)
626 rem += '0';
627 else
628 rem += 'A' - 10;
629 assert(p > PyString_AS_STRING(str));
630 *--p = (char) rem;
631 bitsleft -= basebits;
632 temp >>= basebits;
633 }
634 if (++i >= last) {
635 if (temp == 0) break;
636 bitsleft = 99;
637 /* loop again to pick up final digits */
638 }
639 else {
640 temp = (a->ob_digit[i] << bitsleft) | temp;
641 bitsleft += SHIFT;
642 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000643 }
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000644 }
645 else {
646 Py_INCREF(a);
647 do {
648 digit rem;
649 PyLongObject *temp = divrem1(a, (digit)base, &rem);
650 if (temp == NULL) {
651 Py_DECREF(a);
652 Py_DECREF(str);
653 return NULL;
654 }
655 if (rem < 10)
656 rem += '0';
657 else
658 rem += 'A'-10;
659 assert(p > PyString_AS_STRING(str));
660 *--p = (char) rem;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000661 Py_DECREF(a);
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000662 a = temp;
663 SIGCHECK({
664 Py_DECREF(a);
665 Py_DECREF(str);
666 return NULL;
667 })
668 } while (ABS(a->ob_size) != 0);
669 Py_DECREF(a);
670 }
671
Guido van Rossum2c475421992-08-14 15:13:07 +0000672 if (base == 8) {
673 if (size_a != 0)
674 *--p = '0';
675 }
Guido van Rossum3d3037d1991-10-24 14:55:57 +0000676 else if (base == 16) {
677 *--p = 'x';
678 *--p = '0';
679 }
Guido van Rossumc6913e71991-11-19 20:26:46 +0000680 else if (base != 10) {
681 *--p = '#';
682 *--p = '0' + base%10;
683 if (base > 10)
684 *--p = '0' + base/10;
685 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000686 if (sign)
687 *--p = sign;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000688 if (p != PyString_AS_STRING(str)) {
689 char *q = PyString_AS_STRING(str);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000690 assert(p > q);
691 do {
692 } while ((*q++ = *p++) != '\0');
Guido van Rossumc7ec9c91991-05-28 21:58:16 +0000693 q--;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000694 _PyString_Resize((PyObject **)&str,
695 (int) (q - PyString_AS_STRING(str)));
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000696 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000697 return (PyObject *)str;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000698}
699
Guido van Rossum3535f6e1995-01-17 16:34:13 +0000700#if 0
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000701/* Convert a string to a long int object, in a given base.
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000702 Base zero implies a default depending on the number.
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000703 External linkage: used in compile.c and stropmodule.c. */
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000704
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000705PyObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000706long_scan(str, base)
707 char *str;
708 int base;
709{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000710 return PyLong_FromString(str, (char **)NULL, base);
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000711}
Guido van Rossum3535f6e1995-01-17 16:34:13 +0000712#endif
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000713
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000714PyObject *
715PyLong_FromString(str, pend, base)
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000716 char *str;
717 char **pend;
718 int base;
719{
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000720 int sign = 1;
Guido van Rossume6762971998-06-22 03:54:46 +0000721 char *start;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000722 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000723
Guido van Rossum472c04f1996-12-05 21:57:21 +0000724 if ((base != 0 && base < 2) || base > 36) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000725 PyErr_SetString(PyExc_ValueError,
726 "invalid base for long literal");
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000727 return NULL;
728 }
Guido van Rossum9fa2c111995-02-10 17:00:37 +0000729 while (*str != '\0' && isspace(Py_CHARMASK(*str)))
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000730 str++;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000731 if (*str == '+')
732 ++str;
733 else if (*str == '-') {
734 ++str;
735 sign = -1;
736 }
Guido van Rossum9fa2c111995-02-10 17:00:37 +0000737 while (*str != '\0' && isspace(Py_CHARMASK(*str)))
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000738 str++;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000739 if (base == 0) {
740 if (str[0] != '0')
741 base = 10;
742 else if (str[1] == 'x' || str[1] == 'X')
743 base = 16;
744 else
745 base = 8;
746 }
747 if (base == 16 && str[0] == '0' && (str[1] == 'x' || str[1] == 'X'))
748 str += 2;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000749 z = _PyLong_New(0);
Guido van Rossume6762971998-06-22 03:54:46 +0000750 start = str;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000751 for ( ; z != NULL; ++str) {
752 int k = -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000753 PyLongObject *temp;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000754
755 if (*str <= '9')
756 k = *str - '0';
757 else if (*str >= 'a')
758 k = *str - 'a' + 10;
759 else if (*str >= 'A')
760 k = *str - 'A' + 10;
761 if (k < 0 || k >= base)
762 break;
763 temp = muladd1(z, (digit)base, (digit)k);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000764 Py_DECREF(z);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000765 z = temp;
766 }
Guido van Rossumac6a37a1998-08-04 15:04:06 +0000767 if (z == NULL)
768 return NULL;
Guido van Rossume6762971998-06-22 03:54:46 +0000769 if (str == start) {
770 PyErr_SetString(PyExc_ValueError,
771 "no digits in long int constant");
772 return NULL;
773 }
Guido van Rossumc6913e71991-11-19 20:26:46 +0000774 if (sign < 0 && z != NULL && z->ob_size != 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000775 z->ob_size = -(z->ob_size);
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000776 if (pend)
777 *pend = str;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000778 return (PyObject *) z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000779}
780
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000781static PyLongObject *x_divrem
782 Py_PROTO((PyLongObject *, PyLongObject *, PyLongObject **));
783static PyObject *long_pos Py_PROTO((PyLongObject *));
Guido van Rossum1109fbc1998-04-10 22:16:39 +0000784static int long_divrem Py_PROTO((PyLongObject *, PyLongObject *,
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000785 PyLongObject **, PyLongObject **));
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000786
787/* Long division with remainder, top-level routine */
788
Guido van Rossume32e0141992-01-19 16:31:05 +0000789static int
790long_divrem(a, b, pdiv, prem)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000791 PyLongObject *a, *b;
792 PyLongObject **pdiv;
793 PyLongObject **prem;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000794{
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000795 int size_a = ABS(a->ob_size), size_b = ABS(b->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000796 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000797
798 if (size_b == 0) {
Guido van Rossumbd3a5271998-08-11 15:04:47 +0000799 PyErr_SetString(PyExc_ZeroDivisionError,
800 "long division or modulo");
Guido van Rossume32e0141992-01-19 16:31:05 +0000801 return -1;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000802 }
803 if (size_a < size_b ||
Guido van Rossum472c04f1996-12-05 21:57:21 +0000804 (size_a == size_b &&
805 a->ob_digit[size_a-1] < b->ob_digit[size_b-1])) {
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000806 /* |a| < |b|. */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000807 *pdiv = _PyLong_New(0);
808 Py_INCREF(a);
809 *prem = (PyLongObject *) a;
Guido van Rossume32e0141992-01-19 16:31:05 +0000810 return 0;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000811 }
812 if (size_b == 1) {
813 digit rem = 0;
814 z = divrem1(a, b->ob_digit[0], &rem);
Guido van Rossume32e0141992-01-19 16:31:05 +0000815 if (z == NULL)
816 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000817 *prem = (PyLongObject *) PyLong_FromLong((long)rem);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000818 }
Guido van Rossume32e0141992-01-19 16:31:05 +0000819 else {
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000820 z = x_divrem(a, b, prem);
Guido van Rossume32e0141992-01-19 16:31:05 +0000821 if (z == NULL)
822 return -1;
823 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000824 /* Set the signs.
825 The quotient z has the sign of a*b;
826 the remainder r has the sign of a,
827 so a = b*z + r. */
Guido van Rossume32e0141992-01-19 16:31:05 +0000828 if ((a->ob_size < 0) != (b->ob_size < 0))
829 z->ob_size = -(z->ob_size);
830 if (a->ob_size < 0 && (*prem)->ob_size != 0)
831 (*prem)->ob_size = -((*prem)->ob_size);
832 *pdiv = z;
833 return 0;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000834}
835
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000836/* Unsigned long division with remainder -- the algorithm */
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000837
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000838static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000839x_divrem(v1, w1, prem)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000840 PyLongObject *v1, *w1;
841 PyLongObject **prem;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000842{
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000843 int size_v = ABS(v1->ob_size), size_w = ABS(w1->ob_size);
Guido van Rossum2095d241997-04-09 19:41:24 +0000844 digit d = (digit) ((twodigits)BASE / (w1->ob_digit[size_w-1] + 1));
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000845 PyLongObject *v = mul1(v1, d);
846 PyLongObject *w = mul1(w1, d);
847 PyLongObject *a;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000848 int j, k;
849
850 if (v == NULL || w == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000851 Py_XDECREF(v);
852 Py_XDECREF(w);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000853 return NULL;
854 }
855
856 assert(size_v >= size_w && size_w > 1); /* Assert checks by div() */
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000857 assert(v->ob_refcnt == 1); /* Since v will be used as accumulator! */
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000858 assert(size_w == ABS(w->ob_size)); /* That's how d was calculated */
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000859
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000860 size_v = ABS(v->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000861 a = _PyLong_New(size_v - size_w + 1);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000862
863 for (j = size_v, k = a->ob_size-1; a != NULL && k >= 0; --j, --k) {
864 digit vj = (j >= size_v) ? 0 : v->ob_digit[j];
865 twodigits q;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000866 stwodigits carry = 0;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000867 int i;
868
Guido van Rossumeb1fafc1994-08-29 12:47:19 +0000869 SIGCHECK({
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000870 Py_DECREF(a);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000871 a = NULL;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000872 break;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +0000873 })
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000874 if (vj == w->ob_digit[size_w-1])
875 q = MASK;
876 else
877 q = (((twodigits)vj << SHIFT) + v->ob_digit[j-1]) /
878 w->ob_digit[size_w-1];
879
880 while (w->ob_digit[size_w-2]*q >
881 ((
882 ((twodigits)vj << SHIFT)
883 + v->ob_digit[j-1]
884 - q*w->ob_digit[size_w-1]
885 ) << SHIFT)
886 + v->ob_digit[j-2])
887 --q;
888
889 for (i = 0; i < size_w && i+k < size_v; ++i) {
890 twodigits z = w->ob_digit[i] * q;
Guido van Rossum2095d241997-04-09 19:41:24 +0000891 digit zz = (digit) (z >> SHIFT);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000892 carry += v->ob_digit[i+k] - z
893 + ((twodigits)zz << SHIFT);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000894 v->ob_digit[i+k] = carry & MASK;
895 carry = (carry >> SHIFT) - zz;
896 }
897
898 if (i+k < size_v) {
899 carry += v->ob_digit[i+k];
900 v->ob_digit[i+k] = 0;
901 }
902
903 if (carry == 0)
Guido van Rossum2095d241997-04-09 19:41:24 +0000904 a->ob_digit[k] = (digit) q;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000905 else {
906 assert(carry == -1);
Guido van Rossum2095d241997-04-09 19:41:24 +0000907 a->ob_digit[k] = (digit) q-1;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000908 carry = 0;
909 for (i = 0; i < size_w && i+k < size_v; ++i) {
910 carry += v->ob_digit[i+k] + w->ob_digit[i];
911 v->ob_digit[i+k] = carry & MASK;
912 carry >>= SHIFT;
913 }
914 }
915 } /* for j, k */
916
Guido van Rossumc206c761995-01-10 15:23:19 +0000917 if (a == NULL)
918 *prem = NULL;
919 else {
Guido van Rossumc6913e71991-11-19 20:26:46 +0000920 a = long_normalize(a);
Guido van Rossume32e0141992-01-19 16:31:05 +0000921 *prem = divrem1(v, d, &d);
922 /* d receives the (unused) remainder */
923 if (*prem == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000924 Py_DECREF(a);
Guido van Rossume32e0141992-01-19 16:31:05 +0000925 a = NULL;
Guido van Rossumc6913e71991-11-19 20:26:46 +0000926 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000927 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000928 Py_DECREF(v);
929 Py_DECREF(w);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000930 return a;
931}
932
933/* Methods */
934
Guido van Rossume32e0141992-01-19 16:31:05 +0000935/* Forward */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000936static void long_dealloc Py_PROTO((PyObject *));
937static PyObject *long_repr Py_PROTO((PyObject *));
938static int long_compare Py_PROTO((PyLongObject *, PyLongObject *));
Guido van Rossum065ce5a1998-09-13 15:37:20 +0000939static long long_hash Py_PROTO((PyLongObject *));
Guido van Rossume32e0141992-01-19 16:31:05 +0000940
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000941static PyObject *long_add Py_PROTO((PyLongObject *, PyLongObject *));
942static PyObject *long_sub Py_PROTO((PyLongObject *, PyLongObject *));
943static PyObject *long_mul Py_PROTO((PyLongObject *, PyLongObject *));
944static PyObject *long_div Py_PROTO((PyLongObject *, PyLongObject *));
945static PyObject *long_mod Py_PROTO((PyLongObject *, PyLongObject *));
946static PyObject *long_divmod Py_PROTO((PyLongObject *, PyLongObject *));
947static PyObject *long_pow
948 Py_PROTO((PyLongObject *, PyLongObject *, PyLongObject *));
949static PyObject *long_neg Py_PROTO((PyLongObject *));
950static PyObject *long_pos Py_PROTO((PyLongObject *));
951static PyObject *long_abs Py_PROTO((PyLongObject *));
952static int long_nonzero Py_PROTO((PyLongObject *));
953static PyObject *long_invert Py_PROTO((PyLongObject *));
954static PyObject *long_lshift Py_PROTO((PyLongObject *, PyLongObject *));
955static PyObject *long_rshift Py_PROTO((PyLongObject *, PyLongObject *));
956static PyObject *long_and Py_PROTO((PyLongObject *, PyLongObject *));
957static PyObject *long_xor Py_PROTO((PyLongObject *, PyLongObject *));
958static PyObject *long_or Py_PROTO((PyLongObject *, PyLongObject *));
Guido van Rossume32e0141992-01-19 16:31:05 +0000959
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000960static void
961long_dealloc(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000962 PyObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000963{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000964 PyMem_DEL(v);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000965}
966
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000967static PyObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000968long_repr(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000969 PyObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000970{
Guido van Rossum8b27d921992-03-27 17:27:05 +0000971 return long_format(v, 10);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000972}
973
974static int
975long_compare(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000976 PyLongObject *a, *b;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000977{
978 int sign;
979
Guido van Rossumc6913e71991-11-19 20:26:46 +0000980 if (a->ob_size != b->ob_size) {
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000981 if (ABS(a->ob_size) == 0 && ABS(b->ob_size) == 0)
Guido van Rossumc6913e71991-11-19 20:26:46 +0000982 sign = 0;
983 else
984 sign = a->ob_size - b->ob_size;
985 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000986 else {
Guido van Rossum4c260ff1992-01-14 18:36:43 +0000987 int i = ABS(a->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000988 while (--i >= 0 && a->ob_digit[i] == b->ob_digit[i])
989 ;
990 if (i < 0)
991 sign = 0;
Guido van Rossum0b0db8e1993-01-21 16:07:51 +0000992 else {
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000993 sign = (int)a->ob_digit[i] - (int)b->ob_digit[i];
Guido van Rossum0b0db8e1993-01-21 16:07:51 +0000994 if (a->ob_size < 0)
995 sign = -sign;
996 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000997 }
Guido van Rossumc6913e71991-11-19 20:26:46 +0000998 return sign < 0 ? -1 : sign > 0 ? 1 : 0;
Guido van Rossumedcc38a1991-05-05 20:09:44 +0000999}
1000
Guido van Rossum9bfef441993-03-29 10:43:31 +00001001static long
1002long_hash(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001003 PyLongObject *v;
Guido van Rossum9bfef441993-03-29 10:43:31 +00001004{
1005 long x;
1006 int i, sign;
1007
1008 /* This is designed so that Python ints and longs with the
1009 same value hash to the same value, otherwise comparisons
1010 of mapping keys will turn out weird */
1011 i = v->ob_size;
1012 sign = 1;
1013 x = 0;
1014 if (i < 0) {
1015 sign = -1;
1016 i = -(i);
1017 }
1018 while (--i >= 0) {
1019 /* Force a 32-bit circular shift */
1020 x = ((x << SHIFT) & ~MASK) | ((x >> (32-SHIFT)) & MASK);
1021 x += v->ob_digit[i];
1022 }
1023 x = x * sign;
1024 if (x == -1)
1025 x = -2;
1026 return x;
1027}
1028
1029
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001030/* Add the absolute values of two long integers. */
1031
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001032static PyLongObject *x_add Py_PROTO((PyLongObject *, PyLongObject *));
1033static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001034x_add(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001035 PyLongObject *a, *b;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001036{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001037 int size_a = ABS(a->ob_size), size_b = ABS(b->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001038 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001039 int i;
1040 digit carry = 0;
1041
1042 /* Ensure a is the larger of the two: */
1043 if (size_a < size_b) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001044 { PyLongObject *temp = a; a = b; b = temp; }
1045 { int size_temp = size_a;
1046 size_a = size_b;
1047 size_b = size_temp; }
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001048 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001049 z = _PyLong_New(size_a+1);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001050 if (z == NULL)
1051 return NULL;
1052 for (i = 0; i < size_b; ++i) {
1053 carry += a->ob_digit[i] + b->ob_digit[i];
1054 z->ob_digit[i] = carry & MASK;
1055 /* The following assumes unsigned shifts don't
1056 propagate the sign bit. */
1057 carry >>= SHIFT;
1058 }
1059 for (; i < size_a; ++i) {
1060 carry += a->ob_digit[i];
1061 z->ob_digit[i] = carry & MASK;
1062 carry >>= SHIFT;
1063 }
1064 z->ob_digit[i] = carry;
1065 return long_normalize(z);
1066}
1067
1068/* Subtract the absolute values of two integers. */
1069
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001070static PyLongObject *x_sub Py_PROTO((PyLongObject *, PyLongObject *));
1071static PyLongObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001072x_sub(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001073 PyLongObject *a, *b;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001074{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001075 int size_a = ABS(a->ob_size), size_b = ABS(b->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001076 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001077 int i;
1078 int sign = 1;
1079 digit borrow = 0;
1080
1081 /* Ensure a is the larger of the two: */
1082 if (size_a < size_b) {
1083 sign = -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001084 { PyLongObject *temp = a; a = b; b = temp; }
1085 { int size_temp = size_a;
1086 size_a = size_b;
1087 size_b = size_temp; }
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001088 }
1089 else if (size_a == size_b) {
1090 /* Find highest digit where a and b differ: */
1091 i = size_a;
1092 while (--i >= 0 && a->ob_digit[i] == b->ob_digit[i])
1093 ;
1094 if (i < 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001095 return _PyLong_New(0);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001096 if (a->ob_digit[i] < b->ob_digit[i]) {
1097 sign = -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001098 { PyLongObject *temp = a; a = b; b = temp; }
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001099 }
1100 size_a = size_b = i+1;
1101 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001102 z = _PyLong_New(size_a);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001103 if (z == NULL)
1104 return NULL;
1105 for (i = 0; i < size_b; ++i) {
1106 /* The following assumes unsigned arithmetic
1107 works module 2**N for some N>SHIFT. */
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001108 borrow = a->ob_digit[i] - b->ob_digit[i] - borrow;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001109 z->ob_digit[i] = borrow & MASK;
1110 borrow >>= SHIFT;
1111 borrow &= 1; /* Keep only one sign bit */
1112 }
1113 for (; i < size_a; ++i) {
1114 borrow = a->ob_digit[i] - borrow;
1115 z->ob_digit[i] = borrow & MASK;
1116 borrow >>= SHIFT;
1117 }
1118 assert(borrow == 0);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001119 if (sign < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001120 z->ob_size = -(z->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001121 return long_normalize(z);
1122}
1123
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001124static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001125long_add(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001126 PyLongObject *a;
1127 PyLongObject *b;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001128{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001129 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001130
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001131 if (a->ob_size < 0) {
1132 if (b->ob_size < 0) {
1133 z = x_add(a, b);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001134 if (z != NULL && z->ob_size != 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001135 z->ob_size = -(z->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001136 }
1137 else
1138 z = x_sub(b, a);
1139 }
1140 else {
1141 if (b->ob_size < 0)
1142 z = x_sub(a, b);
1143 else
1144 z = x_add(a, b);
1145 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001146 return (PyObject *)z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001147}
1148
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001149static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001150long_sub(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001151 PyLongObject *a;
1152 PyLongObject *b;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001153{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001154 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001155
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001156 if (a->ob_size < 0) {
1157 if (b->ob_size < 0)
1158 z = x_sub(a, b);
1159 else
1160 z = x_add(a, b);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001161 if (z != NULL && z->ob_size != 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001162 z->ob_size = -(z->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001163 }
1164 else {
1165 if (b->ob_size < 0)
1166 z = x_add(a, b);
1167 else
1168 z = x_sub(a, b);
1169 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001170 return (PyObject *)z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001171}
1172
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001173static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001174long_mul(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001175 PyLongObject *a;
1176 PyLongObject *b;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001177{
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001178 int size_a;
1179 int size_b;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001180 PyLongObject *z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001181 int i;
1182
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001183 size_a = ABS(a->ob_size);
1184 size_b = ABS(b->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001185 z = _PyLong_New(size_a + size_b);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001186 if (z == NULL)
1187 return NULL;
1188 for (i = 0; i < z->ob_size; ++i)
1189 z->ob_digit[i] = 0;
1190 for (i = 0; i < size_a; ++i) {
1191 twodigits carry = 0;
1192 twodigits f = a->ob_digit[i];
1193 int j;
1194
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001195 SIGCHECK({
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001196 Py_DECREF(z);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001197 return NULL;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +00001198 })
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001199 for (j = 0; j < size_b; ++j) {
1200 carry += z->ob_digit[i+j] + b->ob_digit[j] * f;
Guido van Rossum2095d241997-04-09 19:41:24 +00001201 z->ob_digit[i+j] = (digit) (carry & MASK);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001202 carry >>= SHIFT;
1203 }
1204 for (; carry != 0; ++j) {
1205 assert(i+j < z->ob_size);
1206 carry += z->ob_digit[i+j];
Guido van Rossum2095d241997-04-09 19:41:24 +00001207 z->ob_digit[i+j] = (digit) (carry & MASK);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001208 carry >>= SHIFT;
1209 }
1210 }
1211 if (a->ob_size < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001212 z->ob_size = -(z->ob_size);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001213 if (b->ob_size < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001214 z->ob_size = -(z->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001215 return (PyObject *) long_normalize(z);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001216}
1217
Guido van Rossume32e0141992-01-19 16:31:05 +00001218/* The / and % operators are now defined in terms of divmod().
1219 The expression a mod b has the value a - b*floor(a/b).
1220 The long_divrem function gives the remainder after division of
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001221 |a| by |b|, with the sign of a. This is also expressed
1222 as a - b*trunc(a/b), if trunc truncates towards zero.
1223 Some examples:
1224 a b a rem b a mod b
1225 13 10 3 3
1226 -13 10 -3 7
1227 13 -10 3 -7
1228 -13 -10 -3 -3
1229 So, to get from rem to mod, we have to add b if a and b
Guido van Rossum23d6f0e1991-05-14 12:06:49 +00001230 have different signs. We then subtract one from the 'div'
1231 part of the outcome to keep the invariant intact. */
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001232
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001233static int l_divmod Py_PROTO((PyLongObject *, PyLongObject *,
1234 PyLongObject **, PyLongObject **));
Guido van Rossume32e0141992-01-19 16:31:05 +00001235static int
1236l_divmod(v, w, pdiv, pmod)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001237 PyLongObject *v;
1238 PyLongObject *w;
1239 PyLongObject **pdiv;
1240 PyLongObject **pmod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001241{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001242 PyLongObject *div, *mod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001243
1244 if (long_divrem(v, w, &div, &mod) < 0)
1245 return -1;
Guido van Rossum472c04f1996-12-05 21:57:21 +00001246 if ((mod->ob_size < 0 && w->ob_size > 0) ||
1247 (mod->ob_size > 0 && w->ob_size < 0)) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001248 PyLongObject *temp;
1249 PyLongObject *one;
1250 temp = (PyLongObject *) long_add(mod, w);
1251 Py_DECREF(mod);
Guido van Rossume32e0141992-01-19 16:31:05 +00001252 mod = temp;
1253 if (mod == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001254 Py_DECREF(div);
Guido van Rossume32e0141992-01-19 16:31:05 +00001255 return -1;
1256 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001257 one = (PyLongObject *) PyLong_FromLong(1L);
Guido van Rossume32e0141992-01-19 16:31:05 +00001258 if (one == NULL ||
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001259 (temp = (PyLongObject *) long_sub(div, one)) == NULL) {
1260 Py_DECREF(mod);
1261 Py_DECREF(div);
1262 Py_XDECREF(one);
Guido van Rossume32e0141992-01-19 16:31:05 +00001263 return -1;
1264 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001265 Py_DECREF(one);
1266 Py_DECREF(div);
Guido van Rossume32e0141992-01-19 16:31:05 +00001267 div = temp;
1268 }
1269 *pdiv = div;
1270 *pmod = mod;
1271 return 0;
1272}
1273
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001274static PyObject *
Guido van Rossume32e0141992-01-19 16:31:05 +00001275long_div(v, w)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001276 PyLongObject *v;
1277 PyLongObject *w;
Guido van Rossume32e0141992-01-19 16:31:05 +00001278{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001279 PyLongObject *div, *mod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001280 if (l_divmod(v, w, &div, &mod) < 0)
1281 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001282 Py_DECREF(mod);
1283 return (PyObject *)div;
Guido van Rossume32e0141992-01-19 16:31:05 +00001284}
1285
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001286static PyObject *
Guido van Rossume32e0141992-01-19 16:31:05 +00001287long_mod(v, w)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001288 PyLongObject *v;
1289 PyLongObject *w;
Guido van Rossume32e0141992-01-19 16:31:05 +00001290{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001291 PyLongObject *div, *mod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001292 if (l_divmod(v, w, &div, &mod) < 0)
1293 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001294 Py_DECREF(div);
1295 return (PyObject *)mod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001296}
1297
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001298static PyObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001299long_divmod(v, w)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001300 PyLongObject *v;
1301 PyLongObject *w;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001302{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001303 PyObject *z;
1304 PyLongObject *div, *mod;
Guido van Rossume32e0141992-01-19 16:31:05 +00001305 if (l_divmod(v, w, &div, &mod) < 0)
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001306 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001307 z = PyTuple_New(2);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001308 if (z != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001309 PyTuple_SetItem(z, 0, (PyObject *) div);
1310 PyTuple_SetItem(z, 1, (PyObject *) mod);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001311 }
1312 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001313 Py_DECREF(div);
1314 Py_DECREF(mod);
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001315 }
1316 return z;
1317}
1318
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001319static PyObject *
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001320long_pow(a, b, c)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001321 PyLongObject *a;
1322 PyLongObject *b;
1323 PyLongObject *c;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001324{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001325 PyLongObject *z, *div, *mod;
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001326 int size_b, i;
1327
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001328 size_b = b->ob_size;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001329 if (size_b < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001330 PyErr_SetString(PyExc_ValueError,
1331 "long integer to the negative power");
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001332 return NULL;
1333 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001334 z = (PyLongObject *)PyLong_FromLong(1L);
1335 Py_INCREF(a);
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001336 for (i = 0; i < size_b; ++i) {
1337 digit bi = b->ob_digit[i];
1338 int j;
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001339
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001340 for (j = 0; j < SHIFT; ++j) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001341 PyLongObject *temp;
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001342
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001343 if (bi & 1) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001344 temp = (PyLongObject *)long_mul(z, a);
1345 Py_DECREF(z);
1346 if ((PyObject*)c!=Py_None && temp!=NULL) {
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001347 l_divmod(temp, c, &div, &mod);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001348 Py_XDECREF(div);
1349 Py_DECREF(temp);
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001350 temp = mod;
1351 }
1352 z = temp;
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001353 if (z == NULL)
1354 break;
1355 }
1356 bi >>= 1;
1357 if (bi == 0 && i+1 == size_b)
1358 break;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001359 temp = (PyLongObject *)long_mul(a, a);
1360 Py_DECREF(a);
1361 if ((PyObject*)c!=Py_None && temp!=NULL) {
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001362 l_divmod(temp, c, &div, &mod);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001363 Py_XDECREF(div);
1364 Py_DECREF(temp);
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001365 temp = mod;
1366 }
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001367 a = temp;
1368 if (a == NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001369 Py_DECREF(z);
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001370 z = NULL;
1371 break;
1372 }
1373 }
Guido van Rossumc206c761995-01-10 15:23:19 +00001374 if (a == NULL || z == NULL)
Guido van Rossumc7ec9c91991-05-28 21:58:16 +00001375 break;
1376 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001377 Py_XDECREF(a);
1378 if ((PyObject*)c!=Py_None && z!=NULL) {
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001379 l_divmod(z, c, &div, &mod);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001380 Py_XDECREF(div);
1381 Py_DECREF(z);
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001382 z=mod;
1383 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001384 return (PyObject *)z;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001385}
1386
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001387static PyObject *
Guido van Rossumc6913e71991-11-19 20:26:46 +00001388long_invert(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001389 PyLongObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001390{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001391 /* Implement ~x as -(x+1) */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001392 PyLongObject *x;
1393 PyLongObject *w;
1394 w = (PyLongObject *)PyLong_FromLong(1L);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001395 if (w == NULL)
1396 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001397 x = (PyLongObject *) long_add(v, w);
1398 Py_DECREF(w);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001399 if (x == NULL)
1400 return NULL;
1401 if (x->ob_size != 0)
1402 x->ob_size = -(x->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001403 return (PyObject *)x;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001404}
1405
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001406static PyObject *
Guido van Rossumc6913e71991-11-19 20:26:46 +00001407long_pos(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001408 PyLongObject *v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001409{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001410 Py_INCREF(v);
1411 return (PyObject *)v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001412}
1413
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001414static PyObject *
Guido van Rossumc6913e71991-11-19 20:26:46 +00001415long_neg(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001416 PyLongObject *v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001417{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001418 PyLongObject *z;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001419 int i, n;
1420 n = ABS(v->ob_size);
1421 if (n == 0) {
1422 /* -0 == 0 */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001423 Py_INCREF(v);
1424 return (PyObject *) v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001425 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001426 z = _PyLong_New(ABS(n));
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001427 if (z == NULL)
1428 return NULL;
1429 for (i = 0; i < n; i++)
1430 z->ob_digit[i] = v->ob_digit[i];
1431 z->ob_size = -(v->ob_size);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001432 return (PyObject *)z;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001433}
1434
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001435static PyObject *
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001436long_abs(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001437 PyLongObject *v;
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001438{
1439 if (v->ob_size < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001440 return long_neg(v);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001441 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001442 Py_INCREF(v);
1443 return (PyObject *)v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001444 }
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001445}
1446
Guido van Rossum23d6f0e1991-05-14 12:06:49 +00001447static int
1448long_nonzero(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001449 PyLongObject *v;
Guido van Rossum23d6f0e1991-05-14 12:06:49 +00001450{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001451 return ABS(v->ob_size) != 0;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001452}
1453
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001454static PyObject *
Guido van Rossumc6913e71991-11-19 20:26:46 +00001455long_rshift(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001456 PyLongObject *a;
1457 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001458{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001459 PyLongObject *z;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001460 long shiftby;
1461 int newsize, wordshift, loshift, hishift, i, j;
1462 digit lomask, himask;
1463
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001464 if (a->ob_size < 0) {
1465 /* Right shifting negative numbers is harder */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001466 PyLongObject *a1, *a2, *a3;
1467 a1 = (PyLongObject *) long_invert(a);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001468 if (a1 == NULL) return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001469 a2 = (PyLongObject *) long_rshift(a1, b);
1470 Py_DECREF(a1);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001471 if (a2 == NULL) return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001472 a3 = (PyLongObject *) long_invert(a2);
1473 Py_DECREF(a2);
1474 return (PyObject *) a3;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001475 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001476
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001477 shiftby = PyLong_AsLong((PyObject *)b);
1478 if (shiftby == -1L && PyErr_Occurred())
Guido van Rossumc6913e71991-11-19 20:26:46 +00001479 return NULL;
1480 if (shiftby < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001481 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumc6913e71991-11-19 20:26:46 +00001482 return NULL;
1483 }
Guido van Rossumc6913e71991-11-19 20:26:46 +00001484 wordshift = shiftby / SHIFT;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001485 newsize = ABS(a->ob_size) - wordshift;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001486 if (newsize <= 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001487 z = _PyLong_New(0);
1488 return (PyObject *)z;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001489 }
1490 loshift = shiftby % SHIFT;
1491 hishift = SHIFT - loshift;
1492 lomask = ((digit)1 << hishift) - 1;
1493 himask = MASK ^ lomask;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001494 z = _PyLong_New(newsize);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001495 if (z == NULL)
1496 return NULL;
1497 if (a->ob_size < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001498 z->ob_size = -(z->ob_size);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001499 for (i = 0, j = wordshift; i < newsize; i++, j++) {
1500 z->ob_digit[i] = (a->ob_digit[j] >> loshift) & lomask;
1501 if (i+1 < newsize)
1502 z->ob_digit[i] |=
1503 (a->ob_digit[j+1] << hishift) & himask;
1504 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001505 return (PyObject *) long_normalize(z);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001506}
1507
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001508static PyObject *
Guido van Rossumc6913e71991-11-19 20:26:46 +00001509long_lshift(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001510 PyLongObject *a;
1511 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001512{
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001513 /* This version due to Tim Peters */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001514 PyLongObject *z;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001515 long shiftby;
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001516 int oldsize, newsize, wordshift, remshift, i, j;
1517 twodigits accum;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001518
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001519 shiftby = PyLong_AsLong((PyObject *)b);
1520 if (shiftby == -1L && PyErr_Occurred())
Guido van Rossumc6913e71991-11-19 20:26:46 +00001521 return NULL;
1522 if (shiftby < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001523 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumc6913e71991-11-19 20:26:46 +00001524 return NULL;
1525 }
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001526 if ((long)(int)shiftby != shiftby) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001527 PyErr_SetString(PyExc_ValueError,
1528 "outrageous left shift count");
Guido van Rossumc6913e71991-11-19 20:26:46 +00001529 return NULL;
1530 }
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001531 /* wordshift, remshift = divmod(shiftby, SHIFT) */
1532 wordshift = (int)shiftby / SHIFT;
1533 remshift = (int)shiftby - wordshift * SHIFT;
1534
1535 oldsize = ABS(a->ob_size);
1536 newsize = oldsize + wordshift;
1537 if (remshift)
1538 ++newsize;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001539 z = _PyLong_New(newsize);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001540 if (z == NULL)
1541 return NULL;
1542 if (a->ob_size < 0)
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001543 z->ob_size = -(z->ob_size);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001544 for (i = 0; i < wordshift; i++)
1545 z->ob_digit[i] = 0;
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001546 accum = 0;
1547 for (i = wordshift, j = 0; j < oldsize; i++, j++) {
1548 accum |= a->ob_digit[j] << remshift;
1549 z->ob_digit[i] = (digit)(accum & MASK);
1550 accum >>= SHIFT;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001551 }
Guido van Rossumf2e499b1997-03-16 00:37:59 +00001552 if (remshift)
1553 z->ob_digit[newsize-1] = (digit)accum;
1554 else
1555 assert(!accum);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001556 return (PyObject *) long_normalize(z);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001557}
1558
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001559
1560/* Bitwise and/xor/or operations */
1561
Guido van Rossumafbb8db1991-12-31 13:14:13 +00001562#define MAX(x, y) ((x) < (y) ? (y) : (x))
1563#define MIN(x, y) ((x) > (y) ? (y) : (x))
1564
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001565static PyObject *long_bitwise Py_PROTO((PyLongObject *, int, PyLongObject *));
1566static PyObject *
Guido van Rossume32e0141992-01-19 16:31:05 +00001567long_bitwise(a, op, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001568 PyLongObject *a;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001569 int op; /* '&', '|', '^' */
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001570 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001571{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001572 digit maska, maskb; /* 0 or MASK */
1573 int negz;
1574 int size_a, size_b, size_z;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001575 PyLongObject *z;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001576 int i;
Guido van Rossum8b27d921992-03-27 17:27:05 +00001577 digit diga, digb;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001578 PyObject *v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001579
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001580 if (a->ob_size < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001581 a = (PyLongObject *) long_invert(a);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001582 maska = MASK;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001583 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001584 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001585 Py_INCREF(a);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001586 maska = 0;
Guido van Rossumafbb8db1991-12-31 13:14:13 +00001587 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001588 if (b->ob_size < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001589 b = (PyLongObject *) long_invert(b);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001590 maskb = MASK;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001591 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001592 else {
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001593 Py_INCREF(b);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001594 maskb = 0;
1595 }
Guido van Rossumc6913e71991-11-19 20:26:46 +00001596
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001597 negz = 0;
1598 switch (op) {
1599 case '^':
1600 if (maska != maskb) {
1601 maska ^= MASK;
1602 negz = -1;
1603 }
1604 break;
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001605 case '&':
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001606 if (maska && maskb) {
1607 op = '|';
1608 maska ^= MASK;
1609 maskb ^= MASK;
1610 negz = -1;
1611 }
1612 break;
1613 case '|':
1614 if (maska || maskb) {
1615 op = '&';
1616 maska ^= MASK;
1617 maskb ^= MASK;
1618 negz = -1;
1619 }
1620 break;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001621 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001622
Guido van Rossumbd3a5271998-08-11 15:04:47 +00001623 /* JRH: The original logic here was to allocate the result value (z)
1624 as the longer of the two operands. However, there are some cases
1625 where the result is guaranteed to be shorter than that: AND of two
1626 positives, OR of two negatives: use the shorter number. AND with
1627 mixed signs: use the positive number. OR with mixed signs: use the
1628 negative number. After the transformations above, op will be '&'
1629 iff one of these cases applies, and mask will be non-0 for operands
1630 whose length should be ignored.
1631 */
1632
1633 size_a = a->ob_size;
1634 size_b = b->ob_size;
1635 size_z = op == '&'
1636 ? (maska
1637 ? size_b
1638 : (maskb ? size_a : MIN(size_a, size_b)))
1639 : MAX(size_a, size_b);
1640 z = _PyLong_New(size_z);
1641 if (a == NULL || b == NULL || z == NULL) {
1642 Py_XDECREF(a);
1643 Py_XDECREF(b);
1644 Py_XDECREF(z);
1645 return NULL;
1646 }
1647
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001648 for (i = 0; i < size_z; ++i) {
1649 diga = (i < size_a ? a->ob_digit[i] : 0) ^ maska;
1650 digb = (i < size_b ? b->ob_digit[i] : 0) ^ maskb;
1651 switch (op) {
1652 case '&': z->ob_digit[i] = diga & digb; break;
1653 case '|': z->ob_digit[i] = diga | digb; break;
1654 case '^': z->ob_digit[i] = diga ^ digb; break;
1655 }
Guido van Rossumafbb8db1991-12-31 13:14:13 +00001656 }
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001657
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001658 Py_DECREF(a);
1659 Py_DECREF(b);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001660 z = long_normalize(z);
1661 if (negz == 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001662 return (PyObject *) z;
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001663 v = long_invert(z);
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001664 Py_DECREF(z);
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001665 return v;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001666}
1667
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001668static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001669long_and(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001670 PyLongObject *a;
1671 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001672{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001673 return long_bitwise(a, '&', b);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001674}
1675
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001676static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001677long_xor(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001678 PyLongObject *a;
1679 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001680{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001681 return long_bitwise(a, '^', b);
Guido van Rossumc6913e71991-11-19 20:26:46 +00001682}
1683
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001684static PyObject *
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001685long_or(a, b)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001686 PyLongObject *a;
1687 PyLongObject *b;
Guido van Rossumc6913e71991-11-19 20:26:46 +00001688{
Guido van Rossum4c260ff1992-01-14 18:36:43 +00001689 return long_bitwise(a, '|', b);
Guido van Rossum23d6f0e1991-05-14 12:06:49 +00001690}
1691
Guido van Rossum234f9421993-06-17 12:35:49 +00001692static int
Guido van Rossume6eefc21992-08-14 12:06:52 +00001693long_coerce(pv, pw)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001694 PyObject **pv;
1695 PyObject **pw;
Guido van Rossume6eefc21992-08-14 12:06:52 +00001696{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001697 if (PyInt_Check(*pw)) {
1698 *pw = PyLong_FromLong(PyInt_AsLong(*pw));
1699 Py_INCREF(*pv);
Guido van Rossume6eefc21992-08-14 12:06:52 +00001700 return 0;
1701 }
1702 return 1; /* Can't do it */
1703}
1704
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001705static PyObject *
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001706long_int(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001707 PyObject *v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001708{
1709 long x;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001710 x = PyLong_AsLong(v);
1711 if (PyErr_Occurred())
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001712 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001713 return PyInt_FromLong(x);
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001714}
1715
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001716static PyObject *
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001717long_long(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001718 PyObject *v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001719{
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001720 Py_INCREF(v);
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001721 return v;
1722}
1723
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001724static PyObject *
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001725long_float(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001726 PyObject *v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001727{
Guido van Rossum09e6ad01997-02-14 22:54:21 +00001728 double result;
1729 PyFPE_START_PROTECT("long_float", return 0)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001730 result = PyLong_AsDouble(v);
Guido van Rossum45b83911997-03-14 04:32:50 +00001731 PyFPE_END_PROTECT(result)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001732 return PyFloat_FromDouble(result);
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001733}
1734
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001735static PyObject *
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001736long_oct(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001737 PyObject *v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001738{
1739 return long_format(v, 8);
1740}
1741
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001742static PyObject *
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001743long_hex(v)
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001744 PyObject *v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001745{
1746 return long_format(v, 16);
1747}
1748
1749
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001750#define UF (unaryfunc)
1751#define BF (binaryfunc)
1752#define TF (ternaryfunc)
1753#define IF (inquiry)
Guido van Rossum8b27d921992-03-27 17:27:05 +00001754
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001755static PyNumberMethods long_as_number = {
Guido van Rossum8b27d921992-03-27 17:27:05 +00001756 BF long_add, /*nb_add*/
1757 BF long_sub, /*nb_subtract*/
1758 BF long_mul, /*nb_multiply*/
1759 BF long_div, /*nb_divide*/
1760 BF long_mod, /*nb_remainder*/
1761 BF long_divmod, /*nb_divmod*/
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001762 TF long_pow, /*nb_power*/
Guido van Rossum8b27d921992-03-27 17:27:05 +00001763 UF long_neg, /*nb_negative*/
1764 UF long_pos, /*tp_positive*/
1765 UF long_abs, /*tp_absolute*/
1766 IF long_nonzero,/*tp_nonzero*/
1767 UF long_invert, /*nb_invert*/
1768 BF long_lshift, /*nb_lshift*/
1769 BF long_rshift, /*nb_rshift*/
1770 BF long_and, /*nb_and*/
1771 BF long_xor, /*nb_xor*/
1772 BF long_or, /*nb_or*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001773 (int (*) Py_FPROTO((PyObject **, PyObject **)))
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001774 (coercion)long_coerce, /*nb_coerce*/
Guido van Rossum1899c2e1992-09-12 11:09:23 +00001775 UF long_int, /*nb_int*/
1776 UF long_long, /*nb_long*/
1777 UF long_float, /*nb_float*/
1778 UF long_oct, /*nb_oct*/
1779 UF long_hex, /*nb_hex*/
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001780};
1781
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001782PyTypeObject PyLong_Type = {
1783 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001784 0,
1785 "long int",
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001786 sizeof(PyLongObject) - sizeof(digit),
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001787 sizeof(digit),
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001788 (destructor)long_dealloc, /*tp_dealloc*/
Guido van Rossum7066dd71992-09-17 17:54:56 +00001789 0, /*tp_print*/
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001790 0, /*tp_getattr*/
1791 0, /*tp_setattr*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001792 (int (*) Py_FPROTO((PyObject *, PyObject *)))
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001793 (cmpfunc)long_compare, /*tp_compare*/
1794 (reprfunc)long_repr, /*tp_repr*/
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001795 &long_as_number,/*tp_as_number*/
1796 0, /*tp_as_sequence*/
1797 0, /*tp_as_mapping*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001798 (long (*) Py_FPROTO((PyObject *)))
Guido van Rossumeb1fafc1994-08-29 12:47:19 +00001799 (hashfunc)long_hash, /*tp_hash*/
Guido van Rossumedcc38a1991-05-05 20:09:44 +00001800};