blob: 0fe2f774af0aeb0c0c53432b5e0cfcf8931551f7 [file] [log] [blame]
Damien George04b91472014-05-03 23:27:38 +01001/*
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
Paul Sokolovskyda9f0922014-05-13 08:44:45 +03007 * Copyright (c) 2014 Paul Sokolovsky
Damien George04b91472014-05-03 23:27:38 +01008 *
9 * Permission is hereby granted, free of charge, to any person obtaining a copy
10 * of this software and associated documentation files (the "Software"), to deal
11 * in the Software without restriction, including without limitation the rights
12 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
13 * copies of the Software, and to permit persons to whom the Software is
14 * furnished to do so, subject to the following conditions:
15 *
16 * The above copyright notice and this permission notice shall be included in
17 * all copies or substantial portions of the Software.
18 *
19 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
20 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
21 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
22 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
23 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
24 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
25 * THE SOFTWARE.
26 */
27
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020028#include <string.h>
29
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020030#include "mpconfig.h"
Paul Sokolovskyf54bcbf2014-05-02 17:47:01 +030031#include "misc.h"
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020032#include "nlr.h"
33#include "qstr.h"
34#include "obj.h"
35#include "objtuple.h"
36
Damien Georgeee3fd462014-05-24 23:03:12 +010037#if MICROPY_PY_COLLECTIONS
Damien George107c9fb2014-04-26 10:31:15 +010038
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020039typedef struct _mp_obj_namedtuple_type_t {
40 mp_obj_type_t base;
stijn12340142014-12-20 16:37:40 +010041 mp_uint_t n_fields;
42 mp_obj_t fields[];
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020043} mp_obj_namedtuple_type_t;
44
45typedef struct _mp_obj_namedtuple_t {
46 mp_obj_tuple_t tuple;
47} mp_obj_namedtuple_t;
48
stijn12340142014-12-20 16:37:40 +010049STATIC mp_uint_t namedtuple_find_field(mp_obj_namedtuple_type_t *type, qstr name) {
50 for (mp_uint_t i = 0; i < type->n_fields; i++) {
51 if (MP_OBJ_QSTR_VALUE(type->fields[i]) == name) {
52 return i;
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020053 }
54 }
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020055 return -1;
56}
57
58STATIC void namedtuple_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind) {
59 mp_obj_namedtuple_t *o = o_in;
60 print(env, "%s(", qstr_str(o->tuple.base.type->name));
stijn12340142014-12-20 16:37:40 +010061 const mp_obj_t *fields = ((mp_obj_namedtuple_type_t*)o->tuple.base.type)->fields;
Damien George42f3de92014-10-03 17:44:14 +000062 for (mp_uint_t i = 0; i < o->tuple.len; i++) {
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020063 if (i > 0) {
Damien George42f3de92014-10-03 17:44:14 +000064 print(env, ", ");
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020065 }
stijn12340142014-12-20 16:37:40 +010066 print(env, "%s=", qstr_str(MP_OBJ_QSTR_VALUE(fields[i])));
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020067 mp_obj_print_helper(print, env, o->tuple.items[i], PRINT_REPR);
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020068 }
69 print(env, ")");
70}
71
72STATIC void namedtuple_load_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
73 mp_obj_namedtuple_t *self = self_in;
stijn12340142014-12-20 16:37:40 +010074 int id = namedtuple_find_field((mp_obj_namedtuple_type_t*)self->tuple.base.type, attr);
75 if (id == -1) {
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020076 return;
77 }
78 dest[0] = self->tuple.items[id];
79}
80
81STATIC bool namedtuple_store_attr(mp_obj_t self_in, qstr attr, mp_obj_t value) {
Damien Georgeea13f402014-04-05 18:32:08 +010082 nlr_raise(mp_obj_new_exception_msg(&mp_type_AttributeError, "can't set attribute"));
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020083}
84
Damien Georgeecc88e92014-08-30 00:35:11 +010085STATIC mp_obj_t namedtuple_make_new(mp_obj_t type_in, mp_uint_t n_args, mp_uint_t n_kw, const mp_obj_t *args) {
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020086 mp_obj_namedtuple_type_t *type = type_in;
stijn12340142014-12-20 16:37:40 +010087 mp_uint_t num_fields = type->n_fields;
88 if (n_args + n_kw != num_fields) {
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020089 // Counts include implicit "self"
Damien Georgeea13f402014-04-05 18:32:08 +010090 nlr_raise(mp_obj_new_exception_msg_varg(&mp_type_TypeError,
Paul Sokolovskyd08fd682014-02-27 22:22:04 +020091 "__new__() takes %d positional arguments but %d were given",
92 num_fields + 1, n_args + 1));
93 }
94 mp_obj_tuple_t *tuple = mp_obj_new_tuple(n_args, args);
95 tuple->base.type = type_in;
96 return tuple;
97}
98
Damien George07ddab52014-03-29 13:15:08 +000099STATIC const mp_obj_tuple_t namedtuple_base_tuple = {{&mp_type_tuple}, 1, {(mp_obj_t)&mp_type_tuple}};
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200100
stijn12340142014-12-20 16:37:40 +0100101STATIC mp_obj_t mp_obj_new_namedtuple_type(qstr name, mp_uint_t n_fields, mp_obj_t *fields) {
102 mp_obj_namedtuple_type_t *o = m_new_obj_var(mp_obj_namedtuple_type_t, mp_obj_t, n_fields);
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200103 o->base.base.type = &mp_type_type;
104 o->base.name = name;
105 o->base.print = namedtuple_print;
106 o->base.make_new = namedtuple_make_new;
Damien George2323ef92014-05-11 18:00:45 +0100107 o->base.unary_op = mp_obj_tuple_unary_op;
108 o->base.binary_op = mp_obj_tuple_binary_op;
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200109 o->base.load_attr = namedtuple_load_attr;
110 o->base.store_attr = namedtuple_store_attr;
Damien George2323ef92014-05-11 18:00:45 +0100111 o->base.subscr = mp_obj_tuple_subscr;
112 o->base.getiter = mp_obj_tuple_getiter;
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200113 o->base.bases_tuple = (mp_obj_t)&namedtuple_base_tuple;
stijn12340142014-12-20 16:37:40 +0100114 o->n_fields = n_fields;
115 memcpy(o->fields, fields, sizeof(mp_obj_t) * n_fields);
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200116 return o;
117}
118
119STATIC mp_obj_t new_namedtuple_type(mp_obj_t name_in, mp_obj_t fields_in) {
120 qstr name = MP_OBJ_QSTR_VALUE(name_in);
stijn12340142014-12-20 16:37:40 +0100121 mp_uint_t n_fields;
122 mp_obj_t *fields;
123 mp_obj_list_get(fields_in, &n_fields, &fields);
124 return mp_obj_new_namedtuple_type(name, n_fields, fields);
Paul Sokolovskyd08fd682014-02-27 22:22:04 +0200125}
126MP_DEFINE_CONST_FUN_OBJ_2(mp_namedtuple_obj, new_namedtuple_type);
Damien George107c9fb2014-04-26 10:31:15 +0100127
Damien Georgeee3fd462014-05-24 23:03:12 +0100128#endif // MICROPY_PY_COLLECTIONS