blob: b65b8abc2806df0384e96c2ac79a35df66ad7c45 [file] [log] [blame]
Guido van Rossumf70e43a1991-02-19 12:39:46 +00001
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00002/* Tuple object implementation */
3
Guido van Rossumc0b618a1997-05-02 03:12:38 +00004#include "Python.h"
Victor Stinnera15e2602020-04-08 02:01:56 +02005#include "pycore_abstract.h" // _PyIndex_Check()
Victor Stinnere281f7d2018-11-01 02:30:36 +01006#include "pycore_accu.h"
Victor Stinnere5014be2020-04-14 17:52:15 +02007#include "pycore_gc.h" // _PyObject_GC_IS_TRACKED()
8#include "pycore_object.h"
Guido van Rossum85a5fbb1990-10-14 12:07:46 +00009
Serhiy Storchaka0b561592017-03-19 08:47:58 +020010/*[clinic input]
11class tuple "PyTupleObject *" "&PyTuple_Type"
12[clinic start generated code]*/
13/*[clinic end generated code: output=da39a3ee5e6b4b0d input=f051ba3cfdf9a189]*/
14
15#include "clinic/tupleobject.c.h"
16
Guido van Rossum5ce78f82000-04-21 21:15:05 +000017/* Speed optimization to avoid frequent malloc/free of small tuples */
Christian Heimes2202f872008-02-06 14:31:34 +000018#ifndef PyTuple_MAXSAVESIZE
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000019#define PyTuple_MAXSAVESIZE 20 /* Largest tuple to save on free list */
Guido van Rossum5ce78f82000-04-21 21:15:05 +000020#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000021#ifndef PyTuple_MAXFREELIST
Christian Heimes2202f872008-02-06 14:31:34 +000022#define PyTuple_MAXFREELIST 2000 /* Maximum number of tuples of each size to save */
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000023#endif
24
Christian Heimes2202f872008-02-06 14:31:34 +000025#if PyTuple_MAXSAVESIZE > 0
26/* Entries 1 up to PyTuple_MAXSAVESIZE are free lists, entry 0 is the empty
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000027 tuple () of which at most one instance will be allocated.
28*/
Christian Heimes2202f872008-02-06 14:31:34 +000029static PyTupleObject *free_list[PyTuple_MAXSAVESIZE];
30static int numfree[PyTuple_MAXSAVESIZE];
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000031#endif
Antoine Pitrou3a652b12009-03-23 18:52:06 +000032
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050033static inline void
34tuple_gc_track(PyTupleObject *op)
35{
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050036 _PyObject_GC_TRACK(op);
37}
38
David Malcolm49526f42012-06-22 14:55:41 -040039/* Print summary info about the state of the optimized allocator */
40void
41_PyTuple_DebugMallocStats(FILE *out)
42{
43#if PyTuple_MAXSAVESIZE > 0
44 int i;
45 char buf[128];
46 for (i = 1; i < PyTuple_MAXSAVESIZE; i++) {
47 PyOS_snprintf(buf, sizeof(buf),
48 "free %d-sized PyTupleObject", i);
49 _PyDebugAllocatorStats(out,
50 buf,
51 numfree[i], _PyObject_VAR_SIZE(&PyTuple_Type, i));
52 }
53#endif
54}
Antoine Pitrou3a652b12009-03-23 18:52:06 +000055
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050056/* Allocate an uninitialized tuple object. Before making it public following
57 steps must be done:
58 - initialize its items
59 - call tuple_gc_track() on it
60 Because the empty tuple is always reused and it's already tracked by GC,
61 this function must not be called with size == 0 (unless from PyTuple_New()
62 which wraps this function).
63*/
64static PyTupleObject *
65tuple_alloc(Py_ssize_t size)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +000066{
Antoine Pitrou9ed5f272013-08-13 20:18:52 +020067 PyTupleObject *op;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000068 if (size < 0) {
69 PyErr_BadInternalCall();
70 return NULL;
71 }
Christian Heimes2202f872008-02-06 14:31:34 +000072#if PyTuple_MAXSAVESIZE > 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000073 if (size < PyTuple_MAXSAVESIZE && (op = free_list[size]) != NULL) {
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050074 assert(size != 0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000075 free_list[size] = (PyTupleObject *) op->ob_item[0];
76 numfree[size]--;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000077 /* Inline PyObject_InitVar */
Guido van Rossum68055ce1998-12-11 14:56:38 +000078#ifdef Py_TRACE_REFS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000079 Py_SIZE(op) = size;
80 Py_TYPE(op) = &PyTuple_Type;
Guido van Rossum68055ce1998-12-11 14:56:38 +000081#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000082 _Py_NewReference((PyObject *)op);
83 }
84 else
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +000085#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000086 {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000087 /* Check for overflow */
Sergey Fedoseev755d4ef2019-09-10 01:40:58 +050088 if ((size_t)size > ((size_t)PY_SSIZE_T_MAX - (sizeof(PyTupleObject) -
89 sizeof(PyObject *))) / sizeof(PyObject *)) {
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050090 return (PyTupleObject *)PyErr_NoMemory();
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000091 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000092 op = PyObject_GC_NewVar(PyTupleObject, &PyTuple_Type, size);
93 if (op == NULL)
94 return NULL;
95 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +050096 return op;
97}
98
99PyObject *
100PyTuple_New(Py_ssize_t size)
101{
102 PyTupleObject *op;
103#if PyTuple_MAXSAVESIZE > 0
104 if (size == 0 && free_list[0]) {
105 op = free_list[0];
106 Py_INCREF(op);
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500107 return (PyObject *) op;
108 }
109#endif
110 op = tuple_alloc(size);
Zackery Spytz60bd1f82019-09-04 07:58:05 -0600111 if (op == NULL) {
112 return NULL;
113 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500114 for (Py_ssize_t i = 0; i < size; i++) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000115 op->ob_item[i] = NULL;
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500116 }
Christian Heimes2202f872008-02-06 14:31:34 +0000117#if PyTuple_MAXSAVESIZE > 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000118 if (size == 0) {
119 free_list[0] = op;
120 ++numfree[0];
121 Py_INCREF(op); /* extra INCREF so that this is never freed */
122 }
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000123#endif
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500124 tuple_gc_track(op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000125 return (PyObject *) op;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000126}
127
Martin v. Löwis18e16552006-02-15 17:27:45 +0000128Py_ssize_t
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200129PyTuple_Size(PyObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000130{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000131 if (!PyTuple_Check(op)) {
132 PyErr_BadInternalCall();
133 return -1;
134 }
135 else
136 return Py_SIZE(op);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000137}
138
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000139PyObject *
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200140PyTuple_GetItem(PyObject *op, Py_ssize_t i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000141{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000142 if (!PyTuple_Check(op)) {
143 PyErr_BadInternalCall();
144 return NULL;
145 }
146 if (i < 0 || i >= Py_SIZE(op)) {
147 PyErr_SetString(PyExc_IndexError, "tuple index out of range");
148 return NULL;
149 }
150 return ((PyTupleObject *)op) -> ob_item[i];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000151}
152
153int
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200154PyTuple_SetItem(PyObject *op, Py_ssize_t i, PyObject *newitem)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000155{
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200156 PyObject **p;
Victor Stinnera93c51e2020-02-07 00:38:59 +0100157 if (!PyTuple_Check(op) || Py_REFCNT(op) != 1) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000158 Py_XDECREF(newitem);
159 PyErr_BadInternalCall();
160 return -1;
161 }
162 if (i < 0 || i >= Py_SIZE(op)) {
163 Py_XDECREF(newitem);
164 PyErr_SetString(PyExc_IndexError,
165 "tuple assignment index out of range");
166 return -1;
167 }
168 p = ((PyTupleObject *)op) -> ob_item + i;
Serhiy Storchakaec397562016-04-06 09:50:03 +0300169 Py_XSETREF(*p, newitem);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000170 return 0;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000171}
172
Antoine Pitrou3a652b12009-03-23 18:52:06 +0000173void
174_PyTuple_MaybeUntrack(PyObject *op)
175{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000176 PyTupleObject *t;
177 Py_ssize_t i, n;
178
179 if (!PyTuple_CheckExact(op) || !_PyObject_GC_IS_TRACKED(op))
180 return;
181 t = (PyTupleObject *) op;
182 n = Py_SIZE(t);
183 for (i = 0; i < n; i++) {
184 PyObject *elt = PyTuple_GET_ITEM(t, i);
185 /* Tuple with NULL elements aren't
186 fully constructed, don't untrack
187 them yet. */
188 if (!elt ||
189 _PyObject_GC_MAY_BE_TRACKED(elt))
190 return;
191 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000192 _PyObject_GC_UNTRACK(op);
Antoine Pitrou3a652b12009-03-23 18:52:06 +0000193}
194
Raymond Hettingercb2da432003-10-12 18:24:34 +0000195PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000196PyTuple_Pack(Py_ssize_t n, ...)
Raymond Hettingercb2da432003-10-12 18:24:34 +0000197{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000198 Py_ssize_t i;
199 PyObject *o;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000200 PyObject **items;
201 va_list vargs;
Raymond Hettingercb2da432003-10-12 18:24:34 +0000202
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500203 if (n == 0) {
204 return PyTuple_New(0);
205 }
206
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000207 va_start(vargs, n);
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500208 PyTupleObject *result = tuple_alloc(n);
Christian Heimesd5a88042012-09-10 02:54:51 +0200209 if (result == NULL) {
210 va_end(vargs);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000211 return NULL;
Christian Heimesd5a88042012-09-10 02:54:51 +0200212 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500213 items = result->ob_item;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000214 for (i = 0; i < n; i++) {
215 o = va_arg(vargs, PyObject *);
216 Py_INCREF(o);
217 items[i] = o;
218 }
219 va_end(vargs);
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500220 tuple_gc_track(result);
221 return (PyObject *)result;
Raymond Hettingercb2da432003-10-12 18:24:34 +0000222}
223
224
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000225/* Methods */
226
227static void
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200228tupledealloc(PyTupleObject *op)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000229{
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200230 Py_ssize_t i;
231 Py_ssize_t len = Py_SIZE(op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000232 PyObject_GC_UnTrack(op);
Jeroen Demeyer351c6742019-05-10 19:21:11 +0200233 Py_TRASHCAN_BEGIN(op, tupledealloc)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000234 if (len > 0) {
235 i = len;
236 while (--i >= 0)
237 Py_XDECREF(op->ob_item[i]);
Christian Heimes2202f872008-02-06 14:31:34 +0000238#if PyTuple_MAXSAVESIZE > 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000239 if (len < PyTuple_MAXSAVESIZE &&
240 numfree[len] < PyTuple_MAXFREELIST &&
Andy Lesterdffe4c02020-03-04 07:15:20 -0600241 Py_IS_TYPE(op, &PyTuple_Type))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000242 {
243 op->ob_item[0] = (PyObject *) free_list[len];
244 numfree[len]++;
245 free_list[len] = op;
246 goto done; /* return */
247 }
Sjoerd Mullender842d2cc1993-10-15 16:18:48 +0000248#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000249 }
250 Py_TYPE(op)->tp_free((PyObject *)op);
Guido van Rossumd724b232000-03-13 16:01:29 +0000251done:
Jeroen Demeyer351c6742019-05-10 19:21:11 +0200252 Py_TRASHCAN_END
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000253}
254
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000255static PyObject *
Fred Drakeba096332000-07-09 07:04:36 +0000256tuplerepr(PyTupleObject *v)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000257{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000258 Py_ssize_t i, n;
Victor Stinner88a9cd92013-11-19 12:59:46 +0100259 _PyUnicodeWriter writer;
Tim Petersa7259592001-06-16 05:11:17 +0000260
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000261 n = Py_SIZE(v);
262 if (n == 0)
263 return PyUnicode_FromString("()");
Tim Petersa7259592001-06-16 05:11:17 +0000264
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000265 /* While not mutable, it is still possible to end up with a cycle in a
266 tuple through an object that stores itself within a tuple (and thus
267 infinitely asks for the repr of itself). This should only be
268 possible within a type. */
269 i = Py_ReprEnter((PyObject *)v);
270 if (i != 0) {
271 return i > 0 ? PyUnicode_FromString("(...)") : NULL;
272 }
Guido van Rossum8ce8a782007-11-01 19:42:39 +0000273
Victor Stinner88a9cd92013-11-19 12:59:46 +0100274 _PyUnicodeWriter_Init(&writer);
275 writer.overallocate = 1;
276 if (Py_SIZE(v) > 1) {
277 /* "(" + "1" + ", 2" * (len - 1) + ")" */
278 writer.min_length = 1 + 1 + (2 + 1) * (Py_SIZE(v) - 1) + 1;
279 }
280 else {
281 /* "(1,)" */
282 writer.min_length = 4;
283 }
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200284
Victor Stinner88a9cd92013-11-19 12:59:46 +0100285 if (_PyUnicodeWriter_WriteChar(&writer, '(') < 0)
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200286 goto error;
Tim Petersa7259592001-06-16 05:11:17 +0000287
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000288 /* Do repr() on each element. */
289 for (i = 0; i < n; ++i) {
Victor Stinner88a9cd92013-11-19 12:59:46 +0100290 PyObject *s;
291
292 if (i > 0) {
293 if (_PyUnicodeWriter_WriteASCIIString(&writer, ", ", 2) < 0)
294 goto error;
295 }
296
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000297 s = PyObject_Repr(v->ob_item[i]);
Victor Stinner88a9cd92013-11-19 12:59:46 +0100298 if (s == NULL)
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200299 goto error;
Victor Stinner88a9cd92013-11-19 12:59:46 +0100300
301 if (_PyUnicodeWriter_WriteStr(&writer, s) < 0) {
302 Py_DECREF(s);
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200303 goto error;
Victor Stinner88a9cd92013-11-19 12:59:46 +0100304 }
305 Py_DECREF(s);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000306 }
Victor Stinner88a9cd92013-11-19 12:59:46 +0100307
308 writer.overallocate = 0;
309 if (n > 1) {
310 if (_PyUnicodeWriter_WriteChar(&writer, ')') < 0)
311 goto error;
312 }
313 else {
314 if (_PyUnicodeWriter_WriteASCIIString(&writer, ",)", 2) < 0)
315 goto error;
316 }
Tim Petersa7259592001-06-16 05:11:17 +0000317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000318 Py_ReprLeave((PyObject *)v);
Victor Stinner88a9cd92013-11-19 12:59:46 +0100319 return _PyUnicodeWriter_Finish(&writer);
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200320
321error:
Victor Stinner88a9cd92013-11-19 12:59:46 +0100322 _PyUnicodeWriter_Dealloc(&writer);
Antoine Pitroueeb7eea2011-10-06 18:57:27 +0200323 Py_ReprLeave((PyObject *)v);
324 return NULL;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000325}
326
Raymond Hettinger4ec44e82004-06-04 06:35:20 +0000327
jdemeyeraeb1be52018-10-28 02:06:38 +0200328/* Hash for tuples. This is a slightly simplified version of the xxHash
329 non-cryptographic hash:
330 - we do not use any parallellism, there is only 1 accumulator.
331 - we drop the final mixing since this is just a permutation of the
332 output space: it does not help against collisions.
333 - at the end, we mangle the length with a single constant.
334 For the xxHash specification, see
335 https://github.com/Cyan4973/xxHash/blob/master/doc/xxhash_spec.md
Christian Heimes34bdeb52013-01-07 21:24:18 +0100336
jdemeyeraeb1be52018-10-28 02:06:38 +0200337 Below are the official constants from the xxHash specification. Optimizing
338 compilers should emit a single "rotate" instruction for the
339 _PyHASH_XXROTATE() expansion. If that doesn't happen for some important
340 platform, the macro could be changed to expand to a platform-specific rotate
341 spelling instead.
Raymond Hettinger4ec44e82004-06-04 06:35:20 +0000342*/
jdemeyeraeb1be52018-10-28 02:06:38 +0200343#if SIZEOF_PY_UHASH_T > 4
344#define _PyHASH_XXPRIME_1 ((Py_uhash_t)11400714785074694791ULL)
345#define _PyHASH_XXPRIME_2 ((Py_uhash_t)14029467366897019727ULL)
346#define _PyHASH_XXPRIME_5 ((Py_uhash_t)2870177450012600261ULL)
347#define _PyHASH_XXROTATE(x) ((x << 31) | (x >> 33)) /* Rotate left 31 bits */
348#else
349#define _PyHASH_XXPRIME_1 ((Py_uhash_t)2654435761UL)
350#define _PyHASH_XXPRIME_2 ((Py_uhash_t)2246822519UL)
351#define _PyHASH_XXPRIME_5 ((Py_uhash_t)374761393UL)
352#define _PyHASH_XXROTATE(x) ((x << 13) | (x >> 19)) /* Rotate left 13 bits */
353#endif
Raymond Hettinger4ec44e82004-06-04 06:35:20 +0000354
jdemeyeraeb1be52018-10-28 02:06:38 +0200355/* Tests have shown that it's not worth to cache the hash value, see
356 https://bugs.python.org/issue9685 */
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000357static Py_hash_t
Fred Drakeba096332000-07-09 07:04:36 +0000358tuplehash(PyTupleObject *v)
Guido van Rossum9bfef441993-03-29 10:43:31 +0000359{
jdemeyeraeb1be52018-10-28 02:06:38 +0200360 Py_ssize_t i, len = Py_SIZE(v);
361 PyObject **item = v->ob_item;
362
363 Py_uhash_t acc = _PyHASH_XXPRIME_5;
364 for (i = 0; i < len; i++) {
365 Py_uhash_t lane = PyObject_Hash(item[i]);
366 if (lane == (Py_uhash_t)-1) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000367 return -1;
jdemeyeraeb1be52018-10-28 02:06:38 +0200368 }
369 acc += lane * _PyHASH_XXPRIME_2;
370 acc = _PyHASH_XXROTATE(acc);
371 acc *= _PyHASH_XXPRIME_1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000372 }
jdemeyeraeb1be52018-10-28 02:06:38 +0200373
374 /* Add input length, mangled to keep the historical value of hash(()). */
375 acc += len ^ (_PyHASH_XXPRIME_5 ^ 3527539UL);
376
377 if (acc == (Py_uhash_t)-1) {
378 return 1546275796;
379 }
380 return acc;
Guido van Rossum9bfef441993-03-29 10:43:31 +0000381}
382
Martin v. Löwis18e16552006-02-15 17:27:45 +0000383static Py_ssize_t
Fred Drakeba096332000-07-09 07:04:36 +0000384tuplelength(PyTupleObject *a)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000385{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000386 return Py_SIZE(a);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000387}
388
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000389static int
Fred Drakeba096332000-07-09 07:04:36 +0000390tuplecontains(PyTupleObject *a, PyObject *el)
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000391{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000392 Py_ssize_t i;
393 int cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000394
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000395 for (i = 0, cmp = 0 ; cmp == 0 && i < Py_SIZE(a); ++i)
Serhiy Storchaka18b711c2019-08-04 14:12:48 +0300396 cmp = PyObject_RichCompareBool(PyTuple_GET_ITEM(a, i), el, Py_EQ);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000397 return cmp;
Jeremy Hylton37b1a262000-04-27 21:41:03 +0000398}
399
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000400static PyObject *
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200401tupleitem(PyTupleObject *a, Py_ssize_t i)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000402{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000403 if (i < 0 || i >= Py_SIZE(a)) {
404 PyErr_SetString(PyExc_IndexError, "tuple index out of range");
405 return NULL;
406 }
407 Py_INCREF(a->ob_item[i]);
408 return a->ob_item[i];
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000409}
410
Sergey Fedoseev234531b2019-02-25 21:59:12 +0500411PyObject *
412_PyTuple_FromArray(PyObject *const *src, Py_ssize_t n)
413{
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500414 if (n == 0) {
415 return PyTuple_New(0);
416 }
417
418 PyTupleObject *tuple = tuple_alloc(n);
Sergey Fedoseev234531b2019-02-25 21:59:12 +0500419 if (tuple == NULL) {
420 return NULL;
421 }
422 PyObject **dst = tuple->ob_item;
423 for (Py_ssize_t i = 0; i < n; i++) {
424 PyObject *item = src[i];
425 Py_INCREF(item);
426 dst[i] = item;
427 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500428 tuple_gc_track(tuple);
Sergey Fedoseev234531b2019-02-25 21:59:12 +0500429 return (PyObject *)tuple;
430}
431
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000432static PyObject *
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200433tupleslice(PyTupleObject *a, Py_ssize_t ilow,
434 Py_ssize_t ihigh)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000435{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 if (ilow < 0)
437 ilow = 0;
438 if (ihigh > Py_SIZE(a))
439 ihigh = Py_SIZE(a);
440 if (ihigh < ilow)
441 ihigh = ilow;
442 if (ilow == 0 && ihigh == Py_SIZE(a) && PyTuple_CheckExact(a)) {
443 Py_INCREF(a);
444 return (PyObject *)a;
445 }
Sergey Fedoseev234531b2019-02-25 21:59:12 +0500446 return _PyTuple_FromArray(a->ob_item + ilow, ihigh - ilow);
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000447}
448
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000449PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000450PyTuple_GetSlice(PyObject *op, Py_ssize_t i, Py_ssize_t j)
Guido van Rossum7c36ad71992-01-14 18:45:33 +0000451{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000452 if (op == NULL || !PyTuple_Check(op)) {
453 PyErr_BadInternalCall();
454 return NULL;
455 }
456 return tupleslice((PyTupleObject *)op, i, j);
Guido van Rossum7c36ad71992-01-14 18:45:33 +0000457}
458
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000459static PyObject *
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200460tupleconcat(PyTupleObject *a, PyObject *bb)
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000461{
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200462 Py_ssize_t size;
463 Py_ssize_t i;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000464 PyObject **src, **dest;
465 PyTupleObject *np;
Serhiy Storchaka98e80c22017-03-06 23:39:35 +0200466 if (Py_SIZE(a) == 0 && PyTuple_CheckExact(bb)) {
467 Py_INCREF(bb);
468 return bb;
469 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000470 if (!PyTuple_Check(bb)) {
471 PyErr_Format(PyExc_TypeError,
472 "can only concatenate tuple (not \"%.200s\") to tuple",
473 Py_TYPE(bb)->tp_name);
474 return NULL;
475 }
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000476#define b ((PyTupleObject *)bb)
Serhiy Storchaka98e80c22017-03-06 23:39:35 +0200477 if (Py_SIZE(b) == 0 && PyTuple_CheckExact(a)) {
478 Py_INCREF(a);
479 return (PyObject *)a;
480 }
Martin Panterb93d8632016-07-25 02:39:20 +0000481 if (Py_SIZE(a) > PY_SSIZE_T_MAX - Py_SIZE(b))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000482 return PyErr_NoMemory();
Martin Panterb93d8632016-07-25 02:39:20 +0000483 size = Py_SIZE(a) + Py_SIZE(b);
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500484 if (size == 0) {
485 return PyTuple_New(0);
486 }
487
488 np = tuple_alloc(size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000489 if (np == NULL) {
490 return NULL;
491 }
492 src = a->ob_item;
493 dest = np->ob_item;
494 for (i = 0; i < Py_SIZE(a); i++) {
495 PyObject *v = src[i];
496 Py_INCREF(v);
497 dest[i] = v;
498 }
499 src = b->ob_item;
500 dest = np->ob_item + Py_SIZE(a);
501 for (i = 0; i < Py_SIZE(b); i++) {
502 PyObject *v = src[i];
503 Py_INCREF(v);
504 dest[i] = v;
505 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500506 tuple_gc_track(np);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000507 return (PyObject *)np;
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000508#undef b
509}
510
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000511static PyObject *
Martin v. Löwis18e16552006-02-15 17:27:45 +0000512tuplerepeat(PyTupleObject *a, Py_ssize_t n)
Guido van Rossumb8393da1991-06-04 19:35:24 +0000513{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000514 Py_ssize_t i, j;
515 Py_ssize_t size;
516 PyTupleObject *np;
517 PyObject **p, **items;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000518 if (Py_SIZE(a) == 0 || n == 1) {
519 if (PyTuple_CheckExact(a)) {
520 /* Since tuples are immutable, we can return a shared
521 copy in this case */
522 Py_INCREF(a);
523 return (PyObject *)a;
524 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500525 }
526 if (Py_SIZE(a) == 0 || n <= 0) {
527 return PyTuple_New(0);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000528 }
Mark Dickinsonc04ddff2012-10-06 18:04:49 +0100529 if (n > PY_SSIZE_T_MAX / Py_SIZE(a))
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 return PyErr_NoMemory();
Mark Dickinsonc04ddff2012-10-06 18:04:49 +0100531 size = Py_SIZE(a) * n;
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500532 np = tuple_alloc(size);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000533 if (np == NULL)
534 return NULL;
535 p = np->ob_item;
536 items = a->ob_item;
537 for (i = 0; i < n; i++) {
538 for (j = 0; j < Py_SIZE(a); j++) {
539 *p = items[j];
540 Py_INCREF(*p);
541 p++;
542 }
543 }
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500544 tuple_gc_track(np);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000545 return (PyObject *) np;
Guido van Rossumb8393da1991-06-04 19:35:24 +0000546}
547
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200548/*[clinic input]
549tuple.index
Raymond Hettinger65baa342008-02-07 00:41:02 +0000550
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200551 value: object
Serhiy Storchakad4edfc92017-03-30 18:29:23 +0300552 start: slice_index(accept={int}) = 0
553 stop: slice_index(accept={int}, c_default="PY_SSIZE_T_MAX") = sys.maxsize
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200554 /
555
556Return first index of value.
557
558Raises ValueError if the value is not present.
559[clinic start generated code]*/
560
561static PyObject *
562tuple_index_impl(PyTupleObject *self, PyObject *value, Py_ssize_t start,
563 Py_ssize_t stop)
Serhiy Storchakad4edfc92017-03-30 18:29:23 +0300564/*[clinic end generated code: output=07b6f9f3cb5c33eb input=fb39e9874a21fe3f]*/
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200565{
566 Py_ssize_t i;
567
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000568 if (start < 0) {
569 start += Py_SIZE(self);
570 if (start < 0)
571 start = 0;
572 }
573 if (stop < 0) {
574 stop += Py_SIZE(self);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000575 }
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200576 else if (stop > Py_SIZE(self)) {
577 stop = Py_SIZE(self);
578 }
579 for (i = start; i < stop; i++) {
580 int cmp = PyObject_RichCompareBool(self->ob_item[i], value, Py_EQ);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000581 if (cmp > 0)
582 return PyLong_FromSsize_t(i);
583 else if (cmp < 0)
584 return NULL;
585 }
586 PyErr_SetString(PyExc_ValueError, "tuple.index(x): x not in tuple");
587 return NULL;
Raymond Hettinger65baa342008-02-07 00:41:02 +0000588}
589
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200590/*[clinic input]
591tuple.count
592
593 value: object
594 /
595
596Return number of occurrences of value.
597[clinic start generated code]*/
598
Raymond Hettinger65baa342008-02-07 00:41:02 +0000599static PyObject *
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200600tuple_count(PyTupleObject *self, PyObject *value)
601/*[clinic end generated code: output=aa927affc5a97605 input=531721aff65bd772]*/
Raymond Hettinger65baa342008-02-07 00:41:02 +0000602{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000603 Py_ssize_t count = 0;
604 Py_ssize_t i;
Raymond Hettinger65baa342008-02-07 00:41:02 +0000605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000606 for (i = 0; i < Py_SIZE(self); i++) {
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200607 int cmp = PyObject_RichCompareBool(self->ob_item[i], value, Py_EQ);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000608 if (cmp > 0)
609 count++;
610 else if (cmp < 0)
611 return NULL;
612 }
613 return PyLong_FromSsize_t(count);
Raymond Hettinger65baa342008-02-07 00:41:02 +0000614}
615
Jeremy Hylton8caad492000-06-23 14:18:11 +0000616static int
617tupletraverse(PyTupleObject *o, visitproc visit, void *arg)
618{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000619 Py_ssize_t i;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000620
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000621 for (i = Py_SIZE(o); --i >= 0; )
622 Py_VISIT(o->ob_item[i]);
623 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +0000624}
625
Guido van Rossumf77bc622001-01-18 00:00:53 +0000626static PyObject *
627tuplerichcompare(PyObject *v, PyObject *w, int op)
628{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000629 PyTupleObject *vt, *wt;
630 Py_ssize_t i;
631 Py_ssize_t vlen, wlen;
Guido van Rossumf77bc622001-01-18 00:00:53 +0000632
Brian Curtindfc80e32011-08-10 20:28:54 -0500633 if (!PyTuple_Check(v) || !PyTuple_Check(w))
634 Py_RETURN_NOTIMPLEMENTED;
Guido van Rossumf77bc622001-01-18 00:00:53 +0000635
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000636 vt = (PyTupleObject *)v;
637 wt = (PyTupleObject *)w;
Guido van Rossumf77bc622001-01-18 00:00:53 +0000638
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000639 vlen = Py_SIZE(vt);
640 wlen = Py_SIZE(wt);
Guido van Rossumf77bc622001-01-18 00:00:53 +0000641
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000642 /* Note: the corresponding code for lists has an "early out" test
643 * here when op is EQ or NE and the lengths differ. That pays there,
644 * but Tim was unable to find any real code where EQ/NE tuple
645 * compares don't have the same length, so testing for it here would
646 * have cost without benefit.
647 */
Tim Petersd7ed3bf2001-05-15 20:12:59 +0000648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 /* Search for the first index where items are different.
650 * Note that because tuples are immutable, it's safe to reuse
651 * vlen and wlen across the comparison calls.
652 */
653 for (i = 0; i < vlen && i < wlen; i++) {
654 int k = PyObject_RichCompareBool(vt->ob_item[i],
655 wt->ob_item[i], Py_EQ);
656 if (k < 0)
657 return NULL;
658 if (!k)
659 break;
660 }
Guido van Rossumf77bc622001-01-18 00:00:53 +0000661
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000662 if (i >= vlen || i >= wlen) {
663 /* No more items to compare -- compare sizes */
stratakise8b19652017-11-02 11:32:54 +0100664 Py_RETURN_RICHCOMPARE(vlen, wlen, op);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000665 }
Guido van Rossumf77bc622001-01-18 00:00:53 +0000666
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000667 /* We have an item that differs -- shortcuts for EQ/NE */
668 if (op == Py_EQ) {
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200669 Py_RETURN_FALSE;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000670 }
671 if (op == Py_NE) {
Serhiy Storchaka228b12e2017-01-23 09:47:21 +0200672 Py_RETURN_TRUE;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000673 }
Guido van Rossumf77bc622001-01-18 00:00:53 +0000674
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000675 /* Compare the final item again using the proper operator */
676 return PyObject_RichCompare(vt->ob_item[i], wt->ob_item[i], op);
Guido van Rossumf77bc622001-01-18 00:00:53 +0000677}
678
Jeremy Hylton938ace62002-07-17 16:30:39 +0000679static PyObject *
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200680tuple_subtype_new(PyTypeObject *type, PyObject *iterable);
681
682/*[clinic input]
683@classmethod
684tuple.__new__ as tuple_new
685 iterable: object(c_default="NULL") = ()
686 /
687
688Built-in immutable sequence.
689
690If no argument is given, the constructor returns an empty tuple.
691If iterable is specified the tuple is initialized from iterable's items.
692
693If the argument is a tuple, the return value is the same object.
694[clinic start generated code]*/
Guido van Rossumae960af2001-08-30 03:11:59 +0000695
Tim Peters6d6c1a32001-08-02 04:15:00 +0000696static PyObject *
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200697tuple_new_impl(PyTypeObject *type, PyObject *iterable)
698/*[clinic end generated code: output=4546d9f0d469bce7 input=86963bcde633b5a2]*/
Tim Peters6d6c1a32001-08-02 04:15:00 +0000699{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000700 if (type != &PyTuple_Type)
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200701 return tuple_subtype_new(type, iterable);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000702
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200703 if (iterable == NULL)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000704 return PyTuple_New(0);
705 else
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200706 return PySequence_Tuple(iterable);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000707}
708
Guido van Rossumae960af2001-08-30 03:11:59 +0000709static PyObject *
Dong-hee Na9ee88cd2020-03-13 22:57:00 +0900710tuple_vectorcall(PyObject *type, PyObject * const*args,
711 size_t nargsf, PyObject *kwnames)
712{
Dong-hee Na87ec86c2020-03-16 23:06:20 +0900713 if (!_PyArg_NoKwnames("tuple", kwnames)) {
Dong-hee Na9ee88cd2020-03-13 22:57:00 +0900714 return NULL;
715 }
Dong-hee Nac98f87f2020-03-16 23:04:14 +0900716
Dong-hee Na9ee88cd2020-03-13 22:57:00 +0900717 Py_ssize_t nargs = PyVectorcall_NARGS(nargsf);
Dong-hee Nac98f87f2020-03-16 23:04:14 +0900718 if (!_PyArg_CheckPositional("tuple", nargs, 0, 1)) {
Dong-hee Na9ee88cd2020-03-13 22:57:00 +0900719 return NULL;
720 }
721
722 if (nargs) {
723 return tuple_new_impl((PyTypeObject *)type, args[0]);
724 }
725 return PyTuple_New(0);
726}
727
728static PyObject *
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200729tuple_subtype_new(PyTypeObject *type, PyObject *iterable)
Guido van Rossumae960af2001-08-30 03:11:59 +0000730{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000731 PyObject *tmp, *newobj, *item;
732 Py_ssize_t i, n;
Guido van Rossumae960af2001-08-30 03:11:59 +0000733
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000734 assert(PyType_IsSubtype(type, &PyTuple_Type));
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200735 tmp = tuple_new_impl(&PyTuple_Type, iterable);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000736 if (tmp == NULL)
737 return NULL;
738 assert(PyTuple_Check(tmp));
739 newobj = type->tp_alloc(type, n = PyTuple_GET_SIZE(tmp));
Hai Shic81609e2020-03-16 03:37:49 +0800740 if (newobj == NULL) {
741 Py_DECREF(tmp);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000742 return NULL;
Hai Shic81609e2020-03-16 03:37:49 +0800743 }
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000744 for (i = 0; i < n; i++) {
745 item = PyTuple_GET_ITEM(tmp, i);
746 Py_INCREF(item);
747 PyTuple_SET_ITEM(newobj, i, item);
748 }
749 Py_DECREF(tmp);
750 return newobj;
Guido van Rossumae960af2001-08-30 03:11:59 +0000751}
752
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000753static PySequenceMethods tuple_as_sequence = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000754 (lenfunc)tuplelength, /* sq_length */
755 (binaryfunc)tupleconcat, /* sq_concat */
756 (ssizeargfunc)tuplerepeat, /* sq_repeat */
757 (ssizeargfunc)tupleitem, /* sq_item */
758 0, /* sq_slice */
759 0, /* sq_ass_item */
760 0, /* sq_ass_slice */
761 (objobjproc)tuplecontains, /* sq_contains */
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000762};
763
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000764static PyObject*
765tuplesubscript(PyTupleObject* self, PyObject* item)
766{
Victor Stinnera15e2602020-04-08 02:01:56 +0200767 if (_PyIndex_Check(item)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000768 Py_ssize_t i = PyNumber_AsSsize_t(item, PyExc_IndexError);
769 if (i == -1 && PyErr_Occurred())
770 return NULL;
771 if (i < 0)
772 i += PyTuple_GET_SIZE(self);
773 return tupleitem(self, i);
774 }
775 else if (PySlice_Check(item)) {
Zackery Spytz14514d92019-05-17 01:13:03 -0600776 Py_ssize_t start, stop, step, slicelength, i;
777 size_t cur;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000778 PyObject* it;
779 PyObject **src, **dest;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000780
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300781 if (PySlice_Unpack(item, &start, &stop, &step) < 0) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000782 return NULL;
783 }
Serhiy Storchakab879fe82017-04-08 09:53:51 +0300784 slicelength = PySlice_AdjustIndices(PyTuple_GET_SIZE(self), &start,
785 &stop, step);
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000786
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000787 if (slicelength <= 0) {
788 return PyTuple_New(0);
789 }
790 else if (start == 0 && step == 1 &&
791 slicelength == PyTuple_GET_SIZE(self) &&
792 PyTuple_CheckExact(self)) {
793 Py_INCREF(self);
794 return (PyObject *)self;
795 }
796 else {
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500797 PyTupleObject* result = tuple_alloc(slicelength);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000798 if (!result) return NULL;
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000799
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000800 src = self->ob_item;
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500801 dest = result->ob_item;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000802 for (cur = start, i = 0; i < slicelength;
803 cur += step, i++) {
804 it = src[cur];
805 Py_INCREF(it);
806 dest[i] = it;
807 }
808
Sergey Fedoseev4fa10dd2019-08-14 19:10:33 +0500809 tuple_gc_track(result);
810 return (PyObject *)result;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000811 }
812 }
813 else {
814 PyErr_Format(PyExc_TypeError,
Terry Jan Reedyffff1442014-08-02 01:30:37 -0400815 "tuple indices must be integers or slices, not %.200s",
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000816 Py_TYPE(item)->tp_name);
817 return NULL;
818 }
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000819}
820
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200821/*[clinic input]
822tuple.__getnewargs__
823[clinic start generated code]*/
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000824
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200825static PyObject *
826tuple___getnewargs___impl(PyTupleObject *self)
827/*[clinic end generated code: output=25e06e3ee56027e2 input=1aeb4b286a21639a]*/
828{
829 return Py_BuildValue("(N)", tupleslice(self, 0, Py_SIZE(self)));
Guido van Rossum5d9113d2003-01-29 17:58:45 +0000830}
831
832static PyMethodDef tuple_methods[] = {
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200833 TUPLE___GETNEWARGS___METHODDEF
834 TUPLE_INDEX_METHODDEF
835 TUPLE_COUNT_METHODDEF
Guido van Rossum48b069a2020-04-07 09:50:06 -0700836 {"__class_getitem__", (PyCFunction)Py_GenericAlias, METH_O|METH_CLASS, PyDoc_STR("See PEP 585")},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000837 {NULL, NULL} /* sentinel */
Guido van Rossum5d9113d2003-01-29 17:58:45 +0000838};
839
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000840static PyMappingMethods tuple_as_mapping = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000841 (lenfunc)tuplelength,
842 (binaryfunc)tuplesubscript,
843 0
Michael W. Hudson5efaf7e2002-06-11 10:55:12 +0000844};
845
Raymond Hettinger48923c52002-08-09 01:30:17 +0000846static PyObject *tuple_iter(PyObject *seq);
847
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000848PyTypeObject PyTuple_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000849 PyVarObject_HEAD_INIT(&PyType_Type, 0)
850 "tuple",
851 sizeof(PyTupleObject) - sizeof(PyObject *),
852 sizeof(PyObject *),
853 (destructor)tupledealloc, /* tp_dealloc */
Jeroen Demeyer530f5062019-05-31 04:13:39 +0200854 0, /* tp_vectorcall_offset */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000855 0, /* tp_getattr */
856 0, /* tp_setattr */
Jeroen Demeyer530f5062019-05-31 04:13:39 +0200857 0, /* tp_as_async */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000858 (reprfunc)tuplerepr, /* tp_repr */
859 0, /* tp_as_number */
860 &tuple_as_sequence, /* tp_as_sequence */
861 &tuple_as_mapping, /* tp_as_mapping */
862 (hashfunc)tuplehash, /* tp_hash */
863 0, /* tp_call */
864 0, /* tp_str */
865 PyObject_GenericGetAttr, /* tp_getattro */
866 0, /* tp_setattro */
867 0, /* tp_as_buffer */
868 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
869 Py_TPFLAGS_BASETYPE | Py_TPFLAGS_TUPLE_SUBCLASS, /* tp_flags */
Serhiy Storchaka0b561592017-03-19 08:47:58 +0200870 tuple_new__doc__, /* tp_doc */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000871 (traverseproc)tupletraverse, /* tp_traverse */
872 0, /* tp_clear */
873 tuplerichcompare, /* tp_richcompare */
874 0, /* tp_weaklistoffset */
875 tuple_iter, /* tp_iter */
876 0, /* tp_iternext */
877 tuple_methods, /* tp_methods */
878 0, /* tp_members */
879 0, /* tp_getset */
880 0, /* tp_base */
881 0, /* tp_dict */
882 0, /* tp_descr_get */
883 0, /* tp_descr_set */
884 0, /* tp_dictoffset */
885 0, /* tp_init */
886 0, /* tp_alloc */
887 tuple_new, /* tp_new */
888 PyObject_GC_Del, /* tp_free */
Dong-hee Na9ee88cd2020-03-13 22:57:00 +0900889 .tp_vectorcall = tuple_vectorcall,
Guido van Rossum85a5fbb1990-10-14 12:07:46 +0000890};
Guido van Rossum12d12c51993-10-26 17:58:25 +0000891
892/* The following function breaks the notion that tuples are immutable:
893 it changes the size of a tuple. We get away with this only if there
894 is only one module referencing the object. You can also think of it
Neil Schemenauer08b53e62000-10-05 19:36:49 +0000895 as creating a new tuple object and destroying the old one, only more
896 efficiently. In any case, don't use this if the tuple may already be
Tim Peters4324aa32001-05-28 22:30:08 +0000897 known to some other part of the code. */
Guido van Rossum12d12c51993-10-26 17:58:25 +0000898
899int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000900_PyTuple_Resize(PyObject **pv, Py_ssize_t newsize)
Guido van Rossum12d12c51993-10-26 17:58:25 +0000901{
Antoine Pitrou9ed5f272013-08-13 20:18:52 +0200902 PyTupleObject *v;
903 PyTupleObject *sv;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000904 Py_ssize_t i;
905 Py_ssize_t oldsize;
Sjoerd Mullender615194a1993-11-01 13:46:50 +0000906
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000907 v = (PyTupleObject *) *pv;
Andy Lester55728702020-03-06 16:53:17 -0600908 if (v == NULL || !Py_IS_TYPE(v, &PyTuple_Type) ||
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000909 (Py_SIZE(v) != 0 && Py_REFCNT(v) != 1)) {
910 *pv = 0;
911 Py_XDECREF(v);
912 PyErr_BadInternalCall();
913 return -1;
914 }
915 oldsize = Py_SIZE(v);
916 if (oldsize == newsize)
917 return 0;
Neil Schemenauer08b53e62000-10-05 19:36:49 +0000918
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000919 if (oldsize == 0) {
920 /* Empty tuples are often shared, so we should never
921 resize them in-place even if we do own the only
922 (current) reference */
923 Py_DECREF(v);
924 *pv = PyTuple_New(newsize);
925 return *pv == NULL ? -1 : 0;
926 }
Thomas Wouters6a922372001-05-28 13:11:02 +0000927
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000928 /* XXX UNREF/NEWREF interface should be more symmetrical */
Victor Stinner49932fe2020-02-03 17:55:05 +0100929#ifdef Py_REF_DEBUG
930 _Py_RefTotal--;
931#endif
932 if (_PyObject_GC_IS_TRACKED(v)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000933 _PyObject_GC_UNTRACK(v);
Victor Stinner49932fe2020-02-03 17:55:05 +0100934 }
935#ifdef Py_TRACE_REFS
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000936 _Py_ForgetReference((PyObject *) v);
Victor Stinner49932fe2020-02-03 17:55:05 +0100937#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000938 /* DECREF items deleted by shrinkage */
939 for (i = newsize; i < oldsize; i++) {
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200940 Py_CLEAR(v->ob_item[i]);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000941 }
942 sv = PyObject_GC_Resize(PyTupleObject, v, newsize);
943 if (sv == NULL) {
944 *pv = NULL;
945 PyObject_GC_Del(v);
946 return -1;
947 }
948 _Py_NewReference((PyObject *) sv);
949 /* Zero out items added by growing */
950 if (newsize > oldsize)
951 memset(&sv->ob_item[oldsize], 0,
952 sizeof(*sv->ob_item) * (newsize - oldsize));
953 *pv = (PyObject *) sv;
954 _PyObject_GC_TRACK(sv);
955 return 0;
Guido van Rossum12d12c51993-10-26 17:58:25 +0000956}
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000957
Christian Heimesa156e092008-02-16 07:38:31 +0000958int
959PyTuple_ClearFreeList(void)
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000960{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000961 int freelist_size = 0;
Christian Heimes2202f872008-02-06 14:31:34 +0000962#if PyTuple_MAXSAVESIZE > 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000963 int i;
964 for (i = 1; i < PyTuple_MAXSAVESIZE; i++) {
965 PyTupleObject *p, *q;
966 p = free_list[i];
967 freelist_size += numfree[i];
968 free_list[i] = NULL;
969 numfree[i] = 0;
970 while (p) {
971 q = p;
972 p = (PyTupleObject *)(p->ob_item[0]);
973 PyObject_GC_Del(q);
974 }
975 }
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000976#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000977 return freelist_size;
Christian Heimesa156e092008-02-16 07:38:31 +0000978}
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979
Christian Heimesa156e092008-02-16 07:38:31 +0000980void
Victor Stinnerbed48172019-08-27 00:12:32 +0200981_PyTuple_Fini(void)
Christian Heimesa156e092008-02-16 07:38:31 +0000982{
983#if PyTuple_MAXSAVESIZE > 0
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000984 /* empty tuples are used all over the place and applications may
985 * rely on the fact that an empty tuple is a singleton. */
Serhiy Storchaka505ff752014-02-09 13:33:53 +0200986 Py_CLEAR(free_list[0]);
Christian Heimesa156e092008-02-16 07:38:31 +0000987
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000988 (void)PyTuple_ClearFreeList();
Christian Heimesa156e092008-02-16 07:38:31 +0000989#endif
Guido van Rossumfbbd57e1997-08-05 02:16:08 +0000990}
Raymond Hettinger48923c52002-08-09 01:30:17 +0000991
992/*********************** Tuple Iterator **************************/
993
994typedef struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000995 PyObject_HEAD
Victor Stinnera2d56982013-06-05 00:11:34 +0200996 Py_ssize_t it_index;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000997 PyTupleObject *it_seq; /* Set to NULL when iterator is exhausted */
Raymond Hettinger48923c52002-08-09 01:30:17 +0000998} tupleiterobject;
999
Raymond Hettinger48923c52002-08-09 01:30:17 +00001000static void
1001tupleiter_dealloc(tupleiterobject *it)
1002{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 _PyObject_GC_UNTRACK(it);
1004 Py_XDECREF(it->it_seq);
1005 PyObject_GC_Del(it);
Raymond Hettinger48923c52002-08-09 01:30:17 +00001006}
1007
1008static int
1009tupleiter_traverse(tupleiterobject *it, visitproc visit, void *arg)
1010{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001011 Py_VISIT(it->it_seq);
1012 return 0;
Raymond Hettinger48923c52002-08-09 01:30:17 +00001013}
1014
Raymond Hettinger48923c52002-08-09 01:30:17 +00001015static PyObject *
1016tupleiter_next(tupleiterobject *it)
1017{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001018 PyTupleObject *seq;
1019 PyObject *item;
Raymond Hettinger48923c52002-08-09 01:30:17 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 assert(it != NULL);
1022 seq = it->it_seq;
1023 if (seq == NULL)
1024 return NULL;
1025 assert(PyTuple_Check(seq));
Raymond Hettinger48923c52002-08-09 01:30:17 +00001026
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001027 if (it->it_index < PyTuple_GET_SIZE(seq)) {
1028 item = PyTuple_GET_ITEM(seq, it->it_index);
1029 ++it->it_index;
1030 Py_INCREF(item);
1031 return item;
1032 }
Raymond Hettinger48923c52002-08-09 01:30:17 +00001033
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001034 it->it_seq = NULL;
Serhiy Storchakafbb1c5e2016-03-30 20:40:02 +03001035 Py_DECREF(seq);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001036 return NULL;
Raymond Hettinger48923c52002-08-09 01:30:17 +00001037}
1038
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00001039static PyObject *
Siddhesh Poyarekar55edd0c2018-04-30 00:29:33 +05301040tupleiter_len(tupleiterobject *it, PyObject *Py_UNUSED(ignored))
Raymond Hettinger435bf582004-03-18 22:43:10 +00001041{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001042 Py_ssize_t len = 0;
1043 if (it->it_seq)
1044 len = PyTuple_GET_SIZE(it->it_seq) - it->it_index;
1045 return PyLong_FromSsize_t(len);
Raymond Hettinger435bf582004-03-18 22:43:10 +00001046}
1047
Armin Rigof5b3e362006-02-11 21:32:43 +00001048PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00001049
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001050static PyObject *
Siddhesh Poyarekar55edd0c2018-04-30 00:29:33 +05301051tupleiter_reduce(tupleiterobject *it, PyObject *Py_UNUSED(ignored))
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001052{
Serhiy Storchakabb86bf42018-12-11 08:28:18 +02001053 _Py_IDENTIFIER(iter);
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001054 if (it->it_seq)
Serhiy Storchakabb86bf42018-12-11 08:28:18 +02001055 return Py_BuildValue("N(O)n", _PyEval_GetBuiltinId(&PyId_iter),
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001056 it->it_seq, it->it_index);
1057 else
Serhiy Storchakabb86bf42018-12-11 08:28:18 +02001058 return Py_BuildValue("N(())", _PyEval_GetBuiltinId(&PyId_iter));
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001059}
1060
1061static PyObject *
1062tupleiter_setstate(tupleiterobject *it, PyObject *state)
1063{
Victor Stinner7660b882013-06-24 23:59:24 +02001064 Py_ssize_t index = PyLong_AsSsize_t(state);
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001065 if (index == -1 && PyErr_Occurred())
1066 return NULL;
1067 if (it->it_seq != NULL) {
1068 if (index < 0)
1069 index = 0;
Kristján Valur Jónsson25dded02014-03-05 13:47:57 +00001070 else if (index > PyTuple_GET_SIZE(it->it_seq))
1071 index = PyTuple_GET_SIZE(it->it_seq); /* exhausted iterator */
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001072 it->it_index = index;
1073 }
1074 Py_RETURN_NONE;
1075}
1076
1077PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
1078PyDoc_STRVAR(setstate_doc, "Set state information for unpickling.");
1079
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00001080static PyMethodDef tupleiter_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001081 {"__length_hint__", (PyCFunction)tupleiter_len, METH_NOARGS, length_hint_doc},
Kristján Valur Jónsson31668b82012-04-03 10:49:41 +00001082 {"__reduce__", (PyCFunction)tupleiter_reduce, METH_NOARGS, reduce_doc},
1083 {"__setstate__", (PyCFunction)tupleiter_setstate, METH_O, setstate_doc},
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001084 {NULL, NULL} /* sentinel */
Raymond Hettinger435bf582004-03-18 22:43:10 +00001085};
1086
Raymond Hettinger48923c52002-08-09 01:30:17 +00001087PyTypeObject PyTupleIter_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001088 PyVarObject_HEAD_INIT(&PyType_Type, 0)
1089 "tuple_iterator", /* tp_name */
1090 sizeof(tupleiterobject), /* tp_basicsize */
1091 0, /* tp_itemsize */
1092 /* methods */
1093 (destructor)tupleiter_dealloc, /* tp_dealloc */
Jeroen Demeyer530f5062019-05-31 04:13:39 +02001094 0, /* tp_vectorcall_offset */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001095 0, /* tp_getattr */
1096 0, /* tp_setattr */
Jeroen Demeyer530f5062019-05-31 04:13:39 +02001097 0, /* tp_as_async */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001098 0, /* tp_repr */
1099 0, /* tp_as_number */
1100 0, /* tp_as_sequence */
1101 0, /* tp_as_mapping */
1102 0, /* tp_hash */
1103 0, /* tp_call */
1104 0, /* tp_str */
1105 PyObject_GenericGetAttr, /* tp_getattro */
1106 0, /* tp_setattro */
1107 0, /* tp_as_buffer */
1108 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
1109 0, /* tp_doc */
1110 (traverseproc)tupleiter_traverse, /* tp_traverse */
1111 0, /* tp_clear */
1112 0, /* tp_richcompare */
1113 0, /* tp_weaklistoffset */
1114 PyObject_SelfIter, /* tp_iter */
1115 (iternextfunc)tupleiter_next, /* tp_iternext */
1116 tupleiter_methods, /* tp_methods */
1117 0,
Raymond Hettinger48923c52002-08-09 01:30:17 +00001118};
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001119
1120static PyObject *
1121tuple_iter(PyObject *seq)
1122{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 tupleiterobject *it;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001124
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001125 if (!PyTuple_Check(seq)) {
1126 PyErr_BadInternalCall();
1127 return NULL;
1128 }
1129 it = PyObject_GC_New(tupleiterobject, &PyTupleIter_Type);
1130 if (it == NULL)
1131 return NULL;
1132 it->it_index = 0;
1133 Py_INCREF(seq);
1134 it->it_seq = (PyTupleObject *)seq;
1135 _PyObject_GC_TRACK(it);
1136 return (PyObject *)it;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00001137}