#include "nmod_poly.h"
#include "gr_poly.h"
void
_nmod_poly_sqrt_series(nn_ptr g, nn_srcptr h, slong hlen, slong n, nmod_t mod)
{
gr_ctx_t ctx;
_gr_ctx_init_nmod(ctx, &mod);
GR_MUST_SUCCEED(_gr_poly_sqrt_series(g, h, hlen, n, ctx));
}
void
nmod_poly_sqrt_series(nmod_poly_t res, const nmod_poly_t h, slong len)
{
slong hlen;
hlen = h->length;
if (hlen == 0 || len == 0)
{
nmod_poly_zero(res);
return;
}
if (hlen == 1)
len = 1;
if (res == h)
{
nmod_poly_t t;
nmod_poly_init_preinv(t, h->mod.n, h->mod.ninv);
nmod_poly_sqrt_series(t, h, len);
nmod_poly_swap(res, t);
nmod_poly_clear(t);
}
else
{
nmod_poly_fit_length(res, len);
_nmod_poly_sqrt_series(res->coeffs, h->coeffs, h->length, len, h->mod);
_nmod_poly_set_length(res, len);
_nmod_poly_normalise(res);
}
}