Numworks Epsilon  1.4.1
Graphing Calculator Operating System
objattrtuple.c
Go to the documentation of this file.
1 /*
2  * This file is part of the MicroPython project, http://micropython.org/
3  *
4  * The MIT License (MIT)
5  *
6  * Copyright (c) 2015 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 
27 #include "py/objtuple.h"
28 
29 #if MICROPY_PY_ATTRTUPLE || MICROPY_PY_COLLECTIONS
30 
31 // this helper function is used by collections.namedtuple
32 #if !MICROPY_PY_COLLECTIONS
33 STATIC
34 #endif
35 void mp_obj_attrtuple_print_helper(const mp_print_t *print, const qstr *fields, mp_obj_tuple_t *o) {
36  mp_print_str(print, "(");
37  for (size_t i = 0; i < o->len; i++) {
38  if (i > 0) {
39  mp_print_str(print, ", ");
40  }
41  mp_printf(print, "%q=", fields[i]);
42  mp_obj_print_helper(print, o->items[i], PRINT_REPR);
43  }
44  mp_print_str(print, ")");
45 }
46 
47 #endif
48 
49 #if MICROPY_PY_ATTRTUPLE
50 
51 STATIC void mp_obj_attrtuple_print(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind) {
52  (void)kind;
53  mp_obj_tuple_t *o = MP_OBJ_TO_PTR(o_in);
54  const qstr *fields = (const qstr*)MP_OBJ_TO_PTR(o->items[o->len]);
55  mp_obj_attrtuple_print_helper(print, fields, o);
56 }
57 
58 STATIC void mp_obj_attrtuple_attr(mp_obj_t self_in, qstr attr, mp_obj_t *dest) {
59  if (dest[0] == MP_OBJ_NULL) {
60  // load attribute
61  mp_obj_tuple_t *self = MP_OBJ_TO_PTR(self_in);
62  size_t len = self->len;
63  const qstr *fields = (const qstr*)MP_OBJ_TO_PTR(self->items[len]);
64  for (size_t i = 0; i < len; i++) {
65  if (fields[i] == attr) {
66  dest[0] = self->items[i];
67  return;
68  }
69  }
70  }
71 }
72 
73 mp_obj_t mp_obj_new_attrtuple(const qstr *fields, size_t n, const mp_obj_t *items) {
75  o->base.type = &mp_type_attrtuple;
76  o->len = n;
77  for (size_t i = 0; i < n; i++) {
78  o->items[i] = items[i];
79  }
80  o->items[n] = MP_OBJ_FROM_PTR(fields);
81  return MP_OBJ_FROM_PTR(o);
82 }
83 
85  { &mp_type_type },
86  .name = MP_QSTR_tuple, // reuse tuple to save on a qstr
87  .print = mp_obj_attrtuple_print,
88  .unary_op = mp_obj_tuple_unary_op,
89  .binary_op = mp_obj_tuple_binary_op,
90  .attr = mp_obj_attrtuple_attr,
91  .subscr = mp_obj_tuple_subscr,
92  .getiter = mp_obj_tuple_getiter,
93 };
94 
95 #endif // MICROPY_PY_ATTRTUPLE
size_t len
Definition: objtuple.h:33
mp_obj_base_t base
Definition: objtuple.h:32
mp_obj_t mp_obj_tuple_subscr(mp_obj_t self_in, mp_obj_t index, mp_obj_t value)
Definition: objtuple.c:177
STATIC const uint8_t attr[]
Definition: unicode.c:51
mp_obj_t mp_obj_tuple_binary_op(mp_binary_op_t op, mp_obj_t lhs, mp_obj_t rhs)
Definition: objtuple.c:139
int mp_print_str(const mp_print_t *print, const char *str)
Definition: mpprint.c:53
#define MP_OBJ_FROM_PTR(p)
Definition: obj.h:233
mp_obj_t mp_obj_new_attrtuple(const qstr *fields, size_t n, const mp_obj_t *items)
#define STATIC
Definition: mpconfig.h:1178
mp_obj_t items[]
Definition: objtuple.h:34
mp_print_kind_t
Definition: obj.h:412
#define m_new_obj_var(obj_type, var_type, var_num)
Definition: misc.h:62
mp_obj_t mp_obj_tuple_getiter(mp_obj_t o_in, mp_obj_iter_buf_t *iter_buf)
Definition: objtuple.c:284
#define MP_OBJ_NULL
Definition: obj.h:73
void mp_obj_print_helper(const mp_print_t *print, mp_obj_t o_in, mp_print_kind_t kind)
Definition: obj.c:59
size_t qstr
Definition: qstr.h:48
const mp_obj_type_t mp_type_attrtuple
const mp_obj_type_t mp_type_type
Definition: objtype.c:969
#define MP_OBJ_TO_PTR(o)
Definition: obj.h:228
qstr name
Definition: obj.h:478
uint64_t mp_obj_t
Definition: obj.h:39
int mp_printf(const mp_print_t *print, const char *fmt,...)
Definition: mpprint.c:380
mp_obj_t mp_obj_tuple_unary_op(mp_unary_op_t op, mp_obj_t self_in)
Definition: objtuple.c:122