mirror of
				https://github.com/python/cpython.git
				synced 2025-10-31 13:41:24 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			3883 lines
		
	
	
	
		
			116 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			3883 lines
		
	
	
	
		
			116 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| 
 | |
| /* Dictionary object implementation using a hash table */
 | |
| 
 | |
| /* The distribution includes a separate file, Objects/dictnotes.txt,
 | |
|    describing explorations into dictionary design and optimization.
 | |
|    It covers typical dictionary use patterns, the parameters for
 | |
|    tuning dictionaries, and several ideas for possible optimizations.
 | |
| */
 | |
| 
 | |
| 
 | |
| /*
 | |
| There are four kinds of slots in the table:
 | |
| 
 | |
| 1. Unused.  me_key == me_value == NULL
 | |
|    Does not hold an active (key, value) pair now and never did.  Unused can
 | |
|    transition to Active upon key insertion.  This is the only case in which
 | |
|    me_key is NULL, and is each slot's initial state.
 | |
| 
 | |
| 2. Active.  me_key != NULL and me_key != dummy and me_value != NULL
 | |
|    Holds an active (key, value) pair.  Active can transition to Dummy or
 | |
|    Pending upon key deletion (for combined and split tables respectively).
 | |
|    This is the only case in which me_value != NULL.
 | |
| 
 | |
| 3. Dummy.  me_key == dummy and me_value == NULL
 | |
|    Previously held an active (key, value) pair, but that was deleted and an
 | |
|    active pair has not yet overwritten the slot.  Dummy can transition to
 | |
|    Active upon key insertion.  Dummy slots cannot be made Unused again
 | |
|    (cannot have me_key set to NULL), else the probe sequence in case of
 | |
|    collision would have no way to know they were once active.
 | |
| 
 | |
| 4. Pending. Not yet inserted or deleted from a split-table.
 | |
|    key != NULL, key != dummy and value == NULL
 | |
| 
 | |
| The DictObject can be in one of two forms.
 | |
| Either:
 | |
|   A combined table:
 | |
|     ma_values == NULL, dk_refcnt == 1.
 | |
|     Values are stored in the me_value field of the PyDictKeysObject.
 | |
|     Slot kind 4 is not allowed i.e.
 | |
|         key != NULL, key != dummy and value == NULL is illegal.
 | |
| Or:
 | |
|   A split table:
 | |
|     ma_values != NULL, dk_refcnt >= 1
 | |
|     Values are stored in the ma_values array.
 | |
|     Only string (unicode) keys are allowed, no <dummy> keys are present.
 | |
| 
 | |
| Note: .popitem() abuses the me_hash field of an Unused or Dummy slot to
 | |
| hold a search finger.  The me_hash field of Unused or Dummy slots has no
 | |
| meaning otherwise. As a consequence of this popitem always converts the dict
 | |
| to the combined-table form.
 | |
| */
 | |
| 
 | |
| /* PyDict_MINSIZE_SPLIT is the minimum size of a split dictionary.
 | |
|  * It must be a power of 2, and at least 4.
 | |
|  * Resizing of split dictionaries is very rare, so the saving memory is more
 | |
|  * important than the cost of resizing.
 | |
|  */
 | |
| #define PyDict_MINSIZE_SPLIT 4
 | |
| 
 | |
| /* PyDict_MINSIZE_COMBINED is the starting size for any new, non-split dict.
 | |
|  * 8 allows dicts with no more than 5 active entries; experiments suggested
 | |
|  * this suffices for the majority of dicts (consisting mostly of usually-small
 | |
|  * dicts created to pass keyword arguments).
 | |
|  * Making this 8, rather than 4 reduces the number of resizes for most
 | |
|  * dictionaries, without any significant extra memory use.
 | |
|  */
 | |
| #define PyDict_MINSIZE_COMBINED 8
 | |
| 
 | |
| #include "Python.h"
 | |
| #include "stringlib/eq.h"
 | |
| 
 | |
| /*[clinic input]
 | |
| class dict "PyDictObject *" "&PyDict_Type"
 | |
| [clinic start generated code]*/
 | |
| /*[clinic end generated code: output=da39a3ee5e6b4b0d input=f157a5a0ce9589d6]*/
 | |
| 
 | |
| typedef struct {
 | |
|     /* Cached hash code of me_key. */
 | |
|     Py_hash_t me_hash;
 | |
|     PyObject *me_key;
 | |
|     PyObject *me_value; /* This field is only meaningful for combined tables */
 | |
| } PyDictKeyEntry;
 | |
| 
 | |
| typedef PyDictKeyEntry *(*dict_lookup_func)
 | |
| (PyDictObject *mp, PyObject *key, Py_hash_t hash, PyObject ***value_addr);
 | |
| 
 | |
| struct _dictkeysobject {
 | |
|     Py_ssize_t dk_refcnt;
 | |
|     Py_ssize_t dk_size;
 | |
|     dict_lookup_func dk_lookup;
 | |
|     Py_ssize_t dk_usable;
 | |
|     PyDictKeyEntry dk_entries[1];
 | |
| };
 | |
| 
 | |
| 
 | |
| /*
 | |
| To ensure the lookup algorithm terminates, there must be at least one Unused
 | |
| slot (NULL key) in the table.
 | |
| To avoid slowing down lookups on a near-full table, we resize the table when
 | |
| it's USABLE_FRACTION (currently two-thirds) full.
 | |
| */
 | |
| 
 | |
| #define PERTURB_SHIFT 5
 | |
| 
 | |
| /*
 | |
| Major subtleties ahead:  Most hash schemes depend on having a "good" hash
 | |
| function, in the sense of simulating randomness.  Python doesn't:  its most
 | |
| important hash functions (for strings and ints) are very regular in common
 | |
| cases:
 | |
| 
 | |
|   >>> map(hash, (0, 1, 2, 3))
 | |
|   [0, 1, 2, 3]
 | |
|   >>> map(hash, ("namea", "nameb", "namec", "named"))
 | |
|   [-1658398457, -1658398460, -1658398459, -1658398462]
 | |
|   >>>
 | |
| 
 | |
| This isn't necessarily bad!  To the contrary, in a table of size 2**i, taking
 | |
| the low-order i bits as the initial table index is extremely fast, and there
 | |
| are no collisions at all for dicts indexed by a contiguous range of ints.
 | |
| The same is approximately true when keys are "consecutive" strings.  So this
 | |
| gives better-than-random behavior in common cases, and that's very desirable.
 | |
| 
 | |
| OTOH, when collisions occur, the tendency to fill contiguous slices of the
 | |
| hash table makes a good collision resolution strategy crucial.  Taking only
 | |
| the last i bits of the hash code is also vulnerable:  for example, consider
 | |
| the list [i << 16 for i in range(20000)] as a set of keys.  Since ints are
 | |
| their own hash codes, and this fits in a dict of size 2**15, the last 15 bits
 | |
|  of every hash code are all 0:  they *all* map to the same table index.
 | |
| 
 | |
| But catering to unusual cases should not slow the usual ones, so we just take
 | |
| the last i bits anyway.  It's up to collision resolution to do the rest.  If
 | |
| we *usually* find the key we're looking for on the first try (and, it turns
 | |
| out, we usually do -- the table load factor is kept under 2/3, so the odds
 | |
| are solidly in our favor), then it makes best sense to keep the initial index
 | |
| computation dirt cheap.
 | |
| 
 | |
| The first half of collision resolution is to visit table indices via this
 | |
| recurrence:
 | |
| 
 | |
|     j = ((5*j) + 1) mod 2**i
 | |
| 
 | |
| For any initial j in range(2**i), repeating that 2**i times generates each
 | |
| int in range(2**i) exactly once (see any text on random-number generation for
 | |
| proof).  By itself, this doesn't help much:  like linear probing (setting
 | |
| j += 1, or j -= 1, on each loop trip), it scans the table entries in a fixed
 | |
| order.  This would be bad, except that's not the only thing we do, and it's
 | |
| actually *good* in the common cases where hash keys are consecutive.  In an
 | |
| example that's really too small to make this entirely clear, for a table of
 | |
| size 2**3 the order of indices is:
 | |
| 
 | |
|     0 -> 1 -> 6 -> 7 -> 4 -> 5 -> 2 -> 3 -> 0 [and here it's repeating]
 | |
| 
 | |
| If two things come in at index 5, the first place we look after is index 2,
 | |
| not 6, so if another comes in at index 6 the collision at 5 didn't hurt it.
 | |
| Linear probing is deadly in this case because there the fixed probe order
 | |
| is the *same* as the order consecutive keys are likely to arrive.  But it's
 | |
| extremely unlikely hash codes will follow a 5*j+1 recurrence by accident,
 | |
| and certain that consecutive hash codes do not.
 | |
| 
 | |
| The other half of the strategy is to get the other bits of the hash code
 | |
| into play.  This is done by initializing a (unsigned) vrbl "perturb" to the
 | |
| full hash code, and changing the recurrence to:
 | |
| 
 | |
|     j = (5*j) + 1 + perturb;
 | |
|     perturb >>= PERTURB_SHIFT;
 | |
|     use j % 2**i as the next table index;
 | |
| 
 | |
| Now the probe sequence depends (eventually) on every bit in the hash code,
 | |
| and the pseudo-scrambling property of recurring on 5*j+1 is more valuable,
 | |
| because it quickly magnifies small differences in the bits that didn't affect
 | |
| the initial index.  Note that because perturb is unsigned, if the recurrence
 | |
| is executed often enough perturb eventually becomes and remains 0.  At that
 | |
| point (very rarely reached) the recurrence is on (just) 5*j+1 again, and
 | |
| that's certain to find an empty slot eventually (since it generates every int
 | |
| in range(2**i), and we make sure there's always at least one empty slot).
 | |
| 
 | |
| Selecting a good value for PERTURB_SHIFT is a balancing act.  You want it
 | |
| small so that the high bits of the hash code continue to affect the probe
 | |
| sequence across iterations; but you want it large so that in really bad cases
 | |
| the high-order hash bits have an effect on early iterations.  5 was "the
 | |
| best" in minimizing total collisions across experiments Tim Peters ran (on
 | |
| both normal and pathological cases), but 4 and 6 weren't significantly worse.
 | |
| 
 | |
| Historical: Reimer Behrends contributed the idea of using a polynomial-based
 | |
| approach, using repeated multiplication by x in GF(2**n) where an irreducible
 | |
| polynomial for each table size was chosen such that x was a primitive root.
 | |
| Christian Tismer later extended that to use division by x instead, as an
 | |
| efficient way to get the high bits of the hash code into play.  This scheme
 | |
| also gave excellent collision statistics, but was more expensive:  two
 | |
| if-tests were required inside the loop; computing "the next" index took about
 | |
| the same number of operations but without as much potential parallelism
 | |
| (e.g., computing 5*j can go on at the same time as computing 1+perturb in the
 | |
| above, and then shifting perturb can be done while the table index is being
 | |
| masked); and the PyDictObject struct required a member to hold the table's
 | |
| polynomial.  In Tim's experiments the current scheme ran faster, produced
 | |
| equally good collision statistics, needed less code & used less memory.
 | |
| 
 | |
| */
 | |
| 
 | |
| /* Object used as dummy key to fill deleted entries
 | |
|  * This could be any unique object,
 | |
|  * use a custom type in order to minimise coupling.
 | |
| */
 | |
| static PyObject _dummy_struct;
 | |
| 
 | |
| #define dummy (&_dummy_struct)
 | |
| 
 | |
| #ifdef Py_REF_DEBUG
 | |
| PyObject *
 | |
| _PyDict_Dummy(void)
 | |
| {
 | |
|     return dummy;
 | |
| }
 | |
| #endif
 | |
| 
 | |
| /* forward declarations */
 | |
| static PyDictKeyEntry *lookdict(PyDictObject *mp, PyObject *key,
 | |
|                                 Py_hash_t hash, PyObject ***value_addr);
 | |
| static PyDictKeyEntry *lookdict_unicode(PyDictObject *mp, PyObject *key,
 | |
|                                         Py_hash_t hash, PyObject ***value_addr);
 | |
| static PyDictKeyEntry *
 | |
| lookdict_unicode_nodummy(PyDictObject *mp, PyObject *key,
 | |
|                          Py_hash_t hash, PyObject ***value_addr);
 | |
| static PyDictKeyEntry *lookdict_split(PyDictObject *mp, PyObject *key,
 | |
|                                       Py_hash_t hash, PyObject ***value_addr);
 | |
| 
 | |
| static int dictresize(PyDictObject *mp, Py_ssize_t minused);
 | |
| 
 | |
| /* Dictionary reuse scheme to save calls to malloc, free, and memset */
 | |
| #ifndef PyDict_MAXFREELIST
 | |
| #define PyDict_MAXFREELIST 80
 | |
| #endif
 | |
| static PyDictObject *free_list[PyDict_MAXFREELIST];
 | |
| static int numfree = 0;
 | |
| 
 | |
| int
 | |
| PyDict_ClearFreeList(void)
 | |
| {
 | |
|     PyDictObject *op;
 | |
|     int ret = numfree;
 | |
|     while (numfree) {
 | |
|         op = free_list[--numfree];
 | |
|         assert(PyDict_CheckExact(op));
 | |
|         PyObject_GC_Del(op);
 | |
|     }
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| /* Print summary info about the state of the optimized allocator */
 | |
| void
 | |
| _PyDict_DebugMallocStats(FILE *out)
 | |
| {
 | |
|     _PyDebugAllocatorStats(out,
 | |
|                            "free PyDictObject", numfree, sizeof(PyDictObject));
 | |
| }
 | |
| 
 | |
| 
 | |
| void
 | |
| PyDict_Fini(void)
 | |
| {
 | |
|     PyDict_ClearFreeList();
 | |
| }
 | |
| 
 | |
| #define DK_DEBUG_INCREF _Py_INC_REFTOTAL _Py_REF_DEBUG_COMMA
 | |
| #define DK_DEBUG_DECREF _Py_DEC_REFTOTAL _Py_REF_DEBUG_COMMA
 | |
| 
 | |
| #define DK_INCREF(dk) (DK_DEBUG_INCREF ++(dk)->dk_refcnt)
 | |
| #define DK_DECREF(dk) if (DK_DEBUG_DECREF (--(dk)->dk_refcnt) == 0) free_keys_object(dk)
 | |
| #define DK_SIZE(dk) ((dk)->dk_size)
 | |
| #define DK_MASK(dk) (((dk)->dk_size)-1)
 | |
| #define IS_POWER_OF_2(x) (((x) & (x-1)) == 0)
 | |
| 
 | |
| /* USABLE_FRACTION is the maximum dictionary load.
 | |
|  * Currently set to (2n+1)/3. Increasing this ratio makes dictionaries more
 | |
|  * dense resulting in more collisions.  Decreasing it improves sparseness
 | |
|  * at the expense of spreading entries over more cache lines and at the
 | |
|  * cost of total memory consumed.
 | |
|  *
 | |
|  * USABLE_FRACTION must obey the following:
 | |
|  *     (0 < USABLE_FRACTION(n) < n) for all n >= 2
 | |
|  *
 | |
|  * USABLE_FRACTION should be very quick to calculate.
 | |
|  * Fractions around 5/8 to 2/3 seem to work well in practice.
 | |
|  */
 | |
| 
 | |
| /* Use (2n+1)/3 rather than 2n+3 because: it makes no difference for
 | |
|  * combined tables (the two fractions round to the same number n < ),
 | |
|  * but 2*4/3 is 2 whereas (2*4+1)/3 is 3 which potentially saves quite
 | |
|  * a lot of space for small, split tables */
 | |
| #define USABLE_FRACTION(n) ((((n) << 1)+1)/3)
 | |
| 
 | |
| /* Alternative fraction that is otherwise close enough to (2n+1)/3 to make
 | |
|  * little difference. 8 * 2/3 == 8 * 5/8 == 5. 16 * 2/3 == 16 * 5/8 == 10.
 | |
|  * 32 * 2/3 = 21, 32 * 5/8 = 20.
 | |
|  * Its advantage is that it is faster to compute on machines with slow division.
 | |
|  * #define USABLE_FRACTION(n) (((n) >> 1) + ((n) >> 2) - ((n) >> 3))
 | |
| */
 | |
| 
 | |
| /* GROWTH_RATE. Growth rate upon hitting maximum load.
 | |
|  * Currently set to used*2 + capacity/2.
 | |
|  * This means that dicts double in size when growing without deletions,
 | |
|  * but have more head room when the number of deletions is on a par with the
 | |
|  * number of insertions.
 | |
|  * Raising this to used*4 doubles memory consumption depending on the size of
 | |
|  * the dictionary, but results in half the number of resizes, less effort to
 | |
|  * resize.
 | |
|  * GROWTH_RATE was set to used*4 up to version 3.2.
 | |
|  * GROWTH_RATE was set to used*2 in version 3.3.0
 | |
|  */
 | |
| #define GROWTH_RATE(d) (((d)->ma_used*2)+((d)->ma_keys->dk_size>>1))
 | |
| 
 | |
| #define ENSURE_ALLOWS_DELETIONS(d) \
 | |
|     if ((d)->ma_keys->dk_lookup == lookdict_unicode_nodummy) { \
 | |
|         (d)->ma_keys->dk_lookup = lookdict_unicode; \
 | |
|     }
 | |
| 
 | |
| /* This immutable, empty PyDictKeysObject is used for PyDict_Clear()
 | |
|  * (which cannot fail and thus can do no allocation).
 | |
|  */
 | |
| static PyDictKeysObject empty_keys_struct = {
 | |
|         2, /* dk_refcnt 1 for this struct, 1 for dummy_struct */
 | |
|         1, /* dk_size */
 | |
|         lookdict_split, /* dk_lookup */
 | |
|         0, /* dk_usable (immutable) */
 | |
|         {
 | |
|             { 0, 0, 0 } /* dk_entries (empty) */
 | |
|         }
 | |
| };
 | |
| 
 | |
| static PyObject *empty_values[1] = { NULL };
 | |
| 
 | |
| #define Py_EMPTY_KEYS &empty_keys_struct
 | |
| 
 | |
| static PyDictKeysObject *new_keys_object(Py_ssize_t size)
 | |
| {
 | |
|     PyDictKeysObject *dk;
 | |
|     Py_ssize_t i;
 | |
|     PyDictKeyEntry *ep0;
 | |
| 
 | |
|     assert(size >= PyDict_MINSIZE_SPLIT);
 | |
|     assert(IS_POWER_OF_2(size));
 | |
|     dk = PyMem_MALLOC(sizeof(PyDictKeysObject) +
 | |
|                       sizeof(PyDictKeyEntry) * (size-1));
 | |
|     if (dk == NULL) {
 | |
|         PyErr_NoMemory();
 | |
|         return NULL;
 | |
|     }
 | |
|     DK_DEBUG_INCREF dk->dk_refcnt = 1;
 | |
|     dk->dk_size = size;
 | |
|     dk->dk_usable = USABLE_FRACTION(size);
 | |
|     ep0 = &dk->dk_entries[0];
 | |
|     /* Hash value of slot 0 is used by popitem, so it must be initialized */
 | |
|     ep0->me_hash = 0;
 | |
|     for (i = 0; i < size; i++) {
 | |
|         ep0[i].me_key = NULL;
 | |
|         ep0[i].me_value = NULL;
 | |
|     }
 | |
|     dk->dk_lookup = lookdict_unicode_nodummy;
 | |
|     return dk;
 | |
| }
 | |
| 
 | |
| static void
 | |
| free_keys_object(PyDictKeysObject *keys)
 | |
| {
 | |
|     PyDictKeyEntry *entries = &keys->dk_entries[0];
 | |
|     Py_ssize_t i, n;
 | |
|     for (i = 0, n = DK_SIZE(keys); i < n; i++) {
 | |
|         Py_XDECREF(entries[i].me_key);
 | |
|         Py_XDECREF(entries[i].me_value);
 | |
|     }
 | |
|     PyMem_FREE(keys);
 | |
| }
 | |
| 
 | |
| #define new_values(size) PyMem_NEW(PyObject *, size)
 | |
| 
 | |
| #define free_values(values) PyMem_FREE(values)
 | |
| 
 | |
| /* Consumes a reference to the keys object */
 | |
| static PyObject *
 | |
| new_dict(PyDictKeysObject *keys, PyObject **values)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     assert(keys != NULL);
 | |
|     if (numfree) {
 | |
|         mp = free_list[--numfree];
 | |
|         assert (mp != NULL);
 | |
|         assert (Py_TYPE(mp) == &PyDict_Type);
 | |
|         _Py_NewReference((PyObject *)mp);
 | |
|     }
 | |
|     else {
 | |
|         mp = PyObject_GC_New(PyDictObject, &PyDict_Type);
 | |
|         if (mp == NULL) {
 | |
|             DK_DECREF(keys);
 | |
|             free_values(values);
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
|     mp->ma_keys = keys;
 | |
|     mp->ma_values = values;
 | |
|     mp->ma_used = 0;
 | |
|     return (PyObject *)mp;
 | |
| }
 | |
| 
 | |
| /* Consumes a reference to the keys object */
 | |
| static PyObject *
 | |
| new_dict_with_shared_keys(PyDictKeysObject *keys)
 | |
| {
 | |
|     PyObject **values;
 | |
|     Py_ssize_t i, size;
 | |
| 
 | |
|     size = DK_SIZE(keys);
 | |
|     values = new_values(size);
 | |
|     if (values == NULL) {
 | |
|         DK_DECREF(keys);
 | |
|         return PyErr_NoMemory();
 | |
|     }
 | |
|     for (i = 0; i < size; i++) {
 | |
|         values[i] = NULL;
 | |
|     }
 | |
|     return new_dict(keys, values);
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_New(void)
 | |
| {
 | |
|     PyDictKeysObject *keys = new_keys_object(PyDict_MINSIZE_COMBINED);
 | |
|     if (keys == NULL)
 | |
|         return NULL;
 | |
|     return new_dict(keys, NULL);
 | |
| }
 | |
| 
 | |
| /*
 | |
| The basic lookup function used by all operations.
 | |
| This is based on Algorithm D from Knuth Vol. 3, Sec. 6.4.
 | |
| Open addressing is preferred over chaining since the link overhead for
 | |
| chaining would be substantial (100% with typical malloc overhead).
 | |
| 
 | |
| The initial probe index is computed as hash mod the table size. Subsequent
 | |
| probe indices are computed as explained earlier.
 | |
| 
 | |
| All arithmetic on hash should ignore overflow.
 | |
| 
 | |
| The details in this version are due to Tim Peters, building on many past
 | |
| contributions by Reimer Behrends, Jyrki Alakuijala, Vladimir Marangozov and
 | |
| Christian Tismer.
 | |
| 
 | |
| lookdict() is general-purpose, and may return NULL if (and only if) a
 | |
| comparison raises an exception (this was new in Python 2.5).
 | |
| lookdict_unicode() below is specialized to string keys, comparison of which can
 | |
| never raise an exception; that function can never return NULL.
 | |
| lookdict_unicode_nodummy is further specialized for string keys that cannot be
 | |
| the <dummy> value.
 | |
| For both, when the key isn't found a PyDictEntry* is returned
 | |
| where the key would have been found, *value_addr points to the matching value
 | |
| slot.
 | |
| */
 | |
| static PyDictKeyEntry *
 | |
| lookdict(PyDictObject *mp, PyObject *key,
 | |
|          Py_hash_t hash, PyObject ***value_addr)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     PyDictKeyEntry *freeslot;
 | |
|     size_t mask;
 | |
|     PyDictKeyEntry *ep0;
 | |
|     PyDictKeyEntry *ep;
 | |
|     int cmp;
 | |
|     PyObject *startkey;
 | |
| 
 | |
| top:
 | |
|     mask = DK_MASK(mp->ma_keys);
 | |
|     ep0 = &mp->ma_keys->dk_entries[0];
 | |
|     i = (size_t)hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     if (ep->me_key == NULL || ep->me_key == key) {
 | |
|         *value_addr = &ep->me_value;
 | |
|         return ep;
 | |
|     }
 | |
|     if (ep->me_key == dummy)
 | |
|         freeslot = ep;
 | |
|     else {
 | |
|         if (ep->me_hash == hash) {
 | |
|             startkey = ep->me_key;
 | |
|             Py_INCREF(startkey);
 | |
|             cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
 | |
|             Py_DECREF(startkey);
 | |
|             if (cmp < 0)
 | |
|                 return NULL;
 | |
|             if (ep0 == mp->ma_keys->dk_entries && ep->me_key == startkey) {
 | |
|                 if (cmp > 0) {
 | |
|                     *value_addr = &ep->me_value;
 | |
|                     return ep;
 | |
|                 }
 | |
|             }
 | |
|             else {
 | |
|                 /* The dict was mutated, restart */
 | |
|                 goto top;
 | |
|             }
 | |
|         }
 | |
|         freeslot = NULL;
 | |
|     }
 | |
| 
 | |
|     /* In the loop, me_key == dummy is by far (factor of 100s) the
 | |
|        least likely outcome, so test for that last. */
 | |
|     for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|         if (ep->me_key == NULL) {
 | |
|             if (freeslot == NULL) {
 | |
|                 *value_addr = &ep->me_value;
 | |
|                 return ep;
 | |
|             } else {
 | |
|                 *value_addr = &freeslot->me_value;
 | |
|                 return freeslot;
 | |
|             }
 | |
|         }
 | |
|         if (ep->me_key == key) {
 | |
|             *value_addr = &ep->me_value;
 | |
|             return ep;
 | |
|         }
 | |
|         if (ep->me_hash == hash && ep->me_key != dummy) {
 | |
|             startkey = ep->me_key;
 | |
|             Py_INCREF(startkey);
 | |
|             cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
 | |
|             Py_DECREF(startkey);
 | |
|             if (cmp < 0) {
 | |
|                 *value_addr = NULL;
 | |
|                 return NULL;
 | |
|             }
 | |
|             if (ep0 == mp->ma_keys->dk_entries && ep->me_key == startkey) {
 | |
|                 if (cmp > 0) {
 | |
|                     *value_addr = &ep->me_value;
 | |
|                     return ep;
 | |
|                 }
 | |
|             }
 | |
|             else {
 | |
|                 /* The dict was mutated, restart */
 | |
|                 goto top;
 | |
|             }
 | |
|         }
 | |
|         else if (ep->me_key == dummy && freeslot == NULL)
 | |
|             freeslot = ep;
 | |
|     }
 | |
|     assert(0);          /* NOT REACHED */
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /* Specialized version for string-only keys */
 | |
| static PyDictKeyEntry *
 | |
| lookdict_unicode(PyDictObject *mp, PyObject *key,
 | |
|                  Py_hash_t hash, PyObject ***value_addr)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     PyDictKeyEntry *freeslot;
 | |
|     size_t mask = DK_MASK(mp->ma_keys);
 | |
|     PyDictKeyEntry *ep0 = &mp->ma_keys->dk_entries[0];
 | |
|     PyDictKeyEntry *ep;
 | |
| 
 | |
|     /* Make sure this function doesn't have to handle non-unicode keys,
 | |
|        including subclasses of str; e.g., one reason to subclass
 | |
|        unicodes is to override __eq__, and for speed we don't cater to
 | |
|        that here. */
 | |
|     if (!PyUnicode_CheckExact(key)) {
 | |
|         mp->ma_keys->dk_lookup = lookdict;
 | |
|         return lookdict(mp, key, hash, value_addr);
 | |
|     }
 | |
|     i = (size_t)hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     if (ep->me_key == NULL || ep->me_key == key) {
 | |
|         *value_addr = &ep->me_value;
 | |
|         return ep;
 | |
|     }
 | |
|     if (ep->me_key == dummy)
 | |
|         freeslot = ep;
 | |
|     else {
 | |
|         if (ep->me_hash == hash && unicode_eq(ep->me_key, key)) {
 | |
|             *value_addr = &ep->me_value;
 | |
|             return ep;
 | |
|         }
 | |
|         freeslot = NULL;
 | |
|     }
 | |
| 
 | |
|     /* In the loop, me_key == dummy is by far (factor of 100s) the
 | |
|        least likely outcome, so test for that last. */
 | |
|     for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|         if (ep->me_key == NULL) {
 | |
|             if (freeslot == NULL) {
 | |
|                 *value_addr = &ep->me_value;
 | |
|                 return ep;
 | |
|             } else {
 | |
|                 *value_addr = &freeslot->me_value;
 | |
|                 return freeslot;
 | |
|             }
 | |
|         }
 | |
|         if (ep->me_key == key
 | |
|             || (ep->me_hash == hash
 | |
|             && ep->me_key != dummy
 | |
|             && unicode_eq(ep->me_key, key))) {
 | |
|             *value_addr = &ep->me_value;
 | |
|             return ep;
 | |
|         }
 | |
|         if (ep->me_key == dummy && freeslot == NULL)
 | |
|             freeslot = ep;
 | |
|     }
 | |
|     assert(0);          /* NOT REACHED */
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /* Faster version of lookdict_unicode when it is known that no <dummy> keys
 | |
|  * will be present. */
 | |
| static PyDictKeyEntry *
 | |
| lookdict_unicode_nodummy(PyDictObject *mp, PyObject *key,
 | |
|                          Py_hash_t hash, PyObject ***value_addr)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     size_t mask = DK_MASK(mp->ma_keys);
 | |
|     PyDictKeyEntry *ep0 = &mp->ma_keys->dk_entries[0];
 | |
|     PyDictKeyEntry *ep;
 | |
| 
 | |
|     /* Make sure this function doesn't have to handle non-unicode keys,
 | |
|        including subclasses of str; e.g., one reason to subclass
 | |
|        unicodes is to override __eq__, and for speed we don't cater to
 | |
|        that here. */
 | |
|     if (!PyUnicode_CheckExact(key)) {
 | |
|         mp->ma_keys->dk_lookup = lookdict;
 | |
|         return lookdict(mp, key, hash, value_addr);
 | |
|     }
 | |
|     i = (size_t)hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     assert(ep->me_key == NULL || PyUnicode_CheckExact(ep->me_key));
 | |
|     if (ep->me_key == NULL || ep->me_key == key ||
 | |
|         (ep->me_hash == hash && unicode_eq(ep->me_key, key))) {
 | |
|         *value_addr = &ep->me_value;
 | |
|         return ep;
 | |
|     }
 | |
|     for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|         assert(ep->me_key == NULL || PyUnicode_CheckExact(ep->me_key));
 | |
|         if (ep->me_key == NULL || ep->me_key == key ||
 | |
|             (ep->me_hash == hash && unicode_eq(ep->me_key, key))) {
 | |
|             *value_addr = &ep->me_value;
 | |
|             return ep;
 | |
|         }
 | |
|     }
 | |
|     assert(0);          /* NOT REACHED */
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /* Version of lookdict for split tables.
 | |
|  * All split tables and only split tables use this lookup function.
 | |
|  * Split tables only contain unicode keys and no dummy keys,
 | |
|  * so algorithm is the same as lookdict_unicode_nodummy.
 | |
|  */
 | |
| static PyDictKeyEntry *
 | |
| lookdict_split(PyDictObject *mp, PyObject *key,
 | |
|                Py_hash_t hash, PyObject ***value_addr)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     size_t mask = DK_MASK(mp->ma_keys);
 | |
|     PyDictKeyEntry *ep0 = &mp->ma_keys->dk_entries[0];
 | |
|     PyDictKeyEntry *ep;
 | |
| 
 | |
|     if (!PyUnicode_CheckExact(key)) {
 | |
|         ep = lookdict(mp, key, hash, value_addr);
 | |
|         /* lookdict expects a combined-table, so fix value_addr */
 | |
|         i = ep - ep0;
 | |
|         *value_addr = &mp->ma_values[i];
 | |
|         return ep;
 | |
|     }
 | |
|     i = (size_t)hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     assert(ep->me_key == NULL || PyUnicode_CheckExact(ep->me_key));
 | |
|     if (ep->me_key == NULL || ep->me_key == key ||
 | |
|         (ep->me_hash == hash && unicode_eq(ep->me_key, key))) {
 | |
|         *value_addr = &mp->ma_values[i];
 | |
|         return ep;
 | |
|     }
 | |
|     for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|         assert(ep->me_key == NULL || PyUnicode_CheckExact(ep->me_key));
 | |
|         if (ep->me_key == NULL || ep->me_key == key ||
 | |
|             (ep->me_hash == hash && unicode_eq(ep->me_key, key))) {
 | |
|             *value_addr = &mp->ma_values[i & mask];
 | |
|             return ep;
 | |
|         }
 | |
|     }
 | |
|     assert(0);          /* NOT REACHED */
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| int
 | |
| _PyDict_HasOnlyStringKeys(PyObject *dict)
 | |
| {
 | |
|     Py_ssize_t pos = 0;
 | |
|     PyObject *key, *value;
 | |
|     assert(PyDict_Check(dict));
 | |
|     /* Shortcut */
 | |
|     if (((PyDictObject *)dict)->ma_keys->dk_lookup != lookdict)
 | |
|         return 1;
 | |
|     while (PyDict_Next(dict, &pos, &key, &value))
 | |
|         if (!PyUnicode_Check(key))
 | |
|             return 0;
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| #define MAINTAIN_TRACKING(mp, key, value) \
 | |
|     do { \
 | |
|         if (!_PyObject_GC_IS_TRACKED(mp)) { \
 | |
|             if (_PyObject_GC_MAY_BE_TRACKED(key) || \
 | |
|                 _PyObject_GC_MAY_BE_TRACKED(value)) { \
 | |
|                 _PyObject_GC_TRACK(mp); \
 | |
|             } \
 | |
|         } \
 | |
|     } while(0)
 | |
| 
 | |
| void
 | |
| _PyDict_MaybeUntrack(PyObject *op)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     PyObject *value;
 | |
|     Py_ssize_t i, size;
 | |
| 
 | |
|     if (!PyDict_CheckExact(op) || !_PyObject_GC_IS_TRACKED(op))
 | |
|         return;
 | |
| 
 | |
|     mp = (PyDictObject *) op;
 | |
|     size = DK_SIZE(mp->ma_keys);
 | |
|     if (_PyDict_HasSplitTable(mp)) {
 | |
|         for (i = 0; i < size; i++) {
 | |
|             if ((value = mp->ma_values[i]) == NULL)
 | |
|                 continue;
 | |
|             if (_PyObject_GC_MAY_BE_TRACKED(value)) {
 | |
|                 assert(!_PyObject_GC_MAY_BE_TRACKED(
 | |
|                     mp->ma_keys->dk_entries[i].me_key));
 | |
|                 return;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     else {
 | |
|         PyDictKeyEntry *ep0 = &mp->ma_keys->dk_entries[0];
 | |
|         for (i = 0; i < size; i++) {
 | |
|             if ((value = ep0[i].me_value) == NULL)
 | |
|                 continue;
 | |
|             if (_PyObject_GC_MAY_BE_TRACKED(value) ||
 | |
|                 _PyObject_GC_MAY_BE_TRACKED(ep0[i].me_key))
 | |
|                 return;
 | |
|         }
 | |
|     }
 | |
|     _PyObject_GC_UNTRACK(op);
 | |
| }
 | |
| 
 | |
| /* Internal function to find slot for an item from its hash
 | |
|  * when it is known that the key is not present in the dict.
 | |
|  */
 | |
| static PyDictKeyEntry *
 | |
| find_empty_slot(PyDictObject *mp, PyObject *key, Py_hash_t hash,
 | |
|                 PyObject ***value_addr)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     size_t mask = DK_MASK(mp->ma_keys);
 | |
|     PyDictKeyEntry *ep0 = &mp->ma_keys->dk_entries[0];
 | |
|     PyDictKeyEntry *ep;
 | |
| 
 | |
|     assert(key != NULL);
 | |
|     if (!PyUnicode_CheckExact(key))
 | |
|         mp->ma_keys->dk_lookup = lookdict;
 | |
|     i = hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     for (perturb = hash; ep->me_key != NULL; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|     }
 | |
|     assert(ep->me_value == NULL);
 | |
|     if (mp->ma_values)
 | |
|         *value_addr = &mp->ma_values[i & mask];
 | |
|     else
 | |
|         *value_addr = &ep->me_value;
 | |
|     return ep;
 | |
| }
 | |
| 
 | |
| static int
 | |
| insertion_resize(PyDictObject *mp)
 | |
| {
 | |
|     return dictresize(mp, GROWTH_RATE(mp));
 | |
| }
 | |
| 
 | |
| /*
 | |
| Internal routine to insert a new item into the table.
 | |
| Used both by the internal resize routine and by the public insert routine.
 | |
| Returns -1 if an error occurred, or 0 on success.
 | |
| */
 | |
| static int
 | |
| insertdict(PyDictObject *mp, PyObject *key, Py_hash_t hash, PyObject *value)
 | |
| {
 | |
|     PyObject *old_value;
 | |
|     PyObject **value_addr;
 | |
|     PyDictKeyEntry *ep;
 | |
|     assert(key != dummy);
 | |
| 
 | |
|     if (mp->ma_values != NULL && !PyUnicode_CheckExact(key)) {
 | |
|         if (insertion_resize(mp) < 0)
 | |
|             return -1;
 | |
|     }
 | |
| 
 | |
|     ep = mp->ma_keys->dk_lookup(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL) {
 | |
|         return -1;
 | |
|     }
 | |
|     assert(PyUnicode_CheckExact(key) || mp->ma_keys->dk_lookup == lookdict);
 | |
|     Py_INCREF(value);
 | |
|     MAINTAIN_TRACKING(mp, key, value);
 | |
|     old_value = *value_addr;
 | |
|     if (old_value != NULL) {
 | |
|         assert(ep->me_key != NULL && ep->me_key != dummy);
 | |
|         *value_addr = value;
 | |
|         Py_DECREF(old_value); /* which **CAN** re-enter (see issue #22653) */
 | |
|     }
 | |
|     else {
 | |
|         if (ep->me_key == NULL) {
 | |
|             Py_INCREF(key);
 | |
|             if (mp->ma_keys->dk_usable <= 0) {
 | |
|                 /* Need to resize. */
 | |
|                 if (insertion_resize(mp) < 0) {
 | |
|                     Py_DECREF(key);
 | |
|                     Py_DECREF(value);
 | |
|                     return -1;
 | |
|                 }
 | |
|                 ep = find_empty_slot(mp, key, hash, &value_addr);
 | |
|             }
 | |
|             mp->ma_keys->dk_usable--;
 | |
|             assert(mp->ma_keys->dk_usable >= 0);
 | |
|             ep->me_key = key;
 | |
|             ep->me_hash = hash;
 | |
|         }
 | |
|         else {
 | |
|             if (ep->me_key == dummy) {
 | |
|                 Py_INCREF(key);
 | |
|                 ep->me_key = key;
 | |
|                 ep->me_hash = hash;
 | |
|                 Py_DECREF(dummy);
 | |
|             } else {
 | |
|                 assert(_PyDict_HasSplitTable(mp));
 | |
|             }
 | |
|         }
 | |
|         mp->ma_used++;
 | |
|         *value_addr = value;
 | |
|         assert(ep->me_key != NULL && ep->me_key != dummy);
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /*
 | |
| Internal routine used by dictresize() to insert an item which is
 | |
| known to be absent from the dict.  This routine also assumes that
 | |
| the dict contains no deleted entries.  Besides the performance benefit,
 | |
| using insertdict() in dictresize() is dangerous (SF bug #1456209).
 | |
| Note that no refcounts are changed by this routine; if needed, the caller
 | |
| is responsible for incref'ing `key` and `value`.
 | |
| Neither mp->ma_used nor k->dk_usable are modified by this routine; the caller
 | |
| must set them correctly
 | |
| */
 | |
| static void
 | |
| insertdict_clean(PyDictObject *mp, PyObject *key, Py_hash_t hash,
 | |
|                  PyObject *value)
 | |
| {
 | |
|     size_t i;
 | |
|     size_t perturb;
 | |
|     PyDictKeysObject *k = mp->ma_keys;
 | |
|     size_t mask = (size_t)DK_SIZE(k)-1;
 | |
|     PyDictKeyEntry *ep0 = &k->dk_entries[0];
 | |
|     PyDictKeyEntry *ep;
 | |
| 
 | |
|     assert(k->dk_lookup != NULL);
 | |
|     assert(value != NULL);
 | |
|     assert(key != NULL);
 | |
|     assert(key != dummy);
 | |
|     assert(PyUnicode_CheckExact(key) || k->dk_lookup == lookdict);
 | |
|     i = hash & mask;
 | |
|     ep = &ep0[i];
 | |
|     for (perturb = hash; ep->me_key != NULL; perturb >>= PERTURB_SHIFT) {
 | |
|         i = (i << 2) + i + perturb + 1;
 | |
|         ep = &ep0[i & mask];
 | |
|     }
 | |
|     assert(ep->me_value == NULL);
 | |
|     ep->me_key = key;
 | |
|     ep->me_hash = hash;
 | |
|     ep->me_value = value;
 | |
| }
 | |
| 
 | |
| /*
 | |
| Restructure the table by allocating a new table and reinserting all
 | |
| items again.  When entries have been deleted, the new table may
 | |
| actually be smaller than the old one.
 | |
| If a table is split (its keys and hashes are shared, its values are not),
 | |
| then the values are temporarily copied into the table, it is resized as
 | |
| a combined table, then the me_value slots in the old table are NULLed out.
 | |
| After resizing a table is always combined,
 | |
| but can be resplit by make_keys_shared().
 | |
| */
 | |
| static int
 | |
| dictresize(PyDictObject *mp, Py_ssize_t minused)
 | |
| {
 | |
|     Py_ssize_t newsize;
 | |
|     PyDictKeysObject *oldkeys;
 | |
|     PyObject **oldvalues;
 | |
|     Py_ssize_t i, oldsize;
 | |
| 
 | |
| /* Find the smallest table size > minused. */
 | |
|     for (newsize = PyDict_MINSIZE_COMBINED;
 | |
|          newsize <= minused && newsize > 0;
 | |
|          newsize <<= 1)
 | |
|         ;
 | |
|     if (newsize <= 0) {
 | |
|         PyErr_NoMemory();
 | |
|         return -1;
 | |
|     }
 | |
|     oldkeys = mp->ma_keys;
 | |
|     oldvalues = mp->ma_values;
 | |
|     /* Allocate a new table. */
 | |
|     mp->ma_keys = new_keys_object(newsize);
 | |
|     if (mp->ma_keys == NULL) {
 | |
|         mp->ma_keys = oldkeys;
 | |
|         return -1;
 | |
|     }
 | |
|     if (oldkeys->dk_lookup == lookdict)
 | |
|         mp->ma_keys->dk_lookup = lookdict;
 | |
|     oldsize = DK_SIZE(oldkeys);
 | |
|     mp->ma_values = NULL;
 | |
|     /* If empty then nothing to copy so just return */
 | |
|     if (oldsize == 1) {
 | |
|         assert(oldkeys == Py_EMPTY_KEYS);
 | |
|         DK_DECREF(oldkeys);
 | |
|         return 0;
 | |
|     }
 | |
|     /* Main loop below assumes we can transfer refcount to new keys
 | |
|      * and that value is stored in me_value.
 | |
|      * Increment ref-counts and copy values here to compensate
 | |
|      * This (resizing a split table) should be relatively rare */
 | |
|     if (oldvalues != NULL) {
 | |
|         for (i = 0; i < oldsize; i++) {
 | |
|             if (oldvalues[i] != NULL) {
 | |
|                 Py_INCREF(oldkeys->dk_entries[i].me_key);
 | |
|                 oldkeys->dk_entries[i].me_value = oldvalues[i];
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     /* Main loop */
 | |
|     for (i = 0; i < oldsize; i++) {
 | |
|         PyDictKeyEntry *ep = &oldkeys->dk_entries[i];
 | |
|         if (ep->me_value != NULL) {
 | |
|             assert(ep->me_key != dummy);
 | |
|             insertdict_clean(mp, ep->me_key, ep->me_hash, ep->me_value);
 | |
|         }
 | |
|     }
 | |
|     mp->ma_keys->dk_usable -= mp->ma_used;
 | |
|     if (oldvalues != NULL) {
 | |
|         /* NULL out me_value slot in oldkeys, in case it was shared */
 | |
|         for (i = 0; i < oldsize; i++)
 | |
|             oldkeys->dk_entries[i].me_value = NULL;
 | |
|         assert(oldvalues != empty_values);
 | |
|         free_values(oldvalues);
 | |
|         DK_DECREF(oldkeys);
 | |
|     }
 | |
|     else {
 | |
|         assert(oldkeys->dk_lookup != lookdict_split);
 | |
|         if (oldkeys->dk_lookup != lookdict_unicode_nodummy) {
 | |
|             PyDictKeyEntry *ep0 = &oldkeys->dk_entries[0];
 | |
|             for (i = 0; i < oldsize; i++) {
 | |
|                 if (ep0[i].me_key == dummy)
 | |
|                     Py_DECREF(dummy);
 | |
|             }
 | |
|         }
 | |
|         assert(oldkeys->dk_refcnt == 1);
 | |
|         DK_DEBUG_DECREF PyMem_FREE(oldkeys);
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| /* Returns NULL if unable to split table.
 | |
|  * A NULL return does not necessarily indicate an error */
 | |
| static PyDictKeysObject *
 | |
| make_keys_shared(PyObject *op)
 | |
| {
 | |
|     Py_ssize_t i;
 | |
|     Py_ssize_t size;
 | |
|     PyDictObject *mp = (PyDictObject *)op;
 | |
| 
 | |
|     if (!PyDict_CheckExact(op))
 | |
|         return NULL;
 | |
|     if (!_PyDict_HasSplitTable(mp)) {
 | |
|         PyDictKeyEntry *ep0;
 | |
|         PyObject **values;
 | |
|         assert(mp->ma_keys->dk_refcnt == 1);
 | |
|         if (mp->ma_keys->dk_lookup == lookdict) {
 | |
|             return NULL;
 | |
|         }
 | |
|         else if (mp->ma_keys->dk_lookup == lookdict_unicode) {
 | |
|             /* Remove dummy keys */
 | |
|             if (dictresize(mp, DK_SIZE(mp->ma_keys)))
 | |
|                 return NULL;
 | |
|         }
 | |
|         assert(mp->ma_keys->dk_lookup == lookdict_unicode_nodummy);
 | |
|         /* Copy values into a new array */
 | |
|         ep0 = &mp->ma_keys->dk_entries[0];
 | |
|         size = DK_SIZE(mp->ma_keys);
 | |
|         values = new_values(size);
 | |
|         if (values == NULL) {
 | |
|             PyErr_SetString(PyExc_MemoryError,
 | |
|                 "Not enough memory to allocate new values array");
 | |
|             return NULL;
 | |
|         }
 | |
|         for (i = 0; i < size; i++) {
 | |
|             values[i] = ep0[i].me_value;
 | |
|             ep0[i].me_value = NULL;
 | |
|         }
 | |
|         mp->ma_keys->dk_lookup = lookdict_split;
 | |
|         mp->ma_values = values;
 | |
|     }
 | |
|     DK_INCREF(mp->ma_keys);
 | |
|     return mp->ma_keys;
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| _PyDict_NewPresized(Py_ssize_t minused)
 | |
| {
 | |
|     Py_ssize_t newsize;
 | |
|     PyDictKeysObject *new_keys;
 | |
|     for (newsize = PyDict_MINSIZE_COMBINED;
 | |
|          newsize <= minused && newsize > 0;
 | |
|          newsize <<= 1)
 | |
|         ;
 | |
|     new_keys = new_keys_object(newsize);
 | |
|     if (new_keys == NULL)
 | |
|         return NULL;
 | |
|     return new_dict(new_keys, NULL);
 | |
| }
 | |
| 
 | |
| /* Note that, for historical reasons, PyDict_GetItem() suppresses all errors
 | |
|  * that may occur (originally dicts supported only string keys, and exceptions
 | |
|  * weren't possible).  So, while the original intent was that a NULL return
 | |
|  * meant the key wasn't present, in reality it can mean that, or that an error
 | |
|  * (suppressed) occurred while computing the key's hash, or that some error
 | |
|  * (suppressed) occurred when comparing keys in the dict's internal probe
 | |
|  * sequence.  A nasty example of the latter is when a Python-coded comparison
 | |
|  * function hits a stack-depth error, which can cause this to return NULL
 | |
|  * even if the key is present.
 | |
|  */
 | |
| PyObject *
 | |
| PyDict_GetItem(PyObject *op, PyObject *key)
 | |
| {
 | |
|     Py_hash_t hash;
 | |
|     PyDictObject *mp = (PyDictObject *)op;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyThreadState *tstate;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyDict_Check(op))
 | |
|         return NULL;
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1)
 | |
|     {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1) {
 | |
|             PyErr_Clear();
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     /* We can arrive here with a NULL tstate during initialization: try
 | |
|        running "python -Wi" for an example related to string interning.
 | |
|        Let's just hope that no exception occurs then...  This must be
 | |
|        _PyThreadState_Current and not PyThreadState_GET() because in debug
 | |
|        mode, the latter complains if tstate is NULL. */
 | |
|     tstate = (PyThreadState*)_Py_atomic_load_relaxed(
 | |
|         &_PyThreadState_Current);
 | |
|     if (tstate != NULL && tstate->curexc_type != NULL) {
 | |
|         /* preserve the existing exception */
 | |
|         PyObject *err_type, *err_value, *err_tb;
 | |
|         PyErr_Fetch(&err_type, &err_value, &err_tb);
 | |
|         ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|         /* ignore errors */
 | |
|         PyErr_Restore(err_type, err_value, err_tb);
 | |
|         if (ep == NULL)
 | |
|             return NULL;
 | |
|     }
 | |
|     else {
 | |
|         ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|         if (ep == NULL) {
 | |
|             PyErr_Clear();
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
|     return *value_addr;
 | |
| }
 | |
| 
 | |
| /* Variant of PyDict_GetItem() that doesn't suppress exceptions.
 | |
|    This returns NULL *with* an exception set if an exception occurred.
 | |
|    It returns NULL *without* an exception set if the key wasn't present.
 | |
| */
 | |
| PyObject *
 | |
| PyDict_GetItemWithError(PyObject *op, PyObject *key)
 | |
| {
 | |
|     Py_hash_t hash;
 | |
|     PyDictObject*mp = (PyDictObject *)op;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyDict_Check(op)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1)
 | |
|     {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1) {
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     return *value_addr;
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| _PyDict_GetItemIdWithError(PyObject *dp, struct _Py_Identifier *key)
 | |
| {
 | |
|     PyObject *kv;
 | |
|     kv = _PyUnicode_FromId(key); /* borrowed */
 | |
|     if (kv == NULL)
 | |
|         return NULL;
 | |
|     return PyDict_GetItemWithError(dp, kv);
 | |
| }
 | |
| 
 | |
| /* Fast version of global value lookup.
 | |
|  * Lookup in globals, then builtins.
 | |
|  */
 | |
| PyObject *
 | |
| _PyDict_LoadGlobal(PyDictObject *globals, PyDictObject *builtins, PyObject *key)
 | |
| {
 | |
|     PyObject *x;
 | |
|     if (PyUnicode_CheckExact(key)) {
 | |
|         PyObject **value_addr;
 | |
|         Py_hash_t hash = ((PyASCIIObject *)key)->hash;
 | |
|         if (hash != -1) {
 | |
|             PyDictKeyEntry *e;
 | |
|             e = globals->ma_keys->dk_lookup(globals, key, hash, &value_addr);
 | |
|             if (e == NULL) {
 | |
|                 return NULL;
 | |
|             }
 | |
|             x = *value_addr;
 | |
|             if (x != NULL)
 | |
|                 return x;
 | |
|             e = builtins->ma_keys->dk_lookup(builtins, key, hash, &value_addr);
 | |
|             if (e == NULL) {
 | |
|                 return NULL;
 | |
|             }
 | |
|             x = *value_addr;
 | |
|             return x;
 | |
|         }
 | |
|     }
 | |
|     x = PyDict_GetItemWithError((PyObject *)globals, key);
 | |
|     if (x != NULL)
 | |
|         return x;
 | |
|     if (PyErr_Occurred())
 | |
|         return NULL;
 | |
|     return PyDict_GetItemWithError((PyObject *)builtins, key);
 | |
| }
 | |
| 
 | |
| /* CAUTION: PyDict_SetItem() must guarantee that it won't resize the
 | |
|  * dictionary if it's merely replacing the value for an existing key.
 | |
|  * This means that it's safe to loop over a dictionary with PyDict_Next()
 | |
|  * and occasionally replace a value -- but you can't insert new keys or
 | |
|  * remove them.
 | |
|  */
 | |
| int
 | |
| PyDict_SetItem(PyObject *op, PyObject *key, PyObject *value)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     Py_hash_t hash;
 | |
|     if (!PyDict_Check(op)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return -1;
 | |
|     }
 | |
|     assert(key);
 | |
|     assert(value);
 | |
|     mp = (PyDictObject *)op;
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1)
 | |
|     {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return -1;
 | |
|     }
 | |
| 
 | |
|     /* insertdict() handles any resizing that might be necessary */
 | |
|     return insertdict(mp, key, hash, value);
 | |
| }
 | |
| 
 | |
| int
 | |
| PyDict_DelItem(PyObject *op, PyObject *key)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     Py_hash_t hash;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject *old_key, *old_value;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyDict_Check(op)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return -1;
 | |
|     }
 | |
|     assert(key);
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return -1;
 | |
|     }
 | |
|     mp = (PyDictObject *)op;
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return -1;
 | |
|     if (*value_addr == NULL) {
 | |
|         _PyErr_SetKeyError(key);
 | |
|         return -1;
 | |
|     }
 | |
|     old_value = *value_addr;
 | |
|     *value_addr = NULL;
 | |
|     mp->ma_used--;
 | |
|     if (!_PyDict_HasSplitTable(mp)) {
 | |
|         ENSURE_ALLOWS_DELETIONS(mp);
 | |
|         old_key = ep->me_key;
 | |
|         Py_INCREF(dummy);
 | |
|         ep->me_key = dummy;
 | |
|         Py_DECREF(old_key);
 | |
|     }
 | |
|     Py_DECREF(old_value);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| void
 | |
| PyDict_Clear(PyObject *op)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     PyDictKeysObject *oldkeys;
 | |
|     PyObject **oldvalues;
 | |
|     Py_ssize_t i, n;
 | |
| 
 | |
|     if (!PyDict_Check(op))
 | |
|         return;
 | |
|     mp = ((PyDictObject *)op);
 | |
|     oldkeys = mp->ma_keys;
 | |
|     oldvalues = mp->ma_values;
 | |
|     if (oldvalues == empty_values)
 | |
|         return;
 | |
|     /* Empty the dict... */
 | |
|     DK_INCREF(Py_EMPTY_KEYS);
 | |
|     mp->ma_keys = Py_EMPTY_KEYS;
 | |
|     mp->ma_values = empty_values;
 | |
|     mp->ma_used = 0;
 | |
|     /* ...then clear the keys and values */
 | |
|     if (oldvalues != NULL) {
 | |
|         n = DK_SIZE(oldkeys);
 | |
|         for (i = 0; i < n; i++)
 | |
|             Py_CLEAR(oldvalues[i]);
 | |
|         free_values(oldvalues);
 | |
|         DK_DECREF(oldkeys);
 | |
|     }
 | |
|     else {
 | |
|        assert(oldkeys->dk_refcnt == 1);
 | |
|        DK_DECREF(oldkeys);
 | |
|     }
 | |
| }
 | |
| 
 | |
| /* Returns -1 if no more items (or op is not a dict),
 | |
|  * index of item otherwise. Stores value in pvalue
 | |
|  */
 | |
| Py_LOCAL_INLINE(Py_ssize_t)
 | |
| dict_next(PyObject *op, Py_ssize_t i, PyObject **pvalue)
 | |
| {
 | |
|     Py_ssize_t mask, offset;
 | |
|     PyDictObject *mp;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
| 
 | |
|     if (!PyDict_Check(op))
 | |
|         return -1;
 | |
|     mp = (PyDictObject *)op;
 | |
|     if (i < 0)
 | |
|         return -1;
 | |
|     if (mp->ma_values) {
 | |
|         value_ptr = &mp->ma_values[i];
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &mp->ma_keys->dk_entries[i].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     mask = DK_MASK(mp->ma_keys);
 | |
|     while (i <= mask && *value_ptr == NULL) {
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         i++;
 | |
|     }
 | |
|     if (i > mask)
 | |
|         return -1;
 | |
|     if (pvalue)
 | |
|         *pvalue = *value_ptr;
 | |
|     return i;
 | |
| }
 | |
| 
 | |
| /*
 | |
|  * Iterate over a dict.  Use like so:
 | |
|  *
 | |
|  *     Py_ssize_t i;
 | |
|  *     PyObject *key, *value;
 | |
|  *     i = 0;   # important!  i should not otherwise be changed by you
 | |
|  *     while (PyDict_Next(yourdict, &i, &key, &value)) {
 | |
|  *              Refer to borrowed references in key and value.
 | |
|  *     }
 | |
|  *
 | |
|  * CAUTION:  In general, it isn't safe to use PyDict_Next in a loop that
 | |
|  * mutates the dict.  One exception:  it is safe if the loop merely changes
 | |
|  * the values associated with the keys (but doesn't insert new keys or
 | |
|  * delete keys), via PyDict_SetItem().
 | |
|  */
 | |
| int
 | |
| PyDict_Next(PyObject *op, Py_ssize_t *ppos, PyObject **pkey, PyObject **pvalue)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     Py_ssize_t i = dict_next(op, *ppos, pvalue);
 | |
|     if (i < 0)
 | |
|         return 0;
 | |
|     mp = (PyDictObject *)op;
 | |
|     *ppos = i+1;
 | |
|     if (pkey)
 | |
|         *pkey = mp->ma_keys->dk_entries[i].me_key;
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| /* Internal version of PyDict_Next that returns a hash value in addition
 | |
|  * to the key and value.
 | |
|  */
 | |
| int
 | |
| _PyDict_Next(PyObject *op, Py_ssize_t *ppos, PyObject **pkey,
 | |
|              PyObject **pvalue, Py_hash_t *phash)
 | |
| {
 | |
|     PyDictObject *mp;
 | |
|     Py_ssize_t i = dict_next(op, *ppos, pvalue);
 | |
|     if (i < 0)
 | |
|         return 0;
 | |
|     mp = (PyDictObject *)op;
 | |
|     *ppos = i+1;
 | |
|     *phash = mp->ma_keys->dk_entries[i].me_hash;
 | |
|     if (pkey)
 | |
|         *pkey = mp->ma_keys->dk_entries[i].me_key;
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| /* Methods */
 | |
| 
 | |
| static void
 | |
| dict_dealloc(PyDictObject *mp)
 | |
| {
 | |
|     PyObject **values = mp->ma_values;
 | |
|     PyDictKeysObject *keys = mp->ma_keys;
 | |
|     Py_ssize_t i, n;
 | |
|     PyObject_GC_UnTrack(mp);
 | |
|     Py_TRASHCAN_SAFE_BEGIN(mp)
 | |
|     if (values != NULL) {
 | |
|         if (values != empty_values) {
 | |
|             for (i = 0, n = DK_SIZE(mp->ma_keys); i < n; i++) {
 | |
|                 Py_XDECREF(values[i]);
 | |
|             }
 | |
|             free_values(values);
 | |
|         }
 | |
|         DK_DECREF(keys);
 | |
|     }
 | |
|     else if (keys != NULL) {
 | |
|         assert(keys->dk_refcnt == 1);
 | |
|         DK_DECREF(keys);
 | |
|     }
 | |
|     if (numfree < PyDict_MAXFREELIST && Py_TYPE(mp) == &PyDict_Type)
 | |
|         free_list[numfree++] = mp;
 | |
|     else
 | |
|         Py_TYPE(mp)->tp_free((PyObject *)mp);
 | |
|     Py_TRASHCAN_SAFE_END(mp)
 | |
| }
 | |
| 
 | |
| 
 | |
| static PyObject *
 | |
| dict_repr(PyDictObject *mp)
 | |
| {
 | |
|     Py_ssize_t i;
 | |
|     PyObject *key = NULL, *value = NULL;
 | |
|     _PyUnicodeWriter writer;
 | |
|     int first;
 | |
| 
 | |
|     i = Py_ReprEnter((PyObject *)mp);
 | |
|     if (i != 0) {
 | |
|         return i > 0 ? PyUnicode_FromString("{...}") : NULL;
 | |
|     }
 | |
| 
 | |
|     if (mp->ma_used == 0) {
 | |
|         Py_ReprLeave((PyObject *)mp);
 | |
|         return PyUnicode_FromString("{}");
 | |
|     }
 | |
| 
 | |
|     _PyUnicodeWriter_Init(&writer);
 | |
|     writer.overallocate = 1;
 | |
|     /* "{" + "1: 2" + ", 3: 4" * (len - 1) + "}" */
 | |
|     writer.min_length = 1 + 4 + (2 + 4) * (mp->ma_used - 1) + 1;
 | |
| 
 | |
|     if (_PyUnicodeWriter_WriteChar(&writer, '{') < 0)
 | |
|         goto error;
 | |
| 
 | |
|     /* Do repr() on each key+value pair, and insert ": " between them.
 | |
|        Note that repr may mutate the dict. */
 | |
|     i = 0;
 | |
|     first = 1;
 | |
|     while (PyDict_Next((PyObject *)mp, &i, &key, &value)) {
 | |
|         PyObject *s;
 | |
|         int res;
 | |
| 
 | |
|         /* Prevent repr from deleting key or value during key format. */
 | |
|         Py_INCREF(key);
 | |
|         Py_INCREF(value);
 | |
| 
 | |
|         if (!first) {
 | |
|             if (_PyUnicodeWriter_WriteASCIIString(&writer, ", ", 2) < 0)
 | |
|                 goto error;
 | |
|         }
 | |
|         first = 0;
 | |
| 
 | |
|         s = PyObject_Repr(key);
 | |
|         if (s == NULL)
 | |
|             goto error;
 | |
|         res = _PyUnicodeWriter_WriteStr(&writer, s);
 | |
|         Py_DECREF(s);
 | |
|         if (res < 0)
 | |
|             goto error;
 | |
| 
 | |
|         if (_PyUnicodeWriter_WriteASCIIString(&writer, ": ", 2) < 0)
 | |
|             goto error;
 | |
| 
 | |
|         s = PyObject_Repr(value);
 | |
|         if (s == NULL)
 | |
|             goto error;
 | |
|         res = _PyUnicodeWriter_WriteStr(&writer, s);
 | |
|         Py_DECREF(s);
 | |
|         if (res < 0)
 | |
|             goto error;
 | |
| 
 | |
|         Py_CLEAR(key);
 | |
|         Py_CLEAR(value);
 | |
|     }
 | |
| 
 | |
|     writer.overallocate = 0;
 | |
|     if (_PyUnicodeWriter_WriteChar(&writer, '}') < 0)
 | |
|         goto error;
 | |
| 
 | |
|     Py_ReprLeave((PyObject *)mp);
 | |
| 
 | |
|     return _PyUnicodeWriter_Finish(&writer);
 | |
| 
 | |
| error:
 | |
|     Py_ReprLeave((PyObject *)mp);
 | |
|     _PyUnicodeWriter_Dealloc(&writer);
 | |
|     Py_XDECREF(key);
 | |
|     Py_XDECREF(value);
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| static Py_ssize_t
 | |
| dict_length(PyDictObject *mp)
 | |
| {
 | |
|     return mp->ma_used;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_subscript(PyDictObject *mp, PyObject *key)
 | |
| {
 | |
|     PyObject *v;
 | |
|     Py_hash_t hash;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return NULL;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     v = *value_addr;
 | |
|     if (v == NULL) {
 | |
|         if (!PyDict_CheckExact(mp)) {
 | |
|             /* Look up __missing__ method if we're a subclass. */
 | |
|             PyObject *missing, *res;
 | |
|             _Py_IDENTIFIER(__missing__);
 | |
|             missing = _PyObject_LookupSpecial((PyObject *)mp, &PyId___missing__);
 | |
|             if (missing != NULL) {
 | |
|                 res = PyObject_CallFunctionObjArgs(missing,
 | |
|                                                    key, NULL);
 | |
|                 Py_DECREF(missing);
 | |
|                 return res;
 | |
|             }
 | |
|             else if (PyErr_Occurred())
 | |
|                 return NULL;
 | |
|         }
 | |
|         _PyErr_SetKeyError(key);
 | |
|         return NULL;
 | |
|     }
 | |
|     else
 | |
|         Py_INCREF(v);
 | |
|     return v;
 | |
| }
 | |
| 
 | |
| static int
 | |
| dict_ass_sub(PyDictObject *mp, PyObject *v, PyObject *w)
 | |
| {
 | |
|     if (w == NULL)
 | |
|         return PyDict_DelItem((PyObject *)mp, v);
 | |
|     else
 | |
|         return PyDict_SetItem((PyObject *)mp, v, w);
 | |
| }
 | |
| 
 | |
| static PyMappingMethods dict_as_mapping = {
 | |
|     (lenfunc)dict_length, /*mp_length*/
 | |
|     (binaryfunc)dict_subscript, /*mp_subscript*/
 | |
|     (objobjargproc)dict_ass_sub, /*mp_ass_subscript*/
 | |
| };
 | |
| 
 | |
| static PyObject *
 | |
| dict_keys(PyDictObject *mp)
 | |
| {
 | |
|     PyObject *v;
 | |
|     Py_ssize_t i, j;
 | |
|     PyDictKeyEntry *ep;
 | |
|     Py_ssize_t size, n, offset;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|   again:
 | |
|     n = mp->ma_used;
 | |
|     v = PyList_New(n);
 | |
|     if (v == NULL)
 | |
|         return NULL;
 | |
|     if (n != mp->ma_used) {
 | |
|         /* Durnit.  The allocations caused the dict to resize.
 | |
|          * Just start over, this shouldn't normally happen.
 | |
|          */
 | |
|         Py_DECREF(v);
 | |
|         goto again;
 | |
|     }
 | |
|     ep = &mp->ma_keys->dk_entries[0];
 | |
|     size = DK_SIZE(mp->ma_keys);
 | |
|     if (mp->ma_values) {
 | |
|         value_ptr = mp->ma_values;
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &ep[0].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     for (i = 0, j = 0; i < size; i++) {
 | |
|         if (*value_ptr != NULL) {
 | |
|             PyObject *key = ep[i].me_key;
 | |
|             Py_INCREF(key);
 | |
|             PyList_SET_ITEM(v, j, key);
 | |
|             j++;
 | |
|         }
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|     }
 | |
|     assert(j == n);
 | |
|     return v;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_values(PyDictObject *mp)
 | |
| {
 | |
|     PyObject *v;
 | |
|     Py_ssize_t i, j;
 | |
|     Py_ssize_t size, n, offset;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|   again:
 | |
|     n = mp->ma_used;
 | |
|     v = PyList_New(n);
 | |
|     if (v == NULL)
 | |
|         return NULL;
 | |
|     if (n != mp->ma_used) {
 | |
|         /* Durnit.  The allocations caused the dict to resize.
 | |
|          * Just start over, this shouldn't normally happen.
 | |
|          */
 | |
|         Py_DECREF(v);
 | |
|         goto again;
 | |
|     }
 | |
|     size = DK_SIZE(mp->ma_keys);
 | |
|     if (mp->ma_values) {
 | |
|         value_ptr = mp->ma_values;
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &mp->ma_keys->dk_entries[0].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     for (i = 0, j = 0; i < size; i++) {
 | |
|         PyObject *value = *value_ptr;
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         if (value != NULL) {
 | |
|             Py_INCREF(value);
 | |
|             PyList_SET_ITEM(v, j, value);
 | |
|             j++;
 | |
|         }
 | |
|     }
 | |
|     assert(j == n);
 | |
|     return v;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_items(PyDictObject *mp)
 | |
| {
 | |
|     PyObject *v;
 | |
|     Py_ssize_t i, j, n;
 | |
|     Py_ssize_t size, offset;
 | |
|     PyObject *item, *key;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|     /* Preallocate the list of tuples, to avoid allocations during
 | |
|      * the loop over the items, which could trigger GC, which
 | |
|      * could resize the dict. :-(
 | |
|      */
 | |
|   again:
 | |
|     n = mp->ma_used;
 | |
|     v = PyList_New(n);
 | |
|     if (v == NULL)
 | |
|         return NULL;
 | |
|     for (i = 0; i < n; i++) {
 | |
|         item = PyTuple_New(2);
 | |
|         if (item == NULL) {
 | |
|             Py_DECREF(v);
 | |
|             return NULL;
 | |
|         }
 | |
|         PyList_SET_ITEM(v, i, item);
 | |
|     }
 | |
|     if (n != mp->ma_used) {
 | |
|         /* Durnit.  The allocations caused the dict to resize.
 | |
|          * Just start over, this shouldn't normally happen.
 | |
|          */
 | |
|         Py_DECREF(v);
 | |
|         goto again;
 | |
|     }
 | |
|     /* Nothing we do below makes any function calls. */
 | |
|     ep = mp->ma_keys->dk_entries;
 | |
|     size = DK_SIZE(mp->ma_keys);
 | |
|     if (mp->ma_values) {
 | |
|         value_ptr = mp->ma_values;
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &ep[0].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     for (i = 0, j = 0; i < size; i++) {
 | |
|         PyObject *value = *value_ptr;
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         if (value != NULL) {
 | |
|             key = ep[i].me_key;
 | |
|             item = PyList_GET_ITEM(v, j);
 | |
|             Py_INCREF(key);
 | |
|             PyTuple_SET_ITEM(item, 0, key);
 | |
|             Py_INCREF(value);
 | |
|             PyTuple_SET_ITEM(item, 1, value);
 | |
|             j++;
 | |
|         }
 | |
|     }
 | |
|     assert(j == n);
 | |
|     return v;
 | |
| }
 | |
| 
 | |
| /*[clinic input]
 | |
| @classmethod
 | |
| dict.fromkeys
 | |
|     iterable: object
 | |
|     value: object=None
 | |
|     /
 | |
| 
 | |
| Returns a new dict with keys from iterable and values equal to value.
 | |
| [clinic start generated code]*/
 | |
| 
 | |
| PyDoc_STRVAR(dict_fromkeys__doc__,
 | |
| "fromkeys($type, iterable, value=None, /)\n"
 | |
| "--\n"
 | |
| "\n"
 | |
| "Returns a new dict with keys from iterable and values equal to value.");
 | |
| 
 | |
| #define DICT_FROMKEYS_METHODDEF    \
 | |
|     {"fromkeys", (PyCFunction)dict_fromkeys, METH_VARARGS|METH_CLASS, dict_fromkeys__doc__},
 | |
| 
 | |
| static PyObject *
 | |
| dict_fromkeys_impl(PyTypeObject *type, PyObject *iterable, PyObject *value);
 | |
| 
 | |
| static PyObject *
 | |
| dict_fromkeys(PyTypeObject *type, PyObject *args)
 | |
| {
 | |
|     PyObject *return_value = NULL;
 | |
|     PyObject *iterable;
 | |
|     PyObject *value = Py_None;
 | |
| 
 | |
|     if (!PyArg_UnpackTuple(args, "fromkeys",
 | |
|         1, 2,
 | |
|         &iterable, &value))
 | |
|         goto exit;
 | |
|     return_value = dict_fromkeys_impl(type, iterable, value);
 | |
| 
 | |
| exit:
 | |
|     return return_value;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_fromkeys_impl(PyTypeObject *type, PyObject *iterable, PyObject *value)
 | |
| /*[clinic end generated code: output=55f8dc0ffa87406f input=b85a667f9bf4669d]*/
 | |
| {
 | |
|     PyObject *it;       /* iter(seq) */
 | |
|     PyObject *key;
 | |
|     PyObject *d;
 | |
|     int status;
 | |
| 
 | |
|     d = PyObject_CallObject((PyObject *)type, NULL);
 | |
|     if (d == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     if (PyDict_CheckExact(d) && ((PyDictObject *)d)->ma_used == 0) {
 | |
|         if (PyDict_CheckExact(iterable)) {
 | |
|             PyDictObject *mp = (PyDictObject *)d;
 | |
|             PyObject *oldvalue;
 | |
|             Py_ssize_t pos = 0;
 | |
|             PyObject *key;
 | |
|             Py_hash_t hash;
 | |
| 
 | |
|             if (dictresize(mp, Py_SIZE(iterable))) {
 | |
|                 Py_DECREF(d);
 | |
|                 return NULL;
 | |
|             }
 | |
| 
 | |
|             while (_PyDict_Next(iterable, &pos, &key, &oldvalue, &hash)) {
 | |
|                 if (insertdict(mp, key, hash, value)) {
 | |
|                     Py_DECREF(d);
 | |
|                     return NULL;
 | |
|                 }
 | |
|             }
 | |
|             return d;
 | |
|         }
 | |
|         if (PyAnySet_CheckExact(iterable)) {
 | |
|             PyDictObject *mp = (PyDictObject *)d;
 | |
|             Py_ssize_t pos = 0;
 | |
|             PyObject *key;
 | |
|             Py_hash_t hash;
 | |
| 
 | |
|             if (dictresize(mp, PySet_GET_SIZE(iterable))) {
 | |
|                 Py_DECREF(d);
 | |
|                 return NULL;
 | |
|             }
 | |
| 
 | |
|             while (_PySet_NextEntry(iterable, &pos, &key, &hash)) {
 | |
|                 if (insertdict(mp, key, hash, value)) {
 | |
|                     Py_DECREF(d);
 | |
|                     return NULL;
 | |
|                 }
 | |
|             }
 | |
|             return d;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     it = PyObject_GetIter(iterable);
 | |
|     if (it == NULL){
 | |
|         Py_DECREF(d);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     if (PyDict_CheckExact(d)) {
 | |
|         while ((key = PyIter_Next(it)) != NULL) {
 | |
|             status = PyDict_SetItem(d, key, value);
 | |
|             Py_DECREF(key);
 | |
|             if (status < 0)
 | |
|                 goto Fail;
 | |
|         }
 | |
|     } else {
 | |
|         while ((key = PyIter_Next(it)) != NULL) {
 | |
|             status = PyObject_SetItem(d, key, value);
 | |
|             Py_DECREF(key);
 | |
|             if (status < 0)
 | |
|                 goto Fail;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     if (PyErr_Occurred())
 | |
|         goto Fail;
 | |
|     Py_DECREF(it);
 | |
|     return d;
 | |
| 
 | |
| Fail:
 | |
|     Py_DECREF(it);
 | |
|     Py_DECREF(d);
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| static int
 | |
| dict_update_common(PyObject *self, PyObject *args, PyObject *kwds, char *methname)
 | |
| {
 | |
|     PyObject *arg = NULL;
 | |
|     int result = 0;
 | |
| 
 | |
|     if (!PyArg_UnpackTuple(args, methname, 0, 1, &arg))
 | |
|         result = -1;
 | |
| 
 | |
|     else if (arg != NULL) {
 | |
|         _Py_IDENTIFIER(keys);
 | |
|         if (_PyObject_HasAttrId(arg, &PyId_keys))
 | |
|             result = PyDict_Merge(self, arg, 1);
 | |
|         else
 | |
|             result = PyDict_MergeFromSeq2(self, arg, 1);
 | |
|     }
 | |
|     if (result == 0 && kwds != NULL) {
 | |
|         if (PyArg_ValidateKeywordArguments(kwds))
 | |
|             result = PyDict_Merge(self, kwds, 1);
 | |
|         else
 | |
|             result = -1;
 | |
|     }
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_update(PyObject *self, PyObject *args, PyObject *kwds)
 | |
| {
 | |
|     if (dict_update_common(self, args, kwds, "update") != -1)
 | |
|         Py_RETURN_NONE;
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| /* Update unconditionally replaces existing items.
 | |
|    Merge has a 3rd argument 'override'; if set, it acts like Update,
 | |
|    otherwise it leaves existing items unchanged.
 | |
| 
 | |
|    PyDict_{Update,Merge} update/merge from a mapping object.
 | |
| 
 | |
|    PyDict_MergeFromSeq2 updates/merges from any iterable object
 | |
|    producing iterable objects of length 2.
 | |
| */
 | |
| 
 | |
| int
 | |
| PyDict_MergeFromSeq2(PyObject *d, PyObject *seq2, int override)
 | |
| {
 | |
|     PyObject *it;       /* iter(seq2) */
 | |
|     Py_ssize_t i;       /* index into seq2 of current element */
 | |
|     PyObject *item;     /* seq2[i] */
 | |
|     PyObject *fast;     /* item as a 2-tuple or 2-list */
 | |
| 
 | |
|     assert(d != NULL);
 | |
|     assert(PyDict_Check(d));
 | |
|     assert(seq2 != NULL);
 | |
| 
 | |
|     it = PyObject_GetIter(seq2);
 | |
|     if (it == NULL)
 | |
|         return -1;
 | |
| 
 | |
|     for (i = 0; ; ++i) {
 | |
|         PyObject *key, *value;
 | |
|         Py_ssize_t n;
 | |
| 
 | |
|         fast = NULL;
 | |
|         item = PyIter_Next(it);
 | |
|         if (item == NULL) {
 | |
|             if (PyErr_Occurred())
 | |
|                 goto Fail;
 | |
|             break;
 | |
|         }
 | |
| 
 | |
|         /* Convert item to sequence, and verify length 2. */
 | |
|         fast = PySequence_Fast(item, "");
 | |
|         if (fast == NULL) {
 | |
|             if (PyErr_ExceptionMatches(PyExc_TypeError))
 | |
|                 PyErr_Format(PyExc_TypeError,
 | |
|                     "cannot convert dictionary update "
 | |
|                     "sequence element #%zd to a sequence",
 | |
|                     i);
 | |
|             goto Fail;
 | |
|         }
 | |
|         n = PySequence_Fast_GET_SIZE(fast);
 | |
|         if (n != 2) {
 | |
|             PyErr_Format(PyExc_ValueError,
 | |
|                          "dictionary update sequence element #%zd "
 | |
|                          "has length %zd; 2 is required",
 | |
|                          i, n);
 | |
|             goto Fail;
 | |
|         }
 | |
| 
 | |
|         /* Update/merge with this (key, value) pair. */
 | |
|         key = PySequence_Fast_GET_ITEM(fast, 0);
 | |
|         value = PySequence_Fast_GET_ITEM(fast, 1);
 | |
|         if (override || PyDict_GetItem(d, key) == NULL) {
 | |
|             int status = PyDict_SetItem(d, key, value);
 | |
|             if (status < 0)
 | |
|                 goto Fail;
 | |
|         }
 | |
|         Py_DECREF(fast);
 | |
|         Py_DECREF(item);
 | |
|     }
 | |
| 
 | |
|     i = 0;
 | |
|     goto Return;
 | |
| Fail:
 | |
|     Py_XDECREF(item);
 | |
|     Py_XDECREF(fast);
 | |
|     i = -1;
 | |
| Return:
 | |
|     Py_DECREF(it);
 | |
|     return Py_SAFE_DOWNCAST(i, Py_ssize_t, int);
 | |
| }
 | |
| 
 | |
| int
 | |
| PyDict_Update(PyObject *a, PyObject *b)
 | |
| {
 | |
|     return PyDict_Merge(a, b, 1);
 | |
| }
 | |
| 
 | |
| int
 | |
| PyDict_Merge(PyObject *a, PyObject *b, int override)
 | |
| {
 | |
|     PyDictObject *mp, *other;
 | |
|     Py_ssize_t i, n;
 | |
|     PyDictKeyEntry *entry;
 | |
| 
 | |
|     /* We accept for the argument either a concrete dictionary object,
 | |
|      * or an abstract "mapping" object.  For the former, we can do
 | |
|      * things quite efficiently.  For the latter, we only require that
 | |
|      * PyMapping_Keys() and PyObject_GetItem() be supported.
 | |
|      */
 | |
|     if (a == NULL || !PyDict_Check(a) || b == NULL) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return -1;
 | |
|     }
 | |
|     mp = (PyDictObject*)a;
 | |
|     if (PyDict_Check(b)) {
 | |
|         other = (PyDictObject*)b;
 | |
|         if (other == mp || other->ma_used == 0)
 | |
|             /* a.update(a) or a.update({}); nothing to do */
 | |
|             return 0;
 | |
|         if (mp->ma_used == 0)
 | |
|             /* Since the target dict is empty, PyDict_GetItem()
 | |
|              * always returns NULL.  Setting override to 1
 | |
|              * skips the unnecessary test.
 | |
|              */
 | |
|             override = 1;
 | |
|         /* Do one big resize at the start, rather than
 | |
|          * incrementally resizing as we insert new items.  Expect
 | |
|          * that there will be no (or few) overlapping keys.
 | |
|          */
 | |
|         if (mp->ma_keys->dk_usable * 3 < other->ma_used * 2)
 | |
|             if (dictresize(mp, (mp->ma_used + other->ma_used)*2) != 0)
 | |
|                return -1;
 | |
|         for (i = 0, n = DK_SIZE(other->ma_keys); i < n; i++) {
 | |
|             PyObject *key, *value;
 | |
|             Py_hash_t hash;
 | |
|             entry = &other->ma_keys->dk_entries[i];
 | |
|             key = entry->me_key;
 | |
|             hash = entry->me_hash;
 | |
|             if (other->ma_values)
 | |
|                 value = other->ma_values[i];
 | |
|             else
 | |
|                 value = entry->me_value;
 | |
| 
 | |
|             if (value != NULL) {
 | |
|                 int err = 0;
 | |
|                 Py_INCREF(key);
 | |
|                 Py_INCREF(value);
 | |
|                 if (override || PyDict_GetItem(a, key) == NULL)
 | |
|                     err = insertdict(mp, key, hash, value);
 | |
|                 Py_DECREF(value);
 | |
|                 Py_DECREF(key);
 | |
|                 if (err != 0)
 | |
|                     return -1;
 | |
| 
 | |
|                 if (n != DK_SIZE(other->ma_keys)) {
 | |
|                     PyErr_SetString(PyExc_RuntimeError,
 | |
|                                     "dict mutated during update");
 | |
|                     return -1;
 | |
|                 }
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     else {
 | |
|         /* Do it the generic, slower way */
 | |
|         PyObject *keys = PyMapping_Keys(b);
 | |
|         PyObject *iter;
 | |
|         PyObject *key, *value;
 | |
|         int status;
 | |
| 
 | |
|         if (keys == NULL)
 | |
|             /* Docstring says this is equivalent to E.keys() so
 | |
|              * if E doesn't have a .keys() method we want
 | |
|              * AttributeError to percolate up.  Might as well
 | |
|              * do the same for any other error.
 | |
|              */
 | |
|             return -1;
 | |
| 
 | |
|         iter = PyObject_GetIter(keys);
 | |
|         Py_DECREF(keys);
 | |
|         if (iter == NULL)
 | |
|             return -1;
 | |
| 
 | |
|         for (key = PyIter_Next(iter); key; key = PyIter_Next(iter)) {
 | |
|             if (!override && PyDict_GetItem(a, key) != NULL) {
 | |
|                 Py_DECREF(key);
 | |
|                 continue;
 | |
|             }
 | |
|             value = PyObject_GetItem(b, key);
 | |
|             if (value == NULL) {
 | |
|                 Py_DECREF(iter);
 | |
|                 Py_DECREF(key);
 | |
|                 return -1;
 | |
|             }
 | |
|             status = PyDict_SetItem(a, key, value);
 | |
|             Py_DECREF(key);
 | |
|             Py_DECREF(value);
 | |
|             if (status < 0) {
 | |
|                 Py_DECREF(iter);
 | |
|                 return -1;
 | |
|             }
 | |
|         }
 | |
|         Py_DECREF(iter);
 | |
|         if (PyErr_Occurred())
 | |
|             /* Iterator completed, via error */
 | |
|             return -1;
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_copy(PyDictObject *mp)
 | |
| {
 | |
|     return PyDict_Copy((PyObject*)mp);
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_Copy(PyObject *o)
 | |
| {
 | |
|     PyObject *copy;
 | |
|     PyDictObject *mp;
 | |
|     Py_ssize_t i, n;
 | |
| 
 | |
|     if (o == NULL || !PyDict_Check(o)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     mp = (PyDictObject *)o;
 | |
|     if (_PyDict_HasSplitTable(mp)) {
 | |
|         PyDictObject *split_copy;
 | |
|         PyObject **newvalues = new_values(DK_SIZE(mp->ma_keys));
 | |
|         if (newvalues == NULL)
 | |
|             return PyErr_NoMemory();
 | |
|         split_copy = PyObject_GC_New(PyDictObject, &PyDict_Type);
 | |
|         if (split_copy == NULL) {
 | |
|             free_values(newvalues);
 | |
|             return NULL;
 | |
|         }
 | |
|         split_copy->ma_values = newvalues;
 | |
|         split_copy->ma_keys = mp->ma_keys;
 | |
|         split_copy->ma_used = mp->ma_used;
 | |
|         DK_INCREF(mp->ma_keys);
 | |
|         for (i = 0, n = DK_SIZE(mp->ma_keys); i < n; i++) {
 | |
|             PyObject *value = mp->ma_values[i];
 | |
|             Py_XINCREF(value);
 | |
|             split_copy->ma_values[i] = value;
 | |
|         }
 | |
|         if (_PyObject_GC_IS_TRACKED(mp))
 | |
|             _PyObject_GC_TRACK(split_copy);
 | |
|         return (PyObject *)split_copy;
 | |
|     }
 | |
|     copy = PyDict_New();
 | |
|     if (copy == NULL)
 | |
|         return NULL;
 | |
|     if (PyDict_Merge(copy, o, 1) == 0)
 | |
|         return copy;
 | |
|     Py_DECREF(copy);
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| Py_ssize_t
 | |
| PyDict_Size(PyObject *mp)
 | |
| {
 | |
|     if (mp == NULL || !PyDict_Check(mp)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return -1;
 | |
|     }
 | |
|     return ((PyDictObject *)mp)->ma_used;
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_Keys(PyObject *mp)
 | |
| {
 | |
|     if (mp == NULL || !PyDict_Check(mp)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     return dict_keys((PyDictObject *)mp);
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_Values(PyObject *mp)
 | |
| {
 | |
|     if (mp == NULL || !PyDict_Check(mp)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     return dict_values((PyDictObject *)mp);
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_Items(PyObject *mp)
 | |
| {
 | |
|     if (mp == NULL || !PyDict_Check(mp)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     return dict_items((PyDictObject *)mp);
 | |
| }
 | |
| 
 | |
| /* Return 1 if dicts equal, 0 if not, -1 if error.
 | |
|  * Gets out as soon as any difference is detected.
 | |
|  * Uses only Py_EQ comparison.
 | |
|  */
 | |
| static int
 | |
| dict_equal(PyDictObject *a, PyDictObject *b)
 | |
| {
 | |
|     Py_ssize_t i;
 | |
| 
 | |
|     if (a->ma_used != b->ma_used)
 | |
|         /* can't be equal if # of entries differ */
 | |
|         return 0;
 | |
|     /* Same # of entries -- check all of 'em.  Exit early on any diff. */
 | |
|     for (i = 0; i < DK_SIZE(a->ma_keys); i++) {
 | |
|         PyDictKeyEntry *ep = &a->ma_keys->dk_entries[i];
 | |
|         PyObject *aval;
 | |
|         if (a->ma_values)
 | |
|             aval = a->ma_values[i];
 | |
|         else
 | |
|             aval = ep->me_value;
 | |
|         if (aval != NULL) {
 | |
|             int cmp;
 | |
|             PyObject *bval;
 | |
|             PyObject **vaddr;
 | |
|             PyObject *key = ep->me_key;
 | |
|             /* temporarily bump aval's refcount to ensure it stays
 | |
|                alive until we're done with it */
 | |
|             Py_INCREF(aval);
 | |
|             /* ditto for key */
 | |
|             Py_INCREF(key);
 | |
|             /* reuse the known hash value */
 | |
|             if ((b->ma_keys->dk_lookup)(b, key, ep->me_hash, &vaddr) == NULL)
 | |
|                 bval = NULL;
 | |
|             else
 | |
|                 bval = *vaddr;
 | |
|             Py_DECREF(key);
 | |
|             if (bval == NULL) {
 | |
|                 Py_DECREF(aval);
 | |
|                 if (PyErr_Occurred())
 | |
|                     return -1;
 | |
|                 return 0;
 | |
|             }
 | |
|             cmp = PyObject_RichCompareBool(aval, bval, Py_EQ);
 | |
|             Py_DECREF(aval);
 | |
|             if (cmp <= 0)  /* error or not equal */
 | |
|                 return cmp;
 | |
|         }
 | |
|     }
 | |
|     return 1;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_richcompare(PyObject *v, PyObject *w, int op)
 | |
| {
 | |
|     int cmp;
 | |
|     PyObject *res;
 | |
| 
 | |
|     if (!PyDict_Check(v) || !PyDict_Check(w)) {
 | |
|         res = Py_NotImplemented;
 | |
|     }
 | |
|     else if (op == Py_EQ || op == Py_NE) {
 | |
|         cmp = dict_equal((PyDictObject *)v, (PyDictObject *)w);
 | |
|         if (cmp < 0)
 | |
|             return NULL;
 | |
|         res = (cmp == (op == Py_EQ)) ? Py_True : Py_False;
 | |
|     }
 | |
|     else
 | |
|         res = Py_NotImplemented;
 | |
|     Py_INCREF(res);
 | |
|     return res;
 | |
| }
 | |
| 
 | |
| /*[clinic input]
 | |
| 
 | |
| @coexist
 | |
| dict.__contains__
 | |
| 
 | |
|   key: object
 | |
|   /
 | |
| 
 | |
| True if D has a key k, else False.
 | |
| [clinic start generated code]*/
 | |
| 
 | |
| PyDoc_STRVAR(dict___contains____doc__,
 | |
| "__contains__($self, key, /)\n"
 | |
| "--\n"
 | |
| "\n"
 | |
| "True if D has a key k, else False.");
 | |
| 
 | |
| #define DICT___CONTAINS___METHODDEF    \
 | |
|     {"__contains__", (PyCFunction)dict___contains__, METH_O|METH_COEXIST, dict___contains____doc__},
 | |
| 
 | |
| static PyObject *
 | |
| dict___contains__(PyDictObject *self, PyObject *key)
 | |
| /*[clinic end generated code: output=3cf3f8aaf2cc5cc3 input=b852b2a19b51ab24]*/
 | |
| {
 | |
|     register PyDictObject *mp = self;
 | |
|     Py_hash_t hash;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return NULL;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     return PyBool_FromLong(*value_addr != NULL);
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_get(PyDictObject *mp, PyObject *args)
 | |
| {
 | |
|     PyObject *key;
 | |
|     PyObject *failobj = Py_None;
 | |
|     PyObject *val = NULL;
 | |
|     Py_hash_t hash;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyArg_UnpackTuple(args, "get", 1, 2, &key, &failobj))
 | |
|         return NULL;
 | |
| 
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return NULL;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     val = *value_addr;
 | |
|     if (val == NULL)
 | |
|         val = failobj;
 | |
|     Py_INCREF(val);
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| PyObject *
 | |
| PyDict_SetDefault(PyObject *d, PyObject *key, PyObject *defaultobj)
 | |
| {
 | |
|     PyDictObject *mp = (PyDictObject *)d;
 | |
|     PyObject *val = NULL;
 | |
|     Py_hash_t hash;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyDict_Check(d)) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return NULL;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     val = *value_addr;
 | |
|     if (val == NULL) {
 | |
|         if (mp->ma_keys->dk_usable <= 0) {
 | |
|             /* Need to resize. */
 | |
|             if (insertion_resize(mp) < 0)
 | |
|                 return NULL;
 | |
|             ep = find_empty_slot(mp, key, hash, &value_addr);
 | |
|         }
 | |
|         Py_INCREF(defaultobj);
 | |
|         Py_INCREF(key);
 | |
|         MAINTAIN_TRACKING(mp, key, defaultobj);
 | |
|         ep->me_key = key;
 | |
|         ep->me_hash = hash;
 | |
|         *value_addr = defaultobj;
 | |
|         val = defaultobj;
 | |
|         mp->ma_keys->dk_usable--;
 | |
|         mp->ma_used++;
 | |
|     }
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_setdefault(PyDictObject *mp, PyObject *args)
 | |
| {
 | |
|     PyObject *key, *val;
 | |
|     PyObject *defaultobj = Py_None;
 | |
| 
 | |
|     if (!PyArg_UnpackTuple(args, "setdefault", 1, 2, &key, &defaultobj))
 | |
|         return NULL;
 | |
| 
 | |
|     val = PyDict_SetDefault((PyObject *)mp, key, defaultobj);
 | |
|     Py_XINCREF(val);
 | |
|     return val;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_clear(PyDictObject *mp)
 | |
| {
 | |
|     PyDict_Clear((PyObject *)mp);
 | |
|     Py_RETURN_NONE;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_pop(PyDictObject *mp, PyObject *args)
 | |
| {
 | |
|     Py_hash_t hash;
 | |
|     PyObject *old_value, *old_key;
 | |
|     PyObject *key, *deflt = NULL;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if(!PyArg_UnpackTuple(args, "pop", 1, 2, &key, &deflt))
 | |
|         return NULL;
 | |
|     if (mp->ma_used == 0) {
 | |
|         if (deflt) {
 | |
|             Py_INCREF(deflt);
 | |
|             return deflt;
 | |
|         }
 | |
|         _PyErr_SetKeyError(key);
 | |
|         return NULL;
 | |
|     }
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return NULL;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     if (ep == NULL)
 | |
|         return NULL;
 | |
|     old_value = *value_addr;
 | |
|     if (old_value == NULL) {
 | |
|         if (deflt) {
 | |
|             Py_INCREF(deflt);
 | |
|             return deflt;
 | |
|         }
 | |
|         _PyErr_SetKeyError(key);
 | |
|         return NULL;
 | |
|     }
 | |
|     *value_addr = NULL;
 | |
|     mp->ma_used--;
 | |
|     if (!_PyDict_HasSplitTable(mp)) {
 | |
|         ENSURE_ALLOWS_DELETIONS(mp);
 | |
|         old_key = ep->me_key;
 | |
|         Py_INCREF(dummy);
 | |
|         ep->me_key = dummy;
 | |
|         Py_DECREF(old_key);
 | |
|     }
 | |
|     return old_value;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_popitem(PyDictObject *mp)
 | |
| {
 | |
|     Py_hash_t i = 0;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject *res;
 | |
| 
 | |
| 
 | |
|     /* Allocate the result tuple before checking the size.  Believe it
 | |
|      * or not, this allocation could trigger a garbage collection which
 | |
|      * could empty the dict, so if we checked the size first and that
 | |
|      * happened, the result would be an infinite loop (searching for an
 | |
|      * entry that no longer exists).  Note that the usual popitem()
 | |
|      * idiom is "while d: k, v = d.popitem()". so needing to throw the
 | |
|      * tuple away if the dict *is* empty isn't a significant
 | |
|      * inefficiency -- possible, but unlikely in practice.
 | |
|      */
 | |
|     res = PyTuple_New(2);
 | |
|     if (res == NULL)
 | |
|         return NULL;
 | |
|     if (mp->ma_used == 0) {
 | |
|         Py_DECREF(res);
 | |
|         PyErr_SetString(PyExc_KeyError,
 | |
|                         "popitem(): dictionary is empty");
 | |
|         return NULL;
 | |
|     }
 | |
|     /* Convert split table to combined table */
 | |
|     if (mp->ma_keys->dk_lookup == lookdict_split) {
 | |
|         if (dictresize(mp, DK_SIZE(mp->ma_keys))) {
 | |
|             Py_DECREF(res);
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
|     ENSURE_ALLOWS_DELETIONS(mp);
 | |
|     /* Set ep to "the first" dict entry with a value.  We abuse the hash
 | |
|      * field of slot 0 to hold a search finger:
 | |
|      * If slot 0 has a value, use slot 0.
 | |
|      * Else slot 0 is being used to hold a search finger,
 | |
|      * and we use its hash value as the first index to look.
 | |
|      */
 | |
|     ep = &mp->ma_keys->dk_entries[0];
 | |
|     if (ep->me_value == NULL) {
 | |
|         Py_ssize_t mask = DK_MASK(mp->ma_keys);
 | |
|         i = ep->me_hash;
 | |
|         /* The hash field may be a real hash value, or it may be a
 | |
|          * legit search finger, or it may be a once-legit search
 | |
|          * finger that's out of bounds now because it wrapped around
 | |
|          * or the table shrunk -- simply make sure it's in bounds now.
 | |
|          */
 | |
|         if (i > mask || i < 1)
 | |
|             i = 1;              /* skip slot 0 */
 | |
|         while ((ep = &mp->ma_keys->dk_entries[i])->me_value == NULL) {
 | |
|             i++;
 | |
|             if (i > mask)
 | |
|                 i = 1;
 | |
|         }
 | |
|     }
 | |
|     PyTuple_SET_ITEM(res, 0, ep->me_key);
 | |
|     PyTuple_SET_ITEM(res, 1, ep->me_value);
 | |
|     Py_INCREF(dummy);
 | |
|     ep->me_key = dummy;
 | |
|     ep->me_value = NULL;
 | |
|     mp->ma_used--;
 | |
|     assert(mp->ma_keys->dk_entries[0].me_value == NULL);
 | |
|     mp->ma_keys->dk_entries[0].me_hash = i + 1;  /* next place to start */
 | |
|     return res;
 | |
| }
 | |
| 
 | |
| static int
 | |
| dict_traverse(PyObject *op, visitproc visit, void *arg)
 | |
| {
 | |
|     Py_ssize_t i, n;
 | |
|     PyDictObject *mp = (PyDictObject *)op;
 | |
|     if (mp->ma_keys->dk_lookup == lookdict) {
 | |
|         for (i = 0; i < DK_SIZE(mp->ma_keys); i++) {
 | |
|             if (mp->ma_keys->dk_entries[i].me_value != NULL) {
 | |
|                 Py_VISIT(mp->ma_keys->dk_entries[i].me_value);
 | |
|                 Py_VISIT(mp->ma_keys->dk_entries[i].me_key);
 | |
|             }
 | |
|         }
 | |
|     } else {
 | |
|         if (mp->ma_values != NULL) {
 | |
|             for (i = 0, n = DK_SIZE(mp->ma_keys); i < n; i++) {
 | |
|                 Py_VISIT(mp->ma_values[i]);
 | |
|             }
 | |
|         }
 | |
|         else {
 | |
|             for (i = 0, n = DK_SIZE(mp->ma_keys); i < n; i++) {
 | |
|                 Py_VISIT(mp->ma_keys->dk_entries[i].me_value);
 | |
|             }
 | |
|         }
 | |
|     }
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static int
 | |
| dict_tp_clear(PyObject *op)
 | |
| {
 | |
|     PyDict_Clear(op);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static PyObject *dictiter_new(PyDictObject *, PyTypeObject *);
 | |
| 
 | |
| static PyObject *
 | |
| dict_sizeof(PyDictObject *mp)
 | |
| {
 | |
|     Py_ssize_t size, res;
 | |
| 
 | |
|     size = DK_SIZE(mp->ma_keys);
 | |
|     res = sizeof(PyDictObject);
 | |
|     if (mp->ma_values)
 | |
|         res += size * sizeof(PyObject*);
 | |
|     /* If the dictionary is split, the keys portion is accounted-for
 | |
|        in the type object. */
 | |
|     if (mp->ma_keys->dk_refcnt == 1)
 | |
|         res += sizeof(PyDictKeysObject) + (size-1) * sizeof(PyDictKeyEntry);
 | |
|     return PyLong_FromSsize_t(res);
 | |
| }
 | |
| 
 | |
| Py_ssize_t
 | |
| _PyDict_KeysSize(PyDictKeysObject *keys)
 | |
| {
 | |
|     return sizeof(PyDictKeysObject) + (DK_SIZE(keys)-1) * sizeof(PyDictKeyEntry);
 | |
| }
 | |
| 
 | |
| PyDoc_STRVAR(getitem__doc__, "x.__getitem__(y) <==> x[y]");
 | |
| 
 | |
| PyDoc_STRVAR(sizeof__doc__,
 | |
| "D.__sizeof__() -> size of D in memory, in bytes");
 | |
| 
 | |
| PyDoc_STRVAR(get__doc__,
 | |
| "D.get(k[,d]) -> D[k] if k in D, else d.  d defaults to None.");
 | |
| 
 | |
| PyDoc_STRVAR(setdefault_doc__,
 | |
| "D.setdefault(k[,d]) -> D.get(k,d), also set D[k]=d if k not in D");
 | |
| 
 | |
| PyDoc_STRVAR(pop__doc__,
 | |
| "D.pop(k[,d]) -> v, remove specified key and return the corresponding value.\n\
 | |
| If key is not found, d is returned if given, otherwise KeyError is raised");
 | |
| 
 | |
| PyDoc_STRVAR(popitem__doc__,
 | |
| "D.popitem() -> (k, v), remove and return some (key, value) pair as a\n\
 | |
| 2-tuple; but raise KeyError if D is empty.");
 | |
| 
 | |
| PyDoc_STRVAR(update__doc__,
 | |
| "D.update([E, ]**F) -> None.  Update D from dict/iterable E and F.\n\
 | |
| If E is present and has a .keys() method, then does:  for k in E: D[k] = E[k]\n\
 | |
| If E is present and lacks a .keys() method, then does:  for k, v in E: D[k] = v\n\
 | |
| In either case, this is followed by: for k in F:  D[k] = F[k]");
 | |
| 
 | |
| PyDoc_STRVAR(clear__doc__,
 | |
| "D.clear() -> None.  Remove all items from D.");
 | |
| 
 | |
| PyDoc_STRVAR(copy__doc__,
 | |
| "D.copy() -> a shallow copy of D");
 | |
| 
 | |
| /* Forward */
 | |
| static PyObject *dictkeys_new(PyObject *);
 | |
| static PyObject *dictitems_new(PyObject *);
 | |
| static PyObject *dictvalues_new(PyObject *);
 | |
| 
 | |
| PyDoc_STRVAR(keys__doc__,
 | |
|              "D.keys() -> a set-like object providing a view on D's keys");
 | |
| PyDoc_STRVAR(items__doc__,
 | |
|              "D.items() -> a set-like object providing a view on D's items");
 | |
| PyDoc_STRVAR(values__doc__,
 | |
|              "D.values() -> an object providing a view on D's values");
 | |
| 
 | |
| static PyMethodDef mapp_methods[] = {
 | |
|     DICT___CONTAINS___METHODDEF
 | |
|     {"__getitem__", (PyCFunction)dict_subscript,        METH_O | METH_COEXIST,
 | |
|      getitem__doc__},
 | |
|     {"__sizeof__",      (PyCFunction)dict_sizeof,       METH_NOARGS,
 | |
|      sizeof__doc__},
 | |
|     {"get",         (PyCFunction)dict_get,          METH_VARARGS,
 | |
|      get__doc__},
 | |
|     {"setdefault",  (PyCFunction)dict_setdefault,   METH_VARARGS,
 | |
|      setdefault_doc__},
 | |
|     {"pop",         (PyCFunction)dict_pop,          METH_VARARGS,
 | |
|      pop__doc__},
 | |
|     {"popitem",         (PyCFunction)dict_popitem,      METH_NOARGS,
 | |
|      popitem__doc__},
 | |
|     {"keys",            (PyCFunction)dictkeys_new,      METH_NOARGS,
 | |
|     keys__doc__},
 | |
|     {"items",           (PyCFunction)dictitems_new,     METH_NOARGS,
 | |
|     items__doc__},
 | |
|     {"values",          (PyCFunction)dictvalues_new,    METH_NOARGS,
 | |
|     values__doc__},
 | |
|     {"update",          (PyCFunction)dict_update,       METH_VARARGS | METH_KEYWORDS,
 | |
|      update__doc__},
 | |
|     DICT_FROMKEYS_METHODDEF
 | |
|     {"clear",           (PyCFunction)dict_clear,        METH_NOARGS,
 | |
|      clear__doc__},
 | |
|     {"copy",            (PyCFunction)dict_copy,         METH_NOARGS,
 | |
|      copy__doc__},
 | |
|     {NULL,              NULL}   /* sentinel */
 | |
| };
 | |
| 
 | |
| /* Return 1 if `key` is in dict `op`, 0 if not, and -1 on error. */
 | |
| int
 | |
| PyDict_Contains(PyObject *op, PyObject *key)
 | |
| {
 | |
|     Py_hash_t hash;
 | |
|     PyDictObject *mp = (PyDictObject *)op;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     if (!PyUnicode_CheckExact(key) ||
 | |
|         (hash = ((PyASCIIObject *) key)->hash) == -1) {
 | |
|         hash = PyObject_Hash(key);
 | |
|         if (hash == -1)
 | |
|             return -1;
 | |
|     }
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     return (ep == NULL) ? -1 : (*value_addr != NULL);
 | |
| }
 | |
| 
 | |
| /* Internal version of PyDict_Contains used when the hash value is already known */
 | |
| int
 | |
| _PyDict_Contains(PyObject *op, PyObject *key, Py_hash_t hash)
 | |
| {
 | |
|     PyDictObject *mp = (PyDictObject *)op;
 | |
|     PyDictKeyEntry *ep;
 | |
|     PyObject **value_addr;
 | |
| 
 | |
|     ep = (mp->ma_keys->dk_lookup)(mp, key, hash, &value_addr);
 | |
|     return (ep == NULL) ? -1 : (*value_addr != NULL);
 | |
| }
 | |
| 
 | |
| /* Hack to implement "key in dict" */
 | |
| static PySequenceMethods dict_as_sequence = {
 | |
|     0,                          /* sq_length */
 | |
|     0,                          /* sq_concat */
 | |
|     0,                          /* sq_repeat */
 | |
|     0,                          /* sq_item */
 | |
|     0,                          /* sq_slice */
 | |
|     0,                          /* sq_ass_item */
 | |
|     0,                          /* sq_ass_slice */
 | |
|     PyDict_Contains,            /* sq_contains */
 | |
|     0,                          /* sq_inplace_concat */
 | |
|     0,                          /* sq_inplace_repeat */
 | |
| };
 | |
| 
 | |
| static PyObject *
 | |
| dict_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
 | |
| {
 | |
|     PyObject *self;
 | |
|     PyDictObject *d;
 | |
| 
 | |
|     assert(type != NULL && type->tp_alloc != NULL);
 | |
|     self = type->tp_alloc(type, 0);
 | |
|     if (self == NULL)
 | |
|         return NULL;
 | |
|     d = (PyDictObject *)self;
 | |
| 
 | |
|     /* The object has been implicitly tracked by tp_alloc */
 | |
|     if (type == &PyDict_Type)
 | |
|         _PyObject_GC_UNTRACK(d);
 | |
| 
 | |
|     d->ma_used = 0;
 | |
|     d->ma_keys = new_keys_object(PyDict_MINSIZE_COMBINED);
 | |
|     if (d->ma_keys == NULL) {
 | |
|         Py_DECREF(self);
 | |
|         return NULL;
 | |
|     }
 | |
|     return self;
 | |
| }
 | |
| 
 | |
| static int
 | |
| dict_init(PyObject *self, PyObject *args, PyObject *kwds)
 | |
| {
 | |
|     return dict_update_common(self, args, kwds, "dict");
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dict_iter(PyDictObject *dict)
 | |
| {
 | |
|     return dictiter_new(dict, &PyDictIterKey_Type);
 | |
| }
 | |
| 
 | |
| PyDoc_STRVAR(dictionary_doc,
 | |
| "dict() -> new empty dictionary\n"
 | |
| "dict(mapping) -> new dictionary initialized from a mapping object's\n"
 | |
| "    (key, value) pairs\n"
 | |
| "dict(iterable) -> new dictionary initialized as if via:\n"
 | |
| "    d = {}\n"
 | |
| "    for k, v in iterable:\n"
 | |
| "        d[k] = v\n"
 | |
| "dict(**kwargs) -> new dictionary initialized with the name=value pairs\n"
 | |
| "    in the keyword argument list.  For example:  dict(one=1, two=2)");
 | |
| 
 | |
| PyTypeObject PyDict_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict",
 | |
|     sizeof(PyDictObject),
 | |
|     0,
 | |
|     (destructor)dict_dealloc,                   /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     (reprfunc)dict_repr,                        /* tp_repr */
 | |
|     0,                                          /* tp_as_number */
 | |
|     &dict_as_sequence,                          /* tp_as_sequence */
 | |
|     &dict_as_mapping,                           /* tp_as_mapping */
 | |
|     PyObject_HashNotImplemented,                /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
 | |
|         Py_TPFLAGS_BASETYPE | Py_TPFLAGS_DICT_SUBCLASS,         /* tp_flags */
 | |
|     dictionary_doc,                             /* tp_doc */
 | |
|     dict_traverse,                              /* tp_traverse */
 | |
|     dict_tp_clear,                              /* tp_clear */
 | |
|     dict_richcompare,                           /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     (getiterfunc)dict_iter,                     /* tp_iter */
 | |
|     0,                                          /* tp_iternext */
 | |
|     mapp_methods,                               /* tp_methods */
 | |
|     0,                                          /* tp_members */
 | |
|     0,                                          /* tp_getset */
 | |
|     0,                                          /* tp_base */
 | |
|     0,                                          /* tp_dict */
 | |
|     0,                                          /* tp_descr_get */
 | |
|     0,                                          /* tp_descr_set */
 | |
|     0,                                          /* tp_dictoffset */
 | |
|     dict_init,                                  /* tp_init */
 | |
|     PyType_GenericAlloc,                        /* tp_alloc */
 | |
|     dict_new,                                   /* tp_new */
 | |
|     PyObject_GC_Del,                            /* tp_free */
 | |
| };
 | |
| 
 | |
| PyObject *
 | |
| _PyDict_GetItemId(PyObject *dp, struct _Py_Identifier *key)
 | |
| {
 | |
|     PyObject *kv;
 | |
|     kv = _PyUnicode_FromId(key); /* borrowed */
 | |
|     if (kv == NULL) {
 | |
|         PyErr_Clear();
 | |
|         return NULL;
 | |
|     }
 | |
|     return PyDict_GetItem(dp, kv);
 | |
| }
 | |
| 
 | |
| /* For backward compatibility with old dictionary interface */
 | |
| 
 | |
| PyObject *
 | |
| PyDict_GetItemString(PyObject *v, const char *key)
 | |
| {
 | |
|     PyObject *kv, *rv;
 | |
|     kv = PyUnicode_FromString(key);
 | |
|     if (kv == NULL) {
 | |
|         PyErr_Clear();
 | |
|         return NULL;
 | |
|     }
 | |
|     rv = PyDict_GetItem(v, kv);
 | |
|     Py_DECREF(kv);
 | |
|     return rv;
 | |
| }
 | |
| 
 | |
| int
 | |
| _PyDict_SetItemId(PyObject *v, struct _Py_Identifier *key, PyObject *item)
 | |
| {
 | |
|     PyObject *kv;
 | |
|     kv = _PyUnicode_FromId(key); /* borrowed */
 | |
|     if (kv == NULL)
 | |
|         return -1;
 | |
|     return PyDict_SetItem(v, kv, item);
 | |
| }
 | |
| 
 | |
| int
 | |
| PyDict_SetItemString(PyObject *v, const char *key, PyObject *item)
 | |
| {
 | |
|     PyObject *kv;
 | |
|     int err;
 | |
|     kv = PyUnicode_FromString(key);
 | |
|     if (kv == NULL)
 | |
|         return -1;
 | |
|     PyUnicode_InternInPlace(&kv); /* XXX Should we really? */
 | |
|     err = PyDict_SetItem(v, kv, item);
 | |
|     Py_DECREF(kv);
 | |
|     return err;
 | |
| }
 | |
| 
 | |
| int
 | |
| _PyDict_DelItemId(PyObject *v, _Py_Identifier *key)
 | |
| {
 | |
|     PyObject *kv = _PyUnicode_FromId(key); /* borrowed */
 | |
|     if (kv == NULL)
 | |
|         return -1;
 | |
|     return PyDict_DelItem(v, kv);
 | |
| }
 | |
| 
 | |
| int
 | |
| PyDict_DelItemString(PyObject *v, const char *key)
 | |
| {
 | |
|     PyObject *kv;
 | |
|     int err;
 | |
|     kv = PyUnicode_FromString(key);
 | |
|     if (kv == NULL)
 | |
|         return -1;
 | |
|     err = PyDict_DelItem(v, kv);
 | |
|     Py_DECREF(kv);
 | |
|     return err;
 | |
| }
 | |
| 
 | |
| /* Dictionary iterator types */
 | |
| 
 | |
| typedef struct {
 | |
|     PyObject_HEAD
 | |
|     PyDictObject *di_dict; /* Set to NULL when iterator is exhausted */
 | |
|     Py_ssize_t di_used;
 | |
|     Py_ssize_t di_pos;
 | |
|     PyObject* di_result; /* reusable result tuple for iteritems */
 | |
|     Py_ssize_t len;
 | |
| } dictiterobject;
 | |
| 
 | |
| static PyObject *
 | |
| dictiter_new(PyDictObject *dict, PyTypeObject *itertype)
 | |
| {
 | |
|     dictiterobject *di;
 | |
|     di = PyObject_GC_New(dictiterobject, itertype);
 | |
|     if (di == NULL)
 | |
|         return NULL;
 | |
|     Py_INCREF(dict);
 | |
|     di->di_dict = dict;
 | |
|     di->di_used = dict->ma_used;
 | |
|     di->di_pos = 0;
 | |
|     di->len = dict->ma_used;
 | |
|     if (itertype == &PyDictIterItem_Type) {
 | |
|         di->di_result = PyTuple_Pack(2, Py_None, Py_None);
 | |
|         if (di->di_result == NULL) {
 | |
|             Py_DECREF(di);
 | |
|             return NULL;
 | |
|         }
 | |
|     }
 | |
|     else
 | |
|         di->di_result = NULL;
 | |
|     _PyObject_GC_TRACK(di);
 | |
|     return (PyObject *)di;
 | |
| }
 | |
| 
 | |
| static void
 | |
| dictiter_dealloc(dictiterobject *di)
 | |
| {
 | |
|     Py_XDECREF(di->di_dict);
 | |
|     Py_XDECREF(di->di_result);
 | |
|     PyObject_GC_Del(di);
 | |
| }
 | |
| 
 | |
| static int
 | |
| dictiter_traverse(dictiterobject *di, visitproc visit, void *arg)
 | |
| {
 | |
|     Py_VISIT(di->di_dict);
 | |
|     Py_VISIT(di->di_result);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dictiter_len(dictiterobject *di)
 | |
| {
 | |
|     Py_ssize_t len = 0;
 | |
|     if (di->di_dict != NULL && di->di_used == di->di_dict->ma_used)
 | |
|         len = di->len;
 | |
|     return PyLong_FromSize_t(len);
 | |
| }
 | |
| 
 | |
| PyDoc_STRVAR(length_hint_doc,
 | |
|              "Private method returning an estimate of len(list(it)).");
 | |
| 
 | |
| static PyObject *
 | |
| dictiter_reduce(dictiterobject *di);
 | |
| 
 | |
| PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
 | |
| 
 | |
| static PyMethodDef dictiter_methods[] = {
 | |
|     {"__length_hint__", (PyCFunction)dictiter_len, METH_NOARGS,
 | |
|      length_hint_doc},
 | |
|      {"__reduce__", (PyCFunction)dictiter_reduce, METH_NOARGS,
 | |
|      reduce_doc},
 | |
|     {NULL,              NULL}           /* sentinel */
 | |
| };
 | |
| 
 | |
| static PyObject *dictiter_iternextkey(dictiterobject *di)
 | |
| {
 | |
|     PyObject *key;
 | |
|     Py_ssize_t i, mask, offset;
 | |
|     PyDictKeysObject *k;
 | |
|     PyDictObject *d = di->di_dict;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|     if (d == NULL)
 | |
|         return NULL;
 | |
|     assert (PyDict_Check(d));
 | |
| 
 | |
|     if (di->di_used != d->ma_used) {
 | |
|         PyErr_SetString(PyExc_RuntimeError,
 | |
|                         "dictionary changed size during iteration");
 | |
|         di->di_used = -1; /* Make this state sticky */
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     i = di->di_pos;
 | |
|     if (i < 0)
 | |
|         goto fail;
 | |
|     k = d->ma_keys;
 | |
|     if (d->ma_values) {
 | |
|         value_ptr = &d->ma_values[i];
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &k->dk_entries[i].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     mask = DK_SIZE(k)-1;
 | |
|     while (i <= mask && *value_ptr == NULL) {
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         i++;
 | |
|     }
 | |
|     di->di_pos = i+1;
 | |
|     if (i > mask)
 | |
|         goto fail;
 | |
|     di->len--;
 | |
|     key = k->dk_entries[i].me_key;
 | |
|     Py_INCREF(key);
 | |
|     return key;
 | |
| 
 | |
| fail:
 | |
|     Py_DECREF(d);
 | |
|     di->di_dict = NULL;
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| PyTypeObject PyDictIterKey_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_keyiterator",                         /* tp_name */
 | |
|     sizeof(dictiterobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictiter_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     0,                                          /* tp_repr */
 | |
|     0,                                          /* tp_as_number */
 | |
|     0,                                          /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictiter_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     0,                                          /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     PyObject_SelfIter,                          /* tp_iter */
 | |
|     (iternextfunc)dictiter_iternextkey,         /* tp_iternext */
 | |
|     dictiter_methods,                           /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| static PyObject *dictiter_iternextvalue(dictiterobject *di)
 | |
| {
 | |
|     PyObject *value;
 | |
|     Py_ssize_t i, mask, offset;
 | |
|     PyDictObject *d = di->di_dict;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|     if (d == NULL)
 | |
|         return NULL;
 | |
|     assert (PyDict_Check(d));
 | |
| 
 | |
|     if (di->di_used != d->ma_used) {
 | |
|         PyErr_SetString(PyExc_RuntimeError,
 | |
|                         "dictionary changed size during iteration");
 | |
|         di->di_used = -1; /* Make this state sticky */
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     i = di->di_pos;
 | |
|     mask = DK_SIZE(d->ma_keys)-1;
 | |
|     if (i < 0 || i > mask)
 | |
|         goto fail;
 | |
|     if (d->ma_values) {
 | |
|         value_ptr = &d->ma_values[i];
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &d->ma_keys->dk_entries[i].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     while (i <= mask && *value_ptr == NULL) {
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         i++;
 | |
|         if (i > mask)
 | |
|             goto fail;
 | |
|     }
 | |
|     di->di_pos = i+1;
 | |
|     di->len--;
 | |
|     value = *value_ptr;
 | |
|     Py_INCREF(value);
 | |
|     return value;
 | |
| 
 | |
| fail:
 | |
|     Py_DECREF(d);
 | |
|     di->di_dict = NULL;
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| PyTypeObject PyDictIterValue_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_valueiterator",                       /* tp_name */
 | |
|     sizeof(dictiterobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictiter_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     0,                                          /* tp_repr */
 | |
|     0,                                          /* tp_as_number */
 | |
|     0,                                          /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictiter_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     0,                                          /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     PyObject_SelfIter,                          /* tp_iter */
 | |
|     (iternextfunc)dictiter_iternextvalue,       /* tp_iternext */
 | |
|     dictiter_methods,                           /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| static PyObject *dictiter_iternextitem(dictiterobject *di)
 | |
| {
 | |
|     PyObject *key, *value, *result = di->di_result;
 | |
|     Py_ssize_t i, mask, offset;
 | |
|     PyDictObject *d = di->di_dict;
 | |
|     PyObject **value_ptr;
 | |
| 
 | |
|     if (d == NULL)
 | |
|         return NULL;
 | |
|     assert (PyDict_Check(d));
 | |
| 
 | |
|     if (di->di_used != d->ma_used) {
 | |
|         PyErr_SetString(PyExc_RuntimeError,
 | |
|                         "dictionary changed size during iteration");
 | |
|         di->di_used = -1; /* Make this state sticky */
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     i = di->di_pos;
 | |
|     if (i < 0)
 | |
|         goto fail;
 | |
|     mask = DK_SIZE(d->ma_keys)-1;
 | |
|     if (d->ma_values) {
 | |
|         value_ptr = &d->ma_values[i];
 | |
|         offset = sizeof(PyObject *);
 | |
|     }
 | |
|     else {
 | |
|         value_ptr = &d->ma_keys->dk_entries[i].me_value;
 | |
|         offset = sizeof(PyDictKeyEntry);
 | |
|     }
 | |
|     while (i <= mask && *value_ptr == NULL) {
 | |
|         value_ptr = (PyObject **)(((char *)value_ptr) + offset);
 | |
|         i++;
 | |
|     }
 | |
|     di->di_pos = i+1;
 | |
|     if (i > mask)
 | |
|         goto fail;
 | |
| 
 | |
|     if (result->ob_refcnt == 1) {
 | |
|         Py_INCREF(result);
 | |
|         Py_DECREF(PyTuple_GET_ITEM(result, 0));
 | |
|         Py_DECREF(PyTuple_GET_ITEM(result, 1));
 | |
|     } else {
 | |
|         result = PyTuple_New(2);
 | |
|         if (result == NULL)
 | |
|             return NULL;
 | |
|     }
 | |
|     di->len--;
 | |
|     key = d->ma_keys->dk_entries[i].me_key;
 | |
|     value = *value_ptr;
 | |
|     Py_INCREF(key);
 | |
|     Py_INCREF(value);
 | |
|     PyTuple_SET_ITEM(result, 0, key);
 | |
|     PyTuple_SET_ITEM(result, 1, value);
 | |
|     return result;
 | |
| 
 | |
| fail:
 | |
|     Py_DECREF(d);
 | |
|     di->di_dict = NULL;
 | |
|     return NULL;
 | |
| }
 | |
| 
 | |
| PyTypeObject PyDictIterItem_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_itemiterator",                        /* tp_name */
 | |
|     sizeof(dictiterobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictiter_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     0,                                          /* tp_repr */
 | |
|     0,                                          /* tp_as_number */
 | |
|     0,                                          /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictiter_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     0,                                          /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     PyObject_SelfIter,                          /* tp_iter */
 | |
|     (iternextfunc)dictiter_iternextitem,        /* tp_iternext */
 | |
|     dictiter_methods,                           /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| 
 | |
| static PyObject *
 | |
| dictiter_reduce(dictiterobject *di)
 | |
| {
 | |
|     PyObject *list;
 | |
|     dictiterobject tmp;
 | |
| 
 | |
|     list = PyList_New(0);
 | |
|     if (!list)
 | |
|         return NULL;
 | |
| 
 | |
|     /* copy the itertor state */
 | |
|     tmp = *di;
 | |
|     Py_XINCREF(tmp.di_dict);
 | |
| 
 | |
|     /* iterate the temporary into a list */
 | |
|     for(;;) {
 | |
|         PyObject *element = 0;
 | |
|         if (Py_TYPE(di) == &PyDictIterItem_Type)
 | |
|             element = dictiter_iternextitem(&tmp);
 | |
|         else if (Py_TYPE(di) == &PyDictIterKey_Type)
 | |
|             element = dictiter_iternextkey(&tmp);
 | |
|         else if (Py_TYPE(di) == &PyDictIterValue_Type)
 | |
|             element = dictiter_iternextvalue(&tmp);
 | |
|         else
 | |
|             assert(0);
 | |
|         if (element) {
 | |
|             if (PyList_Append(list, element)) {
 | |
|                 Py_DECREF(element);
 | |
|                 Py_DECREF(list);
 | |
|                 Py_XDECREF(tmp.di_dict);
 | |
|                 return NULL;
 | |
|             }
 | |
|             Py_DECREF(element);
 | |
|         } else
 | |
|             break;
 | |
|     }
 | |
|     Py_XDECREF(tmp.di_dict);
 | |
|     /* check for error */
 | |
|     if (tmp.di_dict != NULL) {
 | |
|         /* we have an error */
 | |
|         Py_DECREF(list);
 | |
|         return NULL;
 | |
|     }
 | |
|     return Py_BuildValue("N(N)", _PyObject_GetBuiltin("iter"), list);
 | |
| }
 | |
| 
 | |
| /***********************************************/
 | |
| /* View objects for keys(), items(), values(). */
 | |
| /***********************************************/
 | |
| 
 | |
| /* The instance lay-out is the same for all three; but the type differs. */
 | |
| 
 | |
| typedef struct {
 | |
|     PyObject_HEAD
 | |
|     PyDictObject *dv_dict;
 | |
| } dictviewobject;
 | |
| 
 | |
| 
 | |
| static void
 | |
| dictview_dealloc(dictviewobject *dv)
 | |
| {
 | |
|     Py_XDECREF(dv->dv_dict);
 | |
|     PyObject_GC_Del(dv);
 | |
| }
 | |
| 
 | |
| static int
 | |
| dictview_traverse(dictviewobject *dv, visitproc visit, void *arg)
 | |
| {
 | |
|     Py_VISIT(dv->dv_dict);
 | |
|     return 0;
 | |
| }
 | |
| 
 | |
| static Py_ssize_t
 | |
| dictview_len(dictviewobject *dv)
 | |
| {
 | |
|     Py_ssize_t len = 0;
 | |
|     if (dv->dv_dict != NULL)
 | |
|         len = dv->dv_dict->ma_used;
 | |
|     return len;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dictview_new(PyObject *dict, PyTypeObject *type)
 | |
| {
 | |
|     dictviewobject *dv;
 | |
|     if (dict == NULL) {
 | |
|         PyErr_BadInternalCall();
 | |
|         return NULL;
 | |
|     }
 | |
|     if (!PyDict_Check(dict)) {
 | |
|         /* XXX Get rid of this restriction later */
 | |
|         PyErr_Format(PyExc_TypeError,
 | |
|                      "%s() requires a dict argument, not '%s'",
 | |
|                      type->tp_name, dict->ob_type->tp_name);
 | |
|         return NULL;
 | |
|     }
 | |
|     dv = PyObject_GC_New(dictviewobject, type);
 | |
|     if (dv == NULL)
 | |
|         return NULL;
 | |
|     Py_INCREF(dict);
 | |
|     dv->dv_dict = (PyDictObject *)dict;
 | |
|     _PyObject_GC_TRACK(dv);
 | |
|     return (PyObject *)dv;
 | |
| }
 | |
| 
 | |
| /* TODO(guido): The views objects are not complete:
 | |
| 
 | |
|  * support more set operations
 | |
|  * support arbitrary mappings?
 | |
|    - either these should be static or exported in dictobject.h
 | |
|    - if public then they should probably be in builtins
 | |
| */
 | |
| 
 | |
| /* Return 1 if self is a subset of other, iterating over self;
 | |
|    0 if not; -1 if an error occurred. */
 | |
| static int
 | |
| all_contained_in(PyObject *self, PyObject *other)
 | |
| {
 | |
|     PyObject *iter = PyObject_GetIter(self);
 | |
|     int ok = 1;
 | |
| 
 | |
|     if (iter == NULL)
 | |
|         return -1;
 | |
|     for (;;) {
 | |
|         PyObject *next = PyIter_Next(iter);
 | |
|         if (next == NULL) {
 | |
|             if (PyErr_Occurred())
 | |
|                 ok = -1;
 | |
|             break;
 | |
|         }
 | |
|         ok = PySequence_Contains(other, next);
 | |
|         Py_DECREF(next);
 | |
|         if (ok <= 0)
 | |
|             break;
 | |
|     }
 | |
|     Py_DECREF(iter);
 | |
|     return ok;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dictview_richcompare(PyObject *self, PyObject *other, int op)
 | |
| {
 | |
|     Py_ssize_t len_self, len_other;
 | |
|     int ok;
 | |
|     PyObject *result;
 | |
| 
 | |
|     assert(self != NULL);
 | |
|     assert(PyDictViewSet_Check(self));
 | |
|     assert(other != NULL);
 | |
| 
 | |
|     if (!PyAnySet_Check(other) && !PyDictViewSet_Check(other))
 | |
|         Py_RETURN_NOTIMPLEMENTED;
 | |
| 
 | |
|     len_self = PyObject_Size(self);
 | |
|     if (len_self < 0)
 | |
|         return NULL;
 | |
|     len_other = PyObject_Size(other);
 | |
|     if (len_other < 0)
 | |
|         return NULL;
 | |
| 
 | |
|     ok = 0;
 | |
|     switch(op) {
 | |
| 
 | |
|     case Py_NE:
 | |
|     case Py_EQ:
 | |
|         if (len_self == len_other)
 | |
|             ok = all_contained_in(self, other);
 | |
|         if (op == Py_NE && ok >= 0)
 | |
|             ok = !ok;
 | |
|         break;
 | |
| 
 | |
|     case Py_LT:
 | |
|         if (len_self < len_other)
 | |
|             ok = all_contained_in(self, other);
 | |
|         break;
 | |
| 
 | |
|       case Py_LE:
 | |
|           if (len_self <= len_other)
 | |
|               ok = all_contained_in(self, other);
 | |
|           break;
 | |
| 
 | |
|     case Py_GT:
 | |
|         if (len_self > len_other)
 | |
|             ok = all_contained_in(other, self);
 | |
|         break;
 | |
| 
 | |
|     case Py_GE:
 | |
|         if (len_self >= len_other)
 | |
|             ok = all_contained_in(other, self);
 | |
|         break;
 | |
| 
 | |
|     }
 | |
|     if (ok < 0)
 | |
|         return NULL;
 | |
|     result = ok ? Py_True : Py_False;
 | |
|     Py_INCREF(result);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyObject *
 | |
| dictview_repr(dictviewobject *dv)
 | |
| {
 | |
|     PyObject *seq;
 | |
|     PyObject *result;
 | |
| 
 | |
|     seq = PySequence_List((PyObject *)dv);
 | |
|     if (seq == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     result = PyUnicode_FromFormat("%s(%R)", Py_TYPE(dv)->tp_name, seq);
 | |
|     Py_DECREF(seq);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| /*** dict_keys ***/
 | |
| 
 | |
| static PyObject *
 | |
| dictkeys_iter(dictviewobject *dv)
 | |
| {
 | |
|     if (dv->dv_dict == NULL) {
 | |
|         Py_RETURN_NONE;
 | |
|     }
 | |
|     return dictiter_new(dv->dv_dict, &PyDictIterKey_Type);
 | |
| }
 | |
| 
 | |
| static int
 | |
| dictkeys_contains(dictviewobject *dv, PyObject *obj)
 | |
| {
 | |
|     if (dv->dv_dict == NULL)
 | |
|         return 0;
 | |
|     return PyDict_Contains((PyObject *)dv->dv_dict, obj);
 | |
| }
 | |
| 
 | |
| static PySequenceMethods dictkeys_as_sequence = {
 | |
|     (lenfunc)dictview_len,              /* sq_length */
 | |
|     0,                                  /* sq_concat */
 | |
|     0,                                  /* sq_repeat */
 | |
|     0,                                  /* sq_item */
 | |
|     0,                                  /* sq_slice */
 | |
|     0,                                  /* sq_ass_item */
 | |
|     0,                                  /* sq_ass_slice */
 | |
|     (objobjproc)dictkeys_contains,      /* sq_contains */
 | |
| };
 | |
| 
 | |
| static PyObject*
 | |
| dictviews_sub(PyObject* self, PyObject *other)
 | |
| {
 | |
|     PyObject *result = PySet_New(self);
 | |
|     PyObject *tmp;
 | |
|     _Py_IDENTIFIER(difference_update);
 | |
| 
 | |
|     if (result == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     tmp = _PyObject_CallMethodId(result, &PyId_difference_update, "O", other);
 | |
|     if (tmp == NULL) {
 | |
|         Py_DECREF(result);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     Py_DECREF(tmp);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyObject*
 | |
| dictviews_and(PyObject* self, PyObject *other)
 | |
| {
 | |
|     PyObject *result = PySet_New(self);
 | |
|     PyObject *tmp;
 | |
|     _Py_IDENTIFIER(intersection_update);
 | |
| 
 | |
|     if (result == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     tmp = _PyObject_CallMethodId(result, &PyId_intersection_update, "O", other);
 | |
|     if (tmp == NULL) {
 | |
|         Py_DECREF(result);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     Py_DECREF(tmp);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyObject*
 | |
| dictviews_or(PyObject* self, PyObject *other)
 | |
| {
 | |
|     PyObject *result = PySet_New(self);
 | |
|     PyObject *tmp;
 | |
|     _Py_IDENTIFIER(update);
 | |
| 
 | |
|     if (result == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     tmp = _PyObject_CallMethodId(result, &PyId_update, "O", other);
 | |
|     if (tmp == NULL) {
 | |
|         Py_DECREF(result);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     Py_DECREF(tmp);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyObject*
 | |
| dictviews_xor(PyObject* self, PyObject *other)
 | |
| {
 | |
|     PyObject *result = PySet_New(self);
 | |
|     PyObject *tmp;
 | |
|     _Py_IDENTIFIER(symmetric_difference_update);
 | |
| 
 | |
|     if (result == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     tmp = _PyObject_CallMethodId(result, &PyId_symmetric_difference_update, "O",
 | |
|                               other);
 | |
|     if (tmp == NULL) {
 | |
|         Py_DECREF(result);
 | |
|         return NULL;
 | |
|     }
 | |
| 
 | |
|     Py_DECREF(tmp);
 | |
|     return result;
 | |
| }
 | |
| 
 | |
| static PyNumberMethods dictviews_as_number = {
 | |
|     0,                                  /*nb_add*/
 | |
|     (binaryfunc)dictviews_sub,          /*nb_subtract*/
 | |
|     0,                                  /*nb_multiply*/
 | |
|     0,                                  /*nb_remainder*/
 | |
|     0,                                  /*nb_divmod*/
 | |
|     0,                                  /*nb_power*/
 | |
|     0,                                  /*nb_negative*/
 | |
|     0,                                  /*nb_positive*/
 | |
|     0,                                  /*nb_absolute*/
 | |
|     0,                                  /*nb_bool*/
 | |
|     0,                                  /*nb_invert*/
 | |
|     0,                                  /*nb_lshift*/
 | |
|     0,                                  /*nb_rshift*/
 | |
|     (binaryfunc)dictviews_and,          /*nb_and*/
 | |
|     (binaryfunc)dictviews_xor,          /*nb_xor*/
 | |
|     (binaryfunc)dictviews_or,           /*nb_or*/
 | |
| };
 | |
| 
 | |
| static PyObject*
 | |
| dictviews_isdisjoint(PyObject *self, PyObject *other)
 | |
| {
 | |
|     PyObject *it;
 | |
|     PyObject *item = NULL;
 | |
| 
 | |
|     if (self == other) {
 | |
|         if (dictview_len((dictviewobject *)self) == 0)
 | |
|             Py_RETURN_TRUE;
 | |
|         else
 | |
|             Py_RETURN_FALSE;
 | |
|     }
 | |
| 
 | |
|     /* Iterate over the shorter object (only if other is a set,
 | |
|      * because PySequence_Contains may be expensive otherwise): */
 | |
|     if (PyAnySet_Check(other) || PyDictViewSet_Check(other)) {
 | |
|         Py_ssize_t len_self = dictview_len((dictviewobject *)self);
 | |
|         Py_ssize_t len_other = PyObject_Size(other);
 | |
|         if (len_other == -1)
 | |
|             return NULL;
 | |
| 
 | |
|         if ((len_other > len_self)) {
 | |
|             PyObject *tmp = other;
 | |
|             other = self;
 | |
|             self = tmp;
 | |
|         }
 | |
|     }
 | |
| 
 | |
|     it = PyObject_GetIter(other);
 | |
|     if (it == NULL)
 | |
|         return NULL;
 | |
| 
 | |
|     while ((item = PyIter_Next(it)) != NULL) {
 | |
|         int contains = PySequence_Contains(self, item);
 | |
|         Py_DECREF(item);
 | |
|         if (contains == -1) {
 | |
|             Py_DECREF(it);
 | |
|             return NULL;
 | |
|         }
 | |
| 
 | |
|         if (contains) {
 | |
|             Py_DECREF(it);
 | |
|             Py_RETURN_FALSE;
 | |
|         }
 | |
|     }
 | |
|     Py_DECREF(it);
 | |
|     if (PyErr_Occurred())
 | |
|         return NULL; /* PyIter_Next raised an exception. */
 | |
|     Py_RETURN_TRUE;
 | |
| }
 | |
| 
 | |
| PyDoc_STRVAR(isdisjoint_doc,
 | |
| "Return True if the view and the given iterable have a null intersection.");
 | |
| 
 | |
| static PyMethodDef dictkeys_methods[] = {
 | |
|     {"isdisjoint",      (PyCFunction)dictviews_isdisjoint,  METH_O,
 | |
|      isdisjoint_doc},
 | |
|     {NULL,              NULL}           /* sentinel */
 | |
| };
 | |
| 
 | |
| PyTypeObject PyDictKeys_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_keys",                                /* tp_name */
 | |
|     sizeof(dictviewobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictview_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     (reprfunc)dictview_repr,                    /* tp_repr */
 | |
|     &dictviews_as_number,                       /* tp_as_number */
 | |
|     &dictkeys_as_sequence,                      /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictview_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     dictview_richcompare,                       /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     (getiterfunc)dictkeys_iter,                 /* tp_iter */
 | |
|     0,                                          /* tp_iternext */
 | |
|     dictkeys_methods,                           /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| static PyObject *
 | |
| dictkeys_new(PyObject *dict)
 | |
| {
 | |
|     return dictview_new(dict, &PyDictKeys_Type);
 | |
| }
 | |
| 
 | |
| /*** dict_items ***/
 | |
| 
 | |
| static PyObject *
 | |
| dictitems_iter(dictviewobject *dv)
 | |
| {
 | |
|     if (dv->dv_dict == NULL) {
 | |
|         Py_RETURN_NONE;
 | |
|     }
 | |
|     return dictiter_new(dv->dv_dict, &PyDictIterItem_Type);
 | |
| }
 | |
| 
 | |
| static int
 | |
| dictitems_contains(dictviewobject *dv, PyObject *obj)
 | |
| {
 | |
|     PyObject *key, *value, *found;
 | |
|     if (dv->dv_dict == NULL)
 | |
|         return 0;
 | |
|     if (!PyTuple_Check(obj) || PyTuple_GET_SIZE(obj) != 2)
 | |
|         return 0;
 | |
|     key = PyTuple_GET_ITEM(obj, 0);
 | |
|     value = PyTuple_GET_ITEM(obj, 1);
 | |
|     found = PyDict_GetItem((PyObject *)dv->dv_dict, key);
 | |
|     if (found == NULL) {
 | |
|         if (PyErr_Occurred())
 | |
|             return -1;
 | |
|         return 0;
 | |
|     }
 | |
|     return PyObject_RichCompareBool(value, found, Py_EQ);
 | |
| }
 | |
| 
 | |
| static PySequenceMethods dictitems_as_sequence = {
 | |
|     (lenfunc)dictview_len,              /* sq_length */
 | |
|     0,                                  /* sq_concat */
 | |
|     0,                                  /* sq_repeat */
 | |
|     0,                                  /* sq_item */
 | |
|     0,                                  /* sq_slice */
 | |
|     0,                                  /* sq_ass_item */
 | |
|     0,                                  /* sq_ass_slice */
 | |
|     (objobjproc)dictitems_contains,     /* sq_contains */
 | |
| };
 | |
| 
 | |
| static PyMethodDef dictitems_methods[] = {
 | |
|     {"isdisjoint",      (PyCFunction)dictviews_isdisjoint,  METH_O,
 | |
|      isdisjoint_doc},
 | |
|     {NULL,              NULL}           /* sentinel */
 | |
| };
 | |
| 
 | |
| PyTypeObject PyDictItems_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_items",                               /* tp_name */
 | |
|     sizeof(dictviewobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictview_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     (reprfunc)dictview_repr,                    /* tp_repr */
 | |
|     &dictviews_as_number,                       /* tp_as_number */
 | |
|     &dictitems_as_sequence,                     /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictview_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     dictview_richcompare,                       /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     (getiterfunc)dictitems_iter,                /* tp_iter */
 | |
|     0,                                          /* tp_iternext */
 | |
|     dictitems_methods,                          /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| static PyObject *
 | |
| dictitems_new(PyObject *dict)
 | |
| {
 | |
|     return dictview_new(dict, &PyDictItems_Type);
 | |
| }
 | |
| 
 | |
| /*** dict_values ***/
 | |
| 
 | |
| static PyObject *
 | |
| dictvalues_iter(dictviewobject *dv)
 | |
| {
 | |
|     if (dv->dv_dict == NULL) {
 | |
|         Py_RETURN_NONE;
 | |
|     }
 | |
|     return dictiter_new(dv->dv_dict, &PyDictIterValue_Type);
 | |
| }
 | |
| 
 | |
| static PySequenceMethods dictvalues_as_sequence = {
 | |
|     (lenfunc)dictview_len,              /* sq_length */
 | |
|     0,                                  /* sq_concat */
 | |
|     0,                                  /* sq_repeat */
 | |
|     0,                                  /* sq_item */
 | |
|     0,                                  /* sq_slice */
 | |
|     0,                                  /* sq_ass_item */
 | |
|     0,                                  /* sq_ass_slice */
 | |
|     (objobjproc)0,                      /* sq_contains */
 | |
| };
 | |
| 
 | |
| static PyMethodDef dictvalues_methods[] = {
 | |
|     {NULL,              NULL}           /* sentinel */
 | |
| };
 | |
| 
 | |
| PyTypeObject PyDictValues_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "dict_values",                              /* tp_name */
 | |
|     sizeof(dictviewobject),                     /* tp_basicsize */
 | |
|     0,                                          /* tp_itemsize */
 | |
|     /* methods */
 | |
|     (destructor)dictview_dealloc,               /* tp_dealloc */
 | |
|     0,                                          /* tp_print */
 | |
|     0,                                          /* tp_getattr */
 | |
|     0,                                          /* tp_setattr */
 | |
|     0,                                          /* tp_reserved */
 | |
|     (reprfunc)dictview_repr,                    /* tp_repr */
 | |
|     0,                                          /* tp_as_number */
 | |
|     &dictvalues_as_sequence,                    /* tp_as_sequence */
 | |
|     0,                                          /* tp_as_mapping */
 | |
|     0,                                          /* tp_hash */
 | |
|     0,                                          /* tp_call */
 | |
|     0,                                          /* tp_str */
 | |
|     PyObject_GenericGetAttr,                    /* tp_getattro */
 | |
|     0,                                          /* tp_setattro */
 | |
|     0,                                          /* tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
 | |
|     0,                                          /* tp_doc */
 | |
|     (traverseproc)dictview_traverse,            /* tp_traverse */
 | |
|     0,                                          /* tp_clear */
 | |
|     0,                                          /* tp_richcompare */
 | |
|     0,                                          /* tp_weaklistoffset */
 | |
|     (getiterfunc)dictvalues_iter,               /* tp_iter */
 | |
|     0,                                          /* tp_iternext */
 | |
|     dictvalues_methods,                         /* tp_methods */
 | |
|     0,
 | |
| };
 | |
| 
 | |
| static PyObject *
 | |
| dictvalues_new(PyObject *dict)
 | |
| {
 | |
|     return dictview_new(dict, &PyDictValues_Type);
 | |
| }
 | |
| 
 | |
| /* Returns NULL if cannot allocate a new PyDictKeysObject,
 | |
|    but does not set an error */
 | |
| PyDictKeysObject *
 | |
| _PyDict_NewKeysForClass(void)
 | |
| {
 | |
|     PyDictKeysObject *keys = new_keys_object(PyDict_MINSIZE_SPLIT);
 | |
|     if (keys == NULL)
 | |
|         PyErr_Clear();
 | |
|     else
 | |
|         keys->dk_lookup = lookdict_split;
 | |
|     return keys;
 | |
| }
 | |
| 
 | |
| #define CACHED_KEYS(tp) (((PyHeapTypeObject*)tp)->ht_cached_keys)
 | |
| 
 | |
| PyObject *
 | |
| PyObject_GenericGetDict(PyObject *obj, void *context)
 | |
| {
 | |
|     PyObject *dict, **dictptr = _PyObject_GetDictPtr(obj);
 | |
|     if (dictptr == NULL) {
 | |
|         PyErr_SetString(PyExc_AttributeError,
 | |
|                         "This object has no __dict__");
 | |
|         return NULL;
 | |
|     }
 | |
|     dict = *dictptr;
 | |
|     if (dict == NULL) {
 | |
|         PyTypeObject *tp = Py_TYPE(obj);
 | |
|         if ((tp->tp_flags & Py_TPFLAGS_HEAPTYPE) && CACHED_KEYS(tp)) {
 | |
|             DK_INCREF(CACHED_KEYS(tp));
 | |
|             *dictptr = dict = new_dict_with_shared_keys(CACHED_KEYS(tp));
 | |
|         }
 | |
|         else {
 | |
|             *dictptr = dict = PyDict_New();
 | |
|         }
 | |
|     }
 | |
|     Py_XINCREF(dict);
 | |
|     return dict;
 | |
| }
 | |
| 
 | |
| int
 | |
| _PyObjectDict_SetItem(PyTypeObject *tp, PyObject **dictptr,
 | |
|                      PyObject *key, PyObject *value)
 | |
| {
 | |
|     PyObject *dict;
 | |
|     int res;
 | |
|     PyDictKeysObject *cached;
 | |
| 
 | |
|     assert(dictptr != NULL);
 | |
|     if ((tp->tp_flags & Py_TPFLAGS_HEAPTYPE) && (cached = CACHED_KEYS(tp))) {
 | |
|         assert(dictptr != NULL);
 | |
|         dict = *dictptr;
 | |
|         if (dict == NULL) {
 | |
|             DK_INCREF(cached);
 | |
|             dict = new_dict_with_shared_keys(cached);
 | |
|             if (dict == NULL)
 | |
|                 return -1;
 | |
|             *dictptr = dict;
 | |
|         }
 | |
|         if (value == NULL) {
 | |
|             res = PyDict_DelItem(dict, key);
 | |
|             if (cached != ((PyDictObject *)dict)->ma_keys) {
 | |
|                 CACHED_KEYS(tp) = NULL;
 | |
|                 DK_DECREF(cached);
 | |
|             }
 | |
|         } else {
 | |
|             res = PyDict_SetItem(dict, key, value);
 | |
|             if (cached != ((PyDictObject *)dict)->ma_keys) {
 | |
|                 /* Either update tp->ht_cached_keys or delete it */
 | |
|                 if (cached->dk_refcnt == 1) {
 | |
|                     CACHED_KEYS(tp) = make_keys_shared(dict);
 | |
|                 } else {
 | |
|                     CACHED_KEYS(tp) = NULL;
 | |
|                 }
 | |
|                 DK_DECREF(cached);
 | |
|                 if (CACHED_KEYS(tp) == NULL && PyErr_Occurred())
 | |
|                     return -1;
 | |
|             }
 | |
|         }
 | |
|     } else {
 | |
|         dict = *dictptr;
 | |
|         if (dict == NULL) {
 | |
|             dict = PyDict_New();
 | |
|             if (dict == NULL)
 | |
|                 return -1;
 | |
|             *dictptr = dict;
 | |
|         }
 | |
|         if (value == NULL) {
 | |
|             res = PyDict_DelItem(dict, key);
 | |
|         } else {
 | |
|             res = PyDict_SetItem(dict, key, value);
 | |
|         }
 | |
|     }
 | |
|     return res;
 | |
| }
 | |
| 
 | |
| void
 | |
| _PyDictKeys_DecRef(PyDictKeysObject *keys)
 | |
| {
 | |
|     DK_DECREF(keys);
 | |
| }
 | |
| 
 | |
| 
 | |
| /* ARGSUSED */
 | |
| static PyObject *
 | |
| dummy_repr(PyObject *op)
 | |
| {
 | |
|     return PyUnicode_FromString("<dummy key>");
 | |
| }
 | |
| 
 | |
| /* ARGUSED */
 | |
| static void
 | |
| dummy_dealloc(PyObject* ignore)
 | |
| {
 | |
|     /* This should never get called, but we also don't want to SEGV if
 | |
|      * we accidentally decref dummy-key out of existence.
 | |
|      */
 | |
|     Py_FatalError("deallocating <dummy key>");
 | |
| }
 | |
| 
 | |
| static PyTypeObject PyDictDummy_Type = {
 | |
|     PyVarObject_HEAD_INIT(&PyType_Type, 0)
 | |
|     "<dummy key> type",
 | |
|     0,
 | |
|     0,
 | |
|     dummy_dealloc,      /*tp_dealloc*/ /*never called*/
 | |
|     0,                  /*tp_print*/
 | |
|     0,                  /*tp_getattr*/
 | |
|     0,                  /*tp_setattr*/
 | |
|     0,                  /*tp_reserved*/
 | |
|     dummy_repr,         /*tp_repr*/
 | |
|     0,                  /*tp_as_number*/
 | |
|     0,                  /*tp_as_sequence*/
 | |
|     0,                  /*tp_as_mapping*/
 | |
|     0,                  /*tp_hash */
 | |
|     0,                  /*tp_call */
 | |
|     0,                  /*tp_str */
 | |
|     0,                  /*tp_getattro */
 | |
|     0,                  /*tp_setattro */
 | |
|     0,                  /*tp_as_buffer */
 | |
|     Py_TPFLAGS_DEFAULT, /*tp_flags */
 | |
| };
 | |
| 
 | |
| static PyObject _dummy_struct = {
 | |
|   _PyObject_EXTRA_INIT
 | |
|   2, &PyDictDummy_Type
 | |
| };
 | |
| 
 | 
