py: Add mp_binary_set_val_array_from_int, to store an int directly.

This commit is contained in:
Damien George 2014-04-18 23:28:12 +01:00
parent b11b85adaa
commit 71e9bfa20d
2 changed files with 16 additions and 5 deletions

View File

@ -151,11 +151,21 @@ mp_obj_t mp_binary_get_val(char struct_type, char val_type, byte **ptr) {
}
void mp_binary_set_val_array(char typecode, void *p, int index, mp_obj_t val_in) {
machine_int_t val = 0;
if (MP_OBJ_IS_INT(val_in)) {
val = mp_obj_int_get(val_in);
switch (typecode) {
#if MICROPY_ENABLE_FLOAT
case 'f':
((float*)p)[index] = mp_obj_float_get(val_in);
break;
case 'd':
((double*)p)[index] = mp_obj_float_get(val_in);
break;
#endif
default:
mp_binary_set_val_array_from_int(typecode, p, index, mp_obj_get_int(val_in));
}
}
void mp_binary_set_val_array_from_int(char typecode, void *p, int index, machine_int_t val) {
switch (typecode) {
case 'b':
((int8_t*)p)[index] = val;
@ -187,10 +197,10 @@ void mp_binary_set_val_array(char typecode, void *p, int index, mp_obj_t val_in)
#endif
#if MICROPY_ENABLE_FLOAT
case 'f':
((float*)p)[index] = mp_obj_float_get(val_in);
((float*)p)[index] = val;
break;
case 'd':
((double*)p)[index] = mp_obj_float_get(val_in);
((double*)p)[index] = val;
break;
#endif
}

View File

@ -6,3 +6,4 @@ int mp_binary_get_size(char typecode);
mp_obj_t mp_binary_get_val_array(char typecode, void *p, int index);
mp_obj_t mp_binary_get_val(char struct_type, char val_type, byte **ptr);
void mp_binary_set_val_array(char typecode, void *p, int index, mp_obj_t val_in);
void mp_binary_set_val_array_from_int(char typecode, void *p, int index, machine_int_t val);