summaryrefslogtreecommitdiff
path: root/py/obj.c
diff options
context:
space:
mode:
authorJim Mussared <jim.mussared@gmail.com>2022-04-22 15:32:22 +1000
committerDamien George <damien@micropython.org>2022-05-03 22:23:46 +1000
commit709e8328d9812a2e02da6fba65a03f9a873d60a2 (patch)
tree72d53933f3e251be477f709ad3d94a2609b85fcd /py/obj.c
parent590de399f01cd08aa4825b26b91785e07abcf68c (diff)
py/obj: Introduce mp_obj_malloc macro to allocate, and set object type.
This is to replace the following: mp_foo_obj_t *self = m_new_obj(mp_foo_obj_t); self->base.type = &mp_type_foo; with: mp_foo_obj_t *self = mp_obj_malloc(mp_foo_obj_t, &mp_type_foo); Calling the function is less code than inlining setting the type everywhere, adds up to ~100 bytes on PYBV11. It also helps to avoid an easy mistake of forgetting to set the type. Signed-off-by: Jim Mussared <jim.mussared@gmail.com>
Diffstat (limited to 'py/obj.c')
-rw-r--r--py/obj.c7
1 files changed, 7 insertions, 0 deletions
diff --git a/py/obj.c b/py/obj.c
index 5255e9655..024de3c96 100644
--- a/py/obj.c
+++ b/py/obj.c
@@ -37,6 +37,13 @@
#include "py/stackctrl.h"
#include "py/stream.h" // for mp_obj_print
+// Allocates an object and also sets type, for mp_obj_malloc{,_var} macros.
+void *mp_obj_malloc_helper(size_t num_bytes, const mp_obj_type_t *type) {
+ mp_obj_base_t *base = (mp_obj_base_t *)m_malloc(num_bytes);
+ base->type = type;
+ return base;
+}
+
const mp_obj_type_t *MICROPY_WRAP_MP_OBJ_GET_TYPE(mp_obj_get_type)(mp_const_obj_t o_in) {
#if MICROPY_OBJ_IMMEDIATE_OBJS && MICROPY_OBJ_REPR == MICROPY_OBJ_REPR_A