static PyObject *
Pympany_printf(PyObject *self, PyObject *args)
{
PyObject *result = 0, *x = 0;
char *buffer = 0, *fmtcode = 0;
int buflen;
void *generic;
if (!PyArg_ParseTuple(args, "sO", &fmtcode, &x))
return NULL;
if (CHECK_MPZANY(x) || Pympq_Check(x)) {
if (CHECK_MPZANY(x))
generic = Pympz_AS_MPZ(x);
else
generic = Pympq_AS_MPQ(x);
buflen = gmp_asprintf(&buffer, fmtcode, generic);
result = Py_BuildValue("s", buffer);
PyMem_Free(buffer);
return result;
}
#ifdef WITHMPFR
else if(Pympfr_Check(x)) {
generic = Pympfr_AS_MPFR(x);
buflen = mpfr_asprintf(&buffer, fmtcode, generic);
result = Py_BuildValue("s", buffer);
PyMem_Free(buffer);
return result;
}
#endif
else {
TYPE_ERROR("printf() requires a gmpy2 object as argument");
return NULL;
}
}
static int isOne(PyObject* obj)
{
int overflow = 0;
long temp;
if (!obj)
return 1;
if (Pympq_Check(obj)) {
return (0==mpz_cmp_ui(mpq_denref(Pympq_AS_MPQ(obj)),1)) &&
(0==mpz_cmp_ui(mpq_numref(Pympq_AS_MPQ(obj)),1));
}
else if (Pympz_Check(obj)) {
return 0==mpz_cmp_ui(Pympz_AS_MPZ(obj),1);
}
else if (Pyxmpz_Check(obj)) {
return 0==mpz_cmp_ui(Pyxmpz_AS_MPZ(obj),1);
#ifdef PY2
}
else if (PyInt_Check(obj)) {
return PyInt_AS_LONG(obj)==1;
#endif
}
#ifdef WITHMPFR
else if (Pympfr_Check(obj)) {
return mpfr_get_d(Pympfr_AS_MPFR(obj), context->ctx.mpfr_round)==1.0;
}
#endif
else if (PyFloat_Check(obj)) {
return PyFloat_AS_DOUBLE(obj)==1.0;
}
else if (PyLong_Check(obj)) {
temp = PyLong_AsLongAndOverflow(obj, &overflow);
if (!overflow && temp == 1)
return 1;
else
return 0;
}
return 0;
}
