| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1 | #include "Python.h" | 
| Raymond Hettinger | 691d805 | 2004-05-30 07:26:47 +0000 | [diff] [blame] | 2 | #include "structmember.h" | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 3 |  | 
| Raymond Hettinger | c208308 | 2015-02-28 23:29:16 -0800 | [diff] [blame] | 4 | #ifdef STDC_HEADERS | 
 | 5 | #include <stddef.h> | 
 | 6 | #else | 
 | 7 | #include <sys/types.h>          /* For size_t */ | 
 | 8 | #endif | 
 | 9 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 10 | /* collections module implementation of a deque() datatype | 
 | 11 |    Written and maintained by Raymond D. Hettinger <python@rcn.com> | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 12 | */ | 
 | 13 |  | 
| Raymond Hettinger | 77e8bf1 | 2004-10-01 15:25:53 +0000 | [diff] [blame] | 14 | /* The block length may be set to any number over 1.  Larger numbers | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 15 |  * reduce the number of calls to the memory allocator, give faster | 
| Raymond Hettinger | 30c9074 | 2015-03-02 22:31:35 -0800 | [diff] [blame] | 16 |  * indexing and rotation, and reduce the link to data overhead ratio. | 
| Raymond Hettinger | daf57f2 | 2015-02-26 23:21:29 -0800 | [diff] [blame] | 17 |  * Making the block length a power of two speeds-up the modulo | 
| Raymond Hettinger | 30c9074 | 2015-03-02 22:31:35 -0800 | [diff] [blame] | 18 |  * and division calculations in deque_item() and deque_ass_item(). | 
| Raymond Hettinger | 77e8bf1 | 2004-10-01 15:25:53 +0000 | [diff] [blame] | 19 |  */ | 
 | 20 |  | 
| Raymond Hettinger | daf57f2 | 2015-02-26 23:21:29 -0800 | [diff] [blame] | 21 | #define BLOCKLEN 64 | 
| Raymond Hettinger | 61f05fb | 2004-10-01 06:24:12 +0000 | [diff] [blame] | 22 | #define CENTER ((BLOCKLEN - 1) / 2) | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 23 |  | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 24 | /* Data for deque objects is stored in a doubly-linked list of fixed | 
 | 25 |  * length blocks.  This assures that appends or pops never move any | 
 | 26 |  * other data elements besides the one being appended or popped. | 
 | 27 |  * | 
 | 28 |  * Another advantage is that it completely avoids use of realloc(), | 
 | 29 |  * resulting in more predictable performance. | 
 | 30 |  * | 
 | 31 |  * Textbook implementations of doubly-linked lists store one datum | 
 | 32 |  * per link, but that gives them a 200% memory overhead (a prev and | 
 | 33 |  * next link for each datum) and it costs one malloc() call per data | 
 | 34 |  * element.  By using fixed-length blocks, the link to data ratio is | 
 | 35 |  * significantly improved and there are proportionally fewer calls | 
 | 36 |  * to malloc() and free().  The data blocks of consecutive pointers | 
 | 37 |  * also improve cache locality. | 
 | 38 |  * | 
| Raymond Hettinger | 61f05fb | 2004-10-01 06:24:12 +0000 | [diff] [blame] | 39 |  * The list of blocks is never empty, so d.leftblock and d.rightblock | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 40 |  * are never equal to NULL.  The list is not circular. | 
 | 41 |  * | 
 | 42 |  * A deque d's first element is at d.leftblock[leftindex] | 
 | 43 |  * and its last element is at d.rightblock[rightindex]. | 
| Raymond Hettinger | 61f05fb | 2004-10-01 06:24:12 +0000 | [diff] [blame] | 44 |  * | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 45 |  * Unlike Python slice indices, these indices are inclusive on both | 
 | 46 |  * ends.  This makes the algorithms for left and right operations | 
 | 47 |  * more symmetrical and it simplifies the design. | 
| Raymond Hettinger | 61f05fb | 2004-10-01 06:24:12 +0000 | [diff] [blame] | 48 |  * | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 49 |  * The indices, d.leftindex and d.rightindex are always in the range: | 
 | 50 |  *     0 <= index < BLOCKLEN | 
 | 51 |  * | 
 | 52 |  * And their exact relationship is: | 
 | 53 |  *     (d.leftindex + d.len - 1) % BLOCKLEN == d.rightindex | 
| Raymond Hettinger | 61f05fb | 2004-10-01 06:24:12 +0000 | [diff] [blame] | 54 |  * | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 55 |  * Whenever d.leftblock == d.rightblock, then: | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 56 |  *     d.leftindex + d.len - 1 == d.rightindex | 
| Thomas Wouters | 0e3f591 | 2006-08-11 14:57:12 +0000 | [diff] [blame] | 57 |  * | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 58 |  * However, when d.leftblock != d.rightblock, the d.leftindex and | 
 | 59 |  * d.rightindex become indices into distinct blocks and either may | 
 | 60 |  * be larger than the other. | 
 | 61 |  * | 
 | 62 |  * Empty deques have: | 
 | 63 |  *     d.len == 0 | 
 | 64 |  *     d.leftblock == d.rightblock | 
 | 65 |  *     d.leftindex == CENTER + 1 | 
 | 66 |  *     d.rightindex == CENTER | 
 | 67 |  * | 
 | 68 |  * Checking for d.len == 0 is the intended way to see whether d is empty. | 
| Tim Peters | d8768d3 | 2004-10-01 01:32:53 +0000 | [diff] [blame] | 69 |  */ | 
 | 70 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 71 | typedef struct BLOCK { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 72 |     struct BLOCK *leftlink; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 73 |     PyObject *data[BLOCKLEN]; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 74 |     struct BLOCK *rightlink; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 75 | } block; | 
 | 76 |  | 
| Raymond Hettinger | 30c9074 | 2015-03-02 22:31:35 -0800 | [diff] [blame] | 77 | typedef struct { | 
 | 78 |     PyObject_VAR_HEAD | 
 | 79 |     block *leftblock; | 
 | 80 |     block *rightblock; | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 81 |     Py_ssize_t leftindex;       /* 0 <= leftindex < BLOCKLEN */ | 
 | 82 |     Py_ssize_t rightindex;      /* 0 <= rightindex < BLOCKLEN */ | 
| Raymond Hettinger | f9d9c79 | 2015-03-02 22:47:46 -0800 | [diff] [blame] | 83 |     size_t state;               /* incremented whenever the indices move */ | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 84 |     Py_ssize_t maxlen;          /* maxlen is -1 for unbounded deques */ | 
| Raymond Hettinger | 0f6f947 | 2015-03-21 01:42:10 -0700 | [diff] [blame] | 85 |     PyObject *weakreflist; | 
| Raymond Hettinger | 30c9074 | 2015-03-02 22:31:35 -0800 | [diff] [blame] | 86 | } dequeobject; | 
 | 87 |  | 
 | 88 | static PyTypeObject deque_type; | 
 | 89 |  | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 90 | /* For debug builds, add error checking to track the endpoints | 
 | 91 |  * in the chain of links.  The goal is to make sure that link | 
 | 92 |  * assignments only take place at endpoints so that links already | 
 | 93 |  * in use do not get overwritten. | 
 | 94 |  * | 
 | 95 |  * CHECK_END should happen before each assignment to a block's link field. | 
 | 96 |  * MARK_END should happen whenever a link field becomes a new endpoint. | 
 | 97 |  * This happens when new blocks are added or whenever an existing | 
 | 98 |  * block is freed leaving another existing block as the new endpoint. | 
 | 99 |  */ | 
 | 100 |  | 
| Raymond Hettinger | 3223dd5 | 2013-07-26 23:14:22 -0700 | [diff] [blame] | 101 | #ifndef NDEBUG | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 102 | #define MARK_END(link)  link = NULL; | 
 | 103 | #define CHECK_END(link) assert(link == NULL); | 
 | 104 | #define CHECK_NOT_END(link) assert(link != NULL); | 
 | 105 | #else | 
 | 106 | #define MARK_END(link) | 
 | 107 | #define CHECK_END(link) | 
 | 108 | #define CHECK_NOT_END(link) | 
 | 109 | #endif | 
 | 110 |  | 
 | 111 | /* A simple freelisting scheme is used to minimize calls to the memory | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 112 |    allocator.  It accommodates common use cases where new blocks are being | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 113 |    added at about the same rate as old blocks are being freed. | 
 | 114 |  */ | 
 | 115 |  | 
| Raymond Hettinger | f2b02ce | 2015-09-26 17:47:02 -0700 | [diff] [blame] | 116 | #define MAXFREEBLOCKS 16 | 
| Benjamin Peterson | d631371 | 2008-07-31 16:23:04 +0000 | [diff] [blame] | 117 | static Py_ssize_t numfreeblocks = 0; | 
| Guido van Rossum | 58da931 | 2007-11-10 23:39:45 +0000 | [diff] [blame] | 118 | static block *freeblocks[MAXFREEBLOCKS]; | 
 | 119 |  | 
| Tim Peters | 6f85356 | 2004-10-01 01:04:50 +0000 | [diff] [blame] | 120 | static block * | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 121 | newblock(void) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 122 |     block *b; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 123 |     if (numfreeblocks) { | 
| Raymond Hettinger | 59cf23a | 2013-02-07 00:57:19 -0500 | [diff] [blame] | 124 |         numfreeblocks--; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 125 |         return freeblocks[numfreeblocks]; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 126 |     } | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 127 |     b = PyMem_Malloc(sizeof(block)); | 
 | 128 |     if (b != NULL) { | 
 | 129 |         return b; | 
 | 130 |     } | 
 | 131 |     PyErr_NoMemory(); | 
 | 132 |     return NULL; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 133 | } | 
 | 134 |  | 
| Martin v. Löwis | 59683e8 | 2008-06-13 07:50:45 +0000 | [diff] [blame] | 135 | static void | 
| Guido van Rossum | 58da931 | 2007-11-10 23:39:45 +0000 | [diff] [blame] | 136 | freeblock(block *b) | 
 | 137 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 138 |     if (numfreeblocks < MAXFREEBLOCKS) { | 
 | 139 |         freeblocks[numfreeblocks] = b; | 
 | 140 |         numfreeblocks++; | 
 | 141 |     } else { | 
 | 142 |         PyMem_Free(b); | 
 | 143 |     } | 
| Guido van Rossum | 58da931 | 2007-11-10 23:39:45 +0000 | [diff] [blame] | 144 | } | 
 | 145 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 146 | static PyObject * | 
 | 147 | deque_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | 
 | 148 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 149 |     dequeobject *deque; | 
 | 150 |     block *b; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 151 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 152 |     /* create dequeobject structure */ | 
 | 153 |     deque = (dequeobject *)type->tp_alloc(type, 0); | 
 | 154 |     if (deque == NULL) | 
 | 155 |         return NULL; | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 156 |  | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 157 |     b = newblock(); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 158 |     if (b == NULL) { | 
 | 159 |         Py_DECREF(deque); | 
 | 160 |         return NULL; | 
 | 161 |     } | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 162 |     MARK_END(b->leftlink); | 
 | 163 |     MARK_END(b->rightlink); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 164 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 165 |     assert(BLOCKLEN >= 2); | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 166 |     Py_SIZE(deque) = 0; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 167 |     deque->leftblock = b; | 
 | 168 |     deque->rightblock = b; | 
 | 169 |     deque->leftindex = CENTER + 1; | 
 | 170 |     deque->rightindex = CENTER; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 171 |     deque->state = 0; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 172 |     deque->maxlen = -1; | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 173 |     deque->weakreflist = NULL; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 174 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 175 |     return (PyObject *)deque; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 176 | } | 
 | 177 |  | 
 | 178 | static PyObject * | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 179 | deque_pop(dequeobject *deque, PyObject *unused) | 
 | 180 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 181 |     PyObject *item; | 
 | 182 |     block *prevblock; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 183 |  | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 184 |     if (Py_SIZE(deque) == 0) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 185 |         PyErr_SetString(PyExc_IndexError, "pop from an empty deque"); | 
 | 186 |         return NULL; | 
 | 187 |     } | 
 | 188 |     item = deque->rightblock->data[deque->rightindex]; | 
 | 189 |     deque->rightindex--; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 190 |     Py_SIZE(deque)--; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 191 |     deque->state++; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 192 |  | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 193 |     if (deque->rightindex < 0) { | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 194 |         if (Py_SIZE(deque)) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 195 |             prevblock = deque->rightblock->leftlink; | 
 | 196 |             assert(deque->leftblock != deque->rightblock); | 
 | 197 |             freeblock(deque->rightblock); | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 198 |             CHECK_NOT_END(prevblock); | 
 | 199 |             MARK_END(prevblock->rightlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 200 |             deque->rightblock = prevblock; | 
 | 201 |             deque->rightindex = BLOCKLEN - 1; | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 202 |         } else { | 
 | 203 |             assert(deque->leftblock == deque->rightblock); | 
 | 204 |             assert(deque->leftindex == deque->rightindex+1); | 
 | 205 |             /* re-center instead of freeing a block */ | 
 | 206 |             deque->leftindex = CENTER + 1; | 
 | 207 |             deque->rightindex = CENTER; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 208 |         } | 
 | 209 |     } | 
 | 210 |     return item; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 211 | } | 
 | 212 |  | 
 | 213 | PyDoc_STRVAR(pop_doc, "Remove and return the rightmost element."); | 
 | 214 |  | 
 | 215 | static PyObject * | 
 | 216 | deque_popleft(dequeobject *deque, PyObject *unused) | 
 | 217 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 218 |     PyObject *item; | 
 | 219 |     block *prevblock; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 220 |  | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 221 |     if (Py_SIZE(deque) == 0) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 222 |         PyErr_SetString(PyExc_IndexError, "pop from an empty deque"); | 
 | 223 |         return NULL; | 
 | 224 |     } | 
 | 225 |     assert(deque->leftblock != NULL); | 
 | 226 |     item = deque->leftblock->data[deque->leftindex]; | 
 | 227 |     deque->leftindex++; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 228 |     Py_SIZE(deque)--; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 229 |     deque->state++; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 230 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 231 |     if (deque->leftindex == BLOCKLEN) { | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 232 |         if (Py_SIZE(deque)) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 233 |             assert(deque->leftblock != deque->rightblock); | 
 | 234 |             prevblock = deque->leftblock->rightlink; | 
 | 235 |             freeblock(deque->leftblock); | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 236 |             CHECK_NOT_END(prevblock); | 
 | 237 |             MARK_END(prevblock->leftlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 238 |             deque->leftblock = prevblock; | 
 | 239 |             deque->leftindex = 0; | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 240 |         } else { | 
 | 241 |             assert(deque->leftblock == deque->rightblock); | 
 | 242 |             assert(deque->leftindex == deque->rightindex+1); | 
 | 243 |             /* re-center instead of freeing a block */ | 
 | 244 |             deque->leftindex = CENTER + 1; | 
 | 245 |             deque->rightindex = CENTER; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 246 |         } | 
 | 247 |     } | 
 | 248 |     return item; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 249 | } | 
 | 250 |  | 
 | 251 | PyDoc_STRVAR(popleft_doc, "Remove and return the leftmost element."); | 
 | 252 |  | 
| Raymond Hettinger | f30f5b9 | 2015-03-02 22:23:37 -0800 | [diff] [blame] | 253 | /* The deque's size limit is d.maxlen.  The limit can be zero or positive. | 
 | 254 |  * If there is no limit, then d.maxlen == -1. | 
 | 255 |  * | 
| Raymond Hettinger | a4b13d0 | 2015-10-15 08:05:31 -0700 | [diff] [blame] | 256 |  * After an item is added to a deque, we check to see if the size has | 
 | 257 |  * grown past the limit. If it has, we get the size back down to the limit | 
 | 258 |  * by popping an item off of the opposite end.  The methods that can | 
 | 259 |  * trigger this are append(), appendleft(), extend(), and extendleft(). | 
| Raymond Hettinger | d96db09 | 2015-10-11 22:34:48 -0700 | [diff] [blame] | 260 |  * | 
 | 261 |  * The macro to check whether a deque needs to be trimmed uses a single | 
 | 262 |  * unsigned test that returns true whenever 0 <= maxlen < Py_SIZE(deque). | 
| Raymond Hettinger | f30f5b9 | 2015-03-02 22:23:37 -0800 | [diff] [blame] | 263 |  */ | 
 | 264 |  | 
| Raymond Hettinger | d96db09 | 2015-10-11 22:34:48 -0700 | [diff] [blame] | 265 | #define NEEDS_TRIM(deque, maxlen) ((size_t)(maxlen) < (size_t)(Py_SIZE(deque))) | 
| Raymond Hettinger | f30f5b9 | 2015-03-02 22:23:37 -0800 | [diff] [blame] | 266 |  | 
| doko@ubuntu.com | bc73150 | 2016-05-18 01:06:01 +0200 | [diff] [blame] | 267 | static int | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 268 | deque_append_internal(dequeobject *deque, PyObject *item, Py_ssize_t maxlen) | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 269 | { | 
| Raymond Hettinger | 8dbbae2 | 2015-03-24 21:01:50 -0700 | [diff] [blame] | 270 |     if (deque->rightindex == BLOCKLEN - 1) { | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 271 |         block *b = newblock(); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 272 |         if (b == NULL) | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 273 |             return -1; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 274 |         b->leftlink = deque->rightblock; | 
 | 275 |         CHECK_END(deque->rightblock->rightlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 276 |         deque->rightblock->rightlink = b; | 
 | 277 |         deque->rightblock = b; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 278 |         MARK_END(b->rightlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 279 |         deque->rightindex = -1; | 
 | 280 |     } | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 281 |     Py_SIZE(deque)++; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 282 |     deque->rightindex++; | 
 | 283 |     deque->rightblock->data[deque->rightindex] = item; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 284 |     if (NEEDS_TRIM(deque, maxlen)) { | 
| Raymond Hettinger | 1286d14 | 2015-10-14 23:16:57 -0700 | [diff] [blame] | 285 |         PyObject *olditem = deque_popleft(deque, NULL); | 
 | 286 |         Py_DECREF(olditem); | 
| Raymond Hettinger | 0f43bb1 | 2015-10-20 00:03:33 -0700 | [diff] [blame] | 287 |     } else { | 
 | 288 |         deque->state++; | 
| Raymond Hettinger | d96db09 | 2015-10-11 22:34:48 -0700 | [diff] [blame] | 289 |     } | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 290 |     return 0; | 
 | 291 | } | 
 | 292 |  | 
 | 293 | static PyObject * | 
 | 294 | deque_append(dequeobject *deque, PyObject *item) | 
 | 295 | { | 
 | 296 |     Py_INCREF(item); | 
 | 297 |     if (deque_append_internal(deque, item, deque->maxlen) < 0) | 
 | 298 |         return NULL; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 299 |     Py_RETURN_NONE; | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 300 | } | 
 | 301 |  | 
 | 302 | PyDoc_STRVAR(append_doc, "Add an element to the right side of the deque."); | 
 | 303 |  | 
| doko@ubuntu.com | 17f0e61 | 2016-06-14 07:27:58 +0200 | [diff] [blame] | 304 | static int | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 305 | deque_appendleft_internal(dequeobject *deque, PyObject *item, Py_ssize_t maxlen) | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 306 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 307 |     if (deque->leftindex == 0) { | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 308 |         block *b = newblock(); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 309 |         if (b == NULL) | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 310 |             return -1; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 311 |         b->rightlink = deque->leftblock; | 
 | 312 |         CHECK_END(deque->leftblock->leftlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 313 |         deque->leftblock->leftlink = b; | 
 | 314 |         deque->leftblock = b; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 315 |         MARK_END(b->leftlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 316 |         deque->leftindex = BLOCKLEN; | 
 | 317 |     } | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 318 |     Py_SIZE(deque)++; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 319 |     deque->leftindex--; | 
 | 320 |     deque->leftblock->data[deque->leftindex] = item; | 
| Raymond Hettinger | d96db09 | 2015-10-11 22:34:48 -0700 | [diff] [blame] | 321 |     if (NEEDS_TRIM(deque, deque->maxlen)) { | 
| Raymond Hettinger | 1286d14 | 2015-10-14 23:16:57 -0700 | [diff] [blame] | 322 |         PyObject *olditem = deque_pop(deque, NULL); | 
 | 323 |         Py_DECREF(olditem); | 
| Raymond Hettinger | 0f43bb1 | 2015-10-20 00:03:33 -0700 | [diff] [blame] | 324 |     } else { | 
 | 325 |         deque->state++; | 
| Raymond Hettinger | d96db09 | 2015-10-11 22:34:48 -0700 | [diff] [blame] | 326 |     } | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 327 |     return 0; | 
 | 328 | } | 
 | 329 |  | 
 | 330 | static PyObject * | 
 | 331 | deque_appendleft(dequeobject *deque, PyObject *item) | 
 | 332 | { | 
 | 333 |     Py_INCREF(item); | 
 | 334 |     if (deque_appendleft_internal(deque, item, deque->maxlen) < 0) | 
 | 335 |         return NULL; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 336 |     Py_RETURN_NONE; | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 337 | } | 
 | 338 |  | 
 | 339 | PyDoc_STRVAR(appendleft_doc, "Add an element to the left side of the deque."); | 
 | 340 |  | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 341 | static PyObject* | 
 | 342 | finalize_iterator(PyObject *it) | 
 | 343 | { | 
 | 344 |     if (PyErr_Occurred()) { | 
 | 345 |         if (PyErr_ExceptionMatches(PyExc_StopIteration)) | 
 | 346 |             PyErr_Clear(); | 
 | 347 |         else { | 
 | 348 |             Py_DECREF(it); | 
 | 349 |             return NULL; | 
 | 350 |         } | 
 | 351 |     } | 
 | 352 |     Py_DECREF(it); | 
 | 353 |     Py_RETURN_NONE; | 
 | 354 | } | 
| Raymond Hettinger | 060c7f6 | 2009-03-10 09:36:07 +0000 | [diff] [blame] | 355 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 356 | /* Run an iterator to exhaustion.  Shortcut for | 
| Raymond Hettinger | 060c7f6 | 2009-03-10 09:36:07 +0000 | [diff] [blame] | 357 |    the extend/extendleft methods when maxlen == 0. */ | 
 | 358 | static PyObject* | 
 | 359 | consume_iterator(PyObject *it) | 
 | 360 | { | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 361 |     PyObject *(*iternext)(PyObject *); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 362 |     PyObject *item; | 
| Raymond Hettinger | 060c7f6 | 2009-03-10 09:36:07 +0000 | [diff] [blame] | 363 |  | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 364 |     iternext = *Py_TYPE(it)->tp_iternext; | 
 | 365 |     while ((item = iternext(it)) != NULL) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 366 |         Py_DECREF(item); | 
 | 367 |     } | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 368 |     return finalize_iterator(it); | 
| Raymond Hettinger | 060c7f6 | 2009-03-10 09:36:07 +0000 | [diff] [blame] | 369 | } | 
 | 370 |  | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 371 | static PyObject * | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 372 | deque_extend(dequeobject *deque, PyObject *iterable) | 
 | 373 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 374 |     PyObject *it, *item; | 
| Raymond Hettinger | 7a84552 | 2015-09-26 01:30:51 -0700 | [diff] [blame] | 375 |     PyObject *(*iternext)(PyObject *); | 
| Raymond Hettinger | 6b1e113 | 2015-10-11 09:43:50 -0700 | [diff] [blame] | 376 |     Py_ssize_t maxlen = deque->maxlen; | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 377 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 378 |     /* Handle case where id(deque) == id(iterable) */ | 
 | 379 |     if ((PyObject *)deque == iterable) { | 
 | 380 |         PyObject *result; | 
 | 381 |         PyObject *s = PySequence_List(iterable); | 
 | 382 |         if (s == NULL) | 
 | 383 |             return NULL; | 
 | 384 |         result = deque_extend(deque, s); | 
 | 385 |         Py_DECREF(s); | 
 | 386 |         return result; | 
 | 387 |     } | 
| Raymond Hettinger | 3f9afd8 | 2009-12-10 03:03:02 +0000 | [diff] [blame] | 388 |  | 
| Raymond Hettinger | d79d5b1 | 2016-03-04 09:55:07 -0800 | [diff] [blame] | 389 |     it = PyObject_GetIter(iterable); | 
 | 390 |     if (it == NULL) | 
 | 391 |         return NULL; | 
 | 392 |  | 
 | 393 |     if (maxlen == 0) | 
 | 394 |         return consume_iterator(it); | 
 | 395 |  | 
| Raymond Hettinger | d9c116c | 2013-07-09 00:13:21 -0700 | [diff] [blame] | 396 |     /* Space saving heuristic.  Start filling from the left */ | 
 | 397 |     if (Py_SIZE(deque) == 0) { | 
 | 398 |         assert(deque->leftblock == deque->rightblock); | 
 | 399 |         assert(deque->leftindex == deque->rightindex+1); | 
 | 400 |         deque->leftindex = 1; | 
 | 401 |         deque->rightindex = 0; | 
 | 402 |     } | 
 | 403 |  | 
| Raymond Hettinger | 7a84552 | 2015-09-26 01:30:51 -0700 | [diff] [blame] | 404 |     iternext = *Py_TYPE(it)->tp_iternext; | 
 | 405 |     while ((item = iternext(it)) != NULL) { | 
| Raymond Hettinger | 8805717 | 2016-09-11 22:45:53 -0700 | [diff] [blame] | 406 |         if (deque->rightindex == BLOCKLEN - 1) { | 
 | 407 |             block *b = newblock(); | 
 | 408 |             if (b == NULL) { | 
 | 409 |                 Py_DECREF(item); | 
 | 410 |                 Py_DECREF(it); | 
 | 411 |                 return NULL; | 
 | 412 |             } | 
 | 413 |             b->leftlink = deque->rightblock; | 
 | 414 |             CHECK_END(deque->rightblock->rightlink); | 
 | 415 |             deque->rightblock->rightlink = b; | 
 | 416 |             deque->rightblock = b; | 
 | 417 |             MARK_END(b->rightlink); | 
 | 418 |             deque->rightindex = -1; | 
 | 419 |         } | 
 | 420 |         Py_SIZE(deque)++; | 
 | 421 |         deque->rightindex++; | 
 | 422 |         deque->rightblock->data[deque->rightindex] = item; | 
 | 423 |         if (NEEDS_TRIM(deque, maxlen)) { | 
 | 424 |             PyObject *olditem = deque_popleft(deque, NULL); | 
 | 425 |             Py_DECREF(olditem); | 
 | 426 |         } else { | 
 | 427 |             deque->state++; | 
| Raymond Hettinger | 6b1e113 | 2015-10-11 09:43:50 -0700 | [diff] [blame] | 428 |         } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 429 |     } | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 430 |     return finalize_iterator(it); | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 431 | } | 
 | 432 |  | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 433 | PyDoc_STRVAR(extend_doc, | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 434 | "Extend the right side of the deque with elements from the iterable"); | 
 | 435 |  | 
 | 436 | static PyObject * | 
 | 437 | deque_extendleft(dequeobject *deque, PyObject *iterable) | 
 | 438 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 439 |     PyObject *it, *item; | 
| Raymond Hettinger | 7a84552 | 2015-09-26 01:30:51 -0700 | [diff] [blame] | 440 |     PyObject *(*iternext)(PyObject *); | 
| Raymond Hettinger | 6b1e113 | 2015-10-11 09:43:50 -0700 | [diff] [blame] | 441 |     Py_ssize_t maxlen = deque->maxlen; | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 442 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 443 |     /* Handle case where id(deque) == id(iterable) */ | 
 | 444 |     if ((PyObject *)deque == iterable) { | 
 | 445 |         PyObject *result; | 
 | 446 |         PyObject *s = PySequence_List(iterable); | 
 | 447 |         if (s == NULL) | 
 | 448 |             return NULL; | 
 | 449 |         result = deque_extendleft(deque, s); | 
 | 450 |         Py_DECREF(s); | 
 | 451 |         return result; | 
 | 452 |     } | 
| Raymond Hettinger | 3f9afd8 | 2009-12-10 03:03:02 +0000 | [diff] [blame] | 453 |  | 
| Raymond Hettinger | d79d5b1 | 2016-03-04 09:55:07 -0800 | [diff] [blame] | 454 |     it = PyObject_GetIter(iterable); | 
 | 455 |     if (it == NULL) | 
 | 456 |         return NULL; | 
 | 457 |  | 
 | 458 |     if (maxlen == 0) | 
 | 459 |         return consume_iterator(it); | 
 | 460 |  | 
| Raymond Hettinger | d9c116c | 2013-07-09 00:13:21 -0700 | [diff] [blame] | 461 |     /* Space saving heuristic.  Start filling from the right */ | 
 | 462 |     if (Py_SIZE(deque) == 0) { | 
 | 463 |         assert(deque->leftblock == deque->rightblock); | 
 | 464 |         assert(deque->leftindex == deque->rightindex+1); | 
 | 465 |         deque->leftindex = BLOCKLEN - 1; | 
 | 466 |         deque->rightindex = BLOCKLEN - 2; | 
 | 467 |     } | 
 | 468 |  | 
| Raymond Hettinger | 7a84552 | 2015-09-26 01:30:51 -0700 | [diff] [blame] | 469 |     iternext = *Py_TYPE(it)->tp_iternext; | 
 | 470 |     while ((item = iternext(it)) != NULL) { | 
| Raymond Hettinger | 8805717 | 2016-09-11 22:45:53 -0700 | [diff] [blame] | 471 |         if (deque->leftindex == 0) { | 
 | 472 |             block *b = newblock(); | 
 | 473 |             if (b == NULL) { | 
 | 474 |                 Py_DECREF(item); | 
 | 475 |                 Py_DECREF(it); | 
 | 476 |                 return NULL; | 
 | 477 |             } | 
 | 478 |             b->rightlink = deque->leftblock; | 
 | 479 |             CHECK_END(deque->leftblock->leftlink); | 
 | 480 |             deque->leftblock->leftlink = b; | 
 | 481 |             deque->leftblock = b; | 
 | 482 |             MARK_END(b->leftlink); | 
 | 483 |             deque->leftindex = BLOCKLEN; | 
 | 484 |         } | 
 | 485 |         Py_SIZE(deque)++; | 
 | 486 |         deque->leftindex--; | 
 | 487 |         deque->leftblock->data[deque->leftindex] = item; | 
 | 488 |         if (NEEDS_TRIM(deque, maxlen)) { | 
 | 489 |             PyObject *olditem = deque_pop(deque, NULL); | 
 | 490 |             Py_DECREF(olditem); | 
 | 491 |         } else { | 
 | 492 |             deque->state++; | 
| Raymond Hettinger | 6b1e113 | 2015-10-11 09:43:50 -0700 | [diff] [blame] | 493 |         } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 494 |     } | 
| Raymond Hettinger | fd265f4 | 2015-10-02 23:17:33 -0700 | [diff] [blame] | 495 |     return finalize_iterator(it); | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 496 | } | 
 | 497 |  | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 498 | PyDoc_STRVAR(extendleft_doc, | 
| Raymond Hettinger | 3ba85c2 | 2004-02-06 19:04:56 +0000 | [diff] [blame] | 499 | "Extend the left side of the deque with elements from the iterable"); | 
 | 500 |  | 
| Raymond Hettinger | 3f9afd8 | 2009-12-10 03:03:02 +0000 | [diff] [blame] | 501 | static PyObject * | 
 | 502 | deque_inplace_concat(dequeobject *deque, PyObject *other) | 
 | 503 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 504 |     PyObject *result; | 
| Raymond Hettinger | 3f9afd8 | 2009-12-10 03:03:02 +0000 | [diff] [blame] | 505 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 506 |     result = deque_extend(deque, other); | 
 | 507 |     if (result == NULL) | 
 | 508 |         return result; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 509 |     Py_INCREF(deque); | 
| Raymond Hettinger | 2b2b753 | 2015-09-05 17:05:52 -0700 | [diff] [blame] | 510 |     Py_DECREF(result); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 511 |     return (PyObject *)deque; | 
| Raymond Hettinger | 3f9afd8 | 2009-12-10 03:03:02 +0000 | [diff] [blame] | 512 | } | 
 | 513 |  | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 514 | static PyObject * | 
 | 515 | deque_copy(PyObject *deque) | 
 | 516 | { | 
 | 517 |     dequeobject *old_deque = (dequeobject *)deque; | 
 | 518 |     if (Py_TYPE(deque) == &deque_type) { | 
 | 519 |         dequeobject *new_deque; | 
 | 520 |         PyObject *rv; | 
 | 521 |  | 
 | 522 |         new_deque = (dequeobject *)deque_new(&deque_type, (PyObject *)NULL, (PyObject *)NULL); | 
 | 523 |         if (new_deque == NULL) | 
 | 524 |             return NULL; | 
 | 525 |         new_deque->maxlen = old_deque->maxlen; | 
 | 526 |         /* Fast path for the deque_repeat() common case where len(deque) == 1 */ | 
| Raymond Hettinger | 0443ac2 | 2015-10-05 22:52:37 -0400 | [diff] [blame] | 527 |         if (Py_SIZE(deque) == 1) { | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 528 |             PyObject *item = old_deque->leftblock->data[old_deque->leftindex]; | 
 | 529 |             rv = deque_append(new_deque, item); | 
 | 530 |         } else { | 
 | 531 |             rv = deque_extend(new_deque, deque); | 
 | 532 |         } | 
 | 533 |         if (rv != NULL) { | 
 | 534 |             Py_DECREF(rv); | 
 | 535 |             return (PyObject *)new_deque; | 
 | 536 |         } | 
 | 537 |         Py_DECREF(new_deque); | 
 | 538 |         return NULL; | 
 | 539 |     } | 
 | 540 |     if (old_deque->maxlen < 0) | 
| Victor Stinner | 7bfb42d | 2016-12-05 17:04:32 +0100 | [diff] [blame] | 541 |         return PyObject_CallFunctionObjArgs((PyObject *)(Py_TYPE(deque)), | 
 | 542 |                                             deque, NULL); | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 543 |     else | 
 | 544 |         return PyObject_CallFunction((PyObject *)(Py_TYPE(deque)), "Oi", | 
 | 545 |             deque, old_deque->maxlen, NULL); | 
 | 546 | } | 
 | 547 |  | 
 | 548 | PyDoc_STRVAR(copy_doc, "Return a shallow copy of a deque."); | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 549 |  | 
 | 550 | static PyObject * | 
 | 551 | deque_concat(dequeobject *deque, PyObject *other) | 
 | 552 | { | 
| Benjamin Peterson | 1a62921 | 2015-04-04 10:52:36 -0400 | [diff] [blame] | 553 |     PyObject *new_deque, *result; | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 554 |     int rv; | 
 | 555 |  | 
 | 556 |     rv = PyObject_IsInstance(other, (PyObject *)&deque_type); | 
 | 557 |     if (rv <= 0) { | 
 | 558 |         if (rv == 0) { | 
 | 559 |             PyErr_Format(PyExc_TypeError, | 
 | 560 |                          "can only concatenate deque (not \"%.200s\") to deque", | 
 | 561 |                          other->ob_type->tp_name); | 
 | 562 |         } | 
 | 563 |         return NULL; | 
 | 564 |     } | 
 | 565 |  | 
 | 566 |     new_deque = deque_copy((PyObject *)deque); | 
 | 567 |     if (new_deque == NULL) | 
 | 568 |         return NULL; | 
| Benjamin Peterson | 1a62921 | 2015-04-04 10:52:36 -0400 | [diff] [blame] | 569 |     result = deque_extend((dequeobject *)new_deque, other); | 
 | 570 |     if (result == NULL) { | 
 | 571 |         Py_DECREF(new_deque); | 
 | 572 |         return NULL; | 
 | 573 |     } | 
 | 574 |     Py_DECREF(result); | 
 | 575 |     return new_deque; | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 576 | } | 
 | 577 |  | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 578 | static void | 
 | 579 | deque_clear(dequeobject *deque) | 
 | 580 | { | 
 | 581 |     block *b; | 
 | 582 |     block *prevblock; | 
 | 583 |     block *leftblock; | 
 | 584 |     Py_ssize_t leftindex; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 585 |     Py_ssize_t n, m; | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 586 |     PyObject *item; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 587 |     PyObject **itemptr, **limit; | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 588 |  | 
| Raymond Hettinger | 3803114 | 2015-09-29 22:45:05 -0700 | [diff] [blame] | 589 |     if (Py_SIZE(deque) == 0) | 
 | 590 |         return; | 
 | 591 |  | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 592 |     /* During the process of clearing a deque, decrefs can cause the | 
 | 593 |        deque to mutate.  To avoid fatal confusion, we have to make the | 
 | 594 |        deque empty before clearing the blocks and never refer to | 
 | 595 |        anything via deque->ref while clearing.  (This is the same | 
 | 596 |        technique used for clearing lists, sets, and dicts.) | 
 | 597 |  | 
 | 598 |        Making the deque empty requires allocating a new empty block.  In | 
 | 599 |        the unlikely event that memory is full, we fall back to an | 
 | 600 |        alternate method that doesn't require a new block.  Repeating | 
 | 601 |        pops in a while-loop is slower, possibly re-entrant (and a clever | 
 | 602 |        adversary could cause it to never terminate). | 
 | 603 |     */ | 
 | 604 |  | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 605 |     b = newblock(); | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 606 |     if (b == NULL) { | 
 | 607 |         PyErr_Clear(); | 
 | 608 |         goto alternate_method; | 
 | 609 |     } | 
 | 610 |  | 
 | 611 |     /* Remember the old size, leftblock, and leftindex */ | 
| Raymond Hettinger | 6f86a33 | 2016-03-02 00:30:58 -0800 | [diff] [blame] | 612 |     n = Py_SIZE(deque); | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 613 |     leftblock = deque->leftblock; | 
 | 614 |     leftindex = deque->leftindex; | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 615 |  | 
 | 616 |     /* Set the deque to be empty using the newly allocated block */ | 
 | 617 |     MARK_END(b->leftlink); | 
 | 618 |     MARK_END(b->rightlink); | 
 | 619 |     Py_SIZE(deque) = 0; | 
 | 620 |     deque->leftblock = b; | 
 | 621 |     deque->rightblock = b; | 
 | 622 |     deque->leftindex = CENTER + 1; | 
 | 623 |     deque->rightindex = CENTER; | 
 | 624 |     deque->state++; | 
 | 625 |  | 
 | 626 |     /* Now the old size, leftblock, and leftindex are disconnected from | 
 | 627 |        the empty deque and we can use them to decref the pointers. | 
 | 628 |     */ | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 629 |     m = (BLOCKLEN - leftindex > n) ? n : BLOCKLEN - leftindex; | 
| Raymond Hettinger | 589106b | 2016-03-02 00:06:21 -0800 | [diff] [blame] | 630 |     itemptr = &leftblock->data[leftindex]; | 
| Raymond Hettinger | 6f86a33 | 2016-03-02 00:30:58 -0800 | [diff] [blame] | 631 |     limit = itemptr + m; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 632 |     n -= m; | 
 | 633 |     while (1) { | 
 | 634 |         if (itemptr == limit) { | 
 | 635 |             if (n == 0) | 
 | 636 |                 break; | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 637 |             CHECK_NOT_END(leftblock->rightlink); | 
 | 638 |             prevblock = leftblock; | 
 | 639 |             leftblock = leftblock->rightlink; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 640 |             m = (n > BLOCKLEN) ? BLOCKLEN : n; | 
| Raymond Hettinger | 589106b | 2016-03-02 00:06:21 -0800 | [diff] [blame] | 641 |             itemptr = leftblock->data; | 
| Raymond Hettinger | 6f86a33 | 2016-03-02 00:30:58 -0800 | [diff] [blame] | 642 |             limit = itemptr + m; | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 643 |             n -= m; | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 644 |             freeblock(prevblock); | 
 | 645 |         } | 
| Raymond Hettinger | d84ec22 | 2016-01-24 09:12:06 -0800 | [diff] [blame] | 646 |         item = *(itemptr++); | 
 | 647 |         Py_DECREF(item); | 
| Raymond Hettinger | 8299e9b | 2015-09-26 21:31:23 -0700 | [diff] [blame] | 648 |     } | 
 | 649 |     CHECK_END(leftblock->rightlink); | 
 | 650 |     freeblock(leftblock); | 
 | 651 |     return; | 
 | 652 |  | 
 | 653 |   alternate_method: | 
 | 654 |     while (Py_SIZE(deque)) { | 
 | 655 |         item = deque_pop(deque, NULL); | 
 | 656 |         assert (item != NULL); | 
 | 657 |         Py_DECREF(item); | 
 | 658 |     } | 
 | 659 | } | 
 | 660 |  | 
 | 661 | static PyObject * | 
 | 662 | deque_clearmethod(dequeobject *deque) | 
 | 663 | { | 
 | 664 |     deque_clear(deque); | 
 | 665 |     Py_RETURN_NONE; | 
 | 666 | } | 
 | 667 |  | 
 | 668 | PyDoc_STRVAR(clear_doc, "Remove all elements from the deque."); | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 669 |  | 
 | 670 | static PyObject * | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 671 | deque_inplace_repeat(dequeobject *deque, Py_ssize_t n) | 
 | 672 | { | 
| Raymond Hettinger | c22eee6 | 2015-09-26 02:14:50 -0700 | [diff] [blame] | 673 |     Py_ssize_t i, m, size; | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 674 |     PyObject *seq; | 
 | 675 |     PyObject *rv; | 
 | 676 |  | 
 | 677 |     size = Py_SIZE(deque); | 
 | 678 |     if (size == 0 || n == 1) { | 
 | 679 |         Py_INCREF(deque); | 
 | 680 |         return (PyObject *)deque; | 
 | 681 |     } | 
 | 682 |  | 
 | 683 |     if (n <= 0) { | 
 | 684 |         deque_clear(deque); | 
 | 685 |         Py_INCREF(deque); | 
 | 686 |         return (PyObject *)deque; | 
 | 687 |     } | 
 | 688 |  | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 689 |     if (size == 1) { | 
 | 690 |         /* common case, repeating a single element */ | 
 | 691 |         PyObject *item = deque->leftblock->data[deque->leftindex]; | 
 | 692 |  | 
| Raymond Hettinger | a7f63009 | 2015-10-10 23:56:02 -0400 | [diff] [blame] | 693 |         if (deque->maxlen >= 0 && n > deque->maxlen) | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 694 |             n = deque->maxlen; | 
 | 695 |  | 
| Raymond Hettinger | 67c78b5 | 2015-09-12 11:00:20 -0400 | [diff] [blame] | 696 |         deque->state++; | 
| Raymond Hettinger | ad26225 | 2015-09-14 01:03:04 -0400 | [diff] [blame] | 697 |         for (i = 0 ; i < n-1 ; ) { | 
| Raymond Hettinger | 67c78b5 | 2015-09-12 11:00:20 -0400 | [diff] [blame] | 698 |             if (deque->rightindex == BLOCKLEN - 1) { | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 699 |                 block *b = newblock(); | 
| Raymond Hettinger | 67c78b5 | 2015-09-12 11:00:20 -0400 | [diff] [blame] | 700 |                 if (b == NULL) { | 
 | 701 |                     Py_SIZE(deque) += i; | 
 | 702 |                     return NULL; | 
 | 703 |                 } | 
 | 704 |                 b->leftlink = deque->rightblock; | 
 | 705 |                 CHECK_END(deque->rightblock->rightlink); | 
 | 706 |                 deque->rightblock->rightlink = b; | 
 | 707 |                 deque->rightblock = b; | 
 | 708 |                 MARK_END(b->rightlink); | 
 | 709 |                 deque->rightindex = -1; | 
 | 710 |             } | 
| Raymond Hettinger | c22eee6 | 2015-09-26 02:14:50 -0700 | [diff] [blame] | 711 |             m = n - 1 - i; | 
 | 712 |             if (m > BLOCKLEN - 1 - deque->rightindex) | 
 | 713 |                 m = BLOCKLEN - 1 - deque->rightindex; | 
 | 714 |             i += m; | 
 | 715 |             while (m--) { | 
| Raymond Hettinger | ad26225 | 2015-09-14 01:03:04 -0400 | [diff] [blame] | 716 |                 deque->rightindex++; | 
 | 717 |                 Py_INCREF(item); | 
 | 718 |                 deque->rightblock->data[deque->rightindex] = item; | 
 | 719 |             } | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 720 |         } | 
| Raymond Hettinger | 67c78b5 | 2015-09-12 11:00:20 -0400 | [diff] [blame] | 721 |         Py_SIZE(deque) += i; | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 722 |         Py_INCREF(deque); | 
 | 723 |         return (PyObject *)deque; | 
 | 724 |     } | 
 | 725 |  | 
| Raymond Hettinger | 20151f5 | 2015-10-16 22:47:29 -0700 | [diff] [blame] | 726 |     if ((size_t)size > PY_SSIZE_T_MAX / (size_t)n) { | 
| Raymond Hettinger | f5d72f3 | 2015-09-09 22:39:44 -0400 | [diff] [blame] | 727 |         return PyErr_NoMemory(); | 
 | 728 |     } | 
 | 729 |  | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 730 |     seq = PySequence_List((PyObject *)deque); | 
 | 731 |     if (seq == NULL) | 
 | 732 |         return seq; | 
 | 733 |  | 
| Louie Lu | 357bad7 | 2017-02-24 11:59:49 +0800 | [diff] [blame] | 734 |     /* Reduce the number of repetitions when maxlen would be exceeded */ | 
 | 735 |     if (deque->maxlen >= 0 && n * size > deque->maxlen) | 
 | 736 |         n = (deque->maxlen + size - 1) / size; | 
 | 737 |  | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 738 |     for (i = 0 ; i < n-1 ; i++) { | 
 | 739 |         rv = deque_extend(deque, seq); | 
 | 740 |         if (rv == NULL) { | 
 | 741 |             Py_DECREF(seq); | 
 | 742 |             return NULL; | 
 | 743 |         } | 
 | 744 |         Py_DECREF(rv); | 
 | 745 |     } | 
 | 746 |     Py_INCREF(deque); | 
 | 747 |     Py_DECREF(seq); | 
 | 748 |     return (PyObject *)deque; | 
 | 749 | } | 
 | 750 |  | 
| Raymond Hettinger | f5d72f3 | 2015-09-09 22:39:44 -0400 | [diff] [blame] | 751 | static PyObject * | 
 | 752 | deque_repeat(dequeobject *deque, Py_ssize_t n) | 
 | 753 | { | 
 | 754 |     dequeobject *new_deque; | 
| Raymond Hettinger | 95e2cc5 | 2015-09-13 02:41:18 -0400 | [diff] [blame] | 755 |     PyObject *rv; | 
| Raymond Hettinger | f5d72f3 | 2015-09-09 22:39:44 -0400 | [diff] [blame] | 756 |  | 
 | 757 |     new_deque = (dequeobject *)deque_copy((PyObject *) deque); | 
 | 758 |     if (new_deque == NULL) | 
 | 759 |         return NULL; | 
| Raymond Hettinger | 95e2cc5 | 2015-09-13 02:41:18 -0400 | [diff] [blame] | 760 |     rv = deque_inplace_repeat(new_deque, n); | 
 | 761 |     Py_DECREF(new_deque); | 
 | 762 |     return rv; | 
| Raymond Hettinger | f5d72f3 | 2015-09-09 22:39:44 -0400 | [diff] [blame] | 763 | } | 
 | 764 |  | 
| Raymond Hettinger | 5402315 | 2014-04-23 00:58:48 -0700 | [diff] [blame] | 765 | /* The rotate() method is part of the public API and is used internally | 
 | 766 | as a primitive for other methods. | 
 | 767 |  | 
 | 768 | Rotation by 1 or -1 is a common case, so any optimizations for high | 
 | 769 | volume rotations should take care not to penalize the common case. | 
 | 770 |  | 
 | 771 | Conceptually, a rotate by one is equivalent to a pop on one side and an | 
 | 772 | append on the other.  However, a pop/append pair is unnecessarily slow | 
| Martin Panter | d2ad571 | 2015-11-02 04:20:33 +0000 | [diff] [blame] | 773 | because it requires an incref/decref pair for an object located randomly | 
| Raymond Hettinger | 5402315 | 2014-04-23 00:58:48 -0700 | [diff] [blame] | 774 | in memory.  It is better to just move the object pointer from one block | 
 | 775 | to the next without changing the reference count. | 
 | 776 |  | 
 | 777 | When moving batches of pointers, it is tempting to use memcpy() but that | 
 | 778 | proved to be slower than a simple loop for a variety of reasons. | 
 | 779 | Memcpy() cannot know in advance that we're copying pointers instead of | 
 | 780 | bytes, that the source and destination are pointer aligned and | 
 | 781 | non-overlapping, that moving just one pointer is a common case, that we | 
 | 782 | never need to move more than BLOCKLEN pointers, and that at least one | 
 | 783 | pointer is always moved. | 
 | 784 |  | 
 | 785 | For high volume rotations, newblock() and freeblock() are never called | 
 | 786 | more than once.  Previously emptied blocks are immediately reused as a | 
 | 787 | destination block.  If a block is left-over at the end, it is freed. | 
 | 788 | */ | 
 | 789 |  | 
| Raymond Hettinger | dcb9d94 | 2004-10-09 16:02:18 +0000 | [diff] [blame] | 790 | static int | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 791 | _deque_rotate(dequeobject *deque, Py_ssize_t n) | 
| Raymond Hettinger | 5c5eb86 | 2004-02-07 21:13:00 +0000 | [diff] [blame] | 792 | { | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 793 |     block *b = NULL; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 794 |     block *leftblock = deque->leftblock; | 
 | 795 |     block *rightblock = deque->rightblock; | 
 | 796 |     Py_ssize_t leftindex = deque->leftindex; | 
 | 797 |     Py_ssize_t rightindex = deque->rightindex; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 798 |     Py_ssize_t len=Py_SIZE(deque), halflen=len>>1; | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 799 |     int rv = -1; | 
| Raymond Hettinger | 5c5eb86 | 2004-02-07 21:13:00 +0000 | [diff] [blame] | 800 |  | 
| Raymond Hettinger | 464d89b | 2013-01-11 22:29:50 -0800 | [diff] [blame] | 801 |     if (len <= 1) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 802 |         return 0; | 
 | 803 |     if (n > halflen || n < -halflen) { | 
 | 804 |         n %= len; | 
 | 805 |         if (n > halflen) | 
 | 806 |             n -= len; | 
 | 807 |         else if (n < -halflen) | 
 | 808 |             n += len; | 
 | 809 |     } | 
| Raymond Hettinger | 1f0044c | 2013-02-05 01:30:46 -0500 | [diff] [blame] | 810 |     assert(len > 1); | 
| Raymond Hettinger | a4409c1 | 2013-02-04 00:08:12 -0500 | [diff] [blame] | 811 |     assert(-halflen <= n && n <= halflen); | 
| Raymond Hettinger | 231ee4d | 2013-02-02 11:24:43 -0800 | [diff] [blame] | 812 |  | 
| Raymond Hettinger | 464d89b | 2013-01-11 22:29:50 -0800 | [diff] [blame] | 813 |     deque->state++; | 
| Raymond Hettinger | 1f0044c | 2013-02-05 01:30:46 -0500 | [diff] [blame] | 814 |     while (n > 0) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 815 |         if (leftindex == 0) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 816 |             if (b == NULL) { | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 817 |                 b = newblock(); | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 818 |                 if (b == NULL) | 
 | 819 |                     goto done; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 820 |             } | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 821 |             b->rightlink = leftblock; | 
 | 822 |             CHECK_END(leftblock->leftlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 823 |             leftblock->leftlink = b; | 
 | 824 |             leftblock = b; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 825 |             MARK_END(b->leftlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 826 |             leftindex = BLOCKLEN; | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 827 |             b = NULL; | 
| Raymond Hettinger | 464d89b | 2013-01-11 22:29:50 -0800 | [diff] [blame] | 828 |         } | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 829 |         assert(leftindex > 0); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 830 |         { | 
 | 831 |             PyObject **src, **dest; | 
 | 832 |             Py_ssize_t m = n; | 
| Raymond Hettinger | 21777ac | 2013-02-02 09:56:08 -0800 | [diff] [blame] | 833 |  | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 834 |             if (m > rightindex + 1) | 
 | 835 |                 m = rightindex + 1; | 
 | 836 |             if (m > leftindex) | 
 | 837 |                 m = leftindex; | 
 | 838 |             assert (m > 0 && m <= len); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 839 |             rightindex -= m; | 
 | 840 |             leftindex -= m; | 
| Raymond Hettinger | 0e259f1 | 2015-02-01 22:53:41 -0800 | [diff] [blame] | 841 |             src = &rightblock->data[rightindex + 1]; | 
 | 842 |             dest = &leftblock->data[leftindex]; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 843 |             n -= m; | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 844 |             do { | 
| Raymond Hettinger | 0e259f1 | 2015-02-01 22:53:41 -0800 | [diff] [blame] | 845 |                 *(dest++) = *(src++); | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 846 |             } while (--m); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 847 |         } | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 848 |         if (rightindex < 0) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 849 |             assert(leftblock != rightblock); | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 850 |             assert(b == NULL); | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 851 |             b = rightblock; | 
| Raymond Hettinger | b97cc49 | 2013-07-21 01:51:07 -0700 | [diff] [blame] | 852 |             CHECK_NOT_END(rightblock->leftlink); | 
 | 853 |             rightblock = rightblock->leftlink; | 
 | 854 |             MARK_END(rightblock->rightlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 855 |             rightindex = BLOCKLEN - 1; | 
| Raymond Hettinger | 464d89b | 2013-01-11 22:29:50 -0800 | [diff] [blame] | 856 |         } | 
| Raymond Hettinger | 21777ac | 2013-02-02 09:56:08 -0800 | [diff] [blame] | 857 |     } | 
| Raymond Hettinger | 1f0044c | 2013-02-05 01:30:46 -0500 | [diff] [blame] | 858 |     while (n < 0) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 859 |         if (rightindex == BLOCKLEN - 1) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 860 |             if (b == NULL) { | 
| Raymond Hettinger | db41fd4 | 2015-10-22 22:48:16 -0700 | [diff] [blame] | 861 |                 b = newblock(); | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 862 |                 if (b == NULL) | 
 | 863 |                     goto done; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 864 |             } | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 865 |             b->leftlink = rightblock; | 
 | 866 |             CHECK_END(rightblock->rightlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 867 |             rightblock->rightlink = b; | 
 | 868 |             rightblock = b; | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 869 |             MARK_END(b->rightlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 870 |             rightindex = -1; | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 871 |             b = NULL; | 
| Raymond Hettinger | 464d89b | 2013-01-11 22:29:50 -0800 | [diff] [blame] | 872 |         } | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 873 |         assert (rightindex < BLOCKLEN - 1); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 874 |         { | 
 | 875 |             PyObject **src, **dest; | 
 | 876 |             Py_ssize_t m = -n; | 
| Raymond Hettinger | 21777ac | 2013-02-02 09:56:08 -0800 | [diff] [blame] | 877 |  | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 878 |             if (m > BLOCKLEN - leftindex) | 
 | 879 |                 m = BLOCKLEN - leftindex; | 
 | 880 |             if (m > BLOCKLEN - 1 - rightindex) | 
 | 881 |                 m = BLOCKLEN - 1 - rightindex; | 
 | 882 |             assert (m > 0 && m <= len); | 
 | 883 |             src = &leftblock->data[leftindex]; | 
 | 884 |             dest = &rightblock->data[rightindex + 1]; | 
 | 885 |             leftindex += m; | 
 | 886 |             rightindex += m; | 
 | 887 |             n += m; | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 888 |             do { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 889 |                 *(dest++) = *(src++); | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 890 |             } while (--m); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 891 |         } | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 892 |         if (leftindex == BLOCKLEN) { | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 893 |             assert(leftblock != rightblock); | 
| Raymond Hettinger | 840533b | 2013-07-13 17:03:58 -0700 | [diff] [blame] | 894 |             assert(b == NULL); | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 895 |             b = leftblock; | 
| Raymond Hettinger | b97cc49 | 2013-07-21 01:51:07 -0700 | [diff] [blame] | 896 |             CHECK_NOT_END(leftblock->rightlink); | 
 | 897 |             leftblock = leftblock->rightlink; | 
 | 898 |             MARK_END(leftblock->leftlink); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 899 |             leftindex = 0; | 
| Raymond Hettinger | 21777ac | 2013-02-02 09:56:08 -0800 | [diff] [blame] | 900 |         } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 901 |     } | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 902 |     rv = 0; | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 903 | done: | 
| Raymond Hettinger | 3959af9 | 2013-07-13 02:34:08 -0700 | [diff] [blame] | 904 |     if (b != NULL) | 
 | 905 |         freeblock(b); | 
| Raymond Hettinger | 20b0f87 | 2013-06-23 15:44:33 -0700 | [diff] [blame] | 906 |     deque->leftblock = leftblock; | 
 | 907 |     deque->rightblock = rightblock; | 
 | 908 |     deque->leftindex = leftindex; | 
 | 909 |     deque->rightindex = rightindex; | 
 | 910 |  | 
 | 911 |     return rv; | 
| Raymond Hettinger | dcb9d94 | 2004-10-09 16:02:18 +0000 | [diff] [blame] | 912 | } | 
 | 913 |  | 
 | 914 | static PyObject * | 
| Serhiy Storchaka | 6969eaf | 2017-07-03 21:20:15 +0300 | [diff] [blame] | 915 | deque_rotate(dequeobject *deque, PyObject **args, Py_ssize_t nargs) | 
| Raymond Hettinger | dcb9d94 | 2004-10-09 16:02:18 +0000 | [diff] [blame] | 916 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 917 |     Py_ssize_t n=1; | 
| Raymond Hettinger | dcb9d94 | 2004-10-09 16:02:18 +0000 | [diff] [blame] | 918 |  | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 919 |     if (!_PyArg_ParseStack(args, nargs, "|n:rotate", &n)) { | 
 | 920 |         return NULL; | 
 | 921 |     } | 
 | 922 |  | 
| Raymond Hettinger | 6921c13 | 2015-03-21 02:03:40 -0700 | [diff] [blame] | 923 |     if (!_deque_rotate(deque, n)) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 924 |         Py_RETURN_NONE; | 
 | 925 |     return NULL; | 
| Raymond Hettinger | 5c5eb86 | 2004-02-07 21:13:00 +0000 | [diff] [blame] | 926 | } | 
 | 927 |  | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 928 | PyDoc_STRVAR(rotate_doc, | 
| Raymond Hettinger | ee33b27 | 2004-02-08 04:05:26 +0000 | [diff] [blame] | 929 | "Rotate the deque n steps to the right (default n=1).  If n is negative, rotates left."); | 
| Raymond Hettinger | 5c5eb86 | 2004-02-07 21:13:00 +0000 | [diff] [blame] | 930 |  | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 931 | static PyObject * | 
 | 932 | deque_reverse(dequeobject *deque, PyObject *unused) | 
 | 933 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 934 |     block *leftblock = deque->leftblock; | 
 | 935 |     block *rightblock = deque->rightblock; | 
 | 936 |     Py_ssize_t leftindex = deque->leftindex; | 
 | 937 |     Py_ssize_t rightindex = deque->rightindex; | 
| Raymond Hettinger | cfe5b6c | 2015-07-20 00:25:50 -0400 | [diff] [blame] | 938 |     Py_ssize_t n = Py_SIZE(deque) >> 1; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 939 |     PyObject *tmp; | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 940 |  | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 941 |     while (--n >= 0) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 942 |         /* Validate that pointers haven't met in the middle */ | 
 | 943 |         assert(leftblock != rightblock || leftindex < rightindex); | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 944 |         CHECK_NOT_END(leftblock); | 
 | 945 |         CHECK_NOT_END(rightblock); | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 946 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 947 |         /* Swap */ | 
 | 948 |         tmp = leftblock->data[leftindex]; | 
 | 949 |         leftblock->data[leftindex] = rightblock->data[rightindex]; | 
 | 950 |         rightblock->data[rightindex] = tmp; | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 951 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 952 |         /* Advance left block/index pair */ | 
 | 953 |         leftindex++; | 
 | 954 |         if (leftindex == BLOCKLEN) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 955 |             leftblock = leftblock->rightlink; | 
 | 956 |             leftindex = 0; | 
 | 957 |         } | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 958 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 959 |         /* Step backwards with the right block/index pair */ | 
 | 960 |         rightindex--; | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 961 |         if (rightindex < 0) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 962 |             rightblock = rightblock->leftlink; | 
 | 963 |             rightindex = BLOCKLEN - 1; | 
 | 964 |         } | 
 | 965 |     } | 
 | 966 |     Py_RETURN_NONE; | 
| Raymond Hettinger | e5fdedb | 2009-12-10 00:47:21 +0000 | [diff] [blame] | 967 | } | 
 | 968 |  | 
 | 969 | PyDoc_STRVAR(reverse_doc, | 
 | 970 | "D.reverse() -- reverse *IN PLACE*"); | 
 | 971 |  | 
| Raymond Hettinger | 44459de | 2010-04-03 23:20:46 +0000 | [diff] [blame] | 972 | static PyObject * | 
 | 973 | deque_count(dequeobject *deque, PyObject *v) | 
 | 974 | { | 
| Raymond Hettinger | f3a67b7 | 2013-07-06 17:49:06 -1000 | [diff] [blame] | 975 |     block *b = deque->leftblock; | 
 | 976 |     Py_ssize_t index = deque->leftindex; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 977 |     Py_ssize_t n = Py_SIZE(deque); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 978 |     Py_ssize_t count = 0; | 
| Raymond Hettinger | f9d9c79 | 2015-03-02 22:47:46 -0800 | [diff] [blame] | 979 |     size_t start_state = deque->state; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 980 |     PyObject *item; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 981 |     int cmp; | 
| Raymond Hettinger | 44459de | 2010-04-03 23:20:46 +0000 | [diff] [blame] | 982 |  | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 983 |     while (--n >= 0) { | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 984 |         CHECK_NOT_END(b); | 
| Raymond Hettinger | f3a67b7 | 2013-07-06 17:49:06 -1000 | [diff] [blame] | 985 |         item = b->data[index]; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 986 |         cmp = PyObject_RichCompareBool(item, v, Py_EQ); | 
| Raymond Hettinger | 2b0d646 | 2015-09-23 19:15:44 -0700 | [diff] [blame] | 987 |         if (cmp < 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 988 |             return NULL; | 
| Raymond Hettinger | 2b0d646 | 2015-09-23 19:15:44 -0700 | [diff] [blame] | 989 |         count += cmp; | 
| Raymond Hettinger | 44459de | 2010-04-03 23:20:46 +0000 | [diff] [blame] | 990 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 991 |         if (start_state != deque->state) { | 
 | 992 |             PyErr_SetString(PyExc_RuntimeError, | 
 | 993 |                             "deque mutated during iteration"); | 
 | 994 |             return NULL; | 
 | 995 |         } | 
| Raymond Hettinger | 44459de | 2010-04-03 23:20:46 +0000 | [diff] [blame] | 996 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 997 |         /* Advance left block/index pair */ | 
| Raymond Hettinger | f3a67b7 | 2013-07-06 17:49:06 -1000 | [diff] [blame] | 998 |         index++; | 
 | 999 |         if (index == BLOCKLEN) { | 
 | 1000 |             b = b->rightlink; | 
 | 1001 |             index = 0; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1002 |         } | 
 | 1003 |     } | 
 | 1004 |     return PyLong_FromSsize_t(count); | 
| Raymond Hettinger | 44459de | 2010-04-03 23:20:46 +0000 | [diff] [blame] | 1005 | } | 
 | 1006 |  | 
 | 1007 | PyDoc_STRVAR(count_doc, | 
 | 1008 | "D.count(value) -> integer -- return number of occurrences of value"); | 
 | 1009 |  | 
| Raymond Hettinger | 39dadf7 | 2015-03-20 16:38:56 -0700 | [diff] [blame] | 1010 | static int | 
 | 1011 | deque_contains(dequeobject *deque, PyObject *v) | 
 | 1012 | { | 
 | 1013 |     block *b = deque->leftblock; | 
 | 1014 |     Py_ssize_t index = deque->leftindex; | 
 | 1015 |     Py_ssize_t n = Py_SIZE(deque); | 
| Raymond Hettinger | 39dadf7 | 2015-03-20 16:38:56 -0700 | [diff] [blame] | 1016 |     size_t start_state = deque->state; | 
 | 1017 |     PyObject *item; | 
 | 1018 |     int cmp; | 
 | 1019 |  | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1020 |     while (--n >= 0) { | 
| Raymond Hettinger | 39dadf7 | 2015-03-20 16:38:56 -0700 | [diff] [blame] | 1021 |         CHECK_NOT_END(b); | 
 | 1022 |         item = b->data[index]; | 
 | 1023 |         cmp = PyObject_RichCompareBool(item, v, Py_EQ); | 
 | 1024 |         if (cmp) { | 
 | 1025 |             return cmp; | 
 | 1026 |         } | 
 | 1027 |         if (start_state != deque->state) { | 
 | 1028 |             PyErr_SetString(PyExc_RuntimeError, | 
 | 1029 |                             "deque mutated during iteration"); | 
 | 1030 |             return -1; | 
 | 1031 |         } | 
 | 1032 |         index++; | 
 | 1033 |         if (index == BLOCKLEN) { | 
 | 1034 |             b = b->rightlink; | 
 | 1035 |             index = 0; | 
 | 1036 |         } | 
 | 1037 |     } | 
 | 1038 |     return 0; | 
 | 1039 | } | 
 | 1040 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1041 | static Py_ssize_t | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1042 | deque_len(dequeobject *deque) | 
 | 1043 | { | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1044 |     return Py_SIZE(deque); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1045 | } | 
 | 1046 |  | 
| Raymond Hettinger | 4aec61e | 2005-03-18 21:20:23 +0000 | [diff] [blame] | 1047 | static PyObject * | 
| Serhiy Storchaka | 6969eaf | 2017-07-03 21:20:15 +0300 | [diff] [blame] | 1048 | deque_index(dequeobject *deque, PyObject **args, Py_ssize_t nargs) | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1049 | { | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1050 |     Py_ssize_t i, n, start=0, stop=Py_SIZE(deque); | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1051 |     PyObject *v, *item; | 
 | 1052 |     block *b = deque->leftblock; | 
 | 1053 |     Py_ssize_t index = deque->leftindex; | 
 | 1054 |     size_t start_state = deque->state; | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1055 |     int cmp; | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1056 |  | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1057 |     if (!_PyArg_ParseStack(args, nargs, "O|O&O&:index", &v, | 
| Serhiy Storchaka | d4edfc9 | 2017-03-30 18:29:23 +0300 | [diff] [blame] | 1058 |                            _PyEval_SliceIndexNotNone, &start, | 
 | 1059 |                            _PyEval_SliceIndexNotNone, &stop)) { | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1060 |         return NULL; | 
 | 1061 |     } | 
 | 1062 |  | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1063 |     if (start < 0) { | 
 | 1064 |         start += Py_SIZE(deque); | 
 | 1065 |         if (start < 0) | 
 | 1066 |             start = 0; | 
 | 1067 |     } | 
 | 1068 |     if (stop < 0) { | 
 | 1069 |         stop += Py_SIZE(deque); | 
 | 1070 |         if (stop < 0) | 
 | 1071 |             stop = 0; | 
 | 1072 |     } | 
| Raymond Hettinger | 87674ec | 2015-08-26 08:08:38 -0700 | [diff] [blame] | 1073 |     if (stop > Py_SIZE(deque)) | 
 | 1074 |         stop = Py_SIZE(deque); | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1075 |     if (start > stop) | 
 | 1076 |         start = stop; | 
 | 1077 |     assert(0 <= start && start <= stop && stop <= Py_SIZE(deque)); | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1078 |  | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1079 |     /* XXX Replace this loop with faster code from deque_item() */ | 
 | 1080 |     for (i=0 ; i<start ; i++) { | 
 | 1081 |         index++; | 
 | 1082 |         if (index == BLOCKLEN) { | 
 | 1083 |             b = b->rightlink; | 
 | 1084 |             index = 0; | 
 | 1085 |         } | 
 | 1086 |     } | 
 | 1087 |  | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1088 |     n = stop - i; | 
 | 1089 |     while (--n >= 0) { | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1090 |         CHECK_NOT_END(b); | 
 | 1091 |         item = b->data[index]; | 
 | 1092 |         cmp = PyObject_RichCompareBool(item, v, Py_EQ); | 
 | 1093 |         if (cmp > 0) | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1094 |             return PyLong_FromSsize_t(stop - n - 1); | 
| Raymond Hettinger | df8f5b5 | 2015-11-02 07:27:40 -0500 | [diff] [blame] | 1095 |         if (cmp < 0) | 
| Raymond Hettinger | 67b97b8 | 2015-11-01 23:57:37 -0500 | [diff] [blame] | 1096 |             return NULL; | 
 | 1097 |         if (start_state != deque->state) { | 
 | 1098 |             PyErr_SetString(PyExc_RuntimeError, | 
 | 1099 |                             "deque mutated during iteration"); | 
 | 1100 |             return NULL; | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1101 |         } | 
 | 1102 |         index++; | 
 | 1103 |         if (index == BLOCKLEN) { | 
 | 1104 |             b = b->rightlink; | 
 | 1105 |             index = 0; | 
 | 1106 |         } | 
 | 1107 |     } | 
 | 1108 |     PyErr_Format(PyExc_ValueError, "%R is not in deque", v); | 
 | 1109 |     return NULL; | 
 | 1110 | } | 
 | 1111 |  | 
 | 1112 | PyDoc_STRVAR(index_doc, | 
 | 1113 | "D.index(value, [start, [stop]]) -> integer -- return first index of value.\n" | 
 | 1114 | "Raises ValueError if the value is not present."); | 
 | 1115 |  | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 1116 | /* insert(), remove(), and delitem() are implemented in terms of | 
 | 1117 |    rotate() for simplicity and reasonable performance near the end | 
 | 1118 |    points.  If for some reason these methods become popular, it is not | 
 | 1119 |    hard to re-implement this using direct data movement (similar to | 
 | 1120 |    the code used in list slice assignments) and achieve a performance | 
| Raymond Hettinger | fef9c1b | 2015-03-24 21:12:57 -0700 | [diff] [blame] | 1121 |    boost (by moving each pointer only once instead of twice). | 
| Raymond Hettinger | 551350a | 2015-03-24 00:19:53 -0700 | [diff] [blame] | 1122 | */ | 
 | 1123 |  | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1124 | static PyObject * | 
| Serhiy Storchaka | 6969eaf | 2017-07-03 21:20:15 +0300 | [diff] [blame] | 1125 | deque_insert(dequeobject *deque, PyObject **args, Py_ssize_t nargs) | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1126 | { | 
 | 1127 |     Py_ssize_t index; | 
 | 1128 |     Py_ssize_t n = Py_SIZE(deque); | 
 | 1129 |     PyObject *value; | 
 | 1130 |     PyObject *rv; | 
 | 1131 |  | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1132 |     if (!_PyArg_ParseStack(args, nargs, "nO:insert", &index, &value)) { | 
 | 1133 |         return NULL; | 
 | 1134 |     } | 
 | 1135 |  | 
| Raymond Hettinger | 3743432 | 2016-01-26 21:44:16 -0800 | [diff] [blame] | 1136 |     if (deque->maxlen == Py_SIZE(deque)) { | 
| Raymond Hettinger | a638971 | 2016-02-01 21:21:19 -0800 | [diff] [blame] | 1137 |         PyErr_SetString(PyExc_IndexError, "deque already at its maximum size"); | 
 | 1138 |         return NULL; | 
| Raymond Hettinger | 3743432 | 2016-01-26 21:44:16 -0800 | [diff] [blame] | 1139 |     } | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1140 |     if (index >= n) | 
 | 1141 |         return deque_append(deque, value); | 
 | 1142 |     if (index <= -n || index == 0) | 
 | 1143 |         return deque_appendleft(deque, value); | 
 | 1144 |     if (_deque_rotate(deque, -index)) | 
 | 1145 |         return NULL; | 
 | 1146 |     if (index < 0) | 
 | 1147 |         rv = deque_append(deque, value); | 
 | 1148 |     else | 
 | 1149 |         rv = deque_appendleft(deque, value); | 
 | 1150 |     if (rv == NULL) | 
 | 1151 |         return NULL; | 
 | 1152 |     Py_DECREF(rv); | 
 | 1153 |     if (_deque_rotate(deque, index)) | 
 | 1154 |         return NULL; | 
 | 1155 |     Py_RETURN_NONE; | 
 | 1156 | } | 
 | 1157 |  | 
 | 1158 | PyDoc_STRVAR(insert_doc, | 
 | 1159 | "D.insert(index, object) -- insert object before index"); | 
 | 1160 |  | 
 | 1161 | static PyObject * | 
| Raymond Hettinger | 4aec61e | 2005-03-18 21:20:23 +0000 | [diff] [blame] | 1162 | deque_remove(dequeobject *deque, PyObject *value) | 
 | 1163 | { | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1164 |     Py_ssize_t i, n=Py_SIZE(deque); | 
| Raymond Hettinger | 4aec61e | 2005-03-18 21:20:23 +0000 | [diff] [blame] | 1165 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1166 |     for (i=0 ; i<n ; i++) { | 
 | 1167 |         PyObject *item = deque->leftblock->data[deque->leftindex]; | 
 | 1168 |         int cmp = PyObject_RichCompareBool(item, value, Py_EQ); | 
| Raymond Hettinger | d73202c | 2005-03-19 00:00:51 +0000 | [diff] [blame] | 1169 |  | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1170 |         if (Py_SIZE(deque) != n) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1171 |             PyErr_SetString(PyExc_IndexError, | 
 | 1172 |                 "deque mutated during remove()."); | 
 | 1173 |             return NULL; | 
 | 1174 |         } | 
 | 1175 |         if (cmp > 0) { | 
 | 1176 |             PyObject *tgt = deque_popleft(deque, NULL); | 
 | 1177 |             assert (tgt != NULL); | 
| Raymond Hettinger | 6921c13 | 2015-03-21 02:03:40 -0700 | [diff] [blame] | 1178 |             if (_deque_rotate(deque, i)) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1179 |                 return NULL; | 
| Raymond Hettinger | ac13ad6 | 2015-03-21 01:53:16 -0700 | [diff] [blame] | 1180 |             Py_DECREF(tgt); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1181 |             Py_RETURN_NONE; | 
 | 1182 |         } | 
 | 1183 |         else if (cmp < 0) { | 
 | 1184 |             _deque_rotate(deque, i); | 
 | 1185 |             return NULL; | 
 | 1186 |         } | 
 | 1187 |         _deque_rotate(deque, -1); | 
 | 1188 |     } | 
 | 1189 |     PyErr_SetString(PyExc_ValueError, "deque.remove(x): x not in deque"); | 
 | 1190 |     return NULL; | 
| Raymond Hettinger | 4aec61e | 2005-03-18 21:20:23 +0000 | [diff] [blame] | 1191 | } | 
 | 1192 |  | 
 | 1193 | PyDoc_STRVAR(remove_doc, | 
 | 1194 | "D.remove(value) -- remove first occurrence of value."); | 
 | 1195 |  | 
| Raymond Hettinger | 3c186ba | 2015-03-02 21:45:02 -0800 | [diff] [blame] | 1196 | static int | 
 | 1197 | valid_index(Py_ssize_t i, Py_ssize_t limit) | 
 | 1198 | { | 
| Raymond Hettinger | 12f896c | 2015-07-31 12:03:20 -0700 | [diff] [blame] | 1199 |     /* The cast to size_t lets us use just a single comparison | 
| Raymond Hettinger | 3c186ba | 2015-03-02 21:45:02 -0800 | [diff] [blame] | 1200 |        to check whether i is in the range: 0 <= i < limit */ | 
 | 1201 |     return (size_t) i < (size_t) limit; | 
 | 1202 | } | 
 | 1203 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1204 | static PyObject * | 
| Benjamin Peterson | d631371 | 2008-07-31 16:23:04 +0000 | [diff] [blame] | 1205 | deque_item(dequeobject *deque, Py_ssize_t i) | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1206 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1207 |     block *b; | 
 | 1208 |     PyObject *item; | 
 | 1209 |     Py_ssize_t n, index=i; | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1210 |  | 
| Raymond Hettinger | 3c186ba | 2015-03-02 21:45:02 -0800 | [diff] [blame] | 1211 |     if (!valid_index(i, Py_SIZE(deque))) { | 
 | 1212 |         PyErr_SetString(PyExc_IndexError, "deque index out of range"); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1213 |         return NULL; | 
 | 1214 |     } | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1215 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1216 |     if (i == 0) { | 
 | 1217 |         i = deque->leftindex; | 
 | 1218 |         b = deque->leftblock; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1219 |     } else if (i == Py_SIZE(deque) - 1) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1220 |         i = deque->rightindex; | 
 | 1221 |         b = deque->rightblock; | 
 | 1222 |     } else { | 
 | 1223 |         i += deque->leftindex; | 
| Raymond Hettinger | c208308 | 2015-02-28 23:29:16 -0800 | [diff] [blame] | 1224 |         n = (Py_ssize_t)((size_t) i / BLOCKLEN); | 
 | 1225 |         i = (Py_ssize_t)((size_t) i % BLOCKLEN); | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1226 |         if (index < (Py_SIZE(deque) >> 1)) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1227 |             b = deque->leftblock; | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1228 |             while (--n >= 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1229 |                 b = b->rightlink; | 
 | 1230 |         } else { | 
| Raymond Hettinger | 63d1ff2 | 2015-02-28 07:41:30 -0800 | [diff] [blame] | 1231 |             n = (Py_ssize_t)( | 
| Raymond Hettinger | c208308 | 2015-02-28 23:29:16 -0800 | [diff] [blame] | 1232 |                     ((size_t)(deque->leftindex + Py_SIZE(deque) - 1)) | 
| Raymond Hettinger | 63d1ff2 | 2015-02-28 07:41:30 -0800 | [diff] [blame] | 1233 |                     / BLOCKLEN - n); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1234 |             b = deque->rightblock; | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1235 |             while (--n >= 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1236 |                 b = b->leftlink; | 
 | 1237 |         } | 
 | 1238 |     } | 
 | 1239 |     item = b->data[i]; | 
 | 1240 |     Py_INCREF(item); | 
 | 1241 |     return item; | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1242 | } | 
 | 1243 |  | 
 | 1244 | static int | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1245 | deque_del_item(dequeobject *deque, Py_ssize_t i) | 
| Raymond Hettinger | 0e371f2 | 2004-05-12 20:55:56 +0000 | [diff] [blame] | 1246 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1247 |     PyObject *item; | 
| Raymond Hettinger | ac13ad6 | 2015-03-21 01:53:16 -0700 | [diff] [blame] | 1248 |     int rv; | 
| Raymond Hettinger | 0e371f2 | 2004-05-12 20:55:56 +0000 | [diff] [blame] | 1249 |  | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1250 |     assert (i >= 0 && i < Py_SIZE(deque)); | 
| Raymond Hettinger | 6921c13 | 2015-03-21 02:03:40 -0700 | [diff] [blame] | 1251 |     if (_deque_rotate(deque, -i)) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1252 |         return -1; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1253 |     item = deque_popleft(deque, NULL); | 
| Raymond Hettinger | ac13ad6 | 2015-03-21 01:53:16 -0700 | [diff] [blame] | 1254 |     rv = _deque_rotate(deque, i); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1255 |     assert (item != NULL); | 
 | 1256 |     Py_DECREF(item); | 
| Raymond Hettinger | ac13ad6 | 2015-03-21 01:53:16 -0700 | [diff] [blame] | 1257 |     return rv; | 
| Raymond Hettinger | 0e371f2 | 2004-05-12 20:55:56 +0000 | [diff] [blame] | 1258 | } | 
 | 1259 |  | 
 | 1260 | static int | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1261 | deque_ass_item(dequeobject *deque, Py_ssize_t i, PyObject *v) | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1262 | { | 
| Raymond Hettinger | 3841866 | 2016-02-08 20:34:49 -0800 | [diff] [blame] | 1263 |     PyObject *old_value; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1264 |     block *b; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1265 |     Py_ssize_t n, len=Py_SIZE(deque), halflen=(len+1)>>1, index=i; | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1266 |  | 
| Raymond Hettinger | 3c186ba | 2015-03-02 21:45:02 -0800 | [diff] [blame] | 1267 |     if (!valid_index(i, len)) { | 
 | 1268 |         PyErr_SetString(PyExc_IndexError, "deque index out of range"); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1269 |         return -1; | 
 | 1270 |     } | 
 | 1271 |     if (v == NULL) | 
 | 1272 |         return deque_del_item(deque, i); | 
| Raymond Hettinger | 0e371f2 | 2004-05-12 20:55:56 +0000 | [diff] [blame] | 1273 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1274 |     i += deque->leftindex; | 
| Raymond Hettinger | c208308 | 2015-02-28 23:29:16 -0800 | [diff] [blame] | 1275 |     n = (Py_ssize_t)((size_t) i / BLOCKLEN); | 
 | 1276 |     i = (Py_ssize_t)((size_t) i % BLOCKLEN); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1277 |     if (index <= halflen) { | 
 | 1278 |         b = deque->leftblock; | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1279 |         while (--n >= 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1280 |             b = b->rightlink; | 
 | 1281 |     } else { | 
| Raymond Hettinger | a473b9d | 2015-02-28 17:49:47 -0800 | [diff] [blame] | 1282 |         n = (Py_ssize_t)( | 
| Raymond Hettinger | c208308 | 2015-02-28 23:29:16 -0800 | [diff] [blame] | 1283 |                 ((size_t)(deque->leftindex + Py_SIZE(deque) - 1)) | 
| Raymond Hettinger | a473b9d | 2015-02-28 17:49:47 -0800 | [diff] [blame] | 1284 |                 / BLOCKLEN - n); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1285 |         b = deque->rightblock; | 
| Raymond Hettinger | e1b0287 | 2017-09-04 16:07:06 -0700 | [diff] [blame] | 1286 |         while (--n >= 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1287 |             b = b->leftlink; | 
 | 1288 |     } | 
 | 1289 |     Py_INCREF(v); | 
| Raymond Hettinger | 3841866 | 2016-02-08 20:34:49 -0800 | [diff] [blame] | 1290 |     old_value = b->data[i]; | 
 | 1291 |     b->data[i] = v; | 
 | 1292 |     Py_DECREF(old_value); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1293 |     return 0; | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1294 | } | 
 | 1295 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1296 | static void | 
 | 1297 | deque_dealloc(dequeobject *deque) | 
 | 1298 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1299 |     PyObject_GC_UnTrack(deque); | 
 | 1300 |     if (deque->weakreflist != NULL) | 
 | 1301 |         PyObject_ClearWeakRefs((PyObject *) deque); | 
 | 1302 |     if (deque->leftblock != NULL) { | 
 | 1303 |         deque_clear(deque); | 
 | 1304 |         assert(deque->leftblock != NULL); | 
 | 1305 |         freeblock(deque->leftblock); | 
 | 1306 |     } | 
 | 1307 |     deque->leftblock = NULL; | 
 | 1308 |     deque->rightblock = NULL; | 
 | 1309 |     Py_TYPE(deque)->tp_free(deque); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1310 | } | 
 | 1311 |  | 
 | 1312 | static int | 
| Raymond Hettinger | 0a4977c | 2004-03-01 23:16:22 +0000 | [diff] [blame] | 1313 | deque_traverse(dequeobject *deque, visitproc visit, void *arg) | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1314 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1315 |     block *b; | 
 | 1316 |     PyObject *item; | 
 | 1317 |     Py_ssize_t index; | 
 | 1318 |     Py_ssize_t indexlo = deque->leftindex; | 
| Raymond Hettinger | 1286d14 | 2015-10-14 23:16:57 -0700 | [diff] [blame] | 1319 |     Py_ssize_t indexhigh; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1320 |  | 
| Raymond Hettinger | 5bfa867 | 2013-07-06 11:58:09 -1000 | [diff] [blame] | 1321 |     for (b = deque->leftblock; b != deque->rightblock; b = b->rightlink) { | 
 | 1322 |         for (index = indexlo; index < BLOCKLEN ; index++) { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1323 |             item = b->data[index]; | 
 | 1324 |             Py_VISIT(item); | 
 | 1325 |         } | 
 | 1326 |         indexlo = 0; | 
 | 1327 |     } | 
| Raymond Hettinger | 1286d14 | 2015-10-14 23:16:57 -0700 | [diff] [blame] | 1328 |     indexhigh = deque->rightindex; | 
 | 1329 |     for (index = indexlo; index <= indexhigh; index++) { | 
| Raymond Hettinger | 5bfa867 | 2013-07-06 11:58:09 -1000 | [diff] [blame] | 1330 |         item = b->data[index]; | 
 | 1331 |         Py_VISIT(item); | 
 | 1332 |     } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1333 |     return 0; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1334 | } | 
 | 1335 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1336 | static PyObject * | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1337 | deque_reduce(dequeobject *deque) | 
 | 1338 | { | 
| Serhiy Storchaka | a0d416f | 2016-03-06 08:55:21 +0200 | [diff] [blame] | 1339 |     PyObject *dict, *it; | 
| Martin v. Löwis | bd928fe | 2011-10-14 10:20:37 +0200 | [diff] [blame] | 1340 |     _Py_IDENTIFIER(__dict__); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1341 |  | 
| Martin v. Löwis | 1ee1b6f | 2011-10-10 18:11:30 +0200 | [diff] [blame] | 1342 |     dict = _PyObject_GetAttrId((PyObject *)deque, &PyId___dict__); | 
| Serhiy Storchaka | a0d416f | 2016-03-06 08:55:21 +0200 | [diff] [blame] | 1343 |     if (dict == NULL) { | 
 | 1344 |         if (!PyErr_ExceptionMatches(PyExc_AttributeError)) { | 
 | 1345 |             return NULL; | 
 | 1346 |         } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1347 |         PyErr_Clear(); | 
| Serhiy Storchaka | a0d416f | 2016-03-06 08:55:21 +0200 | [diff] [blame] | 1348 |         dict = Py_None; | 
 | 1349 |         Py_INCREF(dict); | 
 | 1350 |     } | 
 | 1351 |  | 
 | 1352 |     it = PyObject_GetIter((PyObject *)deque); | 
 | 1353 |     if (it == NULL) { | 
 | 1354 |         Py_DECREF(dict); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1355 |         return NULL; | 
 | 1356 |     } | 
| Serhiy Storchaka | a0d416f | 2016-03-06 08:55:21 +0200 | [diff] [blame] | 1357 |  | 
 | 1358 |     if (deque->maxlen < 0) { | 
 | 1359 |         return Py_BuildValue("O()NN", Py_TYPE(deque), dict, it); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1360 |     } | 
| Serhiy Storchaka | a0d416f | 2016-03-06 08:55:21 +0200 | [diff] [blame] | 1361 |     else { | 
 | 1362 |         return Py_BuildValue("O(()n)NN", Py_TYPE(deque), deque->maxlen, dict, it); | 
 | 1363 |     } | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1364 | } | 
 | 1365 |  | 
 | 1366 | PyDoc_STRVAR(reduce_doc, "Return state information for pickling."); | 
 | 1367 |  | 
 | 1368 | static PyObject * | 
 | 1369 | deque_repr(PyObject *deque) | 
 | 1370 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1371 |     PyObject *aslist, *result; | 
 | 1372 |     int i; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1373 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1374 |     i = Py_ReprEnter(deque); | 
 | 1375 |     if (i != 0) { | 
 | 1376 |         if (i < 0) | 
 | 1377 |             return NULL; | 
 | 1378 |         return PyUnicode_FromString("[...]"); | 
 | 1379 |     } | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1380 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1381 |     aslist = PySequence_List(deque); | 
 | 1382 |     if (aslist == NULL) { | 
 | 1383 |         Py_ReprLeave(deque); | 
 | 1384 |         return NULL; | 
 | 1385 |     } | 
| Raymond Hettinger | a7f63009 | 2015-10-10 23:56:02 -0400 | [diff] [blame] | 1386 |     if (((dequeobject *)deque)->maxlen >= 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1387 |         result = PyUnicode_FromFormat("deque(%R, maxlen=%zd)", | 
 | 1388 |                                       aslist, ((dequeobject *)deque)->maxlen); | 
 | 1389 |     else | 
 | 1390 |         result = PyUnicode_FromFormat("deque(%R)", aslist); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1391 |     Py_ReprLeave(deque); | 
| Raymond Hettinger | 2b2b753 | 2015-09-05 17:05:52 -0700 | [diff] [blame] | 1392 |     Py_DECREF(aslist); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1393 |     return result; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1394 | } | 
 | 1395 |  | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1396 | static PyObject * | 
 | 1397 | deque_richcompare(PyObject *v, PyObject *w, int op) | 
 | 1398 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1399 |     PyObject *it1=NULL, *it2=NULL, *x, *y; | 
 | 1400 |     Py_ssize_t vs, ws; | 
 | 1401 |     int b, cmp=-1; | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1402 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1403 |     if (!PyObject_TypeCheck(v, &deque_type) || | 
 | 1404 |         !PyObject_TypeCheck(w, &deque_type)) { | 
| Brian Curtin | dfc80e3 | 2011-08-10 20:28:54 -0500 | [diff] [blame] | 1405 |         Py_RETURN_NOTIMPLEMENTED; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1406 |     } | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1407 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1408 |     /* Shortcuts */ | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1409 |     vs = Py_SIZE((dequeobject *)v); | 
 | 1410 |     ws = Py_SIZE((dequeobject *)w); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1411 |     if (op == Py_EQ) { | 
 | 1412 |         if (v == w) | 
 | 1413 |             Py_RETURN_TRUE; | 
 | 1414 |         if (vs != ws) | 
 | 1415 |             Py_RETURN_FALSE; | 
 | 1416 |     } | 
 | 1417 |     if (op == Py_NE) { | 
 | 1418 |         if (v == w) | 
 | 1419 |             Py_RETURN_FALSE; | 
 | 1420 |         if (vs != ws) | 
 | 1421 |             Py_RETURN_TRUE; | 
 | 1422 |     } | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1423 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1424 |     /* Search for the first index where items are different */ | 
 | 1425 |     it1 = PyObject_GetIter(v); | 
 | 1426 |     if (it1 == NULL) | 
 | 1427 |         goto done; | 
 | 1428 |     it2 = PyObject_GetIter(w); | 
 | 1429 |     if (it2 == NULL) | 
 | 1430 |         goto done; | 
 | 1431 |     for (;;) { | 
 | 1432 |         x = PyIter_Next(it1); | 
 | 1433 |         if (x == NULL && PyErr_Occurred()) | 
 | 1434 |             goto done; | 
 | 1435 |         y = PyIter_Next(it2); | 
 | 1436 |         if (x == NULL || y == NULL) | 
 | 1437 |             break; | 
 | 1438 |         b = PyObject_RichCompareBool(x, y, Py_EQ); | 
 | 1439 |         if (b == 0) { | 
 | 1440 |             cmp = PyObject_RichCompareBool(x, y, op); | 
 | 1441 |             Py_DECREF(x); | 
 | 1442 |             Py_DECREF(y); | 
 | 1443 |             goto done; | 
 | 1444 |         } | 
 | 1445 |         Py_DECREF(x); | 
 | 1446 |         Py_DECREF(y); | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 1447 |         if (b < 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1448 |             goto done; | 
 | 1449 |     } | 
 | 1450 |     /* We reached the end of one deque or both */ | 
 | 1451 |     Py_XDECREF(x); | 
 | 1452 |     Py_XDECREF(y); | 
 | 1453 |     if (PyErr_Occurred()) | 
 | 1454 |         goto done; | 
 | 1455 |     switch (op) { | 
 | 1456 |     case Py_LT: cmp = y != NULL; break;  /* if w was longer */ | 
 | 1457 |     case Py_LE: cmp = x == NULL; break;  /* if v was not longer */ | 
 | 1458 |     case Py_EQ: cmp = x == y;    break;  /* if we reached the end of both */ | 
 | 1459 |     case Py_NE: cmp = x != y;    break;  /* if one deque continues */ | 
 | 1460 |     case Py_GT: cmp = x != NULL; break;  /* if v was longer */ | 
 | 1461 |     case Py_GE: cmp = y == NULL; break;  /* if w was not longer */ | 
 | 1462 |     } | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 1463 |  | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1464 | done: | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1465 |     Py_XDECREF(it1); | 
 | 1466 |     Py_XDECREF(it2); | 
 | 1467 |     if (cmp == 1) | 
 | 1468 |         Py_RETURN_TRUE; | 
 | 1469 |     if (cmp == 0) | 
 | 1470 |         Py_RETURN_FALSE; | 
 | 1471 |     return NULL; | 
| Raymond Hettinger | 738ec90 | 2004-02-29 02:15:56 +0000 | [diff] [blame] | 1472 | } | 
 | 1473 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1474 | static int | 
| Guido van Rossum | 8ce8a78 | 2007-11-01 19:42:39 +0000 | [diff] [blame] | 1475 | deque_init(dequeobject *deque, PyObject *args, PyObject *kwdargs) | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1476 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1477 |     PyObject *iterable = NULL; | 
 | 1478 |     PyObject *maxlenobj = NULL; | 
 | 1479 |     Py_ssize_t maxlen = -1; | 
 | 1480 |     char *kwlist[] = {"iterable", "maxlen", 0}; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1481 |  | 
| Raymond Hettinger | 0d30940 | 2015-09-30 23:15:02 -0700 | [diff] [blame] | 1482 |     if (kwdargs == NULL) { | 
 | 1483 |         if (!PyArg_UnpackTuple(args, "deque()", 0, 2, &iterable, &maxlenobj)) | 
 | 1484 |             return -1; | 
 | 1485 |     } else { | 
 | 1486 |         if (!PyArg_ParseTupleAndKeywords(args, kwdargs, "|OO:deque", kwlist, | 
 | 1487 |                                          &iterable, &maxlenobj)) | 
 | 1488 |             return -1; | 
 | 1489 |     } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1490 |     if (maxlenobj != NULL && maxlenobj != Py_None) { | 
 | 1491 |         maxlen = PyLong_AsSsize_t(maxlenobj); | 
 | 1492 |         if (maxlen == -1 && PyErr_Occurred()) | 
 | 1493 |             return -1; | 
 | 1494 |         if (maxlen < 0) { | 
 | 1495 |             PyErr_SetString(PyExc_ValueError, "maxlen must be non-negative"); | 
 | 1496 |             return -1; | 
 | 1497 |         } | 
 | 1498 |     } | 
 | 1499 |     deque->maxlen = maxlen; | 
| Raymond Hettinger | 0d30940 | 2015-09-30 23:15:02 -0700 | [diff] [blame] | 1500 |     if (Py_SIZE(deque) > 0) | 
 | 1501 |         deque_clear(deque); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1502 |     if (iterable != NULL) { | 
 | 1503 |         PyObject *rv = deque_extend(deque, iterable); | 
 | 1504 |         if (rv == NULL) | 
 | 1505 |             return -1; | 
 | 1506 |         Py_DECREF(rv); | 
 | 1507 |     } | 
 | 1508 |     return 0; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1509 | } | 
 | 1510 |  | 
| Raymond Hettinger | 5bb0f0e | 2009-03-10 12:56:32 +0000 | [diff] [blame] | 1511 | static PyObject * | 
| Jesus Cea | 16e2fca | 2012-08-03 14:49:42 +0200 | [diff] [blame] | 1512 | deque_sizeof(dequeobject *deque, void *unused) | 
 | 1513 | { | 
 | 1514 |     Py_ssize_t res; | 
 | 1515 |     Py_ssize_t blocks; | 
 | 1516 |  | 
| Serhiy Storchaka | 5c4064e | 2015-12-19 20:05:25 +0200 | [diff] [blame] | 1517 |     res = _PyObject_SIZE(Py_TYPE(deque)); | 
| Raymond Hettinger | bc00341 | 2015-10-14 23:33:23 -0700 | [diff] [blame] | 1518 |     blocks = (size_t)(deque->leftindex + Py_SIZE(deque) + BLOCKLEN - 1) / BLOCKLEN; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1519 |     assert(deque->leftindex + Py_SIZE(deque) - 1 == | 
| Jesus Cea | 16e2fca | 2012-08-03 14:49:42 +0200 | [diff] [blame] | 1520 |            (blocks - 1) * BLOCKLEN + deque->rightindex); | 
 | 1521 |     res += blocks * sizeof(block); | 
 | 1522 |     return PyLong_FromSsize_t(res); | 
 | 1523 | } | 
 | 1524 |  | 
 | 1525 | PyDoc_STRVAR(sizeof_doc, | 
 | 1526 | "D.__sizeof__() -- size of D in memory, in bytes"); | 
 | 1527 |  | 
| Raymond Hettinger | 0f1451c | 2015-03-23 23:23:55 -0700 | [diff] [blame] | 1528 | static int | 
 | 1529 | deque_bool(dequeobject *deque) | 
 | 1530 | { | 
 | 1531 |     return Py_SIZE(deque) != 0; | 
 | 1532 | } | 
 | 1533 |  | 
| Jesus Cea | 16e2fca | 2012-08-03 14:49:42 +0200 | [diff] [blame] | 1534 | static PyObject * | 
| Raymond Hettinger | 5bb0f0e | 2009-03-10 12:56:32 +0000 | [diff] [blame] | 1535 | deque_get_maxlen(dequeobject *deque) | 
 | 1536 | { | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 1537 |     if (deque->maxlen < 0) | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1538 |         Py_RETURN_NONE; | 
 | 1539 |     return PyLong_FromSsize_t(deque->maxlen); | 
| Raymond Hettinger | 5bb0f0e | 2009-03-10 12:56:32 +0000 | [diff] [blame] | 1540 | } | 
 | 1541 |  | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 1542 |  | 
 | 1543 | /* deque object ********************************************************/ | 
 | 1544 |  | 
| Raymond Hettinger | 5bb0f0e | 2009-03-10 12:56:32 +0000 | [diff] [blame] | 1545 | static PyGetSetDef deque_getset[] = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1546 |     {"maxlen", (getter)deque_get_maxlen, (setter)NULL, | 
 | 1547 |      "maximum size of a deque or None if unbounded"}, | 
 | 1548 |     {0} | 
| Raymond Hettinger | 5bb0f0e | 2009-03-10 12:56:32 +0000 | [diff] [blame] | 1549 | }; | 
 | 1550 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1551 | static PySequenceMethods deque_as_sequence = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1552 |     (lenfunc)deque_len,                 /* sq_length */ | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 1553 |     (binaryfunc)deque_concat,           /* sq_concat */ | 
 | 1554 |     (ssizeargfunc)deque_repeat,         /* sq_repeat */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1555 |     (ssizeargfunc)deque_item,           /* sq_item */ | 
 | 1556 |     0,                                  /* sq_slice */ | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1557 |     (ssizeobjargproc)deque_ass_item,    /* sq_ass_item */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1558 |     0,                                  /* sq_ass_slice */ | 
| Raymond Hettinger | 39dadf7 | 2015-03-20 16:38:56 -0700 | [diff] [blame] | 1559 |     (objobjproc)deque_contains,         /* sq_contains */ | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1560 |     (binaryfunc)deque_inplace_concat,   /* sq_inplace_concat */ | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 1561 |     (ssizeargfunc)deque_inplace_repeat, /* sq_inplace_repeat */ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1562 | }; | 
 | 1563 |  | 
| Raymond Hettinger | 0f1451c | 2015-03-23 23:23:55 -0700 | [diff] [blame] | 1564 | static PyNumberMethods deque_as_number = { | 
 | 1565 |     0,                                  /* nb_add */ | 
 | 1566 |     0,                                  /* nb_subtract */ | 
 | 1567 |     0,                                  /* nb_multiply */ | 
 | 1568 |     0,                                  /* nb_remainder */ | 
 | 1569 |     0,                                  /* nb_divmod */ | 
 | 1570 |     0,                                  /* nb_power */ | 
 | 1571 |     0,                                  /* nb_negative */ | 
 | 1572 |     0,                                  /* nb_positive */ | 
 | 1573 |     0,                                  /* nb_absolute */ | 
 | 1574 |     (inquiry)deque_bool,                /* nb_bool */ | 
 | 1575 |     0,                                  /* nb_invert */ | 
| Raymond Hettinger | 0f1451c | 2015-03-23 23:23:55 -0700 | [diff] [blame] | 1576 |  }; | 
 | 1577 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1578 | static PyObject *deque_iter(dequeobject *deque); | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1579 | static PyObject *deque_reviter(dequeobject *deque); | 
| Tim Peters | 1065f75 | 2004-10-01 01:03:29 +0000 | [diff] [blame] | 1580 | PyDoc_STRVAR(reversed_doc, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1581 |     "D.__reversed__() -- return a reverse iterator over the deque"); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1582 |  | 
 | 1583 | static PyMethodDef deque_methods[] = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1584 |     {"append",                  (PyCFunction)deque_append, | 
 | 1585 |         METH_O,                  append_doc}, | 
 | 1586 |     {"appendleft",              (PyCFunction)deque_appendleft, | 
 | 1587 |         METH_O,                  appendleft_doc}, | 
 | 1588 |     {"clear",                   (PyCFunction)deque_clearmethod, | 
 | 1589 |         METH_NOARGS,             clear_doc}, | 
 | 1590 |     {"__copy__",                (PyCFunction)deque_copy, | 
 | 1591 |         METH_NOARGS,             copy_doc}, | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1592 |     {"copy",                    (PyCFunction)deque_copy, | 
 | 1593 |         METH_NOARGS,             copy_doc}, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1594 |     {"count",                   (PyCFunction)deque_count, | 
| Raymond Hettinger | 0f6f947 | 2015-03-21 01:42:10 -0700 | [diff] [blame] | 1595 |         METH_O,                  count_doc}, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1596 |     {"extend",                  (PyCFunction)deque_extend, | 
 | 1597 |         METH_O,                  extend_doc}, | 
 | 1598 |     {"extendleft",              (PyCFunction)deque_extendleft, | 
 | 1599 |         METH_O,                  extendleft_doc}, | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1600 |     {"index",                   (PyCFunction)deque_index, | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1601 |         METH_FASTCALL,            index_doc}, | 
| Raymond Hettinger | 32ea165 | 2015-03-21 01:37:37 -0700 | [diff] [blame] | 1602 |     {"insert",                  (PyCFunction)deque_insert, | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1603 |         METH_FASTCALL,            insert_doc}, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1604 |     {"pop",                     (PyCFunction)deque_pop, | 
 | 1605 |         METH_NOARGS,             pop_doc}, | 
 | 1606 |     {"popleft",                 (PyCFunction)deque_popleft, | 
 | 1607 |         METH_NOARGS,             popleft_doc}, | 
| Raymond Hettinger | 0f6f947 | 2015-03-21 01:42:10 -0700 | [diff] [blame] | 1608 |     {"__reduce__",              (PyCFunction)deque_reduce, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1609 |         METH_NOARGS,             reduce_doc}, | 
 | 1610 |     {"remove",                  (PyCFunction)deque_remove, | 
 | 1611 |         METH_O,                  remove_doc}, | 
 | 1612 |     {"__reversed__",            (PyCFunction)deque_reviter, | 
 | 1613 |         METH_NOARGS,             reversed_doc}, | 
 | 1614 |     {"reverse",                 (PyCFunction)deque_reverse, | 
 | 1615 |         METH_NOARGS,             reverse_doc}, | 
 | 1616 |     {"rotate",                  (PyCFunction)deque_rotate, | 
| Victor Stinner | dd407d5 | 2017-02-06 16:06:49 +0100 | [diff] [blame] | 1617 |         METH_FASTCALL,            rotate_doc}, | 
| Jesus Cea | 16e2fca | 2012-08-03 14:49:42 +0200 | [diff] [blame] | 1618 |     {"__sizeof__",              (PyCFunction)deque_sizeof, | 
 | 1619 |         METH_NOARGS,             sizeof_doc}, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1620 |     {NULL,              NULL}   /* sentinel */ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1621 | }; | 
 | 1622 |  | 
 | 1623 | PyDoc_STRVAR(deque_doc, | 
| Andrew Svetlov | 6a5c7c3 | 2012-10-31 11:50:40 +0200 | [diff] [blame] | 1624 | "deque([iterable[, maxlen]]) --> deque object\n\ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1625 | \n\ | 
| Raymond Hettinger | 41290a6 | 2015-03-31 08:12:23 -0700 | [diff] [blame] | 1626 | A list-like sequence optimized for data accesses near its endpoints."); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1627 |  | 
| Neal Norwitz | 87f1013 | 2004-02-29 15:40:53 +0000 | [diff] [blame] | 1628 | static PyTypeObject deque_type = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1629 |     PyVarObject_HEAD_INIT(NULL, 0) | 
 | 1630 |     "collections.deque",                /* tp_name */ | 
 | 1631 |     sizeof(dequeobject),                /* tp_basicsize */ | 
 | 1632 |     0,                                  /* tp_itemsize */ | 
 | 1633 |     /* methods */ | 
 | 1634 |     (destructor)deque_dealloc,          /* tp_dealloc */ | 
 | 1635 |     0,                                  /* tp_print */ | 
 | 1636 |     0,                                  /* tp_getattr */ | 
 | 1637 |     0,                                  /* tp_setattr */ | 
 | 1638 |     0,                                  /* tp_reserved */ | 
 | 1639 |     deque_repr,                         /* tp_repr */ | 
| Raymond Hettinger | 0f1451c | 2015-03-23 23:23:55 -0700 | [diff] [blame] | 1640 |     &deque_as_number,                   /* tp_as_number */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1641 |     &deque_as_sequence,                 /* tp_as_sequence */ | 
 | 1642 |     0,                                  /* tp_as_mapping */ | 
| Georg Brandl | f038b32 | 2010-10-18 07:35:09 +0000 | [diff] [blame] | 1643 |     PyObject_HashNotImplemented,        /* tp_hash */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1644 |     0,                                  /* tp_call */ | 
 | 1645 |     0,                                  /* tp_str */ | 
 | 1646 |     PyObject_GenericGetAttr,            /* tp_getattro */ | 
 | 1647 |     0,                                  /* tp_setattro */ | 
 | 1648 |     0,                                  /* tp_as_buffer */ | 
 | 1649 |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, | 
| Georg Brandl | f038b32 | 2010-10-18 07:35:09 +0000 | [diff] [blame] | 1650 |                                         /* tp_flags */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1651 |     deque_doc,                          /* tp_doc */ | 
 | 1652 |     (traverseproc)deque_traverse,       /* tp_traverse */ | 
 | 1653 |     (inquiry)deque_clear,               /* tp_clear */ | 
 | 1654 |     (richcmpfunc)deque_richcompare,     /* tp_richcompare */ | 
| Georg Brandl | f038b32 | 2010-10-18 07:35:09 +0000 | [diff] [blame] | 1655 |     offsetof(dequeobject, weakreflist), /* tp_weaklistoffset*/ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1656 |     (getiterfunc)deque_iter,            /* tp_iter */ | 
 | 1657 |     0,                                  /* tp_iternext */ | 
 | 1658 |     deque_methods,                      /* tp_methods */ | 
 | 1659 |     0,                                  /* tp_members */ | 
| Georg Brandl | f038b32 | 2010-10-18 07:35:09 +0000 | [diff] [blame] | 1660 |     deque_getset,                       /* tp_getset */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1661 |     0,                                  /* tp_base */ | 
 | 1662 |     0,                                  /* tp_dict */ | 
 | 1663 |     0,                                  /* tp_descr_get */ | 
 | 1664 |     0,                                  /* tp_descr_set */ | 
 | 1665 |     0,                                  /* tp_dictoffset */ | 
 | 1666 |     (initproc)deque_init,               /* tp_init */ | 
 | 1667 |     PyType_GenericAlloc,                /* tp_alloc */ | 
 | 1668 |     deque_new,                          /* tp_new */ | 
 | 1669 |     PyObject_GC_Del,                    /* tp_free */ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1670 | }; | 
 | 1671 |  | 
 | 1672 | /*********************** Deque Iterator **************************/ | 
 | 1673 |  | 
 | 1674 | typedef struct { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1675 |     PyObject_HEAD | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1676 |     block *b; | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1677 |     Py_ssize_t index; | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1678 |     dequeobject *deque; | 
| Raymond Hettinger | f9d9c79 | 2015-03-02 22:47:46 -0800 | [diff] [blame] | 1679 |     size_t state;          /* state when the iterator is created */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1680 |     Py_ssize_t counter;    /* number of items remaining for iteration */ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1681 | } dequeiterobject; | 
 | 1682 |  | 
| Martin v. Löwis | 59683e8 | 2008-06-13 07:50:45 +0000 | [diff] [blame] | 1683 | static PyTypeObject dequeiter_type; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1684 |  | 
 | 1685 | static PyObject * | 
 | 1686 | deque_iter(dequeobject *deque) | 
 | 1687 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1688 |     dequeiterobject *it; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1689 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1690 |     it = PyObject_GC_New(dequeiterobject, &dequeiter_type); | 
 | 1691 |     if (it == NULL) | 
 | 1692 |         return NULL; | 
 | 1693 |     it->b = deque->leftblock; | 
 | 1694 |     it->index = deque->leftindex; | 
 | 1695 |     Py_INCREF(deque); | 
 | 1696 |     it->deque = deque; | 
 | 1697 |     it->state = deque->state; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1698 |     it->counter = Py_SIZE(deque); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1699 |     PyObject_GC_Track(it); | 
 | 1700 |     return (PyObject *)it; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1701 | } | 
 | 1702 |  | 
| Antoine Pitrou | 7ddda78 | 2009-01-01 15:35:33 +0000 | [diff] [blame] | 1703 | static int | 
 | 1704 | dequeiter_traverse(dequeiterobject *dio, visitproc visit, void *arg) | 
 | 1705 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1706 |     Py_VISIT(dio->deque); | 
 | 1707 |     return 0; | 
| Antoine Pitrou | 7ddda78 | 2009-01-01 15:35:33 +0000 | [diff] [blame] | 1708 | } | 
 | 1709 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1710 | static void | 
 | 1711 | dequeiter_dealloc(dequeiterobject *dio) | 
 | 1712 | { | 
| INADA Naoki | a6296d3 | 2017-08-24 14:55:17 +0900 | [diff] [blame] | 1713 |     /* bpo-31095: UnTrack is needed before calling any callbacks */ | 
 | 1714 |     PyObject_GC_UnTrack(dio); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1715 |     Py_XDECREF(dio->deque); | 
 | 1716 |     PyObject_GC_Del(dio); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1717 | } | 
 | 1718 |  | 
 | 1719 | static PyObject * | 
 | 1720 | dequeiter_next(dequeiterobject *it) | 
 | 1721 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1722 |     PyObject *item; | 
| Raymond Hettinger | d1b3d88 | 2004-10-02 00:43:13 +0000 | [diff] [blame] | 1723 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1724 |     if (it->deque->state != it->state) { | 
 | 1725 |         it->counter = 0; | 
 | 1726 |         PyErr_SetString(PyExc_RuntimeError, | 
 | 1727 |                         "deque mutated during iteration"); | 
 | 1728 |         return NULL; | 
 | 1729 |     } | 
 | 1730 |     if (it->counter == 0) | 
 | 1731 |         return NULL; | 
 | 1732 |     assert (!(it->b == it->deque->rightblock && | 
 | 1733 |               it->index > it->deque->rightindex)); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1734 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1735 |     item = it->b->data[it->index]; | 
 | 1736 |     it->index++; | 
 | 1737 |     it->counter--; | 
 | 1738 |     if (it->index == BLOCKLEN && it->counter > 0) { | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 1739 |         CHECK_NOT_END(it->b->rightlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1740 |         it->b = it->b->rightlink; | 
 | 1741 |         it->index = 0; | 
 | 1742 |     } | 
 | 1743 |     Py_INCREF(item); | 
 | 1744 |     return item; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1745 | } | 
 | 1746 |  | 
| Raymond Hettinger | 6b27cda | 2005-09-24 21:23:05 +0000 | [diff] [blame] | 1747 | static PyObject * | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1748 | dequeiter_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | 
 | 1749 | { | 
 | 1750 |     Py_ssize_t i, index=0; | 
 | 1751 |     PyObject *deque; | 
 | 1752 |     dequeiterobject *it; | 
 | 1753 |     if (!PyArg_ParseTuple(args, "O!|n", &deque_type, &deque, &index)) | 
 | 1754 |         return NULL; | 
 | 1755 |     assert(type == &dequeiter_type); | 
 | 1756 |  | 
 | 1757 |     it = (dequeiterobject*)deque_iter((dequeobject *)deque); | 
 | 1758 |     if (!it) | 
 | 1759 |         return NULL; | 
 | 1760 |     /* consume items from the queue */ | 
 | 1761 |     for(i=0; i<index; i++) { | 
 | 1762 |         PyObject *item = dequeiter_next(it); | 
 | 1763 |         if (item) { | 
 | 1764 |             Py_DECREF(item); | 
 | 1765 |         } else { | 
 | 1766 |             if (it->counter) { | 
 | 1767 |                 Py_DECREF(it); | 
 | 1768 |                 return NULL; | 
 | 1769 |             } else | 
 | 1770 |                 break; | 
 | 1771 |         } | 
 | 1772 |     } | 
 | 1773 |     return (PyObject*)it; | 
 | 1774 | } | 
 | 1775 |  | 
 | 1776 | static PyObject * | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1777 | dequeiter_len(dequeiterobject *it) | 
 | 1778 | { | 
| Antoine Pitrou | 554f334 | 2010-08-17 18:30:06 +0000 | [diff] [blame] | 1779 |     return PyLong_FromSsize_t(it->counter); | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1780 | } | 
 | 1781 |  | 
| Armin Rigo | f5b3e36 | 2006-02-11 21:32:43 +0000 | [diff] [blame] | 1782 | PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it))."); | 
| Raymond Hettinger | 6b27cda | 2005-09-24 21:23:05 +0000 | [diff] [blame] | 1783 |  | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1784 | static PyObject * | 
 | 1785 | dequeiter_reduce(dequeiterobject *it) | 
 | 1786 | { | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1787 |     return Py_BuildValue("O(On)", Py_TYPE(it), it->deque, Py_SIZE(it->deque) - it->counter); | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1788 | } | 
 | 1789 |  | 
| Raymond Hettinger | 6b27cda | 2005-09-24 21:23:05 +0000 | [diff] [blame] | 1790 | static PyMethodDef dequeiter_methods[] = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1791 |     {"__length_hint__", (PyCFunction)dequeiter_len, METH_NOARGS, length_hint_doc}, | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1792 |     {"__reduce__", (PyCFunction)dequeiter_reduce, METH_NOARGS, reduce_doc}, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1793 |     {NULL,              NULL}           /* sentinel */ | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1794 | }; | 
 | 1795 |  | 
| Martin v. Löwis | 59683e8 | 2008-06-13 07:50:45 +0000 | [diff] [blame] | 1796 | static PyTypeObject dequeiter_type = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1797 |     PyVarObject_HEAD_INIT(NULL, 0) | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1798 |     "_collections._deque_iterator",             /* tp_name */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1799 |     sizeof(dequeiterobject),                    /* tp_basicsize */ | 
 | 1800 |     0,                                          /* tp_itemsize */ | 
 | 1801 |     /* methods */ | 
 | 1802 |     (destructor)dequeiter_dealloc,              /* tp_dealloc */ | 
 | 1803 |     0,                                          /* tp_print */ | 
 | 1804 |     0,                                          /* tp_getattr */ | 
 | 1805 |     0,                                          /* tp_setattr */ | 
 | 1806 |     0,                                          /* tp_reserved */ | 
 | 1807 |     0,                                          /* tp_repr */ | 
 | 1808 |     0,                                          /* tp_as_number */ | 
 | 1809 |     0,                                          /* tp_as_sequence */ | 
 | 1810 |     0,                                          /* tp_as_mapping */ | 
 | 1811 |     0,                                          /* tp_hash */ | 
 | 1812 |     0,                                          /* tp_call */ | 
 | 1813 |     0,                                          /* tp_str */ | 
 | 1814 |     PyObject_GenericGetAttr,                    /* tp_getattro */ | 
 | 1815 |     0,                                          /* tp_setattro */ | 
 | 1816 |     0,                                          /* tp_as_buffer */ | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1817 |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,    /* tp_flags */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1818 |     0,                                          /* tp_doc */ | 
 | 1819 |     (traverseproc)dequeiter_traverse,           /* tp_traverse */ | 
 | 1820 |     0,                                          /* tp_clear */ | 
 | 1821 |     0,                                          /* tp_richcompare */ | 
 | 1822 |     0,                                          /* tp_weaklistoffset */ | 
 | 1823 |     PyObject_SelfIter,                          /* tp_iter */ | 
 | 1824 |     (iternextfunc)dequeiter_next,               /* tp_iternext */ | 
 | 1825 |     dequeiter_methods,                          /* tp_methods */ | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1826 |     0,                                          /* tp_members */ | 
 | 1827 |     0,                                          /* tp_getset */ | 
 | 1828 |     0,                                          /* tp_base */ | 
 | 1829 |     0,                                          /* tp_dict */ | 
 | 1830 |     0,                                          /* tp_descr_get */ | 
 | 1831 |     0,                                          /* tp_descr_set */ | 
 | 1832 |     0,                                          /* tp_dictoffset */ | 
 | 1833 |     0,                                          /* tp_init */ | 
 | 1834 |     0,                                          /* tp_alloc */ | 
 | 1835 |     dequeiter_new,                              /* tp_new */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1836 |     0, | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 1837 | }; | 
 | 1838 |  | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1839 | /*********************** Deque Reverse Iterator **************************/ | 
 | 1840 |  | 
| Martin v. Löwis | 59683e8 | 2008-06-13 07:50:45 +0000 | [diff] [blame] | 1841 | static PyTypeObject dequereviter_type; | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1842 |  | 
 | 1843 | static PyObject * | 
 | 1844 | deque_reviter(dequeobject *deque) | 
 | 1845 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1846 |     dequeiterobject *it; | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1847 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1848 |     it = PyObject_GC_New(dequeiterobject, &dequereviter_type); | 
 | 1849 |     if (it == NULL) | 
 | 1850 |         return NULL; | 
 | 1851 |     it->b = deque->rightblock; | 
 | 1852 |     it->index = deque->rightindex; | 
 | 1853 |     Py_INCREF(deque); | 
 | 1854 |     it->deque = deque; | 
 | 1855 |     it->state = deque->state; | 
| Raymond Hettinger | df715ba | 2013-07-06 13:01:13 -1000 | [diff] [blame] | 1856 |     it->counter = Py_SIZE(deque); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1857 |     PyObject_GC_Track(it); | 
 | 1858 |     return (PyObject *)it; | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1859 | } | 
 | 1860 |  | 
 | 1861 | static PyObject * | 
 | 1862 | dequereviter_next(dequeiterobject *it) | 
 | 1863 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1864 |     PyObject *item; | 
 | 1865 |     if (it->counter == 0) | 
 | 1866 |         return NULL; | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1867 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1868 |     if (it->deque->state != it->state) { | 
 | 1869 |         it->counter = 0; | 
 | 1870 |         PyErr_SetString(PyExc_RuntimeError, | 
 | 1871 |                         "deque mutated during iteration"); | 
 | 1872 |         return NULL; | 
 | 1873 |     } | 
 | 1874 |     assert (!(it->b == it->deque->leftblock && | 
 | 1875 |               it->index < it->deque->leftindex)); | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1876 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1877 |     item = it->b->data[it->index]; | 
 | 1878 |     it->index--; | 
 | 1879 |     it->counter--; | 
| Raymond Hettinger | d3d2b2c | 2015-09-21 23:41:56 -0700 | [diff] [blame] | 1880 |     if (it->index < 0 && it->counter > 0) { | 
| Raymond Hettinger | 82df925 | 2013-07-07 01:43:42 -1000 | [diff] [blame] | 1881 |         CHECK_NOT_END(it->b->leftlink); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1882 |         it->b = it->b->leftlink; | 
 | 1883 |         it->index = BLOCKLEN - 1; | 
 | 1884 |     } | 
 | 1885 |     Py_INCREF(item); | 
 | 1886 |     return item; | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1887 | } | 
 | 1888 |  | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1889 | static PyObject * | 
 | 1890 | dequereviter_new(PyTypeObject *type, PyObject *args, PyObject *kwds) | 
 | 1891 | { | 
 | 1892 |     Py_ssize_t i, index=0; | 
 | 1893 |     PyObject *deque; | 
 | 1894 |     dequeiterobject *it; | 
 | 1895 |     if (!PyArg_ParseTuple(args, "O!|n", &deque_type, &deque, &index)) | 
 | 1896 |         return NULL; | 
 | 1897 |     assert(type == &dequereviter_type); | 
 | 1898 |  | 
 | 1899 |     it = (dequeiterobject*)deque_reviter((dequeobject *)deque); | 
 | 1900 |     if (!it) | 
 | 1901 |         return NULL; | 
 | 1902 |     /* consume items from the queue */ | 
 | 1903 |     for(i=0; i<index; i++) { | 
 | 1904 |         PyObject *item = dequereviter_next(it); | 
 | 1905 |         if (item) { | 
 | 1906 |             Py_DECREF(item); | 
 | 1907 |         } else { | 
 | 1908 |             if (it->counter) { | 
 | 1909 |                 Py_DECREF(it); | 
 | 1910 |                 return NULL; | 
 | 1911 |             } else | 
 | 1912 |                 break; | 
 | 1913 |         } | 
 | 1914 |     } | 
 | 1915 |     return (PyObject*)it; | 
 | 1916 | } | 
 | 1917 |  | 
| Martin v. Löwis | 59683e8 | 2008-06-13 07:50:45 +0000 | [diff] [blame] | 1918 | static PyTypeObject dequereviter_type = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1919 |     PyVarObject_HEAD_INIT(NULL, 0) | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1920 |     "_collections._deque_reverse_iterator",     /* tp_name */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1921 |     sizeof(dequeiterobject),                    /* tp_basicsize */ | 
 | 1922 |     0,                                          /* tp_itemsize */ | 
 | 1923 |     /* methods */ | 
 | 1924 |     (destructor)dequeiter_dealloc,              /* tp_dealloc */ | 
 | 1925 |     0,                                          /* tp_print */ | 
 | 1926 |     0,                                          /* tp_getattr */ | 
 | 1927 |     0,                                          /* tp_setattr */ | 
 | 1928 |     0,                                          /* tp_reserved */ | 
 | 1929 |     0,                                          /* tp_repr */ | 
 | 1930 |     0,                                          /* tp_as_number */ | 
 | 1931 |     0,                                          /* tp_as_sequence */ | 
 | 1932 |     0,                                          /* tp_as_mapping */ | 
 | 1933 |     0,                                          /* tp_hash */ | 
 | 1934 |     0,                                          /* tp_call */ | 
 | 1935 |     0,                                          /* tp_str */ | 
 | 1936 |     PyObject_GenericGetAttr,                    /* tp_getattro */ | 
 | 1937 |     0,                                          /* tp_setattro */ | 
 | 1938 |     0,                                          /* tp_as_buffer */ | 
| Raymond Hettinger | 87e6912 | 2015-03-02 23:32:02 -0800 | [diff] [blame] | 1939 |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,    /* tp_flags */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1940 |     0,                                          /* tp_doc */ | 
 | 1941 |     (traverseproc)dequeiter_traverse,           /* tp_traverse */ | 
 | 1942 |     0,                                          /* tp_clear */ | 
 | 1943 |     0,                                          /* tp_richcompare */ | 
 | 1944 |     0,                                          /* tp_weaklistoffset */ | 
 | 1945 |     PyObject_SelfIter,                          /* tp_iter */ | 
 | 1946 |     (iternextfunc)dequereviter_next,            /* tp_iternext */ | 
 | 1947 |     dequeiter_methods,                          /* tp_methods */ | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 1948 |     0,                                          /* tp_members */ | 
 | 1949 |     0,                                          /* tp_getset */ | 
 | 1950 |     0,                                          /* tp_base */ | 
 | 1951 |     0,                                          /* tp_dict */ | 
 | 1952 |     0,                                          /* tp_descr_get */ | 
 | 1953 |     0,                                          /* tp_descr_set */ | 
 | 1954 |     0,                                          /* tp_dictoffset */ | 
 | 1955 |     0,                                          /* tp_init */ | 
 | 1956 |     0,                                          /* tp_alloc */ | 
 | 1957 |     dequereviter_new,                           /* tp_new */ | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1958 |     0, | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 1959 | }; | 
 | 1960 |  | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 1961 | /* defaultdict type *********************************************************/ | 
 | 1962 |  | 
 | 1963 | typedef struct { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1964 |     PyDictObject dict; | 
 | 1965 |     PyObject *default_factory; | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 1966 | } defdictobject; | 
 | 1967 |  | 
 | 1968 | static PyTypeObject defdict_type; /* Forward */ | 
 | 1969 |  | 
 | 1970 | PyDoc_STRVAR(defdict_missing_doc, | 
 | 1971 | "__missing__(key) # Called by __getitem__ for missing key; pseudo-code:\n\ | 
| Guido van Rossum | d8faa36 | 2007-04-27 19:54:29 +0000 | [diff] [blame] | 1972 |   if self.default_factory is None: raise KeyError((key,))\n\ | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 1973 |   self[key] = value = self.default_factory()\n\ | 
 | 1974 |   return value\n\ | 
 | 1975 | "); | 
 | 1976 |  | 
 | 1977 | static PyObject * | 
 | 1978 | defdict_missing(defdictobject *dd, PyObject *key) | 
 | 1979 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 1980 |     PyObject *factory = dd->default_factory; | 
 | 1981 |     PyObject *value; | 
 | 1982 |     if (factory == NULL || factory == Py_None) { | 
 | 1983 |         /* XXX Call dict.__missing__(key) */ | 
 | 1984 |         PyObject *tup; | 
 | 1985 |         tup = PyTuple_Pack(1, key); | 
 | 1986 |         if (!tup) return NULL; | 
 | 1987 |         PyErr_SetObject(PyExc_KeyError, tup); | 
 | 1988 |         Py_DECREF(tup); | 
 | 1989 |         return NULL; | 
 | 1990 |     } | 
 | 1991 |     value = PyEval_CallObject(factory, NULL); | 
 | 1992 |     if (value == NULL) | 
 | 1993 |         return value; | 
 | 1994 |     if (PyObject_SetItem((PyObject *)dd, key, value) < 0) { | 
 | 1995 |         Py_DECREF(value); | 
 | 1996 |         return NULL; | 
 | 1997 |     } | 
 | 1998 |     return value; | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 1999 | } | 
 | 2000 |  | 
 | 2001 | PyDoc_STRVAR(defdict_copy_doc, "D.copy() -> a shallow copy of D."); | 
 | 2002 |  | 
 | 2003 | static PyObject * | 
 | 2004 | defdict_copy(defdictobject *dd) | 
 | 2005 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2006 |     /* This calls the object's class.  That only works for subclasses | 
 | 2007 |        whose class constructor has the same signature.  Subclasses that | 
 | 2008 |        define a different constructor signature must override copy(). | 
 | 2009 |     */ | 
| Raymond Hettinger | 54628fa | 2009-08-04 19:16:39 +0000 | [diff] [blame] | 2010 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2011 |     if (dd->default_factory == NULL) | 
 | 2012 |         return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), Py_None, dd, NULL); | 
 | 2013 |     return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), | 
 | 2014 |                                         dd->default_factory, dd, NULL); | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2015 | } | 
 | 2016 |  | 
 | 2017 | static PyObject * | 
 | 2018 | defdict_reduce(defdictobject *dd) | 
 | 2019 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2020 |     /* __reduce__ must return a 5-tuple as follows: | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2021 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2022 |        - factory function | 
 | 2023 |        - tuple of args for the factory function | 
 | 2024 |        - additional state (here None) | 
 | 2025 |        - sequence iterator (here None) | 
 | 2026 |        - dictionary iterator (yielding successive (key, value) pairs | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2027 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2028 |        This API is used by pickle.py and copy.py. | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2029 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2030 |        For this to be useful with pickle.py, the default_factory | 
 | 2031 |        must be picklable; e.g., None, a built-in, or a global | 
 | 2032 |        function in a module or package. | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2033 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2034 |        Both shallow and deep copying are supported, but for deep | 
 | 2035 |        copying, the default_factory must be deep-copyable; e.g. None, | 
 | 2036 |        or a built-in (functions are not copyable at this time). | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2037 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2038 |        This only works for subclasses as long as their constructor | 
 | 2039 |        signature is compatible; the first argument must be the | 
 | 2040 |        optional default_factory, defaulting to None. | 
 | 2041 |     */ | 
 | 2042 |     PyObject *args; | 
 | 2043 |     PyObject *items; | 
 | 2044 |     PyObject *iter; | 
 | 2045 |     PyObject *result; | 
| Martin v. Löwis | bd928fe | 2011-10-14 10:20:37 +0200 | [diff] [blame] | 2046 |     _Py_IDENTIFIER(items); | 
| Martin v. Löwis | afe55bb | 2011-10-09 10:38:36 +0200 | [diff] [blame] | 2047 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2048 |     if (dd->default_factory == NULL || dd->default_factory == Py_None) | 
 | 2049 |         args = PyTuple_New(0); | 
 | 2050 |     else | 
 | 2051 |         args = PyTuple_Pack(1, dd->default_factory); | 
 | 2052 |     if (args == NULL) | 
 | 2053 |         return NULL; | 
| Victor Stinner | ad8c83a | 2016-09-05 17:53:15 -0700 | [diff] [blame] | 2054 |     items = _PyObject_CallMethodId((PyObject *)dd, &PyId_items, NULL); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2055 |     if (items == NULL) { | 
 | 2056 |         Py_DECREF(args); | 
 | 2057 |         return NULL; | 
 | 2058 |     } | 
 | 2059 |     iter = PyObject_GetIter(items); | 
 | 2060 |     if (iter == NULL) { | 
 | 2061 |         Py_DECREF(items); | 
 | 2062 |         Py_DECREF(args); | 
 | 2063 |         return NULL; | 
 | 2064 |     } | 
 | 2065 |     result = PyTuple_Pack(5, Py_TYPE(dd), args, | 
 | 2066 |                           Py_None, Py_None, iter); | 
 | 2067 |     Py_DECREF(iter); | 
 | 2068 |     Py_DECREF(items); | 
 | 2069 |     Py_DECREF(args); | 
 | 2070 |     return result; | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2071 | } | 
 | 2072 |  | 
 | 2073 | static PyMethodDef defdict_methods[] = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2074 |     {"__missing__", (PyCFunction)defdict_missing, METH_O, | 
 | 2075 |      defdict_missing_doc}, | 
 | 2076 |     {"copy", (PyCFunction)defdict_copy, METH_NOARGS, | 
 | 2077 |      defdict_copy_doc}, | 
 | 2078 |     {"__copy__", (PyCFunction)defdict_copy, METH_NOARGS, | 
 | 2079 |      defdict_copy_doc}, | 
 | 2080 |     {"__reduce__", (PyCFunction)defdict_reduce, METH_NOARGS, | 
 | 2081 |      reduce_doc}, | 
 | 2082 |     {NULL} | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2083 | }; | 
 | 2084 |  | 
 | 2085 | static PyMemberDef defdict_members[] = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2086 |     {"default_factory", T_OBJECT, | 
 | 2087 |      offsetof(defdictobject, default_factory), 0, | 
 | 2088 |      PyDoc_STR("Factory for default value called by __missing__().")}, | 
 | 2089 |     {NULL} | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2090 | }; | 
 | 2091 |  | 
 | 2092 | static void | 
 | 2093 | defdict_dealloc(defdictobject *dd) | 
 | 2094 | { | 
| INADA Naoki | a6296d3 | 2017-08-24 14:55:17 +0900 | [diff] [blame] | 2095 |     /* bpo-31095: UnTrack is needed before calling any callbacks */ | 
 | 2096 |     PyObject_GC_UnTrack(dd); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2097 |     Py_CLEAR(dd->default_factory); | 
 | 2098 |     PyDict_Type.tp_dealloc((PyObject *)dd); | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2099 | } | 
 | 2100 |  | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2101 | static PyObject * | 
 | 2102 | defdict_repr(defdictobject *dd) | 
 | 2103 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2104 |     PyObject *baserepr; | 
 | 2105 |     PyObject *defrepr; | 
 | 2106 |     PyObject *result; | 
 | 2107 |     baserepr = PyDict_Type.tp_repr((PyObject *)dd); | 
 | 2108 |     if (baserepr == NULL) | 
 | 2109 |         return NULL; | 
 | 2110 |     if (dd->default_factory == NULL) | 
 | 2111 |         defrepr = PyUnicode_FromString("None"); | 
 | 2112 |     else | 
 | 2113 |     { | 
 | 2114 |         int status = Py_ReprEnter(dd->default_factory); | 
 | 2115 |         if (status != 0) { | 
| Antoine Pitrou | f5f1fe0 | 2012-02-15 02:42:46 +0100 | [diff] [blame] | 2116 |             if (status < 0) { | 
 | 2117 |                 Py_DECREF(baserepr); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2118 |                 return NULL; | 
| Antoine Pitrou | f5f1fe0 | 2012-02-15 02:42:46 +0100 | [diff] [blame] | 2119 |             } | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2120 |             defrepr = PyUnicode_FromString("..."); | 
 | 2121 |         } | 
 | 2122 |         else | 
 | 2123 |             defrepr = PyObject_Repr(dd->default_factory); | 
 | 2124 |         Py_ReprLeave(dd->default_factory); | 
 | 2125 |     } | 
 | 2126 |     if (defrepr == NULL) { | 
 | 2127 |         Py_DECREF(baserepr); | 
 | 2128 |         return NULL; | 
 | 2129 |     } | 
 | 2130 |     result = PyUnicode_FromFormat("defaultdict(%U, %U)", | 
 | 2131 |                                   defrepr, baserepr); | 
 | 2132 |     Py_DECREF(defrepr); | 
 | 2133 |     Py_DECREF(baserepr); | 
 | 2134 |     return result; | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2135 | } | 
 | 2136 |  | 
 | 2137 | static int | 
 | 2138 | defdict_traverse(PyObject *self, visitproc visit, void *arg) | 
 | 2139 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2140 |     Py_VISIT(((defdictobject *)self)->default_factory); | 
 | 2141 |     return PyDict_Type.tp_traverse(self, visit, arg); | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2142 | } | 
 | 2143 |  | 
 | 2144 | static int | 
 | 2145 | defdict_tp_clear(defdictobject *dd) | 
 | 2146 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2147 |     Py_CLEAR(dd->default_factory); | 
 | 2148 |     return PyDict_Type.tp_clear((PyObject *)dd); | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2149 | } | 
 | 2150 |  | 
 | 2151 | static int | 
 | 2152 | defdict_init(PyObject *self, PyObject *args, PyObject *kwds) | 
 | 2153 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2154 |     defdictobject *dd = (defdictobject *)self; | 
 | 2155 |     PyObject *olddefault = dd->default_factory; | 
 | 2156 |     PyObject *newdefault = NULL; | 
 | 2157 |     PyObject *newargs; | 
 | 2158 |     int result; | 
 | 2159 |     if (args == NULL || !PyTuple_Check(args)) | 
 | 2160 |         newargs = PyTuple_New(0); | 
 | 2161 |     else { | 
 | 2162 |         Py_ssize_t n = PyTuple_GET_SIZE(args); | 
 | 2163 |         if (n > 0) { | 
 | 2164 |             newdefault = PyTuple_GET_ITEM(args, 0); | 
 | 2165 |             if (!PyCallable_Check(newdefault) && newdefault != Py_None) { | 
 | 2166 |                 PyErr_SetString(PyExc_TypeError, | 
| Raymond Hettinger | 239aba7 | 2015-07-20 03:09:22 -0400 | [diff] [blame] | 2167 |                     "first argument must be callable or None"); | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2168 |                 return -1; | 
 | 2169 |             } | 
 | 2170 |         } | 
 | 2171 |         newargs = PySequence_GetSlice(args, 1, n); | 
 | 2172 |     } | 
 | 2173 |     if (newargs == NULL) | 
 | 2174 |         return -1; | 
 | 2175 |     Py_XINCREF(newdefault); | 
 | 2176 |     dd->default_factory = newdefault; | 
 | 2177 |     result = PyDict_Type.tp_init(self, newargs, kwds); | 
 | 2178 |     Py_DECREF(newargs); | 
 | 2179 |     Py_XDECREF(olddefault); | 
 | 2180 |     return result; | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2181 | } | 
 | 2182 |  | 
 | 2183 | PyDoc_STRVAR(defdict_doc, | 
| Benjamin Peterson | 9cb33b7 | 2014-01-13 23:56:05 -0500 | [diff] [blame] | 2184 | "defaultdict(default_factory[, ...]) --> dict with default factory\n\ | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2185 | \n\ | 
 | 2186 | The default factory is called without arguments to produce\n\ | 
 | 2187 | a new value when a key is not present, in __getitem__ only.\n\ | 
 | 2188 | A defaultdict compares equal to a dict with the same items.\n\ | 
| Benjamin Peterson | 9cb33b7 | 2014-01-13 23:56:05 -0500 | [diff] [blame] | 2189 | All remaining arguments are treated the same as if they were\n\ | 
 | 2190 | passed to the dict constructor, including keyword arguments.\n\ | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2191 | "); | 
 | 2192 |  | 
| Thomas Wouters | 49fd7fa | 2006-04-21 10:40:58 +0000 | [diff] [blame] | 2193 | /* See comment in xxsubtype.c */ | 
 | 2194 | #define DEFERRED_ADDRESS(ADDR) 0 | 
 | 2195 |  | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2196 | static PyTypeObject defdict_type = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2197 |     PyVarObject_HEAD_INIT(DEFERRED_ADDRESS(&PyType_Type), 0) | 
 | 2198 |     "collections.defaultdict",          /* tp_name */ | 
 | 2199 |     sizeof(defdictobject),              /* tp_basicsize */ | 
 | 2200 |     0,                                  /* tp_itemsize */ | 
 | 2201 |     /* methods */ | 
 | 2202 |     (destructor)defdict_dealloc,        /* tp_dealloc */ | 
 | 2203 |     0,                                  /* tp_print */ | 
 | 2204 |     0,                                  /* tp_getattr */ | 
 | 2205 |     0,                                  /* tp_setattr */ | 
 | 2206 |     0,                                  /* tp_reserved */ | 
 | 2207 |     (reprfunc)defdict_repr,             /* tp_repr */ | 
 | 2208 |     0,                                  /* tp_as_number */ | 
 | 2209 |     0,                                  /* tp_as_sequence */ | 
 | 2210 |     0,                                  /* tp_as_mapping */ | 
 | 2211 |     0,                                  /* tp_hash */ | 
 | 2212 |     0,                                  /* tp_call */ | 
 | 2213 |     0,                                  /* tp_str */ | 
 | 2214 |     PyObject_GenericGetAttr,            /* tp_getattro */ | 
 | 2215 |     0,                                  /* tp_setattro */ | 
 | 2216 |     0,                                  /* tp_as_buffer */ | 
 | 2217 |     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC, | 
 | 2218 |                                     /* tp_flags */ | 
 | 2219 |     defdict_doc,                        /* tp_doc */ | 
 | 2220 |     defdict_traverse,                   /* tp_traverse */ | 
 | 2221 |     (inquiry)defdict_tp_clear,          /* tp_clear */ | 
 | 2222 |     0,                                  /* tp_richcompare */ | 
 | 2223 |     0,                                  /* tp_weaklistoffset*/ | 
 | 2224 |     0,                                  /* tp_iter */ | 
 | 2225 |     0,                                  /* tp_iternext */ | 
 | 2226 |     defdict_methods,                    /* tp_methods */ | 
 | 2227 |     defdict_members,                    /* tp_members */ | 
 | 2228 |     0,                                  /* tp_getset */ | 
 | 2229 |     DEFERRED_ADDRESS(&PyDict_Type),     /* tp_base */ | 
 | 2230 |     0,                                  /* tp_dict */ | 
 | 2231 |     0,                                  /* tp_descr_get */ | 
 | 2232 |     0,                                  /* tp_descr_set */ | 
 | 2233 |     0,                                  /* tp_dictoffset */ | 
 | 2234 |     defdict_init,                       /* tp_init */ | 
 | 2235 |     PyType_GenericAlloc,                /* tp_alloc */ | 
 | 2236 |     0,                                  /* tp_new */ | 
 | 2237 |     PyObject_GC_Del,                    /* tp_free */ | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2238 | }; | 
 | 2239 |  | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2240 | /* helper function for Counter  *********************************************/ | 
 | 2241 |  | 
 | 2242 | PyDoc_STRVAR(_count_elements_doc, | 
 | 2243 | "_count_elements(mapping, iterable) -> None\n\ | 
 | 2244 | \n\ | 
| Raymond Hettinger | a24dca6 | 2017-01-12 22:25:25 -0800 | [diff] [blame] | 2245 | Count elements in the iterable, updating the mapping"); | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2246 |  | 
 | 2247 | static PyObject * | 
 | 2248 | _count_elements(PyObject *self, PyObject *args) | 
 | 2249 | { | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2250 |     _Py_IDENTIFIER(get); | 
| Raymond Hettinger | 2ff2190 | 2013-10-01 00:55:43 -0700 | [diff] [blame] | 2251 |     _Py_IDENTIFIER(__setitem__); | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2252 |     PyObject *it, *iterable, *mapping, *oldval; | 
 | 2253 |     PyObject *newval = NULL; | 
 | 2254 |     PyObject *key = NULL; | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2255 |     PyObject *bound_get = NULL; | 
 | 2256 |     PyObject *mapping_get; | 
 | 2257 |     PyObject *dict_get; | 
| Raymond Hettinger | 2ff2190 | 2013-10-01 00:55:43 -0700 | [diff] [blame] | 2258 |     PyObject *mapping_setitem; | 
| Raymond Hettinger | 2ff2190 | 2013-10-01 00:55:43 -0700 | [diff] [blame] | 2259 |     PyObject *dict_setitem; | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2260 |  | 
 | 2261 |     if (!PyArg_UnpackTuple(args, "_count_elements", 2, 2, &mapping, &iterable)) | 
 | 2262 |         return NULL; | 
 | 2263 |  | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2264 |     it = PyObject_GetIter(iterable); | 
 | 2265 |     if (it == NULL) | 
 | 2266 |         return NULL; | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2267 |  | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2268 |     /* Only take the fast path when get() and __setitem__() | 
 | 2269 |      * have not been overridden. | 
 | 2270 |      */ | 
 | 2271 |     mapping_get = _PyType_LookupId(Py_TYPE(mapping), &PyId_get); | 
 | 2272 |     dict_get = _PyType_LookupId(&PyDict_Type, &PyId_get); | 
| Raymond Hettinger | 2ff2190 | 2013-10-01 00:55:43 -0700 | [diff] [blame] | 2273 |     mapping_setitem = _PyType_LookupId(Py_TYPE(mapping), &PyId___setitem__); | 
 | 2274 |     dict_setitem = _PyType_LookupId(&PyDict_Type, &PyId___setitem__); | 
 | 2275 |  | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2276 |     if (mapping_get != NULL && mapping_get == dict_get && | 
 | 2277 |         mapping_setitem != NULL && mapping_setitem == dict_setitem) { | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2278 |         while (1) { | 
| Raymond Hettinger | 507d997 | 2014-05-18 21:32:40 +0100 | [diff] [blame] | 2279 |             /* Fast path advantages: | 
 | 2280 |                    1. Eliminate double hashing | 
 | 2281 |                       (by re-using the same hash for both the get and set) | 
 | 2282 |                    2. Avoid argument overhead of PyObject_CallFunctionObjArgs | 
 | 2283 |                       (argument tuple creation and parsing) | 
 | 2284 |                    3. Avoid indirection through a bound method object | 
 | 2285 |                       (creates another argument tuple) | 
 | 2286 |                    4. Avoid initial increment from zero | 
 | 2287 |                       (reuse an existing one-object instead) | 
 | 2288 |             */ | 
| Raymond Hettinger | 4b0b1ac | 2014-05-03 16:41:19 -0700 | [diff] [blame] | 2289 |             Py_hash_t hash; | 
 | 2290 |  | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2291 |             key = PyIter_Next(it); | 
| Victor Stinner | a154b5c | 2011-04-20 23:23:52 +0200 | [diff] [blame] | 2292 |             if (key == NULL) | 
 | 2293 |                 break; | 
| Raymond Hettinger | 4b0b1ac | 2014-05-03 16:41:19 -0700 | [diff] [blame] | 2294 |  | 
 | 2295 |             if (!PyUnicode_CheckExact(key) || | 
 | 2296 |                 (hash = ((PyASCIIObject *) key)->hash) == -1) | 
 | 2297 |             { | 
 | 2298 |                 hash = PyObject_Hash(key); | 
| Raymond Hettinger | da2850f | 2015-02-27 12:42:54 -0800 | [diff] [blame] | 2299 |                 if (hash == -1) | 
| Raymond Hettinger | 4b0b1ac | 2014-05-03 16:41:19 -0700 | [diff] [blame] | 2300 |                     goto done; | 
 | 2301 |             } | 
 | 2302 |  | 
 | 2303 |             oldval = _PyDict_GetItem_KnownHash(mapping, key, hash); | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2304 |             if (oldval == NULL) { | 
| Serhiy Storchaka | f0b311b | 2016-11-06 13:18:24 +0200 | [diff] [blame] | 2305 |                 if (PyErr_Occurred()) | 
 | 2306 |                     goto done; | 
| Serhiy Storchaka | ba85d69 | 2017-03-30 09:09:41 +0300 | [diff] [blame] | 2307 |                 if (_PyDict_SetItem_KnownHash(mapping, key, _PyLong_One, hash) < 0) | 
| Raymond Hettinger | 507d997 | 2014-05-18 21:32:40 +0100 | [diff] [blame] | 2308 |                     goto done; | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2309 |             } else { | 
| Serhiy Storchaka | ba85d69 | 2017-03-30 09:09:41 +0300 | [diff] [blame] | 2310 |                 newval = PyNumber_Add(oldval, _PyLong_One); | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2311 |                 if (newval == NULL) | 
| Raymond Hettinger | 507d997 | 2014-05-18 21:32:40 +0100 | [diff] [blame] | 2312 |                     goto done; | 
| Raymond Hettinger | 28c995d | 2015-08-14 02:07:41 -0700 | [diff] [blame] | 2313 |                 if (_PyDict_SetItem_KnownHash(mapping, key, newval, hash) < 0) | 
| Raymond Hettinger | 507d997 | 2014-05-18 21:32:40 +0100 | [diff] [blame] | 2314 |                     goto done; | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2315 |                 Py_CLEAR(newval); | 
 | 2316 |             } | 
 | 2317 |             Py_DECREF(key); | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2318 |         } | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2319 |     } else { | 
| Victor Stinner | e7f516c | 2013-11-06 23:52:55 +0100 | [diff] [blame] | 2320 |         bound_get = _PyObject_GetAttrId(mapping, &PyId_get); | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2321 |         if (bound_get == NULL) | 
| Raymond Hettinger | 224c87d | 2013-10-01 21:36:09 -0700 | [diff] [blame] | 2322 |             goto done; | 
 | 2323 |  | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2324 |         while (1) { | 
 | 2325 |             key = PyIter_Next(it); | 
| Victor Stinner | a154b5c | 2011-04-20 23:23:52 +0200 | [diff] [blame] | 2326 |             if (key == NULL) | 
 | 2327 |                 break; | 
| Serhiy Storchaka | ba85d69 | 2017-03-30 09:09:41 +0300 | [diff] [blame] | 2328 |             oldval = PyObject_CallFunctionObjArgs(bound_get, key, _PyLong_Zero, NULL); | 
| Raymond Hettinger | 224c87d | 2013-10-01 21:36:09 -0700 | [diff] [blame] | 2329 |             if (oldval == NULL) | 
 | 2330 |                 break; | 
| Serhiy Storchaka | ba85d69 | 2017-03-30 09:09:41 +0300 | [diff] [blame] | 2331 |             newval = PyNumber_Add(oldval, _PyLong_One); | 
| Raymond Hettinger | 224c87d | 2013-10-01 21:36:09 -0700 | [diff] [blame] | 2332 |             Py_DECREF(oldval); | 
 | 2333 |             if (newval == NULL) | 
 | 2334 |                 break; | 
| Raymond Hettinger | 28c995d | 2015-08-14 02:07:41 -0700 | [diff] [blame] | 2335 |             if (PyObject_SetItem(mapping, key, newval) < 0) | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2336 |                 break; | 
 | 2337 |             Py_CLEAR(newval); | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2338 |             Py_DECREF(key); | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2339 |         } | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2340 |     } | 
| Raymond Hettinger | 426e052 | 2011-01-03 02:12:02 +0000 | [diff] [blame] | 2341 |  | 
| Raymond Hettinger | 224c87d | 2013-10-01 21:36:09 -0700 | [diff] [blame] | 2342 | done: | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2343 |     Py_DECREF(it); | 
 | 2344 |     Py_XDECREF(key); | 
 | 2345 |     Py_XDECREF(newval); | 
| Raymond Hettinger | cb1d96f | 2013-10-04 16:51:02 -0700 | [diff] [blame] | 2346 |     Py_XDECREF(bound_get); | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2347 |     if (PyErr_Occurred()) | 
 | 2348 |         return NULL; | 
 | 2349 |     Py_RETURN_NONE; | 
 | 2350 | } | 
 | 2351 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2352 | /* module level code ********************************************************/ | 
 | 2353 |  | 
 | 2354 | PyDoc_STRVAR(module_doc, | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2355 | "High performance data structures.\n\ | 
 | 2356 | - deque:        ordered collection accessible from endpoints only\n\ | 
 | 2357 | - defaultdict:  dict subclass with a default value factory\n\ | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2358 | "); | 
 | 2359 |  | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2360 | static struct PyMethodDef module_functions[] = { | 
 | 2361 |     {"_count_elements", _count_elements,    METH_VARARGS,   _count_elements_doc}, | 
 | 2362 |     {NULL,       NULL}          /* sentinel */ | 
 | 2363 | }; | 
| Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 2364 |  | 
 | 2365 | static struct PyModuleDef _collectionsmodule = { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2366 |     PyModuleDef_HEAD_INIT, | 
 | 2367 |     "_collections", | 
 | 2368 |     module_doc, | 
 | 2369 |     -1, | 
| Raymond Hettinger | 96f3410 | 2010-12-15 16:30:37 +0000 | [diff] [blame] | 2370 |     module_functions, | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2371 |     NULL, | 
 | 2372 |     NULL, | 
 | 2373 |     NULL, | 
 | 2374 |     NULL | 
| Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 2375 | }; | 
 | 2376 |  | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2377 | PyMODINIT_FUNC | 
| Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 2378 | PyInit__collections(void) | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2379 | { | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2380 |     PyObject *m; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2381 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2382 |     m = PyModule_Create(&_collectionsmodule); | 
 | 2383 |     if (m == NULL) | 
 | 2384 |         return NULL; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2385 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2386 |     if (PyType_Ready(&deque_type) < 0) | 
 | 2387 |         return NULL; | 
 | 2388 |     Py_INCREF(&deque_type); | 
 | 2389 |     PyModule_AddObject(m, "deque", (PyObject *)&deque_type); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2390 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2391 |     defdict_type.tp_base = &PyDict_Type; | 
 | 2392 |     if (PyType_Ready(&defdict_type) < 0) | 
 | 2393 |         return NULL; | 
 | 2394 |     Py_INCREF(&defdict_type); | 
 | 2395 |     PyModule_AddObject(m, "defaultdict", (PyObject *)&defdict_type); | 
| Guido van Rossum | 1968ad3 | 2006-02-25 22:38:04 +0000 | [diff] [blame] | 2396 |  | 
| Eric Snow | 47db717 | 2015-05-29 22:21:39 -0600 | [diff] [blame] | 2397 |     Py_INCREF(&PyODict_Type); | 
 | 2398 |     PyModule_AddObject(m, "OrderedDict", (PyObject *)&PyODict_Type); | 
 | 2399 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2400 |     if (PyType_Ready(&dequeiter_type) < 0) | 
 | 2401 |         return NULL; | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 2402 |     Py_INCREF(&dequeiter_type); | 
 | 2403 |     PyModule_AddObject(m, "_deque_iterator", (PyObject *)&dequeiter_type); | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2404 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2405 |     if (PyType_Ready(&dequereviter_type) < 0) | 
 | 2406 |         return NULL; | 
| Kristján Valur Jónsson | 31668b8 | 2012-04-03 10:49:41 +0000 | [diff] [blame] | 2407 |     Py_INCREF(&dequereviter_type); | 
 | 2408 |     PyModule_AddObject(m, "_deque_reverse_iterator", (PyObject *)&dequereviter_type); | 
| Raymond Hettinger | 1e5809f | 2004-03-18 11:04:57 +0000 | [diff] [blame] | 2409 |  | 
| Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 2410 |     return m; | 
| Raymond Hettinger | 756b3f3 | 2004-01-29 06:37:52 +0000 | [diff] [blame] | 2411 | } |