#include "arb_poly.h"
#include "acb_poly.h"
#include "gr_poly.h"
void
_acb_poly_pow_ui_trunc_binexp(acb_ptr res,
acb_srcptr f, slong flen, ulong exp, slong len, slong prec)
{
if (exp <= 2)
{
if (exp == 0)
acb_one(res);
else if (exp == 1)
_acb_vec_set_round(res, f, len, prec);
else
_acb_poly_mullow(res, f, flen, f, flen, len, prec);
}
else if (!_acb_vec_is_finite(f, flen))
{
_acb_vec_indeterminate(res, len);
}
else
{
gr_ctx_t ctx;
gr_ctx_init_complex_acb(ctx, prec);
GR_MUST_SUCCEED(_gr_poly_pow_series_ui_binexp(res, f, flen, exp, len, ctx));
}
}
void
acb_poly_pow_ui_trunc_binexp(acb_poly_t res,
const acb_poly_t poly, ulong exp, slong len, slong prec)
{
slong flen, rlen;
flen = poly->length;
if (exp == 0 && len != 0)
{
acb_poly_one(res);
}
else if (flen == 0 || len == 0)
{
acb_poly_zero(res);
}
else
{
rlen = poly_pow_length(flen, exp, len);
if (res != poly)
{
acb_poly_fit_length(res, rlen);
_acb_poly_pow_ui_trunc_binexp(res->coeffs,
poly->coeffs, flen, exp, rlen, prec);
_acb_poly_set_length(res, rlen);
_acb_poly_normalise(res);
}
else
{
acb_poly_t t;
acb_poly_init2(t, rlen);
_acb_poly_pow_ui_trunc_binexp(t->coeffs,
poly->coeffs, flen, exp, rlen, prec);
_acb_poly_set_length(t, rlen);
_acb_poly_normalise(t);
acb_poly_swap(res, t);
acb_poly_clear(t);
}
}
}