summaryrefslogtreecommitdiffstatshomepage
diff options
context:
space:
mode:
authorPaul Sokolovsky <pfalcon@users.sourceforge.net>2014-01-20 18:37:30 +0200
committerPaul Sokolovsky <pfalcon@users.sourceforge.net>2014-01-20 18:42:08 +0200
commitdff3f896d7f7ef20762f0be7c8d5aa457c04b71a (patch)
treeb0b6ed8d244e498fd2f3bd9824445d5f3ce6a3cd
parent3754c4a04035d18b0bbb780fa0aef273e1c8f033 (diff)
downloadmicropython-dff3f896d7f7ef20762f0be7c8d5aa457c04b71a.tar.gz
micropython-dff3f896d7f7ef20762f0be7c8d5aa457c04b71a.zip
mp_identity(): Add generic identity function.
Useful as getiter method for objects which are their own iterators, etc.
-rw-r--r--py/obj.c6
-rw-r--r--py/obj.h2
2 files changed, 8 insertions, 0 deletions
diff --git a/py/obj.c b/py/obj.c
index dd081ee35e..7dd9cc0a24 100644
--- a/py/obj.c
+++ b/py/obj.c
@@ -288,3 +288,9 @@ mp_obj_t mp_obj_len_maybe(mp_obj_t o_in) {
}
return MP_OBJ_NEW_SMALL_INT(len);
}
+
+// Return input argument. Useful as .getiter for objects which are
+// their own iterators, etc.
+mp_obj_t mp_identity(mp_obj_t self) {
+ return self;
+}
diff --git a/py/obj.h b/py/obj.h
index faf0231474..d7154a743c 100644
--- a/py/obj.h
+++ b/py/obj.h
@@ -351,6 +351,8 @@ extern const mp_obj_type_t fun_native_type;
extern const mp_obj_type_t fun_bc_type;
void mp_obj_fun_bc_get(mp_obj_t self_in, int *n_args, uint *n_state, const byte **code);
+mp_obj_t mp_identity(mp_obj_t self);
+
// generator
extern const mp_obj_type_t gen_instance_type;