#include "nmod_vec.h"
#include "nmod_poly.h"
void _nmod_poly_mulmod(nn_ptr res, nn_srcptr poly1, slong len1,
nn_srcptr poly2, slong len2, nn_srcptr f,
slong lenf, nmod_t mod)
{
nn_ptr T, Q;
slong lenT, lenQ;
TMP_INIT;
if (lenf == 2 && len1 == 1 && len2 == 1)
{
res[0] = nmod_mul(poly1[0], poly2[0], mod);
return;
}
lenT = len1 + len2 - 1;
lenQ = lenT - lenf + 1;
FLINT_ASSERT(lenQ >= 1);
TMP_START;
T = TMP_ALLOC(sizeof(ulong) * (lenT + lenQ));
Q = T + lenT;
if (len1 >= len2)
_nmod_poly_mul(T, poly1, len1, poly2, len2, mod);
else
_nmod_poly_mul(T, poly2, len2, poly1, len1, mod);
_nmod_poly_divrem(Q, res, T, lenT, f, lenf, mod);
TMP_END;
}
void
nmod_poly_mulmod(nmod_poly_t res,
const nmod_poly_t poly1, const nmod_poly_t poly2, const nmod_poly_t f)
{
slong len1, len2, lenf;
nn_ptr fcoeffs;
lenf = f->length;
len1 = poly1->length;
len2 = poly2->length;
if (lenf == 0)
{
flint_throw(FLINT_ERROR, "Exception (nmod_poly_mulmod). Divide by zero.\n");
}
if (lenf == 1 || len1 == 0 || len2 == 0)
{
nmod_poly_zero(res);
return;
}
if (len1 + len2 - lenf > 0)
{
if (f == res)
{
fcoeffs = flint_malloc(sizeof(ulong) * lenf);
_nmod_vec_set(fcoeffs, f->coeffs, lenf);
}
else
fcoeffs = f->coeffs;
nmod_poly_fit_length(res, lenf - 1);
_nmod_poly_mulmod(res->coeffs, poly1->coeffs, len1,
poly2->coeffs, len2,
fcoeffs, lenf,
res->mod);
if (f == res)
flint_free(fcoeffs);
res->length = lenf - 1;
_nmod_poly_normalise(res);
}
else
{
nmod_poly_mul(res, poly1, poly2);
}
}