diff --git a/py/objnamedtuple.c b/py/objnamedtuple.c index 18931a16c2..6f50d6cf96 100644 --- a/py/objnamedtuple.c +++ b/py/objnamedtuple.c @@ -28,22 +28,13 @@ #include #include "py/nlr.h" +#include "py/objnamedtuple.h" #include "py/objtuple.h" #include "py/runtime.h" #include "py/objstr.h" #if MICROPY_PY_COLLECTIONS -typedef struct _mp_obj_namedtuple_type_t { - mp_obj_type_t base; - mp_uint_t n_fields; - qstr fields[]; -} mp_obj_namedtuple_type_t; - -typedef struct _mp_obj_namedtuple_t { - mp_obj_tuple_t tuple; -} mp_obj_namedtuple_t; - STATIC mp_uint_t namedtuple_find_field(const mp_obj_namedtuple_type_t *type, qstr name) { for (mp_uint_t i = 0; i < type->n_fields; i++) { if (type->fields[i] == name) { @@ -53,7 +44,7 @@ STATIC mp_uint_t namedtuple_find_field(const mp_obj_namedtuple_type_t *type, qst return -1; } -STATIC void namedtuple_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind) { +void namedtuple_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind) { (void)kind; mp_obj_namedtuple_t *o = MP_OBJ_TO_PTR(o_in); mp_printf(print, "%q", o->tuple.base.type->name); @@ -61,7 +52,7 @@ STATIC void namedtuple_print(const mp_print_t *print, mp_obj_t o_in, mp_print_ki mp_obj_attrtuple_print_helper(print, fields, &o->tuple); } -STATIC void namedtuple_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) { +void namedtuple_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) { if (dest[0] == MP_OBJ_NULL) { // load attribute mp_obj_namedtuple_t *self = MP_OBJ_TO_PTR(self_in); @@ -77,7 +68,7 @@ STATIC void namedtuple_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) { } } -STATIC mp_obj_t namedtuple_make_new(const mp_obj_type_t *type_in, size_t n_args, size_t n_kw, const mp_obj_t *args) { +mp_obj_t namedtuple_make_new(const mp_obj_type_t *type_in, size_t n_args, size_t n_kw, const mp_obj_t *args) { const mp_obj_namedtuple_type_t *type = (const mp_obj_namedtuple_type_t*)type_in; size_t num_fields = type->n_fields; if (n_args + n_kw != num_fields) { @@ -134,7 +125,7 @@ STATIC mp_obj_t namedtuple_make_new(const mp_obj_type_t *type_in, size_t n_args, return MP_OBJ_FROM_PTR(tuple); } -STATIC const mp_rom_obj_tuple_t namedtuple_base_tuple = {{&mp_type_tuple}, 1, {MP_ROM_PTR(&mp_type_tuple)}}; +const mp_rom_obj_tuple_t namedtuple_base_tuple = {{&mp_type_tuple}, 1, {MP_ROM_PTR(&mp_type_tuple)}}; STATIC mp_obj_t mp_obj_new_namedtuple_type(qstr name, mp_uint_t n_fields, mp_obj_t *fields) { mp_obj_namedtuple_type_t *o = m_new_obj_var(mp_obj_namedtuple_type_t, qstr, n_fields); diff --git a/py/objnamedtuple.h b/py/objnamedtuple.h new file mode 100644 index 0000000000..f6621642ae --- /dev/null +++ b/py/objnamedtuple.h @@ -0,0 +1,55 @@ +/* + * This file is part of the Micro Python project, http://micropython.org/ + * + * The MIT License (MIT) + * + * Copyright (c) 2013, 2014 Damien P. George + * Copyright (c) 2014 Paul Sokolovsky + * + * Permission is hereby granted, free of charge, to any person obtaining a copy + * of this software and associated documentation files (the "Software"), to deal + * in the Software without restriction, including without limitation the rights + * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell + * copies of the Software, and to permit persons to whom the Software is + * furnished to do so, subject to the following conditions: + * + * The above copyright notice and this permission notice shall be included in + * all copies or substantial portions of the Software. + * + * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR + * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, + * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE + * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER + * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, + * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN + * THE SOFTWARE. + */ + +#include + +#include "py/nlr.h" +#include "py/objtuple.h" +#include "py/runtime.h" +#include "py/objstr.h" + +#if MICROPY_PY_COLLECTIONS + +typedef struct _mp_obj_namedtuple_type_t { + mp_obj_type_t base; + mp_uint_t n_fields; + qstr fields[]; +} mp_obj_namedtuple_type_t; + +typedef struct _mp_obj_namedtuple_t { + mp_obj_tuple_t tuple; +} mp_obj_namedtuple_t; + +void namedtuple_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind); + +void namedtuple_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest); + +mp_obj_t namedtuple_make_new(const mp_obj_type_t *type_in, size_t n_args, size_t n_kw, const mp_obj_t *args); + +const mp_rom_obj_tuple_t namedtuple_base_tuple; + +#endif // MICROPY_PY_COLLECTIONS diff --git a/shared-bindings/nativeio/DigitalInOut.c b/shared-bindings/nativeio/DigitalInOut.c index 1ffb5c7f3f..6c15944eb5 100644 --- a/shared-bindings/nativeio/DigitalInOut.c +++ b/shared-bindings/nativeio/DigitalInOut.c @@ -198,7 +198,7 @@ mp_obj_property_t nativeio_digitalinout_direction_obj = { //| .. attribute:: value //| -//| Get or set the digital logic level of the pin. +//| The digital logic level of the pin. //| STATIC mp_obj_t nativeio_digitalinout_obj_get_value(mp_obj_t self_in) { nativeio_digitalinout_obj_t *self = MP_OBJ_TO_PTR(self_in); diff --git a/shared-bindings/time/__init__.c b/shared-bindings/time/__init__.c index cd41884fd1..1fdacc0750 100644 --- a/shared-bindings/time/__init__.c +++ b/shared-bindings/time/__init__.c @@ -28,12 +28,8 @@ #include -//#include "py/nlr.h" #include "py/obj.h" -//#include "py/gc.h" -//#include "py/runtime.h" -//#include "py/mphal.h" -//#include "py/smallint.h" +#include "py/objnamedtuple.h" #include "shared-bindings/time/__init__.h" //| :mod:`time` --- time and timing related functions @@ -76,11 +72,71 @@ STATIC mp_obj_t time_sleep(mp_obj_t seconds_o) { } MP_DEFINE_CONST_FUN_OBJ_1(time_sleep_obj, time_sleep); +#if MICROPY_PY_COLLECTIONS +mp_obj_t struct_time_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) { + if (n_args != 1) { + nlr_raise(mp_obj_new_exception_msg(&mp_type_TypeError, "time.struct_time() takes exactly 1 argument")); + } + if (!MP_OBJ_IS_TYPE(args[0], &mp_type_tuple) || ((mp_obj_tuple_t*) MP_OBJ_TO_PTR(args[0]))->len != 9) { + nlr_raise(mp_obj_new_exception_msg(&mp_type_TypeError, "time.struct_time() takes a 9-sequence")); + } + + mp_obj_tuple_t* tuple = MP_OBJ_TO_PTR(args[0]); + return namedtuple_make_new(type, 9, 0, tuple->items); +} + +//| .. class:: struct_time((tm_year, tm_mon, tm_mday, tm_hour, tm_min, tm_sec, tm_wday, tm_yday, tm_isdst)) +//| +//| Structure used to capture a date and time. Note that it takes a tuple! +//| +//| :param int tm_year: the year, 2017 for example +//| :param int tm_mon: the month, range [1, 12] +//| :param int tm_mday: the day of the month, range [1, 31] +//| :param int tm_hour: the hour, range [0, 23] +//| :param int tm_min: the minute, range [0, 59] +//| :param int tm_sec: the second, range [0, 61] +//| :param int tm_wday: the day of the week, range [0, 6], Monday is 0 +//| :param int tm_yday: the day of the year, range [1, 366], -1 indicates not known +//| :param int tm_isdst: 1 when in daylight savings, 0 when not, -1 if unknown. +//| +mp_obj_namedtuple_type_t struct_time_type_obj = { + .base = { + .base = { + .type = &mp_type_type + }, + .name = MP_QSTR_struct_time, + .print = namedtuple_print, + .make_new = struct_time_make_new, + .unary_op = mp_obj_tuple_unary_op, + .binary_op = mp_obj_tuple_binary_op, + .attr = namedtuple_attr, + .subscr = mp_obj_tuple_subscr, + .getiter = mp_obj_tuple_getiter, + .bases_tuple = (mp_obj_tuple_t*)(mp_rom_obj_tuple_t*)&namedtuple_base_tuple, + }, + .n_fields = 9, + .fields = { + MP_QSTR_tm_year, + MP_QSTR_tm_mon, + MP_QSTR_tm_mday, + MP_QSTR_tm_hour, + MP_QSTR_tm_min, + MP_QSTR_tm_sec, + MP_QSTR_tm_wday, + MP_QSTR_tm_yday, + MP_QSTR_tm_isdst + }, +}; +#endif + STATIC const mp_map_elem_t time_module_globals_table[] = { { MP_OBJ_NEW_QSTR(MP_QSTR___name__), MP_OBJ_NEW_QSTR(MP_QSTR_time) }, { MP_OBJ_NEW_QSTR(MP_QSTR_monotonic), (mp_obj_t)&time_monotonic_obj }, { MP_OBJ_NEW_QSTR(MP_QSTR_sleep), (mp_obj_t)&time_sleep_obj }, + #if MICROPY_PY_COLLECTIONS + { MP_OBJ_NEW_QSTR(MP_QSTR_struct_time), (mp_obj_t)&struct_time_type_obj }, + #endif }; STATIC MP_DEFINE_CONST_DICT(time_module_globals, time_module_globals_table);