/* todo: do this using underscore methods */
static void
bsplit(acb_poly_t A, acb_poly_t B, acb_poly_t C,
acb_srcptr a, slong p, acb_srcptr b, slong q,
const acb_t z, slong an, slong bn, slong prec)
{
if (bn - an == 1)
{
evaluate(A, a, p, z, an, prec);
evaluate(B, b, q, NULL, an, prec);
acb_poly_set(C, B);
}
else if (bn - an == 2) /* inlined */
{
acb_poly_t A2, B2;
acb_poly_init(A2);
acb_poly_init(B2);
evaluate(A, a, p, z, an, prec);
evaluate(A2, a, p, z, an + 1, prec);
evaluate(B, b, q, NULL, an, prec);
evaluate(B2, b, q, NULL, an + 1, prec);
acb_poly_mul(C, B, B2, prec);
acb_poly_set(B, C);
acb_poly_mul(C, A, B2, prec);
acb_poly_add(C, C, B, prec);
acb_poly_mul(A2, A, A2, prec);
acb_poly_swap(A, A2);
acb_poly_clear(A2);
acb_poly_clear(B2);
}
else
{
slong m = an + (bn - an) / 2;
acb_poly_t A2, B2, C2, T;
acb_poly_init(A2);
acb_poly_init(B2);
acb_poly_init(C2);
acb_poly_init(T);
bsplit(A, B, C, a, p, b, q, z, an, m, prec);
bsplit(A2, B2, C2, a, p, b, q, z, m, bn, prec);
acb_poly_mul(T, B2, C, prec);
acb_poly_mul(C, A, C2, prec);
acb_poly_add(C, C, T, prec);
acb_poly_mul(C2, B, B2, prec);
acb_poly_swap(B, C2);
acb_poly_mul(B2, A, A2, prec);
acb_poly_swap(A, B2);
acb_poly_clear(A2);
acb_poly_clear(B2);
acb_poly_clear(C2);
acb_poly_clear(T);
}
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("cos_pi_series....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
slong m, n1, n2, bits1, bits2, bits3;
acb_poly_t S, A, B, C;
acb_t pi;
bits1 = 2 + n_randint(state, 200);
bits2 = 2 + n_randint(state, 200);
bits3 = 2 + n_randint(state, 200);
m = 1 + n_randint(state, 30);
n1 = 1 + n_randint(state, 30);
n2 = 1 + n_randint(state, 30);
acb_poly_init(S);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
acb_init(pi);
acb_poly_randtest(S, state, m, bits1, 3);
acb_poly_randtest(A, state, m, bits1, 3);
acb_poly_randtest(B, state, m, bits1, 3);
acb_poly_cos_pi_series(A, S, n1, bits2);
acb_const_pi(pi, bits3);
acb_poly_set_acb(B, pi);
acb_poly_mul(B, S, B, bits3);
acb_poly_cos_series(B, B, n2, bits3);
acb_poly_set(C, A);
acb_poly_truncate(C, FLINT_MIN(n1, n2));
acb_poly_truncate(B, FLINT_MIN(n1, n2));
if (!acb_poly_overlaps(B, C))
{
flint_printf("FAIL\n\n");
flint_printf("S = "); acb_poly_printd(S, 15); flint_printf("\n\n");
flint_printf("A = "); acb_poly_printd(A, 15); flint_printf("\n\n");
flint_printf("B = "); acb_poly_printd(B, 15); flint_printf("\n\n");
abort();
}
acb_poly_cos_pi_series(S, S, n1, bits2);
if (!acb_poly_overlaps(A, S))
{
flint_printf("FAIL (aliasing)\n\n");
abort();
}
acb_poly_clear(S);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
acb_clear(pi);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
int main()
{
slong iter;
flint_rand_t state;
flint_printf("find_roots....");
fflush(stdout);
flint_randinit(state);
for (iter = 0; iter < 1000 * arb_test_multiplier(); iter++)
{
acb_poly_t A;
acb_poly_t B;
acb_poly_t C;
acb_t t;
acb_ptr roots;
slong i, deg, isolated;
slong prec = 10 + n_randint(state, 400);
acb_init(t);
acb_poly_init(A);
acb_poly_init(B);
acb_poly_init(C);
do {
acb_poly_randtest(A, state, 2 + n_randint(state, 15), prec, 5);
} while (A->length == 0);
deg = A->length - 1;
roots = _acb_vec_init(deg);
isolated = acb_poly_find_roots(roots, A, NULL, 0, prec);
if (isolated == deg)
{
acb_poly_fit_length(B, 1);
acb_set(B->coeffs, A->coeffs + deg);
_acb_poly_set_length(B, 1);
for (i = 0; i < deg; i++)
{
acb_poly_fit_length(C, 2);
acb_one(C->coeffs + 1);
acb_neg(C->coeffs + 0, roots + i);
_acb_poly_set_length(C, 2);
acb_poly_mul(B, B, C, prec);
}
if (!acb_poly_contains(B, A))
{
flint_printf("FAIL: product does not equal polynomial\n");
acb_poly_printd(A, 15); flint_printf("\n\n");
acb_poly_printd(B, 15); flint_printf("\n\n");
flint_abort();
}
}
for (i = 0; i < isolated; i++)
{
acb_poly_evaluate(t, A, roots + i, prec);
if (!acb_contains_zero(t))
{
flint_printf("FAIL: poly(root) does not contain zero\n");
acb_poly_printd(A, 15); flint_printf("\n\n");
acb_printd(roots + i, 15); flint_printf("\n\n");
acb_printd(t, 15); flint_printf("\n\n");
flint_abort();
}
}
_acb_vec_clear(roots, deg);
acb_clear(t);
acb_poly_clear(A);
acb_poly_clear(B);
acb_poly_clear(C);
}
flint_randclear(state);
flint_cleanup();
flint_printf("PASS\n");
return EXIT_SUCCESS;
}
