commit e9ce00d874ef6c93a56e726bb1503341f0448bf9
author Damien George <damien.p.george@gmail.com> Sat Jun 13 22:40:50 2015 +0100
committer Damien George <damien.p.george@gmail.com> Sat Jun 13 23:38:28 2015 +0100
tree 0ab159361f833c6af960fbf36728f8ec307a35e6
parent c5029bcbf37fd1a3fb145d9fa9e4a5094478f17b

py: Implement divmod for mpz bignum.

py/objint_mpz.c

diff --git a/py/objint_mpz.c b/py/objint_mpz.c
index 936e2cb..27f1ddb 100644
--- a/py/objint_mpz.c
+++ b/py/objint_mpz.c
@@ -262,6 +262,17 @@
                 mpz_pow_inpl(&res->mpz, zlhs, zrhs);
                 break;
 
+            case MP_BINARY_OP_DIVMOD: {
+                mp_obj_int_t *quo = mp_obj_int_new_mpz();
+                mpz_divmod_inpl(&quo->mpz, &res->mpz, zlhs, zrhs);
+                // Check signs and do Python style modulo
+                if (zlhs->neg != zrhs->neg) {
+                    mpz_add_inpl(&res->mpz, &res->mpz, zrhs);
+                }
+                mp_obj_t tuple[2] = {quo, res};
+                return mp_obj_new_tuple(2, tuple);
+            }
+
             default:
                 return MP_OBJ_NULL; // op not supported
         }