void
zeta_series_em_choose_param(fmpr_t bound, ulong * N, ulong * M, const fmpcb_t s, const fmpcb_t a, long d, long target, long prec)
{
ulong A, B, C;
fmpr_t Abound, Bbound, Cbound, tol;
fmpr_init(Abound);
fmpr_init(Bbound);
fmpr_init(Cbound);
fmpr_init(tol);
fmpr_set_si_2exp_si(tol, 1, -target);
A = 1;
B = 2;
zeta_series_em_bound(Bbound, s, a, B, choose_M(B, target), d, prec);
if (fmpr_cmp(Bbound, tol) > 0)
{
while (fmpr_cmp(Bbound, tol) > 0)
{
fmpr_set(Abound, Bbound);
A *= 2;
B *= 2;
if (B == 0) abort();
zeta_series_em_bound(Bbound, s, a, B, choose_M(B, target), d, prec);
}
/* bisect (-A, B] */
while (B > A + 4)
{
C = A + (B - A) / 2;
zeta_series_em_bound(Cbound, s, a, C, choose_M(C, target), d, prec);
if (fmpr_cmp(Cbound, tol) < 0)
{
B = C;
fmpr_set(Bbound, Cbound);
}
else
{
A = C;
fmpr_set(Abound, Cbound);
}
}
}
fmpr_set(bound, Bbound);
*N = B;
*M = choose_M(B, target);
fmpr_clear(Abound);
fmpr_clear(Bbound);
fmpr_clear(Cbound);
fmpr_clear(tol);
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("cmp....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000 * arb_test_multiplier(); iter++)
{
fmpr_t x, y;
mag_t xb, yb;
int c1, c2;
fmpr_init(x);
fmpr_init(y);
mag_init(xb);
mag_init(yb);
mag_randtest_special(xb, state, 100);
mag_randtest_special(yb, state, 100);
mag_get_fmpr(x, xb);
mag_get_fmpr(y, yb);
c1 = fmpr_cmp(x, y);
c2 = mag_cmp(xb, yb);
if (c1 != c2)
{
flint_printf("FAIL\n\n");
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("xb = "); mag_print(xb); flint_printf("\n\n");
flint_printf("yb = "); mag_print(yb); flint_printf("\n\n");
abort();
}
fmpr_clear(x);
fmpr_clear(y);
mag_clear(xb);
mag_clear(yb);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("cmp_2exp_si....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000; iter++)
{
slong bits, e;
fmpr_t x, y;
int cmp1, cmp2;
bits = 2 + n_randint(state, 1000);
e = n_randtest(state);
fmpr_init(x);
fmpr_init(y);
fmpr_randtest_special(x, state, bits, 100);
fmpr_set_ui_2exp_si(y, 1, e);
cmp1 = fmpr_cmp(x, y);
cmp2 = fmpr_cmp_2exp_si(x, e);
if (cmp1 != cmp2)
{
flint_printf("FAIL\n\n");
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("cmp1 = %d, cmp2 = %d\n\n", cmp1, cmp2);
abort();
}
fmpr_clear(x);
fmpr_clear(y);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
long iter;
flint_rand_t state;
printf("set_interval_fmpr....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 100000; iter++)
{
fmprb_t x;
fmpr_t a, b;
fmprb_init(x);
fmpr_init(a);
fmpr_init(b);
fmpr_randtest_special(a, state, 200, 10);
fmpr_randtest_special(b, state, 200, 10);
if (fmpr_cmp(a, b) > 0)
fmpr_swap(a, b);
fmprb_set_interval_fmpr(x, a, b, 2 + n_randint(state, 200));
if (!fmprb_contains_fmpr(x, a) || !fmprb_contains_fmpr(x, b))
{
printf("FAIL:\n\n");
printf("x = "); fmprb_print(x); printf("\n\n");
printf("a = "); fmpr_print(a); printf("\n\n");
printf("b = "); fmpr_print(b); printf("\n\n");
abort();
}
fmprb_clear(x);
fmpr_clear(a);
fmpr_clear(b);
}
flint_randclear(state);
flint_cleanup();
printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("sub....");
fflush(stdout);
flint_randinit(state);
/* test exact subtraction: (x - y) - z == (x - z) - y */
for (iter = 0; iter < 100000; iter++)
{
slong bits, res1, res2, res3, res4;
fmpr_t x, y, z, t, u;
bits = 2 + n_randint(state, 200);
fmpr_init(x);
fmpr_init(y);
fmpr_init(z);
fmpr_init(t);
fmpr_init(u);
fmpr_randtest_special(x, state, bits, 10);
fmpr_randtest_special(y, state, bits, 10);
fmpr_randtest_special(z, state, bits, 10);
fmpr_randtest_special(t, state, bits, 10);
fmpr_randtest_special(u, state, bits, 10);
res1 = fmpr_sub(t, x, y, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res2 = fmpr_sub(t, t, z, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res3 = fmpr_sub(u, x, z, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res4 = fmpr_sub(u, u, y, FMPR_PREC_EXACT, FMPR_RND_DOWN);
if (!fmpr_equal(t, u) ||
res1 != FMPR_RESULT_EXACT || res2 != FMPR_RESULT_EXACT ||
res3 != FMPR_RESULT_EXACT || res4 != FMPR_RESULT_EXACT)
{
flint_printf("FAIL\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("t = "); fmpr_print(t); flint_printf("\n\n");
flint_printf("u = "); fmpr_print(u); flint_printf("\n\n");
abort();
}
fmpr_clear(x);
fmpr_clear(y);
fmpr_clear(z);
fmpr_clear(t);
fmpr_clear(u);
}
/* compare rounding with mpfr */
for (iter = 0; iter < 100000; iter++)
{
slong bits, res;
int mpfr_res;
fmpr_t x, y, z, w;
mpfr_t X, Y, Z;
bits = 2 + n_randint(state, 200);
fmpr_init(x);
fmpr_init(y);
fmpr_init(z);
fmpr_init(w);
mpfr_init2(X, bits + 100);
mpfr_init2(Y, bits + 100);
mpfr_init2(Z, bits);
fmpr_randtest_special(x, state, bits + n_randint(state, 100), 10);
fmpr_randtest_special(y, state, bits + n_randint(state, 100), 10);
fmpr_randtest_special(z, state, bits, 10);
fmpr_get_mpfr(X, x, MPFR_RNDN);
fmpr_get_mpfr(Y, y, MPFR_RNDN);
switch (n_randint(state, 4))
{
case 0:
mpfr_res = mpfr_sub(Z, X, Y, MPFR_RNDZ);
res = fmpr_sub(z, x, y, bits, FMPR_RND_DOWN);
break;
case 1:
mpfr_res = mpfr_sub(Z, X, Y, MPFR_RNDA);
res = fmpr_sub(z, x, y, bits, FMPR_RND_UP);
break;
case 2:
mpfr_res = mpfr_sub(Z, X, Y, MPFR_RNDD);
res = fmpr_sub(z, x, y, bits, FMPR_RND_FLOOR);
break;
default:
mpfr_res = mpfr_sub(Z, X, Y, MPFR_RNDU);
res = fmpr_sub(z, x, y, bits, FMPR_RND_CEIL);
break;
}
fmpr_set_mpfr(w, Z);
if (!fmpr_equal(z, w) || (res == FMPR_RESULT_EXACT) != (mpfr_res == 0)
|| !fmpr_check_ulp(z, res, bits))
{
flint_printf("FAIL\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("w = "); fmpr_print(w); flint_printf("\n\n");
flint_printf("returned: %wd, %d\n", res, mpfr_res);
abort();
}
/* check error bound */
if (!fmpr_is_nan(z) && !fmpr_is_inf(z))
{
fmpr_t z_exact, error_bound, true_error;
fmpr_init(z_exact);
fmpr_init(error_bound);
fmpr_init(true_error);
fmpr_set_error_result(error_bound, z, res);
fmpr_sub(z_exact, x, y, FMPR_PREC_EXACT, FMPR_RND_DOWN);
fmpr_sub(true_error, z, z_exact, FMPR_PREC_EXACT, FMPR_RND_DOWN);
fmpr_abs(true_error, true_error);
if (fmpr_is_zero(error_bound) != fmpr_is_zero(true_error) ||
fmpr_cmp(true_error, error_bound) > 0)
{
flint_printf("FAIL: error bound\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("z_exact = "); fmpr_print(z_exact); flint_printf("\n\n");
flint_printf("true_error = "); fmpr_print(true_error); flint_printf("\n\n");
flint_printf("error_bound = "); fmpr_print(error_bound); flint_printf("\n\n");
abort();
}
fmpr_clear(z_exact);
fmpr_clear(error_bound);
fmpr_clear(true_error);
}
fmpr_clear(x);
fmpr_clear(y);
fmpr_clear(z);
fmpr_clear(w);
mpfr_clear(X);
mpfr_clear(Y);
mpfr_clear(Z);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
/* convert to an fmpz poly with a common exponent and coefficients
at most prec bits, also bounding input error plus rounding error */
void _fmprb_poly_get_fmpz_poly_2exp(fmpr_t error, fmpz_t exp, fmpz * coeffs,
fmprb_srcptr A, long lenA, long prec)
{
fmpz_t top_exp, bot_exp;
long shift;
long i;
int rounding;
fmpz_init(top_exp);
fmpz_init(bot_exp);
if (!_fmprb_poly_mid_get_hull(bot_exp, top_exp, A, lenA))
{
fmpz_zero(exp);
_fmpz_vec_zero(coeffs, lenA);
fmpr_zero(error);
for (i = 0; i < lenA; i++)
{
if (fmpr_cmp(fmprb_radref(A + i), error) > 0)
fmpr_set(error, fmprb_radref(A + i));
}
return; /* no need to clear fmpzs */
}
/* only take as much precision as necessary */
shift = _fmpz_sub_small(top_exp, bot_exp);
prec = FLINT_MIN(prec, shift);
fmpz_sub_ui(exp, top_exp, prec);
/* extract integer polynomial */
rounding = 0;
for (i = 0; i < lenA; i++)
rounding |= fmpr_get_fmpz_fixed_fmpz(coeffs + i,
fmprb_midref(A + i), exp);
fmpr_zero(error);
/* compute maximum of input errors */
for (i = 0; i < lenA; i++)
{
if (fmpr_cmp(fmprb_radref(A + i), error) > 0)
fmpr_set(error, fmprb_radref(A + i));
}
/* add rounding error */
if (rounding)
{
fmpr_t t;
fmpr_init(t);
fmpz_set_ui(fmpr_manref(t), 1UL);
fmpz_set(fmpr_expref(t), exp);
fmpr_add(error, error, t, FMPRB_RAD_PREC, FMPR_RND_UP);
fmpr_clear(t);
}
fmpz_clear(top_exp);
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("add....");
fflush(stdout);
flint_randinit(state);
/* test exact addition: (x + y) + z == x + (y + z) */
for (iter = 0; iter < 100000 * arb_test_multiplier(); iter++)
{
slong bits, res1, res2, res3, res4;
fmpr_t x, y, z, t, u;
bits = 2 + n_randint(state, 200);
fmpr_init(x);
fmpr_init(y);
fmpr_init(z);
fmpr_init(t);
fmpr_init(u);
fmpr_randtest_special(x, state, bits, 10);
fmpr_randtest_special(y, state, bits, 10);
fmpr_randtest_special(z, state, bits, 10);
fmpr_randtest_special(t, state, bits, 10);
fmpr_randtest_special(u, state, bits, 10);
res1 = fmpr_add(t, x, y, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res2 = fmpr_add(t, t, z, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res3 = fmpr_add(u, y, z, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res4 = fmpr_add(u, x, u, FMPR_PREC_EXACT, FMPR_RND_DOWN);
if (!fmpr_equal(t, u) ||
res1 != FMPR_RESULT_EXACT || res2 != FMPR_RESULT_EXACT ||
res3 != FMPR_RESULT_EXACT || res4 != FMPR_RESULT_EXACT)
{
flint_printf("FAIL\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("t = "); fmpr_print(t); flint_printf("\n\n");
flint_printf("u = "); fmpr_print(u); flint_printf("\n\n");
abort();
}
fmpr_clear(x);
fmpr_clear(y);
fmpr_clear(z);
fmpr_clear(t);
fmpr_clear(u);
}
/* compare with add_naive */
for (iter = 0; iter < 1000000 * arb_test_multiplier(); iter++)
{
slong prec, ret1, ret2;
fmpr_t x, y, z, w;
fmpr_rnd_t rnd;
fmpr_init(x);
fmpr_init(y);
fmpr_init(z);
fmpr_init(w);
fmpr_randtest_special(x, state, 1 + n_randint(state, 1000), 1 + n_randint(state, 200));
fmpr_randtest_special(y, state, 1 + n_randint(state, 1000), 1 + n_randint(state, 200));
fmpr_randtest_special(z, state, 1 + n_randint(state, 1000), 1 + n_randint(state, 200));
prec = 2 + n_randint(state, 1000);
if (n_randint(state, 10) == 0 &&
fmpz_bits(fmpr_expref(x)) < 10 &&
fmpz_bits(fmpr_expref(y)) < 10)
prec = FMPR_PREC_EXACT;
switch (n_randint(state, 4))
{
case 0: rnd = FMPR_RND_DOWN; break;
case 1: rnd = FMPR_RND_UP; break;
case 2: rnd = FMPR_RND_FLOOR; break;
default: rnd = FMPR_RND_CEIL; break;
}
ret1 = fmpr_add(z, x, y, prec, rnd);
ret2 = fmpr_add_naive(w, x, y, prec, rnd);
if (!fmpr_equal(z, w) || ret1 != ret2 ||
!fmpr_check_ulp(z, ret1, prec) || !fmpr_check_ulp(w, ret2, prec))
{
flint_printf("FAIL\n\n");
flint_printf("iter %wd\n", iter);
flint_printf("prec = %wd\n", prec);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("w = "); fmpr_print(w); flint_printf("\n\n");
flint_printf("ret1 = %wd, ret2 = %wd\n", ret1, ret2);
abort();
}
fmpr_clear(x);
fmpr_clear(y);
fmpr_clear(z);
fmpr_clear(w);
}
/* compare rounding with mpfr */
for (iter = 0; iter < 1000000 * arb_test_multiplier(); iter++)
{
slong bits, res;
int mpfr_res;
fmpr_t x, y, z, w;
mpfr_t X, Y, Z;
bits = 2 + n_randint(state, 500);
fmpr_init(x);
fmpr_init(y);
fmpr_init(z);
fmpr_init(w);
mpfr_init2(X, bits + 500);
mpfr_init2(Y, bits + 500);
mpfr_init2(Z, bits);
fmpr_randtest_special(x, state, bits + n_randint(state, 500), 10);
fmpr_randtest_special(y, state, bits + n_randint(state, 500), 10);
fmpr_randtest_special(z, state, bits, 10);
fmpr_get_mpfr(X, x, MPFR_RNDN);
fmpr_get_mpfr(Y, y, MPFR_RNDN);
switch (n_randint(state, 4))
{
case 0:
mpfr_res = mpfr_add(Z, X, Y, MPFR_RNDZ);
res = fmpr_add(z, x, y, bits, FMPR_RND_DOWN);
break;
case 1:
mpfr_res = mpfr_add(Z, X, Y, MPFR_RNDA);
res = fmpr_add(z, x, y, bits, FMPR_RND_UP);
break;
case 2:
mpfr_res = mpfr_add(Z, X, Y, MPFR_RNDD);
res = fmpr_add(z, x, y, bits, FMPR_RND_FLOOR);
break;
default:
mpfr_res = mpfr_add(Z, X, Y, MPFR_RNDU);
res = fmpr_add(z, x, y, bits, FMPR_RND_CEIL);
break;
}
fmpr_set_mpfr(w, Z);
if (!fmpr_equal(z, w) || (res == FMPR_RESULT_EXACT) != (mpfr_res == 0)
|| !fmpr_check_ulp(z, res, bits))
{
flint_printf("FAIL\n\n");
flint_printf("iter %wd\n", iter);
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("w = "); fmpr_print(w); flint_printf("\n\n");
flint_printf("returned: %wd, %d\n", res, mpfr_res);
abort();
}
/* check error bound */
if (!fmpr_is_nan(z) && !fmpr_is_inf(z))
{
fmpr_t z_exact, error_bound, true_error;
fmpr_init(z_exact);
fmpr_init(error_bound);
fmpr_init(true_error);
fmpr_set_error_result(error_bound, z, res);
fmpr_add(z_exact, x, y, FMPR_PREC_EXACT, FMPR_RND_DOWN);
fmpr_sub(true_error, z, z_exact, FMPR_PREC_EXACT, FMPR_RND_DOWN);
fmpr_abs(true_error, true_error);
if (fmpr_is_zero(error_bound) != fmpr_is_zero(true_error) ||
fmpr_cmp(true_error, error_bound) > 0)
{
flint_printf("FAIL: error bound\n\n");
flint_printf("bits = %wd\n", bits);
flint_printf("x = "); fmpr_print(x); flint_printf("\n\n");
flint_printf("y = "); fmpr_print(y); flint_printf("\n\n");
flint_printf("z = "); fmpr_print(z); flint_printf("\n\n");
flint_printf("z_exact = "); fmpr_print(z_exact); flint_printf("\n\n");
flint_printf("true_error = "); fmpr_print(true_error); flint_printf("\n\n");
flint_printf("error_bound = "); fmpr_print(error_bound); flint_printf("\n\n");
abort();
}
fmpr_clear(z_exact);
fmpr_clear(error_bound);
fmpr_clear(true_error);
}
fmpr_clear(x);
fmpr_clear(y);
fmpr_clear(z);
fmpr_clear(w);
mpfr_clear(X);
mpfr_clear(Y);
mpfr_clear(Z);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
void
fmpcb_calc_cauchy_bound(fmprb_t bound, fmpcb_calc_func_t func, void * param,
const fmpcb_t x, const fmprb_t radius, long maxdepth, long prec)
{
long i, n, depth, wp;
fmprb_t pi, theta, v, s1, c1, s2, c2, st, ct;
fmpcb_t t, u;
fmprb_t b;
fmprb_init(pi);
fmprb_init(theta);
fmprb_init(v);
fmprb_init(s1);
fmprb_init(c1);
fmprb_init(s2);
fmprb_init(c2);
fmprb_init(st);
fmprb_init(ct);
fmpcb_init(t);
fmpcb_init(u);
fmprb_init(b);
wp = prec + 20;
fmprb_const_pi(pi, wp);
fmprb_zero_pm_inf(b);
for (depth = 0, n = 16; depth < maxdepth; n *= 2, depth++)
{
fmprb_zero(b);
/* theta = 2 pi / n */
fmprb_div_ui(theta, pi, n, wp);
fmprb_mul_2exp_si(theta, theta, 1);
/* sine and cosine of i*theta and (i+1)*theta */
fmprb_zero(s1);
fmprb_one(c1);
fmprb_sin_cos(st, ct, theta, wp);
fmprb_set(s2, st);
fmprb_set(c2, ct);
for (i = 0; i < n; i++)
{
/* sine and cosine of 2 pi ([i,i+1]/n) */
/* since we use power of two subdivision points, the
sine and cosine are monotone on each subinterval */
fmprb_union(fmpcb_realref(t), c1, c2, wp);
fmprb_union(fmpcb_imagref(t), s1, s2, wp);
fmpcb_mul_fmprb(t, t, radius, wp);
fmpcb_add(t, t, x, prec);
/* next angle */
fmprb_mul(v, c2, ct, wp);
fmprb_mul(c1, s2, st, wp);
fmprb_sub(c1, v, c1, wp);
fmprb_mul(v, c2, st, wp);
fmprb_mul(s1, s2, ct, wp);
fmprb_add(s1, v, s1, wp);
fmprb_swap(c1, c2);
fmprb_swap(s1, s2);
func(u, t, param, 1, prec);
fmpcb_abs(v, u, prec);
fmprb_add(b, b, v, prec);
}
fmprb_div_ui(b, b, n, prec);
if (fmprb_is_exact(b) || fmpr_cmp(fmprb_radref(b), fmprb_midref(b)) < 0)
break;
}
fmprb_set(bound, b);
fmprb_clear(pi);
fmprb_clear(theta);
fmprb_clear(v);
fmpcb_clear(t);
fmpcb_clear(u);
fmprb_clear(b);
fmprb_clear(s1);
fmprb_clear(c1);
fmprb_clear(s2);
fmprb_clear(c2);
fmprb_clear(st);
fmprb_clear(ct);
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("sum....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000000 * arb_test_multiplier(); iter++)
{
slong i, len, prec, bits, expbits, res1, res2;
fmpr_t s1, s2, s3, err, err_bound;
fmpr_struct terms[20];
fmpr_rnd_t rnd;
len = n_randint(state, 20);
bits = 2 + n_randint(state, 1000);
prec = 2 + n_randint(state, 1000);
expbits = n_randint(state, 14);
fmpr_init(s1);
fmpr_init(s2);
fmpr_init(s3);
fmpr_init(err);
fmpr_init(err_bound);
for (i = 0; i < len; i++)
{
fmpr_init(terms + i);
fmpr_randtest_special(terms + i, state, bits, expbits);
}
switch (n_randint(state, 4))
{
case 0: rnd = FMPR_RND_DOWN; break;
case 1: rnd = FMPR_RND_UP; break;
case 2: rnd = FMPR_RND_FLOOR; break;
default: rnd = FMPR_RND_CEIL; break;
}
res1 = fmpr_sum(s1, terms, len, prec, rnd);
fmpr_zero(s2);
for (i = 0; i < len; i++)
fmpr_add(s2, s2, terms + i, FMPR_PREC_EXACT, FMPR_RND_DOWN);
res2 = fmpr_set_round(s3, s2, prec, rnd);
if (!fmpr_equal(s1, s3) || res1 != res2 ||
!fmpr_check_ulp(s1, res1, prec) || !fmpr_check_ulp(s3, res2, prec))
{
flint_printf("FAIL (%wd)\n\n", iter);
flint_printf("prec = %wd\n\n", prec);
for (i = 0; i < len; i++)
{
flint_printf("terms[%wd] = ", i); fmpr_print(terms + i); flint_printf("\n\n");
}
flint_printf("s1 = "); fmpr_print(s1); flint_printf("\n\n");
flint_printf("s2 = "); fmpr_print(s2); flint_printf("\n\n");
flint_printf("s3 = "); fmpr_print(s3); flint_printf("\n\n");
flint_printf("res1 = %wd, res2 = %wd\n\n", res1, res2);
abort();
}
fmpr_sub(err, s1, s2, FMPR_PREC_EXACT, FMPR_RND_DOWN);
fmpr_abs(err, err);
fmpr_set_error_result(err_bound, s1, res1);
if (fmpr_cmp(err, err_bound) > 0)
{
flint_printf("FAIL (error bound)!\n");
flint_printf("prec = %wd\n\n", prec);
for (i = 0; i < len; i++)
{
flint_printf("terms[%wd] = ", i); fmpr_print(terms + i); flint_printf("\n\n");
}
flint_printf("s1 = "); fmpr_print(s1); flint_printf("\n\n");
flint_printf("s2 = "); fmpr_print(s2); flint_printf("\n\n");
flint_printf("s3 = "); fmpr_print(s3); flint_printf("\n\n");
flint_printf("error: "); fmpr_print(err); flint_printf("\n\n");
flint_printf("error bound: "); fmpr_print(err_bound); flint_printf("\n\n");
flint_printf("res1 = %wd, res2 = %wd\n\n", res1, res2);
abort();
}
fmpr_clear(s1);
fmpr_clear(s2);
fmpr_clear(s3);
fmpr_clear(err);
fmpr_clear(err_bound);
for (i = 0; i < len; i++)
fmpr_clear(terms + i);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int
fmprb_contains(const fmprb_t x, const fmprb_t y)
{
fmpr_t t;
fmpr_t u;
fmpr_struct tmp[4];
int left_ok, right_ok;
if (fmprb_is_exact(y))
return fmprb_contains_fmpr(x, fmprb_midref(y));
if (fmpr_is_nan(fmprb_midref(y)))
return fmpr_is_nan(fmprb_midref(x));
fmpr_init(t);
fmpr_init(u);
/* fast check */
fmpr_sub(t, fmprb_midref(x), fmprb_radref(x), 30, FMPR_RND_CEIL);
fmpr_sub(u, fmprb_midref(y), fmprb_radref(y), 30, FMPR_RND_FLOOR);
left_ok = fmpr_cmp(t, u) <= 0;
/* exact check */
if (!left_ok)
{
fmpr_init(tmp + 0);
fmpr_init(tmp + 1);
fmpr_init(tmp + 2);
fmpr_init(tmp + 3);
fmpr_set(tmp + 0, fmprb_midref(x));
fmpr_neg(tmp + 1, fmprb_radref(x));
fmpr_neg(tmp + 2, fmprb_midref(y));
fmpr_set(tmp + 3, fmprb_radref(y));
fmpr_sum(t, tmp, 4, 30, FMPR_RND_DOWN);
left_ok = fmpr_sgn(t) <= 0;
fmpr_clear(tmp + 0);
fmpr_clear(tmp + 1);
fmpr_clear(tmp + 2);
fmpr_clear(tmp + 3);
}
/* fast check */
fmpr_add(t, fmprb_midref(x), fmprb_radref(x), 30, FMPR_RND_FLOOR);
fmpr_add(u, fmprb_midref(y), fmprb_radref(y), 30, FMPR_RND_CEIL);
right_ok = (fmpr_cmp(t, u) >= 0);
/* exact check */
if (!right_ok)
{
fmpr_init(tmp + 0);
fmpr_init(tmp + 1);
fmpr_init(tmp + 2);
fmpr_init(tmp + 3);
fmpr_set(tmp + 0, fmprb_midref(x));
fmpr_set(tmp + 1, fmprb_radref(x));
fmpr_neg(tmp + 2, fmprb_midref(y));
fmpr_neg(tmp + 3, fmprb_radref(y));
fmpr_sum(t, tmp, 4, 30, FMPR_RND_DOWN);
right_ok = fmpr_sgn(t) >= 0;
fmpr_clear(tmp + 0);
fmpr_clear(tmp + 1);
fmpr_clear(tmp + 2);
fmpr_clear(tmp + 3);
}
fmpr_clear(t);
fmpr_clear(u);
return left_ok && right_ok;
}
