djsundog/circuitpython
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

py, vm: Make unum a local variable for each opcode that uses it.

Browse Source 
This makes no change to the generated code, but it's now easier to
understand since unum is not a "global" variable anymore.
This commit is contained in:
Damien George 2014-12-02 19:25:10 +00:00
parent 57c70d6073
commit cd97a43f8d
1 changed files with 83 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
py/vm.c
View File

@ -61,36 +61,35 @@ typedef enum {
UNWIND_JUMP, UNWIND_JUMP,
} mp_unwind_reason_t; } mp_unwind_reason_t;
#define DECODE_UINT do { \ #define DECODE_UINT \
unum = 0; \ mp_uint_t unum = 0; \
do { \ do { \
unum = (unum << 7) + (*ip & 0x7f); \ unum = (unum << 7) + (*ip & 0x7f); \
} while ((*ip++ & 0x80) != 0); \ } while ((*ip++ & 0x80) != 0)
} while (0) #define DECODE_ULABEL mp_uint_t ulab = (ip[0] | (ip[1] << 8)); ip += 2
#define DECODE_ULABEL do { unum = (ip[0] | (ip[1] << 8)); ip += 2; } while (0) #define DECODE_SLABEL mp_uint_t slab = (ip[0] | (ip[1] << 8)) - 0x8000; ip += 2
#define DECODE_SLABEL do { unum = (ip[0] | (ip[1] << 8)) - 0x8000; ip += 2; } while (0)
#define DECODE_QSTR qstr qst = 0; \ #define DECODE_QSTR qstr qst = 0; \
do { \ do { \
qst = (qst << 7) + (*ip & 0x7f); \ qst = (qst << 7) + (*ip & 0x7f); \
} while ((*ip++ & 0x80) != 0) } while ((*ip++ & 0x80) != 0)
#define DECODE_PTR do { \ #define DECODE_PTR \
ip = (byte*)(((mp_uint_t)ip + sizeof(mp_uint_t) - 1) & (~(sizeof(mp_uint_t) - 1))); /* align ip */ \ ip = (byte*)(((mp_uint_t)ip + sizeof(mp_uint_t) - 1) & (~(sizeof(mp_uint_t) - 1))); /* align ip */ \
unum = *(mp_uint_t*)ip; \ void *ptr = (void*)*(mp_uint_t*)ip; \
ip += sizeof(mp_uint_t); \ ip += sizeof(mp_uint_t)
} while (0)
#define PUSH(val) *++sp = (val) #define PUSH(val) *++sp = (val)
#define POP() (*sp--) #define POP() (*sp--)
#define TOP() (*sp) #define TOP() (*sp)
#define SET_TOP(val) *sp = (val) #define SET_TOP(val) *sp = (val)
#define PUSH_EXC_BLOCK() \ #define PUSH_EXC_BLOCK() do { \
DECODE_ULABEL; /* except labels are always forward */ \ DECODE_ULABEL; /* except labels are always forward */ \
++exc_sp; \ ++exc_sp; \
exc_sp->opcode = *code_state->ip; \ exc_sp->opcode = *code_state->ip; \
exc_sp->handler = ip + unum; \ exc_sp->handler = ip + ulab; \
exc_sp->val_sp = MP_TAGPTR_MAKE(sp, currently_in_except_block); \ exc_sp->val_sp = MP_TAGPTR_MAKE(sp, currently_in_except_block); \
exc_sp->prev_exc = MP_OBJ_NULL; \ exc_sp->prev_exc = MP_OBJ_NULL; \
currently_in_except_block = 0; /* in a try block now */ currently_in_except_block = 0; /* in a try block now */ \
} while (0)
#define POP_EXC_BLOCK() \ #define POP_EXC_BLOCK() \
currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); /* restore previous state */ \ currently_in_except_block = MP_TAGPTR_TAG(exc_sp->val_sp); /* restore previous state */ \
@ -142,7 +141,6 @@ outer_dispatch_loop:
// local variables that are not visible to the exception handler // local variables that are not visible to the exception handler
const byte *ip = code_state->ip; const byte *ip = code_state->ip;
mp_obj_t *sp = code_state->sp; mp_obj_t *sp = code_state->sp;
mp_uint_t unum;
mp_obj_t obj_shared; mp_obj_t obj_shared;
// If we have exception to inject, now that we finish setting up // If we have exception to inject, now that we finish setting up
@ -225,7 +223,7 @@ dispatch_loop:
PUSH(MP_OBJ_NULL); PUSH(MP_OBJ_NULL);
DISPATCH(); DISPATCH();
ENTRY(MP_BC_LOAD_FAST_N): ENTRY(MP_BC_LOAD_FAST_N): {
DECODE_UINT; DECODE_UINT;
obj_shared = fastn[-unum]; obj_shared = fastn[-unum];
load_check: load_check:
@ -237,11 +235,13 @@ dispatch_loop:
} }
PUSH(obj_shared); PUSH(obj_shared);
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_LOAD_DEREF): ENTRY(MP_BC_LOAD_DEREF): {
DECODE_UINT; DECODE_UINT;
obj_shared = mp_obj_cell_get(fastn[-unum]); obj_shared = mp_obj_cell_get(fastn[-unum]);
goto load_check; goto load_check;
}
ENTRY(MP_BC_LOAD_NAME): { ENTRY(MP_BC_LOAD_NAME): {
DECODE_QSTR; DECODE_QSTR;
@ -278,15 +278,17 @@ dispatch_loop:
DISPATCH(); DISPATCH();
} }
ENTRY(MP_BC_STORE_FAST_N): ENTRY(MP_BC_STORE_FAST_N): {
DECODE_UINT; DECODE_UINT;
fastn[-unum] = POP(); fastn[-unum] = POP();
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_STORE_DEREF): ENTRY(MP_BC_STORE_DEREF): {
DECODE_UINT; DECODE_UINT;
mp_obj_cell_set(fastn[-unum], POP()); mp_obj_cell_set(fastn[-unum], POP());
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_STORE_NAME): { ENTRY(MP_BC_STORE_NAME): {
DECODE_QSTR; DECODE_QSTR;
@ -312,21 +314,23 @@ dispatch_loop:
sp -= 3; sp -= 3;
DISPATCH(); DISPATCH();
ENTRY(MP_BC_DELETE_FAST): ENTRY(MP_BC_DELETE_FAST): {
DECODE_UINT; DECODE_UINT;
if (fastn[-unum] == MP_OBJ_NULL) { if (fastn[-unum] == MP_OBJ_NULL) {
goto local_name_error; goto local_name_error;
} }
fastn[-unum] = MP_OBJ_NULL; fastn[-unum] = MP_OBJ_NULL;
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_DELETE_DEREF): ENTRY(MP_BC_DELETE_DEREF): {
DECODE_UINT; DECODE_UINT;
if (mp_obj_cell_get(fastn[-unum]) == MP_OBJ_NULL) { if (mp_obj_cell_get(fastn[-unum]) == MP_OBJ_NULL) {
goto local_name_error; goto local_name_error;
} }
mp_obj_cell_set(fastn[-unum], MP_OBJ_NULL); mp_obj_cell_set(fastn[-unum], MP_OBJ_NULL);
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_DELETE_NAME): { ENTRY(MP_BC_DELETE_NAME): {
DECODE_QSTR; DECODE_QSTR;
@ -371,42 +375,47 @@ dispatch_loop:
DISPATCH(); DISPATCH();
} }
ENTRY(MP_BC_JUMP): ENTRY(MP_BC_JUMP): {
DECODE_SLABEL; DECODE_SLABEL;
ip += unum; ip += slab;
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
ENTRY(MP_BC_POP_JUMP_IF_TRUE): ENTRY(MP_BC_POP_JUMP_IF_TRUE): {
DECODE_SLABEL; DECODE_SLABEL;
if (mp_obj_is_true(POP())) { if (mp_obj_is_true(POP())) {
ip += unum; ip += slab;
} }
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
ENTRY(MP_BC_POP_JUMP_IF_FALSE): ENTRY(MP_BC_POP_JUMP_IF_FALSE): {
DECODE_SLABEL; DECODE_SLABEL;
if (!mp_obj_is_true(POP())) { if (!mp_obj_is_true(POP())) {
ip += unum; ip += slab;
} }
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
ENTRY(MP_BC_JUMP_IF_TRUE_OR_POP): ENTRY(MP_BC_JUMP_IF_TRUE_OR_POP): {
DECODE_SLABEL; DECODE_SLABEL;
if (mp_obj_is_true(TOP())) { if (mp_obj_is_true(TOP())) {
ip += unum; ip += slab;
} else { } else {
sp--; sp--;
} }
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
ENTRY(MP_BC_JUMP_IF_FALSE_OR_POP): ENTRY(MP_BC_JUMP_IF_FALSE_OR_POP): {
DECODE_SLABEL; DECODE_SLABEL;
if (mp_obj_is_true(TOP())) { if (mp_obj_is_true(TOP())) {
sp--; sp--;
} else { } else {
ip += unum; ip += slab;
} }
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
ENTRY(MP_BC_SETUP_WITH): { ENTRY(MP_BC_SETUP_WITH): {
mp_obj_t obj = TOP(); mp_obj_t obj = TOP();
@ -478,12 +487,12 @@ dispatch_loop:
DISPATCH(); DISPATCH();
} }
ENTRY(MP_BC_UNWIND_JUMP): ENTRY(MP_BC_UNWIND_JUMP): {
DECODE_SLABEL; DECODE_SLABEL;
PUSH((void*)(ip + unum)); // push destination ip for jump PUSH((void*)(ip + slab)); // push destination ip for jump
PUSH((void*)(mp_uint_t)(*ip)); // push number of exception handlers to unwind (0x80 bit set if we also need to pop stack) PUSH((void*)(mp_uint_t)(*ip)); // push number of exception handlers to unwind (0x80 bit set if we also need to pop stack)
unwind_jump: unwind_jump:;
unum = (mp_uint_t)POP(); // get number of exception handlers to unwind mp_uint_t unum = (mp_uint_t)POP(); // get number of exception handlers to unwind
while ((unum & 0x7f) > 0) { while ((unum & 0x7f) > 0) {
unum -= 1; unum -= 1;
assert(exc_sp >= exc_stack); assert(exc_sp >= exc_stack);
@ -505,12 +514,14 @@ unwind_jump:
sp--; sp--;
} }
DISPATCH_WITH_PEND_EXC_CHECK(); DISPATCH_WITH_PEND_EXC_CHECK();
}
// matched against: POP_BLOCK or POP_EXCEPT (anything else?) // matched against: POP_BLOCK or POP_EXCEPT (anything else?)
ENTRY(MP_BC_SETUP_EXCEPT): ENTRY(MP_BC_SETUP_EXCEPT):
ENTRY(MP_BC_SETUP_FINALLY): ENTRY(MP_BC_SETUP_FINALLY): {
PUSH_EXC_BLOCK(); PUSH_EXC_BLOCK();
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_END_FINALLY): ENTRY(MP_BC_END_FINALLY):
// not fully implemented // not fully implemented
@ -550,7 +561,7 @@ unwind_jump:
mp_obj_t value = mp_iternext_allow_raise(TOP()); mp_obj_t value = mp_iternext_allow_raise(TOP());
if (value == MP_OBJ_STOP_ITERATION) { if (value == MP_OBJ_STOP_ITERATION) {
--sp; // pop the exhausted iterator --sp; // pop the exhausted iterator
ip += unum; // jump to after for-block ip += ulab; // jump to after for-block
} else { } else {
PUSH(value); // push the next iteration value PUSH(value); // push the next iteration value
} }
@ -584,59 +595,66 @@ unwind_jump:
} }
DISPATCH(); DISPATCH();
ENTRY(MP_BC_BUILD_TUPLE): ENTRY(MP_BC_BUILD_TUPLE): {
DECODE_UINT; DECODE_UINT;
sp -= unum - 1; sp -= unum - 1;
SET_TOP(mp_obj_new_tuple(unum, sp)); SET_TOP(mp_obj_new_tuple(unum, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_BUILD_LIST): ENTRY(MP_BC_BUILD_LIST): {
DECODE_UINT; DECODE_UINT;
sp -= unum - 1; sp -= unum - 1;
SET_TOP(mp_obj_new_list(unum, sp)); SET_TOP(mp_obj_new_list(unum, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_LIST_APPEND): ENTRY(MP_BC_LIST_APPEND): {
DECODE_UINT; DECODE_UINT;
// I think it's guaranteed by the compiler that sp[unum] is a list // I think it's guaranteed by the compiler that sp[unum] is a list
mp_obj_list_append(sp[-unum], sp[0]); mp_obj_list_append(sp[-unum], sp[0]);
sp--; sp--;
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_BUILD_MAP): ENTRY(MP_BC_BUILD_MAP): {
DECODE_UINT; DECODE_UINT;
PUSH(mp_obj_new_dict(unum)); PUSH(mp_obj_new_dict(unum));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_STORE_MAP): ENTRY(MP_BC_STORE_MAP):
sp -= 2; sp -= 2;
mp_obj_dict_store(sp[0], sp[2], sp[1]); mp_obj_dict_store(sp[0], sp[2], sp[1]);
DISPATCH(); DISPATCH();
ENTRY(MP_BC_MAP_ADD): ENTRY(MP_BC_MAP_ADD): {
DECODE_UINT; DECODE_UINT;
// I think it's guaranteed by the compiler that sp[-unum - 1] is a map // I think it's guaranteed by the compiler that sp[-unum - 1] is a map
mp_obj_dict_store(sp[-unum - 1], sp[0], sp[-1]); mp_obj_dict_store(sp[-unum - 1], sp[0], sp[-1]);
sp -= 2; sp -= 2;
DISPATCH(); DISPATCH();
}
#if MICROPY_PY_BUILTINS_SET #if MICROPY_PY_BUILTINS_SET
ENTRY(MP_BC_BUILD_SET): ENTRY(MP_BC_BUILD_SET): {
DECODE_UINT; DECODE_UINT;
sp -= unum - 1; sp -= unum - 1;
SET_TOP(mp_obj_new_set(unum, sp)); SET_TOP(mp_obj_new_set(unum, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_SET_ADD): ENTRY(MP_BC_SET_ADD): {
DECODE_UINT; DECODE_UINT;
// I think it's guaranteed by the compiler that sp[-unum] is a set // I think it's guaranteed by the compiler that sp[-unum] is a set
mp_obj_set_store(sp[-unum], sp[0]); mp_obj_set_store(sp[-unum], sp[0]);
sp--; sp--;
DISPATCH(); DISPATCH();
}
#endif #endif
#if MICROPY_PY_BUILTINS_SLICE #if MICROPY_PY_BUILTINS_SLICE
ENTRY(MP_BC_BUILD_SLICE): ENTRY(MP_BC_BUILD_SLICE): {
DECODE_UINT; DECODE_UINT;
if (unum == 2) { if (unum == 2) {
mp_obj_t stop = POP(); mp_obj_t stop = POP();
@ -649,30 +667,34 @@ unwind_jump:
SET_TOP(mp_obj_new_slice(start, stop, step)); SET_TOP(mp_obj_new_slice(start, stop, step));
} }
DISPATCH(); DISPATCH();
}
#endif #endif
ENTRY(MP_BC_UNPACK_SEQUENCE): ENTRY(MP_BC_UNPACK_SEQUENCE): {
DECODE_UINT; DECODE_UINT;
mp_unpack_sequence(sp[0], unum, sp); mp_unpack_sequence(sp[0], unum, sp);
sp += unum - 1; sp += unum - 1;
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_UNPACK_EX): ENTRY(MP_BC_UNPACK_EX): {
DECODE_UINT; DECODE_UINT;
mp_unpack_ex(sp[0], unum, sp); mp_unpack_ex(sp[0], unum, sp);
sp += (unum & 0xff) + ((unum >> 8) & 0xff); sp += (unum & 0xff) + ((unum >> 8) & 0xff);
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_MAKE_FUNCTION): ENTRY(MP_BC_MAKE_FUNCTION): {
DECODE_PTR; DECODE_PTR;
PUSH(mp_make_function_from_raw_code((mp_raw_code_t*)unum, MP_OBJ_NULL, MP_OBJ_NULL)); PUSH(mp_make_function_from_raw_code(ptr, MP_OBJ_NULL, MP_OBJ_NULL));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_MAKE_FUNCTION_DEFARGS): { ENTRY(MP_BC_MAKE_FUNCTION_DEFARGS): {
DECODE_PTR; DECODE_PTR;
// Stack layout: def_tuple def_dict <- TOS // Stack layout: def_tuple def_dict <- TOS
mp_obj_t def_dict = POP(); mp_obj_t def_dict = POP();
SET_TOP(mp_make_function_from_raw_code((mp_raw_code_t*)unum, TOP(), def_dict)); SET_TOP(mp_make_function_from_raw_code(ptr, TOP(), def_dict));
DISPATCH(); DISPATCH();
} }
@ -681,7 +703,7 @@ unwind_jump:
mp_uint_t n_closed_over = *ip++; mp_uint_t n_closed_over = *ip++;
// Stack layout: closed_overs <- TOS // Stack layout: closed_overs <- TOS
sp -= n_closed_over - 1; sp -= n_closed_over - 1;
SET_TOP(mp_make_closure_from_raw_code((mp_raw_code_t*)unum, n_closed_over, sp)); SET_TOP(mp_make_closure_from_raw_code(ptr, n_closed_over, sp));
DISPATCH(); DISPATCH();
} }
@ -690,19 +712,20 @@ unwind_jump:
mp_uint_t n_closed_over = *ip++; mp_uint_t n_closed_over = *ip++;
// Stack layout: def_tuple def_dict closed_overs <- TOS // Stack layout: def_tuple def_dict closed_overs <- TOS
sp -= 2 + n_closed_over - 1; sp -= 2 + n_closed_over - 1;
SET_TOP(mp_make_closure_from_raw_code((mp_raw_code_t*)unum, 0x100 | n_closed_over, sp)); SET_TOP(mp_make_closure_from_raw_code(ptr, 0x100 | n_closed_over, sp));
DISPATCH(); DISPATCH();
} }
ENTRY(MP_BC_CALL_FUNCTION): ENTRY(MP_BC_CALL_FUNCTION): {
DECODE_UINT; DECODE_UINT;
// unum & 0xff == n_positional // unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword // (unum >> 8) & 0xff == n_keyword
sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe); sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe);
SET_TOP(mp_call_function_n_kw(*sp, unum & 0xff, (unum >> 8) & 0xff, sp + 1)); SET_TOP(mp_call_function_n_kw(*sp, unum & 0xff, (unum >> 8) & 0xff, sp + 1));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_CALL_FUNCTION_VAR_KW): ENTRY(MP_BC_CALL_FUNCTION_VAR_KW): {
DECODE_UINT; DECODE_UINT;
// unum & 0xff == n_positional // unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword // (unum >> 8) & 0xff == n_keyword
@ -711,16 +734,18 @@ unwind_jump:
sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 2; sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 2;
SET_TOP(mp_call_method_n_kw_var(false, unum, sp)); SET_TOP(mp_call_method_n_kw_var(false, unum, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_CALL_METHOD): ENTRY(MP_BC_CALL_METHOD): {
DECODE_UINT; DECODE_UINT;
// unum & 0xff == n_positional // unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword // (unum >> 8) & 0xff == n_keyword
sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 1; sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 1;
SET_TOP(mp_call_method_n_kw(unum & 0xff, (unum >> 8) & 0xff, sp)); SET_TOP(mp_call_method_n_kw(unum & 0xff, (unum >> 8) & 0xff, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_CALL_METHOD_VAR_KW): ENTRY(MP_BC_CALL_METHOD_VAR_KW): {
DECODE_UINT; DECODE_UINT;
// unum & 0xff == n_positional // unum & 0xff == n_positional
// (unum >> 8) & 0xff == n_keyword // (unum >> 8) & 0xff == n_keyword
@ -729,6 +754,7 @@ unwind_jump:
sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 3; sp -= (unum & 0xff) + ((unum >> 7) & 0x1fe) + 3;
SET_TOP(mp_call_method_n_kw_var(true, unum, sp)); SET_TOP(mp_call_method_n_kw_var(true, unum, sp));
DISPATCH(); DISPATCH();
}
ENTRY(MP_BC_RETURN_VALUE): ENTRY(MP_BC_RETURN_VALUE):
unwind_return: unwind_return:
@ -754,7 +780,7 @@ unwind_return:
return MP_VM_RETURN_NORMAL; return MP_VM_RETURN_NORMAL;
ENTRY(MP_BC_RAISE_VARARGS): { ENTRY(MP_BC_RAISE_VARARGS): {
unum = *ip++; mp_uint_t unum = *ip++;
mp_obj_t obj; mp_obj_t obj;
assert(unum <= 1); assert(unum <= 1);
if (unum == 0) { if (unum == 0) {
@ -928,9 +954,8 @@ exception_handler:
// check if it's a StopIteration within a for block // check if it's a StopIteration within a for block
if (*code_state->ip == MP_BC_FOR_ITER && mp_obj_is_subclass_fast(mp_obj_get_type(nlr.ret_val), &mp_type_StopIteration)) { if (*code_state->ip == MP_BC_FOR_ITER && mp_obj_is_subclass_fast(mp_obj_get_type(nlr.ret_val), &mp_type_StopIteration)) {
const byte *ip = code_state->ip + 1; const byte *ip = code_state->ip + 1;
mp_uint_t unum;
DECODE_ULABEL; // the jump offset if iteration finishes; for labels are always forward DECODE_ULABEL; // the jump offset if iteration finishes; for labels are always forward
code_state->ip = ip + unum; // jump to after for-block code_state->ip = ip + ulab; // jump to after for-block
code_state->sp -= 1; // pop the exhausted iterator code_state->sp -= 1; // pop the exhausted iterator
goto outer_dispatch_loop; // continue with dispatch loop goto outer_dispatch_loop; // continue with dispatch loop
} }