Damien George | 04b9147 | 2014-05-03 23:27:38 +0100 | [diff] [blame] | 1 | /* |
| 2 | * This file is part of the Micro Python project, http://micropython.org/ |
| 3 | * |
| 4 | * The MIT License (MIT) |
| 5 | * |
| 6 | * Copyright (c) 2013, 2014 Damien P. George |
| 7 | * |
| 8 | * Permission is hereby granted, free of charge, to any person obtaining a copy |
| 9 | * of this software and associated documentation files (the "Software"), to deal |
| 10 | * in the Software without restriction, including without limitation the rights |
| 11 | * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
| 12 | * copies of the Software, and to permit persons to whom the Software is |
| 13 | * furnished to do so, subject to the following conditions: |
| 14 | * |
| 15 | * The above copyright notice and this permission notice shall be included in |
| 16 | * all copies or substantial portions of the Software. |
| 17 | * |
| 18 | * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
| 19 | * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
| 20 | * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
| 21 | * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
| 22 | * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
| 23 | * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN |
| 24 | * THE SOFTWARE. |
| 25 | */ |
| 26 | |
xbe | efe3422 | 2014-03-16 00:14:26 -0700 | [diff] [blame] | 27 | #include <stdbool.h> |
John R. Lenton | 3b0bd87 | 2014-01-12 15:56:25 +0000 | [diff] [blame] | 28 | #include <string.h> |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 29 | #include <assert.h> |
| 30 | |
Damien George | 51dfcb4 | 2015-01-01 20:27:54 +0000 | [diff] [blame] | 31 | #include "py/nlr.h" |
| 32 | #include "py/runtime.h" |
| 33 | #include "py/runtime0.h" |
| 34 | #include "py/builtin.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 35 | |
Damien George | 3ebd4d0 | 2014-06-01 13:46:47 +0100 | [diff] [blame] | 36 | #if MICROPY_PY_BUILTINS_SET |
| 37 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 38 | typedef struct _mp_obj_set_t { |
| 39 | mp_obj_base_t base; |
| 40 | mp_set_t set; |
| 41 | } mp_obj_set_t; |
| 42 | |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 43 | typedef struct _mp_obj_set_it_t { |
| 44 | mp_obj_base_t base; |
| 45 | mp_obj_set_t *set; |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 46 | mp_uint_t cur; |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 47 | } mp_obj_set_it_t; |
| 48 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 49 | STATIC mp_obj_t set_it_iternext(mp_obj_t self_in); |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 50 | |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 51 | STATIC bool is_set_or_frozenset(mp_obj_t o) { |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 52 | return MP_OBJ_IS_TYPE(o, &mp_type_set) |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 53 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 54 | || MP_OBJ_IS_TYPE(o, &mp_type_frozenset) |
| 55 | #endif |
| 56 | ; |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 57 | } |
| 58 | |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 59 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 60 | STATIC void check_set_or_frozenset(mp_obj_t o) { |
| 61 | if (!is_set_or_frozenset(o)) { |
| 62 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "'set' object required")); |
| 63 | } |
| 64 | } |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 65 | #else |
| 66 | #define check_set_or_frozenset(o) check_set(o) |
| 67 | #endif |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 68 | |
| 69 | STATIC void check_set(mp_obj_t o) { |
| 70 | if (!MP_OBJ_IS_TYPE(o, &mp_type_set)) { |
| 71 | // Emulate CPython behavior |
| 72 | // AttributeError: 'frozenset' object has no attribute 'add' |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 73 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 74 | if (MP_OBJ_IS_TYPE(o, &mp_type_frozenset)) { |
| 75 | nlr_raise(mp_obj_new_exception_msg(&mp_type_AttributeError, "'frozenset' has no such attribute")); |
| 76 | } |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 77 | #endif |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 78 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "'set' object required")); |
| 79 | } |
| 80 | } |
| 81 | |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 82 | STATIC void set_print(const mp_print_t *print, mp_obj_t self_in, mp_print_kind_t kind) { |
Damien George | ff8dd3f | 2015-01-20 12:47:20 +0000 | [diff] [blame] | 83 | (void)kind; |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 84 | mp_obj_set_t *self = self_in; |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 85 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 86 | bool is_frozen = MP_OBJ_IS_TYPE(self_in, &mp_type_frozenset); |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 87 | #endif |
John R. Lenton | 7244a14 | 2014-01-12 23:37:45 +0000 | [diff] [blame] | 88 | if (self->set.used == 0) { |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 89 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 90 | if (is_frozen) { |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 91 | mp_print_str(print, "frozen"); |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 92 | } |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 93 | #endif |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 94 | mp_print_str(print, "set()"); |
John R. Lenton | 7244a14 | 2014-01-12 23:37:45 +0000 | [diff] [blame] | 95 | return; |
| 96 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 97 | bool first = true; |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 98 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 99 | if (is_frozen) { |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 100 | mp_print_str(print, "frozenset("); |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 101 | } |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 102 | #endif |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 103 | mp_print_str(print, "{"); |
Damien George | 93965e7 | 2014-08-30 13:23:35 +0100 | [diff] [blame] | 104 | for (mp_uint_t i = 0; i < self->set.alloc; i++) { |
Damien George | 8b0535e | 2014-04-05 21:53:54 +0100 | [diff] [blame] | 105 | if (MP_SET_SLOT_IS_FILLED(&self->set, i)) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 106 | if (!first) { |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 107 | mp_print_str(print, ", "); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 108 | } |
| 109 | first = false; |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 110 | mp_obj_print_helper(print, self->set.table[i], PRINT_REPR); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 111 | } |
| 112 | } |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 113 | mp_print_str(print, "}"); |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 114 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 115 | if (is_frozen) { |
Damien George | 7f9d1d6 | 2015-04-09 23:56:15 +0100 | [diff] [blame] | 116 | mp_print_str(print, ")"); |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 117 | } |
Paul Sokolovsky | d80e247 | 2014-05-10 16:11:04 +0300 | [diff] [blame] | 118 | #endif |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 119 | } |
| 120 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 121 | STATIC mp_obj_t set_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args) { |
Damien George | 1d34e32 | 2014-05-11 18:28:48 +0100 | [diff] [blame] | 122 | mp_arg_check_num(n_args, n_kw, 0, 1, false); |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 123 | |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 124 | switch (n_args) { |
Damien George | 1d34e32 | 2014-05-11 18:28:48 +0100 | [diff] [blame] | 125 | case 0: { |
| 126 | // create a new, empty set |
| 127 | mp_obj_set_t *set = mp_obj_new_set(0, NULL); |
| 128 | // set actual set/frozenset type |
| 129 | set->base.type = type_in; |
| 130 | return set; |
| 131 | } |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 132 | |
| 133 | case 1: |
Damien George | 1d34e32 | 2014-05-11 18:28:48 +0100 | [diff] [blame] | 134 | default: { // can only be 0 or 1 arg |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 135 | // 1 argument, an iterable from which we make a new set |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 136 | mp_obj_set_t *set = mp_obj_new_set(0, NULL); |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 137 | mp_obj_t iterable = mp_getiter(args[0]); |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 138 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 139 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 140 | mp_obj_set_store(set, item); |
| 141 | } |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 142 | // Set actual set/frozenset type |
| 143 | set->base.type = type_in; |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 144 | return set; |
| 145 | } |
Damien George | 71c5181 | 2014-01-04 20:21:15 +0000 | [diff] [blame] | 146 | } |
| 147 | } |
| 148 | |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 149 | const mp_obj_type_t mp_type_set_it = { |
Damien George | c596612 | 2014-02-15 16:10:44 +0000 | [diff] [blame] | 150 | { &mp_type_type }, |
Damien George | a71c83a | 2014-02-15 11:34:50 +0000 | [diff] [blame] | 151 | .name = MP_QSTR_iterator, |
Paul Sokolovsky | f7eaf60 | 2014-03-30 22:00:12 +0300 | [diff] [blame] | 152 | .getiter = mp_identity, |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 153 | .iternext = set_it_iternext, |
| 154 | }; |
| 155 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 156 | STATIC mp_obj_t set_it_iternext(mp_obj_t self_in) { |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 157 | assert(MP_OBJ_IS_TYPE(self_in, &mp_type_set_it)); |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 158 | mp_obj_set_it_t *self = self_in; |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 159 | mp_uint_t max = self->set->set.alloc; |
Damien George | 8b0535e | 2014-04-05 21:53:54 +0100 | [diff] [blame] | 160 | mp_set_t *set = &self->set->set; |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 161 | |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 162 | for (mp_uint_t i = self->cur; i < max; i++) { |
Damien George | 8b0535e | 2014-04-05 21:53:54 +0100 | [diff] [blame] | 163 | if (MP_SET_SLOT_IS_FILLED(set, i)) { |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 164 | self->cur = i + 1; |
Damien George | 8b0535e | 2014-04-05 21:53:54 +0100 | [diff] [blame] | 165 | return set->table[i]; |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 166 | } |
| 167 | } |
| 168 | |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 169 | return MP_OBJ_STOP_ITERATION; |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 170 | } |
| 171 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 172 | STATIC mp_obj_t set_getiter(mp_obj_t set_in) { |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 173 | mp_obj_set_it_t *o = m_new_obj(mp_obj_set_it_t); |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 174 | o->base.type = &mp_type_set_it; |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 175 | o->set = (mp_obj_set_t *)set_in; |
| 176 | o->cur = 0; |
| 177 | return o; |
| 178 | } |
| 179 | |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 180 | |
| 181 | /******************************************************************************/ |
| 182 | /* set methods */ |
| 183 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 184 | STATIC mp_obj_t set_add(mp_obj_t self_in, mp_obj_t item) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 185 | check_set(self_in); |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 186 | mp_obj_set_t *self = self_in; |
John R. Lenton | 2a24172 | 2014-01-12 16:39:39 +0000 | [diff] [blame] | 187 | mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 188 | return mp_const_none; |
| 189 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 190 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_add_obj, set_add); |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 191 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 192 | STATIC mp_obj_t set_clear(mp_obj_t self_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 193 | check_set(self_in); |
John R. Lenton | 1d7fb2f | 2014-01-12 15:44:26 +0000 | [diff] [blame] | 194 | mp_obj_set_t *self = self_in; |
| 195 | |
| 196 | mp_set_clear(&self->set); |
| 197 | |
| 198 | return mp_const_none; |
| 199 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 200 | STATIC MP_DEFINE_CONST_FUN_OBJ_1(set_clear_obj, set_clear); |
John R. Lenton | 1d7fb2f | 2014-01-12 15:44:26 +0000 | [diff] [blame] | 201 | |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 202 | STATIC mp_obj_t set_copy_as_mutable(mp_obj_t self_in) { |
John R. Lenton | 3b0bd87 | 2014-01-12 15:56:25 +0000 | [diff] [blame] | 203 | mp_obj_set_t *self = self_in; |
| 204 | |
| 205 | mp_obj_set_t *other = m_new_obj(mp_obj_set_t); |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 206 | other->base.type = &mp_type_set; |
Paul Sokolovsky | 46bd12d | 2014-04-07 03:07:21 +0300 | [diff] [blame] | 207 | mp_set_init(&other->set, self->set.alloc); |
John R. Lenton | 3b0bd87 | 2014-01-12 15:56:25 +0000 | [diff] [blame] | 208 | other->set.used = self->set.used; |
| 209 | memcpy(other->set.table, self->set.table, self->set.alloc * sizeof(mp_obj_t)); |
| 210 | |
| 211 | return other; |
| 212 | } |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 213 | |
| 214 | STATIC mp_obj_t set_copy(mp_obj_t self_in) { |
| 215 | check_set_or_frozenset(self_in); |
| 216 | mp_obj_set_t *self = self_in; |
| 217 | |
| 218 | mp_obj_set_t *other = set_copy_as_mutable(self); |
| 219 | other->base.type = self->base.type; |
| 220 | |
| 221 | return other; |
| 222 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 223 | STATIC MP_DEFINE_CONST_FUN_OBJ_1(set_copy_obj, set_copy); |
John R. Lenton | 3b0bd87 | 2014-01-12 15:56:25 +0000 | [diff] [blame] | 224 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 225 | STATIC mp_obj_t set_discard(mp_obj_t self_in, mp_obj_t item) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 226 | check_set(self_in); |
John R. Lenton | 2a24172 | 2014-01-12 16:39:39 +0000 | [diff] [blame] | 227 | mp_obj_set_t *self = self_in; |
| 228 | mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_REMOVE_IF_FOUND); |
| 229 | return mp_const_none; |
| 230 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 231 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_discard_obj, set_discard); |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 232 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 233 | STATIC mp_obj_t set_diff_int(mp_uint_t n_args, const mp_obj_t *args, bool update) { |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 234 | assert(n_args > 0); |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 235 | |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 236 | mp_obj_set_t *self; |
| 237 | if (update) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 238 | check_set(args[0]); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 239 | self = args[0]; |
| 240 | } else { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 241 | check_set_or_frozenset(args[0]); |
| 242 | self = set_copy_as_mutable(args[0]); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 243 | } |
| 244 | |
| 245 | |
Damien George | 93965e7 | 2014-08-30 13:23:35 +0100 | [diff] [blame] | 246 | for (mp_uint_t i = 1; i < n_args; i++) { |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 247 | mp_obj_t other = args[i]; |
| 248 | if (self == other) { |
| 249 | set_clear(self); |
| 250 | } else { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 251 | mp_obj_t iter = mp_getiter(other); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 252 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 253 | while ((next = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 254 | set_discard(self, next); |
| 255 | } |
| 256 | } |
| 257 | } |
| 258 | |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 259 | self->base.type = ((mp_obj_set_t*)args[0])->base.type; |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 260 | return self; |
| 261 | } |
| 262 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 263 | STATIC mp_obj_t set_diff(mp_uint_t n_args, const mp_obj_t *args) { |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 264 | return set_diff_int(n_args, args, false); |
| 265 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 266 | STATIC MP_DEFINE_CONST_FUN_OBJ_VAR(set_diff_obj, 1, set_diff); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 267 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 268 | STATIC mp_obj_t set_diff_update(mp_uint_t n_args, const mp_obj_t *args) { |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 269 | set_diff_int(n_args, args, true); |
| 270 | return mp_const_none; |
| 271 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 272 | STATIC MP_DEFINE_CONST_FUN_OBJ_VAR(set_diff_update_obj, 1, set_diff_update); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 273 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 274 | STATIC mp_obj_t set_intersect_int(mp_obj_t self_in, mp_obj_t other, bool update) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 275 | if (update) { |
| 276 | check_set(self_in); |
| 277 | } else { |
| 278 | check_set_or_frozenset(self_in); |
| 279 | } |
| 280 | |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 281 | if (self_in == other) { |
| 282 | return update ? mp_const_none : set_copy(self_in); |
| 283 | } |
| 284 | |
| 285 | mp_obj_set_t *self = self_in; |
| 286 | mp_obj_set_t *out = mp_obj_new_set(0, NULL); |
| 287 | |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 288 | mp_obj_t iter = mp_getiter(other); |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 289 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 290 | while ((next = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 291 | if (mp_set_lookup(&self->set, next, MP_MAP_LOOKUP)) { |
| 292 | set_add(out, next); |
| 293 | } |
| 294 | } |
| 295 | |
| 296 | if (update) { |
| 297 | m_del(mp_obj_t, self->set.table, self->set.alloc); |
| 298 | self->set.alloc = out->set.alloc; |
| 299 | self->set.used = out->set.used; |
| 300 | self->set.table = out->set.table; |
| 301 | } |
| 302 | |
| 303 | return update ? mp_const_none : out; |
| 304 | } |
| 305 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 306 | STATIC mp_obj_t set_intersect(mp_obj_t self_in, mp_obj_t other) { |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 307 | return set_intersect_int(self_in, other, false); |
| 308 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 309 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_intersect_obj, set_intersect); |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 310 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 311 | STATIC mp_obj_t set_intersect_update(mp_obj_t self_in, mp_obj_t other) { |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 312 | return set_intersect_int(self_in, other, true); |
| 313 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 314 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_intersect_update_obj, set_intersect_update); |
John R. Lenton | f1ae6b4 | 2014-01-12 17:54:03 +0000 | [diff] [blame] | 315 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 316 | STATIC mp_obj_t set_isdisjoint(mp_obj_t self_in, mp_obj_t other) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 317 | check_set_or_frozenset(self_in); |
John R. Lenton | 4a08067 | 2014-01-12 18:03:21 +0000 | [diff] [blame] | 318 | mp_obj_set_t *self = self_in; |
| 319 | |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 320 | mp_obj_t iter = mp_getiter(other); |
John R. Lenton | 4a08067 | 2014-01-12 18:03:21 +0000 | [diff] [blame] | 321 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 322 | while ((next = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
John R. Lenton | 4a08067 | 2014-01-12 18:03:21 +0000 | [diff] [blame] | 323 | if (mp_set_lookup(&self->set, next, MP_MAP_LOOKUP)) { |
| 324 | return mp_const_false; |
| 325 | } |
| 326 | } |
| 327 | return mp_const_true; |
| 328 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 329 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_isdisjoint_obj, set_isdisjoint); |
John R. Lenton | 4a08067 | 2014-01-12 18:03:21 +0000 | [diff] [blame] | 330 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 331 | STATIC mp_obj_t set_issubset_internal(mp_obj_t self_in, mp_obj_t other_in, bool proper) { |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 332 | mp_obj_set_t *self; |
| 333 | bool cleanup_self = false; |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 334 | if (is_set_or_frozenset(self_in)) { |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 335 | self = self_in; |
| 336 | } else { |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 337 | self = set_make_new((mp_obj_t)&mp_type_set, 1, 0, &self_in); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 338 | cleanup_self = true; |
| 339 | } |
| 340 | |
| 341 | mp_obj_set_t *other; |
| 342 | bool cleanup_other = false; |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 343 | if (is_set_or_frozenset(other_in)) { |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 344 | other = other_in; |
| 345 | } else { |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 346 | other = set_make_new((mp_obj_t)&mp_type_set, 1, 0, &other_in); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 347 | cleanup_other = true; |
| 348 | } |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 349 | bool out = true; |
| 350 | if (proper && self->set.used == other->set.used) { |
| 351 | out = false; |
| 352 | } else { |
| 353 | mp_obj_t iter = set_getiter(self); |
| 354 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 355 | while ((next = set_it_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 356 | if (!mp_set_lookup(&other->set, next, MP_MAP_LOOKUP)) { |
| 357 | out = false; |
| 358 | break; |
| 359 | } |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 360 | } |
| 361 | } |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 362 | // TODO: Should free objects altogether |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 363 | if (cleanup_self) { |
| 364 | set_clear(self); |
| 365 | } |
| 366 | if (cleanup_other) { |
| 367 | set_clear(other); |
| 368 | } |
Paul Sokolovsky | 1b586f3 | 2015-10-11 12:09:43 +0300 | [diff] [blame^] | 369 | return mp_obj_new_bool(out); |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 370 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 371 | STATIC mp_obj_t set_issubset(mp_obj_t self_in, mp_obj_t other_in) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 372 | return set_issubset_internal(self_in, other_in, false); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 373 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 374 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_issubset_obj, set_issubset); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 375 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 376 | STATIC mp_obj_t set_issubset_proper(mp_obj_t self_in, mp_obj_t other_in) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 377 | return set_issubset_internal(self_in, other_in, true); |
| 378 | } |
| 379 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 380 | STATIC mp_obj_t set_issuperset(mp_obj_t self_in, mp_obj_t other_in) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 381 | return set_issubset_internal(other_in, self_in, false); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 382 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 383 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_issuperset_obj, set_issuperset); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 384 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 385 | STATIC mp_obj_t set_issuperset_proper(mp_obj_t self_in, mp_obj_t other_in) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 386 | return set_issubset_internal(other_in, self_in, true); |
| 387 | } |
| 388 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 389 | STATIC mp_obj_t set_equal(mp_obj_t self_in, mp_obj_t other_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 390 | check_set_or_frozenset(self_in); |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 391 | mp_obj_set_t *self = self_in; |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 392 | if (!is_set_or_frozenset(other_in)) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 393 | return mp_const_false; |
| 394 | } |
| 395 | mp_obj_set_t *other = other_in; |
| 396 | if (self->set.used != other->set.used) { |
| 397 | return mp_const_false; |
| 398 | } |
| 399 | return set_issubset(self_in, other_in); |
| 400 | } |
| 401 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 402 | STATIC mp_obj_t set_pop(mp_obj_t self_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 403 | check_set(self_in); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 404 | mp_obj_set_t *self = self_in; |
Damien George | 95004e5 | 2014-04-05 17:17:19 +0100 | [diff] [blame] | 405 | mp_obj_t obj = mp_set_remove_first(&self->set); |
| 406 | if (obj == MP_OBJ_NULL) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 407 | nlr_raise(mp_obj_new_exception_msg(&mp_type_KeyError, "pop from an empty set")); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 408 | } |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 409 | return obj; |
| 410 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 411 | STATIC MP_DEFINE_CONST_FUN_OBJ_1(set_pop_obj, set_pop); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 412 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 413 | STATIC mp_obj_t set_remove(mp_obj_t self_in, mp_obj_t item) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 414 | check_set(self_in); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 415 | mp_obj_set_t *self = self_in; |
| 416 | if (mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_REMOVE_IF_FOUND) == MP_OBJ_NULL) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 417 | nlr_raise(mp_obj_new_exception(&mp_type_KeyError)); |
John R. Lenton | ae00d33 | 2014-01-12 18:23:36 +0000 | [diff] [blame] | 418 | } |
| 419 | return mp_const_none; |
| 420 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 421 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_remove_obj, set_remove); |
John R. Lenton | 032129f | 2014-01-12 17:07:17 +0000 | [diff] [blame] | 422 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 423 | STATIC mp_obj_t set_symmetric_difference_update(mp_obj_t self_in, mp_obj_t other_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 424 | check_set(self_in); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 425 | mp_obj_set_t *self = self_in; |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 426 | mp_obj_t iter = mp_getiter(other_in); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 427 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 428 | while ((next = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
Damien George | d1cee02 | 2015-03-20 17:41:37 +0000 | [diff] [blame] | 429 | mp_set_lookup(&self->set, next, MP_MAP_LOOKUP_ADD_IF_NOT_FOUND_OR_REMOVE_IF_FOUND); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 430 | } |
| 431 | return mp_const_none; |
| 432 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 433 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_symmetric_difference_update_obj, set_symmetric_difference_update); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 434 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 435 | STATIC mp_obj_t set_symmetric_difference(mp_obj_t self_in, mp_obj_t other_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 436 | check_set_or_frozenset(self_in); |
| 437 | mp_obj_set_t *self_out = set_copy_as_mutable(self_in); |
| 438 | set_symmetric_difference_update(self_out, other_in); |
| 439 | self_out->base.type = ((mp_obj_set_t*)self_in)->base.type; |
| 440 | return self_out; |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 441 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 442 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_symmetric_difference_obj, set_symmetric_difference); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 443 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 444 | STATIC void set_update_int(mp_obj_set_t *self, mp_obj_t other_in) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 445 | mp_obj_t iter = mp_getiter(other_in); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 446 | mp_obj_t next; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 447 | while ((next = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) { |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 448 | mp_set_lookup(&self->set, next, MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); |
| 449 | } |
| 450 | } |
| 451 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 452 | STATIC mp_obj_t set_update(mp_uint_t n_args, const mp_obj_t *args) { |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 453 | assert(n_args > 0); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 454 | |
Damien George | 93965e7 | 2014-08-30 13:23:35 +0100 | [diff] [blame] | 455 | for (mp_uint_t i = 1; i < n_args; i++) { |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 456 | set_update_int(args[0], args[i]); |
| 457 | } |
| 458 | |
| 459 | return mp_const_none; |
| 460 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 461 | STATIC MP_DEFINE_CONST_FUN_OBJ_VAR(set_update_obj, 1, set_update); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 462 | |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 463 | STATIC mp_obj_t set_union(mp_obj_t self_in, mp_obj_t other_in) { |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 464 | check_set_or_frozenset(self_in); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 465 | mp_obj_set_t *self = set_copy(self_in); |
| 466 | set_update_int(self, other_in); |
| 467 | return self; |
| 468 | } |
Paul Sokolovsky | d5df6cd | 2014-02-12 18:15:40 +0200 | [diff] [blame] | 469 | STATIC MP_DEFINE_CONST_FUN_OBJ_2(set_union_obj, set_union); |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 470 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 471 | STATIC mp_obj_t set_unary_op(mp_uint_t op, mp_obj_t self_in) { |
Damien George | 95004e5 | 2014-04-05 17:17:19 +0100 | [diff] [blame] | 472 | mp_obj_set_t *self = self_in; |
| 473 | switch (op) { |
Paul Sokolovsky | 1b586f3 | 2015-10-11 12:09:43 +0300 | [diff] [blame^] | 474 | case MP_UNARY_OP_BOOL: return mp_obj_new_bool(self->set.used != 0); |
Damien George | bb4c6f3 | 2014-07-31 10:49:14 +0100 | [diff] [blame] | 475 | case MP_UNARY_OP_LEN: return MP_OBJ_NEW_SMALL_INT(self->set.used); |
Paul Sokolovsky | 8b3b2d0 | 2015-08-28 22:31:31 +0300 | [diff] [blame] | 476 | #if MICROPY_PY_BUILTINS_FROZENSET |
| 477 | case MP_UNARY_OP_HASH: |
| 478 | if (MP_OBJ_IS_TYPE(self_in, &mp_type_frozenset)) { |
| 479 | // start hash with unique value |
| 480 | mp_int_t hash = (mp_int_t)&mp_type_frozenset; |
| 481 | mp_uint_t max = self->set.alloc; |
| 482 | mp_set_t *set = &self->set; |
| 483 | |
| 484 | for (mp_uint_t i = 0; i < max; i++) { |
| 485 | if (MP_SET_SLOT_IS_FILLED(set, i)) { |
| 486 | hash += MP_OBJ_SMALL_INT_VALUE(mp_unary_op(MP_UNARY_OP_HASH, set->table[i])); |
| 487 | } |
| 488 | } |
| 489 | return MP_OBJ_NEW_SMALL_INT(hash); |
| 490 | } |
| 491 | #endif |
Damien George | 6ac5dce | 2014-05-21 19:42:43 +0100 | [diff] [blame] | 492 | default: return MP_OBJ_NULL; // op not supported |
Damien George | 95004e5 | 2014-04-05 17:17:19 +0100 | [diff] [blame] | 493 | } |
| 494 | } |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 495 | |
Damien George | ecc88e9 | 2014-08-30 00:35:11 +0100 | [diff] [blame] | 496 | STATIC mp_obj_t set_binary_op(mp_uint_t op, mp_obj_t lhs, mp_obj_t rhs) { |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 497 | mp_obj_t args[] = {lhs, rhs}; |
| 498 | switch (op) { |
Damien George | d0a5bf3 | 2014-05-10 13:55:11 +0100 | [diff] [blame] | 499 | case MP_BINARY_OP_OR: |
| 500 | return set_union(lhs, rhs); |
| 501 | case MP_BINARY_OP_XOR: |
| 502 | return set_symmetric_difference(lhs, rhs); |
| 503 | case MP_BINARY_OP_AND: |
| 504 | return set_intersect(lhs, rhs); |
| 505 | case MP_BINARY_OP_SUBTRACT: |
| 506 | return set_diff(2, args); |
| 507 | case MP_BINARY_OP_INPLACE_OR: |
| 508 | return set_union(lhs, rhs); |
| 509 | case MP_BINARY_OP_INPLACE_XOR: |
| 510 | return set_symmetric_difference(lhs, rhs); |
| 511 | case MP_BINARY_OP_INPLACE_AND: |
| 512 | return set_intersect(lhs, rhs); |
| 513 | case MP_BINARY_OP_INPLACE_SUBTRACT: |
| 514 | return set_diff(2, args); |
| 515 | case MP_BINARY_OP_LESS: |
| 516 | return set_issubset_proper(lhs, rhs); |
| 517 | case MP_BINARY_OP_MORE: |
| 518 | return set_issuperset_proper(lhs, rhs); |
| 519 | case MP_BINARY_OP_EQUAL: |
| 520 | return set_equal(lhs, rhs); |
| 521 | case MP_BINARY_OP_LESS_EQUAL: |
| 522 | return set_issubset(lhs, rhs); |
| 523 | case MP_BINARY_OP_MORE_EQUAL: |
| 524 | return set_issuperset(lhs, rhs); |
| 525 | case MP_BINARY_OP_IN: { |
| 526 | mp_obj_set_t *o = lhs; |
| 527 | mp_obj_t elem = mp_set_lookup(&o->set, rhs, MP_MAP_LOOKUP); |
Paul Sokolovsky | 1b586f3 | 2015-10-11 12:09:43 +0300 | [diff] [blame^] | 528 | return mp_obj_new_bool(elem != NULL); |
Damien George | d0a5bf3 | 2014-05-10 13:55:11 +0100 | [diff] [blame] | 529 | } |
| 530 | default: |
Damien George | 6ac5dce | 2014-05-21 19:42:43 +0100 | [diff] [blame] | 531 | return MP_OBJ_NULL; // op not supported |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 532 | } |
| 533 | } |
John R. Lenton | 0de386b | 2014-01-12 19:39:48 +0000 | [diff] [blame] | 534 | |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 535 | /******************************************************************************/ |
| 536 | /* set constructors & public C API */ |
| 537 | |
| 538 | |
Damien George | 9b196cd | 2014-03-26 21:47:19 +0000 | [diff] [blame] | 539 | STATIC const mp_map_elem_t set_locals_dict_table[] = { |
| 540 | { MP_OBJ_NEW_QSTR(MP_QSTR_add), (mp_obj_t)&set_add_obj }, |
| 541 | { MP_OBJ_NEW_QSTR(MP_QSTR_clear), (mp_obj_t)&set_clear_obj }, |
| 542 | { MP_OBJ_NEW_QSTR(MP_QSTR_copy), (mp_obj_t)&set_copy_obj }, |
| 543 | { MP_OBJ_NEW_QSTR(MP_QSTR_discard), (mp_obj_t)&set_discard_obj }, |
| 544 | { MP_OBJ_NEW_QSTR(MP_QSTR_difference), (mp_obj_t)&set_diff_obj }, |
| 545 | { MP_OBJ_NEW_QSTR(MP_QSTR_difference_update), (mp_obj_t)&set_diff_update_obj }, |
| 546 | { MP_OBJ_NEW_QSTR(MP_QSTR_intersection), (mp_obj_t)&set_intersect_obj }, |
| 547 | { MP_OBJ_NEW_QSTR(MP_QSTR_intersection_update), (mp_obj_t)&set_intersect_update_obj }, |
| 548 | { MP_OBJ_NEW_QSTR(MP_QSTR_isdisjoint), (mp_obj_t)&set_isdisjoint_obj }, |
| 549 | { MP_OBJ_NEW_QSTR(MP_QSTR_issubset), (mp_obj_t)&set_issubset_obj }, |
| 550 | { MP_OBJ_NEW_QSTR(MP_QSTR_issuperset), (mp_obj_t)&set_issuperset_obj }, |
| 551 | { MP_OBJ_NEW_QSTR(MP_QSTR_pop), (mp_obj_t)&set_pop_obj }, |
| 552 | { MP_OBJ_NEW_QSTR(MP_QSTR_remove), (mp_obj_t)&set_remove_obj }, |
| 553 | { MP_OBJ_NEW_QSTR(MP_QSTR_symmetric_difference), (mp_obj_t)&set_symmetric_difference_obj }, |
| 554 | { MP_OBJ_NEW_QSTR(MP_QSTR_symmetric_difference_update), (mp_obj_t)&set_symmetric_difference_update_obj }, |
| 555 | { MP_OBJ_NEW_QSTR(MP_QSTR_union), (mp_obj_t)&set_union_obj }, |
| 556 | { MP_OBJ_NEW_QSTR(MP_QSTR_update), (mp_obj_t)&set_update_obj }, |
Paul Sokolovsky | 68e7c51 | 2014-04-13 11:53:15 +0300 | [diff] [blame] | 557 | { MP_OBJ_NEW_QSTR(MP_QSTR___contains__), (mp_obj_t)&mp_op_contains_obj }, |
John R. Lenton | 19b14d3 | 2014-01-12 15:29:11 +0000 | [diff] [blame] | 558 | }; |
| 559 | |
Damien George | 9b196cd | 2014-03-26 21:47:19 +0000 | [diff] [blame] | 560 | STATIC MP_DEFINE_CONST_DICT(set_locals_dict, set_locals_dict_table); |
| 561 | |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 562 | const mp_obj_type_t mp_type_set = { |
Damien George | c596612 | 2014-02-15 16:10:44 +0000 | [diff] [blame] | 563 | { &mp_type_type }, |
Damien George | a71c83a | 2014-02-15 11:34:50 +0000 | [diff] [blame] | 564 | .name = MP_QSTR_set, |
Damien George | 97209d3 | 2014-01-07 15:58:30 +0000 | [diff] [blame] | 565 | .print = set_print, |
| 566 | .make_new = set_make_new, |
Damien George | 95004e5 | 2014-04-05 17:17:19 +0100 | [diff] [blame] | 567 | .unary_op = set_unary_op, |
John R. Lenton | be790f9 | 2014-01-12 23:09:10 +0000 | [diff] [blame] | 568 | .binary_op = set_binary_op, |
John R. Lenton | 0ce03b4 | 2014-01-12 15:17:42 +0000 | [diff] [blame] | 569 | .getiter = set_getiter, |
Damien George | 9b196cd | 2014-03-26 21:47:19 +0000 | [diff] [blame] | 570 | .locals_dict = (mp_obj_t)&set_locals_dict, |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 571 | }; |
| 572 | |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 573 | #if MICROPY_PY_BUILTINS_FROZENSET |
Paul Sokolovsky | b181b58 | 2014-05-10 16:02:17 +0300 | [diff] [blame] | 574 | const mp_obj_type_t mp_type_frozenset = { |
| 575 | { &mp_type_type }, |
| 576 | .name = MP_QSTR_frozenset, |
| 577 | .print = set_print, |
| 578 | .make_new = set_make_new, |
| 579 | .unary_op = set_unary_op, |
| 580 | .binary_op = set_binary_op, |
| 581 | .getiter = set_getiter, |
| 582 | .locals_dict = (mp_obj_t)&set_locals_dict, |
| 583 | }; |
| 584 | #endif |
| 585 | |
Damien George | 93965e7 | 2014-08-30 13:23:35 +0100 | [diff] [blame] | 586 | mp_obj_t mp_obj_new_set(mp_uint_t n_args, mp_obj_t *items) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 587 | mp_obj_set_t *o = m_new_obj(mp_obj_set_t); |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 588 | o->base.type = &mp_type_set; |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 589 | mp_set_init(&o->set, n_args); |
Damien George | 93965e7 | 2014-08-30 13:23:35 +0100 | [diff] [blame] | 590 | for (mp_uint_t i = 0; i < n_args; i++) { |
John R. Lenton | 2a24172 | 2014-01-12 16:39:39 +0000 | [diff] [blame] | 591 | mp_set_lookup(&o->set, items[i], MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 592 | } |
| 593 | return o; |
| 594 | } |
| 595 | |
| 596 | void mp_obj_set_store(mp_obj_t self_in, mp_obj_t item) { |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 597 | assert(MP_OBJ_IS_TYPE(self_in, &mp_type_set)); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 598 | mp_obj_set_t *self = self_in; |
John R. Lenton | 2a24172 | 2014-01-12 16:39:39 +0000 | [diff] [blame] | 599 | mp_set_lookup(&self->set, item, MP_MAP_LOOKUP_ADD_IF_NOT_FOUND); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 600 | } |
Damien George | 3ebd4d0 | 2014-06-01 13:46:47 +0100 | [diff] [blame] | 601 | |
| 602 | #endif // MICROPY_PY_BUILTINS_SET |