| /* 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. |
| */ |
| |
| /* PyDictKeysObject |
| |
| This implements the dictionary's hashtable. |
| |
| As of Python 3.6, this is compact and ordered. Basic idea is described here: |
| * https://mail.python.org/pipermail/python-dev/2012-December/123028.html |
| * https://morepypy.blogspot.com/2015/01/faster-more-memory-efficient-and-more.html |
| |
| layout: |
| |
| +---------------+ |
| | dk_refcnt | |
| | dk_size | |
| | dk_lookup | |
| | dk_usable | |
| | dk_nentries | |
| +---------------+ |
| | dk_indices | |
| | | |
| +---------------+ |
| | dk_entries | |
| | | |
| +---------------+ |
| |
| dk_indices is actual hashtable. It holds index in entries, or DKIX_EMPTY(-1) |
| or DKIX_DUMMY(-2). |
| Size of indices is dk_size. Type of each index in indices is vary on dk_size: |
| |
| * int8 for dk_size <= 128 |
| * int16 for 256 <= dk_size <= 2**15 |
| * int32 for 2**16 <= dk_size <= 2**31 |
| * int64 for 2**32 <= dk_size |
| |
| dk_entries is array of PyDictKeyEntry. Its size is USABLE_FRACTION(dk_size). |
| DK_ENTRIES(dk) can be used to get pointer to entries. |
| |
| NOTE: Since negative value is used for DKIX_EMPTY and DKIX_DUMMY, type of |
| dk_indices entry is signed integer and int16 is used for table which |
| dk_size == 256. |
| */ |
| |
| |
| /* |
| 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. |
| Or: |
| A split table: |
| ma_values != NULL, dk_refcnt >= 1 |
| Values are stored in the ma_values array. |
| Only string (unicode) keys are allowed. |
| All dicts sharing same key must have same insertion order. |
| |
| There are four kinds of slots in the table (slot is index, and |
| DK_ENTRIES(keys)[index] if index >= 0): |
| |
| 1. Unused. index == DKIX_EMPTY |
| Does not hold an active (key, value) pair now and never did. Unused can |
| transition to Active upon key insertion. This is each slot's initial state. |
| |
| 2. Active. index >= 0, me_key != NULL 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. index == DKIX_DUMMY (combined only) |
| 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 |
| else the probe sequence in case of collision would have no way to know |
| they were once active. |
| |
| 4. Pending. index >= 0, key != NULL, and value == NULL (split only) |
| Not yet inserted in split-table. |
| */ |
| |
| /* |
| Preserving insertion order |
| |
| It's simple for combined table. Since dk_entries is mostly append only, we can |
| get insertion order by just iterating dk_entries. |
| |
| One exception is .popitem(). It removes last item in dk_entries and decrement |
| dk_nentries to achieve amortized O(1). Since there are DKIX_DUMMY remains in |
| dk_indices, we can't increment dk_usable even though dk_nentries is |
| decremented. |
| |
| In split table, inserting into pending entry is allowed only for dk_entries[ix] |
| where ix == mp->ma_used. Inserting into other index and deleting item cause |
| converting the dict to the combined table. |
| */ |
| |
| /* PyDict_MINSIZE is the starting size for any new 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 8 |
| |
| #include "Python.h" |
| #include "pycore_gc.h" // _PyObject_GC_IS_TRACKED() |
| #include "pycore_object.h" // _PyObject_GC_TRACK() |
| #include "pycore_pyerrors.h" // _PyErr_Fetch() |
| #include "pycore_pystate.h" // _PyThreadState_GET() |
| #include "dict-common.h" |
| #include "stringlib/eq.h" // unicode_eq() |
| |
| /*[clinic input] |
| class dict "PyDictObject *" "&PyDict_Type" |
| [clinic start generated code]*/ |
| /*[clinic end generated code: output=da39a3ee5e6b4b0d input=f157a5a0ce9589d6]*/ |
| |
| |
| /* |
| 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 ints) are very regular in common |
| cases: |
| |
| >>>[hash(i) for i in range(4)] |
| [0, 1, 2, 3] |
| |
| 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. 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: |
| |
| perturb >>= PERTURB_SHIFT; |
| j = (5*j) + 1 + perturb; |
| 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. |
| |
| */ |
| |
| /* forward declarations */ |
| static Py_ssize_t lookdict(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr); |
| static Py_ssize_t lookdict_unicode(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr); |
| static Py_ssize_t |
| lookdict_unicode_nodummy(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr); |
| static Py_ssize_t lookdict_split(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr); |
| |
| static int dictresize(PyDictObject *mp, Py_ssize_t minused); |
| |
| static PyObject* dict_iter(PyDictObject *dict); |
| |
| /*Global counter used to set ma_version_tag field of dictionary. |
| * It is incremented each time that a dictionary is created and each |
| * time that a dictionary is modified. */ |
| static uint64_t pydict_global_version = 0; |
| |
| #define DICT_NEXT_VERSION() (++pydict_global_version) |
| |
| #include "clinic/dictobject.c.h" |
| |
| |
| static struct _Py_dict_state * |
| get_dict_state(void) |
| { |
| PyInterpreterState *interp = _PyInterpreterState_GET(); |
| return &interp->dict_state; |
| } |
| |
| |
| void |
| _PyDict_ClearFreeList(PyThreadState *tstate) |
| { |
| struct _Py_dict_state *state = &tstate->interp->dict_state; |
| while (state->numfree) { |
| PyDictObject *op = state->free_list[--state->numfree]; |
| assert(PyDict_CheckExact(op)); |
| PyObject_GC_Del(op); |
| } |
| while (state->keys_numfree) { |
| PyObject_FREE(state->keys_free_list[--state->keys_numfree]); |
| } |
| } |
| |
| |
| void |
| _PyDict_Fini(PyThreadState *tstate) |
| { |
| _PyDict_ClearFreeList(tstate); |
| #ifdef Py_DEBUG |
| struct _Py_dict_state *state = get_dict_state(); |
| state->numfree = -1; |
| state->keys_numfree = -1; |
| #endif |
| } |
| |
| |
| /* Print summary info about the state of the optimized allocator */ |
| void |
| _PyDict_DebugMallocStats(FILE *out) |
| { |
| struct _Py_dict_state *state = get_dict_state(); |
| _PyDebugAllocatorStats(out, "free PyDictObject", |
| state->numfree, sizeof(PyDictObject)); |
| } |
| |
| |
| #define DK_SIZE(dk) ((dk)->dk_size) |
| #if SIZEOF_VOID_P > 4 |
| #define DK_IXSIZE(dk) \ |
| (DK_SIZE(dk) <= 0xff ? \ |
| 1 : DK_SIZE(dk) <= 0xffff ? \ |
| 2 : DK_SIZE(dk) <= 0xffffffff ? \ |
| 4 : sizeof(int64_t)) |
| #else |
| #define DK_IXSIZE(dk) \ |
| (DK_SIZE(dk) <= 0xff ? \ |
| 1 : DK_SIZE(dk) <= 0xffff ? \ |
| 2 : sizeof(int32_t)) |
| #endif |
| #define DK_ENTRIES(dk) \ |
| ((PyDictKeyEntry*)(&((int8_t*)((dk)->dk_indices))[DK_SIZE(dk) * DK_IXSIZE(dk)])) |
| |
| #define DK_MASK(dk) (((dk)->dk_size)-1) |
| #define IS_POWER_OF_2(x) (((x) & (x-1)) == 0) |
| |
| static void free_keys_object(PyDictKeysObject *keys); |
| |
| static inline void |
| dictkeys_incref(PyDictKeysObject *dk) |
| { |
| #ifdef Py_REF_DEBUG |
| _Py_RefTotal++; |
| #endif |
| dk->dk_refcnt++; |
| } |
| |
| static inline void |
| dictkeys_decref(PyDictKeysObject *dk) |
| { |
| assert(dk->dk_refcnt > 0); |
| #ifdef Py_REF_DEBUG |
| _Py_RefTotal--; |
| #endif |
| if (--dk->dk_refcnt == 0) { |
| free_keys_object(dk); |
| } |
| } |
| |
| /* lookup indices. returns DKIX_EMPTY, DKIX_DUMMY, or ix >=0 */ |
| static inline Py_ssize_t |
| dictkeys_get_index(const PyDictKeysObject *keys, Py_ssize_t i) |
| { |
| Py_ssize_t s = DK_SIZE(keys); |
| Py_ssize_t ix; |
| |
| if (s <= 0xff) { |
| const int8_t *indices = (const int8_t*)(keys->dk_indices); |
| ix = indices[i]; |
| } |
| else if (s <= 0xffff) { |
| const int16_t *indices = (const int16_t*)(keys->dk_indices); |
| ix = indices[i]; |
| } |
| #if SIZEOF_VOID_P > 4 |
| else if (s > 0xffffffff) { |
| const int64_t *indices = (const int64_t*)(keys->dk_indices); |
| ix = indices[i]; |
| } |
| #endif |
| else { |
| const int32_t *indices = (const int32_t*)(keys->dk_indices); |
| ix = indices[i]; |
| } |
| assert(ix >= DKIX_DUMMY); |
| return ix; |
| } |
| |
| /* write to indices. */ |
| static inline void |
| dictkeys_set_index(PyDictKeysObject *keys, Py_ssize_t i, Py_ssize_t ix) |
| { |
| Py_ssize_t s = DK_SIZE(keys); |
| |
| assert(ix >= DKIX_DUMMY); |
| |
| if (s <= 0xff) { |
| int8_t *indices = (int8_t*)(keys->dk_indices); |
| assert(ix <= 0x7f); |
| indices[i] = (char)ix; |
| } |
| else if (s <= 0xffff) { |
| int16_t *indices = (int16_t*)(keys->dk_indices); |
| assert(ix <= 0x7fff); |
| indices[i] = (int16_t)ix; |
| } |
| #if SIZEOF_VOID_P > 4 |
| else if (s > 0xffffffff) { |
| int64_t *indices = (int64_t*)(keys->dk_indices); |
| indices[i] = ix; |
| } |
| #endif |
| else { |
| int32_t *indices = (int32_t*)(keys->dk_indices); |
| assert(ix <= 0x7fffffff); |
| indices[i] = (int32_t)ix; |
| } |
| } |
| |
| |
| /* USABLE_FRACTION is the maximum dictionary load. |
| * Increasing this ratio makes dictionaries more dense resulting in more |
| * collisions. Decreasing it improves sparseness at the expense of spreading |
| * indices 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 quick to calculate. |
| * Fractions around 1/2 to 2/3 seem to work well in practice. |
| */ |
| #define USABLE_FRACTION(n) (((n) << 1)/3) |
| |
| /* ESTIMATE_SIZE is reverse function of USABLE_FRACTION. |
| * This can be used to reserve enough size to insert n entries without |
| * resizing. |
| */ |
| #define ESTIMATE_SIZE(n) (((n)*3+1) >> 1) |
| |
| /* Alternative fraction that is otherwise close enough to 2n/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*3. |
| * 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. See also bpo-17563 and bpo-33205. |
| * |
| * GROWTH_RATE was set to used*4 up to version 3.2. |
| * GROWTH_RATE was set to used*2 in version 3.3.0 |
| * GROWTH_RATE was set to used*2 + capacity/2 in 3.4.0-3.6.0. |
| */ |
| #define GROWTH_RATE(d) ((d)->ma_used*3) |
| |
| #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 = { |
| 1, /* dk_refcnt */ |
| 1, /* dk_size */ |
| lookdict_split, /* dk_lookup */ |
| 0, /* dk_usable (immutable) */ |
| 0, /* dk_nentries */ |
| {DKIX_EMPTY, DKIX_EMPTY, DKIX_EMPTY, DKIX_EMPTY, |
| DKIX_EMPTY, DKIX_EMPTY, DKIX_EMPTY, DKIX_EMPTY}, /* dk_indices */ |
| }; |
| |
| static PyObject *empty_values[1] = { NULL }; |
| |
| #define Py_EMPTY_KEYS &empty_keys_struct |
| |
| /* Uncomment to check the dict content in _PyDict_CheckConsistency() */ |
| /* #define DEBUG_PYDICT */ |
| |
| #ifdef DEBUG_PYDICT |
| # define ASSERT_CONSISTENT(op) assert(_PyDict_CheckConsistency((PyObject *)(op), 1)) |
| #else |
| # define ASSERT_CONSISTENT(op) assert(_PyDict_CheckConsistency((PyObject *)(op), 0)) |
| #endif |
| |
| |
| int |
| _PyDict_CheckConsistency(PyObject *op, int check_content) |
| { |
| #define CHECK(expr) \ |
| do { if (!(expr)) { _PyObject_ASSERT_FAILED_MSG(op, Py_STRINGIFY(expr)); } } while (0) |
| |
| assert(op != NULL); |
| CHECK(PyDict_Check(op)); |
| PyDictObject *mp = (PyDictObject *)op; |
| |
| PyDictKeysObject *keys = mp->ma_keys; |
| int splitted = _PyDict_HasSplitTable(mp); |
| Py_ssize_t usable = USABLE_FRACTION(keys->dk_size); |
| |
| CHECK(0 <= mp->ma_used && mp->ma_used <= usable); |
| CHECK(IS_POWER_OF_2(keys->dk_size)); |
| CHECK(0 <= keys->dk_usable && keys->dk_usable <= usable); |
| CHECK(0 <= keys->dk_nentries && keys->dk_nentries <= usable); |
| CHECK(keys->dk_usable + keys->dk_nentries <= usable); |
| |
| if (!splitted) { |
| /* combined table */ |
| CHECK(keys->dk_refcnt == 1); |
| } |
| |
| if (check_content) { |
| PyDictKeyEntry *entries = DK_ENTRIES(keys); |
| Py_ssize_t i; |
| |
| for (i=0; i < keys->dk_size; i++) { |
| Py_ssize_t ix = dictkeys_get_index(keys, i); |
| CHECK(DKIX_DUMMY <= ix && ix <= usable); |
| } |
| |
| for (i=0; i < usable; i++) { |
| PyDictKeyEntry *entry = &entries[i]; |
| PyObject *key = entry->me_key; |
| |
| if (key != NULL) { |
| if (PyUnicode_CheckExact(key)) { |
| Py_hash_t hash = ((PyASCIIObject *)key)->hash; |
| CHECK(hash != -1); |
| CHECK(entry->me_hash == hash); |
| } |
| else { |
| /* test_dict fails if PyObject_Hash() is called again */ |
| CHECK(entry->me_hash != -1); |
| } |
| if (!splitted) { |
| CHECK(entry->me_value != NULL); |
| } |
| } |
| |
| if (splitted) { |
| CHECK(entry->me_value == NULL); |
| } |
| } |
| |
| if (splitted) { |
| /* splitted table */ |
| for (i=0; i < mp->ma_used; i++) { |
| CHECK(mp->ma_values[i] != NULL); |
| } |
| } |
| } |
| return 1; |
| |
| #undef CHECK |
| } |
| |
| |
| static PyDictKeysObject* |
| new_keys_object(Py_ssize_t size) |
| { |
| PyDictKeysObject *dk; |
| Py_ssize_t es, usable; |
| |
| assert(size >= PyDict_MINSIZE); |
| assert(IS_POWER_OF_2(size)); |
| |
| usable = USABLE_FRACTION(size); |
| if (size <= 0xff) { |
| es = 1; |
| } |
| else if (size <= 0xffff) { |
| es = 2; |
| } |
| #if SIZEOF_VOID_P > 4 |
| else if (size <= 0xffffffff) { |
| es = 4; |
| } |
| #endif |
| else { |
| es = sizeof(Py_ssize_t); |
| } |
| |
| struct _Py_dict_state *state = get_dict_state(); |
| #ifdef Py_DEBUG |
| // new_keys_object() must not be called after _PyDict_Fini() |
| assert(state->keys_numfree != -1); |
| #endif |
| if (size == PyDict_MINSIZE && state->keys_numfree > 0) { |
| dk = state->keys_free_list[--state->keys_numfree]; |
| } |
| else |
| { |
| dk = PyObject_MALLOC(sizeof(PyDictKeysObject) |
| + es * size |
| + sizeof(PyDictKeyEntry) * usable); |
| if (dk == NULL) { |
| PyErr_NoMemory(); |
| return NULL; |
| } |
| } |
| #ifdef Py_REF_DEBUG |
| _Py_RefTotal++; |
| #endif |
| dk->dk_refcnt = 1; |
| dk->dk_size = size; |
| dk->dk_usable = usable; |
| dk->dk_lookup = lookdict_unicode_nodummy; |
| dk->dk_nentries = 0; |
| memset(&dk->dk_indices[0], 0xff, es * size); |
| memset(DK_ENTRIES(dk), 0, sizeof(PyDictKeyEntry) * usable); |
| return dk; |
| } |
| |
| static void |
| free_keys_object(PyDictKeysObject *keys) |
| { |
| PyDictKeyEntry *entries = DK_ENTRIES(keys); |
| Py_ssize_t i, n; |
| for (i = 0, n = keys->dk_nentries; i < n; i++) { |
| Py_XDECREF(entries[i].me_key); |
| Py_XDECREF(entries[i].me_value); |
| } |
| struct _Py_dict_state *state = get_dict_state(); |
| #ifdef Py_DEBUG |
| // free_keys_object() must not be called after _PyDict_Fini() |
| assert(state->keys_numfree != -1); |
| #endif |
| if (keys->dk_size == PyDict_MINSIZE && state->keys_numfree < PyDict_MAXFREELIST) { |
| state->keys_free_list[state->keys_numfree++] = keys; |
| return; |
| } |
| PyObject_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); |
| struct _Py_dict_state *state = get_dict_state(); |
| #ifdef Py_DEBUG |
| // new_dict() must not be called after _PyDict_Fini() |
| assert(state->numfree != -1); |
| #endif |
| if (state->numfree) { |
| mp = state->free_list[--state->numfree]; |
| assert (mp != NULL); |
| assert (Py_IS_TYPE(mp, &PyDict_Type)); |
| _Py_NewReference((PyObject *)mp); |
| } |
| else { |
| mp = PyObject_GC_New(PyDictObject, &PyDict_Type); |
| if (mp == NULL) { |
| dictkeys_decref(keys); |
| if (values != empty_values) { |
| free_values(values); |
| } |
| return NULL; |
| } |
| } |
| mp->ma_keys = keys; |
| mp->ma_values = values; |
| mp->ma_used = 0; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| ASSERT_CONSISTENT(mp); |
| 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 = USABLE_FRACTION(DK_SIZE(keys)); |
| values = new_values(size); |
| if (values == NULL) { |
| dictkeys_decref(keys); |
| return PyErr_NoMemory(); |
| } |
| for (i = 0; i < size; i++) { |
| values[i] = NULL; |
| } |
| return new_dict(keys, values); |
| } |
| |
| |
| static PyObject * |
| clone_combined_dict(PyDictObject *orig) |
| { |
| assert(PyDict_CheckExact(orig)); |
| assert(orig->ma_values == NULL); |
| assert(orig->ma_keys->dk_refcnt == 1); |
| |
| Py_ssize_t keys_size = _PyDict_KeysSize(orig->ma_keys); |
| PyDictKeysObject *keys = PyObject_Malloc(keys_size); |
| if (keys == NULL) { |
| PyErr_NoMemory(); |
| return NULL; |
| } |
| |
| memcpy(keys, orig->ma_keys, keys_size); |
| |
| /* After copying key/value pairs, we need to incref all |
| keys and values and they are about to be co-owned by a |
| new dict object. */ |
| PyDictKeyEntry *ep0 = DK_ENTRIES(keys); |
| Py_ssize_t n = keys->dk_nentries; |
| for (Py_ssize_t i = 0; i < n; i++) { |
| PyDictKeyEntry *entry = &ep0[i]; |
| PyObject *value = entry->me_value; |
| if (value != NULL) { |
| Py_INCREF(value); |
| Py_INCREF(entry->me_key); |
| } |
| } |
| |
| PyDictObject *new = (PyDictObject *)new_dict(keys, NULL); |
| if (new == NULL) { |
| /* In case of an error, `new_dict()` takes care of |
| cleaning up `keys`. */ |
| return NULL; |
| } |
| new->ma_used = orig->ma_used; |
| ASSERT_CONSISTENT(new); |
| if (_PyObject_GC_IS_TRACKED(orig)) { |
| /* Maintain tracking. */ |
| _PyObject_GC_TRACK(new); |
| } |
| |
| /* Since we copied the keys table we now have an extra reference |
| in the system. Manually call increment _Py_RefTotal to signal that |
| we have it now; calling dictkeys_incref would be an error as |
| keys->dk_refcnt is already set to 1 (after memcpy). */ |
| #ifdef Py_REF_DEBUG |
| _Py_RefTotal++; |
| #endif |
| |
| return (PyObject *)new; |
| } |
| |
| PyObject * |
| PyDict_New(void) |
| { |
| dictkeys_incref(Py_EMPTY_KEYS); |
| return new_dict(Py_EMPTY_KEYS, empty_values); |
| } |
| |
| /* Search index of hash table from offset of entry table */ |
| static Py_ssize_t |
| lookdict_index(PyDictKeysObject *k, Py_hash_t hash, Py_ssize_t index) |
| { |
| size_t mask = DK_MASK(k); |
| size_t perturb = (size_t)hash; |
| size_t i = (size_t)hash & mask; |
| |
| for (;;) { |
| Py_ssize_t ix = dictkeys_get_index(k, i); |
| if (ix == index) { |
| return i; |
| } |
| if (ix == DKIX_EMPTY) { |
| return DKIX_EMPTY; |
| } |
| perturb >>= PERTURB_SHIFT; |
| i = mask & (i*5 + perturb + 1); |
| } |
| Py_UNREACHABLE(); |
| } |
| |
| /* |
| 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 DKIX_ERROR if (and only if) a |
| comparison raises an exception. |
| lookdict_unicode() below is specialized to string keys, comparison of which can |
| never raise an exception; that function can never return DKIX_ERROR when key |
| is string. Otherwise, it falls back to lookdict(). |
| lookdict_unicode_nodummy is further specialized for string keys that cannot be |
| the <dummy> value. |
| For both, when the key isn't found a DKIX_EMPTY is returned. |
| */ |
| static Py_ssize_t _Py_HOT_FUNCTION |
| lookdict(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr) |
| { |
| size_t i, mask, perturb; |
| PyDictKeysObject *dk; |
| PyDictKeyEntry *ep0; |
| |
| top: |
| dk = mp->ma_keys; |
| ep0 = DK_ENTRIES(dk); |
| mask = DK_MASK(dk); |
| perturb = hash; |
| i = (size_t)hash & mask; |
| |
| for (;;) { |
| Py_ssize_t ix = dictkeys_get_index(dk, i); |
| if (ix == DKIX_EMPTY) { |
| *value_addr = NULL; |
| return ix; |
| } |
| if (ix >= 0) { |
| PyDictKeyEntry *ep = &ep0[ix]; |
| assert(ep->me_key != NULL); |
| if (ep->me_key == key) { |
| *value_addr = ep->me_value; |
| return ix; |
| } |
| if (ep->me_hash == hash) { |
| PyObject *startkey = ep->me_key; |
| Py_INCREF(startkey); |
| int cmp = PyObject_RichCompareBool(startkey, key, Py_EQ); |
| Py_DECREF(startkey); |
| if (cmp < 0) { |
| *value_addr = NULL; |
| return DKIX_ERROR; |
| } |
| if (dk == mp->ma_keys && ep->me_key == startkey) { |
| if (cmp > 0) { |
| *value_addr = ep->me_value; |
| return ix; |
| } |
| } |
| else { |
| /* The dict was mutated, restart */ |
| goto top; |
| } |
| } |
| } |
| perturb >>= PERTURB_SHIFT; |
| i = (i*5 + perturb + 1) & mask; |
| } |
| Py_UNREACHABLE(); |
| } |
| |
| /* Specialized version for string-only keys */ |
| static Py_ssize_t _Py_HOT_FUNCTION |
| lookdict_unicode(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr) |
| { |
| assert(mp->ma_values == NULL); |
| /* 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); |
| } |
| |
| PyDictKeyEntry *ep0 = DK_ENTRIES(mp->ma_keys); |
| size_t mask = DK_MASK(mp->ma_keys); |
| size_t perturb = (size_t)hash; |
| size_t i = (size_t)hash & mask; |
| |
| for (;;) { |
| Py_ssize_t ix = dictkeys_get_index(mp->ma_keys, i); |
| if (ix == DKIX_EMPTY) { |
| *value_addr = NULL; |
| return DKIX_EMPTY; |
| } |
| if (ix >= 0) { |
| PyDictKeyEntry *ep = &ep0[ix]; |
| assert(ep->me_key != NULL); |
| assert(PyUnicode_CheckExact(ep->me_key)); |
| if (ep->me_key == key || |
| (ep->me_hash == hash && unicode_eq(ep->me_key, key))) { |
| *value_addr = ep->me_value; |
| return ix; |
| } |
| } |
| perturb >>= PERTURB_SHIFT; |
| i = mask & (i*5 + perturb + 1); |
| } |
| Py_UNREACHABLE(); |
| } |
| |
| /* Faster version of lookdict_unicode when it is known that no <dummy> keys |
| * will be present. */ |
| static Py_ssize_t _Py_HOT_FUNCTION |
| lookdict_unicode_nodummy(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr) |
| { |
| assert(mp->ma_values == NULL); |
| /* 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); |
| } |
| |
| PyDictKeyEntry *ep0 = DK_ENTRIES(mp->ma_keys); |
| size_t mask = DK_MASK(mp->ma_keys); |
| size_t perturb = (size_t)hash; |
| size_t i = (size_t)hash & mask; |
| |
| for (;;) { |
| Py_ssize_t ix = dictkeys_get_index(mp->ma_keys, i); |
| assert (ix != DKIX_DUMMY); |
| if (ix == DKIX_EMPTY) { |
| *value_addr = NULL; |
| return DKIX_EMPTY; |
| } |
| PyDictKeyEntry *ep = &ep0[ix]; |
| assert(ep->me_key != NULL); |
| assert(PyUnicode_CheckExact(ep->me_key)); |
| if (ep->me_key == key || |
| (ep->me_hash == hash && unicode_eq(ep->me_key, key))) { |
| *value_addr = ep->me_value; |
| return ix; |
| } |
| perturb >>= PERTURB_SHIFT; |
| i = mask & (i*5 + perturb + 1); |
| } |
| Py_UNREACHABLE(); |
| } |
| |
| /* 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 Py_ssize_t _Py_HOT_FUNCTION |
| lookdict_split(PyDictObject *mp, PyObject *key, |
| Py_hash_t hash, PyObject **value_addr) |
| { |
| /* mp must split table */ |
| assert(mp->ma_values != NULL); |
| if (!PyUnicode_CheckExact(key)) { |
| Py_ssize_t ix = lookdict(mp, key, hash, value_addr); |
| if (ix >= 0) { |
| *value_addr = mp->ma_values[ix]; |
| } |
| return ix; |
| } |
| |
| PyDictKeyEntry *ep0 = DK_ENTRIES(mp->ma_keys); |
| size_t mask = DK_MASK(mp->ma_keys); |
| size_t perturb = (size_t)hash; |
| size_t i = (size_t)hash & mask; |
| |
| for (;;) { |
| Py_ssize_t ix = dictkeys_get_index(mp->ma_keys, i); |
| assert (ix != DKIX_DUMMY); |
| if (ix == DKIX_EMPTY) { |
| *value_addr = NULL; |
| return DKIX_EMPTY; |
| } |
| PyDictKeyEntry *ep = &ep0[ix]; |
| assert(ep->me_key != NULL); |
| assert(PyUnicode_CheckExact(ep->me_key)); |
| if (ep->me_key == key || |
| (ep->me_hash == hash && unicode_eq(ep->me_key, key))) { |
| *value_addr = mp->ma_values[ix]; |
| return ix; |
| } |
| perturb >>= PERTURB_SHIFT; |
| i = mask & (i*5 + perturb + 1); |
| } |
| Py_UNREACHABLE(); |
| } |
| |
| 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, numentries; |
| PyDictKeyEntry *ep0; |
| |
| if (!PyDict_CheckExact(op) || !_PyObject_GC_IS_TRACKED(op)) |
| return; |
| |
| mp = (PyDictObject *) op; |
| ep0 = DK_ENTRIES(mp->ma_keys); |
| numentries = mp->ma_keys->dk_nentries; |
| if (_PyDict_HasSplitTable(mp)) { |
| for (i = 0; i < numentries; i++) { |
| if ((value = mp->ma_values[i]) == NULL) |
| continue; |
| if (_PyObject_GC_MAY_BE_TRACKED(value)) { |
| assert(!_PyObject_GC_MAY_BE_TRACKED(ep0[i].me_key)); |
| return; |
| } |
| } |
| } |
| else { |
| for (i = 0; i < numentries; 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. |
| |
| The dict must be combined. */ |
| static Py_ssize_t |
| find_empty_slot(PyDictKeysObject *keys, Py_hash_t hash) |
| { |
| assert(keys != NULL); |
| |
| const size_t mask = DK_MASK(keys); |
| size_t i = hash & mask; |
| Py_ssize_t ix = dictkeys_get_index(keys, i); |
| for (size_t perturb = hash; ix >= 0;) { |
| perturb >>= PERTURB_SHIFT; |
| i = (i*5 + perturb + 1) & mask; |
| ix = dictkeys_get_index(keys, i); |
| } |
| return i; |
| } |
| |
| 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; |
| PyDictKeyEntry *ep; |
| |
| Py_INCREF(key); |
| Py_INCREF(value); |
| if (mp->ma_values != NULL && !PyUnicode_CheckExact(key)) { |
| if (insertion_resize(mp) < 0) |
| goto Fail; |
| } |
| |
| Py_ssize_t ix = mp->ma_keys->dk_lookup(mp, key, hash, &old_value); |
| if (ix == DKIX_ERROR) |
| goto Fail; |
| |
| assert(PyUnicode_CheckExact(key) || mp->ma_keys->dk_lookup == lookdict); |
| MAINTAIN_TRACKING(mp, key, value); |
| |
| /* When insertion order is different from shared key, we can't share |
| * the key anymore. Convert this instance to combine table. |
| */ |
| if (_PyDict_HasSplitTable(mp) && |
| ((ix >= 0 && old_value == NULL && mp->ma_used != ix) || |
| (ix == DKIX_EMPTY && mp->ma_used != mp->ma_keys->dk_nentries))) { |
| if (insertion_resize(mp) < 0) |
| goto Fail; |
| ix = DKIX_EMPTY; |
| } |
| |
| if (ix == DKIX_EMPTY) { |
| /* Insert into new slot. */ |
| assert(old_value == NULL); |
| if (mp->ma_keys->dk_usable <= 0) { |
| /* Need to resize. */ |
| if (insertion_resize(mp) < 0) |
| goto Fail; |
| } |
| Py_ssize_t hashpos = find_empty_slot(mp->ma_keys, hash); |
| ep = &DK_ENTRIES(mp->ma_keys)[mp->ma_keys->dk_nentries]; |
| dictkeys_set_index(mp->ma_keys, hashpos, mp->ma_keys->dk_nentries); |
| ep->me_key = key; |
| ep->me_hash = hash; |
| if (mp->ma_values) { |
| assert (mp->ma_values[mp->ma_keys->dk_nentries] == NULL); |
| mp->ma_values[mp->ma_keys->dk_nentries] = value; |
| } |
| else { |
| ep->me_value = value; |
| } |
| mp->ma_used++; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| mp->ma_keys->dk_usable--; |
| mp->ma_keys->dk_nentries++; |
| assert(mp->ma_keys->dk_usable >= 0); |
| ASSERT_CONSISTENT(mp); |
| return 0; |
| } |
| |
| if (old_value != value) { |
| if (_PyDict_HasSplitTable(mp)) { |
| mp->ma_values[ix] = value; |
| if (old_value == NULL) { |
| /* pending state */ |
| assert(ix == mp->ma_used); |
| mp->ma_used++; |
| } |
| } |
| else { |
| assert(old_value != NULL); |
| DK_ENTRIES(mp->ma_keys)[ix].me_value = value; |
| } |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| } |
| Py_XDECREF(old_value); /* which **CAN** re-enter (see issue #22653) */ |
| ASSERT_CONSISTENT(mp); |
| Py_DECREF(key); |
| return 0; |
| |
| Fail: |
| Py_DECREF(value); |
| Py_DECREF(key); |
| return -1; |
| } |
| |
| // Same to insertdict but specialized for ma_keys = Py_EMPTY_KEYS. |
| static int |
| insert_to_emptydict(PyDictObject *mp, PyObject *key, Py_hash_t hash, |
| PyObject *value) |
| { |
| assert(mp->ma_keys == Py_EMPTY_KEYS); |
| |
| PyDictKeysObject *newkeys = new_keys_object(PyDict_MINSIZE); |
| if (newkeys == NULL) { |
| return -1; |
| } |
| if (!PyUnicode_CheckExact(key)) { |
| newkeys->dk_lookup = lookdict; |
| } |
| dictkeys_decref(Py_EMPTY_KEYS); |
| mp->ma_keys = newkeys; |
| mp->ma_values = NULL; |
| |
| Py_INCREF(key); |
| Py_INCREF(value); |
| MAINTAIN_TRACKING(mp, key, value); |
| |
| size_t hashpos = (size_t)hash & (PyDict_MINSIZE-1); |
| PyDictKeyEntry *ep = DK_ENTRIES(mp->ma_keys); |
| dictkeys_set_index(mp->ma_keys, hashpos, 0); |
| ep->me_key = key; |
| ep->me_hash = hash; |
| ep->me_value = value; |
| mp->ma_used++; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| mp->ma_keys->dk_usable--; |
| mp->ma_keys->dk_nentries++; |
| return 0; |
| } |
| |
| /* |
| Internal routine used by dictresize() to build a hashtable of entries. |
| */ |
| static void |
| build_indices(PyDictKeysObject *keys, PyDictKeyEntry *ep, Py_ssize_t n) |
| { |
| size_t mask = (size_t)DK_SIZE(keys) - 1; |
| for (Py_ssize_t ix = 0; ix != n; ix++, ep++) { |
| Py_hash_t hash = ep->me_hash; |
| size_t i = hash & mask; |
| for (size_t perturb = hash; dictkeys_get_index(keys, i) != DKIX_EMPTY;) { |
| perturb >>= PERTURB_SHIFT; |
| i = mask & (i*5 + perturb + 1); |
| } |
| dictkeys_set_index(keys, i, ix); |
| } |
| } |
| |
| /* |
| 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 minsize) |
| { |
| Py_ssize_t newsize, numentries; |
| PyDictKeysObject *oldkeys; |
| PyObject **oldvalues; |
| PyDictKeyEntry *oldentries, *newentries; |
| |
| /* Find the smallest table size > minused. */ |
| for (newsize = PyDict_MINSIZE; |
| newsize < minsize && newsize > 0; |
| newsize <<= 1) |
| ; |
| if (newsize <= 0) { |
| PyErr_NoMemory(); |
| return -1; |
| } |
| |
| oldkeys = mp->ma_keys; |
| |
| /* NOTE: Current odict checks mp->ma_keys to detect resize happen. |
| * So we can't reuse oldkeys even if oldkeys->dk_size == newsize. |
| * TODO: Try reusing oldkeys when reimplement odict. |
| */ |
| |
| /* Allocate a new table. */ |
| mp->ma_keys = new_keys_object(newsize); |
| if (mp->ma_keys == NULL) { |
| mp->ma_keys = oldkeys; |
| return -1; |
| } |
| // New table must be large enough. |
| assert(mp->ma_keys->dk_usable >= mp->ma_used); |
| if (oldkeys->dk_lookup == lookdict) |
| mp->ma_keys->dk_lookup = lookdict; |
| |
| numentries = mp->ma_used; |
| oldentries = DK_ENTRIES(oldkeys); |
| newentries = DK_ENTRIES(mp->ma_keys); |
| oldvalues = mp->ma_values; |
| if (oldvalues != NULL) { |
| /* Convert split table into new combined table. |
| * We must incref keys; we can transfer values. |
| * Note that values of split table is always dense. |
| */ |
| for (Py_ssize_t i = 0; i < numentries; i++) { |
| assert(oldvalues[i] != NULL); |
| PyDictKeyEntry *ep = &oldentries[i]; |
| PyObject *key = ep->me_key; |
| Py_INCREF(key); |
| newentries[i].me_key = key; |
| newentries[i].me_hash = ep->me_hash; |
| newentries[i].me_value = oldvalues[i]; |
| } |
| |
| dictkeys_decref(oldkeys); |
| mp->ma_values = NULL; |
| if (oldvalues != empty_values) { |
| free_values(oldvalues); |
| } |
| } |
| else { // combined table. |
| if (oldkeys->dk_nentries == numentries) { |
| memcpy(newentries, oldentries, numentries * sizeof(PyDictKeyEntry)); |
| } |
| else { |
| PyDictKeyEntry *ep = oldentries; |
| for (Py_ssize_t i = 0; i < numentries; i++) { |
| while (ep->me_value == NULL) |
| ep++; |
| newentries[i] = *ep++; |
| } |
| } |
| |
| assert(oldkeys->dk_lookup != lookdict_split); |
| assert(oldkeys->dk_refcnt == 1); |
| #ifdef Py_REF_DEBUG |
| _Py_RefTotal--; |
| #endif |
| struct _Py_dict_state *state = get_dict_state(); |
| #ifdef Py_DEBUG |
| // dictresize() must not be called after _PyDict_Fini() |
| assert(state->keys_numfree != -1); |
| #endif |
| if (oldkeys->dk_size == PyDict_MINSIZE && |
| state->keys_numfree < PyDict_MAXFREELIST) |
| { |
| state->keys_free_list[state->keys_numfree++] = oldkeys; |
| } |
| else { |
| PyObject_FREE(oldkeys); |
| } |
| } |
| |
| build_indices(mp->ma_keys, newentries, numentries); |
| mp->ma_keys->dk_usable -= numentries; |
| mp->ma_keys->dk_nentries = numentries; |
| 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 = DK_ENTRIES(mp->ma_keys); |
| size = USABLE_FRACTION(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; |
| } |
| dictkeys_incref(mp->ma_keys); |
| return mp->ma_keys; |
| } |
| |
| PyObject * |
| _PyDict_NewPresized(Py_ssize_t minused) |
| { |
| const Py_ssize_t max_presize = 128 * 1024; |
| Py_ssize_t newsize; |
| PyDictKeysObject *new_keys; |
| |
| if (minused <= USABLE_FRACTION(PyDict_MINSIZE)) { |
| return PyDict_New(); |
| } |
| /* There are no strict guarantee that returned dict can contain minused |
| * items without resize. So we create medium size dict instead of very |
| * large dict or MemoryError. |
| */ |
| if (minused > USABLE_FRACTION(max_presize)) { |
| newsize = max_presize; |
| } |
| else { |
| Py_ssize_t minsize = ESTIMATE_SIZE(minused); |
| newsize = PyDict_MINSIZE*2; |
| while (newsize < minsize) { |
| newsize <<= 1; |
| } |
| } |
| assert(IS_POWER_OF_2(newsize)); |
| |
| 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) |
| { |
| if (!PyDict_Check(op)) { |
| return NULL; |
| } |
| PyDictObject *mp = (PyDictObject *)op; |
| |
| Py_hash_t hash; |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) |
| { |
| hash = PyObject_Hash(key); |
| if (hash == -1) { |
| PyErr_Clear(); |
| return NULL; |
| } |
| } |
| |
| PyThreadState *tstate = _PyThreadState_GET(); |
| #ifdef Py_DEBUG |
| // bpo-40839: Before Python 3.10, it was possible to call PyDict_GetItem() |
| // with the GIL released. |
| _Py_EnsureTstateNotNULL(tstate); |
| #endif |
| |
| /* Preserve the existing exception */ |
| PyObject *exc_type, *exc_value, *exc_tb; |
| PyObject *value; |
| Py_ssize_t ix; |
| |
| _PyErr_Fetch(tstate, &exc_type, &exc_value, &exc_tb); |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| |
| /* Ignore any exception raised by the lookup */ |
| _PyErr_Restore(tstate, exc_type, exc_value, exc_tb); |
| |
| if (ix < 0) { |
| return NULL; |
| } |
| return value; |
| } |
| |
| /* Same as PyDict_GetItemWithError() but with hash supplied by caller. |
| 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_GetItem_KnownHash(PyObject *op, PyObject *key, Py_hash_t hash) |
| { |
| Py_ssize_t ix; |
| PyDictObject *mp = (PyDictObject *)op; |
| PyObject *value; |
| |
| if (!PyDict_Check(op)) { |
| PyErr_BadInternalCall(); |
| return NULL; |
| } |
| |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix < 0) { |
| return NULL; |
| } |
| return value; |
| } |
| |
| /* 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_ssize_t ix; |
| Py_hash_t hash; |
| PyDictObject*mp = (PyDictObject *)op; |
| PyObject *value; |
| |
| 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; |
| } |
| } |
| |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix < 0) |
| return NULL; |
| return value; |
| } |
| |
| PyObject * |
| _PyDict_GetItemIdWithError(PyObject *dp, struct _Py_Identifier *key) |
| { |
| PyObject *kv; |
| kv = _PyUnicode_FromId(key); /* borrowed */ |
| if (kv == NULL) |
| return NULL; |
| Py_hash_t hash = ((PyASCIIObject *) kv)->hash; |
| assert (hash != -1); /* interned strings have their hash value initialised */ |
| return _PyDict_GetItem_KnownHash(dp, kv, hash); |
| } |
| |
| PyObject * |
| _PyDict_GetItemStringWithError(PyObject *v, const char *key) |
| { |
| PyObject *kv, *rv; |
| kv = PyUnicode_FromString(key); |
| if (kv == NULL) { |
| return NULL; |
| } |
| rv = PyDict_GetItemWithError(v, kv); |
| Py_DECREF(kv); |
| return rv; |
| } |
| |
| /* Fast version of global value lookup (LOAD_GLOBAL). |
| * Lookup in globals, then builtins. |
| * |
| * Raise an exception and return NULL if an error occurred (ex: computing the |
| * key hash failed, key comparison failed, ...). Return NULL if the key doesn't |
| * exist. Return the value if the key exists. |
| */ |
| PyObject * |
| _PyDict_LoadGlobal(PyDictObject *globals, PyDictObject *builtins, PyObject *key) |
| { |
| Py_ssize_t ix; |
| Py_hash_t hash; |
| PyObject *value; |
| |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) |
| { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return NULL; |
| } |
| |
| /* namespace 1: globals */ |
| ix = globals->ma_keys->dk_lookup(globals, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| if (ix != DKIX_EMPTY && value != NULL) |
| return value; |
| |
| /* namespace 2: builtins */ |
| ix = builtins->ma_keys->dk_lookup(builtins, key, hash, &value); |
| if (ix < 0) |
| return NULL; |
| return value; |
| } |
| |
| /* 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; |
| } |
| |
| if (mp->ma_keys == Py_EMPTY_KEYS) { |
| return insert_to_emptydict(mp, key, hash, value); |
| } |
| /* insertdict() handles any resizing that might be necessary */ |
| return insertdict(mp, key, hash, value); |
| } |
| |
| int |
| _PyDict_SetItem_KnownHash(PyObject *op, PyObject *key, PyObject *value, |
| Py_hash_t hash) |
| { |
| PyDictObject *mp; |
| |
| if (!PyDict_Check(op)) { |
| PyErr_BadInternalCall(); |
| return -1; |
| } |
| assert(key); |
| assert(value); |
| assert(hash != -1); |
| mp = (PyDictObject *)op; |
| |
| if (mp->ma_keys == Py_EMPTY_KEYS) { |
| return insert_to_emptydict(mp, key, hash, value); |
| } |
| /* insertdict() handles any resizing that might be necessary */ |
| return insertdict(mp, key, hash, value); |
| } |
| |
| static int |
| delitem_common(PyDictObject *mp, Py_hash_t hash, Py_ssize_t ix, |
| PyObject *old_value) |
| { |
| PyObject *old_key; |
| PyDictKeyEntry *ep; |
| |
| Py_ssize_t hashpos = lookdict_index(mp->ma_keys, hash, ix); |
| assert(hashpos >= 0); |
| |
| mp->ma_used--; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| ep = &DK_ENTRIES(mp->ma_keys)[ix]; |
| dictkeys_set_index(mp->ma_keys, hashpos, DKIX_DUMMY); |
| ENSURE_ALLOWS_DELETIONS(mp); |
| old_key = ep->me_key; |
| ep->me_key = NULL; |
| ep->me_value = NULL; |
| Py_DECREF(old_key); |
| Py_DECREF(old_value); |
| |
| ASSERT_CONSISTENT(mp); |
| return 0; |
| } |
| |
| int |
| PyDict_DelItem(PyObject *op, PyObject *key) |
| { |
| Py_hash_t hash; |
| assert(key); |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return -1; |
| } |
| |
| return _PyDict_DelItem_KnownHash(op, key, hash); |
| } |
| |
| int |
| _PyDict_DelItem_KnownHash(PyObject *op, PyObject *key, Py_hash_t hash) |
| { |
| Py_ssize_t ix; |
| PyDictObject *mp; |
| PyObject *old_value; |
| |
| if (!PyDict_Check(op)) { |
| PyErr_BadInternalCall(); |
| return -1; |
| } |
| assert(key); |
| assert(hash != -1); |
| mp = (PyDictObject *)op; |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| if (ix == DKIX_ERROR) |
| return -1; |
| if (ix == DKIX_EMPTY || old_value == NULL) { |
| _PyErr_SetKeyError(key); |
| return -1; |
| } |
| |
| // Split table doesn't allow deletion. Combine it. |
| if (_PyDict_HasSplitTable(mp)) { |
| if (dictresize(mp, DK_SIZE(mp->ma_keys))) { |
| return -1; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| assert(ix >= 0); |
| } |
| |
| return delitem_common(mp, hash, ix, old_value); |
| } |
| |
| /* This function promises that the predicate -> deletion sequence is atomic |
| * (i.e. protected by the GIL), assuming the predicate itself doesn't |
| * release the GIL. |
| */ |
| int |
| _PyDict_DelItemIf(PyObject *op, PyObject *key, |
| int (*predicate)(PyObject *value)) |
| { |
| Py_ssize_t hashpos, ix; |
| PyDictObject *mp; |
| Py_hash_t hash; |
| PyObject *old_value; |
| int res; |
| |
| if (!PyDict_Check(op)) { |
| PyErr_BadInternalCall(); |
| return -1; |
| } |
| assert(key); |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return -1; |
| mp = (PyDictObject *)op; |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| if (ix == DKIX_ERROR) |
| return -1; |
| if (ix == DKIX_EMPTY || old_value == NULL) { |
| _PyErr_SetKeyError(key); |
| return -1; |
| } |
| |
| // Split table doesn't allow deletion. Combine it. |
| if (_PyDict_HasSplitTable(mp)) { |
| if (dictresize(mp, DK_SIZE(mp->ma_keys))) { |
| return -1; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| assert(ix >= 0); |
| } |
| |
| res = predicate(old_value); |
| if (res == -1) |
| return -1; |
| |
| hashpos = lookdict_index(mp->ma_keys, hash, ix); |
| assert(hashpos >= 0); |
| |
| if (res > 0) |
| return delitem_common(mp, hashpos, ix, old_value); |
| else |
| 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... */ |
| dictkeys_incref(Py_EMPTY_KEYS); |
| mp->ma_keys = Py_EMPTY_KEYS; |
| mp->ma_values = empty_values; |
| mp->ma_used = 0; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| /* ...then clear the keys and values */ |
| if (oldvalues != NULL) { |
| n = oldkeys->dk_nentries; |
| for (i = 0; i < n; i++) |
| Py_CLEAR(oldvalues[i]); |
| free_values(oldvalues); |
| dictkeys_decref(oldkeys); |
| } |
| else { |
| assert(oldkeys->dk_refcnt == 1); |
| dictkeys_decref(oldkeys); |
| } |
| ASSERT_CONSISTENT(mp); |
| } |
| |
| /* Internal version of PyDict_Next that returns a hash value in addition |
| * to the key and value. |
| * Return 1 on success, return 0 when the reached the end of the dictionary |
| * (or if op is not a dictionary) |
| */ |
| int |
| _PyDict_Next(PyObject *op, Py_ssize_t *ppos, PyObject **pkey, |
| PyObject **pvalue, Py_hash_t *phash) |
| { |
| Py_ssize_t i; |
| PyDictObject *mp; |
| PyDictKeyEntry *entry_ptr; |
| PyObject *value; |
| |
| if (!PyDict_Check(op)) |
| return 0; |
| mp = (PyDictObject *)op; |
| i = *ppos; |
| if (mp->ma_values) { |
| if (i < 0 || i >= mp->ma_used) |
| return 0; |
| /* values of split table is always dense */ |
| entry_ptr = &DK_ENTRIES(mp->ma_keys)[i]; |
| value = mp->ma_values[i]; |
| assert(value != NULL); |
| } |
| else { |
| Py_ssize_t n = mp->ma_keys->dk_nentries; |
| if (i < 0 || i >= n) |
| return 0; |
| entry_ptr = &DK_ENTRIES(mp->ma_keys)[i]; |
| while (i < n && entry_ptr->me_value == NULL) { |
| entry_ptr++; |
| i++; |
| } |
| if (i >= n) |
| return 0; |
| value = entry_ptr->me_value; |
| } |
| *ppos = i+1; |
| if (pkey) |
| *pkey = entry_ptr->me_key; |
| if (phash) |
| *phash = entry_ptr->me_hash; |
| if (pvalue) |
| *pvalue = value; |
| return 1; |
| } |
| |
| /* |
| * 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. |
| * } |
| * |
| * Return 1 on success, return 0 when the reached the end of the dictionary |
| * (or if op is not a dictionary) |
| * |
| * 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) |
| { |
| return _PyDict_Next(op, ppos, pkey, pvalue, NULL); |
| } |
| |
| /* Internal version of dict.pop(). */ |
| PyObject * |
| _PyDict_Pop_KnownHash(PyObject *dict, PyObject *key, Py_hash_t hash, PyObject *deflt) |
| { |
| Py_ssize_t ix, hashpos; |
| PyObject *old_value, *old_key; |
| PyDictKeyEntry *ep; |
| PyDictObject *mp; |
| |
| assert(PyDict_Check(dict)); |
| mp = (PyDictObject *)dict; |
| |
| if (mp->ma_used == 0) { |
| if (deflt) { |
| Py_INCREF(deflt); |
| return deflt; |
| } |
| _PyErr_SetKeyError(key); |
| return NULL; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| if (ix == DKIX_EMPTY || old_value == NULL) { |
| if (deflt) { |
| Py_INCREF(deflt); |
| return deflt; |
| } |
| _PyErr_SetKeyError(key); |
| return NULL; |
| } |
| |
| // Split table doesn't allow deletion. Combine it. |
| if (_PyDict_HasSplitTable(mp)) { |
| if (dictresize(mp, DK_SIZE(mp->ma_keys))) { |
| return NULL; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &old_value); |
| assert(ix >= 0); |
| } |
| |
| hashpos = lookdict_index(mp->ma_keys, hash, ix); |
| assert(hashpos >= 0); |
| assert(old_value != NULL); |
| mp->ma_used--; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| dictkeys_set_index(mp->ma_keys, hashpos, DKIX_DUMMY); |
| ep = &DK_ENTRIES(mp->ma_keys)[ix]; |
| ENSURE_ALLOWS_DELETIONS(mp); |
| old_key = ep->me_key; |
| ep->me_key = NULL; |
| ep->me_value = NULL; |
| Py_DECREF(old_key); |
| |
| ASSERT_CONSISTENT(mp); |
| return old_value; |
| } |
| |
| PyObject * |
| _PyDict_Pop(PyObject *dict, PyObject *key, PyObject *deflt) |
| { |
| Py_hash_t hash; |
| |
| if (((PyDictObject *)dict)->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; |
| } |
| return _PyDict_Pop_KnownHash(dict, key, hash, deflt); |
| } |
| |
| /* Internal version of dict.from_keys(). It is subclass-friendly. */ |
| PyObject * |
| _PyDict_FromKeys(PyObject *cls, PyObject *iterable, PyObject *value) |
| { |
| PyObject *it; /* iter(iterable) */ |
| PyObject *key; |
| PyObject *d; |
| int status; |
| |
| d = _PyObject_CallNoArg(cls); |
| 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, ESTIMATE_SIZE(PyDict_GET_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, ESTIMATE_SIZE(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; |
| } |
| |
| /* Methods */ |
| |
| static void |
| dict_dealloc(PyDictObject *mp) |
| { |
| PyObject **values = mp->ma_values; |
| PyDictKeysObject *keys = mp->ma_keys; |
| Py_ssize_t i, n; |
| |
| /* bpo-31095: UnTrack is needed before calling any callbacks */ |
| PyObject_GC_UnTrack(mp); |
| Py_TRASHCAN_BEGIN(mp, dict_dealloc) |
| if (values != NULL) { |
| if (values != empty_values) { |
| for (i = 0, n = mp->ma_keys->dk_nentries; i < n; i++) { |
| Py_XDECREF(values[i]); |
| } |
| free_values(values); |
| } |
| dictkeys_decref(keys); |
| } |
| else if (keys != NULL) { |
| assert(keys->dk_refcnt == 1); |
| dictkeys_decref(keys); |
| } |
| struct _Py_dict_state *state = get_dict_state(); |
| #ifdef Py_DEBUG |
| // new_dict() must not be called after _PyDict_Fini() |
| assert(state->numfree != -1); |
| #endif |
| if (state->numfree < PyDict_MAXFREELIST && Py_IS_TYPE(mp, &PyDict_Type)) { |
| state->free_list[state->numfree++] = mp; |
| } |
| else { |
| Py_TYPE(mp)->tp_free((PyObject *)mp); |
| } |
| Py_TRASHCAN_END |
| } |
| |
| |
| 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) |
| { |
| Py_ssize_t ix; |
| Py_hash_t hash; |
| PyObject *value; |
| |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return NULL; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| if (ix == DKIX_EMPTY || value == 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_CallOneArg(missing, key); |
| Py_DECREF(missing); |
| return res; |
| } |
| else if (PyErr_Occurred()) |
| return NULL; |
| } |
| _PyErr_SetKeyError(key); |
| return NULL; |
| } |
| Py_INCREF(value); |
| return value; |
| } |
| |
| 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 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 = DK_ENTRIES(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; j < n; 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; |
| PyDictKeyEntry *ep; |
| Py_ssize_t 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 = DK_ENTRIES(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; j < n; 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 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 = DK_ENTRIES(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; j < n; 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 |
| / |
| |
| Create a new dictionary with keys from iterable and values set to value. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict_fromkeys_impl(PyTypeObject *type, PyObject *iterable, PyObject *value) |
| /*[clinic end generated code: output=8fb98e4b10384999 input=382ba4855d0f74c3]*/ |
| { |
| return _PyDict_FromKeys((PyObject *)type, iterable, value); |
| } |
| |
| /* Single-arg dict update; used by dict_update_common and operators. */ |
| static int |
| dict_update_arg(PyObject *self, PyObject *arg) |
| { |
| if (PyDict_CheckExact(arg)) { |
| return PyDict_Merge(self, arg, 1); |
| } |
| _Py_IDENTIFIER(keys); |
| PyObject *func; |
| if (_PyObject_LookupAttrId(arg, &PyId_keys, &func) < 0) { |
| return -1; |
| } |
| if (func != NULL) { |
| Py_DECREF(func); |
| return PyDict_Merge(self, arg, 1); |
| } |
| return PyDict_MergeFromSeq2(self, arg, 1); |
| } |
| |
| static int |
| dict_update_common(PyObject *self, PyObject *args, PyObject *kwds, |
| const char *methname) |
| { |
| PyObject *arg = NULL; |
| int result = 0; |
| |
| if (!PyArg_UnpackTuple(args, methname, 0, 1, &arg)) { |
| result = -1; |
| } |
| else if (arg != NULL) { |
| result = dict_update_arg(self, arg); |
| } |
| |
| if (result == 0 && kwds != NULL) { |
| if (PyArg_ValidateKeywordArguments(kwds)) |
| result = PyDict_Merge(self, kwds, 1); |
| else |
| result = -1; |
| } |
| return result; |
| } |
| |
| /* Note: dict.update() uses the METH_VARARGS|METH_KEYWORDS calling convention. |
| Using METH_FASTCALL|METH_KEYWORDS would make dict.update(**dict2) calls |
| slower, see the issue #29312. */ |
| 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); |
| Py_INCREF(key); |
| Py_INCREF(value); |
| if (override) { |
| if (PyDict_SetItem(d, key, value) < 0) { |
| Py_DECREF(key); |
| Py_DECREF(value); |
| goto Fail; |
| } |
| } |
| else if (PyDict_GetItemWithError(d, key) == NULL) { |
| if (PyErr_Occurred() || PyDict_SetItem(d, key, value) < 0) { |
| Py_DECREF(key); |
| Py_DECREF(value); |
| goto Fail; |
| } |
| } |
| |
| Py_DECREF(key); |
| Py_DECREF(value); |
| Py_DECREF(fast); |
| Py_DECREF(item); |
| } |
| |
| i = 0; |
| ASSERT_CONSISTENT(d); |
| goto Return; |
| Fail: |
| Py_XDECREF(item); |
| Py_XDECREF(fast); |
| i = -1; |
| Return: |
| Py_DECREF(it); |
| return Py_SAFE_DOWNCAST(i, Py_ssize_t, int); |
| } |
| |
| static int |
| dict_merge(PyObject *a, PyObject *b, int override) |
| { |
| PyDictObject *mp, *other; |
| Py_ssize_t i, n; |
| PyDictKeyEntry *entry, *ep0; |
| |
| assert(0 <= override && override <= 2); |
| |
| /* 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) && (Py_TYPE(b)->tp_iter == (getiterfunc)dict_iter)) { |
| 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 (USABLE_FRACTION(mp->ma_keys->dk_size) < other->ma_used) { |
| if (dictresize(mp, ESTIMATE_SIZE(mp->ma_used + other->ma_used))) { |
| return -1; |
| } |
| } |
| ep0 = DK_ENTRIES(other->ma_keys); |
| for (i = 0, n = other->ma_keys->dk_nentries; i < n; i++) { |
| PyObject *key, *value; |
| Py_hash_t hash; |
| entry = &ep0[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 == 1) |
| err = insertdict(mp, key, hash, value); |
| else if (_PyDict_GetItem_KnownHash(a, key, hash) == NULL) { |
| if (PyErr_Occurred()) { |
| Py_DECREF(value); |
| Py_DECREF(key); |
| return -1; |
| } |
| err = insertdict(mp, key, hash, value); |
| } |
| else if (override != 0) { |
| _PyErr_SetKeyError(key); |
| Py_DECREF(value); |
| Py_DECREF(key); |
| return -1; |
| } |
| Py_DECREF(value); |
| Py_DECREF(key); |
| if (err != 0) |
| return -1; |
| |
| if (n != other->ma_keys->dk_nentries) { |
| 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 != 1) { |
| if (PyDict_GetItemWithError(a, key) != NULL) { |
| if (override != 0) { |
| _PyErr_SetKeyError(key); |
| Py_DECREF(key); |
| Py_DECREF(iter); |
| return -1; |
| } |
| Py_DECREF(key); |
| continue; |
| } |
| else if (PyErr_Occurred()) { |
| Py_DECREF(key); |
| Py_DECREF(iter); |
| return -1; |
| } |
| } |
| 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; |
| } |
| ASSERT_CONSISTENT(a); |
| return 0; |
| } |
| |
| int |
| PyDict_Update(PyObject *a, PyObject *b) |
| { |
| return dict_merge(a, b, 1); |
| } |
| |
| int |
| PyDict_Merge(PyObject *a, PyObject *b, int override) |
| { |
| /* XXX Deprecate override not in (0, 1). */ |
| return dict_merge(a, b, override != 0); |
| } |
| |
| int |
| _PyDict_MergeEx(PyObject *a, PyObject *b, int override) |
| { |
| return dict_merge(a, b, override); |
| } |
| |
| static PyObject * |
| dict_copy(PyDictObject *mp, PyObject *Py_UNUSED(ignored)) |
| { |
| 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 (mp->ma_used == 0) { |
| /* The dict is empty; just return a new dict. */ |
| return PyDict_New(); |
| } |
| |
| if (_PyDict_HasSplitTable(mp)) { |
| PyDictObject *split_copy; |
| Py_ssize_t size = USABLE_FRACTION(DK_SIZE(mp->ma_keys)); |
| PyObject **newvalues; |
| newvalues = new_values(size); |
| 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; |
| split_copy->ma_version_tag = DICT_NEXT_VERSION(); |
| dictkeys_incref(mp->ma_keys); |
| for (i = 0, n = size; 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; |
| } |
| |
| if (PyDict_CheckExact(mp) && mp->ma_values == NULL && |
| (mp->ma_used >= (mp->ma_keys->dk_nentries * 2) / 3)) |
| { |
| /* Use fast-copy if: |
| |
| (1) 'mp' is an instance of a subclassed dict; and |
| |
| (2) 'mp' is not a split-dict; and |
| |
| (3) if 'mp' is non-compact ('del' operation does not resize dicts), |
| do fast-copy only if it has at most 1/3 non-used keys. |
| |
| The last condition (3) is important to guard against a pathological |
| case when a large dict is almost emptied with multiple del/pop |
| operations and copied after that. In cases like this, we defer to |
| PyDict_Merge, which produces a compacted copy. |
| */ |
| return clone_combined_dict(mp); |
| } |
| |
| 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 < a->ma_keys->dk_nentries; i++) { |
| PyDictKeyEntry *ep = &DK_ENTRIES(a->ma_keys)[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 *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 */ |
| b->ma_keys->dk_lookup(b, key, ep->me_hash, &bval); |
| if (bval == NULL) { |
| Py_DECREF(key); |
| Py_DECREF(aval); |
| if (PyErr_Occurred()) |
| return -1; |
| return 0; |
| } |
| Py_INCREF(bval); |
| cmp = PyObject_RichCompareBool(aval, bval, Py_EQ); |
| Py_DECREF(key); |
| Py_DECREF(aval); |
| Py_DECREF(bval); |
| 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 the dictionary has the specified key, else False. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict___contains__(PyDictObject *self, PyObject *key) |
| /*[clinic end generated code: output=a3d03db709ed6e6b input=fe1cb42ad831e820]*/ |
| { |
| register PyDictObject *mp = self; |
| Py_hash_t hash; |
| Py_ssize_t ix; |
| PyObject *value; |
| |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return NULL; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| if (ix == DKIX_EMPTY || value == NULL) |
| Py_RETURN_FALSE; |
| Py_RETURN_TRUE; |
| } |
| |
| /*[clinic input] |
| dict.get |
| |
| key: object |
| default: object = None |
| / |
| |
| Return the value for key if key is in the dictionary, else default. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict_get_impl(PyDictObject *self, PyObject *key, PyObject *default_value) |
| /*[clinic end generated code: output=bba707729dee05bf input=279ddb5790b6b107]*/ |
| { |
| PyObject *val = NULL; |
| Py_hash_t hash; |
| Py_ssize_t ix; |
| |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return NULL; |
| } |
| ix = (self->ma_keys->dk_lookup) (self, key, hash, &val); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| if (ix == DKIX_EMPTY || val == NULL) { |
| val = default_value; |
| } |
| Py_INCREF(val); |
| return val; |
| } |
| |
| PyObject * |
| PyDict_SetDefault(PyObject *d, PyObject *key, PyObject *defaultobj) |
| { |
| PyDictObject *mp = (PyDictObject *)d; |
| PyObject *value; |
| Py_hash_t hash; |
| |
| 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; |
| } |
| if (mp->ma_keys == Py_EMPTY_KEYS) { |
| if (insert_to_emptydict(mp, key, hash, defaultobj) < 0) { |
| return NULL; |
| } |
| return defaultobj; |
| } |
| |
| if (mp->ma_values != NULL && !PyUnicode_CheckExact(key)) { |
| if (insertion_resize(mp) < 0) |
| return NULL; |
| } |
| |
| Py_ssize_t ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return NULL; |
| |
| if (_PyDict_HasSplitTable(mp) && |
| ((ix >= 0 && value == NULL && mp->ma_used != ix) || |
| (ix == DKIX_EMPTY && mp->ma_used != mp->ma_keys->dk_nentries))) { |
| if (insertion_resize(mp) < 0) { |
| return NULL; |
| } |
| ix = DKIX_EMPTY; |
| } |
| |
| if (ix == DKIX_EMPTY) { |
| PyDictKeyEntry *ep, *ep0; |
| value = defaultobj; |
| if (mp->ma_keys->dk_usable <= 0) { |
| if (insertion_resize(mp) < 0) { |
| return NULL; |
| } |
| } |
| Py_ssize_t hashpos = find_empty_slot(mp->ma_keys, hash); |
| ep0 = DK_ENTRIES(mp->ma_keys); |
| ep = &ep0[mp->ma_keys->dk_nentries]; |
| dictkeys_set_index(mp->ma_keys, hashpos, mp->ma_keys->dk_nentries); |
| Py_INCREF(key); |
| Py_INCREF(value); |
| MAINTAIN_TRACKING(mp, key, value); |
| ep->me_key = key; |
| ep->me_hash = hash; |
| if (_PyDict_HasSplitTable(mp)) { |
| assert(mp->ma_values[mp->ma_keys->dk_nentries] == NULL); |
| mp->ma_values[mp->ma_keys->dk_nentries] = value; |
| } |
| else { |
| ep->me_value = value; |
| } |
| mp->ma_used++; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| mp->ma_keys->dk_usable--; |
| mp->ma_keys->dk_nentries++; |
| assert(mp->ma_keys->dk_usable >= 0); |
| } |
| else if (value == NULL) { |
| value = defaultobj; |
| assert(_PyDict_HasSplitTable(mp)); |
| assert(ix == mp->ma_used); |
| Py_INCREF(value); |
| MAINTAIN_TRACKING(mp, key, value); |
| mp->ma_values[ix] = value; |
| mp->ma_used++; |
| mp->ma_version_tag = DICT_NEXT_VERSION(); |
| } |
| |
| ASSERT_CONSISTENT(mp); |
| return value; |
| } |
| |
| /*[clinic input] |
| dict.setdefault |
| |
| key: object |
| default: object = None |
| / |
| |
| Insert key with a value of default if key is not in the dictionary. |
| |
| Return the value for key if key is in the dictionary, else default. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict_setdefault_impl(PyDictObject *self, PyObject *key, |
| PyObject *default_value) |
| /*[clinic end generated code: output=f8c1101ebf69e220 input=0f063756e815fd9d]*/ |
| { |
| PyObject *val; |
| |
| val = PyDict_SetDefault((PyObject *)self, key, default_value); |
| Py_XINCREF(val); |
| return val; |
| } |
| |
| static PyObject * |
| dict_clear(PyDictObject *mp, PyObject *Py_UNUSED(ignored)) |
| { |
| PyDict_Clear((PyObject *)mp); |
| Py_RETURN_NONE; |
| } |
| |
| /*[clinic input] |
| dict.pop |
| |
| key: object |
| default: object = NULL |
| / |
| |
| D.pop(k[,d]) -> v, remove specified key and return the corresponding value. |
| |
| If the key is not found, return the default if given; otherwise, |
| raise a KeyError. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict_pop_impl(PyDictObject *self, PyObject *key, PyObject *default_value) |
| /*[clinic end generated code: output=3abb47b89f24c21c input=e221baa01044c44c]*/ |
| { |
| return _PyDict_Pop((PyObject*)self, key, default_value); |
| } |
| |
| /*[clinic input] |
| dict.popitem |
| |
| Remove and return a (key, value) pair as a 2-tuple. |
| |
| Pairs are returned in LIFO (last-in, first-out) order. |
| Raises KeyError if the dict is empty. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict_popitem_impl(PyDictObject *self) |
| /*[clinic end generated code: output=e65fcb04420d230d input=1c38a49f21f64941]*/ |
| { |
| Py_ssize_t i, j; |
| PyDictKeyEntry *ep0, *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 (self->ma_used == 0) { |
| Py_DECREF(res); |
| PyErr_SetString(PyExc_KeyError, "popitem(): dictionary is empty"); |
| return NULL; |
| } |
| /* Convert split table to combined table */ |
| if (self->ma_keys->dk_lookup == lookdict_split) { |
| if (dictresize(self, DK_SIZE(self->ma_keys))) { |
| Py_DECREF(res); |
| return NULL; |
| } |
| } |
| ENSURE_ALLOWS_DELETIONS(self); |
| |
| /* Pop last item */ |
| ep0 = DK_ENTRIES(self->ma_keys); |
| i = self->ma_keys->dk_nentries - 1; |
| while (i >= 0 && ep0[i].me_value == NULL) { |
| i--; |
| } |
| assert(i >= 0); |
| |
| ep = &ep0[i]; |
| j = lookdict_index(self->ma_keys, ep->me_hash, i); |
| assert(j >= 0); |
| assert(dictkeys_get_index(self->ma_keys, j) == i); |
| dictkeys_set_index(self->ma_keys, j, DKIX_DUMMY); |
| |
| PyTuple_SET_ITEM(res, 0, ep->me_key); |
| PyTuple_SET_ITEM(res, 1, ep->me_value); |
| ep->me_key = NULL; |
| ep->me_value = NULL; |
| /* We can't dk_usable++ since there is DKIX_DUMMY in indices */ |
| self->ma_keys->dk_nentries = i; |
| self->ma_used--; |
| self->ma_version_tag = DICT_NEXT_VERSION(); |
| ASSERT_CONSISTENT(self); |
| return res; |
| } |
| |
| static int |
| dict_traverse(PyObject *op, visitproc visit, void *arg) |
| { |
| PyDictObject *mp = (PyDictObject *)op; |
| PyDictKeysObject *keys = mp->ma_keys; |
| PyDictKeyEntry *entries = DK_ENTRIES(keys); |
| Py_ssize_t i, n = keys->dk_nentries; |
| |
| if (keys->dk_lookup == lookdict) { |
| for (i = 0; i < n; i++) { |
| if (entries[i].me_value != NULL) { |
| Py_VISIT(entries[i].me_value); |
| Py_VISIT(entries[i].me_key); |
| } |
| } |
| } |
| else { |
| if (mp->ma_values != NULL) { |
| for (i = 0; i < n; i++) { |
| Py_VISIT(mp->ma_values[i]); |
| } |
| } |
| else { |
| for (i = 0; i < n; i++) { |
| Py_VISIT(entries[i].me_value); |
| } |
| } |
| } |
| return 0; |
| } |
| |
| static int |
| dict_tp_clear(PyObject *op) |
| { |
| PyDict_Clear(op); |
| return 0; |
| } |
| |
| static PyObject *dictiter_new(PyDictObject *, PyTypeObject *); |
| |
| Py_ssize_t |
| _PyDict_SizeOf(PyDictObject *mp) |
| { |
| Py_ssize_t size, usable, res; |
| |
| size = DK_SIZE(mp->ma_keys); |
| usable = USABLE_FRACTION(size); |
| |
| res = _PyObject_SIZE(Py_TYPE(mp)); |
| if (mp->ma_values) |
| res += usable * 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) |
| + DK_IXSIZE(mp->ma_keys) * size |
| + sizeof(PyDictKeyEntry) * usable); |
| return res; |
| } |
| |
| Py_ssize_t |
| _PyDict_KeysSize(PyDictKeysObject *keys) |
| { |
| return (sizeof(PyDictKeysObject) |
| + DK_IXSIZE(keys) * DK_SIZE(keys) |
| + USABLE_FRACTION(DK_SIZE(keys)) * sizeof(PyDictKeyEntry)); |
| } |
| |
| static PyObject * |
| dict_sizeof(PyDictObject *mp, PyObject *Py_UNUSED(ignored)) |
| { |
| return PyLong_FromSsize_t(_PyDict_SizeOf(mp)); |
| } |
| |
| static PyObject * |
| dict_or(PyObject *self, PyObject *other) |
| { |
| if (!PyDict_Check(self) || !PyDict_Check(other)) { |
| Py_RETURN_NOTIMPLEMENTED; |
| } |
| PyObject *new = PyDict_Copy(self); |
| if (new == NULL) { |
| return NULL; |
| } |
| if (dict_update_arg(new, other)) { |
| Py_DECREF(new); |
| return NULL; |
| } |
| return new; |
| } |
| |
| static PyObject * |
| dict_ior(PyObject *self, PyObject *other) |
| { |
| if (dict_update_arg(self, other)) { |
| return NULL; |
| } |
| Py_INCREF(self); |
| return self; |
| } |
| |
| PyDoc_STRVAR(getitem__doc__, "x.__getitem__(y) <==> x[y]"); |
| |
| PyDoc_STRVAR(sizeof__doc__, |
| "D.__sizeof__() -> size of D in memory, in bytes"); |
| |
| 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 *, PyObject *); |
| static PyObject *dictitems_new(PyObject *, PyObject *); |
| static PyObject *dictvalues_new(PyObject *, 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)(void(*)(void))dict_subscript, METH_O | METH_COEXIST, |
| getitem__doc__}, |
| {"__sizeof__", (PyCFunction)(void(*)(void))dict_sizeof, METH_NOARGS, |
| sizeof__doc__}, |
| DICT_GET_METHODDEF |
| DICT_SETDEFAULT_METHODDEF |
| DICT_POP_METHODDEF |
| DICT_POPITEM_METHODDEF |
| {"keys", dictkeys_new, METH_NOARGS, |
| keys__doc__}, |
| {"items", dictitems_new, METH_NOARGS, |
| items__doc__}, |
| {"values", dictvalues_new, METH_NOARGS, |
| values__doc__}, |
| {"update", (PyCFunction)(void(*)(void))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__}, |
| DICT___REVERSED___METHODDEF |
| {"__class_getitem__", (PyCFunction)Py_GenericAlias, METH_O|METH_CLASS, PyDoc_STR("See PEP 585")}, |
| {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; |
| Py_ssize_t ix; |
| PyDictObject *mp = (PyDictObject *)op; |
| PyObject *value; |
| |
| if (!PyUnicode_CheckExact(key) || |
| (hash = ((PyASCIIObject *) key)->hash) == -1) { |
| hash = PyObject_Hash(key); |
| if (hash == -1) |
| return -1; |
| } |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return -1; |
| return (ix != DKIX_EMPTY && value != 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; |
| PyObject *value; |
| Py_ssize_t ix; |
| |
| ix = (mp->ma_keys->dk_lookup)(mp, key, hash, &value); |
| if (ix == DKIX_ERROR) |
| return -1; |
| return (ix != DKIX_EMPTY && value != 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 PyNumberMethods dict_as_number = { |
| .nb_or = dict_or, |
| .nb_inplace_or = dict_ior, |
| }; |
| |
| 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_version_tag = DICT_NEXT_VERSION(); |
| d->ma_keys = new_keys_object(PyDict_MINSIZE); |
| if (d->ma_keys == NULL) { |
| Py_DECREF(self); |
| return NULL; |
| } |
| ASSERT_CONSISTENT(d); |
| return self; |
| } |
| |
| static int |
| dict_init(PyObject *self, PyObject *args, PyObject *kwds) |
| { |
| return dict_update_common(self, args, kwds, "dict"); |
| } |
| |
| static PyObject * |
| dict_vectorcall(PyObject *type, PyObject * const*args, |
| size_t nargsf, PyObject *kwnames) |
| { |
| assert(PyType_Check(type)); |
| Py_ssize_t nargs = PyVectorcall_NARGS(nargsf); |
| if (!_PyArg_CheckPositional("dict", nargs, 0, 1)) { |
| return NULL; |
| } |
| |
| PyObject *self = dict_new((PyTypeObject *)type, NULL, NULL); |
| if (self == NULL) { |
| return NULL; |
| } |
| if (nargs == 1) { |
| if (dict_update_arg(self, args[0]) < 0) { |
| Py_DECREF(self); |
| return NULL; |
| } |
| args++; |
| } |
| if (kwnames != NULL) { |
| for (Py_ssize_t i = 0; i < PyTuple_GET_SIZE(kwnames); i++) { |
| if (PyDict_SetItem(self, PyTuple_GET_ITEM(kwnames, i), args[i]) < 0) { |
| Py_DECREF(self); |
| return NULL; |
| } |
| } |
| } |
| return self; |
| } |
| |
| 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_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| (reprfunc)dict_repr, /* tp_repr */ |
| &dict_as_number, /* 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 */ |
| .tp_vectorcall = dict_vectorcall, |
| }; |
| |
| 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->len = dict->ma_used; |
| if (itertype == &PyDictRevIterKey_Type || |
| itertype == &PyDictRevIterItem_Type || |
| itertype == &PyDictRevIterValue_Type) { |
| if (dict->ma_values) { |
| di->di_pos = dict->ma_used - 1; |
| } |
| else { |
| di->di_pos = dict->ma_keys->dk_nentries - 1; |
| } |
| } |
| else { |
| di->di_pos = 0; |
| } |
| if (itertype == &PyDictIterItem_Type || |
| itertype == &PyDictRevIterItem_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) |
| { |
| /* bpo-31095: UnTrack is needed before calling any callbacks */ |
| _PyObject_GC_UNTRACK(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, PyObject *Py_UNUSED(ignored)) |
| { |
| 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, PyObject *Py_UNUSED(ignored)); |
| |
| PyDoc_STRVAR(reduce_doc, "Return state information for pickling."); |
| |
| static PyMethodDef dictiter_methods[] = { |
| {"__length_hint__", (PyCFunction)(void(*)(void))dictiter_len, METH_NOARGS, |
| length_hint_doc}, |
| {"__reduce__", (PyCFunction)(void(*)(void))dictiter_reduce, METH_NOARGS, |
| reduce_doc}, |
| {NULL, NULL} /* sentinel */ |
| }; |
| |
| static PyObject* |
| dictiter_iternextkey(dictiterobject *di) |
| { |
| PyObject *key; |
| Py_ssize_t i; |
| PyDictKeysObject *k; |
| PyDictObject *d = di->di_dict; |
| |
| 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; |
| k = d->ma_keys; |
| assert(i >= 0); |
| if (d->ma_values) { |
| if (i >= d->ma_used) |
| goto fail; |
| key = DK_ENTRIES(k)[i].me_key; |
| assert(d->ma_values[i] != NULL); |
| } |
| else { |
| Py_ssize_t n = k->dk_nentries; |
| PyDictKeyEntry *entry_ptr = &DK_ENTRIES(k)[i]; |
| while (i < n && entry_ptr->me_value == NULL) { |
| entry_ptr++; |
| i++; |
| } |
| if (i >= n) |
| goto fail; |
| key = entry_ptr->me_key; |
| } |
| // We found an element (key), but did not expect it |
| if (di->len == 0) { |
| PyErr_SetString(PyExc_RuntimeError, |
| "dictionary keys changed during iteration"); |
| goto fail; |
| } |
| di->di_pos = i+1; |
| di->len--; |
| Py_INCREF(key); |
| return key; |
| |
| fail: |
| di->di_dict = NULL; |
| Py_DECREF(d); |
| 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_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| 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; |
| PyDictObject *d = di->di_dict; |
| |
| 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; |
| assert(i >= 0); |
| if (d->ma_values) { |
| if (i >= d->ma_used) |
| goto fail; |
| value = d->ma_values[i]; |
| assert(value != NULL); |
| } |
| else { |
| Py_ssize_t n = d->ma_keys->dk_nentries; |
| PyDictKeyEntry *entry_ptr = &DK_ENTRIES(d->ma_keys)[i]; |
| while (i < n && entry_ptr->me_value == NULL) { |
| entry_ptr++; |
| i++; |
| } |
| if (i >= n) |
| goto fail; |
| value = entry_ptr->me_value; |
| } |
| // We found an element, but did not expect it |
| if (di->len == 0) { |
| PyErr_SetString(PyExc_RuntimeError, |
| "dictionary keys changed during iteration"); |
| goto fail; |
| } |
| di->di_pos = i+1; |
| di->len--; |
| Py_INCREF(value); |
| return value; |
| |
| fail: |
| di->di_dict = NULL; |
| Py_DECREF(d); |
| 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_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| 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; |
| Py_ssize_t i; |
| PyDictObject *d = di->di_dict; |
| |
| 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; |
| assert(i >= 0); |
| if (d->ma_values) { |
| if (i >= d->ma_used) |
| goto fail; |
| key = DK_ENTRIES(d->ma_keys)[i].me_key; |
| value = d->ma_values[i]; |
| assert(value != NULL); |
| } |
| else { |
| Py_ssize_t n = d->ma_keys->dk_nentries; |
| PyDictKeyEntry *entry_ptr = &DK_ENTRIES(d->ma_keys)[i]; |
| while (i < n && entry_ptr->me_value == NULL) { |
| entry_ptr++; |
| i++; |
| } |
| if (i >= n) |
| goto fail; |
| key = entry_ptr->me_key; |
| value = entry_ptr->me_value; |
| } |
| // We found an element, but did not expect it |
| if (di->len == 0) { |
| PyErr_SetString(PyExc_RuntimeError, |
| "dictionary keys changed during iteration"); |
| goto fail; |
| } |
| di->di_pos = i+1; |
| di->len--; |
| Py_INCREF(key); |
| Py_INCREF(value); |
| result = di->di_result; |
| if (Py_REFCNT(result) == 1) { |
| PyObject *oldkey = PyTuple_GET_ITEM(result, 0); |
| PyObject *oldvalue = PyTuple_GET_ITEM(result, 1); |
| PyTuple_SET_ITEM(result, 0, key); /* steals reference */ |
| PyTuple_SET_ITEM(result, 1, value); /* steals reference */ |
| Py_INCREF(result); |
| Py_DECREF(oldkey); |
| Py_DECREF(oldvalue); |
| } |
| else { |
| result = PyTuple_New(2); |
| if (result == NULL) |
| return NULL; |
| PyTuple_SET_ITEM(result, 0, key); /* steals reference */ |
| PyTuple_SET_ITEM(result, 1, value); /* steals reference */ |
| } |
| return result; |
| |
| fail: |
| di->di_dict = NULL; |
| Py_DECREF(d); |
| 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_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| 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, |
| }; |
| |
| |
| /* dictreviter */ |
| |
| static PyObject * |
| dictreviter_iternext(dictiterobject *di) |
| { |
| PyDictObject *d = di->di_dict; |
| |
| 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; |
| } |
| |
| Py_ssize_t i = di->di_pos; |
| PyDictKeysObject *k = d->ma_keys; |
| PyObject *key, *value, *result; |
| |
| if (i < 0) { |
| goto fail; |
| } |
| if (d->ma_values) { |
| key = DK_ENTRIES(k)[i].me_key; |
| value = d->ma_values[i]; |
| assert (value != NULL); |
| } |
| else { |
| PyDictKeyEntry *entry_ptr = &DK_ENTRIES(k)[i]; |
| while (entry_ptr->me_value == NULL) { |
| if (--i < 0) { |
| goto fail; |
| } |
| entry_ptr--; |
| } |
| key = entry_ptr->me_key; |
| value = entry_ptr->me_value; |
| } |
| di->di_pos = i-1; |
| di->len--; |
| |
| if (Py_IS_TYPE(di, &PyDictRevIterKey_Type)) { |
| Py_INCREF(key); |
| return key; |
| } |
| else if (Py_IS_TYPE(di, &PyDictRevIterValue_Type)) { |
| Py_INCREF(value); |
| return value; |
| } |
| else if (Py_IS_TYPE(di, &PyDictRevIterItem_Type)) { |
| Py_INCREF(key); |
| Py_INCREF(value); |
| result = di->di_result; |
| if (Py_REFCNT(result) == 1) { |
| PyObject *oldkey = PyTuple_GET_ITEM(result, 0); |
| PyObject *oldvalue = PyTuple_GET_ITEM(result, 1); |
| PyTuple_SET_ITEM(result, 0, key); /* steals reference */ |
| PyTuple_SET_ITEM(result, 1, value); /* steals reference */ |
| Py_INCREF(result); |
| Py_DECREF(oldkey); |
| Py_DECREF(oldvalue); |
| } |
| else { |
| result = PyTuple_New(2); |
| if (result == NULL) { |
| return NULL; |
| } |
| PyTuple_SET_ITEM(result, 0, key); /* steals reference */ |
| PyTuple_SET_ITEM(result, 1, value); /* steals reference */ |
| } |
| return result; |
| } |
| else { |
| Py_UNREACHABLE(); |
| } |
| |
| fail: |
| di->di_dict = NULL; |
| Py_DECREF(d); |
| return NULL; |
| } |
| |
| PyTypeObject PyDictRevIterKey_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_reversekeyiterator", |
| sizeof(dictiterobject), |
| .tp_dealloc = (destructor)dictiter_dealloc, |
| .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, |
| .tp_traverse = (traverseproc)dictiter_traverse, |
| .tp_iter = PyObject_SelfIter, |
| .tp_iternext = (iternextfunc)dictreviter_iternext, |
| .tp_methods = dictiter_methods |
| }; |
| |
| |
| /*[clinic input] |
| dict.__reversed__ |
| |
| Return a reverse iterator over the dict keys. |
| [clinic start generated code]*/ |
| |
| static PyObject * |
| dict___reversed___impl(PyDictObject *self) |
| /*[clinic end generated code: output=e674483336d1ed51 input=23210ef3477d8c4d]*/ |
| { |
| assert (PyDict_Check(self)); |
| return dictiter_new(self, &PyDictRevIterKey_Type); |
| } |
| |
| static PyObject * |
| dictiter_reduce(dictiterobject *di, PyObject *Py_UNUSED(ignored)) |
| { |
| _Py_IDENTIFIER(iter); |
| /* copy the iterator state */ |
| dictiterobject tmp = *di; |
| Py_XINCREF(tmp.di_dict); |
| |
| PyObject *list = PySequence_List((PyObject*)&tmp); |
| Py_XDECREF(tmp.di_dict); |
| if (list == NULL) { |
| return NULL; |
| } |
| return Py_BuildValue("N(N)", _PyEval_GetBuiltinId(&PyId_iter), list); |
| } |
| |
| PyTypeObject PyDictRevIterItem_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_reverseitemiterator", |
| sizeof(dictiterobject), |
| .tp_dealloc = (destructor)dictiter_dealloc, |
| .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, |
| .tp_traverse = (traverseproc)dictiter_traverse, |
| .tp_iter = PyObject_SelfIter, |
| .tp_iternext = (iternextfunc)dictreviter_iternext, |
| .tp_methods = dictiter_methods |
| }; |
| |
| PyTypeObject PyDictRevIterValue_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_reversevalueiterator", |
| sizeof(dictiterobject), |
| .tp_dealloc = (destructor)dictiter_dealloc, |
| .tp_flags = Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, |
| .tp_traverse = (traverseproc)dictiter_traverse, |
| .tp_iter = PyObject_SelfIter, |
| .tp_iternext = (iternextfunc)dictreviter_iternext, |
| .tp_methods = dictiter_methods |
| }; |
| |
| /***********************************************/ |
| /* View objects for keys(), items(), values(). */ |
| /***********************************************/ |
| |
| /* The instance lay-out is the same for all three; but the type differs. */ |
| |
| static void |
| dictview_dealloc(_PyDictViewObject *dv) |
| { |
| /* bpo-31095: UnTrack is needed before calling any callbacks */ |
| _PyObject_GC_UNTRACK(dv); |
| Py_XDECREF(dv->dv_dict); |
| PyObject_GC_Del(dv); |
| } |
| |
| static int |
| dictview_traverse(_PyDictViewObject *dv, visitproc visit, void *arg) |
| { |
| Py_VISIT(dv->dv_dict); |
| return 0; |
| } |
| |
| static Py_ssize_t |
| dictview_len(_PyDictViewObject *dv) |
| { |
| Py_ssize_t len = 0; |
| if (dv->dv_dict != NULL) |
| len = dv->dv_dict->ma_used; |
| return len; |
| } |
| |
| PyObject * |
| _PyDictView_New(PyObject *dict, PyTypeObject *type) |
| { |
| _PyDictViewObject *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, Py_TYPE(dict)->tp_name); |
| return NULL; |
| } |
| dv = PyObject_GC_New(_PyDictViewObject, type); |
| if (dv == NULL) |
| return NULL; |
| Py_INCREF(dict); |
| dv->dv_dict = (PyDictObject *)dict; |
| _PyObject_GC_TRACK(dv); |
| return (PyObject *)dv; |
| } |
| |
| static PyObject * |
| dictview_mapping(PyObject *view, void *Py_UNUSED(ignored)) { |
| assert(view != NULL); |
| assert(PyDictKeys_Check(view) |
| || PyDictValues_Check(view) |
| || PyDictItems_Check(view)); |
| PyObject *mapping = (PyObject *)((_PyDictViewObject *)view)->dv_dict; |
| return PyDictProxy_New(mapping); |
| } |
| |
| static PyGetSetDef dictview_getset[] = { |
| {"mapping", dictview_mapping, (setter)NULL, |
| "dictionary that this view refers to", NULL}, |
| {0} |
| }; |
| |
| /* 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(_PyDictViewObject *dv) |
| { |
| PyObject *seq; |
| PyObject *result = NULL; |
| Py_ssize_t rc; |
| |
| rc = Py_ReprEnter((PyObject *)dv); |
| if (rc != 0) { |
| return rc > 0 ? PyUnicode_FromString("...") : NULL; |
| } |
| seq = PySequence_List((PyObject *)dv); |
| if (seq == NULL) { |
| goto Done; |
| } |
| result = PyUnicode_FromFormat("%s(%R)", Py_TYPE(dv)->tp_name, seq); |
| Py_DECREF(seq); |
| |
| Done: |
| Py_ReprLeave((PyObject *)dv); |
| return result; |
| } |
| |
| /*** dict_keys ***/ |
| |
| static PyObject * |
| dictkeys_iter(_PyDictViewObject *dv) |
| { |
| if (dv->dv_dict == NULL) { |
| Py_RETURN_NONE; |
| } |
| return dictiter_new(dv->dv_dict, &PyDictIterKey_Type); |
| } |
| |
| static int |
| dictkeys_contains(_PyDictViewObject *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 */ |
| }; |
| |
| // Create an set object from dictviews object. |
| // Returns a new reference. |
| // This utility function is used by set operations. |
| static PyObject* |
| dictviews_to_set(PyObject *self) |
| { |
| PyObject *left = self; |
| if (PyDictKeys_Check(self)) { |
| // PySet_New() has fast path for the dict object. |
| PyObject *dict = (PyObject *)((_PyDictViewObject *)self)->dv_dict; |
| if (PyDict_CheckExact(dict)) { |
| left = dict; |
| } |
| } |
| return PySet_New(left); |
| } |
| |
| static PyObject* |
| dictviews_sub(PyObject *self, PyObject *other) |
| { |
| PyObject *result = dictviews_to_set(self); |
| if (result == NULL) { |
| return NULL; |
| } |
| |
| _Py_IDENTIFIER(difference_update); |
| PyObject *tmp = _PyObject_CallMethodIdOneArg( |
| result, &PyId_difference_update, other); |
| if (tmp == NULL) { |
| Py_DECREF(result); |
| return NULL; |
| } |
| |
| Py_DECREF(tmp); |
| return result; |
| } |
| |
| static int |
| dictitems_contains(_PyDictViewObject *dv, PyObject *obj); |
| |
| PyObject * |
| _PyDictView_Intersect(PyObject* self, PyObject *other) |
| { |
| PyObject *result; |
| PyObject *it; |
| PyObject *key; |
| Py_ssize_t len_self; |
| int rv; |
| int (*dict_contains)(_PyDictViewObject *, PyObject *); |
| |
| /* Python interpreter swaps parameters when dict view |
| is on right side of & */ |
| if (!PyDictViewSet_Check(self)) { |
| PyObject *tmp = other; |
| other = self; |
| self = tmp; |
| } |
| |
| len_self = dictview_len((_PyDictViewObject *)self); |
| |
| /* if other is a set and self is smaller than other, |
| reuse set intersection logic */ |
| if (Py_IS_TYPE(other, &PySet_Type) && len_self <= PyObject_Size(other)) { |
| _Py_IDENTIFIER(intersection); |
| return _PyObject_CallMethodIdObjArgs(other, &PyId_intersection, self, NULL); |
| } |
| |
| /* if other is another dict view, and it is bigger than self, |
| swap them */ |
| if (PyDictViewSet_Check(other)) { |
| Py_ssize_t len_other = dictview_len((_PyDictViewObject *)other); |
| if (len_other > len_self) { |
| PyObject *tmp = other; |
| other = self; |
| self = tmp; |
| } |
| } |
| |
| /* at this point, two things should be true |
| 1. self is a dictview |
| 2. if other is a dictview then it is smaller than self */ |
| result = PySet_New(NULL); |
| if (result == NULL) |
| return NULL; |
| |
| it = PyObject_GetIter(other); |
| if (it == NULL) { |
| Py_DECREF(result); |
| return NULL; |
| } |
| |
| if (PyDictKeys_Check(self)) { |
| dict_contains = dictkeys_contains; |
| } |
| /* else PyDictItems_Check(self) */ |
| else { |
| dict_contains = dictitems_contains; |
| } |
| |
| while ((key = PyIter_Next(it)) != NULL) { |
| rv = dict_contains((_PyDictViewObject *)self, key); |
| if (rv < 0) { |
| goto error; |
| } |
| if (rv) { |
| if (PySet_Add(result, key)) { |
| goto error; |
| } |
| } |
| Py_DECREF(key); |
| } |
| Py_DECREF(it); |
| if (PyErr_Occurred()) { |
| Py_DECREF(result); |
| return NULL; |
| } |
| return result; |
| |
| error: |
| Py_DECREF(it); |
| Py_DECREF(result); |
| Py_DECREF(key); |
| return NULL; |
| } |
| |
| static PyObject* |
| dictviews_or(PyObject* self, PyObject *other) |
| { |
| PyObject *result = dictviews_to_set(self); |
| if (result == NULL) { |
| return NULL; |
| } |
| |
| if (_PySet_Update(result, other) < 0) { |
| Py_DECREF(result); |
| return NULL; |
| } |
| return result; |
| } |
| |
| static PyObject * |
| dictitems_xor(PyObject *self, PyObject *other) |
| { |
| assert(PyDictItems_Check(self)); |
| assert(PyDictItems_Check(other)); |
| PyObject *d1 = (PyObject *)((_PyDictViewObject *)self)->dv_dict; |
| PyObject *d2 = (PyObject *)((_PyDictViewObject *)other)->dv_dict; |
| |
| PyObject *temp_dict = PyDict_Copy(d1); |
| if (temp_dict == NULL) { |
| return NULL; |
| } |
| PyObject *result_set = PySet_New(NULL); |
| if (result_set == NULL) { |
| Py_CLEAR(temp_dict); |
| return NULL; |
| } |
| |
| PyObject *key = NULL, *val1 = NULL, *val2 = NULL; |
| Py_ssize_t pos = 0; |
| Py_hash_t hash; |
| |
| while (_PyDict_Next(d2, &pos, &key, &val2, &hash)) { |
| Py_INCREF(key); |
| Py_INCREF(val2); |
| val1 = _PyDict_GetItem_KnownHash(temp_dict, key, hash); |
| |
| int to_delete; |
| if (val1 == NULL) { |
| if (PyErr_Occurred()) { |
| goto error; |
| } |
| to_delete = 0; |
| } |
| else { |
| Py_INCREF(val1); |
| to_delete = PyObject_RichCompareBool(val1, val2, Py_EQ); |
| if (to_delete < 0) { |
| goto error; |
| } |
| } |
| |
| if (to_delete) { |
| if (_PyDict_DelItem_KnownHash(temp_dict, key, hash) < 0) { |
| goto error; |
| } |
| } |
| else { |
| PyObject *pair = PyTuple_Pack(2, key, val2); |
| if (pair == NULL) { |
| goto error; |
| } |
| if (PySet_Add(result_set, pair) < 0) { |
| Py_DECREF(pair); |
| goto error; |
| } |
| Py_DECREF(pair); |
| } |
| Py_DECREF(key); |
| Py_XDECREF(val1); |
| Py_DECREF(val2); |
| } |
| key = val1 = val2 = NULL; |
| |
| _Py_IDENTIFIER(items); |
| PyObject *remaining_pairs = _PyObject_CallMethodIdNoArgs(temp_dict, |
| &PyId_items); |
| if (remaining_pairs == NULL) { |
| goto error; |
| } |
| if (_PySet_Update(result_set, remaining_pairs) < 0) { |
| Py_DECREF(remaining_pairs); |
| goto error; |
| } |
| Py_DECREF(temp_dict); |
| Py_DECREF(remaining_pairs); |
| return result_set; |
| |
| error: |
| Py_XDECREF(temp_dict); |
| Py_XDECREF(result_set); |
| Py_XDECREF(key); |
| Py_XDECREF(val1); |
| Py_XDECREF(val2); |
| return NULL; |
| } |
| |
| static PyObject* |
| dictviews_xor(PyObject* self, PyObject *other) |
| { |
| if (PyDictItems_Check(self) && PyDictItems_Check(other)) { |
| return dictitems_xor(self, other); |
| } |
| PyObject *result = dictviews_to_set(self); |
| if (result == NULL) { |
| return NULL; |
| } |
| |
| _Py_IDENTIFIER(symmetric_difference_update); |
| PyObject *tmp = _PyObject_CallMethodIdOneArg( |
| result, &PyId_symmetric_difference_update, 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)_PyDictView_Intersect, /*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((_PyDictViewObject *)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((_PyDictViewObject *)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 PyObject* dictkeys_reversed(_PyDictViewObject *dv, PyObject *Py_UNUSED(ignored)); |
| |
| PyDoc_STRVAR(reversed_keys_doc, |
| "Return a reverse iterator over the dict keys."); |
| |
| static PyMethodDef dictkeys_methods[] = { |
| {"isdisjoint", (PyCFunction)dictviews_isdisjoint, METH_O, |
| isdisjoint_doc}, |
| {"__reversed__", (PyCFunction)(void(*)(void))dictkeys_reversed, METH_NOARGS, |
| reversed_keys_doc}, |
| {NULL, NULL} /* sentinel */ |
| }; |
| |
| PyTypeObject PyDictKeys_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_keys", /* tp_name */ |
| sizeof(_PyDictViewObject), /* tp_basicsize */ |
| 0, /* tp_itemsize */ |
| /* methods */ |
| (destructor)dictview_dealloc, /* tp_dealloc */ |
| 0, /* tp_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| (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 */ |
| .tp_getset = dictview_getset, |
| }; |
| |
| static PyObject * |
| dictkeys_new(PyObject *dict, PyObject *Py_UNUSED(ignored)) |
| { |
| return _PyDictView_New(dict, &PyDictKeys_Type); |
| } |
| |
| static PyObject * |
| dictkeys_reversed(_PyDictViewObject *dv, PyObject *Py_UNUSED(ignored)) |
| { |
| if (dv->dv_dict == NULL) { |
| Py_RETURN_NONE; |
| } |
| return dictiter_new(dv->dv_dict, &PyDictRevIterKey_Type); |
| } |
| |
| /*** dict_items ***/ |
| |
| static PyObject * |
| dictitems_iter(_PyDictViewObject *dv) |
| { |
| if (dv->dv_dict == NULL) { |
| Py_RETURN_NONE; |
| } |
| return dictiter_new(dv->dv_dict, &PyDictIterItem_Type); |
| } |
| |
| static int |
| dictitems_contains(_PyDictViewObject *dv, PyObject *obj) |
| { |
| int result; |
| 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_GetItemWithError((PyObject *)dv->dv_dict, key); |
| if (found == NULL) { |
| if (PyErr_Occurred()) |
| return -1; |
| return 0; |
| } |
| Py_INCREF(found); |
| result = PyObject_RichCompareBool(found, value, Py_EQ); |
| Py_DECREF(found); |
| return result; |
| } |
| |
| 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 PyObject* dictitems_reversed(_PyDictViewObject *dv); |
| |
| PyDoc_STRVAR(reversed_items_doc, |
| "Return a reverse iterator over the dict items."); |
| |
| static PyMethodDef dictitems_methods[] = { |
| {"isdisjoint", (PyCFunction)dictviews_isdisjoint, METH_O, |
| isdisjoint_doc}, |
| {"__reversed__", (PyCFunction)(void(*)(void))dictitems_reversed, METH_NOARGS, |
| reversed_items_doc}, |
| {NULL, NULL} /* sentinel */ |
| }; |
| |
| PyTypeObject PyDictItems_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_items", /* tp_name */ |
| sizeof(_PyDictViewObject), /* tp_basicsize */ |
| 0, /* tp_itemsize */ |
| /* methods */ |
| (destructor)dictview_dealloc, /* tp_dealloc */ |
| 0, /* tp_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| (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 */ |
| .tp_getset = dictview_getset, |
| }; |
| |
| static PyObject * |
| dictitems_new(PyObject *dict, PyObject *Py_UNUSED(ignored)) |
| { |
| return _PyDictView_New(dict, &PyDictItems_Type); |
| } |
| |
| static PyObject * |
| dictitems_reversed(_PyDictViewObject *dv) |
| { |
| if (dv->dv_dict == NULL) { |
| Py_RETURN_NONE; |
| } |
| return dictiter_new(dv->dv_dict, &PyDictRevIterItem_Type); |
| } |
| |
| /*** dict_values ***/ |
| |
| static PyObject * |
| dictvalues_iter(_PyDictViewObject *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 PyObject* dictvalues_reversed(_PyDictViewObject *dv); |
| |
| PyDoc_STRVAR(reversed_values_doc, |
| "Return a reverse iterator over the dict values."); |
| |
| static PyMethodDef dictvalues_methods[] = { |
| {"__reversed__", (PyCFunction)(void(*)(void))dictvalues_reversed, METH_NOARGS, |
| reversed_values_doc}, |
| {NULL, NULL} /* sentinel */ |
| }; |
| |
| PyTypeObject PyDictValues_Type = { |
| PyVarObject_HEAD_INIT(&PyType_Type, 0) |
| "dict_values", /* tp_name */ |
| sizeof(_PyDictViewObject), /* tp_basicsize */ |
| 0, /* tp_itemsize */ |
| /* methods */ |
| (destructor)dictview_dealloc, /* tp_dealloc */ |
| 0, /* tp_vectorcall_offset */ |
| 0, /* tp_getattr */ |
| 0, /* tp_setattr */ |
| 0, /* tp_as_async */ |
| (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 */ |
| .tp_getset = dictview_getset, |
| }; |
| |
| static PyObject * |
| dictvalues_new(PyObject *dict, PyObject *Py_UNUSED(ignored)) |
| { |
| return _PyDictView_New(dict, &PyDictValues_Type); |
| } |
| |
| static PyObject * |
| dictvalues_reversed(_PyDictViewObject *dv) |
| { |
| if (dv->dv_dict == NULL) { |
| Py_RETURN_NONE; |
| } |
| return dictiter_new(dv->dv_dict, &PyDictRevIterValue_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); |
| 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)) { |
| dictkeys_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) { |
| dictkeys_incref(cached); |
| dict = new_dict_with_shared_keys(cached); |
| if (dict == NULL) |
| return -1; |
| *dictptr = dict; |
| } |
| if (value == NULL) { |
| res = PyDict_DelItem(dict, key); |
| // Since key sharing dict doesn't allow deletion, PyDict_DelItem() |
| // always converts dict to combined form. |
| if ((cached = CACHED_KEYS(tp)) != NULL) { |
| CACHED_KEYS(tp) = NULL; |
| dictkeys_decref(cached); |
| } |
| } |
| else { |
| int was_shared = (cached == ((PyDictObject *)dict)->ma_keys); |
| res = PyDict_SetItem(dict, key, value); |
| if (was_shared && |
| (cached = CACHED_KEYS(tp)) != NULL && |
| cached != ((PyDictObject *)dict)->ma_keys) { |
| /* PyDict_SetItem() may call dictresize and convert split table |
| * into combined table. In such case, convert it to split |
| * table again and update type's shared key only when this is |
| * the only dict sharing key with the type. |
| * |
| * This is to allow using shared key in class like this: |
| * |
| * class C: |
| * def __init__(self): |
| * # one dict resize happens |
| * self.a, self.b, self.c = 1, 2, 3 |
| * self.d, self.e, self.f = 4, 5, 6 |
| * a = C() |
| */ |
| if (cached->dk_refcnt == 1) { |
| CACHED_KEYS(tp) = make_keys_shared(dict); |
| } |
| else { |
| CACHED_KEYS(tp) = NULL; |
| } |
| dictkeys_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) |
| { |
| dictkeys_decref(keys); |
| } |