aboutsummaryrefslogtreecommitdiff
path: root/py/objclosure.c
diff options
context:
space:
mode:
authorDamien George <damien.p.george@gmail.com>2014-01-18 14:10:48 +0000
committerDamien George <damien.p.george@gmail.com>2014-01-18 14:10:48 +0000
commit20006dbba9d2d84ead036fdfab7190e88b2337ce (patch)
treea83dc966464cbbee17a397bda96360e9d238f8cf /py/objclosure.c
parent8655065f8cec8b978d075adae1f65ffdfa9b51d8 (diff)
Make VM stack grow upwards, and so no reversed args arrays.
Change state layout in VM so the stack starts at state[0] and grows upwards. Locals are at the top end of the state and number downwards. This cleans up a lot of the interface connecting the VM to C: now all functions that take an array of Micro Python objects are in order (ie no longer in reverse). Also clean up C API with keyword arguments (call_n and call_n_kw replaced with single call method that takes keyword arguments). And now make_new takes keyword arguments. emitnative.c has not yet been changed to comply with the new order of stack layout.
Diffstat (limited to 'py/objclosure.c')
-rw-r--r--py/objclosure.c31
1 files changed, 19 insertions, 12 deletions
diff --git a/py/objclosure.c b/py/objclosure.c
index b372ee6fe..7f6bcf4ac 100644
--- a/py/objclosure.c
+++ b/py/objclosure.c
@@ -16,26 +16,33 @@ typedef struct _mp_obj_closure_t {
mp_obj_t *closed;
} mp_obj_closure_t;
-// args are in reverse order in the array
-mp_obj_t closure_call_n(mp_obj_t self_in, int n_args, const mp_obj_t *args) {
+mp_obj_t closure_call(mp_obj_t self_in, uint n_args, uint n_kw, const mp_obj_t *args) {
mp_obj_closure_t *self = self_in;
- // concatenate args and closed-over-vars, in reverse order
- // TODO perhaps cache this array so we don't need to create it each time we are called
- mp_obj_t *args2 = m_new(mp_obj_t, self->n_closed + n_args);
- memcpy(args2, args, n_args * sizeof(mp_obj_t));
- for (int i = 0; i < self->n_closed; i++) {
- args2[n_args + i] = self->closed[self->n_closed - 1 - i];
- }
+ // need to concatenate closed-over-vars and args
- // call the function with the new vars array
- return rt_call_function_n(self->fun, n_args + self->n_closed, args2);
+ int n_total = self->n_closed + n_args + 2 * n_kw;
+ if (n_total <= 5) {
+ // use stack to allocate temporary args array
+ mp_obj_t args2[5];
+ memcpy(args2, self->closed, self->n_closed * sizeof(mp_obj_t));
+ memcpy(args2 + self->n_closed, args, (n_args + 2 * n_kw) * sizeof(mp_obj_t));
+ return rt_call_function_n_kw(self->fun, self->n_closed + n_args, n_kw, args2);
+ } else {
+ // use heap to allocate temporary args array
+ mp_obj_t *args2 = m_new(mp_obj_t, n_total);
+ memcpy(args2, self->closed, self->n_closed * sizeof(mp_obj_t));
+ memcpy(args2 + self->n_closed, args, (n_args + 2 * n_kw) * sizeof(mp_obj_t));
+ mp_obj_t res = rt_call_function_n_kw(self->fun, self->n_closed + n_args, n_kw, args2);
+ m_del(mp_obj_t, args2, n_total);
+ return res;
+ }
}
const mp_obj_type_t closure_type = {
{ &mp_const_type },
"closure",
- .call_n = closure_call_n,
+ .call = closure_call,
};
mp_obj_t mp_obj_new_closure(mp_obj_t fun, mp_obj_t closure_tuple) {