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 | |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 27 | #include <stdio.h> |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 28 | #include <assert.h> |
| 29 | |
Paul Sokolovsky | f54bcbf | 2014-05-02 17:47:01 +0300 | [diff] [blame] | 30 | #include "mpconfig.h" |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 31 | #include "nlr.h" |
| 32 | #include "misc.h" |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 33 | #include "qstr.h" |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 34 | #include "obj.h" |
Damien George | 897fe0c | 2014-04-15 22:03:55 +0100 | [diff] [blame] | 35 | #include "objstr.h" |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 36 | #include "runtime0.h" |
| 37 | #include "runtime.h" |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 38 | #include "builtin.h" |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 39 | #include "stream.h" |
| 40 | #include "pfenv.h" |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 41 | |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 42 | #if MICROPY_PY_BUILTINS_FLOAT |
Damien George | 0c36da0 | 2014-03-08 15:24:39 +0000 | [diff] [blame] | 43 | #include <math.h> |
| 44 | #endif |
| 45 | |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 46 | // args[0] is function from class body |
| 47 | // args[1] is class name |
| 48 | // args[2:] are base objects |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 49 | STATIC mp_obj_t mp_builtin___build_class__(uint n_args, const mp_obj_t *args) { |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 50 | assert(2 <= n_args); |
| 51 | |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 52 | // set the new classes __locals__ object |
| 53 | mp_obj_dict_t *old_locals = mp_locals_get(); |
Damien George | 062478e | 2014-01-09 20:57:50 +0000 | [diff] [blame] | 54 | mp_obj_t class_locals = mp_obj_new_dict(0); |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 55 | mp_locals_set(class_locals); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 56 | |
| 57 | // call the class code |
Damien George | 882b363 | 2014-04-02 15:56:31 +0100 | [diff] [blame] | 58 | mp_obj_t cell = mp_call_function_0(args[0]); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 59 | |
| 60 | // restore old __locals__ object |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 61 | mp_locals_set(old_locals); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 62 | |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 63 | // get the class type (meta object) from the base objects |
| 64 | mp_obj_t meta; |
| 65 | if (n_args == 2) { |
| 66 | // no explicit bases, so use 'type' |
Damien George | c596612 | 2014-02-15 16:10:44 +0000 | [diff] [blame] | 67 | meta = (mp_obj_t)&mp_type_type; |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 68 | } else { |
| 69 | // use type of first base object |
| 70 | meta = mp_obj_get_type(args[2]); |
| 71 | } |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 72 | |
| 73 | // TODO do proper metaclass resolution for multiple base objects |
| 74 | |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 75 | // create the new class using a call to the meta object |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 76 | mp_obj_t meta_args[3]; |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 77 | meta_args[0] = args[1]; // class name |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 78 | meta_args[1] = mp_obj_new_tuple(n_args - 2, args + 2); // tuple of bases |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 79 | meta_args[2] = class_locals; // dict of members |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 80 | mp_obj_t new_class = mp_call_function_n_kw(meta, 3, 0, meta_args); |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 81 | |
| 82 | // store into cell if neede |
| 83 | if (cell != mp_const_none) { |
| 84 | mp_obj_cell_set(cell, new_class); |
| 85 | } |
| 86 | |
| 87 | return new_class; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 88 | } |
| 89 | |
Damien George | b97669a | 2014-01-08 11:47:55 +0000 | [diff] [blame] | 90 | MP_DEFINE_CONST_FUN_OBJ_VAR(mp_builtin___build_class___obj, 2, mp_builtin___build_class__); |
| 91 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 92 | STATIC mp_obj_t mp_builtin___repl_print__(mp_obj_t o) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 93 | if (o != mp_const_none) { |
Paul Sokolovsky | 76d982e | 2014-01-13 19:19:16 +0200 | [diff] [blame] | 94 | mp_obj_print(o, PRINT_REPR); |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 95 | printf("\n"); |
| 96 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 97 | return mp_const_none; |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 98 | } |
| 99 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 100 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin___repl_print___obj, mp_builtin___repl_print__); |
| 101 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 102 | mp_obj_t mp_builtin_abs(mp_obj_t o_in) { |
| 103 | if (MP_OBJ_IS_SMALL_INT(o_in)) { |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 104 | mp_int_t val = MP_OBJ_SMALL_INT_VALUE(o_in); |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 105 | if (val < 0) { |
| 106 | val = -val; |
| 107 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 108 | return MP_OBJ_NEW_SMALL_INT(val); |
Damien George | fb510b3 | 2014-06-01 13:32:54 +0100 | [diff] [blame] | 109 | #if MICROPY_PY_BUILTINS_FLOAT |
Damien George | 0c36da0 | 2014-03-08 15:24:39 +0000 | [diff] [blame] | 110 | } else if (MP_OBJ_IS_TYPE(o_in, &mp_type_float)) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 111 | mp_float_t value = mp_obj_float_get(o_in); |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 112 | // TODO check for NaN etc |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 113 | if (value < 0) { |
| 114 | return mp_obj_new_float(-value); |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 115 | } else { |
| 116 | return o_in; |
| 117 | } |
Paul Sokolovsky | 3b6f7b9 | 2014-06-20 01:48:35 +0300 | [diff] [blame] | 118 | #if MICROPY_PY_BUILTINS_COMPLEX |
Damien George | 0c36da0 | 2014-03-08 15:24:39 +0000 | [diff] [blame] | 119 | } else if (MP_OBJ_IS_TYPE(o_in, &mp_type_complex)) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 120 | mp_float_t real, imag; |
| 121 | mp_obj_complex_get(o_in, &real, &imag); |
Damien George | 0c36da0 | 2014-03-08 15:24:39 +0000 | [diff] [blame] | 122 | return mp_obj_new_float(MICROPY_FLOAT_C_FUN(sqrt)(real*real + imag*imag)); |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 123 | #endif |
Paul Sokolovsky | 3b6f7b9 | 2014-06-20 01:48:35 +0300 | [diff] [blame] | 124 | #endif |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 125 | } else { |
| 126 | assert(0); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 127 | return mp_const_none; |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 128 | } |
| 129 | } |
| 130 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 131 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_abs_obj, mp_builtin_abs); |
| 132 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 133 | STATIC mp_obj_t mp_builtin_all(mp_obj_t o_in) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 134 | mp_obj_t iterable = mp_getiter(o_in); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 135 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 136 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 137 | if (!mp_obj_is_true(item)) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 138 | return mp_const_false; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 139 | } |
| 140 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 141 | return mp_const_true; |
Damien | 660365e | 2013-12-17 18:27:24 +0000 | [diff] [blame] | 142 | } |
| 143 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 144 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_all_obj, mp_builtin_all); |
| 145 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 146 | STATIC mp_obj_t mp_builtin_any(mp_obj_t o_in) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 147 | mp_obj_t iterable = mp_getiter(o_in); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 148 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 149 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 150 | if (mp_obj_is_true(item)) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 151 | return mp_const_true; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 152 | } |
| 153 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 154 | return mp_const_false; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 155 | } |
| 156 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 157 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_any_obj, mp_builtin_any); |
| 158 | |
Damien George | 897fe0c | 2014-04-15 22:03:55 +0100 | [diff] [blame] | 159 | STATIC mp_obj_t mp_builtin_bin(mp_obj_t o_in) { |
| 160 | mp_obj_t args[] = { MP_OBJ_NEW_QSTR(MP_QSTR__brace_open__colon__hash_b_brace_close_), o_in }; |
Emmanuel Blot | f6932d6 | 2014-06-19 18:54:34 +0200 | [diff] [blame] | 161 | return mp_obj_str_format(MP_ARRAY_SIZE(args), args); |
Damien George | 897fe0c | 2014-04-15 22:03:55 +0100 | [diff] [blame] | 162 | } |
| 163 | |
| 164 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_bin_obj, mp_builtin_bin); |
| 165 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 166 | STATIC mp_obj_t mp_builtin_callable(mp_obj_t o_in) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 167 | if (mp_obj_is_callable(o_in)) { |
| 168 | return mp_const_true; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 169 | } else { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 170 | return mp_const_false; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 171 | } |
| 172 | } |
| 173 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 174 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_callable_obj, mp_builtin_callable); |
| 175 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 176 | STATIC mp_obj_t mp_builtin_chr(mp_obj_t o_in) { |
Paul Sokolovsky | 42a5251 | 2014-06-13 02:39:37 +0300 | [diff] [blame] | 177 | #if MICROPY_PY_BUILTINS_STR_UNICODE |
Damien George | c9aa58e | 2014-07-31 13:41:43 +0000 | [diff] [blame] | 178 | mp_uint_t c = mp_obj_get_int(o_in); |
Chris Angelico | 9a1a4be | 2014-06-04 05:28:12 +1000 | [diff] [blame] | 179 | char str[4]; |
| 180 | int len = 0; |
| 181 | if (c < 0x80) { |
| 182 | *str = c; len = 1; |
| 183 | } else if (c < 0x800) { |
| 184 | str[0] = (c >> 6) | 0xC0; |
| 185 | str[1] = (c & 0x3F) | 0x80; |
| 186 | len = 2; |
| 187 | } else if (c < 0x10000) { |
| 188 | str[0] = (c >> 12) | 0xE0; |
| 189 | str[1] = ((c >> 6) & 0x3F) | 0x80; |
| 190 | str[2] = (c & 0x3F) | 0x80; |
| 191 | len = 3; |
| 192 | } else if (c < 0x110000) { |
| 193 | str[0] = (c >> 18) | 0xF0; |
| 194 | str[1] = ((c >> 12) & 0x3F) | 0x80; |
| 195 | str[2] = ((c >> 6) & 0x3F) | 0x80; |
| 196 | str[3] = (c & 0x3F) | 0x80; |
| 197 | len = 4; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 198 | } else { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 199 | nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "chr() arg not in range(0x110000)")); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 200 | } |
Chris Angelico | 9a1a4be | 2014-06-04 05:28:12 +1000 | [diff] [blame] | 201 | return mp_obj_new_str(str, len, true); |
Paul Sokolovsky | 42a5251 | 2014-06-13 02:39:37 +0300 | [diff] [blame] | 202 | #else |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 203 | mp_int_t ord = mp_obj_get_int(o_in); |
Paul Sokolovsky | 42a5251 | 2014-06-13 02:39:37 +0300 | [diff] [blame] | 204 | if (0 <= ord && ord <= 0x10ffff) { |
| 205 | char str[1] = {ord}; |
| 206 | return mp_obj_new_str(str, 1, true); |
| 207 | } else { |
| 208 | nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "chr() arg not in range(0x110000)")); |
| 209 | } |
| 210 | #endif |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 211 | } |
| 212 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 213 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_chr_obj, mp_builtin_chr); |
| 214 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 215 | STATIC mp_obj_t mp_builtin_dir(uint n_args, const mp_obj_t *args) { |
Damien George | 4acb245 | 2014-02-02 22:07:44 +0000 | [diff] [blame] | 216 | // TODO make this function more general and less of a hack |
| 217 | |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 218 | mp_obj_dict_t *dict = NULL; |
Damien George | 4acb245 | 2014-02-02 22:07:44 +0000 | [diff] [blame] | 219 | if (n_args == 0) { |
| 220 | // make a list of names in the local name space |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 221 | dict = mp_locals_get(); |
Damien George | 4acb245 | 2014-02-02 22:07:44 +0000 | [diff] [blame] | 222 | } else { // n_args == 1 |
| 223 | // make a list of names in the given object |
Damien George | 6022d9d | 2014-03-26 22:35:00 +0000 | [diff] [blame] | 224 | if (MP_OBJ_IS_TYPE(args[0], &mp_type_module)) { |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 225 | dict = mp_obj_module_get_globals(args[0]); |
Damien George | 6022d9d | 2014-03-26 22:35:00 +0000 | [diff] [blame] | 226 | } else { |
| 227 | mp_obj_type_t *type; |
| 228 | if (MP_OBJ_IS_TYPE(args[0], &mp_type_type)) { |
| 229 | type = args[0]; |
| 230 | } else { |
| 231 | type = mp_obj_get_type(args[0]); |
| 232 | } |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 233 | if (type->locals_dict != MP_OBJ_NULL && MP_OBJ_IS_TYPE(type->locals_dict, &mp_type_dict)) { |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 234 | dict = type->locals_dict; |
Damien George | 6022d9d | 2014-03-26 22:35:00 +0000 | [diff] [blame] | 235 | } |
Damien George | badc9d4 | 2014-03-23 00:03:11 +0000 | [diff] [blame] | 236 | } |
Damien George | 4acb245 | 2014-02-02 22:07:44 +0000 | [diff] [blame] | 237 | } |
| 238 | |
| 239 | mp_obj_t dir = mp_obj_new_list(0, NULL); |
Damien George | 7efc5b3 | 2014-04-05 22:36:42 +0100 | [diff] [blame] | 240 | if (dict != NULL) { |
| 241 | for (uint i = 0; i < dict->map.alloc; i++) { |
| 242 | if (MP_MAP_SLOT_IS_FILLED(&dict->map, i)) { |
| 243 | mp_obj_list_append(dir, dict->map.table[i].key); |
Damien George | badc9d4 | 2014-03-23 00:03:11 +0000 | [diff] [blame] | 244 | } |
| 245 | } |
| 246 | } |
Damien George | 9b196cd | 2014-03-26 21:47:19 +0000 | [diff] [blame] | 247 | |
Damien George | 4acb245 | 2014-02-02 22:07:44 +0000 | [diff] [blame] | 248 | return dir; |
| 249 | } |
| 250 | |
| 251 | MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_dir_obj, 0, 1, mp_builtin_dir); |
| 252 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 253 | STATIC mp_obj_t mp_builtin_divmod(mp_obj_t o1_in, mp_obj_t o2_in) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 254 | if (MP_OBJ_IS_SMALL_INT(o1_in) && MP_OBJ_IS_SMALL_INT(o2_in)) { |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 255 | mp_int_t i1 = MP_OBJ_SMALL_INT_VALUE(o1_in); |
| 256 | mp_int_t i2 = MP_OBJ_SMALL_INT_VALUE(o2_in); |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 257 | mp_obj_t args[2]; |
| 258 | args[0] = MP_OBJ_NEW_SMALL_INT(i1 / i2); |
| 259 | args[1] = MP_OBJ_NEW_SMALL_INT(i1 % i2); |
Damien George | 15d1806 | 2014-03-31 16:28:13 +0100 | [diff] [blame] | 260 | return mp_obj_new_tuple(2, args); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 261 | } else { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 262 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "unsupported operand type(s) for divmod(): '%s' and '%s'", mp_obj_get_type_str(o1_in), mp_obj_get_type_str(o2_in))); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 263 | } |
| 264 | } |
| 265 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 266 | MP_DEFINE_CONST_FUN_OBJ_2(mp_builtin_divmod_obj, mp_builtin_divmod); |
| 267 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 268 | STATIC mp_obj_t mp_builtin_hash(mp_obj_t o_in) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 269 | // TODO hash will generally overflow small integer; can we safely truncate it? |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 270 | return mp_obj_new_int(mp_obj_hash(o_in)); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 271 | } |
| 272 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 273 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_hash_obj, mp_builtin_hash); |
| 274 | |
Damien George | 5805111 | 2014-04-15 12:42:52 +0100 | [diff] [blame] | 275 | STATIC mp_obj_t mp_builtin_hex(mp_obj_t o_in) { |
Damien George | b013aea | 2014-04-15 12:50:21 +0100 | [diff] [blame] | 276 | return mp_binary_op(MP_BINARY_OP_MODULO, MP_OBJ_NEW_QSTR(MP_QSTR__percent__hash_x), o_in); |
Damien George | 5805111 | 2014-04-15 12:42:52 +0100 | [diff] [blame] | 277 | } |
| 278 | |
| 279 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_hex_obj, mp_builtin_hex); |
| 280 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 281 | STATIC mp_obj_t mp_builtin_iter(mp_obj_t o_in) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 282 | return mp_getiter(o_in); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 283 | } |
| 284 | |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 285 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_iter_obj, mp_builtin_iter); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 286 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 287 | STATIC mp_obj_t mp_builtin_len(mp_obj_t o_in) { |
John R. Lenton | 4bee76e | 2014-01-10 11:25:03 +0000 | [diff] [blame] | 288 | mp_obj_t len = mp_obj_len_maybe(o_in); |
| 289 | if (len == NULL) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 290 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "object of type '%s' has no len()", mp_obj_get_type_str(o_in))); |
John R. Lenton | 4bee76e | 2014-01-10 11:25:03 +0000 | [diff] [blame] | 291 | } else { |
| 292 | return len; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 293 | } |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 294 | } |
| 295 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 296 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_len_obj, mp_builtin_len); |
| 297 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 298 | STATIC mp_obj_t mp_builtin_max(uint n_args, const mp_obj_t *args) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 299 | if (n_args == 1) { |
| 300 | // given an iterable |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 301 | mp_obj_t iterable = mp_getiter(args[0]); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 302 | mp_obj_t max_obj = NULL; |
| 303 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 304 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Andrew Scheller | 3706766 | 2014-05-01 21:21:43 +0100 | [diff] [blame] | 305 | if (max_obj == NULL || (mp_binary_op(MP_BINARY_OP_LESS, max_obj, item) == mp_const_true)) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 306 | max_obj = item; |
| 307 | } |
| 308 | } |
| 309 | if (max_obj == NULL) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 310 | nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "max() arg is an empty sequence")); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 311 | } |
| 312 | return max_obj; |
| 313 | } else { |
| 314 | // given many args |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 315 | mp_obj_t max_obj = args[0]; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 316 | for (int i = 1; i < n_args; i++) { |
Andrew Scheller | 3706766 | 2014-05-01 21:21:43 +0100 | [diff] [blame] | 317 | if (mp_binary_op(MP_BINARY_OP_LESS, max_obj, args[i]) == mp_const_true) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 318 | max_obj = args[i]; |
| 319 | } |
| 320 | } |
| 321 | return max_obj; |
| 322 | } |
| 323 | } |
| 324 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 325 | MP_DEFINE_CONST_FUN_OBJ_VAR(mp_builtin_max_obj, 1, mp_builtin_max); |
| 326 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 327 | STATIC mp_obj_t mp_builtin_min(uint n_args, const mp_obj_t *args) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 328 | if (n_args == 1) { |
| 329 | // given an iterable |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 330 | mp_obj_t iterable = mp_getiter(args[0]); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 331 | mp_obj_t min_obj = NULL; |
| 332 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 333 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Andrew Scheller | 3706766 | 2014-05-01 21:21:43 +0100 | [diff] [blame] | 334 | if (min_obj == NULL || (mp_binary_op(MP_BINARY_OP_LESS, item, min_obj) == mp_const_true)) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 335 | min_obj = item; |
| 336 | } |
| 337 | } |
| 338 | if (min_obj == NULL) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 339 | nlr_raise(mp_obj_new_exception_msg(&mp_type_ValueError, "min() arg is an empty sequence")); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 340 | } |
| 341 | return min_obj; |
| 342 | } else { |
| 343 | // given many args |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 344 | mp_obj_t min_obj = args[0]; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 345 | for (int i = 1; i < n_args; i++) { |
Andrew Scheller | 3706766 | 2014-05-01 21:21:43 +0100 | [diff] [blame] | 346 | if (mp_binary_op(MP_BINARY_OP_LESS, args[i], min_obj) == mp_const_true) { |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 347 | min_obj = args[i]; |
| 348 | } |
| 349 | } |
| 350 | return min_obj; |
| 351 | } |
| 352 | } |
| 353 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 354 | MP_DEFINE_CONST_FUN_OBJ_VAR(mp_builtin_min_obj, 1, mp_builtin_min); |
| 355 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 356 | STATIC mp_obj_t mp_builtin_next(mp_obj_t o) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 357 | mp_obj_t ret = mp_iternext_allow_raise(o); |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 358 | if (ret == MP_OBJ_STOP_ITERATION) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 359 | nlr_raise(mp_obj_new_exception(&mp_type_StopIteration)); |
Damien | d9d6201 | 2013-12-21 18:38:03 +0000 | [diff] [blame] | 360 | } else { |
| 361 | return ret; |
| 362 | } |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 363 | } |
| 364 | |
| 365 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_next_obj, mp_builtin_next); |
| 366 | |
Damien George | 897fe0c | 2014-04-15 22:03:55 +0100 | [diff] [blame] | 367 | STATIC mp_obj_t mp_builtin_oct(mp_obj_t o_in) { |
| 368 | return mp_binary_op(MP_BINARY_OP_MODULO, MP_OBJ_NEW_QSTR(MP_QSTR__percent__hash_o), o_in); |
| 369 | } |
| 370 | |
| 371 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_oct_obj, mp_builtin_oct); |
| 372 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 373 | STATIC mp_obj_t mp_builtin_ord(mp_obj_t o_in) { |
Damien George | 55baff4 | 2014-01-21 21:40:13 +0000 | [diff] [blame] | 374 | uint len; |
Damien George | 698ec21 | 2014-02-08 18:17:23 +0000 | [diff] [blame] | 375 | const char *str = mp_obj_str_get_data(o_in, &len); |
Paul Sokolovsky | 42a5251 | 2014-06-13 02:39:37 +0300 | [diff] [blame] | 376 | #if MICROPY_PY_BUILTINS_STR_UNICODE |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 377 | mp_uint_t charlen = unichar_charlen(str, len); |
Chris Angelico | 9a1a4be | 2014-06-04 05:28:12 +1000 | [diff] [blame] | 378 | if (charlen == 1) { |
| 379 | if (MP_OBJ_IS_STR(o_in) && UTF8_IS_NONASCII(*str)) { |
Damien George | 40f3c02 | 2014-07-03 13:25:24 +0100 | [diff] [blame] | 380 | mp_int_t ord = *str++ & 0x7F; |
| 381 | for (mp_int_t mask = 0x40; ord & mask; mask >>= 1) { |
Chris Angelico | 9a1a4be | 2014-06-04 05:28:12 +1000 | [diff] [blame] | 382 | ord &= ~mask; |
| 383 | } |
| 384 | while (UTF8_IS_CONT(*str)) { |
| 385 | ord = (ord << 6) | (*str++ & 0x3F); |
| 386 | } |
| 387 | return mp_obj_new_int(ord); |
| 388 | } else { |
| 389 | return mp_obj_new_int(((const byte*)str)[0]); |
| 390 | } |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 391 | } else { |
Chris Angelico | 9a1a4be | 2014-06-04 05:28:12 +1000 | [diff] [blame] | 392 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "ord() expected a character, but string of length %d found", charlen)); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 393 | } |
Paul Sokolovsky | 42a5251 | 2014-06-13 02:39:37 +0300 | [diff] [blame] | 394 | #else |
| 395 | if (len == 1) { |
| 396 | // don't sign extend when converting to ord |
| 397 | return mp_obj_new_int(((const byte*)str)[0]); |
| 398 | } else { |
| 399 | nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError, "ord() expected a character, but string of length %d found", len)); |
| 400 | } |
| 401 | #endif |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 402 | } |
| 403 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 404 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_ord_obj, mp_builtin_ord); |
| 405 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 406 | STATIC mp_obj_t mp_builtin_pow(uint n_args, const mp_obj_t *args) { |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 407 | assert(2 <= n_args && n_args <= 3); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 408 | switch (n_args) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 409 | case 2: return mp_binary_op(MP_BINARY_OP_POWER, args[0], args[1]); |
| 410 | default: return mp_binary_op(MP_BINARY_OP_MODULO, mp_binary_op(MP_BINARY_OP_POWER, args[0], args[1]), args[2]); // TODO optimise... |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 411 | } |
| 412 | } |
| 413 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 414 | MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_pow_obj, 2, 3, mp_builtin_pow); |
| 415 | |
Damien George | 4881566 | 2014-04-02 10:34:44 +0100 | [diff] [blame] | 416 | STATIC mp_obj_t mp_builtin_print(uint n_args, const mp_obj_t *args, mp_map_t *kwargs) { |
| 417 | mp_map_elem_t *sep_elem = mp_map_lookup(kwargs, MP_OBJ_NEW_QSTR(MP_QSTR_sep), MP_MAP_LOOKUP); |
| 418 | mp_map_elem_t *end_elem = mp_map_lookup(kwargs, MP_OBJ_NEW_QSTR(MP_QSTR_end), MP_MAP_LOOKUP); |
| 419 | const char *sep_data = " "; |
| 420 | uint sep_len = 1; |
| 421 | const char *end_data = "\n"; |
| 422 | uint end_len = 1; |
| 423 | if (sep_elem != NULL && sep_elem->value != mp_const_none) { |
| 424 | sep_data = mp_obj_str_get_data(sep_elem->value, &sep_len); |
| 425 | } |
| 426 | if (end_elem != NULL && end_elem->value != mp_const_none) { |
| 427 | end_data = mp_obj_str_get_data(end_elem->value, &end_len); |
| 428 | } |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 429 | #if MICROPY_PY_IO |
| 430 | mp_obj_t stream_obj = &mp_sys_stdout_obj; |
| 431 | mp_map_elem_t *file_elem = mp_map_lookup(kwargs, MP_OBJ_NEW_QSTR(MP_QSTR_file), MP_MAP_LOOKUP); |
| 432 | if (file_elem != NULL && file_elem->value != mp_const_none) { |
| 433 | stream_obj = file_elem->value; |
| 434 | } |
| 435 | |
| 436 | pfenv_t pfenv; |
| 437 | pfenv.data = stream_obj; |
| 438 | pfenv.print_strn = (void (*)(void *, const char *, unsigned int))mp_stream_write; |
| 439 | #endif |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 440 | for (int i = 0; i < n_args; i++) { |
| 441 | if (i > 0) { |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 442 | #if MICROPY_PY_IO |
| 443 | mp_stream_write(stream_obj, sep_data, sep_len); |
| 444 | #else |
Damien George | 4881566 | 2014-04-02 10:34:44 +0100 | [diff] [blame] | 445 | printf("%.*s", sep_len, sep_data); |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 446 | #endif |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 447 | } |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 448 | #if MICROPY_PY_IO |
| 449 | mp_obj_print_helper((void (*)(void *env, const char *fmt, ...))pfenv_printf, &pfenv, args[i], PRINT_STR); |
| 450 | #else |
Paul Sokolovsky | 76d982e | 2014-01-13 19:19:16 +0200 | [diff] [blame] | 451 | mp_obj_print(args[i], PRINT_STR); |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 452 | #endif |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 453 | } |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 454 | #if MICROPY_PY_IO |
| 455 | mp_stream_write(stream_obj, end_data, end_len); |
| 456 | #else |
Damien George | 4881566 | 2014-04-02 10:34:44 +0100 | [diff] [blame] | 457 | printf("%.*s", end_len, end_data); |
Paul Sokolovsky | cb66f41 | 2014-07-13 23:07:42 +0300 | [diff] [blame] | 458 | #endif |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 459 | return mp_const_none; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 460 | } |
| 461 | |
Damien George | 4881566 | 2014-04-02 10:34:44 +0100 | [diff] [blame] | 462 | MP_DEFINE_CONST_FUN_OBJ_KW(mp_builtin_print_obj, 0, mp_builtin_print); |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 463 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 464 | STATIC mp_obj_t mp_builtin_repr(mp_obj_t o_in) { |
Damien George | e2fb2ba | 2014-01-15 21:40:48 +0000 | [diff] [blame] | 465 | vstr_t *vstr = vstr_new(); |
Damien George | 4899ff9 | 2014-01-15 22:39:03 +0000 | [diff] [blame] | 466 | mp_obj_print_helper((void (*)(void *env, const char *fmt, ...))vstr_printf, vstr, o_in, PRINT_REPR); |
Damien George | 2617eeb | 2014-05-25 22:27:57 +0100 | [diff] [blame] | 467 | mp_obj_t s = mp_obj_new_str(vstr->buf, vstr->len, false); |
Damien George | 5fa93b6 | 2014-01-22 14:35:10 +0000 | [diff] [blame] | 468 | vstr_free(vstr); |
| 469 | return s; |
Damien George | e2fb2ba | 2014-01-15 21:40:48 +0000 | [diff] [blame] | 470 | } |
| 471 | |
| 472 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_repr_obj, mp_builtin_repr); |
| 473 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 474 | STATIC mp_obj_t mp_builtin_sum(uint n_args, const mp_obj_t *args) { |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 475 | assert(1 <= n_args && n_args <= 2); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 476 | mp_obj_t value; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 477 | switch (n_args) { |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 478 | case 1: value = mp_obj_new_int(0); break; |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 479 | default: value = args[1]; break; |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 480 | } |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 481 | mp_obj_t iterable = mp_getiter(args[0]); |
Damien | d99b052 | 2013-12-21 18:17:45 +0000 | [diff] [blame] | 482 | mp_obj_t item; |
Damien George | ea8d06c | 2014-04-17 23:19:36 +0100 | [diff] [blame] | 483 | while ((item = mp_iternext(iterable)) != MP_OBJ_STOP_ITERATION) { |
Damien George | d17926d | 2014-03-30 13:35:08 +0100 | [diff] [blame] | 484 | value = mp_binary_op(MP_BINARY_OP_ADD, value, item); |
Damien | a3dcd9e | 2013-12-17 21:35:38 +0000 | [diff] [blame] | 485 | } |
| 486 | return value; |
| 487 | } |
Damien George | e2fb2ba | 2014-01-15 21:40:48 +0000 | [diff] [blame] | 488 | |
Damien George | 2300537 | 2014-01-13 19:39:01 +0000 | [diff] [blame] | 489 | MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_sum_obj, 1, 2, mp_builtin_sum); |
John R. Lenton | 5c76839 | 2014-01-13 05:12:50 +0000 | [diff] [blame] | 490 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 491 | STATIC mp_obj_t mp_builtin_sorted(uint n_args, const mp_obj_t *args, mp_map_t *kwargs) { |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 492 | assert(n_args >= 1); |
| 493 | if (n_args > 1) { |
Damien George | ea13f40 | 2014-04-05 18:32:08 +0100 | [diff] [blame] | 494 | nlr_raise(mp_obj_new_exception_msg(&mp_type_TypeError, |
John R. Lenton | 5c76839 | 2014-01-13 05:12:50 +0000 | [diff] [blame] | 495 | "must use keyword argument for key function")); |
| 496 | } |
Damien George | 3e1a5c1 | 2014-03-29 13:43:38 +0000 | [diff] [blame] | 497 | mp_obj_t self = mp_type_list.make_new((mp_obj_t)&mp_type_list, 1, 0, args); |
Damien George | 20006db | 2014-01-18 14:10:48 +0000 | [diff] [blame] | 498 | mp_obj_list_sort(1, &self, kwargs); |
John R. Lenton | 5c76839 | 2014-01-13 05:12:50 +0000 | [diff] [blame] | 499 | |
| 500 | return self; |
| 501 | } |
Damien George | e2fb2ba | 2014-01-15 21:40:48 +0000 | [diff] [blame] | 502 | |
John R. Lenton | 88cb1e6 | 2014-01-13 19:55:18 +0000 | [diff] [blame] | 503 | MP_DEFINE_CONST_FUN_OBJ_KW(mp_builtin_sorted_obj, 1, mp_builtin_sorted); |
Paul Sokolovsky | 36c4499 | 2014-01-13 19:20:46 +0200 | [diff] [blame] | 504 | |
Paul Sokolovsky | 520e2f5 | 2014-02-12 18:31:30 +0200 | [diff] [blame] | 505 | STATIC mp_obj_t mp_builtin_id(mp_obj_t o_in) { |
Paul Sokolovsky | 645582f | 2014-07-23 01:38:19 +0300 | [diff] [blame] | 506 | mp_int_t id = (mp_int_t)o_in; |
| 507 | if (!MP_OBJ_IS_OBJ(o_in)) { |
| 508 | return mp_obj_new_int(id); |
| 509 | } else if (id >= 0) { |
| 510 | // Many OSes and CPUs have affinity for putting "user" memories |
| 511 | // into low half of address space, and "system" into upper half. |
| 512 | // We're going to take advantage of that and return small int |
| 513 | // (signed) for such "user" addresses. |
| 514 | return MP_OBJ_NEW_SMALL_INT(id); |
| 515 | } else { |
| 516 | // If that didn't work, well, let's return long int, just as |
| 517 | // a (big) positve value, so it will never clash with the range |
| 518 | // of small int returned in previous case. |
| 519 | return mp_obj_new_int_from_uint((mp_uint_t)id); |
| 520 | } |
xbe | 0ebf853 | 2014-02-01 19:00:41 -0800 | [diff] [blame] | 521 | } |
| 522 | |
| 523 | MP_DEFINE_CONST_FUN_OBJ_1(mp_builtin_id_obj, mp_builtin_id); |
Paul Sokolovsky | e9137b9 | 2014-03-26 23:35:13 +0200 | [diff] [blame] | 524 | |
Paul Sokolovsky | 36dd19a | 2014-04-06 02:12:03 +0300 | [diff] [blame] | 525 | // See mp_load_attr() if making any changes |
| 526 | STATIC inline mp_obj_t mp_load_attr_default(mp_obj_t base, qstr attr, mp_obj_t defval) { |
| 527 | mp_obj_t dest[2]; |
| 528 | // use load_method, raising or not raising exception |
| 529 | ((defval == MP_OBJ_NULL) ? mp_load_method : mp_load_method_maybe)(base, attr, dest); |
| 530 | if (dest[0] == MP_OBJ_NULL) { |
| 531 | return defval; |
| 532 | } else if (dest[1] == MP_OBJ_NULL) { |
| 533 | // load_method returned just a normal attribute |
| 534 | return dest[0]; |
| 535 | } else { |
| 536 | // load_method returned a method, so build a bound method object |
| 537 | return mp_obj_new_bound_meth(dest[0], dest[1]); |
| 538 | } |
| 539 | } |
| 540 | |
Paul Sokolovsky | bfb7d6a | 2014-04-05 13:33:04 +0300 | [diff] [blame] | 541 | STATIC mp_obj_t mp_builtin_getattr(uint n_args, const mp_obj_t *args) { |
Paul Sokolovsky | b4efac1 | 2014-06-08 01:13:35 +0300 | [diff] [blame] | 542 | mp_obj_t attr = args[1]; |
| 543 | if (MP_OBJ_IS_TYPE(attr, &mp_type_str)) { |
| 544 | attr = mp_obj_str_intern(attr); |
| 545 | } else if (!MP_OBJ_IS_QSTR(attr)) { |
| 546 | nlr_raise(mp_obj_new_exception_msg(&mp_type_TypeError, "string required")); |
| 547 | } |
Paul Sokolovsky | bfb7d6a | 2014-04-05 13:33:04 +0300 | [diff] [blame] | 548 | mp_obj_t defval = MP_OBJ_NULL; |
| 549 | if (n_args > 2) { |
| 550 | defval = args[2]; |
| 551 | } |
Paul Sokolovsky | b4efac1 | 2014-06-08 01:13:35 +0300 | [diff] [blame] | 552 | return mp_load_attr_default(args[0], MP_OBJ_QSTR_VALUE(attr), defval); |
Paul Sokolovsky | e9137b9 | 2014-03-26 23:35:13 +0200 | [diff] [blame] | 553 | } |
| 554 | |
Paul Sokolovsky | bfb7d6a | 2014-04-05 13:33:04 +0300 | [diff] [blame] | 555 | MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(mp_builtin_getattr_obj, 2, 3, mp_builtin_getattr); |
Paul Sokolovsky | cc0af3d | 2014-04-06 01:00:46 +0300 | [diff] [blame] | 556 | |
Paul Sokolovsky | ff30666 | 2014-05-11 19:05:42 +0300 | [diff] [blame] | 557 | STATIC mp_obj_t mp_builtin_hasattr(mp_obj_t object_in, mp_obj_t attr_in) { |
| 558 | assert(MP_OBJ_IS_QSTR(attr_in)); |
| 559 | |
| 560 | mp_obj_t dest[2]; |
Chris Angelico | daf973a | 2014-06-06 03:51:03 +1000 | [diff] [blame] | 561 | // TODO: https://docs.python.org/3/library/functions.html?highlight=hasattr#hasattr |
Paul Sokolovsky | ff30666 | 2014-05-11 19:05:42 +0300 | [diff] [blame] | 562 | // explicitly says "This is implemented by calling getattr(object, name) and seeing |
| 563 | // whether it raises an AttributeError or not.", so we should explicitly wrap this |
| 564 | // in nlr_push and handle exception. |
| 565 | mp_load_method_maybe(object_in, MP_OBJ_QSTR_VALUE(attr_in), dest); |
| 566 | |
| 567 | return MP_BOOL(dest[0] != MP_OBJ_NULL); |
| 568 | } |
| 569 | |
| 570 | MP_DEFINE_CONST_FUN_OBJ_2(mp_builtin_hasattr_obj, mp_builtin_hasattr); |
| 571 | |
Paul Sokolovsky | 080d99b | 2014-04-06 01:18:19 +0300 | [diff] [blame] | 572 | // These two are defined in terms of MicroPython API functions right away |
| 573 | MP_DEFINE_CONST_FUN_OBJ_0(mp_builtin_globals_obj, mp_globals_get); |
| 574 | MP_DEFINE_CONST_FUN_OBJ_0(mp_builtin_locals_obj, mp_locals_get); |