17db96d56Sopenharmony_ci/* Ordered Dictionary object implementation. 27db96d56Sopenharmony_ci 37db96d56Sopenharmony_ciThis implementation is necessarily explicitly equivalent to the pure Python 47db96d56Sopenharmony_ciOrderedDict class in Lib/collections/__init__.py. The strategy there 57db96d56Sopenharmony_ciinvolves using a doubly-linked-list to capture the order. We keep to that 67db96d56Sopenharmony_cistrategy, using a lower-level linked-list. 77db96d56Sopenharmony_ci 87db96d56Sopenharmony_ciAbout the Linked-List 97db96d56Sopenharmony_ci===================== 107db96d56Sopenharmony_ci 117db96d56Sopenharmony_ciFor the linked list we use a basic doubly-linked-list. Using a circularly- 127db96d56Sopenharmony_cilinked-list does have some benefits, but they don't apply so much here 137db96d56Sopenharmony_cisince OrderedDict is focused on the ends of the list (for the most part). 147db96d56Sopenharmony_ciFurthermore, there are some features of generic linked-lists that we simply 157db96d56Sopenharmony_cidon't need for OrderedDict. Thus a simple custom implementation meets our 167db96d56Sopenharmony_cineeds. Alternatives to our simple approach include the QCIRCLE_* 177db96d56Sopenharmony_cimacros from BSD's queue.h, and the linux's list.h. 187db96d56Sopenharmony_ci 197db96d56Sopenharmony_ciGetting O(1) Node Lookup 207db96d56Sopenharmony_ci------------------------ 217db96d56Sopenharmony_ci 227db96d56Sopenharmony_ciOne invariant of Python's OrderedDict is that it preserves time complexity 237db96d56Sopenharmony_ciof dict's methods, particularly the O(1) operations. Simply adding a 247db96d56Sopenharmony_cilinked-list on top of dict is not sufficient here; operations for nodes in 257db96d56Sopenharmony_cithe middle of the linked-list implicitly require finding the node first. 267db96d56Sopenharmony_ciWith a simple linked-list like we're using, that is an O(n) operation. 277db96d56Sopenharmony_ciConsequently, methods like __delitem__() would change from O(1) to O(n), 287db96d56Sopenharmony_ciwhich is unacceptable. 297db96d56Sopenharmony_ci 307db96d56Sopenharmony_ciIn order to preserve O(1) performance for node removal (finding nodes), we 317db96d56Sopenharmony_cimust do better than just looping through the linked-list. Here are options 327db96d56Sopenharmony_ciwe've considered: 337db96d56Sopenharmony_ci 347db96d56Sopenharmony_ci1. use a second dict to map keys to nodes (a la the pure Python version). 357db96d56Sopenharmony_ci2. keep a simple hash table mirroring the order of dict's, mapping each key 367db96d56Sopenharmony_ci to the corresponding node in the linked-list. 377db96d56Sopenharmony_ci3. use a version of shared keys (split dict) that allows non-unicode keys. 387db96d56Sopenharmony_ci4. have the value stored for each key be a (value, node) pair, and adjust 397db96d56Sopenharmony_ci __getitem__(), get(), etc. accordingly. 407db96d56Sopenharmony_ci 417db96d56Sopenharmony_ciThe approach with the least performance impact (time and space) is #2, 427db96d56Sopenharmony_cimirroring the key order of dict's dk_entries with an array of node pointers. 437db96d56Sopenharmony_ciWhile _Py_dict_lookup() does not give us the index into the array, 447db96d56Sopenharmony_ciwe make use of pointer arithmetic to get that index. An alternative would 457db96d56Sopenharmony_cibe to refactor _Py_dict_lookup() to provide the index, explicitly exposing 467db96d56Sopenharmony_cithe implementation detail. We could even just use a custom lookup function 477db96d56Sopenharmony_cifor OrderedDict that facilitates our need. However, both approaches are 487db96d56Sopenharmony_cisignificantly more complicated than just using pointer arithmetic. 497db96d56Sopenharmony_ci 507db96d56Sopenharmony_ciThe catch with mirroring the hash table ordering is that we have to keep 517db96d56Sopenharmony_cithe ordering in sync through any dict resizes. However, that order only 527db96d56Sopenharmony_cimatters during node lookup. We can simply defer any potential resizing 537db96d56Sopenharmony_ciuntil we need to do a lookup. 547db96d56Sopenharmony_ci 557db96d56Sopenharmony_ciLinked-List Nodes 567db96d56Sopenharmony_ci----------------- 577db96d56Sopenharmony_ci 587db96d56Sopenharmony_ciThe current implementation stores a pointer to the associated key only. 597db96d56Sopenharmony_ciOne alternative would be to store a pointer to the PyDictKeyEntry instead. 607db96d56Sopenharmony_ciThis would save one pointer de-reference per item, which is nice during 617db96d56Sopenharmony_cicalls to values() and items(). However, it adds unnecessary overhead 627db96d56Sopenharmony_ciotherwise, so we stick with just the key. 637db96d56Sopenharmony_ci 647db96d56Sopenharmony_ciLinked-List API 657db96d56Sopenharmony_ci--------------- 667db96d56Sopenharmony_ci 677db96d56Sopenharmony_ciAs noted, the linked-list implemented here does not have all the bells and 687db96d56Sopenharmony_ciwhistles. However, we recognize that the implementation may need to 697db96d56Sopenharmony_cichange to accommodate performance improvements or extra functionality. To 707db96d56Sopenharmony_cithat end, we use a simple API to interact with the linked-list. Here's a 717db96d56Sopenharmony_cisummary of the methods/macros: 727db96d56Sopenharmony_ci 737db96d56Sopenharmony_ciNode info: 747db96d56Sopenharmony_ci 757db96d56Sopenharmony_ci* _odictnode_KEY(node) 767db96d56Sopenharmony_ci* _odictnode_VALUE(od, node) 777db96d56Sopenharmony_ci* _odictnode_PREV(node) 787db96d56Sopenharmony_ci* _odictnode_NEXT(node) 797db96d56Sopenharmony_ci 807db96d56Sopenharmony_ciLinked-List info: 817db96d56Sopenharmony_ci 827db96d56Sopenharmony_ci* _odict_FIRST(od) 837db96d56Sopenharmony_ci* _odict_LAST(od) 847db96d56Sopenharmony_ci* _odict_EMPTY(od) 857db96d56Sopenharmony_ci* _odict_FOREACH(od, node) - used in place of `for (node=...)` 867db96d56Sopenharmony_ci 877db96d56Sopenharmony_ciFor adding nodes: 887db96d56Sopenharmony_ci 897db96d56Sopenharmony_ci* _odict_add_head(od, node) 907db96d56Sopenharmony_ci* _odict_add_tail(od, node) 917db96d56Sopenharmony_ci* _odict_add_new_node(od, key, hash) 927db96d56Sopenharmony_ci 937db96d56Sopenharmony_ciFor removing nodes: 947db96d56Sopenharmony_ci 957db96d56Sopenharmony_ci* _odict_clear_node(od, node, key, hash) 967db96d56Sopenharmony_ci* _odict_clear_nodes(od, clear_each) 977db96d56Sopenharmony_ci 987db96d56Sopenharmony_ciOthers: 997db96d56Sopenharmony_ci 1007db96d56Sopenharmony_ci* _odict_find_node_hash(od, key, hash) 1017db96d56Sopenharmony_ci* _odict_find_node(od, key) 1027db96d56Sopenharmony_ci* _odict_keys_equal(od1, od2) 1037db96d56Sopenharmony_ci 1047db96d56Sopenharmony_ciAnd here's a look at how the linked-list relates to the OrderedDict API: 1057db96d56Sopenharmony_ci 1067db96d56Sopenharmony_ci============ === === ==== ==== ==== === ==== ===== ==== ==== === ==== === === 1077db96d56Sopenharmony_cimethod key val prev next mem 1st last empty iter find add rmv clr keq 1087db96d56Sopenharmony_ci============ === === ==== ==== ==== === ==== ===== ==== ==== === ==== === === 1097db96d56Sopenharmony_ci__del__ ~ X 1107db96d56Sopenharmony_ci__delitem__ free ~ node 1117db96d56Sopenharmony_ci__eq__ ~ X 1127db96d56Sopenharmony_ci__iter__ X X 1137db96d56Sopenharmony_ci__new__ X X 1147db96d56Sopenharmony_ci__reduce__ X ~ X 1157db96d56Sopenharmony_ci__repr__ X X X 1167db96d56Sopenharmony_ci__reversed__ X X 1177db96d56Sopenharmony_ci__setitem__ key 1187db96d56Sopenharmony_ci__sizeof__ size X 1197db96d56Sopenharmony_ciclear ~ ~ X 1207db96d56Sopenharmony_cicopy X X X 1217db96d56Sopenharmony_ciitems X X X 1227db96d56Sopenharmony_cikeys X X 1237db96d56Sopenharmony_cimove_to_end X X X ~ h/t key 1247db96d56Sopenharmony_cipop free key 1257db96d56Sopenharmony_cipopitem X X free X X node 1267db96d56Sopenharmony_cisetdefault ~ ? ~ 1277db96d56Sopenharmony_civalues X X 1287db96d56Sopenharmony_ci============ === === ==== ==== ==== === ==== ===== ==== ==== === ==== === === 1297db96d56Sopenharmony_ci 1307db96d56Sopenharmony_ci__delitem__ is the only method that directly relies on finding an arbitrary 1317db96d56Sopenharmony_cinode in the linked-list. Everything else is iteration or relates to the 1327db96d56Sopenharmony_ciends of the linked-list. 1337db96d56Sopenharmony_ci 1347db96d56Sopenharmony_ciSituation that Endangers Consistency 1357db96d56Sopenharmony_ci------------------------------------ 1367db96d56Sopenharmony_ciUsing a raw linked-list for OrderedDict exposes a key situation that can 1377db96d56Sopenharmony_cicause problems. If a node is stored in a variable, there is a chance that 1387db96d56Sopenharmony_cithe node may have been deallocated before the variable gets used, thus 1397db96d56Sopenharmony_cipotentially leading to a segmentation fault. A key place where this shows 1407db96d56Sopenharmony_ciup is during iteration through the linked list (via _odict_FOREACH or 1417db96d56Sopenharmony_ciotherwise). 1427db96d56Sopenharmony_ci 1437db96d56Sopenharmony_ciA number of solutions are available to resolve this situation: 1447db96d56Sopenharmony_ci 1457db96d56Sopenharmony_ci* defer looking up the node until as late as possible and certainly after 1467db96d56Sopenharmony_ci any code that could possibly result in a deletion; 1477db96d56Sopenharmony_ci* if the node is needed both before and after a point where the node might 1487db96d56Sopenharmony_ci be removed, do a check before using the node at the "after" location to 1497db96d56Sopenharmony_ci see if the node is still valid; 1507db96d56Sopenharmony_ci* like the last one, but simply pull the node again to ensure it's right; 1517db96d56Sopenharmony_ci* keep the key in the variable instead of the node and then look up the 1527db96d56Sopenharmony_ci node using the key at the point where the node is needed (this is what 1537db96d56Sopenharmony_ci we do for the iterators). 1547db96d56Sopenharmony_ci 1557db96d56Sopenharmony_ciAnother related problem, preserving consistent ordering during iteration, 1567db96d56Sopenharmony_ciis described below. That one is not exclusive to using linked-lists. 1577db96d56Sopenharmony_ci 1587db96d56Sopenharmony_ci 1597db96d56Sopenharmony_ciChallenges from Subclassing dict 1607db96d56Sopenharmony_ci================================ 1617db96d56Sopenharmony_ci 1627db96d56Sopenharmony_ciOrderedDict subclasses dict, which is an unusual relationship between two 1637db96d56Sopenharmony_cibuiltin types (other than the base object type). Doing so results in 1647db96d56Sopenharmony_cisome complication and deserves further explanation. There are two things 1657db96d56Sopenharmony_cito consider here. First, in what circumstances or with what adjustments 1667db96d56Sopenharmony_cican OrderedDict be used as a drop-in replacement for dict (at the C level)? 1677db96d56Sopenharmony_ciSecond, how can the OrderedDict implementation leverage the dict 1687db96d56Sopenharmony_ciimplementation effectively without introducing unnecessary coupling or 1697db96d56Sopenharmony_ciinefficiencies? 1707db96d56Sopenharmony_ci 1717db96d56Sopenharmony_ciThis second point is reflected here and in the implementation, so the 1727db96d56Sopenharmony_cifurther focus is on the first point. It is worth noting that for 1737db96d56Sopenharmony_cioverridden methods, the dict implementation is deferred to as much as 1747db96d56Sopenharmony_cipossible. Furthermore, coupling is limited to as little as is reasonable. 1757db96d56Sopenharmony_ci 1767db96d56Sopenharmony_ciConcrete API Compatibility 1777db96d56Sopenharmony_ci-------------------------- 1787db96d56Sopenharmony_ci 1797db96d56Sopenharmony_ciUse of the concrete C-API for dict (PyDict_*) with OrderedDict is 1807db96d56Sopenharmony_ciproblematic. (See http://bugs.python.org/issue10977.) The concrete API 1817db96d56Sopenharmony_cihas a number of hard-coded assumptions tied to the dict implementation. 1827db96d56Sopenharmony_ciThis is, in part, due to performance reasons, which is understandable 1837db96d56Sopenharmony_cigiven the part dict plays in Python. 1847db96d56Sopenharmony_ci 1857db96d56Sopenharmony_ciAny attempt to replace dict with OrderedDict for any role in the 1867db96d56Sopenharmony_ciinterpreter (e.g. **kwds) faces a challenge. Such any effort must 1877db96d56Sopenharmony_cirecognize that the instances in affected locations currently interact with 1887db96d56Sopenharmony_cithe concrete API. 1897db96d56Sopenharmony_ci 1907db96d56Sopenharmony_ciHere are some ways to address this challenge: 1917db96d56Sopenharmony_ci 1927db96d56Sopenharmony_ci1. Change the relevant usage of the concrete API in CPython and add 1937db96d56Sopenharmony_ci PyDict_CheckExact() calls to each of the concrete API functions. 1947db96d56Sopenharmony_ci2. Adjust the relevant concrete API functions to explicitly accommodate 1957db96d56Sopenharmony_ci OrderedDict. 1967db96d56Sopenharmony_ci3. As with #1, add the checks, but improve the abstract API with smart fast 1977db96d56Sopenharmony_ci paths for dict and OrderedDict, and refactor CPython to use the abstract 1987db96d56Sopenharmony_ci API. Improvements to the abstract API would be valuable regardless. 1997db96d56Sopenharmony_ci 2007db96d56Sopenharmony_ciAdding the checks to the concrete API would help make any interpreter 2017db96d56Sopenharmony_ciswitch to OrderedDict less painful for extension modules. However, this 2027db96d56Sopenharmony_ciwon't work. The equivalent C API call to `dict.__setitem__(obj, k, v)` 2037db96d56Sopenharmony_ciis 'PyDict_SetItem(obj, k, v)`. This illustrates how subclasses in C call 2047db96d56Sopenharmony_cithe base class's methods, since there is no equivalent of super() in the 2057db96d56Sopenharmony_ciC API. Calling into Python for parent class API would work, but some 2067db96d56Sopenharmony_ciextension modules already rely on this feature of the concrete API. 2077db96d56Sopenharmony_ci 2087db96d56Sopenharmony_ciFor reference, here is a breakdown of some of the dict concrete API: 2097db96d56Sopenharmony_ci 2107db96d56Sopenharmony_ci========================== ============= ======================= 2117db96d56Sopenharmony_ciconcrete API uses abstract API 2127db96d56Sopenharmony_ci========================== ============= ======================= 2137db96d56Sopenharmony_ciPyDict_Check PyMapping_Check 2147db96d56Sopenharmony_ci(PyDict_CheckExact) - 2157db96d56Sopenharmony_ci(PyDict_New) - 2167db96d56Sopenharmony_ci(PyDictProxy_New) - 2177db96d56Sopenharmony_ciPyDict_Clear - 2187db96d56Sopenharmony_ciPyDict_Contains PySequence_Contains 2197db96d56Sopenharmony_ciPyDict_Copy - 2207db96d56Sopenharmony_ciPyDict_SetItem PyObject_SetItem 2217db96d56Sopenharmony_ciPyDict_SetItemString PyMapping_SetItemString 2227db96d56Sopenharmony_ciPyDict_DelItem PyMapping_DelItem 2237db96d56Sopenharmony_ciPyDict_DelItemString PyMapping_DelItemString 2247db96d56Sopenharmony_ciPyDict_GetItem - 2257db96d56Sopenharmony_ciPyDict_GetItemWithError PyObject_GetItem 2267db96d56Sopenharmony_ci_PyDict_GetItemIdWithError - 2277db96d56Sopenharmony_ciPyDict_GetItemString PyMapping_GetItemString 2287db96d56Sopenharmony_ciPyDict_Items PyMapping_Items 2297db96d56Sopenharmony_ciPyDict_Keys PyMapping_Keys 2307db96d56Sopenharmony_ciPyDict_Values PyMapping_Values 2317db96d56Sopenharmony_ciPyDict_Size PyMapping_Size 2327db96d56Sopenharmony_ci PyMapping_Length 2337db96d56Sopenharmony_ciPyDict_Next PyIter_Next 2347db96d56Sopenharmony_ci_PyDict_Next - 2357db96d56Sopenharmony_ciPyDict_Merge - 2367db96d56Sopenharmony_ciPyDict_Update - 2377db96d56Sopenharmony_ciPyDict_MergeFromSeq2 - 2387db96d56Sopenharmony_ciPyDict_ClearFreeList - 2397db96d56Sopenharmony_ci- PyMapping_HasKeyString 2407db96d56Sopenharmony_ci- PyMapping_HasKey 2417db96d56Sopenharmony_ci========================== ============= ======================= 2427db96d56Sopenharmony_ci 2437db96d56Sopenharmony_ci 2447db96d56Sopenharmony_ciThe dict Interface Relative to OrderedDict 2457db96d56Sopenharmony_ci========================================== 2467db96d56Sopenharmony_ci 2477db96d56Sopenharmony_ciSince OrderedDict subclasses dict, understanding the various methods and 2487db96d56Sopenharmony_ciattributes of dict is important for implementing OrderedDict. 2497db96d56Sopenharmony_ci 2507db96d56Sopenharmony_ciRelevant Type Slots 2517db96d56Sopenharmony_ci------------------- 2527db96d56Sopenharmony_ci 2537db96d56Sopenharmony_ci================= ================ =================== ================ 2547db96d56Sopenharmony_cislot attribute object dict 2557db96d56Sopenharmony_ci================= ================ =================== ================ 2567db96d56Sopenharmony_citp_dealloc - object_dealloc dict_dealloc 2577db96d56Sopenharmony_citp_repr __repr__ object_repr dict_repr 2587db96d56Sopenharmony_cisq_contains __contains__ - dict_contains 2597db96d56Sopenharmony_cimp_length __len__ - dict_length 2607db96d56Sopenharmony_cimp_subscript __getitem__ - dict_subscript 2617db96d56Sopenharmony_cimp_ass_subscript __setitem__ - dict_ass_sub 2627db96d56Sopenharmony_ci __delitem__ 2637db96d56Sopenharmony_citp_hash __hash__ _Py_HashPointer ..._HashNotImpl 2647db96d56Sopenharmony_citp_str __str__ object_str - 2657db96d56Sopenharmony_citp_getattro __getattribute__ ..._GenericGetAttr (repeated) 2667db96d56Sopenharmony_ci __getattr__ 2677db96d56Sopenharmony_citp_setattro __setattr__ ..._GenericSetAttr (disabled) 2687db96d56Sopenharmony_citp_doc __doc__ (literal) dictionary_doc 2697db96d56Sopenharmony_citp_traverse - - dict_traverse 2707db96d56Sopenharmony_citp_clear - - dict_tp_clear 2717db96d56Sopenharmony_citp_richcompare __eq__ object_richcompare dict_richcompare 2727db96d56Sopenharmony_ci __ne__ 2737db96d56Sopenharmony_citp_weaklistoffset (__weakref__) - - 2747db96d56Sopenharmony_citp_iter __iter__ - dict_iter 2757db96d56Sopenharmony_citp_dictoffset (__dict__) - - 2767db96d56Sopenharmony_citp_init __init__ object_init dict_init 2777db96d56Sopenharmony_citp_alloc - PyType_GenericAlloc (repeated) 2787db96d56Sopenharmony_citp_new __new__ object_new dict_new 2797db96d56Sopenharmony_citp_free - PyObject_Del PyObject_GC_Del 2807db96d56Sopenharmony_ci================= ================ =================== ================ 2817db96d56Sopenharmony_ci 2827db96d56Sopenharmony_ciRelevant Methods 2837db96d56Sopenharmony_ci---------------- 2847db96d56Sopenharmony_ci 2857db96d56Sopenharmony_ci================ =================== =============== 2867db96d56Sopenharmony_cimethod object dict 2877db96d56Sopenharmony_ci================ =================== =============== 2887db96d56Sopenharmony_ci__reduce__ object_reduce - 2897db96d56Sopenharmony_ci__sizeof__ object_sizeof dict_sizeof 2907db96d56Sopenharmony_ciclear - dict_clear 2917db96d56Sopenharmony_cicopy - dict_copy 2927db96d56Sopenharmony_cifromkeys - dict_fromkeys 2937db96d56Sopenharmony_ciget - dict_get 2947db96d56Sopenharmony_ciitems - dictitems_new 2957db96d56Sopenharmony_cikeys - dictkeys_new 2967db96d56Sopenharmony_cipop - dict_pop 2977db96d56Sopenharmony_cipopitem - dict_popitem 2987db96d56Sopenharmony_cisetdefault - dict_setdefault 2997db96d56Sopenharmony_ciupdate - dict_update 3007db96d56Sopenharmony_civalues - dictvalues_new 3017db96d56Sopenharmony_ci================ =================== =============== 3027db96d56Sopenharmony_ci 3037db96d56Sopenharmony_ci 3047db96d56Sopenharmony_ciPure Python OrderedDict 3057db96d56Sopenharmony_ci======================= 3067db96d56Sopenharmony_ci 3077db96d56Sopenharmony_ciAs already noted, compatibility with the pure Python OrderedDict 3087db96d56Sopenharmony_ciimplementation is a key goal of this C implementation. To further that 3097db96d56Sopenharmony_cigoal, here's a summary of how OrderedDict-specific methods are implemented 3107db96d56Sopenharmony_ciin collections/__init__.py. Also provided is an indication of which 3117db96d56Sopenharmony_cimethods directly mutate or iterate the object, as well as any relationship 3127db96d56Sopenharmony_ciwith the underlying linked-list. 3137db96d56Sopenharmony_ci 3147db96d56Sopenharmony_ci============= ============== == ================ === === ==== 3157db96d56Sopenharmony_cimethod impl used ll uses inq mut iter 3167db96d56Sopenharmony_ci============= ============== == ================ === === ==== 3177db96d56Sopenharmony_ci__contains__ dict - - X 3187db96d56Sopenharmony_ci__delitem__ OrderedDict Y dict.__delitem__ X 3197db96d56Sopenharmony_ci__eq__ OrderedDict N OrderedDict ~ 3207db96d56Sopenharmony_ci dict.__eq__ 3217db96d56Sopenharmony_ci __iter__ 3227db96d56Sopenharmony_ci__getitem__ dict - - X 3237db96d56Sopenharmony_ci__iter__ OrderedDict Y - X 3247db96d56Sopenharmony_ci__init__ OrderedDict N update 3257db96d56Sopenharmony_ci__len__ dict - - X 3267db96d56Sopenharmony_ci__ne__ MutableMapping - __eq__ ~ 3277db96d56Sopenharmony_ci__reduce__ OrderedDict N OrderedDict ~ 3287db96d56Sopenharmony_ci __iter__ 3297db96d56Sopenharmony_ci __getitem__ 3307db96d56Sopenharmony_ci__repr__ OrderedDict N __class__ ~ 3317db96d56Sopenharmony_ci items 3327db96d56Sopenharmony_ci__reversed__ OrderedDict Y - X 3337db96d56Sopenharmony_ci__setitem__ OrderedDict Y __contains__ X 3347db96d56Sopenharmony_ci dict.__setitem__ 3357db96d56Sopenharmony_ci__sizeof__ OrderedDict Y __len__ ~ 3367db96d56Sopenharmony_ci __dict__ 3377db96d56Sopenharmony_ciclear OrderedDict Y dict.clear X 3387db96d56Sopenharmony_cicopy OrderedDict N __class__ 3397db96d56Sopenharmony_ci __init__ 3407db96d56Sopenharmony_cifromkeys OrderedDict N __setitem__ 3417db96d56Sopenharmony_ciget dict - - ~ 3427db96d56Sopenharmony_ciitems MutableMapping - ItemsView X 3437db96d56Sopenharmony_cikeys MutableMapping - KeysView X 3447db96d56Sopenharmony_cimove_to_end OrderedDict Y - X 3457db96d56Sopenharmony_cipop OrderedDict N __contains__ X 3467db96d56Sopenharmony_ci __getitem__ 3477db96d56Sopenharmony_ci __delitem__ 3487db96d56Sopenharmony_cipopitem OrderedDict Y dict.pop X 3497db96d56Sopenharmony_cisetdefault OrderedDict N __contains__ ~ 3507db96d56Sopenharmony_ci __getitem__ 3517db96d56Sopenharmony_ci __setitem__ 3527db96d56Sopenharmony_ciupdate MutableMapping - __setitem__ ~ 3537db96d56Sopenharmony_civalues MutableMapping - ValuesView X 3547db96d56Sopenharmony_ci============= ============== == ================ === === ==== 3557db96d56Sopenharmony_ci 3567db96d56Sopenharmony_ci__reversed__ and move_to_end are both exclusive to OrderedDict. 3577db96d56Sopenharmony_ci 3587db96d56Sopenharmony_ci 3597db96d56Sopenharmony_ciC OrderedDict Implementation 3607db96d56Sopenharmony_ci============================ 3617db96d56Sopenharmony_ci 3627db96d56Sopenharmony_ci================= ================ 3637db96d56Sopenharmony_cislot impl 3647db96d56Sopenharmony_ci================= ================ 3657db96d56Sopenharmony_citp_dealloc odict_dealloc 3667db96d56Sopenharmony_citp_repr odict_repr 3677db96d56Sopenharmony_cimp_ass_subscript odict_ass_sub 3687db96d56Sopenharmony_citp_doc odict_doc 3697db96d56Sopenharmony_citp_traverse odict_traverse 3707db96d56Sopenharmony_citp_clear odict_tp_clear 3717db96d56Sopenharmony_citp_richcompare odict_richcompare 3727db96d56Sopenharmony_citp_weaklistoffset (offset) 3737db96d56Sopenharmony_citp_iter odict_iter 3747db96d56Sopenharmony_citp_dictoffset (offset) 3757db96d56Sopenharmony_citp_init odict_init 3767db96d56Sopenharmony_citp_alloc (repeated) 3777db96d56Sopenharmony_ci================= ================ 3787db96d56Sopenharmony_ci 3797db96d56Sopenharmony_ci================= ================ 3807db96d56Sopenharmony_cimethod impl 3817db96d56Sopenharmony_ci================= ================ 3827db96d56Sopenharmony_ci__reduce__ odict_reduce 3837db96d56Sopenharmony_ci__sizeof__ odict_sizeof 3847db96d56Sopenharmony_ciclear odict_clear 3857db96d56Sopenharmony_cicopy odict_copy 3867db96d56Sopenharmony_cifromkeys odict_fromkeys 3877db96d56Sopenharmony_ciitems odictitems_new 3887db96d56Sopenharmony_cikeys odictkeys_new 3897db96d56Sopenharmony_cipop odict_pop 3907db96d56Sopenharmony_cipopitem odict_popitem 3917db96d56Sopenharmony_cisetdefault odict_setdefault 3927db96d56Sopenharmony_ciupdate odict_update 3937db96d56Sopenharmony_civalues odictvalues_new 3947db96d56Sopenharmony_ci================= ================ 3957db96d56Sopenharmony_ci 3967db96d56Sopenharmony_ciInherited unchanged from object/dict: 3977db96d56Sopenharmony_ci 3987db96d56Sopenharmony_ci================ ========================== 3997db96d56Sopenharmony_cimethod type field 4007db96d56Sopenharmony_ci================ ========================== 4017db96d56Sopenharmony_ci- tp_free 4027db96d56Sopenharmony_ci__contains__ tp_as_sequence.sq_contains 4037db96d56Sopenharmony_ci__getattr__ tp_getattro 4047db96d56Sopenharmony_ci__getattribute__ tp_getattro 4057db96d56Sopenharmony_ci__getitem__ tp_as_mapping.mp_subscript 4067db96d56Sopenharmony_ci__hash__ tp_hash 4077db96d56Sopenharmony_ci__len__ tp_as_mapping.mp_length 4087db96d56Sopenharmony_ci__setattr__ tp_setattro 4097db96d56Sopenharmony_ci__str__ tp_str 4107db96d56Sopenharmony_ciget - 4117db96d56Sopenharmony_ci================ ========================== 4127db96d56Sopenharmony_ci 4137db96d56Sopenharmony_ci 4147db96d56Sopenharmony_ciOther Challenges 4157db96d56Sopenharmony_ci================ 4167db96d56Sopenharmony_ci 4177db96d56Sopenharmony_ciPreserving Ordering During Iteration 4187db96d56Sopenharmony_ci------------------------------------ 4197db96d56Sopenharmony_ciDuring iteration through an OrderedDict, it is possible that items could 4207db96d56Sopenharmony_ciget added, removed, or reordered. For a linked-list implementation, as 4217db96d56Sopenharmony_ciwith some other implementations, that situation may lead to undefined 4227db96d56Sopenharmony_cibehavior. The documentation for dict mentions this in the `iter()` section 4237db96d56Sopenharmony_ciof http://docs.python.org/3.4/library/stdtypes.html#dictionary-view-objects. 4247db96d56Sopenharmony_ciIn this implementation we follow dict's lead (as does the pure Python 4257db96d56Sopenharmony_ciimplementation) for __iter__(), keys(), values(), and items(). 4267db96d56Sopenharmony_ci 4277db96d56Sopenharmony_ciFor internal iteration (using _odict_FOREACH or not), there is still the 4287db96d56Sopenharmony_cirisk that not all nodes that we expect to be seen in the loop actually get 4297db96d56Sopenharmony_ciseen. Thus, we are careful in each of those places to ensure that they 4307db96d56Sopenharmony_ciare. This comes, of course, at a small price at each location. The 4317db96d56Sopenharmony_cisolutions are much the same as those detailed in the `Situation that 4327db96d56Sopenharmony_ciEndangers Consistency` section above. 4337db96d56Sopenharmony_ci 4347db96d56Sopenharmony_ci 4357db96d56Sopenharmony_ciPotential Optimizations 4367db96d56Sopenharmony_ci======================= 4377db96d56Sopenharmony_ci 4387db96d56Sopenharmony_ci* Allocate the nodes as a block via od_fast_nodes instead of individually. 4397db96d56Sopenharmony_ci - Set node->key to NULL to indicate the node is not-in-use. 4407db96d56Sopenharmony_ci - Add _odict_EXISTS()? 4417db96d56Sopenharmony_ci - How to maintain consistency across resizes? Existing node pointers 4427db96d56Sopenharmony_ci would be invalidated after a resize, which is particularly problematic 4437db96d56Sopenharmony_ci for the iterators. 4447db96d56Sopenharmony_ci* Use a more stream-lined implementation of update() and, likely indirectly, 4457db96d56Sopenharmony_ci __init__(). 4467db96d56Sopenharmony_ci 4477db96d56Sopenharmony_ci*/ 4487db96d56Sopenharmony_ci 4497db96d56Sopenharmony_ci/* TODO 4507db96d56Sopenharmony_ci 4517db96d56Sopenharmony_cisooner: 4527db96d56Sopenharmony_ci- reentrancy (make sure everything is at a thread-safe state when calling 4537db96d56Sopenharmony_ci into Python). I've already checked this multiple times, but want to 4547db96d56Sopenharmony_ci make one more pass. 4557db96d56Sopenharmony_ci- add unit tests for reentrancy? 4567db96d56Sopenharmony_ci 4577db96d56Sopenharmony_cilater: 4587db96d56Sopenharmony_ci- make the dict views support the full set API (the pure Python impl does) 4597db96d56Sopenharmony_ci- implement a fuller MutableMapping API in C? 4607db96d56Sopenharmony_ci- move the MutableMapping implementation to abstract.c? 4617db96d56Sopenharmony_ci- optimize mutablemapping_update 4627db96d56Sopenharmony_ci- use PyObject_Malloc (small object allocator) for odict nodes? 4637db96d56Sopenharmony_ci- support subclasses better (e.g. in odict_richcompare) 4647db96d56Sopenharmony_ci 4657db96d56Sopenharmony_ci*/ 4667db96d56Sopenharmony_ci 4677db96d56Sopenharmony_ci#include "Python.h" 4687db96d56Sopenharmony_ci#include "pycore_call.h" // _PyObject_CallNoArgs() 4697db96d56Sopenharmony_ci#include "pycore_object.h" // _PyObject_GC_UNTRACK() 4707db96d56Sopenharmony_ci#include "pycore_dict.h" // _Py_dict_lookup() 4717db96d56Sopenharmony_ci#include <stddef.h> // offsetof() 4727db96d56Sopenharmony_ci 4737db96d56Sopenharmony_ci#include "clinic/odictobject.c.h" 4747db96d56Sopenharmony_ci 4757db96d56Sopenharmony_ci/*[clinic input] 4767db96d56Sopenharmony_ciclass OrderedDict "PyODictObject *" "&PyODict_Type" 4777db96d56Sopenharmony_ci[clinic start generated code]*/ 4787db96d56Sopenharmony_ci/*[clinic end generated code: output=da39a3ee5e6b4b0d input=ca0641cf6143d4af]*/ 4797db96d56Sopenharmony_ci 4807db96d56Sopenharmony_ci 4817db96d56Sopenharmony_citypedef struct _odictnode _ODictNode; 4827db96d56Sopenharmony_ci 4837db96d56Sopenharmony_ci/* PyODictObject */ 4847db96d56Sopenharmony_cistruct _odictobject { 4857db96d56Sopenharmony_ci PyDictObject od_dict; /* the underlying dict */ 4867db96d56Sopenharmony_ci _ODictNode *od_first; /* first node in the linked list, if any */ 4877db96d56Sopenharmony_ci _ODictNode *od_last; /* last node in the linked list, if any */ 4887db96d56Sopenharmony_ci /* od_fast_nodes, od_fast_nodes_size and od_resize_sentinel are managed 4897db96d56Sopenharmony_ci * by _odict_resize(). 4907db96d56Sopenharmony_ci * Note that we rely on implementation details of dict for both. */ 4917db96d56Sopenharmony_ci _ODictNode **od_fast_nodes; /* hash table that mirrors the dict table */ 4927db96d56Sopenharmony_ci Py_ssize_t od_fast_nodes_size; 4937db96d56Sopenharmony_ci void *od_resize_sentinel; /* changes if odict should be resized */ 4947db96d56Sopenharmony_ci 4957db96d56Sopenharmony_ci size_t od_state; /* incremented whenever the LL changes */ 4967db96d56Sopenharmony_ci PyObject *od_inst_dict; /* OrderedDict().__dict__ */ 4977db96d56Sopenharmony_ci PyObject *od_weakreflist; /* holds weakrefs to the odict */ 4987db96d56Sopenharmony_ci}; 4997db96d56Sopenharmony_ci 5007db96d56Sopenharmony_ci 5017db96d56Sopenharmony_ci/* ---------------------------------------------- 5027db96d56Sopenharmony_ci * odict keys (a simple doubly-linked list) 5037db96d56Sopenharmony_ci */ 5047db96d56Sopenharmony_ci 5057db96d56Sopenharmony_cistruct _odictnode { 5067db96d56Sopenharmony_ci PyObject *key; 5077db96d56Sopenharmony_ci Py_hash_t hash; 5087db96d56Sopenharmony_ci _ODictNode *next; 5097db96d56Sopenharmony_ci _ODictNode *prev; 5107db96d56Sopenharmony_ci}; 5117db96d56Sopenharmony_ci 5127db96d56Sopenharmony_ci#define _odictnode_KEY(node) \ 5137db96d56Sopenharmony_ci (node->key) 5147db96d56Sopenharmony_ci#define _odictnode_HASH(node) \ 5157db96d56Sopenharmony_ci (node->hash) 5167db96d56Sopenharmony_ci/* borrowed reference */ 5177db96d56Sopenharmony_ci#define _odictnode_VALUE(node, od) \ 5187db96d56Sopenharmony_ci PyODict_GetItemWithError((PyObject *)od, _odictnode_KEY(node)) 5197db96d56Sopenharmony_ci#define _odictnode_PREV(node) (node->prev) 5207db96d56Sopenharmony_ci#define _odictnode_NEXT(node) (node->next) 5217db96d56Sopenharmony_ci 5227db96d56Sopenharmony_ci#define _odict_FIRST(od) (((PyODictObject *)od)->od_first) 5237db96d56Sopenharmony_ci#define _odict_LAST(od) (((PyODictObject *)od)->od_last) 5247db96d56Sopenharmony_ci#define _odict_EMPTY(od) (_odict_FIRST(od) == NULL) 5257db96d56Sopenharmony_ci#define _odict_FOREACH(od, node) \ 5267db96d56Sopenharmony_ci for (node = _odict_FIRST(od); node != NULL; node = _odictnode_NEXT(node)) 5277db96d56Sopenharmony_ci 5287db96d56Sopenharmony_ci/* Return the index into the hash table, regardless of a valid node. */ 5297db96d56Sopenharmony_cistatic Py_ssize_t 5307db96d56Sopenharmony_ci_odict_get_index_raw(PyODictObject *od, PyObject *key, Py_hash_t hash) 5317db96d56Sopenharmony_ci{ 5327db96d56Sopenharmony_ci PyObject *value = NULL; 5337db96d56Sopenharmony_ci PyDictKeysObject *keys = ((PyDictObject *)od)->ma_keys; 5347db96d56Sopenharmony_ci Py_ssize_t ix; 5357db96d56Sopenharmony_ci 5367db96d56Sopenharmony_ci ix = _Py_dict_lookup((PyDictObject *)od, key, hash, &value); 5377db96d56Sopenharmony_ci if (ix == DKIX_EMPTY) { 5387db96d56Sopenharmony_ci return keys->dk_nentries; /* index of new entry */ 5397db96d56Sopenharmony_ci } 5407db96d56Sopenharmony_ci if (ix < 0) 5417db96d56Sopenharmony_ci return -1; 5427db96d56Sopenharmony_ci /* We use pointer arithmetic to get the entry's index into the table. */ 5437db96d56Sopenharmony_ci return ix; 5447db96d56Sopenharmony_ci} 5457db96d56Sopenharmony_ci 5467db96d56Sopenharmony_ci#define ONE ((Py_ssize_t)1) 5477db96d56Sopenharmony_ci 5487db96d56Sopenharmony_ci/* Replace od->od_fast_nodes with a new table matching the size of dict's. */ 5497db96d56Sopenharmony_cistatic int 5507db96d56Sopenharmony_ci_odict_resize(PyODictObject *od) 5517db96d56Sopenharmony_ci{ 5527db96d56Sopenharmony_ci Py_ssize_t size, i; 5537db96d56Sopenharmony_ci _ODictNode **fast_nodes, *node; 5547db96d56Sopenharmony_ci 5557db96d56Sopenharmony_ci /* Initialize a new "fast nodes" table. */ 5567db96d56Sopenharmony_ci size = ONE << (((PyDictObject *)od)->ma_keys->dk_log2_size); 5577db96d56Sopenharmony_ci fast_nodes = PyMem_NEW(_ODictNode *, size); 5587db96d56Sopenharmony_ci if (fast_nodes == NULL) { 5597db96d56Sopenharmony_ci PyErr_NoMemory(); 5607db96d56Sopenharmony_ci return -1; 5617db96d56Sopenharmony_ci } 5627db96d56Sopenharmony_ci for (i = 0; i < size; i++) 5637db96d56Sopenharmony_ci fast_nodes[i] = NULL; 5647db96d56Sopenharmony_ci 5657db96d56Sopenharmony_ci /* Copy the current nodes into the table. */ 5667db96d56Sopenharmony_ci _odict_FOREACH(od, node) { 5677db96d56Sopenharmony_ci i = _odict_get_index_raw(od, _odictnode_KEY(node), 5687db96d56Sopenharmony_ci _odictnode_HASH(node)); 5697db96d56Sopenharmony_ci if (i < 0) { 5707db96d56Sopenharmony_ci PyMem_Free(fast_nodes); 5717db96d56Sopenharmony_ci return -1; 5727db96d56Sopenharmony_ci } 5737db96d56Sopenharmony_ci fast_nodes[i] = node; 5747db96d56Sopenharmony_ci } 5757db96d56Sopenharmony_ci 5767db96d56Sopenharmony_ci /* Replace the old fast nodes table. */ 5777db96d56Sopenharmony_ci PyMem_Free(od->od_fast_nodes); 5787db96d56Sopenharmony_ci od->od_fast_nodes = fast_nodes; 5797db96d56Sopenharmony_ci od->od_fast_nodes_size = size; 5807db96d56Sopenharmony_ci od->od_resize_sentinel = ((PyDictObject *)od)->ma_keys; 5817db96d56Sopenharmony_ci return 0; 5827db96d56Sopenharmony_ci} 5837db96d56Sopenharmony_ci 5847db96d56Sopenharmony_ci/* Return the index into the hash table, regardless of a valid node. */ 5857db96d56Sopenharmony_cistatic Py_ssize_t 5867db96d56Sopenharmony_ci_odict_get_index(PyODictObject *od, PyObject *key, Py_hash_t hash) 5877db96d56Sopenharmony_ci{ 5887db96d56Sopenharmony_ci PyDictKeysObject *keys; 5897db96d56Sopenharmony_ci 5907db96d56Sopenharmony_ci assert(key != NULL); 5917db96d56Sopenharmony_ci keys = ((PyDictObject *)od)->ma_keys; 5927db96d56Sopenharmony_ci 5937db96d56Sopenharmony_ci /* Ensure od_fast_nodes and dk_entries are in sync. */ 5947db96d56Sopenharmony_ci if (od->od_resize_sentinel != keys || 5957db96d56Sopenharmony_ci od->od_fast_nodes_size != (ONE << (keys->dk_log2_size))) { 5967db96d56Sopenharmony_ci int resize_res = _odict_resize(od); 5977db96d56Sopenharmony_ci if (resize_res < 0) 5987db96d56Sopenharmony_ci return -1; 5997db96d56Sopenharmony_ci } 6007db96d56Sopenharmony_ci 6017db96d56Sopenharmony_ci return _odict_get_index_raw(od, key, hash); 6027db96d56Sopenharmony_ci} 6037db96d56Sopenharmony_ci 6047db96d56Sopenharmony_ci/* Returns NULL if there was some error or the key was not found. */ 6057db96d56Sopenharmony_cistatic _ODictNode * 6067db96d56Sopenharmony_ci_odict_find_node_hash(PyODictObject *od, PyObject *key, Py_hash_t hash) 6077db96d56Sopenharmony_ci{ 6087db96d56Sopenharmony_ci Py_ssize_t index; 6097db96d56Sopenharmony_ci 6107db96d56Sopenharmony_ci if (_odict_EMPTY(od)) 6117db96d56Sopenharmony_ci return NULL; 6127db96d56Sopenharmony_ci index = _odict_get_index(od, key, hash); 6137db96d56Sopenharmony_ci if (index < 0) 6147db96d56Sopenharmony_ci return NULL; 6157db96d56Sopenharmony_ci assert(od->od_fast_nodes != NULL); 6167db96d56Sopenharmony_ci return od->od_fast_nodes[index]; 6177db96d56Sopenharmony_ci} 6187db96d56Sopenharmony_ci 6197db96d56Sopenharmony_cistatic _ODictNode * 6207db96d56Sopenharmony_ci_odict_find_node(PyODictObject *od, PyObject *key) 6217db96d56Sopenharmony_ci{ 6227db96d56Sopenharmony_ci Py_ssize_t index; 6237db96d56Sopenharmony_ci Py_hash_t hash; 6247db96d56Sopenharmony_ci 6257db96d56Sopenharmony_ci if (_odict_EMPTY(od)) 6267db96d56Sopenharmony_ci return NULL; 6277db96d56Sopenharmony_ci hash = PyObject_Hash(key); 6287db96d56Sopenharmony_ci if (hash == -1) 6297db96d56Sopenharmony_ci return NULL; 6307db96d56Sopenharmony_ci index = _odict_get_index(od, key, hash); 6317db96d56Sopenharmony_ci if (index < 0) 6327db96d56Sopenharmony_ci return NULL; 6337db96d56Sopenharmony_ci assert(od->od_fast_nodes != NULL); 6347db96d56Sopenharmony_ci return od->od_fast_nodes[index]; 6357db96d56Sopenharmony_ci} 6367db96d56Sopenharmony_ci 6377db96d56Sopenharmony_cistatic void 6387db96d56Sopenharmony_ci_odict_add_head(PyODictObject *od, _ODictNode *node) 6397db96d56Sopenharmony_ci{ 6407db96d56Sopenharmony_ci _odictnode_PREV(node) = NULL; 6417db96d56Sopenharmony_ci _odictnode_NEXT(node) = _odict_FIRST(od); 6427db96d56Sopenharmony_ci if (_odict_FIRST(od) == NULL) 6437db96d56Sopenharmony_ci _odict_LAST(od) = node; 6447db96d56Sopenharmony_ci else 6457db96d56Sopenharmony_ci _odictnode_PREV(_odict_FIRST(od)) = node; 6467db96d56Sopenharmony_ci _odict_FIRST(od) = node; 6477db96d56Sopenharmony_ci od->od_state++; 6487db96d56Sopenharmony_ci} 6497db96d56Sopenharmony_ci 6507db96d56Sopenharmony_cistatic void 6517db96d56Sopenharmony_ci_odict_add_tail(PyODictObject *od, _ODictNode *node) 6527db96d56Sopenharmony_ci{ 6537db96d56Sopenharmony_ci _odictnode_PREV(node) = _odict_LAST(od); 6547db96d56Sopenharmony_ci _odictnode_NEXT(node) = NULL; 6557db96d56Sopenharmony_ci if (_odict_LAST(od) == NULL) 6567db96d56Sopenharmony_ci _odict_FIRST(od) = node; 6577db96d56Sopenharmony_ci else 6587db96d56Sopenharmony_ci _odictnode_NEXT(_odict_LAST(od)) = node; 6597db96d56Sopenharmony_ci _odict_LAST(od) = node; 6607db96d56Sopenharmony_ci od->od_state++; 6617db96d56Sopenharmony_ci} 6627db96d56Sopenharmony_ci 6637db96d56Sopenharmony_ci/* adds the node to the end of the list */ 6647db96d56Sopenharmony_cistatic int 6657db96d56Sopenharmony_ci_odict_add_new_node(PyODictObject *od, PyObject *key, Py_hash_t hash) 6667db96d56Sopenharmony_ci{ 6677db96d56Sopenharmony_ci Py_ssize_t i; 6687db96d56Sopenharmony_ci _ODictNode *node; 6697db96d56Sopenharmony_ci 6707db96d56Sopenharmony_ci Py_INCREF(key); 6717db96d56Sopenharmony_ci i = _odict_get_index(od, key, hash); 6727db96d56Sopenharmony_ci if (i < 0) { 6737db96d56Sopenharmony_ci if (!PyErr_Occurred()) 6747db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 6757db96d56Sopenharmony_ci Py_DECREF(key); 6767db96d56Sopenharmony_ci return -1; 6777db96d56Sopenharmony_ci } 6787db96d56Sopenharmony_ci assert(od->od_fast_nodes != NULL); 6797db96d56Sopenharmony_ci if (od->od_fast_nodes[i] != NULL) { 6807db96d56Sopenharmony_ci /* We already have a node for the key so there's no need to add one. */ 6817db96d56Sopenharmony_ci Py_DECREF(key); 6827db96d56Sopenharmony_ci return 0; 6837db96d56Sopenharmony_ci } 6847db96d56Sopenharmony_ci 6857db96d56Sopenharmony_ci /* must not be added yet */ 6867db96d56Sopenharmony_ci node = (_ODictNode *)PyMem_Malloc(sizeof(_ODictNode)); 6877db96d56Sopenharmony_ci if (node == NULL) { 6887db96d56Sopenharmony_ci Py_DECREF(key); 6897db96d56Sopenharmony_ci PyErr_NoMemory(); 6907db96d56Sopenharmony_ci return -1; 6917db96d56Sopenharmony_ci } 6927db96d56Sopenharmony_ci 6937db96d56Sopenharmony_ci _odictnode_KEY(node) = key; 6947db96d56Sopenharmony_ci _odictnode_HASH(node) = hash; 6957db96d56Sopenharmony_ci _odict_add_tail(od, node); 6967db96d56Sopenharmony_ci od->od_fast_nodes[i] = node; 6977db96d56Sopenharmony_ci return 0; 6987db96d56Sopenharmony_ci} 6997db96d56Sopenharmony_ci 7007db96d56Sopenharmony_ci/* Putting the decref after the free causes problems. */ 7017db96d56Sopenharmony_ci#define _odictnode_DEALLOC(node) \ 7027db96d56Sopenharmony_ci do { \ 7037db96d56Sopenharmony_ci Py_DECREF(_odictnode_KEY(node)); \ 7047db96d56Sopenharmony_ci PyMem_Free((void *)node); \ 7057db96d56Sopenharmony_ci } while (0) 7067db96d56Sopenharmony_ci 7077db96d56Sopenharmony_ci/* Repeated calls on the same node are no-ops. */ 7087db96d56Sopenharmony_cistatic void 7097db96d56Sopenharmony_ci_odict_remove_node(PyODictObject *od, _ODictNode *node) 7107db96d56Sopenharmony_ci{ 7117db96d56Sopenharmony_ci if (_odict_FIRST(od) == node) 7127db96d56Sopenharmony_ci _odict_FIRST(od) = _odictnode_NEXT(node); 7137db96d56Sopenharmony_ci else if (_odictnode_PREV(node) != NULL) 7147db96d56Sopenharmony_ci _odictnode_NEXT(_odictnode_PREV(node)) = _odictnode_NEXT(node); 7157db96d56Sopenharmony_ci 7167db96d56Sopenharmony_ci if (_odict_LAST(od) == node) 7177db96d56Sopenharmony_ci _odict_LAST(od) = _odictnode_PREV(node); 7187db96d56Sopenharmony_ci else if (_odictnode_NEXT(node) != NULL) 7197db96d56Sopenharmony_ci _odictnode_PREV(_odictnode_NEXT(node)) = _odictnode_PREV(node); 7207db96d56Sopenharmony_ci 7217db96d56Sopenharmony_ci _odictnode_PREV(node) = NULL; 7227db96d56Sopenharmony_ci _odictnode_NEXT(node) = NULL; 7237db96d56Sopenharmony_ci od->od_state++; 7247db96d56Sopenharmony_ci} 7257db96d56Sopenharmony_ci 7267db96d56Sopenharmony_ci/* If someone calls PyDict_DelItem() directly on an OrderedDict, we'll 7277db96d56Sopenharmony_ci get all sorts of problems here. In PyODict_DelItem we make sure to 7287db96d56Sopenharmony_ci call _odict_clear_node first. 7297db96d56Sopenharmony_ci 7307db96d56Sopenharmony_ci This matters in the case of colliding keys. Suppose we add 3 keys: 7317db96d56Sopenharmony_ci [A, B, C], where the hash of C collides with A and the next possible 7327db96d56Sopenharmony_ci index in the hash table is occupied by B. If we remove B then for C 7337db96d56Sopenharmony_ci the dict's looknode func will give us the old index of B instead of 7347db96d56Sopenharmony_ci the index we got before deleting B. However, the node for C in 7357db96d56Sopenharmony_ci od_fast_nodes is still at the old dict index of C. Thus to be sure 7367db96d56Sopenharmony_ci things don't get out of sync, we clear the node in od_fast_nodes 7377db96d56Sopenharmony_ci *before* calling PyDict_DelItem. 7387db96d56Sopenharmony_ci 7397db96d56Sopenharmony_ci The same must be done for any other OrderedDict operations where 7407db96d56Sopenharmony_ci we modify od_fast_nodes. 7417db96d56Sopenharmony_ci*/ 7427db96d56Sopenharmony_cistatic int 7437db96d56Sopenharmony_ci_odict_clear_node(PyODictObject *od, _ODictNode *node, PyObject *key, 7447db96d56Sopenharmony_ci Py_hash_t hash) 7457db96d56Sopenharmony_ci{ 7467db96d56Sopenharmony_ci Py_ssize_t i; 7477db96d56Sopenharmony_ci 7487db96d56Sopenharmony_ci assert(key != NULL); 7497db96d56Sopenharmony_ci if (_odict_EMPTY(od)) { 7507db96d56Sopenharmony_ci /* Let later code decide if this is a KeyError. */ 7517db96d56Sopenharmony_ci return 0; 7527db96d56Sopenharmony_ci } 7537db96d56Sopenharmony_ci 7547db96d56Sopenharmony_ci i = _odict_get_index(od, key, hash); 7557db96d56Sopenharmony_ci if (i < 0) 7567db96d56Sopenharmony_ci return PyErr_Occurred() ? -1 : 0; 7577db96d56Sopenharmony_ci 7587db96d56Sopenharmony_ci assert(od->od_fast_nodes != NULL); 7597db96d56Sopenharmony_ci if (node == NULL) 7607db96d56Sopenharmony_ci node = od->od_fast_nodes[i]; 7617db96d56Sopenharmony_ci assert(node == od->od_fast_nodes[i]); 7627db96d56Sopenharmony_ci if (node == NULL) { 7637db96d56Sopenharmony_ci /* Let later code decide if this is a KeyError. */ 7647db96d56Sopenharmony_ci return 0; 7657db96d56Sopenharmony_ci } 7667db96d56Sopenharmony_ci 7677db96d56Sopenharmony_ci // Now clear the node. 7687db96d56Sopenharmony_ci od->od_fast_nodes[i] = NULL; 7697db96d56Sopenharmony_ci _odict_remove_node(od, node); 7707db96d56Sopenharmony_ci _odictnode_DEALLOC(node); 7717db96d56Sopenharmony_ci return 0; 7727db96d56Sopenharmony_ci} 7737db96d56Sopenharmony_ci 7747db96d56Sopenharmony_cistatic void 7757db96d56Sopenharmony_ci_odict_clear_nodes(PyODictObject *od) 7767db96d56Sopenharmony_ci{ 7777db96d56Sopenharmony_ci _ODictNode *node, *next; 7787db96d56Sopenharmony_ci 7797db96d56Sopenharmony_ci PyMem_Free(od->od_fast_nodes); 7807db96d56Sopenharmony_ci od->od_fast_nodes = NULL; 7817db96d56Sopenharmony_ci od->od_fast_nodes_size = 0; 7827db96d56Sopenharmony_ci od->od_resize_sentinel = NULL; 7837db96d56Sopenharmony_ci 7847db96d56Sopenharmony_ci node = _odict_FIRST(od); 7857db96d56Sopenharmony_ci _odict_FIRST(od) = NULL; 7867db96d56Sopenharmony_ci _odict_LAST(od) = NULL; 7877db96d56Sopenharmony_ci while (node != NULL) { 7887db96d56Sopenharmony_ci next = _odictnode_NEXT(node); 7897db96d56Sopenharmony_ci _odictnode_DEALLOC(node); 7907db96d56Sopenharmony_ci node = next; 7917db96d56Sopenharmony_ci } 7927db96d56Sopenharmony_ci} 7937db96d56Sopenharmony_ci 7947db96d56Sopenharmony_ci/* There isn't any memory management of nodes past this point. */ 7957db96d56Sopenharmony_ci#undef _odictnode_DEALLOC 7967db96d56Sopenharmony_ci 7977db96d56Sopenharmony_cistatic int 7987db96d56Sopenharmony_ci_odict_keys_equal(PyODictObject *a, PyODictObject *b) 7997db96d56Sopenharmony_ci{ 8007db96d56Sopenharmony_ci _ODictNode *node_a, *node_b; 8017db96d56Sopenharmony_ci 8027db96d56Sopenharmony_ci node_a = _odict_FIRST(a); 8037db96d56Sopenharmony_ci node_b = _odict_FIRST(b); 8047db96d56Sopenharmony_ci while (1) { 8057db96d56Sopenharmony_ci if (node_a == NULL && node_b == NULL) 8067db96d56Sopenharmony_ci /* success: hit the end of each at the same time */ 8077db96d56Sopenharmony_ci return 1; 8087db96d56Sopenharmony_ci else if (node_a == NULL || node_b == NULL) 8097db96d56Sopenharmony_ci /* unequal length */ 8107db96d56Sopenharmony_ci return 0; 8117db96d56Sopenharmony_ci else { 8127db96d56Sopenharmony_ci int res = PyObject_RichCompareBool( 8137db96d56Sopenharmony_ci (PyObject *)_odictnode_KEY(node_a), 8147db96d56Sopenharmony_ci (PyObject *)_odictnode_KEY(node_b), 8157db96d56Sopenharmony_ci Py_EQ); 8167db96d56Sopenharmony_ci if (res < 0) 8177db96d56Sopenharmony_ci return res; 8187db96d56Sopenharmony_ci else if (res == 0) 8197db96d56Sopenharmony_ci return 0; 8207db96d56Sopenharmony_ci 8217db96d56Sopenharmony_ci /* otherwise it must match, so move on to the next one */ 8227db96d56Sopenharmony_ci node_a = _odictnode_NEXT(node_a); 8237db96d56Sopenharmony_ci node_b = _odictnode_NEXT(node_b); 8247db96d56Sopenharmony_ci } 8257db96d56Sopenharmony_ci } 8267db96d56Sopenharmony_ci} 8277db96d56Sopenharmony_ci 8287db96d56Sopenharmony_ci 8297db96d56Sopenharmony_ci/* ---------------------------------------------- 8307db96d56Sopenharmony_ci * OrderedDict mapping methods 8317db96d56Sopenharmony_ci */ 8327db96d56Sopenharmony_ci 8337db96d56Sopenharmony_ci/* mp_ass_subscript: __setitem__() and __delitem__() */ 8347db96d56Sopenharmony_ci 8357db96d56Sopenharmony_cistatic int 8367db96d56Sopenharmony_ciodict_mp_ass_sub(PyODictObject *od, PyObject *v, PyObject *w) 8377db96d56Sopenharmony_ci{ 8387db96d56Sopenharmony_ci if (w == NULL) 8397db96d56Sopenharmony_ci return PyODict_DelItem((PyObject *)od, v); 8407db96d56Sopenharmony_ci else 8417db96d56Sopenharmony_ci return PyODict_SetItem((PyObject *)od, v, w); 8427db96d56Sopenharmony_ci} 8437db96d56Sopenharmony_ci 8447db96d56Sopenharmony_ci/* tp_as_mapping */ 8457db96d56Sopenharmony_ci 8467db96d56Sopenharmony_cistatic PyMappingMethods odict_as_mapping = { 8477db96d56Sopenharmony_ci 0, /*mp_length*/ 8487db96d56Sopenharmony_ci 0, /*mp_subscript*/ 8497db96d56Sopenharmony_ci (objobjargproc)odict_mp_ass_sub, /*mp_ass_subscript*/ 8507db96d56Sopenharmony_ci}; 8517db96d56Sopenharmony_ci 8527db96d56Sopenharmony_ci 8537db96d56Sopenharmony_ci/* ---------------------------------------------- 8547db96d56Sopenharmony_ci * OrderedDict number methods 8557db96d56Sopenharmony_ci */ 8567db96d56Sopenharmony_ci 8577db96d56Sopenharmony_cistatic int mutablemapping_update_arg(PyObject*, PyObject*); 8587db96d56Sopenharmony_ci 8597db96d56Sopenharmony_cistatic PyObject * 8607db96d56Sopenharmony_ciodict_or(PyObject *left, PyObject *right) 8617db96d56Sopenharmony_ci{ 8627db96d56Sopenharmony_ci PyTypeObject *type; 8637db96d56Sopenharmony_ci PyObject *other; 8647db96d56Sopenharmony_ci if (PyODict_Check(left)) { 8657db96d56Sopenharmony_ci type = Py_TYPE(left); 8667db96d56Sopenharmony_ci other = right; 8677db96d56Sopenharmony_ci } 8687db96d56Sopenharmony_ci else { 8697db96d56Sopenharmony_ci type = Py_TYPE(right); 8707db96d56Sopenharmony_ci other = left; 8717db96d56Sopenharmony_ci } 8727db96d56Sopenharmony_ci if (!PyDict_Check(other)) { 8737db96d56Sopenharmony_ci Py_RETURN_NOTIMPLEMENTED; 8747db96d56Sopenharmony_ci } 8757db96d56Sopenharmony_ci PyObject *new = PyObject_CallOneArg((PyObject*)type, left); 8767db96d56Sopenharmony_ci if (!new) { 8777db96d56Sopenharmony_ci return NULL; 8787db96d56Sopenharmony_ci } 8797db96d56Sopenharmony_ci if (mutablemapping_update_arg(new, right) < 0) { 8807db96d56Sopenharmony_ci Py_DECREF(new); 8817db96d56Sopenharmony_ci return NULL; 8827db96d56Sopenharmony_ci } 8837db96d56Sopenharmony_ci return new; 8847db96d56Sopenharmony_ci} 8857db96d56Sopenharmony_ci 8867db96d56Sopenharmony_cistatic PyObject * 8877db96d56Sopenharmony_ciodict_inplace_or(PyObject *self, PyObject *other) 8887db96d56Sopenharmony_ci{ 8897db96d56Sopenharmony_ci if (mutablemapping_update_arg(self, other) < 0) { 8907db96d56Sopenharmony_ci return NULL; 8917db96d56Sopenharmony_ci } 8927db96d56Sopenharmony_ci Py_INCREF(self); 8937db96d56Sopenharmony_ci return self; 8947db96d56Sopenharmony_ci} 8957db96d56Sopenharmony_ci 8967db96d56Sopenharmony_ci/* tp_as_number */ 8977db96d56Sopenharmony_ci 8987db96d56Sopenharmony_cistatic PyNumberMethods odict_as_number = { 8997db96d56Sopenharmony_ci .nb_or = odict_or, 9007db96d56Sopenharmony_ci .nb_inplace_or = odict_inplace_or, 9017db96d56Sopenharmony_ci}; 9027db96d56Sopenharmony_ci 9037db96d56Sopenharmony_ci 9047db96d56Sopenharmony_ci/* ---------------------------------------------- 9057db96d56Sopenharmony_ci * OrderedDict methods 9067db96d56Sopenharmony_ci */ 9077db96d56Sopenharmony_ci 9087db96d56Sopenharmony_ci/* fromkeys() */ 9097db96d56Sopenharmony_ci 9107db96d56Sopenharmony_ci/*[clinic input] 9117db96d56Sopenharmony_ci@classmethod 9127db96d56Sopenharmony_ciOrderedDict.fromkeys 9137db96d56Sopenharmony_ci 9147db96d56Sopenharmony_ci iterable as seq: object 9157db96d56Sopenharmony_ci value: object = None 9167db96d56Sopenharmony_ci 9177db96d56Sopenharmony_ciCreate a new ordered dictionary with keys from iterable and values set to value. 9187db96d56Sopenharmony_ci[clinic start generated code]*/ 9197db96d56Sopenharmony_ci 9207db96d56Sopenharmony_cistatic PyObject * 9217db96d56Sopenharmony_ciOrderedDict_fromkeys_impl(PyTypeObject *type, PyObject *seq, PyObject *value) 9227db96d56Sopenharmony_ci/*[clinic end generated code: output=c10390d452d78d6d input=1a0476c229c597b3]*/ 9237db96d56Sopenharmony_ci{ 9247db96d56Sopenharmony_ci return _PyDict_FromKeys((PyObject *)type, seq, value); 9257db96d56Sopenharmony_ci} 9267db96d56Sopenharmony_ci 9277db96d56Sopenharmony_ci/* __sizeof__() */ 9287db96d56Sopenharmony_ci 9297db96d56Sopenharmony_ci/* OrderedDict.__sizeof__() does not have a docstring. */ 9307db96d56Sopenharmony_ciPyDoc_STRVAR(odict_sizeof__doc__, ""); 9317db96d56Sopenharmony_ci 9327db96d56Sopenharmony_cistatic PyObject * 9337db96d56Sopenharmony_ciodict_sizeof(PyODictObject *od, PyObject *Py_UNUSED(ignored)) 9347db96d56Sopenharmony_ci{ 9357db96d56Sopenharmony_ci Py_ssize_t res = _PyDict_SizeOf((PyDictObject *)od); 9367db96d56Sopenharmony_ci res += sizeof(_ODictNode *) * od->od_fast_nodes_size; /* od_fast_nodes */ 9377db96d56Sopenharmony_ci if (!_odict_EMPTY(od)) { 9387db96d56Sopenharmony_ci res += sizeof(_ODictNode) * PyODict_SIZE(od); /* linked-list */ 9397db96d56Sopenharmony_ci } 9407db96d56Sopenharmony_ci return PyLong_FromSsize_t(res); 9417db96d56Sopenharmony_ci} 9427db96d56Sopenharmony_ci 9437db96d56Sopenharmony_ci/* __reduce__() */ 9447db96d56Sopenharmony_ci 9457db96d56Sopenharmony_ciPyDoc_STRVAR(odict_reduce__doc__, "Return state information for pickling"); 9467db96d56Sopenharmony_ci 9477db96d56Sopenharmony_cistatic PyObject * 9487db96d56Sopenharmony_ciodict_reduce(register PyODictObject *od, PyObject *Py_UNUSED(ignored)) 9497db96d56Sopenharmony_ci{ 9507db96d56Sopenharmony_ci PyObject *state, *result = NULL; 9517db96d56Sopenharmony_ci PyObject *items_iter, *items, *args = NULL; 9527db96d56Sopenharmony_ci 9537db96d56Sopenharmony_ci /* capture any instance state */ 9547db96d56Sopenharmony_ci state = _PyObject_GetState((PyObject *)od); 9557db96d56Sopenharmony_ci if (state == NULL) 9567db96d56Sopenharmony_ci goto Done; 9577db96d56Sopenharmony_ci 9587db96d56Sopenharmony_ci /* build the result */ 9597db96d56Sopenharmony_ci args = PyTuple_New(0); 9607db96d56Sopenharmony_ci if (args == NULL) 9617db96d56Sopenharmony_ci goto Done; 9627db96d56Sopenharmony_ci 9637db96d56Sopenharmony_ci items = PyObject_CallMethodNoArgs((PyObject *)od, &_Py_ID(items)); 9647db96d56Sopenharmony_ci if (items == NULL) 9657db96d56Sopenharmony_ci goto Done; 9667db96d56Sopenharmony_ci 9677db96d56Sopenharmony_ci items_iter = PyObject_GetIter(items); 9687db96d56Sopenharmony_ci Py_DECREF(items); 9697db96d56Sopenharmony_ci if (items_iter == NULL) 9707db96d56Sopenharmony_ci goto Done; 9717db96d56Sopenharmony_ci 9727db96d56Sopenharmony_ci result = PyTuple_Pack(5, Py_TYPE(od), args, state, Py_None, items_iter); 9737db96d56Sopenharmony_ci Py_DECREF(items_iter); 9747db96d56Sopenharmony_ci 9757db96d56Sopenharmony_ciDone: 9767db96d56Sopenharmony_ci Py_XDECREF(state); 9777db96d56Sopenharmony_ci Py_XDECREF(args); 9787db96d56Sopenharmony_ci 9797db96d56Sopenharmony_ci return result; 9807db96d56Sopenharmony_ci} 9817db96d56Sopenharmony_ci 9827db96d56Sopenharmony_ci/* setdefault(): Skips __missing__() calls. */ 9837db96d56Sopenharmony_ci 9847db96d56Sopenharmony_ci 9857db96d56Sopenharmony_ci/*[clinic input] 9867db96d56Sopenharmony_ciOrderedDict.setdefault 9877db96d56Sopenharmony_ci 9887db96d56Sopenharmony_ci key: object 9897db96d56Sopenharmony_ci default: object = None 9907db96d56Sopenharmony_ci 9917db96d56Sopenharmony_ciInsert key with a value of default if key is not in the dictionary. 9927db96d56Sopenharmony_ci 9937db96d56Sopenharmony_ciReturn the value for key if key is in the dictionary, else default. 9947db96d56Sopenharmony_ci[clinic start generated code]*/ 9957db96d56Sopenharmony_ci 9967db96d56Sopenharmony_cistatic PyObject * 9977db96d56Sopenharmony_ciOrderedDict_setdefault_impl(PyODictObject *self, PyObject *key, 9987db96d56Sopenharmony_ci PyObject *default_value) 9997db96d56Sopenharmony_ci/*[clinic end generated code: output=97537cb7c28464b6 input=38e098381c1efbc6]*/ 10007db96d56Sopenharmony_ci{ 10017db96d56Sopenharmony_ci PyObject *result = NULL; 10027db96d56Sopenharmony_ci 10037db96d56Sopenharmony_ci if (PyODict_CheckExact(self)) { 10047db96d56Sopenharmony_ci result = PyODict_GetItemWithError(self, key); /* borrowed */ 10057db96d56Sopenharmony_ci if (result == NULL) { 10067db96d56Sopenharmony_ci if (PyErr_Occurred()) 10077db96d56Sopenharmony_ci return NULL; 10087db96d56Sopenharmony_ci assert(_odict_find_node(self, key) == NULL); 10097db96d56Sopenharmony_ci if (PyODict_SetItem((PyObject *)self, key, default_value) >= 0) { 10107db96d56Sopenharmony_ci result = default_value; 10117db96d56Sopenharmony_ci Py_INCREF(result); 10127db96d56Sopenharmony_ci } 10137db96d56Sopenharmony_ci } 10147db96d56Sopenharmony_ci else { 10157db96d56Sopenharmony_ci Py_INCREF(result); 10167db96d56Sopenharmony_ci } 10177db96d56Sopenharmony_ci } 10187db96d56Sopenharmony_ci else { 10197db96d56Sopenharmony_ci int exists = PySequence_Contains((PyObject *)self, key); 10207db96d56Sopenharmony_ci if (exists < 0) { 10217db96d56Sopenharmony_ci return NULL; 10227db96d56Sopenharmony_ci } 10237db96d56Sopenharmony_ci else if (exists) { 10247db96d56Sopenharmony_ci result = PyObject_GetItem((PyObject *)self, key); 10257db96d56Sopenharmony_ci } 10267db96d56Sopenharmony_ci else if (PyObject_SetItem((PyObject *)self, key, default_value) >= 0) { 10277db96d56Sopenharmony_ci result = default_value; 10287db96d56Sopenharmony_ci Py_INCREF(result); 10297db96d56Sopenharmony_ci } 10307db96d56Sopenharmony_ci } 10317db96d56Sopenharmony_ci 10327db96d56Sopenharmony_ci return result; 10337db96d56Sopenharmony_ci} 10347db96d56Sopenharmony_ci 10357db96d56Sopenharmony_ci/* pop() */ 10367db96d56Sopenharmony_ci 10377db96d56Sopenharmony_cistatic PyObject * 10387db96d56Sopenharmony_ci_odict_popkey_hash(PyObject *od, PyObject *key, PyObject *failobj, 10397db96d56Sopenharmony_ci Py_hash_t hash) 10407db96d56Sopenharmony_ci{ 10417db96d56Sopenharmony_ci PyObject *value = NULL; 10427db96d56Sopenharmony_ci 10437db96d56Sopenharmony_ci _ODictNode *node = _odict_find_node_hash((PyODictObject *)od, key, hash); 10447db96d56Sopenharmony_ci if (node != NULL) { 10457db96d56Sopenharmony_ci /* Pop the node first to avoid a possible dict resize (due to 10467db96d56Sopenharmony_ci eval loop reentrancy) and complications due to hash collision 10477db96d56Sopenharmony_ci resolution. */ 10487db96d56Sopenharmony_ci int res = _odict_clear_node((PyODictObject *)od, node, key, hash); 10497db96d56Sopenharmony_ci if (res < 0) { 10507db96d56Sopenharmony_ci return NULL; 10517db96d56Sopenharmony_ci } 10527db96d56Sopenharmony_ci /* Now delete the value from the dict. */ 10537db96d56Sopenharmony_ci value = _PyDict_Pop_KnownHash(od, key, hash, failobj); 10547db96d56Sopenharmony_ci } 10557db96d56Sopenharmony_ci else if (value == NULL && !PyErr_Occurred()) { 10567db96d56Sopenharmony_ci /* Apply the fallback value, if necessary. */ 10577db96d56Sopenharmony_ci if (failobj) { 10587db96d56Sopenharmony_ci value = failobj; 10597db96d56Sopenharmony_ci Py_INCREF(failobj); 10607db96d56Sopenharmony_ci } 10617db96d56Sopenharmony_ci else { 10627db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 10637db96d56Sopenharmony_ci } 10647db96d56Sopenharmony_ci } 10657db96d56Sopenharmony_ci 10667db96d56Sopenharmony_ci return value; 10677db96d56Sopenharmony_ci} 10687db96d56Sopenharmony_ci 10697db96d56Sopenharmony_ci/* Skips __missing__() calls. */ 10707db96d56Sopenharmony_ci/*[clinic input] 10717db96d56Sopenharmony_ciOrderedDict.pop 10727db96d56Sopenharmony_ci 10737db96d56Sopenharmony_ci key: object 10747db96d56Sopenharmony_ci default: object = NULL 10757db96d56Sopenharmony_ci 10767db96d56Sopenharmony_ciod.pop(key[,default]) -> v, remove specified key and return the corresponding value. 10777db96d56Sopenharmony_ci 10787db96d56Sopenharmony_ciIf the key is not found, return the default if given; otherwise, 10797db96d56Sopenharmony_ciraise a KeyError. 10807db96d56Sopenharmony_ci[clinic start generated code]*/ 10817db96d56Sopenharmony_ci 10827db96d56Sopenharmony_cistatic PyObject * 10837db96d56Sopenharmony_ciOrderedDict_pop_impl(PyODictObject *self, PyObject *key, 10847db96d56Sopenharmony_ci PyObject *default_value) 10857db96d56Sopenharmony_ci/*[clinic end generated code: output=7a6447d104e7494b input=7efe36601007dff7]*/ 10867db96d56Sopenharmony_ci{ 10877db96d56Sopenharmony_ci Py_hash_t hash = PyObject_Hash(key); 10887db96d56Sopenharmony_ci if (hash == -1) 10897db96d56Sopenharmony_ci return NULL; 10907db96d56Sopenharmony_ci return _odict_popkey_hash((PyObject *)self, key, default_value, hash); 10917db96d56Sopenharmony_ci} 10927db96d56Sopenharmony_ci 10937db96d56Sopenharmony_ci 10947db96d56Sopenharmony_ci/* popitem() */ 10957db96d56Sopenharmony_ci 10967db96d56Sopenharmony_ci/*[clinic input] 10977db96d56Sopenharmony_ciOrderedDict.popitem 10987db96d56Sopenharmony_ci 10997db96d56Sopenharmony_ci last: bool = True 11007db96d56Sopenharmony_ci 11017db96d56Sopenharmony_ciRemove and return a (key, value) pair from the dictionary. 11027db96d56Sopenharmony_ci 11037db96d56Sopenharmony_ciPairs are returned in LIFO order if last is true or FIFO order if false. 11047db96d56Sopenharmony_ci[clinic start generated code]*/ 11057db96d56Sopenharmony_ci 11067db96d56Sopenharmony_cistatic PyObject * 11077db96d56Sopenharmony_ciOrderedDict_popitem_impl(PyODictObject *self, int last) 11087db96d56Sopenharmony_ci/*[clinic end generated code: output=98e7d986690d49eb input=d992ac5ee8305e1a]*/ 11097db96d56Sopenharmony_ci{ 11107db96d56Sopenharmony_ci PyObject *key, *value, *item = NULL; 11117db96d56Sopenharmony_ci _ODictNode *node; 11127db96d56Sopenharmony_ci 11137db96d56Sopenharmony_ci /* pull the item */ 11147db96d56Sopenharmony_ci 11157db96d56Sopenharmony_ci if (_odict_EMPTY(self)) { 11167db96d56Sopenharmony_ci PyErr_SetString(PyExc_KeyError, "dictionary is empty"); 11177db96d56Sopenharmony_ci return NULL; 11187db96d56Sopenharmony_ci } 11197db96d56Sopenharmony_ci 11207db96d56Sopenharmony_ci node = last ? _odict_LAST(self) : _odict_FIRST(self); 11217db96d56Sopenharmony_ci key = _odictnode_KEY(node); 11227db96d56Sopenharmony_ci Py_INCREF(key); 11237db96d56Sopenharmony_ci value = _odict_popkey_hash((PyObject *)self, key, NULL, _odictnode_HASH(node)); 11247db96d56Sopenharmony_ci if (value == NULL) 11257db96d56Sopenharmony_ci return NULL; 11267db96d56Sopenharmony_ci item = PyTuple_Pack(2, key, value); 11277db96d56Sopenharmony_ci Py_DECREF(key); 11287db96d56Sopenharmony_ci Py_DECREF(value); 11297db96d56Sopenharmony_ci return item; 11307db96d56Sopenharmony_ci} 11317db96d56Sopenharmony_ci 11327db96d56Sopenharmony_ci/* keys() */ 11337db96d56Sopenharmony_ci 11347db96d56Sopenharmony_ci/* MutableMapping.keys() does not have a docstring. */ 11357db96d56Sopenharmony_ciPyDoc_STRVAR(odict_keys__doc__, ""); 11367db96d56Sopenharmony_ci 11377db96d56Sopenharmony_cistatic PyObject * odictkeys_new(PyObject *od, PyObject *Py_UNUSED(ignored)); /* forward */ 11387db96d56Sopenharmony_ci 11397db96d56Sopenharmony_ci/* values() */ 11407db96d56Sopenharmony_ci 11417db96d56Sopenharmony_ci/* MutableMapping.values() does not have a docstring. */ 11427db96d56Sopenharmony_ciPyDoc_STRVAR(odict_values__doc__, ""); 11437db96d56Sopenharmony_ci 11447db96d56Sopenharmony_cistatic PyObject * odictvalues_new(PyObject *od, PyObject *Py_UNUSED(ignored)); /* forward */ 11457db96d56Sopenharmony_ci 11467db96d56Sopenharmony_ci/* items() */ 11477db96d56Sopenharmony_ci 11487db96d56Sopenharmony_ci/* MutableMapping.items() does not have a docstring. */ 11497db96d56Sopenharmony_ciPyDoc_STRVAR(odict_items__doc__, ""); 11507db96d56Sopenharmony_ci 11517db96d56Sopenharmony_cistatic PyObject * odictitems_new(PyObject *od, PyObject *Py_UNUSED(ignored)); /* forward */ 11527db96d56Sopenharmony_ci 11537db96d56Sopenharmony_ci/* update() */ 11547db96d56Sopenharmony_ci 11557db96d56Sopenharmony_ci/* MutableMapping.update() does not have a docstring. */ 11567db96d56Sopenharmony_ciPyDoc_STRVAR(odict_update__doc__, ""); 11577db96d56Sopenharmony_ci 11587db96d56Sopenharmony_ci/* forward */ 11597db96d56Sopenharmony_cistatic PyObject * mutablemapping_update(PyObject *, PyObject *, PyObject *); 11607db96d56Sopenharmony_ci 11617db96d56Sopenharmony_ci#define odict_update mutablemapping_update 11627db96d56Sopenharmony_ci 11637db96d56Sopenharmony_ci/* clear() */ 11647db96d56Sopenharmony_ci 11657db96d56Sopenharmony_ciPyDoc_STRVAR(odict_clear__doc__, 11667db96d56Sopenharmony_ci "od.clear() -> None. Remove all items from od."); 11677db96d56Sopenharmony_ci 11687db96d56Sopenharmony_cistatic PyObject * 11697db96d56Sopenharmony_ciodict_clear(register PyODictObject *od, PyObject *Py_UNUSED(ignored)) 11707db96d56Sopenharmony_ci{ 11717db96d56Sopenharmony_ci PyDict_Clear((PyObject *)od); 11727db96d56Sopenharmony_ci _odict_clear_nodes(od); 11737db96d56Sopenharmony_ci Py_RETURN_NONE; 11747db96d56Sopenharmony_ci} 11757db96d56Sopenharmony_ci 11767db96d56Sopenharmony_ci/* copy() */ 11777db96d56Sopenharmony_ci 11787db96d56Sopenharmony_ci/* forward */ 11797db96d56Sopenharmony_cistatic int _PyODict_SetItem_KnownHash(PyObject *, PyObject *, PyObject *, 11807db96d56Sopenharmony_ci Py_hash_t); 11817db96d56Sopenharmony_ci 11827db96d56Sopenharmony_ciPyDoc_STRVAR(odict_copy__doc__, "od.copy() -> a shallow copy of od"); 11837db96d56Sopenharmony_ci 11847db96d56Sopenharmony_cistatic PyObject * 11857db96d56Sopenharmony_ciodict_copy(register PyODictObject *od, PyObject *Py_UNUSED(ignored)) 11867db96d56Sopenharmony_ci{ 11877db96d56Sopenharmony_ci _ODictNode *node; 11887db96d56Sopenharmony_ci PyObject *od_copy; 11897db96d56Sopenharmony_ci 11907db96d56Sopenharmony_ci if (PyODict_CheckExact(od)) 11917db96d56Sopenharmony_ci od_copy = PyODict_New(); 11927db96d56Sopenharmony_ci else 11937db96d56Sopenharmony_ci od_copy = _PyObject_CallNoArgs((PyObject *)Py_TYPE(od)); 11947db96d56Sopenharmony_ci if (od_copy == NULL) 11957db96d56Sopenharmony_ci return NULL; 11967db96d56Sopenharmony_ci 11977db96d56Sopenharmony_ci if (PyODict_CheckExact(od)) { 11987db96d56Sopenharmony_ci _odict_FOREACH(od, node) { 11997db96d56Sopenharmony_ci PyObject *key = _odictnode_KEY(node); 12007db96d56Sopenharmony_ci PyObject *value = _odictnode_VALUE(node, od); 12017db96d56Sopenharmony_ci if (value == NULL) { 12027db96d56Sopenharmony_ci if (!PyErr_Occurred()) 12037db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 12047db96d56Sopenharmony_ci goto fail; 12057db96d56Sopenharmony_ci } 12067db96d56Sopenharmony_ci if (_PyODict_SetItem_KnownHash((PyObject *)od_copy, key, value, 12077db96d56Sopenharmony_ci _odictnode_HASH(node)) != 0) 12087db96d56Sopenharmony_ci goto fail; 12097db96d56Sopenharmony_ci } 12107db96d56Sopenharmony_ci } 12117db96d56Sopenharmony_ci else { 12127db96d56Sopenharmony_ci _odict_FOREACH(od, node) { 12137db96d56Sopenharmony_ci int res; 12147db96d56Sopenharmony_ci PyObject *value = PyObject_GetItem((PyObject *)od, 12157db96d56Sopenharmony_ci _odictnode_KEY(node)); 12167db96d56Sopenharmony_ci if (value == NULL) 12177db96d56Sopenharmony_ci goto fail; 12187db96d56Sopenharmony_ci res = PyObject_SetItem((PyObject *)od_copy, 12197db96d56Sopenharmony_ci _odictnode_KEY(node), value); 12207db96d56Sopenharmony_ci Py_DECREF(value); 12217db96d56Sopenharmony_ci if (res != 0) 12227db96d56Sopenharmony_ci goto fail; 12237db96d56Sopenharmony_ci } 12247db96d56Sopenharmony_ci } 12257db96d56Sopenharmony_ci return od_copy; 12267db96d56Sopenharmony_ci 12277db96d56Sopenharmony_cifail: 12287db96d56Sopenharmony_ci Py_DECREF(od_copy); 12297db96d56Sopenharmony_ci return NULL; 12307db96d56Sopenharmony_ci} 12317db96d56Sopenharmony_ci 12327db96d56Sopenharmony_ci/* __reversed__() */ 12337db96d56Sopenharmony_ci 12347db96d56Sopenharmony_ciPyDoc_STRVAR(odict_reversed__doc__, "od.__reversed__() <==> reversed(od)"); 12357db96d56Sopenharmony_ci 12367db96d56Sopenharmony_ci#define _odict_ITER_REVERSED 1 12377db96d56Sopenharmony_ci#define _odict_ITER_KEYS 2 12387db96d56Sopenharmony_ci#define _odict_ITER_VALUES 4 12397db96d56Sopenharmony_ci#define _odict_ITER_ITEMS (_odict_ITER_KEYS|_odict_ITER_VALUES) 12407db96d56Sopenharmony_ci 12417db96d56Sopenharmony_ci/* forward */ 12427db96d56Sopenharmony_cistatic PyObject * odictiter_new(PyODictObject *, int); 12437db96d56Sopenharmony_ci 12447db96d56Sopenharmony_cistatic PyObject * 12457db96d56Sopenharmony_ciodict_reversed(PyODictObject *od, PyObject *Py_UNUSED(ignored)) 12467db96d56Sopenharmony_ci{ 12477db96d56Sopenharmony_ci return odictiter_new(od, _odict_ITER_KEYS|_odict_ITER_REVERSED); 12487db96d56Sopenharmony_ci} 12497db96d56Sopenharmony_ci 12507db96d56Sopenharmony_ci 12517db96d56Sopenharmony_ci/* move_to_end() */ 12527db96d56Sopenharmony_ci 12537db96d56Sopenharmony_ci/*[clinic input] 12547db96d56Sopenharmony_ciOrderedDict.move_to_end 12557db96d56Sopenharmony_ci 12567db96d56Sopenharmony_ci key: object 12577db96d56Sopenharmony_ci last: bool = True 12587db96d56Sopenharmony_ci 12597db96d56Sopenharmony_ciMove an existing element to the end (or beginning if last is false). 12607db96d56Sopenharmony_ci 12617db96d56Sopenharmony_ciRaise KeyError if the element does not exist. 12627db96d56Sopenharmony_ci[clinic start generated code]*/ 12637db96d56Sopenharmony_ci 12647db96d56Sopenharmony_cistatic PyObject * 12657db96d56Sopenharmony_ciOrderedDict_move_to_end_impl(PyODictObject *self, PyObject *key, int last) 12667db96d56Sopenharmony_ci/*[clinic end generated code: output=fafa4c5cc9b92f20 input=d6ceff7132a2fcd7]*/ 12677db96d56Sopenharmony_ci{ 12687db96d56Sopenharmony_ci _ODictNode *node; 12697db96d56Sopenharmony_ci 12707db96d56Sopenharmony_ci if (_odict_EMPTY(self)) { 12717db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 12727db96d56Sopenharmony_ci return NULL; 12737db96d56Sopenharmony_ci } 12747db96d56Sopenharmony_ci node = last ? _odict_LAST(self) : _odict_FIRST(self); 12757db96d56Sopenharmony_ci if (key != _odictnode_KEY(node)) { 12767db96d56Sopenharmony_ci node = _odict_find_node(self, key); 12777db96d56Sopenharmony_ci if (node == NULL) { 12787db96d56Sopenharmony_ci if (!PyErr_Occurred()) 12797db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 12807db96d56Sopenharmony_ci return NULL; 12817db96d56Sopenharmony_ci } 12827db96d56Sopenharmony_ci if (last) { 12837db96d56Sopenharmony_ci /* Only move if not already the last one. */ 12847db96d56Sopenharmony_ci if (node != _odict_LAST(self)) { 12857db96d56Sopenharmony_ci _odict_remove_node(self, node); 12867db96d56Sopenharmony_ci _odict_add_tail(self, node); 12877db96d56Sopenharmony_ci } 12887db96d56Sopenharmony_ci } 12897db96d56Sopenharmony_ci else { 12907db96d56Sopenharmony_ci /* Only move if not already the first one. */ 12917db96d56Sopenharmony_ci if (node != _odict_FIRST(self)) { 12927db96d56Sopenharmony_ci _odict_remove_node(self, node); 12937db96d56Sopenharmony_ci _odict_add_head(self, node); 12947db96d56Sopenharmony_ci } 12957db96d56Sopenharmony_ci } 12967db96d56Sopenharmony_ci } 12977db96d56Sopenharmony_ci Py_RETURN_NONE; 12987db96d56Sopenharmony_ci} 12997db96d56Sopenharmony_ci 13007db96d56Sopenharmony_ci 13017db96d56Sopenharmony_ci/* tp_methods */ 13027db96d56Sopenharmony_ci 13037db96d56Sopenharmony_cistatic PyMethodDef odict_methods[] = { 13047db96d56Sopenharmony_ci 13057db96d56Sopenharmony_ci /* overridden dict methods */ 13067db96d56Sopenharmony_ci ORDEREDDICT_FROMKEYS_METHODDEF 13077db96d56Sopenharmony_ci {"__sizeof__", (PyCFunction)odict_sizeof, METH_NOARGS, 13087db96d56Sopenharmony_ci odict_sizeof__doc__}, 13097db96d56Sopenharmony_ci {"__reduce__", (PyCFunction)odict_reduce, METH_NOARGS, 13107db96d56Sopenharmony_ci odict_reduce__doc__}, 13117db96d56Sopenharmony_ci ORDEREDDICT_SETDEFAULT_METHODDEF 13127db96d56Sopenharmony_ci ORDEREDDICT_POP_METHODDEF 13137db96d56Sopenharmony_ci ORDEREDDICT_POPITEM_METHODDEF 13147db96d56Sopenharmony_ci {"keys", odictkeys_new, METH_NOARGS, 13157db96d56Sopenharmony_ci odict_keys__doc__}, 13167db96d56Sopenharmony_ci {"values", odictvalues_new, METH_NOARGS, 13177db96d56Sopenharmony_ci odict_values__doc__}, 13187db96d56Sopenharmony_ci {"items", odictitems_new, METH_NOARGS, 13197db96d56Sopenharmony_ci odict_items__doc__}, 13207db96d56Sopenharmony_ci {"update", _PyCFunction_CAST(odict_update), METH_VARARGS | METH_KEYWORDS, 13217db96d56Sopenharmony_ci odict_update__doc__}, 13227db96d56Sopenharmony_ci {"clear", (PyCFunction)odict_clear, METH_NOARGS, 13237db96d56Sopenharmony_ci odict_clear__doc__}, 13247db96d56Sopenharmony_ci {"copy", (PyCFunction)odict_copy, METH_NOARGS, 13257db96d56Sopenharmony_ci odict_copy__doc__}, 13267db96d56Sopenharmony_ci 13277db96d56Sopenharmony_ci /* new methods */ 13287db96d56Sopenharmony_ci {"__reversed__", (PyCFunction)odict_reversed, METH_NOARGS, 13297db96d56Sopenharmony_ci odict_reversed__doc__}, 13307db96d56Sopenharmony_ci ORDEREDDICT_MOVE_TO_END_METHODDEF 13317db96d56Sopenharmony_ci 13327db96d56Sopenharmony_ci {NULL, NULL} /* sentinel */ 13337db96d56Sopenharmony_ci}; 13347db96d56Sopenharmony_ci 13357db96d56Sopenharmony_ci 13367db96d56Sopenharmony_ci/* ---------------------------------------------- 13377db96d56Sopenharmony_ci * OrderedDict members 13387db96d56Sopenharmony_ci */ 13397db96d56Sopenharmony_ci 13407db96d56Sopenharmony_ci/* tp_getset */ 13417db96d56Sopenharmony_ci 13427db96d56Sopenharmony_cistatic PyGetSetDef odict_getset[] = { 13437db96d56Sopenharmony_ci {"__dict__", PyObject_GenericGetDict, PyObject_GenericSetDict}, 13447db96d56Sopenharmony_ci {NULL} 13457db96d56Sopenharmony_ci}; 13467db96d56Sopenharmony_ci 13477db96d56Sopenharmony_ci/* ---------------------------------------------- 13487db96d56Sopenharmony_ci * OrderedDict type slot methods 13497db96d56Sopenharmony_ci */ 13507db96d56Sopenharmony_ci 13517db96d56Sopenharmony_ci/* tp_dealloc */ 13527db96d56Sopenharmony_ci 13537db96d56Sopenharmony_cistatic void 13547db96d56Sopenharmony_ciodict_dealloc(PyODictObject *self) 13557db96d56Sopenharmony_ci{ 13567db96d56Sopenharmony_ci PyObject_GC_UnTrack(self); 13577db96d56Sopenharmony_ci Py_TRASHCAN_BEGIN(self, odict_dealloc) 13587db96d56Sopenharmony_ci 13597db96d56Sopenharmony_ci Py_XDECREF(self->od_inst_dict); 13607db96d56Sopenharmony_ci if (self->od_weakreflist != NULL) 13617db96d56Sopenharmony_ci PyObject_ClearWeakRefs((PyObject *)self); 13627db96d56Sopenharmony_ci 13637db96d56Sopenharmony_ci _odict_clear_nodes(self); 13647db96d56Sopenharmony_ci PyDict_Type.tp_dealloc((PyObject *)self); 13657db96d56Sopenharmony_ci 13667db96d56Sopenharmony_ci Py_TRASHCAN_END 13677db96d56Sopenharmony_ci} 13687db96d56Sopenharmony_ci 13697db96d56Sopenharmony_ci/* tp_repr */ 13707db96d56Sopenharmony_ci 13717db96d56Sopenharmony_cistatic PyObject * 13727db96d56Sopenharmony_ciodict_repr(PyODictObject *self) 13737db96d56Sopenharmony_ci{ 13747db96d56Sopenharmony_ci int i; 13757db96d56Sopenharmony_ci PyObject *pieces = NULL, *result = NULL; 13767db96d56Sopenharmony_ci 13777db96d56Sopenharmony_ci if (PyODict_SIZE(self) == 0) 13787db96d56Sopenharmony_ci return PyUnicode_FromFormat("%s()", _PyType_Name(Py_TYPE(self))); 13797db96d56Sopenharmony_ci 13807db96d56Sopenharmony_ci i = Py_ReprEnter((PyObject *)self); 13817db96d56Sopenharmony_ci if (i != 0) { 13827db96d56Sopenharmony_ci return i > 0 ? PyUnicode_FromString("...") : NULL; 13837db96d56Sopenharmony_ci } 13847db96d56Sopenharmony_ci 13857db96d56Sopenharmony_ci if (PyODict_CheckExact(self)) { 13867db96d56Sopenharmony_ci Py_ssize_t count = 0; 13877db96d56Sopenharmony_ci _ODictNode *node; 13887db96d56Sopenharmony_ci pieces = PyList_New(PyODict_SIZE(self)); 13897db96d56Sopenharmony_ci if (pieces == NULL) 13907db96d56Sopenharmony_ci goto Done; 13917db96d56Sopenharmony_ci 13927db96d56Sopenharmony_ci _odict_FOREACH(self, node) { 13937db96d56Sopenharmony_ci PyObject *pair; 13947db96d56Sopenharmony_ci PyObject *key = _odictnode_KEY(node); 13957db96d56Sopenharmony_ci PyObject *value = _odictnode_VALUE(node, self); 13967db96d56Sopenharmony_ci if (value == NULL) { 13977db96d56Sopenharmony_ci if (!PyErr_Occurred()) 13987db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 13997db96d56Sopenharmony_ci goto Done; 14007db96d56Sopenharmony_ci } 14017db96d56Sopenharmony_ci pair = PyTuple_Pack(2, key, value); 14027db96d56Sopenharmony_ci if (pair == NULL) 14037db96d56Sopenharmony_ci goto Done; 14047db96d56Sopenharmony_ci 14057db96d56Sopenharmony_ci if (count < PyList_GET_SIZE(pieces)) 14067db96d56Sopenharmony_ci PyList_SET_ITEM(pieces, count, pair); /* steals reference */ 14077db96d56Sopenharmony_ci else { 14087db96d56Sopenharmony_ci if (PyList_Append(pieces, pair) < 0) { 14097db96d56Sopenharmony_ci Py_DECREF(pair); 14107db96d56Sopenharmony_ci goto Done; 14117db96d56Sopenharmony_ci } 14127db96d56Sopenharmony_ci Py_DECREF(pair); 14137db96d56Sopenharmony_ci } 14147db96d56Sopenharmony_ci count++; 14157db96d56Sopenharmony_ci } 14167db96d56Sopenharmony_ci if (count < PyList_GET_SIZE(pieces)) { 14177db96d56Sopenharmony_ci Py_SET_SIZE(pieces, count); 14187db96d56Sopenharmony_ci } 14197db96d56Sopenharmony_ci } 14207db96d56Sopenharmony_ci else { 14217db96d56Sopenharmony_ci PyObject *items = PyObject_CallMethodNoArgs( 14227db96d56Sopenharmony_ci (PyObject *)self, &_Py_ID(items)); 14237db96d56Sopenharmony_ci if (items == NULL) 14247db96d56Sopenharmony_ci goto Done; 14257db96d56Sopenharmony_ci pieces = PySequence_List(items); 14267db96d56Sopenharmony_ci Py_DECREF(items); 14277db96d56Sopenharmony_ci if (pieces == NULL) 14287db96d56Sopenharmony_ci goto Done; 14297db96d56Sopenharmony_ci } 14307db96d56Sopenharmony_ci 14317db96d56Sopenharmony_ci result = PyUnicode_FromFormat("%s(%R)", 14327db96d56Sopenharmony_ci _PyType_Name(Py_TYPE(self)), pieces); 14337db96d56Sopenharmony_ci 14347db96d56Sopenharmony_ciDone: 14357db96d56Sopenharmony_ci Py_XDECREF(pieces); 14367db96d56Sopenharmony_ci Py_ReprLeave((PyObject *)self); 14377db96d56Sopenharmony_ci return result; 14387db96d56Sopenharmony_ci} 14397db96d56Sopenharmony_ci 14407db96d56Sopenharmony_ci/* tp_doc */ 14417db96d56Sopenharmony_ci 14427db96d56Sopenharmony_ciPyDoc_STRVAR(odict_doc, 14437db96d56Sopenharmony_ci "Dictionary that remembers insertion order"); 14447db96d56Sopenharmony_ci 14457db96d56Sopenharmony_ci/* tp_traverse */ 14467db96d56Sopenharmony_ci 14477db96d56Sopenharmony_cistatic int 14487db96d56Sopenharmony_ciodict_traverse(PyODictObject *od, visitproc visit, void *arg) 14497db96d56Sopenharmony_ci{ 14507db96d56Sopenharmony_ci _ODictNode *node; 14517db96d56Sopenharmony_ci 14527db96d56Sopenharmony_ci Py_VISIT(od->od_inst_dict); 14537db96d56Sopenharmony_ci _odict_FOREACH(od, node) { 14547db96d56Sopenharmony_ci Py_VISIT(_odictnode_KEY(node)); 14557db96d56Sopenharmony_ci } 14567db96d56Sopenharmony_ci return PyDict_Type.tp_traverse((PyObject *)od, visit, arg); 14577db96d56Sopenharmony_ci} 14587db96d56Sopenharmony_ci 14597db96d56Sopenharmony_ci/* tp_clear */ 14607db96d56Sopenharmony_ci 14617db96d56Sopenharmony_cistatic int 14627db96d56Sopenharmony_ciodict_tp_clear(PyODictObject *od) 14637db96d56Sopenharmony_ci{ 14647db96d56Sopenharmony_ci Py_CLEAR(od->od_inst_dict); 14657db96d56Sopenharmony_ci PyDict_Clear((PyObject *)od); 14667db96d56Sopenharmony_ci _odict_clear_nodes(od); 14677db96d56Sopenharmony_ci return 0; 14687db96d56Sopenharmony_ci} 14697db96d56Sopenharmony_ci 14707db96d56Sopenharmony_ci/* tp_richcompare */ 14717db96d56Sopenharmony_ci 14727db96d56Sopenharmony_cistatic PyObject * 14737db96d56Sopenharmony_ciodict_richcompare(PyObject *v, PyObject *w, int op) 14747db96d56Sopenharmony_ci{ 14757db96d56Sopenharmony_ci if (!PyODict_Check(v) || !PyDict_Check(w)) { 14767db96d56Sopenharmony_ci Py_RETURN_NOTIMPLEMENTED; 14777db96d56Sopenharmony_ci } 14787db96d56Sopenharmony_ci 14797db96d56Sopenharmony_ci if (op == Py_EQ || op == Py_NE) { 14807db96d56Sopenharmony_ci PyObject *res, *cmp; 14817db96d56Sopenharmony_ci int eq; 14827db96d56Sopenharmony_ci 14837db96d56Sopenharmony_ci cmp = PyDict_Type.tp_richcompare(v, w, op); 14847db96d56Sopenharmony_ci if (cmp == NULL) 14857db96d56Sopenharmony_ci return NULL; 14867db96d56Sopenharmony_ci if (!PyODict_Check(w)) 14877db96d56Sopenharmony_ci return cmp; 14887db96d56Sopenharmony_ci if (op == Py_EQ && cmp == Py_False) 14897db96d56Sopenharmony_ci return cmp; 14907db96d56Sopenharmony_ci if (op == Py_NE && cmp == Py_True) 14917db96d56Sopenharmony_ci return cmp; 14927db96d56Sopenharmony_ci Py_DECREF(cmp); 14937db96d56Sopenharmony_ci 14947db96d56Sopenharmony_ci /* Try comparing odict keys. */ 14957db96d56Sopenharmony_ci eq = _odict_keys_equal((PyODictObject *)v, (PyODictObject *)w); 14967db96d56Sopenharmony_ci if (eq < 0) 14977db96d56Sopenharmony_ci return NULL; 14987db96d56Sopenharmony_ci 14997db96d56Sopenharmony_ci res = (eq == (op == Py_EQ)) ? Py_True : Py_False; 15007db96d56Sopenharmony_ci Py_INCREF(res); 15017db96d56Sopenharmony_ci return res; 15027db96d56Sopenharmony_ci } else { 15037db96d56Sopenharmony_ci Py_RETURN_NOTIMPLEMENTED; 15047db96d56Sopenharmony_ci } 15057db96d56Sopenharmony_ci} 15067db96d56Sopenharmony_ci 15077db96d56Sopenharmony_ci/* tp_iter */ 15087db96d56Sopenharmony_ci 15097db96d56Sopenharmony_cistatic PyObject * 15107db96d56Sopenharmony_ciodict_iter(PyODictObject *od) 15117db96d56Sopenharmony_ci{ 15127db96d56Sopenharmony_ci return odictiter_new(od, _odict_ITER_KEYS); 15137db96d56Sopenharmony_ci} 15147db96d56Sopenharmony_ci 15157db96d56Sopenharmony_ci/* tp_init */ 15167db96d56Sopenharmony_ci 15177db96d56Sopenharmony_cistatic int 15187db96d56Sopenharmony_ciodict_init(PyObject *self, PyObject *args, PyObject *kwds) 15197db96d56Sopenharmony_ci{ 15207db96d56Sopenharmony_ci PyObject *res; 15217db96d56Sopenharmony_ci Py_ssize_t len = PyObject_Length(args); 15227db96d56Sopenharmony_ci 15237db96d56Sopenharmony_ci if (len == -1) 15247db96d56Sopenharmony_ci return -1; 15257db96d56Sopenharmony_ci if (len > 1) { 15267db96d56Sopenharmony_ci const char *msg = "expected at most 1 arguments, got %zd"; 15277db96d56Sopenharmony_ci PyErr_Format(PyExc_TypeError, msg, len); 15287db96d56Sopenharmony_ci return -1; 15297db96d56Sopenharmony_ci } 15307db96d56Sopenharmony_ci 15317db96d56Sopenharmony_ci /* __init__() triggering update() is just the way things are! */ 15327db96d56Sopenharmony_ci res = odict_update(self, args, kwds); 15337db96d56Sopenharmony_ci if (res == NULL) { 15347db96d56Sopenharmony_ci return -1; 15357db96d56Sopenharmony_ci } else { 15367db96d56Sopenharmony_ci Py_DECREF(res); 15377db96d56Sopenharmony_ci return 0; 15387db96d56Sopenharmony_ci } 15397db96d56Sopenharmony_ci} 15407db96d56Sopenharmony_ci 15417db96d56Sopenharmony_ci/* PyODict_Type */ 15427db96d56Sopenharmony_ci 15437db96d56Sopenharmony_ciPyTypeObject PyODict_Type = { 15447db96d56Sopenharmony_ci PyVarObject_HEAD_INIT(&PyType_Type, 0) 15457db96d56Sopenharmony_ci "collections.OrderedDict", /* tp_name */ 15467db96d56Sopenharmony_ci sizeof(PyODictObject), /* tp_basicsize */ 15477db96d56Sopenharmony_ci 0, /* tp_itemsize */ 15487db96d56Sopenharmony_ci (destructor)odict_dealloc, /* tp_dealloc */ 15497db96d56Sopenharmony_ci 0, /* tp_vectorcall_offset */ 15507db96d56Sopenharmony_ci 0, /* tp_getattr */ 15517db96d56Sopenharmony_ci 0, /* tp_setattr */ 15527db96d56Sopenharmony_ci 0, /* tp_as_async */ 15537db96d56Sopenharmony_ci (reprfunc)odict_repr, /* tp_repr */ 15547db96d56Sopenharmony_ci &odict_as_number, /* tp_as_number */ 15557db96d56Sopenharmony_ci 0, /* tp_as_sequence */ 15567db96d56Sopenharmony_ci &odict_as_mapping, /* tp_as_mapping */ 15577db96d56Sopenharmony_ci 0, /* tp_hash */ 15587db96d56Sopenharmony_ci 0, /* tp_call */ 15597db96d56Sopenharmony_ci 0, /* tp_str */ 15607db96d56Sopenharmony_ci 0, /* tp_getattro */ 15617db96d56Sopenharmony_ci 0, /* tp_setattro */ 15627db96d56Sopenharmony_ci 0, /* tp_as_buffer */ 15637db96d56Sopenharmony_ci Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE | Py_TPFLAGS_HAVE_GC,/* tp_flags */ 15647db96d56Sopenharmony_ci odict_doc, /* tp_doc */ 15657db96d56Sopenharmony_ci (traverseproc)odict_traverse, /* tp_traverse */ 15667db96d56Sopenharmony_ci (inquiry)odict_tp_clear, /* tp_clear */ 15677db96d56Sopenharmony_ci (richcmpfunc)odict_richcompare, /* tp_richcompare */ 15687db96d56Sopenharmony_ci offsetof(PyODictObject, od_weakreflist), /* tp_weaklistoffset */ 15697db96d56Sopenharmony_ci (getiterfunc)odict_iter, /* tp_iter */ 15707db96d56Sopenharmony_ci 0, /* tp_iternext */ 15717db96d56Sopenharmony_ci odict_methods, /* tp_methods */ 15727db96d56Sopenharmony_ci 0, /* tp_members */ 15737db96d56Sopenharmony_ci odict_getset, /* tp_getset */ 15747db96d56Sopenharmony_ci &PyDict_Type, /* tp_base */ 15757db96d56Sopenharmony_ci 0, /* tp_dict */ 15767db96d56Sopenharmony_ci 0, /* tp_descr_get */ 15777db96d56Sopenharmony_ci 0, /* tp_descr_set */ 15787db96d56Sopenharmony_ci offsetof(PyODictObject, od_inst_dict), /* tp_dictoffset */ 15797db96d56Sopenharmony_ci (initproc)odict_init, /* tp_init */ 15807db96d56Sopenharmony_ci PyType_GenericAlloc, /* tp_alloc */ 15817db96d56Sopenharmony_ci 0, /* tp_new */ 15827db96d56Sopenharmony_ci 0, /* tp_free */ 15837db96d56Sopenharmony_ci}; 15847db96d56Sopenharmony_ci 15857db96d56Sopenharmony_ci 15867db96d56Sopenharmony_ci/* ---------------------------------------------- 15877db96d56Sopenharmony_ci * the public OrderedDict API 15887db96d56Sopenharmony_ci */ 15897db96d56Sopenharmony_ci 15907db96d56Sopenharmony_ciPyObject * 15917db96d56Sopenharmony_ciPyODict_New(void) 15927db96d56Sopenharmony_ci{ 15937db96d56Sopenharmony_ci return PyDict_Type.tp_new(&PyODict_Type, NULL, NULL); 15947db96d56Sopenharmony_ci} 15957db96d56Sopenharmony_ci 15967db96d56Sopenharmony_cistatic int 15977db96d56Sopenharmony_ci_PyODict_SetItem_KnownHash(PyObject *od, PyObject *key, PyObject *value, 15987db96d56Sopenharmony_ci Py_hash_t hash) 15997db96d56Sopenharmony_ci{ 16007db96d56Sopenharmony_ci int res = _PyDict_SetItem_KnownHash(od, key, value, hash); 16017db96d56Sopenharmony_ci if (res == 0) { 16027db96d56Sopenharmony_ci res = _odict_add_new_node((PyODictObject *)od, key, hash); 16037db96d56Sopenharmony_ci if (res < 0) { 16047db96d56Sopenharmony_ci /* Revert setting the value on the dict */ 16057db96d56Sopenharmony_ci PyObject *exc, *val, *tb; 16067db96d56Sopenharmony_ci PyErr_Fetch(&exc, &val, &tb); 16077db96d56Sopenharmony_ci (void) _PyDict_DelItem_KnownHash(od, key, hash); 16087db96d56Sopenharmony_ci _PyErr_ChainExceptions(exc, val, tb); 16097db96d56Sopenharmony_ci } 16107db96d56Sopenharmony_ci } 16117db96d56Sopenharmony_ci return res; 16127db96d56Sopenharmony_ci} 16137db96d56Sopenharmony_ci 16147db96d56Sopenharmony_ciint 16157db96d56Sopenharmony_ciPyODict_SetItem(PyObject *od, PyObject *key, PyObject *value) 16167db96d56Sopenharmony_ci{ 16177db96d56Sopenharmony_ci Py_hash_t hash = PyObject_Hash(key); 16187db96d56Sopenharmony_ci if (hash == -1) 16197db96d56Sopenharmony_ci return -1; 16207db96d56Sopenharmony_ci return _PyODict_SetItem_KnownHash(od, key, value, hash); 16217db96d56Sopenharmony_ci} 16227db96d56Sopenharmony_ci 16237db96d56Sopenharmony_ciint 16247db96d56Sopenharmony_ciPyODict_DelItem(PyObject *od, PyObject *key) 16257db96d56Sopenharmony_ci{ 16267db96d56Sopenharmony_ci int res; 16277db96d56Sopenharmony_ci Py_hash_t hash = PyObject_Hash(key); 16287db96d56Sopenharmony_ci if (hash == -1) 16297db96d56Sopenharmony_ci return -1; 16307db96d56Sopenharmony_ci res = _odict_clear_node((PyODictObject *)od, NULL, key, hash); 16317db96d56Sopenharmony_ci if (res < 0) 16327db96d56Sopenharmony_ci return -1; 16337db96d56Sopenharmony_ci return _PyDict_DelItem_KnownHash(od, key, hash); 16347db96d56Sopenharmony_ci} 16357db96d56Sopenharmony_ci 16367db96d56Sopenharmony_ci 16377db96d56Sopenharmony_ci/* ------------------------------------------- 16387db96d56Sopenharmony_ci * The OrderedDict views (keys/values/items) 16397db96d56Sopenharmony_ci */ 16407db96d56Sopenharmony_ci 16417db96d56Sopenharmony_citypedef struct { 16427db96d56Sopenharmony_ci PyObject_HEAD 16437db96d56Sopenharmony_ci int kind; 16447db96d56Sopenharmony_ci PyODictObject *di_odict; 16457db96d56Sopenharmony_ci Py_ssize_t di_size; 16467db96d56Sopenharmony_ci size_t di_state; 16477db96d56Sopenharmony_ci PyObject *di_current; 16487db96d56Sopenharmony_ci PyObject *di_result; /* reusable result tuple for iteritems */ 16497db96d56Sopenharmony_ci} odictiterobject; 16507db96d56Sopenharmony_ci 16517db96d56Sopenharmony_cistatic void 16527db96d56Sopenharmony_ciodictiter_dealloc(odictiterobject *di) 16537db96d56Sopenharmony_ci{ 16547db96d56Sopenharmony_ci _PyObject_GC_UNTRACK(di); 16557db96d56Sopenharmony_ci Py_XDECREF(di->di_odict); 16567db96d56Sopenharmony_ci Py_XDECREF(di->di_current); 16577db96d56Sopenharmony_ci if ((di->kind & _odict_ITER_ITEMS) == _odict_ITER_ITEMS) { 16587db96d56Sopenharmony_ci Py_DECREF(di->di_result); 16597db96d56Sopenharmony_ci } 16607db96d56Sopenharmony_ci PyObject_GC_Del(di); 16617db96d56Sopenharmony_ci} 16627db96d56Sopenharmony_ci 16637db96d56Sopenharmony_cistatic int 16647db96d56Sopenharmony_ciodictiter_traverse(odictiterobject *di, visitproc visit, void *arg) 16657db96d56Sopenharmony_ci{ 16667db96d56Sopenharmony_ci Py_VISIT(di->di_odict); 16677db96d56Sopenharmony_ci Py_VISIT(di->di_current); /* A key could be any type, not just str. */ 16687db96d56Sopenharmony_ci Py_VISIT(di->di_result); 16697db96d56Sopenharmony_ci return 0; 16707db96d56Sopenharmony_ci} 16717db96d56Sopenharmony_ci 16727db96d56Sopenharmony_ci/* In order to protect against modifications during iteration, we track 16737db96d56Sopenharmony_ci * the current key instead of the current node. */ 16747db96d56Sopenharmony_cistatic PyObject * 16757db96d56Sopenharmony_ciodictiter_nextkey(odictiterobject *di) 16767db96d56Sopenharmony_ci{ 16777db96d56Sopenharmony_ci PyObject *key = NULL; 16787db96d56Sopenharmony_ci _ODictNode *node; 16797db96d56Sopenharmony_ci int reversed = di->kind & _odict_ITER_REVERSED; 16807db96d56Sopenharmony_ci 16817db96d56Sopenharmony_ci if (di->di_odict == NULL) 16827db96d56Sopenharmony_ci return NULL; 16837db96d56Sopenharmony_ci if (di->di_current == NULL) 16847db96d56Sopenharmony_ci goto done; /* We're already done. */ 16857db96d56Sopenharmony_ci 16867db96d56Sopenharmony_ci /* Check for unsupported changes. */ 16877db96d56Sopenharmony_ci if (di->di_odict->od_state != di->di_state) { 16887db96d56Sopenharmony_ci PyErr_SetString(PyExc_RuntimeError, 16897db96d56Sopenharmony_ci "OrderedDict mutated during iteration"); 16907db96d56Sopenharmony_ci goto done; 16917db96d56Sopenharmony_ci } 16927db96d56Sopenharmony_ci if (di->di_size != PyODict_SIZE(di->di_odict)) { 16937db96d56Sopenharmony_ci PyErr_SetString(PyExc_RuntimeError, 16947db96d56Sopenharmony_ci "OrderedDict changed size during iteration"); 16957db96d56Sopenharmony_ci di->di_size = -1; /* Make this state sticky */ 16967db96d56Sopenharmony_ci return NULL; 16977db96d56Sopenharmony_ci } 16987db96d56Sopenharmony_ci 16997db96d56Sopenharmony_ci /* Get the key. */ 17007db96d56Sopenharmony_ci node = _odict_find_node(di->di_odict, di->di_current); 17017db96d56Sopenharmony_ci if (node == NULL) { 17027db96d56Sopenharmony_ci if (!PyErr_Occurred()) 17037db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, di->di_current); 17047db96d56Sopenharmony_ci /* Must have been deleted. */ 17057db96d56Sopenharmony_ci Py_CLEAR(di->di_current); 17067db96d56Sopenharmony_ci return NULL; 17077db96d56Sopenharmony_ci } 17087db96d56Sopenharmony_ci key = di->di_current; 17097db96d56Sopenharmony_ci 17107db96d56Sopenharmony_ci /* Advance to the next key. */ 17117db96d56Sopenharmony_ci node = reversed ? _odictnode_PREV(node) : _odictnode_NEXT(node); 17127db96d56Sopenharmony_ci if (node == NULL) { 17137db96d56Sopenharmony_ci /* Reached the end. */ 17147db96d56Sopenharmony_ci di->di_current = NULL; 17157db96d56Sopenharmony_ci } 17167db96d56Sopenharmony_ci else { 17177db96d56Sopenharmony_ci di->di_current = _odictnode_KEY(node); 17187db96d56Sopenharmony_ci Py_INCREF(di->di_current); 17197db96d56Sopenharmony_ci } 17207db96d56Sopenharmony_ci 17217db96d56Sopenharmony_ci return key; 17227db96d56Sopenharmony_ci 17237db96d56Sopenharmony_cidone: 17247db96d56Sopenharmony_ci Py_CLEAR(di->di_odict); 17257db96d56Sopenharmony_ci return key; 17267db96d56Sopenharmony_ci} 17277db96d56Sopenharmony_ci 17287db96d56Sopenharmony_cistatic PyObject * 17297db96d56Sopenharmony_ciodictiter_iternext(odictiterobject *di) 17307db96d56Sopenharmony_ci{ 17317db96d56Sopenharmony_ci PyObject *result, *value; 17327db96d56Sopenharmony_ci PyObject *key = odictiter_nextkey(di); /* new reference */ 17337db96d56Sopenharmony_ci 17347db96d56Sopenharmony_ci if (key == NULL) 17357db96d56Sopenharmony_ci return NULL; 17367db96d56Sopenharmony_ci 17377db96d56Sopenharmony_ci /* Handle the keys case. */ 17387db96d56Sopenharmony_ci if (! (di->kind & _odict_ITER_VALUES)) { 17397db96d56Sopenharmony_ci return key; 17407db96d56Sopenharmony_ci } 17417db96d56Sopenharmony_ci 17427db96d56Sopenharmony_ci value = PyODict_GetItem((PyObject *)di->di_odict, key); /* borrowed */ 17437db96d56Sopenharmony_ci if (value == NULL) { 17447db96d56Sopenharmony_ci if (!PyErr_Occurred()) 17457db96d56Sopenharmony_ci PyErr_SetObject(PyExc_KeyError, key); 17467db96d56Sopenharmony_ci Py_DECREF(key); 17477db96d56Sopenharmony_ci goto done; 17487db96d56Sopenharmony_ci } 17497db96d56Sopenharmony_ci Py_INCREF(value); 17507db96d56Sopenharmony_ci 17517db96d56Sopenharmony_ci /* Handle the values case. */ 17527db96d56Sopenharmony_ci if (!(di->kind & _odict_ITER_KEYS)) { 17537db96d56Sopenharmony_ci Py_DECREF(key); 17547db96d56Sopenharmony_ci return value; 17557db96d56Sopenharmony_ci } 17567db96d56Sopenharmony_ci 17577db96d56Sopenharmony_ci /* Handle the items case. */ 17587db96d56Sopenharmony_ci result = di->di_result; 17597db96d56Sopenharmony_ci 17607db96d56Sopenharmony_ci if (Py_REFCNT(result) == 1) { 17617db96d56Sopenharmony_ci /* not in use so we can reuse it 17627db96d56Sopenharmony_ci * (the common case during iteration) */ 17637db96d56Sopenharmony_ci Py_INCREF(result); 17647db96d56Sopenharmony_ci Py_DECREF(PyTuple_GET_ITEM(result, 0)); /* borrowed */ 17657db96d56Sopenharmony_ci Py_DECREF(PyTuple_GET_ITEM(result, 1)); /* borrowed */ 17667db96d56Sopenharmony_ci // bpo-42536: The GC may have untracked this result tuple. Since we're 17677db96d56Sopenharmony_ci // recycling it, make sure it's tracked again: 17687db96d56Sopenharmony_ci if (!_PyObject_GC_IS_TRACKED(result)) { 17697db96d56Sopenharmony_ci _PyObject_GC_TRACK(result); 17707db96d56Sopenharmony_ci } 17717db96d56Sopenharmony_ci } 17727db96d56Sopenharmony_ci else { 17737db96d56Sopenharmony_ci result = PyTuple_New(2); 17747db96d56Sopenharmony_ci if (result == NULL) { 17757db96d56Sopenharmony_ci Py_DECREF(key); 17767db96d56Sopenharmony_ci Py_DECREF(value); 17777db96d56Sopenharmony_ci goto done; 17787db96d56Sopenharmony_ci } 17797db96d56Sopenharmony_ci } 17807db96d56Sopenharmony_ci 17817db96d56Sopenharmony_ci PyTuple_SET_ITEM(result, 0, key); /* steals reference */ 17827db96d56Sopenharmony_ci PyTuple_SET_ITEM(result, 1, value); /* steals reference */ 17837db96d56Sopenharmony_ci return result; 17847db96d56Sopenharmony_ci 17857db96d56Sopenharmony_cidone: 17867db96d56Sopenharmony_ci Py_CLEAR(di->di_current); 17877db96d56Sopenharmony_ci Py_CLEAR(di->di_odict); 17887db96d56Sopenharmony_ci return NULL; 17897db96d56Sopenharmony_ci} 17907db96d56Sopenharmony_ci 17917db96d56Sopenharmony_ci/* No need for tp_clear because odictiterobject is not mutable. */ 17927db96d56Sopenharmony_ci 17937db96d56Sopenharmony_ciPyDoc_STRVAR(reduce_doc, "Return state information for pickling"); 17947db96d56Sopenharmony_ci 17957db96d56Sopenharmony_cistatic PyObject * 17967db96d56Sopenharmony_ciodictiter_reduce(odictiterobject *di, PyObject *Py_UNUSED(ignored)) 17977db96d56Sopenharmony_ci{ 17987db96d56Sopenharmony_ci /* copy the iterator state */ 17997db96d56Sopenharmony_ci odictiterobject tmp = *di; 18007db96d56Sopenharmony_ci Py_XINCREF(tmp.di_odict); 18017db96d56Sopenharmony_ci Py_XINCREF(tmp.di_current); 18027db96d56Sopenharmony_ci 18037db96d56Sopenharmony_ci /* iterate the temporary into a list */ 18047db96d56Sopenharmony_ci PyObject *list = PySequence_List((PyObject*)&tmp); 18057db96d56Sopenharmony_ci Py_XDECREF(tmp.di_odict); 18067db96d56Sopenharmony_ci Py_XDECREF(tmp.di_current); 18077db96d56Sopenharmony_ci if (list == NULL) { 18087db96d56Sopenharmony_ci return NULL; 18097db96d56Sopenharmony_ci } 18107db96d56Sopenharmony_ci return Py_BuildValue("N(N)", _PyEval_GetBuiltin(&_Py_ID(iter)), list); 18117db96d56Sopenharmony_ci} 18127db96d56Sopenharmony_ci 18137db96d56Sopenharmony_cistatic PyMethodDef odictiter_methods[] = { 18147db96d56Sopenharmony_ci {"__reduce__", (PyCFunction)odictiter_reduce, METH_NOARGS, reduce_doc}, 18157db96d56Sopenharmony_ci {NULL, NULL} /* sentinel */ 18167db96d56Sopenharmony_ci}; 18177db96d56Sopenharmony_ci 18187db96d56Sopenharmony_ciPyTypeObject PyODictIter_Type = { 18197db96d56Sopenharmony_ci PyVarObject_HEAD_INIT(&PyType_Type, 0) 18207db96d56Sopenharmony_ci "odict_iterator", /* tp_name */ 18217db96d56Sopenharmony_ci sizeof(odictiterobject), /* tp_basicsize */ 18227db96d56Sopenharmony_ci 0, /* tp_itemsize */ 18237db96d56Sopenharmony_ci /* methods */ 18247db96d56Sopenharmony_ci (destructor)odictiter_dealloc, /* tp_dealloc */ 18257db96d56Sopenharmony_ci 0, /* tp_vectorcall_offset */ 18267db96d56Sopenharmony_ci 0, /* tp_getattr */ 18277db96d56Sopenharmony_ci 0, /* tp_setattr */ 18287db96d56Sopenharmony_ci 0, /* tp_as_async */ 18297db96d56Sopenharmony_ci 0, /* tp_repr */ 18307db96d56Sopenharmony_ci 0, /* tp_as_number */ 18317db96d56Sopenharmony_ci 0, /* tp_as_sequence */ 18327db96d56Sopenharmony_ci 0, /* tp_as_mapping */ 18337db96d56Sopenharmony_ci 0, /* tp_hash */ 18347db96d56Sopenharmony_ci 0, /* tp_call */ 18357db96d56Sopenharmony_ci 0, /* tp_str */ 18367db96d56Sopenharmony_ci PyObject_GenericGetAttr, /* tp_getattro */ 18377db96d56Sopenharmony_ci 0, /* tp_setattro */ 18387db96d56Sopenharmony_ci 0, /* tp_as_buffer */ 18397db96d56Sopenharmony_ci Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC, /* tp_flags */ 18407db96d56Sopenharmony_ci 0, /* tp_doc */ 18417db96d56Sopenharmony_ci (traverseproc)odictiter_traverse, /* tp_traverse */ 18427db96d56Sopenharmony_ci 0, /* tp_clear */ 18437db96d56Sopenharmony_ci 0, /* tp_richcompare */ 18447db96d56Sopenharmony_ci 0, /* tp_weaklistoffset */ 18457db96d56Sopenharmony_ci PyObject_SelfIter, /* tp_iter */ 18467db96d56Sopenharmony_ci (iternextfunc)odictiter_iternext, /* tp_iternext */ 18477db96d56Sopenharmony_ci odictiter_methods, /* tp_methods */ 18487db96d56Sopenharmony_ci 0, 18497db96d56Sopenharmony_ci}; 18507db96d56Sopenharmony_ci 18517db96d56Sopenharmony_cistatic PyObject * 18527db96d56Sopenharmony_ciodictiter_new(PyODictObject *od, int kind) 18537db96d56Sopenharmony_ci{ 18547db96d56Sopenharmony_ci odictiterobject *di; 18557db96d56Sopenharmony_ci _ODictNode *node; 18567db96d56Sopenharmony_ci int reversed = kind & _odict_ITER_REVERSED; 18577db96d56Sopenharmony_ci 18587db96d56Sopenharmony_ci di = PyObject_GC_New(odictiterobject, &PyODictIter_Type); 18597db96d56Sopenharmony_ci if (di == NULL) 18607db96d56Sopenharmony_ci return NULL; 18617db96d56Sopenharmony_ci 18627db96d56Sopenharmony_ci if ((kind & _odict_ITER_ITEMS) == _odict_ITER_ITEMS) { 18637db96d56Sopenharmony_ci di->di_result = PyTuple_Pack(2, Py_None, Py_None); 18647db96d56Sopenharmony_ci if (di->di_result == NULL) { 18657db96d56Sopenharmony_ci Py_DECREF(di); 18667db96d56Sopenharmony_ci return NULL; 18677db96d56Sopenharmony_ci } 18687db96d56Sopenharmony_ci } 18697db96d56Sopenharmony_ci else { 18707db96d56Sopenharmony_ci di->di_result = NULL; 18717db96d56Sopenharmony_ci } 18727db96d56Sopenharmony_ci 18737db96d56Sopenharmony_ci di->kind = kind; 18747db96d56Sopenharmony_ci node = reversed ? _odict_LAST(od) : _odict_FIRST(od); 18757db96d56Sopenharmony_ci di->di_current = node ? _odictnode_KEY(node) : NULL; 18767db96d56Sopenharmony_ci Py_XINCREF(di->di_current); 18777db96d56Sopenharmony_ci di->di_size = PyODict_SIZE(od); 18787db96d56Sopenharmony_ci di->di_state = od->od_state; 18797db96d56Sopenharmony_ci di->di_odict = od; 18807db96d56Sopenharmony_ci Py_INCREF(od); 18817db96d56Sopenharmony_ci 18827db96d56Sopenharmony_ci _PyObject_GC_TRACK(di); 18837db96d56Sopenharmony_ci return (PyObject *)di; 18847db96d56Sopenharmony_ci} 18857db96d56Sopenharmony_ci 18867db96d56Sopenharmony_ci/* keys() */ 18877db96d56Sopenharmony_ci 18887db96d56Sopenharmony_cistatic PyObject * 18897db96d56Sopenharmony_ciodictkeys_iter(_PyDictViewObject *dv) 18907db96d56Sopenharmony_ci{ 18917db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 18927db96d56Sopenharmony_ci Py_RETURN_NONE; 18937db96d56Sopenharmony_ci } 18947db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 18957db96d56Sopenharmony_ci _odict_ITER_KEYS); 18967db96d56Sopenharmony_ci} 18977db96d56Sopenharmony_ci 18987db96d56Sopenharmony_cistatic PyObject * 18997db96d56Sopenharmony_ciodictkeys_reversed(_PyDictViewObject *dv, PyObject *Py_UNUSED(ignored)) 19007db96d56Sopenharmony_ci{ 19017db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 19027db96d56Sopenharmony_ci Py_RETURN_NONE; 19037db96d56Sopenharmony_ci } 19047db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 19057db96d56Sopenharmony_ci _odict_ITER_KEYS|_odict_ITER_REVERSED); 19067db96d56Sopenharmony_ci} 19077db96d56Sopenharmony_ci 19087db96d56Sopenharmony_cistatic PyMethodDef odictkeys_methods[] = { 19097db96d56Sopenharmony_ci {"__reversed__", (PyCFunction)odictkeys_reversed, METH_NOARGS, NULL}, 19107db96d56Sopenharmony_ci {NULL, NULL} /* sentinel */ 19117db96d56Sopenharmony_ci}; 19127db96d56Sopenharmony_ci 19137db96d56Sopenharmony_ciPyTypeObject PyODictKeys_Type = { 19147db96d56Sopenharmony_ci PyVarObject_HEAD_INIT(&PyType_Type, 0) 19157db96d56Sopenharmony_ci "odict_keys", /* tp_name */ 19167db96d56Sopenharmony_ci 0, /* tp_basicsize */ 19177db96d56Sopenharmony_ci 0, /* tp_itemsize */ 19187db96d56Sopenharmony_ci 0, /* tp_dealloc */ 19197db96d56Sopenharmony_ci 0, /* tp_vectorcall_offset */ 19207db96d56Sopenharmony_ci 0, /* tp_getattr */ 19217db96d56Sopenharmony_ci 0, /* tp_setattr */ 19227db96d56Sopenharmony_ci 0, /* tp_as_async */ 19237db96d56Sopenharmony_ci 0, /* tp_repr */ 19247db96d56Sopenharmony_ci 0, /* tp_as_number */ 19257db96d56Sopenharmony_ci 0, /* tp_as_sequence */ 19267db96d56Sopenharmony_ci 0, /* tp_as_mapping */ 19277db96d56Sopenharmony_ci 0, /* tp_hash */ 19287db96d56Sopenharmony_ci 0, /* tp_call */ 19297db96d56Sopenharmony_ci 0, /* tp_str */ 19307db96d56Sopenharmony_ci 0, /* tp_getattro */ 19317db96d56Sopenharmony_ci 0, /* tp_setattro */ 19327db96d56Sopenharmony_ci 0, /* tp_as_buffer */ 19337db96d56Sopenharmony_ci 0, /* tp_flags */ 19347db96d56Sopenharmony_ci 0, /* tp_doc */ 19357db96d56Sopenharmony_ci 0, /* tp_traverse */ 19367db96d56Sopenharmony_ci 0, /* tp_clear */ 19377db96d56Sopenharmony_ci 0, /* tp_richcompare */ 19387db96d56Sopenharmony_ci 0, /* tp_weaklistoffset */ 19397db96d56Sopenharmony_ci (getiterfunc)odictkeys_iter, /* tp_iter */ 19407db96d56Sopenharmony_ci 0, /* tp_iternext */ 19417db96d56Sopenharmony_ci odictkeys_methods, /* tp_methods */ 19427db96d56Sopenharmony_ci 0, /* tp_members */ 19437db96d56Sopenharmony_ci 0, /* tp_getset */ 19447db96d56Sopenharmony_ci &PyDictKeys_Type, /* tp_base */ 19457db96d56Sopenharmony_ci}; 19467db96d56Sopenharmony_ci 19477db96d56Sopenharmony_cistatic PyObject * 19487db96d56Sopenharmony_ciodictkeys_new(PyObject *od, PyObject *Py_UNUSED(ignored)) 19497db96d56Sopenharmony_ci{ 19507db96d56Sopenharmony_ci return _PyDictView_New(od, &PyODictKeys_Type); 19517db96d56Sopenharmony_ci} 19527db96d56Sopenharmony_ci 19537db96d56Sopenharmony_ci/* items() */ 19547db96d56Sopenharmony_ci 19557db96d56Sopenharmony_cistatic PyObject * 19567db96d56Sopenharmony_ciodictitems_iter(_PyDictViewObject *dv) 19577db96d56Sopenharmony_ci{ 19587db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 19597db96d56Sopenharmony_ci Py_RETURN_NONE; 19607db96d56Sopenharmony_ci } 19617db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 19627db96d56Sopenharmony_ci _odict_ITER_KEYS|_odict_ITER_VALUES); 19637db96d56Sopenharmony_ci} 19647db96d56Sopenharmony_ci 19657db96d56Sopenharmony_cistatic PyObject * 19667db96d56Sopenharmony_ciodictitems_reversed(_PyDictViewObject *dv, PyObject *Py_UNUSED(ignored)) 19677db96d56Sopenharmony_ci{ 19687db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 19697db96d56Sopenharmony_ci Py_RETURN_NONE; 19707db96d56Sopenharmony_ci } 19717db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 19727db96d56Sopenharmony_ci _odict_ITER_KEYS|_odict_ITER_VALUES|_odict_ITER_REVERSED); 19737db96d56Sopenharmony_ci} 19747db96d56Sopenharmony_ci 19757db96d56Sopenharmony_cistatic PyMethodDef odictitems_methods[] = { 19767db96d56Sopenharmony_ci {"__reversed__", (PyCFunction)odictitems_reversed, METH_NOARGS, NULL}, 19777db96d56Sopenharmony_ci {NULL, NULL} /* sentinel */ 19787db96d56Sopenharmony_ci}; 19797db96d56Sopenharmony_ci 19807db96d56Sopenharmony_ciPyTypeObject PyODictItems_Type = { 19817db96d56Sopenharmony_ci PyVarObject_HEAD_INIT(&PyType_Type, 0) 19827db96d56Sopenharmony_ci "odict_items", /* tp_name */ 19837db96d56Sopenharmony_ci 0, /* tp_basicsize */ 19847db96d56Sopenharmony_ci 0, /* tp_itemsize */ 19857db96d56Sopenharmony_ci 0, /* tp_dealloc */ 19867db96d56Sopenharmony_ci 0, /* tp_vectorcall_offset */ 19877db96d56Sopenharmony_ci 0, /* tp_getattr */ 19887db96d56Sopenharmony_ci 0, /* tp_setattr */ 19897db96d56Sopenharmony_ci 0, /* tp_as_async */ 19907db96d56Sopenharmony_ci 0, /* tp_repr */ 19917db96d56Sopenharmony_ci 0, /* tp_as_number */ 19927db96d56Sopenharmony_ci 0, /* tp_as_sequence */ 19937db96d56Sopenharmony_ci 0, /* tp_as_mapping */ 19947db96d56Sopenharmony_ci 0, /* tp_hash */ 19957db96d56Sopenharmony_ci 0, /* tp_call */ 19967db96d56Sopenharmony_ci 0, /* tp_str */ 19977db96d56Sopenharmony_ci 0, /* tp_getattro */ 19987db96d56Sopenharmony_ci 0, /* tp_setattro */ 19997db96d56Sopenharmony_ci 0, /* tp_as_buffer */ 20007db96d56Sopenharmony_ci 0, /* tp_flags */ 20017db96d56Sopenharmony_ci 0, /* tp_doc */ 20027db96d56Sopenharmony_ci 0, /* tp_traverse */ 20037db96d56Sopenharmony_ci 0, /* tp_clear */ 20047db96d56Sopenharmony_ci 0, /* tp_richcompare */ 20057db96d56Sopenharmony_ci 0, /* tp_weaklistoffset */ 20067db96d56Sopenharmony_ci (getiterfunc)odictitems_iter, /* tp_iter */ 20077db96d56Sopenharmony_ci 0, /* tp_iternext */ 20087db96d56Sopenharmony_ci odictitems_methods, /* tp_methods */ 20097db96d56Sopenharmony_ci 0, /* tp_members */ 20107db96d56Sopenharmony_ci 0, /* tp_getset */ 20117db96d56Sopenharmony_ci &PyDictItems_Type, /* tp_base */ 20127db96d56Sopenharmony_ci}; 20137db96d56Sopenharmony_ci 20147db96d56Sopenharmony_cistatic PyObject * 20157db96d56Sopenharmony_ciodictitems_new(PyObject *od, PyObject *Py_UNUSED(ignored)) 20167db96d56Sopenharmony_ci{ 20177db96d56Sopenharmony_ci return _PyDictView_New(od, &PyODictItems_Type); 20187db96d56Sopenharmony_ci} 20197db96d56Sopenharmony_ci 20207db96d56Sopenharmony_ci/* values() */ 20217db96d56Sopenharmony_ci 20227db96d56Sopenharmony_cistatic PyObject * 20237db96d56Sopenharmony_ciodictvalues_iter(_PyDictViewObject *dv) 20247db96d56Sopenharmony_ci{ 20257db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 20267db96d56Sopenharmony_ci Py_RETURN_NONE; 20277db96d56Sopenharmony_ci } 20287db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 20297db96d56Sopenharmony_ci _odict_ITER_VALUES); 20307db96d56Sopenharmony_ci} 20317db96d56Sopenharmony_ci 20327db96d56Sopenharmony_cistatic PyObject * 20337db96d56Sopenharmony_ciodictvalues_reversed(_PyDictViewObject *dv, PyObject *Py_UNUSED(ignored)) 20347db96d56Sopenharmony_ci{ 20357db96d56Sopenharmony_ci if (dv->dv_dict == NULL) { 20367db96d56Sopenharmony_ci Py_RETURN_NONE; 20377db96d56Sopenharmony_ci } 20387db96d56Sopenharmony_ci return odictiter_new((PyODictObject *)dv->dv_dict, 20397db96d56Sopenharmony_ci _odict_ITER_VALUES|_odict_ITER_REVERSED); 20407db96d56Sopenharmony_ci} 20417db96d56Sopenharmony_ci 20427db96d56Sopenharmony_cistatic PyMethodDef odictvalues_methods[] = { 20437db96d56Sopenharmony_ci {"__reversed__", (PyCFunction)odictvalues_reversed, METH_NOARGS, NULL}, 20447db96d56Sopenharmony_ci {NULL, NULL} /* sentinel */ 20457db96d56Sopenharmony_ci}; 20467db96d56Sopenharmony_ci 20477db96d56Sopenharmony_ciPyTypeObject PyODictValues_Type = { 20487db96d56Sopenharmony_ci PyVarObject_HEAD_INIT(&PyType_Type, 0) 20497db96d56Sopenharmony_ci "odict_values", /* tp_name */ 20507db96d56Sopenharmony_ci 0, /* tp_basicsize */ 20517db96d56Sopenharmony_ci 0, /* tp_itemsize */ 20527db96d56Sopenharmony_ci 0, /* tp_dealloc */ 20537db96d56Sopenharmony_ci 0, /* tp_vectorcall_offset */ 20547db96d56Sopenharmony_ci 0, /* tp_getattr */ 20557db96d56Sopenharmony_ci 0, /* tp_setattr */ 20567db96d56Sopenharmony_ci 0, /* tp_as_async */ 20577db96d56Sopenharmony_ci 0, /* tp_repr */ 20587db96d56Sopenharmony_ci 0, /* tp_as_number */ 20597db96d56Sopenharmony_ci 0, /* tp_as_sequence */ 20607db96d56Sopenharmony_ci 0, /* tp_as_mapping */ 20617db96d56Sopenharmony_ci 0, /* tp_hash */ 20627db96d56Sopenharmony_ci 0, /* tp_call */ 20637db96d56Sopenharmony_ci 0, /* tp_str */ 20647db96d56Sopenharmony_ci 0, /* tp_getattro */ 20657db96d56Sopenharmony_ci 0, /* tp_setattro */ 20667db96d56Sopenharmony_ci 0, /* tp_as_buffer */ 20677db96d56Sopenharmony_ci 0, /* tp_flags */ 20687db96d56Sopenharmony_ci 0, /* tp_doc */ 20697db96d56Sopenharmony_ci 0, /* tp_traverse */ 20707db96d56Sopenharmony_ci 0, /* tp_clear */ 20717db96d56Sopenharmony_ci 0, /* tp_richcompare */ 20727db96d56Sopenharmony_ci 0, /* tp_weaklistoffset */ 20737db96d56Sopenharmony_ci (getiterfunc)odictvalues_iter, /* tp_iter */ 20747db96d56Sopenharmony_ci 0, /* tp_iternext */ 20757db96d56Sopenharmony_ci odictvalues_methods, /* tp_methods */ 20767db96d56Sopenharmony_ci 0, /* tp_members */ 20777db96d56Sopenharmony_ci 0, /* tp_getset */ 20787db96d56Sopenharmony_ci &PyDictValues_Type, /* tp_base */ 20797db96d56Sopenharmony_ci}; 20807db96d56Sopenharmony_ci 20817db96d56Sopenharmony_cistatic PyObject * 20827db96d56Sopenharmony_ciodictvalues_new(PyObject *od, PyObject *Py_UNUSED(ignored)) 20837db96d56Sopenharmony_ci{ 20847db96d56Sopenharmony_ci return _PyDictView_New(od, &PyODictValues_Type); 20857db96d56Sopenharmony_ci} 20867db96d56Sopenharmony_ci 20877db96d56Sopenharmony_ci 20887db96d56Sopenharmony_ci/* ---------------------------------------------- 20897db96d56Sopenharmony_ci MutableMapping implementations 20907db96d56Sopenharmony_ci 20917db96d56Sopenharmony_ciMapping: 20927db96d56Sopenharmony_ci 20937db96d56Sopenharmony_ci============ =========== 20947db96d56Sopenharmony_cimethod uses 20957db96d56Sopenharmony_ci============ =========== 20967db96d56Sopenharmony_ci__contains__ __getitem__ 20977db96d56Sopenharmony_ci__eq__ items 20987db96d56Sopenharmony_ci__getitem__ + 20997db96d56Sopenharmony_ci__iter__ + 21007db96d56Sopenharmony_ci__len__ + 21017db96d56Sopenharmony_ci__ne__ __eq__ 21027db96d56Sopenharmony_ciget __getitem__ 21037db96d56Sopenharmony_ciitems ItemsView 21047db96d56Sopenharmony_cikeys KeysView 21057db96d56Sopenharmony_civalues ValuesView 21067db96d56Sopenharmony_ci============ =========== 21077db96d56Sopenharmony_ci 21087db96d56Sopenharmony_ciItemsView uses __len__, __iter__, and __getitem__. 21097db96d56Sopenharmony_ciKeysView uses __len__, __iter__, and __contains__. 21107db96d56Sopenharmony_ciValuesView uses __len__, __iter__, and __getitem__. 21117db96d56Sopenharmony_ci 21127db96d56Sopenharmony_ciMutableMapping: 21137db96d56Sopenharmony_ci 21147db96d56Sopenharmony_ci============ =========== 21157db96d56Sopenharmony_cimethod uses 21167db96d56Sopenharmony_ci============ =========== 21177db96d56Sopenharmony_ci__delitem__ + 21187db96d56Sopenharmony_ci__setitem__ + 21197db96d56Sopenharmony_ciclear popitem 21207db96d56Sopenharmony_cipop __getitem__ 21217db96d56Sopenharmony_ci __delitem__ 21227db96d56Sopenharmony_cipopitem __iter__ 21237db96d56Sopenharmony_ci _getitem__ 21247db96d56Sopenharmony_ci __delitem__ 21257db96d56Sopenharmony_cisetdefault __getitem__ 21267db96d56Sopenharmony_ci __setitem__ 21277db96d56Sopenharmony_ciupdate __setitem__ 21287db96d56Sopenharmony_ci============ =========== 21297db96d56Sopenharmony_ci*/ 21307db96d56Sopenharmony_ci 21317db96d56Sopenharmony_cistatic int 21327db96d56Sopenharmony_cimutablemapping_add_pairs(PyObject *self, PyObject *pairs) 21337db96d56Sopenharmony_ci{ 21347db96d56Sopenharmony_ci PyObject *pair, *iterator, *unexpected; 21357db96d56Sopenharmony_ci int res = 0; 21367db96d56Sopenharmony_ci 21377db96d56Sopenharmony_ci iterator = PyObject_GetIter(pairs); 21387db96d56Sopenharmony_ci if (iterator == NULL) 21397db96d56Sopenharmony_ci return -1; 21407db96d56Sopenharmony_ci PyErr_Clear(); 21417db96d56Sopenharmony_ci 21427db96d56Sopenharmony_ci while ((pair = PyIter_Next(iterator)) != NULL) { 21437db96d56Sopenharmony_ci /* could be more efficient (see UNPACK_SEQUENCE in ceval.c) */ 21447db96d56Sopenharmony_ci PyObject *key = NULL, *value = NULL; 21457db96d56Sopenharmony_ci PyObject *pair_iterator = PyObject_GetIter(pair); 21467db96d56Sopenharmony_ci if (pair_iterator == NULL) 21477db96d56Sopenharmony_ci goto Done; 21487db96d56Sopenharmony_ci 21497db96d56Sopenharmony_ci key = PyIter_Next(pair_iterator); 21507db96d56Sopenharmony_ci if (key == NULL) { 21517db96d56Sopenharmony_ci if (!PyErr_Occurred()) 21527db96d56Sopenharmony_ci PyErr_SetString(PyExc_ValueError, 21537db96d56Sopenharmony_ci "need more than 0 values to unpack"); 21547db96d56Sopenharmony_ci goto Done; 21557db96d56Sopenharmony_ci } 21567db96d56Sopenharmony_ci 21577db96d56Sopenharmony_ci value = PyIter_Next(pair_iterator); 21587db96d56Sopenharmony_ci if (value == NULL) { 21597db96d56Sopenharmony_ci if (!PyErr_Occurred()) 21607db96d56Sopenharmony_ci PyErr_SetString(PyExc_ValueError, 21617db96d56Sopenharmony_ci "need more than 1 value to unpack"); 21627db96d56Sopenharmony_ci goto Done; 21637db96d56Sopenharmony_ci } 21647db96d56Sopenharmony_ci 21657db96d56Sopenharmony_ci unexpected = PyIter_Next(pair_iterator); 21667db96d56Sopenharmony_ci if (unexpected != NULL) { 21677db96d56Sopenharmony_ci Py_DECREF(unexpected); 21687db96d56Sopenharmony_ci PyErr_SetString(PyExc_ValueError, 21697db96d56Sopenharmony_ci "too many values to unpack (expected 2)"); 21707db96d56Sopenharmony_ci goto Done; 21717db96d56Sopenharmony_ci } 21727db96d56Sopenharmony_ci else if (PyErr_Occurred()) 21737db96d56Sopenharmony_ci goto Done; 21747db96d56Sopenharmony_ci 21757db96d56Sopenharmony_ci res = PyObject_SetItem(self, key, value); 21767db96d56Sopenharmony_ci 21777db96d56Sopenharmony_ciDone: 21787db96d56Sopenharmony_ci Py_DECREF(pair); 21797db96d56Sopenharmony_ci Py_XDECREF(pair_iterator); 21807db96d56Sopenharmony_ci Py_XDECREF(key); 21817db96d56Sopenharmony_ci Py_XDECREF(value); 21827db96d56Sopenharmony_ci if (PyErr_Occurred()) 21837db96d56Sopenharmony_ci break; 21847db96d56Sopenharmony_ci } 21857db96d56Sopenharmony_ci Py_DECREF(iterator); 21867db96d56Sopenharmony_ci 21877db96d56Sopenharmony_ci if (res < 0 || PyErr_Occurred() != NULL) 21887db96d56Sopenharmony_ci return -1; 21897db96d56Sopenharmony_ci else 21907db96d56Sopenharmony_ci return 0; 21917db96d56Sopenharmony_ci} 21927db96d56Sopenharmony_ci 21937db96d56Sopenharmony_cistatic int 21947db96d56Sopenharmony_cimutablemapping_update_arg(PyObject *self, PyObject *arg) 21957db96d56Sopenharmony_ci{ 21967db96d56Sopenharmony_ci int res = 0; 21977db96d56Sopenharmony_ci if (PyDict_CheckExact(arg)) { 21987db96d56Sopenharmony_ci PyObject *items = PyDict_Items(arg); 21997db96d56Sopenharmony_ci if (items == NULL) { 22007db96d56Sopenharmony_ci return -1; 22017db96d56Sopenharmony_ci } 22027db96d56Sopenharmony_ci res = mutablemapping_add_pairs(self, items); 22037db96d56Sopenharmony_ci Py_DECREF(items); 22047db96d56Sopenharmony_ci return res; 22057db96d56Sopenharmony_ci } 22067db96d56Sopenharmony_ci PyObject *func; 22077db96d56Sopenharmony_ci if (_PyObject_LookupAttr(arg, &_Py_ID(keys), &func) < 0) { 22087db96d56Sopenharmony_ci return -1; 22097db96d56Sopenharmony_ci } 22107db96d56Sopenharmony_ci if (func != NULL) { 22117db96d56Sopenharmony_ci PyObject *keys = _PyObject_CallNoArgs(func); 22127db96d56Sopenharmony_ci Py_DECREF(func); 22137db96d56Sopenharmony_ci if (keys == NULL) { 22147db96d56Sopenharmony_ci return -1; 22157db96d56Sopenharmony_ci } 22167db96d56Sopenharmony_ci PyObject *iterator = PyObject_GetIter(keys); 22177db96d56Sopenharmony_ci Py_DECREF(keys); 22187db96d56Sopenharmony_ci if (iterator == NULL) { 22197db96d56Sopenharmony_ci return -1; 22207db96d56Sopenharmony_ci } 22217db96d56Sopenharmony_ci PyObject *key; 22227db96d56Sopenharmony_ci while (res == 0 && (key = PyIter_Next(iterator))) { 22237db96d56Sopenharmony_ci PyObject *value = PyObject_GetItem(arg, key); 22247db96d56Sopenharmony_ci if (value != NULL) { 22257db96d56Sopenharmony_ci res = PyObject_SetItem(self, key, value); 22267db96d56Sopenharmony_ci Py_DECREF(value); 22277db96d56Sopenharmony_ci } 22287db96d56Sopenharmony_ci else { 22297db96d56Sopenharmony_ci res = -1; 22307db96d56Sopenharmony_ci } 22317db96d56Sopenharmony_ci Py_DECREF(key); 22327db96d56Sopenharmony_ci } 22337db96d56Sopenharmony_ci Py_DECREF(iterator); 22347db96d56Sopenharmony_ci if (res != 0 || PyErr_Occurred()) { 22357db96d56Sopenharmony_ci return -1; 22367db96d56Sopenharmony_ci } 22377db96d56Sopenharmony_ci return 0; 22387db96d56Sopenharmony_ci } 22397db96d56Sopenharmony_ci if (_PyObject_LookupAttr(arg, &_Py_ID(items), &func) < 0) { 22407db96d56Sopenharmony_ci return -1; 22417db96d56Sopenharmony_ci } 22427db96d56Sopenharmony_ci if (func != NULL) { 22437db96d56Sopenharmony_ci PyObject *items = _PyObject_CallNoArgs(func); 22447db96d56Sopenharmony_ci Py_DECREF(func); 22457db96d56Sopenharmony_ci if (items == NULL) { 22467db96d56Sopenharmony_ci return -1; 22477db96d56Sopenharmony_ci } 22487db96d56Sopenharmony_ci res = mutablemapping_add_pairs(self, items); 22497db96d56Sopenharmony_ci Py_DECREF(items); 22507db96d56Sopenharmony_ci return res; 22517db96d56Sopenharmony_ci } 22527db96d56Sopenharmony_ci res = mutablemapping_add_pairs(self, arg); 22537db96d56Sopenharmony_ci return res; 22547db96d56Sopenharmony_ci} 22557db96d56Sopenharmony_ci 22567db96d56Sopenharmony_cistatic PyObject * 22577db96d56Sopenharmony_cimutablemapping_update(PyObject *self, PyObject *args, PyObject *kwargs) 22587db96d56Sopenharmony_ci{ 22597db96d56Sopenharmony_ci int res; 22607db96d56Sopenharmony_ci /* first handle args, if any */ 22617db96d56Sopenharmony_ci assert(args == NULL || PyTuple_Check(args)); 22627db96d56Sopenharmony_ci Py_ssize_t len = (args != NULL) ? PyTuple_GET_SIZE(args) : 0; 22637db96d56Sopenharmony_ci if (len > 1) { 22647db96d56Sopenharmony_ci const char *msg = "update() takes at most 1 positional argument (%zd given)"; 22657db96d56Sopenharmony_ci PyErr_Format(PyExc_TypeError, msg, len); 22667db96d56Sopenharmony_ci return NULL; 22677db96d56Sopenharmony_ci } 22687db96d56Sopenharmony_ci 22697db96d56Sopenharmony_ci if (len) { 22707db96d56Sopenharmony_ci PyObject *other = PyTuple_GET_ITEM(args, 0); /* borrowed reference */ 22717db96d56Sopenharmony_ci assert(other != NULL); 22727db96d56Sopenharmony_ci Py_INCREF(other); 22737db96d56Sopenharmony_ci res = mutablemapping_update_arg(self, other); 22747db96d56Sopenharmony_ci Py_DECREF(other); 22757db96d56Sopenharmony_ci if (res < 0) { 22767db96d56Sopenharmony_ci return NULL; 22777db96d56Sopenharmony_ci } 22787db96d56Sopenharmony_ci } 22797db96d56Sopenharmony_ci 22807db96d56Sopenharmony_ci /* now handle kwargs */ 22817db96d56Sopenharmony_ci assert(kwargs == NULL || PyDict_Check(kwargs)); 22827db96d56Sopenharmony_ci if (kwargs != NULL && PyDict_GET_SIZE(kwargs)) { 22837db96d56Sopenharmony_ci PyObject *items = PyDict_Items(kwargs); 22847db96d56Sopenharmony_ci if (items == NULL) 22857db96d56Sopenharmony_ci return NULL; 22867db96d56Sopenharmony_ci res = mutablemapping_add_pairs(self, items); 22877db96d56Sopenharmony_ci Py_DECREF(items); 22887db96d56Sopenharmony_ci if (res == -1) 22897db96d56Sopenharmony_ci return NULL; 22907db96d56Sopenharmony_ci } 22917db96d56Sopenharmony_ci 22927db96d56Sopenharmony_ci Py_RETURN_NONE; 22937db96d56Sopenharmony_ci} 2294