blob: 282da201b82ea075b2f0935b739b2ffe0b6bbf56 [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 Rossumc0b618a1997-05-02 03:12:38 +000025static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +000026err_ovf(char *msg)
Guido van Rossum165e67e1990-10-14 20:02:26 +000027{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000028 PyErr_SetString(PyExc_OverflowError, msg);
Guido van Rossum165e67e1990-10-14 20:02:26 +000029 return NULL;
30}
31
Guido van Rossum3f5da241990-12-20 15:06:42 +000032/* Integers are quite normal objects, to make object handling uniform.
33 (Using odd pointers to represent integers would save much space
34 but require extra checks for this special case throughout the code.)
35 Since, a typical Python program spends much of its time allocating
36 and deallocating integers, these operations should be very fast.
37 Therefore we use a dedicated allocation scheme with a much lower
38 overhead (in space and time) than straight malloc(): a simple
39 dedicated free list, filled when necessary with memory from malloc().
40*/
41
42#define BLOCK_SIZE 1000 /* 1K less typical malloc overhead */
Guido van Rossum3fce8831999-03-12 19:43:17 +000043#define BHEAD_SIZE 8 /* Enough for a 64-bit pointer */
44#define N_INTOBJECTS ((BLOCK_SIZE - BHEAD_SIZE) / sizeof(PyIntObject))
Guido van Rossumda084ed1999-03-10 22:55:24 +000045
Guido van Rossum3fce8831999-03-12 19:43:17 +000046struct _intblock {
47 struct _intblock *next;
48 PyIntObject objects[N_INTOBJECTS];
49};
50
51typedef struct _intblock PyIntBlock;
52
53static PyIntBlock *block_list = NULL;
54static PyIntObject *free_list = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000055
Guido van Rossumc0b618a1997-05-02 03:12:38 +000056static PyIntObject *
Fred Drakea2f55112000-07-09 15:16:51 +000057fill_free_list(void)
Guido van Rossum3f5da241990-12-20 15:06:42 +000058{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000059 PyIntObject *p, *q;
Guido van Rossumb18618d2000-05-03 23:44:39 +000060 /* XXX Int blocks escape the object heap. Use PyObject_MALLOC ??? */
61 p = (PyIntObject *) PyMem_MALLOC(sizeof(PyIntBlock));
Guido van Rossum3f5da241990-12-20 15:06:42 +000062 if (p == NULL)
Guido van Rossumb18618d2000-05-03 23:44:39 +000063 return (PyIntObject *) PyErr_NoMemory();
Guido van Rossum3fce8831999-03-12 19:43:17 +000064 ((PyIntBlock *)p)->next = block_list;
65 block_list = (PyIntBlock *)p;
66 p = &((PyIntBlock *)p)->objects[0];
Guido van Rossum3f5da241990-12-20 15:06:42 +000067 q = p + N_INTOBJECTS;
68 while (--q > p)
Guido van Rossumda084ed1999-03-10 22:55:24 +000069 q->ob_type = (struct _typeobject *)(q-1);
70 q->ob_type = NULL;
Guido van Rossum3f5da241990-12-20 15:06:42 +000071 return p + N_INTOBJECTS - 1;
72}
73
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000074#ifndef NSMALLPOSINTS
75#define NSMALLPOSINTS 100
76#endif
77#ifndef NSMALLNEGINTS
78#define NSMALLNEGINTS 1
79#endif
80#if NSMALLNEGINTS + NSMALLPOSINTS > 0
81/* References to small integers are saved in this array so that they
82 can be shared.
83 The integers that are saved are those in the range
84 -NSMALLNEGINTS (inclusive) to NSMALLPOSINTS (not inclusive).
85*/
Guido van Rossumc0b618a1997-05-02 03:12:38 +000086static PyIntObject *small_ints[NSMALLNEGINTS + NSMALLPOSINTS];
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000087#endif
88#ifdef COUNT_ALLOCS
89int quick_int_allocs, quick_neg_int_allocs;
90#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +000091
Guido van Rossumc0b618a1997-05-02 03:12:38 +000092PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +000093PyInt_FromLong(long ival)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000094{
Guido van Rossumc0b618a1997-05-02 03:12:38 +000095 register PyIntObject *v;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000096#if NSMALLNEGINTS + NSMALLPOSINTS > 0
97 if (-NSMALLNEGINTS <= ival && ival < NSMALLPOSINTS &&
98 (v = small_ints[ival + NSMALLNEGINTS]) != NULL) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +000099 Py_INCREF(v);
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000100#ifdef COUNT_ALLOCS
101 if (ival >= 0)
102 quick_int_allocs++;
103 else
104 quick_neg_int_allocs++;
105#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000106 return (PyObject *) v;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000107 }
108#endif
Guido van Rossum3f5da241990-12-20 15:06:42 +0000109 if (free_list == NULL) {
110 if ((free_list = fill_free_list()) == NULL)
111 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000112 }
Guido van Rossumb18618d2000-05-03 23:44:39 +0000113 /* PyObject_New is inlined */
Guido van Rossum3f5da241990-12-20 15:06:42 +0000114 v = free_list;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000115 free_list = (PyIntObject *)v->ob_type;
Guido van Rossumb18618d2000-05-03 23:44:39 +0000116 PyObject_INIT(v, &PyInt_Type);
Guido van Rossum3f5da241990-12-20 15:06:42 +0000117 v->ob_ival = ival;
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000118#if NSMALLNEGINTS + NSMALLPOSINTS > 0
119 if (-NSMALLNEGINTS <= ival && ival < NSMALLPOSINTS) {
120 /* save this one for a following allocation */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000121 Py_INCREF(v);
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000122 small_ints[ival + NSMALLNEGINTS] = v;
123 }
124#endif
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000125 return (PyObject *) v;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000126}
127
128static void
Fred Drakea2f55112000-07-09 15:16:51 +0000129int_dealloc(PyIntObject *v)
Guido van Rossum3f5da241990-12-20 15:06:42 +0000130{
Guido van Rossumda084ed1999-03-10 22:55:24 +0000131 v->ob_type = (struct _typeobject *)free_list;
Guido van Rossum3f5da241990-12-20 15:06:42 +0000132 free_list = v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000133}
134
135long
Fred Drakea2f55112000-07-09 15:16:51 +0000136PyInt_AsLong(register PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000137{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000138 PyNumberMethods *nb;
139 PyIntObject *io;
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000140 long val;
141
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000142 if (op && PyInt_Check(op))
143 return PyInt_AS_LONG((PyIntObject*) op);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000144
145 if (op == NULL || (nb = op->ob_type->tp_as_number) == NULL ||
146 nb->nb_int == NULL) {
Guido van Rossumc18a6f42000-05-09 14:27:48 +0000147 PyErr_SetString(PyExc_TypeError, "an integer is required");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000148 return -1;
149 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000150
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000151 io = (PyIntObject*) (*nb->nb_int) (op);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000152 if (io == NULL)
153 return -1;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000154 if (!PyInt_Check(io)) {
155 PyErr_SetString(PyExc_TypeError,
156 "nb_int should return int object");
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000157 return -1;
158 }
159
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000160 val = PyInt_AS_LONG(io);
161 Py_DECREF(io);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000162
163 return val;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000164}
165
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000166PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000167PyInt_FromString(char *s, char **pend, int base)
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000168{
169 char *end;
170 long x;
171 char buffer[256]; /* For errors */
172
173 if ((base != 0 && base < 2) || base > 36) {
Fred Drake661ea262000-10-24 19:57:45 +0000174 PyErr_SetString(PyExc_ValueError, "int() base must be >= 2 and <= 36");
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000175 return NULL;
176 }
177
178 while (*s && isspace(Py_CHARMASK(*s)))
179 s++;
180 errno = 0;
181 if (base == 0 && s[0] == '0')
182 x = (long) PyOS_strtoul(s, &end, base);
183 else
184 x = PyOS_strtol(s, &end, base);
Martin v. Löwis2b6727b2001-03-06 12:12:02 +0000185 if (end == s || !isalnum(Py_CHARMASK(end[-1])))
Barry Warsaw226ae6c1999-10-12 19:54:53 +0000186 goto bad;
187 while (*end && isspace(Py_CHARMASK(*end)))
188 end++;
189 if (*end != '\0') {
190 bad:
191 sprintf(buffer, "invalid literal for int(): %.200s", s);
192 PyErr_SetString(PyExc_ValueError, buffer);
193 return NULL;
194 }
195 else if (errno != 0) {
196 sprintf(buffer, "int() literal too large: %.200s", s);
197 PyErr_SetString(PyExc_ValueError, buffer);
198 return NULL;
199 }
200 if (pend)
201 *pend = end;
202 return PyInt_FromLong(x);
203}
204
Guido van Rossum9e896b32000-04-05 20:11:21 +0000205PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000206PyInt_FromUnicode(Py_UNICODE *s, int length, int base)
Guido van Rossum9e896b32000-04-05 20:11:21 +0000207{
208 char buffer[256];
209
210 if (length >= sizeof(buffer)) {
211 PyErr_SetString(PyExc_ValueError,
212 "int() literal too large to convert");
213 return NULL;
214 }
215 if (PyUnicode_EncodeDecimal(s, length, buffer, NULL))
216 return NULL;
217 return PyInt_FromString(buffer, NULL, base);
218}
219
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000220/* Methods */
221
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000222/* Integers are seen as the "smallest" of all numeric types and thus
223 don't have any knowledge about conversion of other types to
224 integers. */
225
226#define CONVERT_TO_LONG(obj, lng) \
227 if (PyInt_Check(obj)) { \
228 lng = PyInt_AS_LONG(obj); \
229 } \
230 else { \
231 Py_INCREF(Py_NotImplemented); \
232 return Py_NotImplemented; \
233 }
234
Guido van Rossum719f5fa1992-03-27 17:31:02 +0000235/* ARGSUSED */
Guido van Rossum90933611991-06-07 16:10:43 +0000236static int
Fred Drakea2f55112000-07-09 15:16:51 +0000237int_print(PyIntObject *v, FILE *fp, int flags)
238 /* flags -- not used but required by interface */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000239{
240 fprintf(fp, "%ld", v->ob_ival);
Guido van Rossum90933611991-06-07 16:10:43 +0000241 return 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000242}
243
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000244static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000245int_repr(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000246{
247 char buf[20];
248 sprintf(buf, "%ld", v->ob_ival);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000249 return PyString_FromString(buf);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000250}
251
252static int
Fred Drakea2f55112000-07-09 15:16:51 +0000253int_compare(PyIntObject *v, PyIntObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000254{
255 register long i = v->ob_ival;
256 register long j = w->ob_ival;
257 return (i < j) ? -1 : (i > j) ? 1 : 0;
258}
259
Guido van Rossum9bfef441993-03-29 10:43:31 +0000260static long
Fred Drakea2f55112000-07-09 15:16:51 +0000261int_hash(PyIntObject *v)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000262{
Guido van Rossum541cdd81997-01-06 22:53:20 +0000263 /* XXX If this is changed, you also need to change the way
264 Python's long, float and complex types are hashed. */
Guido van Rossum9bfef441993-03-29 10:43:31 +0000265 long x = v -> ob_ival;
266 if (x == -1)
267 x = -2;
268 return x;
269}
270
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000271static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000272int_add(PyIntObject *v, PyIntObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000273{
274 register long a, b, x;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000275 CONVERT_TO_LONG(v, a);
276 CONVERT_TO_LONG(w, b);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000277 x = a + b;
Guido van Rossum165e67e1990-10-14 20:02:26 +0000278 if ((x^a) < 0 && (x^b) < 0)
Guido van Rossum3a628451991-12-10 13:57:36 +0000279 return err_ovf("integer addition");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000280 return PyInt_FromLong(x);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000281}
282
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000283static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000284int_sub(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 Rossum165e67e1990-10-14 20:02:26 +0000290 if ((x^a) < 0 && (x^~b) < 0)
Guido van Rossum3a628451991-12-10 13:57:36 +0000291 return err_ovf("integer subtraction");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000292 return PyInt_FromLong(x);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000293}
294
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000295/*
296Integer overflow checking used to be done using a double, but on 64
297bit machines (where both long and double are 64 bit) this fails
Thomas Wouters7e474022000-07-16 12:04:32 +0000298because the double doesn't have enough precision. John Tromp suggests
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000299the following algorithm:
300
301Suppose again we normalize a and b to be nonnegative.
302Let ah and al (bh and bl) be the high and low 32 bits of a (b, resp.).
303Now we test ah and bh against zero and get essentially 3 possible outcomes.
304
3051) both ah and bh > 0 : then report overflow
306
3072) both ah and bh = 0 : then compute a*b and report overflow if it comes out
308 negative
309
3103) ah > 0 and bh = 0 : compute ah*bl and report overflow if it's >= 2^31
311 compute al*bl and report overflow if it's negative
312 add (ah*bl)<<32 to al*bl and report overflow if
313 it's negative
314
315In case of no overflow the result is then negated if necessary.
316
317The majority of cases will be 2), in which case this method is the same as
318what I suggested before. If multiplication is expensive enough, then the
319other method is faster on case 3), but also more work to program, so I
320guess the above is the preferred solution.
321
322*/
323
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000324static PyObject *
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000325int_mul(PyObject *v, PyObject *w)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000326{
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000327 long a, b, ah, bh, x, y;
328 int s = 1;
329
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000330 if (v->ob_type->tp_as_sequence &&
331 v->ob_type->tp_as_sequence->sq_repeat) {
332 /* sequence * int */
333 a = PyInt_AsLong(w);
334 return (*v->ob_type->tp_as_sequence->sq_repeat)(v, a);
335 }
336 else if (w->ob_type->tp_as_sequence &&
337 w->ob_type->tp_as_sequence->sq_repeat) {
338 /* int * sequence */
339 a = PyInt_AsLong(v);
340 return (*w->ob_type->tp_as_sequence->sq_repeat)(w, a);
341 }
342
343 CONVERT_TO_LONG(v, a);
344 CONVERT_TO_LONG(w, b);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000345 ah = a >> (LONG_BIT/2);
346 bh = b >> (LONG_BIT/2);
347
348 /* Quick test for common case: two small positive ints */
349
350 if (ah == 0 && bh == 0) {
351 x = a*b;
352 if (x < 0)
353 goto bad;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000354 return PyInt_FromLong(x);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000355 }
356
357 /* Arrange that a >= b >= 0 */
358
359 if (a < 0) {
360 a = -a;
361 if (a < 0) {
362 /* Largest negative */
363 if (b == 0 || b == 1) {
364 x = a*b;
365 goto ok;
366 }
367 else
368 goto bad;
369 }
370 s = -s;
371 ah = a >> (LONG_BIT/2);
372 }
373 if (b < 0) {
374 b = -b;
375 if (b < 0) {
376 /* Largest negative */
Guido van Rossum9478dd41996-12-06 20:14:43 +0000377 if (a == 0 || (a == 1 && s == 1)) {
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000378 x = a*b;
379 goto ok;
380 }
381 else
382 goto bad;
383 }
384 s = -s;
385 bh = b >> (LONG_BIT/2);
386 }
387
388 /* 1) both ah and bh > 0 : then report overflow */
389
390 if (ah != 0 && bh != 0)
391 goto bad;
392
393 /* 2) both ah and bh = 0 : then compute a*b and report
394 overflow if it comes out negative */
395
396 if (ah == 0 && bh == 0) {
397 x = a*b;
398 if (x < 0)
399 goto bad;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000400 return PyInt_FromLong(x*s);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000401 }
402
403 if (a < b) {
404 /* Swap */
405 x = a;
406 a = b;
407 b = x;
408 ah = bh;
409 /* bh not used beyond this point */
410 }
411
412 /* 3) ah > 0 and bh = 0 : compute ah*bl and report overflow if
413 it's >= 2^31
414 compute al*bl and report overflow if it's negative
415 add (ah*bl)<<32 to al*bl and report overflow if
416 it's negative
417 (NB b == bl in this case, and we make a = al) */
418
419 y = ah*b;
Guido van Rossum541cdd81997-01-06 22:53:20 +0000420 if (y >= (1L << (LONG_BIT/2 - 1)))
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000421 goto bad;
422 a &= (1L << (LONG_BIT/2)) - 1;
423 x = a*b;
424 if (x < 0)
425 goto bad;
Guido van Rossum541cdd81997-01-06 22:53:20 +0000426 x += y << (LONG_BIT/2);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000427 if (x < 0)
428 goto bad;
429 ok:
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000430 return PyInt_FromLong(x * s);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000431
432 bad:
433 return err_ovf("integer multiplication");
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000434}
435
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000436static int
Tim Peters1dad6a82001-06-18 19:21:11 +0000437i_divmod(register long x, register long y,
Fred Drakea2f55112000-07-09 15:16:51 +0000438 long *p_xdivy, long *p_xmody)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000439{
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000440 long xdivy, xmody;
441
Tim Peters1dad6a82001-06-18 19:21:11 +0000442 if (y == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000443 PyErr_SetString(PyExc_ZeroDivisionError,
Fred Drake661ea262000-10-24 19:57:45 +0000444 "integer division or modulo by zero");
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000445 return -1;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000446 }
Tim Peters1dad6a82001-06-18 19:21:11 +0000447 /* (-sys.maxint-1)/-1 is the only overflow case. */
448 if (y == -1 && x < 0 && x == -x) {
449 err_ovf("integer division");
450 return -1;
Guido van Rossum00466951991-05-05 20:08:27 +0000451 }
Tim Peters1dad6a82001-06-18 19:21:11 +0000452 xdivy = x / y;
453 xmody = x - xdivy * y;
454 /* If the signs of x and y differ, and the remainder is non-0,
455 * C89 doesn't define whether xdivy is now the floor or the
456 * ceiling of the infinitely precise quotient. We want the floor,
457 * and we have it iff the remainder's sign matches y's.
458 */
459 if (xmody && ((y ^ xmody) < 0) /* i.e. and signs differ */) {
460 xmody += y;
461 --xdivy;
462 assert(xmody && ((y ^ xmody) >= 0));
Guido van Rossum00466951991-05-05 20:08:27 +0000463 }
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000464 *p_xdivy = xdivy;
465 *p_xmody = xmody;
466 return 0;
467}
468
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000469static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000470int_div(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000471{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000472 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000473 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000474 CONVERT_TO_LONG(x, xi);
475 CONVERT_TO_LONG(y, yi);
476 if (i_divmod(xi, yi, &d, &m) < 0)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000477 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000478 return PyInt_FromLong(d);
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000479}
480
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000481static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000482int_mod(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000483{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000484 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000485 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000486 CONVERT_TO_LONG(x, xi);
487 CONVERT_TO_LONG(y, yi);
488 if (i_divmod(xi, yi, &d, &m) < 0)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000489 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000490 return PyInt_FromLong(m);
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000491}
492
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000493static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000494int_divmod(PyIntObject *x, PyIntObject *y)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000495{
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000496 long xi, yi;
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000497 long d, m;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000498 CONVERT_TO_LONG(x, xi);
499 CONVERT_TO_LONG(y, yi);
500 if (i_divmod(xi, yi, &d, &m) < 0)
Guido van Rossum2b16a6f1992-01-19 16:28:51 +0000501 return NULL;
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000502 return Py_BuildValue("(ll)", d, m);
Guido van Rossum00466951991-05-05 20:08:27 +0000503}
504
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000505static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000506int_pow(PyIntObject *v, PyIntObject *w, PyIntObject *z)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000507{
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000508#if 1
Guido van Rossum9478dd41996-12-06 20:14:43 +0000509 register long iv, iw, iz=0, ix, temp, prev;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000510 CONVERT_TO_LONG(v, iv);
511 CONVERT_TO_LONG(w, iw);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000512 if (iw < 0) {
Guido van Rossumb82fedc2001-07-12 11:19:45 +0000513 /* Return a float. This works because we know that
514 this calls float_pow() which converts its
515 arguments to double. */
516 return PyFloat_Type.tp_as_number->nb_power(
517 (PyObject *)v, (PyObject *)w, (PyObject *)z);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000518 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000519 if ((PyObject *)z != Py_None) {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000520 CONVERT_TO_LONG(z, iz);
Guido van Rossum9478dd41996-12-06 20:14:43 +0000521 if (iz == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000522 PyErr_SetString(PyExc_ValueError,
Fred Drake661ea262000-10-24 19:57:45 +0000523 "pow() arg 3 cannot be 0");
Guido van Rossum9478dd41996-12-06 20:14:43 +0000524 return NULL;
525 }
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000526 }
527 /*
528 * XXX: The original exponentiation code stopped looping
529 * when temp hit zero; this code will continue onwards
530 * unnecessarily, but at least it won't cause any errors.
531 * Hopefully the speed improvement from the fast exponentiation
532 * will compensate for the slight inefficiency.
533 * XXX: Better handling of overflows is desperately needed.
534 */
535 temp = iv;
536 ix = 1;
537 while (iw > 0) {
538 prev = ix; /* Save value for overflow check */
539 if (iw & 1) {
540 ix = ix*temp;
541 if (temp == 0)
542 break; /* Avoid ix / 0 */
543 if (ix / temp != prev)
Guido van Rossumfb4574e2000-02-15 14:51:46 +0000544 return err_ovf("integer exponentiation");
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000545 }
546 iw >>= 1; /* Shift exponent down by 1 bit */
547 if (iw==0) break;
548 prev = temp;
549 temp *= temp; /* Square the value of temp */
550 if (prev!=0 && temp/prev!=prev)
Guido van Rossumfb4574e2000-02-15 14:51:46 +0000551 return err_ovf("integer exponentiation");
Guido van Rossum9478dd41996-12-06 20:14:43 +0000552 if (iz) {
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000553 /* If we did a multiplication, perform a modulo */
554 ix = ix % iz;
555 temp = temp % iz;
556 }
557 }
Guido van Rossum9478dd41996-12-06 20:14:43 +0000558 if (iz) {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000559 long div, mod;
560 if (i_divmod(ix, iz, &div, &mod) < 0)
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000561 return(NULL);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000562 ix=mod;
563 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000564 return PyInt_FromLong(ix);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000565#else
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000566 register long iv, iw, ix;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000567 CONVERT_TO_LONG(v, iv);
568 CONVERT_TO_LONG(w, iw);
Guido van Rossum00466951991-05-05 20:08:27 +0000569 if (iw < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000570 PyErr_SetString(PyExc_ValueError,
571 "integer to the negative power");
Guido van Rossum00466951991-05-05 20:08:27 +0000572 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000573 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000574 if ((PyObject *)z != Py_None) {
575 PyErr_SetString(PyExc_TypeError,
576 "pow(int, int, int) not yet supported");
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000577 return NULL;
578 }
Guido van Rossum00466951991-05-05 20:08:27 +0000579 ix = 1;
580 while (--iw >= 0) {
581 long prev = ix;
582 ix = ix * iv;
583 if (iv == 0)
584 break; /* 0 to some power -- avoid ix / 0 */
585 if (ix / iv != prev)
Guido van Rossumfb4574e2000-02-15 14:51:46 +0000586 return err_ovf("integer exponentiation");
Guido van Rossum00466951991-05-05 20:08:27 +0000587 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000588 return PyInt_FromLong(ix);
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000589#endif
590}
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000591
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000592static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000593int_neg(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000594{
595 register long a, x;
596 a = v->ob_ival;
597 x = -a;
Guido van Rossum165e67e1990-10-14 20:02:26 +0000598 if (a < 0 && x < 0)
Guido van Rossum3a628451991-12-10 13:57:36 +0000599 return err_ovf("integer negation");
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000600 return PyInt_FromLong(x);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000601}
602
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000603static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000604int_pos(PyIntObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000605{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000606 Py_INCREF(v);
607 return (PyObject *)v;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000608}
609
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000610static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000611int_abs(PyIntObject *v)
Guido van Rossum00466951991-05-05 20:08:27 +0000612{
613 if (v->ob_ival >= 0)
614 return int_pos(v);
615 else
616 return int_neg(v);
617}
618
Guido van Rossum0bff0151991-05-14 12:05:32 +0000619static int
Fred Drakea2f55112000-07-09 15:16:51 +0000620int_nonzero(PyIntObject *v)
Guido van Rossum0bff0151991-05-14 12:05:32 +0000621{
622 return v->ob_ival != 0;
623}
624
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000625static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000626int_invert(PyIntObject *v)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000627{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000628 return PyInt_FromLong(~v->ob_ival);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000629}
630
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000631static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000632int_lshift(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000633{
634 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000635 CONVERT_TO_LONG(v, a);
636 CONVERT_TO_LONG(w, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000637 if (b < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000638 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000639 return NULL;
640 }
641 if (a == 0 || b == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000642 Py_INCREF(v);
643 return (PyObject *) v;
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000644 }
Guido van Rossum72481a31993-10-26 15:21:51 +0000645 if (b >= LONG_BIT) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000646 return PyInt_FromLong(0L);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000647 }
Fred Drake1bc8fab2001-07-19 21:49:38 +0000648 a = (long)((unsigned long)a << b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000649 return PyInt_FromLong(a);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000650}
651
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000652static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000653int_rshift(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000654{
655 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000656 CONVERT_TO_LONG(v, a);
657 CONVERT_TO_LONG(w, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000658 if (b < 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000659 PyErr_SetString(PyExc_ValueError, "negative shift count");
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000660 return NULL;
661 }
662 if (a == 0 || b == 0) {
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000663 Py_INCREF(v);
664 return (PyObject *) v;
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000665 }
Guido van Rossum72481a31993-10-26 15:21:51 +0000666 if (b >= LONG_BIT) {
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000667 if (a < 0)
668 a = -1;
669 else
670 a = 0;
671 }
672 else {
Tim Peters7d3a5112000-07-08 04:17:21 +0000673 a = Py_ARITHMETIC_RIGHT_SHIFT(long, a, b);
Guido van Rossumf3b351f1992-01-14 18:33:22 +0000674 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000675 return PyInt_FromLong(a);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000676}
677
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000678static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000679int_and(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000680{
681 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000682 CONVERT_TO_LONG(v, a);
683 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000684 return PyInt_FromLong(a & b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000685}
686
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000687static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000688int_xor(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000689{
690 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000691 CONVERT_TO_LONG(v, a);
692 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000693 return PyInt_FromLong(a ^ b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000694}
695
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000696static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000697int_or(PyIntObject *v, PyIntObject *w)
Guido van Rossum7928cd71991-10-24 14:59:31 +0000698{
699 register long a, b;
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000700 CONVERT_TO_LONG(v, a);
701 CONVERT_TO_LONG(w, b);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000702 return PyInt_FromLong(a | b);
Guido van Rossum7928cd71991-10-24 14:59:31 +0000703}
704
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000705static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000706int_int(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000707{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000708 Py_INCREF(v);
709 return (PyObject *)v;
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000710}
711
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000712static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000713int_long(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000714{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000715 return PyLong_FromLong((v -> ob_ival));
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000716}
717
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000718static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000719int_float(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000720{
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000721 return PyFloat_FromDouble((double)(v -> ob_ival));
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000722}
723
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000724static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000725int_oct(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000726{
Guido van Rossum6f72f971997-01-14 15:43:41 +0000727 char buf[100];
Guido van Rossum9bfef441993-03-29 10:43:31 +0000728 long x = v -> ob_ival;
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000729 if (x == 0)
730 strcpy(buf, "0");
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000731 else
Guido van Rossumebee0251997-01-12 19:48:03 +0000732 sprintf(buf, "0%lo", x);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000733 return PyString_FromString(buf);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000734}
735
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000736static PyObject *
Fred Drakea2f55112000-07-09 15:16:51 +0000737int_hex(PyIntObject *v)
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000738{
Guido van Rossum6f72f971997-01-14 15:43:41 +0000739 char buf[100];
Guido van Rossum9bfef441993-03-29 10:43:31 +0000740 long x = v -> ob_ival;
Guido van Rossumebee0251997-01-12 19:48:03 +0000741 sprintf(buf, "0x%lx", x);
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000742 return PyString_FromString(buf);
Guido van Rossum1899c2e1992-09-12 11:09:23 +0000743}
744
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000745static PyNumberMethods int_as_number = {
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000746 (binaryfunc)int_add, /*nb_add*/
747 (binaryfunc)int_sub, /*nb_subtract*/
748 (binaryfunc)int_mul, /*nb_multiply*/
749 (binaryfunc)int_div, /*nb_divide*/
750 (binaryfunc)int_mod, /*nb_remainder*/
751 (binaryfunc)int_divmod, /*nb_divmod*/
752 (ternaryfunc)int_pow, /*nb_power*/
753 (unaryfunc)int_neg, /*nb_negative*/
754 (unaryfunc)int_pos, /*nb_positive*/
755 (unaryfunc)int_abs, /*nb_absolute*/
756 (inquiry)int_nonzero, /*nb_nonzero*/
757 (unaryfunc)int_invert, /*nb_invert*/
758 (binaryfunc)int_lshift, /*nb_lshift*/
759 (binaryfunc)int_rshift, /*nb_rshift*/
760 (binaryfunc)int_and, /*nb_and*/
761 (binaryfunc)int_xor, /*nb_xor*/
762 (binaryfunc)int_or, /*nb_or*/
763 0, /*nb_coerce*/
764 (unaryfunc)int_int, /*nb_int*/
765 (unaryfunc)int_long, /*nb_long*/
766 (unaryfunc)int_float, /*nb_float*/
767 (unaryfunc)int_oct, /*nb_oct*/
768 (unaryfunc)int_hex, /*nb_hex*/
769 0, /*nb_inplace_add*/
770 0, /*nb_inplace_subtract*/
771 0, /*nb_inplace_multiply*/
772 0, /*nb_inplace_divide*/
773 0, /*nb_inplace_remainder*/
774 0, /*nb_inplace_power*/
775 0, /*nb_inplace_lshift*/
776 0, /*nb_inplace_rshift*/
777 0, /*nb_inplace_and*/
778 0, /*nb_inplace_xor*/
779 0, /*nb_inplace_or*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000780};
781
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000782PyTypeObject PyInt_Type = {
783 PyObject_HEAD_INIT(&PyType_Type)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000784 0,
785 "int",
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000786 sizeof(PyIntObject),
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000787 0,
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000788 (destructor)int_dealloc, /*tp_dealloc*/
789 (printfunc)int_print, /*tp_print*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000790 0, /*tp_getattr*/
791 0, /*tp_setattr*/
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000792 (cmpfunc)int_compare, /*tp_compare*/
793 (reprfunc)int_repr, /*tp_repr*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000794 &int_as_number, /*tp_as_number*/
795 0, /*tp_as_sequence*/
796 0, /*tp_as_mapping*/
Guido van Rossumbf8c0e31994-08-29 12:48:32 +0000797 (hashfunc)int_hash, /*tp_hash*/
Neil Schemenauer139e72a2001-01-04 01:45:33 +0000798 0, /*tp_call*/
799 0, /*tp_str*/
800 0, /*tp_getattro*/
801 0, /*tp_setattro*/
802 0, /*tp_as_buffer*/
Guido van Rossum3968e4c2001-01-17 15:32:23 +0000803 Py_TPFLAGS_CHECKTYPES /*tp_flags*/
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000804};
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000805
806void
Fred Drakea2f55112000-07-09 15:16:51 +0000807PyInt_Fini(void)
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000808{
Guido van Rossum3fce8831999-03-12 19:43:17 +0000809 PyIntObject *p;
810 PyIntBlock *list, *next;
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000811 int i;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000812 int bc, bf; /* block count, number of freed blocks */
813 int irem, isum; /* remaining unfreed ints per block, total */
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000814
Guido van Rossumda084ed1999-03-10 22:55:24 +0000815#if NSMALLNEGINTS + NSMALLPOSINTS > 0
816 PyIntObject **q;
817
818 i = NSMALLNEGINTS + NSMALLPOSINTS;
819 q = small_ints;
820 while (--i >= 0) {
821 Py_XDECREF(*q);
822 *q++ = NULL;
823 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000824#endif
Guido van Rossumda084ed1999-03-10 22:55:24 +0000825 bc = 0;
826 bf = 0;
827 isum = 0;
828 list = block_list;
829 block_list = NULL;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000830 free_list = NULL;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000831 while (list != NULL) {
Guido van Rossumda084ed1999-03-10 22:55:24 +0000832 bc++;
833 irem = 0;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000834 for (i = 0, p = &list->objects[0];
835 i < N_INTOBJECTS;
836 i++, p++) {
Guido van Rossumda084ed1999-03-10 22:55:24 +0000837 if (PyInt_Check(p) && p->ob_refcnt != 0)
838 irem++;
839 }
Guido van Rossum3fce8831999-03-12 19:43:17 +0000840 next = list->next;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000841 if (irem) {
Guido van Rossum3fce8831999-03-12 19:43:17 +0000842 list->next = block_list;
843 block_list = list;
Guido van Rossum51288bc1999-03-19 20:30:39 +0000844 for (i = 0, p = &list->objects[0];
845 i < N_INTOBJECTS;
846 i++, p++) {
847 if (!PyInt_Check(p) || p->ob_refcnt == 0) {
848 p->ob_type = (struct _typeobject *)
849 free_list;
850 free_list = p;
851 }
852#if NSMALLNEGINTS + NSMALLPOSINTS > 0
853 else if (-NSMALLNEGINTS <= p->ob_ival &&
854 p->ob_ival < NSMALLPOSINTS &&
855 small_ints[p->ob_ival +
856 NSMALLNEGINTS] == NULL) {
857 Py_INCREF(p);
858 small_ints[p->ob_ival +
859 NSMALLNEGINTS] = p;
860 }
861#endif
862 }
Guido van Rossumda084ed1999-03-10 22:55:24 +0000863 }
864 else {
Guido van Rossumb18618d2000-05-03 23:44:39 +0000865 PyMem_FREE(list); /* XXX PyObject_FREE ??? */
Guido van Rossumda084ed1999-03-10 22:55:24 +0000866 bf++;
867 }
868 isum += irem;
Guido van Rossum3fce8831999-03-12 19:43:17 +0000869 list = next;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000870 }
Guido van Rossum3fce8831999-03-12 19:43:17 +0000871 if (!Py_VerboseFlag)
872 return;
873 fprintf(stderr, "# cleanup ints");
874 if (!isum) {
875 fprintf(stderr, "\n");
876 }
877 else {
878 fprintf(stderr,
879 ": %d unfreed int%s in %d out of %d block%s\n",
880 isum, isum == 1 ? "" : "s",
881 bc - bf, bc, bc == 1 ? "" : "s");
882 }
883 if (Py_VerboseFlag > 1) {
884 list = block_list;
885 while (list != NULL) {
Guido van Rossum51288bc1999-03-19 20:30:39 +0000886 for (i = 0, p = &list->objects[0];
887 i < N_INTOBJECTS;
888 i++, p++) {
Guido van Rossum3fce8831999-03-12 19:43:17 +0000889 if (PyInt_Check(p) && p->ob_refcnt != 0)
890 fprintf(stderr,
Fred Drakea44d3532000-06-30 15:01:00 +0000891 "# <int at %p, refcnt=%d, val=%ld>\n",
892 p, p->ob_refcnt, p->ob_ival);
Guido van Rossum3fce8831999-03-12 19:43:17 +0000893 }
894 list = list->next;
Guido van Rossumda084ed1999-03-10 22:55:24 +0000895 }
896 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000897}