#include "fq_nmod.h"
#include "n_poly.h"
#include "mpoly.h"
#include "fq_nmod_mpoly.h"
void fq_nmod_mpoly_sub_fq_nmod(
fq_nmod_mpoly_t A,
const fq_nmod_mpoly_t B,
const fq_nmod_t c,
const fq_nmod_mpoly_ctx_t ctx)
{
slong d = fq_nmod_ctx_degree(ctx->fqctx);
slong N = mpoly_words_per_exp(B->bits, ctx->minfo);
slong Blen = B->length;
fq_nmod_mpoly_assert_canonical(B, ctx);
if (fq_nmod_is_zero(c, ctx->fqctx))
{
fq_nmod_mpoly_set(A, B, ctx);
return;
}
if (Blen < 1)
{
fq_nmod_mpoly_set_fq_nmod(A, c, ctx);
fq_nmod_mpoly_neg(A, A, ctx);
return;
}
if (mpoly_monomial_is_zero(B->exps + (Blen - 1)*N, N))
{
if (A != B)
{
fq_nmod_mpoly_fit_length_reset_bits(A, Blen, B->bits, ctx);
_nmod_vec_set(A->coeffs, B->coeffs, d*(Blen - 1));
mpoly_copy_monomials(A->exps, B->exps, Blen, N);
_fq_nmod_mpoly_set_length(A, Blen, ctx);
}
n_fq_sub_fq_nmod(A->coeffs + d*(Blen - 1), B->coeffs + d*(Blen - 1), c, ctx->fqctx);
if (_n_fq_is_zero(A->coeffs + d*(Blen - 1), d))
_fq_nmod_mpoly_set_length(A, Blen - 1, ctx);
}
else
{
if (A != B)
{
fq_nmod_mpoly_fit_length_reset_bits(A, Blen + 1, B->bits, ctx);
_nmod_vec_set(A->coeffs, B->coeffs, d*Blen);
mpoly_copy_monomials(A->exps, B->exps, Blen, N);
}
else
{
fq_nmod_mpoly_fit_length(A, Blen + 1, ctx);
}
mpoly_monomial_zero(A->exps + N*Blen, N);
n_fq_set_fq_nmod(A->coeffs + d*Blen, c, ctx->fqctx);
_n_fq_neg(A->coeffs + d*Blen, A->coeffs + d*Blen, d, fq_nmod_ctx_mod(ctx->fqctx));
_fq_nmod_mpoly_set_length(A, Blen + 1, ctx);
}
}