blob: f038ee3fdeb7dd6d07f58bf32422f503a206bcab [file] [log] [blame]
Raymond Hettingerc991db22005-08-11 07:58:45 +00001
Raymond Hettingera9d99362005-08-05 00:01:15 +00002/* set object implementation
3 Written and maintained by Raymond D. Hettinger <python@rcn.com>
4 Derived from Lib/sets.py and Objects/dictobject.c.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00005
Martin v. Löwis72d20672006-04-11 09:04:12 +00006 Copyright (c) 2003-6 Python Software Foundation.
Raymond Hettingera9d99362005-08-05 00:01:15 +00007 All rights reserved.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00008*/
9
Raymond Hettingera690a992003-11-16 16:17:49 +000010#include "Python.h"
Raymond Hettingera9d99362005-08-05 00:01:15 +000011#include "structmember.h"
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000012
Raymond Hettinger775ebe22006-12-08 18:12:24 +000013/* Set a key error with the specified argument, wrapping it in a
14 * tuple automatically so that tuple keys are not unpacked as the
15 * exception arguments. */
16static void
17set_key_error(PyObject *arg)
18{
19 PyObject *tup;
20 tup = PyTuple_Pack(1, arg);
21 if (!tup)
22 return; /* caller will expect error to be set anyway */
23 PyErr_SetObject(PyExc_KeyError, tup);
24 Py_DECREF(tup);
25}
26
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000027/* This must be >= 1. */
28#define PERTURB_SHIFT 5
29
30/* Object used as dummy key to fill deleted entries */
Raymond Hettingera9d99362005-08-05 00:01:15 +000031static PyObject *dummy = NULL; /* Initialized by first call to make_new_set() */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000032
Armin Rigoe1709372006-04-12 17:06:05 +000033#ifdef Py_REF_DEBUG
34PyObject *
35_PySet_Dummy(void)
36{
37 return dummy;
38}
39#endif
40
Raymond Hettingerbc841a12005-08-07 13:02:53 +000041#define INIT_NONZERO_SET_SLOTS(so) do { \
42 (so)->table = (so)->smalltable; \
43 (so)->mask = PySet_MINSIZE - 1; \
44 (so)->hash = -1; \
45 } while(0)
46
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000047#define EMPTY_TO_MINSIZE(so) do { \
48 memset((so)->smalltable, 0, sizeof((so)->smalltable)); \
49 (so)->used = (so)->fill = 0; \
Raymond Hettingerbc841a12005-08-07 13:02:53 +000050 INIT_NONZERO_SET_SLOTS(so); \
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000051 } while(0)
52
Raymond Hettingerbc841a12005-08-07 13:02:53 +000053/* Reuse scheme to save calls to malloc, free, and memset */
54#define MAXFREESETS 80
55static PySetObject *free_sets[MAXFREESETS];
56static int num_free_sets = 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000057
58/*
59The basic lookup function used by all operations.
60This is based on Algorithm D from Knuth Vol. 3, Sec. 6.4.
61Open addressing is preferred over chaining since the link overhead for
62chaining would be substantial (100% with typical malloc overhead).
63
64The initial probe index is computed as hash mod the table size. Subsequent
Raymond Hettingerbc841a12005-08-07 13:02:53 +000065probe indices are computed as explained in Objects/dictobject.c.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000066
67All arithmetic on hash should ignore overflow.
68
Raymond Hettinger9bda1d62005-09-16 07:14:21 +000069Unlike the dictionary implementation, the lookkey functions can return
70NULL if the rich comparison returns an error.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000071*/
72
73static setentry *
74set_lookkey(PySetObject *so, PyObject *key, register long hash)
75{
Martin v. Löwis18e16552006-02-15 17:27:45 +000076 register Py_ssize_t i;
77 register size_t perturb;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000078 register setentry *freeslot;
Neal Norwitz0f2783c2006-06-19 05:40:44 +000079 register size_t mask = so->mask;
Raymond Hettingera580c472005-08-05 17:19:54 +000080 setentry *table = so->table;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +000081 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000082 register int cmp;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000083 PyObject *startkey;
84
85 i = hash & mask;
Raymond Hettingera580c472005-08-05 17:19:54 +000086 entry = &table[i];
Raymond Hettinger06d8cf82005-07-31 15:36:06 +000087 if (entry->key == NULL || entry->key == key)
88 return entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000089
Raymond Hettinger06d8cf82005-07-31 15:36:06 +000090 if (entry->key == dummy)
91 freeslot = entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000092 else {
Raymond Hettinger06d8cf82005-07-31 15:36:06 +000093 if (entry->hash == hash) {
Raymond Hettinger06d8cf82005-07-31 15:36:06 +000094 startkey = entry->key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000095 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
96 if (cmp < 0)
Raymond Hettinger9bda1d62005-09-16 07:14:21 +000097 return NULL;
Raymond Hettingera580c472005-08-05 17:19:54 +000098 if (table == so->table && entry->key == startkey) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000099 if (cmp > 0)
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000100 return entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000101 }
102 else {
103 /* The compare did major nasty stuff to the
104 * set: start over.
105 */
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000106 return set_lookkey(so, key, hash);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000107 }
108 }
109 freeslot = NULL;
110 }
111
112 /* In the loop, key == dummy is by far (factor of 100s) the
113 least likely outcome, so test for that last. */
114 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
115 i = (i << 2) + i + perturb + 1;
Raymond Hettingera580c472005-08-05 17:19:54 +0000116 entry = &table[i & mask];
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000117 if (entry->key == NULL) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000118 if (freeslot != NULL)
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000119 entry = freeslot;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000120 break;
121 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000122 if (entry->key == key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000123 break;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000124 if (entry->hash == hash && entry->key != dummy) {
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000125 startkey = entry->key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000126 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
127 if (cmp < 0)
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000128 return NULL;
Raymond Hettingera580c472005-08-05 17:19:54 +0000129 if (table == so->table && entry->key == startkey) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000130 if (cmp > 0)
131 break;
132 }
133 else {
134 /* The compare did major nasty stuff to the
135 * set: start over.
136 */
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000137 return set_lookkey(so, key, hash);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000138 }
139 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000140 else if (entry->key == dummy && freeslot == NULL)
141 freeslot = entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000142 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000143 return entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000144}
145
146/*
147 * Hacked up version of set_lookkey which can assume keys are always strings;
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000148 * This means we can always use _PyString_Eq directly and not have to check to
149 * see if the comparison altered the table.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000150 */
151static setentry *
152set_lookkey_string(PySetObject *so, PyObject *key, register long hash)
153{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000154 register Py_ssize_t i;
155 register size_t perturb;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000156 register setentry *freeslot;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000157 register size_t mask = so->mask;
Raymond Hettingera580c472005-08-05 17:19:54 +0000158 setentry *table = so->table;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000159 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000160
161 /* Make sure this function doesn't have to handle non-string keys,
162 including subclasses of str; e.g., one reason to subclass
163 strings is to override __eq__, and for speed we don't cater to
164 that here. */
165 if (!PyString_CheckExact(key)) {
166 so->lookup = set_lookkey;
167 return set_lookkey(so, key, hash);
168 }
169 i = hash & mask;
Raymond Hettingera580c472005-08-05 17:19:54 +0000170 entry = &table[i];
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000171 if (entry->key == NULL || entry->key == key)
172 return entry;
Raymond Hettingered6c1ef2005-08-13 08:28:03 +0000173 if (entry->key == dummy)
174 freeslot = entry;
175 else {
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000176 if (entry->hash == hash && _PyString_Eq(entry->key, key))
177 return entry;
Raymond Hettingered6c1ef2005-08-13 08:28:03 +0000178 freeslot = NULL;
179 }
180
181 /* In the loop, key == dummy is by far (factor of 100s) the
182 least likely outcome, so test for that last. */
183 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
184 i = (i << 2) + i + perturb + 1;
185 entry = &table[i & mask];
186 if (entry->key == NULL)
187 return freeslot == NULL ? entry : freeslot;
188 if (entry->key == key
189 || (entry->hash == hash
190 && entry->key != dummy
191 && _PyString_Eq(entry->key, key)))
192 return entry;
193 if (entry->key == dummy && freeslot == NULL)
194 freeslot = entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000195 }
Neal Norwitz7e3ec042006-10-28 21:37:16 +0000196 assert(0); /* NOT REACHED */
197 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000198}
199
200/*
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000201Internal routine to insert a new key into the table.
Raymond Hettinger775ebe22006-12-08 18:12:24 +0000202Used by the public insert routine.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000203Eats a reference to key.
204*/
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000205static int
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000206set_insert_key(register PySetObject *so, PyObject *key, long hash)
207{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000208 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000209 typedef setentry *(*lookupfunc)(PySetObject *, PyObject *, long);
210
211 assert(so->lookup != NULL);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000212 entry = so->lookup(so, key, hash);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000213 if (entry == NULL)
214 return -1;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000215 if (entry->key == NULL) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000216 /* UNUSED */
217 so->fill++;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000218 entry->key = key;
219 entry->hash = hash;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000220 so->used++;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000221 } else if (entry->key == dummy) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000222 /* DUMMY */
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000223 entry->key = key;
224 entry->hash = hash;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000225 so->used++;
226 Py_DECREF(dummy);
227 } else {
228 /* ACTIVE */
229 Py_DECREF(key);
230 }
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000231 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000232}
233
234/*
Raymond Hettinger775ebe22006-12-08 18:12:24 +0000235Internal routine used by set_table_resize() to insert an item which is
236known to be absent from the set. This routine also assumes that
237the set contains no deleted entries. Besides the performance benefit,
238using set_insert_clean() in set_table_resize() is dangerous (SF bug #1456209).
239Note that no refcounts are changed by this routine; if needed, the caller
240is responsible for incref'ing `key`.
241*/
242static void
243set_insert_clean(register PySetObject *so, PyObject *key, long hash)
244{
245 register size_t i;
246 register size_t perturb;
247 register size_t mask = (size_t)so->mask;
248 setentry *table = so->table;
249 register setentry *entry;
250
251 i = hash & mask;
252 entry = &table[i];
253 for (perturb = hash; entry->key != NULL; perturb >>= PERTURB_SHIFT) {
254 i = (i << 2) + i + perturb + 1;
255 entry = &table[i & mask];
256 }
257 so->fill++;
258 entry->key = key;
259 entry->hash = hash;
260 so->used++;
261}
262
263/*
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000264Restructure the table by allocating a new table and reinserting all
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000265keys again. When entries have been deleted, the new table may
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000266actually be smaller than the old one.
267*/
268static int
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000269set_table_resize(PySetObject *so, Py_ssize_t minused)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000270{
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000271 Py_ssize_t newsize;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000272 setentry *oldtable, *newtable, *entry;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000273 Py_ssize_t i;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000274 int is_oldtable_malloced;
275 setentry small_copy[PySet_MINSIZE];
276
277 assert(minused >= 0);
278
279 /* Find the smallest table size > minused. */
280 for (newsize = PySet_MINSIZE;
281 newsize <= minused && newsize > 0;
282 newsize <<= 1)
283 ;
284 if (newsize <= 0) {
285 PyErr_NoMemory();
286 return -1;
287 }
288
289 /* Get space for a new table. */
290 oldtable = so->table;
291 assert(oldtable != NULL);
292 is_oldtable_malloced = oldtable != so->smalltable;
293
294 if (newsize == PySet_MINSIZE) {
295 /* A large table is shrinking, or we can't get any smaller. */
296 newtable = so->smalltable;
297 if (newtable == oldtable) {
298 if (so->fill == so->used) {
299 /* No dummies, so no point doing anything. */
300 return 0;
301 }
302 /* We're not going to resize it, but rebuild the
303 table anyway to purge old dummy entries.
304 Subtle: This is *necessary* if fill==size,
305 as set_lookkey needs at least one virgin slot to
306 terminate failing searches. If fill < size, it's
307 merely desirable, as dummies slow searches. */
308 assert(so->fill > so->used);
309 memcpy(small_copy, oldtable, sizeof(small_copy));
310 oldtable = small_copy;
311 }
312 }
313 else {
314 newtable = PyMem_NEW(setentry, newsize);
315 if (newtable == NULL) {
316 PyErr_NoMemory();
317 return -1;
318 }
319 }
320
321 /* Make the set empty, using the new table. */
322 assert(newtable != oldtable);
323 so->table = newtable;
324 so->mask = newsize - 1;
325 memset(newtable, 0, sizeof(setentry) * newsize);
326 so->used = 0;
327 i = so->fill;
328 so->fill = 0;
329
330 /* Copy the data over; this is refcount-neutral for active entries;
331 dummy entries aren't copied over, of course */
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000332 for (entry = oldtable; i > 0; entry++) {
333 if (entry->key == NULL) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000334 /* UNUSED */
335 ;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000336 } else if (entry->key == dummy) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000337 /* DUMMY */
338 --i;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000339 assert(entry->key == dummy);
340 Py_DECREF(entry->key);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000341 } else {
342 /* ACTIVE */
343 --i;
Raymond Hettinger775ebe22006-12-08 18:12:24 +0000344 set_insert_clean(so, entry->key, entry->hash);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000345 }
346 }
347
348 if (is_oldtable_malloced)
349 PyMem_DEL(oldtable);
350 return 0;
351}
352
Raymond Hettingerc991db22005-08-11 07:58:45 +0000353/* CAUTION: set_add_key/entry() must guarantee it won't resize the table */
354
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000355static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000356set_add_entry(register PySetObject *so, setentry *entry)
357{
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000358 register Py_ssize_t n_used;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000359
360 assert(so->fill <= so->mask); /* at least one empty slot */
361 n_used = so->used;
362 Py_INCREF(entry->key);
Georg Brandl8de403a2006-09-08 06:02:26 +0000363 if (set_insert_key(so, entry->key, entry->hash) == -1) {
364 Py_DECREF(entry->key);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000365 return -1;
Georg Brandl8de403a2006-09-08 06:02:26 +0000366 }
Raymond Hettingerc991db22005-08-11 07:58:45 +0000367 if (!(so->used > n_used && so->fill*3 >= (so->mask+1)*2))
368 return 0;
369 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
370}
371
372static int
373set_add_key(register PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000374{
375 register long hash;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000376 register Py_ssize_t n_used;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000377
Raymond Hettingerc991db22005-08-11 07:58:45 +0000378 if (!PyString_CheckExact(key) ||
379 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000380 hash = PyObject_Hash(key);
381 if (hash == -1)
382 return -1;
383 }
384 assert(so->fill <= so->mask); /* at least one empty slot */
385 n_used = so->used;
386 Py_INCREF(key);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000387 if (set_insert_key(so, key, hash) == -1) {
388 Py_DECREF(key);
389 return -1;
390 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000391 if (!(so->used > n_used && so->fill*3 >= (so->mask+1)*2))
392 return 0;
Raymond Hettingerbc841a12005-08-07 13:02:53 +0000393 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000394}
395
396#define DISCARD_NOTFOUND 0
397#define DISCARD_FOUND 1
398
399static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000400set_discard_entry(PySetObject *so, setentry *oldentry)
401{ register setentry *entry;
402 PyObject *old_key;
403
404 entry = (so->lookup)(so, oldentry->key, oldentry->hash);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000405 if (entry == NULL)
406 return -1;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000407 if (entry->key == NULL || entry->key == dummy)
408 return DISCARD_NOTFOUND;
409 old_key = entry->key;
410 Py_INCREF(dummy);
411 entry->key = dummy;
412 so->used--;
413 Py_DECREF(old_key);
414 return DISCARD_FOUND;
415}
416
417static int
418set_discard_key(PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000419{
420 register long hash;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000421 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000422 PyObject *old_key;
423
424 assert (PyAnySet_Check(so));
425 if (!PyString_CheckExact(key) ||
426 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
427 hash = PyObject_Hash(key);
428 if (hash == -1)
429 return -1;
430 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000431 entry = (so->lookup)(so, key, hash);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000432 if (entry == NULL)
433 return -1;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000434 if (entry->key == NULL || entry->key == dummy)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000435 return DISCARD_NOTFOUND;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000436 old_key = entry->key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000437 Py_INCREF(dummy);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000438 entry->key = dummy;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000439 so->used--;
440 Py_DECREF(old_key);
441 return DISCARD_FOUND;
442}
443
Raymond Hettingerfe889f32005-08-06 05:43:39 +0000444static int
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000445set_clear_internal(PySetObject *so)
446{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000447 setentry *entry, *table;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000448 int table_is_malloced;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000449 Py_ssize_t fill;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000450 setentry small_copy[PySet_MINSIZE];
451#ifdef Py_DEBUG
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000452 Py_ssize_t i, n;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000453 assert (PyAnySet_Check(so));
Raymond Hettingera580c472005-08-05 17:19:54 +0000454
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000455 n = so->mask + 1;
456 i = 0;
457#endif
458
459 table = so->table;
460 assert(table != NULL);
461 table_is_malloced = table != so->smalltable;
462
463 /* This is delicate. During the process of clearing the set,
464 * decrefs can cause the set to mutate. To avoid fatal confusion
465 * (voice of experience), we have to make the set empty before
Raymond Hettingerfe889f32005-08-06 05:43:39 +0000466 * clearing the slots, and never refer to anything via so->ref while
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000467 * clearing.
468 */
469 fill = so->fill;
470 if (table_is_malloced)
471 EMPTY_TO_MINSIZE(so);
472
473 else if (fill > 0) {
474 /* It's a small table with something that needs to be cleared.
475 * Afraid the only safe way is to copy the set entries into
476 * another small table first.
477 */
478 memcpy(small_copy, table, sizeof(small_copy));
479 table = small_copy;
480 EMPTY_TO_MINSIZE(so);
481 }
482 /* else it's a small table that's already empty */
483
484 /* Now we can finally clear things. If C had refcounts, we could
485 * assert that the refcount on table is 1 now, i.e. that this function
486 * has unique access to it, so decref side-effects can't alter it.
487 */
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000488 for (entry = table; fill > 0; ++entry) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000489#ifdef Py_DEBUG
490 assert(i < n);
491 ++i;
492#endif
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000493 if (entry->key) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000494 --fill;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000495 Py_DECREF(entry->key);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000496 }
497#ifdef Py_DEBUG
498 else
Raymond Hettinger334b5b22006-03-26 03:11:29 +0000499 assert(entry->key == NULL);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000500#endif
501 }
502
503 if (table_is_malloced)
504 PyMem_DEL(table);
Raymond Hettingerfe889f32005-08-06 05:43:39 +0000505 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000506}
507
508/*
509 * Iterate over a set table. Use like so:
510 *
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000511 * Py_ssize_t pos;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000512 * setentry *entry;
Raymond Hettingerd7946662005-08-01 21:39:29 +0000513 * pos = 0; # important! pos should not otherwise be changed by you
Raymond Hettingerc991db22005-08-11 07:58:45 +0000514 * while (set_next(yourset, &pos, &entry)) {
515 * Refer to borrowed reference in entry->key.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000516 * }
517 *
Raymond Hettingerc991db22005-08-11 07:58:45 +0000518 * CAUTION: In general, it isn't safe to use set_next in a loop that
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000519 * mutates the table.
520 */
521static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000522set_next(PySetObject *so, Py_ssize_t *pos_ptr, setentry **entry_ptr)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000523{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000524 Py_ssize_t i;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000525 Py_ssize_t mask;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000526 register setentry *table;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000527
528 assert (PyAnySet_Check(so));
Raymond Hettingerc991db22005-08-11 07:58:45 +0000529 i = *pos_ptr;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000530 assert(i >= 0);
Raymond Hettingerc991db22005-08-11 07:58:45 +0000531 table = so->table;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000532 mask = so->mask;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000533 while (i <= mask && (table[i].key == NULL || table[i].key == dummy))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000534 i++;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000535 *pos_ptr = i+1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000536 if (i > mask)
537 return 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000538 assert(table[i].key != NULL);
539 *entry_ptr = &table[i];
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000540 return 1;
541}
542
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000543static void
544set_dealloc(PySetObject *so)
545{
546 register setentry *entry;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000547 Py_ssize_t fill = so->fill;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000548 PyObject_GC_UnTrack(so);
549 Py_TRASHCAN_SAFE_BEGIN(so)
550 if (so->weakreflist != NULL)
551 PyObject_ClearWeakRefs((PyObject *) so);
552
553 for (entry = so->table; fill > 0; entry++) {
554 if (entry->key) {
555 --fill;
556 Py_DECREF(entry->key);
557 }
558 }
559 if (so->table != so->smalltable)
560 PyMem_DEL(so->table);
561 if (num_free_sets < MAXFREESETS && PyAnySet_CheckExact(so))
562 free_sets[num_free_sets++] = so;
563 else
564 so->ob_type->tp_free(so);
565 Py_TRASHCAN_SAFE_END(so)
566}
567
568static int
569set_tp_print(PySetObject *so, FILE *fp, int flags)
570{
571 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000572 Py_ssize_t pos=0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000573 char *emit = ""; /* No separator emitted on first pass */
574 char *separator = ", ";
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000575 int status = Py_ReprEnter((PyObject*)so);
576
577 if (status != 0) {
578 if (status < 0)
579 return status;
580 fprintf(fp, "%s(...)", so->ob_type->tp_name);
581 return 0;
582 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000583
584 fprintf(fp, "%s([", so->ob_type->tp_name);
585 while (set_next(so, &pos, &entry)) {
586 fputs(emit, fp);
587 emit = separator;
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000588 if (PyObject_Print(entry->key, fp, 0) != 0) {
589 Py_ReprLeave((PyObject*)so);
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000590 return -1;
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000591 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000592 }
593 fputs("])", fp);
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000594 Py_ReprLeave((PyObject*)so);
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000595 return 0;
596}
597
598static PyObject *
599set_repr(PySetObject *so)
600{
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000601 PyObject *keys, *result=NULL, *listrepr;
602 int status = Py_ReprEnter((PyObject*)so);
603
604 if (status != 0) {
605 if (status < 0)
606 return NULL;
607 return PyString_FromFormat("%s(...)", so->ob_type->tp_name);
608 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000609
610 keys = PySequence_List((PyObject *)so);
611 if (keys == NULL)
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000612 goto done;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000613 listrepr = PyObject_Repr(keys);
614 Py_DECREF(keys);
615 if (listrepr == NULL)
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000616 goto done;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000617
618 result = PyString_FromFormat("%s(%s)", so->ob_type->tp_name,
619 PyString_AS_STRING(listrepr));
620 Py_DECREF(listrepr);
Raymond Hettinger9cdf7032006-12-29 18:49:13 +0000621done:
622 Py_ReprLeave((PyObject*)so);
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000623 return result;
624}
625
Martin v. Löwis18e16552006-02-15 17:27:45 +0000626static Py_ssize_t
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000627set_len(PyObject *so)
628{
629 return ((PySetObject *)so)->used;
630}
631
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000632static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000633set_merge(PySetObject *so, PyObject *otherset)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000634{
Raymond Hettingerd7946662005-08-01 21:39:29 +0000635 PySetObject *other;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000636 register Py_ssize_t i;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000637 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000638
639 assert (PyAnySet_Check(so));
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000640 assert (PyAnySet_Check(otherset));
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000641
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000642 other = (PySetObject*)otherset;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000643 if (other == so || other->used == 0)
644 /* a.update(a) or a.update({}); nothing to do */
645 return 0;
646 /* Do one big resize at the start, rather than
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000647 * incrementally resizing as we insert new keys. Expect
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000648 * that there will be no (or few) overlapping keys.
649 */
650 if ((so->fill + other->used)*3 >= (so->mask+1)*2) {
651 if (set_table_resize(so, (so->used + other->used)*2) != 0)
652 return -1;
653 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000654 for (i = 0; i <= other->mask; i++) {
655 entry = &other->table[i];
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000656 if (entry->key != NULL &&
657 entry->key != dummy) {
658 Py_INCREF(entry->key);
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000659 if (set_insert_key(so, entry->key, entry->hash) == -1) {
660 Py_DECREF(entry->key);
661 return -1;
662 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000663 }
664 }
665 return 0;
666}
667
668static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000669set_contains_key(PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000670{
671 long hash;
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000672 setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000673
674 if (!PyString_CheckExact(key) ||
675 (hash = ((PyStringObject *) key)->ob_shash) == -1) {
676 hash = PyObject_Hash(key);
677 if (hash == -1)
678 return -1;
679 }
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000680 entry = (so->lookup)(so, key, hash);
681 if (entry == NULL)
682 return -1;
683 key = entry->key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000684 return key != NULL && key != dummy;
685}
686
Raymond Hettingerc991db22005-08-11 07:58:45 +0000687static int
688set_contains_entry(PySetObject *so, setentry *entry)
689{
690 PyObject *key;
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000691 setentry *lu_entry;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000692
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000693 lu_entry = (so->lookup)(so, entry->key, entry->hash);
694 if (lu_entry == NULL)
695 return -1;
696 key = lu_entry->key;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000697 return key != NULL && key != dummy;
698}
699
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000700static PyObject *
701set_pop(PySetObject *so)
702{
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000703 register Py_ssize_t i = 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000704 register setentry *entry;
705 PyObject *key;
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000706
707 assert (PyAnySet_Check(so));
708 if (so->used == 0) {
709 PyErr_SetString(PyExc_KeyError, "pop from an empty set");
710 return NULL;
711 }
712
713 /* Set entry to "the first" unused or dummy set entry. We abuse
714 * the hash field of slot 0 to hold a search finger:
715 * If slot 0 has a value, use slot 0.
716 * Else slot 0 is being used to hold a search finger,
717 * and we use its hash value as the first index to look.
718 */
719 entry = &so->table[0];
720 if (entry->key == NULL || entry->key == dummy) {
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000721 i = entry->hash;
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000722 /* The hash field may be a real hash value, or it may be a
723 * legit search finger, or it may be a once-legit search
724 * finger that's out of bounds now because it wrapped around
725 * or the table shrunk -- simply make sure it's in bounds now.
726 */
727 if (i > so->mask || i < 1)
728 i = 1; /* skip slot 0 */
729 while ((entry = &so->table[i])->key == NULL || entry->key==dummy) {
730 i++;
731 if (i > so->mask)
732 i = 1;
733 }
734 }
735 key = entry->key;
736 Py_INCREF(dummy);
737 entry->key = dummy;
738 so->used--;
739 so->table[0].hash = i + 1; /* next place to start */
740 return key;
741}
742
743PyDoc_STRVAR(pop_doc, "Remove and return an arbitrary set element.");
744
745static int
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000746set_traverse(PySetObject *so, visitproc visit, void *arg)
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000747{
Martin v. Löwis18e16552006-02-15 17:27:45 +0000748 Py_ssize_t pos = 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000749 setentry *entry;
750
751 while (set_next(so, &pos, &entry))
752 Py_VISIT(entry->key);
753 return 0;
754}
755
756static long
757frozenset_hash(PyObject *self)
758{
759 PySetObject *so = (PySetObject *)self;
760 long h, hash = 1927868237L;
761 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000762 Py_ssize_t pos = 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000763
764 if (so->hash != -1)
765 return so->hash;
766
767 hash *= PySet_GET_SIZE(self) + 1;
768 while (set_next(so, &pos, &entry)) {
769 /* Work to increase the bit dispersion for closely spaced hash
770 values. The is important because some use cases have many
771 combinations of a small number of elements with nearby
772 hashes so that many distinct combinations collapse to only
773 a handful of distinct hash values. */
774 h = entry->hash;
775 hash ^= (h ^ (h << 16) ^ 89869747L) * 3644798167u;
776 }
777 hash = hash * 69069L + 907133923L;
778 if (hash == -1)
779 hash = 590923713L;
780 so->hash = hash;
781 return hash;
782}
783
784static long
785set_nohash(PyObject *self)
786{
787 PyErr_SetString(PyExc_TypeError, "set objects are unhashable");
788 return -1;
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000789}
790
Raymond Hettingera9d99362005-08-05 00:01:15 +0000791/***** Set iterator type ***********************************************/
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000792
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000793typedef struct {
794 PyObject_HEAD
795 PySetObject *si_set; /* Set to NULL when iterator is exhausted */
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000796 Py_ssize_t si_used;
797 Py_ssize_t si_pos;
798 Py_ssize_t len;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000799} setiterobject;
800
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000801static void
802setiter_dealloc(setiterobject *si)
803{
804 Py_XDECREF(si->si_set);
805 PyObject_Del(si);
806}
807
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000808static PyObject *
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000809setiter_len(setiterobject *si)
810{
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000811 Py_ssize_t len = 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000812 if (si->si_set != NULL && si->si_used == si->si_set->used)
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000813 len = si->len;
814 return PyInt_FromLong(len);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000815}
816
Armin Rigof5b3e362006-02-11 21:32:43 +0000817PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000818
819static PyMethodDef setiter_methods[] = {
Armin Rigof5b3e362006-02-11 21:32:43 +0000820 {"__length_hint__", (PyCFunction)setiter_len, METH_NOARGS, length_hint_doc},
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000821 {NULL, NULL} /* sentinel */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000822};
823
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000824static PyObject *setiter_iternext(setiterobject *si)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000825{
826 PyObject *key;
Neal Norwitz0f2783c2006-06-19 05:40:44 +0000827 register Py_ssize_t i, mask;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000828 register setentry *entry;
829 PySetObject *so = si->si_set;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000830
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000831 if (so == NULL)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000832 return NULL;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000833 assert (PyAnySet_Check(so));
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000834
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000835 if (si->si_used != so->used) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000836 PyErr_SetString(PyExc_RuntimeError,
837 "Set changed size during iteration");
838 si->si_used = -1; /* Make this state sticky */
839 return NULL;
840 }
841
842 i = si->si_pos;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000843 assert(i>=0);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000844 entry = so->table;
845 mask = so->mask;
846 while (i <= mask && (entry[i].key == NULL || entry[i].key == dummy))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000847 i++;
848 si->si_pos = i+1;
849 if (i > mask)
850 goto fail;
851 si->len--;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000852 key = entry[i].key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000853 Py_INCREF(key);
854 return key;
855
856fail:
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000857 Py_DECREF(so);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000858 si->si_set = NULL;
859 return NULL;
860}
861
Hye-Shik Change2956762005-08-01 05:26:41 +0000862static PyTypeObject PySetIter_Type = {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000863 PyObject_HEAD_INIT(&PyType_Type)
864 0, /* ob_size */
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000865 "setiterator", /* tp_name */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000866 sizeof(setiterobject), /* tp_basicsize */
867 0, /* tp_itemsize */
868 /* methods */
869 (destructor)setiter_dealloc, /* tp_dealloc */
870 0, /* tp_print */
871 0, /* tp_getattr */
872 0, /* tp_setattr */
873 0, /* tp_compare */
874 0, /* tp_repr */
875 0, /* tp_as_number */
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000876 0, /* tp_as_sequence */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000877 0, /* tp_as_mapping */
878 0, /* tp_hash */
879 0, /* tp_call */
880 0, /* tp_str */
881 PyObject_GenericGetAttr, /* tp_getattro */
882 0, /* tp_setattro */
883 0, /* tp_as_buffer */
884 Py_TPFLAGS_DEFAULT, /* tp_flags */
885 0, /* tp_doc */
886 0, /* tp_traverse */
887 0, /* tp_clear */
888 0, /* tp_richcompare */
889 0, /* tp_weaklistoffset */
890 PyObject_SelfIter, /* tp_iter */
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000891 (iternextfunc)setiter_iternext, /* tp_iternext */
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000892 setiter_methods, /* tp_methods */
893 0,
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000894};
895
Martin v. Löwis72d20672006-04-11 09:04:12 +0000896static PyObject *
897set_iter(PySetObject *so)
898{
899 setiterobject *si = PyObject_New(setiterobject, &PySetIter_Type);
900 if (si == NULL)
901 return NULL;
902 Py_INCREF(so);
903 si->si_set = so;
904 si->si_used = so->used;
905 si->si_pos = 0;
906 si->len = so->used;
907 return (PyObject *)si;
908}
909
Raymond Hettingerd7946662005-08-01 21:39:29 +0000910static int
Raymond Hettingerd7946662005-08-01 21:39:29 +0000911set_update_internal(PySetObject *so, PyObject *other)
Raymond Hettingera690a992003-11-16 16:17:49 +0000912{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000913 PyObject *key, *it;
Raymond Hettingera690a992003-11-16 16:17:49 +0000914
Raymond Hettingerd7946662005-08-01 21:39:29 +0000915 if (PyAnySet_Check(other))
Raymond Hettingerc991db22005-08-11 07:58:45 +0000916 return set_merge(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +0000917
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000918 if (PyDict_Check(other)) {
Neal Norwitz0c6e2f12006-01-08 06:13:44 +0000919 PyObject *value;
Martin v. Löwis18e16552006-02-15 17:27:45 +0000920 Py_ssize_t pos = 0;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000921 while (PyDict_Next(other, &pos, &key, &value)) {
Raymond Hettingerc991db22005-08-11 07:58:45 +0000922 if (set_add_key(so, key) == -1)
Raymond Hettingerd7946662005-08-01 21:39:29 +0000923 return -1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000924 }
Raymond Hettingerd7946662005-08-01 21:39:29 +0000925 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000926 }
927
Raymond Hettingera38123e2003-11-24 22:18:49 +0000928 it = PyObject_GetIter(other);
929 if (it == NULL)
Raymond Hettingerd7946662005-08-01 21:39:29 +0000930 return -1;
Raymond Hettingera690a992003-11-16 16:17:49 +0000931
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000932 while ((key = PyIter_Next(it)) != NULL) {
Raymond Hettingerc991db22005-08-11 07:58:45 +0000933 if (set_add_key(so, key) == -1) {
Raymond Hettingera38123e2003-11-24 22:18:49 +0000934 Py_DECREF(it);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000935 Py_DECREF(key);
Raymond Hettingerd7946662005-08-01 21:39:29 +0000936 return -1;
Raymond Hettingera690a992003-11-16 16:17:49 +0000937 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000938 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +0000939 }
Raymond Hettingera38123e2003-11-24 22:18:49 +0000940 Py_DECREF(it);
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +0000941 if (PyErr_Occurred())
Raymond Hettingerd7946662005-08-01 21:39:29 +0000942 return -1;
943 return 0;
944}
945
946static PyObject *
947set_update(PySetObject *so, PyObject *other)
948{
949 if (set_update_internal(so, other) == -1)
Raymond Hettingera38123e2003-11-24 22:18:49 +0000950 return NULL;
951 Py_RETURN_NONE;
952}
953
954PyDoc_STRVAR(update_doc,
955"Update a set with the union of itself and another.");
956
957static PyObject *
958make_new_set(PyTypeObject *type, PyObject *iterable)
959{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000960 register PySetObject *so = NULL;
Raymond Hettingera38123e2003-11-24 22:18:49 +0000961
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000962 if (dummy == NULL) { /* Auto-initialize dummy */
963 dummy = PyString_FromString("<dummy key>");
964 if (dummy == NULL)
965 return NULL;
966 }
Raymond Hettingera690a992003-11-16 16:17:49 +0000967
968 /* create PySetObject structure */
Raymond Hettingerbc841a12005-08-07 13:02:53 +0000969 if (num_free_sets &&
970 (type == &PySet_Type || type == &PyFrozenSet_Type)) {
971 so = free_sets[--num_free_sets];
972 assert (so != NULL && PyAnySet_CheckExact(so));
973 so->ob_type = type;
974 _Py_NewReference((PyObject *)so);
975 EMPTY_TO_MINSIZE(so);
976 PyObject_GC_Track(so);
977 } else {
978 so = (PySetObject *)type->tp_alloc(type, 0);
979 if (so == NULL)
980 return NULL;
981 /* tp_alloc has already zeroed the structure */
982 assert(so->table == NULL && so->fill == 0 && so->used == 0);
983 INIT_NONZERO_SET_SLOTS(so);
984 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000985
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000986 so->lookup = set_lookkey_string;
Raymond Hettinger691d8052004-05-30 07:26:47 +0000987 so->weakreflist = NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +0000988
Raymond Hettingera38123e2003-11-24 22:18:49 +0000989 if (iterable != NULL) {
Raymond Hettingerd7946662005-08-01 21:39:29 +0000990 if (set_update_internal(so, iterable) == -1) {
Raymond Hettingera38123e2003-11-24 22:18:49 +0000991 Py_DECREF(so);
992 return NULL;
993 }
Raymond Hettingera38123e2003-11-24 22:18:49 +0000994 }
995
Raymond Hettingera690a992003-11-16 16:17:49 +0000996 return (PyObject *)so;
997}
998
Raymond Hettingerd7946662005-08-01 21:39:29 +0000999/* The empty frozenset is a singleton */
1000static PyObject *emptyfrozenset = NULL;
1001
Raymond Hettingera690a992003-11-16 16:17:49 +00001002static PyObject *
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001003frozenset_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Raymond Hettingera690a992003-11-16 16:17:49 +00001004{
Raymond Hettingerd7946662005-08-01 21:39:29 +00001005 PyObject *iterable = NULL, *result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001006
Raymond Hettinger535989e2007-01-11 18:21:04 +00001007 if (type == &PyFrozenSet_Type && !_PyArg_NoKeywords("frozenset()", kwds))
Georg Brandl02c42872005-08-26 06:42:30 +00001008 return NULL;
1009
Raymond Hettingera690a992003-11-16 16:17:49 +00001010 if (!PyArg_UnpackTuple(args, type->tp_name, 0, 1, &iterable))
1011 return NULL;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001012
1013 if (type != &PyFrozenSet_Type)
1014 return make_new_set(type, iterable);
1015
1016 if (iterable != NULL) {
1017 /* frozenset(f) is idempotent */
1018 if (PyFrozenSet_CheckExact(iterable)) {
1019 Py_INCREF(iterable);
1020 return iterable;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001021 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001022 result = make_new_set(type, iterable);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001023 if (result == NULL || PySet_GET_SIZE(result))
Raymond Hettingerd7946662005-08-01 21:39:29 +00001024 return result;
1025 Py_DECREF(result);
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001026 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001027 /* The empty frozenset is a singleton */
1028 if (emptyfrozenset == NULL)
1029 emptyfrozenset = make_new_set(type, NULL);
1030 Py_XINCREF(emptyfrozenset);
1031 return emptyfrozenset;
1032}
1033
1034void
1035PySet_Fini(void)
1036{
Raymond Hettingerbc841a12005-08-07 13:02:53 +00001037 PySetObject *so;
1038
1039 while (num_free_sets) {
1040 num_free_sets--;
1041 so = free_sets[num_free_sets];
1042 PyObject_GC_Del(so);
1043 }
Martin v. Löwised8f7832006-04-15 12:47:23 +00001044 Py_CLEAR(dummy);
1045 Py_CLEAR(emptyfrozenset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001046}
1047
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001048static PyObject *
1049set_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1050{
Raymond Hettinger535989e2007-01-11 18:21:04 +00001051 if (type == &PySet_Type && !_PyArg_NoKeywords("set()", kwds))
Georg Brandl02c42872005-08-26 06:42:30 +00001052 return NULL;
1053
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001054 return make_new_set(type, NULL);
1055}
1056
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001057/* set_swap_bodies() switches the contents of any two sets by moving their
1058 internal data pointers and, if needed, copying the internal smalltables.
1059 Semantically equivalent to:
1060
1061 t=set(a); a.clear(); a.update(b); b.clear(); b.update(t); del t
1062
1063 The function always succeeds and it leaves both objects in a stable state.
1064 Useful for creating temporary frozensets from sets for membership testing
1065 in __contains__(), discard(), and remove(). Also useful for operations
1066 that update in-place (by allowing an intermediate result to be swapped
Raymond Hettinger9dcb17c2005-07-31 13:09:28 +00001067 into one of the original inputs).
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001068*/
1069
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001070static void
1071set_swap_bodies(PySetObject *a, PySetObject *b)
Raymond Hettingera690a992003-11-16 16:17:49 +00001072{
Neal Norwitz0f2783c2006-06-19 05:40:44 +00001073 Py_ssize_t t;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001074 setentry *u;
1075 setentry *(*f)(PySetObject *so, PyObject *key, long hash);
1076 setentry tab[PySet_MINSIZE];
1077 long h;
1078
1079 t = a->fill; a->fill = b->fill; b->fill = t;
1080 t = a->used; a->used = b->used; b->used = t;
1081 t = a->mask; a->mask = b->mask; b->mask = t;
1082
1083 u = a->table;
1084 if (a->table == a->smalltable)
1085 u = b->smalltable;
1086 a->table = b->table;
1087 if (b->table == b->smalltable)
1088 a->table = a->smalltable;
1089 b->table = u;
1090
1091 f = a->lookup; a->lookup = b->lookup; b->lookup = f;
1092
1093 if (a->table == a->smalltable || b->table == b->smalltable) {
1094 memcpy(tab, a->smalltable, sizeof(tab));
1095 memcpy(a->smalltable, b->smalltable, sizeof(tab));
1096 memcpy(b->smalltable, tab, sizeof(tab));
1097 }
1098
Raymond Hettingera580c472005-08-05 17:19:54 +00001099 if (PyType_IsSubtype(a->ob_type, &PyFrozenSet_Type) &&
1100 PyType_IsSubtype(b->ob_type, &PyFrozenSet_Type)) {
1101 h = a->hash; a->hash = b->hash; b->hash = h;
1102 } else {
1103 a->hash = -1;
1104 b->hash = -1;
1105 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001106}
1107
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001108static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001109set_copy(PySetObject *so)
1110{
Raymond Hettingera38123e2003-11-24 22:18:49 +00001111 return make_new_set(so->ob_type, (PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001112}
1113
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001114static PyObject *
1115frozenset_copy(PySetObject *so)
1116{
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001117 if (PyFrozenSet_CheckExact(so)) {
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001118 Py_INCREF(so);
1119 return (PyObject *)so;
1120 }
1121 return set_copy(so);
1122}
1123
Raymond Hettingera690a992003-11-16 16:17:49 +00001124PyDoc_STRVAR(copy_doc, "Return a shallow copy of a set.");
1125
1126static PyObject *
Raymond Hettingerc991db22005-08-11 07:58:45 +00001127set_clear(PySetObject *so)
1128{
1129 set_clear_internal(so);
1130 Py_RETURN_NONE;
1131}
1132
1133PyDoc_STRVAR(clear_doc, "Remove all elements from this set.");
1134
1135static PyObject *
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001136set_union(PySetObject *so, PyObject *other)
1137{
1138 PySetObject *result;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001139
1140 result = (PySetObject *)set_copy(so);
1141 if (result == NULL)
1142 return NULL;
Raymond Hettingerd8e13382005-08-17 12:27:17 +00001143 if ((PyObject *)so == other)
1144 return (PyObject *)result;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001145 if (set_update_internal(result, other) == -1) {
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001146 Py_DECREF(result);
1147 return NULL;
1148 }
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001149 return (PyObject *)result;
1150}
1151
1152PyDoc_STRVAR(union_doc,
1153 "Return the union of two sets as a new set.\n\
1154\n\
1155(i.e. all elements that are in either set.)");
1156
1157static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001158set_or(PySetObject *so, PyObject *other)
1159{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001160 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001161 Py_INCREF(Py_NotImplemented);
1162 return Py_NotImplemented;
1163 }
1164 return set_union(so, other);
1165}
1166
1167static PyObject *
1168set_ior(PySetObject *so, PyObject *other)
1169{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001170 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001171 Py_INCREF(Py_NotImplemented);
1172 return Py_NotImplemented;
1173 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001174 if (set_update_internal(so, other) == -1)
Raymond Hettingera690a992003-11-16 16:17:49 +00001175 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001176 Py_INCREF(so);
1177 return (PyObject *)so;
1178}
1179
1180static PyObject *
1181set_intersection(PySetObject *so, PyObject *other)
1182{
1183 PySetObject *result;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001184 PyObject *key, *it, *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001185
Raymond Hettingerd8e13382005-08-17 12:27:17 +00001186 if ((PyObject *)so == other)
1187 return set_copy(so);
Raymond Hettingerc991db22005-08-11 07:58:45 +00001188
Raymond Hettingera690a992003-11-16 16:17:49 +00001189 result = (PySetObject *)make_new_set(so->ob_type, NULL);
1190 if (result == NULL)
1191 return NULL;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001192
Raymond Hettingerc991db22005-08-11 07:58:45 +00001193 if (PyAnySet_Check(other)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +00001194 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001195 setentry *entry;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001196
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001197 if (PySet_GET_SIZE(other) > PySet_GET_SIZE(so)) {
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001198 tmp = (PyObject *)so;
1199 so = (PySetObject *)other;
1200 other = tmp;
1201 }
1202
Raymond Hettingerc991db22005-08-11 07:58:45 +00001203 while (set_next((PySetObject *)other, &pos, &entry)) {
Georg Brandl8de403a2006-09-08 06:02:26 +00001204 int rv = set_contains_entry(so, entry);
1205 if (rv == -1) {
1206 Py_DECREF(result);
1207 return NULL;
1208 }
1209 if (rv) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001210 if (set_add_entry(result, entry) == -1) {
Raymond Hettingera3b11e72003-12-31 14:08:58 +00001211 Py_DECREF(result);
1212 return NULL;
1213 }
1214 }
1215 }
1216 return (PyObject *)result;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001217 }
1218
Raymond Hettingera690a992003-11-16 16:17:49 +00001219 it = PyObject_GetIter(other);
1220 if (it == NULL) {
1221 Py_DECREF(result);
1222 return NULL;
1223 }
1224
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001225 while ((key = PyIter_Next(it)) != NULL) {
Raymond Hettinger775ebe22006-12-08 18:12:24 +00001226 int rv;
1227 setentry entry;
1228 long hash = PyObject_Hash(key);
1229
1230 if (hash == -1) {
1231 Py_DECREF(it);
1232 Py_DECREF(result);
1233 Py_DECREF(key);
1234 return NULL;
1235 }
1236 entry.hash = hash;
1237 entry.key = key;
1238 rv = set_contains_entry(so, &entry);
Georg Brandl8de403a2006-09-08 06:02:26 +00001239 if (rv == -1) {
1240 Py_DECREF(it);
1241 Py_DECREF(result);
1242 Py_DECREF(key);
1243 return NULL;
1244 }
1245 if (rv) {
Raymond Hettinger775ebe22006-12-08 18:12:24 +00001246 if (set_add_entry(result, &entry) == -1) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001247 Py_DECREF(it);
1248 Py_DECREF(result);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001249 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +00001250 return NULL;
1251 }
1252 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001253 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +00001254 }
1255 Py_DECREF(it);
1256 if (PyErr_Occurred()) {
1257 Py_DECREF(result);
1258 return NULL;
1259 }
1260 return (PyObject *)result;
1261}
1262
1263PyDoc_STRVAR(intersection_doc,
1264"Return the intersection of two sets as a new set.\n\
1265\n\
1266(i.e. all elements that are in both sets.)");
1267
1268static PyObject *
1269set_intersection_update(PySetObject *so, PyObject *other)
1270{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001271 PyObject *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001272
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001273 tmp = set_intersection(so, other);
1274 if (tmp == NULL)
Raymond Hettingera690a992003-11-16 16:17:49 +00001275 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001276 set_swap_bodies(so, (PySetObject *)tmp);
Raymond Hettingera690a992003-11-16 16:17:49 +00001277 Py_DECREF(tmp);
1278 Py_RETURN_NONE;
1279}
1280
1281PyDoc_STRVAR(intersection_update_doc,
1282"Update a set with the intersection of itself and another.");
1283
1284static PyObject *
1285set_and(PySetObject *so, PyObject *other)
1286{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001287 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001288 Py_INCREF(Py_NotImplemented);
1289 return Py_NotImplemented;
1290 }
1291 return set_intersection(so, other);
1292}
1293
1294static PyObject *
1295set_iand(PySetObject *so, PyObject *other)
1296{
1297 PyObject *result;
1298
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001299 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001300 Py_INCREF(Py_NotImplemented);
1301 return Py_NotImplemented;
1302 }
1303 result = set_intersection_update(so, other);
1304 if (result == NULL)
1305 return NULL;
1306 Py_DECREF(result);
1307 Py_INCREF(so);
1308 return (PyObject *)so;
1309}
1310
Neal Norwitz6576bd82005-11-13 18:41:28 +00001311static int
Raymond Hettingerc991db22005-08-11 07:58:45 +00001312set_difference_update_internal(PySetObject *so, PyObject *other)
1313{
1314 if ((PyObject *)so == other)
1315 return set_clear_internal(so);
1316
1317 if (PyAnySet_Check(other)) {
1318 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001319 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001320
1321 while (set_next((PySetObject *)other, &pos, &entry))
Georg Brandl8de403a2006-09-08 06:02:26 +00001322 if (set_discard_entry(so, entry) == -1)
1323 return -1;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001324 } else {
1325 PyObject *key, *it;
1326 it = PyObject_GetIter(other);
1327 if (it == NULL)
1328 return -1;
1329
1330 while ((key = PyIter_Next(it)) != NULL) {
1331 if (set_discard_key(so, key) == -1) {
1332 Py_DECREF(it);
1333 Py_DECREF(key);
1334 return -1;
1335 }
1336 Py_DECREF(key);
1337 }
1338 Py_DECREF(it);
1339 if (PyErr_Occurred())
1340 return -1;
1341 }
1342 /* If more than 1/5 are dummies, then resize them away. */
1343 if ((so->fill - so->used) * 5 < so->mask)
1344 return 0;
1345 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
1346}
1347
Raymond Hettingera690a992003-11-16 16:17:49 +00001348static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001349set_difference_update(PySetObject *so, PyObject *other)
1350{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001351 if (set_difference_update_internal(so, other) != -1)
Raymond Hettingerbc841a12005-08-07 13:02:53 +00001352 Py_RETURN_NONE;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001353 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001354}
1355
1356PyDoc_STRVAR(difference_update_doc,
1357"Remove all elements of another set from this set.");
1358
1359static PyObject *
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001360set_difference(PySetObject *so, PyObject *other)
1361{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001362 PyObject *result;
1363 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001364 Py_ssize_t pos = 0;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001365
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001366 if (!PyAnySet_Check(other) && !PyDict_Check(other)) {
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001367 result = set_copy(so);
1368 if (result == NULL)
Raymond Hettingerc991db22005-08-11 07:58:45 +00001369 return NULL;
1370 if (set_difference_update_internal((PySetObject *)result, other) != -1)
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001371 return result;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001372 Py_DECREF(result);
1373 return NULL;
1374 }
1375
1376 result = make_new_set(so->ob_type, NULL);
1377 if (result == NULL)
1378 return NULL;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001379
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001380 if (PyDict_Check(other)) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001381 while (set_next(so, &pos, &entry)) {
1382 setentry entrycopy;
1383 entrycopy.hash = entry->hash;
1384 entrycopy.key = entry->key;
1385 if (!PyDict_Contains(other, entry->key)) {
Georg Brandl8de403a2006-09-08 06:02:26 +00001386 if (set_add_entry((PySetObject *)result, &entrycopy) == -1) {
1387 Py_DECREF(result);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001388 return NULL;
Georg Brandl8de403a2006-09-08 06:02:26 +00001389 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001390 }
1391 }
1392 return result;
1393 }
1394
Raymond Hettingerc991db22005-08-11 07:58:45 +00001395 while (set_next(so, &pos, &entry)) {
Georg Brandl8de403a2006-09-08 06:02:26 +00001396 int rv = set_contains_entry((PySetObject *)other, entry);
1397 if (rv == -1) {
1398 Py_DECREF(result);
1399 return NULL;
1400 }
1401 if (!rv) {
1402 if (set_add_entry((PySetObject *)result, entry) == -1) {
1403 Py_DECREF(result);
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001404 return NULL;
Georg Brandl8de403a2006-09-08 06:02:26 +00001405 }
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001406 }
1407 }
1408 return result;
1409}
1410
1411PyDoc_STRVAR(difference_doc,
1412"Return the difference of two sets as a new set.\n\
1413\n\
1414(i.e. all elements that are in this set but not the other.)");
1415static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001416set_sub(PySetObject *so, PyObject *other)
1417{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001418 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001419 Py_INCREF(Py_NotImplemented);
1420 return Py_NotImplemented;
1421 }
1422 return set_difference(so, other);
1423}
1424
1425static PyObject *
1426set_isub(PySetObject *so, PyObject *other)
1427{
1428 PyObject *result;
1429
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001430 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001431 Py_INCREF(Py_NotImplemented);
1432 return Py_NotImplemented;
1433 }
1434 result = set_difference_update(so, other);
1435 if (result == NULL)
1436 return NULL;
1437 Py_DECREF(result);
1438 Py_INCREF(so);
1439 return (PyObject *)so;
1440}
1441
1442static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001443set_symmetric_difference_update(PySetObject *so, PyObject *other)
1444{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001445 PySetObject *otherset;
1446 PyObject *key;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001447 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001448 setentry *entry;
1449
1450 if ((PyObject *)so == other)
1451 return set_clear(so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001452
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001453 if (PyDict_Check(other)) {
1454 PyObject *value;
1455 int rv;
1456 while (PyDict_Next(other, &pos, &key, &value)) {
Raymond Hettinger775ebe22006-12-08 18:12:24 +00001457 setentry an_entry;
1458 long hash = PyObject_Hash(key);
1459
1460 if (hash == -1)
1461 return NULL;
1462 an_entry.hash = hash;
1463 an_entry.key = key;
1464 rv = set_discard_entry(so, &an_entry);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001465 if (rv == -1)
1466 return NULL;
1467 if (rv == DISCARD_NOTFOUND) {
Raymond Hettinger775ebe22006-12-08 18:12:24 +00001468 if (set_add_entry(so, &an_entry) == -1)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001469 return NULL;
1470 }
1471 }
1472 Py_RETURN_NONE;
1473 }
1474
1475 if (PyAnySet_Check(other)) {
1476 Py_INCREF(other);
1477 otherset = (PySetObject *)other;
1478 } else {
Raymond Hettingera690a992003-11-16 16:17:49 +00001479 otherset = (PySetObject *)make_new_set(so->ob_type, other);
1480 if (otherset == NULL)
1481 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001482 }
1483
Raymond Hettingerc991db22005-08-11 07:58:45 +00001484 while (set_next(otherset, &pos, &entry)) {
1485 int rv = set_discard_entry(so, entry);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001486 if (rv == -1) {
Neal Norwitz04e39ec2006-07-17 00:57:15 +00001487 Py_DECREF(otherset);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001488 return NULL;
1489 }
1490 if (rv == DISCARD_NOTFOUND) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001491 if (set_add_entry(so, entry) == -1) {
Neal Norwitz04e39ec2006-07-17 00:57:15 +00001492 Py_DECREF(otherset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001493 return NULL;
1494 }
1495 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001496 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001497 Py_DECREF(otherset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001498 Py_RETURN_NONE;
1499}
1500
1501PyDoc_STRVAR(symmetric_difference_update_doc,
1502"Update a set with the symmetric difference of itself and another.");
1503
1504static PyObject *
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001505set_symmetric_difference(PySetObject *so, PyObject *other)
1506{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001507 PyObject *rv;
1508 PySetObject *otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001509
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001510 otherset = (PySetObject *)make_new_set(so->ob_type, other);
1511 if (otherset == NULL)
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001512 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001513 rv = set_symmetric_difference_update(otherset, (PyObject *)so);
1514 if (rv == NULL)
1515 return NULL;
1516 Py_DECREF(rv);
1517 return (PyObject *)otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001518}
1519
1520PyDoc_STRVAR(symmetric_difference_doc,
1521"Return the symmetric difference of two sets as a new set.\n\
1522\n\
1523(i.e. all elements that are in exactly one of the sets.)");
1524
1525static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001526set_xor(PySetObject *so, PyObject *other)
1527{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001528 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001529 Py_INCREF(Py_NotImplemented);
1530 return Py_NotImplemented;
1531 }
1532 return set_symmetric_difference(so, other);
1533}
1534
1535static PyObject *
1536set_ixor(PySetObject *so, PyObject *other)
1537{
1538 PyObject *result;
1539
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001540 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001541 Py_INCREF(Py_NotImplemented);
1542 return Py_NotImplemented;
1543 }
1544 result = set_symmetric_difference_update(so, other);
1545 if (result == NULL)
1546 return NULL;
1547 Py_DECREF(result);
1548 Py_INCREF(so);
1549 return (PyObject *)so;
1550}
1551
1552static PyObject *
1553set_issubset(PySetObject *so, PyObject *other)
1554{
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001555 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001556 Py_ssize_t pos = 0;
Raymond Hettingera690a992003-11-16 16:17:49 +00001557
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001558 if (!PyAnySet_Check(other)) {
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001559 PyObject *tmp, *result;
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001560 tmp = make_new_set(&PySet_Type, other);
1561 if (tmp == NULL)
1562 return NULL;
1563 result = set_issubset(so, tmp);
1564 Py_DECREF(tmp);
1565 return result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001566 }
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001567 if (PySet_GET_SIZE(so) > PySet_GET_SIZE(other))
Raymond Hettingera690a992003-11-16 16:17:49 +00001568 Py_RETURN_FALSE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001569
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001570 while (set_next(so, &pos, &entry)) {
Georg Brandl8de403a2006-09-08 06:02:26 +00001571 int rv = set_contains_entry((PySetObject *)other, entry);
1572 if (rv == -1)
1573 return NULL;
1574 if (!rv)
Raymond Hettingera690a992003-11-16 16:17:49 +00001575 Py_RETURN_FALSE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001576 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001577 Py_RETURN_TRUE;
1578}
1579
1580PyDoc_STRVAR(issubset_doc, "Report whether another set contains this set.");
1581
1582static PyObject *
1583set_issuperset(PySetObject *so, PyObject *other)
1584{
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001585 PyObject *tmp, *result;
1586
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001587 if (!PyAnySet_Check(other)) {
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001588 tmp = make_new_set(&PySet_Type, other);
1589 if (tmp == NULL)
1590 return NULL;
1591 result = set_issuperset(so, tmp);
1592 Py_DECREF(tmp);
1593 return result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001594 }
1595 return set_issubset((PySetObject *)other, (PyObject *)so);
1596}
1597
1598PyDoc_STRVAR(issuperset_doc, "Report whether this set contains another set.");
1599
Raymond Hettingera690a992003-11-16 16:17:49 +00001600static PyObject *
1601set_richcompare(PySetObject *v, PyObject *w, int op)
1602{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001603 PyObject *r1, *r2;
1604
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001605 if(!PyAnySet_Check(w)) {
1606 if (op == Py_EQ)
1607 Py_RETURN_FALSE;
1608 if (op == Py_NE)
1609 Py_RETURN_TRUE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001610 PyErr_SetString(PyExc_TypeError, "can only compare to a set");
1611 return NULL;
1612 }
1613 switch (op) {
1614 case Py_EQ:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001615 if (PySet_GET_SIZE(v) != PySet_GET_SIZE(w))
Raymond Hettingera690a992003-11-16 16:17:49 +00001616 Py_RETURN_FALSE;
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00001617 if (v->hash != -1 &&
1618 ((PySetObject *)w)->hash != -1 &&
1619 v->hash != ((PySetObject *)w)->hash)
1620 Py_RETURN_FALSE;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001621 return set_issubset(v, w);
1622 case Py_NE:
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00001623 r1 = set_richcompare(v, w, Py_EQ);
1624 if (r1 == NULL)
1625 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001626 r2 = PyBool_FromLong(PyObject_Not(r1));
1627 Py_DECREF(r1);
1628 return r2;
1629 case Py_LE:
1630 return set_issubset(v, w);
1631 case Py_GE:
1632 return set_issuperset(v, w);
1633 case Py_LT:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001634 if (PySet_GET_SIZE(v) >= PySet_GET_SIZE(w))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001635 Py_RETURN_FALSE;
1636 return set_issubset(v, w);
1637 case Py_GT:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001638 if (PySet_GET_SIZE(v) <= PySet_GET_SIZE(w))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001639 Py_RETURN_FALSE;
1640 return set_issuperset(v, w);
Raymond Hettingera690a992003-11-16 16:17:49 +00001641 }
1642 Py_INCREF(Py_NotImplemented);
1643 return Py_NotImplemented;
1644}
1645
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001646static int
Georg Brandl347b3002006-03-30 11:57:00 +00001647set_nocmp(PyObject *self, PyObject *other)
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001648{
1649 PyErr_SetString(PyExc_TypeError, "cannot compare sets using cmp()");
1650 return -1;
1651}
1652
Raymond Hettingera690a992003-11-16 16:17:49 +00001653static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001654set_add(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001655{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001656 if (set_add_key(so, key) == -1)
Raymond Hettingera690a992003-11-16 16:17:49 +00001657 return NULL;
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001658 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001659}
1660
1661PyDoc_STRVAR(add_doc,
1662"Add an element to a set.\n\
1663\n\
1664This has no effect if the element is already present.");
1665
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001666static int
1667set_contains(PySetObject *so, PyObject *key)
1668{
1669 PyObject *tmpkey;
1670 int rv;
1671
1672 rv = set_contains_key(so, key);
1673 if (rv == -1) {
1674 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1675 return -1;
1676 PyErr_Clear();
1677 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1678 if (tmpkey == NULL)
1679 return -1;
1680 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
1681 rv = set_contains(so, tmpkey);
1682 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
1683 Py_DECREF(tmpkey);
1684 }
1685 return rv;
1686}
1687
1688static PyObject *
1689set_direct_contains(PySetObject *so, PyObject *key)
1690{
1691 long result;
1692
1693 result = set_contains(so, key);
1694 if (result == -1)
1695 return NULL;
1696 return PyBool_FromLong(result);
1697}
1698
1699PyDoc_STRVAR(contains_doc, "x.__contains__(y) <==> y in x.");
1700
Raymond Hettingera690a992003-11-16 16:17:49 +00001701static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001702set_remove(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001703{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001704 PyObject *tmpkey, *result;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001705 int rv;
Raymond Hettingerbfd334a2003-11-22 03:55:23 +00001706
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001707 rv = set_discard_key(so, key);
1708 if (rv == -1) {
1709 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1710 return NULL;
1711 PyErr_Clear();
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001712 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1713 if (tmpkey == NULL)
Raymond Hettingerbfd334a2003-11-22 03:55:23 +00001714 return NULL;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001715 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001716 result = set_remove(so, tmpkey);
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001717 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001718 Py_DECREF(tmpkey);
Raymond Hettinger0deab622003-12-13 18:53:18 +00001719 return result;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001720 } else if (rv == DISCARD_NOTFOUND) {
Raymond Hettinger775ebe22006-12-08 18:12:24 +00001721 set_key_error(key);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001722 return NULL;
1723 }
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001724 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001725}
1726
1727PyDoc_STRVAR(remove_doc,
1728"Remove an element from a set; it must be a member.\n\
1729\n\
1730If the element is not a member, raise a KeyError.");
1731
1732static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001733set_discard(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001734{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001735 PyObject *tmpkey, *result;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001736 int rv;
Raymond Hettinger0deab622003-12-13 18:53:18 +00001737
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001738 rv = set_discard_key(so, key);
1739 if (rv == -1) {
1740 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1741 return NULL;
1742 PyErr_Clear();
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001743 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1744 if (tmpkey == NULL)
Raymond Hettinger0deab622003-12-13 18:53:18 +00001745 return NULL;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001746 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001747 result = set_discard(so, tmpkey);
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001748 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001749 Py_DECREF(tmpkey);
Raymond Hettinger0deab622003-12-13 18:53:18 +00001750 return result;
1751 }
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001752 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001753}
1754
1755PyDoc_STRVAR(discard_doc,
1756"Remove an element from a set if it is a member.\n\
1757\n\
1758If the element is not a member, do nothing.");
1759
1760static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001761set_reduce(PySetObject *so)
1762{
Raymond Hettinger15056a52004-11-09 07:25:31 +00001763 PyObject *keys=NULL, *args=NULL, *result=NULL, *dict=NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001764
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001765 keys = PySequence_List((PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001766 if (keys == NULL)
1767 goto done;
1768 args = PyTuple_Pack(1, keys);
1769 if (args == NULL)
1770 goto done;
Raymond Hettinger15056a52004-11-09 07:25:31 +00001771 dict = PyObject_GetAttrString((PyObject *)so, "__dict__");
1772 if (dict == NULL) {
1773 PyErr_Clear();
1774 dict = Py_None;
1775 Py_INCREF(dict);
1776 }
1777 result = PyTuple_Pack(3, so->ob_type, args, dict);
Raymond Hettingera690a992003-11-16 16:17:49 +00001778done:
1779 Py_XDECREF(args);
1780 Py_XDECREF(keys);
Raymond Hettinger15056a52004-11-09 07:25:31 +00001781 Py_XDECREF(dict);
Raymond Hettingera690a992003-11-16 16:17:49 +00001782 return result;
1783}
1784
1785PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
1786
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001787static int
1788set_init(PySetObject *self, PyObject *args, PyObject *kwds)
1789{
1790 PyObject *iterable = NULL;
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001791
1792 if (!PyAnySet_Check(self))
1793 return -1;
1794 if (!PyArg_UnpackTuple(args, self->ob_type->tp_name, 0, 1, &iterable))
1795 return -1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001796 set_clear_internal(self);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001797 self->hash = -1;
1798 if (iterable == NULL)
1799 return 0;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001800 return set_update_internal(self, iterable);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001801}
1802
Raymond Hettingera690a992003-11-16 16:17:49 +00001803static PySequenceMethods set_as_sequence = {
Georg Brandl347b3002006-03-30 11:57:00 +00001804 set_len, /* sq_length */
Raymond Hettingera690a992003-11-16 16:17:49 +00001805 0, /* sq_concat */
1806 0, /* sq_repeat */
1807 0, /* sq_item */
1808 0, /* sq_slice */
1809 0, /* sq_ass_item */
1810 0, /* sq_ass_slice */
1811 (objobjproc)set_contains, /* sq_contains */
1812};
1813
1814/* set object ********************************************************/
1815
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00001816#ifdef Py_DEBUG
1817static PyObject *test_c_api(PySetObject *so);
1818
1819PyDoc_STRVAR(test_c_api_doc, "Exercises C API. Returns True.\n\
1820All is well if assertions don't fail.");
1821#endif
1822
Raymond Hettingera690a992003-11-16 16:17:49 +00001823static PyMethodDef set_methods[] = {
1824 {"add", (PyCFunction)set_add, METH_O,
1825 add_doc},
1826 {"clear", (PyCFunction)set_clear, METH_NOARGS,
1827 clear_doc},
Raymond Hettinger0deab622003-12-13 18:53:18 +00001828 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001829 contains_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001830 {"copy", (PyCFunction)set_copy, METH_NOARGS,
1831 copy_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001832 {"discard", (PyCFunction)set_discard, METH_O,
1833 discard_doc},
1834 {"difference", (PyCFunction)set_difference, METH_O,
1835 difference_doc},
1836 {"difference_update", (PyCFunction)set_difference_update, METH_O,
1837 difference_update_doc},
1838 {"intersection",(PyCFunction)set_intersection, METH_O,
1839 intersection_doc},
1840 {"intersection_update",(PyCFunction)set_intersection_update, METH_O,
1841 intersection_update_doc},
1842 {"issubset", (PyCFunction)set_issubset, METH_O,
1843 issubset_doc},
1844 {"issuperset", (PyCFunction)set_issuperset, METH_O,
1845 issuperset_doc},
1846 {"pop", (PyCFunction)set_pop, METH_NOARGS,
1847 pop_doc},
1848 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
1849 reduce_doc},
1850 {"remove", (PyCFunction)set_remove, METH_O,
1851 remove_doc},
1852 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
1853 symmetric_difference_doc},
1854 {"symmetric_difference_update",(PyCFunction)set_symmetric_difference_update, METH_O,
1855 symmetric_difference_update_doc},
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00001856#ifdef Py_DEBUG
1857 {"test_c_api", (PyCFunction)test_c_api, METH_NOARGS,
1858 test_c_api_doc},
1859#endif
Raymond Hettingera690a992003-11-16 16:17:49 +00001860 {"union", (PyCFunction)set_union, METH_O,
1861 union_doc},
Raymond Hettingera38123e2003-11-24 22:18:49 +00001862 {"update", (PyCFunction)set_update, METH_O,
1863 update_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001864 {NULL, NULL} /* sentinel */
1865};
1866
1867static PyNumberMethods set_as_number = {
1868 0, /*nb_add*/
1869 (binaryfunc)set_sub, /*nb_subtract*/
1870 0, /*nb_multiply*/
1871 0, /*nb_divide*/
1872 0, /*nb_remainder*/
1873 0, /*nb_divmod*/
1874 0, /*nb_power*/
1875 0, /*nb_negative*/
1876 0, /*nb_positive*/
1877 0, /*nb_absolute*/
1878 0, /*nb_nonzero*/
1879 0, /*nb_invert*/
1880 0, /*nb_lshift*/
1881 0, /*nb_rshift*/
1882 (binaryfunc)set_and, /*nb_and*/
1883 (binaryfunc)set_xor, /*nb_xor*/
1884 (binaryfunc)set_or, /*nb_or*/
1885 0, /*nb_coerce*/
1886 0, /*nb_int*/
1887 0, /*nb_long*/
1888 0, /*nb_float*/
1889 0, /*nb_oct*/
1890 0, /*nb_hex*/
1891 0, /*nb_inplace_add*/
1892 (binaryfunc)set_isub, /*nb_inplace_subtract*/
1893 0, /*nb_inplace_multiply*/
1894 0, /*nb_inplace_divide*/
1895 0, /*nb_inplace_remainder*/
1896 0, /*nb_inplace_power*/
1897 0, /*nb_inplace_lshift*/
1898 0, /*nb_inplace_rshift*/
1899 (binaryfunc)set_iand, /*nb_inplace_and*/
1900 (binaryfunc)set_ixor, /*nb_inplace_xor*/
1901 (binaryfunc)set_ior, /*nb_inplace_or*/
1902};
1903
1904PyDoc_STRVAR(set_doc,
1905"set(iterable) --> set object\n\
1906\n\
Andrew M. Kuchling52740be2006-07-29 15:10:32 +00001907Build an unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00001908
1909PyTypeObject PySet_Type = {
1910 PyObject_HEAD_INIT(&PyType_Type)
1911 0, /* ob_size */
1912 "set", /* tp_name */
1913 sizeof(PySetObject), /* tp_basicsize */
1914 0, /* tp_itemsize */
1915 /* methods */
1916 (destructor)set_dealloc, /* tp_dealloc */
1917 (printfunc)set_tp_print, /* tp_print */
1918 0, /* tp_getattr */
1919 0, /* tp_setattr */
Georg Brandl347b3002006-03-30 11:57:00 +00001920 set_nocmp, /* tp_compare */
Raymond Hettingera690a992003-11-16 16:17:49 +00001921 (reprfunc)set_repr, /* tp_repr */
1922 &set_as_number, /* tp_as_number */
1923 &set_as_sequence, /* tp_as_sequence */
1924 0, /* tp_as_mapping */
1925 set_nohash, /* tp_hash */
1926 0, /* tp_call */
1927 0, /* tp_str */
1928 PyObject_GenericGetAttr, /* tp_getattro */
1929 0, /* tp_setattro */
1930 0, /* tp_as_buffer */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001931 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_CHECKTYPES |
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001932 Py_TPFLAGS_BASETYPE, /* tp_flags */
Raymond Hettingera690a992003-11-16 16:17:49 +00001933 set_doc, /* tp_doc */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001934 (traverseproc)set_traverse, /* tp_traverse */
Raymond Hettingerfe889f32005-08-06 05:43:39 +00001935 (inquiry)set_clear_internal, /* tp_clear */
Raymond Hettingera690a992003-11-16 16:17:49 +00001936 (richcmpfunc)set_richcompare, /* tp_richcompare */
Raymond Hettinger691d8052004-05-30 07:26:47 +00001937 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001938 (getiterfunc)set_iter, /* tp_iter */
Raymond Hettingera690a992003-11-16 16:17:49 +00001939 0, /* tp_iternext */
1940 set_methods, /* tp_methods */
1941 0, /* tp_members */
1942 0, /* tp_getset */
1943 0, /* tp_base */
1944 0, /* tp_dict */
1945 0, /* tp_descr_get */
1946 0, /* tp_descr_set */
1947 0, /* tp_dictoffset */
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001948 (initproc)set_init, /* tp_init */
Raymond Hettingera690a992003-11-16 16:17:49 +00001949 PyType_GenericAlloc, /* tp_alloc */
1950 set_new, /* tp_new */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001951 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00001952};
1953
1954/* frozenset object ********************************************************/
1955
1956
1957static PyMethodDef frozenset_methods[] = {
Raymond Hettinger0deab622003-12-13 18:53:18 +00001958 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001959 contains_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001960 {"copy", (PyCFunction)frozenset_copy, METH_NOARGS,
Raymond Hettingera690a992003-11-16 16:17:49 +00001961 copy_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001962 {"difference", (PyCFunction)set_difference, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001963 difference_doc},
1964 {"intersection",(PyCFunction)set_intersection, METH_O,
1965 intersection_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001966 {"issubset", (PyCFunction)set_issubset, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001967 issubset_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001968 {"issuperset", (PyCFunction)set_issuperset, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001969 issuperset_doc},
1970 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
1971 reduce_doc},
1972 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
1973 symmetric_difference_doc},
1974 {"union", (PyCFunction)set_union, METH_O,
1975 union_doc},
1976 {NULL, NULL} /* sentinel */
1977};
1978
1979static PyNumberMethods frozenset_as_number = {
1980 0, /*nb_add*/
1981 (binaryfunc)set_sub, /*nb_subtract*/
1982 0, /*nb_multiply*/
1983 0, /*nb_divide*/
1984 0, /*nb_remainder*/
1985 0, /*nb_divmod*/
1986 0, /*nb_power*/
1987 0, /*nb_negative*/
1988 0, /*nb_positive*/
1989 0, /*nb_absolute*/
1990 0, /*nb_nonzero*/
1991 0, /*nb_invert*/
1992 0, /*nb_lshift*/
1993 0, /*nb_rshift*/
1994 (binaryfunc)set_and, /*nb_and*/
1995 (binaryfunc)set_xor, /*nb_xor*/
1996 (binaryfunc)set_or, /*nb_or*/
1997};
1998
1999PyDoc_STRVAR(frozenset_doc,
2000"frozenset(iterable) --> frozenset object\n\
2001\n\
Andrew M. Kuchling52740be2006-07-29 15:10:32 +00002002Build an immutable unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00002003
2004PyTypeObject PyFrozenSet_Type = {
2005 PyObject_HEAD_INIT(&PyType_Type)
2006 0, /* ob_size */
2007 "frozenset", /* tp_name */
2008 sizeof(PySetObject), /* tp_basicsize */
Raymond Hettingera3b11e72003-12-31 14:08:58 +00002009 0, /* tp_itemsize */
2010 /* methods */
Raymond Hettingera690a992003-11-16 16:17:49 +00002011 (destructor)set_dealloc, /* tp_dealloc */
2012 (printfunc)set_tp_print, /* tp_print */
2013 0, /* tp_getattr */
2014 0, /* tp_setattr */
Georg Brandl347b3002006-03-30 11:57:00 +00002015 set_nocmp, /* tp_compare */
Raymond Hettingera690a992003-11-16 16:17:49 +00002016 (reprfunc)set_repr, /* tp_repr */
2017 &frozenset_as_number, /* tp_as_number */
2018 &set_as_sequence, /* tp_as_sequence */
2019 0, /* tp_as_mapping */
2020 frozenset_hash, /* tp_hash */
2021 0, /* tp_call */
2022 0, /* tp_str */
2023 PyObject_GenericGetAttr, /* tp_getattro */
2024 0, /* tp_setattro */
2025 0, /* tp_as_buffer */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002026 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_CHECKTYPES |
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00002027 Py_TPFLAGS_BASETYPE, /* tp_flags */
Raymond Hettingera690a992003-11-16 16:17:49 +00002028 frozenset_doc, /* tp_doc */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002029 (traverseproc)set_traverse, /* tp_traverse */
Raymond Hettingerfe889f32005-08-06 05:43:39 +00002030 (inquiry)set_clear_internal, /* tp_clear */
Raymond Hettingera690a992003-11-16 16:17:49 +00002031 (richcmpfunc)set_richcompare, /* tp_richcompare */
Raymond Hettinger691d8052004-05-30 07:26:47 +00002032 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
Raymond Hettingera690a992003-11-16 16:17:49 +00002033 (getiterfunc)set_iter, /* tp_iter */
2034 0, /* tp_iternext */
2035 frozenset_methods, /* tp_methods */
2036 0, /* tp_members */
2037 0, /* tp_getset */
2038 0, /* tp_base */
2039 0, /* tp_dict */
2040 0, /* tp_descr_get */
2041 0, /* tp_descr_set */
2042 0, /* tp_dictoffset */
2043 0, /* tp_init */
2044 PyType_GenericAlloc, /* tp_alloc */
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00002045 frozenset_new, /* tp_new */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002046 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00002047};
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002048
2049
2050/***** C API functions *************************************************/
2051
2052PyObject *
2053PySet_New(PyObject *iterable)
2054{
2055 return make_new_set(&PySet_Type, iterable);
2056}
2057
2058PyObject *
2059PyFrozenSet_New(PyObject *iterable)
2060{
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002061 PyObject *args, *result;
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002062
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002063 if (iterable == NULL)
2064 args = PyTuple_New(0);
2065 else
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002066 args = PyTuple_Pack(1, iterable);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002067 if (args == NULL)
2068 return NULL;
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002069 result = frozenset_new(&PyFrozenSet_Type, args, NULL);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002070 Py_DECREF(args);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002071 return result;
2072}
2073
Neal Norwitz8c49c822006-03-04 18:41:19 +00002074Py_ssize_t
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002075PySet_Size(PyObject *anyset)
2076{
2077 if (!PyAnySet_Check(anyset)) {
2078 PyErr_BadInternalCall();
2079 return -1;
2080 }
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00002081 return PySet_GET_SIZE(anyset);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002082}
2083
2084int
Barry Warsaw176014f2006-03-30 22:45:35 +00002085PySet_Clear(PyObject *set)
2086{
2087 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2088 PyErr_BadInternalCall();
2089 return -1;
2090 }
2091 return set_clear_internal((PySetObject *)set);
2092}
2093
2094int
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002095PySet_Contains(PyObject *anyset, PyObject *key)
2096{
2097 if (!PyAnySet_Check(anyset)) {
2098 PyErr_BadInternalCall();
2099 return -1;
2100 }
2101 return set_contains_key((PySetObject *)anyset, key);
2102}
2103
2104int
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002105PySet_Discard(PyObject *set, PyObject *key)
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002106{
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002107 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002108 PyErr_BadInternalCall();
2109 return -1;
2110 }
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002111 return set_discard_key((PySetObject *)set, key);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002112}
2113
2114int
2115PySet_Add(PyObject *set, PyObject *key)
2116{
2117 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2118 PyErr_BadInternalCall();
2119 return -1;
2120 }
2121 return set_add_key((PySetObject *)set, key);
2122}
2123
Barry Warsaw176014f2006-03-30 22:45:35 +00002124int
2125_PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **entry)
2126{
2127 setentry *entry_ptr;
2128
2129 if (!PyAnySet_Check(set)) {
2130 PyErr_BadInternalCall();
2131 return -1;
2132 }
2133 if (set_next((PySetObject *)set, pos, &entry_ptr) == 0)
2134 return 0;
2135 *entry = entry_ptr->key;
2136 return 1;
2137}
2138
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002139PyObject *
2140PySet_Pop(PyObject *set)
2141{
2142 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2143 PyErr_BadInternalCall();
2144 return NULL;
2145 }
2146 return set_pop((PySetObject *)set);
2147}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002148
Barry Warsaw176014f2006-03-30 22:45:35 +00002149int
2150_PySet_Update(PyObject *set, PyObject *iterable)
2151{
2152 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2153 PyErr_BadInternalCall();
2154 return -1;
2155 }
2156 return set_update_internal((PySetObject *)set, iterable);
2157}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002158
2159#ifdef Py_DEBUG
2160
2161/* Test code to be called with any three element set.
2162 Returns True and original set is restored. */
2163
2164#define assertRaises(call_return_value, exception) \
2165 do { \
2166 assert(call_return_value); \
2167 assert(PyErr_ExceptionMatches(exception)); \
2168 PyErr_Clear(); \
2169 } while(0)
2170
2171static PyObject *
2172test_c_api(PySetObject *so)
2173{
Neal Norwitz0f2783c2006-06-19 05:40:44 +00002174 Py_ssize_t count;
Barry Warsaw176014f2006-03-30 22:45:35 +00002175 char *s;
2176 Py_ssize_t i;
2177 PyObject *elem, *dup, *t, *f, *dup2;
2178 PyObject *ob = (PyObject *)so;
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002179
2180 /* Verify preconditions and exercise type/size checks */
2181 assert(PyAnySet_Check(ob));
2182 assert(PyAnySet_CheckExact(ob));
2183 assert(!PyFrozenSet_CheckExact(ob));
2184 assert(PySet_Size(ob) == 3);
2185 assert(PySet_GET_SIZE(ob) == 3);
2186
2187 /* Raise TypeError for non-iterable constructor arguments */
2188 assertRaises(PySet_New(Py_None) == NULL, PyExc_TypeError);
2189 assertRaises(PyFrozenSet_New(Py_None) == NULL, PyExc_TypeError);
2190
2191 /* Raise TypeError for unhashable key */
2192 dup = PySet_New(ob);
2193 assertRaises(PySet_Discard(ob, dup) == -1, PyExc_TypeError);
2194 assertRaises(PySet_Contains(ob, dup) == -1, PyExc_TypeError);
2195 assertRaises(PySet_Add(ob, dup) == -1, PyExc_TypeError);
2196
2197 /* Exercise successful pop, contains, add, and discard */
2198 elem = PySet_Pop(ob);
2199 assert(PySet_Contains(ob, elem) == 0);
2200 assert(PySet_GET_SIZE(ob) == 2);
2201 assert(PySet_Add(ob, elem) == 0);
2202 assert(PySet_Contains(ob, elem) == 1);
2203 assert(PySet_GET_SIZE(ob) == 3);
2204 assert(PySet_Discard(ob, elem) == 1);
2205 assert(PySet_GET_SIZE(ob) == 2);
2206 assert(PySet_Discard(ob, elem) == 0);
2207 assert(PySet_GET_SIZE(ob) == 2);
2208
Barry Warsaw176014f2006-03-30 22:45:35 +00002209 /* Exercise clear */
2210 dup2 = PySet_New(dup);
2211 assert(PySet_Clear(dup2) == 0);
2212 assert(PySet_Size(dup2) == 0);
2213 Py_DECREF(dup2);
2214
2215 /* Raise SystemError on clear or update of frozen set */
2216 f = PyFrozenSet_New(dup);
2217 assertRaises(PySet_Clear(f) == -1, PyExc_SystemError);
2218 assertRaises(_PySet_Update(f, dup) == -1, PyExc_SystemError);
2219 Py_DECREF(f);
2220
2221 /* Exercise direct iteration */
2222 i = 0, count = 0;
2223 while (_PySet_Next((PyObject *)dup, &i, &elem)) {
2224 s = PyString_AsString(elem);
2225 assert(s && (s[0] == 'a' || s[0] == 'b' || s[0] == 'c'));
2226 count++;
2227 }
2228 assert(count == 3);
2229
2230 /* Exercise updates */
2231 dup2 = PySet_New(NULL);
2232 assert(_PySet_Update(dup2, dup) == 0);
2233 assert(PySet_Size(dup2) == 3);
2234 assert(_PySet_Update(dup2, dup) == 0);
2235 assert(PySet_Size(dup2) == 3);
2236 Py_DECREF(dup2);
2237
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002238 /* Raise SystemError when self argument is not a set or frozenset. */
2239 t = PyTuple_New(0);
2240 assertRaises(PySet_Size(t) == -1, PyExc_SystemError);
2241 assertRaises(PySet_Contains(t, elem) == -1, PyExc_SystemError);
2242 Py_DECREF(t);
2243
2244 /* Raise SystemError when self argument is not a set. */
2245 f = PyFrozenSet_New(dup);
2246 assert(PySet_Size(f) == 3);
2247 assert(PyFrozenSet_CheckExact(f));
2248 assertRaises(PySet_Add(f, elem) == -1, PyExc_SystemError);
2249 assertRaises(PySet_Discard(f, elem) == -1, PyExc_SystemError);
2250 assertRaises(PySet_Pop(f) == NULL, PyExc_SystemError);
2251 Py_DECREF(f);
2252
2253 /* Raise KeyError when popping from an empty set */
Raymond Hettingerd8e13382005-08-17 12:27:17 +00002254 assert(PyNumber_InPlaceSubtract(ob, ob) == ob);
2255 Py_DECREF(ob);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002256 assert(PySet_GET_SIZE(ob) == 0);
2257 assertRaises(PySet_Pop(ob) == NULL, PyExc_KeyError);
2258
Raymond Hettingerd8e13382005-08-17 12:27:17 +00002259 /* Restore the set from the copy using the PyNumber API */
2260 assert(PyNumber_InPlaceOr(ob, dup) == ob);
2261 Py_DECREF(ob);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002262
2263 /* Verify constructors accept NULL arguments */
2264 f = PySet_New(NULL);
2265 assert(f != NULL);
2266 assert(PySet_GET_SIZE(f) == 0);
2267 Py_DECREF(f);
2268 f = PyFrozenSet_New(NULL);
2269 assert(f != NULL);
2270 assert(PyFrozenSet_CheckExact(f));
2271 assert(PySet_GET_SIZE(f) == 0);
2272 Py_DECREF(f);
2273
2274 Py_DECREF(elem);
2275 Py_DECREF(dup);
2276 Py_RETURN_TRUE;
2277}
2278
Raymond Hettinger9bda1d62005-09-16 07:14:21 +00002279#undef assertRaises
2280
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002281#endif