static PyObject *
GMPy_Complex_Is_NAN(PyObject *x, CTXT_Object *context)
{
MPC_Object *tempx;
int res;
if (MPC_Check(x)) {
res = MPC_IS_NAN_P(x);
}
else {
CHECK_CONTEXT(context);
if (!(tempx = GMPy_MPC_From_Complex(x, 1, 1, context))) {
return NULL;
}
res = MPC_IS_NAN_P(tempx);
Py_DECREF((PyObject*)tempx);
}
if (res) {
Py_RETURN_TRUE;
}
else {
Py_RETURN_FALSE;
}
}
static PyObject *
GMPy_Real_Pow(PyObject *base, PyObject *exp, PyObject *mod, CTXT_Object *context)
{
MPFR_Object *tempb = NULL, *tempe = NULL, *result = NULL;
MPZ_Object *tempz = NULL;
MPC_Object *mpc_result = NULL;
if (mod != Py_None) {
TYPE_ERROR("pow() 3rd argument not allowed unless all arguments are integers");
return NULL;
}
CHECK_CONTEXT(context);
result = GMPy_MPFR_New(0, context);
tempb = GMPy_MPFR_From_Real(base, 1, context);
if (!result || !tempb) {
goto err;
}
mpfr_clear_flags();
if (PyIntOrLong_Check(exp)) {
int error;
long temp_exp = GMPy_Integer_AsLongAndError(exp, &error);
if (!error) {
result->rc = mpfr_pow_si(result->f, tempb->f, temp_exp, GET_MPFR_ROUND(context));
}
else {
mpz_t tempzz;
mpz_inoc(tempzz);
mpz_set_PyIntOrLong(tempzz, exp);
result->rc = mpfr_pow_z(result->f, tempb->f, tempzz, GET_MPFR_ROUND(context));
mpz_cloc(tempzz);
}
}
else if (IS_INTEGER(exp)) {
if (!(tempz = GMPy_MPZ_From_Integer(exp, context))) {
goto err;
}
result->rc = mpfr_pow_z(result->f, tempb->f, tempz->z, GET_MPFR_ROUND(context));
}
else {
if (!(tempe = GMPy_MPFR_From_Real(exp, 1, context))) {
goto err;
}
result->rc = mpfr_pow(result->f, tempb->f, tempe->f, GET_MPFR_ROUND(context));
}
/* If the result is NaN, check if a complex result works. */
if (result && mpfr_nanflag_p() && context->ctx.allow_complex) {
mpc_result = (MPC_Object*)GMPy_Complex_Pow(base, exp, Py_None, context);
if (!mpc_result || MPC_IS_NAN_P(mpc_result)) {
Py_XDECREF((PyObject*)mpc_result);
context->ctx.invalid = 1;
GMPY_INVALID("pow() invalid operation");
goto err;
}
/* return a valid complex result */
Py_XDECREF((PyObject*)tempe);
Py_XDECREF((PyObject*)tempz);
Py_XDECREF((PyObject*)tempb);
Py_XDECREF((PyObject*)result);
return (PyObject*)mpc_result;
}
GMPY_MPFR_CLEANUP(result, context, "pow()");
Py_XDECREF((PyObject*)tempz);
Py_XDECREF((PyObject*)tempe);
Py_XDECREF((PyObject*)tempb);
return (PyObject*)result;
err:
Py_XDECREF((PyObject*)result);
Py_XDECREF((PyObject*)tempz);
Py_XDECREF((PyObject*)tempe);
Py_XDECREF((PyObject*)tempb);
return NULL;
}
