From ac0134d427677438ec07f76f9b28b3f514cebde7 Mon Sep 17 00:00:00 2001 From: Paul Sokolovsky Date: Mon, 10 Feb 2014 07:10:55 +0200 Subject: Factor out mp_seq_count_obj() and implement tuple.count(). --- py/sequence.c | 12 ++++++++++++ 1 file changed, 12 insertions(+) (limited to 'py/sequence.c') diff --git a/py/sequence.c b/py/sequence.c index b3fb9fcd3..d8cfd9f3e 100644 --- a/py/sequence.c +++ b/py/sequence.c @@ -164,3 +164,15 @@ mp_obj_t mp_seq_index_obj(const mp_obj_t *items, uint len, uint n_args, const mp nlr_jump(mp_obj_new_exception_msg(MP_QSTR_ValueError, "object not in sequence")); } + +mp_obj_t mp_seq_count_obj(const mp_obj_t *items, uint len, mp_obj_t value) { + uint count = 0; + for (uint i = 0; i < len; i++) { + if (mp_obj_equal(items[i], value)) { + count++; + } + } + + // Common sense says this cannot overflow small int + return MP_OBJ_NEW_SMALL_INT(count); +} -- cgit v1.2.3