blob: d1f241d7ed87fff7189393ca54fce612d2d9a160 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Integer object implementation */
3
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Barry Warsaw226ae6c1999-10-12 19:54:53 +00005#include <ctype.h>
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00006
Guido van Rossum2e1d4331993-12-24 10:22:45 +00007long
Fred Drakea2f55112000-07-09 15:16:51 +00008PyInt_GetMax(void)
Guido van Rossum2e1d4331993-12-24 10:22:45 +00009{
10 return LONG_MAX; /* To initialize sys.maxint */
11}
12
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000013/* Standard Booleans */
Guido van Rossum3f5da241990-12-20 15:06:42 +000014
Guido van Rossumc0b618a1997-05-02 03:12:38 +000015PyIntObject _Py_ZeroStruct = {
16 PyObject_HEAD_INIT(&PyInt_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000017 0
18};
Guido van Rossum3f5da241990-12-20 15:06:42 +000019
Guido van Rossumc0b618a1997-05-02 03:12:38 +000020PyIntObject _Py_TrueStruct = {
21 PyObject_HEAD_INIT(&PyInt_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000022 1
23};
24
Guido van Rossume27f7952001-08-23 02:59:04 +000025/* Return 1 if exception raised, 0 if caller should retry using longs */
26static int
Fred Drakea2f55112000-07-09 15:16:51 +000027err_ovf(char *msg)
Guido van Rossum165e67e1990-10-14 20:02:26 +000028{
Guido van Rossume27f7952001-08-23 02:59:04 +000029 if (PyErr_Warn(PyExc_OverflowWarning, msg) < 0) {
Guido van Rossum0b131162001-08-23 21:32:40 +000030 if (PyErr_ExceptionMatches(PyExc_OverflowWarning))
31 PyErr_SetString(PyExc_OverflowError, msg);
Guido van Rossume27f7952001-08-23 02:59:04 +000032 return 1;
33 }
34 else
35 return 0;
Guido van Rossum165e67e1990-10-14 20:02:26 +000036}
37
Guido van Rossum3f5da241990-12-20 15:06:42 +000038/* Integers are quite normal objects, to make object handling uniform.
39 (Using odd pointers to represent integers would save much space
40 but require extra checks for this special case throughout the code.)
41 Since, a typical Python program spends much of its time allocating
42 and deallocating integers, these operations should be very fast.
43 Therefore we use a dedicated allocation scheme with a much lower
44 overhead (in space and time) than straight malloc(): a simple
45 dedicated free list, filled when necessary with memory from malloc().
46*/
47
48#define BLOCK_SIZE 1000 /* 1K less typical malloc overhead */
Guido van Rossum3fce8831999-03-12 19:43:17 +000049#define BHEAD_SIZE 8 /* Enough for a 64-bit pointer */
50#define N_INTOBJECTS ((BLOCK_SIZE - BHEAD_SIZE) / sizeof(PyIntObject))
Guido van Rossumda084ed1999-03-10 22:55:24 +000051
Guido van Rossum3fce8831999-03-12 19:43:17 +000052struct _intblock {
53 struct _intblock *next;
54 PyIntObject objects[N_INTOBJECTS];
55};
56
57typedef struct _intblock PyIntBlock;
58
59static PyIntBlock *block_list = NULL;
60static PyIntObject *free_list = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000061
Guido van Rossumc0b618a1997-05-02 03:12:38 +000062static PyIntObject *
Fred Drakea2f55112000-07-09 15:16:51 +000063fill_free_list(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +000064{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000065 PyIntObject *p, *q;
Guido van Rossumb18618d2000-05-03 23:44:39 +000066 /* XXX Int blocks escape the object heap. Use PyObject_MALLOC ??? */
67 p = (PyIntObject *) PyMem_MALLOC(sizeof(PyIntBlock));
Guido van Rossum3f5da241990-12-20 15:06:42 +000068 if (p == NULL)
Guido van Rossumb18618d2000-05-03 23:44:39 +000069 return (PyIntObject *) PyErr_NoMemory();
Guido van Rossum3fce8831999-03-12 19:43:17 +000070 ((PyIntBlock *)p)->next = block_list;
71 block_list = (PyIntBlock *)p;
72 p = &((PyIntBlock *)p)->objects[0];
Guido van Rossum3f5da241990-12-20 15:06:42 +000073 q = p + N_INTOBJECTS;
74 while (--q > p)
Guido van Rossumda084ed1999-03-10 22:55:24 +000075 q->ob_type = (struct _typeobject *)(q-1);
76 q->ob_type = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000077 return p + N_INTOBJECTS - 1;
78}
79
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000080#ifndef NSMALLPOSINTS
81#define NSMALLPOSINTS 100
82#endif
83#ifndef NSMALLNEGINTS
84#define NSMALLNEGINTS 1
85#endif
86#if NSMALLNEGINTS + NSMALLPOSINTS > 0
87/* References to small integers are saved in this array so that they
88 can be shared.
89 The integers that are saved are those in the range
90 -NSMALLNEGINTS (inclusive) to NSMALLPOSINTS (not inclusive).
91*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +000092static PyIntObject *small_ints[NSMALLNEGINTS + NSMALLPOSINTS];
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000093#endif
94#ifdef COUNT_ALLOCS
95int quick_int_allocs, quick_neg_int_allocs;
96#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +000097
Guido van Rossumc0b618a1997-05-02 03:12:38 +000098PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +000099PyInt_FromLong(long ival)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000100{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000101 register PyIntObject *v;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000102#if NSMALLNEGINTS + NSMALLPOSINTS > 0
103 if (-NSMALLNEGINTS <= ival && ival < NSMALLPOSINTS &&
104 (v = small_ints[ival + NSMALLNEGINTS]) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000105 Py_INCREF(v);
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000106#ifdef COUNT_ALLOCS
107 if (ival >= 0)
108 quick_int_allocs++;
109 else
110 quick_neg_int_allocs++;
111#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000112 return (PyObject *) v;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000113 }
114#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +0000115 if (free_list == NULL) {
116 if ((free_list = fill_free_list()) == NULL)
117 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000118 }
Guido van Rossumb18618d2000-05-03 23:44:39 +0000119 /* PyObject_New is inlined */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000120 v = free_list;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000121 free_list = (PyIntObject *)v->ob_type;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000122 PyObject_INIT(v, &PyInt_Type);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000123 v->ob_ival = ival;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000124#if NSMALLNEGINTS + NSMALLPOSINTS > 0
125 if (-NSMALLNEGINTS <= ival && ival < NSMALLPOSINTS) {
126 /* save this one for a following allocation */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000127 Py_INCREF(v);
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000128 small_ints[ival + NSMALLNEGINTS] = v;
129 }
130#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000131 return (PyObject *) v;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000132}
133
134static void
Fred Drakea2f55112000-07-09 15:16:51 +0000135int_dealloc(PyIntObject *v)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000136{
Guido van Rossumbef14172001-08-29 15:47:46 +0000137 if (v->ob_type == &PyInt_Type) {
138 v->ob_type = (struct _typeobject *)free_list;
139 free_list = v;
140 }
141 else
142 v->ob_type->tp_free((PyObject *)v);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000143}
144
145long
Fred Drakea2f55112000-07-09 15:16:51 +0000146PyInt_AsLong(register PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000147{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000148 PyNumberMethods *nb;
149 PyIntObject *io;
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000150 long val;
151
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000152 if (op && PyInt_Check(op))
153 return PyInt_AS_LONG((PyIntObject*) op);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000154
155 if (op == NULL || (nb = op->ob_type->tp_as_number) == NULL ||
156 nb->nb_int == NULL) {
Guido van Rossumc18a6f42000-05-09 14:27:48 +0000157 PyErr_SetString(PyExc_TypeError, "an integer is required");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000158 return -1;
159 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000160
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000161 io = (PyIntObject*) (*nb->nb_int) (op);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000162 if (io == NULL)
163 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000164 if (!PyInt_Check(io)) {
165 PyErr_SetString(PyExc_TypeError,
166 "nb_int should return int object");
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000167 return -1;
168 }
169
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000170 val = PyInt_AS_LONG(io);
171 Py_DECREF(io);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000172
173 return val;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000174}
175
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000176PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000177PyInt_FromString(char *s, char **pend, int base)
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000178{
179 char *end;
180 long x;
181 char buffer[256]; /* For errors */
182
183 if ((base != 0 && base < 2) || base > 36) {
Fred Drake661ea262000-10-24 19:57:45 +0000184 PyErr_SetString(PyExc_ValueError, "int() base must be >= 2 and <= 36");
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000185 return NULL;
186 }
187
188 while (*s && isspace(Py_CHARMASK(*s)))
189 s++;
190 errno = 0;
191 if (base == 0 && s[0] == '0')
192 x = (long) PyOS_strtoul(s, &end, base);
193 else
194 x = PyOS_strtol(s, &end, base);
Martin v. Löwis2b6727b2001-03-06 12:12:02 +0000195 if (end == s || !isalnum(Py_CHARMASK(end[-1])))
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000196 goto bad;
197 while (*end && isspace(Py_CHARMASK(*end)))
198 end++;
199 if (*end != '\0') {
200 bad:
201 sprintf(buffer, "invalid literal for int(): %.200s", s);
202 PyErr_SetString(PyExc_ValueError, buffer);
203 return NULL;
204 }
205 else if (errno != 0) {
206 sprintf(buffer, "int() literal too large: %.200s", s);
207 PyErr_SetString(PyExc_ValueError, buffer);
208 return NULL;
209 }
210 if (pend)
211 *pend = end;
212 return PyInt_FromLong(x);
213}
214
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000215#ifdef Py_USING_UNICODE
Guido van Rossum9e896b32000-04-05 20:11:21 +0000216PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000217PyInt_FromUnicode(Py_UNICODE *s, int length, int base)
Guido van Rossum9e896b32000-04-05 20:11:21 +0000218{
219 char buffer[256];
220
221 if (length >= sizeof(buffer)) {
222 PyErr_SetString(PyExc_ValueError,
223 "int() literal too large to convert");
224 return NULL;
225 }
226 if (PyUnicode_EncodeDecimal(s, length, buffer, NULL))
227 return NULL;
228 return PyInt_FromString(buffer, NULL, base);
229}
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000230#endif
Guido van Rossum9e896b32000-04-05 20:11:21 +0000231
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000232/* Methods */
233
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000234/* Integers are seen as the "smallest" of all numeric types and thus
235 don't have any knowledge about conversion of other types to
236 integers. */
237
238#define CONVERT_TO_LONG(obj, lng) \
239 if (PyInt_Check(obj)) { \
240 lng = PyInt_AS_LONG(obj); \
241 } \
242 else { \
243 Py_INCREF(Py_NotImplemented); \
244 return Py_NotImplemented; \
245 }
246
Guido van Rossum719f5fa1992-03-27 17:31:02 +0000247/* ARGSUSED */
Guido van Rossum90933611991-06-07 16:10:43 +0000248static int
Fred Drakea2f55112000-07-09 15:16:51 +0000249int_print(PyIntObject *v, FILE *fp, int flags)
250 /* flags -- not used but required by interface */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000251{
252 fprintf(fp, "%ld", v->ob_ival);
Guido van Rossum90933611991-06-07 16:10:43 +0000253 return 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000254}
255
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000256static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000257int_repr(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000258{
259 char buf[20];
260 sprintf(buf, "%ld", v->ob_ival);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000261 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000262}
263
264static int
Fred Drakea2f55112000-07-09 15:16:51 +0000265int_compare(PyIntObject *v, PyIntObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000266{
267 register long i = v->ob_ival;
268 register long j = w->ob_ival;
269 return (i < j) ? -1 : (i > j) ? 1 : 0;
270}
271
Guido van Rossum9bfef441993-03-29 10:43:31 +0000272static long
Fred Drakea2f55112000-07-09 15:16:51 +0000273int_hash(PyIntObject *v)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000274{
Guido van Rossum541cdd81997-01-06 22:53:20 +0000275 /* XXX If this is changed, you also need to change the way
276 Python's long, float and complex types are hashed. */
Guido van Rossum9bfef441993-03-29 10:43:31 +0000277 long x = v -> ob_ival;
278 if (x == -1)
279 x = -2;
280 return x;
281}
282
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000283static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000284int_add(PyIntObject *v, PyIntObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000285{
286 register long a, b, x;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000287 CONVERT_TO_LONG(v, a);
288 CONVERT_TO_LONG(w, b);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000289 x = a + b;
Guido van Rossume27f7952001-08-23 02:59:04 +0000290 if ((x^a) >= 0 || (x^b) >= 0)
291 return PyInt_FromLong(x);
292 if (err_ovf("integer addition"))
293 return NULL;
294 return PyLong_Type.tp_as_number->nb_add((PyObject *)v, (PyObject *)w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000295}
296
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000297static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000298int_sub(PyIntObject *v, PyIntObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000299{
300 register long a, b, x;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000301 CONVERT_TO_LONG(v, a);
302 CONVERT_TO_LONG(w, b);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000303 x = a - b;
Guido van Rossume27f7952001-08-23 02:59:04 +0000304 if ((x^a) >= 0 || (x^~b) >= 0)
305 return PyInt_FromLong(x);
306 if (err_ovf("integer subtraction"))
307 return NULL;
308 return PyLong_Type.tp_as_number->nb_subtract((PyObject *)v,
309 (PyObject *)w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000310}
311
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000312/*
313Integer overflow checking used to be done using a double, but on 64
314bit machines (where both long and double are 64 bit) this fails
Thomas Wouters7e474022000-07-16 12:04:32 +0000315because the double doesn't have enough precision. John Tromp suggests
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000316the following algorithm:
317
318Suppose again we normalize a and b to be nonnegative.
319Let ah and al (bh and bl) be the high and low 32 bits of a (b, resp.).
320Now we test ah and bh against zero and get essentially 3 possible outcomes.
321
3221) both ah and bh > 0 : then report overflow
323
3242) both ah and bh = 0 : then compute a*b and report overflow if it comes out
325 negative
326
3273) ah > 0 and bh = 0 : compute ah*bl and report overflow if it's >= 2^31
328 compute al*bl and report overflow if it's negative
329 add (ah*bl)<<32 to al*bl and report overflow if
330 it's negative
331
332In case of no overflow the result is then negated if necessary.
333
334The majority of cases will be 2), in which case this method is the same as
335what I suggested before. If multiplication is expensive enough, then the
336other method is faster on case 3), but also more work to program, so I
337guess the above is the preferred solution.
338
339*/
340
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000341static PyObject *
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000342int_mul(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000343{
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000344 long a, b, ah, bh, x, y;
345 int s = 1;
346
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000347 if (v->ob_type->tp_as_sequence &&
348 v->ob_type->tp_as_sequence->sq_repeat) {
349 /* sequence * int */
350 a = PyInt_AsLong(w);
351 return (*v->ob_type->tp_as_sequence->sq_repeat)(v, a);
352 }
353 else if (w->ob_type->tp_as_sequence &&
354 w->ob_type->tp_as_sequence->sq_repeat) {
355 /* int * sequence */
356 a = PyInt_AsLong(v);
357 return (*w->ob_type->tp_as_sequence->sq_repeat)(w, a);
358 }
359
360 CONVERT_TO_LONG(v, a);
361 CONVERT_TO_LONG(w, b);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000362 ah = a >> (LONG_BIT/2);
363 bh = b >> (LONG_BIT/2);
364
365 /* Quick test for common case: two small positive ints */
366
367 if (ah == 0 && bh == 0) {
368 x = a*b;
369 if (x < 0)
370 goto bad;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000371 return PyInt_FromLong(x);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000372 }
373
374 /* Arrange that a >= b >= 0 */
375
376 if (a < 0) {
377 a = -a;
378 if (a < 0) {
379 /* Largest negative */
380 if (b == 0 || b == 1) {
381 x = a*b;
382 goto ok;
383 }
384 else
385 goto bad;
386 }
387 s = -s;
388 ah = a >> (LONG_BIT/2);
389 }
390 if (b < 0) {
391 b = -b;
392 if (b < 0) {
393 /* Largest negative */
Guido van Rossum9478dd41996-12-06 20:14:43 +0000394 if (a == 0 || (a == 1 && s == 1)) {
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000395 x = a*b;
396 goto ok;
397 }
398 else
399 goto bad;
400 }
401 s = -s;
402 bh = b >> (LONG_BIT/2);
403 }
404
405 /* 1) both ah and bh > 0 : then report overflow */
406
407 if (ah != 0 && bh != 0)
408 goto bad;
409
410 /* 2) both ah and bh = 0 : then compute a*b and report
411 overflow if it comes out negative */
412
413 if (ah == 0 && bh == 0) {
414 x = a*b;
415 if (x < 0)
416 goto bad;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000417 return PyInt_FromLong(x*s);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000418 }
419
420 if (a < b) {
421 /* Swap */
422 x = a;
423 a = b;
424 b = x;
425 ah = bh;
426 /* bh not used beyond this point */
427 }
428
429 /* 3) ah > 0 and bh = 0 : compute ah*bl and report overflow if
430 it's >= 2^31
431 compute al*bl and report overflow if it's negative
432 add (ah*bl)<<32 to al*bl and report overflow if
433 it's negative
434 (NB b == bl in this case, and we make a = al) */
435
436 y = ah*b;
Guido van Rossum541cdd81997-01-06 22:53:20 +0000437 if (y >= (1L << (LONG_BIT/2 - 1)))
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000438 goto bad;
439 a &= (1L << (LONG_BIT/2)) - 1;
440 x = a*b;
441 if (x < 0)
442 goto bad;
Guido van Rossum541cdd81997-01-06 22:53:20 +0000443 x += y << (LONG_BIT/2);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000444 if (x < 0)
445 goto bad;
446 ok:
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000447 return PyInt_FromLong(x * s);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000448
449 bad:
Guido van Rossume27f7952001-08-23 02:59:04 +0000450 if (err_ovf("integer multiplication"))
451 return NULL;
452 return PyLong_Type.tp_as_number->nb_multiply(v, w);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000453}
454
Guido van Rossume27f7952001-08-23 02:59:04 +0000455/* Return type of i_divmod */
456enum divmod_result {
457 DIVMOD_OK, /* Correct result */
458 DIVMOD_OVERFLOW, /* Overflow, try again using longs */
459 DIVMOD_ERROR /* Exception raised */
460};
461
462static enum divmod_result
Tim Peters1dad6a82001-06-18 19:21:11 +0000463i_divmod(register long x, register long y,
Fred Drakea2f55112000-07-09 15:16:51 +0000464 long *p_xdivy, long *p_xmody)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000465{
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000466 long xdivy, xmody;
467
Tim Peters1dad6a82001-06-18 19:21:11 +0000468 if (y == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000469 PyErr_SetString(PyExc_ZeroDivisionError,
Fred Drake661ea262000-10-24 19:57:45 +0000470 "integer division or modulo by zero");
Guido van Rossume27f7952001-08-23 02:59:04 +0000471 return DIVMOD_ERROR;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000472 }
Tim Peters1dad6a82001-06-18 19:21:11 +0000473 /* (-sys.maxint-1)/-1 is the only overflow case. */
474 if (y == -1 && x < 0 && x == -x) {
Guido van Rossume27f7952001-08-23 02:59:04 +0000475 if (err_ovf("integer division"))
476 return DIVMOD_ERROR;
477 return DIVMOD_OVERFLOW;
Guido van Rossum00466951991-05-05 20:08:27 +0000478 }
Tim Peters1dad6a82001-06-18 19:21:11 +0000479 xdivy = x / y;
480 xmody = x - xdivy * y;
481 /* If the signs of x and y differ, and the remainder is non-0,
482 * C89 doesn't define whether xdivy is now the floor or the
483 * ceiling of the infinitely precise quotient. We want the floor,
484 * and we have it iff the remainder's sign matches y's.
485 */
486 if (xmody && ((y ^ xmody) < 0) /* i.e. and signs differ */) {
487 xmody += y;
488 --xdivy;
489 assert(xmody && ((y ^ xmody) >= 0));
Guido van Rossum00466951991-05-05 20:08:27 +0000490 }
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000491 *p_xdivy = xdivy;
492 *p_xmody = xmody;
Guido van Rossume27f7952001-08-23 02:59:04 +0000493 return DIVMOD_OK;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000494}
495
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000496static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000497int_div(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000498{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000499 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000500 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000501 CONVERT_TO_LONG(x, xi);
502 CONVERT_TO_LONG(y, yi);
Guido van Rossume27f7952001-08-23 02:59:04 +0000503 switch (i_divmod(xi, yi, &d, &m)) {
504 case DIVMOD_OK:
505 return PyInt_FromLong(d);
506 case DIVMOD_OVERFLOW:
507 return PyLong_Type.tp_as_number->nb_divide((PyObject *)x,
508 (PyObject *)y);
509 default:
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000510 return NULL;
Guido van Rossume27f7952001-08-23 02:59:04 +0000511 }
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000512}
513
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000514static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000515int_mod(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000516{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000517 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000518 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000519 CONVERT_TO_LONG(x, xi);
520 CONVERT_TO_LONG(y, yi);
Guido van Rossume27f7952001-08-23 02:59:04 +0000521 switch (i_divmod(xi, yi, &d, &m)) {
522 case DIVMOD_OK:
523 return PyInt_FromLong(m);
524 case DIVMOD_OVERFLOW:
525 return PyLong_Type.tp_as_number->nb_remainder((PyObject *)x,
526 (PyObject *)y);
527 default:
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000528 return NULL;
Guido van Rossume27f7952001-08-23 02:59:04 +0000529 }
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000530}
531
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000532static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000533int_divmod(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000534{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000535 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000536 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000537 CONVERT_TO_LONG(x, xi);
538 CONVERT_TO_LONG(y, yi);
Guido van Rossume27f7952001-08-23 02:59:04 +0000539 switch (i_divmod(xi, yi, &d, &m)) {
540 case DIVMOD_OK:
541 return Py_BuildValue("(ll)", d, m);
542 case DIVMOD_OVERFLOW:
543 return PyLong_Type.tp_as_number->nb_divmod((PyObject *)x,
544 (PyObject *)y);
545 default:
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000546 return NULL;
Guido van Rossume27f7952001-08-23 02:59:04 +0000547 }
Guido van Rossum00466951991-05-05 20:08:27 +0000548}
549
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000550static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000551int_pow(PyIntObject *v, PyIntObject *w, PyIntObject *z)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000552{
Guido van Rossum9478dd41996-12-06 20:14:43 +0000553 register long iv, iw, iz=0, ix, temp, prev;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000554 CONVERT_TO_LONG(v, iv);
555 CONVERT_TO_LONG(w, iw);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000556 if (iw < 0) {
Guido van Rossumb82fedc2001-07-12 11:19:45 +0000557 /* Return a float. This works because we know that
558 this calls float_pow() which converts its
559 arguments to double. */
560 return PyFloat_Type.tp_as_number->nb_power(
561 (PyObject *)v, (PyObject *)w, (PyObject *)z);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000562 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000563 if ((PyObject *)z != Py_None) {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000564 CONVERT_TO_LONG(z, iz);
Guido van Rossum9478dd41996-12-06 20:14:43 +0000565 if (iz == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000566 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +0000567 "pow() arg 3 cannot be 0");
Guido van Rossum9478dd41996-12-06 20:14:43 +0000568 return NULL;
569 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000570 }
571 /*
572 * XXX: The original exponentiation code stopped looping
573 * when temp hit zero; this code will continue onwards
574 * unnecessarily, but at least it won't cause any errors.
575 * Hopefully the speed improvement from the fast exponentiation
576 * will compensate for the slight inefficiency.
577 * XXX: Better handling of overflows is desperately needed.
578 */
579 temp = iv;
580 ix = 1;
581 while (iw > 0) {
582 prev = ix; /* Save value for overflow check */
583 if (iw & 1) {
584 ix = ix*temp;
585 if (temp == 0)
586 break; /* Avoid ix / 0 */
Guido van Rossume27f7952001-08-23 02:59:04 +0000587 if (ix / temp != prev) {
588 if (err_ovf("integer exponentiation"))
589 return NULL;
590 return PyLong_Type.tp_as_number->nb_power(
591 (PyObject *)v,
592 (PyObject *)w,
Tim Peters31960db2001-08-23 21:28:33 +0000593 (PyObject *)z);
Guido van Rossume27f7952001-08-23 02:59:04 +0000594 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000595 }
596 iw >>= 1; /* Shift exponent down by 1 bit */
597 if (iw==0) break;
598 prev = temp;
599 temp *= temp; /* Square the value of temp */
Guido van Rossume27f7952001-08-23 02:59:04 +0000600 if (prev!=0 && temp/prev!=prev) {
601 if (err_ovf("integer exponentiation"))
602 return NULL;
603 return PyLong_Type.tp_as_number->nb_power(
604 (PyObject *)v, (PyObject *)w, (PyObject *)z);
605 }
Guido van Rossum9478dd41996-12-06 20:14:43 +0000606 if (iz) {
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000607 /* If we did a multiplication, perform a modulo */
608 ix = ix % iz;
609 temp = temp % iz;
610 }
611 }
Guido van Rossum9478dd41996-12-06 20:14:43 +0000612 if (iz) {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000613 long div, mod;
Guido van Rossume27f7952001-08-23 02:59:04 +0000614 switch (i_divmod(ix, iz, &div, &mod)) {
615 case DIVMOD_OK:
616 ix = mod;
617 break;
618 case DIVMOD_OVERFLOW:
619 return PyLong_Type.tp_as_number->nb_power(
620 (PyObject *)v, (PyObject *)w, (PyObject *)z);
621 default:
622 return NULL;
623 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000624 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000625 return PyInt_FromLong(ix);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000626}
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000627
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000628static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000629int_neg(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000630{
631 register long a, x;
632 a = v->ob_ival;
633 x = -a;
Guido van Rossume27f7952001-08-23 02:59:04 +0000634 if (a < 0 && x < 0) {
635 if (err_ovf("integer negation"))
636 return NULL;
637 return PyNumber_Negative(PyLong_FromLong(a));
638 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000639 return PyInt_FromLong(x);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000640}
641
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000642static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000643int_pos(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000644{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000645 Py_INCREF(v);
646 return (PyObject *)v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000647}
648
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000649static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000650int_abs(PyIntObject *v)
Guido van Rossum00466951991-05-05 20:08:27 +0000651{
652 if (v->ob_ival >= 0)
653 return int_pos(v);
654 else
655 return int_neg(v);
656}
657
Guido van Rossum0bff0151991-05-14 12:05:32 +0000658static int
Fred Drakea2f55112000-07-09 15:16:51 +0000659int_nonzero(PyIntObject *v)
Guido van Rossum0bff0151991-05-14 12:05:32 +0000660{
661 return v->ob_ival != 0;
662}
663
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000664static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000665int_invert(PyIntObject *v)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000666{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000667 return PyInt_FromLong(~v->ob_ival);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000668}
669
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000670static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000671int_lshift(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000672{
673 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000674 CONVERT_TO_LONG(v, a);
675 CONVERT_TO_LONG(w, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000676 if (b < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000677 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000678 return NULL;
679 }
680 if (a == 0 || b == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000681 Py_INCREF(v);
682 return (PyObject *) v;
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000683 }
Guido van Rossum72481a31993-10-26 15:21:51 +0000684 if (b >= LONG_BIT) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000685 return PyInt_FromLong(0L);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000686 }
Fred Drake1bc8fab2001-07-19 21:49:38 +0000687 a = (long)((unsigned long)a << b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000688 return PyInt_FromLong(a);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000689}
690
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000691static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000692int_rshift(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000693{
694 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000695 CONVERT_TO_LONG(v, a);
696 CONVERT_TO_LONG(w, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000697 if (b < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000698 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000699 return NULL;
700 }
701 if (a == 0 || b == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000702 Py_INCREF(v);
703 return (PyObject *) v;
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000704 }
Guido van Rossum72481a31993-10-26 15:21:51 +0000705 if (b >= LONG_BIT) {
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000706 if (a < 0)
707 a = -1;
708 else
709 a = 0;
710 }
711 else {
Tim Peters7d3a5112000-07-08 04:17:21 +0000712 a = Py_ARITHMETIC_RIGHT_SHIFT(long, a, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000713 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000714 return PyInt_FromLong(a);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000715}
716
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000717static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000718int_and(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000719{
720 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000721 CONVERT_TO_LONG(v, a);
722 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000723 return PyInt_FromLong(a & b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000724}
725
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000726static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000727int_xor(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000728{
729 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000730 CONVERT_TO_LONG(v, a);
731 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000732 return PyInt_FromLong(a ^ b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000733}
734
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000735static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000736int_or(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000737{
738 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000739 CONVERT_TO_LONG(v, a);
740 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000741 return PyInt_FromLong(a | b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000742}
743
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000744static PyObject *
Guido van Rossum4668b002001-08-08 05:00:18 +0000745int_true_divide(PyObject *v, PyObject *w)
746{
747 return PyFloat_Type.tp_as_number->nb_divide(v, w);
748}
749
750static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000751int_int(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000752{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000753 Py_INCREF(v);
754 return (PyObject *)v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000755}
756
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000757static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000758int_long(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000759{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000760 return PyLong_FromLong((v -> ob_ival));
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000761}
762
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000763static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000764int_float(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000765{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000766 return PyFloat_FromDouble((double)(v -> ob_ival));
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000767}
768
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000769static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000770int_oct(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000771{
Guido van Rossum6f72f971997-01-14 15:43:41 +0000772 char buf[100];
Guido van Rossum9bfef441993-03-29 10:43:31 +0000773 long x = v -> ob_ival;
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000774 if (x == 0)
775 strcpy(buf, "0");
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000776 else
Guido van Rossumebee0251997-01-12 19:48:03 +0000777 sprintf(buf, "0%lo", x);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000778 return PyString_FromString(buf);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000779}
780
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000781static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000782int_hex(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000783{
Guido van Rossum6f72f971997-01-14 15:43:41 +0000784 char buf[100];
Guido van Rossum9bfef441993-03-29 10:43:31 +0000785 long x = v -> ob_ival;
Guido van Rossumebee0251997-01-12 19:48:03 +0000786 sprintf(buf, "0x%lx", x);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000787 return PyString_FromString(buf);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000788}
789
Guido van Rossumbef14172001-08-29 15:47:46 +0000790staticforward PyObject *
791int_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds);
792
Tim Peters6d6c1a32001-08-02 04:15:00 +0000793static PyObject *
794int_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
795{
796 PyObject *x = NULL;
797 int base = -909;
798 static char *kwlist[] = {"x", "base", 0};
799
Guido van Rossumbef14172001-08-29 15:47:46 +0000800 if (type != &PyInt_Type)
801 return int_subtype_new(type, args, kwds); /* Wimp out */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000802 if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Oi:int", kwlist,
803 &x, &base))
804 return NULL;
805 if (x == NULL)
806 return PyInt_FromLong(0L);
807 if (base == -909)
808 return PyNumber_Int(x);
809 if (PyString_Check(x))
810 return PyInt_FromString(PyString_AS_STRING(x), NULL, base);
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000811#ifdef Py_USING_UNICODE
Tim Peters6d6c1a32001-08-02 04:15:00 +0000812 if (PyUnicode_Check(x))
813 return PyInt_FromUnicode(PyUnicode_AS_UNICODE(x),
814 PyUnicode_GET_SIZE(x),
815 base);
Martin v. Löwis339d0f72001-08-17 18:39:25 +0000816#endif
Tim Peters6d6c1a32001-08-02 04:15:00 +0000817 PyErr_SetString(PyExc_TypeError,
818 "int() can't convert non-string with explicit base");
819 return NULL;
820}
821
Guido van Rossumbef14172001-08-29 15:47:46 +0000822/* Wimpy, slow approach to tp_new calls for subtypes of int:
823 first create a regular int from whatever arguments we got,
824 then allocate a subtype instance and initialize its ob_ival
825 from the regular int. The regular int is then thrown away.
826*/
827static PyObject *
828int_subtype_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
829{
830 PyObject *tmp, *new;
831
832 assert(PyType_IsSubtype(type, &PyInt_Type));
833 tmp = int_new(&PyInt_Type, args, kwds);
834 if (tmp == NULL)
835 return NULL;
836 assert(PyInt_Check(tmp));
Guido van Rossumd93dce12001-08-30 03:09:31 +0000837 new = type->tp_alloc(type, 0);
Guido van Rossumbef14172001-08-29 15:47:46 +0000838 if (new == NULL)
839 return NULL;
840 ((PyIntObject *)new)->ob_ival = ((PyIntObject *)tmp)->ob_ival;
841 Py_DECREF(tmp);
842 return new;
843}
844
Tim Peters6d6c1a32001-08-02 04:15:00 +0000845static char int_doc[] =
846"int(x[, base]) -> integer\n\
847\n\
848Convert a string or number to an integer, if possible. A floating point\n\
849argument will be truncated towards zero (this does not include a string\n\
850representation of a floating point number!) When converting a string, use\n\
851the optional base. It is an error to supply a base when converting a\n\
852non-string.";
853
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000854static PyNumberMethods int_as_number = {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000855 (binaryfunc)int_add, /*nb_add*/
856 (binaryfunc)int_sub, /*nb_subtract*/
857 (binaryfunc)int_mul, /*nb_multiply*/
858 (binaryfunc)int_div, /*nb_divide*/
859 (binaryfunc)int_mod, /*nb_remainder*/
860 (binaryfunc)int_divmod, /*nb_divmod*/
861 (ternaryfunc)int_pow, /*nb_power*/
862 (unaryfunc)int_neg, /*nb_negative*/
863 (unaryfunc)int_pos, /*nb_positive*/
864 (unaryfunc)int_abs, /*nb_absolute*/
865 (inquiry)int_nonzero, /*nb_nonzero*/
866 (unaryfunc)int_invert, /*nb_invert*/
867 (binaryfunc)int_lshift, /*nb_lshift*/
868 (binaryfunc)int_rshift, /*nb_rshift*/
869 (binaryfunc)int_and, /*nb_and*/
870 (binaryfunc)int_xor, /*nb_xor*/
871 (binaryfunc)int_or, /*nb_or*/
872 0, /*nb_coerce*/
873 (unaryfunc)int_int, /*nb_int*/
874 (unaryfunc)int_long, /*nb_long*/
875 (unaryfunc)int_float, /*nb_float*/
876 (unaryfunc)int_oct, /*nb_oct*/
877 (unaryfunc)int_hex, /*nb_hex*/
878 0, /*nb_inplace_add*/
879 0, /*nb_inplace_subtract*/
880 0, /*nb_inplace_multiply*/
881 0, /*nb_inplace_divide*/
882 0, /*nb_inplace_remainder*/
883 0, /*nb_inplace_power*/
884 0, /*nb_inplace_lshift*/
885 0, /*nb_inplace_rshift*/
886 0, /*nb_inplace_and*/
887 0, /*nb_inplace_xor*/
888 0, /*nb_inplace_or*/
Guido van Rossum4668b002001-08-08 05:00:18 +0000889 (binaryfunc)int_div, /* nb_floor_divide */
890 int_true_divide, /* nb_true_divide */
891 0, /* nb_inplace_floor_divide */
892 0, /* nb_inplace_true_divide */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000893};
894
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000895PyTypeObject PyInt_Type = {
896 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000897 0,
898 "int",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000899 sizeof(PyIntObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000900 0,
Tim Peters6d6c1a32001-08-02 04:15:00 +0000901 (destructor)int_dealloc, /* tp_dealloc */
902 (printfunc)int_print, /* tp_print */
903 0, /* tp_getattr */
904 0, /* tp_setattr */
905 (cmpfunc)int_compare, /* tp_compare */
906 (reprfunc)int_repr, /* tp_repr */
907 &int_as_number, /* tp_as_number */
908 0, /* tp_as_sequence */
909 0, /* tp_as_mapping */
910 (hashfunc)int_hash, /* tp_hash */
911 0, /* tp_call */
912 0, /* tp_str */
913 PyObject_GenericGetAttr, /* tp_getattro */
914 0, /* tp_setattro */
915 0, /* tp_as_buffer */
Guido van Rossumbef14172001-08-29 15:47:46 +0000916 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_CHECKTYPES |
917 Py_TPFLAGS_BASETYPE, /* tp_flags */
Tim Peters6d6c1a32001-08-02 04:15:00 +0000918 int_doc, /* tp_doc */
919 0, /* tp_traverse */
920 0, /* tp_clear */
921 0, /* tp_richcompare */
922 0, /* tp_weaklistoffset */
923 0, /* tp_iter */
924 0, /* tp_iternext */
925 0, /* tp_methods */
926 0, /* tp_members */
927 0, /* tp_getset */
928 0, /* tp_base */
929 0, /* tp_dict */
930 0, /* tp_descr_get */
931 0, /* tp_descr_set */
932 0, /* tp_dictoffset */
933 0, /* tp_init */
934 0, /* tp_alloc */
935 int_new, /* tp_new */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000936};
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000937
938void
Fred Drakea2f55112000-07-09 15:16:51 +0000939PyInt_Fini(void)
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000940{
Guido van Rossum3fce8831999-03-12 19:43:17 +0000941 PyIntObject *p;
942 PyIntBlock *list, *next;
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000943 int i;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000944 int bc, bf; /* block count, number of freed blocks */
945 int irem, isum; /* remaining unfreed ints per block, total */
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000946
Guido van Rossumda084ed1999-03-10 22:55:24 +0000947#if NSMALLNEGINTS + NSMALLPOSINTS > 0
948 PyIntObject **q;
949
950 i = NSMALLNEGINTS + NSMALLPOSINTS;
951 q = small_ints;
952 while (--i >= 0) {
953 Py_XDECREF(*q);
954 *q++ = NULL;
955 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000956#endif
Guido van Rossumda084ed1999-03-10 22:55:24 +0000957 bc = 0;
958 bf = 0;
959 isum = 0;
960 list = block_list;
961 block_list = NULL;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000962 free_list = NULL;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000963 while (list != NULL) {
Guido van Rossumda084ed1999-03-10 22:55:24 +0000964 bc++;
965 irem = 0;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000966 for (i = 0, p = &list->objects[0];
967 i < N_INTOBJECTS;
968 i++, p++) {
Guido van Rossumbef14172001-08-29 15:47:46 +0000969 if (p->ob_type == &PyInt_Type && p->ob_refcnt != 0)
Guido van Rossumda084ed1999-03-10 22:55:24 +0000970 irem++;
971 }
Guido van Rossum3fce8831999-03-12 19:43:17 +0000972 next = list->next;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000973 if (irem) {
Guido van Rossum3fce8831999-03-12 19:43:17 +0000974 list->next = block_list;
975 block_list = list;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000976 for (i = 0, p = &list->objects[0];
977 i < N_INTOBJECTS;
978 i++, p++) {
Guido van Rossumbef14172001-08-29 15:47:46 +0000979 if (p->ob_type != &PyInt_Type ||
980 p->ob_refcnt == 0) {
Guido van Rossum51288bc1999-03-19 20:30:39 +0000981 p->ob_type = (struct _typeobject *)
982 free_list;
983 free_list = p;
984 }
985#if NSMALLNEGINTS + NSMALLPOSINTS > 0
986 else if (-NSMALLNEGINTS <= p->ob_ival &&
987 p->ob_ival < NSMALLPOSINTS &&
988 small_ints[p->ob_ival +
989 NSMALLNEGINTS] == NULL) {
990 Py_INCREF(p);
991 small_ints[p->ob_ival +
992 NSMALLNEGINTS] = p;
993 }
994#endif
995 }
Guido van Rossumda084ed1999-03-10 22:55:24 +0000996 }
997 else {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000998 PyMem_FREE(list); /* XXX PyObject_FREE ??? */
Guido van Rossumda084ed1999-03-10 22:55:24 +0000999 bf++;
1000 }
1001 isum += irem;
Guido van Rossum3fce8831999-03-12 19:43:17 +00001002 list = next;
Guido van Rossumda084ed1999-03-10 22:55:24 +00001003 }
Guido van Rossum3fce8831999-03-12 19:43:17 +00001004 if (!Py_VerboseFlag)
1005 return;
1006 fprintf(stderr, "# cleanup ints");
1007 if (!isum) {
1008 fprintf(stderr, "\n");
1009 }
1010 else {
1011 fprintf(stderr,
1012 ": %d unfreed int%s in %d out of %d block%s\n",
1013 isum, isum == 1 ? "" : "s",
1014 bc - bf, bc, bc == 1 ? "" : "s");
1015 }
1016 if (Py_VerboseFlag > 1) {
1017 list = block_list;
1018 while (list != NULL) {
Guido van Rossum51288bc1999-03-19 20:30:39 +00001019 for (i = 0, p = &list->objects[0];
1020 i < N_INTOBJECTS;
1021 i++, p++) {
Guido van Rossumbef14172001-08-29 15:47:46 +00001022 if (p->ob_type == &PyInt_Type && p->ob_refcnt != 0)
Guido van Rossum3fce8831999-03-12 19:43:17 +00001023 fprintf(stderr,
Fred Drakea44d3532000-06-30 15:01:00 +00001024 "# <int at %p, refcnt=%d, val=%ld>\n",
1025 p, p->ob_refcnt, p->ob_ival);
Guido van Rossum3fce8831999-03-12 19:43:17 +00001026 }
1027 list = list->next;
Guido van Rossumda084ed1999-03-10 22:55:24 +00001028 }
1029 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +00001030}