void
fmprb_agm(fmprb_t z, const fmprb_t x, const fmprb_t y, long prec)
{
fmprb_t t, u, v, w;
if (fmprb_contains_negative(x) || fmprb_contains_negative(y))
{
fmprb_indeterminate(z);
return;
}
if (fmprb_is_zero(x) || fmprb_is_zero(y))
{
fmprb_zero(z);
return;
}
fmprb_init(t);
fmprb_init(u);
fmprb_init(v);
fmprb_init(w);
fmprb_set(t, x);
fmprb_set(u, y);
while (!fmprb_overlaps(t, u) &&
!fmprb_contains_nonpositive(t) &&
!fmprb_contains_nonpositive(u))
{
fmprb_add(v, t, u, prec);
fmprb_mul_2exp_si(v, v, -1);
fmprb_mul(w, t, u, prec);
fmprb_sqrt(w, w, prec);
fmprb_swap(v, t);
fmprb_swap(w, u);
}
if (!fmprb_is_finite(t) || !fmprb_is_finite(u))
{
fmprb_indeterminate(z);
}
else
{
fmprb_union(z, t, u, prec);
}
fmprb_clear(t);
fmprb_clear(u);
fmprb_clear(v);
fmprb_clear(w);
}
void
fmprb_sqrtpos(fmprb_t z, const fmprb_t x, slong prec)
{
if (!fmprb_is_finite(x))
{
if (fmpr_is_zero(fmprb_radref(x)) && fmpr_is_pos_inf(fmprb_midref(x)))
fmprb_pos_inf(z);
else
fmprb_zero_pm_inf(z);
}
else if (fmprb_contains_nonpositive(x))
{
fmpr_t t;
fmpr_init(t);
fmpr_add(t, fmprb_midref(x), fmprb_radref(x),
FMPRB_RAD_PREC, FMPR_RND_CEIL);
if (fmpr_sgn(t) <= 0)
{
fmprb_zero(z);
}
else
{
fmpr_sqrt(t, t, FMPRB_RAD_PREC, FMPR_RND_CEIL);
fmpr_mul_2exp_si(t, t, -1);
fmpr_set(fmprb_midref(z), t);
fmpr_set(fmprb_radref(z), t);
}
fmpr_clear(t);
}
else
{
fmprb_sqrt(z, x, prec);
}
fmprb_nonnegative_part(z, z, prec);
}
