You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
314 lines
9.3 KiB
314 lines
9.3 KiB
/*
|
|
* This file is part of the MicroPython project, http://micropython.org/
|
|
*
|
|
* The MIT License (MIT)
|
|
*
|
|
* Copyright (c) 2018 Paul Sokolovsky
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*/
|
|
|
|
#include <unistd.h> // for ssize_t
|
|
|
|
#include "py/runtime.h"
|
|
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE
|
|
|
|
typedef struct _mp_obj_deque_t {
|
|
mp_obj_base_t base;
|
|
size_t alloc;
|
|
size_t i_get;
|
|
size_t i_put;
|
|
mp_obj_t *items;
|
|
uint32_t flags;
|
|
#define FLAG_CHECK_OVERFLOW 1
|
|
} mp_obj_deque_t;
|
|
|
|
static mp_obj_t mp_obj_deque_append(mp_obj_t self_in, mp_obj_t arg);
|
|
static mp_obj_t mp_obj_deque_extend(mp_obj_t self_in, mp_obj_t arg_in);
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE_ITER
|
|
static mp_obj_t mp_obj_new_deque_it(mp_obj_t deque, mp_obj_iter_buf_t *iter_buf);
|
|
#endif
|
|
|
|
static mp_obj_t deque_make_new(const mp_obj_type_t *type, size_t n_args, size_t n_kw, const mp_obj_t *args) {
|
|
mp_arg_check_num(n_args, n_kw, 2, 3, false);
|
|
|
|
// Protect against -1 leading to zero-length allocation and bad array access
|
|
mp_int_t maxlen = mp_obj_get_int(args[1]);
|
|
if (maxlen < 0) {
|
|
mp_raise_ValueError(NULL);
|
|
}
|
|
|
|
mp_obj_deque_t *o = mp_obj_malloc(mp_obj_deque_t, type);
|
|
o->alloc = maxlen + 1;
|
|
o->i_get = o->i_put = 0;
|
|
o->items = m_new0(mp_obj_t, o->alloc);
|
|
|
|
if (n_args > 2) {
|
|
o->flags = mp_obj_get_int(args[2]);
|
|
}
|
|
|
|
mp_obj_deque_extend(MP_OBJ_FROM_PTR(o), args[0]);
|
|
|
|
return MP_OBJ_FROM_PTR(o);
|
|
}
|
|
|
|
static size_t deque_len(mp_obj_deque_t *self) {
|
|
ssize_t len = self->i_put - self->i_get;
|
|
if (len < 0) {
|
|
len += self->alloc;
|
|
}
|
|
return len;
|
|
}
|
|
|
|
static mp_obj_t deque_unary_op(mp_unary_op_t op, mp_obj_t self_in) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
switch (op) {
|
|
case MP_UNARY_OP_BOOL:
|
|
return mp_obj_new_bool(self->i_get != self->i_put);
|
|
case MP_UNARY_OP_LEN:
|
|
return MP_OBJ_NEW_SMALL_INT(deque_len(self));
|
|
|
|
#if MICROPY_PY_SYS_GETSIZEOF
|
|
case MP_UNARY_OP_SIZEOF: {
|
|
size_t sz = sizeof(*self) + sizeof(mp_obj_t) * self->alloc;
|
|
return MP_OBJ_NEW_SMALL_INT(sz);
|
|
}
|
|
#endif
|
|
default:
|
|
return MP_OBJ_NULL; // op not supported
|
|
}
|
|
}
|
|
|
|
static mp_obj_t mp_obj_deque_append(mp_obj_t self_in, mp_obj_t arg) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
size_t new_i_put = self->i_put + 1;
|
|
if (new_i_put == self->alloc) {
|
|
new_i_put = 0;
|
|
}
|
|
|
|
if (self->flags & FLAG_CHECK_OVERFLOW && new_i_put == self->i_get) {
|
|
mp_raise_msg(&mp_type_IndexError, MP_ERROR_TEXT("full"));
|
|
}
|
|
|
|
self->items[self->i_put] = arg;
|
|
self->i_put = new_i_put;
|
|
|
|
if (self->i_get == new_i_put) {
|
|
if (++self->i_get == self->alloc) {
|
|
self->i_get = 0;
|
|
}
|
|
}
|
|
|
|
return mp_const_none;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_2(deque_append_obj, mp_obj_deque_append);
|
|
|
|
static mp_obj_t mp_obj_deque_appendleft(mp_obj_t self_in, mp_obj_t arg) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
size_t new_i_get = self->i_get - 1;
|
|
if (self->i_get == 0) {
|
|
new_i_get = self->alloc - 1;
|
|
}
|
|
|
|
if (self->flags & FLAG_CHECK_OVERFLOW && new_i_get == self->i_put) {
|
|
mp_raise_msg(&mp_type_IndexError, MP_ERROR_TEXT("full"));
|
|
}
|
|
|
|
self->i_get = new_i_get;
|
|
self->items[self->i_get] = arg;
|
|
|
|
// overwriting first element in deque
|
|
if (self->i_put == new_i_get) {
|
|
if (self->i_put == 0) {
|
|
self->i_put = self->alloc - 1;
|
|
} else {
|
|
self->i_put--;
|
|
}
|
|
}
|
|
|
|
return mp_const_none;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_2(deque_appendleft_obj, mp_obj_deque_appendleft);
|
|
|
|
static mp_obj_t mp_obj_deque_extend(mp_obj_t self_in, mp_obj_t arg_in) {
|
|
mp_obj_iter_buf_t iter_buf;
|
|
mp_obj_t iter = mp_getiter(arg_in, &iter_buf);
|
|
mp_obj_t item;
|
|
while ((item = mp_iternext(iter)) != MP_OBJ_STOP_ITERATION) {
|
|
mp_obj_deque_append(self_in, item);
|
|
}
|
|
return mp_const_none;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_2(deque_extend_obj, mp_obj_deque_extend);
|
|
|
|
static mp_obj_t deque_popleft(mp_obj_t self_in) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
if (self->i_get == self->i_put) {
|
|
mp_raise_msg(&mp_type_IndexError, MP_ERROR_TEXT("empty"));
|
|
}
|
|
|
|
mp_obj_t ret = self->items[self->i_get];
|
|
self->items[self->i_get] = MP_OBJ_NULL;
|
|
|
|
if (++self->i_get == self->alloc) {
|
|
self->i_get = 0;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_1(deque_popleft_obj, deque_popleft);
|
|
|
|
static mp_obj_t deque_pop(mp_obj_t self_in) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
if (self->i_get == self->i_put) {
|
|
mp_raise_msg(&mp_type_IndexError, MP_ERROR_TEXT("empty"));
|
|
}
|
|
|
|
if (self->i_put == 0) {
|
|
self->i_put = self->alloc - 1;
|
|
} else {
|
|
self->i_put--;
|
|
}
|
|
|
|
mp_obj_t ret = self->items[self->i_put];
|
|
self->items[self->i_put] = MP_OBJ_NULL;
|
|
|
|
return ret;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_1(deque_pop_obj, deque_pop);
|
|
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE_SUBSCR
|
|
static mp_obj_t deque_subscr(mp_obj_t self_in, mp_obj_t index, mp_obj_t value) {
|
|
if (value == MP_OBJ_NULL) {
|
|
// delete not supported, fall back to mp_obj_subscr() error message
|
|
return MP_OBJ_NULL;
|
|
}
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
|
|
size_t offset = mp_get_index(self->base.type, deque_len(self), index, false);
|
|
size_t index_val = self->i_get + offset;
|
|
if (index_val > self->alloc) {
|
|
index_val -= self->alloc;
|
|
}
|
|
|
|
if (value == MP_OBJ_SENTINEL) {
|
|
// load
|
|
return self->items[index_val];
|
|
} else {
|
|
// store into deque
|
|
self->items[index_val] = value;
|
|
return mp_const_none;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
#if 0
|
|
static mp_obj_t deque_clear(mp_obj_t self_in) {
|
|
mp_obj_deque_t *self = MP_OBJ_TO_PTR(self_in);
|
|
self->i_get = self->i_put = 0;
|
|
mp_seq_clear(self->items, 0, self->alloc, sizeof(*self->items));
|
|
return mp_const_none;
|
|
}
|
|
static MP_DEFINE_CONST_FUN_OBJ_1(deque_clear_obj, deque_clear);
|
|
#endif
|
|
|
|
static const mp_rom_map_elem_t deque_locals_dict_table[] = {
|
|
{ MP_ROM_QSTR(MP_QSTR_append), MP_ROM_PTR(&deque_append_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_appendleft), MP_ROM_PTR(&deque_appendleft_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_extend), MP_ROM_PTR(&deque_extend_obj) },
|
|
#if 0
|
|
{ MP_ROM_QSTR(MP_QSTR_clear), MP_ROM_PTR(&deque_clear_obj) },
|
|
#endif
|
|
{ MP_ROM_QSTR(MP_QSTR_pop), MP_ROM_PTR(&deque_pop_obj) },
|
|
{ MP_ROM_QSTR(MP_QSTR_popleft), MP_ROM_PTR(&deque_popleft_obj) },
|
|
};
|
|
|
|
static MP_DEFINE_CONST_DICT(deque_locals_dict, deque_locals_dict_table);
|
|
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE_ITER
|
|
#define DEQUE_TYPE_FLAGS MP_TYPE_FLAG_ITER_IS_GETITER
|
|
#define DEQUE_TYPE_ITER iter, mp_obj_new_deque_it,
|
|
#else
|
|
#define DEQUE_TYPE_FLAGS MP_TYPE_FLAG_NONE
|
|
#define DEQUE_TYPE_ITER
|
|
#endif
|
|
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE_SUBSCR
|
|
#define DEQUE_TYPE_SUBSCR subscr, deque_subscr,
|
|
#else
|
|
#define DEQUE_TYPE_SUBSCR
|
|
#endif
|
|
|
|
MP_DEFINE_CONST_OBJ_TYPE(
|
|
mp_type_deque,
|
|
MP_QSTR_deque,
|
|
MP_TYPE_FLAG_ITER_IS_GETITER,
|
|
make_new, deque_make_new,
|
|
unary_op, deque_unary_op,
|
|
DEQUE_TYPE_SUBSCR
|
|
DEQUE_TYPE_ITER
|
|
locals_dict, &deque_locals_dict
|
|
);
|
|
|
|
/******************************************************************************/
|
|
/* deque iterator */
|
|
|
|
#if MICROPY_PY_COLLECTIONS_DEQUE_ITER
|
|
|
|
typedef struct _mp_obj_deque_it_t {
|
|
mp_obj_base_t base;
|
|
mp_fun_1_t iternext;
|
|
mp_obj_t deque;
|
|
size_t cur;
|
|
} mp_obj_deque_it_t;
|
|
|
|
static mp_obj_t deque_it_iternext(mp_obj_t self_in) {
|
|
mp_obj_deque_it_t *self = MP_OBJ_TO_PTR(self_in);
|
|
mp_obj_deque_t *deque = MP_OBJ_TO_PTR(self->deque);
|
|
if (self->cur != deque->i_put) {
|
|
mp_obj_t o_out = deque->items[self->cur];
|
|
if (++self->cur == deque->alloc) {
|
|
self->cur = 0;
|
|
}
|
|
return o_out;
|
|
} else {
|
|
return MP_OBJ_STOP_ITERATION;
|
|
}
|
|
}
|
|
|
|
static mp_obj_t mp_obj_new_deque_it(mp_obj_t deque, mp_obj_iter_buf_t *iter_buf) {
|
|
mp_obj_deque_t *deque_ = MP_OBJ_TO_PTR(deque);
|
|
size_t i_get = deque_->i_get;
|
|
assert(sizeof(mp_obj_deque_it_t) <= sizeof(mp_obj_iter_buf_t));
|
|
mp_obj_deque_it_t *o = (mp_obj_deque_it_t *)iter_buf;
|
|
o->base.type = &mp_type_polymorph_iter;
|
|
o->iternext = deque_it_iternext;
|
|
o->deque = deque;
|
|
o->cur = i_get;
|
|
return MP_OBJ_FROM_PTR(o);
|
|
}
|
|
|
|
#endif
|
|
|
|
#endif // MICROPY_PY_COLLECTIONS_DEQUE
|
|
|