/* * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * gmpy2_mpc.c * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * Python interface to the GMP or MPIR, MPFR, and MPC multiple precision * * libraries. * * * * Copyright 2000 - 2009 Alex Martelli * * * * Copyright 2008 - 2022 Case Van Horsen * * * * This file is part of GMPY2. * * * * GMPY2 is free software: you can redistribute it and/or modify it under * * the terms of the GNU Lesser General Public License as published by the * * Free Software Foundation, either version 3 of the License, or (at your * * option) any later version. * * * * GMPY2 is distributed in the hope that it will be useful, but WITHOUT * * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or * * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public * * License for more details. * * * * You should have received a copy of the GNU Lesser General Public * * License along with GMPY2; if not, see * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * * */ static void _GMPy_MPC_Cleanup(MPC_Object **v, CTXT_Object *ctext) { /* GMPY_MPC_CHECK_RANGE(V, CTX) */ { int rcr, rci; rcr = MPC_INEX_RE((*v)->rc); rci = MPC_INEX_IM((*v)->rc); if (mpfr_regular_p(mpc_realref((*v)->c)) && (!((mpc_realref((*v)->c)->_mpfr_exp >= ctext->ctx.emin) && (mpc_realref((*v)->c)->_mpfr_exp <= ctext->ctx.emax)))) { mpfr_exp_t _oldemin, _oldemax; _oldemin = mpfr_get_emin(); _oldemax = mpfr_get_emax(); mpfr_set_emin(ctext->ctx.emin); mpfr_set_emax(ctext->ctx.emax); rcr = mpfr_check_range(mpc_realref((*v)->c), rcr, GET_REAL_ROUND(ctext)); mpfr_set_emin(_oldemin); mpfr_set_emax(_oldemax); } if (mpfr_regular_p(mpc_imagref((*v)->c)) && (!((mpc_imagref((*v)->c)->_mpfr_exp >= ctext->ctx.emin) && (mpc_imagref((*v)->c)->_mpfr_exp <= ctext->ctx.emax)))) { mpfr_exp_t _oldemin, _oldemax; _oldemin = mpfr_get_emin(); _oldemax = mpfr_get_emax(); mpfr_set_emin(ctext->ctx.emin); mpfr_set_emax(ctext->ctx.emax); rci = mpfr_check_range(mpc_imagref((*v)->c), rci, GET_IMAG_ROUND(ctext)); mpfr_set_emin(_oldemin); mpfr_set_emax(_oldemax); } (*v)->rc = MPC_INEX(rcr, rci); } /* GMPY_MPC_SUBNORMALIZE(V, CTX) */ { int rcr, rci; rcr = MPC_INEX_RE((*v)->rc); rci = MPC_INEX_IM((*v)->rc); if (ctext->ctx.subnormalize && (!((mpc_realref((*v)->c)->_mpfr_exp >= ctext->ctx.emin) && (mpc_realref((*v)->c)->_mpfr_exp <= ctext->ctx.emin + mpfr_get_prec(mpc_realref((*v)->c)) - 2)))) { mpfr_exp_t _oldemin, _oldemax; _oldemin = mpfr_get_emin(); _oldemax = mpfr_get_emax(); mpfr_set_emin(ctext->ctx.emin); mpfr_set_emax(ctext->ctx.emax); rcr = mpfr_subnormalize(mpc_realref((*v)->c), rcr, GET_REAL_ROUND(ctext)); mpfr_set_emin(_oldemin); mpfr_set_emax(_oldemax); } if (ctext->ctx.subnormalize && (!((mpc_imagref((*v)->c)->_mpfr_exp >= ctext->ctx.emin) && (mpc_imagref((*v)->c)->_mpfr_exp <= ctext->ctx.emin + mpfr_get_prec(mpc_imagref((*v)->c)) - 2)))) { mpfr_exp_t _oldemin, _oldemax; _oldemin = mpfr_get_emin(); _oldemax = mpfr_get_emax(); mpfr_set_emin(ctext->ctx.emin); mpfr_set_emax(ctext->ctx.emax); rci = mpfr_check_range(mpc_imagref((*v)->c), rci, GET_IMAG_ROUND(ctext)); mpfr_set_emin(_oldemin); mpfr_set_emax(_oldemax); } (*v)->rc = MPC_INEX(rcr, rci); } /* GMPY_MPC_EXCEPTIONS(V, CTX) */ { int _invalid = 0, _underflow = 0, _overflow = 0, _inexact = 0; int rcr, rci; rcr = MPC_INEX_RE((*v)->rc); rci = MPC_INEX_IM((*v)->rc); if (MPC_IS_NAN_P(*v)) { ctext->ctx.invalid = 1; _invalid = 1; } if ((*v)->rc) { ctext->ctx.inexact = 1; _inexact = 1; } if ((rcr && mpfr_zero_p(mpc_realref((*v)->c))) || (rci && mpfr_zero_p(mpc_imagref((*v)->c)))) { ctext->ctx.underflow = 1; _underflow = 1; } if ((rcr && mpfr_inf_p(mpc_realref((*v)->c))) || (rci && mpfr_inf_p(mpc_imagref((*v)->c)))) { ctext->ctx.overflow = 1; _overflow = 1; } if (ctext->ctx.traps) { if ((ctext->ctx.traps & TRAP_UNDERFLOW) && _underflow) { \ GMPY_UNDERFLOW("underflow"); Py_XDECREF((PyObject*)(*v)); (*v) = NULL; } if ((ctext->ctx.traps & TRAP_OVERFLOW) && _overflow) { GMPY_OVERFLOW("overflow"); Py_XDECREF((PyObject*)(*v)); (*v) = NULL; } if ((ctext->ctx.traps & TRAP_INEXACT) && _inexact) { GMPY_INEXACT("inexact result"); Py_XDECREF((PyObject*)(*v)); (*v) = NULL; } if ((ctext->ctx.traps & TRAP_INVALID) && _invalid) { GMPY_INVALID("invalid operation"); Py_XDECREF((PyObject*)(*v)); (*v) = NULL; } } } } PyDoc_STRVAR(GMPy_doc_mpc, "mpc() -> mpc(0.0+0.0j)\n\n" " If no argument is given, return mpc(0.0+0.0j).\n\n" "mpc(c [, precision=0]) -> mpc\n\n" " Return a new 'mpc' object from an existing complex number (either\n" " a Python complex object or another 'mpc' object).\n\n" "mpc(real [,imag=0 [, precision=0]]) -> mpc\n\n" " Return a new 'mpc' object by converting two non-complex numbers\n" " into the real and imaginary components of an 'mpc' object.\n\n" "mpc(s [, precision=0 [, base=10]]) -> mpc\n\n" " Return a new 'mpc' object by converting a string s into a complex\n" " number. If base is omitted, then a base-10 representation is\n" " assumed otherwise the base must be in the interval [2,36].\n\n" "Note: The precision can be specified either a single number that\n" " is used for both the real and imaginary components, or as a\n" " tuple that can specify different precisions for the real\n" " and imaginary components.\n\n" " If a precision greater than or equal to 2 is specified, then it\n" " is used.\n\n" " A precision of 0 (the default) implies the precision of the\n" " current context is used.\n\n" " A precision of 1 minimizes the loss of precision by following\n" " these rules:\n" " 1) If n is a radix-2 floating point number, then the full\n" " precision of n is retained.\n" " 2) If n is an integer, then the precision is the bit length\n" " of the integer.\n" ); static PyMethodDef Pympc_methods[] = { { "__complex__", GMPy_PyComplex_From_MPC, METH_NOARGS, GMPy_doc_mpc_complex }, { "__format__", GMPy_MPC_Format, METH_VARARGS, GMPy_doc_mpc_format }, { "__sizeof__", GMPy_MPC_SizeOf_Method, METH_NOARGS, GMPy_doc_mpc_sizeof_method }, { "conjugate", GMPy_MPC_Conjugate_Method, METH_NOARGS, GMPy_doc_mpc_conjugate_method }, { "digits", GMPy_MPC_Digits_Method, METH_VARARGS, GMPy_doc_mpc_digits_method }, { "is_finite", GMPy_Number_Method_Is_Finite, METH_NOARGS, GMPy_doc_method_is_finite }, { "is_infinite", GMPy_Number_Method_Is_Infinite, METH_NOARGS, GMPy_doc_method_is_infinite }, { "is_nan", GMPy_Number_Method_Is_NAN, METH_NOARGS, GMPy_doc_method_is_nan }, { "is_zero", GMPy_Number_Method_Is_Zero, METH_NOARGS, GMPy_doc_method_is_zero }, { NULL, NULL, 1 } }; #ifdef PY3 static PyNumberMethods mpc_number_methods = { (binaryfunc) GMPy_Number_Add_Slot, /* nb_add */ (binaryfunc) GMPy_Number_Sub_Slot, /* nb_subtract */ (binaryfunc) GMPy_Number_Mul_Slot, /* nb_multiply */ (binaryfunc) GMPy_Number_Mod_Slot, /* nb_remainder */ (binaryfunc) GMPy_Number_DivMod_Slot, /* nb_divmod */ (ternaryfunc) GMPy_Number_Pow_Slot, /* nb_power */ (unaryfunc) GMPy_MPC_Minus_Slot, /* nb_negative */ (unaryfunc) GMPy_MPC_Plus_Slot, /* nb_positive */ (unaryfunc) GMPy_MPC_Abs_Slot, /* nb_absolute */ (inquiry) GMPy_MPC_NonZero_Slot, /* nb_bool */ 0, /* nb_invert */ 0, /* nb_lshift */ 0, /* nb_rshift */ 0, /* nb_and */ 0, /* nb_xor */ 0, /* nb_or */ (unaryfunc) GMPy_MPC_Int_Slot, /* nb_int */ 0, /* nb_reserved */ (unaryfunc) GMPy_MPC_Float_Slot, /* nb_float */ 0, /* nb_inplace_add */ 0, /* nb_inplace_subtract */ 0, /* nb_inplace_multiply */ 0, /* nb_inplace_remainder */ 0, /* nb_inplace_power */ 0, /* nb_inplace_lshift */ 0, /* nb_inplace_rshift */ 0, /* nb_inplace_and */ 0, /* nb_inplace_xor */ 0, /* nb_inplace_or */ (binaryfunc) GMPy_Number_FloorDiv_Slot, /* nb_floor_divide */ (binaryfunc) GMPy_Number_TrueDiv_Slot, /* nb_true_divide */ 0, /* nb_inplace_floor_divide */ 0, /* nb_inplace_true_divide */ 0, /* nb_index */ }; #else static PyNumberMethods mpc_number_methods = { (binaryfunc) GMPy_Number_Add_Slot, /* nb_add */ (binaryfunc) GMPy_Number_Sub_Slot, /* nb_subtract */ (binaryfunc) GMPy_Number_Mul_Slot, /* nb_multiply */ (binaryfunc) GMPy_Number_TrueDiv_Slot, /* nb_divide */ (binaryfunc) GMPy_Number_Mod_Slot, /* nb_remainder */ (binaryfunc) GMPy_Number_DivMod_Slot, /* nb_divmod */ (ternaryfunc) GMPy_Number_Pow_Slot, /* nb_power */ (unaryfunc) GMPy_MPC_Minus_Slot, /* nb_negative */ (unaryfunc) GMPy_MPC_Plus_Slot, /* nb_positive */ (unaryfunc) GMPy_MPC_Abs_Slot, /* nb_absolute */ (inquiry) GMPy_MPC_NonZero_Slot, /* nb_bool */ 0, /* nb_invert */ 0, /* nb_lshift */ 0, /* nb_rshift */ 0, /* nb_and */ 0, /* nb_xor */ 0, /* nb_or */ 0, /* nb_coerce */ (unaryfunc) GMPy_MPC_Int_Slot, /* nb_int */ (unaryfunc) GMPy_MPC_Long_Slot, /* nb_long */ (unaryfunc) GMPy_MPC_Float_Slot, /* nb_float */ 0, /* nb_oct */ 0, /* nb_hex */ 0, /* nb_inplace_add */ 0, /* nb_inplace_subtract */ 0, /* nb_inplace_multiply */ 0, /* nb_inplace_divide */ 0, /* nb_inplace_remainder */ 0, /* nb_inplace_power */ 0, /* nb_inplace_lshift */ 0, /* nb_inplace_rshift */ 0, /* nb_inplace_and */ 0, /* nb_inplace_xor */ 0, /* nb_inplace_or */ (binaryfunc) GMPy_Number_FloorDiv_Slot, /* nb_floor_divide */ (binaryfunc) GMPy_Number_TrueDiv_Slot, /* nb_true_divide */ 0, /* nb_inplace_floor_divide */ 0, /* nb_inplace_true_divide */ }; #endif static PyGetSetDef Pympc_getseters[] = { {"precision", (getter)GMPy_MPC_GetPrec_Attrib, NULL, "precision in bits", NULL}, {"rc", (getter)GMPy_MPC_GetRc_Attrib, NULL, "return code", NULL}, {"imag", (getter)GMPy_MPC_GetImag_Attrib, NULL, "imaginary component", NULL}, {"real", (getter)GMPy_MPC_GetReal_Attrib, NULL, "real component", NULL}, {NULL} }; static PyTypeObject MPC_Type = { /* PyObject_HEAD_INIT(&PyType_Type) */ #ifdef PY3 PyVarObject_HEAD_INIT(NULL, 0) #else PyObject_HEAD_INIT(0) 0, /* ob_size */ #endif "mpc", /* tp_name */ sizeof(MPC_Object), /* tp_basicsize */ 0, /* tp_itemsize */ /* methods */ (destructor) GMPy_MPC_Dealloc, /* tp_dealloc */ 0, /* tp_print */ 0, /* tp_getattr */ 0, /* tp_setattr */ 0, /* tp_reserved */ (reprfunc) GMPy_MPC_Repr_Slot, /* tp_repr */ &mpc_number_methods, /* tp_as_number */ 0, /* tp_as_sequence */ 0, /* tp_as_mapping */ (hashfunc) GMPy_MPC_Hash_Slot, /* tp_hash */ 0, /* tp_call */ (reprfunc) GMPy_MPC_Str_Slot, /* tp_str */ 0, /* tp_getattro */ 0, /* tp_setattro */ 0, /* tp_as_buffer */ #ifdef PY3 Py_TPFLAGS_DEFAULT, /* tp_flags */ #else Py_TPFLAGS_HAVE_RICHCOMPARE|Py_TPFLAGS_CHECKTYPES, /* tp_flags */ #endif GMPy_doc_mpc, /* tp_doc */ 0, /* tp_traverse */ 0, /* tp_clear */ (richcmpfunc)&GMPy_RichCompare_Slot, /* tp_richcompare */ 0, /* tp_weaklistoffset*/ 0, /* tp_iter */ 0, /* tp_iternext */ Pympc_methods, /* tp_methods */ 0, /* tp_members */ Pympc_getseters, /* tp_getset */ 0, /* tp_base */ 0, /* tp_dict */ 0, /* tp_descr_get */ 0, /* tp_descr_set */ 0, /* tp_dictoffset */ 0, /* tp_init */ 0, /* tp_alloc */ GMPy_MPC_NewInit, /* tp_new */ 0, /* tp_free */ };