#include "mpn_extras.h"
#include "fmpz.h"
#include "fmpz_mod.h"
#include "fmpz_mod_poly.h"
void fmpz_mod_poly_set_coeff_si(fmpz_mod_poly_t poly, slong n, slong x,
const fmpz_mod_ctx_t ctx)
{
fmpz_mod_poly_fit_length(poly, n + 1, ctx);
if (n + 1 > poly->length)
{
flint_mpn_zero((nn_ptr) (poly->coeffs + poly->length), n - poly->length);
poly->length = n + 1;
}
fmpz_set_si(poly->coeffs + n, x);
fmpz_mod(poly->coeffs + n, poly->coeffs + n, fmpz_mod_ctx_modulus(ctx));
_fmpz_mod_poly_normalise(poly);
}
void fmpz_mod_poly_set_coeff_ui(fmpz_mod_poly_t poly, slong n, ulong x,
const fmpz_mod_ctx_t ctx)
{
if (x == 0)
{
if (n >= poly->length)
return;
fmpz_zero(poly->coeffs + n);
}
else
{
fmpz_mod_poly_fit_length(poly, n + 1, ctx);
if (n + 1 > poly->length)
{
flint_mpn_zero((nn_ptr) (poly->coeffs + poly->length), n - poly->length);
poly->length = n + 1;
}
fmpz_set_ui(poly->coeffs + n, x);
fmpz_mod(poly->coeffs + n, poly->coeffs + n, fmpz_mod_ctx_modulus(ctx));
}
if (n == poly->length - 1)
_fmpz_mod_poly_normalise(poly);
}
void fmpz_mod_poly_set_coeff_fmpz(fmpz_mod_poly_t poly, slong n, const fmpz_t x,
const fmpz_mod_ctx_t ctx)
{
if (fmpz_is_zero(x))
{
if (n >= poly->length)
return;
fmpz_zero(poly->coeffs + n);
}
else
{
fmpz_mod_poly_fit_length(poly, n + 1, ctx);
if (n + 1 > poly->length)
{
flint_mpn_zero((nn_ptr) (poly->coeffs + poly->length), n - poly->length);
poly->length = n + 1;
}
fmpz_mod_set_fmpz(poly->coeffs + n, x, ctx);
}
if (n == poly->length - 1)
_fmpz_mod_poly_normalise(poly);
}
void fmpz_mod_poly_get_coeff_fmpz(fmpz_t x, const fmpz_mod_poly_t poly,
slong n, const fmpz_mod_ctx_t FLINT_UNUSED(ctx))
{
if (n < poly->length)
fmpz_set(x, poly->coeffs + n);
else
fmpz_zero(x);
}