int main()
{
long iter;
flint_rand_t state;
printf("add_si_inline....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000; iter++)
{
fmpz_t a, c, d;
long b;
fmpz_init(a);
fmpz_init(c);
fmpz_init(d);
fmpz_randtest(a, state, 1 + n_randint(state, 200));
fmpz_randtest(c, state, 1 + n_randint(state, 200));
fmpz_randtest(d, state, 1 + n_randint(state, 200));
b = n_randtest(state);
fmpz_add_si(c, a, b);
fmpz_add_si_inline(d, a, b);
if (!fmpz_equal(c, d))
{
printf("FAIL\n");
fmpz_print(a); printf("\n\n");
printf("%ld", b); printf("\n\n");
fmpz_print(c); printf("\n\n");
fmpz_print(d); printf("\n\n");
abort();
}
fmpz_add_si_inline(a, a, b);
if (!fmpz_equal(c, a))
{
printf("FAIL (aliasing)\n");
fmpz_print(a); printf("\n\n");
printf("%ld", b); printf("\n\n");
fmpz_print(c); printf("\n\n");
fmpz_print(d); printf("\n\n");
abort();
}
fmpz_clear(a);
fmpz_clear(c);
fmpz_clear(d);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int
arf_cmp_2exp_si(const arf_t x, long e)
{
if (arf_is_special(x))
{
if (arf_is_zero(x)) return -1;
if (arf_is_pos_inf(x)) return 1;
if (arf_is_neg_inf(x)) return -1;
return 0;
}
if (ARF_SGNBIT(x))
return -1;
/* Fast path. */
if (!COEFF_IS_MPZ(ARF_EXP(x)))
{
if (ARF_IS_POW2(x) && (ARF_EXP(x) - 1 == e))
return 0;
else
return (ARF_EXP(x) <= e) ? -1 : 1;
}
if (ARF_IS_POW2(x))
{
fmpz_t t;
fmpz_init(t);
fmpz_one(t);
fmpz_add_si(t, t, e);
if (fmpz_equal(ARF_EXPREF(x), t))
{
fmpz_clear(t);
return 0;
}
fmpz_clear(t);
}
return (fmpz_cmp_si(ARF_EXPREF(x), e) <= 0) ? -1 : 1;
}
int
arf_root(arf_ptr z, arf_srcptr x, ulong k, slong prec, arf_rnd_t rnd)
{
mp_size_t xn, zn, val;
mp_srcptr xptr;
mp_ptr tmp, zptr;
mpfr_t xf, zf;
fmpz_t q, r;
int inexact;
if (k == 0)
{
arf_nan(z);
return 0;
}
if (k == 1)
return arf_set_round(z, x, prec, rnd);
if (k == 2)
return arf_sqrt(z, x, prec, rnd);
if (arf_is_special(x))
{
if (arf_is_neg_inf(x))
arf_nan(z);
else
arf_set(z, x);
return 0;
}
if (ARF_SGNBIT(x))
{
arf_nan(z);
return 0;
}
fmpz_init(q);
fmpz_init(r);
/* x = m * 2^e where e = qk + r */
/* x^(1/k) = (m * 2^(qk+r))^(1/k) */
/* x^(1/k) = (m * 2^r)^(1/k) * 2^q */
fmpz_set_ui(r, k);
fmpz_fdiv_qr(q, r, ARF_EXPREF(x), r);
ARF_GET_MPN_READONLY(xptr, xn, x);
zn = (prec + FLINT_BITS - 1) / FLINT_BITS;
zf->_mpfr_d = tmp = flint_malloc(zn * sizeof(mp_limb_t));
zf->_mpfr_prec = prec;
zf->_mpfr_sign = 1;
zf->_mpfr_exp = 0;
xf->_mpfr_d = (mp_ptr) xptr;
xf->_mpfr_prec = xn * FLINT_BITS;
xf->_mpfr_sign = 1;
xf->_mpfr_exp = fmpz_get_ui(r);
inexact = mpfr_root(zf, xf, k, arf_rnd_to_mpfr(rnd));
inexact = (inexact != 0);
val = 0;
while (tmp[val] == 0)
val++;
ARF_GET_MPN_WRITE(zptr, zn - val, z);
flint_mpn_copyi(zptr, tmp + val, zn - val);
fmpz_add_si(ARF_EXPREF(z), q, zf->_mpfr_exp);
flint_free(tmp);
fmpz_clear(q);
fmpz_clear(r);
return inexact;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("add2_fmpz_si_inline....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000; iter++)
{
fmpz_t a, b, c, d;
slong e;
fmpz_init(a);
fmpz_init(b);
fmpz_init(c);
fmpz_init(d);
fmpz_randtest(a, state, 1 + n_randint(state, 200));
fmpz_randtest(b, state, 1 + n_randint(state, 200));
fmpz_randtest(c, state, 1 + n_randint(state, 200));
fmpz_randtest(d, state, 1 + n_randint(state, 200));
e = n_randtest(state);
fmpz_add(c, a, b);
fmpz_add_si(c, c, e);
fmpz_add2_fmpz_si_inline(d, a, b, e);
if (!fmpz_equal(c, d))
{
flint_printf("FAIL\n");
fmpz_print(a); flint_printf("\n\n");
fmpz_print(b); flint_printf("\n\n");
fmpz_print(c); flint_printf("\n\n");
fmpz_print(d); flint_printf("\n\n");
flint_printf("%wd\n\n", e);
abort();
}
fmpz_add2_fmpz_si_inline(a, a, b, e);
if (!fmpz_equal(c, a))
{
flint_printf("FAIL (aliasing 1)\n");
fmpz_print(a); flint_printf("\n\n");
fmpz_print(b); flint_printf("\n\n");
fmpz_print(c); flint_printf("\n\n");
fmpz_print(d); flint_printf("\n\n");
flint_printf("%wd\n\n", e);
abort();
}
fmpz_randtest(a, state, 1 + n_randint(state, 200));
fmpz_add(c, a, b);
fmpz_add_si(c, c, e);
fmpz_add2_fmpz_si_inline(b, a, b, e);
if (!fmpz_equal(c, b))
{
flint_printf("FAIL (aliasing 2)\n");
fmpz_print(a); flint_printf("\n\n");
fmpz_print(b); flint_printf("\n\n");
fmpz_print(c); flint_printf("\n\n");
fmpz_print(d); flint_printf("\n\n");
flint_printf("%wd\n\n", e);
abort();
}
fmpz_add(d, a, a);
fmpz_add_si(d, d, e);
fmpz_add2_fmpz_si_inline(c, a, a, e);
if (!fmpz_equal(c, d))
{
flint_printf("FAIL (aliasing 3)\n");
fmpz_print(a); flint_printf("\n\n");
fmpz_print(b); flint_printf("\n\n");
fmpz_print(c); flint_printf("\n\n");
fmpz_print(d); flint_printf("\n\n");
flint_printf("%wd\n\n", e);
abort();
}
fmpz_add(d, a, a);
fmpz_add_si(d, d, e);
fmpz_add2_fmpz_si_inline(a, a, a, e);
if (!fmpz_equal(d, a))
{
flint_printf("FAIL (aliasing 4)\n");
fmpz_print(a); flint_printf("\n\n");
fmpz_print(b); flint_printf("\n\n");
fmpz_print(c); flint_printf("\n\n");
fmpz_print(d); flint_printf("\n\n");
flint_printf("%wd\n\n", e);
abort();
}
fmpz_clear(a);
fmpz_clear(b);
fmpz_clear(c);
fmpz_clear(d);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
