blob: 9d19b1db13156d19bf786112ad6fe7957b5a3cad [file] [log] [blame]
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001
Guido van Rossum2bc13791999-03-24 19:06:42 +00002/* Dictionary object implementation using a hash table */
Guido van Rossum9bfef441993-03-29 10:43:31 +00003
Raymond Hettinger930427b2003-05-03 06:51:59 +00004/* The distribution includes a separate file, Objects/dictnotes.txt,
Tim Peters60b29962006-01-01 01:19:23 +00005 describing explorations into dictionary design and optimization.
Raymond Hettinger930427b2003-05-03 06:51:59 +00006 It covers typical dictionary use patterns, the parameters for
7 tuning dictionaries, and several ideas for possible optimizations.
8*/
9
Guido van Rossumc0b618a1997-05-02 03:12:38 +000010#include "Python.h"
Guido van Rossum4b1302b1993-03-27 18:11:32 +000011
Guido van Rossum16e93a81997-01-28 00:00:11 +000012
Georg Brandlb9f4ad32006-10-29 18:31:42 +000013/* Set a key error with the specified argument, wrapping it in a
14 * tuple automatically so that tuple keys are not unpacked as the
15 * exception arguments. */
16static void
17set_key_error(PyObject *arg)
18{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +000019 PyObject *tup;
20 tup = PyTuple_Pack(1, arg);
21 if (!tup)
22 return; /* caller will expect error to be set anyway */
23 PyErr_SetObject(PyExc_KeyError, tup);
24 Py_DECREF(tup);
Georg Brandlb9f4ad32006-10-29 18:31:42 +000025}
26
Tim Peterseb28ef22001-06-02 05:27:19 +000027/* Define this out if you don't want conversion statistics on exit. */
Fred Drake1bff34a2000-08-31 19:31:38 +000028#undef SHOW_CONVERSION_COUNTS
29
Tim Peterseb28ef22001-06-02 05:27:19 +000030/* See large comment block below. This must be >= 1. */
31#define PERTURB_SHIFT 5
32
Guido van Rossum16e93a81997-01-28 00:00:11 +000033/*
Tim Peterseb28ef22001-06-02 05:27:19 +000034Major subtleties ahead: Most hash schemes depend on having a "good" hash
35function, in the sense of simulating randomness. Python doesn't: its most
36important hash functions (for strings and ints) are very regular in common
37cases:
Tim Peters15d49292001-05-27 07:39:22 +000038
Tim Peterseb28ef22001-06-02 05:27:19 +000039>>> map(hash, (0, 1, 2, 3))
40[0, 1, 2, 3]
41>>> map(hash, ("namea", "nameb", "namec", "named"))
42[-1658398457, -1658398460, -1658398459, -1658398462]
43>>>
Tim Peters15d49292001-05-27 07:39:22 +000044
Tim Peterseb28ef22001-06-02 05:27:19 +000045This isn't necessarily bad! To the contrary, in a table of size 2**i, taking
46the low-order i bits as the initial table index is extremely fast, and there
47are no collisions at all for dicts indexed by a contiguous range of ints.
48The same is approximately true when keys are "consecutive" strings. So this
49gives better-than-random behavior in common cases, and that's very desirable.
Tim Peters15d49292001-05-27 07:39:22 +000050
Tim Peterseb28ef22001-06-02 05:27:19 +000051OTOH, when collisions occur, the tendency to fill contiguous slices of the
52hash table makes a good collision resolution strategy crucial. Taking only
53the last i bits of the hash code is also vulnerable: for example, consider
54[i << 16 for i in range(20000)] as a set of keys. Since ints are their own
55hash codes, and this fits in a dict of size 2**15, the last 15 bits of every
56hash code are all 0: they *all* map to the same table index.
Tim Peters15d49292001-05-27 07:39:22 +000057
Tim Peterseb28ef22001-06-02 05:27:19 +000058But catering to unusual cases should not slow the usual ones, so we just take
59the last i bits anyway. It's up to collision resolution to do the rest. If
60we *usually* find the key we're looking for on the first try (and, it turns
61out, we usually do -- the table load factor is kept under 2/3, so the odds
62are solidly in our favor), then it makes best sense to keep the initial index
63computation dirt cheap.
Tim Peters15d49292001-05-27 07:39:22 +000064
Tim Peterseb28ef22001-06-02 05:27:19 +000065The first half of collision resolution is to visit table indices via this
66recurrence:
Tim Peters15d49292001-05-27 07:39:22 +000067
Tim Peterseb28ef22001-06-02 05:27:19 +000068 j = ((5*j) + 1) mod 2**i
Tim Peters15d49292001-05-27 07:39:22 +000069
Tim Peterseb28ef22001-06-02 05:27:19 +000070For any initial j in range(2**i), repeating that 2**i times generates each
71int in range(2**i) exactly once (see any text on random-number generation for
72proof). By itself, this doesn't help much: like linear probing (setting
73j += 1, or j -= 1, on each loop trip), it scans the table entries in a fixed
74order. This would be bad, except that's not the only thing we do, and it's
75actually *good* in the common cases where hash keys are consecutive. In an
76example that's really too small to make this entirely clear, for a table of
77size 2**3 the order of indices is:
Tim Peters15d49292001-05-27 07:39:22 +000078
Tim Peterseb28ef22001-06-02 05:27:19 +000079 0 -> 1 -> 6 -> 7 -> 4 -> 5 -> 2 -> 3 -> 0 [and here it's repeating]
80
81If two things come in at index 5, the first place we look after is index 2,
82not 6, so if another comes in at index 6 the collision at 5 didn't hurt it.
83Linear probing is deadly in this case because there the fixed probe order
84is the *same* as the order consecutive keys are likely to arrive. But it's
85extremely unlikely hash codes will follow a 5*j+1 recurrence by accident,
86and certain that consecutive hash codes do not.
87
88The other half of the strategy is to get the other bits of the hash code
89into play. This is done by initializing a (unsigned) vrbl "perturb" to the
90full hash code, and changing the recurrence to:
91
92 j = (5*j) + 1 + perturb;
93 perturb >>= PERTURB_SHIFT;
94 use j % 2**i as the next table index;
95
96Now the probe sequence depends (eventually) on every bit in the hash code,
97and the pseudo-scrambling property of recurring on 5*j+1 is more valuable,
98because it quickly magnifies small differences in the bits that didn't affect
99the initial index. Note that because perturb is unsigned, if the recurrence
100is executed often enough perturb eventually becomes and remains 0. At that
101point (very rarely reached) the recurrence is on (just) 5*j+1 again, and
102that's certain to find an empty slot eventually (since it generates every int
103in range(2**i), and we make sure there's always at least one empty slot).
104
105Selecting a good value for PERTURB_SHIFT is a balancing act. You want it
106small so that the high bits of the hash code continue to affect the probe
107sequence across iterations; but you want it large so that in really bad cases
108the high-order hash bits have an effect on early iterations. 5 was "the
109best" in minimizing total collisions across experiments Tim Peters ran (on
110both normal and pathological cases), but 4 and 6 weren't significantly worse.
111
112Historical: Reimer Behrends contributed the idea of using a polynomial-based
113approach, using repeated multiplication by x in GF(2**n) where an irreducible
114polynomial for each table size was chosen such that x was a primitive root.
115Christian Tismer later extended that to use division by x instead, as an
116efficient way to get the high bits of the hash code into play. This scheme
117also gave excellent collision statistics, but was more expensive: two
118if-tests were required inside the loop; computing "the next" index took about
119the same number of operations but without as much potential parallelism
120(e.g., computing 5*j can go on at the same time as computing 1+perturb in the
121above, and then shifting perturb can be done while the table index is being
Brett Cannon77ae87c2007-10-10 00:07:50 +0000122masked); and the PyDictObject struct required a member to hold the table's
Tim Peterseb28ef22001-06-02 05:27:19 +0000123polynomial. In Tim's experiments the current scheme ran faster, produced
124equally good collision statistics, needed less code & used less memory.
Tim Peters9b10f7e2006-05-30 04:16:25 +0000125
126Theoretical Python 2.5 headache: hash codes are only C "long", but
127sizeof(Py_ssize_t) > sizeof(long) may be possible. In that case, and if a
128dict is genuinely huge, then only the slots directly reachable via indexing
129by a C long can be the first slot in a probe sequence. The probe sequence
130will still eventually reach every slot in the table, but the collision rate
131on initial probes may be much higher than this scheme was designed for.
132Getting a hash code as fat as Py_ssize_t is the only real cure. But in
133practice, this probably won't make a lick of difference for many years (at
134which point everyone will have terabytes of RAM on 64-bit boxes).
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000135*/
Tim Petersdea48ec2001-05-22 20:40:22 +0000136
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000137/* Object used as dummy key to fill deleted entries */
Brett Cannon77ae87c2007-10-10 00:07:50 +0000138static PyObject *dummy = NULL; /* Initialized by first call to newPyDictObject() */
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000139
Armin Rigoe1709372006-04-12 17:06:05 +0000140#ifdef Py_REF_DEBUG
141PyObject *
142_PyDict_Dummy(void)
143{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000144 return dummy;
Armin Rigoe1709372006-04-12 17:06:05 +0000145}
146#endif
147
Fred Drake1bff34a2000-08-31 19:31:38 +0000148/* forward declarations */
Brett Cannon77ae87c2007-10-10 00:07:50 +0000149static PyDictEntry *
150lookdict_string(PyDictObject *mp, PyObject *key, long hash);
Fred Drake1bff34a2000-08-31 19:31:38 +0000151
152#ifdef SHOW_CONVERSION_COUNTS
153static long created = 0L;
154static long converted = 0L;
155
156static void
157show_counts(void)
158{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000159 fprintf(stderr, "created %ld string dicts\n", created);
160 fprintf(stderr, "converted %ld to normal dicts\n", converted);
161 fprintf(stderr, "%.2f%% conversion rate\n", (100.0*converted)/created);
Fred Drake1bff34a2000-08-31 19:31:38 +0000162}
163#endif
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000164
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000165/* Debug statistic to compare allocations with reuse through the free list */
166#undef SHOW_ALLOC_COUNT
167#ifdef SHOW_ALLOC_COUNT
168static size_t count_alloc = 0;
169static size_t count_reuse = 0;
170
171static void
172show_alloc(void)
173{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000174 fprintf(stderr, "Dict allocations: %" PY_FORMAT_SIZE_T "d\n",
175 count_alloc);
176 fprintf(stderr, "Dict reuse through freelist: %" PY_FORMAT_SIZE_T
177 "d\n", count_reuse);
178 fprintf(stderr, "%.2f%% reuse rate\n\n",
179 (100.0*count_reuse/(count_alloc+count_reuse)));
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000180}
181#endif
182
Tim Peters6d6c1a32001-08-02 04:15:00 +0000183/* Initialization macros.
184 There are two ways to create a dict: PyDict_New() is the main C API
185 function, and the tp_new slot maps to dict_new(). In the latter case we
186 can save a little time over what PyDict_New does because it's guaranteed
187 that the PyDictObject struct is already zeroed out.
188 Everyone except dict_new() should use EMPTY_TO_MINSIZE (unless they have
189 an excellent reason not to).
190*/
191
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000192#define INIT_NONZERO_DICT_SLOTS(mp) do { \
193 (mp)->ma_table = (mp)->ma_smalltable; \
194 (mp)->ma_mask = PyDict_MINSIZE - 1; \
Tim Peters6d6c1a32001-08-02 04:15:00 +0000195 } while(0)
196
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000197#define EMPTY_TO_MINSIZE(mp) do { \
198 memset((mp)->ma_smalltable, 0, sizeof((mp)->ma_smalltable)); \
199 (mp)->ma_used = (mp)->ma_fill = 0; \
200 INIT_NONZERO_DICT_SLOTS(mp); \
Tim Petersdea48ec2001-05-22 20:40:22 +0000201 } while(0)
202
Raymond Hettinger43442782004-03-17 21:55:03 +0000203/* Dictionary reuse scheme to save calls to malloc, free, and memset */
Christian Heimes5b970ad2008-02-06 13:33:44 +0000204#ifndef PyDict_MAXFREELIST
205#define PyDict_MAXFREELIST 80
206#endif
207static PyDictObject *free_list[PyDict_MAXFREELIST];
208static int numfree = 0;
Raymond Hettinger43442782004-03-17 21:55:03 +0000209
Christian Heimesf75dbef2008-02-08 00:11:31 +0000210void
211PyDict_Fini(void)
212{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000213 PyDictObject *op;
Christian Heimesf75dbef2008-02-08 00:11:31 +0000214
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000215 while (numfree) {
216 op = free_list[--numfree];
217 assert(PyDict_CheckExact(op));
218 PyObject_GC_Del(op);
219 }
Christian Heimesf75dbef2008-02-08 00:11:31 +0000220}
221
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000222PyObject *
Thomas Wouters78890102000-07-22 19:25:51 +0000223PyDict_New(void)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000224{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000225 register PyDictObject *mp;
226 if (dummy == NULL) { /* Auto-initialize dummy */
227 dummy = PyString_FromString("<dummy key>");
228 if (dummy == NULL)
229 return NULL;
Fred Drake1bff34a2000-08-31 19:31:38 +0000230#ifdef SHOW_CONVERSION_COUNTS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000231 Py_AtExit(show_counts);
Fred Drake1bff34a2000-08-31 19:31:38 +0000232#endif
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000233#ifdef SHOW_ALLOC_COUNT
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000234 Py_AtExit(show_alloc);
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000235#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000236 }
237 if (numfree) {
238 mp = free_list[--numfree];
239 assert (mp != NULL);
240 assert (Py_TYPE(mp) == &PyDict_Type);
241 _Py_NewReference((PyObject *)mp);
242 if (mp->ma_fill) {
243 EMPTY_TO_MINSIZE(mp);
244 } else {
245 /* At least set ma_table and ma_mask; these are wrong
246 if an empty but presized dict is added to freelist */
247 INIT_NONZERO_DICT_SLOTS(mp);
248 }
249 assert (mp->ma_used == 0);
250 assert (mp->ma_table == mp->ma_smalltable);
251 assert (mp->ma_mask == PyDict_MINSIZE - 1);
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000252#ifdef SHOW_ALLOC_COUNT
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000253 count_reuse++;
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000254#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000255 } else {
256 mp = PyObject_GC_New(PyDictObject, &PyDict_Type);
257 if (mp == NULL)
258 return NULL;
259 EMPTY_TO_MINSIZE(mp);
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000260#ifdef SHOW_ALLOC_COUNT
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000261 count_alloc++;
Christian Heimesb4ee4a12008-02-07 17:15:30 +0000262#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000263 }
264 mp->ma_lookup = lookdict_string;
Fred Drake1bff34a2000-08-31 19:31:38 +0000265#ifdef SHOW_CONVERSION_COUNTS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000266 ++created;
Fred Drake1bff34a2000-08-31 19:31:38 +0000267#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000268 _PyObject_GC_TRACK(mp);
269 return (PyObject *)mp;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000270}
271
272/*
273The basic lookup function used by all operations.
Guido van Rossum16e93a81997-01-28 00:00:11 +0000274This is based on Algorithm D from Knuth Vol. 3, Sec. 6.4.
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000275Open addressing is preferred over chaining since the link overhead for
276chaining would be substantial (100% with typical malloc overhead).
277
Tim Peterseb28ef22001-06-02 05:27:19 +0000278The initial probe index is computed as hash mod the table size. Subsequent
279probe indices are computed as explained earlier.
Guido van Rossum2bc13791999-03-24 19:06:42 +0000280
281All arithmetic on hash should ignore overflow.
Guido van Rossum16e93a81997-01-28 00:00:11 +0000282
Tim Peterseb28ef22001-06-02 05:27:19 +0000283(The details in this version are due to Tim Peters, building on many past
284contributions by Reimer Behrends, Jyrki Alakuijala, Vladimir Marangozov and
285Christian Tismer).
Fred Drake1bff34a2000-08-31 19:31:38 +0000286
Tim Petersd770ebd2006-06-01 15:50:44 +0000287lookdict() is general-purpose, and may return NULL if (and only if) a
288comparison raises an exception (this was new in Python 2.5).
289lookdict_string() below is specialized to string keys, comparison of which can
290never raise an exception; that function can never return NULL. For both, when
Brett Cannon77ae87c2007-10-10 00:07:50 +0000291the key isn't found a PyDictEntry* is returned for which the me_value field is
Tim Petersd770ebd2006-06-01 15:50:44 +0000292NULL; this is the slot in the dict at which the key would have been found, and
293the caller can (if it wishes) add the <key, value> pair to the returned
Brett Cannon77ae87c2007-10-10 00:07:50 +0000294PyDictEntry*.
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000295*/
Brett Cannon77ae87c2007-10-10 00:07:50 +0000296static PyDictEntry *
297lookdict(PyDictObject *mp, PyObject *key, register long hash)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000298{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000299 register size_t i;
300 register size_t perturb;
301 register PyDictEntry *freeslot;
302 register size_t mask = (size_t)mp->ma_mask;
303 PyDictEntry *ep0 = mp->ma_table;
304 register PyDictEntry *ep;
305 register int cmp;
306 PyObject *startkey;
Tim Peterseb28ef22001-06-02 05:27:19 +0000307
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000308 i = (size_t)hash & mask;
309 ep = &ep0[i];
310 if (ep->me_key == NULL || ep->me_key == key)
311 return ep;
Tim Peters7b5d0af2001-06-03 04:14:43 +0000312
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000313 if (ep->me_key == dummy)
314 freeslot = ep;
315 else {
316 if (ep->me_hash == hash) {
317 startkey = ep->me_key;
318 Py_INCREF(startkey);
319 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
320 Py_DECREF(startkey);
321 if (cmp < 0)
322 return NULL;
323 if (ep0 == mp->ma_table && ep->me_key == startkey) {
324 if (cmp > 0)
325 return ep;
326 }
327 else {
328 /* The compare did major nasty stuff to the
329 * dict: start over.
330 * XXX A clever adversary could prevent this
331 * XXX from terminating.
332 */
333 return lookdict(mp, key, hash);
334 }
335 }
336 freeslot = NULL;
337 }
Tim Peters15d49292001-05-27 07:39:22 +0000338
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000339 /* In the loop, me_key == dummy is by far (factor of 100s) the
340 least likely outcome, so test for that last. */
341 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
342 i = (i << 2) + i + perturb + 1;
343 ep = &ep0[i & mask];
344 if (ep->me_key == NULL)
345 return freeslot == NULL ? ep : freeslot;
346 if (ep->me_key == key)
347 return ep;
348 if (ep->me_hash == hash && ep->me_key != dummy) {
349 startkey = ep->me_key;
350 Py_INCREF(startkey);
351 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
352 Py_DECREF(startkey);
353 if (cmp < 0)
354 return NULL;
355 if (ep0 == mp->ma_table && ep->me_key == startkey) {
356 if (cmp > 0)
357 return ep;
358 }
359 else {
360 /* The compare did major nasty stuff to the
361 * dict: start over.
362 * XXX A clever adversary could prevent this
363 * XXX from terminating.
364 */
365 return lookdict(mp, key, hash);
366 }
367 }
368 else if (ep->me_key == dummy && freeslot == NULL)
369 freeslot = ep;
370 }
371 assert(0); /* NOT REACHED */
372 return 0;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000373}
374
375/*
Fred Drake1bff34a2000-08-31 19:31:38 +0000376 * Hacked up version of lookdict which can assume keys are always strings;
Tim Petersd770ebd2006-06-01 15:50:44 +0000377 * this assumption allows testing for errors during PyObject_RichCompareBool()
378 * to be dropped; string-string comparisons never raise exceptions. This also
379 * means we don't need to go through PyObject_RichCompareBool(); we can always
Gregory P. Smithdd96db62008-06-09 04:58:54 +0000380 * use _PyString_Eq() directly.
Fred Drake1bff34a2000-08-31 19:31:38 +0000381 *
Tim Petersd770ebd2006-06-01 15:50:44 +0000382 * This is valuable because dicts with only string keys are very common.
Fred Drake1bff34a2000-08-31 19:31:38 +0000383 */
Brett Cannon77ae87c2007-10-10 00:07:50 +0000384static PyDictEntry *
385lookdict_string(PyDictObject *mp, PyObject *key, register long hash)
Fred Drake1bff34a2000-08-31 19:31:38 +0000386{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000387 register size_t i;
388 register size_t perturb;
389 register PyDictEntry *freeslot;
390 register size_t mask = (size_t)mp->ma_mask;
391 PyDictEntry *ep0 = mp->ma_table;
392 register PyDictEntry *ep;
Fred Drake1bff34a2000-08-31 19:31:38 +0000393
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000394 /* Make sure this function doesn't have to handle non-string keys,
395 including subclasses of str; e.g., one reason to subclass
396 strings is to override __eq__, and for speed we don't cater to
397 that here. */
398 if (!PyString_CheckExact(key)) {
Fred Drake1bff34a2000-08-31 19:31:38 +0000399#ifdef SHOW_CONVERSION_COUNTS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000400 ++converted;
Fred Drake1bff34a2000-08-31 19:31:38 +0000401#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000402 mp->ma_lookup = lookdict;
403 return lookdict(mp, key, hash);
404 }
405 i = hash & mask;
406 ep = &ep0[i];
407 if (ep->me_key == NULL || ep->me_key == key)
408 return ep;
409 if (ep->me_key == dummy)
410 freeslot = ep;
411 else {
412 if (ep->me_hash == hash && _PyString_Eq(ep->me_key, key))
413 return ep;
414 freeslot = NULL;
415 }
Tim Peters15d49292001-05-27 07:39:22 +0000416
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000417 /* In the loop, me_key == dummy is by far (factor of 100s) the
418 least likely outcome, so test for that last. */
419 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
420 i = (i << 2) + i + perturb + 1;
421 ep = &ep0[i & mask];
422 if (ep->me_key == NULL)
423 return freeslot == NULL ? ep : freeslot;
424 if (ep->me_key == key
425 || (ep->me_hash == hash
426 && ep->me_key != dummy
427 && _PyString_Eq(ep->me_key, key)))
428 return ep;
429 if (ep->me_key == dummy && freeslot == NULL)
430 freeslot = ep;
431 }
432 assert(0); /* NOT REACHED */
433 return 0;
Fred Drake1bff34a2000-08-31 19:31:38 +0000434}
435
436/*
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000437Internal routine to insert a new item into the table.
438Used both by the internal resize routine and by the public insert routine.
439Eats a reference to key and one to value.
Tim Petersd770ebd2006-06-01 15:50:44 +0000440Returns -1 if an error occurred, or 0 on success.
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000441*/
Armin Rigo35f6d362006-06-01 13:19:12 +0000442static int
Brett Cannon77ae87c2007-10-10 00:07:50 +0000443insertdict(register PyDictObject *mp, PyObject *key, long hash, PyObject *value)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000444{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000445 PyObject *old_value;
446 register PyDictEntry *ep;
447 typedef PyDictEntry *(*lookupfunc)(PyDictObject *, PyObject *, long);
Tim Peters6d6c1a32001-08-02 04:15:00 +0000448
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000449 assert(mp->ma_lookup != NULL);
450 ep = mp->ma_lookup(mp, key, hash);
451 if (ep == NULL) {
452 Py_DECREF(key);
453 Py_DECREF(value);
454 return -1;
455 }
456 if (ep->me_value != NULL) {
457 old_value = ep->me_value;
458 ep->me_value = value;
459 Py_DECREF(old_value); /* which **CAN** re-enter */
460 Py_DECREF(key);
461 }
462 else {
463 if (ep->me_key == NULL)
464 mp->ma_fill++;
465 else {
466 assert(ep->me_key == dummy);
467 Py_DECREF(dummy);
468 }
469 ep->me_key = key;
470 ep->me_hash = (Py_ssize_t)hash;
471 ep->me_value = value;
472 mp->ma_used++;
473 }
474 return 0;
Armin Rigo35f6d362006-06-01 13:19:12 +0000475}
476
477/*
478Internal routine used by dictresize() to insert an item which is
479known to be absent from the dict. This routine also assumes that
480the dict contains no deleted entries. Besides the performance benefit,
481using insertdict() in dictresize() is dangerous (SF bug #1456209).
Tim Petersd770ebd2006-06-01 15:50:44 +0000482Note that no refcounts are changed by this routine; if needed, the caller
483is responsible for incref'ing `key` and `value`.
Armin Rigo35f6d362006-06-01 13:19:12 +0000484*/
485static void
Brett Cannon77ae87c2007-10-10 00:07:50 +0000486insertdict_clean(register PyDictObject *mp, PyObject *key, long hash,
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000487 PyObject *value)
Armin Rigo35f6d362006-06-01 13:19:12 +0000488{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000489 register size_t i;
490 register size_t perturb;
491 register size_t mask = (size_t)mp->ma_mask;
492 PyDictEntry *ep0 = mp->ma_table;
493 register PyDictEntry *ep;
Armin Rigo35f6d362006-06-01 13:19:12 +0000494
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000495 i = hash & mask;
496 ep = &ep0[i];
497 for (perturb = hash; ep->me_key != NULL; perturb >>= PERTURB_SHIFT) {
498 i = (i << 2) + i + perturb + 1;
499 ep = &ep0[i & mask];
500 }
501 assert(ep->me_value == NULL);
502 mp->ma_fill++;
503 ep->me_key = key;
504 ep->me_hash = (Py_ssize_t)hash;
505 ep->me_value = value;
506 mp->ma_used++;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000507}
508
509/*
510Restructure the table by allocating a new table and reinserting all
511items again. When entries have been deleted, the new table may
512actually be smaller than the old one.
513*/
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000514static int
Brett Cannon77ae87c2007-10-10 00:07:50 +0000515dictresize(PyDictObject *mp, Py_ssize_t minused)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000516{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000517 Py_ssize_t newsize;
518 PyDictEntry *oldtable, *newtable, *ep;
519 Py_ssize_t i;
520 int is_oldtable_malloced;
521 PyDictEntry small_copy[PyDict_MINSIZE];
Tim Peters91a364d2001-05-19 07:04:38 +0000522
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000523 assert(minused >= 0);
Tim Peters0c6010b2001-05-23 23:33:57 +0000524
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000525 /* Find the smallest table size > minused. */
526 for (newsize = PyDict_MINSIZE;
527 newsize <= minused && newsize > 0;
528 newsize <<= 1)
529 ;
530 if (newsize <= 0) {
531 PyErr_NoMemory();
532 return -1;
533 }
Tim Peters0c6010b2001-05-23 23:33:57 +0000534
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000535 /* Get space for a new table. */
536 oldtable = mp->ma_table;
537 assert(oldtable != NULL);
538 is_oldtable_malloced = oldtable != mp->ma_smalltable;
Tim Peters0c6010b2001-05-23 23:33:57 +0000539
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000540 if (newsize == PyDict_MINSIZE) {
541 /* A large table is shrinking, or we can't get any smaller. */
542 newtable = mp->ma_smalltable;
543 if (newtable == oldtable) {
544 if (mp->ma_fill == mp->ma_used) {
545 /* No dummies, so no point doing anything. */
546 return 0;
547 }
548 /* We're not going to resize it, but rebuild the
549 table anyway to purge old dummy entries.
550 Subtle: This is *necessary* if fill==size,
551 as lookdict needs at least one virgin slot to
552 terminate failing searches. If fill < size, it's
553 merely desirable, as dummies slow searches. */
554 assert(mp->ma_fill > mp->ma_used);
555 memcpy(small_copy, oldtable, sizeof(small_copy));
556 oldtable = small_copy;
557 }
558 }
559 else {
560 newtable = PyMem_NEW(PyDictEntry, newsize);
561 if (newtable == NULL) {
562 PyErr_NoMemory();
563 return -1;
564 }
565 }
Tim Peters0c6010b2001-05-23 23:33:57 +0000566
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000567 /* Make the dict empty, using the new table. */
568 assert(newtable != oldtable);
569 mp->ma_table = newtable;
570 mp->ma_mask = newsize - 1;
571 memset(newtable, 0, sizeof(PyDictEntry) * newsize);
572 mp->ma_used = 0;
573 i = mp->ma_fill;
574 mp->ma_fill = 0;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000575
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000576 /* Copy the data over; this is refcount-neutral for active entries;
577 dummy entries aren't copied over, of course */
578 for (ep = oldtable; i > 0; ep++) {
579 if (ep->me_value != NULL) { /* active entry */
580 --i;
581 insertdict_clean(mp, ep->me_key, (long)ep->me_hash,
582 ep->me_value);
583 }
584 else if (ep->me_key != NULL) { /* dummy entry */
585 --i;
586 assert(ep->me_key == dummy);
587 Py_DECREF(ep->me_key);
588 }
589 /* else key == value == NULL: nothing to do */
590 }
Guido van Rossumd7047b31995-01-02 19:07:15 +0000591
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000592 if (is_oldtable_malloced)
593 PyMem_DEL(oldtable);
594 return 0;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000595}
596
Raymond Hettingerfd7ed402007-12-18 21:24:09 +0000597/* Create a new dictionary pre-sized to hold an estimated number of elements.
598 Underestimates are okay because the dictionary will resize as necessary.
599 Overestimates just mean the dictionary will be more sparse than usual.
600*/
601
602PyObject *
603_PyDict_NewPresized(Py_ssize_t minused)
604{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000605 PyObject *op = PyDict_New();
Raymond Hettingerfd7ed402007-12-18 21:24:09 +0000606
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000607 if (minused>5 && op != NULL && dictresize((PyDictObject *)op, minused) == -1) {
608 Py_DECREF(op);
609 return NULL;
610 }
611 return op;
Raymond Hettingerfd7ed402007-12-18 21:24:09 +0000612}
613
Tim Petersd770ebd2006-06-01 15:50:44 +0000614/* Note that, for historical reasons, PyDict_GetItem() suppresses all errors
615 * that may occur (originally dicts supported only string keys, and exceptions
616 * weren't possible). So, while the original intent was that a NULL return
Andrew M. Kuchling0067b5f2006-08-04 20:37:43 +0000617 * meant the key wasn't present, in reality it can mean that, or that an error
Tim Petersd770ebd2006-06-01 15:50:44 +0000618 * (suppressed) occurred while computing the key's hash, or that some error
619 * (suppressed) occurred when comparing keys in the dict's internal probe
620 * sequence. A nasty example of the latter is when a Python-coded comparison
621 * function hits a stack-depth error, which can cause this to return NULL
622 * even if the key is present.
623 */
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000624PyObject *
Tim Peters1f5871e2000-07-04 17:44:48 +0000625PyDict_GetItem(PyObject *op, PyObject *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000626{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000627 long hash;
628 PyDictObject *mp = (PyDictObject *)op;
629 PyDictEntry *ep;
630 PyThreadState *tstate;
631 if (!PyDict_Check(op))
632 return NULL;
633 if (!PyString_CheckExact(key) ||
634 (hash = ((PyStringObject *) key)->ob_shash) == -1)
635 {
636 hash = PyObject_Hash(key);
637 if (hash == -1) {
638 PyErr_Clear();
639 return NULL;
640 }
641 }
Armin Rigo35f6d362006-06-01 13:19:12 +0000642
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000643 /* We can arrive here with a NULL tstate during initialization:
644 try running "python -Wi" for an example related to string
645 interning. Let's just hope that no exception occurs then... */
646 tstate = _PyThreadState_Current;
647 if (tstate != NULL && tstate->curexc_type != NULL) {
648 /* preserve the existing exception */
649 PyObject *err_type, *err_value, *err_tb;
650 PyErr_Fetch(&err_type, &err_value, &err_tb);
651 ep = (mp->ma_lookup)(mp, key, hash);
652 /* ignore errors */
653 PyErr_Restore(err_type, err_value, err_tb);
654 if (ep == NULL)
655 return NULL;
656 }
657 else {
658 ep = (mp->ma_lookup)(mp, key, hash);
659 if (ep == NULL) {
660 PyErr_Clear();
661 return NULL;
662 }
663 }
664 return ep->me_value;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000665}
666
Guido van Rossuma4dd0112001-04-15 22:16:26 +0000667/* CAUTION: PyDict_SetItem() must guarantee that it won't resize the
Tim Peters60b29962006-01-01 01:19:23 +0000668 * dictionary if it's merely replacing the value for an existing key.
669 * This means that it's safe to loop over a dictionary with PyDict_Next()
670 * and occasionally replace a value -- but you can't insert new keys or
671 * remove them.
Guido van Rossuma4dd0112001-04-15 22:16:26 +0000672 */
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000673int
Tim Peters1f5871e2000-07-04 17:44:48 +0000674PyDict_SetItem(register PyObject *op, PyObject *key, PyObject *value)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000675{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000676 register PyDictObject *mp;
677 register long hash;
678 register Py_ssize_t n_used;
Guido van Rossuma4dd0112001-04-15 22:16:26 +0000679
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000680 if (!PyDict_Check(op)) {
681 PyErr_BadInternalCall();
682 return -1;
683 }
684 assert(key);
685 assert(value);
686 mp = (PyDictObject *)op;
687 if (PyString_CheckExact(key)) {
688 hash = ((PyStringObject *)key)->ob_shash;
689 if (hash == -1)
690 hash = PyObject_Hash(key);
691 }
692 else {
693 hash = PyObject_Hash(key);
694 if (hash == -1)
695 return -1;
696 }
697 assert(mp->ma_fill <= mp->ma_mask); /* at least one empty slot */
698 n_used = mp->ma_used;
699 Py_INCREF(value);
700 Py_INCREF(key);
701 if (insertdict(mp, key, hash, value) != 0)
702 return -1;
703 /* If we added a key, we can safely resize. Otherwise just return!
704 * If fill >= 2/3 size, adjust size. Normally, this doubles or
705 * quaduples the size, but it's also possible for the dict to shrink
706 * (if ma_fill is much larger than ma_used, meaning a lot of dict
707 * keys have been * deleted).
708 *
709 * Quadrupling the size improves average dictionary sparseness
710 * (reducing collisions) at the cost of some memory and iteration
711 * speed (which loops over every possible entry). It also halves
712 * the number of expensive resize operations in a growing dictionary.
713 *
714 * Very large dictionaries (over 50K items) use doubling instead.
715 * This may help applications with severe memory constraints.
716 */
717 if (!(mp->ma_used > n_used && mp->ma_fill*3 >= (mp->ma_mask+1)*2))
718 return 0;
719 return dictresize(mp, (mp->ma_used > 50000 ? 2 : 4) * mp->ma_used);
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000720}
721
722int
Tim Peters1f5871e2000-07-04 17:44:48 +0000723PyDict_DelItem(PyObject *op, PyObject *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000724{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000725 register PyDictObject *mp;
726 register long hash;
727 register PyDictEntry *ep;
728 PyObject *old_value, *old_key;
Guido van Rossumd7047b31995-01-02 19:07:15 +0000729
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000730 if (!PyDict_Check(op)) {
731 PyErr_BadInternalCall();
732 return -1;
733 }
734 assert(key);
735 if (!PyString_CheckExact(key) ||
736 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
737 hash = PyObject_Hash(key);
738 if (hash == -1)
739 return -1;
740 }
741 mp = (PyDictObject *)op;
742 ep = (mp->ma_lookup)(mp, key, hash);
743 if (ep == NULL)
744 return -1;
745 if (ep->me_value == NULL) {
746 set_key_error(key);
747 return -1;
748 }
749 old_key = ep->me_key;
750 Py_INCREF(dummy);
751 ep->me_key = dummy;
752 old_value = ep->me_value;
753 ep->me_value = NULL;
754 mp->ma_used--;
755 Py_DECREF(old_value);
756 Py_DECREF(old_key);
757 return 0;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000758}
759
Guido van Rossum25831651993-05-19 14:50:45 +0000760void
Tim Peters1f5871e2000-07-04 17:44:48 +0000761PyDict_Clear(PyObject *op)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000762{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000763 PyDictObject *mp;
764 PyDictEntry *ep, *table;
765 int table_is_malloced;
766 Py_ssize_t fill;
767 PyDictEntry small_copy[PyDict_MINSIZE];
Tim Petersdea48ec2001-05-22 20:40:22 +0000768#ifdef Py_DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000769 Py_ssize_t i, n;
Tim Petersdea48ec2001-05-22 20:40:22 +0000770#endif
771
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000772 if (!PyDict_Check(op))
773 return;
774 mp = (PyDictObject *)op;
Tim Petersdea48ec2001-05-22 20:40:22 +0000775#ifdef Py_DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000776 n = mp->ma_mask + 1;
777 i = 0;
Tim Petersdea48ec2001-05-22 20:40:22 +0000778#endif
779
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000780 table = mp->ma_table;
781 assert(table != NULL);
782 table_is_malloced = table != mp->ma_smalltable;
Tim Petersdea48ec2001-05-22 20:40:22 +0000783
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000784 /* This is delicate. During the process of clearing the dict,
785 * decrefs can cause the dict to mutate. To avoid fatal confusion
786 * (voice of experience), we have to make the dict empty before
787 * clearing the slots, and never refer to anything via mp->xxx while
788 * clearing.
789 */
790 fill = mp->ma_fill;
791 if (table_is_malloced)
792 EMPTY_TO_MINSIZE(mp);
Tim Petersdea48ec2001-05-22 20:40:22 +0000793
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000794 else if (fill > 0) {
795 /* It's a small table with something that needs to be cleared.
796 * Afraid the only safe way is to copy the dict entries into
797 * another small table first.
798 */
799 memcpy(small_copy, table, sizeof(small_copy));
800 table = small_copy;
801 EMPTY_TO_MINSIZE(mp);
802 }
803 /* else it's a small table that's already empty */
Tim Petersdea48ec2001-05-22 20:40:22 +0000804
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000805 /* Now we can finally clear things. If C had refcounts, we could
806 * assert that the refcount on table is 1 now, i.e. that this function
807 * has unique access to it, so decref side-effects can't alter it.
808 */
809 for (ep = table; fill > 0; ++ep) {
Tim Petersdea48ec2001-05-22 20:40:22 +0000810#ifdef Py_DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000811 assert(i < n);
812 ++i;
Tim Petersdea48ec2001-05-22 20:40:22 +0000813#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000814 if (ep->me_key) {
815 --fill;
816 Py_DECREF(ep->me_key);
817 Py_XDECREF(ep->me_value);
818 }
Tim Petersdea48ec2001-05-22 20:40:22 +0000819#ifdef Py_DEBUG
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000820 else
821 assert(ep->me_value == NULL);
Tim Petersdea48ec2001-05-22 20:40:22 +0000822#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000823 }
Tim Petersdea48ec2001-05-22 20:40:22 +0000824
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000825 if (table_is_malloced)
826 PyMem_DEL(table);
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000827}
828
Tim Peters080c88b2003-02-15 03:01:11 +0000829/*
830 * Iterate over a dict. Use like so:
831 *
Tim Peters9b10f7e2006-05-30 04:16:25 +0000832 * Py_ssize_t i;
Tim Peters080c88b2003-02-15 03:01:11 +0000833 * PyObject *key, *value;
834 * i = 0; # important! i should not otherwise be changed by you
Neal Norwitz07323012003-02-15 14:45:12 +0000835 * while (PyDict_Next(yourdict, &i, &key, &value)) {
Tim Peters080c88b2003-02-15 03:01:11 +0000836 * Refer to borrowed references in key and value.
837 * }
838 *
839 * CAUTION: In general, it isn't safe to use PyDict_Next in a loop that
Tim Peters67830702001-03-21 19:23:56 +0000840 * mutates the dict. One exception: it is safe if the loop merely changes
841 * the values associated with the keys (but doesn't insert new keys or
842 * delete keys), via PyDict_SetItem().
843 */
Guido van Rossum25831651993-05-19 14:50:45 +0000844int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000845PyDict_Next(PyObject *op, Py_ssize_t *ppos, PyObject **pkey, PyObject **pvalue)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000846{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000847 register Py_ssize_t i;
848 register Py_ssize_t mask;
849 register PyDictEntry *ep;
Raymond Hettinger43442782004-03-17 21:55:03 +0000850
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000851 if (!PyDict_Check(op))
852 return 0;
853 i = *ppos;
854 if (i < 0)
855 return 0;
856 ep = ((PyDictObject *)op)->ma_table;
857 mask = ((PyDictObject *)op)->ma_mask;
858 while (i <= mask && ep[i].me_value == NULL)
859 i++;
860 *ppos = i+1;
861 if (i > mask)
862 return 0;
863 if (pkey)
864 *pkey = ep[i].me_key;
865 if (pvalue)
866 *pvalue = ep[i].me_value;
867 return 1;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000868}
869
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +0000870/* Internal version of PyDict_Next that returns a hash value in addition to the key and value.*/
871int
872_PyDict_Next(PyObject *op, Py_ssize_t *ppos, PyObject **pkey, PyObject **pvalue, long *phash)
873{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000874 register Py_ssize_t i;
875 register Py_ssize_t mask;
876 register PyDictEntry *ep;
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +0000877
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000878 if (!PyDict_Check(op))
879 return 0;
880 i = *ppos;
881 if (i < 0)
882 return 0;
883 ep = ((PyDictObject *)op)->ma_table;
884 mask = ((PyDictObject *)op)->ma_mask;
885 while (i <= mask && ep[i].me_value == NULL)
886 i++;
887 *ppos = i+1;
888 if (i > mask)
889 return 0;
890 *phash = (long)(ep[i].me_hash);
891 if (pkey)
892 *pkey = ep[i].me_key;
893 if (pvalue)
894 *pvalue = ep[i].me_value;
895 return 1;
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +0000896}
897
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000898/* Methods */
899
900static void
Brett Cannon77ae87c2007-10-10 00:07:50 +0000901dict_dealloc(register PyDictObject *mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000902{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000903 register PyDictEntry *ep;
904 Py_ssize_t fill = mp->ma_fill;
905 PyObject_GC_UnTrack(mp);
906 Py_TRASHCAN_SAFE_BEGIN(mp)
907 for (ep = mp->ma_table; fill > 0; ep++) {
908 if (ep->me_key) {
909 --fill;
910 Py_DECREF(ep->me_key);
911 Py_XDECREF(ep->me_value);
912 }
913 }
914 if (mp->ma_table != mp->ma_smalltable)
915 PyMem_DEL(mp->ma_table);
916 if (numfree < PyDict_MAXFREELIST && Py_TYPE(mp) == &PyDict_Type)
917 free_list[numfree++] = mp;
918 else
919 Py_TYPE(mp)->tp_free((PyObject *)mp);
920 Py_TRASHCAN_SAFE_END(mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000921}
922
923static int
Brett Cannon77ae87c2007-10-10 00:07:50 +0000924dict_print(register PyDictObject *mp, register FILE *fp, register int flags)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000925{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000926 register Py_ssize_t i;
927 register Py_ssize_t any;
928 int status;
Guido van Rossum255443b1998-04-10 22:47:14 +0000929
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000930 status = Py_ReprEnter((PyObject*)mp);
931 if (status != 0) {
932 if (status < 0)
933 return status;
934 Py_BEGIN_ALLOW_THREADS
935 fprintf(fp, "{...}");
936 Py_END_ALLOW_THREADS
937 return 0;
938 }
Guido van Rossum255443b1998-04-10 22:47:14 +0000939
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000940 Py_BEGIN_ALLOW_THREADS
941 fprintf(fp, "{");
942 Py_END_ALLOW_THREADS
943 any = 0;
944 for (i = 0; i <= mp->ma_mask; i++) {
945 PyDictEntry *ep = mp->ma_table + i;
946 PyObject *pvalue = ep->me_value;
947 if (pvalue != NULL) {
948 /* Prevent PyObject_Repr from deleting value during
949 key format */
950 Py_INCREF(pvalue);
951 if (any++ > 0) {
952 Py_BEGIN_ALLOW_THREADS
953 fprintf(fp, ", ");
954 Py_END_ALLOW_THREADS
955 }
956 if (PyObject_Print((PyObject *)ep->me_key, fp, 0)!=0) {
957 Py_DECREF(pvalue);
958 Py_ReprLeave((PyObject*)mp);
959 return -1;
960 }
961 Py_BEGIN_ALLOW_THREADS
962 fprintf(fp, ": ");
963 Py_END_ALLOW_THREADS
964 if (PyObject_Print(pvalue, fp, 0) != 0) {
965 Py_DECREF(pvalue);
966 Py_ReprLeave((PyObject*)mp);
967 return -1;
968 }
969 Py_DECREF(pvalue);
970 }
971 }
972 Py_BEGIN_ALLOW_THREADS
973 fprintf(fp, "}");
974 Py_END_ALLOW_THREADS
975 Py_ReprLeave((PyObject*)mp);
976 return 0;
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000977}
978
Guido van Rossumc0b618a1997-05-02 03:12:38 +0000979static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +0000980dict_repr(PyDictObject *mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +0000981{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000982 Py_ssize_t i;
983 PyObject *s, *temp, *colon = NULL;
984 PyObject *pieces = NULL, *result = NULL;
985 PyObject *key, *value;
Guido van Rossum255443b1998-04-10 22:47:14 +0000986
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000987 i = Py_ReprEnter((PyObject *)mp);
988 if (i != 0) {
989 return i > 0 ? PyString_FromString("{...}") : NULL;
990 }
Guido van Rossum255443b1998-04-10 22:47:14 +0000991
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000992 if (mp->ma_used == 0) {
993 result = PyString_FromString("{}");
994 goto Done;
995 }
Tim Petersa7259592001-06-16 05:11:17 +0000996
Antoine Pitrouc7c96a92010-05-09 15:15:40 +0000997 pieces = PyList_New(0);
998 if (pieces == NULL)
999 goto Done;
Tim Petersa7259592001-06-16 05:11:17 +00001000
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001001 colon = PyString_FromString(": ");
1002 if (colon == NULL)
1003 goto Done;
Tim Petersa7259592001-06-16 05:11:17 +00001004
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001005 /* Do repr() on each key+value pair, and insert ": " between them.
1006 Note that repr may mutate the dict. */
1007 i = 0;
1008 while (PyDict_Next((PyObject *)mp, &i, &key, &value)) {
1009 int status;
1010 /* Prevent repr from deleting value during key format. */
1011 Py_INCREF(value);
1012 s = PyObject_Repr(key);
1013 PyString_Concat(&s, colon);
1014 PyString_ConcatAndDel(&s, PyObject_Repr(value));
1015 Py_DECREF(value);
1016 if (s == NULL)
1017 goto Done;
1018 status = PyList_Append(pieces, s);
1019 Py_DECREF(s); /* append created a new ref */
1020 if (status < 0)
1021 goto Done;
1022 }
Tim Petersa7259592001-06-16 05:11:17 +00001023
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001024 /* Add "{}" decorations to the first and last items. */
1025 assert(PyList_GET_SIZE(pieces) > 0);
1026 s = PyString_FromString("{");
1027 if (s == NULL)
1028 goto Done;
1029 temp = PyList_GET_ITEM(pieces, 0);
1030 PyString_ConcatAndDel(&s, temp);
1031 PyList_SET_ITEM(pieces, 0, s);
1032 if (s == NULL)
1033 goto Done;
Tim Petersa7259592001-06-16 05:11:17 +00001034
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001035 s = PyString_FromString("}");
1036 if (s == NULL)
1037 goto Done;
1038 temp = PyList_GET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1);
1039 PyString_ConcatAndDel(&temp, s);
1040 PyList_SET_ITEM(pieces, PyList_GET_SIZE(pieces) - 1, temp);
1041 if (temp == NULL)
1042 goto Done;
Tim Petersa7259592001-06-16 05:11:17 +00001043
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001044 /* Paste them all together with ", " between. */
1045 s = PyString_FromString(", ");
1046 if (s == NULL)
1047 goto Done;
1048 result = _PyString_Join(s, pieces);
1049 Py_DECREF(s);
Tim Petersa7259592001-06-16 05:11:17 +00001050
1051Done:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001052 Py_XDECREF(pieces);
1053 Py_XDECREF(colon);
1054 Py_ReprLeave((PyObject *)mp);
1055 return result;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001056}
1057
Martin v. Löwis18e16552006-02-15 17:27:45 +00001058static Py_ssize_t
Brett Cannon77ae87c2007-10-10 00:07:50 +00001059dict_length(PyDictObject *mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001060{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001061 return mp->ma_used;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001062}
1063
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001064static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001065dict_subscript(PyDictObject *mp, register PyObject *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001066{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001067 PyObject *v;
1068 long hash;
1069 PyDictEntry *ep;
1070 assert(mp->ma_table != NULL);
1071 if (!PyString_CheckExact(key) ||
1072 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
1073 hash = PyObject_Hash(key);
1074 if (hash == -1)
1075 return NULL;
1076 }
1077 ep = (mp->ma_lookup)(mp, key, hash);
1078 if (ep == NULL)
1079 return NULL;
1080 v = ep->me_value;
1081 if (v == NULL) {
1082 if (!PyDict_CheckExact(mp)) {
1083 /* Look up __missing__ method if we're a subclass. */
1084 PyObject *missing;
1085 static PyObject *missing_str = NULL;
1086 if (missing_str == NULL)
1087 missing_str =
1088 PyString_InternFromString("__missing__");
1089 missing = _PyType_Lookup(Py_TYPE(mp), missing_str);
1090 if (missing != NULL)
1091 return PyObject_CallFunctionObjArgs(missing,
1092 (PyObject *)mp, key, NULL);
1093 }
1094 set_key_error(key);
1095 return NULL;
1096 }
1097 else
1098 Py_INCREF(v);
1099 return v;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001100}
1101
1102static int
Brett Cannon77ae87c2007-10-10 00:07:50 +00001103dict_ass_sub(PyDictObject *mp, PyObject *v, PyObject *w)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001104{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001105 if (w == NULL)
1106 return PyDict_DelItem((PyObject *)mp, v);
1107 else
1108 return PyDict_SetItem((PyObject *)mp, v, w);
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001109}
1110
Guido van Rossuma9e7a811997-05-13 21:02:11 +00001111static PyMappingMethods dict_as_mapping = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001112 (lenfunc)dict_length, /*mp_length*/
1113 (binaryfunc)dict_subscript, /*mp_subscript*/
1114 (objobjargproc)dict_ass_sub, /*mp_ass_subscript*/
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001115};
1116
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001117static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001118dict_keys(register PyDictObject *mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001119{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001120 register PyObject *v;
1121 register Py_ssize_t i, j;
1122 PyDictEntry *ep;
1123 Py_ssize_t mask, n;
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001124
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001125 again:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001126 n = mp->ma_used;
1127 v = PyList_New(n);
1128 if (v == NULL)
1129 return NULL;
1130 if (n != mp->ma_used) {
1131 /* Durnit. The allocations caused the dict to resize.
1132 * Just start over, this shouldn't normally happen.
1133 */
1134 Py_DECREF(v);
1135 goto again;
1136 }
1137 ep = mp->ma_table;
1138 mask = mp->ma_mask;
1139 for (i = 0, j = 0; i <= mask; i++) {
1140 if (ep[i].me_value != NULL) {
1141 PyObject *key = ep[i].me_key;
1142 Py_INCREF(key);
1143 PyList_SET_ITEM(v, j, key);
1144 j++;
1145 }
1146 }
1147 assert(j == n);
1148 return v;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001149}
1150
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001151static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001152dict_values(register PyDictObject *mp)
Guido van Rossum25831651993-05-19 14:50:45 +00001153{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001154 register PyObject *v;
1155 register Py_ssize_t i, j;
1156 PyDictEntry *ep;
1157 Py_ssize_t mask, n;
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001158
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001159 again:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001160 n = mp->ma_used;
1161 v = PyList_New(n);
1162 if (v == NULL)
1163 return NULL;
1164 if (n != mp->ma_used) {
1165 /* Durnit. The allocations caused the dict to resize.
1166 * Just start over, this shouldn't normally happen.
1167 */
1168 Py_DECREF(v);
1169 goto again;
1170 }
1171 ep = mp->ma_table;
1172 mask = mp->ma_mask;
1173 for (i = 0, j = 0; i <= mask; i++) {
1174 if (ep[i].me_value != NULL) {
1175 PyObject *value = ep[i].me_value;
1176 Py_INCREF(value);
1177 PyList_SET_ITEM(v, j, value);
1178 j++;
1179 }
1180 }
1181 assert(j == n);
1182 return v;
Guido van Rossum25831651993-05-19 14:50:45 +00001183}
1184
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001185static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001186dict_items(register PyDictObject *mp)
Guido van Rossum25831651993-05-19 14:50:45 +00001187{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001188 register PyObject *v;
1189 register Py_ssize_t i, j, n;
1190 Py_ssize_t mask;
1191 PyObject *item, *key, *value;
1192 PyDictEntry *ep;
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001193
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001194 /* Preallocate the list of tuples, to avoid allocations during
1195 * the loop over the items, which could trigger GC, which
1196 * could resize the dict. :-(
1197 */
Guido van Rossuma4dd0112001-04-15 22:16:26 +00001198 again:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001199 n = mp->ma_used;
1200 v = PyList_New(n);
1201 if (v == NULL)
1202 return NULL;
1203 for (i = 0; i < n; i++) {
1204 item = PyTuple_New(2);
1205 if (item == NULL) {
1206 Py_DECREF(v);
1207 return NULL;
1208 }
1209 PyList_SET_ITEM(v, i, item);
1210 }
1211 if (n != mp->ma_used) {
1212 /* Durnit. The allocations caused the dict to resize.
1213 * Just start over, this shouldn't normally happen.
1214 */
1215 Py_DECREF(v);
1216 goto again;
1217 }
1218 /* Nothing we do below makes any function calls. */
1219 ep = mp->ma_table;
1220 mask = mp->ma_mask;
1221 for (i = 0, j = 0; i <= mask; i++) {
1222 if ((value=ep[i].me_value) != NULL) {
1223 key = ep[i].me_key;
1224 item = PyList_GET_ITEM(v, j);
1225 Py_INCREF(key);
1226 PyTuple_SET_ITEM(item, 0, key);
1227 Py_INCREF(value);
1228 PyTuple_SET_ITEM(item, 1, value);
1229 j++;
1230 }
1231 }
1232 assert(j == n);
1233 return v;
Guido van Rossum25831651993-05-19 14:50:45 +00001234}
1235
Guido van Rossume3f5b9c1997-05-28 19:15:28 +00001236static PyObject *
Tim Petersbca1cbc2002-12-09 22:56:13 +00001237dict_fromkeys(PyObject *cls, PyObject *args)
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001238{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001239 PyObject *seq;
1240 PyObject *value = Py_None;
1241 PyObject *it; /* iter(seq) */
1242 PyObject *key;
1243 PyObject *d;
1244 int status;
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001245
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001246 if (!PyArg_UnpackTuple(args, "fromkeys", 1, 2, &seq, &value))
1247 return NULL;
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001248
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001249 d = PyObject_CallObject(cls, NULL);
1250 if (d == NULL)
1251 return NULL;
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001252
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001253 if (PyDict_CheckExact(d) && PyDict_CheckExact(seq)) {
1254 PyDictObject *mp = (PyDictObject *)d;
1255 PyObject *oldvalue;
1256 Py_ssize_t pos = 0;
1257 PyObject *key;
1258 long hash;
Raymond Hettingercdcf8872007-11-07 02:26:17 +00001259
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001260 if (dictresize(mp, Py_SIZE(seq)))
1261 return NULL;
Raymond Hettingercdcf8872007-11-07 02:26:17 +00001262
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001263 while (_PyDict_Next(seq, &pos, &key, &oldvalue, &hash)) {
1264 Py_INCREF(key);
1265 Py_INCREF(value);
1266 if (insertdict(mp, key, hash, value))
1267 return NULL;
1268 }
1269 return d;
1270 }
Raymond Hettingercdcf8872007-11-07 02:26:17 +00001271
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001272 if (PyDict_CheckExact(d) && PyAnySet_CheckExact(seq)) {
1273 PyDictObject *mp = (PyDictObject *)d;
1274 Py_ssize_t pos = 0;
1275 PyObject *key;
1276 long hash;
Raymond Hettinger0bbbfc42007-03-20 21:27:24 +00001277
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001278 if (dictresize(mp, PySet_GET_SIZE(seq)))
1279 return NULL;
Raymond Hettinger0bbbfc42007-03-20 21:27:24 +00001280
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001281 while (_PySet_NextEntry(seq, &pos, &key, &hash)) {
1282 Py_INCREF(key);
1283 Py_INCREF(value);
1284 if (insertdict(mp, key, hash, value))
1285 return NULL;
1286 }
1287 return d;
1288 }
Raymond Hettinger0bbbfc42007-03-20 21:27:24 +00001289
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001290 it = PyObject_GetIter(seq);
1291 if (it == NULL){
1292 Py_DECREF(d);
1293 return NULL;
1294 }
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001295
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001296 if (PyDict_CheckExact(d)) {
1297 while ((key = PyIter_Next(it)) != NULL) {
1298 status = PyDict_SetItem(d, key, value);
1299 Py_DECREF(key);
1300 if (status < 0)
1301 goto Fail;
1302 }
1303 } else {
1304 while ((key = PyIter_Next(it)) != NULL) {
1305 status = PyObject_SetItem(d, key, value);
1306 Py_DECREF(key);
1307 if (status < 0)
1308 goto Fail;
1309 }
1310 }
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001311
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001312 if (PyErr_Occurred())
1313 goto Fail;
1314 Py_DECREF(it);
1315 return d;
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001316
1317Fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001318 Py_DECREF(it);
1319 Py_DECREF(d);
1320 return NULL;
Raymond Hettingere33d3df2002-11-27 07:29:33 +00001321}
1322
Raymond Hettinger31017ae2004-03-04 08:25:44 +00001323static int
1324dict_update_common(PyObject *self, PyObject *args, PyObject *kwds, char *methname)
Guido van Rossume3f5b9c1997-05-28 19:15:28 +00001325{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001326 PyObject *arg = NULL;
1327 int result = 0;
Raymond Hettinger31017ae2004-03-04 08:25:44 +00001328
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001329 if (!PyArg_UnpackTuple(args, methname, 0, 1, &arg))
1330 result = -1;
Raymond Hettinger31017ae2004-03-04 08:25:44 +00001331
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001332 else if (arg != NULL) {
1333 if (PyObject_HasAttrString(arg, "keys"))
1334 result = PyDict_Merge(self, arg, 1);
1335 else
1336 result = PyDict_MergeFromSeq2(self, arg, 1);
1337 }
1338 if (result == 0 && kwds != NULL)
1339 result = PyDict_Merge(self, kwds, 1);
1340 return result;
Raymond Hettinger31017ae2004-03-04 08:25:44 +00001341}
1342
1343static PyObject *
1344dict_update(PyObject *self, PyObject *args, PyObject *kwds)
1345{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001346 if (dict_update_common(self, args, kwds, "update") != -1)
1347 Py_RETURN_NONE;
1348 return NULL;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001349}
1350
Guido van Rossum05ac6de2001-08-10 20:28:28 +00001351/* Update unconditionally replaces existing items.
1352 Merge has a 3rd argument 'override'; if set, it acts like Update,
Tim Peters1fc240e2001-10-26 05:06:50 +00001353 otherwise it leaves existing items unchanged.
1354
1355 PyDict_{Update,Merge} update/merge from a mapping object.
1356
Tim Petersf582b822001-12-11 18:51:08 +00001357 PyDict_MergeFromSeq2 updates/merges from any iterable object
Tim Peters1fc240e2001-10-26 05:06:50 +00001358 producing iterable objects of length 2.
1359*/
1360
Tim Petersf582b822001-12-11 18:51:08 +00001361int
Tim Peters1fc240e2001-10-26 05:06:50 +00001362PyDict_MergeFromSeq2(PyObject *d, PyObject *seq2, int override)
1363{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001364 PyObject *it; /* iter(seq2) */
1365 Py_ssize_t i; /* index into seq2 of current element */
1366 PyObject *item; /* seq2[i] */
1367 PyObject *fast; /* item as a 2-tuple or 2-list */
Tim Peters1fc240e2001-10-26 05:06:50 +00001368
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001369 assert(d != NULL);
1370 assert(PyDict_Check(d));
1371 assert(seq2 != NULL);
Tim Peters1fc240e2001-10-26 05:06:50 +00001372
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001373 it = PyObject_GetIter(seq2);
1374 if (it == NULL)
1375 return -1;
Tim Peters1fc240e2001-10-26 05:06:50 +00001376
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001377 for (i = 0; ; ++i) {
1378 PyObject *key, *value;
1379 Py_ssize_t n;
Tim Peters1fc240e2001-10-26 05:06:50 +00001380
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001381 fast = NULL;
1382 item = PyIter_Next(it);
1383 if (item == NULL) {
1384 if (PyErr_Occurred())
1385 goto Fail;
1386 break;
1387 }
Tim Peters1fc240e2001-10-26 05:06:50 +00001388
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001389 /* Convert item to sequence, and verify length 2. */
1390 fast = PySequence_Fast(item, "");
1391 if (fast == NULL) {
1392 if (PyErr_ExceptionMatches(PyExc_TypeError))
1393 PyErr_Format(PyExc_TypeError,
1394 "cannot convert dictionary update "
1395 "sequence element #%zd to a sequence",
1396 i);
1397 goto Fail;
1398 }
1399 n = PySequence_Fast_GET_SIZE(fast);
1400 if (n != 2) {
1401 PyErr_Format(PyExc_ValueError,
1402 "dictionary update sequence element #%zd "
1403 "has length %zd; 2 is required",
1404 i, n);
1405 goto Fail;
1406 }
Tim Peters1fc240e2001-10-26 05:06:50 +00001407
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001408 /* Update/merge with this (key, value) pair. */
1409 key = PySequence_Fast_GET_ITEM(fast, 0);
1410 value = PySequence_Fast_GET_ITEM(fast, 1);
1411 if (override || PyDict_GetItem(d, key) == NULL) {
1412 int status = PyDict_SetItem(d, key, value);
1413 if (status < 0)
1414 goto Fail;
1415 }
1416 Py_DECREF(fast);
1417 Py_DECREF(item);
1418 }
Tim Peters1fc240e2001-10-26 05:06:50 +00001419
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001420 i = 0;
1421 goto Return;
Tim Peters1fc240e2001-10-26 05:06:50 +00001422Fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001423 Py_XDECREF(item);
1424 Py_XDECREF(fast);
1425 i = -1;
Tim Peters1fc240e2001-10-26 05:06:50 +00001426Return:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001427 Py_DECREF(it);
1428 return Py_SAFE_DOWNCAST(i, Py_ssize_t, int);
Tim Peters1fc240e2001-10-26 05:06:50 +00001429}
1430
Tim Peters6d6c1a32001-08-02 04:15:00 +00001431int
1432PyDict_Update(PyObject *a, PyObject *b)
1433{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001434 return PyDict_Merge(a, b, 1);
Guido van Rossum05ac6de2001-08-10 20:28:28 +00001435}
1436
1437int
1438PyDict_Merge(PyObject *a, PyObject *b, int override)
1439{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001440 register PyDictObject *mp, *other;
1441 register Py_ssize_t i;
1442 PyDictEntry *entry;
Tim Peters6d6c1a32001-08-02 04:15:00 +00001443
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001444 /* We accept for the argument either a concrete dictionary object,
1445 * or an abstract "mapping" object. For the former, we can do
1446 * things quite efficiently. For the latter, we only require that
1447 * PyMapping_Keys() and PyObject_GetItem() be supported.
1448 */
1449 if (a == NULL || !PyDict_Check(a) || b == NULL) {
1450 PyErr_BadInternalCall();
1451 return -1;
1452 }
1453 mp = (PyDictObject*)a;
1454 if (PyDict_Check(b)) {
1455 other = (PyDictObject*)b;
1456 if (other == mp || other->ma_used == 0)
1457 /* a.update(a) or a.update({}); nothing to do */
1458 return 0;
1459 if (mp->ma_used == 0)
1460 /* Since the target dict is empty, PyDict_GetItem()
1461 * always returns NULL. Setting override to 1
1462 * skips the unnecessary test.
1463 */
1464 override = 1;
1465 /* Do one big resize at the start, rather than
1466 * incrementally resizing as we insert new items. Expect
1467 * that there will be no (or few) overlapping keys.
1468 */
1469 if ((mp->ma_fill + other->ma_used)*3 >= (mp->ma_mask+1)*2) {
1470 if (dictresize(mp, (mp->ma_used + other->ma_used)*2) != 0)
1471 return -1;
1472 }
1473 for (i = 0; i <= other->ma_mask; i++) {
1474 entry = &other->ma_table[i];
1475 if (entry->me_value != NULL &&
1476 (override ||
1477 PyDict_GetItem(a, entry->me_key) == NULL)) {
1478 Py_INCREF(entry->me_key);
1479 Py_INCREF(entry->me_value);
1480 if (insertdict(mp, entry->me_key,
1481 (long)entry->me_hash,
1482 entry->me_value) != 0)
1483 return -1;
1484 }
1485 }
1486 }
1487 else {
1488 /* Do it the generic, slower way */
1489 PyObject *keys = PyMapping_Keys(b);
1490 PyObject *iter;
1491 PyObject *key, *value;
1492 int status;
Barry Warsaw66a0d1d2001-06-26 20:08:32 +00001493
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001494 if (keys == NULL)
1495 /* Docstring says this is equivalent to E.keys() so
1496 * if E doesn't have a .keys() method we want
1497 * AttributeError to percolate up. Might as well
1498 * do the same for any other error.
1499 */
1500 return -1;
Barry Warsaw66a0d1d2001-06-26 20:08:32 +00001501
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001502 iter = PyObject_GetIter(keys);
1503 Py_DECREF(keys);
1504 if (iter == NULL)
1505 return -1;
Barry Warsaw66a0d1d2001-06-26 20:08:32 +00001506
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001507 for (key = PyIter_Next(iter); key; key = PyIter_Next(iter)) {
1508 if (!override && PyDict_GetItem(a, key) != NULL) {
1509 Py_DECREF(key);
1510 continue;
1511 }
1512 value = PyObject_GetItem(b, key);
1513 if (value == NULL) {
1514 Py_DECREF(iter);
1515 Py_DECREF(key);
1516 return -1;
1517 }
1518 status = PyDict_SetItem(a, key, value);
1519 Py_DECREF(key);
1520 Py_DECREF(value);
1521 if (status < 0) {
1522 Py_DECREF(iter);
1523 return -1;
1524 }
1525 }
1526 Py_DECREF(iter);
1527 if (PyErr_Occurred())
1528 /* Iterator completed, via error */
1529 return -1;
1530 }
1531 return 0;
Guido van Rossume3f5b9c1997-05-28 19:15:28 +00001532}
1533
1534static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001535dict_copy(register PyDictObject *mp)
Guido van Rossume3f5b9c1997-05-28 19:15:28 +00001536{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001537 return PyDict_Copy((PyObject*)mp);
Jeremy Hyltona12c7a72000-03-30 22:27:31 +00001538}
1539
1540PyObject *
Tim Peters1f5871e2000-07-04 17:44:48 +00001541PyDict_Copy(PyObject *o)
Jeremy Hyltona12c7a72000-03-30 22:27:31 +00001542{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001543 PyObject *copy;
Jeremy Hyltona12c7a72000-03-30 22:27:31 +00001544
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001545 if (o == NULL || !PyDict_Check(o)) {
1546 PyErr_BadInternalCall();
1547 return NULL;
1548 }
1549 copy = PyDict_New();
1550 if (copy == NULL)
1551 return NULL;
1552 if (PyDict_Merge(copy, o, 1) == 0)
1553 return copy;
1554 Py_DECREF(copy);
1555 return NULL;
Guido van Rossume3f5b9c1997-05-28 19:15:28 +00001556}
1557
Martin v. Löwis18e16552006-02-15 17:27:45 +00001558Py_ssize_t
Tim Peters1f5871e2000-07-04 17:44:48 +00001559PyDict_Size(PyObject *mp)
Guido van Rossum4199fac1993-11-05 10:18:44 +00001560{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001561 if (mp == NULL || !PyDict_Check(mp)) {
1562 PyErr_BadInternalCall();
1563 return -1;
1564 }
1565 return ((PyDictObject *)mp)->ma_used;
Guido van Rossum4199fac1993-11-05 10:18:44 +00001566}
1567
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001568PyObject *
Tim Peters1f5871e2000-07-04 17:44:48 +00001569PyDict_Keys(PyObject *mp)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001570{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001571 if (mp == NULL || !PyDict_Check(mp)) {
1572 PyErr_BadInternalCall();
1573 return NULL;
1574 }
1575 return dict_keys((PyDictObject *)mp);
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001576}
1577
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001578PyObject *
Tim Peters1f5871e2000-07-04 17:44:48 +00001579PyDict_Values(PyObject *mp)
Guido van Rossum25831651993-05-19 14:50:45 +00001580{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001581 if (mp == NULL || !PyDict_Check(mp)) {
1582 PyErr_BadInternalCall();
1583 return NULL;
1584 }
1585 return dict_values((PyDictObject *)mp);
Guido van Rossum25831651993-05-19 14:50:45 +00001586}
1587
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001588PyObject *
Tim Peters1f5871e2000-07-04 17:44:48 +00001589PyDict_Items(PyObject *mp)
Guido van Rossum25831651993-05-19 14:50:45 +00001590{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001591 if (mp == NULL || !PyDict_Check(mp)) {
1592 PyErr_BadInternalCall();
1593 return NULL;
1594 }
1595 return dict_items((PyDictObject *)mp);
Guido van Rossum25831651993-05-19 14:50:45 +00001596}
1597
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001598/* Subroutine which returns the smallest key in a for which b's value
1599 is different or absent. The value is returned too, through the
Tim Peters95bf9392001-05-10 08:32:44 +00001600 pval argument. Both are NULL if no key in a is found for which b's status
1601 differs. The refcounts on (and only on) non-NULL *pval and function return
1602 values must be decremented by the caller (characterize() increments them
1603 to ensure that mutating comparison and PyDict_GetItem calls can't delete
1604 them before the caller is done looking at them). */
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001605
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001606static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001607characterize(PyDictObject *a, PyDictObject *b, PyObject **pval)
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001608{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001609 PyObject *akey = NULL; /* smallest key in a s.t. a[akey] != b[akey] */
1610 PyObject *aval = NULL; /* a[akey] */
1611 Py_ssize_t i;
1612 int cmp;
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001613
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001614 for (i = 0; i <= a->ma_mask; i++) {
1615 PyObject *thiskey, *thisaval, *thisbval;
1616 if (a->ma_table[i].me_value == NULL)
1617 continue;
1618 thiskey = a->ma_table[i].me_key;
1619 Py_INCREF(thiskey); /* keep alive across compares */
1620 if (akey != NULL) {
1621 cmp = PyObject_RichCompareBool(akey, thiskey, Py_LT);
1622 if (cmp < 0) {
1623 Py_DECREF(thiskey);
1624 goto Fail;
1625 }
1626 if (cmp > 0 ||
1627 i > a->ma_mask ||
1628 a->ma_table[i].me_value == NULL)
1629 {
1630 /* Not the *smallest* a key; or maybe it is
1631 * but the compare shrunk the dict so we can't
1632 * find its associated value anymore; or
1633 * maybe it is but the compare deleted the
1634 * a[thiskey] entry.
1635 */
1636 Py_DECREF(thiskey);
1637 continue;
1638 }
1639 }
Tim Peters95bf9392001-05-10 08:32:44 +00001640
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001641 /* Compare a[thiskey] to b[thiskey]; cmp <- true iff equal. */
1642 thisaval = a->ma_table[i].me_value;
1643 assert(thisaval);
1644 Py_INCREF(thisaval); /* keep alive */
1645 thisbval = PyDict_GetItem((PyObject *)b, thiskey);
1646 if (thisbval == NULL)
1647 cmp = 0;
1648 else {
1649 /* both dicts have thiskey: same values? */
1650 cmp = PyObject_RichCompareBool(
1651 thisaval, thisbval, Py_EQ);
1652 if (cmp < 0) {
1653 Py_DECREF(thiskey);
1654 Py_DECREF(thisaval);
1655 goto Fail;
1656 }
1657 }
1658 if (cmp == 0) {
1659 /* New winner. */
1660 Py_XDECREF(akey);
1661 Py_XDECREF(aval);
1662 akey = thiskey;
1663 aval = thisaval;
1664 }
1665 else {
1666 Py_DECREF(thiskey);
1667 Py_DECREF(thisaval);
1668 }
1669 }
1670 *pval = aval;
1671 return akey;
Tim Peters95bf9392001-05-10 08:32:44 +00001672
1673Fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001674 Py_XDECREF(akey);
1675 Py_XDECREF(aval);
1676 *pval = NULL;
1677 return NULL;
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001678}
1679
1680static int
Brett Cannon77ae87c2007-10-10 00:07:50 +00001681dict_compare(PyDictObject *a, PyDictObject *b)
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001682{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001683 PyObject *adiff, *bdiff, *aval, *bval;
1684 int res;
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001685
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001686 /* Compare lengths first */
1687 if (a->ma_used < b->ma_used)
1688 return -1; /* a is shorter */
1689 else if (a->ma_used > b->ma_used)
1690 return 1; /* b is shorter */
Tim Peters95bf9392001-05-10 08:32:44 +00001691
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001692 /* Same length -- check all keys */
1693 bdiff = bval = NULL;
1694 adiff = characterize(a, b, &aval);
1695 if (adiff == NULL) {
1696 assert(!aval);
1697 /* Either an error, or a is a subset with the same length so
1698 * must be equal.
1699 */
1700 res = PyErr_Occurred() ? -1 : 0;
1701 goto Finished;
1702 }
1703 bdiff = characterize(b, a, &bval);
1704 if (bdiff == NULL && PyErr_Occurred()) {
1705 assert(!bval);
1706 res = -1;
1707 goto Finished;
1708 }
1709 res = 0;
1710 if (bdiff) {
1711 /* bdiff == NULL "should be" impossible now, but perhaps
1712 * the last comparison done by the characterize() on a had
1713 * the side effect of making the dicts equal!
1714 */
1715 res = PyObject_Compare(adiff, bdiff);
1716 }
1717 if (res == 0 && bval != NULL)
1718 res = PyObject_Compare(aval, bval);
Tim Peters95bf9392001-05-10 08:32:44 +00001719
1720Finished:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001721 Py_XDECREF(adiff);
1722 Py_XDECREF(bdiff);
1723 Py_XDECREF(aval);
1724 Py_XDECREF(bval);
1725 return res;
Guido van Rossuma0a69b81996-12-05 21:55:55 +00001726}
1727
Tim Peterse63415e2001-05-08 04:38:29 +00001728/* Return 1 if dicts equal, 0 if not, -1 if error.
1729 * Gets out as soon as any difference is detected.
1730 * Uses only Py_EQ comparison.
1731 */
1732static int
Brett Cannon77ae87c2007-10-10 00:07:50 +00001733dict_equal(PyDictObject *a, PyDictObject *b)
Tim Peterse63415e2001-05-08 04:38:29 +00001734{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001735 Py_ssize_t i;
Tim Peterse63415e2001-05-08 04:38:29 +00001736
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001737 if (a->ma_used != b->ma_used)
1738 /* can't be equal if # of entries differ */
1739 return 0;
Tim Peterseb28ef22001-06-02 05:27:19 +00001740
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001741 /* Same # of entries -- check all of 'em. Exit early on any diff. */
1742 for (i = 0; i <= a->ma_mask; i++) {
1743 PyObject *aval = a->ma_table[i].me_value;
1744 if (aval != NULL) {
1745 int cmp;
1746 PyObject *bval;
1747 PyObject *key = a->ma_table[i].me_key;
1748 /* temporarily bump aval's refcount to ensure it stays
1749 alive until we're done with it */
1750 Py_INCREF(aval);
1751 /* ditto for key */
1752 Py_INCREF(key);
1753 bval = PyDict_GetItem((PyObject *)b, key);
1754 Py_DECREF(key);
1755 if (bval == NULL) {
1756 Py_DECREF(aval);
1757 return 0;
1758 }
1759 cmp = PyObject_RichCompareBool(aval, bval, Py_EQ);
1760 Py_DECREF(aval);
1761 if (cmp <= 0) /* error or not equal */
1762 return cmp;
1763 }
1764 }
1765 return 1;
Tim Peterse63415e2001-05-08 04:38:29 +00001766 }
1767
1768static PyObject *
1769dict_richcompare(PyObject *v, PyObject *w, int op)
1770{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001771 int cmp;
1772 PyObject *res;
Tim Peterse63415e2001-05-08 04:38:29 +00001773
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001774 if (!PyDict_Check(v) || !PyDict_Check(w)) {
1775 res = Py_NotImplemented;
1776 }
1777 else if (op == Py_EQ || op == Py_NE) {
1778 cmp = dict_equal((PyDictObject *)v, (PyDictObject *)w);
1779 if (cmp < 0)
1780 return NULL;
1781 res = (cmp == (op == Py_EQ)) ? Py_True : Py_False;
1782 }
1783 else {
1784 /* Py3K warning if comparison isn't == or != */
1785 if (PyErr_WarnPy3k("dict inequality comparisons not supported "
1786 "in 3.x", 1) < 0) {
1787 return NULL;
1788 }
1789 res = Py_NotImplemented;
1790 }
1791 Py_INCREF(res);
1792 return res;
Tim Peterse63415e2001-05-08 04:38:29 +00001793 }
1794
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001795static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001796dict_contains(register PyDictObject *mp, PyObject *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001797{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001798 long hash;
1799 PyDictEntry *ep;
Tim Petersd770ebd2006-06-01 15:50:44 +00001800
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001801 if (!PyString_CheckExact(key) ||
1802 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
1803 hash = PyObject_Hash(key);
1804 if (hash == -1)
1805 return NULL;
1806 }
1807 ep = (mp->ma_lookup)(mp, key, hash);
1808 if (ep == NULL)
1809 return NULL;
1810 return PyBool_FromLong(ep->me_value != NULL);
Guido van Rossum4b1302b1993-03-27 18:11:32 +00001811}
1812
Guido van Rossumc0b618a1997-05-02 03:12:38 +00001813static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001814dict_has_key(register PyDictObject *mp, PyObject *key)
Neal Norwitz8b2bfbc2007-05-23 06:35:32 +00001815{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001816 if (PyErr_WarnPy3k("dict.has_key() not supported in 3.x; "
1817 "use the in operator", 1) < 0)
1818 return NULL;
1819 return dict_contains(mp, key);
Neal Norwitz8b2bfbc2007-05-23 06:35:32 +00001820}
1821
1822static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001823dict_get(register PyDictObject *mp, PyObject *args)
Barry Warsawc38c5da1997-10-06 17:49:20 +00001824{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001825 PyObject *key;
1826 PyObject *failobj = Py_None;
1827 PyObject *val = NULL;
1828 long hash;
1829 PyDictEntry *ep;
Barry Warsawc38c5da1997-10-06 17:49:20 +00001830
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001831 if (!PyArg_UnpackTuple(args, "get", 1, 2, &key, &failobj))
1832 return NULL;
Barry Warsawc38c5da1997-10-06 17:49:20 +00001833
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001834 if (!PyString_CheckExact(key) ||
1835 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
1836 hash = PyObject_Hash(key);
1837 if (hash == -1)
1838 return NULL;
1839 }
1840 ep = (mp->ma_lookup)(mp, key, hash);
1841 if (ep == NULL)
1842 return NULL;
1843 val = ep->me_value;
1844 if (val == NULL)
1845 val = failobj;
1846 Py_INCREF(val);
1847 return val;
Barry Warsawc38c5da1997-10-06 17:49:20 +00001848}
1849
1850
1851static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001852dict_setdefault(register PyDictObject *mp, PyObject *args)
Guido van Rossum164452c2000-08-08 16:12:54 +00001853{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001854 PyObject *key;
1855 PyObject *failobj = Py_None;
1856 PyObject *val = NULL;
1857 long hash;
1858 PyDictEntry *ep;
Guido van Rossum164452c2000-08-08 16:12:54 +00001859
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001860 if (!PyArg_UnpackTuple(args, "setdefault", 1, 2, &key, &failobj))
1861 return NULL;
Guido van Rossum164452c2000-08-08 16:12:54 +00001862
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001863 if (!PyString_CheckExact(key) ||
1864 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
1865 hash = PyObject_Hash(key);
1866 if (hash == -1)
1867 return NULL;
1868 }
1869 ep = (mp->ma_lookup)(mp, key, hash);
1870 if (ep == NULL)
1871 return NULL;
1872 val = ep->me_value;
1873 if (val == NULL) {
1874 val = failobj;
1875 if (PyDict_SetItem((PyObject*)mp, key, failobj))
1876 val = NULL;
1877 }
1878 Py_XINCREF(val);
1879 return val;
Guido van Rossum164452c2000-08-08 16:12:54 +00001880}
1881
1882
1883static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001884dict_clear(register PyDictObject *mp)
Guido van Rossumfb8f1ca1997-03-21 21:55:12 +00001885{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001886 PyDict_Clear((PyObject *)mp);
1887 Py_RETURN_NONE;
Guido van Rossumfb8f1ca1997-03-21 21:55:12 +00001888}
1889
Guido van Rossumba6ab842000-12-12 22:02:18 +00001890static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001891dict_pop(PyDictObject *mp, PyObject *args)
Guido van Rossume027d982002-04-12 15:11:59 +00001892{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001893 long hash;
1894 PyDictEntry *ep;
1895 PyObject *old_value, *old_key;
1896 PyObject *key, *deflt = NULL;
Guido van Rossume027d982002-04-12 15:11:59 +00001897
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001898 if(!PyArg_UnpackTuple(args, "pop", 1, 2, &key, &deflt))
1899 return NULL;
1900 if (mp->ma_used == 0) {
1901 if (deflt) {
1902 Py_INCREF(deflt);
1903 return deflt;
1904 }
1905 PyErr_SetString(PyExc_KeyError,
1906 "pop(): dictionary is empty");
1907 return NULL;
1908 }
1909 if (!PyString_CheckExact(key) ||
1910 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
1911 hash = PyObject_Hash(key);
1912 if (hash == -1)
1913 return NULL;
1914 }
1915 ep = (mp->ma_lookup)(mp, key, hash);
1916 if (ep == NULL)
1917 return NULL;
1918 if (ep->me_value == NULL) {
1919 if (deflt) {
1920 Py_INCREF(deflt);
1921 return deflt;
1922 }
1923 set_key_error(key);
1924 return NULL;
1925 }
1926 old_key = ep->me_key;
1927 Py_INCREF(dummy);
1928 ep->me_key = dummy;
1929 old_value = ep->me_value;
1930 ep->me_value = NULL;
1931 mp->ma_used--;
1932 Py_DECREF(old_key);
1933 return old_value;
Guido van Rossume027d982002-04-12 15:11:59 +00001934}
1935
1936static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00001937dict_popitem(PyDictObject *mp)
Guido van Rossumba6ab842000-12-12 22:02:18 +00001938{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001939 Py_ssize_t i = 0;
1940 PyDictEntry *ep;
1941 PyObject *res;
Guido van Rossumba6ab842000-12-12 22:02:18 +00001942
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001943 /* Allocate the result tuple before checking the size. Believe it
1944 * or not, this allocation could trigger a garbage collection which
1945 * could empty the dict, so if we checked the size first and that
1946 * happened, the result would be an infinite loop (searching for an
1947 * entry that no longer exists). Note that the usual popitem()
1948 * idiom is "while d: k, v = d.popitem()". so needing to throw the
1949 * tuple away if the dict *is* empty isn't a significant
1950 * inefficiency -- possible, but unlikely in practice.
1951 */
1952 res = PyTuple_New(2);
1953 if (res == NULL)
1954 return NULL;
1955 if (mp->ma_used == 0) {
1956 Py_DECREF(res);
1957 PyErr_SetString(PyExc_KeyError,
1958 "popitem(): dictionary is empty");
1959 return NULL;
1960 }
1961 /* Set ep to "the first" dict entry with a value. We abuse the hash
1962 * field of slot 0 to hold a search finger:
1963 * If slot 0 has a value, use slot 0.
1964 * Else slot 0 is being used to hold a search finger,
1965 * and we use its hash value as the first index to look.
1966 */
1967 ep = &mp->ma_table[0];
1968 if (ep->me_value == NULL) {
1969 i = ep->me_hash;
1970 /* The hash field may be a real hash value, or it may be a
1971 * legit search finger, or it may be a once-legit search
1972 * finger that's out of bounds now because it wrapped around
1973 * or the table shrunk -- simply make sure it's in bounds now.
1974 */
1975 if (i > mp->ma_mask || i < 1)
1976 i = 1; /* skip slot 0 */
1977 while ((ep = &mp->ma_table[i])->me_value == NULL) {
1978 i++;
1979 if (i > mp->ma_mask)
1980 i = 1;
1981 }
1982 }
1983 PyTuple_SET_ITEM(res, 0, ep->me_key);
1984 PyTuple_SET_ITEM(res, 1, ep->me_value);
1985 Py_INCREF(dummy);
1986 ep->me_key = dummy;
1987 ep->me_value = NULL;
1988 mp->ma_used--;
1989 assert(mp->ma_table[0].me_value == NULL);
1990 mp->ma_table[0].me_hash = i + 1; /* next place to start */
1991 return res;
Guido van Rossumba6ab842000-12-12 22:02:18 +00001992}
1993
Jeremy Hylton8caad492000-06-23 14:18:11 +00001994static int
1995dict_traverse(PyObject *op, visitproc visit, void *arg)
1996{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00001997 Py_ssize_t i = 0;
1998 PyObject *pk;
1999 PyObject *pv;
Jeremy Hylton8caad492000-06-23 14:18:11 +00002000
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002001 while (PyDict_Next(op, &i, &pk, &pv)) {
2002 Py_VISIT(pk);
2003 Py_VISIT(pv);
2004 }
2005 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00002006}
2007
2008static int
2009dict_tp_clear(PyObject *op)
2010{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002011 PyDict_Clear(op);
2012 return 0;
Jeremy Hylton8caad492000-06-23 14:18:11 +00002013}
2014
Tim Petersf7f88b12000-12-13 23:18:45 +00002015
Raymond Hettinger019a1482004-03-18 02:41:19 +00002016extern PyTypeObject PyDictIterKey_Type; /* Forward */
2017extern PyTypeObject PyDictIterValue_Type; /* Forward */
2018extern PyTypeObject PyDictIterItem_Type; /* Forward */
Brett Cannon77ae87c2007-10-10 00:07:50 +00002019static PyObject *dictiter_new(PyDictObject *, PyTypeObject *);
Guido van Rossum09e563a2001-05-01 12:10:21 +00002020
2021static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00002022dict_iterkeys(PyDictObject *dict)
Guido van Rossum09e563a2001-05-01 12:10:21 +00002023{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002024 return dictiter_new(dict, &PyDictIterKey_Type);
Guido van Rossum09e563a2001-05-01 12:10:21 +00002025}
2026
2027static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00002028dict_itervalues(PyDictObject *dict)
Guido van Rossum09e563a2001-05-01 12:10:21 +00002029{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002030 return dictiter_new(dict, &PyDictIterValue_Type);
Guido van Rossum09e563a2001-05-01 12:10:21 +00002031}
2032
2033static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00002034dict_iteritems(PyDictObject *dict)
Guido van Rossum09e563a2001-05-01 12:10:21 +00002035{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002036 return dictiter_new(dict, &PyDictIterItem_Type);
Guido van Rossum09e563a2001-05-01 12:10:21 +00002037}
2038
Robert Schuppenies51df0642008-06-01 16:16:17 +00002039static PyObject *
2040dict_sizeof(PyDictObject *mp)
2041{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002042 Py_ssize_t res;
Robert Schuppenies51df0642008-06-01 16:16:17 +00002043
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002044 res = sizeof(PyDictObject);
2045 if (mp->ma_table != mp->ma_smalltable)
2046 res = res + (mp->ma_mask + 1) * sizeof(PyDictEntry);
2047 return PyInt_FromSsize_t(res);
Robert Schuppenies51df0642008-06-01 16:16:17 +00002048}
Guido van Rossum09e563a2001-05-01 12:10:21 +00002049
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002050PyDoc_STRVAR(has_key__doc__,
Raymond Hettinger574aa322003-09-01 22:12:08 +00002051"D.has_key(k) -> True if D has a key k, else False");
Tim Petersf7f88b12000-12-13 23:18:45 +00002052
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00002053PyDoc_STRVAR(contains__doc__,
2054"D.__contains__(k) -> True if D has a key k, else False");
2055
2056PyDoc_STRVAR(getitem__doc__, "x.__getitem__(y) <==> x[y]");
2057
Robert Schuppenies51df0642008-06-01 16:16:17 +00002058PyDoc_STRVAR(sizeof__doc__,
Georg Brandl7a6de8b2008-06-01 16:42:16 +00002059"D.__sizeof__() -> size of D in memory, in bytes");
Robert Schuppenies51df0642008-06-01 16:16:17 +00002060
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002061PyDoc_STRVAR(get__doc__,
Guido van Rossumefae8862002-09-04 11:29:45 +00002062"D.get(k[,d]) -> D[k] if k in D, else d. d defaults to None.");
Tim Petersf7f88b12000-12-13 23:18:45 +00002063
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002064PyDoc_STRVAR(setdefault_doc__,
Guido van Rossumefae8862002-09-04 11:29:45 +00002065"D.setdefault(k[,d]) -> D.get(k,d), also set D[k]=d if k not in D");
Tim Petersf7f88b12000-12-13 23:18:45 +00002066
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002067PyDoc_STRVAR(pop__doc__,
Georg Brandl4aef7032008-11-07 08:56:27 +00002068"D.pop(k[,d]) -> v, remove specified key and return the corresponding value.\n\
Raymond Hettingera3e1e4c2003-03-06 23:54:28 +00002069If key is not found, d is returned if given, otherwise KeyError is raised");
Guido van Rossume027d982002-04-12 15:11:59 +00002070
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002071PyDoc_STRVAR(popitem__doc__,
Tim Petersf7f88b12000-12-13 23:18:45 +00002072"D.popitem() -> (k, v), remove and return some (key, value) pair as a\n\
Georg Brandl4aef7032008-11-07 08:56:27 +000020732-tuple; but raise KeyError if D is empty.");
Tim Petersf7f88b12000-12-13 23:18:45 +00002074
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002075PyDoc_STRVAR(keys__doc__,
2076"D.keys() -> list of D's keys");
Tim Petersf7f88b12000-12-13 23:18:45 +00002077
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002078PyDoc_STRVAR(items__doc__,
2079"D.items() -> list of D's (key, value) pairs, as 2-tuples");
Tim Petersf7f88b12000-12-13 23:18:45 +00002080
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002081PyDoc_STRVAR(values__doc__,
2082"D.values() -> list of D's values");
Tim Petersf7f88b12000-12-13 23:18:45 +00002083
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002084PyDoc_STRVAR(update__doc__,
Georg Brandl4aef7032008-11-07 08:56:27 +00002085"D.update(E, **F) -> None. Update D from dict/iterable E and F.\n"
2086"If E has a .keys() method, does: for k in E: D[k] = E[k]\n\
2087If E lacks .keys() method, does: for (k, v) in E: D[k] = v\n\
2088In either case, this is followed by: for k in F: D[k] = F[k]");
Tim Petersf7f88b12000-12-13 23:18:45 +00002089
Raymond Hettingere33d3df2002-11-27 07:29:33 +00002090PyDoc_STRVAR(fromkeys__doc__,
2091"dict.fromkeys(S[,v]) -> New dict with keys from S and values equal to v.\n\
2092v defaults to None.");
2093
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002094PyDoc_STRVAR(clear__doc__,
2095"D.clear() -> None. Remove all items from D.");
Tim Petersf7f88b12000-12-13 23:18:45 +00002096
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002097PyDoc_STRVAR(copy__doc__,
2098"D.copy() -> a shallow copy of D");
Tim Petersf7f88b12000-12-13 23:18:45 +00002099
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002100PyDoc_STRVAR(iterkeys__doc__,
2101"D.iterkeys() -> an iterator over the keys of D");
Guido van Rossum09e563a2001-05-01 12:10:21 +00002102
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002103PyDoc_STRVAR(itervalues__doc__,
2104"D.itervalues() -> an iterator over the values of D");
Guido van Rossum09e563a2001-05-01 12:10:21 +00002105
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002106PyDoc_STRVAR(iteritems__doc__,
2107"D.iteritems() -> an iterator over the (key, value) items of D");
Guido van Rossum09e563a2001-05-01 12:10:21 +00002108
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002109static PyMethodDef mapp_methods[] = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002110 {"__contains__",(PyCFunction)dict_contains, METH_O | METH_COEXIST,
2111 contains__doc__},
2112 {"__getitem__", (PyCFunction)dict_subscript, METH_O | METH_COEXIST,
2113 getitem__doc__},
2114 {"__sizeof__", (PyCFunction)dict_sizeof, METH_NOARGS,
2115 sizeof__doc__},
2116 {"has_key", (PyCFunction)dict_has_key, METH_O,
2117 has_key__doc__},
2118 {"get", (PyCFunction)dict_get, METH_VARARGS,
2119 get__doc__},
2120 {"setdefault", (PyCFunction)dict_setdefault, METH_VARARGS,
2121 setdefault_doc__},
2122 {"pop", (PyCFunction)dict_pop, METH_VARARGS,
2123 pop__doc__},
2124 {"popitem", (PyCFunction)dict_popitem, METH_NOARGS,
2125 popitem__doc__},
2126 {"keys", (PyCFunction)dict_keys, METH_NOARGS,
2127 keys__doc__},
2128 {"items", (PyCFunction)dict_items, METH_NOARGS,
2129 items__doc__},
2130 {"values", (PyCFunction)dict_values, METH_NOARGS,
2131 values__doc__},
2132 {"update", (PyCFunction)dict_update, METH_VARARGS | METH_KEYWORDS,
2133 update__doc__},
2134 {"fromkeys", (PyCFunction)dict_fromkeys, METH_VARARGS | METH_CLASS,
2135 fromkeys__doc__},
2136 {"clear", (PyCFunction)dict_clear, METH_NOARGS,
2137 clear__doc__},
2138 {"copy", (PyCFunction)dict_copy, METH_NOARGS,
2139 copy__doc__},
2140 {"iterkeys", (PyCFunction)dict_iterkeys, METH_NOARGS,
2141 iterkeys__doc__},
2142 {"itervalues", (PyCFunction)dict_itervalues, METH_NOARGS,
2143 itervalues__doc__},
2144 {"iteritems", (PyCFunction)dict_iteritems, METH_NOARGS,
2145 iteritems__doc__},
2146 {NULL, NULL} /* sentinel */
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002147};
2148
Tim Petersd770ebd2006-06-01 15:50:44 +00002149/* Return 1 if `key` is in dict `op`, 0 if not, and -1 on error. */
Raymond Hettingerbc0f2ab2003-11-25 21:12:14 +00002150int
2151PyDict_Contains(PyObject *op, PyObject *key)
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +00002152{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002153 long hash;
2154 PyDictObject *mp = (PyDictObject *)op;
2155 PyDictEntry *ep;
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +00002156
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002157 if (!PyString_CheckExact(key) ||
2158 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
2159 hash = PyObject_Hash(key);
2160 if (hash == -1)
2161 return -1;
2162 }
2163 ep = (mp->ma_lookup)(mp, key, hash);
2164 return ep == NULL ? -1 : (ep->me_value != NULL);
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +00002165}
2166
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +00002167/* Internal version of PyDict_Contains used when the hash value is already known */
2168int
2169_PyDict_Contains(PyObject *op, PyObject *key, long hash)
2170{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002171 PyDictObject *mp = (PyDictObject *)op;
2172 PyDictEntry *ep;
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +00002173
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002174 ep = (mp->ma_lookup)(mp, key, hash);
2175 return ep == NULL ? -1 : (ep->me_value != NULL);
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +00002176}
2177
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +00002178/* Hack to implement "key in dict" */
2179static PySequenceMethods dict_as_sequence = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002180 0, /* sq_length */
2181 0, /* sq_concat */
2182 0, /* sq_repeat */
2183 0, /* sq_item */
2184 0, /* sq_slice */
2185 0, /* sq_ass_item */
2186 0, /* sq_ass_slice */
2187 PyDict_Contains, /* sq_contains */
2188 0, /* sq_inplace_concat */
2189 0, /* sq_inplace_repeat */
Guido van Rossum0dbb4fb2001-04-20 16:50:40 +00002190};
2191
Guido van Rossum09e563a2001-05-01 12:10:21 +00002192static PyObject *
Tim Peters6d6c1a32001-08-02 04:15:00 +00002193dict_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
2194{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002195 PyObject *self;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002196
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002197 assert(type != NULL && type->tp_alloc != NULL);
2198 self = type->tp_alloc(type, 0);
2199 if (self != NULL) {
2200 PyDictObject *d = (PyDictObject *)self;
2201 /* It's guaranteed that tp->alloc zeroed out the struct. */
2202 assert(d->ma_table == NULL && d->ma_fill == 0 && d->ma_used == 0);
2203 INIT_NONZERO_DICT_SLOTS(d);
2204 d->ma_lookup = lookdict_string;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002205#ifdef SHOW_CONVERSION_COUNTS
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002206 ++created;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002207#endif
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002208 }
2209 return self;
Tim Peters6d6c1a32001-08-02 04:15:00 +00002210}
2211
Tim Peters25786c02001-09-02 08:22:48 +00002212static int
2213dict_init(PyObject *self, PyObject *args, PyObject *kwds)
2214{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002215 return dict_update_common(self, args, kwds, "dict");
Tim Peters25786c02001-09-02 08:22:48 +00002216}
2217
Tim Peters6d6c1a32001-08-02 04:15:00 +00002218static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00002219dict_iter(PyDictObject *dict)
Guido van Rossum09e563a2001-05-01 12:10:21 +00002220{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002221 return dictiter_new(dict, &PyDictIterKey_Type);
Guido van Rossum09e563a2001-05-01 12:10:21 +00002222}
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002223
Martin v. Löwis14f8b4c2002-06-13 20:33:02 +00002224PyDoc_STRVAR(dictionary_doc,
Ezio Melotti56f16822010-03-01 04:05:56 +00002225"dict() -> new empty dictionary\n"
Tim Petersa427a2b2001-10-29 22:25:45 +00002226"dict(mapping) -> new dictionary initialized from a mapping object's\n"
Ezio Melotti56f16822010-03-01 04:05:56 +00002227" (key, value) pairs\n"
2228"dict(iterable) -> new dictionary initialized as if via:\n"
Tim Peters4d859532001-10-27 18:27:48 +00002229" d = {}\n"
Ezio Melotti56f16822010-03-01 04:05:56 +00002230" for k, v in iterable:\n"
Just van Rossuma797d812002-11-23 09:45:04 +00002231" d[k] = v\n"
2232"dict(**kwargs) -> new dictionary initialized with the name=value pairs\n"
2233" in the keyword argument list. For example: dict(one=1, two=2)");
Tim Peters25786c02001-09-02 08:22:48 +00002234
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002235PyTypeObject PyDict_Type = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002236 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2237 "dict",
2238 sizeof(PyDictObject),
2239 0,
2240 (destructor)dict_dealloc, /* tp_dealloc */
2241 (printfunc)dict_print, /* tp_print */
2242 0, /* tp_getattr */
2243 0, /* tp_setattr */
2244 (cmpfunc)dict_compare, /* tp_compare */
2245 (reprfunc)dict_repr, /* tp_repr */
2246 0, /* tp_as_number */
2247 &dict_as_sequence, /* tp_as_sequence */
2248 &dict_as_mapping, /* tp_as_mapping */
2249 (hashfunc)PyObject_HashNotImplemented, /* tp_hash */
2250 0, /* tp_call */
2251 0, /* tp_str */
2252 PyObject_GenericGetAttr, /* tp_getattro */
2253 0, /* tp_setattro */
2254 0, /* tp_as_buffer */
2255 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
2256 Py_TPFLAGS_BASETYPE | Py_TPFLAGS_DICT_SUBCLASS, /* tp_flags */
2257 dictionary_doc, /* tp_doc */
2258 dict_traverse, /* tp_traverse */
2259 dict_tp_clear, /* tp_clear */
2260 dict_richcompare, /* tp_richcompare */
2261 0, /* tp_weaklistoffset */
2262 (getiterfunc)dict_iter, /* tp_iter */
2263 0, /* tp_iternext */
2264 mapp_methods, /* tp_methods */
2265 0, /* tp_members */
2266 0, /* tp_getset */
2267 0, /* tp_base */
2268 0, /* tp_dict */
2269 0, /* tp_descr_get */
2270 0, /* tp_descr_set */
2271 0, /* tp_dictoffset */
2272 dict_init, /* tp_init */
2273 PyType_GenericAlloc, /* tp_alloc */
2274 dict_new, /* tp_new */
2275 PyObject_GC_Del, /* tp_free */
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002276};
2277
Guido van Rossum3cca2451997-05-16 14:23:33 +00002278/* For backward compatibility with old dictionary interface */
2279
Guido van Rossumc0b618a1997-05-02 03:12:38 +00002280PyObject *
Martin v. Löwis32b4a1b2002-12-11 13:21:12 +00002281PyDict_GetItemString(PyObject *v, const char *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002282{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002283 PyObject *kv, *rv;
2284 kv = PyString_FromString(key);
2285 if (kv == NULL)
2286 return NULL;
2287 rv = PyDict_GetItem(v, kv);
2288 Py_DECREF(kv);
2289 return rv;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002290}
2291
2292int
Martin v. Löwis32b4a1b2002-12-11 13:21:12 +00002293PyDict_SetItemString(PyObject *v, const char *key, PyObject *item)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002294{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002295 PyObject *kv;
2296 int err;
2297 kv = PyString_FromString(key);
2298 if (kv == NULL)
2299 return -1;
2300 PyString_InternInPlace(&kv); /* XXX Should we really? */
2301 err = PyDict_SetItem(v, kv, item);
2302 Py_DECREF(kv);
2303 return err;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002304}
2305
2306int
Martin v. Löwis32b4a1b2002-12-11 13:21:12 +00002307PyDict_DelItemString(PyObject *v, const char *key)
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002308{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002309 PyObject *kv;
2310 int err;
2311 kv = PyString_FromString(key);
2312 if (kv == NULL)
2313 return -1;
2314 err = PyDict_DelItem(v, kv);
2315 Py_DECREF(kv);
2316 return err;
Guido van Rossum4b1302b1993-03-27 18:11:32 +00002317}
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002318
Raymond Hettinger019a1482004-03-18 02:41:19 +00002319/* Dictionary iterator types */
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002320
2321typedef struct {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002322 PyObject_HEAD
2323 PyDictObject *di_dict; /* Set to NULL when iterator is exhausted */
2324 Py_ssize_t di_used;
2325 Py_ssize_t di_pos;
2326 PyObject* di_result; /* reusable result tuple for iteritems */
2327 Py_ssize_t len;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002328} dictiterobject;
2329
2330static PyObject *
Brett Cannon77ae87c2007-10-10 00:07:50 +00002331dictiter_new(PyDictObject *dict, PyTypeObject *itertype)
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002332{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002333 dictiterobject *di;
2334 di = PyObject_GC_New(dictiterobject, itertype);
2335 if (di == NULL)
2336 return NULL;
2337 Py_INCREF(dict);
2338 di->di_dict = dict;
2339 di->di_used = dict->ma_used;
2340 di->di_pos = 0;
2341 di->len = dict->ma_used;
2342 if (itertype == &PyDictIterItem_Type) {
2343 di->di_result = PyTuple_Pack(2, Py_None, Py_None);
2344 if (di->di_result == NULL) {
2345 Py_DECREF(di);
2346 return NULL;
2347 }
2348 }
2349 else
2350 di->di_result = NULL;
2351 _PyObject_GC_TRACK(di);
2352 return (PyObject *)di;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002353}
2354
2355static void
2356dictiter_dealloc(dictiterobject *di)
2357{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002358 Py_XDECREF(di->di_dict);
2359 Py_XDECREF(di->di_result);
2360 PyObject_GC_Del(di);
Georg Brandl47fe9812009-01-01 15:46:10 +00002361}
2362
2363static int
2364dictiter_traverse(dictiterobject *di, visitproc visit, void *arg)
2365{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002366 Py_VISIT(di->di_dict);
2367 Py_VISIT(di->di_result);
2368 return 0;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002369}
2370
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00002371static PyObject *
Raymond Hettinger0ce6dc82004-03-18 08:38:00 +00002372dictiter_len(dictiterobject *di)
2373{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002374 Py_ssize_t len = 0;
2375 if (di->di_dict != NULL && di->di_used == di->di_dict->ma_used)
2376 len = di->len;
2377 return PyInt_FromSize_t(len);
Raymond Hettinger0ce6dc82004-03-18 08:38:00 +00002378}
2379
Armin Rigof5b3e362006-02-11 21:32:43 +00002380PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
Raymond Hettinger6b27cda2005-09-24 21:23:05 +00002381
2382static PyMethodDef dictiter_methods[] = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002383 {"__length_hint__", (PyCFunction)dictiter_len, METH_NOARGS, length_hint_doc},
2384 {NULL, NULL} /* sentinel */
Raymond Hettinger0ce6dc82004-03-18 08:38:00 +00002385};
2386
Raymond Hettinger019a1482004-03-18 02:41:19 +00002387static PyObject *dictiter_iternextkey(dictiterobject *di)
Guido van Rossum213c7a62001-04-23 14:08:49 +00002388{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002389 PyObject *key;
2390 register Py_ssize_t i, mask;
2391 register PyDictEntry *ep;
2392 PyDictObject *d = di->di_dict;
Guido van Rossum213c7a62001-04-23 14:08:49 +00002393
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002394 if (d == NULL)
2395 return NULL;
2396 assert (PyDict_Check(d));
Guido van Rossum2147df72002-07-16 20:30:22 +00002397
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002398 if (di->di_used != d->ma_used) {
2399 PyErr_SetString(PyExc_RuntimeError,
2400 "dictionary changed size during iteration");
2401 di->di_used = -1; /* Make this state sticky */
2402 return NULL;
2403 }
Guido van Rossum2147df72002-07-16 20:30:22 +00002404
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002405 i = di->di_pos;
2406 if (i < 0)
2407 goto fail;
2408 ep = d->ma_table;
2409 mask = d->ma_mask;
2410 while (i <= mask && ep[i].me_value == NULL)
2411 i++;
2412 di->di_pos = i+1;
2413 if (i > mask)
2414 goto fail;
2415 di->len--;
2416 key = ep[i].me_key;
2417 Py_INCREF(key);
2418 return key;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002419
2420fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002421 Py_DECREF(d);
2422 di->di_dict = NULL;
2423 return NULL;
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002424}
2425
Raymond Hettinger019a1482004-03-18 02:41:19 +00002426PyTypeObject PyDictIterKey_Type = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002427 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2428 "dictionary-keyiterator", /* tp_name */
2429 sizeof(dictiterobject), /* tp_basicsize */
2430 0, /* tp_itemsize */
2431 /* methods */
2432 (destructor)dictiter_dealloc, /* tp_dealloc */
2433 0, /* tp_print */
2434 0, /* tp_getattr */
2435 0, /* tp_setattr */
2436 0, /* tp_compare */
2437 0, /* tp_repr */
2438 0, /* tp_as_number */
2439 0, /* tp_as_sequence */
2440 0, /* tp_as_mapping */
2441 0, /* tp_hash */
2442 0, /* tp_call */
2443 0, /* tp_str */
2444 PyObject_GenericGetAttr, /* tp_getattro */
2445 0, /* tp_setattro */
2446 0, /* tp_as_buffer */
2447 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2448 0, /* tp_doc */
2449 (traverseproc)dictiter_traverse, /* tp_traverse */
2450 0, /* tp_clear */
2451 0, /* tp_richcompare */
2452 0, /* tp_weaklistoffset */
2453 PyObject_SelfIter, /* tp_iter */
2454 (iternextfunc)dictiter_iternextkey, /* tp_iternext */
2455 dictiter_methods, /* tp_methods */
2456 0,
Raymond Hettinger019a1482004-03-18 02:41:19 +00002457};
2458
2459static PyObject *dictiter_iternextvalue(dictiterobject *di)
2460{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002461 PyObject *value;
2462 register Py_ssize_t i, mask;
2463 register PyDictEntry *ep;
2464 PyDictObject *d = di->di_dict;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002465
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002466 if (d == NULL)
2467 return NULL;
2468 assert (PyDict_Check(d));
Raymond Hettinger019a1482004-03-18 02:41:19 +00002469
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002470 if (di->di_used != d->ma_used) {
2471 PyErr_SetString(PyExc_RuntimeError,
2472 "dictionary changed size during iteration");
2473 di->di_used = -1; /* Make this state sticky */
2474 return NULL;
2475 }
Raymond Hettinger019a1482004-03-18 02:41:19 +00002476
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002477 i = di->di_pos;
2478 mask = d->ma_mask;
2479 if (i < 0 || i > mask)
2480 goto fail;
2481 ep = d->ma_table;
2482 while ((value=ep[i].me_value) == NULL) {
2483 i++;
2484 if (i > mask)
2485 goto fail;
2486 }
2487 di->di_pos = i+1;
2488 di->len--;
2489 Py_INCREF(value);
2490 return value;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002491
2492fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002493 Py_DECREF(d);
2494 di->di_dict = NULL;
2495 return NULL;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002496}
2497
2498PyTypeObject PyDictIterValue_Type = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002499 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2500 "dictionary-valueiterator", /* tp_name */
2501 sizeof(dictiterobject), /* tp_basicsize */
2502 0, /* tp_itemsize */
2503 /* methods */
2504 (destructor)dictiter_dealloc, /* tp_dealloc */
2505 0, /* tp_print */
2506 0, /* tp_getattr */
2507 0, /* tp_setattr */
2508 0, /* tp_compare */
2509 0, /* tp_repr */
2510 0, /* tp_as_number */
2511 0, /* tp_as_sequence */
2512 0, /* tp_as_mapping */
2513 0, /* tp_hash */
2514 0, /* tp_call */
2515 0, /* tp_str */
2516 PyObject_GenericGetAttr, /* tp_getattro */
2517 0, /* tp_setattro */
2518 0, /* tp_as_buffer */
2519 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2520 0, /* tp_doc */
2521 (traverseproc)dictiter_traverse, /* tp_traverse */
2522 0, /* tp_clear */
2523 0, /* tp_richcompare */
2524 0, /* tp_weaklistoffset */
2525 PyObject_SelfIter, /* tp_iter */
2526 (iternextfunc)dictiter_iternextvalue, /* tp_iternext */
2527 dictiter_methods, /* tp_methods */
2528 0,
Raymond Hettinger019a1482004-03-18 02:41:19 +00002529};
2530
2531static PyObject *dictiter_iternextitem(dictiterobject *di)
2532{
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002533 PyObject *key, *value, *result = di->di_result;
2534 register Py_ssize_t i, mask;
2535 register PyDictEntry *ep;
2536 PyDictObject *d = di->di_dict;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002537
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002538 if (d == NULL)
2539 return NULL;
2540 assert (PyDict_Check(d));
Raymond Hettinger019a1482004-03-18 02:41:19 +00002541
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002542 if (di->di_used != d->ma_used) {
2543 PyErr_SetString(PyExc_RuntimeError,
2544 "dictionary changed size during iteration");
2545 di->di_used = -1; /* Make this state sticky */
2546 return NULL;
2547 }
Raymond Hettinger019a1482004-03-18 02:41:19 +00002548
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002549 i = di->di_pos;
2550 if (i < 0)
2551 goto fail;
2552 ep = d->ma_table;
2553 mask = d->ma_mask;
2554 while (i <= mask && ep[i].me_value == NULL)
2555 i++;
2556 di->di_pos = i+1;
2557 if (i > mask)
2558 goto fail;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002559
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002560 if (result->ob_refcnt == 1) {
2561 Py_INCREF(result);
2562 Py_DECREF(PyTuple_GET_ITEM(result, 0));
2563 Py_DECREF(PyTuple_GET_ITEM(result, 1));
2564 } else {
2565 result = PyTuple_New(2);
2566 if (result == NULL)
2567 return NULL;
2568 }
2569 di->len--;
2570 key = ep[i].me_key;
2571 value = ep[i].me_value;
2572 Py_INCREF(key);
2573 Py_INCREF(value);
2574 PyTuple_SET_ITEM(result, 0, key);
2575 PyTuple_SET_ITEM(result, 1, value);
2576 return result;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002577
2578fail:
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002579 Py_DECREF(d);
2580 di->di_dict = NULL;
2581 return NULL;
Raymond Hettinger019a1482004-03-18 02:41:19 +00002582}
2583
2584PyTypeObject PyDictIterItem_Type = {
Antoine Pitrouc7c96a92010-05-09 15:15:40 +00002585 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2586 "dictionary-itemiterator", /* tp_name */
2587 sizeof(dictiterobject), /* tp_basicsize */
2588 0, /* tp_itemsize */
2589 /* methods */
2590 (destructor)dictiter_dealloc, /* tp_dealloc */
2591 0, /* tp_print */
2592 0, /* tp_getattr */
2593 0, /* tp_setattr */
2594 0, /* tp_compare */
2595 0, /* tp_repr */
2596 0, /* tp_as_number */
2597 0, /* tp_as_sequence */
2598 0, /* tp_as_mapping */
2599 0, /* tp_hash */
2600 0, /* tp_call */
2601 0, /* tp_str */
2602 PyObject_GenericGetAttr, /* tp_getattro */
2603 0, /* tp_setattro */
2604 0, /* tp_as_buffer */
2605 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
2606 0, /* tp_doc */
2607 (traverseproc)dictiter_traverse, /* tp_traverse */
2608 0, /* tp_clear */
2609 0, /* tp_richcompare */
2610 0, /* tp_weaklistoffset */
2611 PyObject_SelfIter, /* tp_iter */
2612 (iternextfunc)dictiter_iternextitem, /* tp_iternext */
2613 dictiter_methods, /* tp_methods */
2614 0,
Guido van Rossum59d1d2b2001-04-20 19:13:02 +00002615};