#include "fmpz.h"
#include "nf.h"
void nf_init(nf_t nf, const fmpq_poly_t pol)
{
slong i, j;
slong len = pol->length, deg = len - 1;
fmpz * pow, * lead = fmpq_poly_numref(pol) + len - 1;
fmpq_poly_init(nf->pol);
fmpq_poly_set(nf->pol, pol);
if (fmpz_is_one(fmpq_poly_denref(pol))
&& fmpz_is_one(lead))
nf->flag = NF_MONIC;
else
{
fmpz_preinvn_init(nf->pinv.qq, lead);
nf->flag = NF_GENERIC;
}
if (len < 2)
{
flint_throw(FLINT_ERROR, "Exception (nf_init). Degree must be at least 1.\n");
} else if (len == 2)
nf->flag |= NF_LINEAR;
else if (len == 3)
{
nf->flag |= NF_QUADRATIC;
if (fmpz_is_one(pol->coeffs + 0) && fmpz_is_zero(pol->coeffs + 1) &&
fmpz_is_one(pol->coeffs + 2) && fmpz_is_one(pol->den))
nf->flag |= NF_GAUSSIAN;
}
else if (len <= NF_POWERS_CUTOFF)
{
if (nf->flag & NF_MONIC)
{
nf->powers.zz->powers = _fmpz_poly_powers_precompute(fmpq_poly_numref(pol),
len);
nf->powers.zz->len = len;
}
else
{
nf->powers.qq->powers = _fmpq_poly_powers_precompute(fmpq_poly_numref(pol),
fmpq_poly_denref(pol), len);
nf->powers.qq->len = len;
}
}
fmpq_poly_init2(nf->traces, deg);
pow = fmpq_poly_denref(nf->traces);
for (i = 1; i < deg; i++)
{
fmpz_mul_si(fmpq_poly_numref(nf->traces) + i,
fmpq_poly_numref(pol) + deg - i, i);
for (j = i - 1; j >= 1; j--)
{
fmpz_mul(fmpq_poly_numref(nf->traces) + i,
fmpq_poly_numref(nf->traces) + i, lead);
fmpz_addmul(fmpq_poly_numref(nf->traces) + i,
fmpq_poly_numref(pol) + deg - j,
fmpq_poly_numref(nf->traces) + i - j);
}
fmpz_neg(fmpq_poly_numref(nf->traces) + i,
fmpq_poly_numref(nf->traces) + i);
}
for (i = 1; i < deg; i++)
{
fmpz_mul(fmpq_poly_numref(nf->traces) + deg - i,
fmpq_poly_numref(nf->traces) + deg - i, pow);
fmpz_mul(pow, pow, lead);
}
fmpz_mul_si(fmpq_poly_numref(nf->traces), pow, deg);
}