#include "gr_vec.h"
#include "gr_poly.h"
int
_gr_poly_divrem_newton_n_preinv(
gr_ptr Q,
gr_ptr R,
gr_srcptr A, slong lenA,
gr_srcptr B, slong lenB,
gr_srcptr Binv, slong lenBinv,
gr_ctx_t ctx)
{
slong lenQ = lenA - lenB + 1;
int status = GR_SUCCESS;
status |= _gr_poly_div_newton_n_preinv(Q, A, lenA, B, lenB, Binv, lenBinv, ctx);
if (lenB > 1)
{
status |= _gr_poly_mullow(R, Q, lenQ, B, lenB - 1, lenB - 1, ctx);
status |= _gr_vec_sub(R, A, R, lenB - 1, ctx);
}
return status;
}
int
gr_poly_divrem_newton_n_preinv(gr_poly_t Q,
gr_poly_t R,
const gr_poly_t A,
const gr_poly_t B,
const gr_poly_t Binv,
gr_ctx_t ctx)
{
slong lenA = A->length, lenB = B->length, lenQ = lenA - lenB + 1;
slong lenBinv = Binv->length;
gr_poly_t tQ, tR;
gr_ptr q, r;
int status = GR_SUCCESS;
if (lenB == 0)
return GR_DOMAIN;
if (lenA < lenB)
{
status |= gr_poly_set(R, A, ctx);
status |= gr_poly_zero(Q, ctx);
return status;
}
if (lenA > 2 * lenB - 2)
return GR_UNABLE;
if (Q == A || Q == B || Q == Binv)
{
gr_poly_init2(tQ, lenQ, ctx);
q = tQ->coeffs;
}
else
{
gr_poly_fit_length(Q, lenQ, ctx);
q = Q->coeffs;
}
if (R == A || R == B || R == Binv)
{
gr_poly_init2(tR, lenB - 1, ctx);
r = tR->coeffs;
}
else
{
gr_poly_fit_length(R, lenB - 1, ctx);
r = R->coeffs;
}
status |= _gr_poly_divrem_newton_n_preinv(q, r, A->coeffs, lenA, B->coeffs, lenB, Binv->coeffs, lenBinv, ctx);
if (Q == A || Q == B || Q == Binv)
{
gr_poly_swap(tQ, Q, ctx);
gr_poly_clear(tQ, ctx);
}
else
{
_gr_poly_set_length(Q, lenQ, ctx);
}
if (R == A || R == B || R == Binv)
{
gr_poly_swap(tR, R, ctx);
gr_poly_clear(tR, ctx);
}
_gr_poly_set_length(Q, lenA - lenB + 1, ctx);
_gr_poly_set_length(R, lenB - 1, ctx);
_gr_poly_normalise(Q, ctx);
_gr_poly_normalise(R, ctx);
return status;
}