mirrors/micropython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

py: Swap around the double return value of mp_obj_gen_resume.

Browse Source 
Just to keep things consistent :)
This commit is contained in:
Damien George 2014-03-26 19:33:23 +00:00
parent 66eaf84b8c
commit 69b3ba0df3
2 changed files with 33 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
py/objgenerator.c
View File

@ -74,11 +74,11 @@ mp_obj_t gen_instance_getiter(mp_obj_t self_in) {
return self_in; return self_in;
} }
mp_obj_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw_value, mp_vm_return_kind_t *ret_kind) { mp_vm_return_kind_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw_value, mp_obj_t *ret_val) {
mp_obj_gen_instance_t *self = self_in; mp_obj_gen_instance_t *self = self_in;
if (self->ip == 0) { if (self->ip == 0) {
*ret_kind = MP_VM_RETURN_NORMAL; *ret_val = MP_OBJ_NULL;
return MP_OBJ_NULL; return MP_VM_RETURN_NORMAL;
} }
if (self->sp == self->state - 1) { if (self->sp == self->state - 1) {
if (send_value != mp_const_none) { if (send_value != mp_const_none) {
@ -87,11 +87,11 @@ mp_obj_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw
} else { } else {
*self->sp = send_value; *self->sp = send_value;
} }
*ret_kind = mp_execute_byte_code_2(self->code_info, &self->ip, mp_vm_return_kind_t ret_kind = mp_execute_byte_code_2(self->code_info, &self->ip,
&self->state[self->n_state - 1], &self->sp, (mp_exc_stack*)(self->state + self->n_state), &self->state[self->n_state - 1], &self->sp, (mp_exc_stack*)(self->state + self->n_state),
&self->exc_sp, throw_value); &self->exc_sp, throw_value);
switch (*ret_kind) { switch (ret_kind) {
case MP_VM_RETURN_NORMAL: case MP_VM_RETURN_NORMAL:
// Explicitly mark generator as completed. If we don't do this, // Explicitly mark generator as completed. If we don't do this,
// subsequent next() may re-execute statements after last yield // subsequent next() may re-execute statements after last yield
@ -99,26 +99,30 @@ mp_obj_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw
// TODO: check how return with value behaves under such conditions // TODO: check how return with value behaves under such conditions
// in CPython. // in CPython.
self->ip = 0; self->ip = 0;
return *self->sp; *ret_val = *self->sp;
break;
case MP_VM_RETURN_YIELD: case MP_VM_RETURN_YIELD:
return *self->sp; *ret_val = *self->sp;
break;
case MP_VM_RETURN_EXCEPTION: case MP_VM_RETURN_EXCEPTION:
self->ip = 0; self->ip = 0;
return self->state[self->n_state - 1]; *ret_val = self->state[self->n_state - 1];
break;
default: default:
assert(0); assert(0);
return mp_const_none; *ret_val = mp_const_none;
break;
} }
return ret_kind;
} }
STATIC mp_obj_t gen_resume_and_raise(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw_value) { STATIC mp_obj_t gen_resume_and_raise(mp_obj_t self_in, mp_obj_t send_value, mp_obj_t throw_value) {
mp_vm_return_kind_t ret_kind; mp_obj_t ret;
mp_obj_t ret = mp_obj_gen_resume(self_in, send_value, throw_value, &ret_kind); switch (mp_obj_gen_resume(self_in, send_value, throw_value, &ret)) {
switch (ret_kind) {
case MP_VM_RETURN_NORMAL: case MP_VM_RETURN_NORMAL:
// Optimize return w/o value in case generator is used in for loop // Optimize return w/o value in case generator is used in for loop
if (ret == mp_const_none) { if (ret == mp_const_none) {
@ -166,24 +170,24 @@ STATIC mp_obj_t gen_instance_throw(uint n_args, const mp_obj_t *args) {
STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(gen_instance_throw_obj, 2, 4, gen_instance_throw); STATIC MP_DEFINE_CONST_FUN_OBJ_VAR_BETWEEN(gen_instance_throw_obj, 2, 4, gen_instance_throw);
STATIC mp_obj_t gen_instance_close(mp_obj_t self_in) { STATIC mp_obj_t gen_instance_close(mp_obj_t self_in) {
mp_vm_return_kind_t ret_kind; mp_obj_t ret;
mp_obj_t ret = mp_obj_gen_resume(self_in, mp_const_none, (mp_obj_t)&mp_type_GeneratorExit, &ret_kind); switch (mp_obj_gen_resume(self_in, mp_const_none, (mp_obj_t)&mp_type_GeneratorExit, &ret)) {
case MP_VM_RETURN_YIELD:
nlr_jump(mp_obj_new_exception_msg(&mp_type_RuntimeError, "generator ignored GeneratorExit"));
if (ret_kind == MP_VM_RETURN_YIELD) { // Swallow StopIteration & GeneratorExit (== successful close), and re-raise any other
nlr_jump(mp_obj_new_exception_msg(&mp_type_RuntimeError, "generator ignored GeneratorExit")); case MP_VM_RETURN_EXCEPTION:
} // ret should always be an instance of an exception class
// Swallow StopIteration & GeneratorExit (== successful close), and re-raise any other if (mp_obj_is_subclass_fast(mp_obj_get_type(ret), &mp_type_GeneratorExit) ||
if (ret_kind == MP_VM_RETURN_EXCEPTION) { mp_obj_is_subclass_fast(mp_obj_get_type(ret), &mp_type_StopIteration)) {
// ret should always be an instance of an exception class return mp_const_none;
if (mp_obj_is_subclass_fast(mp_obj_get_type(ret), &mp_type_GeneratorExit) || }
mp_obj_is_subclass_fast(mp_obj_get_type(ret), &mp_type_StopIteration)) { nlr_jump(ret);
default:
// The only choice left is MP_VM_RETURN_NORMAL which is successful close
return mp_const_none; return mp_const_none;
}
nlr_jump(ret);
} }
// The only choice left is MP_VM_RETURN_NORMAL which is successful close
return mp_const_none;
} }
STATIC MP_DEFINE_CONST_FUN_OBJ_1(gen_instance_close_obj, gen_instance_close); STATIC MP_DEFINE_CONST_FUN_OBJ_1(gen_instance_close_obj, gen_instance_close);

2
py/objgenerator.h
View File

@ -1 +1 @@
mp_obj_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_val, mp_obj_t throw_val, mp_vm_return_kind_t *ret_kind); mp_vm_return_kind_t mp_obj_gen_resume(mp_obj_t self_in, mp_obj_t send_val, mp_obj_t throw_val, mp_obj_t *ret_val);