You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

1636 lines
49 KiB

  1. #include "Python.h"
  2. #include "structmember.h"
  3. /* collections module implementation of a deque() datatype
  4. Written and maintained by Raymond D. Hettinger <python@rcn.com>
  5. Copyright (c) 2004 Python Software Foundation.
  6. All rights reserved.
  7. */
  8. /* The block length may be set to any number over 1. Larger numbers
  9. * reduce the number of calls to the memory allocator but take more
  10. * memory. Ideally, BLOCKLEN should be set with an eye to the
  11. * length of a cache line.
  12. */
  13. #define BLOCKLEN 62
  14. #define CENTER ((BLOCKLEN - 1) / 2)
  15. /* A `dequeobject` is composed of a doubly-linked list of `block` nodes.
  16. * This list is not circular (the leftmost block has leftlink==NULL,
  17. * and the rightmost block has rightlink==NULL). A deque d's first
  18. * element is at d.leftblock[leftindex] and its last element is at
  19. * d.rightblock[rightindex]; note that, unlike as for Python slice
  20. * indices, these indices are inclusive on both ends. By being inclusive
  21. * on both ends, algorithms for left and right operations become
  22. * symmetrical which simplifies the design.
  23. *
  24. * The list of blocks is never empty, so d.leftblock and d.rightblock
  25. * are never equal to NULL.
  26. *
  27. * The indices, d.leftindex and d.rightindex are always in the range
  28. * 0 <= index < BLOCKLEN.
  29. * Their exact relationship is:
  30. * (d.leftindex + d.len - 1) % BLOCKLEN == d.rightindex.
  31. *
  32. * Empty deques have d.len == 0; d.leftblock==d.rightblock;
  33. * d.leftindex == CENTER+1; and d.rightindex == CENTER.
  34. * Checking for d.len == 0 is the intended way to see whether d is empty.
  35. *
  36. * Whenever d.leftblock == d.rightblock,
  37. * d.leftindex + d.len - 1 == d.rightindex.
  38. *
  39. * However, when d.leftblock != d.rightblock, d.leftindex and d.rightindex
  40. * become indices into distinct blocks and either may be larger than the
  41. * other.
  42. */
  43. typedef struct BLOCK {
  44. struct BLOCK *leftlink;
  45. struct BLOCK *rightlink;
  46. PyObject *data[BLOCKLEN];
  47. } block;
  48. #define MAXFREEBLOCKS 10
  49. static Py_ssize_t numfreeblocks = 0;
  50. static block *freeblocks[MAXFREEBLOCKS];
  51. static block *
  52. newblock(block *leftlink, block *rightlink, Py_ssize_t len) {
  53. block *b;
  54. /* To prevent len from overflowing PY_SSIZE_T_MAX on 64-bit machines, we
  55. * refuse to allocate new blocks if the current len is dangerously
  56. * close. There is some extra margin to prevent spurious arithmetic
  57. * overflows at various places. The following check ensures that
  58. * the blocks allocated to the deque, in the worst case, can only
  59. * have PY_SSIZE_T_MAX-2 entries in total.
  60. */
  61. if (len >= PY_SSIZE_T_MAX - 2*BLOCKLEN) {
  62. PyErr_SetString(PyExc_OverflowError,
  63. "cannot add more blocks to the deque");
  64. return NULL;
  65. }
  66. if (numfreeblocks) {
  67. numfreeblocks -= 1;
  68. b = freeblocks[numfreeblocks];
  69. } else {
  70. b = PyMem_Malloc(sizeof(block));
  71. if (b == NULL) {
  72. PyErr_NoMemory();
  73. return NULL;
  74. }
  75. }
  76. b->leftlink = leftlink;
  77. b->rightlink = rightlink;
  78. return b;
  79. }
  80. static void
  81. freeblock(block *b)
  82. {
  83. if (numfreeblocks < MAXFREEBLOCKS) {
  84. freeblocks[numfreeblocks] = b;
  85. numfreeblocks++;
  86. } else {
  87. PyMem_Free(b);
  88. }
  89. }
  90. typedef struct {
  91. PyObject_HEAD
  92. block *leftblock;
  93. block *rightblock;
  94. Py_ssize_t leftindex; /* in range(BLOCKLEN) */
  95. Py_ssize_t rightindex; /* in range(BLOCKLEN) */
  96. Py_ssize_t len;
  97. Py_ssize_t maxlen;
  98. long state; /* incremented whenever the indices move */
  99. PyObject *weakreflist; /* List of weak references */
  100. } dequeobject;
  101. /* The deque's size limit is d.maxlen. The limit can be zero or positive.
  102. * If there is no limit, then d.maxlen == -1.
  103. *
  104. * After an item is added to a deque, we check to see if the size has grown past
  105. * the limit. If it has, we get the size back down to the limit by popping an
  106. * item off of the opposite end. The methods that can trigger this are append(),
  107. * appendleft(), extend(), and extendleft().
  108. */
  109. #define TRIM(d, popfunction) \
  110. if (d->maxlen != -1 && d->len > d->maxlen) { \
  111. PyObject *rv = popfunction(d, NULL); \
  112. assert(rv != NULL && d->len <= d->maxlen); \
  113. Py_DECREF(rv); \
  114. }
  115. static PyTypeObject deque_type;
  116. static PyObject *
  117. deque_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
  118. {
  119. dequeobject *deque;
  120. block *b;
  121. /* create dequeobject structure */
  122. deque = (dequeobject *)type->tp_alloc(type, 0);
  123. if (deque == NULL)
  124. return NULL;
  125. b = newblock(NULL, NULL, 0);
  126. if (b == NULL) {
  127. Py_DECREF(deque);
  128. return NULL;
  129. }
  130. assert(BLOCKLEN >= 2);
  131. deque->leftblock = b;
  132. deque->rightblock = b;
  133. deque->leftindex = CENTER + 1;
  134. deque->rightindex = CENTER;
  135. deque->len = 0;
  136. deque->state = 0;
  137. deque->weakreflist = NULL;
  138. deque->maxlen = -1;
  139. return (PyObject *)deque;
  140. }
  141. static PyObject *
  142. deque_pop(dequeobject *deque, PyObject *unused)
  143. {
  144. PyObject *item;
  145. block *prevblock;
  146. if (deque->len == 0) {
  147. PyErr_SetString(PyExc_IndexError, "pop from an empty deque");
  148. return NULL;
  149. }
  150. item = deque->rightblock->data[deque->rightindex];
  151. deque->rightindex--;
  152. deque->len--;
  153. deque->state++;
  154. if (deque->rightindex == -1) {
  155. if (deque->len == 0) {
  156. assert(deque->leftblock == deque->rightblock);
  157. assert(deque->leftindex == deque->rightindex+1);
  158. /* re-center instead of freeing a block */
  159. deque->leftindex = CENTER + 1;
  160. deque->rightindex = CENTER;
  161. } else {
  162. prevblock = deque->rightblock->leftlink;
  163. assert(deque->leftblock != deque->rightblock);
  164. freeblock(deque->rightblock);
  165. prevblock->rightlink = NULL;
  166. deque->rightblock = prevblock;
  167. deque->rightindex = BLOCKLEN - 1;
  168. }
  169. }
  170. return item;
  171. }
  172. PyDoc_STRVAR(pop_doc, "Remove and return the rightmost element.");
  173. static PyObject *
  174. deque_popleft(dequeobject *deque, PyObject *unused)
  175. {
  176. PyObject *item;
  177. block *prevblock;
  178. if (deque->len == 0) {
  179. PyErr_SetString(PyExc_IndexError, "pop from an empty deque");
  180. return NULL;
  181. }
  182. assert(deque->leftblock != NULL);
  183. item = deque->leftblock->data[deque->leftindex];
  184. deque->leftindex++;
  185. deque->len--;
  186. deque->state++;
  187. if (deque->leftindex == BLOCKLEN) {
  188. if (deque->len == 0) {
  189. assert(deque->leftblock == deque->rightblock);
  190. assert(deque->leftindex == deque->rightindex+1);
  191. /* re-center instead of freeing a block */
  192. deque->leftindex = CENTER + 1;
  193. deque->rightindex = CENTER;
  194. } else {
  195. assert(deque->leftblock != deque->rightblock);
  196. prevblock = deque->leftblock->rightlink;
  197. freeblock(deque->leftblock);
  198. assert(prevblock != NULL);
  199. prevblock->leftlink = NULL;
  200. deque->leftblock = prevblock;
  201. deque->leftindex = 0;
  202. }
  203. }
  204. return item;
  205. }
  206. PyDoc_STRVAR(popleft_doc, "Remove and return the leftmost element.");
  207. static PyObject *
  208. deque_append(dequeobject *deque, PyObject *item)
  209. {
  210. deque->state++;
  211. if (deque->rightindex == BLOCKLEN-1) {
  212. block *b = newblock(deque->rightblock, NULL, deque->len);
  213. if (b == NULL)
  214. return NULL;
  215. assert(deque->rightblock->rightlink == NULL);
  216. deque->rightblock->rightlink = b;
  217. deque->rightblock = b;
  218. deque->rightindex = -1;
  219. }
  220. Py_INCREF(item);
  221. deque->len++;
  222. deque->rightindex++;
  223. deque->rightblock->data[deque->rightindex] = item;
  224. TRIM(deque, deque_popleft);
  225. Py_RETURN_NONE;
  226. }
  227. PyDoc_STRVAR(append_doc, "Add an element to the right side of the deque.");
  228. static PyObject *
  229. deque_appendleft(dequeobject *deque, PyObject *item)
  230. {
  231. deque->state++;
  232. if (deque->leftindex == 0) {
  233. block *b = newblock(NULL, deque->leftblock, deque->len);
  234. if (b == NULL)
  235. return NULL;
  236. assert(deque->leftblock->leftlink == NULL);
  237. deque->leftblock->leftlink = b;
  238. deque->leftblock = b;
  239. deque->leftindex = BLOCKLEN;
  240. }
  241. Py_INCREF(item);
  242. deque->len++;
  243. deque->leftindex--;
  244. deque->leftblock->data[deque->leftindex] = item;
  245. TRIM(deque, deque_pop);
  246. Py_RETURN_NONE;
  247. }
  248. PyDoc_STRVAR(appendleft_doc, "Add an element to the left side of the deque.");
  249. /* Run an iterator to exhaustion. Shortcut for
  250. the extend/extendleft methods when maxlen == 0. */
  251. static PyObject*
  252. consume_iterator(PyObject *it)
  253. {
  254. PyObject *item;
  255. while ((item = PyIter_Next(it)) != NULL) {
  256. Py_DECREF(item);
  257. }
  258. Py_DECREF(it);
  259. if (PyErr_Occurred())
  260. return NULL;
  261. Py_RETURN_NONE;
  262. }
  263. static PyObject *
  264. deque_extend(dequeobject *deque, PyObject *iterable)
  265. {
  266. PyObject *it, *item;
  267. /* Handle case where id(deque) == id(iterable) */
  268. if ((PyObject *)deque == iterable) {
  269. PyObject *result;
  270. PyObject *s = PySequence_List(iterable);
  271. if (s == NULL)
  272. return NULL;
  273. result = deque_extend(deque, s);
  274. Py_DECREF(s);
  275. return result;
  276. }
  277. it = PyObject_GetIter(iterable);
  278. if (it == NULL)
  279. return NULL;
  280. if (deque->maxlen == 0)
  281. return consume_iterator(it);
  282. while ((item = PyIter_Next(it)) != NULL) {
  283. deque->state++;
  284. if (deque->rightindex == BLOCKLEN-1) {
  285. block *b = newblock(deque->rightblock, NULL,
  286. deque->len);
  287. if (b == NULL) {
  288. Py_DECREF(item);
  289. Py_DECREF(it);
  290. return NULL;
  291. }
  292. assert(deque->rightblock->rightlink == NULL);
  293. deque->rightblock->rightlink = b;
  294. deque->rightblock = b;
  295. deque->rightindex = -1;
  296. }
  297. deque->len++;
  298. deque->rightindex++;
  299. deque->rightblock->data[deque->rightindex] = item;
  300. TRIM(deque, deque_popleft);
  301. }
  302. Py_DECREF(it);
  303. if (PyErr_Occurred())
  304. return NULL;
  305. Py_RETURN_NONE;
  306. }
  307. PyDoc_STRVAR(extend_doc,
  308. "Extend the right side of the deque with elements from the iterable");
  309. static PyObject *
  310. deque_extendleft(dequeobject *deque, PyObject *iterable)
  311. {
  312. PyObject *it, *item;
  313. /* Handle case where id(deque) == id(iterable) */
  314. if ((PyObject *)deque == iterable) {
  315. PyObject *result;
  316. PyObject *s = PySequence_List(iterable);
  317. if (s == NULL)
  318. return NULL;
  319. result = deque_extendleft(deque, s);
  320. Py_DECREF(s);
  321. return result;
  322. }
  323. it = PyObject_GetIter(iterable);
  324. if (it == NULL)
  325. return NULL;
  326. if (deque->maxlen == 0)
  327. return consume_iterator(it);
  328. while ((item = PyIter_Next(it)) != NULL) {
  329. deque->state++;
  330. if (deque->leftindex == 0) {
  331. block *b = newblock(NULL, deque->leftblock,
  332. deque->len);
  333. if (b == NULL) {
  334. Py_DECREF(item);
  335. Py_DECREF(it);
  336. return NULL;
  337. }
  338. assert(deque->leftblock->leftlink == NULL);
  339. deque->leftblock->leftlink = b;
  340. deque->leftblock = b;
  341. deque->leftindex = BLOCKLEN;
  342. }
  343. deque->len++;
  344. deque->leftindex--;
  345. deque->leftblock->data[deque->leftindex] = item;
  346. TRIM(deque, deque_pop);
  347. }
  348. Py_DECREF(it);
  349. if (PyErr_Occurred())
  350. return NULL;
  351. Py_RETURN_NONE;
  352. }
  353. PyDoc_STRVAR(extendleft_doc,
  354. "Extend the left side of the deque with elements from the iterable");
  355. static PyObject *
  356. deque_inplace_concat(dequeobject *deque, PyObject *other)
  357. {
  358. PyObject *result;
  359. result = deque_extend(deque, other);
  360. if (result == NULL)
  361. return result;
  362. Py_DECREF(result);
  363. Py_INCREF(deque);
  364. return (PyObject *)deque;
  365. }
  366. static int
  367. _deque_rotate(dequeobject *deque, Py_ssize_t n)
  368. {
  369. Py_ssize_t i, len=deque->len, halflen=(len+1)>>1;
  370. PyObject *item, *rv;
  371. if (len == 0)
  372. return 0;
  373. if (n > halflen || n < -halflen) {
  374. n %= len;
  375. if (n > halflen)
  376. n -= len;
  377. else if (n < -halflen)
  378. n += len;
  379. }
  380. for (i=0 ; i<n ; i++) {
  381. item = deque_pop(deque, NULL);
  382. assert (item != NULL);
  383. rv = deque_appendleft(deque, item);
  384. Py_DECREF(item);
  385. if (rv == NULL)
  386. return -1;
  387. Py_DECREF(rv);
  388. }
  389. for (i=0 ; i>n ; i--) {
  390. item = deque_popleft(deque, NULL);
  391. assert (item != NULL);
  392. rv = deque_append(deque, item);
  393. Py_DECREF(item);
  394. if (rv == NULL)
  395. return -1;
  396. Py_DECREF(rv);
  397. }
  398. return 0;
  399. }
  400. static PyObject *
  401. deque_rotate(dequeobject *deque, PyObject *args)
  402. {
  403. Py_ssize_t n=1;
  404. if (!PyArg_ParseTuple(args, "|n:rotate", &n))
  405. return NULL;
  406. if (_deque_rotate(deque, n) == 0)
  407. Py_RETURN_NONE;
  408. return NULL;
  409. }
  410. PyDoc_STRVAR(rotate_doc,
  411. "Rotate the deque n steps to the right (default n=1). If n is negative, rotates left.");
  412. static PyObject *
  413. deque_reverse(dequeobject *deque, PyObject *unused)
  414. {
  415. block *leftblock = deque->leftblock;
  416. block *rightblock = deque->rightblock;
  417. Py_ssize_t leftindex = deque->leftindex;
  418. Py_ssize_t rightindex = deque->rightindex;
  419. Py_ssize_t n = (deque->len)/2;
  420. Py_ssize_t i;
  421. PyObject *tmp;
  422. for (i=0 ; i<n ; i++) {
  423. /* Validate that pointers haven't met in the middle */
  424. assert(leftblock != rightblock || leftindex < rightindex);
  425. /* Swap */
  426. tmp = leftblock->data[leftindex];
  427. leftblock->data[leftindex] = rightblock->data[rightindex];
  428. rightblock->data[rightindex] = tmp;
  429. /* Advance left block/index pair */
  430. leftindex++;
  431. if (leftindex == BLOCKLEN) {
  432. if (leftblock->rightlink == NULL)
  433. break;
  434. leftblock = leftblock->rightlink;
  435. leftindex = 0;
  436. }
  437. /* Step backwards with the right block/index pair */
  438. rightindex--;
  439. if (rightindex == -1) {
  440. if (rightblock->leftlink == NULL)
  441. break;
  442. rightblock = rightblock->leftlink;
  443. rightindex = BLOCKLEN - 1;
  444. }
  445. }
  446. Py_RETURN_NONE;
  447. }
  448. PyDoc_STRVAR(reverse_doc,
  449. "D.reverse() -- reverse *IN PLACE*");
  450. static PyObject *
  451. deque_count(dequeobject *deque, PyObject *v)
  452. {
  453. block *leftblock = deque->leftblock;
  454. Py_ssize_t leftindex = deque->leftindex;
  455. Py_ssize_t n = deque->len;
  456. Py_ssize_t i;
  457. Py_ssize_t count = 0;
  458. PyObject *item;
  459. long start_state = deque->state;
  460. int cmp;
  461. for (i=0 ; i<n ; i++) {
  462. item = leftblock->data[leftindex];
  463. cmp = PyObject_RichCompareBool(item, v, Py_EQ);
  464. if (cmp > 0)
  465. count++;
  466. else if (cmp < 0)
  467. return NULL;
  468. if (start_state != deque->state) {
  469. PyErr_SetString(PyExc_RuntimeError,
  470. "deque mutated during iteration");
  471. return NULL;
  472. }
  473. /* Advance left block/index pair */
  474. leftindex++;
  475. if (leftindex == BLOCKLEN) {
  476. if (leftblock->rightlink == NULL) /* can occur when i==n-1 */
  477. break;
  478. leftblock = leftblock->rightlink;
  479. leftindex = 0;
  480. }
  481. }
  482. return PyInt_FromSsize_t(count);
  483. }
  484. PyDoc_STRVAR(count_doc,
  485. "D.count(value) -> integer -- return number of occurrences of value");
  486. static Py_ssize_t
  487. deque_len(dequeobject *deque)
  488. {
  489. return deque->len;
  490. }
  491. static PyObject *
  492. deque_remove(dequeobject *deque, PyObject *value)
  493. {
  494. Py_ssize_t i, n=deque->len;
  495. for (i=0 ; i<n ; i++) {
  496. PyObject *item = deque->leftblock->data[deque->leftindex];
  497. int cmp = PyObject_RichCompareBool(item, value, Py_EQ);
  498. if (deque->len != n) {
  499. PyErr_SetString(PyExc_IndexError,
  500. "deque mutated during remove().");
  501. return NULL;
  502. }
  503. if (cmp > 0) {
  504. PyObject *tgt = deque_popleft(deque, NULL);
  505. assert (tgt != NULL);
  506. Py_DECREF(tgt);
  507. if (_deque_rotate(deque, i) == -1)
  508. return NULL;
  509. Py_RETURN_NONE;
  510. }
  511. else if (cmp < 0) {
  512. _deque_rotate(deque, i);
  513. return NULL;
  514. }
  515. _deque_rotate(deque, -1);
  516. }
  517. PyErr_SetString(PyExc_ValueError, "deque.remove(x): x not in deque");
  518. return NULL;
  519. }
  520. PyDoc_STRVAR(remove_doc,
  521. "D.remove(value) -- remove first occurrence of value.");
  522. static int
  523. deque_clear(dequeobject *deque)
  524. {
  525. PyObject *item;
  526. while (deque->len) {
  527. item = deque_pop(deque, NULL);
  528. assert (item != NULL);
  529. Py_DECREF(item);
  530. }
  531. assert(deque->leftblock == deque->rightblock &&
  532. deque->leftindex - 1 == deque->rightindex &&
  533. deque->len == 0);
  534. return 0;
  535. }
  536. static PyObject *
  537. deque_item(dequeobject *deque, Py_ssize_t i)
  538. {
  539. block *b;
  540. PyObject *item;
  541. Py_ssize_t n, index=i;
  542. if (i < 0 || i >= deque->len) {
  543. PyErr_SetString(PyExc_IndexError,
  544. "deque index out of range");
  545. return NULL;
  546. }
  547. if (i == 0) {
  548. i = deque->leftindex;
  549. b = deque->leftblock;
  550. } else if (i == deque->len - 1) {
  551. i = deque->rightindex;
  552. b = deque->rightblock;
  553. } else {
  554. i += deque->leftindex;
  555. n = i / BLOCKLEN;
  556. i %= BLOCKLEN;
  557. if (index < (deque->len >> 1)) {
  558. b = deque->leftblock;
  559. while (n--)
  560. b = b->rightlink;
  561. } else {
  562. n = (deque->leftindex + deque->len - 1) / BLOCKLEN - n;
  563. b = deque->rightblock;
  564. while (n--)
  565. b = b->leftlink;
  566. }
  567. }
  568. item = b->data[i];
  569. Py_INCREF(item);
  570. return item;
  571. }
  572. /* delitem() implemented in terms of rotate for simplicity and reasonable
  573. performance near the end points. If for some reason this method becomes
  574. popular, it is not hard to re-implement this using direct data movement
  575. (similar to code in list slice assignment) and achieve a two or threefold
  576. performance boost.
  577. */
  578. static int
  579. deque_del_item(dequeobject *deque, Py_ssize_t i)
  580. {
  581. PyObject *item;
  582. assert (i >= 0 && i < deque->len);
  583. if (_deque_rotate(deque, -i) == -1)
  584. return -1;
  585. item = deque_popleft(deque, NULL);
  586. assert (item != NULL);
  587. Py_DECREF(item);
  588. return _deque_rotate(deque, i);
  589. }
  590. static int
  591. deque_ass_item(dequeobject *deque, Py_ssize_t i, PyObject *v)
  592. {
  593. PyObject *old_value;
  594. block *b;
  595. Py_ssize_t n, len=deque->len, halflen=(len+1)>>1, index=i;
  596. if (i < 0 || i >= len) {
  597. PyErr_SetString(PyExc_IndexError,
  598. "deque index out of range");
  599. return -1;
  600. }
  601. if (v == NULL)
  602. return deque_del_item(deque, i);
  603. i += deque->leftindex;
  604. n = i / BLOCKLEN;
  605. i %= BLOCKLEN;
  606. if (index <= halflen) {
  607. b = deque->leftblock;
  608. while (n--)
  609. b = b->rightlink;
  610. } else {
  611. n = (deque->leftindex + len - 1) / BLOCKLEN - n;
  612. b = deque->rightblock;
  613. while (n--)
  614. b = b->leftlink;
  615. }
  616. Py_INCREF(v);
  617. old_value = b->data[i];
  618. b->data[i] = v;
  619. Py_DECREF(old_value);
  620. return 0;
  621. }
  622. static PyObject *
  623. deque_clearmethod(dequeobject *deque)
  624. {
  625. int rv;
  626. rv = deque_clear(deque);
  627. assert (rv != -1);
  628. Py_RETURN_NONE;
  629. }
  630. PyDoc_STRVAR(clear_doc, "Remove all elements from the deque.");
  631. static void
  632. deque_dealloc(dequeobject *deque)
  633. {
  634. PyObject_GC_UnTrack(deque);
  635. if (deque->weakreflist != NULL)
  636. PyObject_ClearWeakRefs((PyObject *) deque);
  637. if (deque->leftblock != NULL) {
  638. deque_clear(deque);
  639. assert(deque->leftblock != NULL);
  640. freeblock(deque->leftblock);
  641. }
  642. deque->leftblock = NULL;
  643. deque->rightblock = NULL;
  644. Py_TYPE(deque)->tp_free(deque);
  645. }
  646. static int
  647. deque_traverse(dequeobject *deque, visitproc visit, void *arg)
  648. {
  649. block *b;
  650. PyObject *item;
  651. Py_ssize_t index;
  652. Py_ssize_t indexlo = deque->leftindex;
  653. for (b = deque->leftblock; b != NULL; b = b->rightlink) {
  654. const Py_ssize_t indexhi = b == deque->rightblock ?
  655. deque->rightindex :
  656. BLOCKLEN - 1;
  657. for (index = indexlo; index <= indexhi; ++index) {
  658. item = b->data[index];
  659. Py_VISIT(item);
  660. }
  661. indexlo = 0;
  662. }
  663. return 0;
  664. }
  665. static PyObject *
  666. deque_copy(PyObject *deque)
  667. {
  668. if (((dequeobject *)deque)->maxlen == -1)
  669. return PyObject_CallFunction((PyObject *)(Py_TYPE(deque)), "O", deque, NULL);
  670. else
  671. return PyObject_CallFunction((PyObject *)(Py_TYPE(deque)), "Oi",
  672. deque, ((dequeobject *)deque)->maxlen, NULL);
  673. }
  674. PyDoc_STRVAR(copy_doc, "Return a shallow copy of a deque.");
  675. static PyObject *
  676. deque_reduce(dequeobject *deque)
  677. {
  678. PyObject *dict, *result, *aslist;
  679. dict = PyObject_GetAttrString((PyObject *)deque, "__dict__");
  680. if (dict == NULL)
  681. PyErr_Clear();
  682. aslist = PySequence_List((PyObject *)deque);
  683. if (aslist == NULL) {
  684. Py_XDECREF(dict);
  685. return NULL;
  686. }
  687. if (dict == NULL) {
  688. if (deque->maxlen == -1)
  689. result = Py_BuildValue("O(O)", Py_TYPE(deque), aslist);
  690. else
  691. result = Py_BuildValue("O(On)", Py_TYPE(deque), aslist, deque->maxlen);
  692. } else {
  693. if (deque->maxlen == -1)
  694. result = Py_BuildValue("O(OO)O", Py_TYPE(deque), aslist, Py_None, dict);
  695. else
  696. result = Py_BuildValue("O(On)O", Py_TYPE(deque), aslist, deque->maxlen, dict);
  697. }
  698. Py_XDECREF(dict);
  699. Py_DECREF(aslist);
  700. return result;
  701. }
  702. PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
  703. static PyObject *
  704. deque_repr(PyObject *deque)
  705. {
  706. PyObject *aslist, *result, *fmt;
  707. int i;
  708. i = Py_ReprEnter(deque);
  709. if (i != 0) {
  710. if (i < 0)
  711. return NULL;
  712. return PyString_FromString("[...]");
  713. }
  714. aslist = PySequence_List(deque);
  715. if (aslist == NULL) {
  716. Py_ReprLeave(deque);
  717. return NULL;
  718. }
  719. if (((dequeobject *)deque)->maxlen != -1)
  720. fmt = PyString_FromFormat("deque(%%r, maxlen=%zd)",
  721. ((dequeobject *)deque)->maxlen);
  722. else
  723. fmt = PyString_FromString("deque(%r)");
  724. if (fmt == NULL) {
  725. Py_DECREF(aslist);
  726. Py_ReprLeave(deque);
  727. return NULL;
  728. }
  729. result = PyString_Format(fmt, aslist);
  730. Py_DECREF(fmt);
  731. Py_DECREF(aslist);
  732. Py_ReprLeave(deque);
  733. return result;
  734. }
  735. static int
  736. deque_tp_print(PyObject *deque, FILE *fp, int flags)
  737. {
  738. PyObject *it, *item;
  739. char *emit = ""; /* No separator emitted on first pass */
  740. char *separator = ", ";
  741. int i;
  742. i = Py_ReprEnter(deque);
  743. if (i != 0) {
  744. if (i < 0)
  745. return i;
  746. Py_BEGIN_ALLOW_THREADS
  747. fputs("[...]", fp);
  748. Py_END_ALLOW_THREADS
  749. return 0;
  750. }
  751. it = PyObject_GetIter(deque);
  752. if (it == NULL)
  753. return -1;
  754. Py_BEGIN_ALLOW_THREADS
  755. fputs("deque([", fp);
  756. Py_END_ALLOW_THREADS
  757. while ((item = PyIter_Next(it)) != NULL) {
  758. Py_BEGIN_ALLOW_THREADS
  759. fputs(emit, fp);
  760. Py_END_ALLOW_THREADS
  761. emit = separator;
  762. if (PyObject_Print(item, fp, 0) != 0) {
  763. Py_DECREF(item);
  764. Py_DECREF(it);
  765. Py_ReprLeave(deque);
  766. return -1;
  767. }
  768. Py_DECREF(item);
  769. }
  770. Py_ReprLeave(deque);
  771. Py_DECREF(it);
  772. if (PyErr_Occurred())
  773. return -1;
  774. Py_BEGIN_ALLOW_THREADS
  775. if (((dequeobject *)deque)->maxlen == -1)
  776. fputs("])", fp);
  777. else
  778. fprintf(fp, "], maxlen=%" PY_FORMAT_SIZE_T "d)", ((dequeobject *)deque)->maxlen);
  779. Py_END_ALLOW_THREADS
  780. return 0;
  781. }
  782. static PyObject *
  783. deque_richcompare(PyObject *v, PyObject *w, int op)
  784. {
  785. PyObject *it1=NULL, *it2=NULL, *x, *y;
  786. Py_ssize_t vs, ws;
  787. int b, cmp=-1;
  788. if (!PyObject_TypeCheck(v, &deque_type) ||
  789. !PyObject_TypeCheck(w, &deque_type)) {
  790. Py_INCREF(Py_NotImplemented);
  791. return Py_NotImplemented;
  792. }
  793. /* Shortcuts */
  794. vs = ((dequeobject *)v)->len;
  795. ws = ((dequeobject *)w)->len;
  796. if (op == Py_EQ) {
  797. if (v == w)
  798. Py_RETURN_TRUE;
  799. if (vs != ws)
  800. Py_RETURN_FALSE;
  801. }
  802. if (op == Py_NE) {
  803. if (v == w)
  804. Py_RETURN_FALSE;
  805. if (vs != ws)
  806. Py_RETURN_TRUE;
  807. }
  808. /* Search for the first index where items are different */
  809. it1 = PyObject_GetIter(v);
  810. if (it1 == NULL)
  811. goto done;
  812. it2 = PyObject_GetIter(w);
  813. if (it2 == NULL)
  814. goto done;
  815. for (;;) {
  816. x = PyIter_Next(it1);
  817. if (x == NULL && PyErr_Occurred())
  818. goto done;
  819. y = PyIter_Next(it2);
  820. if (x == NULL || y == NULL)
  821. break;
  822. b = PyObject_RichCompareBool(x, y, Py_EQ);
  823. if (b == 0) {
  824. cmp = PyObject_RichCompareBool(x, y, op);
  825. Py_DECREF(x);
  826. Py_DECREF(y);
  827. goto done;
  828. }
  829. Py_DECREF(x);
  830. Py_DECREF(y);
  831. if (b == -1)
  832. goto done;
  833. }
  834. /* We reached the end of one deque or both */
  835. Py_XDECREF(x);
  836. Py_XDECREF(y);
  837. if (PyErr_Occurred())
  838. goto done;
  839. switch (op) {
  840. case Py_LT: cmp = y != NULL; break; /* if w was longer */
  841. case Py_LE: cmp = x == NULL; break; /* if v was not longer */
  842. case Py_EQ: cmp = x == y; break; /* if we reached the end of both */
  843. case Py_NE: cmp = x != y; break; /* if one deque continues */
  844. case Py_GT: cmp = x != NULL; break; /* if v was longer */
  845. case Py_GE: cmp = y == NULL; break; /* if w was not longer */
  846. }
  847. done:
  848. Py_XDECREF(it1);
  849. Py_XDECREF(it2);
  850. if (cmp == 1)
  851. Py_RETURN_TRUE;
  852. if (cmp == 0)
  853. Py_RETURN_FALSE;
  854. return NULL;
  855. }
  856. static int
  857. deque_init(dequeobject *deque, PyObject *args, PyObject *kwdargs)
  858. {
  859. PyObject *iterable = NULL;
  860. PyObject *maxlenobj = NULL;
  861. Py_ssize_t maxlen = -1;
  862. char *kwlist[] = {"iterable", "maxlen", 0};
  863. if (!PyArg_ParseTupleAndKeywords(args, kwdargs, "|OO:deque", kwlist, &iterable, &maxlenobj))
  864. return -1;
  865. if (maxlenobj != NULL && maxlenobj != Py_None) {
  866. maxlen = PyInt_AsSsize_t(maxlenobj);
  867. if (maxlen == -1 && PyErr_Occurred())
  868. return -1;
  869. if (maxlen < 0) {
  870. PyErr_SetString(PyExc_ValueError, "maxlen must be non-negative");
  871. return -1;
  872. }
  873. }
  874. deque->maxlen = maxlen;
  875. deque_clear(deque);
  876. if (iterable != NULL) {
  877. PyObject *rv = deque_extend(deque, iterable);
  878. if (rv == NULL)
  879. return -1;
  880. Py_DECREF(rv);
  881. }
  882. return 0;
  883. }
  884. static PyObject *
  885. deque_get_maxlen(dequeobject *deque)
  886. {
  887. if (deque->maxlen == -1)
  888. Py_RETURN_NONE;
  889. return PyInt_FromSsize_t(deque->maxlen);
  890. }
  891. static PyGetSetDef deque_getset[] = {
  892. {"maxlen", (getter)deque_get_maxlen, (setter)NULL,
  893. "maximum size of a deque or None if unbounded"},
  894. {0}
  895. };
  896. static PySequenceMethods deque_as_sequence = {
  897. (lenfunc)deque_len, /* sq_length */
  898. 0, /* sq_concat */
  899. 0, /* sq_repeat */
  900. (ssizeargfunc)deque_item, /* sq_item */
  901. 0, /* sq_slice */
  902. (ssizeobjargproc)deque_ass_item, /* sq_ass_item */
  903. 0, /* sq_ass_slice */
  904. 0, /* sq_contains */
  905. (binaryfunc)deque_inplace_concat, /* sq_inplace_concat */
  906. 0, /* sq_inplace_repeat */
  907. };
  908. /* deque object ********************************************************/
  909. static PyObject *deque_iter(dequeobject *deque);
  910. static PyObject *deque_reviter(dequeobject *deque);
  911. PyDoc_STRVAR(reversed_doc,
  912. "D.__reversed__() -- return a reverse iterator over the deque");
  913. static PyMethodDef deque_methods[] = {
  914. {"append", (PyCFunction)deque_append,
  915. METH_O, append_doc},
  916. {"appendleft", (PyCFunction)deque_appendleft,
  917. METH_O, appendleft_doc},
  918. {"clear", (PyCFunction)deque_clearmethod,
  919. METH_NOARGS, clear_doc},
  920. {"__copy__", (PyCFunction)deque_copy,
  921. METH_NOARGS, copy_doc},
  922. {"count", (PyCFunction)deque_count,
  923. METH_O, count_doc},
  924. {"extend", (PyCFunction)deque_extend,
  925. METH_O, extend_doc},
  926. {"extendleft", (PyCFunction)deque_extendleft,
  927. METH_O, extendleft_doc},
  928. {"pop", (PyCFunction)deque_pop,
  929. METH_NOARGS, pop_doc},
  930. {"popleft", (PyCFunction)deque_popleft,
  931. METH_NOARGS, popleft_doc},
  932. {"__reduce__", (PyCFunction)deque_reduce,
  933. METH_NOARGS, reduce_doc},
  934. {"remove", (PyCFunction)deque_remove,
  935. METH_O, remove_doc},
  936. {"__reversed__", (PyCFunction)deque_reviter,
  937. METH_NOARGS, reversed_doc},
  938. {"reverse", (PyCFunction)deque_reverse,
  939. METH_NOARGS, reverse_doc},
  940. {"rotate", (PyCFunction)deque_rotate,
  941. METH_VARARGS, rotate_doc},
  942. {NULL, NULL} /* sentinel */
  943. };
  944. PyDoc_STRVAR(deque_doc,
  945. "deque(iterable[, maxlen]) --> deque object\n\
  946. \n\
  947. Build an ordered collection with optimized access from its endpoints.");
  948. static PyTypeObject deque_type = {
  949. PyVarObject_HEAD_INIT(NULL, 0)
  950. "collections.deque", /* tp_name */
  951. sizeof(dequeobject), /* tp_basicsize */
  952. 0, /* tp_itemsize */
  953. /* methods */
  954. (destructor)deque_dealloc, /* tp_dealloc */
  955. deque_tp_print, /* tp_print */
  956. 0, /* tp_getattr */
  957. 0, /* tp_setattr */
  958. 0, /* tp_compare */
  959. deque_repr, /* tp_repr */
  960. 0, /* tp_as_number */
  961. &deque_as_sequence, /* tp_as_sequence */
  962. 0, /* tp_as_mapping */
  963. (hashfunc)PyObject_HashNotImplemented, /* tp_hash */
  964. 0, /* tp_call */
  965. 0, /* tp_str */
  966. PyObject_GenericGetAttr, /* tp_getattro */
  967. 0, /* tp_setattro */
  968. 0, /* tp_as_buffer */
  969. Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC |
  970. Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */
  971. deque_doc, /* tp_doc */
  972. (traverseproc)deque_traverse, /* tp_traverse */
  973. (inquiry)deque_clear, /* tp_clear */
  974. (richcmpfunc)deque_richcompare, /* tp_richcompare */
  975. offsetof(dequeobject, weakreflist), /* tp_weaklistoffset*/
  976. (getiterfunc)deque_iter, /* tp_iter */
  977. 0, /* tp_iternext */
  978. deque_methods, /* tp_methods */
  979. 0, /* tp_members */
  980. deque_getset, /* tp_getset */
  981. 0, /* tp_base */
  982. 0, /* tp_dict */
  983. 0, /* tp_descr_get */
  984. 0, /* tp_descr_set */
  985. 0, /* tp_dictoffset */
  986. (initproc)deque_init, /* tp_init */
  987. PyType_GenericAlloc, /* tp_alloc */
  988. deque_new, /* tp_new */
  989. PyObject_GC_Del, /* tp_free */
  990. };
  991. /*********************** Deque Iterator **************************/
  992. typedef struct {
  993. PyObject_HEAD
  994. Py_ssize_t index;
  995. block *b;
  996. dequeobject *deque;
  997. long state; /* state when the iterator is created */
  998. Py_ssize_t counter; /* number of items remaining for iteration */
  999. } dequeiterobject;
  1000. static PyTypeObject dequeiter_type;
  1001. static PyObject *
  1002. deque_iter(dequeobject *deque)
  1003. {
  1004. dequeiterobject *it;
  1005. it = PyObject_GC_New(dequeiterobject, &dequeiter_type);
  1006. if (it == NULL)
  1007. return NULL;
  1008. it->b = deque->leftblock;
  1009. it->index = deque->leftindex;
  1010. Py_INCREF(deque);
  1011. it->deque = deque;
  1012. it->state = deque->state;
  1013. it->counter = deque->len;
  1014. PyObject_GC_Track(it);
  1015. return (PyObject *)it;
  1016. }
  1017. static int
  1018. dequeiter_traverse(dequeiterobject *dio, visitproc visit, void *arg)
  1019. {
  1020. Py_VISIT(dio->deque);
  1021. return 0;
  1022. }
  1023. static void
  1024. dequeiter_dealloc(dequeiterobject *dio)
  1025. {
  1026. Py_XDECREF(dio->deque);
  1027. PyObject_GC_Del(dio);
  1028. }
  1029. static PyObject *
  1030. dequeiter_next(dequeiterobject *it)
  1031. {
  1032. PyObject *item;
  1033. if (it->deque->state != it->state) {
  1034. it->counter = 0;
  1035. PyErr_SetString(PyExc_RuntimeError,
  1036. "deque mutated during iteration");
  1037. return NULL;
  1038. }
  1039. if (it->counter == 0)
  1040. return NULL;
  1041. assert (!(it->b == it->deque->rightblock &&
  1042. it->index > it->deque->rightindex));
  1043. item = it->b->data[it->index];
  1044. it->index++;
  1045. it->counter--;
  1046. if (it->index == BLOCKLEN && it->counter > 0) {
  1047. assert (it->b->rightlink != NULL);
  1048. it->b = it->b->rightlink;
  1049. it->index = 0;
  1050. }
  1051. Py_INCREF(item);
  1052. return item;
  1053. }
  1054. static PyObject *
  1055. dequeiter_len(dequeiterobject *it)
  1056. {
  1057. return PyInt_FromLong(it->counter);
  1058. }
  1059. PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
  1060. static PyMethodDef dequeiter_methods[] = {
  1061. {"__length_hint__", (PyCFunction)dequeiter_len, METH_NOARGS, length_hint_doc},
  1062. {NULL, NULL} /* sentinel */
  1063. };
  1064. static PyTypeObject dequeiter_type = {
  1065. PyVarObject_HEAD_INIT(NULL, 0)
  1066. "deque_iterator", /* tp_name */
  1067. sizeof(dequeiterobject), /* tp_basicsize */
  1068. 0, /* tp_itemsize */
  1069. /* methods */
  1070. (destructor)dequeiter_dealloc, /* tp_dealloc */
  1071. 0, /* tp_print */
  1072. 0, /* tp_getattr */
  1073. 0, /* tp_setattr */
  1074. 0, /* tp_compare */
  1075. 0, /* tp_repr */
  1076. 0, /* tp_as_number */
  1077. 0, /* tp_as_sequence */
  1078. 0, /* tp_as_mapping */
  1079. 0, /* tp_hash */
  1080. 0, /* tp_call */
  1081. 0, /* tp_str */
  1082. PyObject_GenericGetAttr, /* tp_getattro */
  1083. 0, /* tp_setattro */
  1084. 0, /* tp_as_buffer */
  1085. Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
  1086. 0, /* tp_doc */
  1087. (traverseproc)dequeiter_traverse, /* tp_traverse */
  1088. 0, /* tp_clear */
  1089. 0, /* tp_richcompare */
  1090. 0, /* tp_weaklistoffset */
  1091. PyObject_SelfIter, /* tp_iter */
  1092. (iternextfunc)dequeiter_next, /* tp_iternext */
  1093. dequeiter_methods, /* tp_methods */
  1094. 0,
  1095. };
  1096. /*********************** Deque Reverse Iterator **************************/
  1097. static PyTypeObject dequereviter_type;
  1098. static PyObject *
  1099. deque_reviter(dequeobject *deque)
  1100. {
  1101. dequeiterobject *it;
  1102. it = PyObject_GC_New(dequeiterobject, &dequereviter_type);
  1103. if (it == NULL)
  1104. return NULL;
  1105. it->b = deque->rightblock;
  1106. it->index = deque->rightindex;
  1107. Py_INCREF(deque);
  1108. it->deque = deque;
  1109. it->state = deque->state;
  1110. it->counter = deque->len;
  1111. PyObject_GC_Track(it);
  1112. return (PyObject *)it;
  1113. }
  1114. static PyObject *
  1115. dequereviter_next(dequeiterobject *it)
  1116. {
  1117. PyObject *item;
  1118. if (it->counter == 0)
  1119. return NULL;
  1120. if (it->deque->state != it->state) {
  1121. it->counter = 0;
  1122. PyErr_SetString(PyExc_RuntimeError,
  1123. "deque mutated during iteration");
  1124. return NULL;
  1125. }
  1126. assert (!(it->b == it->deque->leftblock &&
  1127. it->index < it->deque->leftindex));
  1128. item = it->b->data[it->index];
  1129. it->index--;
  1130. it->counter--;
  1131. if (it->index == -1 && it->counter > 0) {
  1132. assert (it->b->leftlink != NULL);
  1133. it->b = it->b->leftlink;
  1134. it->index = BLOCKLEN - 1;
  1135. }
  1136. Py_INCREF(item);
  1137. return item;
  1138. }
  1139. static PyTypeObject dequereviter_type = {
  1140. PyVarObject_HEAD_INIT(NULL, 0)
  1141. "deque_reverse_iterator", /* tp_name */
  1142. sizeof(dequeiterobject), /* tp_basicsize */
  1143. 0, /* tp_itemsize */
  1144. /* methods */
  1145. (destructor)dequeiter_dealloc, /* tp_dealloc */
  1146. 0, /* tp_print */
  1147. 0, /* tp_getattr */
  1148. 0, /* tp_setattr */
  1149. 0, /* tp_compare */
  1150. 0, /* tp_repr */
  1151. 0, /* tp_as_number */
  1152. 0, /* tp_as_sequence */
  1153. 0, /* tp_as_mapping */
  1154. 0, /* tp_hash */
  1155. 0, /* tp_call */
  1156. 0, /* tp_str */
  1157. PyObject_GenericGetAttr, /* tp_getattro */
  1158. 0, /* tp_setattro */
  1159. 0, /* tp_as_buffer */
  1160. Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
  1161. 0, /* tp_doc */
  1162. (traverseproc)dequeiter_traverse, /* tp_traverse */
  1163. 0, /* tp_clear */
  1164. 0, /* tp_richcompare */
  1165. 0, /* tp_weaklistoffset */
  1166. PyObject_SelfIter, /* tp_iter */
  1167. (iternextfunc)dequereviter_next, /* tp_iternext */
  1168. dequeiter_methods, /* tp_methods */
  1169. 0,
  1170. };
  1171. /* defaultdict type *********************************************************/
  1172. typedef struct {
  1173. PyDictObject dict;
  1174. PyObject *default_factory;
  1175. } defdictobject;
  1176. static PyTypeObject defdict_type; /* Forward */
  1177. PyDoc_STRVAR(defdict_missing_doc,
  1178. "__missing__(key) # Called by __getitem__ for missing key; pseudo-code:\n\
  1179. if self.default_factory is None: raise KeyError((key,))\n\
  1180. self[key] = value = self.default_factory()\n\
  1181. return value\n\
  1182. ");
  1183. static PyObject *
  1184. defdict_missing(defdictobject *dd, PyObject *key)
  1185. {
  1186. PyObject *factory = dd->default_factory;
  1187. PyObject *value;
  1188. if (factory == NULL || factory == Py_None) {
  1189. /* XXX Call dict.__missing__(key) */
  1190. PyObject *tup;
  1191. tup = PyTuple_Pack(1, key);
  1192. if (!tup) return NULL;
  1193. PyErr_SetObject(PyExc_KeyError, tup);
  1194. Py_DECREF(tup);
  1195. return NULL;
  1196. }
  1197. value = PyEval_CallObject(factory, NULL);
  1198. if (value == NULL)
  1199. return value;
  1200. if (PyObject_SetItem((PyObject *)dd, key, value) < 0) {
  1201. Py_DECREF(value);
  1202. return NULL;
  1203. }
  1204. return value;
  1205. }
  1206. PyDoc_STRVAR(defdict_copy_doc, "D.copy() -> a shallow copy of D.");
  1207. static PyObject *
  1208. defdict_copy(defdictobject *dd)
  1209. {
  1210. /* This calls the object's class. That only works for subclasses
  1211. whose class constructor has the same signature. Subclasses that
  1212. define a different constructor signature must override copy().
  1213. */
  1214. if (dd->default_factory == NULL)
  1215. return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd), Py_None, dd, NULL);
  1216. return PyObject_CallFunctionObjArgs((PyObject*)Py_TYPE(dd),
  1217. dd->default_factory, dd, NULL);
  1218. }
  1219. static PyObject *
  1220. defdict_reduce(defdictobject *dd)
  1221. {
  1222. /* __reduce__ must return a 5-tuple as follows:
  1223. - factory function
  1224. - tuple of args for the factory function
  1225. - additional state (here None)
  1226. - sequence iterator (here None)
  1227. - dictionary iterator (yielding successive (key, value) pairs
  1228. This API is used by pickle.py and copy.py.
  1229. For this to be useful with pickle.py, the default_factory
  1230. must be picklable; e.g., None, a built-in, or a global
  1231. function in a module or package.
  1232. Both shallow and deep copying are supported, but for deep
  1233. copying, the default_factory must be deep-copyable; e.g. None,
  1234. or a built-in (functions are not copyable at this time).
  1235. This only works for subclasses as long as their constructor
  1236. signature is compatible; the first argument must be the
  1237. optional default_factory, defaulting to None.
  1238. */
  1239. PyObject *args;
  1240. PyObject *items;
  1241. PyObject *result;
  1242. if (dd->default_factory == NULL || dd->default_factory == Py_None)
  1243. args = PyTuple_New(0);
  1244. else
  1245. args = PyTuple_Pack(1, dd->default_factory);
  1246. if (args == NULL)
  1247. return NULL;
  1248. items = PyObject_CallMethod((PyObject *)dd, "iteritems", "()");
  1249. if (items == NULL) {
  1250. Py_DECREF(args);
  1251. return NULL;
  1252. }
  1253. result = PyTuple_Pack(5, Py_TYPE(dd), args,
  1254. Py_None, Py_None, items);
  1255. Py_DECREF(items);
  1256. Py_DECREF(args);
  1257. return result;
  1258. }
  1259. static PyMethodDef defdict_methods[] = {
  1260. {"__missing__", (PyCFunction)defdict_missing, METH_O,
  1261. defdict_missing_doc},
  1262. {"copy", (PyCFunction)defdict_copy, METH_NOARGS,
  1263. defdict_copy_doc},
  1264. {"__copy__", (PyCFunction)defdict_copy, METH_NOARGS,
  1265. defdict_copy_doc},
  1266. {"__reduce__", (PyCFunction)defdict_reduce, METH_NOARGS,
  1267. reduce_doc},
  1268. {NULL}
  1269. };
  1270. static PyMemberDef defdict_members[] = {
  1271. {"default_factory", T_OBJECT,
  1272. offsetof(defdictobject, default_factory), 0,
  1273. PyDoc_STR("Factory for default value called by __missing__().")},
  1274. {NULL}
  1275. };
  1276. static void
  1277. defdict_dealloc(defdictobject *dd)
  1278. {
  1279. Py_CLEAR(dd->default_factory);
  1280. PyDict_Type.tp_dealloc((PyObject *)dd);
  1281. }
  1282. static int
  1283. defdict_print(defdictobject *dd, FILE *fp, int flags)
  1284. {
  1285. int sts;
  1286. Py_BEGIN_ALLOW_THREADS
  1287. fprintf(fp, "defaultdict(");
  1288. Py_END_ALLOW_THREADS
  1289. if (dd->default_factory == NULL) {
  1290. Py_BEGIN_ALLOW_THREADS
  1291. fprintf(fp, "None");
  1292. Py_END_ALLOW_THREADS
  1293. } else {
  1294. PyObject_Print(dd->default_factory, fp, 0);
  1295. }
  1296. Py_BEGIN_ALLOW_THREADS
  1297. fprintf(fp, ", ");
  1298. Py_END_ALLOW_THREADS
  1299. sts = PyDict_Type.tp_print((PyObject *)dd, fp, 0);
  1300. Py_BEGIN_ALLOW_THREADS
  1301. fprintf(fp, ")");
  1302. Py_END_ALLOW_THREADS
  1303. return sts;
  1304. }
  1305. static PyObject *
  1306. defdict_repr(defdictobject *dd)
  1307. {
  1308. PyObject *defrepr;
  1309. PyObject *baserepr;
  1310. PyObject *result;
  1311. baserepr = PyDict_Type.tp_repr((PyObject *)dd);
  1312. if (baserepr == NULL)
  1313. return NULL;
  1314. if (dd->default_factory == NULL)
  1315. defrepr = PyString_FromString("None");
  1316. else
  1317. {
  1318. int status = Py_ReprEnter(dd->default_factory);
  1319. if (status != 0) {
  1320. if (status < 0) {
  1321. Py_DECREF(baserepr);
  1322. return NULL;
  1323. }
  1324. defrepr = PyString_FromString("...");
  1325. }
  1326. else
  1327. defrepr = PyObject_Repr(dd->default_factory);
  1328. Py_ReprLeave(dd->default_factory);
  1329. }
  1330. if (defrepr == NULL) {
  1331. Py_DECREF(baserepr);
  1332. return NULL;
  1333. }
  1334. result = PyString_FromFormat("defaultdict(%s, %s)",
  1335. PyString_AS_STRING(defrepr),
  1336. PyString_AS_STRING(baserepr));
  1337. Py_DECREF(defrepr);
  1338. Py_DECREF(baserepr);
  1339. return result;
  1340. }
  1341. static int
  1342. defdict_traverse(PyObject *self, visitproc visit, void *arg)
  1343. {
  1344. Py_VISIT(((defdictobject *)self)->default_factory);
  1345. return PyDict_Type.tp_traverse(self, visit, arg);
  1346. }
  1347. static int
  1348. defdict_tp_clear(defdictobject *dd)
  1349. {
  1350. Py_CLEAR(dd->default_factory);
  1351. return PyDict_Type.tp_clear((PyObject *)dd);
  1352. }
  1353. static int
  1354. defdict_init(PyObject *self, PyObject *args, PyObject *kwds)
  1355. {
  1356. defdictobject *dd = (defdictobject *)self;
  1357. PyObject *olddefault = dd->default_factory;
  1358. PyObject *newdefault = NULL;
  1359. PyObject *newargs;
  1360. int result;
  1361. if (args == NULL || !PyTuple_Check(args))
  1362. newargs = PyTuple_New(0);
  1363. else {
  1364. Py_ssize_t n = PyTuple_GET_SIZE(args);
  1365. if (n > 0) {
  1366. newdefault = PyTuple_GET_ITEM(args, 0);
  1367. if (!PyCallable_Check(newdefault) && newdefault != Py_None) {
  1368. PyErr_SetString(PyExc_TypeError,
  1369. "first argument must be callable");
  1370. return -1;
  1371. }
  1372. }
  1373. newargs = PySequence_GetSlice(args, 1, n);
  1374. }
  1375. if (newargs == NULL)
  1376. return -1;
  1377. Py_XINCREF(newdefault);
  1378. dd->default_factory = newdefault;
  1379. result = PyDict_Type.tp_init(self, newargs, kwds);
  1380. Py_DECREF(newargs);
  1381. Py_XDECREF(olddefault);
  1382. return result;
  1383. }
  1384. PyDoc_STRVAR(defdict_doc,
  1385. "defaultdict(default_factory) --> dict with default factory\n\
  1386. \n\
  1387. The default factory is called without arguments to produce\n\
  1388. a new value when a key is not present, in __getitem__ only.\n\
  1389. A defaultdict compares equal to a dict with the same items.\n\
  1390. ");
  1391. /* See comment in xxsubtype.c */
  1392. #define DEFERRED_ADDRESS(ADDR) 0
  1393. static PyTypeObject defdict_type = {
  1394. PyVarObject_HEAD_INIT(DEFERRED_ADDRESS(&PyType_Type), 0)
  1395. "collections.defaultdict", /* tp_name */
  1396. sizeof(defdictobject), /* tp_basicsize */
  1397. 0, /* tp_itemsize */
  1398. /* methods */
  1399. (destructor)defdict_dealloc, /* tp_dealloc */
  1400. (printfunc)defdict_print, /* tp_print */
  1401. 0, /* tp_getattr */
  1402. 0, /* tp_setattr */
  1403. 0, /* tp_compare */
  1404. (reprfunc)defdict_repr, /* tp_repr */
  1405. 0, /* tp_as_number */
  1406. 0, /* tp_as_sequence */
  1407. 0, /* tp_as_mapping */
  1408. 0, /* tp_hash */
  1409. 0, /* tp_call */
  1410. 0, /* tp_str */
  1411. PyObject_GenericGetAttr, /* tp_getattro */
  1412. 0, /* tp_setattro */
  1413. 0, /* tp_as_buffer */
  1414. Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC |
  1415. Py_TPFLAGS_HAVE_WEAKREFS, /* tp_flags */
  1416. defdict_doc, /* tp_doc */
  1417. defdict_traverse, /* tp_traverse */
  1418. (inquiry)defdict_tp_clear, /* tp_clear */
  1419. 0, /* tp_richcompare */
  1420. 0, /* tp_weaklistoffset*/
  1421. 0, /* tp_iter */
  1422. 0, /* tp_iternext */
  1423. defdict_methods, /* tp_methods */
  1424. defdict_members, /* tp_members */
  1425. 0, /* tp_getset */
  1426. DEFERRED_ADDRESS(&PyDict_Type), /* tp_base */
  1427. 0, /* tp_dict */
  1428. 0, /* tp_descr_get */
  1429. 0, /* tp_descr_set */
  1430. 0, /* tp_dictoffset */
  1431. defdict_init, /* tp_init */
  1432. PyType_GenericAlloc, /* tp_alloc */
  1433. 0, /* tp_new */
  1434. PyObject_GC_Del, /* tp_free */
  1435. };
  1436. /* module level code ********************************************************/
  1437. PyDoc_STRVAR(module_doc,
  1438. "High performance data structures.\n\
  1439. - deque: ordered collection accessible from endpoints only\n\
  1440. - defaultdict: dict subclass with a default value factory\n\
  1441. ");
  1442. PyMODINIT_FUNC
  1443. init_collections(void)
  1444. {
  1445. PyObject *m;
  1446. m = Py_InitModule3("_collections", NULL, module_doc);
  1447. if (m == NULL)
  1448. return;
  1449. if (PyType_Ready(&deque_type) < 0)
  1450. return;
  1451. Py_INCREF(&deque_type);
  1452. PyModule_AddObject(m, "deque", (PyObject *)&deque_type);
  1453. defdict_type.tp_base = &PyDict_Type;
  1454. if (PyType_Ready(&defdict_type) < 0)
  1455. return;
  1456. Py_INCREF(&defdict_type);
  1457. PyModule_AddObject(m, "defaultdict", (PyObject *)&defdict_type);
  1458. if (PyType_Ready(&dequeiter_type) < 0)
  1459. return;
  1460. if (PyType_Ready(&dequereviter_type) < 0)
  1461. return;
  1462. return;
  1463. }