catacomb/__init__.py, mp.c: Use implicit-conversion rules for rationals.

[catacomb-python] / mp.c

diff --git a/mp.c b/mp.c
index ce04819..3abbcd5 100644 (file)
--- a/mp.c
+++ b/mp.c
@@ -603,6 +603,22 @@ end:
   return ((PyObject *)zz);
 }
 
+#define IMPLICIT(pre)                                                  \
+  static PyObject *pre##meth__implicit(PyObject *me, PyObject *arg)    \
+  {                                                                    \
+    PyObject *x, *rc = 0;                                              \
+    mp *y = MP_NEW;                                                    \
+    if (!PyArg_ParseTuple(arg, "O:_implicit", &x)) goto end;           \
+    y = implicit##pre(x);                                              \
+    if (!y) TYERR("can't convert implicitly to " #pre);                        \
+    rc = pre##_pywrap(y);                                              \
+  end:                                                                 \
+    return (rc);                                                       \
+  }
+IMPLICIT(mp)
+IMPLICIT(gf)
+#undef IMPLICIT
+
 Py_hash_t mphash(mp *x)
 {
   PyObject *l = mp_topylong(x);
@@ -931,6 +947,7 @@ static const PyMethodDef mp_pymethods[] = {
     "  Parse STR as a large integer, according to RADIX.  If RADIX is\n"
     "  zero, read a prefix from STR to decide radix: allow `0b' for binary,\n"
     "  `0' or `0o' for octal, `0x' for hex, or `R_' for other radix R.")
+  SMTH (_implicit,     0)
   SMTH (factorial,     "factorial(I) -> I!: compute factorial")
   SMTH (fibonacci,     "fibonacci(I) -> F(I): compute Fibonacci number")
   SMTH (loadl,         "loadl(STR) -> X: read little-endian bytes")
@@ -2123,6 +2140,7 @@ static const PyMethodDef gf_pymethods[] = {
     "  Parse STR as a binary polynomial, according to RADIX.  If RADIX is\n"
     "  zero, read a prefix from STR to decide radix: allow `0b' for binary,\n"
     "  `0' or `0o' for octal, `0x' for hex, or `R_' for other radix R.")
+  SMTH (_implicit,     0)
   SMTH (loadl,         "loadl(STR) -> X: read little-endian bytes")
   SMTH (loadb,         "loadb(STR) -> X: read big-endian bytes")
   SMTH (frombuf,       "frombuf(STR) -> (X, REST): read buffer format")