blob: a896d937faec3046179dc5323534953a25768483 [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 Hettinger9c14ffb2006-12-08 04:57:50 +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 Norwitza5ccda92006-10-28 21:16:54 +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 Hettinger0c850862006-12-08 04:24:33 +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 Hettinger0c850862006-12-08 04:24:33 +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 Hettinger0c850862006-12-08 04:24:33 +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);
Raymond Hettingerc563a1c2006-09-07 02:42:48 +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;
Raymond Hettingerc563a1c2006-09-07 02:42:48 +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 Hettinger53999102006-12-30 04:01:17 +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 Hettinger53999102006-12-30 04:01:17 +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 Hettinger53999102006-12-30 04:01:17 +0000591 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000592 }
593 fputs("])", fp);
Raymond Hettinger53999102006-12-30 04:01:17 +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 Hettinger53999102006-12-30 04:01:17 +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 Hettinger53999102006-12-30 04:01:17 +0000612 goto done;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000613 listrepr = PyObject_Repr(keys);
614 Py_DECREF(keys);
615 if (listrepr == NULL)
Raymond Hettinger53999102006-12-30 04:01:17 +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 Hettinger53999102006-12-30 04:01:17 +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 Hettinger0e7a6322007-02-08 00:50:39 +0000915 if (PyAnySet_CheckExact(other))
Raymond Hettingerc991db22005-08-11 07:58:45 +0000916 return set_merge(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +0000917
Raymond Hettingerdb67aef2007-02-01 21:02:59 +0000918 if (PyDict_CheckExact(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 Hettingerd6fc72a2007-02-19 02:03:19 +0000921 long hash;
Raymond Hettinger15cade02007-02-19 20:44:04 +0000922 Py_ssize_t dictsize = PyDict_Size(other);
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +0000923
Raymond Hettinger15cade02007-02-19 20:44:04 +0000924 /* Do one big resize at the start, rather than
925 * incrementally resizing as we insert new keys. Expect
926 * that there will be no (or few) overlapping keys.
927 */
928 if (dictsize == -1)
929 return -1;
930 if ((so->fill + dictsize)*3 >= (so->mask+1)*2) {
931 if (set_table_resize(so, (so->used + dictsize)*2) != 0)
932 return -1;
933 }
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +0000934 while (_PyDict_Next(other, &pos, &key, &value, &hash)) {
935 setentry an_entry;
936
937 an_entry.hash = hash;
938 an_entry.key = key;
939 if (set_add_entry(so, &an_entry) == -1)
Raymond Hettingerd7946662005-08-01 21:39:29 +0000940 return -1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000941 }
Raymond Hettingerd7946662005-08-01 21:39:29 +0000942 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000943 }
944
Raymond Hettingera38123e2003-11-24 22:18:49 +0000945 it = PyObject_GetIter(other);
946 if (it == NULL)
Raymond Hettingerd7946662005-08-01 21:39:29 +0000947 return -1;
Raymond Hettingera690a992003-11-16 16:17:49 +0000948
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000949 while ((key = PyIter_Next(it)) != NULL) {
Raymond Hettingerc991db22005-08-11 07:58:45 +0000950 if (set_add_key(so, key) == -1) {
Raymond Hettingera38123e2003-11-24 22:18:49 +0000951 Py_DECREF(it);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000952 Py_DECREF(key);
Raymond Hettingerd7946662005-08-01 21:39:29 +0000953 return -1;
Raymond Hettingera690a992003-11-16 16:17:49 +0000954 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000955 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +0000956 }
Raymond Hettingera38123e2003-11-24 22:18:49 +0000957 Py_DECREF(it);
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +0000958 if (PyErr_Occurred())
Raymond Hettingerd7946662005-08-01 21:39:29 +0000959 return -1;
960 return 0;
961}
962
963static PyObject *
964set_update(PySetObject *so, PyObject *other)
965{
966 if (set_update_internal(so, other) == -1)
Raymond Hettingera38123e2003-11-24 22:18:49 +0000967 return NULL;
968 Py_RETURN_NONE;
969}
970
971PyDoc_STRVAR(update_doc,
972"Update a set with the union of itself and another.");
973
974static PyObject *
975make_new_set(PyTypeObject *type, PyObject *iterable)
976{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000977 register PySetObject *so = NULL;
Raymond Hettingera38123e2003-11-24 22:18:49 +0000978
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000979 if (dummy == NULL) { /* Auto-initialize dummy */
980 dummy = PyString_FromString("<dummy key>");
981 if (dummy == NULL)
982 return NULL;
983 }
Raymond Hettingera690a992003-11-16 16:17:49 +0000984
985 /* create PySetObject structure */
Raymond Hettingerbc841a12005-08-07 13:02:53 +0000986 if (num_free_sets &&
987 (type == &PySet_Type || type == &PyFrozenSet_Type)) {
988 so = free_sets[--num_free_sets];
989 assert (so != NULL && PyAnySet_CheckExact(so));
990 so->ob_type = type;
991 _Py_NewReference((PyObject *)so);
992 EMPTY_TO_MINSIZE(so);
993 PyObject_GC_Track(so);
994 } else {
995 so = (PySetObject *)type->tp_alloc(type, 0);
996 if (so == NULL)
997 return NULL;
998 /* tp_alloc has already zeroed the structure */
999 assert(so->table == NULL && so->fill == 0 && so->used == 0);
1000 INIT_NONZERO_SET_SLOTS(so);
1001 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001002
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001003 so->lookup = set_lookkey_string;
Raymond Hettinger691d8052004-05-30 07:26:47 +00001004 so->weakreflist = NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001005
Raymond Hettingera38123e2003-11-24 22:18:49 +00001006 if (iterable != NULL) {
Raymond Hettingerd7946662005-08-01 21:39:29 +00001007 if (set_update_internal(so, iterable) == -1) {
Raymond Hettingera38123e2003-11-24 22:18:49 +00001008 Py_DECREF(so);
1009 return NULL;
1010 }
Raymond Hettingera38123e2003-11-24 22:18:49 +00001011 }
1012
Raymond Hettingera690a992003-11-16 16:17:49 +00001013 return (PyObject *)so;
1014}
1015
Raymond Hettingerd7946662005-08-01 21:39:29 +00001016/* The empty frozenset is a singleton */
1017static PyObject *emptyfrozenset = NULL;
1018
Raymond Hettingera690a992003-11-16 16:17:49 +00001019static PyObject *
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001020frozenset_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Raymond Hettingera690a992003-11-16 16:17:49 +00001021{
Raymond Hettingerd7946662005-08-01 21:39:29 +00001022 PyObject *iterable = NULL, *result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001023
Raymond Hettinger9fdfadb2007-01-11 18:22:55 +00001024 if (type == &PyFrozenSet_Type && !_PyArg_NoKeywords("frozenset()", kwds))
Georg Brandl02c42872005-08-26 06:42:30 +00001025 return NULL;
1026
Raymond Hettingera690a992003-11-16 16:17:49 +00001027 if (!PyArg_UnpackTuple(args, type->tp_name, 0, 1, &iterable))
1028 return NULL;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001029
1030 if (type != &PyFrozenSet_Type)
1031 return make_new_set(type, iterable);
1032
1033 if (iterable != NULL) {
1034 /* frozenset(f) is idempotent */
1035 if (PyFrozenSet_CheckExact(iterable)) {
1036 Py_INCREF(iterable);
1037 return iterable;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001038 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001039 result = make_new_set(type, iterable);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001040 if (result == NULL || PySet_GET_SIZE(result))
Raymond Hettingerd7946662005-08-01 21:39:29 +00001041 return result;
1042 Py_DECREF(result);
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001043 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001044 /* The empty frozenset is a singleton */
1045 if (emptyfrozenset == NULL)
1046 emptyfrozenset = make_new_set(type, NULL);
1047 Py_XINCREF(emptyfrozenset);
1048 return emptyfrozenset;
1049}
1050
1051void
1052PySet_Fini(void)
1053{
Raymond Hettingerbc841a12005-08-07 13:02:53 +00001054 PySetObject *so;
1055
1056 while (num_free_sets) {
1057 num_free_sets--;
1058 so = free_sets[num_free_sets];
1059 PyObject_GC_Del(so);
1060 }
Martin v. Löwised8f7832006-04-15 12:47:23 +00001061 Py_CLEAR(dummy);
1062 Py_CLEAR(emptyfrozenset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001063}
1064
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001065static PyObject *
1066set_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1067{
Raymond Hettinger9fdfadb2007-01-11 18:22:55 +00001068 if (type == &PySet_Type && !_PyArg_NoKeywords("set()", kwds))
Georg Brandl02c42872005-08-26 06:42:30 +00001069 return NULL;
1070
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001071 return make_new_set(type, NULL);
1072}
1073
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001074/* set_swap_bodies() switches the contents of any two sets by moving their
1075 internal data pointers and, if needed, copying the internal smalltables.
1076 Semantically equivalent to:
1077
1078 t=set(a); a.clear(); a.update(b); b.clear(); b.update(t); del t
1079
1080 The function always succeeds and it leaves both objects in a stable state.
1081 Useful for creating temporary frozensets from sets for membership testing
1082 in __contains__(), discard(), and remove(). Also useful for operations
1083 that update in-place (by allowing an intermediate result to be swapped
Raymond Hettinger9dcb17c2005-07-31 13:09:28 +00001084 into one of the original inputs).
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001085*/
1086
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001087static void
1088set_swap_bodies(PySetObject *a, PySetObject *b)
Raymond Hettingera690a992003-11-16 16:17:49 +00001089{
Neal Norwitz0f2783c2006-06-19 05:40:44 +00001090 Py_ssize_t t;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001091 setentry *u;
1092 setentry *(*f)(PySetObject *so, PyObject *key, long hash);
1093 setentry tab[PySet_MINSIZE];
1094 long h;
1095
1096 t = a->fill; a->fill = b->fill; b->fill = t;
1097 t = a->used; a->used = b->used; b->used = t;
1098 t = a->mask; a->mask = b->mask; b->mask = t;
1099
1100 u = a->table;
1101 if (a->table == a->smalltable)
1102 u = b->smalltable;
1103 a->table = b->table;
1104 if (b->table == b->smalltable)
1105 a->table = a->smalltable;
1106 b->table = u;
1107
1108 f = a->lookup; a->lookup = b->lookup; b->lookup = f;
1109
1110 if (a->table == a->smalltable || b->table == b->smalltable) {
1111 memcpy(tab, a->smalltable, sizeof(tab));
1112 memcpy(a->smalltable, b->smalltable, sizeof(tab));
1113 memcpy(b->smalltable, tab, sizeof(tab));
1114 }
1115
Raymond Hettingera580c472005-08-05 17:19:54 +00001116 if (PyType_IsSubtype(a->ob_type, &PyFrozenSet_Type) &&
1117 PyType_IsSubtype(b->ob_type, &PyFrozenSet_Type)) {
1118 h = a->hash; a->hash = b->hash; b->hash = h;
1119 } else {
1120 a->hash = -1;
1121 b->hash = -1;
1122 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001123}
1124
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001125static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001126set_copy(PySetObject *so)
1127{
Raymond Hettingera38123e2003-11-24 22:18:49 +00001128 return make_new_set(so->ob_type, (PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001129}
1130
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001131static PyObject *
1132frozenset_copy(PySetObject *so)
1133{
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001134 if (PyFrozenSet_CheckExact(so)) {
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001135 Py_INCREF(so);
1136 return (PyObject *)so;
1137 }
1138 return set_copy(so);
1139}
1140
Raymond Hettingera690a992003-11-16 16:17:49 +00001141PyDoc_STRVAR(copy_doc, "Return a shallow copy of a set.");
1142
1143static PyObject *
Raymond Hettingerc991db22005-08-11 07:58:45 +00001144set_clear(PySetObject *so)
1145{
1146 set_clear_internal(so);
1147 Py_RETURN_NONE;
1148}
1149
1150PyDoc_STRVAR(clear_doc, "Remove all elements from this set.");
1151
1152static PyObject *
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001153set_union(PySetObject *so, PyObject *other)
1154{
1155 PySetObject *result;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001156
1157 result = (PySetObject *)set_copy(so);
1158 if (result == NULL)
1159 return NULL;
Raymond Hettingerd8e13382005-08-17 12:27:17 +00001160 if ((PyObject *)so == other)
1161 return (PyObject *)result;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001162 if (set_update_internal(result, other) == -1) {
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001163 Py_DECREF(result);
1164 return NULL;
1165 }
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001166 return (PyObject *)result;
1167}
1168
1169PyDoc_STRVAR(union_doc,
1170 "Return the union of two sets as a new set.\n\
1171\n\
1172(i.e. all elements that are in either set.)");
1173
1174static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001175set_or(PySetObject *so, PyObject *other)
1176{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001177 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001178 Py_INCREF(Py_NotImplemented);
1179 return Py_NotImplemented;
1180 }
1181 return set_union(so, other);
1182}
1183
1184static PyObject *
1185set_ior(PySetObject *so, PyObject *other)
1186{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001187 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001188 Py_INCREF(Py_NotImplemented);
1189 return Py_NotImplemented;
1190 }
Raymond Hettingerd7946662005-08-01 21:39:29 +00001191 if (set_update_internal(so, other) == -1)
Raymond Hettingera690a992003-11-16 16:17:49 +00001192 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001193 Py_INCREF(so);
1194 return (PyObject *)so;
1195}
1196
1197static PyObject *
1198set_intersection(PySetObject *so, PyObject *other)
1199{
1200 PySetObject *result;
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001201 PyObject *key, *it, *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001202
Raymond Hettingerd8e13382005-08-17 12:27:17 +00001203 if ((PyObject *)so == other)
1204 return set_copy(so);
Raymond Hettingerc991db22005-08-11 07:58:45 +00001205
Raymond Hettingera690a992003-11-16 16:17:49 +00001206 result = (PySetObject *)make_new_set(so->ob_type, NULL);
1207 if (result == NULL)
1208 return NULL;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001209
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001210 if (PyAnySet_CheckExact(other)) {
Martin v. Löwis18e16552006-02-15 17:27:45 +00001211 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001212 setentry *entry;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001213
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001214 if (PySet_GET_SIZE(other) > PySet_GET_SIZE(so)) {
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001215 tmp = (PyObject *)so;
1216 so = (PySetObject *)other;
1217 other = tmp;
1218 }
1219
Raymond Hettingerc991db22005-08-11 07:58:45 +00001220 while (set_next((PySetObject *)other, &pos, &entry)) {
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001221 int rv = set_contains_entry(so, entry);
1222 if (rv == -1) {
1223 Py_DECREF(result);
1224 return NULL;
1225 }
1226 if (rv) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001227 if (set_add_entry(result, entry) == -1) {
Raymond Hettingera3b11e72003-12-31 14:08:58 +00001228 Py_DECREF(result);
1229 return NULL;
1230 }
1231 }
1232 }
1233 return (PyObject *)result;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001234 }
1235
Raymond Hettingera690a992003-11-16 16:17:49 +00001236 it = PyObject_GetIter(other);
1237 if (it == NULL) {
1238 Py_DECREF(result);
1239 return NULL;
1240 }
1241
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001242 while ((key = PyIter_Next(it)) != NULL) {
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001243 int rv;
1244 setentry entry;
1245 long hash = PyObject_Hash(key);
1246
1247 if (hash == -1) {
1248 Py_DECREF(it);
1249 Py_DECREF(result);
1250 Py_DECREF(key);
1251 return NULL;
1252 }
1253 entry.hash = hash;
1254 entry.key = key;
1255 rv = set_contains_entry(so, &entry);
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001256 if (rv == -1) {
1257 Py_DECREF(it);
1258 Py_DECREF(result);
1259 Py_DECREF(key);
1260 return NULL;
1261 }
1262 if (rv) {
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001263 if (set_add_entry(result, &entry) == -1) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001264 Py_DECREF(it);
1265 Py_DECREF(result);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001266 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +00001267 return NULL;
1268 }
1269 }
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001270 Py_DECREF(key);
Raymond Hettingera690a992003-11-16 16:17:49 +00001271 }
1272 Py_DECREF(it);
1273 if (PyErr_Occurred()) {
1274 Py_DECREF(result);
1275 return NULL;
1276 }
1277 return (PyObject *)result;
1278}
1279
1280PyDoc_STRVAR(intersection_doc,
1281"Return the intersection of two sets as a new set.\n\
1282\n\
1283(i.e. all elements that are in both sets.)");
1284
1285static PyObject *
1286set_intersection_update(PySetObject *so, PyObject *other)
1287{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001288 PyObject *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001289
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001290 tmp = set_intersection(so, other);
1291 if (tmp == NULL)
Raymond Hettingera690a992003-11-16 16:17:49 +00001292 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001293 set_swap_bodies(so, (PySetObject *)tmp);
Raymond Hettingera690a992003-11-16 16:17:49 +00001294 Py_DECREF(tmp);
1295 Py_RETURN_NONE;
1296}
1297
1298PyDoc_STRVAR(intersection_update_doc,
1299"Update a set with the intersection of itself and another.");
1300
1301static PyObject *
1302set_and(PySetObject *so, PyObject *other)
1303{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001304 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001305 Py_INCREF(Py_NotImplemented);
1306 return Py_NotImplemented;
1307 }
1308 return set_intersection(so, other);
1309}
1310
1311static PyObject *
1312set_iand(PySetObject *so, PyObject *other)
1313{
1314 PyObject *result;
1315
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001316 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001317 Py_INCREF(Py_NotImplemented);
1318 return Py_NotImplemented;
1319 }
1320 result = set_intersection_update(so, other);
1321 if (result == NULL)
1322 return NULL;
1323 Py_DECREF(result);
1324 Py_INCREF(so);
1325 return (PyObject *)so;
1326}
1327
Neal Norwitz6576bd82005-11-13 18:41:28 +00001328static int
Raymond Hettingerc991db22005-08-11 07:58:45 +00001329set_difference_update_internal(PySetObject *so, PyObject *other)
1330{
1331 if ((PyObject *)so == other)
1332 return set_clear_internal(so);
1333
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001334 if (PyAnySet_CheckExact(other)) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001335 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001336 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001337
1338 while (set_next((PySetObject *)other, &pos, &entry))
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001339 if (set_discard_entry(so, entry) == -1)
1340 return -1;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001341 } else {
1342 PyObject *key, *it;
1343 it = PyObject_GetIter(other);
1344 if (it == NULL)
1345 return -1;
1346
1347 while ((key = PyIter_Next(it)) != NULL) {
1348 if (set_discard_key(so, key) == -1) {
1349 Py_DECREF(it);
1350 Py_DECREF(key);
1351 return -1;
1352 }
1353 Py_DECREF(key);
1354 }
1355 Py_DECREF(it);
1356 if (PyErr_Occurred())
1357 return -1;
1358 }
1359 /* If more than 1/5 are dummies, then resize them away. */
1360 if ((so->fill - so->used) * 5 < so->mask)
1361 return 0;
1362 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
1363}
1364
Raymond Hettingera690a992003-11-16 16:17:49 +00001365static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001366set_difference_update(PySetObject *so, PyObject *other)
1367{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001368 if (set_difference_update_internal(so, other) != -1)
Raymond Hettingerbc841a12005-08-07 13:02:53 +00001369 Py_RETURN_NONE;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001370 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001371}
1372
1373PyDoc_STRVAR(difference_update_doc,
1374"Remove all elements of another set from this set.");
1375
1376static PyObject *
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001377set_difference(PySetObject *so, PyObject *other)
1378{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001379 PyObject *result;
1380 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001381 Py_ssize_t pos = 0;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001382
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001383 if (!PyAnySet_CheckExact(other) && !PyDict_CheckExact(other)) {
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001384 result = set_copy(so);
1385 if (result == NULL)
Raymond Hettingerc991db22005-08-11 07:58:45 +00001386 return NULL;
1387 if (set_difference_update_internal((PySetObject *)result, other) != -1)
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001388 return result;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001389 Py_DECREF(result);
1390 return NULL;
1391 }
1392
1393 result = make_new_set(so->ob_type, NULL);
1394 if (result == NULL)
1395 return NULL;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001396
Raymond Hettingerdb67aef2007-02-01 21:02:59 +00001397 if (PyDict_CheckExact(other)) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001398 while (set_next(so, &pos, &entry)) {
1399 setentry entrycopy;
1400 entrycopy.hash = entry->hash;
1401 entrycopy.key = entry->key;
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +00001402 if (!_PyDict_Contains(other, entry->key, entry->hash)) {
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001403 if (set_add_entry((PySetObject *)result, &entrycopy) == -1) {
1404 Py_DECREF(result);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001405 return NULL;
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001406 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001407 }
1408 }
1409 return result;
1410 }
1411
Raymond Hettingerc991db22005-08-11 07:58:45 +00001412 while (set_next(so, &pos, &entry)) {
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001413 int rv = set_contains_entry((PySetObject *)other, entry);
1414 if (rv == -1) {
1415 Py_DECREF(result);
1416 return NULL;
1417 }
1418 if (!rv) {
1419 if (set_add_entry((PySetObject *)result, entry) == -1) {
1420 Py_DECREF(result);
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001421 return NULL;
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001422 }
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001423 }
1424 }
1425 return result;
1426}
1427
1428PyDoc_STRVAR(difference_doc,
1429"Return the difference of two sets as a new set.\n\
1430\n\
1431(i.e. all elements that are in this set but not the other.)");
1432static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001433set_sub(PySetObject *so, PyObject *other)
1434{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001435 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001436 Py_INCREF(Py_NotImplemented);
1437 return Py_NotImplemented;
1438 }
1439 return set_difference(so, other);
1440}
1441
1442static PyObject *
1443set_isub(PySetObject *so, PyObject *other)
1444{
1445 PyObject *result;
1446
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001447 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001448 Py_INCREF(Py_NotImplemented);
1449 return Py_NotImplemented;
1450 }
1451 result = set_difference_update(so, other);
1452 if (result == NULL)
1453 return NULL;
1454 Py_DECREF(result);
1455 Py_INCREF(so);
1456 return (PyObject *)so;
1457}
1458
1459static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001460set_symmetric_difference_update(PySetObject *so, PyObject *other)
1461{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001462 PySetObject *otherset;
1463 PyObject *key;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001464 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001465 setentry *entry;
1466
1467 if ((PyObject *)so == other)
1468 return set_clear(so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001469
Raymond Hettingerdb67aef2007-02-01 21:02:59 +00001470 if (PyDict_CheckExact(other)) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001471 PyObject *value;
1472 int rv;
Raymond Hettingerd6fc72a2007-02-19 02:03:19 +00001473 long hash;
1474 while (_PyDict_Next(other, &pos, &key, &value, &hash)) {
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001475 setentry an_entry;
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001476
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001477 an_entry.hash = hash;
1478 an_entry.key = key;
1479 rv = set_discard_entry(so, &an_entry);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001480 if (rv == -1)
1481 return NULL;
1482 if (rv == DISCARD_NOTFOUND) {
Raymond Hettingerf31e1752006-12-08 03:17:18 +00001483 if (set_add_entry(so, &an_entry) == -1)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001484 return NULL;
1485 }
1486 }
1487 Py_RETURN_NONE;
1488 }
1489
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001490 if (PyAnySet_CheckExact(other)) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001491 Py_INCREF(other);
1492 otherset = (PySetObject *)other;
1493 } else {
Raymond Hettingera690a992003-11-16 16:17:49 +00001494 otherset = (PySetObject *)make_new_set(so->ob_type, other);
1495 if (otherset == NULL)
1496 return NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001497 }
1498
Raymond Hettingerc991db22005-08-11 07:58:45 +00001499 while (set_next(otherset, &pos, &entry)) {
1500 int rv = set_discard_entry(so, entry);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001501 if (rv == -1) {
Neal Norwitz04e39ec2006-07-17 00:57:15 +00001502 Py_DECREF(otherset);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001503 return NULL;
1504 }
1505 if (rv == DISCARD_NOTFOUND) {
Raymond Hettingerc991db22005-08-11 07:58:45 +00001506 if (set_add_entry(so, entry) == -1) {
Neal Norwitz04e39ec2006-07-17 00:57:15 +00001507 Py_DECREF(otherset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001508 return NULL;
1509 }
1510 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001511 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001512 Py_DECREF(otherset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001513 Py_RETURN_NONE;
1514}
1515
1516PyDoc_STRVAR(symmetric_difference_update_doc,
1517"Update a set with the symmetric difference of itself and another.");
1518
1519static PyObject *
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001520set_symmetric_difference(PySetObject *so, PyObject *other)
1521{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001522 PyObject *rv;
1523 PySetObject *otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001524
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001525 otherset = (PySetObject *)make_new_set(so->ob_type, other);
1526 if (otherset == NULL)
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001527 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001528 rv = set_symmetric_difference_update(otherset, (PyObject *)so);
1529 if (rv == NULL)
1530 return NULL;
1531 Py_DECREF(rv);
1532 return (PyObject *)otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001533}
1534
1535PyDoc_STRVAR(symmetric_difference_doc,
1536"Return the symmetric difference of two sets as a new set.\n\
1537\n\
1538(i.e. all elements that are in exactly one of the sets.)");
1539
1540static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001541set_xor(PySetObject *so, PyObject *other)
1542{
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001543 if (!PyAnySet_Check(so) || !PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001544 Py_INCREF(Py_NotImplemented);
1545 return Py_NotImplemented;
1546 }
1547 return set_symmetric_difference(so, other);
1548}
1549
1550static PyObject *
1551set_ixor(PySetObject *so, PyObject *other)
1552{
1553 PyObject *result;
1554
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001555 if (!PyAnySet_Check(other)) {
Raymond Hettingera690a992003-11-16 16:17:49 +00001556 Py_INCREF(Py_NotImplemented);
1557 return Py_NotImplemented;
1558 }
1559 result = set_symmetric_difference_update(so, other);
1560 if (result == NULL)
1561 return NULL;
1562 Py_DECREF(result);
1563 Py_INCREF(so);
1564 return (PyObject *)so;
1565}
1566
1567static PyObject *
1568set_issubset(PySetObject *so, PyObject *other)
1569{
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001570 setentry *entry;
Martin v. Löwis18e16552006-02-15 17:27:45 +00001571 Py_ssize_t pos = 0;
Raymond Hettingera690a992003-11-16 16:17:49 +00001572
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001573 if (!PyAnySet_CheckExact(other)) {
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001574 PyObject *tmp, *result;
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001575 tmp = make_new_set(&PySet_Type, other);
1576 if (tmp == NULL)
1577 return NULL;
1578 result = set_issubset(so, tmp);
1579 Py_DECREF(tmp);
1580 return result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001581 }
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001582 if (PySet_GET_SIZE(so) > PySet_GET_SIZE(other))
Raymond Hettingera690a992003-11-16 16:17:49 +00001583 Py_RETURN_FALSE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001584
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001585 while (set_next(so, &pos, &entry)) {
Raymond Hettingerc563a1c2006-09-07 02:42:48 +00001586 int rv = set_contains_entry((PySetObject *)other, entry);
1587 if (rv == -1)
1588 return NULL;
1589 if (!rv)
Raymond Hettingera690a992003-11-16 16:17:49 +00001590 Py_RETURN_FALSE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001591 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001592 Py_RETURN_TRUE;
1593}
1594
1595PyDoc_STRVAR(issubset_doc, "Report whether another set contains this set.");
1596
1597static PyObject *
1598set_issuperset(PySetObject *so, PyObject *other)
1599{
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001600 PyObject *tmp, *result;
1601
Raymond Hettinger0e7a6322007-02-08 00:50:39 +00001602 if (!PyAnySet_CheckExact(other)) {
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001603 tmp = make_new_set(&PySet_Type, other);
1604 if (tmp == NULL)
1605 return NULL;
1606 result = set_issuperset(so, tmp);
1607 Py_DECREF(tmp);
1608 return result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001609 }
1610 return set_issubset((PySetObject *)other, (PyObject *)so);
1611}
1612
1613PyDoc_STRVAR(issuperset_doc, "Report whether this set contains another set.");
1614
Raymond Hettingera690a992003-11-16 16:17:49 +00001615static PyObject *
1616set_richcompare(PySetObject *v, PyObject *w, int op)
1617{
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001618 PyObject *r1, *r2;
1619
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001620 if(!PyAnySet_Check(w)) {
1621 if (op == Py_EQ)
1622 Py_RETURN_FALSE;
1623 if (op == Py_NE)
1624 Py_RETURN_TRUE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001625 PyErr_SetString(PyExc_TypeError, "can only compare to a set");
1626 return NULL;
1627 }
1628 switch (op) {
1629 case Py_EQ:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001630 if (PySet_GET_SIZE(v) != PySet_GET_SIZE(w))
Raymond Hettingera690a992003-11-16 16:17:49 +00001631 Py_RETURN_FALSE;
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00001632 if (v->hash != -1 &&
1633 ((PySetObject *)w)->hash != -1 &&
1634 v->hash != ((PySetObject *)w)->hash)
1635 Py_RETURN_FALSE;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001636 return set_issubset(v, w);
1637 case Py_NE:
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00001638 r1 = set_richcompare(v, w, Py_EQ);
1639 if (r1 == NULL)
1640 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001641 r2 = PyBool_FromLong(PyObject_Not(r1));
1642 Py_DECREF(r1);
1643 return r2;
1644 case Py_LE:
1645 return set_issubset(v, w);
1646 case Py_GE:
1647 return set_issuperset(v, w);
1648 case Py_LT:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001649 if (PySet_GET_SIZE(v) >= PySet_GET_SIZE(w))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001650 Py_RETURN_FALSE;
1651 return set_issubset(v, w);
1652 case Py_GT:
Raymond Hettingerbeb31012005-08-16 03:47:52 +00001653 if (PySet_GET_SIZE(v) <= PySet_GET_SIZE(w))
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001654 Py_RETURN_FALSE;
1655 return set_issuperset(v, w);
Raymond Hettingera690a992003-11-16 16:17:49 +00001656 }
1657 Py_INCREF(Py_NotImplemented);
1658 return Py_NotImplemented;
1659}
1660
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001661static int
Georg Brandl347b3002006-03-30 11:57:00 +00001662set_nocmp(PyObject *self, PyObject *other)
Raymond Hettingered6c1ef2005-08-13 08:28:03 +00001663{
1664 PyErr_SetString(PyExc_TypeError, "cannot compare sets using cmp()");
1665 return -1;
1666}
1667
Raymond Hettingera690a992003-11-16 16:17:49 +00001668static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001669set_add(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001670{
Raymond Hettingerc991db22005-08-11 07:58:45 +00001671 if (set_add_key(so, key) == -1)
Raymond Hettingera690a992003-11-16 16:17:49 +00001672 return NULL;
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001673 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001674}
1675
1676PyDoc_STRVAR(add_doc,
1677"Add an element to a set.\n\
1678\n\
1679This has no effect if the element is already present.");
1680
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001681static int
1682set_contains(PySetObject *so, PyObject *key)
1683{
1684 PyObject *tmpkey;
1685 int rv;
1686
1687 rv = set_contains_key(so, key);
1688 if (rv == -1) {
1689 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1690 return -1;
1691 PyErr_Clear();
1692 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1693 if (tmpkey == NULL)
1694 return -1;
1695 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
1696 rv = set_contains(so, tmpkey);
1697 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
1698 Py_DECREF(tmpkey);
1699 }
1700 return rv;
1701}
1702
1703static PyObject *
1704set_direct_contains(PySetObject *so, PyObject *key)
1705{
1706 long result;
1707
1708 result = set_contains(so, key);
1709 if (result == -1)
1710 return NULL;
1711 return PyBool_FromLong(result);
1712}
1713
1714PyDoc_STRVAR(contains_doc, "x.__contains__(y) <==> y in x.");
1715
Raymond Hettingera690a992003-11-16 16:17:49 +00001716static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001717set_remove(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001718{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001719 PyObject *tmpkey, *result;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001720 int rv;
Raymond Hettingerbfd334a2003-11-22 03:55:23 +00001721
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001722 rv = set_discard_key(so, key);
1723 if (rv == -1) {
1724 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1725 return NULL;
1726 PyErr_Clear();
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001727 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1728 if (tmpkey == NULL)
Raymond Hettingerbfd334a2003-11-22 03:55:23 +00001729 return NULL;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001730 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001731 result = set_remove(so, tmpkey);
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001732 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001733 Py_DECREF(tmpkey);
Raymond Hettinger0deab622003-12-13 18:53:18 +00001734 return result;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001735 } else if (rv == DISCARD_NOTFOUND) {
Raymond Hettinger9c14ffb2006-12-08 04:57:50 +00001736 set_key_error(key);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001737 return NULL;
1738 }
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001739 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001740}
1741
1742PyDoc_STRVAR(remove_doc,
1743"Remove an element from a set; it must be a member.\n\
1744\n\
1745If the element is not a member, raise a KeyError.");
1746
1747static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001748set_discard(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001749{
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001750 PyObject *tmpkey, *result;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001751 int rv;
Raymond Hettinger0deab622003-12-13 18:53:18 +00001752
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001753 rv = set_discard_key(so, key);
1754 if (rv == -1) {
1755 if (!PyAnySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1756 return NULL;
1757 PyErr_Clear();
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001758 tmpkey = make_new_set(&PyFrozenSet_Type, NULL);
1759 if (tmpkey == NULL)
Raymond Hettinger0deab622003-12-13 18:53:18 +00001760 return NULL;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001761 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001762 result = set_discard(so, tmpkey);
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001763 set_swap_bodies((PySetObject *)tmpkey, (PySetObject *)key);
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001764 Py_DECREF(tmpkey);
Raymond Hettinger0deab622003-12-13 18:53:18 +00001765 return result;
1766 }
Raymond Hettinger438e02d2003-12-13 19:38:47 +00001767 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001768}
1769
1770PyDoc_STRVAR(discard_doc,
1771"Remove an element from a set if it is a member.\n\
1772\n\
1773If the element is not a member, do nothing.");
1774
1775static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001776set_reduce(PySetObject *so)
1777{
Raymond Hettinger15056a52004-11-09 07:25:31 +00001778 PyObject *keys=NULL, *args=NULL, *result=NULL, *dict=NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001779
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001780 keys = PySequence_List((PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001781 if (keys == NULL)
1782 goto done;
1783 args = PyTuple_Pack(1, keys);
1784 if (args == NULL)
1785 goto done;
Raymond Hettinger15056a52004-11-09 07:25:31 +00001786 dict = PyObject_GetAttrString((PyObject *)so, "__dict__");
1787 if (dict == NULL) {
1788 PyErr_Clear();
1789 dict = Py_None;
1790 Py_INCREF(dict);
1791 }
1792 result = PyTuple_Pack(3, so->ob_type, args, dict);
Raymond Hettingera690a992003-11-16 16:17:49 +00001793done:
1794 Py_XDECREF(args);
1795 Py_XDECREF(keys);
Raymond Hettinger15056a52004-11-09 07:25:31 +00001796 Py_XDECREF(dict);
Raymond Hettingera690a992003-11-16 16:17:49 +00001797 return result;
1798}
1799
1800PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
1801
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001802static int
1803set_init(PySetObject *self, PyObject *args, PyObject *kwds)
1804{
1805 PyObject *iterable = NULL;
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001806
1807 if (!PyAnySet_Check(self))
1808 return -1;
1809 if (!PyArg_UnpackTuple(args, self->ob_type->tp_name, 0, 1, &iterable))
1810 return -1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001811 set_clear_internal(self);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001812 self->hash = -1;
1813 if (iterable == NULL)
1814 return 0;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001815 return set_update_internal(self, iterable);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001816}
1817
Raymond Hettingera690a992003-11-16 16:17:49 +00001818static PySequenceMethods set_as_sequence = {
Georg Brandl347b3002006-03-30 11:57:00 +00001819 set_len, /* sq_length */
Raymond Hettingera690a992003-11-16 16:17:49 +00001820 0, /* sq_concat */
1821 0, /* sq_repeat */
1822 0, /* sq_item */
1823 0, /* sq_slice */
1824 0, /* sq_ass_item */
1825 0, /* sq_ass_slice */
1826 (objobjproc)set_contains, /* sq_contains */
1827};
1828
1829/* set object ********************************************************/
1830
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00001831#ifdef Py_DEBUG
1832static PyObject *test_c_api(PySetObject *so);
1833
1834PyDoc_STRVAR(test_c_api_doc, "Exercises C API. Returns True.\n\
1835All is well if assertions don't fail.");
1836#endif
1837
Raymond Hettingera690a992003-11-16 16:17:49 +00001838static PyMethodDef set_methods[] = {
1839 {"add", (PyCFunction)set_add, METH_O,
1840 add_doc},
1841 {"clear", (PyCFunction)set_clear, METH_NOARGS,
1842 clear_doc},
Raymond Hettinger0deab622003-12-13 18:53:18 +00001843 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001844 contains_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001845 {"copy", (PyCFunction)set_copy, METH_NOARGS,
1846 copy_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001847 {"discard", (PyCFunction)set_discard, METH_O,
1848 discard_doc},
1849 {"difference", (PyCFunction)set_difference, METH_O,
1850 difference_doc},
1851 {"difference_update", (PyCFunction)set_difference_update, METH_O,
1852 difference_update_doc},
1853 {"intersection",(PyCFunction)set_intersection, METH_O,
1854 intersection_doc},
1855 {"intersection_update",(PyCFunction)set_intersection_update, METH_O,
1856 intersection_update_doc},
1857 {"issubset", (PyCFunction)set_issubset, METH_O,
1858 issubset_doc},
1859 {"issuperset", (PyCFunction)set_issuperset, METH_O,
1860 issuperset_doc},
1861 {"pop", (PyCFunction)set_pop, METH_NOARGS,
1862 pop_doc},
1863 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
1864 reduce_doc},
1865 {"remove", (PyCFunction)set_remove, METH_O,
1866 remove_doc},
1867 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
1868 symmetric_difference_doc},
1869 {"symmetric_difference_update",(PyCFunction)set_symmetric_difference_update, METH_O,
1870 symmetric_difference_update_doc},
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00001871#ifdef Py_DEBUG
1872 {"test_c_api", (PyCFunction)test_c_api, METH_NOARGS,
1873 test_c_api_doc},
1874#endif
Raymond Hettingera690a992003-11-16 16:17:49 +00001875 {"union", (PyCFunction)set_union, METH_O,
1876 union_doc},
Raymond Hettingera38123e2003-11-24 22:18:49 +00001877 {"update", (PyCFunction)set_update, METH_O,
1878 update_doc},
Raymond Hettingera690a992003-11-16 16:17:49 +00001879 {NULL, NULL} /* sentinel */
1880};
1881
1882static PyNumberMethods set_as_number = {
1883 0, /*nb_add*/
1884 (binaryfunc)set_sub, /*nb_subtract*/
1885 0, /*nb_multiply*/
1886 0, /*nb_divide*/
1887 0, /*nb_remainder*/
1888 0, /*nb_divmod*/
1889 0, /*nb_power*/
1890 0, /*nb_negative*/
1891 0, /*nb_positive*/
1892 0, /*nb_absolute*/
1893 0, /*nb_nonzero*/
1894 0, /*nb_invert*/
1895 0, /*nb_lshift*/
1896 0, /*nb_rshift*/
1897 (binaryfunc)set_and, /*nb_and*/
1898 (binaryfunc)set_xor, /*nb_xor*/
1899 (binaryfunc)set_or, /*nb_or*/
1900 0, /*nb_coerce*/
1901 0, /*nb_int*/
1902 0, /*nb_long*/
1903 0, /*nb_float*/
1904 0, /*nb_oct*/
1905 0, /*nb_hex*/
1906 0, /*nb_inplace_add*/
1907 (binaryfunc)set_isub, /*nb_inplace_subtract*/
1908 0, /*nb_inplace_multiply*/
1909 0, /*nb_inplace_divide*/
1910 0, /*nb_inplace_remainder*/
1911 0, /*nb_inplace_power*/
1912 0, /*nb_inplace_lshift*/
1913 0, /*nb_inplace_rshift*/
1914 (binaryfunc)set_iand, /*nb_inplace_and*/
1915 (binaryfunc)set_ixor, /*nb_inplace_xor*/
1916 (binaryfunc)set_ior, /*nb_inplace_or*/
1917};
1918
1919PyDoc_STRVAR(set_doc,
1920"set(iterable) --> set object\n\
1921\n\
Andrew M. Kuchling52740be2006-07-29 15:10:32 +00001922Build an unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00001923
1924PyTypeObject PySet_Type = {
1925 PyObject_HEAD_INIT(&PyType_Type)
1926 0, /* ob_size */
1927 "set", /* tp_name */
1928 sizeof(PySetObject), /* tp_basicsize */
1929 0, /* tp_itemsize */
1930 /* methods */
1931 (destructor)set_dealloc, /* tp_dealloc */
1932 (printfunc)set_tp_print, /* tp_print */
1933 0, /* tp_getattr */
1934 0, /* tp_setattr */
Georg Brandl347b3002006-03-30 11:57:00 +00001935 set_nocmp, /* tp_compare */
Raymond Hettingera690a992003-11-16 16:17:49 +00001936 (reprfunc)set_repr, /* tp_repr */
1937 &set_as_number, /* tp_as_number */
1938 &set_as_sequence, /* tp_as_sequence */
1939 0, /* tp_as_mapping */
1940 set_nohash, /* tp_hash */
1941 0, /* tp_call */
1942 0, /* tp_str */
1943 PyObject_GenericGetAttr, /* tp_getattro */
1944 0, /* tp_setattro */
1945 0, /* tp_as_buffer */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001946 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_CHECKTYPES |
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001947 Py_TPFLAGS_BASETYPE, /* tp_flags */
Raymond Hettingera690a992003-11-16 16:17:49 +00001948 set_doc, /* tp_doc */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001949 (traverseproc)set_traverse, /* tp_traverse */
Raymond Hettingerfe889f32005-08-06 05:43:39 +00001950 (inquiry)set_clear_internal, /* tp_clear */
Raymond Hettingera690a992003-11-16 16:17:49 +00001951 (richcmpfunc)set_richcompare, /* tp_richcompare */
Raymond Hettinger691d8052004-05-30 07:26:47 +00001952 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001953 (getiterfunc)set_iter, /* tp_iter */
Raymond Hettingera690a992003-11-16 16:17:49 +00001954 0, /* tp_iternext */
1955 set_methods, /* tp_methods */
1956 0, /* tp_members */
1957 0, /* tp_getset */
1958 0, /* tp_base */
1959 0, /* tp_dict */
1960 0, /* tp_descr_get */
1961 0, /* tp_descr_set */
1962 0, /* tp_dictoffset */
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001963 (initproc)set_init, /* tp_init */
Raymond Hettingera690a992003-11-16 16:17:49 +00001964 PyType_GenericAlloc, /* tp_alloc */
1965 set_new, /* tp_new */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00001966 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00001967};
1968
1969/* frozenset object ********************************************************/
1970
1971
1972static PyMethodDef frozenset_methods[] = {
Raymond Hettinger0deab622003-12-13 18:53:18 +00001973 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001974 contains_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001975 {"copy", (PyCFunction)frozenset_copy, METH_NOARGS,
Raymond Hettingera690a992003-11-16 16:17:49 +00001976 copy_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001977 {"difference", (PyCFunction)set_difference, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001978 difference_doc},
1979 {"intersection",(PyCFunction)set_intersection, METH_O,
1980 intersection_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001981 {"issubset", (PyCFunction)set_issubset, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001982 issubset_doc},
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001983 {"issuperset", (PyCFunction)set_issuperset, METH_O,
Raymond Hettingera690a992003-11-16 16:17:49 +00001984 issuperset_doc},
1985 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
1986 reduce_doc},
1987 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
1988 symmetric_difference_doc},
1989 {"union", (PyCFunction)set_union, METH_O,
1990 union_doc},
1991 {NULL, NULL} /* sentinel */
1992};
1993
1994static PyNumberMethods frozenset_as_number = {
1995 0, /*nb_add*/
1996 (binaryfunc)set_sub, /*nb_subtract*/
1997 0, /*nb_multiply*/
1998 0, /*nb_divide*/
1999 0, /*nb_remainder*/
2000 0, /*nb_divmod*/
2001 0, /*nb_power*/
2002 0, /*nb_negative*/
2003 0, /*nb_positive*/
2004 0, /*nb_absolute*/
2005 0, /*nb_nonzero*/
2006 0, /*nb_invert*/
2007 0, /*nb_lshift*/
2008 0, /*nb_rshift*/
2009 (binaryfunc)set_and, /*nb_and*/
2010 (binaryfunc)set_xor, /*nb_xor*/
2011 (binaryfunc)set_or, /*nb_or*/
2012};
2013
2014PyDoc_STRVAR(frozenset_doc,
2015"frozenset(iterable) --> frozenset object\n\
2016\n\
Andrew M. Kuchling52740be2006-07-29 15:10:32 +00002017Build an immutable unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00002018
2019PyTypeObject PyFrozenSet_Type = {
2020 PyObject_HEAD_INIT(&PyType_Type)
2021 0, /* ob_size */
2022 "frozenset", /* tp_name */
2023 sizeof(PySetObject), /* tp_basicsize */
Raymond Hettingera3b11e72003-12-31 14:08:58 +00002024 0, /* tp_itemsize */
2025 /* methods */
Raymond Hettingera690a992003-11-16 16:17:49 +00002026 (destructor)set_dealloc, /* tp_dealloc */
2027 (printfunc)set_tp_print, /* tp_print */
2028 0, /* tp_getattr */
2029 0, /* tp_setattr */
Georg Brandl347b3002006-03-30 11:57:00 +00002030 set_nocmp, /* tp_compare */
Raymond Hettingera690a992003-11-16 16:17:49 +00002031 (reprfunc)set_repr, /* tp_repr */
2032 &frozenset_as_number, /* tp_as_number */
2033 &set_as_sequence, /* tp_as_sequence */
2034 0, /* tp_as_mapping */
2035 frozenset_hash, /* tp_hash */
2036 0, /* tp_call */
2037 0, /* tp_str */
2038 PyObject_GenericGetAttr, /* tp_getattro */
2039 0, /* tp_setattro */
2040 0, /* tp_as_buffer */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002041 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC | Py_TPFLAGS_CHECKTYPES |
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00002042 Py_TPFLAGS_BASETYPE, /* tp_flags */
Raymond Hettingera690a992003-11-16 16:17:49 +00002043 frozenset_doc, /* tp_doc */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002044 (traverseproc)set_traverse, /* tp_traverse */
Raymond Hettingerfe889f32005-08-06 05:43:39 +00002045 (inquiry)set_clear_internal, /* tp_clear */
Raymond Hettingera690a992003-11-16 16:17:49 +00002046 (richcmpfunc)set_richcompare, /* tp_richcompare */
Raymond Hettinger691d8052004-05-30 07:26:47 +00002047 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
Raymond Hettingera690a992003-11-16 16:17:49 +00002048 (getiterfunc)set_iter, /* tp_iter */
2049 0, /* tp_iternext */
2050 frozenset_methods, /* tp_methods */
2051 0, /* tp_members */
2052 0, /* tp_getset */
2053 0, /* tp_base */
2054 0, /* tp_dict */
2055 0, /* tp_descr_get */
2056 0, /* tp_descr_set */
2057 0, /* tp_dictoffset */
2058 0, /* tp_init */
2059 PyType_GenericAlloc, /* tp_alloc */
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00002060 frozenset_new, /* tp_new */
Raymond Hettingerbb999b52005-06-18 21:00:26 +00002061 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00002062};
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002063
2064
2065/***** C API functions *************************************************/
2066
2067PyObject *
2068PySet_New(PyObject *iterable)
2069{
2070 return make_new_set(&PySet_Type, iterable);
2071}
2072
2073PyObject *
2074PyFrozenSet_New(PyObject *iterable)
2075{
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002076 PyObject *args, *result;
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002077
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002078 if (iterable == NULL)
2079 args = PyTuple_New(0);
2080 else
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002081 args = PyTuple_Pack(1, iterable);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002082 if (args == NULL)
2083 return NULL;
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002084 result = frozenset_new(&PyFrozenSet_Type, args, NULL);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002085 Py_DECREF(args);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002086 return result;
2087}
2088
Neal Norwitz8c49c822006-03-04 18:41:19 +00002089Py_ssize_t
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002090PySet_Size(PyObject *anyset)
2091{
2092 if (!PyAnySet_Check(anyset)) {
2093 PyErr_BadInternalCall();
2094 return -1;
2095 }
Raymond Hettinger9c1491f2005-08-24 00:24:40 +00002096 return PySet_GET_SIZE(anyset);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002097}
2098
2099int
Barry Warsaw176014f2006-03-30 22:45:35 +00002100PySet_Clear(PyObject *set)
2101{
2102 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2103 PyErr_BadInternalCall();
2104 return -1;
2105 }
2106 return set_clear_internal((PySetObject *)set);
2107}
2108
2109int
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002110PySet_Contains(PyObject *anyset, PyObject *key)
2111{
2112 if (!PyAnySet_Check(anyset)) {
2113 PyErr_BadInternalCall();
2114 return -1;
2115 }
2116 return set_contains_key((PySetObject *)anyset, key);
2117}
2118
2119int
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002120PySet_Discard(PyObject *set, PyObject *key)
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002121{
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002122 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002123 PyErr_BadInternalCall();
2124 return -1;
2125 }
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002126 return set_discard_key((PySetObject *)set, key);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002127}
2128
2129int
2130PySet_Add(PyObject *set, PyObject *key)
2131{
2132 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2133 PyErr_BadInternalCall();
2134 return -1;
2135 }
2136 return set_add_key((PySetObject *)set, key);
2137}
2138
Barry Warsaw176014f2006-03-30 22:45:35 +00002139int
Raymond Hettinger0bbbfc42007-03-20 21:27:24 +00002140_PySet_Next(PyObject *set, Py_ssize_t *pos, PyObject **key)
Barry Warsaw176014f2006-03-30 22:45:35 +00002141{
2142 setentry *entry_ptr;
2143
2144 if (!PyAnySet_Check(set)) {
2145 PyErr_BadInternalCall();
2146 return -1;
2147 }
2148 if (set_next((PySetObject *)set, pos, &entry_ptr) == 0)
2149 return 0;
Raymond Hettinger0bbbfc42007-03-20 21:27:24 +00002150 *key = entry_ptr->key;
2151 return 1;
2152}
2153
2154int
2155_PySet_NextEntry(PyObject *set, Py_ssize_t *pos, PyObject **key, long *hash)
2156{
2157 setentry *entry;
2158
2159 if (!PyAnySet_Check(set)) {
2160 PyErr_BadInternalCall();
2161 return -1;
2162 }
2163 if (set_next((PySetObject *)set, pos, &entry) == 0)
2164 return 0;
2165 *key = entry->key;
2166 *hash = entry->hash;
Barry Warsaw176014f2006-03-30 22:45:35 +00002167 return 1;
2168}
2169
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002170PyObject *
2171PySet_Pop(PyObject *set)
2172{
2173 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2174 PyErr_BadInternalCall();
2175 return NULL;
2176 }
2177 return set_pop((PySetObject *)set);
2178}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002179
Barry Warsaw176014f2006-03-30 22:45:35 +00002180int
2181_PySet_Update(PyObject *set, PyObject *iterable)
2182{
2183 if (!PyType_IsSubtype(set->ob_type, &PySet_Type)) {
2184 PyErr_BadInternalCall();
2185 return -1;
2186 }
2187 return set_update_internal((PySetObject *)set, iterable);
2188}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002189
2190#ifdef Py_DEBUG
2191
2192/* Test code to be called with any three element set.
2193 Returns True and original set is restored. */
2194
2195#define assertRaises(call_return_value, exception) \
2196 do { \
2197 assert(call_return_value); \
2198 assert(PyErr_ExceptionMatches(exception)); \
2199 PyErr_Clear(); \
2200 } while(0)
2201
2202static PyObject *
2203test_c_api(PySetObject *so)
2204{
Neal Norwitz0f2783c2006-06-19 05:40:44 +00002205 Py_ssize_t count;
Barry Warsaw176014f2006-03-30 22:45:35 +00002206 char *s;
2207 Py_ssize_t i;
2208 PyObject *elem, *dup, *t, *f, *dup2;
2209 PyObject *ob = (PyObject *)so;
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002210
2211 /* Verify preconditions and exercise type/size checks */
2212 assert(PyAnySet_Check(ob));
2213 assert(PyAnySet_CheckExact(ob));
2214 assert(!PyFrozenSet_CheckExact(ob));
2215 assert(PySet_Size(ob) == 3);
2216 assert(PySet_GET_SIZE(ob) == 3);
2217
2218 /* Raise TypeError for non-iterable constructor arguments */
2219 assertRaises(PySet_New(Py_None) == NULL, PyExc_TypeError);
2220 assertRaises(PyFrozenSet_New(Py_None) == NULL, PyExc_TypeError);
2221
2222 /* Raise TypeError for unhashable key */
2223 dup = PySet_New(ob);
2224 assertRaises(PySet_Discard(ob, dup) == -1, PyExc_TypeError);
2225 assertRaises(PySet_Contains(ob, dup) == -1, PyExc_TypeError);
2226 assertRaises(PySet_Add(ob, dup) == -1, PyExc_TypeError);
2227
2228 /* Exercise successful pop, contains, add, and discard */
2229 elem = PySet_Pop(ob);
2230 assert(PySet_Contains(ob, elem) == 0);
2231 assert(PySet_GET_SIZE(ob) == 2);
2232 assert(PySet_Add(ob, elem) == 0);
2233 assert(PySet_Contains(ob, elem) == 1);
2234 assert(PySet_GET_SIZE(ob) == 3);
2235 assert(PySet_Discard(ob, elem) == 1);
2236 assert(PySet_GET_SIZE(ob) == 2);
2237 assert(PySet_Discard(ob, elem) == 0);
2238 assert(PySet_GET_SIZE(ob) == 2);
2239
Barry Warsaw176014f2006-03-30 22:45:35 +00002240 /* Exercise clear */
2241 dup2 = PySet_New(dup);
2242 assert(PySet_Clear(dup2) == 0);
2243 assert(PySet_Size(dup2) == 0);
2244 Py_DECREF(dup2);
2245
2246 /* Raise SystemError on clear or update of frozen set */
2247 f = PyFrozenSet_New(dup);
2248 assertRaises(PySet_Clear(f) == -1, PyExc_SystemError);
2249 assertRaises(_PySet_Update(f, dup) == -1, PyExc_SystemError);
2250 Py_DECREF(f);
2251
2252 /* Exercise direct iteration */
2253 i = 0, count = 0;
2254 while (_PySet_Next((PyObject *)dup, &i, &elem)) {
2255 s = PyString_AsString(elem);
2256 assert(s && (s[0] == 'a' || s[0] == 'b' || s[0] == 'c'));
2257 count++;
2258 }
2259 assert(count == 3);
2260
2261 /* Exercise updates */
2262 dup2 = PySet_New(NULL);
2263 assert(_PySet_Update(dup2, dup) == 0);
2264 assert(PySet_Size(dup2) == 3);
2265 assert(_PySet_Update(dup2, dup) == 0);
2266 assert(PySet_Size(dup2) == 3);
2267 Py_DECREF(dup2);
2268
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002269 /* Raise SystemError when self argument is not a set or frozenset. */
2270 t = PyTuple_New(0);
2271 assertRaises(PySet_Size(t) == -1, PyExc_SystemError);
2272 assertRaises(PySet_Contains(t, elem) == -1, PyExc_SystemError);
2273 Py_DECREF(t);
2274
2275 /* Raise SystemError when self argument is not a set. */
2276 f = PyFrozenSet_New(dup);
2277 assert(PySet_Size(f) == 3);
2278 assert(PyFrozenSet_CheckExact(f));
2279 assertRaises(PySet_Add(f, elem) == -1, PyExc_SystemError);
2280 assertRaises(PySet_Discard(f, elem) == -1, PyExc_SystemError);
2281 assertRaises(PySet_Pop(f) == NULL, PyExc_SystemError);
2282 Py_DECREF(f);
2283
2284 /* Raise KeyError when popping from an empty set */
Raymond Hettingerd8e13382005-08-17 12:27:17 +00002285 assert(PyNumber_InPlaceSubtract(ob, ob) == ob);
2286 Py_DECREF(ob);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002287 assert(PySet_GET_SIZE(ob) == 0);
2288 assertRaises(PySet_Pop(ob) == NULL, PyExc_KeyError);
2289
Raymond Hettingerd8e13382005-08-17 12:27:17 +00002290 /* Restore the set from the copy using the PyNumber API */
2291 assert(PyNumber_InPlaceOr(ob, dup) == ob);
2292 Py_DECREF(ob);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002293
2294 /* Verify constructors accept NULL arguments */
2295 f = PySet_New(NULL);
2296 assert(f != NULL);
2297 assert(PySet_GET_SIZE(f) == 0);
2298 Py_DECREF(f);
2299 f = PyFrozenSet_New(NULL);
2300 assert(f != NULL);
2301 assert(PyFrozenSet_CheckExact(f));
2302 assert(PySet_GET_SIZE(f) == 0);
2303 Py_DECREF(f);
2304
2305 Py_DECREF(elem);
2306 Py_DECREF(dup);
2307 Py_RETURN_TRUE;
2308}
2309
Raymond Hettinger9bda1d62005-09-16 07:14:21 +00002310#undef assertRaises
2311
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002312#endif