kopia lustrzana https://github.com/micropython/micropython
objarray: Refactor to use array accessors from binary.c .
Paul Sokolovsky
rodzic
8bc3516389
commit
c203324e6c
30
py/binary.c
30
py/binary.c
|
|
@ -11,6 +11,36 @@
|
|||
|
||||
// Helpers to work with binary-encoded data
|
||||
|
||||
int mp_binary_get_size(char typecode) {
|
||||
// This assumes that unsigned and signed types are of the same type,
|
||||
// which is invariant for [u]intN_t.
|
||||
switch (typecode) {
|
||||
case BYTEARRAY_TYPECODE:
|
||||
case 'b':
|
||||
case 'B':
|
||||
return sizeof(int8_t);
|
||||
case 'h':
|
||||
case 'H':
|
||||
return sizeof(int16_t);
|
||||
case 'i':
|
||||
case 'I':
|
||||
return sizeof(int32_t);
|
||||
case 'l':
|
||||
case 'L':
|
||||
return sizeof(int32_t);
|
||||
case 'q':
|
||||
case 'Q':
|
||||
return sizeof(long long);
|
||||
#if MICROPY_ENABLE_FLOAT
|
||||
case 'f':
|
||||
return sizeof(float);
|
||||
case 'd':
|
||||
return sizeof(double);
|
||||
#endif
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
mp_obj_t mp_binary_get_val(char typecode, void *p, int index) {
|
||||
int val = 0;
|
||||
switch (typecode) {
|
||||
|
|
|
|||
110
py/objarray.c
110
py/objarray.c
|
|
@ -12,10 +12,7 @@
|
|||
#include "map.h"
|
||||
#include "runtime0.h"
|
||||
#include "runtime.h"
|
||||
|
||||
// Use special typecode to differentiate repr() of bytearray vs array.array('B')
|
||||
// (underlyingly they're same).
|
||||
#define BYTEARRAY_TYPECODE 0
|
||||
#include "binary.h"
|
||||
|
||||
typedef struct _mp_obj_array_t {
|
||||
mp_obj_base_t base;
|
||||
|
|
@ -36,91 +33,6 @@ STATIC mp_obj_t array_append(mp_obj_t self_in, mp_obj_t arg);
|
|||
/******************************************************************************/
|
||||
/* array */
|
||||
|
||||
STATIC machine_int_t array_get_el_size(char typecode) {
|
||||
// This assumes that unsigned and signed types are of the same type,
|
||||
// which is invariant for [u]intN_t.
|
||||
switch (typecode) {
|
||||
case BYTEARRAY_TYPECODE:
|
||||
case 'b':
|
||||
case 'B':
|
||||
return sizeof(int8_t);
|
||||
case 'h':
|
||||
case 'H':
|
||||
return sizeof(int16_t);
|
||||
case 'i':
|
||||
case 'I':
|
||||
return sizeof(int32_t);
|
||||
case 'l':
|
||||
case 'L':
|
||||
return sizeof(int32_t);
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
STATIC machine_int_t array_get_el(mp_obj_array_t *o, int index) {
|
||||
machine_int_t val = 0;
|
||||
switch (o->typecode) {
|
||||
case 'b':
|
||||
val = ((int8_t*)o->items)[index];
|
||||
break;
|
||||
case BYTEARRAY_TYPECODE:
|
||||
case 'B':
|
||||
val = ((uint8_t*)o->items)[index];
|
||||
break;
|
||||
case 'h':
|
||||
val = ((int16_t*)o->items)[index];
|
||||
break;
|
||||
case 'H':
|
||||
val = ((uint16_t*)o->items)[index];
|
||||
break;
|
||||
case 'i':
|
||||
val = ((int32_t*)o->items)[index];
|
||||
break;
|
||||
case 'I':
|
||||
val = ((uint32_t*)o->items)[index];
|
||||
break;
|
||||
case 'l':
|
||||
val = ((int32_t*)o->items)[index];
|
||||
break;
|
||||
case 'L':
|
||||
val = ((uint32_t*)o->items)[index];
|
||||
break;
|
||||
}
|
||||
return val;
|
||||
}
|
||||
|
||||
STATIC void array_set_el(mp_obj_array_t *o, int index, mp_obj_t val_in) {
|
||||
machine_int_t val = mp_obj_int_get(val_in);
|
||||
switch (o->typecode) {
|
||||
case 'b':
|
||||
((int8_t*)o->items)[index] = val;
|
||||
break;
|
||||
case BYTEARRAY_TYPECODE:
|
||||
case 'B':
|
||||
((uint8_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'h':
|
||||
((int16_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'H':
|
||||
((uint16_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'i':
|
||||
((int32_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'I':
|
||||
((uint32_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'l':
|
||||
((int32_t*)o->items)[index] = val;
|
||||
break;
|
||||
case 'L':
|
||||
((uint32_t*)o->items)[index] = val;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
STATIC void array_print(void (*print)(void *env, const char *fmt, ...), void *env, mp_obj_t o_in, mp_print_kind_t kind) {
|
||||
mp_obj_array_t *o = o_in;
|
||||
if (o->typecode == BYTEARRAY_TYPECODE) {
|
||||
|
|
@ -134,7 +46,7 @@ STATIC void array_print(void (*print)(void *env, const char *fmt, ...), void *en
|
|||
if (i > 0) {
|
||||
print(env, ", ");
|
||||
}
|
||||
print(env, "%d", array_get_el(o, i));
|
||||
mp_obj_print_helper(print, env, mp_binary_get_val(o->typecode, o->items, i), PRINT_REPR);
|
||||
}
|
||||
print(env, "]");
|
||||
}
|
||||
|
|
@ -161,7 +73,7 @@ STATIC mp_obj_t array_construct(char typecode, mp_obj_t initializer) {
|
|||
if (len == 0) {
|
||||
array_append(array, item);
|
||||
} else {
|
||||
array_set_el(array, i++, item);
|
||||
mp_binary_set_val(typecode, array->items, i++, item);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
@ -204,8 +116,7 @@ STATIC mp_obj_t array_binary_op(int op, mp_obj_t lhs, mp_obj_t rhs) {
|
|||
case RT_BINARY_OP_SUBSCR:
|
||||
{
|
||||
uint index = mp_get_index(o->base.type, o->len, rhs);
|
||||
machine_int_t val = array_get_el(o, index);
|
||||
return mp_obj_new_int(val);
|
||||
return mp_binary_get_val(o->typecode, o->items, index);
|
||||
}
|
||||
|
||||
default:
|
||||
|
|
@ -218,12 +129,12 @@ STATIC mp_obj_t array_append(mp_obj_t self_in, mp_obj_t arg) {
|
|||
assert(MP_OBJ_IS_TYPE(self_in, &array_type));
|
||||
mp_obj_array_t *self = self_in;
|
||||
if (self->free == 0) {
|
||||
int item_sz = array_get_el_size(self->typecode);
|
||||
int item_sz = mp_binary_get_size(self->typecode);
|
||||
// TODO: alloc policy
|
||||
self->free = 8;
|
||||
self->items = m_realloc(self->items, item_sz * self->len, item_sz * (self->len + self->free));
|
||||
}
|
||||
array_set_el(self, self->len++, arg);
|
||||
mp_binary_set_val(self->typecode, self->items, self->len++, arg);
|
||||
self->free--;
|
||||
return mp_const_none; // return None, as per CPython
|
||||
}
|
||||
|
|
@ -232,14 +143,14 @@ STATIC MP_DEFINE_CONST_FUN_OBJ_2(array_append_obj, array_append);
|
|||
STATIC bool array_store_item(mp_obj_t self_in, mp_obj_t index_in, mp_obj_t value) {
|
||||
mp_obj_array_t *o = self_in;
|
||||
uint index = mp_get_index(o->base.type, o->len, index_in);
|
||||
array_set_el(o, index, value);
|
||||
mp_binary_set_val(o->typecode, o->items, index, value);
|
||||
return true;
|
||||
}
|
||||
|
||||
STATIC machine_int_t array_get_buffer(mp_obj_t o_in, buffer_info_t *bufinfo, int flags) {
|
||||
mp_obj_array_t *o = o_in;
|
||||
bufinfo->buf = o->items;
|
||||
bufinfo->len = o->len * array_get_el_size(o->typecode);
|
||||
bufinfo->len = o->len * mp_binary_get_size(o->typecode);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
|
@ -267,7 +178,7 @@ STATIC mp_obj_array_t *array_new(char typecode, uint n) {
|
|||
o->typecode = typecode;
|
||||
o->free = 0;
|
||||
o->len = n;
|
||||
o->items = m_malloc(array_get_el_size(typecode) * o->len);
|
||||
o->items = m_malloc(mp_binary_get_size(typecode) * o->len);
|
||||
return o;
|
||||
}
|
||||
|
||||
|
|
@ -304,8 +215,7 @@ typedef struct _mp_obj_array_it_t {
|
|||
mp_obj_t array_it_iternext(mp_obj_t self_in) {
|
||||
mp_obj_array_it_t *self = self_in;
|
||||
if (self->cur < self->array->len) {
|
||||
machine_int_t val = array_get_el(self->array, self->cur++);
|
||||
return mp_obj_new_int(val);
|
||||
return mp_binary_get_val(self->array->typecode, self->array->items, self->cur++);
|
||||
} else {
|
||||
return mp_const_stop_iteration;
|
||||
}
|
||||
|
|
|
|||
Ładowanie…
Reference in New Issue