blob: 242abf2afd024fa6bee910cd597e82d0be6910fa [file] [log] [blame]
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001#include "Python.h"
Raymond Hettinger691d8052004-05-30 07:26:47 +00002#include "structmember.h"
Raymond Hettinger756b3f32004-01-29 06:37:52 +00003
4/* collections module implementation of a deque() datatype
5 Written and maintained by Raymond D. Hettinger <python@rcn.com>
6 Copyright (c) 2004 Python Software Foundation.
7 All rights reserved.
8*/
9
Raymond Hettinger77e8bf12004-10-01 15:25:53 +000010/* The block length may be set to any number over 1. Larger numbers
11 * reduce the number of calls to the memory allocator but take more
12 * memory. Ideally, BLOCKLEN should be set with an eye to the
Tim Peters5566e962006-07-28 00:23:15 +000013 * length of a cache line.
Raymond Hettinger77e8bf12004-10-01 15:25:53 +000014 */
15
Raymond Hettinger7d112df2004-11-02 02:11:35 +000016#define BLOCKLEN 62
Raymond Hettinger61f05fb2004-10-01 06:24:12 +000017#define CENTER ((BLOCKLEN - 1) / 2)
Raymond Hettinger756b3f32004-01-29 06:37:52 +000018
Tim Petersd8768d32004-10-01 01:32:53 +000019/* A `dequeobject` is composed of a doubly-linked list of `block` nodes.
20 * This list is not circular (the leftmost block has leftlink==NULL,
21 * and the rightmost block has rightlink==NULL). A deque d's first
22 * element is at d.leftblock[leftindex] and its last element is at
23 * d.rightblock[rightindex]; note that, unlike as for Python slice
Raymond Hettinger61f05fb2004-10-01 06:24:12 +000024 * indices, these indices are inclusive on both ends. By being inclusive
Tim Peters5566e962006-07-28 00:23:15 +000025 * on both ends, algorithms for left and right operations become
Raymond Hettinger61f05fb2004-10-01 06:24:12 +000026 * symmetrical which simplifies the design.
Tim Peters5566e962006-07-28 00:23:15 +000027 *
Raymond Hettinger61f05fb2004-10-01 06:24:12 +000028 * The list of blocks is never empty, so d.leftblock and d.rightblock
29 * are never equal to NULL.
30 *
31 * The indices, d.leftindex and d.rightindex are always in the range
32 * 0 <= index < BLOCKLEN.
Raymond Hettinger4ca4c7c2004-10-01 15:14:39 +000033 * Their exact relationship is:
34 * (d.leftindex + d.len - 1) % BLOCKLEN == d.rightindex.
Raymond Hettinger61f05fb2004-10-01 06:24:12 +000035 *
36 * Empty deques have d.len == 0; d.leftblock==d.rightblock;
37 * d.leftindex == CENTER+1; and d.rightindex == CENTER.
38 * Checking for d.len == 0 is the intended way to see whether d is empty.
39 *
Tim Peters5566e962006-07-28 00:23:15 +000040 * Whenever d.leftblock == d.rightblock,
Raymond Hettinger4ca4c7c2004-10-01 15:14:39 +000041 * d.leftindex + d.len - 1 == d.rightindex.
Tim Peters5566e962006-07-28 00:23:15 +000042 *
Raymond Hettinger4ca4c7c2004-10-01 15:14:39 +000043 * However, when d.leftblock != d.rightblock, d.leftindex and d.rightindex
Tim Peters5566e962006-07-28 00:23:15 +000044 * become indices into distinct blocks and either may be larger than the
Raymond Hettinger4ca4c7c2004-10-01 15:14:39 +000045 * other.
Tim Petersd8768d32004-10-01 01:32:53 +000046 */
47
Raymond Hettinger756b3f32004-01-29 06:37:52 +000048typedef struct BLOCK {
Antoine Pitrouc83ea132010-05-09 14:46:46 +000049 struct BLOCK *leftlink;
50 struct BLOCK *rightlink;
51 PyObject *data[BLOCKLEN];
Raymond Hettinger756b3f32004-01-29 06:37:52 +000052} block;
53
Raymond Hettingerd3ffd342007-11-10 01:54:03 +000054#define MAXFREEBLOCKS 10
Raymond Hettinger33fcf9d2008-07-24 00:08:18 +000055static Py_ssize_t numfreeblocks = 0;
Raymond Hettingerd3ffd342007-11-10 01:54:03 +000056static block *freeblocks[MAXFREEBLOCKS];
57
Tim Peters6f853562004-10-01 01:04:50 +000058static block *
Raymond Hettinger33fcf9d2008-07-24 00:08:18 +000059newblock(block *leftlink, block *rightlink, Py_ssize_t len) {
Antoine Pitrouc83ea132010-05-09 14:46:46 +000060 block *b;
61 /* To prevent len from overflowing PY_SSIZE_T_MAX on 64-bit machines, we
62 * refuse to allocate new blocks if the current len is dangerously
63 * close. There is some extra margin to prevent spurious arithmetic
64 * overflows at various places. The following check ensures that
65 * the blocks allocated to the deque, in the worst case, can only
66 * have PY_SSIZE_T_MAX-2 entries in total.
67 */
68 if (len >= PY_SSIZE_T_MAX - 2*BLOCKLEN) {
69 PyErr_SetString(PyExc_OverflowError,
70 "cannot add more blocks to the deque");
71 return NULL;
72 }
73 if (numfreeblocks) {
74 numfreeblocks -= 1;
75 b = freeblocks[numfreeblocks];
76 } else {
77 b = PyMem_Malloc(sizeof(block));
78 if (b == NULL) {
79 PyErr_NoMemory();
80 return NULL;
81 }
82 }
83 b->leftlink = leftlink;
84 b->rightlink = rightlink;
85 return b;
Raymond Hettinger756b3f32004-01-29 06:37:52 +000086}
87
Martin v. Löwis111c1802008-06-13 07:47:47 +000088static void
Raymond Hettingerd3ffd342007-11-10 01:54:03 +000089freeblock(block *b)
90{
Antoine Pitrouc83ea132010-05-09 14:46:46 +000091 if (numfreeblocks < MAXFREEBLOCKS) {
92 freeblocks[numfreeblocks] = b;
93 numfreeblocks++;
94 } else {
95 PyMem_Free(b);
96 }
Raymond Hettingerd3ffd342007-11-10 01:54:03 +000097}
98
Raymond Hettinger756b3f32004-01-29 06:37:52 +000099typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000100 PyObject_HEAD
101 block *leftblock;
102 block *rightblock;
103 Py_ssize_t leftindex; /* in range(BLOCKLEN) */
104 Py_ssize_t rightindex; /* in range(BLOCKLEN) */
105 Py_ssize_t len;
106 Py_ssize_t maxlen;
107 long state; /* incremented whenever the indices move */
108 PyObject *weakreflist; /* List of weak references */
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000109} dequeobject;
110
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000111/* The deque's size limit is d.maxlen. The limit can be zero or positive.
112 * If there is no limit, then d.maxlen == -1.
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000113 *
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000114 * After an item is added to a deque, we check to see if the size has grown past
115 * the limit. If it has, we get the size back down to the limit by popping an
116 * item off of the opposite end. The methods that can trigger this are append(),
117 * appendleft(), extend(), and extendleft().
118 */
119
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000120#define TRIM(d, popfunction) \
121 if (d->maxlen != -1 && d->len > d->maxlen) { \
122 PyObject *rv = popfunction(d, NULL); \
123 assert(rv != NULL && d->len <= d->maxlen); \
124 Py_DECREF(rv); \
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000125 }
126
Neal Norwitz87f10132004-02-29 15:40:53 +0000127static PyTypeObject deque_type;
Raymond Hettinger738ec902004-02-29 02:15:56 +0000128
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000129static PyObject *
130deque_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
131{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000132 dequeobject *deque;
133 block *b;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000134
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000135 /* create dequeobject structure */
136 deque = (dequeobject *)type->tp_alloc(type, 0);
137 if (deque == NULL)
138 return NULL;
Tim Peters1065f752004-10-01 01:03:29 +0000139
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000140 b = newblock(NULL, NULL, 0);
141 if (b == NULL) {
142 Py_DECREF(deque);
143 return NULL;
144 }
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000145
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000146 assert(BLOCKLEN >= 2);
147 deque->leftblock = b;
148 deque->rightblock = b;
149 deque->leftindex = CENTER + 1;
150 deque->rightindex = CENTER;
151 deque->len = 0;
152 deque->state = 0;
153 deque->weakreflist = NULL;
154 deque->maxlen = -1;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000155
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000156 return (PyObject *)deque;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000157}
158
159static PyObject *
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000160deque_pop(dequeobject *deque, PyObject *unused)
161{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000162 PyObject *item;
163 block *prevblock;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000164
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000165 if (deque->len == 0) {
166 PyErr_SetString(PyExc_IndexError, "pop from an empty deque");
167 return NULL;
168 }
169 item = deque->rightblock->data[deque->rightindex];
170 deque->rightindex--;
171 deque->len--;
172 deque->state++;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000173
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000174 if (deque->rightindex == -1) {
175 if (deque->len == 0) {
176 assert(deque->leftblock == deque->rightblock);
177 assert(deque->leftindex == deque->rightindex+1);
178 /* re-center instead of freeing a block */
179 deque->leftindex = CENTER + 1;
180 deque->rightindex = CENTER;
181 } else {
182 prevblock = deque->rightblock->leftlink;
183 assert(deque->leftblock != deque->rightblock);
184 freeblock(deque->rightblock);
185 prevblock->rightlink = NULL;
186 deque->rightblock = prevblock;
187 deque->rightindex = BLOCKLEN - 1;
188 }
189 }
190 return item;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000191}
192
193PyDoc_STRVAR(pop_doc, "Remove and return the rightmost element.");
194
195static PyObject *
196deque_popleft(dequeobject *deque, PyObject *unused)
197{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000198 PyObject *item;
199 block *prevblock;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000200
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000201 if (deque->len == 0) {
202 PyErr_SetString(PyExc_IndexError, "pop from an empty deque");
203 return NULL;
204 }
205 assert(deque->leftblock != NULL);
206 item = deque->leftblock->data[deque->leftindex];
207 deque->leftindex++;
208 deque->len--;
209 deque->state++;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000210
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000211 if (deque->leftindex == BLOCKLEN) {
212 if (deque->len == 0) {
213 assert(deque->leftblock == deque->rightblock);
214 assert(deque->leftindex == deque->rightindex+1);
215 /* re-center instead of freeing a block */
216 deque->leftindex = CENTER + 1;
217 deque->rightindex = CENTER;
218 } else {
219 assert(deque->leftblock != deque->rightblock);
220 prevblock = deque->leftblock->rightlink;
221 freeblock(deque->leftblock);
222 assert(prevblock != NULL);
223 prevblock->leftlink = NULL;
224 deque->leftblock = prevblock;
225 deque->leftindex = 0;
226 }
227 }
228 return item;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000229}
230
231PyDoc_STRVAR(popleft_doc, "Remove and return the leftmost element.");
232
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000233static PyObject *
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000234deque_append(dequeobject *deque, PyObject *item)
235{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000236 deque->state++;
237 if (deque->rightindex == BLOCKLEN-1) {
238 block *b = newblock(deque->rightblock, NULL, deque->len);
239 if (b == NULL)
240 return NULL;
241 assert(deque->rightblock->rightlink == NULL);
242 deque->rightblock->rightlink = b;
243 deque->rightblock = b;
244 deque->rightindex = -1;
245 }
246 Py_INCREF(item);
247 deque->len++;
248 deque->rightindex++;
249 deque->rightblock->data[deque->rightindex] = item;
250 TRIM(deque, deque_popleft);
251 Py_RETURN_NONE;
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000252}
253
254PyDoc_STRVAR(append_doc, "Add an element to the right side of the deque.");
255
256static PyObject *
257deque_appendleft(dequeobject *deque, PyObject *item)
258{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000259 deque->state++;
260 if (deque->leftindex == 0) {
261 block *b = newblock(NULL, deque->leftblock, deque->len);
262 if (b == NULL)
263 return NULL;
264 assert(deque->leftblock->leftlink == NULL);
265 deque->leftblock->leftlink = b;
266 deque->leftblock = b;
267 deque->leftindex = BLOCKLEN;
268 }
269 Py_INCREF(item);
270 deque->len++;
271 deque->leftindex--;
272 deque->leftblock->data[deque->leftindex] = item;
273 TRIM(deque, deque_pop);
274 Py_RETURN_NONE;
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000275}
276
277PyDoc_STRVAR(appendleft_doc, "Add an element to the left side of the deque.");
278
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000279
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000280/* Run an iterator to exhaustion. Shortcut for
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000281 the extend/extendleft methods when maxlen == 0. */
282static PyObject*
283consume_iterator(PyObject *it)
284{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000285 PyObject *item;
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000286
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000287 while ((item = PyIter_Next(it)) != NULL) {
288 Py_DECREF(item);
289 }
290 Py_DECREF(it);
291 if (PyErr_Occurred())
292 return NULL;
293 Py_RETURN_NONE;
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000294}
295
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000296static PyObject *
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000297deque_extend(dequeobject *deque, PyObject *iterable)
298{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000299 PyObject *it, *item;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000300
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000301 /* Handle case where id(deque) == id(iterable) */
302 if ((PyObject *)deque == iterable) {
303 PyObject *result;
304 PyObject *s = PySequence_List(iterable);
305 if (s == NULL)
306 return NULL;
307 result = deque_extend(deque, s);
308 Py_DECREF(s);
309 return result;
310 }
Raymond Hettinger0b3263b2009-12-10 06:00:33 +0000311
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000312 it = PyObject_GetIter(iterable);
313 if (it == NULL)
314 return NULL;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000315
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000316 if (deque->maxlen == 0)
317 return consume_iterator(it);
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000318
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000319 while ((item = PyIter_Next(it)) != NULL) {
320 deque->state++;
321 if (deque->rightindex == BLOCKLEN-1) {
322 block *b = newblock(deque->rightblock, NULL,
323 deque->len);
324 if (b == NULL) {
325 Py_DECREF(item);
326 Py_DECREF(it);
327 return NULL;
328 }
329 assert(deque->rightblock->rightlink == NULL);
330 deque->rightblock->rightlink = b;
331 deque->rightblock = b;
332 deque->rightindex = -1;
333 }
334 deque->len++;
335 deque->rightindex++;
336 deque->rightblock->data[deque->rightindex] = item;
337 TRIM(deque, deque_popleft);
338 }
339 Py_DECREF(it);
340 if (PyErr_Occurred())
341 return NULL;
342 Py_RETURN_NONE;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000343}
344
Tim Peters1065f752004-10-01 01:03:29 +0000345PyDoc_STRVAR(extend_doc,
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000346"Extend the right side of the deque with elements from the iterable");
347
348static PyObject *
349deque_extendleft(dequeobject *deque, PyObject *iterable)
350{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000351 PyObject *it, *item;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000352
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000353 /* Handle case where id(deque) == id(iterable) */
354 if ((PyObject *)deque == iterable) {
355 PyObject *result;
356 PyObject *s = PySequence_List(iterable);
357 if (s == NULL)
358 return NULL;
359 result = deque_extendleft(deque, s);
360 Py_DECREF(s);
361 return result;
362 }
Raymond Hettinger0b3263b2009-12-10 06:00:33 +0000363
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000364 it = PyObject_GetIter(iterable);
365 if (it == NULL)
366 return NULL;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000367
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000368 if (deque->maxlen == 0)
369 return consume_iterator(it);
Raymond Hettingerbac769b2009-03-10 09:31:48 +0000370
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000371 while ((item = PyIter_Next(it)) != NULL) {
372 deque->state++;
373 if (deque->leftindex == 0) {
374 block *b = newblock(NULL, deque->leftblock,
375 deque->len);
376 if (b == NULL) {
377 Py_DECREF(item);
378 Py_DECREF(it);
379 return NULL;
380 }
381 assert(deque->leftblock->leftlink == NULL);
382 deque->leftblock->leftlink = b;
383 deque->leftblock = b;
384 deque->leftindex = BLOCKLEN;
385 }
386 deque->len++;
387 deque->leftindex--;
388 deque->leftblock->data[deque->leftindex] = item;
389 TRIM(deque, deque_pop);
390 }
391 Py_DECREF(it);
392 if (PyErr_Occurred())
393 return NULL;
394 Py_RETURN_NONE;
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000395}
396
Tim Peters1065f752004-10-01 01:03:29 +0000397PyDoc_STRVAR(extendleft_doc,
Raymond Hettinger3ba85c22004-02-06 19:04:56 +0000398"Extend the left side of the deque with elements from the iterable");
399
Raymond Hettinger0b3263b2009-12-10 06:00:33 +0000400static PyObject *
401deque_inplace_concat(dequeobject *deque, PyObject *other)
402{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000403 PyObject *result;
Raymond Hettinger0b3263b2009-12-10 06:00:33 +0000404
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000405 result = deque_extend(deque, other);
406 if (result == NULL)
407 return result;
408 Py_DECREF(result);
409 Py_INCREF(deque);
410 return (PyObject *)deque;
Raymond Hettinger0b3263b2009-12-10 06:00:33 +0000411}
412
Raymond Hettingerdcb9d942004-10-09 16:02:18 +0000413static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000414_deque_rotate(dequeobject *deque, Py_ssize_t n)
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000415{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000416 Py_ssize_t i, len=deque->len, halflen=(len+1)>>1;
417 PyObject *item, *rv;
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000418
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000419 if (len == 0)
420 return 0;
421 if (n > halflen || n < -halflen) {
422 n %= len;
423 if (n > halflen)
424 n -= len;
425 else if (n < -halflen)
426 n += len;
427 }
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000428
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000429 for (i=0 ; i<n ; i++) {
430 item = deque_pop(deque, NULL);
431 assert (item != NULL);
432 rv = deque_appendleft(deque, item);
433 Py_DECREF(item);
434 if (rv == NULL)
435 return -1;
436 Py_DECREF(rv);
437 }
438 for (i=0 ; i>n ; i--) {
439 item = deque_popleft(deque, NULL);
440 assert (item != NULL);
441 rv = deque_append(deque, item);
442 Py_DECREF(item);
443 if (rv == NULL)
444 return -1;
445 Py_DECREF(rv);
446 }
447 return 0;
Raymond Hettingerdcb9d942004-10-09 16:02:18 +0000448}
449
450static PyObject *
451deque_rotate(dequeobject *deque, PyObject *args)
452{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000453 Py_ssize_t n=1;
Raymond Hettingerdcb9d942004-10-09 16:02:18 +0000454
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000455 if (!PyArg_ParseTuple(args, "|n:rotate", &n))
456 return NULL;
457 if (_deque_rotate(deque, n) == 0)
458 Py_RETURN_NONE;
459 return NULL;
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000460}
461
Tim Peters1065f752004-10-01 01:03:29 +0000462PyDoc_STRVAR(rotate_doc,
Raymond Hettingeree33b272004-02-08 04:05:26 +0000463"Rotate the deque n steps to the right (default n=1). If n is negative, rotates left.");
Raymond Hettinger5c5eb862004-02-07 21:13:00 +0000464
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000465static PyObject *
466deque_reverse(dequeobject *deque, PyObject *unused)
467{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000468 block *leftblock = deque->leftblock;
469 block *rightblock = deque->rightblock;
470 Py_ssize_t leftindex = deque->leftindex;
471 Py_ssize_t rightindex = deque->rightindex;
472 Py_ssize_t n = (deque->len)/2;
473 Py_ssize_t i;
474 PyObject *tmp;
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000475
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000476 for (i=0 ; i<n ; i++) {
477 /* Validate that pointers haven't met in the middle */
478 assert(leftblock != rightblock || leftindex < rightindex);
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000479
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000480 /* Swap */
481 tmp = leftblock->data[leftindex];
482 leftblock->data[leftindex] = rightblock->data[rightindex];
483 rightblock->data[rightindex] = tmp;
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000484
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000485 /* Advance left block/index pair */
486 leftindex++;
487 if (leftindex == BLOCKLEN) {
Raymond Hettinger57a86892011-01-25 21:43:29 +0000488 if (leftblock->rightlink == NULL)
489 break;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000490 leftblock = leftblock->rightlink;
491 leftindex = 0;
492 }
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000493
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000494 /* Step backwards with the right block/index pair */
495 rightindex--;
496 if (rightindex == -1) {
Raymond Hettinger57a86892011-01-25 21:43:29 +0000497 if (rightblock->leftlink == NULL)
498 break;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000499 rightblock = rightblock->leftlink;
500 rightindex = BLOCKLEN - 1;
501 }
502 }
503 Py_RETURN_NONE;
Raymond Hettingera5fd24e2009-12-10 06:42:54 +0000504}
505
506PyDoc_STRVAR(reverse_doc,
507"D.reverse() -- reverse *IN PLACE*");
508
Raymond Hettinger5f516ed2010-04-03 18:10:37 +0000509static PyObject *
510deque_count(dequeobject *deque, PyObject *v)
511{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000512 block *leftblock = deque->leftblock;
513 Py_ssize_t leftindex = deque->leftindex;
Raymond Hettinger57a86892011-01-25 21:43:29 +0000514 Py_ssize_t n = deque->len;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000515 Py_ssize_t i;
516 Py_ssize_t count = 0;
517 PyObject *item;
518 long start_state = deque->state;
519 int cmp;
Raymond Hettinger5f516ed2010-04-03 18:10:37 +0000520
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000521 for (i=0 ; i<n ; i++) {
522 item = leftblock->data[leftindex];
523 cmp = PyObject_RichCompareBool(item, v, Py_EQ);
524 if (cmp > 0)
525 count++;
526 else if (cmp < 0)
527 return NULL;
Raymond Hettinger5f516ed2010-04-03 18:10:37 +0000528
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000529 if (start_state != deque->state) {
530 PyErr_SetString(PyExc_RuntimeError,
531 "deque mutated during iteration");
532 return NULL;
533 }
Raymond Hettinger5f516ed2010-04-03 18:10:37 +0000534
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000535 /* Advance left block/index pair */
536 leftindex++;
537 if (leftindex == BLOCKLEN) {
Raymond Hettinger57a86892011-01-25 21:43:29 +0000538 if (leftblock->rightlink == NULL) /* can occur when i==n-1 */
539 break;
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000540 leftblock = leftblock->rightlink;
541 leftindex = 0;
542 }
543 }
544 return PyInt_FromSsize_t(count);
Raymond Hettinger5f516ed2010-04-03 18:10:37 +0000545}
546
547PyDoc_STRVAR(count_doc,
548"D.count(value) -> integer -- return number of occurrences of value");
549
Martin v. Löwis18e16552006-02-15 17:27:45 +0000550static Py_ssize_t
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000551deque_len(dequeobject *deque)
552{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000553 return deque->len;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000554}
555
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000556static PyObject *
557deque_remove(dequeobject *deque, PyObject *value)
558{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000559 Py_ssize_t i, n=deque->len;
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000560
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000561 for (i=0 ; i<n ; i++) {
562 PyObject *item = deque->leftblock->data[deque->leftindex];
563 int cmp = PyObject_RichCompareBool(item, value, Py_EQ);
Raymond Hettingerd73202c2005-03-19 00:00:51 +0000564
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000565 if (deque->len != n) {
566 PyErr_SetString(PyExc_IndexError,
567 "deque mutated during remove().");
568 return NULL;
569 }
570 if (cmp > 0) {
571 PyObject *tgt = deque_popleft(deque, NULL);
572 assert (tgt != NULL);
573 Py_DECREF(tgt);
574 if (_deque_rotate(deque, i) == -1)
575 return NULL;
576 Py_RETURN_NONE;
577 }
578 else if (cmp < 0) {
579 _deque_rotate(deque, i);
580 return NULL;
581 }
582 _deque_rotate(deque, -1);
583 }
584 PyErr_SetString(PyExc_ValueError, "deque.remove(x): x not in deque");
585 return NULL;
Raymond Hettinger4aec61e2005-03-18 21:20:23 +0000586}
587
588PyDoc_STRVAR(remove_doc,
589"D.remove(value) -- remove first occurrence of value.");
590
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000591static int
592deque_clear(dequeobject *deque)
593{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000594 PyObject *item;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000595
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000596 while (deque->len) {
597 item = deque_pop(deque, NULL);
598 assert (item != NULL);
599 Py_DECREF(item);
600 }
601 assert(deque->leftblock == deque->rightblock &&
602 deque->leftindex - 1 == deque->rightindex &&
603 deque->len == 0);
604 return 0;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000605}
606
607static PyObject *
Raymond Hettinger33fcf9d2008-07-24 00:08:18 +0000608deque_item(dequeobject *deque, Py_ssize_t i)
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000609{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000610 block *b;
611 PyObject *item;
612 Py_ssize_t n, index=i;
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000613
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000614 if (i < 0 || i >= deque->len) {
615 PyErr_SetString(PyExc_IndexError,
616 "deque index out of range");
617 return NULL;
618 }
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000619
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000620 if (i == 0) {
621 i = deque->leftindex;
622 b = deque->leftblock;
623 } else if (i == deque->len - 1) {
624 i = deque->rightindex;
625 b = deque->rightblock;
626 } else {
627 i += deque->leftindex;
628 n = i / BLOCKLEN;
629 i %= BLOCKLEN;
630 if (index < (deque->len >> 1)) {
631 b = deque->leftblock;
632 while (n--)
633 b = b->rightlink;
634 } else {
635 n = (deque->leftindex + deque->len - 1) / BLOCKLEN - n;
636 b = deque->rightblock;
637 while (n--)
638 b = b->leftlink;
639 }
640 }
641 item = b->data[i];
642 Py_INCREF(item);
643 return item;
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000644}
645
Raymond Hettinger616f4f62004-06-26 04:42:06 +0000646/* delitem() implemented in terms of rotate for simplicity and reasonable
647 performance near the end points. If for some reason this method becomes
Tim Peters1065f752004-10-01 01:03:29 +0000648 popular, it is not hard to re-implement this using direct data movement
Raymond Hettinger616f4f62004-06-26 04:42:06 +0000649 (similar to code in list slice assignment) and achieve a two or threefold
650 performance boost.
651*/
652
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000653static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000654deque_del_item(dequeobject *deque, Py_ssize_t i)
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000655{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000656 PyObject *item;
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000657
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000658 assert (i >= 0 && i < deque->len);
659 if (_deque_rotate(deque, -i) == -1)
660 return -1;
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000661
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000662 item = deque_popleft(deque, NULL);
663 assert (item != NULL);
664 Py_DECREF(item);
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000665
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000666 return _deque_rotate(deque, i);
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000667}
668
669static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000670deque_ass_item(dequeobject *deque, Py_ssize_t i, PyObject *v)
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000671{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000672 PyObject *old_value;
673 block *b;
674 Py_ssize_t n, len=deque->len, halflen=(len+1)>>1, index=i;
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000675
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000676 if (i < 0 || i >= len) {
677 PyErr_SetString(PyExc_IndexError,
678 "deque index out of range");
679 return -1;
680 }
681 if (v == NULL)
682 return deque_del_item(deque, i);
Raymond Hettinger0e371f22004-05-12 20:55:56 +0000683
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000684 i += deque->leftindex;
685 n = i / BLOCKLEN;
686 i %= BLOCKLEN;
687 if (index <= halflen) {
688 b = deque->leftblock;
689 while (n--)
690 b = b->rightlink;
691 } else {
692 n = (deque->leftindex + len - 1) / BLOCKLEN - n;
693 b = deque->rightblock;
694 while (n--)
695 b = b->leftlink;
696 }
697 Py_INCREF(v);
698 old_value = b->data[i];
699 b->data[i] = v;
700 Py_DECREF(old_value);
701 return 0;
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000702}
703
704static PyObject *
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000705deque_clearmethod(dequeobject *deque)
706{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000707 int rv;
Raymond Hettingera435c532004-07-09 04:10:20 +0000708
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000709 rv = deque_clear(deque);
710 assert (rv != -1);
711 Py_RETURN_NONE;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000712}
713
714PyDoc_STRVAR(clear_doc, "Remove all elements from the deque.");
715
716static void
717deque_dealloc(dequeobject *deque)
718{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000719 PyObject_GC_UnTrack(deque);
720 if (deque->weakreflist != NULL)
721 PyObject_ClearWeakRefs((PyObject *) deque);
722 if (deque->leftblock != NULL) {
723 deque_clear(deque);
724 assert(deque->leftblock != NULL);
725 freeblock(deque->leftblock);
726 }
727 deque->leftblock = NULL;
728 deque->rightblock = NULL;
729 Py_TYPE(deque)->tp_free(deque);
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000730}
731
732static int
Raymond Hettinger0a4977c2004-03-01 23:16:22 +0000733deque_traverse(dequeobject *deque, visitproc visit, void *arg)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000734{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000735 block *b;
736 PyObject *item;
737 Py_ssize_t index;
738 Py_ssize_t indexlo = deque->leftindex;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000739
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000740 for (b = deque->leftblock; b != NULL; b = b->rightlink) {
741 const Py_ssize_t indexhi = b == deque->rightblock ?
742 deque->rightindex :
743 BLOCKLEN - 1;
Tim Peters10c7e862004-10-01 02:01:04 +0000744
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000745 for (index = indexlo; index <= indexhi; ++index) {
746 item = b->data[index];
747 Py_VISIT(item);
748 }
749 indexlo = 0;
750 }
751 return 0;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000752}
753
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000754static PyObject *
755deque_copy(PyObject *deque)
756{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000757 if (((dequeobject *)deque)->maxlen == -1)
758 return PyObject_CallFunction((PyObject *)(Py_TYPE(deque)), "O", deque, NULL);
759 else
760 return PyObject_CallFunction((PyObject *)(Py_TYPE(deque)), "Oi",
761 deque, ((dequeobject *)deque)->maxlen, NULL);
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000762}
763
764PyDoc_STRVAR(copy_doc, "Return a shallow copy of a deque.");
765
766static PyObject *
767deque_reduce(dequeobject *deque)
768{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000769 PyObject *dict, *result, *aslist;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000770
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000771 dict = PyObject_GetAttrString((PyObject *)deque, "__dict__");
772 if (dict == NULL)
773 PyErr_Clear();
774 aslist = PySequence_List((PyObject *)deque);
775 if (aslist == NULL) {
776 Py_XDECREF(dict);
777 return NULL;
778 }
779 if (dict == NULL) {
780 if (deque->maxlen == -1)
781 result = Py_BuildValue("O(O)", Py_TYPE(deque), aslist);
782 else
783 result = Py_BuildValue("O(On)", Py_TYPE(deque), aslist, deque->maxlen);
784 } else {
785 if (deque->maxlen == -1)
786 result = Py_BuildValue("O(OO)O", Py_TYPE(deque), aslist, Py_None, dict);
787 else
788 result = Py_BuildValue("O(On)O", Py_TYPE(deque), aslist, deque->maxlen, dict);
789 }
790 Py_XDECREF(dict);
791 Py_DECREF(aslist);
792 return result;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000793}
794
795PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
796
797static PyObject *
798deque_repr(PyObject *deque)
799{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000800 PyObject *aslist, *result, *fmt;
801 int i;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000802
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000803 i = Py_ReprEnter(deque);
804 if (i != 0) {
805 if (i < 0)
806 return NULL;
807 return PyString_FromString("[...]");
808 }
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000809
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000810 aslist = PySequence_List(deque);
811 if (aslist == NULL) {
812 Py_ReprLeave(deque);
813 return NULL;
814 }
815 if (((dequeobject *)deque)->maxlen != -1)
816 fmt = PyString_FromFormat("deque(%%r, maxlen=%zd)",
817 ((dequeobject *)deque)->maxlen);
818 else
819 fmt = PyString_FromString("deque(%r)");
820 if (fmt == NULL) {
821 Py_DECREF(aslist);
822 Py_ReprLeave(deque);
823 return NULL;
824 }
825 result = PyString_Format(fmt, aslist);
826 Py_DECREF(fmt);
827 Py_DECREF(aslist);
828 Py_ReprLeave(deque);
829 return result;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000830}
831
832static int
833deque_tp_print(PyObject *deque, FILE *fp, int flags)
834{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000835 PyObject *it, *item;
836 char *emit = ""; /* No separator emitted on first pass */
837 char *separator = ", ";
838 int i;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000839
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000840 i = Py_ReprEnter(deque);
841 if (i != 0) {
842 if (i < 0)
843 return i;
844 Py_BEGIN_ALLOW_THREADS
845 fputs("[...]", fp);
846 Py_END_ALLOW_THREADS
847 return 0;
848 }
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000849
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000850 it = PyObject_GetIter(deque);
851 if (it == NULL)
852 return -1;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000853
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000854 Py_BEGIN_ALLOW_THREADS
855 fputs("deque([", fp);
856 Py_END_ALLOW_THREADS
857 while ((item = PyIter_Next(it)) != NULL) {
858 Py_BEGIN_ALLOW_THREADS
859 fputs(emit, fp);
860 Py_END_ALLOW_THREADS
861 emit = separator;
862 if (PyObject_Print(item, fp, 0) != 0) {
863 Py_DECREF(item);
864 Py_DECREF(it);
865 Py_ReprLeave(deque);
866 return -1;
867 }
868 Py_DECREF(item);
869 }
870 Py_ReprLeave(deque);
871 Py_DECREF(it);
872 if (PyErr_Occurred())
873 return -1;
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000874
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000875 Py_BEGIN_ALLOW_THREADS
876 if (((dequeobject *)deque)->maxlen == -1)
877 fputs("])", fp);
878 else
879 fprintf(fp, "], maxlen=%" PY_FORMAT_SIZE_T "d)", ((dequeobject *)deque)->maxlen);
880 Py_END_ALLOW_THREADS
881 return 0;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000882}
883
Raymond Hettinger738ec902004-02-29 02:15:56 +0000884static PyObject *
885deque_richcompare(PyObject *v, PyObject *w, int op)
886{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000887 PyObject *it1=NULL, *it2=NULL, *x, *y;
888 Py_ssize_t vs, ws;
889 int b, cmp=-1;
Raymond Hettinger738ec902004-02-29 02:15:56 +0000890
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000891 if (!PyObject_TypeCheck(v, &deque_type) ||
892 !PyObject_TypeCheck(w, &deque_type)) {
893 Py_INCREF(Py_NotImplemented);
894 return Py_NotImplemented;
895 }
Raymond Hettinger738ec902004-02-29 02:15:56 +0000896
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000897 /* Shortcuts */
898 vs = ((dequeobject *)v)->len;
899 ws = ((dequeobject *)w)->len;
900 if (op == Py_EQ) {
901 if (v == w)
902 Py_RETURN_TRUE;
903 if (vs != ws)
904 Py_RETURN_FALSE;
905 }
906 if (op == Py_NE) {
907 if (v == w)
908 Py_RETURN_FALSE;
909 if (vs != ws)
910 Py_RETURN_TRUE;
911 }
Raymond Hettinger738ec902004-02-29 02:15:56 +0000912
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000913 /* Search for the first index where items are different */
914 it1 = PyObject_GetIter(v);
915 if (it1 == NULL)
916 goto done;
917 it2 = PyObject_GetIter(w);
918 if (it2 == NULL)
919 goto done;
920 for (;;) {
921 x = PyIter_Next(it1);
922 if (x == NULL && PyErr_Occurred())
923 goto done;
924 y = PyIter_Next(it2);
925 if (x == NULL || y == NULL)
926 break;
927 b = PyObject_RichCompareBool(x, y, Py_EQ);
928 if (b == 0) {
929 cmp = PyObject_RichCompareBool(x, y, op);
930 Py_DECREF(x);
931 Py_DECREF(y);
932 goto done;
933 }
934 Py_DECREF(x);
935 Py_DECREF(y);
936 if (b == -1)
937 goto done;
938 }
939 /* We reached the end of one deque or both */
940 Py_XDECREF(x);
941 Py_XDECREF(y);
942 if (PyErr_Occurred())
943 goto done;
944 switch (op) {
945 case Py_LT: cmp = y != NULL; break; /* if w was longer */
946 case Py_LE: cmp = x == NULL; break; /* if v was not longer */
947 case Py_EQ: cmp = x == y; break; /* if we reached the end of both */
948 case Py_NE: cmp = x != y; break; /* if one deque continues */
949 case Py_GT: cmp = x != NULL; break; /* if v was longer */
950 case Py_GE: cmp = y == NULL; break; /* if w was not longer */
951 }
Tim Peters1065f752004-10-01 01:03:29 +0000952
Raymond Hettinger738ec902004-02-29 02:15:56 +0000953done:
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000954 Py_XDECREF(it1);
955 Py_XDECREF(it2);
956 if (cmp == 1)
957 Py_RETURN_TRUE;
958 if (cmp == 0)
959 Py_RETURN_FALSE;
960 return NULL;
Raymond Hettinger738ec902004-02-29 02:15:56 +0000961}
962
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000963static int
Raymond Hettingera7fc4b12007-10-05 02:47:07 +0000964deque_init(dequeobject *deque, PyObject *args, PyObject *kwdargs)
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000965{
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000966 PyObject *iterable = NULL;
967 PyObject *maxlenobj = NULL;
968 Py_ssize_t maxlen = -1;
969 char *kwlist[] = {"iterable", "maxlen", 0};
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000970
Antoine Pitrouc83ea132010-05-09 14:46:46 +0000971 if (!PyArg_ParseTupleAndKeywords(args, kwdargs, "|OO:deque", kwlist, &iterable, &maxlenobj))
972 return -1;
973 if (maxlenobj != NULL && maxlenobj != Py_None) {
974 maxlen = PyInt_AsSsize_t(maxlenobj);
975 if (maxlen == -1 && PyErr_Occurred())
976 return -1;
977 if (maxlen < 0) {
978 PyErr_SetString(PyExc_ValueError, "maxlen must be non-negative");
979 return -1;
980 }
981 }
982 deque->maxlen = maxlen;
983 deque_clear(deque);
984 if (iterable != NULL) {
985 PyObject *rv = deque_extend(deque, iterable);
986 if (rv == NULL)
987 return -1;
988 Py_DECREF(rv);
989 }
990 return 0;
Raymond Hettinger756b3f32004-01-29 06:37:52 +0000991}
992
Raymond Hettinger56411aa2009-03-10 12:50:59 +0000993static PyObject *
Jesus Cead4e58dc2012-08-03 14:48:23 +0200994deque_sizeof(dequeobject *deque, void *unused)
995{
996 Py_ssize_t res;
997 Py_ssize_t blocks;
998
999 res = sizeof(dequeobject);
1000 blocks = (deque->leftindex + deque->len + BLOCKLEN - 1) / BLOCKLEN;
1001 assert(deque->leftindex + deque->len - 1 ==
1002 (blocks - 1) * BLOCKLEN + deque->rightindex);
1003 res += blocks * sizeof(block);
1004 return PyLong_FromSsize_t(res);
1005}
1006
1007PyDoc_STRVAR(sizeof_doc,
1008"D.__sizeof__() -- size of D in memory, in bytes");
1009
1010static PyObject *
Raymond Hettinger56411aa2009-03-10 12:50:59 +00001011deque_get_maxlen(dequeobject *deque)
1012{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001013 if (deque->maxlen == -1)
1014 Py_RETURN_NONE;
1015 return PyInt_FromSsize_t(deque->maxlen);
Raymond Hettinger56411aa2009-03-10 12:50:59 +00001016}
1017
1018static PyGetSetDef deque_getset[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001019 {"maxlen", (getter)deque_get_maxlen, (setter)NULL,
1020 "maximum size of a deque or None if unbounded"},
1021 {0}
Raymond Hettinger56411aa2009-03-10 12:50:59 +00001022};
1023
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001024static PySequenceMethods deque_as_sequence = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001025 (lenfunc)deque_len, /* sq_length */
1026 0, /* sq_concat */
1027 0, /* sq_repeat */
1028 (ssizeargfunc)deque_item, /* sq_item */
1029 0, /* sq_slice */
1030 (ssizeobjargproc)deque_ass_item, /* sq_ass_item */
1031 0, /* sq_ass_slice */
1032 0, /* sq_contains */
1033 (binaryfunc)deque_inplace_concat, /* sq_inplace_concat */
1034 0, /* sq_inplace_repeat */
Raymond Hettinger0b3263b2009-12-10 06:00:33 +00001035
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001036};
1037
1038/* deque object ********************************************************/
1039
1040static PyObject *deque_iter(dequeobject *deque);
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001041static PyObject *deque_reviter(dequeobject *deque);
Tim Peters1065f752004-10-01 01:03:29 +00001042PyDoc_STRVAR(reversed_doc,
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001043 "D.__reversed__() -- return a reverse iterator over the deque");
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001044
1045static PyMethodDef deque_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001046 {"append", (PyCFunction)deque_append,
1047 METH_O, append_doc},
1048 {"appendleft", (PyCFunction)deque_appendleft,
1049 METH_O, appendleft_doc},
1050 {"clear", (PyCFunction)deque_clearmethod,
1051 METH_NOARGS, clear_doc},
1052 {"__copy__", (PyCFunction)deque_copy,
1053 METH_NOARGS, copy_doc},
1054 {"count", (PyCFunction)deque_count,
1055 METH_O, count_doc},
1056 {"extend", (PyCFunction)deque_extend,
1057 METH_O, extend_doc},
1058 {"extendleft", (PyCFunction)deque_extendleft,
1059 METH_O, extendleft_doc},
1060 {"pop", (PyCFunction)deque_pop,
1061 METH_NOARGS, pop_doc},
1062 {"popleft", (PyCFunction)deque_popleft,
1063 METH_NOARGS, popleft_doc},
1064 {"__reduce__", (PyCFunction)deque_reduce,
1065 METH_NOARGS, reduce_doc},
1066 {"remove", (PyCFunction)deque_remove,
1067 METH_O, remove_doc},
1068 {"__reversed__", (PyCFunction)deque_reviter,
1069 METH_NOARGS, reversed_doc},
1070 {"reverse", (PyCFunction)deque_reverse,
1071 METH_NOARGS, reverse_doc},
1072 {"rotate", (PyCFunction)deque_rotate,
Jesus Cead4e58dc2012-08-03 14:48:23 +02001073 METH_VARARGS, rotate_doc},
1074 {"__sizeof__", (PyCFunction)deque_sizeof,
1075 METH_NOARGS, sizeof_doc},
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001076 {NULL, NULL} /* sentinel */
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001077};
1078
1079PyDoc_STRVAR(deque_doc,
Andrew Svetlov227f59b2012-10-31 11:50:00 +02001080"deque([iterable[, maxlen]]) --> deque object\n\
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001081\n\
Raymond Hettingerdb9d64b2011-03-29 17:28:25 -07001082Build an ordered collection with optimized access from its endpoints.");
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001083
Neal Norwitz87f10132004-02-29 15:40:53 +00001084static PyTypeObject deque_type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001085 PyVarObject_HEAD_INIT(NULL, 0)
1086 "collections.deque", /* tp_name */
1087 sizeof(dequeobject), /* tp_basicsize */
1088 0, /* tp_itemsize */
1089 /* methods */
1090 (destructor)deque_dealloc, /* tp_dealloc */
1091 deque_tp_print, /* tp_print */
1092 0, /* tp_getattr */
1093 0, /* tp_setattr */
1094 0, /* tp_compare */
1095 deque_repr, /* tp_repr */
1096 0, /* tp_as_number */
1097 &deque_as_sequence, /* tp_as_sequence */
1098 0, /* tp_as_mapping */
1099 (hashfunc)PyObject_HashNotImplemented, /* tp_hash */
1100 0, /* tp_call */
1101 0, /* tp_str */
1102 PyObject_GenericGetAttr, /* tp_getattro */
1103 0, /* tp_setattro */
1104 0, /* tp_as_buffer */
1105 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC |
1106 Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */
1107 deque_doc, /* tp_doc */
1108 (traverseproc)deque_traverse, /* tp_traverse */
1109 (inquiry)deque_clear, /* tp_clear */
1110 (richcmpfunc)deque_richcompare, /* tp_richcompare */
1111 offsetof(dequeobject, weakreflist), /* tp_weaklistoffset*/
1112 (getiterfunc)deque_iter, /* tp_iter */
1113 0, /* tp_iternext */
1114 deque_methods, /* tp_methods */
1115 0, /* tp_members */
1116 deque_getset, /* tp_getset */
1117 0, /* tp_base */
1118 0, /* tp_dict */
1119 0, /* tp_descr_get */
1120 0, /* tp_descr_set */
1121 0, /* tp_dictoffset */
1122 (initproc)deque_init, /* tp_init */
1123 PyType_GenericAlloc, /* tp_alloc */
1124 deque_new, /* tp_new */
1125 PyObject_GC_Del, /* tp_free */
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001126};
1127
1128/*********************** Deque Iterator **************************/
1129
1130typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001131 PyObject_HEAD
1132 Py_ssize_t index;
1133 block *b;
1134 dequeobject *deque;
1135 long state; /* state when the iterator is created */
1136 Py_ssize_t counter; /* number of items remaining for iteration */
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001137} dequeiterobject;
1138
Martin v. Löwis111c1802008-06-13 07:47:47 +00001139static PyTypeObject dequeiter_type;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001140
1141static PyObject *
1142deque_iter(dequeobject *deque)
1143{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001144 dequeiterobject *it;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001145
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001146 it = PyObject_GC_New(dequeiterobject, &dequeiter_type);
1147 if (it == NULL)
1148 return NULL;
1149 it->b = deque->leftblock;
1150 it->index = deque->leftindex;
1151 Py_INCREF(deque);
1152 it->deque = deque;
1153 it->state = deque->state;
1154 it->counter = deque->len;
1155 PyObject_GC_Track(it);
1156 return (PyObject *)it;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001157}
1158
Antoine Pitrouaa687902009-01-01 14:11:22 +00001159static int
1160dequeiter_traverse(dequeiterobject *dio, visitproc visit, void *arg)
1161{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001162 Py_VISIT(dio->deque);
1163 return 0;
Antoine Pitrouaa687902009-01-01 14:11:22 +00001164}
1165
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001166static void
1167dequeiter_dealloc(dequeiterobject *dio)
1168{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001169 Py_XDECREF(dio->deque);
1170 PyObject_GC_Del(dio);
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001171}
1172
1173static PyObject *
1174dequeiter_next(dequeiterobject *it)
1175{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001176 PyObject *item;
Raymond Hettingerd1b3d882004-10-02 00:43:13 +00001177
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001178 if (it->deque->state != it->state) {
1179 it->counter = 0;
1180 PyErr_SetString(PyExc_RuntimeError,
1181 "deque mutated during iteration");
1182 return NULL;
1183 }
1184 if (it->counter == 0)
1185 return NULL;
1186 assert (!(it->b == it->deque->rightblock &&
1187 it->index > it->deque->rightindex));
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001188
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001189 item = it->b->data[it->index];
1190 it->index++;
1191 it->counter--;
1192 if (it->index == BLOCKLEN && it->counter > 0) {
1193 assert (it->b->rightlink != NULL);
1194 it->b = it->b->rightlink;
1195 it->index = 0;
1196 }
1197 Py_INCREF(item);
1198 return item;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001199}
1200
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00001201static PyObject *
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001202dequeiter_len(dequeiterobject *it)
1203{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001204 return PyInt_FromLong(it->counter);
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001205}
1206
Armin Rigof5b3e362006-02-11 21:32:43 +00001207PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00001208
1209static PyMethodDef dequeiter_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001210 {"__length_hint__", (PyCFunction)dequeiter_len, METH_NOARGS, length_hint_doc},
1211 {NULL, NULL} /* sentinel */
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001212};
1213
Martin v. Löwis111c1802008-06-13 07:47:47 +00001214static PyTypeObject dequeiter_type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001215 PyVarObject_HEAD_INIT(NULL, 0)
1216 "deque_iterator", /* tp_name */
1217 sizeof(dequeiterobject), /* tp_basicsize */
1218 0, /* tp_itemsize */
1219 /* methods */
1220 (destructor)dequeiter_dealloc, /* tp_dealloc */
1221 0, /* tp_print */
1222 0, /* tp_getattr */
1223 0, /* tp_setattr */
1224 0, /* tp_compare */
1225 0, /* tp_repr */
1226 0, /* tp_as_number */
1227 0, /* tp_as_sequence */
1228 0, /* tp_as_mapping */
1229 0, /* tp_hash */
1230 0, /* tp_call */
1231 0, /* tp_str */
1232 PyObject_GenericGetAttr, /* tp_getattro */
1233 0, /* tp_setattro */
1234 0, /* tp_as_buffer */
1235 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
1236 0, /* tp_doc */
1237 (traverseproc)dequeiter_traverse, /* tp_traverse */
1238 0, /* tp_clear */
1239 0, /* tp_richcompare */
1240 0, /* tp_weaklistoffset */
1241 PyObject_SelfIter, /* tp_iter */
1242 (iternextfunc)dequeiter_next, /* tp_iternext */
1243 dequeiter_methods, /* tp_methods */
1244 0,
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001245};
1246
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001247/*********************** Deque Reverse Iterator **************************/
1248
Martin v. Löwis111c1802008-06-13 07:47:47 +00001249static PyTypeObject dequereviter_type;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001250
1251static PyObject *
1252deque_reviter(dequeobject *deque)
1253{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001254 dequeiterobject *it;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001255
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001256 it = PyObject_GC_New(dequeiterobject, &dequereviter_type);
1257 if (it == NULL)
1258 return NULL;
1259 it->b = deque->rightblock;
1260 it->index = deque->rightindex;
1261 Py_INCREF(deque);
1262 it->deque = deque;
1263 it->state = deque->state;
1264 it->counter = deque->len;
1265 PyObject_GC_Track(it);
1266 return (PyObject *)it;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001267}
1268
1269static PyObject *
1270dequereviter_next(dequeiterobject *it)
1271{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001272 PyObject *item;
1273 if (it->counter == 0)
1274 return NULL;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001275
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001276 if (it->deque->state != it->state) {
1277 it->counter = 0;
1278 PyErr_SetString(PyExc_RuntimeError,
1279 "deque mutated during iteration");
1280 return NULL;
1281 }
1282 assert (!(it->b == it->deque->leftblock &&
1283 it->index < it->deque->leftindex));
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001284
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001285 item = it->b->data[it->index];
1286 it->index--;
1287 it->counter--;
1288 if (it->index == -1 && it->counter > 0) {
1289 assert (it->b->leftlink != NULL);
1290 it->b = it->b->leftlink;
1291 it->index = BLOCKLEN - 1;
1292 }
1293 Py_INCREF(item);
1294 return item;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001295}
1296
Martin v. Löwis111c1802008-06-13 07:47:47 +00001297static PyTypeObject dequereviter_type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001298 PyVarObject_HEAD_INIT(NULL, 0)
1299 "deque_reverse_iterator", /* tp_name */
1300 sizeof(dequeiterobject), /* tp_basicsize */
1301 0, /* tp_itemsize */
1302 /* methods */
1303 (destructor)dequeiter_dealloc, /* tp_dealloc */
1304 0, /* tp_print */
1305 0, /* tp_getattr */
1306 0, /* tp_setattr */
1307 0, /* tp_compare */
1308 0, /* tp_repr */
1309 0, /* tp_as_number */
1310 0, /* tp_as_sequence */
1311 0, /* tp_as_mapping */
1312 0, /* tp_hash */
1313 0, /* tp_call */
1314 0, /* tp_str */
1315 PyObject_GenericGetAttr, /* tp_getattro */
1316 0, /* tp_setattro */
1317 0, /* tp_as_buffer */
1318 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
1319 0, /* tp_doc */
1320 (traverseproc)dequeiter_traverse, /* tp_traverse */
1321 0, /* tp_clear */
1322 0, /* tp_richcompare */
1323 0, /* tp_weaklistoffset */
1324 PyObject_SelfIter, /* tp_iter */
1325 (iternextfunc)dequereviter_next, /* tp_iternext */
1326 dequeiter_methods, /* tp_methods */
1327 0,
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001328};
1329
Guido van Rossum1968ad32006-02-25 22:38:04 +00001330/* defaultdict type *********************************************************/
1331
1332typedef struct {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001333 PyDictObject dict;
1334 PyObject *default_factory;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001335} defdictobject;
1336
1337static PyTypeObject defdict_type; /* Forward */
1338
1339PyDoc_STRVAR(defdict_missing_doc,
1340"__missing__(key) # Called by __getitem__ for missing key; pseudo-code:\n\
Georg Brandlb51a57e2007-03-06 13:32:52 +00001341 if self.default_factory is None: raise KeyError((key,))\n\
Guido van Rossum1968ad32006-02-25 22:38:04 +00001342 self[key] = value = self.default_factory()\n\
1343 return value\n\
1344");
1345
1346static PyObject *
1347defdict_missing(defdictobject *dd, PyObject *key)
1348{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001349 PyObject *factory = dd->default_factory;
1350 PyObject *value;
1351 if (factory == NULL || factory == Py_None) {
1352 /* XXX Call dict.__missing__(key) */
1353 PyObject *tup;
1354 tup = PyTuple_Pack(1, key);
1355 if (!tup) return NULL;
1356 PyErr_SetObject(PyExc_KeyError, tup);
1357 Py_DECREF(tup);
1358 return NULL;
1359 }
1360 value = PyEval_CallObject(factory, NULL);
1361 if (value == NULL)
1362 return value;
1363 if (PyObject_SetItem((PyObject *)dd, key, value) < 0) {
1364 Py_DECREF(value);
1365 return NULL;
1366 }
1367 return value;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001368}
1369
1370PyDoc_STRVAR(defdict_copy_doc, "D.copy() -> a shallow copy of D.");
1371
1372static PyObject *
1373defdict_copy(defdictobject *dd)
1374{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001375 /* This calls the object's class. That only works for subclasses
1376 whose class constructor has the same signature. Subclasses that
1377 define a different constructor signature must override copy().
1378 */
Raymond Hettinger8fdab952009-08-04 19:08:05 +00001379
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001380 if (dd->default_factory == NULL)
1381 return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), Py_None, dd, NULL);
1382 return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd),
1383 dd->default_factory, dd, NULL);
Guido van Rossum1968ad32006-02-25 22:38:04 +00001384}
1385
1386static PyObject *
1387defdict_reduce(defdictobject *dd)
1388{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001389 /* __reduce__ must return a 5-tuple as follows:
Guido van Rossum1968ad32006-02-25 22:38:04 +00001390
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001391 - factory function
1392 - tuple of args for the factory function
1393 - additional state (here None)
1394 - sequence iterator (here None)
1395 - dictionary iterator (yielding successive (key, value) pairs
Guido van Rossum1968ad32006-02-25 22:38:04 +00001396
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001397 This API is used by pickle.py and copy.py.
Guido van Rossum1968ad32006-02-25 22:38:04 +00001398
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001399 For this to be useful with pickle.py, the default_factory
1400 must be picklable; e.g., None, a built-in, or a global
1401 function in a module or package.
Guido van Rossum1968ad32006-02-25 22:38:04 +00001402
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001403 Both shallow and deep copying are supported, but for deep
1404 copying, the default_factory must be deep-copyable; e.g. None,
1405 or a built-in (functions are not copyable at this time).
Guido van Rossum1968ad32006-02-25 22:38:04 +00001406
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001407 This only works for subclasses as long as their constructor
1408 signature is compatible; the first argument must be the
1409 optional default_factory, defaulting to None.
1410 */
1411 PyObject *args;
1412 PyObject *items;
1413 PyObject *result;
1414 if (dd->default_factory == NULL || dd->default_factory == Py_None)
1415 args = PyTuple_New(0);
1416 else
1417 args = PyTuple_Pack(1, dd->default_factory);
1418 if (args == NULL)
1419 return NULL;
1420 items = PyObject_CallMethod((PyObject *)dd, "iteritems", "()");
1421 if (items == NULL) {
1422 Py_DECREF(args);
1423 return NULL;
1424 }
1425 result = PyTuple_Pack(5, Py_TYPE(dd), args,
1426 Py_None, Py_None, items);
1427 Py_DECREF(items);
1428 Py_DECREF(args);
1429 return result;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001430}
1431
1432static PyMethodDef defdict_methods[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001433 {"__missing__", (PyCFunction)defdict_missing, METH_O,
1434 defdict_missing_doc},
1435 {"copy", (PyCFunction)defdict_copy, METH_NOARGS,
1436 defdict_copy_doc},
1437 {"__copy__", (PyCFunction)defdict_copy, METH_NOARGS,
1438 defdict_copy_doc},
1439 {"__reduce__", (PyCFunction)defdict_reduce, METH_NOARGS,
1440 reduce_doc},
1441 {NULL}
Guido van Rossum1968ad32006-02-25 22:38:04 +00001442};
1443
1444static PyMemberDef defdict_members[] = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001445 {"default_factory", T_OBJECT,
1446 offsetof(defdictobject, default_factory), 0,
1447 PyDoc_STR("Factory for default value called by __missing__().")},
1448 {NULL}
Guido van Rossum1968ad32006-02-25 22:38:04 +00001449};
1450
1451static void
1452defdict_dealloc(defdictobject *dd)
1453{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001454 Py_CLEAR(dd->default_factory);
1455 PyDict_Type.tp_dealloc((PyObject *)dd);
Guido van Rossum1968ad32006-02-25 22:38:04 +00001456}
1457
1458static int
1459defdict_print(defdictobject *dd, FILE *fp, int flags)
1460{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001461 int sts;
1462 Py_BEGIN_ALLOW_THREADS
1463 fprintf(fp, "defaultdict(");
1464 Py_END_ALLOW_THREADS
1465 if (dd->default_factory == NULL) {
1466 Py_BEGIN_ALLOW_THREADS
1467 fprintf(fp, "None");
1468 Py_END_ALLOW_THREADS
1469 } else {
1470 PyObject_Print(dd->default_factory, fp, 0);
1471 }
1472 Py_BEGIN_ALLOW_THREADS
1473 fprintf(fp, ", ");
1474 Py_END_ALLOW_THREADS
1475 sts = PyDict_Type.tp_print((PyObject *)dd, fp, 0);
1476 Py_BEGIN_ALLOW_THREADS
1477 fprintf(fp, ")");
1478 Py_END_ALLOW_THREADS
1479 return sts;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001480}
1481
1482static PyObject *
1483defdict_repr(defdictobject *dd)
1484{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001485 PyObject *defrepr;
1486 PyObject *baserepr;
1487 PyObject *result;
1488 baserepr = PyDict_Type.tp_repr((PyObject *)dd);
1489 if (baserepr == NULL)
1490 return NULL;
1491 if (dd->default_factory == NULL)
1492 defrepr = PyString_FromString("None");
1493 else
1494 {
1495 int status = Py_ReprEnter(dd->default_factory);
1496 if (status != 0) {
Antoine Pitrouc39cd782012-02-15 02:42:46 +01001497 if (status < 0) {
1498 Py_DECREF(baserepr);
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001499 return NULL;
Antoine Pitrouc39cd782012-02-15 02:42:46 +01001500 }
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001501 defrepr = PyString_FromString("...");
1502 }
1503 else
1504 defrepr = PyObject_Repr(dd->default_factory);
1505 Py_ReprLeave(dd->default_factory);
1506 }
1507 if (defrepr == NULL) {
1508 Py_DECREF(baserepr);
1509 return NULL;
1510 }
1511 result = PyString_FromFormat("defaultdict(%s, %s)",
1512 PyString_AS_STRING(defrepr),
1513 PyString_AS_STRING(baserepr));
1514 Py_DECREF(defrepr);
1515 Py_DECREF(baserepr);
1516 return result;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001517}
1518
1519static int
1520defdict_traverse(PyObject *self, visitproc visit, void *arg)
1521{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001522 Py_VISIT(((defdictobject *)self)->default_factory);
1523 return PyDict_Type.tp_traverse(self, visit, arg);
Guido van Rossum1968ad32006-02-25 22:38:04 +00001524}
1525
1526static int
1527defdict_tp_clear(defdictobject *dd)
1528{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001529 Py_CLEAR(dd->default_factory);
1530 return PyDict_Type.tp_clear((PyObject *)dd);
Guido van Rossum1968ad32006-02-25 22:38:04 +00001531}
1532
1533static int
1534defdict_init(PyObject *self, PyObject *args, PyObject *kwds)
1535{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001536 defdictobject *dd = (defdictobject *)self;
1537 PyObject *olddefault = dd->default_factory;
1538 PyObject *newdefault = NULL;
1539 PyObject *newargs;
1540 int result;
1541 if (args == NULL || !PyTuple_Check(args))
1542 newargs = PyTuple_New(0);
1543 else {
1544 Py_ssize_t n = PyTuple_GET_SIZE(args);
1545 if (n > 0) {
1546 newdefault = PyTuple_GET_ITEM(args, 0);
1547 if (!PyCallable_Check(newdefault) && newdefault != Py_None) {
1548 PyErr_SetString(PyExc_TypeError,
1549 "first argument must be callable");
1550 return -1;
1551 }
1552 }
1553 newargs = PySequence_GetSlice(args, 1, n);
1554 }
1555 if (newargs == NULL)
1556 return -1;
1557 Py_XINCREF(newdefault);
1558 dd->default_factory = newdefault;
1559 result = PyDict_Type.tp_init(self, newargs, kwds);
1560 Py_DECREF(newargs);
1561 Py_XDECREF(olddefault);
1562 return result;
Guido van Rossum1968ad32006-02-25 22:38:04 +00001563}
1564
1565PyDoc_STRVAR(defdict_doc,
1566"defaultdict(default_factory) --> dict with default factory\n\
1567\n\
1568The default factory is called without arguments to produce\n\
1569a new value when a key is not present, in __getitem__ only.\n\
1570A defaultdict compares equal to a dict with the same items.\n\
1571");
1572
Anthony Baxter3b8ff312006-04-04 15:05:23 +00001573/* See comment in xxsubtype.c */
1574#define DEFERRED_ADDRESS(ADDR) 0
1575
Guido van Rossum1968ad32006-02-25 22:38:04 +00001576static PyTypeObject defdict_type = {
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001577 PyVarObject_HEAD_INIT(DEFERRED_ADDRESS(&PyType_Type), 0)
1578 "collections.defaultdict", /* tp_name */
1579 sizeof(defdictobject), /* tp_basicsize */
1580 0, /* tp_itemsize */
1581 /* methods */
1582 (destructor)defdict_dealloc, /* tp_dealloc */
1583 (printfunc)defdict_print, /* tp_print */
1584 0, /* tp_getattr */
1585 0, /* tp_setattr */
1586 0, /* tp_compare */
1587 (reprfunc)defdict_repr, /* tp_repr */
1588 0, /* tp_as_number */
1589 0, /* tp_as_sequence */
1590 0, /* tp_as_mapping */
1591 0, /* tp_hash */
1592 0, /* tp_call */
1593 0, /* tp_str */
1594 PyObject_GenericGetAttr, /* tp_getattro */
1595 0, /* tp_setattro */
1596 0, /* tp_as_buffer */
1597 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC |
1598 Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */
1599 defdict_doc, /* tp_doc */
1600 defdict_traverse, /* tp_traverse */
1601 (inquiry)defdict_tp_clear, /* tp_clear */
1602 0, /* tp_richcompare */
1603 0, /* tp_weaklistoffset*/
1604 0, /* tp_iter */
1605 0, /* tp_iternext */
1606 defdict_methods, /* tp_methods */
1607 defdict_members, /* tp_members */
1608 0, /* tp_getset */
1609 DEFERRED_ADDRESS(&PyDict_Type), /* tp_base */
1610 0, /* tp_dict */
1611 0, /* tp_descr_get */
1612 0, /* tp_descr_set */
1613 0, /* tp_dictoffset */
1614 defdict_init, /* tp_init */
1615 PyType_GenericAlloc, /* tp_alloc */
1616 0, /* tp_new */
1617 PyObject_GC_Del, /* tp_free */
Guido van Rossum1968ad32006-02-25 22:38:04 +00001618};
1619
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001620/* module level code ********************************************************/
1621
1622PyDoc_STRVAR(module_doc,
Guido van Rossum1968ad32006-02-25 22:38:04 +00001623"High performance data structures.\n\
1624- deque: ordered collection accessible from endpoints only\n\
1625- defaultdict: dict subclass with a default value factory\n\
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001626");
1627
1628PyMODINIT_FUNC
Raymond Hettingereb979882007-02-28 18:37:52 +00001629init_collections(void)
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001630{
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001631 PyObject *m;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001632
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001633 m = Py_InitModule3("_collections", NULL, module_doc);
1634 if (m == NULL)
1635 return;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001636
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001637 if (PyType_Ready(&deque_type) < 0)
1638 return;
1639 Py_INCREF(&deque_type);
1640 PyModule_AddObject(m, "deque", (PyObject *)&deque_type);
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001641
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001642 defdict_type.tp_base = &PyDict_Type;
1643 if (PyType_Ready(&defdict_type) < 0)
1644 return;
1645 Py_INCREF(&defdict_type);
1646 PyModule_AddObject(m, "defaultdict", (PyObject *)&defdict_type);
Guido van Rossum1968ad32006-02-25 22:38:04 +00001647
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001648 if (PyType_Ready(&dequeiter_type) < 0)
1649 return;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001650
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001651 if (PyType_Ready(&dequereviter_type) < 0)
1652 return;
Raymond Hettinger1e5809f2004-03-18 11:04:57 +00001653
Antoine Pitrouc83ea132010-05-09 14:46:46 +00001654 return;
Raymond Hettinger756b3f32004-01-29 06:37:52 +00001655}