static PyObject *
GMPy_MPZ_Xor_Slot(PyObject *self, PyObject *other)
{
MPZ_Object *result;
if (CHECK_MPZANY(self)) {
if (CHECK_MPZANY(other)) {
if (!(result = GMPy_MPZ_New(NULL)))
return NULL;
mpz_xor(result->z, MPZ(self), MPZ(other));
}
else {
if (!(result = GMPy_MPZ_From_Integer(other, NULL)))
return NULL;
mpz_xor(result->z, MPZ(self), result->z);
}
}
else if (CHECK_MPZANY(other)) {
if (!(result = GMPy_MPZ_From_Integer(self, NULL)))
return NULL;
mpz_xor(result->z, result->z, MPZ(other));
}
else {
Py_RETURN_NOTIMPLEMENTED;
}
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Method_IsCongruent(PyObject *self, PyObject *args)
{
int res;
MPZ_Object *tempy = NULL, *tempm = NULL;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("is_congruent() requires 2 integer arguments");
return NULL;
}
if (!(tempy = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL)) ||
!(tempm = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL))) {
Py_XDECREF((PyObject*)tempy);
Py_XDECREF((PyObject*)tempm);
TYPE_ERROR("is_congruent() requires 2 integer arguments");
return NULL;
}
res = mpz_congruent_p(MPZ(self), tempy->z, tempm->z);
Py_DECREF((PyObject*)tempy);
Py_DECREF((PyObject*)tempm);
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
static PyObject *
GMPy_MPZ_hamdist(PyObject *self, PyObject *args)
{
PyObject *result = NULL;
MPZ_Object *tempx = NULL, *tempy = NULL;
if (PyTuple_GET_SIZE(args) != 2)
goto err;
tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL);
tempy = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL);
if (!tempx || !tempy)
goto err;
result = PyIntOrLong_FromMpBitCnt(mpz_hamdist(tempx->z, tempy->z));
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
return result;
err:
TYPE_ERROR("hamdist() requires 'mpz','mpz' arguments");
Py_XDECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)tempy);
return NULL;
}
static PyObject *
GMPy_MPZ_Function_Legendre(PyObject *self, PyObject *args)
{
MPZ_Object *tempx = NULL, *tempy = NULL;
long res;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("legendre() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL)) ||
!(tempy = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL))) {
Py_XDECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)tempy);
return NULL;
}
if (mpz_sgn(tempy->z) <= 0 || mpz_even_p(tempy->z)) {
VALUE_ERROR("y must be odd, prime, and >0");
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
return NULL;
}
res = (long)(mpz_legendre(tempx->z, tempy->z));
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
return PyIntOrLong_FromLong(res);
}
static PyObject *
GMPy_MPZ_Function_GCDext(PyObject *self, PyObject *args)
{
PyObject *arg0, *arg1, *result = NULL;
MPZ_Object *g = NULL, *s = NULL, *t = NULL, *tempa = NULL, *tempb = NULL;
if(PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("gcdext() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(result = PyTuple_New(3)) ||
!(g = GMPy_MPZ_New(NULL)) ||
!(s = GMPy_MPZ_New(NULL)) ||
!(t = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
Py_XDECREF((PyObject*)g);
Py_XDECREF((PyObject*)s);
Py_XDECREF((PyObject*)t);
Py_XDECREF(result);
return NULL;
/* LCOV_EXCL_STOP */
}
arg0 = PyTuple_GET_ITEM(args, 0);
arg1 = PyTuple_GET_ITEM(args, 1);
if (MPZ_Check(arg0) && MPZ_Check(arg1)) {
mpz_gcdext(g->z, s->z, t->z, MPZ(arg0), MPZ(arg1));
}
else {
if(!(tempa = GMPy_MPZ_From_Integer(arg0, NULL)) ||
!(tempb = GMPy_MPZ_From_Integer(arg1, NULL))) {
TYPE_ERROR("gcdext() requires 'mpz','mpz' arguments");
Py_XDECREF((PyObject*)tempa);
Py_XDECREF((PyObject*)tempb);
Py_DECREF((PyObject*)g);
Py_DECREF((PyObject*)s);
Py_DECREF((PyObject*)t);
Py_DECREF(result);
return NULL;
}
mpz_gcdext(g->z, s->z, t->z, tempa->z, tempb->z);
Py_DECREF((PyObject*)tempa);
Py_DECREF((PyObject*)tempb);
}
PyTuple_SET_ITEM(result, 0, (PyObject*)g);
PyTuple_SET_ITEM(result, 1, (PyObject*)s);
PyTuple_SET_ITEM(result, 2, (PyObject*)t);
return result;
}
static PyObject *
GMPy_MPZ_Function_Remove(PyObject *self, PyObject *args)
{
MPZ_Object *result = NULL, *tempx = NULL, *tempf = NULL;
PyObject *x, *f;
size_t multiplicity;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("remove() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(result = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
return NULL;
/* LCOV_EXCL_STOP */
}
x = PyTuple_GET_ITEM(args, 0);
f = PyTuple_GET_ITEM(args, 1);
if (MPZ_Check(x) && MPZ_Check(f)) {
if (mpz_cmp_si(MPZ(f), 2) < 0) {
VALUE_ERROR("factor must be > 1");
Py_DECREF((PyObject*)result);
return NULL;
}
multiplicity = mpz_remove(result->z, MPZ(x), MPZ(f));
return Py_BuildValue("(Nk)", result, multiplicity);
}
else {
if (!(tempx = GMPy_MPZ_From_Integer(x, NULL)) ||
!(tempf = GMPy_MPZ_From_Integer(f, NULL))) {
TYPE_ERROR("remove() requires 'mpz','mpz' arguments");
Py_XDECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)tempf);
Py_DECREF((PyObject*)result);
return NULL;
}
if (mpz_cmp_si(MPZ(tempf), 2) < 0) {
VALUE_ERROR("factor must be > 1");
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempf);
Py_DECREF((PyObject*)result);
return NULL;
}
multiplicity = mpz_remove(result->z, tempx->z, tempf->z);
return Py_BuildValue("(Nk)", result, multiplicity);
}
}
static PyObject *
GMPy_MPZ_Function_Divexact(PyObject *self, PyObject *args)
{
PyObject *x, *y;
MPZ_Object *result, *tempx= NULL, *tempy = NULL;
if(PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("divexact() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(result = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
return NULL;
/* LCOV_EXCL_STOP */
}
x = PyTuple_GET_ITEM(args, 0);
y = PyTuple_GET_ITEM(args, 1);
if (MPZ_Check(x) && MPZ_Check(y)) {
if (mpz_sgn(MPZ(y)) == 0) {
ZERO_ERROR("divexact() division by 0");
Py_DECREF((PyObject*)result);
return NULL;
}
mpz_divexact(result->z, MPZ(x), MPZ(y));
}
else {
if (!(tempx = GMPy_MPZ_From_Integer(x, NULL)) ||
!(tempy = GMPy_MPZ_From_Integer(y, NULL))) {
TYPE_ERROR("divexact() requires 'mpz','mpz' arguments");
Py_XDECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)tempy);
Py_DECREF((PyObject*)result);
return NULL;
}
if (mpz_sgn(MPZ(tempy)) == 0) {
ZERO_ERROR("divexact() division by 0");
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
Py_DECREF((PyObject*)result);
return NULL;
}
mpz_divexact(result->z, tempx->z, tempy->z);
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
}
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Method_IsDivisible(PyObject *self, PyObject *other)
{
unsigned long temp;
int error, res;
MPZ_Object *tempd;
temp = GMPy_Integer_AsUnsignedLongAndError(other, &error);
if (!error) {
res = mpz_divisible_ui_p(MPZ(self), temp);
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
if (!(tempd = GMPy_MPZ_From_Integer(other, NULL))) {
TYPE_ERROR("is_divisible() requires integer argument");
return NULL;
}
res = mpz_divisible_p(MPZ(self), tempd->z);
Py_DECREF((PyObject*)tempd);
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
static PyObject *
GMPy_RandomState_Factory(PyObject *self, PyObject *args)
{
RandomState_Object *result;
MPZ_Object *temp;
if (!(result = GMPy_RandomState_New())) {
return NULL;
}
if (PyTuple_GET_SIZE(args) == 0) {
gmp_randseed_ui(result->state, 0);
}
else if (PyTuple_GET_SIZE(args) == 1) {
if (!(temp = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args,0), NULL))) {
Py_DECREF((PyObject*)result);
TYPE_ERROR("seed must be an integer");
return NULL;
}
gmp_randseed(result->state, temp->z);
Py_DECREF((PyObject*)temp);
}
else {
Py_DECREF((PyObject*)result);
TYPE_ERROR("random_state() requires 0 or 1 integer arguments");
return NULL;
}
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Function_NumDigits(PyObject *self, PyObject *args)
{
long base = 10;
Py_ssize_t argc;
MPZ_Object *temp;
PyObject *result;
argc = PyTuple_GET_SIZE(args);
if (argc == 0 || argc > 2) {
TYPE_ERROR("num_digits() requires 'mpz',['int'] arguments");
return NULL;
}
if (argc == 2) {
base = PyIntOrLong_AsLong(PyTuple_GET_ITEM(args, 1));
if (base == -1 && PyErr_Occurred()) {
return NULL;
}
}
if ((base < 2) || (base > 62)) {
VALUE_ERROR("base must be in the interval [2, 62]");
return NULL;
}
if (!(temp = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
return NULL;
}
result = PyIntOrLong_FromSize_t(mpz_sizeinbase(temp->z, (int)base));
Py_DECREF((PyObject*)temp);
return result;
}
static PyObject *
GMPy_Integer_Abs(PyObject *x, CTXT_Object *context)
{
MPZ_Object *result = NULL;
if (MPZ_Check(x)) {
if (mpz_sgn(MPZ(x)) >= 0) {
Py_INCREF(x);
return x;
}
else {
if ((result = GMPy_MPZ_New(context)))
mpz_abs(result->z, MPZ(x));
return (PyObject*)result;
}
}
/* This is safe because result is not an incremented reference to an
* existing value. Why?
* 1) No values are interned like Python's integers.
* 2) MPZ is already handled so GMPy_MPZ_From_Integer() can't return
* an incremented reference to an existing value (which it would do
* if passed an MPZ).
*/
if ((result = GMPy_MPZ_From_Integer(x, context))) {
mpz_abs(result->z, result->z);
}
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Function_IsOdd(PyObject *self, PyObject *other)
{
int res;
MPZ_Object *tempx;
if (CHECK_MPZANY(other)) {
res = mpz_odd_p(MPZ(other));
}
else {
if (!(tempx = GMPy_MPZ_From_Integer(other, NULL))) {
TYPE_ERROR("is_odd() requires 'mpz' argument");
return NULL;
}
else {
res = mpz_odd_p(tempx->z);
Py_DECREF((PyObject*)tempx);
}
}
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
static PyObject *
GMPy_MPZ_bit_flip_function(PyObject *self, PyObject *args)
{
mp_bitcnt_t bit_index;
MPZ_Object *result = NULL, *tempx = NULL;
if (PyTuple_GET_SIZE(args) != 2)
goto err;
if (!(result = GMPy_MPZ_New(NULL)))
return NULL;
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL)))
goto err;
bit_index = mp_bitcnt_t_From_Integer(PyTuple_GET_ITEM(args, 1));
if (bit_index == (mp_bitcnt_t)(-1) && PyErr_Occurred())
goto err_index;
mpz_set(result->z, tempx->z);
mpz_combit(result->z, bit_index);
err:
TYPE_ERROR("bit_flip() requires 'mpz','int' arguments");
err_index:
Py_XDECREF((PyObject*)result);
Py_XDECREF((PyObject*)tempx);
return NULL;
}
static PyObject *
GMPy_MPZ_random_Function(PyObject *self, PyObject *args)
{
MPZ_Object *result, *temp;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("mpz_random() requires 2 arguments");
return NULL;
}
if (!RandomState_Check(PyTuple_GET_ITEM(args, 0))) {
TYPE_ERROR("mpz_random() requires 'random_state' and 'int' arguments");
return NULL;
}
if (!(temp = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL))) {
TYPE_ERROR("mpz_random() requires 'random_state' and 'int' arguments");
return NULL;
}
if ((result = GMPy_MPZ_New(NULL))) {
mpz_urandomm(result->z, RANDOM_STATE(PyTuple_GET_ITEM(args, 0)), temp->z);
}
Py_DECREF((PyObject*)temp);
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Function_Isqrt(PyObject *self, PyObject *other)
{
MPZ_Object *result;
if (CHECK_MPZANY(other)) {
if (mpz_sgn(MPZ(other)) < 0) {
VALUE_ERROR("isqrt() of negative number");
return NULL;
}
if ((result = GMPy_MPZ_New(NULL))) {
mpz_sqrt(result->z, MPZ(other));
}
}
else {
if (!(result = GMPy_MPZ_From_Integer(other, NULL))) {
TYPE_ERROR("isqrt() requires 'mpz' argument");
return NULL;
}
if (mpz_sgn(result->z) < 0) {
VALUE_ERROR("isqrt() of negative number");
Py_DECREF((PyObject*)result);
return NULL;
}
mpz_sqrt(result->z, result->z);
}
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Lshift_Slot(PyObject *self, PyObject *other)
{
mp_bitcnt_t count;
MPZ_Object *result, *tempx;
count = mp_bitcnt_t_From_Integer(other);
if ((count == (mp_bitcnt_t)(-1)) && PyErr_Occurred())
return NULL;
if (!(result = GMPy_MPZ_New(NULL)))
return NULL;
if (CHECK_MPZANY(self)) {
mpz_mul_2exp(result->z, MPZ(self), count);
return (PyObject*)result;
}
else {
if (!(tempx = GMPy_MPZ_From_Integer(self, NULL))) {
Py_XDECREF((PyObject*)result);
Py_XDECREF((PyObject*)tempx);
return NULL;
}
mpz_mul_2exp(result->z, tempx->z, count);
Py_DECREF((PyObject*)tempx);
return (PyObject*)result;
}
}
static PyObject *
GMPy_MPZ_bit_test_function(PyObject *self, PyObject *args)
{
mp_bitcnt_t bit_index;
int temp;
MPZ_Object *tempx = NULL;
if (PyTuple_GET_SIZE(args) != 2) {
goto err;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
goto err;
}
bit_index = mp_bitcnt_t_From_Integer(PyTuple_GET_ITEM(args, 1));
if (bit_index == (mp_bitcnt_t)(-1) && PyErr_Occurred()) {
goto err_index;
}
temp = mpz_tstbit(tempx->z, bit_index);
Py_DECREF((PyObject*)tempx);
if (temp)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
err:
TYPE_ERROR("bit_test() requires 'mpz','int' arguments");
err_index:
Py_DECREF((PyObject*)tempx);
return NULL;
}
static PyObject *
GMPy_MPZ_c_div_2exp(PyObject *self, PyObject *args)
{
mp_bitcnt_t nbits;
MPZ_Object *result, *tempx;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("c_div_2exp() requires 'mpz','int' arguments");
return NULL;
}
nbits = mp_bitcnt_t_From_Integer(PyTuple_GET_ITEM(args, 1));
if (nbits == (mp_bitcnt_t)(-1) && PyErr_Occurred()) {
return NULL;
}
tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL);
result = GMPy_MPZ_New(NULL);
if (!tempx || !result) {
Py_XDECREF((PyObject*)result);
Py_XDECREF((PyObject*)tempx);
return NULL;
}
mpz_cdiv_q_2exp(result->z, tempx->z, nbits);
Py_DECREF((PyObject*)tempx);
return (PyObject*)result;
}
static PyObject *
GMPy_Context_Digits(PyObject *self, PyObject *args)
{
PyObject *arg0, *tuple, *temp, *result;
Py_ssize_t argc;
argc = PyTuple_GET_SIZE(args);
if (argc == 0) {
TYPE_ERROR("digits() requires at least one argument");
return NULL;
}
if (argc > 3) {
TYPE_ERROR("digits() accepts at most three arguments");
return NULL;
}
arg0 = PyTuple_GET_ITEM(args, 0);
if (!(tuple = PyTuple_GetSlice(args, 1, argc))) {
return NULL;
}
if (IS_INTEGER(arg0)) {
temp = (PyObject*)GMPy_MPZ_From_Integer(arg0, NULL);
result = GMPy_MPZ_Digits_Method(temp, tuple);
Py_DECREF(temp);
Py_DECREF(tuple);
return result;
}
if (IS_RATIONAL(arg0)) {
temp = (PyObject*)GMPy_MPQ_From_Rational(arg0, NULL);
result = GMPy_MPQ_Digits_Method(temp, tuple);
Py_DECREF(temp);
Py_DECREF(tuple);
return result;
}
if (IS_REAL(arg0)) {
temp = (PyObject*)GMPy_MPFR_From_Real(arg0, 1, NULL);
result = GMPy_MPFR_Digits_Method(temp, tuple);
Py_DECREF(temp);
Py_DECREF(tuple);
return result;
}
if (IS_COMPLEX(arg0)) {
temp = (PyObject*)GMPy_MPC_From_Complex(arg0, 1, 1, NULL);
result = GMPy_MPC_Digits_Method(temp, tuple);
Py_DECREF(temp);
Py_DECREF(tuple);
return result;
}
TYPE_ERROR("digits() argument type not supported");
return NULL;
}
static PyObject *
GMPy_MPZ_Function_Iroot(PyObject *self, PyObject *args)
{
unsigned long n;
int exact;
MPZ_Object *root = NULL, *tempx = NULL;
PyObject *result = NULL;
if ((PyTuple_GET_SIZE(args) != 2) ||
((!IS_INTEGER(PyTuple_GET_ITEM(args, 0))) ||
(!IS_INTEGER(PyTuple_GET_ITEM(args, 1))))) {
TYPE_ERROR("iroot() requires 'int','int' arguments");
return NULL;
}
n = c_ulong_From_Integer(PyTuple_GET_ITEM(args, 1));
if ((n == 0) || ((n == (unsigned long)(-1)) && PyErr_Occurred())) {
VALUE_ERROR("n must be > 0");
return NULL;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
/* LCOV_EXCL_START */
return NULL;
/* LCOV_EXCL_STOP */
}
if (mpz_sgn(tempx->z) < 0) {
VALUE_ERROR("iroot() of negative number");
Py_DECREF((PyObject*)tempx);
return NULL;
}
if (!(result = PyTuple_New(2)) ||
!(root = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
Py_DECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)root);
Py_XDECREF(result);
return NULL;
/* LCOV_EXCL_STOP */
}
exact = mpz_root(root->z, tempx->z, n);
Py_DECREF((PyObject*)tempx);
PyTuple_SET_ITEM(result, 0, (PyObject*)root);
PyTuple_SET_ITEM(result, 1, (PyObject*)PyBool_FromLong(exact));
return result;
}
static PyObject *
GMPy_MPZ_Function_IsDivisible(PyObject *self, PyObject *args)
{
unsigned long temp;
int error, res;
MPZ_Object *tempx, *tempd;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("is_divisible() requires 2 integer arguments");
return NULL;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
return NULL;
}
temp = GMPy_Integer_AsUnsignedLongAndError(PyTuple_GET_ITEM(args, 1), &error);
if (!error) {
res = mpz_divisible_ui_p(tempx->z, temp);
Py_DECREF((PyObject*)tempx);
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
if (!(tempd = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL))) {
TYPE_ERROR("is_divisible() requires 2 integer arguments");
Py_DECREF((PyObject*)tempx);
return NULL;
}
res = mpz_divisible_p(tempx->z, tempd->z);
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempd);
if (res)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
static PyObject *
GMPy_MPZ_Function_LCM(PyObject *self, PyObject *args)
{
PyObject *arg0, *arg1;
MPZ_Object *result = NULL, *tempa = NULL, *tempb = NULL;
if(PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("lcm() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(result = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
return NULL;
/* LCOV_EXCL_STOP */
}
arg0 = PyTuple_GET_ITEM(args, 0);
arg1 = PyTuple_GET_ITEM(args, 1);
if (MPZ_Check(arg0) && MPZ_Check(arg1)) {
mpz_lcm(result->z, MPZ(arg0), MPZ(arg1));
}
else {
if (!(tempa = GMPy_MPZ_From_Integer(arg0, NULL)) ||
!(tempb = GMPy_MPZ_From_Integer(arg1, NULL))) {
TYPE_ERROR("lcm() requires 'mpz','mpz' arguments");
Py_XDECREF((PyObject*)tempa);
Py_XDECREF((PyObject*)tempb);
Py_DECREF((PyObject*)result);
return NULL;
}
mpz_lcm(result->z, tempa->z, tempb->z);
Py_DECREF((PyObject*)tempa);
Py_DECREF((PyObject*)tempb);
}
return (PyObject*)result;
}
static PyObject *
GMPy_Integer_Minus(PyObject *x, CTXT_Object *context)
{
PyObject *result, *tempx;
if (!(tempx = (PyObject*)GMPy_MPZ_From_Integer(x, context))) {
return NULL;
}
result = _GMPy_MPZ_Minus(tempx, context);
Py_DECREF(tempx);
return result;
}
static PyObject *
GMPy_Integer_Sign(PyObject *x, CTXT_Object *context)
{
long res;
MPZ_Object *tempx;
if (!(tempx = GMPy_MPZ_From_Integer(x, context))) {
return NULL;
}
else {
res = mpz_sgn(tempx->z);
Py_DECREF((PyObject*)tempx);
return PyIntOrLong_FromLong(res);
}
}
static PyObject *
GMPy_MPZ_Function_Kronecker(PyObject *self, PyObject *args)
{
MPZ_Object *tempx = NULL, *tempy = NULL;
long res;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("kronecker() requires 'mpz','mpz' arguments");
return NULL;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL)) ||
!(tempy = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 1), NULL))) {
Py_XDECREF((PyObject*)tempx);
Py_XDECREF((PyObject*)tempy);
return NULL;
}
res = (long)(mpz_kronecker(tempx->z, tempy->z));
Py_DECREF((PyObject*)tempx);
Py_DECREF((PyObject*)tempy);
return PyIntOrLong_FromLong(res);
}
static PyObject *
GMPy_MPZ_bit_length_function(PyObject *self, PyObject *other)
{
mp_bitcnt_t n = 0;
MPZ_Object* tempx;
if (!(tempx = GMPy_MPZ_From_Integer(other, NULL))) {
TYPE_ERROR("bit_length() requires 'mpz' argument");
return NULL;
}
if (mpz_size(MPZ(tempx)))
n = mpz_sizeinbase(tempx->z, 2);
Py_DECREF((PyObject*)tempx);
return PyIntOrLong_FromMpBitCnt(n);
}
static PyObject *
GMPy_MPZ_popcount(PyObject *self, PyObject *other)
{
mp_bitcnt_t n;
MPZ_Object *tempx;
if ((tempx = GMPy_MPZ_From_Integer(other, NULL))) {
n = mpz_popcount(tempx->z);
Py_DECREF((PyObject*)tempx);
if (n == (mp_bitcnt_t)(-1))
return PyLong_FromLong(-1);
else
return PyIntOrLong_FromMpBitCnt(n);
}
else {
TYPE_ERROR("popcount() requires 'mpz' argument");
return NULL;
}
}
static PyObject *
GMPy_MPZ_Function_Bincoef(PyObject *self, PyObject *args)
{
MPZ_Object *result = NULL, *tempx;
unsigned long n, k;
if (PyTuple_GET_SIZE(args) != 2) {
TYPE_ERROR("bincoef() requires two integer arguments");
return NULL;
}
if (!(result = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
return NULL;
/* LCOV_EXCL_STOP */
}
k = c_ulong_From_Integer(PyTuple_GET_ITEM(args, 1));
if (k == (unsigned long)(-1) && PyErr_Occurred()) {
Py_DECREF((PyObject*)result);
return NULL;
}
n = c_ulong_From_Integer(PyTuple_GET_ITEM(args, 0));
if (!(n == (unsigned long)(-1) && PyErr_Occurred())) {
/* Use mpz_bin_uiui which should be faster. */
mpz_bin_uiui(result->z, n, k);
return (PyObject*)result;
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
Py_DECREF((PyObject*)result);
return NULL;
}
mpz_bin_ui(result->z, tempx->z, k);
Py_DECREF((PyObject*)tempx);
return (PyObject*)result;
}
static PyObject *
GMPy_MPZ_Function_IsqrtRem(PyObject *self, PyObject *other)
{
MPZ_Object *root = NULL, *rem = NULL, *temp = NULL;
PyObject *result;
if (!(temp = GMPy_MPZ_From_Integer(other, NULL))) {
TYPE_ERROR("isqrt_rem() requires 'mpz' argument");
return NULL;
}
if (mpz_sgn(temp->z) < 0) {
VALUE_ERROR("isqrt_rem() of negative number");
Py_DECREF((PyObject*)temp);
return NULL;
}
if (!(result = PyTuple_New(2)) ||
!(root = GMPy_MPZ_New(NULL)) ||
!(rem = GMPy_MPZ_New(NULL))) {
/* LCOV_EXCL_START */
Py_DECREF((PyObject*)temp);
Py_XDECREF(result);
Py_XDECREF((PyObject*)root);
Py_XDECREF((PyObject*)rem);
return NULL;
/* LCOV_EXCL_STOP */
}
mpz_sqrtrem(root->z, rem->z, temp->z);
Py_DECREF((PyObject*)temp);
PyTuple_SET_ITEM(result, 0, (PyObject*)root);
PyTuple_SET_ITEM(result, 1, (PyObject*)rem);
return result;
}
static PyObject *
GMPy_MPZ_Function_IsPrime(PyObject *self, PyObject *args)
{
int i;
unsigned long reps = 25;
MPZ_Object* tempx;
Py_ssize_t argc;
argc = PyTuple_GET_SIZE(args);
if (argc == 0 || argc > 2) {
TYPE_ERROR("is_prime() requires 'mpz'[,'int'] arguments");
return NULL;
}
if (PyTuple_GET_SIZE(args) == 2) {
reps = c_ulong_From_Integer(PyTuple_GET_ITEM(args, 1));
if (reps == -1 && PyErr_Occurred()) {
return NULL;
}
/* Silently limit n to a reasonable value. */
if (reps > 1000) {
reps = 1000;
}
}
if (!(tempx = GMPy_MPZ_From_Integer(PyTuple_GET_ITEM(args, 0), NULL))) {
return NULL;
}
i = mpz_probab_prime_p(tempx->z, (int)reps);
Py_DECREF((PyObject*)tempx);
if (i)
Py_RETURN_TRUE;
else
Py_RETURN_FALSE;
}
