#include "fmpz.h"
#include "fmpz_vec.h"
#include "n_poly.h"
#include "mpoly.h"
#include "fq_nmod_mpoly.h"
int fq_nmod_mpoly_pow_fmpz(fq_nmod_mpoly_t A, const fq_nmod_mpoly_t B,
const fmpz_t k, const fq_nmod_mpoly_ctx_t ctx)
{
slong i;
fmpz * maxBfields;
flint_bitcnt_t exp_bits;
TMP_INIT;
if (fmpz_sgn(k) < 0)
flint_throw(FLINT_ERROR, "fq_nmod_mpoly_pow_fmpz: power is negative");
if (fmpz_fits_si(k))
return fq_nmod_mpoly_pow_ui(A, B, fmpz_get_ui(k), ctx);
if (B->length == 0)
{
fq_nmod_mpoly_zero(A, ctx);
return 1;
}
if (B->length != 1)
return 0;
TMP_START;
maxBfields = (fmpz *) TMP_ALLOC(ctx->minfo->nfields*sizeof(fmpz));
for (i = 0; i < ctx->minfo->nfields; i++)
fmpz_init(maxBfields + i);
mpoly_max_fields_fmpz(maxBfields, B->exps, B->length, B->bits, ctx->minfo);
_fmpz_vec_scalar_mul_fmpz(maxBfields, maxBfields, ctx->minfo->nfields, k);
exp_bits = 1 + _fmpz_vec_max_bits(maxBfields, ctx->minfo->nfields);
exp_bits = mpoly_fix_bits(exp_bits, ctx->minfo);
fq_nmod_mpoly_fit_length_reset_bits(A, 1, exp_bits, ctx);
n_fq_pow_fmpz(A->coeffs, B->coeffs, k, ctx->fqctx);
mpoly_pack_vec_fmpz(A->exps, maxBfields, exp_bits, ctx->minfo->nfields, 1);
_fq_nmod_mpoly_set_length(A, 1, ctx);
for (i = 0; i < ctx->minfo->nfields; i++)
fmpz_clear(maxBfields + i);
TMP_END;
return 1;
}