#include "gr_vec.h"
#include "gr_poly.h"
int
_gr_poly_divexact_series_basecase_noinv(gr_ptr Q,
gr_srcptr A, slong Alen,
gr_srcptr B, slong Blen, slong len, gr_ctx_t ctx)
{
int status = GR_SUCCESS;
slong sz = ctx->sizeof_elem;
slong i, l;
if (len == 0)
return GR_SUCCESS;
if (Blen == 0)
return GR_DOMAIN;
Alen = FLINT_MIN(Alen, len);
Blen = FLINT_MIN(Blen, len);
if (Blen == 1)
{
status |= _gr_vec_divexact_scalar(Q, A, Alen, B, ctx);
status |= _gr_vec_zero(GR_ENTRY(Q, Alen, sz), len - Alen, ctx);
return status;
}
status = gr_divexact(Q, A, B, ctx);
if (status == GR_SUCCESS)
{
for (i = 1; i < len; i++)
{
l = FLINT_MIN(i, Blen - 1);
status |= _gr_vec_dot_rev(GR_ENTRY(Q, i, sz), (i < Alen) ? GR_ENTRY(A, i, sz) : NULL, 1, GR_ENTRY(B, 1, sz), GR_ENTRY(Q, i - l, sz), l, ctx);
status |= gr_divexact(GR_ENTRY(Q, i, sz), GR_ENTRY(Q, i, sz), B, ctx);
if (status != GR_SUCCESS)
break;
}
}
return status;
}
int
_gr_poly_divexact_series_basecase(gr_ptr Q,
gr_srcptr A, slong Alen,
gr_srcptr B, slong Blen, slong len, gr_ctx_t ctx)
{
int status = GR_SUCCESS;
slong sz = ctx->sizeof_elem;
if (len == 0)
return GR_SUCCESS;
if (Blen == 0)
return GR_DOMAIN;
Alen = FLINT_MIN(Alen, len);
Blen = FLINT_MIN(Blen, len);
if (Blen == 1)
{
status |= _gr_vec_divexact_scalar(Q, A, Alen, B, ctx);
status |= _gr_vec_zero(GR_ENTRY(Q, Alen, sz), len - Alen, ctx);
return status;
}
{
gr_ptr q;
GR_TMP_INIT(q, ctx);
status = gr_inv(q, B, ctx);
if (status == GR_SUCCESS)
status |= _gr_poly_div_series_basecase_preinv1(Q, A, Alen, B, Blen, q, len, ctx);
else
status = _gr_poly_divexact_series_basecase_noinv(Q, A, Alen, B, Blen, len, ctx);
GR_TMP_CLEAR(q, ctx);
}
return status;
}
int
gr_poly_divexact_series_basecase(gr_poly_t Q, const gr_poly_t A, const gr_poly_t B, slong len, gr_ctx_t ctx)
{
int status = GR_SUCCESS;
if (len == 0)
return gr_poly_zero(Q, ctx);
if (B->length == 0)
return GR_DOMAIN;
if (A->length == 0)
{
truth_t is_zero = gr_poly_is_zero(B, ctx);
if (is_zero == T_FALSE)
return gr_poly_zero(Q, ctx);
return GR_UNABLE;
}
if (Q == A || Q == B)
{
gr_poly_t t;
gr_poly_init(t, ctx);
status = gr_poly_divexact_series_basecase(t, A, B, len, ctx);
gr_poly_swap(Q, t, ctx);
gr_poly_clear(t, ctx);
return status;
}
gr_poly_fit_length(Q, len, ctx);
status = _gr_poly_divexact_series_basecase(Q->coeffs, A->coeffs, A->length, B->coeffs, B->length, len, ctx);
_gr_poly_set_length(Q, len, ctx);
_gr_poly_normalise(Q, ctx);
return status;
}