blob: d1bad27d56ea4e903a63be56f8199002bec8373d [file] [log] [blame]
Raymond Hettingerc991db22005-08-11 07:58:45 +00001
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002/* set object implementation
Raymond Hettingera9d99362005-08-05 00:01:15 +00003 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
Raymond Hettinger2212e522008-11-30 23:43:36 +00006 Copyright (c) 2003-2008 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"
Christian Heimes0ded5b52007-12-10 15:50:56 +000012#include "stringlib/eq.h"
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000013
Thomas Wouters89f507f2006-12-13 04:49:30 +000014/* Set a key error with the specified argument, wrapping it in a
15 * tuple automatically so that tuple keys are not unpacked as the
16 * exception arguments. */
17static void
18set_key_error(PyObject *arg)
19{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000020 PyObject *tup;
21 tup = PyTuple_Pack(1, arg);
22 if (!tup)
23 return; /* caller will expect error to be set anyway */
24 PyErr_SetObject(PyExc_KeyError, tup);
25 Py_DECREF(tup);
Thomas Wouters89f507f2006-12-13 04:49:30 +000026}
27
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000028/* This must be >= 1. */
29#define PERTURB_SHIFT 5
30
31/* Object used as dummy key to fill deleted entries */
Raymond Hettingera9d99362005-08-05 00:01:15 +000032static PyObject *dummy = NULL; /* Initialized by first call to make_new_set() */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000033
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000034#ifdef Py_REF_DEBUG
35PyObject *
36_PySet_Dummy(void)
37{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000038 return dummy;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +000039}
40#endif
41
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000042#define INIT_NONZERO_SET_SLOTS(so) do { \
43 (so)->table = (so)->smalltable; \
44 (so)->mask = PySet_MINSIZE - 1; \
45 (so)->hash = -1; \
Raymond Hettingerbc841a12005-08-07 13:02:53 +000046 } while(0)
47
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000048#define EMPTY_TO_MINSIZE(so) do { \
49 memset((so)->smalltable, 0, sizeof((so)->smalltable)); \
50 (so)->used = (so)->fill = 0; \
51 INIT_NONZERO_SET_SLOTS(so); \
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000052 } while(0)
53
Raymond Hettingerbc841a12005-08-07 13:02:53 +000054/* Reuse scheme to save calls to malloc, free, and memset */
Christian Heimes2202f872008-02-06 14:31:34 +000055#ifndef PySet_MAXFREELIST
56#define PySet_MAXFREELIST 80
57#endif
58static PySetObject *free_list[PySet_MAXFREELIST];
59static int numfree = 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000060
Christian Heimes0ded5b52007-12-10 15:50:56 +000061
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000062/*
63The basic lookup function used by all operations.
64This is based on Algorithm D from Knuth Vol. 3, Sec. 6.4.
65Open addressing is preferred over chaining since the link overhead for
66chaining would be substantial (100% with typical malloc overhead).
67
68The initial probe index is computed as hash mod the table size. Subsequent
Raymond Hettingerbc841a12005-08-07 13:02:53 +000069probe indices are computed as explained in Objects/dictobject.c.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000070
71All arithmetic on hash should ignore overflow.
72
Raymond Hettinger9bda1d62005-09-16 07:14:21 +000073Unlike the dictionary implementation, the lookkey functions can return
74NULL if the rich comparison returns an error.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000075*/
76
77static setentry *
Benjamin Peterson8f67d082010-10-17 20:54:53 +000078set_lookkey(PySetObject *so, PyObject *key, register Py_hash_t hash)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000079{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000080 register Py_ssize_t i;
81 register size_t perturb;
82 register setentry *freeslot;
83 register size_t mask = so->mask;
84 setentry *table = so->table;
85 register setentry *entry;
86 register int cmp;
87 PyObject *startkey;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000088
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000089 i = hash & mask;
90 entry = &table[i];
91 if (entry->key == NULL || entry->key == key)
92 return entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +000093
Antoine Pitrouf95a1b32010-05-09 15:52:27 +000094 if (entry->key == dummy)
95 freeslot = entry;
96 else {
97 if (entry->hash == hash) {
98 startkey = entry->key;
99 Py_INCREF(startkey);
100 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
101 Py_DECREF(startkey);
102 if (cmp < 0)
103 return NULL;
104 if (table == so->table && entry->key == startkey) {
105 if (cmp > 0)
106 return entry;
107 }
108 else {
109 /* The compare did major nasty stuff to the
110 * set: start over.
111 */
112 return set_lookkey(so, key, hash);
113 }
114 }
115 freeslot = NULL;
116 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000117
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000118 /* In the loop, key == dummy is by far (factor of 100s) the
119 least likely outcome, so test for that last. */
120 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
121 i = (i << 2) + i + perturb + 1;
122 entry = &table[i & mask];
123 if (entry->key == NULL) {
124 if (freeslot != NULL)
125 entry = freeslot;
126 break;
127 }
128 if (entry->key == key)
129 break;
130 if (entry->hash == hash && entry->key != dummy) {
131 startkey = entry->key;
132 Py_INCREF(startkey);
133 cmp = PyObject_RichCompareBool(startkey, key, Py_EQ);
134 Py_DECREF(startkey);
135 if (cmp < 0)
136 return NULL;
137 if (table == so->table && entry->key == startkey) {
138 if (cmp > 0)
139 break;
140 }
141 else {
142 /* The compare did major nasty stuff to the
143 * set: start over.
144 */
145 return set_lookkey(so, key, hash);
146 }
147 }
148 else if (entry->key == dummy && freeslot == NULL)
149 freeslot = entry;
150 }
151 return entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000152}
153
154/*
Christian Heimes0ded5b52007-12-10 15:50:56 +0000155 * Hacked up version of set_lookkey which can assume keys are always unicode;
156 * This means we can always use unicode_eq directly and not have to check to
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000157 * see if the comparison altered the table.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000158 */
159static setentry *
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000160set_lookkey_unicode(PySetObject *so, PyObject *key, register Py_hash_t hash)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000161{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000162 register Py_ssize_t i;
163 register size_t perturb;
164 register setentry *freeslot;
165 register size_t mask = so->mask;
166 setentry *table = so->table;
167 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000168
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000169 /* Make sure this function doesn't have to handle non-unicode keys,
170 including subclasses of str; e.g., one reason to subclass
171 strings is to override __eq__, and for speed we don't cater to
172 that here. */
173 if (!PyUnicode_CheckExact(key)) {
174 so->lookup = set_lookkey;
175 return set_lookkey(so, key, hash);
176 }
177 i = hash & mask;
178 entry = &table[i];
179 if (entry->key == NULL || entry->key == key)
180 return entry;
181 if (entry->key == dummy)
182 freeslot = entry;
183 else {
184 if (entry->hash == hash && unicode_eq(entry->key, key))
185 return entry;
186 freeslot = NULL;
187 }
Raymond Hettingered6c1ef2005-08-13 08:28:03 +0000188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000189 /* In the loop, key == dummy is by far (factor of 100s) the
190 least likely outcome, so test for that last. */
191 for (perturb = hash; ; perturb >>= PERTURB_SHIFT) {
192 i = (i << 2) + i + perturb + 1;
193 entry = &table[i & mask];
194 if (entry->key == NULL)
195 return freeslot == NULL ? entry : freeslot;
196 if (entry->key == key
197 || (entry->hash == hash
198 && entry->key != dummy
199 && unicode_eq(entry->key, key)))
200 return entry;
201 if (entry->key == dummy && freeslot == NULL)
202 freeslot = entry;
203 }
204 assert(0); /* NOT REACHED */
205 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000206}
207
208/*
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000209Internal routine to insert a new key into the table.
Thomas Wouters89f507f2006-12-13 04:49:30 +0000210Used by the public insert routine.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000211Eats a reference to key.
212*/
Raymond Hettinger9bda1d62005-09-16 07:14:21 +0000213static int
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000214set_insert_key(register PySetObject *so, PyObject *key, Py_hash_t hash)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000215{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000216 register setentry *entry;
Antoine Pitroufbb1c612010-10-23 17:37:54 +0000217 typedef setentry *(*lookupfunc)(PySetObject *, PyObject *, Py_hash_t);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000218
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000219 assert(so->lookup != NULL);
220 entry = so->lookup(so, key, hash);
221 if (entry == NULL)
222 return -1;
223 if (entry->key == NULL) {
224 /* UNUSED */
225 so->fill++;
226 entry->key = key;
227 entry->hash = hash;
228 so->used++;
229 } else if (entry->key == dummy) {
230 /* DUMMY */
231 entry->key = key;
232 entry->hash = hash;
233 so->used++;
234 Py_DECREF(dummy);
235 } else {
236 /* ACTIVE */
237 Py_DECREF(key);
238 }
239 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000240}
241
242/*
Thomas Wouters89f507f2006-12-13 04:49:30 +0000243Internal routine used by set_table_resize() to insert an item which is
244known to be absent from the set. This routine also assumes that
245the set contains no deleted entries. Besides the performance benefit,
246using set_insert_clean() in set_table_resize() is dangerous (SF bug #1456209).
247Note that no refcounts are changed by this routine; if needed, the caller
248is responsible for incref'ing `key`.
249*/
250static void
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000251set_insert_clean(register PySetObject *so, PyObject *key, Py_hash_t hash)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000252{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000253 register size_t i;
254 register size_t perturb;
255 register size_t mask = (size_t)so->mask;
256 setentry *table = so->table;
257 register setentry *entry;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000259 i = hash & mask;
260 entry = &table[i];
261 for (perturb = hash; entry->key != NULL; perturb >>= PERTURB_SHIFT) {
262 i = (i << 2) + i + perturb + 1;
263 entry = &table[i & mask];
264 }
265 so->fill++;
266 entry->key = key;
267 entry->hash = hash;
268 so->used++;
Thomas Wouters89f507f2006-12-13 04:49:30 +0000269}
270
271/*
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000272Restructure the table by allocating a new table and reinserting all
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000273keys again. When entries have been deleted, the new table may
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000274actually be smaller than the old one.
275*/
276static int
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000277set_table_resize(PySetObject *so, Py_ssize_t minused)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000278{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000279 Py_ssize_t newsize;
280 setentry *oldtable, *newtable, *entry;
281 Py_ssize_t i;
282 int is_oldtable_malloced;
283 setentry small_copy[PySet_MINSIZE];
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000284
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000285 assert(minused >= 0);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000287 /* Find the smallest table size > minused. */
288 for (newsize = PySet_MINSIZE;
289 newsize <= minused && newsize > 0;
290 newsize <<= 1)
291 ;
292 if (newsize <= 0) {
293 PyErr_NoMemory();
294 return -1;
295 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000296
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000297 /* Get space for a new table. */
298 oldtable = so->table;
299 assert(oldtable != NULL);
300 is_oldtable_malloced = oldtable != so->smalltable;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000301
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000302 if (newsize == PySet_MINSIZE) {
303 /* A large table is shrinking, or we can't get any smaller. */
304 newtable = so->smalltable;
305 if (newtable == oldtable) {
306 if (so->fill == so->used) {
307 /* No dummies, so no point doing anything. */
308 return 0;
309 }
310 /* We're not going to resize it, but rebuild the
311 table anyway to purge old dummy entries.
312 Subtle: This is *necessary* if fill==size,
313 as set_lookkey needs at least one virgin slot to
314 terminate failing searches. If fill < size, it's
315 merely desirable, as dummies slow searches. */
316 assert(so->fill > so->used);
317 memcpy(small_copy, oldtable, sizeof(small_copy));
318 oldtable = small_copy;
319 }
320 }
321 else {
322 newtable = PyMem_NEW(setentry, newsize);
323 if (newtable == NULL) {
324 PyErr_NoMemory();
325 return -1;
326 }
327 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000328
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000329 /* Make the set empty, using the new table. */
330 assert(newtable != oldtable);
331 so->table = newtable;
332 so->mask = newsize - 1;
333 memset(newtable, 0, sizeof(setentry) * newsize);
334 so->used = 0;
335 i = so->fill;
336 so->fill = 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000337
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000338 /* Copy the data over; this is refcount-neutral for active entries;
339 dummy entries aren't copied over, of course */
340 for (entry = oldtable; i > 0; entry++) {
341 if (entry->key == NULL) {
342 /* UNUSED */
343 ;
344 } else if (entry->key == dummy) {
345 /* DUMMY */
346 --i;
347 assert(entry->key == dummy);
348 Py_DECREF(entry->key);
349 } else {
350 /* ACTIVE */
351 --i;
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000352 set_insert_clean(so, entry->key, entry->hash);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000353 }
354 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000355
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000356 if (is_oldtable_malloced)
357 PyMem_DEL(oldtable);
358 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000359}
360
Raymond Hettingerc991db22005-08-11 07:58:45 +0000361/* CAUTION: set_add_key/entry() must guarantee it won't resize the table */
362
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000363static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000364set_add_entry(register PySetObject *so, setentry *entry)
365{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000366 register Py_ssize_t n_used;
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000367 PyObject *key = entry->key;
Éric Araujobbcfc1f2011-03-23 03:43:22 +0100368 Py_hash_t hash = entry->hash;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000369
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000370 assert(so->fill <= so->mask); /* at least one empty slot */
371 n_used = so->used;
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000372 Py_INCREF(key);
Éric Araujo48049912011-03-23 02:08:07 +0100373 if (set_insert_key(so, key, hash) == -1) {
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000374 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000375 return -1;
376 }
377 if (!(so->used > n_used && so->fill*3 >= (so->mask+1)*2))
378 return 0;
379 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
Raymond Hettingerc991db22005-08-11 07:58:45 +0000380}
381
382static int
383set_add_key(register PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000384{
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000385 register Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000386 register Py_ssize_t n_used;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000387
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000388 if (!PyUnicode_CheckExact(key) ||
389 (hash = ((PyUnicodeObject *) key)->hash) == -1) {
390 hash = PyObject_Hash(key);
391 if (hash == -1)
392 return -1;
393 }
394 assert(so->fill <= so->mask); /* at least one empty slot */
395 n_used = so->used;
396 Py_INCREF(key);
397 if (set_insert_key(so, key, hash) == -1) {
398 Py_DECREF(key);
399 return -1;
400 }
401 if (!(so->used > n_used && so->fill*3 >= (so->mask+1)*2))
402 return 0;
403 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000404}
405
406#define DISCARD_NOTFOUND 0
407#define DISCARD_FOUND 1
408
409static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000410set_discard_entry(PySetObject *so, setentry *oldentry)
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000411{ register setentry *entry;
412 PyObject *old_key;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000413
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000414 entry = (so->lookup)(so, oldentry->key, oldentry->hash);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000415 if (entry == NULL)
416 return -1;
417 if (entry->key == NULL || entry->key == dummy)
418 return DISCARD_NOTFOUND;
419 old_key = entry->key;
420 Py_INCREF(dummy);
421 entry->key = dummy;
422 so->used--;
423 Py_DECREF(old_key);
424 return DISCARD_FOUND;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000425}
426
427static int
428set_discard_key(PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000429{
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000430 register Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000431 register setentry *entry;
432 PyObject *old_key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000433
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000434 assert (PyAnySet_Check(so));
Christian Heimes0ded5b52007-12-10 15:50:56 +0000435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000436 if (!PyUnicode_CheckExact(key) ||
437 (hash = ((PyUnicodeObject *) key)->hash) == -1) {
438 hash = PyObject_Hash(key);
439 if (hash == -1)
440 return -1;
441 }
442 entry = (so->lookup)(so, key, hash);
443 if (entry == NULL)
444 return -1;
445 if (entry->key == NULL || entry->key == dummy)
446 return DISCARD_NOTFOUND;
447 old_key = entry->key;
448 Py_INCREF(dummy);
449 entry->key = dummy;
450 so->used--;
451 Py_DECREF(old_key);
452 return DISCARD_FOUND;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000453}
454
Raymond Hettingerfe889f32005-08-06 05:43:39 +0000455static int
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000456set_clear_internal(PySetObject *so)
457{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000458 setentry *entry, *table;
459 int table_is_malloced;
460 Py_ssize_t fill;
461 setentry small_copy[PySet_MINSIZE];
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000462#ifdef Py_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000463 Py_ssize_t i, n;
464 assert (PyAnySet_Check(so));
Raymond Hettingera580c472005-08-05 17:19:54 +0000465
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000466 n = so->mask + 1;
467 i = 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000468#endif
469
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000470 table = so->table;
471 assert(table != NULL);
472 table_is_malloced = table != so->smalltable;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000473
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000474 /* This is delicate. During the process of clearing the set,
475 * decrefs can cause the set to mutate. To avoid fatal confusion
476 * (voice of experience), we have to make the set empty before
477 * clearing the slots, and never refer to anything via so->ref while
478 * clearing.
479 */
480 fill = so->fill;
481 if (table_is_malloced)
482 EMPTY_TO_MINSIZE(so);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000483
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000484 else if (fill > 0) {
485 /* It's a small table with something that needs to be cleared.
486 * Afraid the only safe way is to copy the set entries into
487 * another small table first.
488 */
489 memcpy(small_copy, table, sizeof(small_copy));
490 table = small_copy;
491 EMPTY_TO_MINSIZE(so);
492 }
493 /* else it's a small table that's already empty */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000494
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000495 /* Now we can finally clear things. If C had refcounts, we could
496 * assert that the refcount on table is 1 now, i.e. that this function
497 * has unique access to it, so decref side-effects can't alter it.
498 */
499 for (entry = table; fill > 0; ++entry) {
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000500#ifdef Py_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000501 assert(i < n);
502 ++i;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000503#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000504 if (entry->key) {
505 --fill;
506 Py_DECREF(entry->key);
507 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000508#ifdef Py_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000509 else
510 assert(entry->key == NULL);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000511#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000512 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000514 if (table_is_malloced)
515 PyMem_DEL(table);
516 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000517}
518
519/*
520 * Iterate over a set table. Use like so:
521 *
Thomas Wouters0e3f5912006-08-11 14:57:12 +0000522 * Py_ssize_t pos;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000523 * setentry *entry;
Raymond Hettingerd7946662005-08-01 21:39:29 +0000524 * pos = 0; # important! pos should not otherwise be changed by you
Raymond Hettingerc991db22005-08-11 07:58:45 +0000525 * while (set_next(yourset, &pos, &entry)) {
526 * Refer to borrowed reference in entry->key.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000527 * }
528 *
Raymond Hettingerc991db22005-08-11 07:58:45 +0000529 * CAUTION: In general, it isn't safe to use set_next in a loop that
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000530 * mutates the table.
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000531 */
532static int
Martin v. Löwis18e16552006-02-15 17:27:45 +0000533set_next(PySetObject *so, Py_ssize_t *pos_ptr, setentry **entry_ptr)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000534{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000535 Py_ssize_t i;
536 Py_ssize_t mask;
537 register setentry *table;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000538
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000539 assert (PyAnySet_Check(so));
540 i = *pos_ptr;
541 assert(i >= 0);
542 table = so->table;
543 mask = so->mask;
544 while (i <= mask && (table[i].key == NULL || table[i].key == dummy))
545 i++;
546 *pos_ptr = i+1;
547 if (i > mask)
548 return 0;
549 assert(table[i].key != NULL);
550 *entry_ptr = &table[i];
551 return 1;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000552}
553
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000554static void
555set_dealloc(PySetObject *so)
556{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000557 register setentry *entry;
558 Py_ssize_t fill = so->fill;
559 PyObject_GC_UnTrack(so);
560 Py_TRASHCAN_SAFE_BEGIN(so)
561 if (so->weakreflist != NULL)
562 PyObject_ClearWeakRefs((PyObject *) so);
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000563
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000564 for (entry = so->table; fill > 0; entry++) {
565 if (entry->key) {
566 --fill;
567 Py_DECREF(entry->key);
568 }
569 }
570 if (so->table != so->smalltable)
571 PyMem_DEL(so->table);
572 if (numfree < PySet_MAXFREELIST && PyAnySet_CheckExact(so))
573 free_list[numfree++] = so;
574 else
575 Py_TYPE(so)->tp_free(so);
576 Py_TRASHCAN_SAFE_END(so)
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000577}
578
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000579static PyObject *
580set_repr(PySetObject *so)
581{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000582 PyObject *keys, *result=NULL;
583 Py_UNICODE *u;
584 int status = Py_ReprEnter((PyObject*)so);
585 PyObject *listrepr;
586 Py_ssize_t newsize;
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000587
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000588 if (status != 0) {
589 if (status < 0)
590 return NULL;
591 return PyUnicode_FromFormat("%s(...)", Py_TYPE(so)->tp_name);
592 }
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000593
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000594 /* shortcut for the empty set */
595 if (!so->used) {
596 Py_ReprLeave((PyObject*)so);
597 return PyUnicode_FromFormat("%s()", Py_TYPE(so)->tp_name);
598 }
Georg Brandlc4996ba2006-08-28 19:37:11 +0000599
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000600 keys = PySequence_List((PyObject *)so);
601 if (keys == NULL)
602 goto done;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000603
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000604 listrepr = PyObject_Repr(keys);
605 Py_DECREF(keys);
606 if (listrepr == NULL)
607 goto done;
608 newsize = PyUnicode_GET_SIZE(listrepr);
609 result = PyUnicode_FromUnicode(NULL, newsize);
Victor Stinnera1a807b2011-05-26 14:24:30 +0200610 if (result == NULL)
611 goto done;
612
613 u = PyUnicode_AS_UNICODE(result);
614 *u++ = '{';
615 /* Omit the brackets from the listrepr */
616 Py_UNICODE_COPY(u, PyUnicode_AS_UNICODE(listrepr)+1,
617 newsize-2);
618 u += newsize-2;
Victor Stinner4f2dab52011-05-27 16:46:51 +0200619 *u++ = '}';
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000620 Py_DECREF(listrepr);
Victor Stinnera1a807b2011-05-26 14:24:30 +0200621
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000622 if (Py_TYPE(so) != &PySet_Type) {
623 PyObject *tmp = PyUnicode_FromFormat("%s(%U)",
624 Py_TYPE(so)->tp_name,
625 result);
626 Py_DECREF(result);
627 result = tmp;
628 }
Thomas Wouters902d6eb2007-01-09 23:18:33 +0000629done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000630 Py_ReprLeave((PyObject*)so);
631 return result;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000632}
633
Martin v. Löwis18e16552006-02-15 17:27:45 +0000634static Py_ssize_t
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000635set_len(PyObject *so)
636{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000637 return ((PySetObject *)so)->used;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000638}
639
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000640static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000641set_merge(PySetObject *so, PyObject *otherset)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000642{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000643 PySetObject *other;
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000644 PyObject *key;
Éric Araujobbcfc1f2011-03-23 03:43:22 +0100645 Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000646 register Py_ssize_t i;
647 register setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000648
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000649 assert (PyAnySet_Check(so));
650 assert (PyAnySet_Check(otherset));
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000651
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000652 other = (PySetObject*)otherset;
653 if (other == so || other->used == 0)
654 /* a.update(a) or a.update({}); nothing to do */
655 return 0;
656 /* Do one big resize at the start, rather than
657 * incrementally resizing as we insert new keys. Expect
658 * that there will be no (or few) overlapping keys.
659 */
660 if ((so->fill + other->used)*3 >= (so->mask+1)*2) {
661 if (set_table_resize(so, (so->used + other->used)*2) != 0)
662 return -1;
663 }
664 for (i = 0; i <= other->mask; i++) {
665 entry = &other->table[i];
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000666 key = entry->key;
Éric Araujo48049912011-03-23 02:08:07 +0100667 hash = entry->hash;
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000668 if (key != NULL &&
669 key != dummy) {
670 Py_INCREF(key);
Éric Araujo48049912011-03-23 02:08:07 +0100671 if (set_insert_key(so, key, hash) == -1) {
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +0000672 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000673 return -1;
674 }
675 }
676 }
677 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000678}
679
680static int
Raymond Hettingerc991db22005-08-11 07:58:45 +0000681set_contains_key(PySetObject *so, PyObject *key)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000682{
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000683 Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000684 setentry *entry;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000686 if (!PyUnicode_CheckExact(key) ||
687 (hash = ((PyUnicodeObject *) key)->hash) == -1) {
688 hash = PyObject_Hash(key);
689 if (hash == -1)
690 return -1;
691 }
692 entry = (so->lookup)(so, key, hash);
693 if (entry == NULL)
694 return -1;
695 key = entry->key;
696 return key != NULL && key != dummy;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000697}
698
Raymond Hettingerc991db22005-08-11 07:58:45 +0000699static int
700set_contains_entry(PySetObject *so, setentry *entry)
701{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000702 PyObject *key;
703 setentry *lu_entry;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000704
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000705 lu_entry = (so->lookup)(so, entry->key, entry->hash);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000706 if (lu_entry == NULL)
707 return -1;
708 key = lu_entry->key;
709 return key != NULL && key != dummy;
Raymond Hettingerc991db22005-08-11 07:58:45 +0000710}
711
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000712static PyObject *
713set_pop(PySetObject *so)
714{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000715 register Py_ssize_t i = 0;
716 register setentry *entry;
717 PyObject *key;
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000718
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000719 assert (PyAnySet_Check(so));
720 if (so->used == 0) {
721 PyErr_SetString(PyExc_KeyError, "pop from an empty set");
722 return NULL;
723 }
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000724
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000725 /* Set entry to "the first" unused or dummy set entry. We abuse
726 * the hash field of slot 0 to hold a search finger:
727 * If slot 0 has a value, use slot 0.
728 * Else slot 0 is being used to hold a search finger,
729 * and we use its hash value as the first index to look.
730 */
731 entry = &so->table[0];
732 if (entry->key == NULL || entry->key == dummy) {
733 i = entry->hash;
734 /* The hash field may be a real hash value, or it may be a
735 * legit search finger, or it may be a once-legit search
736 * finger that's out of bounds now because it wrapped around
737 * or the table shrunk -- simply make sure it's in bounds now.
738 */
739 if (i > so->mask || i < 1)
740 i = 1; /* skip slot 0 */
741 while ((entry = &so->table[i])->key == NULL || entry->key==dummy) {
742 i++;
743 if (i > so->mask)
744 i = 1;
745 }
746 }
747 key = entry->key;
748 Py_INCREF(dummy);
749 entry->key = dummy;
750 so->used--;
751 so->table[0].hash = i + 1; /* next place to start */
752 return key;
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000753}
754
Benjamin Petersonf10a79a2008-10-11 00:49:57 +0000755PyDoc_STRVAR(pop_doc, "Remove and return an arbitrary set element.\n\
756Raises KeyError if the set is empty.");
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000757
758static int
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000759set_traverse(PySetObject *so, visitproc visit, void *arg)
Raymond Hettingerce8185e2005-08-13 09:28:48 +0000760{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000761 Py_ssize_t pos = 0;
762 setentry *entry;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000763
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000764 while (set_next(so, &pos, &entry))
765 Py_VISIT(entry->key);
766 return 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000767}
768
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000769static Py_hash_t
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000770frozenset_hash(PyObject *self)
771{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000772 PySetObject *so = (PySetObject *)self;
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000773 Py_hash_t h, hash = 1927868237L;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000774 setentry *entry;
775 Py_ssize_t pos = 0;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000776
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000777 if (so->hash != -1)
778 return so->hash;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000779
Antoine Pitroufbb1c612010-10-23 17:37:54 +0000780 hash *= PySet_GET_SIZE(self) + 1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000781 while (set_next(so, &pos, &entry)) {
782 /* Work to increase the bit dispersion for closely spaced hash
783 values. The is important because some use cases have many
784 combinations of a small number of elements with nearby
785 hashes so that many distinct combinations collapse to only
786 a handful of distinct hash values. */
Antoine Pitroufbb1c612010-10-23 17:37:54 +0000787 h = entry->hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000788 hash ^= (h ^ (h << 16) ^ 89869747L) * 3644798167u;
789 }
790 hash = hash * 69069L + 907133923L;
791 if (hash == -1)
792 hash = 590923713L;
793 so->hash = hash;
794 return hash;
Raymond Hettingerf408ddf2005-08-17 00:27:42 +0000795}
796
Raymond Hettingera9d99362005-08-05 00:01:15 +0000797/***** Set iterator type ***********************************************/
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000798
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000799typedef struct {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000800 PyObject_HEAD
801 PySetObject *si_set; /* Set to NULL when iterator is exhausted */
802 Py_ssize_t si_used;
803 Py_ssize_t si_pos;
804 Py_ssize_t len;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000805} setiterobject;
806
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000807static void
808setiter_dealloc(setiterobject *si)
809{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000810 Py_XDECREF(si->si_set);
811 PyObject_GC_Del(si);
Antoine Pitrou7ddda782009-01-01 15:35:33 +0000812}
813
814static int
815setiter_traverse(setiterobject *si, visitproc visit, void *arg)
816{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000817 Py_VISIT(si->si_set);
818 return 0;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000819}
820
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000821static PyObject *
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000822setiter_len(setiterobject *si)
823{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000824 Py_ssize_t len = 0;
825 if (si->si_set != NULL && si->si_used == si->si_set->used)
826 len = si->len;
Antoine Pitrou671b4d92010-08-17 17:55:07 +0000827 return PyLong_FromSsize_t(len);
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000828}
829
Armin Rigof5b3e362006-02-11 21:32:43 +0000830PyDoc_STRVAR(length_hint_doc, "Private method returning an estimate of len(list(it)).");
Raymond Hettinger6b27cda2005-09-24 21:23:05 +0000831
832static PyMethodDef setiter_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000833 {"__length_hint__", (PyCFunction)setiter_len, METH_NOARGS, length_hint_doc},
834 {NULL, NULL} /* sentinel */
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000835};
836
Raymond Hettinger06d8cf82005-07-31 15:36:06 +0000837static PyObject *setiter_iternext(setiterobject *si)
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000838{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000839 PyObject *key;
840 register Py_ssize_t i, mask;
841 register setentry *entry;
842 PySetObject *so = si->si_set;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000843
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000844 if (so == NULL)
845 return NULL;
846 assert (PyAnySet_Check(so));
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000847
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000848 if (si->si_used != so->used) {
849 PyErr_SetString(PyExc_RuntimeError,
850 "Set changed size during iteration");
851 si->si_used = -1; /* Make this state sticky */
852 return NULL;
853 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000854
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000855 i = si->si_pos;
856 assert(i>=0);
857 entry = so->table;
858 mask = so->mask;
859 while (i <= mask && (entry[i].key == NULL || entry[i].key == dummy))
860 i++;
861 si->si_pos = i+1;
862 if (i > mask)
863 goto fail;
864 si->len--;
865 key = entry[i].key;
866 Py_INCREF(key);
867 return key;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000868
869fail:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000870 Py_DECREF(so);
871 si->si_set = NULL;
872 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000873}
874
Christian Heimesa22e8bd2007-11-29 22:35:39 +0000875PyTypeObject PySetIter_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000876 PyVarObject_HEAD_INIT(&PyType_Type, 0)
877 "set_iterator", /* tp_name */
878 sizeof(setiterobject), /* tp_basicsize */
879 0, /* tp_itemsize */
880 /* methods */
881 (destructor)setiter_dealloc, /* tp_dealloc */
882 0, /* tp_print */
883 0, /* tp_getattr */
884 0, /* tp_setattr */
885 0, /* tp_reserved */
886 0, /* tp_repr */
887 0, /* tp_as_number */
888 0, /* tp_as_sequence */
889 0, /* tp_as_mapping */
890 0, /* tp_hash */
891 0, /* tp_call */
892 0, /* tp_str */
893 PyObject_GenericGetAttr, /* tp_getattro */
894 0, /* tp_setattro */
895 0, /* tp_as_buffer */
896 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC,/* tp_flags */
897 0, /* tp_doc */
898 (traverseproc)setiter_traverse, /* tp_traverse */
899 0, /* tp_clear */
900 0, /* tp_richcompare */
901 0, /* tp_weaklistoffset */
902 PyObject_SelfIter, /* tp_iter */
903 (iternextfunc)setiter_iternext, /* tp_iternext */
904 setiter_methods, /* tp_methods */
905 0,
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000906};
907
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000908static PyObject *
909set_iter(PySetObject *so)
910{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000911 setiterobject *si = PyObject_GC_New(setiterobject, &PySetIter_Type);
912 if (si == NULL)
913 return NULL;
914 Py_INCREF(so);
915 si->si_set = so;
916 si->si_used = so->used;
917 si->si_pos = 0;
918 si->len = so->used;
919 _PyObject_GC_TRACK(si);
920 return (PyObject *)si;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +0000921}
922
Raymond Hettingerd7946662005-08-01 21:39:29 +0000923static int
Raymond Hettingerd7946662005-08-01 21:39:29 +0000924set_update_internal(PySetObject *so, PyObject *other)
Raymond Hettingera690a992003-11-16 16:17:49 +0000925{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000926 PyObject *key, *it;
Raymond Hettingera690a992003-11-16 16:17:49 +0000927
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000928 if (PyAnySet_Check(other))
929 return set_merge(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +0000930
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000931 if (PyDict_CheckExact(other)) {
932 PyObject *value;
933 Py_ssize_t pos = 0;
Benjamin Peterson8f67d082010-10-17 20:54:53 +0000934 Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000935 Py_ssize_t dictsize = PyDict_Size(other);
Thomas Wouterscf297e42007-02-23 15:07:44 +0000936
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000937 /* Do one big resize at the start, rather than
938 * incrementally resizing as we insert new keys. Expect
939 * that there will be no (or few) overlapping keys.
940 */
941 if (dictsize == -1)
942 return -1;
943 if ((so->fill + dictsize)*3 >= (so->mask+1)*2) {
944 if (set_table_resize(so, (so->used + dictsize)*2) != 0)
945 return -1;
946 }
947 while (_PyDict_Next(other, &pos, &key, &value, &hash)) {
948 setentry an_entry;
Thomas Wouterscf297e42007-02-23 15:07:44 +0000949
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000950 an_entry.hash = hash;
951 an_entry.key = key;
952 if (set_add_entry(so, &an_entry) == -1)
953 return -1;
954 }
955 return 0;
956 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +0000957
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000958 it = PyObject_GetIter(other);
959 if (it == NULL)
960 return -1;
Raymond Hettingera690a992003-11-16 16:17:49 +0000961
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000962 while ((key = PyIter_Next(it)) != NULL) {
963 if (set_add_key(so, key) == -1) {
964 Py_DECREF(it);
965 Py_DECREF(key);
966 return -1;
967 }
968 Py_DECREF(key);
969 }
970 Py_DECREF(it);
971 if (PyErr_Occurred())
972 return -1;
973 return 0;
Raymond Hettingerd7946662005-08-01 21:39:29 +0000974}
975
976static PyObject *
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000977set_update(PySetObject *so, PyObject *args)
Raymond Hettingerd7946662005-08-01 21:39:29 +0000978{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000979 Py_ssize_t i;
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000980
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000981 for (i=0 ; i<PyTuple_GET_SIZE(args) ; i++) {
982 PyObject *other = PyTuple_GET_ITEM(args, i);
983 if (set_update_internal(so, other) == -1)
984 return NULL;
985 }
986 Py_RETURN_NONE;
Raymond Hettingera38123e2003-11-24 22:18:49 +0000987}
988
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000989PyDoc_STRVAR(update_doc,
Georg Brandlc28e1fa2008-06-10 19:20:26 +0000990"Update a set with the union of itself and others.");
Raymond Hettingera38123e2003-11-24 22:18:49 +0000991
992static PyObject *
993make_new_set(PyTypeObject *type, PyObject *iterable)
994{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000995 register PySetObject *so = NULL;
Raymond Hettingera38123e2003-11-24 22:18:49 +0000996
Antoine Pitrouf95a1b32010-05-09 15:52:27 +0000997 if (dummy == NULL) { /* Auto-initialize dummy */
998 dummy = PyUnicode_FromString("<dummy key>");
999 if (dummy == NULL)
1000 return NULL;
1001 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001002
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001003 /* create PySetObject structure */
1004 if (numfree &&
1005 (type == &PySet_Type || type == &PyFrozenSet_Type)) {
1006 so = free_list[--numfree];
1007 assert (so != NULL && PyAnySet_CheckExact(so));
1008 Py_TYPE(so) = type;
1009 _Py_NewReference((PyObject *)so);
1010 EMPTY_TO_MINSIZE(so);
1011 PyObject_GC_Track(so);
1012 } else {
1013 so = (PySetObject *)type->tp_alloc(type, 0);
1014 if (so == NULL)
1015 return NULL;
1016 /* tp_alloc has already zeroed the structure */
1017 assert(so->table == NULL && so->fill == 0 && so->used == 0);
1018 INIT_NONZERO_SET_SLOTS(so);
1019 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001020
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001021 so->lookup = set_lookkey_unicode;
1022 so->weakreflist = NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001023
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001024 if (iterable != NULL) {
1025 if (set_update_internal(so, iterable) == -1) {
1026 Py_DECREF(so);
1027 return NULL;
1028 }
1029 }
Raymond Hettingera38123e2003-11-24 22:18:49 +00001030
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001031 return (PyObject *)so;
Raymond Hettingera690a992003-11-16 16:17:49 +00001032}
1033
Raymond Hettinger7d99f092008-11-16 11:44:54 +00001034static PyObject *
1035make_new_set_basetype(PyTypeObject *type, PyObject *iterable)
1036{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001037 if (type != &PySet_Type && type != &PyFrozenSet_Type) {
1038 if (PyType_IsSubtype(type, &PySet_Type))
1039 type = &PySet_Type;
1040 else
1041 type = &PyFrozenSet_Type;
1042 }
1043 return make_new_set(type, iterable);
Raymond Hettinger7d99f092008-11-16 11:44:54 +00001044}
1045
Raymond Hettingerd7946662005-08-01 21:39:29 +00001046/* The empty frozenset is a singleton */
1047static PyObject *emptyfrozenset = NULL;
1048
Raymond Hettingera690a992003-11-16 16:17:49 +00001049static PyObject *
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001050frozenset_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
Raymond Hettingera690a992003-11-16 16:17:49 +00001051{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001052 PyObject *iterable = NULL, *result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001053
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001054 if (type == &PyFrozenSet_Type && !_PyArg_NoKeywords("frozenset()", kwds))
1055 return NULL;
Georg Brandl02c42872005-08-26 06:42:30 +00001056
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001057 if (!PyArg_UnpackTuple(args, type->tp_name, 0, 1, &iterable))
1058 return NULL;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001059
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001060 if (type != &PyFrozenSet_Type)
1061 return make_new_set(type, iterable);
Raymond Hettingerd7946662005-08-01 21:39:29 +00001062
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001063 if (iterable != NULL) {
1064 /* frozenset(f) is idempotent */
1065 if (PyFrozenSet_CheckExact(iterable)) {
1066 Py_INCREF(iterable);
1067 return iterable;
1068 }
1069 result = make_new_set(type, iterable);
1070 if (result == NULL || PySet_GET_SIZE(result))
1071 return result;
1072 Py_DECREF(result);
1073 }
1074 /* The empty frozenset is a singleton */
1075 if (emptyfrozenset == NULL)
1076 emptyfrozenset = make_new_set(type, NULL);
1077 Py_XINCREF(emptyfrozenset);
1078 return emptyfrozenset;
Raymond Hettingerd7946662005-08-01 21:39:29 +00001079}
1080
1081void
1082PySet_Fini(void)
1083{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001084 PySetObject *so;
Raymond Hettingerbc841a12005-08-07 13:02:53 +00001085
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001086 while (numfree) {
1087 numfree--;
1088 so = free_list[numfree];
1089 PyObject_GC_Del(so);
1090 }
1091 Py_CLEAR(dummy);
1092 Py_CLEAR(emptyfrozenset);
Raymond Hettingera690a992003-11-16 16:17:49 +00001093}
1094
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001095static PyObject *
1096set_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
1097{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001098 if (type == &PySet_Type && !_PyArg_NoKeywords("set()", kwds))
1099 return NULL;
1100
1101 return make_new_set(type, NULL);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001102}
1103
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001104/* set_swap_bodies() switches the contents of any two sets by moving their
1105 internal data pointers and, if needed, copying the internal smalltables.
1106 Semantically equivalent to:
1107
1108 t=set(a); a.clear(); a.update(b); b.clear(); b.update(t); del t
1109
1110 The function always succeeds and it leaves both objects in a stable state.
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001111 Useful for creating temporary frozensets from sets for membership testing
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001112 in __contains__(), discard(), and remove(). Also useful for operations
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001113 that update in-place (by allowing an intermediate result to be swapped
Raymond Hettinger9dcb17c2005-07-31 13:09:28 +00001114 into one of the original inputs).
Raymond Hettinger934d63e2005-07-31 01:33:10 +00001115*/
1116
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001117static void
1118set_swap_bodies(PySetObject *a, PySetObject *b)
Raymond Hettingera690a992003-11-16 16:17:49 +00001119{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001120 Py_ssize_t t;
1121 setentry *u;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00001122 setentry *(*f)(PySetObject *so, PyObject *key, Py_ssize_t hash);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001123 setentry tab[PySet_MINSIZE];
Antoine Pitroufbb1c612010-10-23 17:37:54 +00001124 Py_hash_t h;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001125
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001126 t = a->fill; a->fill = b->fill; b->fill = t;
1127 t = a->used; a->used = b->used; b->used = t;
1128 t = a->mask; a->mask = b->mask; b->mask = t;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001129
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001130 u = a->table;
1131 if (a->table == a->smalltable)
1132 u = b->smalltable;
1133 a->table = b->table;
1134 if (b->table == b->smalltable)
1135 a->table = a->smalltable;
1136 b->table = u;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001137
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001138 f = a->lookup; a->lookup = b->lookup; b->lookup = f;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001139
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001140 if (a->table == a->smalltable || b->table == b->smalltable) {
1141 memcpy(tab, a->smalltable, sizeof(tab));
1142 memcpy(a->smalltable, b->smalltable, sizeof(tab));
1143 memcpy(b->smalltable, tab, sizeof(tab));
1144 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001145
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001146 if (PyType_IsSubtype(Py_TYPE(a), &PyFrozenSet_Type) &&
1147 PyType_IsSubtype(Py_TYPE(b), &PyFrozenSet_Type)) {
1148 h = a->hash; a->hash = b->hash; b->hash = h;
1149 } else {
1150 a->hash = -1;
1151 b->hash = -1;
1152 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001153}
1154
Raymond Hettinger8f5cdaa2003-12-13 11:26:12 +00001155static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001156set_copy(PySetObject *so)
1157{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001158 return make_new_set_basetype(Py_TYPE(so), (PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001159}
1160
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001161static PyObject *
1162frozenset_copy(PySetObject *so)
1163{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001164 if (PyFrozenSet_CheckExact(so)) {
1165 Py_INCREF(so);
1166 return (PyObject *)so;
1167 }
1168 return set_copy(so);
Raymond Hettinger49ba4c32003-11-23 02:49:05 +00001169}
1170
Raymond Hettingera690a992003-11-16 16:17:49 +00001171PyDoc_STRVAR(copy_doc, "Return a shallow copy of a set.");
1172
1173static PyObject *
Raymond Hettingerc991db22005-08-11 07:58:45 +00001174set_clear(PySetObject *so)
1175{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001176 set_clear_internal(so);
1177 Py_RETURN_NONE;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001178}
1179
1180PyDoc_STRVAR(clear_doc, "Remove all elements from this set.");
1181
1182static PyObject *
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001183set_union(PySetObject *so, PyObject *args)
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001184{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001185 PySetObject *result;
1186 PyObject *other;
1187 Py_ssize_t i;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001188
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001189 result = (PySetObject *)set_copy(so);
1190 if (result == NULL)
1191 return NULL;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001192
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001193 for (i=0 ; i<PyTuple_GET_SIZE(args) ; i++) {
1194 other = PyTuple_GET_ITEM(args, i);
1195 if ((PyObject *)so == other)
1196 continue;
1197 if (set_update_internal(result, other) == -1) {
1198 Py_DECREF(result);
1199 return NULL;
1200 }
1201 }
1202 return (PyObject *)result;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001203}
1204
1205PyDoc_STRVAR(union_doc,
1206 "Return the union of sets as a new set.\n\
1207\n\
1208(i.e. all elements that are in either set.)");
1209
1210static PyObject *
1211set_or(PySetObject *so, PyObject *other)
1212{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001213 PySetObject *result;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001214
Brian Curtindfc80e32011-08-10 20:28:54 -05001215 if (!PyAnySet_Check(so) || !PyAnySet_Check(other))
1216 Py_RETURN_NOTIMPLEMENTED;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001217
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001218 result = (PySetObject *)set_copy(so);
1219 if (result == NULL)
1220 return NULL;
1221 if ((PyObject *)so == other)
1222 return (PyObject *)result;
1223 if (set_update_internal(result, other) == -1) {
1224 Py_DECREF(result);
1225 return NULL;
1226 }
1227 return (PyObject *)result;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001228}
1229
Raymond Hettingera690a992003-11-16 16:17:49 +00001230static PyObject *
1231set_ior(PySetObject *so, PyObject *other)
1232{
Brian Curtindfc80e32011-08-10 20:28:54 -05001233 if (!PyAnySet_Check(other))
1234 Py_RETURN_NOTIMPLEMENTED;
1235
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001236 if (set_update_internal(so, other) == -1)
1237 return NULL;
1238 Py_INCREF(so);
1239 return (PyObject *)so;
Raymond Hettingera690a992003-11-16 16:17:49 +00001240}
1241
1242static PyObject *
1243set_intersection(PySetObject *so, PyObject *other)
1244{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001245 PySetObject *result;
1246 PyObject *key, *it, *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001247
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001248 if ((PyObject *)so == other)
1249 return set_copy(so);
Raymond Hettingerc991db22005-08-11 07:58:45 +00001250
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001251 result = (PySetObject *)make_new_set_basetype(Py_TYPE(so), NULL);
1252 if (result == NULL)
1253 return NULL;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001254
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001255 if (PyAnySet_Check(other)) {
1256 Py_ssize_t pos = 0;
1257 setentry *entry;
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001258
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001259 if (PySet_GET_SIZE(other) > PySet_GET_SIZE(so)) {
1260 tmp = (PyObject *)so;
1261 so = (PySetObject *)other;
1262 other = tmp;
1263 }
Raymond Hettingerb02c35e2005-08-12 20:48:39 +00001264
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001265 while (set_next((PySetObject *)other, &pos, &entry)) {
1266 int rv = set_contains_entry(so, entry);
1267 if (rv == -1) {
1268 Py_DECREF(result);
1269 return NULL;
1270 }
1271 if (rv) {
1272 if (set_add_entry(result, entry) == -1) {
1273 Py_DECREF(result);
1274 return NULL;
1275 }
1276 }
1277 }
1278 return (PyObject *)result;
1279 }
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001280
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001281 it = PyObject_GetIter(other);
1282 if (it == NULL) {
1283 Py_DECREF(result);
1284 return NULL;
1285 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001286
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001287 while ((key = PyIter_Next(it)) != NULL) {
1288 int rv;
1289 setentry entry;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00001290 Py_hash_t hash = PyObject_Hash(key);
Thomas Wouters89f507f2006-12-13 04:49:30 +00001291
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001292 if (hash == -1) {
1293 Py_DECREF(it);
1294 Py_DECREF(result);
1295 Py_DECREF(key);
1296 return NULL;
1297 }
1298 entry.hash = hash;
1299 entry.key = key;
1300 rv = set_contains_entry(so, &entry);
1301 if (rv == -1) {
1302 Py_DECREF(it);
1303 Py_DECREF(result);
1304 Py_DECREF(key);
1305 return NULL;
1306 }
1307 if (rv) {
1308 if (set_add_entry(result, &entry) == -1) {
1309 Py_DECREF(it);
1310 Py_DECREF(result);
1311 Py_DECREF(key);
1312 return NULL;
1313 }
1314 }
1315 Py_DECREF(key);
1316 }
1317 Py_DECREF(it);
1318 if (PyErr_Occurred()) {
1319 Py_DECREF(result);
1320 return NULL;
1321 }
1322 return (PyObject *)result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001323}
1324
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001325static PyObject *
1326set_intersection_multi(PySetObject *so, PyObject *args)
1327{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001328 Py_ssize_t i;
1329 PyObject *result = (PyObject *)so;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001330
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001331 if (PyTuple_GET_SIZE(args) == 0)
1332 return set_copy(so);
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001333
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001334 Py_INCREF(so);
1335 for (i=0 ; i<PyTuple_GET_SIZE(args) ; i++) {
1336 PyObject *other = PyTuple_GET_ITEM(args, i);
1337 PyObject *newresult = set_intersection((PySetObject *)result, other);
1338 if (newresult == NULL) {
1339 Py_DECREF(result);
1340 return NULL;
1341 }
1342 Py_DECREF(result);
1343 result = newresult;
1344 }
1345 return result;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001346}
1347
Raymond Hettingera690a992003-11-16 16:17:49 +00001348PyDoc_STRVAR(intersection_doc,
1349"Return the intersection of two sets as a new set.\n\
1350\n\
1351(i.e. all elements that are in both sets.)");
1352
1353static PyObject *
1354set_intersection_update(PySetObject *so, PyObject *other)
1355{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001356 PyObject *tmp;
Raymond Hettingera690a992003-11-16 16:17:49 +00001357
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001358 tmp = set_intersection(so, other);
1359 if (tmp == NULL)
1360 return NULL;
1361 set_swap_bodies(so, (PySetObject *)tmp);
1362 Py_DECREF(tmp);
1363 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001364}
1365
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001366static PyObject *
1367set_intersection_update_multi(PySetObject *so, PyObject *args)
1368{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001369 PyObject *tmp;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001370
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001371 tmp = set_intersection_multi(so, args);
1372 if (tmp == NULL)
1373 return NULL;
1374 set_swap_bodies(so, (PySetObject *)tmp);
1375 Py_DECREF(tmp);
1376 Py_RETURN_NONE;
Georg Brandlc28e1fa2008-06-10 19:20:26 +00001377}
1378
Raymond Hettingera690a992003-11-16 16:17:49 +00001379PyDoc_STRVAR(intersection_update_doc,
1380"Update a set with the intersection of itself and another.");
1381
1382static PyObject *
1383set_and(PySetObject *so, PyObject *other)
1384{
Brian Curtindfc80e32011-08-10 20:28:54 -05001385 if (!PyAnySet_Check(so) || !PyAnySet_Check(other))
1386 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001387 return set_intersection(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +00001388}
1389
1390static PyObject *
1391set_iand(PySetObject *so, PyObject *other)
1392{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001393 PyObject *result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001394
Brian Curtindfc80e32011-08-10 20:28:54 -05001395 if (!PyAnySet_Check(other))
1396 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001397 result = set_intersection_update(so, other);
1398 if (result == NULL)
1399 return NULL;
1400 Py_DECREF(result);
1401 Py_INCREF(so);
1402 return (PyObject *)so;
Raymond Hettingera690a992003-11-16 16:17:49 +00001403}
1404
Guido van Rossum58da9312007-11-10 23:39:45 +00001405static PyObject *
1406set_isdisjoint(PySetObject *so, PyObject *other)
1407{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001408 PyObject *key, *it, *tmp;
Guido van Rossum58da9312007-11-10 23:39:45 +00001409
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001410 if ((PyObject *)so == other) {
1411 if (PySet_GET_SIZE(so) == 0)
1412 Py_RETURN_TRUE;
1413 else
1414 Py_RETURN_FALSE;
1415 }
Guido van Rossum58da9312007-11-10 23:39:45 +00001416
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001417 if (PyAnySet_CheckExact(other)) {
1418 Py_ssize_t pos = 0;
1419 setentry *entry;
Guido van Rossum58da9312007-11-10 23:39:45 +00001420
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001421 if (PySet_GET_SIZE(other) > PySet_GET_SIZE(so)) {
1422 tmp = (PyObject *)so;
1423 so = (PySetObject *)other;
1424 other = tmp;
1425 }
1426 while (set_next((PySetObject *)other, &pos, &entry)) {
1427 int rv = set_contains_entry(so, entry);
1428 if (rv == -1)
1429 return NULL;
1430 if (rv)
1431 Py_RETURN_FALSE;
1432 }
1433 Py_RETURN_TRUE;
1434 }
Guido van Rossum58da9312007-11-10 23:39:45 +00001435
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001436 it = PyObject_GetIter(other);
1437 if (it == NULL)
1438 return NULL;
Guido van Rossum58da9312007-11-10 23:39:45 +00001439
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001440 while ((key = PyIter_Next(it)) != NULL) {
1441 int rv;
1442 setentry entry;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00001443 Py_hash_t hash = PyObject_Hash(key);
Guido van Rossum58da9312007-11-10 23:39:45 +00001444
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001445 if (hash == -1) {
1446 Py_DECREF(key);
1447 Py_DECREF(it);
1448 return NULL;
1449 }
1450 entry.hash = hash;
1451 entry.key = key;
1452 rv = set_contains_entry(so, &entry);
1453 Py_DECREF(key);
1454 if (rv == -1) {
1455 Py_DECREF(it);
1456 return NULL;
1457 }
1458 if (rv) {
1459 Py_DECREF(it);
1460 Py_RETURN_FALSE;
1461 }
1462 }
1463 Py_DECREF(it);
1464 if (PyErr_Occurred())
1465 return NULL;
1466 Py_RETURN_TRUE;
Guido van Rossum58da9312007-11-10 23:39:45 +00001467}
1468
1469PyDoc_STRVAR(isdisjoint_doc,
1470"Return True if two sets have a null intersection.");
1471
Neal Norwitz6576bd82005-11-13 18:41:28 +00001472static int
Raymond Hettingerc991db22005-08-11 07:58:45 +00001473set_difference_update_internal(PySetObject *so, PyObject *other)
1474{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001475 if ((PyObject *)so == other)
1476 return set_clear_internal(so);
Raymond Hettingerc991db22005-08-11 07:58:45 +00001477
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001478 if (PyAnySet_Check(other)) {
1479 setentry *entry;
1480 Py_ssize_t pos = 0;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001481
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001482 while (set_next((PySetObject *)other, &pos, &entry))
1483 if (set_discard_entry(so, entry) == -1)
1484 return -1;
1485 } else {
1486 PyObject *key, *it;
1487 it = PyObject_GetIter(other);
1488 if (it == NULL)
1489 return -1;
1490
1491 while ((key = PyIter_Next(it)) != NULL) {
1492 if (set_discard_key(so, key) == -1) {
1493 Py_DECREF(it);
1494 Py_DECREF(key);
1495 return -1;
1496 }
1497 Py_DECREF(key);
1498 }
1499 Py_DECREF(it);
1500 if (PyErr_Occurred())
1501 return -1;
1502 }
1503 /* If more than 1/5 are dummies, then resize them away. */
1504 if ((so->fill - so->used) * 5 < so->mask)
1505 return 0;
1506 return set_table_resize(so, so->used>50000 ? so->used*2 : so->used*4);
Raymond Hettingerc991db22005-08-11 07:58:45 +00001507}
1508
Raymond Hettingera690a992003-11-16 16:17:49 +00001509static PyObject *
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001510set_difference_update(PySetObject *so, PyObject *args)
Raymond Hettingera690a992003-11-16 16:17:49 +00001511{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001512 Py_ssize_t i;
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001513
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001514 for (i=0 ; i<PyTuple_GET_SIZE(args) ; i++) {
1515 PyObject *other = PyTuple_GET_ITEM(args, i);
1516 if (set_difference_update_internal(so, other) == -1)
1517 return NULL;
1518 }
1519 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001520}
1521
1522PyDoc_STRVAR(difference_update_doc,
1523"Remove all elements of another set from this set.");
1524
1525static PyObject *
Antoine Pitrou715f3cd2010-11-30 22:23:20 +00001526set_copy_and_difference(PySetObject *so, PyObject *other)
1527{
1528 PyObject *result;
1529
1530 result = set_copy(so);
1531 if (result == NULL)
1532 return NULL;
1533 if (set_difference_update_internal((PySetObject *) result, other) != -1)
1534 return result;
1535 Py_DECREF(result);
1536 return NULL;
1537}
1538
1539static PyObject *
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001540set_difference(PySetObject *so, PyObject *other)
1541{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001542 PyObject *result;
1543 setentry *entry;
1544 Py_ssize_t pos = 0;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001545
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001546 if (!PyAnySet_Check(other) && !PyDict_CheckExact(other)) {
Antoine Pitrou715f3cd2010-11-30 22:23:20 +00001547 return set_copy_and_difference(so, other);
1548 }
1549
1550 /* If len(so) much more than len(other), it's more efficient to simply copy
1551 * so and then iterate other looking for common elements. */
1552 if ((PySet_GET_SIZE(so) >> 2) > PyObject_Size(other)) {
1553 return set_copy_and_difference(so, other);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001554 }
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001555
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001556 result = make_new_set_basetype(Py_TYPE(so), NULL);
1557 if (result == NULL)
1558 return NULL;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001559
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001560 if (PyDict_CheckExact(other)) {
1561 while (set_next(so, &pos, &entry)) {
1562 setentry entrycopy;
1563 entrycopy.hash = entry->hash;
1564 entrycopy.key = entry->key;
Antoine Pitroufbb1c612010-10-23 17:37:54 +00001565 if (!_PyDict_Contains(other, entry->key, entry->hash)) {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001566 if (set_add_entry((PySetObject *)result, &entrycopy) == -1) {
1567 Py_DECREF(result);
1568 return NULL;
1569 }
1570 }
1571 }
1572 return result;
1573 }
1574
Antoine Pitrou715f3cd2010-11-30 22:23:20 +00001575 /* Iterate over so, checking for common elements in other. */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001576 while (set_next(so, &pos, &entry)) {
1577 int rv = set_contains_entry((PySetObject *)other, entry);
1578 if (rv == -1) {
1579 Py_DECREF(result);
1580 return NULL;
1581 }
1582 if (!rv) {
1583 if (set_add_entry((PySetObject *)result, entry) == -1) {
1584 Py_DECREF(result);
1585 return NULL;
1586 }
1587 }
1588 }
1589 return result;
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001590}
1591
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001592static PyObject *
1593set_difference_multi(PySetObject *so, PyObject *args)
1594{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001595 Py_ssize_t i;
1596 PyObject *result, *other;
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001597
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001598 if (PyTuple_GET_SIZE(args) == 0)
1599 return set_copy(so);
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001600
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001601 other = PyTuple_GET_ITEM(args, 0);
1602 result = set_difference(so, other);
1603 if (result == NULL)
1604 return NULL;
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001605
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001606 for (i=1 ; i<PyTuple_GET_SIZE(args) ; i++) {
1607 other = PyTuple_GET_ITEM(args, i);
1608 if (set_difference_update_internal((PySetObject *)result, other) == -1) {
1609 Py_DECREF(result);
1610 return NULL;
1611 }
1612 }
1613 return result;
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001614}
1615
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001616PyDoc_STRVAR(difference_doc,
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001617"Return the difference of two or more sets as a new set.\n\
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001618\n\
Amaury Forgeot d'Arcfdfe62d2008-06-17 20:36:03 +00001619(i.e. all elements that are in this set but not the others.)");
Raymond Hettingerfb4e33a2003-12-15 13:23:55 +00001620static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001621set_sub(PySetObject *so, PyObject *other)
1622{
Brian Curtindfc80e32011-08-10 20:28:54 -05001623 if (!PyAnySet_Check(so) || !PyAnySet_Check(other))
1624 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001625 return set_difference(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +00001626}
1627
1628static PyObject *
1629set_isub(PySetObject *so, PyObject *other)
1630{
Brian Curtindfc80e32011-08-10 20:28:54 -05001631 if (!PyAnySet_Check(other))
1632 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001633 if (set_difference_update_internal(so, other) == -1)
1634 return NULL;
1635 Py_INCREF(so);
1636 return (PyObject *)so;
Raymond Hettingera690a992003-11-16 16:17:49 +00001637}
1638
1639static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001640set_symmetric_difference_update(PySetObject *so, PyObject *other)
1641{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001642 PySetObject *otherset;
1643 PyObject *key;
1644 Py_ssize_t pos = 0;
1645 setentry *entry;
Raymond Hettingerc991db22005-08-11 07:58:45 +00001646
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001647 if ((PyObject *)so == other)
1648 return set_clear(so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001649
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001650 if (PyDict_CheckExact(other)) {
1651 PyObject *value;
1652 int rv;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00001653 Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001654 while (_PyDict_Next(other, &pos, &key, &value, &hash)) {
1655 setentry an_entry;
Thomas Wouters89f507f2006-12-13 04:49:30 +00001656
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +00001657 Py_INCREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001658 an_entry.hash = hash;
1659 an_entry.key = key;
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +00001660
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001661 rv = set_discard_entry(so, &an_entry);
Georg Brandl2d444492010-09-03 10:52:55 +00001662 if (rv == -1) {
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +00001663 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001664 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001665 }
Raymond Hettingerfaf7b7f2010-09-03 10:00:50 +00001666 if (rv == DISCARD_NOTFOUND) {
1667 if (set_add_entry(so, &an_entry) == -1) {
1668 Py_DECREF(key);
1669 return NULL;
1670 }
1671 }
Georg Brandl2d444492010-09-03 10:52:55 +00001672 Py_DECREF(key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001673 }
1674 Py_RETURN_NONE;
1675 }
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001676
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001677 if (PyAnySet_Check(other)) {
1678 Py_INCREF(other);
1679 otherset = (PySetObject *)other;
1680 } else {
1681 otherset = (PySetObject *)make_new_set_basetype(Py_TYPE(so), other);
1682 if (otherset == NULL)
1683 return NULL;
1684 }
Raymond Hettingera690a992003-11-16 16:17:49 +00001685
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001686 while (set_next(otherset, &pos, &entry)) {
1687 int rv = set_discard_entry(so, entry);
1688 if (rv == -1) {
1689 Py_DECREF(otherset);
1690 return NULL;
1691 }
1692 if (rv == DISCARD_NOTFOUND) {
1693 if (set_add_entry(so, entry) == -1) {
1694 Py_DECREF(otherset);
1695 return NULL;
1696 }
1697 }
1698 }
1699 Py_DECREF(otherset);
1700 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001701}
1702
1703PyDoc_STRVAR(symmetric_difference_update_doc,
1704"Update a set with the symmetric difference of itself and another.");
1705
1706static PyObject *
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001707set_symmetric_difference(PySetObject *so, PyObject *other)
1708{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001709 PyObject *rv;
1710 PySetObject *otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001711
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001712 otherset = (PySetObject *)make_new_set_basetype(Py_TYPE(so), other);
1713 if (otherset == NULL)
1714 return NULL;
1715 rv = set_symmetric_difference_update(otherset, (PyObject *)so);
1716 if (rv == NULL)
1717 return NULL;
1718 Py_DECREF(rv);
1719 return (PyObject *)otherset;
Raymond Hettingerf5f41bf2003-11-24 02:57:33 +00001720}
1721
1722PyDoc_STRVAR(symmetric_difference_doc,
1723"Return the symmetric difference of two sets as a new set.\n\
1724\n\
1725(i.e. all elements that are in exactly one of the sets.)");
1726
1727static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001728set_xor(PySetObject *so, PyObject *other)
1729{
Brian Curtindfc80e32011-08-10 20:28:54 -05001730 if (!PyAnySet_Check(so) || !PyAnySet_Check(other))
1731 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001732 return set_symmetric_difference(so, other);
Raymond Hettingera690a992003-11-16 16:17:49 +00001733}
1734
1735static PyObject *
1736set_ixor(PySetObject *so, PyObject *other)
1737{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001738 PyObject *result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001739
Brian Curtindfc80e32011-08-10 20:28:54 -05001740 if (!PyAnySet_Check(other))
1741 Py_RETURN_NOTIMPLEMENTED;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001742 result = set_symmetric_difference_update(so, other);
1743 if (result == NULL)
1744 return NULL;
1745 Py_DECREF(result);
1746 Py_INCREF(so);
1747 return (PyObject *)so;
Raymond Hettingera690a992003-11-16 16:17:49 +00001748}
1749
1750static PyObject *
1751set_issubset(PySetObject *so, PyObject *other)
1752{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001753 setentry *entry;
1754 Py_ssize_t pos = 0;
Raymond Hettingera690a992003-11-16 16:17:49 +00001755
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001756 if (!PyAnySet_Check(other)) {
1757 PyObject *tmp, *result;
1758 tmp = make_new_set(&PySet_Type, other);
1759 if (tmp == NULL)
1760 return NULL;
1761 result = set_issubset(so, tmp);
1762 Py_DECREF(tmp);
1763 return result;
1764 }
1765 if (PySet_GET_SIZE(so) > PySet_GET_SIZE(other))
1766 Py_RETURN_FALSE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001767
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001768 while (set_next(so, &pos, &entry)) {
1769 int rv = set_contains_entry((PySetObject *)other, entry);
1770 if (rv == -1)
1771 return NULL;
1772 if (!rv)
1773 Py_RETURN_FALSE;
1774 }
1775 Py_RETURN_TRUE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001776}
1777
1778PyDoc_STRVAR(issubset_doc, "Report whether another set contains this set.");
1779
1780static PyObject *
1781set_issuperset(PySetObject *so, PyObject *other)
1782{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001783 PyObject *tmp, *result;
Raymond Hettinger3fbec702003-11-21 07:56:36 +00001784
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001785 if (!PyAnySet_Check(other)) {
1786 tmp = make_new_set(&PySet_Type, other);
1787 if (tmp == NULL)
1788 return NULL;
1789 result = set_issuperset(so, tmp);
1790 Py_DECREF(tmp);
1791 return result;
1792 }
1793 return set_issubset((PySetObject *)other, (PyObject *)so);
Raymond Hettingera690a992003-11-16 16:17:49 +00001794}
1795
1796PyDoc_STRVAR(issuperset_doc, "Report whether this set contains another set.");
1797
Raymond Hettingera690a992003-11-16 16:17:49 +00001798static PyObject *
1799set_richcompare(PySetObject *v, PyObject *w, int op)
1800{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001801 PyObject *r1, *r2;
Raymond Hettinger9f1a6792005-07-31 01:16:36 +00001802
Brian Curtindfc80e32011-08-10 20:28:54 -05001803 if(!PyAnySet_Check(w))
1804 Py_RETURN_NOTIMPLEMENTED;
1805
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001806 switch (op) {
1807 case Py_EQ:
1808 if (PySet_GET_SIZE(v) != PySet_GET_SIZE(w))
1809 Py_RETURN_FALSE;
1810 if (v->hash != -1 &&
1811 ((PySetObject *)w)->hash != -1 &&
1812 v->hash != ((PySetObject *)w)->hash)
1813 Py_RETURN_FALSE;
1814 return set_issubset(v, w);
1815 case Py_NE:
1816 r1 = set_richcompare(v, w, Py_EQ);
1817 if (r1 == NULL)
1818 return NULL;
1819 r2 = PyBool_FromLong(PyObject_Not(r1));
1820 Py_DECREF(r1);
1821 return r2;
1822 case Py_LE:
1823 return set_issubset(v, w);
1824 case Py_GE:
1825 return set_issuperset(v, w);
1826 case Py_LT:
1827 if (PySet_GET_SIZE(v) >= PySet_GET_SIZE(w))
1828 Py_RETURN_FALSE;
1829 return set_issubset(v, w);
1830 case Py_GT:
1831 if (PySet_GET_SIZE(v) <= PySet_GET_SIZE(w))
1832 Py_RETURN_FALSE;
1833 return set_issuperset(v, w);
1834 }
Brian Curtindfc80e32011-08-10 20:28:54 -05001835 Py_RETURN_NOTIMPLEMENTED;
Raymond Hettingera690a992003-11-16 16:17:49 +00001836}
1837
Raymond Hettingera690a992003-11-16 16:17:49 +00001838static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001839set_add(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001840{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001841 if (set_add_key(so, key) == -1)
1842 return NULL;
1843 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001844}
1845
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001846PyDoc_STRVAR(add_doc,
Raymond Hettingera690a992003-11-16 16:17:49 +00001847"Add an element to a set.\n\
1848\n\
1849This has no effect if the element is already present.");
1850
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001851static int
1852set_contains(PySetObject *so, PyObject *key)
1853{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001854 PyObject *tmpkey;
1855 int rv;
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001856
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001857 rv = set_contains_key(so, key);
1858 if (rv == -1) {
1859 if (!PySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1860 return -1;
1861 PyErr_Clear();
Raymond Hettinger38bf2cc2010-08-06 09:52:17 +00001862 tmpkey = make_new_set(&PyFrozenSet_Type, key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001863 if (tmpkey == NULL)
1864 return -1;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001865 rv = set_contains(so, tmpkey);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001866 Py_DECREF(tmpkey);
1867 }
1868 return rv;
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001869}
1870
1871static PyObject *
1872set_direct_contains(PySetObject *so, PyObject *key)
1873{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001874 long result;
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001875
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001876 result = set_contains(so, key);
1877 if (result == -1)
1878 return NULL;
1879 return PyBool_FromLong(result);
Raymond Hettingerce8185e2005-08-13 09:28:48 +00001880}
1881
1882PyDoc_STRVAR(contains_doc, "x.__contains__(y) <==> y in x.");
1883
Raymond Hettingera690a992003-11-16 16:17:49 +00001884static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001885set_remove(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001886{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001887 PyObject *tmpkey;
1888 int rv;
Raymond Hettingerbfd334a2003-11-22 03:55:23 +00001889
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001890 rv = set_discard_key(so, key);
1891 if (rv == -1) {
1892 if (!PySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1893 return NULL;
1894 PyErr_Clear();
Raymond Hettinger38bf2cc2010-08-06 09:52:17 +00001895 tmpkey = make_new_set(&PyFrozenSet_Type, key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001896 if (tmpkey == NULL)
1897 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001898 rv = set_discard_key(so, tmpkey);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001899 Py_DECREF(tmpkey);
1900 if (rv == -1)
1901 return NULL;
1902 }
Benjamin Petersonf10a79a2008-10-11 00:49:57 +00001903
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001904 if (rv == DISCARD_NOTFOUND) {
1905 set_key_error(key);
1906 return NULL;
1907 }
1908 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001909}
1910
1911PyDoc_STRVAR(remove_doc,
1912"Remove an element from a set; it must be a member.\n\
1913\n\
1914If the element is not a member, raise a KeyError.");
1915
1916static PyObject *
Raymond Hettinger06d8cf82005-07-31 15:36:06 +00001917set_discard(PySetObject *so, PyObject *key)
Raymond Hettingera690a992003-11-16 16:17:49 +00001918{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001919 PyObject *tmpkey, *result;
1920 int rv;
Raymond Hettinger0deab622003-12-13 18:53:18 +00001921
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001922 rv = set_discard_key(so, key);
1923 if (rv == -1) {
1924 if (!PySet_Check(key) || !PyErr_ExceptionMatches(PyExc_TypeError))
1925 return NULL;
1926 PyErr_Clear();
Raymond Hettinger38bf2cc2010-08-06 09:52:17 +00001927 tmpkey = make_new_set(&PyFrozenSet_Type, key);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001928 if (tmpkey == NULL)
1929 return NULL;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001930 result = set_discard(so, tmpkey);
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001931 Py_DECREF(tmpkey);
1932 return result;
1933 }
1934 Py_RETURN_NONE;
Raymond Hettingera690a992003-11-16 16:17:49 +00001935}
1936
1937PyDoc_STRVAR(discard_doc,
1938"Remove an element from a set if it is a member.\n\
1939\n\
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001940If the element is not a member, do nothing.");
Raymond Hettingera690a992003-11-16 16:17:49 +00001941
1942static PyObject *
Raymond Hettingera690a992003-11-16 16:17:49 +00001943set_reduce(PySetObject *so)
1944{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001945 PyObject *keys=NULL, *args=NULL, *result=NULL, *dict=NULL;
Raymond Hettingera690a992003-11-16 16:17:49 +00001946
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001947 keys = PySequence_List((PyObject *)so);
1948 if (keys == NULL)
1949 goto done;
1950 args = PyTuple_Pack(1, keys);
1951 if (args == NULL)
1952 goto done;
1953 dict = PyObject_GetAttrString((PyObject *)so, "__dict__");
1954 if (dict == NULL) {
1955 PyErr_Clear();
1956 dict = Py_None;
1957 Py_INCREF(dict);
1958 }
1959 result = PyTuple_Pack(3, Py_TYPE(so), args, dict);
Raymond Hettingera690a992003-11-16 16:17:49 +00001960done:
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001961 Py_XDECREF(args);
1962 Py_XDECREF(keys);
1963 Py_XDECREF(dict);
1964 return result;
Raymond Hettingera690a992003-11-16 16:17:49 +00001965}
1966
1967PyDoc_STRVAR(reduce_doc, "Return state information for pickling.");
1968
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00001969static PyObject *
1970set_sizeof(PySetObject *so)
1971{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001972 Py_ssize_t res;
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00001973
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001974 res = sizeof(PySetObject);
1975 if (so->table != so->smalltable)
1976 res = res + (so->mask + 1) * sizeof(setentry);
1977 return PyLong_FromSsize_t(res);
Robert Schuppeniesfbe94c52008-07-14 10:13:31 +00001978}
1979
1980PyDoc_STRVAR(sizeof_doc, "S.__sizeof__() -> size of S in memory, in bytes");
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001981static int
1982set_init(PySetObject *self, PyObject *args, PyObject *kwds)
1983{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001984 PyObject *iterable = NULL;
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001985
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00001986 if (!PyAnySet_Check(self))
1987 return -1;
1988 if (PySet_Check(self) && !_PyArg_NoKeywords("set()", kwds))
1989 return -1;
1990 if (!PyArg_UnpackTuple(args, Py_TYPE(self)->tp_name, 0, 1, &iterable))
1991 return -1;
1992 set_clear_internal(self);
1993 self->hash = -1;
1994 if (iterable == NULL)
1995 return 0;
1996 return set_update_internal(self, iterable);
Raymond Hettinger50a4bb32003-11-17 16:42:33 +00001997}
1998
Raymond Hettingera690a992003-11-16 16:17:49 +00001999static PySequenceMethods set_as_sequence = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002000 set_len, /* sq_length */
2001 0, /* sq_concat */
2002 0, /* sq_repeat */
2003 0, /* sq_item */
2004 0, /* sq_slice */
2005 0, /* sq_ass_item */
2006 0, /* sq_ass_slice */
2007 (objobjproc)set_contains, /* sq_contains */
Raymond Hettingera690a992003-11-16 16:17:49 +00002008};
2009
2010/* set object ********************************************************/
2011
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002012#ifdef Py_DEBUG
2013static PyObject *test_c_api(PySetObject *so);
2014
2015PyDoc_STRVAR(test_c_api_doc, "Exercises C API. Returns True.\n\
2016All is well if assertions don't fail.");
2017#endif
2018
Raymond Hettingera690a992003-11-16 16:17:49 +00002019static PyMethodDef set_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002020 {"add", (PyCFunction)set_add, METH_O,
2021 add_doc},
2022 {"clear", (PyCFunction)set_clear, METH_NOARGS,
2023 clear_doc},
2024 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
2025 contains_doc},
2026 {"copy", (PyCFunction)set_copy, METH_NOARGS,
2027 copy_doc},
2028 {"discard", (PyCFunction)set_discard, METH_O,
2029 discard_doc},
2030 {"difference", (PyCFunction)set_difference_multi, METH_VARARGS,
2031 difference_doc},
2032 {"difference_update", (PyCFunction)set_difference_update, METH_VARARGS,
2033 difference_update_doc},
2034 {"intersection",(PyCFunction)set_intersection_multi, METH_VARARGS,
2035 intersection_doc},
2036 {"intersection_update",(PyCFunction)set_intersection_update_multi, METH_VARARGS,
2037 intersection_update_doc},
2038 {"isdisjoint", (PyCFunction)set_isdisjoint, METH_O,
2039 isdisjoint_doc},
2040 {"issubset", (PyCFunction)set_issubset, METH_O,
2041 issubset_doc},
2042 {"issuperset", (PyCFunction)set_issuperset, METH_O,
2043 issuperset_doc},
2044 {"pop", (PyCFunction)set_pop, METH_NOARGS,
2045 pop_doc},
2046 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
2047 reduce_doc},
2048 {"remove", (PyCFunction)set_remove, METH_O,
2049 remove_doc},
2050 {"__sizeof__", (PyCFunction)set_sizeof, METH_NOARGS,
2051 sizeof_doc},
2052 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
2053 symmetric_difference_doc},
2054 {"symmetric_difference_update",(PyCFunction)set_symmetric_difference_update, METH_O,
2055 symmetric_difference_update_doc},
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002056#ifdef Py_DEBUG
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002057 {"test_c_api", (PyCFunction)test_c_api, METH_NOARGS,
2058 test_c_api_doc},
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002059#endif
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002060 {"union", (PyCFunction)set_union, METH_VARARGS,
2061 union_doc},
2062 {"update", (PyCFunction)set_update, METH_VARARGS,
2063 update_doc},
2064 {NULL, NULL} /* sentinel */
Raymond Hettingera690a992003-11-16 16:17:49 +00002065};
2066
2067static PyNumberMethods set_as_number = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002068 0, /*nb_add*/
2069 (binaryfunc)set_sub, /*nb_subtract*/
2070 0, /*nb_multiply*/
2071 0, /*nb_remainder*/
2072 0, /*nb_divmod*/
2073 0, /*nb_power*/
2074 0, /*nb_negative*/
2075 0, /*nb_positive*/
2076 0, /*nb_absolute*/
2077 0, /*nb_bool*/
2078 0, /*nb_invert*/
2079 0, /*nb_lshift*/
2080 0, /*nb_rshift*/
2081 (binaryfunc)set_and, /*nb_and*/
2082 (binaryfunc)set_xor, /*nb_xor*/
2083 (binaryfunc)set_or, /*nb_or*/
2084 0, /*nb_int*/
2085 0, /*nb_reserved*/
2086 0, /*nb_float*/
2087 0, /*nb_inplace_add*/
2088 (binaryfunc)set_isub, /*nb_inplace_subtract*/
2089 0, /*nb_inplace_multiply*/
2090 0, /*nb_inplace_remainder*/
2091 0, /*nb_inplace_power*/
2092 0, /*nb_inplace_lshift*/
2093 0, /*nb_inplace_rshift*/
2094 (binaryfunc)set_iand, /*nb_inplace_and*/
2095 (binaryfunc)set_ixor, /*nb_inplace_xor*/
2096 (binaryfunc)set_ior, /*nb_inplace_or*/
Raymond Hettingera690a992003-11-16 16:17:49 +00002097};
2098
2099PyDoc_STRVAR(set_doc,
Ezio Melotti7f807b72010-03-01 04:08:34 +00002100"set() -> new empty set object\n\
2101set(iterable) -> new set object\n\
Raymond Hettingera690a992003-11-16 16:17:49 +00002102\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002103Build an unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00002104
2105PyTypeObject PySet_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002106 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2107 "set", /* tp_name */
2108 sizeof(PySetObject), /* tp_basicsize */
2109 0, /* tp_itemsize */
2110 /* methods */
2111 (destructor)set_dealloc, /* tp_dealloc */
2112 0, /* tp_print */
2113 0, /* tp_getattr */
2114 0, /* tp_setattr */
2115 0, /* tp_reserved */
2116 (reprfunc)set_repr, /* tp_repr */
2117 &set_as_number, /* tp_as_number */
2118 &set_as_sequence, /* tp_as_sequence */
2119 0, /* tp_as_mapping */
Georg Brandl00da4e02010-10-18 07:32:48 +00002120 PyObject_HashNotImplemented, /* tp_hash */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002121 0, /* tp_call */
2122 0, /* tp_str */
2123 PyObject_GenericGetAttr, /* tp_getattro */
2124 0, /* tp_setattro */
2125 0, /* tp_as_buffer */
2126 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
2127 Py_TPFLAGS_BASETYPE, /* tp_flags */
2128 set_doc, /* tp_doc */
2129 (traverseproc)set_traverse, /* tp_traverse */
2130 (inquiry)set_clear_internal, /* tp_clear */
2131 (richcmpfunc)set_richcompare, /* tp_richcompare */
Georg Brandl00da4e02010-10-18 07:32:48 +00002132 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
2133 (getiterfunc)set_iter, /* tp_iter */
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002134 0, /* tp_iternext */
2135 set_methods, /* tp_methods */
2136 0, /* tp_members */
2137 0, /* tp_getset */
2138 0, /* tp_base */
2139 0, /* tp_dict */
2140 0, /* tp_descr_get */
2141 0, /* tp_descr_set */
2142 0, /* tp_dictoffset */
2143 (initproc)set_init, /* tp_init */
2144 PyType_GenericAlloc, /* tp_alloc */
2145 set_new, /* tp_new */
2146 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00002147};
2148
2149/* frozenset object ********************************************************/
2150
2151
2152static PyMethodDef frozenset_methods[] = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002153 {"__contains__",(PyCFunction)set_direct_contains, METH_O | METH_COEXIST,
2154 contains_doc},
2155 {"copy", (PyCFunction)frozenset_copy, METH_NOARGS,
2156 copy_doc},
2157 {"difference", (PyCFunction)set_difference_multi, METH_VARARGS,
2158 difference_doc},
2159 {"intersection",(PyCFunction)set_intersection_multi, METH_VARARGS,
2160 intersection_doc},
2161 {"isdisjoint", (PyCFunction)set_isdisjoint, METH_O,
2162 isdisjoint_doc},
2163 {"issubset", (PyCFunction)set_issubset, METH_O,
2164 issubset_doc},
2165 {"issuperset", (PyCFunction)set_issuperset, METH_O,
2166 issuperset_doc},
2167 {"__reduce__", (PyCFunction)set_reduce, METH_NOARGS,
2168 reduce_doc},
2169 {"__sizeof__", (PyCFunction)set_sizeof, METH_NOARGS,
2170 sizeof_doc},
2171 {"symmetric_difference",(PyCFunction)set_symmetric_difference, METH_O,
2172 symmetric_difference_doc},
2173 {"union", (PyCFunction)set_union, METH_VARARGS,
2174 union_doc},
2175 {NULL, NULL} /* sentinel */
Raymond Hettingera690a992003-11-16 16:17:49 +00002176};
2177
2178static PyNumberMethods frozenset_as_number = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002179 0, /*nb_add*/
2180 (binaryfunc)set_sub, /*nb_subtract*/
2181 0, /*nb_multiply*/
2182 0, /*nb_remainder*/
2183 0, /*nb_divmod*/
2184 0, /*nb_power*/
2185 0, /*nb_negative*/
2186 0, /*nb_positive*/
2187 0, /*nb_absolute*/
2188 0, /*nb_bool*/
2189 0, /*nb_invert*/
2190 0, /*nb_lshift*/
2191 0, /*nb_rshift*/
2192 (binaryfunc)set_and, /*nb_and*/
2193 (binaryfunc)set_xor, /*nb_xor*/
2194 (binaryfunc)set_or, /*nb_or*/
Raymond Hettingera690a992003-11-16 16:17:49 +00002195};
2196
2197PyDoc_STRVAR(frozenset_doc,
Ezio Melotti7f807b72010-03-01 04:08:34 +00002198"frozenset() -> empty frozenset object\n\
2199frozenset(iterable) -> frozenset object\n\
Raymond Hettingera690a992003-11-16 16:17:49 +00002200\n\
Thomas Wouters0e3f5912006-08-11 14:57:12 +00002201Build an immutable unordered collection of unique elements.");
Raymond Hettingera690a992003-11-16 16:17:49 +00002202
2203PyTypeObject PyFrozenSet_Type = {
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002204 PyVarObject_HEAD_INIT(&PyType_Type, 0)
2205 "frozenset", /* tp_name */
2206 sizeof(PySetObject), /* tp_basicsize */
2207 0, /* tp_itemsize */
2208 /* methods */
2209 (destructor)set_dealloc, /* tp_dealloc */
2210 0, /* tp_print */
2211 0, /* tp_getattr */
2212 0, /* tp_setattr */
2213 0, /* tp_reserved */
2214 (reprfunc)set_repr, /* tp_repr */
2215 &frozenset_as_number, /* tp_as_number */
2216 &set_as_sequence, /* tp_as_sequence */
2217 0, /* tp_as_mapping */
2218 frozenset_hash, /* tp_hash */
2219 0, /* tp_call */
2220 0, /* tp_str */
2221 PyObject_GenericGetAttr, /* tp_getattro */
2222 0, /* tp_setattro */
2223 0, /* tp_as_buffer */
2224 Py_TPFLAGS_DEFAULT | Py_TPFLAGS_HAVE_GC |
2225 Py_TPFLAGS_BASETYPE, /* tp_flags */
2226 frozenset_doc, /* tp_doc */
2227 (traverseproc)set_traverse, /* tp_traverse */
2228 (inquiry)set_clear_internal, /* tp_clear */
2229 (richcmpfunc)set_richcompare, /* tp_richcompare */
2230 offsetof(PySetObject, weakreflist), /* tp_weaklistoffset */
2231 (getiterfunc)set_iter, /* tp_iter */
2232 0, /* tp_iternext */
2233 frozenset_methods, /* tp_methods */
2234 0, /* tp_members */
2235 0, /* tp_getset */
2236 0, /* tp_base */
2237 0, /* tp_dict */
2238 0, /* tp_descr_get */
2239 0, /* tp_descr_set */
2240 0, /* tp_dictoffset */
2241 0, /* tp_init */
2242 PyType_GenericAlloc, /* tp_alloc */
2243 frozenset_new, /* tp_new */
2244 PyObject_GC_Del, /* tp_free */
Raymond Hettingera690a992003-11-16 16:17:49 +00002245};
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002246
2247
2248/***** C API functions *************************************************/
2249
2250PyObject *
2251PySet_New(PyObject *iterable)
2252{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002253 return make_new_set(&PySet_Type, iterable);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002254}
2255
2256PyObject *
2257PyFrozenSet_New(PyObject *iterable)
2258{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002259 return make_new_set(&PyFrozenSet_Type, iterable);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002260}
2261
Neal Norwitz8c49c822006-03-04 18:41:19 +00002262Py_ssize_t
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002263PySet_Size(PyObject *anyset)
2264{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002265 if (!PyAnySet_Check(anyset)) {
2266 PyErr_BadInternalCall();
2267 return -1;
2268 }
2269 return PySet_GET_SIZE(anyset);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002270}
2271
2272int
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002273PySet_Clear(PyObject *set)
2274{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002275 if (!PySet_Check(set)) {
2276 PyErr_BadInternalCall();
2277 return -1;
2278 }
2279 return set_clear_internal((PySetObject *)set);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002280}
2281
2282int
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002283PySet_Contains(PyObject *anyset, PyObject *key)
2284{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002285 if (!PyAnySet_Check(anyset)) {
2286 PyErr_BadInternalCall();
2287 return -1;
2288 }
2289 return set_contains_key((PySetObject *)anyset, key);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002290}
2291
2292int
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002293PySet_Discard(PyObject *set, PyObject *key)
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002294{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002295 if (!PySet_Check(set)) {
2296 PyErr_BadInternalCall();
2297 return -1;
2298 }
2299 return set_discard_key((PySetObject *)set, key);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002300}
2301
2302int
Christian Heimesfd66e512008-01-29 12:18:50 +00002303PySet_Add(PyObject *anyset, PyObject *key)
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002304{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002305 if (!PySet_Check(anyset) &&
2306 (!PyFrozenSet_Check(anyset) || Py_REFCNT(anyset) != 1)) {
2307 PyErr_BadInternalCall();
2308 return -1;
2309 }
2310 return set_add_key((PySetObject *)anyset, key);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002311}
2312
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002313int
Benjamin Peterson8f67d082010-10-17 20:54:53 +00002314_PySet_NextEntry(PyObject *set, Py_ssize_t *pos, PyObject **key, Py_hash_t *hash)
Guido van Rossumd8faa362007-04-27 19:54:29 +00002315{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002316 setentry *entry;
Guido van Rossumd8faa362007-04-27 19:54:29 +00002317
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002318 if (!PyAnySet_Check(set)) {
2319 PyErr_BadInternalCall();
2320 return -1;
2321 }
2322 if (set_next((PySetObject *)set, pos, &entry) == 0)
2323 return 0;
2324 *key = entry->key;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00002325 *hash = entry->hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002326 return 1;
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002327}
2328
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002329PyObject *
2330PySet_Pop(PyObject *set)
2331{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002332 if (!PySet_Check(set)) {
2333 PyErr_BadInternalCall();
2334 return NULL;
2335 }
2336 return set_pop((PySetObject *)set);
Raymond Hettingerbeb31012005-08-16 03:47:52 +00002337}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002338
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002339int
2340_PySet_Update(PyObject *set, PyObject *iterable)
2341{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002342 if (!PySet_Check(set)) {
2343 PyErr_BadInternalCall();
2344 return -1;
2345 }
2346 return set_update_internal((PySetObject *)set, iterable);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002347}
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002348
2349#ifdef Py_DEBUG
2350
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002351/* Test code to be called with any three element set.
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002352 Returns True and original set is restored. */
2353
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002354#define assertRaises(call_return_value, exception) \
2355 do { \
2356 assert(call_return_value); \
2357 assert(PyErr_ExceptionMatches(exception)); \
2358 PyErr_Clear(); \
2359 } while(0)
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002360
2361static PyObject *
2362test_c_api(PySetObject *so)
2363{
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002364 Py_ssize_t count;
2365 char *s;
2366 Py_ssize_t i;
2367 PyObject *elem=NULL, *dup=NULL, *t, *f, *dup2, *x;
2368 PyObject *ob = (PyObject *)so;
Benjamin Peterson8f67d082010-10-17 20:54:53 +00002369 Py_hash_t hash;
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002370 PyObject *str;
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002371
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002372 /* Verify preconditions */
2373 assert(PyAnySet_Check(ob));
2374 assert(PyAnySet_CheckExact(ob));
2375 assert(!PyFrozenSet_CheckExact(ob));
Victor Stinner08b36bd2010-03-13 00:19:17 +00002376
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002377 /* so.clear(); so |= set("abc"); */
2378 str = PyUnicode_FromString("abc");
2379 if (str == NULL)
2380 return NULL;
2381 set_clear_internal(so);
2382 if (set_update_internal(so, str) == -1) {
2383 Py_DECREF(str);
2384 return NULL;
2385 }
2386 Py_DECREF(str);
Victor Stinner08b36bd2010-03-13 00:19:17 +00002387
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002388 /* Exercise type/size checks */
2389 assert(PySet_Size(ob) == 3);
2390 assert(PySet_GET_SIZE(ob) == 3);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002391
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002392 /* Raise TypeError for non-iterable constructor arguments */
2393 assertRaises(PySet_New(Py_None) == NULL, PyExc_TypeError);
2394 assertRaises(PyFrozenSet_New(Py_None) == NULL, PyExc_TypeError);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002395
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002396 /* Raise TypeError for unhashable key */
2397 dup = PySet_New(ob);
2398 assertRaises(PySet_Discard(ob, dup) == -1, PyExc_TypeError);
2399 assertRaises(PySet_Contains(ob, dup) == -1, PyExc_TypeError);
2400 assertRaises(PySet_Add(ob, dup) == -1, PyExc_TypeError);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002401
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002402 /* Exercise successful pop, contains, add, and discard */
2403 elem = PySet_Pop(ob);
2404 assert(PySet_Contains(ob, elem) == 0);
2405 assert(PySet_GET_SIZE(ob) == 2);
2406 assert(PySet_Add(ob, elem) == 0);
2407 assert(PySet_Contains(ob, elem) == 1);
2408 assert(PySet_GET_SIZE(ob) == 3);
2409 assert(PySet_Discard(ob, elem) == 1);
2410 assert(PySet_GET_SIZE(ob) == 2);
2411 assert(PySet_Discard(ob, elem) == 0);
2412 assert(PySet_GET_SIZE(ob) == 2);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002413
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002414 /* Exercise clear */
2415 dup2 = PySet_New(dup);
2416 assert(PySet_Clear(dup2) == 0);
2417 assert(PySet_Size(dup2) == 0);
2418 Py_DECREF(dup2);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002419
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002420 /* Raise SystemError on clear or update of frozen set */
2421 f = PyFrozenSet_New(dup);
2422 assertRaises(PySet_Clear(f) == -1, PyExc_SystemError);
2423 assertRaises(_PySet_Update(f, dup) == -1, PyExc_SystemError);
2424 assert(PySet_Add(f, elem) == 0);
2425 Py_INCREF(f);
2426 assertRaises(PySet_Add(f, elem) == -1, PyExc_SystemError);
2427 Py_DECREF(f);
2428 Py_DECREF(f);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002429
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002430 /* Exercise direct iteration */
2431 i = 0, count = 0;
2432 while (_PySet_NextEntry((PyObject *)dup, &i, &x, &hash)) {
2433 s = _PyUnicode_AsString(x);
2434 assert(s && (s[0] == 'a' || s[0] == 'b' || s[0] == 'c'));
2435 count++;
2436 }
2437 assert(count == 3);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002438
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002439 /* Exercise updates */
2440 dup2 = PySet_New(NULL);
2441 assert(_PySet_Update(dup2, dup) == 0);
2442 assert(PySet_Size(dup2) == 3);
2443 assert(_PySet_Update(dup2, dup) == 0);
2444 assert(PySet_Size(dup2) == 3);
2445 Py_DECREF(dup2);
Thomas Wouters49fd7fa2006-04-21 10:40:58 +00002446
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002447 /* Raise SystemError when self argument is not a set or frozenset. */
2448 t = PyTuple_New(0);
2449 assertRaises(PySet_Size(t) == -1, PyExc_SystemError);
2450 assertRaises(PySet_Contains(t, elem) == -1, PyExc_SystemError);
2451 Py_DECREF(t);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002452
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002453 /* Raise SystemError when self argument is not a set. */
2454 f = PyFrozenSet_New(dup);
2455 assert(PySet_Size(f) == 3);
2456 assert(PyFrozenSet_CheckExact(f));
2457 assertRaises(PySet_Discard(f, elem) == -1, PyExc_SystemError);
2458 assertRaises(PySet_Pop(f) == NULL, PyExc_SystemError);
2459 Py_DECREF(f);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002460
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002461 /* Raise KeyError when popping from an empty set */
2462 assert(PyNumber_InPlaceSubtract(ob, ob) == ob);
2463 Py_DECREF(ob);
2464 assert(PySet_GET_SIZE(ob) == 0);
2465 assertRaises(PySet_Pop(ob) == NULL, PyExc_KeyError);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002466
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002467 /* Restore the set from the copy using the PyNumber API */
2468 assert(PyNumber_InPlaceOr(ob, dup) == ob);
2469 Py_DECREF(ob);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002470
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002471 /* Verify constructors accept NULL arguments */
2472 f = PySet_New(NULL);
2473 assert(f != NULL);
2474 assert(PySet_GET_SIZE(f) == 0);
2475 Py_DECREF(f);
2476 f = PyFrozenSet_New(NULL);
2477 assert(f != NULL);
2478 assert(PyFrozenSet_CheckExact(f));
2479 assert(PySet_GET_SIZE(f) == 0);
2480 Py_DECREF(f);
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002481
Antoine Pitrouf95a1b32010-05-09 15:52:27 +00002482 Py_DECREF(elem);
2483 Py_DECREF(dup);
2484 Py_RETURN_TRUE;
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002485}
2486
Raymond Hettinger9bda1d62005-09-16 07:14:21 +00002487#undef assertRaises
2488
Raymond Hettingerc47e01d2005-08-16 10:44:15 +00002489#endif