#include "test_helpers.h"
#include "mpn_extras.h"
TEST_FUNCTION_START(flint_mpn_divrem_preinvn, state)
{
int i, result;
mpz_t a, d, q1, q2, r1, r2;
gmp_randstate_t st;
mp_ptr dinv;
mp_size_t size, size2;
flint_bitcnt_t norm;
mpz_init(a);
mpz_init(d);
mpz_init(q1);
mpz_init(r1);
gmp_randinit_default(st);
for (i = 0; i < 10000 * flint_test_multiplier(); i++)
{
int alias = n_randint(state, 2);
size = n_randint(state, 20) + 1;
size2 = n_randint(state, 20) + size;
mpz_rrandomb(a, st, size2 * FLINT_BITS);
do {
mpz_rrandomb(d, st, size * FLINT_BITS);
} while (mpz_sgn(d) == 0);
norm = flint_clz(d->_mp_d[d->_mp_size - 1]);
mpz_mul_2exp(d, d, norm);
mpz_mul_2exp(a, a, norm);
size2 = a->_mp_size;
q2->_mp_size = size2 - size + 1;
q2->_mp_d = flint_malloc(q2->_mp_size*sizeof(mp_limb_t));
r2->_mp_size = size2;
r2->_mp_d = flint_malloc(r2->_mp_size*sizeof(mp_limb_t));
mpz_fdiv_qr(q1, r1, a, d);
dinv = flint_malloc(size*sizeof(mp_limb_t));
flint_mpn_preinvn(dinv, d->_mp_d, size);
if (alias)
{
q2->_mp_d[q2->_mp_size - 1] = flint_mpn_divrem_preinvn(q2->_mp_d, a->_mp_d, a->_mp_d, size2, d->_mp_d, size, dinv);
while (size && a->_mp_d[size - 1] == 0) size--;
a->_mp_size = size;
}
else
{
q2->_mp_d[q2->_mp_size - 1] = flint_mpn_divrem_preinvn(q2->_mp_d, r2->_mp_d, a->_mp_d, size2, d->_mp_d, size, dinv);
while (size && r2->_mp_d[size - 1] == 0) size--;
r2->_mp_size = size;
}
size2 = q2->_mp_size;
while (size2 && q2->_mp_d[size2 - 1] == 0) size2--;
q2->_mp_size = size2;
result = (mpz_cmp(r1, alias ? a : r2) == 0 && mpz_cmp(q1, q2) == 0);
if (!result)
TEST_FUNCTION_FAIL(
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"%{mpz}\n"
"size = %wd\n"
"size2 = %wd\n",
a, d, r1, r2, q1, q2, size, size2);
flint_free(dinv);
flint_free(q2->_mp_d);
flint_free(r2->_mp_d);
}
mpz_clear(a);
mpz_clear(d);
mpz_clear(q1);
mpz_clear(r1);
gmp_randclear(st);
TEST_FUNCTION_END(state);
}