#include "ulong_extras.h"
#include "nmod_poly.h"
#include "nf_elem.h"
void _nf_elem_get_nmod_poly(nmod_poly_t pol, const nf_elem_t a, const nf_t nf)
{
if (nf_elem_is_zero(a, nf))
{
nmod_poly_zero(pol);
return;
}
if (nf->flag & NF_LINEAR)
{
{
nmod_poly_fit_length(pol, 1);
pol->coeffs[0] = fmpz_get_nmod(LNF_ELEM_NUMREF(a), pol->mod);
_nmod_poly_set_length(pol, 1);
_nmod_poly_normalise(pol);
}
} else if (nf->flag & NF_QUADRATIC)
{
nmod_poly_fit_length(pol, 3);
pol->coeffs[0] = fmpz_get_nmod(QNF_ELEM_NUMREF(a), pol->mod);
pol->coeffs[1] = fmpz_get_nmod(QNF_ELEM_NUMREF(a) + 1, pol->mod);
pol->coeffs[2] = fmpz_get_nmod(QNF_ELEM_NUMREF(a) + 2, pol->mod);
_nmod_poly_set_length(pol, 3);
_nmod_poly_normalise(pol);
} else
{
slong len = NF_ELEM(a)->length;
slong i;
nmod_poly_fit_length(pol, len);
for (i = 0; i < len; i++)
pol->coeffs[i] = fmpz_get_nmod(NF_ELEM(a)->coeffs + i, pol->mod);
_nmod_poly_set_length(pol, len);
_nmod_poly_normalise(pol);
}
}
void nf_elem_get_nmod_poly_den(nmod_poly_t pol, const nf_elem_t a, const nf_t nf, int den)
{
_nf_elem_get_nmod_poly(pol, a, nf);
if (den)
{
if (nf->flag & NF_LINEAR)
nmod_poly_scalar_mul_nmod(pol, pol, n_invmod(fmpz_get_nmod(LNF_ELEM_DENREF(a), pol->mod), pol->mod.n));
else if (nf->flag & NF_QUADRATIC)
nmod_poly_scalar_mul_nmod(pol, pol, n_invmod(fmpz_get_nmod(QNF_ELEM_DENREF(a), pol->mod), pol->mod.n));
else
nmod_poly_scalar_mul_nmod(pol, pol, n_invmod(fmpz_get_nmod(NF_ELEM_DENREF(a), pol->mod), pol->mod.n));
}
}
void nf_elem_get_nmod_poly(nmod_poly_t pol, const nf_elem_t a, const nf_t nf)
{
nf_elem_get_nmod_poly_den(pol, a, nf, 1);
}