diff options
author | Jim Mussared <jim.mussared@gmail.com> | 2022-04-22 15:32:22 +1000 |
---|---|---|
committer | Damien George <damien@micropython.org> | 2022-05-03 22:23:46 +1000 |
commit | 709e8328d9812a2e02da6fba65a03f9a873d60a2 (patch) | |
tree | 72d53933f3e251be477f709ad3d94a2609b85fcd /py/obj.c | |
parent | 590de399f01cd08aa4825b26b91785e07abcf68c (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.c | 7 |
1 files changed, 7 insertions, 0 deletions
@@ -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 |