TonBERRY/micropython
3
0
Fork 0
Code Pull Requests Activity

py/modbuiltins: Implement abs() by dispatching to MP_UNARY_OP_ABS.

Browse Source 
This allows user classes to implement __abs__ special method, and saves
code size (104 bytes for x86_64), even though during refactor, an issue
was fixed and few optimizations were made:

* abs() of minimum (negative) small int value is calculated properly.
* objint_longlong and objint_mpz avoid allocating new object is the
  argument is already non-negative.
This commit is contained in:
Paul Sokolovsky
2017-09-18 00:06:43 +03:00
parent 72491b3e40
commit 9dce823cfd
10 changed files with 45 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
py/objcomplex.c
View File

@@ -124,6 +124,11 @@ STATIC mp_obj_t complex_unary_op(mp_unary_op_t op, mp_obj_t o_in) {
case MP_UNARY_OP_HASH: return MP_OBJ_NEW_SMALL_INT(mp_float_hash(o->real) ^ mp_float_hash(o->imag));
case MP_UNARY_OP_POSITIVE: return o_in;
case MP_UNARY_OP_NEGATIVE: return mp_obj_new_complex(-o->real, -o->imag);
case MP_UNARY_OP_ABS: {
mp_float_t real, imag;
mp_obj_complex_get(o_in, &real, &imag);
return mp_obj_new_float(MICROPY_FLOAT_C_FUN(sqrt)(real*real + imag*imag));
}
default: return MP_OBJ_NULL; // op not supported
}
}