#include "pari.h"
#include "paripriv.h"
GEN
eltreltoabs(GEN rnfeq, GEN x)
{
long i, k, v;
pari_sp av = avma;
GEN T, pol, teta, a, s;
pol = gel(rnfeq,1);
a = gel(rnfeq,2);
k = itos(gel(rnfeq,3));
T = gel(rnfeq,4);
v = varn(pol);
if (varncmp(gvar(x), v) > 0) x = scalarpol(x,v);
x = RgX_nffix("eltreltoabs", T, x, 1);
teta = gadd(pol_x(v), gmulsg(-k,a));
s = gen_0;
for (i=lg(x)-1; i>1; i--)
{
GEN c = gel(x,i);
if (typ(c) == t_POL) c = RgX_RgXQ_eval(c, a, pol);
s = RgX_rem(gadd(c, gmul(teta,s)), pol);
}
return gerepileupto(av, s);
}
GEN
rnfeltreltoabs(GEN rnf,GEN x)
{
const char *f = "rnfeltreltoabs";
GEN pol;
checkrnf(rnf);
pol = rnf_get_polabs(rnf);
switch(typ(x))
{
case t_INT: return icopy(x);
case t_FRAC: return gcopy(x);
case t_POLMOD:
if (RgX_equal_var(gel(x,1), pol))
{
if (rnf_get_nfdegree(rnf) == 1)
{
GEN y = gel(x,2);
pari_sp av = avma;
y = simplify_shallow(liftpol_shallow(y));
return gerepilecopy(av, mkpolmod(y, pol));
}
return gcopy(x);
}
x = polmod_nffix(f,rnf,x,0);
if (typ(x) == t_POLMOD) return rnfeltup(rnf,x);
retmkpolmod(eltreltoabs(rnf_get_map(rnf), x), ZX_copy(pol));
case t_POL:
if (varn(x) == rnf_get_nfvarn(rnf)) return rnfeltup(rnf,x);
retmkpolmod(eltreltoabs(rnf_get_map(rnf), x), ZX_copy(pol));
}
pari_err_TYPE(f,x); return NULL;
}
GEN
eltabstorel_lift(GEN rnfeq, GEN P)
{
GEN k, T = gel(rnfeq,4), relpol = gel(rnfeq,5);
if (is_scalar_t(typ(P))) return P;
k = gel(rnfeq,3);
P = lift_shallow(P);
if (signe(k)) P = RgXQX_translate(P, deg1pol_shallow(k, gen_0, varn(T)), T);
P = RgXQX_rem(P, relpol, T);
return QXQX_to_mod_shallow(P, T);
}
GEN
eltabstorel(GEN rnfeq, GEN P)
{
GEN T = gel(rnfeq,4), relpol = gel(rnfeq,5);
return mkpolmod(eltabstorel_lift(rnfeq,P), QXQX_to_mod_shallow(relpol,T));
}
GEN
rnfeltabstorel(GEN rnf,GEN x)
{
const char *f = "rnfeltabstorel";
pari_sp av = avma;
GEN pol, T, P, NF;
checkrnf(rnf);
T = rnf_get_nfpol(rnf);
P = rnf_get_pol(rnf);
pol = rnf_get_polabs(rnf);
switch(typ(x))
{
case t_INT: return icopy(x);
case t_FRAC: return gcopy(x);
case t_POLMOD:
if (RgX_equal_var(P, gel(x,1)))
{
x = polmod_nffix(f, rnf, x, 0);
P = QXQX_to_mod_shallow(P,T);
return gerepilecopy(av, mkpolmod(x,P));
}
if (RgX_equal_var(T, gel(x,1))) { x = Rg_nffix(f, T, x, 0); goto END; }
if (!RgX_equal_var(pol, gel(x,1))) pari_err_MODULUS(f, gel(x,1),pol);
x = gel(x,2); break;
case t_POL: break;
case t_COL:
NF = obj_check(rnf, rnf_NFABS);
if (!NF) pari_err_TYPE("rnfeltabstorel, apply nfinit(rnf)",x);
x = nf_to_scalar_or_alg(NF,x); break;
default:
pari_err_TYPE(f,x);
return NULL;
}
switch(typ(x))
{
case t_INT: return icopy(x);
case t_FRAC: return gcopy(x);
case t_POL: break;
default: pari_err_TYPE(f, x);
}
RgX_check_QX(x,f);
if (varn(x) != varn(pol))
{
if (varn(x) == varn(T)) { x = Rg_nffix(f,T,x,0); goto END; }
pari_err_VAR(f, x,pol);
}
switch(lg(x))
{
case 2: avma = av; return gen_0;
case 3: return gerepilecopy(av, gel(x,2));
}
END:
return gerepilecopy(av, eltabstorel(rnf_get_map(rnf), x));
}
static GEN
modulereltoabs(GEN rnf, GEN x)
{
GEN W=gel(x,1), I=gel(x,2), rnfeq = rnf_get_map(rnf), polabs = gel(rnfeq,1);
long i, j, k, m, N = lg(W)-1;
GEN zknf, dzknf, M;
if (!N) return cgetg(1, t_VEC);
zknf = rnf_get_nfzk(rnf);
dzknf = gel(zknf,1);
m = rnf_get_nfdegree(rnf);
M = cgetg(N*m+1, t_VEC);
for (k=i=1; i<=N; i++)
{
GEN c0, cid, w = gel(W,i), id = gel(I,i);
if (lg(id) == 1) continue;
id = Q_primitive_part(id, &cid);
w = Q_primitive_part(eltreltoabs(rnfeq,w), &c0);
c0 = mul_content(c0, mul_content(cid,inv_content(dzknf)));
if (typ(id) == t_INT)
for (j=1; j<=m; j++)
{
GEN z = RgX_rem(gmul(w, gel(zknf,j)), polabs);
if (c0) z = RgX_Rg_mul(z, c0);
gel(M,k++) = z;
}
else
for (j=1; j<=m; j++)
{
GEN c, z = Q_primitive_part(RgV_RgC_mul(zknf,gel(id,j)), &c);
z = RgX_rem(gmul(w, z), polabs);
c = mul_content(c, c0); if (c) z = RgX_Rg_mul(z, c);
gel(M,k++) = z;
}
}
setlg(M, k); return M;
}
GEN
rnf_zkabs(GEN rnf)
{
GEN d, M = modulereltoabs(rnf, rnf_get_zk(rnf));
GEN T = rnf_get_polabs(rnf);
long n = degpol(T);
M = Q_remove_denom(M, &d);
if (d)
{
M = RgXV_to_RgM(M,n);
M = ZM_hnfmodall(M, d, hnf_MODID|hnf_CENTER);
M = RgM_Rg_div(M, d);
}
else
M = matid(n);
return mkvec2(T, RgM_to_RgXV(M, varn(T)));
}
static GEN
mknfabs(GEN rnf, long prec)
{
GEN NF;
if ((NF = obj_check(rnf,rnf_NFABS)))
{ if (nf_get_prec(NF) < prec) NF = nfnewprec_shallow(NF,prec); }
else
NF = nfinit(rnf_zkabs(rnf), prec);
return NF;
}
static GEN
mkupdown(GEN rnf)
{
GEN NF = obj_check(rnf, rnf_NFABS), M, zknf, dzknf;
long i, l;
zknf = rnf_get_nfzk(rnf);
dzknf = gel(zknf,1); if (gequal1(dzknf)) dzknf = NULL;
l = lg(zknf); M = cgetg(l, t_MAT);
gel(M,1) = vec_ei(nf_get_degree(NF), 1);
for (i = 2; i < l; i++)
{
GEN c = poltobasis(NF, gel(zknf,i));
if (dzknf) c = gdiv(c, dzknf);
gel(M,i) = c;
}
return Qevproj_init(M);
}
GEN
rnf_build_nfabs(GEN rnf, long prec)
{
GEN NF = obj_checkbuild_prec(rnf, rnf_NFABS, &mknfabs, &nf_get_prec, prec);
(void)obj_checkbuild(rnf, rnf_MAPS, &mkupdown);
return NF;
}
void
rnfcomplete(GEN rnf)
{ (void)rnf_build_nfabs(rnf, nf_get_prec(rnf_get_nf(rnf))); }
GEN
nf_nfzk(GEN nf, GEN rnfeq)
{
GEN pol = gel(rnfeq,1), a = gel(rnfeq,2);
return Q_primpart(QXV_QXQ_eval(nf_get_zkprimpart(nf), a, pol));
}
GEN
check_polrel(GEN nf, GEN P, ulong *lim)
{
if (typ(P) != t_VEC || lg(P) != 3) *lim = 0;
else { *lim = gtou(gel(P,2)); P = gel(P,1); }
if (typ(P) != t_POL) pari_err_TYPE("rnfinit",P);
P = RgX_nffix("rnfinit", nf_get_pol(nf), P, 0);
if (!gequal1(leading_coeff(P)))
pari_err_IMPL("non-monic relative polynomials");
return P;
}
GEN
rnfinit0(GEN nf, GEN T, long flag)
{
pari_sp av = avma;
GEN bas, D, f, B, T0, rnfeq, rnf = obj_init(11, 2);
ulong lim;
nf = checknf(nf);
T0 = check_polrel(nf, T, &lim);
T = lift_shallow(T0);
gel(rnf,11) = rnfeq = nf_rnfeq(nf,T);
gel(rnf,2) = nf_nfzk(nf, rnfeq);
bas = rnfallbase(nf, T0, lim, rnf, &D, &f);
B = matbasistoalg(nf,gel(bas,1));
gel(bas,1) = lift_if_rational( RgM_to_RgXV(B,varn(T)) );
gel(rnf,1) = T;
gel(rnf,3) = D;
gel(rnf,4) = f;
gel(rnf,5) = cgetg(1, t_VEC);
gel(rnf,6) = cgetg(1, t_VEC);
gel(rnf,7) = bas;
gel(rnf,8) = lift_if_rational( RgM_inv_upper(B) );
gel(rnf,9) = typ(f) == t_INT? powiu(f, nf_get_degree(nf))
: RgM_det_triangular(f);
gel(rnf,10)= nf;
rnf = gerepilecopy(av, rnf);
if (flag) rnfcomplete(rnf);
return rnf;
}
GEN
rnfinit(GEN nf, GEN T) { return rnfinit0(nf,T,0); }
GEN
rnfeltup0(GEN rnf, GEN x, long flag)
{
pari_sp av = avma;
GEN zknf, nf, NF, POL;
long tx = typ(x);
checkrnf(rnf);
if (flag) rnfcomplete(rnf);
NF = obj_check(rnf,rnf_NFABS);
POL = rnf_get_polabs(rnf);
if (tx == t_POLMOD && RgX_equal_var(gel(x,1), POL))
{
if (flag) x = nf_to_scalar_or_basis(NF,x);
return gerepilecopy(av, x);
}
if (NF && tx == t_COL && lg(x)-1 == nf_get_degree(NF))
{
x = flag? nf_to_scalar_or_basis(NF,x)
: mkpolmod(nf_to_scalar_or_alg(NF,x), POL);
return gerepilecopy(av, x);
}
nf = rnf_get_nf(rnf);
if (NF)
{
GEN d, proj;
x = nf_to_scalar_or_basis(nf, x);
if (typ(x) != t_COL) return gerepilecopy(av, x);
proj = obj_check(rnf,rnf_MAPS);
x = Q_remove_denom(x,&d);
x = ZM_ZC_mul(gel(proj,1), x);
if (d) x = gdiv(x,d);
if (!flag) x = basistoalg(NF,x);
}
else
{
zknf = rnf_get_nfzk(rnf);
x = nfeltup(nf, x, zknf);
if (typ(x) == t_POL) x = mkpolmod(x, POL);
}
return gerepilecopy(av, x);
}
GEN
rnfeltup(GEN rnf, GEN x) { return rnfeltup0(rnf,x,0); }
GEN
nfeltup(GEN nf, GEN x, GEN zknf)
{
GEN c, dzknf = gel(zknf,1);
x = nf_to_scalar_or_basis(nf, x);
if (typ(x) != t_COL) return x;
x = Q_primitive_part(x, &c);
if (!RgV_is_ZV(x)) pari_err_TYPE("rnfeltup", x);
if (gequal1(dzknf)) dzknf = NULL;
c = mul_content(c, inv_content(dzknf));
x = RgV_RgC_mul(zknf, x); if (c) x = RgX_Rg_mul(x, c);
return x;
}
static void
fail(const char *f, GEN x)
{ pari_err_DOMAIN(f,"element","not in", strtoGENstr("the base field"),x); }
static GEN
eltdown(GEN rnf, GEN x, long flag)
{
GEN z,y, d, proj = obj_check(rnf,rnf_MAPS);
GEN M= gel(proj,1), iM=gel(proj,2), diM=gel(proj,3), perm=gel(proj,4);
x = Q_remove_denom(x,&d);
if (!RgV_is_ZV(x)) pari_err_TYPE("rnfeltdown", x);
y = ZM_ZC_mul(iM, vecpermute(x, perm));
z = ZM_ZC_mul(M,y);
if (!isint1(diM)) z = ZC_Z_mul(z,diM);
if (!ZV_equal(z,x)) fail("rnfeltdown",x);
d = mul_denom(d, diM);
if (d) y = gdiv(y,d);
if (!flag) y = basistoalg(rnf_get_nf(rnf), y);
return y;
}
GEN
rnfeltdown0(GEN rnf, GEN x, long flag)
{
const char *f = "rnfeltdown";
pari_sp av = avma;
GEN z, T, NF, nf;
long v;
checkrnf(rnf);
NF = obj_check(rnf,rnf_NFABS);
nf = rnf_get_nf(rnf);
T = nf_get_pol(nf);
v = varn(T);
switch(typ(x))
{
case t_INT: return icopy(x);
case t_FRAC:return gcopy(x);
case t_POLMOD:
if (RgX_equal_var(gel(x,1), rnf_get_polabs(rnf)))
{
if (degpol(T) == 1)
{
x = simplify_shallow(liftpol_shallow(gel(x,2)));
if (typ(x) != t_POL) return gerepilecopy(av,x);
}
break;
}
x = polmod_nffix(f,rnf,x,0);
if (typ(x) == t_POL) fail(f,x);
if (flag) x = nf_to_scalar_or_basis(nf,x);
return gerepilecopy(av, x);
case t_POL:
if (varn(x) != v) break;
x = Rg_nffix(f,T,x,0);
if (flag) x = nf_to_scalar_or_basis(nf,x);
return gerepilecopy(av, x);
case t_COL:
{
long n = lg(x)-1;
if (n == degpol(T) && RgV_is_QV(x))
{
if (RgV_isscalar(x)) return gcopy(gel(x,1));
if (!flag) return gcopy(x);
return basistoalg(nf,x);
}
if (NF) break;
}
default: pari_err_TYPE(f, x);
}
if (NF)
{
x = nf_to_scalar_or_basis(NF, x);
if (typ(x) == t_COL) x = eltdown(rnf,x,flag);
return gerepilecopy(av, x);
}
z = rnfeltabstorel(rnf,x);
switch(typ(z))
{
case t_INT:
case t_FRAC: return z;
}
z = gel(z,2);
if (typ(z) == t_POL)
{
if (lg(z) != 3) fail(f,x);
z = gel(z,2);
}
return gerepilecopy(av, z);
}
GEN
rnfeltdown(GEN rnf, GEN x) { return rnfeltdown0(rnf,x,0); }
static GEN
rnfV_to_nfM(GEN rnf, GEN x)
{
long i, l = lg(x);
GEN y = cgetg(l, t_MAT);
for (i = 1; i < l; i++) gel(y,i) = rnfalgtobasis(rnf,gel(x,i));
return y;
}
static GEN
rnfprincipaltohnf(GEN rnf,GEN x)
{
pari_sp av = avma;
GEN bas = rnf_get_zk(rnf), nf = rnf_get_nf(rnf);
x = rnfbasistoalg(rnf,x);
x = gmul(x, gmodulo(gel(bas,1), rnf_get_pol(rnf)));
return gerepileupto(av, nfhnf(nf, mkvec2(rnfV_to_nfM(rnf,x), gel(bas,2))));
}
static GEN
rnfideal0(void) { retmkvec2(cgetg(1,t_MAT),cgetg(1,t_VEC)); }
GEN
rnfidealhnf(GEN rnf, GEN x)
{
GEN z, nf, bas;
checkrnf(rnf); nf = rnf_get_nf(rnf);
switch(typ(x))
{
case t_INT: case t_FRAC:
if (isintzero(x)) return rnfideal0();
bas = rnf_get_zk(rnf); z = cgetg(3,t_VEC);
gel(z,1) = matid(rnf_get_degree(rnf));
gel(z,2) = gmul(x, gel(bas,2)); return z;
case t_VEC:
if (lg(x) == 3 && typ(gel(x,1)) == t_MAT) return nfhnf(nf, x);
case t_MAT:
return rnfidealabstorel(rnf, x);
case t_POLMOD: case t_POL: case t_COL:
return rnfprincipaltohnf(rnf,x);
}
pari_err_TYPE("rnfidealhnf",x);
return NULL;
}
static GEN
prodidnorm(GEN nf, GEN I)
{
long i, l = lg(I);
GEN z;
if (l == 1) return gen_1;
z = idealnorm(nf, gel(I,1));
for (i=2; i<l; i++) z = gmul(z, idealnorm(nf, gel(I,i)));
return z;
}
GEN
rnfidealnormrel(GEN rnf, GEN id)
{
pari_sp av = avma;
GEN nf, z = gel(rnfidealhnf(rnf,id), 2);
if (lg(z) == 1) return cgetg(1, t_MAT);
nf = rnf_get_nf(rnf); z = idealprod(nf, z);
return gerepileupto(av, idealmul(nf,z, rnf_get_index(rnf)));
}
GEN
rnfidealnormabs(GEN rnf, GEN id)
{
pari_sp av = avma;
GEN nf, z = gel(rnfidealhnf(rnf,id), 2);
if (lg(z) == 1) return gen_0;
nf = rnf_get_nf(rnf); z = prodidnorm(nf, z);
return gerepileupto(av, gmul(z, gel(rnf,9)));
}
static GEN
rnfidealreltoabs_i(GEN rnf, GEN x)
{
long i, l;
GEN w;
x = rnfidealhnf(rnf,x);
w = gel(x,1); l = lg(w); settyp(w, t_VEC);
for (i=1; i<l; i++) gel(w,i) = lift_shallow( rnfbasistoalg(rnf, gel(w,i)) );
return modulereltoabs(rnf, x);
}
GEN
rnfidealreltoabs(GEN rnf, GEN x)
{
pari_sp av = avma;
return gerepilecopy(av, rnfidealreltoabs_i(rnf,x));
}
GEN
rnfidealreltoabs0(GEN rnf, GEN x, long flag)
{
pari_sp av = avma;
long i, l;
GEN NF;
x = rnfidealreltoabs_i(rnf, x);
if (!flag) return gerepilecopy(av,x);
rnfcomplete(rnf);
NF = obj_check(rnf,rnf_NFABS);
l = lg(x); settyp(x, t_MAT);
for (i=1; i<l; i++) gel(x,i) = algtobasis(NF, gel(x,i));
return gerepileupto(av, idealhnf(NF,x));
}
GEN
rnfidealabstorel(GEN rnf, GEN x)
{
long n, N, j, tx = typ(x);
pari_sp av = avma;
GEN A, I, invbas;
checkrnf(rnf);
invbas = rnf_get_invzk(rnf);
if (tx != t_VEC && tx != t_MAT) pari_err_TYPE("rnfidealabstorel",x);
N = lg(x)-1;
if (N != rnf_get_absdegree(rnf))
{
if (!N) return rnfideal0();
pari_err_DIM("rnfidealabstorel");
}
n = rnf_get_degree(rnf);
A = cgetg(N+1,t_MAT);
I = cgetg(N+1,t_VEC);
for (j=1; j<=N; j++)
{
GEN t = lift_shallow( rnfeltabstorel(rnf, gel(x,j)) );
if (typ(t) == t_POL)
t = RgM_RgX_mul(invbas, t);
else
t = scalarcol_shallow(t, n);
gel(A,j) = t;
gel(I,j) = gen_1;
}
return gerepileupto(av, nfhnf(rnf_get_nf(rnf), mkvec2(A,I)));
}
GEN
rnfidealdown(GEN rnf,GEN x)
{
pari_sp av = avma;
GEN I;
if (typ(x) == t_MAT)
{
GEN d;
x = Q_remove_denom(x,&d);
if (RgM_is_ZM(x))
{
GEN NF = obj_check(rnf,rnf_NFABS);
if (NF)
{
GEN z, proj = obj_check(rnf,rnf_MAPS), ZK = gel(proj,1);
long i, lz, l;
x = idealhnf(NF,x);
if (lg(x) == 1) { avma = av; return cgetg(1,t_MAT); }
z = ZM_lll(shallowconcat(ZK,x), 0.99, LLL_KER);
lz = lg(z); l = lg(ZK);
for (i = 1; i < lz; i++) setlg(gel(z,i), l);
z = ZM_hnfmodid(z, gcoeff(x,1,1));
if (d) z = gdiv(z,d);
return gerepileupto(av, z);
}
}
}
x = rnfidealhnf(rnf,x); I = gel(x,2);
if (lg(I) == 1) { avma = av; return cgetg(1,t_MAT); }
return gerepilecopy(av, gel(I,1));
}
GEN
rnfidealup(GEN rnf,GEN x)
{
pari_sp av = avma;
long i, n;
GEN nf, bas, bas2, I, x2;
checkrnf(rnf); nf = rnf_get_nf(rnf);
n = rnf_get_degree(rnf);
bas = rnf_get_zk(rnf); bas2 = gel(bas,2);
(void)idealtyp(&x, &I);
x2 = idealtwoelt(nf,x);
I = cgetg(n+1,t_VEC);
for (i=1; i<=n; i++)
{
GEN c = gel(bas2,i), d;
if (typ(c) == t_MAT)
{
c = Q_remove_denom(c,&d);
c = idealHNF_mul(nf,c,x2);
if (d) c = gdiv(c,d);
}
else
c = idealmul(nf,c,x);
gel(I,i) = c;
}
return gerepilecopy(av, modulereltoabs(rnf, mkvec2(gel(bas,1), I)));
}
GEN
rnfidealup0(GEN rnf,GEN x, long flag)
{
pari_sp av = avma;
GEN NF, nf, proj, d, x2;
if (!flag) return rnfidealup(rnf,x);
checkrnf(rnf); nf = rnf_get_nf(rnf);
rnfcomplete(rnf);
proj = obj_check(rnf,rnf_MAPS);
NF = obj_check(rnf,rnf_NFABS);
(void)idealtyp(&x, &d);
x2 = idealtwoelt(nf,x);
x2 = Q_remove_denom(x2,&d);
if (typ(gel(x2,2)) == t_COL) gel(x2,2) = ZM_ZC_mul(gel(proj,1),gel(x2,2));
x2 = idealhnf_two(NF, x2);
if (d) x2 = gdiv(x2,d);
return gerepileupto(av, x2);
}
GEN
rnfidealtwoelement(GEN rnf, GEN x)
{
pari_sp av = avma;
GEN y, cy, z, NF;
y = rnfidealreltoabs_i(rnf,x);
rnfcomplete(rnf);
NF = obj_check(rnf,rnf_NFABS);
y = matalgtobasis(NF, y); settyp(y, t_MAT);
y = Q_primitive_part(y, &cy);
y = ZM_hnf(y);
if (lg(y) == 1) { avma = av; return mkvec2(gen_0, gen_0); }
y = idealtwoelt(NF, y);
if (cy) y = RgV_Rg_mul(y, cy);
z = gel(y,2);
if (typ(z) == t_COL) z = rnfeltabstorel(rnf, nf_to_scalar_or_alg(NF, z));
return gerepilecopy(av, mkvec2(gel(y,1), z));
}
GEN
rnfidealmul(GEN rnf,GEN x,GEN y)
{
pari_sp av = avma;
GEN nf, z, x1, x2, p1, p2, bas;
y = rnfidealtwoelement(rnf,y);
if (isintzero(gel(y,1))) { avma = av; return rnfideal0(); }
nf = rnf_get_nf(rnf);
bas = rnf_get_zk(rnf);
x = rnfidealhnf(rnf,x);
x1 = gmodulo(gmul(gel(bas,1), matbasistoalg(nf,gel(x,1))), rnf_get_pol(rnf));
x2 = gel(x,2);
p1 = gmul(gel(y,1), gel(x,1));
p2 = rnfV_to_nfM(rnf, gmul(gel(y,2), x1));
z = mkvec2(shallowconcat(p1, p2), shallowconcat(x2, x2));
return gerepileupto(av, nfhnf(nf,z));
}
static GEN
rnfidealprimedec_1(GEN rnf, GEN SL, GEN prK)
{
GEN v, piL = rnfeltup0(rnf, pr_get_gen(prK), 1);
long i, c, l;
if (typ(piL) != t_COL) return SL;
v = cgetg_copy(SL, &l);
for (i = c = 1; i < l; i++)
{
GEN P = gel(SL,i);
if (ZC_prdvd(piL, P)) gel(v,c++) = P;
}
setlg(v, c); return v;
}
GEN
rnfidealprimedec(GEN rnf, GEN pr)
{
pari_sp av = avma;
GEN p, z, NF, nf, SL;
checkrnf(rnf);
rnfcomplete(rnf);
NF = obj_check(rnf,rnf_NFABS);
nf = rnf_get_nf(rnf);
if (typ(pr) == t_INT)
{
p = pr;
pr = NULL;
}
else
{
checkprid(pr);
p = pr_get_p(pr);
}
SL = idealprimedec(NF, p);
if (pr) z = rnfidealprimedec_1(rnf, SL, pr);
else
{
GEN vK = idealprimedec(nf, p), vL;
long l = lg(vK), i;
vL = cgetg(l, t_VEC);
for (i = 1; i < l; i++) gel(vL,i) = rnfidealprimedec_1(rnf, SL, gel(vK,i));
z = mkvec2(vK, vL);
}
return gerepilecopy(av, z);
}
GEN
rnfidealfactor(GEN rnf, GEN x)
{
pari_sp av = avma;
GEN NF;
checkrnf(rnf);
rnfcomplete(rnf);
NF = obj_check(rnf,rnf_NFABS);
return gerepileupto(av, idealfactor(NF, rnfidealreltoabs0(rnf, x, 1)));
}
GEN
rnfequationall(GEN A, GEN B, long *pk, GEN *pLPRS)
{
long lA, lB;
GEN nf, C;
A = get_nfpol(A, &nf); lA = lg(A);
if (!nf) {
if (lA<=3) pari_err_CONSTPOL("rnfequation");
RgX_check_ZX(A,"rnfequation");
}
B = RgX_nffix("rnfequation", A,B,1); lB = lg(B);
if (lB<=3) pari_err_CONSTPOL("rnfequation");
B = Q_primpart(B);
if (!nfissquarefree(A,B))
pari_err_DOMAIN("rnfequation","issquarefree(B)","=",gen_0,B);
*pk = 0; C = ZX_ZXY_resultant_all(A, B, pk, pLPRS);
if (signe(leading_coeff(C)) < 0) C = ZX_neg(C);
*pk = -*pk; return Q_primpart(C);
}
GEN
rnfequation0(GEN A, GEN B, long flall)
{
pari_sp av = avma;
GEN LPRS, C;
long k;
C = rnfequationall(A, B, &k, flall? &LPRS: NULL);
if (flall)
{
GEN a, mH0 = RgX_neg(gel(LPRS,1)), H1 = gel(LPRS,2);
a = RgXQ_mul(mH0, QXQ_inv(H1, C), C);
C = mkvec3(C, mkpolmod(a, C), stoi(k));
}
return gerepilecopy(av, C);
}
GEN
rnfequation(GEN nf, GEN pol) { return rnfequation0(nf,pol,0); }
GEN
rnfequation2(GEN nf, GEN pol) { return rnfequation0(nf,pol,1); }
GEN
nf_rnfeq(GEN nf, GEN relpol)
{
GEN pol, a, k, junk, eq;
relpol = liftpol_shallow(relpol);
eq = rnfequation2(nf, relpol);
pol = gel(eq,1);
a = gel(eq,2); if (typ(a) == t_POLMOD) a = gel(a,2);
k = gel(eq,3);
return mkvec5(pol,a,k,get_nfpol(nf, &junk),relpol);
}
GEN
nf_rnfeqsimple(GEN nf, GEN relpol)
{
long sa;
GEN junk, pol;
relpol = liftpol_shallow(relpol);
pol = rnfequationall(nf, relpol, &sa, NULL);
return mkvec5(pol,gen_0,stoi(sa),get_nfpol(nf, &junk),relpol);
}
static GEN
nftau(long r1, GEN x)
{
long i, l = lg(x);
GEN s = r1? gel(x,1): gmul2n(real_i(gel(x,1)),1);
for (i=2; i<=r1; i++) s = gadd(s, gel(x,i));
for ( ; i < l; i++) s = gadd(s, gmul2n(real_i(gel(x,i)),1));
return s;
}
static GEN
initmat(long l)
{
GEN x = cgetg(l, t_MAT);
long i;
for (i = 1; i < l; i++) gel(x,i) = cgetg(l, t_COL);
return x;
}
static GEN
nftocomplex(GEN nf, GEN x)
{
GEN M = nf_get_M(nf);
x = nf_to_scalar_or_basis(nf,x);
if (typ(x) != t_COL) return const_col(nbrows(M), x);
return RgM_RgC_mul(M, x);
}
static GEN
mattocomplex(GEN nf, GEN x)
{
long i,j, l = lg(x);
GEN v = cgetg(l, t_VEC);
for (j=1; j<l; j++)
{
GEN c = gel(x,j), b = cgetg(l, t_MAT);
for (i=1; i<l; i++) gel(b,i) = nftocomplex(nf, gel(c,i));
b = shallowtrans(b); settyp(b, t_COL);
gel(v,j) = b;
}
return v;
}
static GEN
nf_all_roots(GEN nf, GEN x, long prec)
{
long i, j, l = lg(x), ru = lg(nf_get_roots(nf));
GEN y = cgetg(l, t_POL), v, z;
x = RgX_to_nfX(nf, x);
y[1] = x[1];
for (i=2; i<l; i++) gel(y,i) = nftocomplex(nf, gel(x,i));
i = gprecision(y); if (i && i <= 3) return NULL;
v = cgetg(ru, t_VEC);
z = cgetg(l, t_POL); z[1] = x[1];
for (i=1; i<ru; i++)
{
for (j = 2; j < l; j++) gel(z,j) = gmael(y,j,i);
gel(v,i) = cleanroots(z, prec);
}
return v;
}
static GEN
rnfscal(GEN m, GEN x, GEN y)
{
long i, l = lg(m);
GEN z = cgetg(l, t_COL);
for (i = 1; i < l; i++)
gel(z,i) = gmul(conj_i(shallowtrans(gel(x,i))), gmul(gel(m,i), gel(y,i)));
return z;
}
static GEN
findmin(GEN nf, GEN x, GEN muf)
{
pari_sp av = avma;
long e;
GEN cx, y, m, M = nf_get_M(nf);
x = Q_primitive_part(x, &cx);
if (gequal1(gcoeff(x,1,1))) y = M;
else
{
GEN G = nf_get_G(nf);
m = lllfp(RgM_mul(G,x), 0.75, 0);
if (typ(m) != t_MAT)
{
x = ZM_lll(x, 0.75, LLL_INPLACE);
m = lllfp(RgM_mul(G,x), 0.75, 0);
if (typ(m) != t_MAT) pari_err_PREC("rnflllgram");
}
x = ZM_mul(x, m);
y = RgM_mul(M, x);
}
m = RgM_solve_realimag(y, muf);
if (!m) return NULL;
if (cx) m = RgC_Rg_div(m, cx);
m = grndtoi(m, &e);
if (e >= 0) return NULL;
m = ZM_ZC_mul(x, m);
if (cx) m = ZC_Q_mul(m, cx);
return gerepileupto(av, m);
}
static int
RED(long k, long l, GEN U, GEN mu, GEN MC, GEN nf, GEN I, GEN *Ik_inv)
{
GEN x, xc, ideal;
long i;
if (!*Ik_inv) *Ik_inv = idealinv(nf, gel(I,k));
ideal = idealmul(nf,gel(I,l), *Ik_inv);
x = findmin(nf, ideal, gcoeff(mu,k,l));
if (!x) return 0;
if (gequal0(x)) return 1;
xc = nftocomplex(nf,x);
gel(MC,k) = gsub(gel(MC,k), vecmul(xc,gel(MC,l)));
gel(U,k) = gsub(gel(U,k), gmul(coltoalg(nf,x), gel(U,l)));
gcoeff(mu,k,l) = gsub(gcoeff(mu,k,l), xc);
for (i=1; i<l; i++)
gcoeff(mu,k,i) = gsub(gcoeff(mu,k,i), vecmul(xc,gcoeff(mu,l,i)));
return 1;
}
static int
check_0(GEN B)
{
long i, l = lg(B);
for (i = 1; i < l; i++)
if (gsigne(gel(B,i)) <= 0) return 1;
return 0;
}
static int
do_SWAP(GEN I, GEN MC, GEN MCS, GEN h, GEN mu, GEN B, long kmax, long k,
const long alpha, long r1)
{
GEN p1, p2, muf, mufc, Bf, temp;
long i, j;
p1 = nftau(r1, gadd(gel(B,k),
gmul(gnorml2(gcoeff(mu,k,k-1)), gel(B,k-1))));
p2 = nftau(r1, gel(B,k-1));
if (gcmp(gmulsg(alpha,p1), gmulsg(alpha-1,p2)) > 0) return 0;
swap(gel(MC,k-1),gel(MC,k));
swap(gel(h,k-1), gel(h,k));
swap(gel(I,k-1), gel(I,k));
for (j=1; j<=k-2; j++) swap(gcoeff(mu,k-1,j),gcoeff(mu,k,j));
muf = gcoeff(mu,k,k-1);
mufc = conj_i(muf);
Bf = gadd(gel(B,k), vecmul(real_i(vecmul(muf,mufc)), gel(B,k-1)));
if (check_0(Bf)) return 1;
p1 = vecdiv(gel(B,k-1),Bf);
gcoeff(mu,k,k-1) = vecmul(mufc,p1);
temp = gel(MCS,k-1);
gel(MCS,k-1) = gadd(gel(MCS,k), vecmul(muf,gel(MCS,k-1)));
gel(MCS,k) = gsub(vecmul(vecdiv(gel(B,k),Bf), temp),
vecmul(gcoeff(mu,k,k-1), gel(MCS,k)));
gel(B,k) = vecmul(gel(B,k),p1);
gel(B,k-1) = Bf;
for (i=k+1; i<=kmax; i++)
{
temp = gcoeff(mu,i,k);
gcoeff(mu,i,k) = gsub(gcoeff(mu,i,k-1), vecmul(muf, gcoeff(mu,i,k)));
gcoeff(mu,i,k-1) = gadd(temp, vecmul(gcoeff(mu,k,k-1),gcoeff(mu,i,k)));
}
return 1;
}
static GEN
rel_T2(GEN nf, GEN pol, long lx, long prec)
{
long ru, i, j, k, l;
GEN T2, s, unro, roorder, powreorder;
roorder = nf_all_roots(nf, pol, prec);
if (!roorder) return NULL;
ru = lg(roorder);
unro = cgetg(lx,t_COL); for (i=1; i<lx; i++) gel(unro,i) = gen_1;
powreorder = cgetg(lx,t_MAT); gel(powreorder,1) = unro;
T2 = cgetg(ru, t_VEC);
for (i = 1; i < ru; i++)
{
GEN ro = gel(roorder,i);
GEN m = initmat(lx);
for (k=2; k<lx; k++)
{
GEN c = cgetg(lx, t_COL); gel(powreorder,k) = c;
for (j=1; j < lx; j++)
gel(c,j) = gmul(gel(ro,j), gmael(powreorder,k-1,j));
}
for (l = 1; l < lx; l++)
for (k = 1; k <= l; k++)
{
s = gen_0;
for (j = 1; j < lx; j++)
s = gadd(s, gmul(conj_i(gmael(powreorder,k,j)),
gmael(powreorder,l,j)));
if (l == k)
gcoeff(m, l, l) = real_i(s);
else
{
gcoeff(m, k, l) = s;
gcoeff(m, l, k) = conj_i(s);
}
}
gel(T2,i) = m;
}
return T2;
}
GEN
rnflllgram(GEN nf, GEN pol, GEN order,long prec)
{
pari_sp av = avma;
long j, k, l, kmax, r1, lx, count = 0;
GEN M, I, h, H, mth, MC, MPOL, MCS, B, mu;
const long alpha = 10, MAX_COUNT = 4;
nf = checknf(nf); r1 = nf_get_r1(nf);
check_ZKmodule(order, "rnflllgram");
M = gel(order,1);
I = gel(order,2); lx = lg(I);
if (lx < 3) return gcopy(order);
if (lx-1 != degpol(pol)) pari_err_DIM("rnflllgram");
I = leafcopy(I);
H = NULL;
MPOL = matbasistoalg(nf, M);
MCS = matid(lx-1);
PRECNF:
if (count == MAX_COUNT)
{
prec = precdbl(prec); count = 0;
if (DEBUGLEVEL) pari_warn(warnprec,"rnflllgram",prec);
nf = nfnewprec_shallow(nf,prec);
}
mth = rel_T2(nf, pol, lx, prec);
if (!mth) { count = MAX_COUNT; goto PRECNF; }
h = NULL;
PRECPB:
if (h)
{
if (++count == MAX_COUNT || RgM_isidentity(h)) {count = MAX_COUNT; goto PRECNF;}
H = H? gmul(H, h): h;
MPOL = gmul(MPOL, h);
}
h = matid(lx-1);
MC = mattocomplex(nf, MPOL);
mu = cgetg(lx,t_MAT);
B = cgetg(lx,t_COL);
for (j=1; j<lx; j++)
{
gel(mu,j) = zerocol(lx - 1);
gel(B,j) = gen_0;
}
if (DEBUGLEVEL) err_printf("k = ");
gel(B,1) = real_i(rnfscal(mth,gel(MC,1),gel(MC,1)));
gel(MCS,1) = gel(MC,1);
kmax = 1; k = 2;
do
{
GEN Ik_inv = NULL;
if (DEBUGLEVEL) err_printf("%ld ",k);
if (k > kmax)
{
kmax = k; gel(MCS,k) = gel(MC,k);
for (j=1; j<k; j++)
{
gcoeff(mu,k,j) = vecdiv(rnfscal(mth,gel(MCS,j),gel(MC,k)),
gel(B,j));
gel(MCS,k) = gsub(gel(MCS,k), vecmul(gcoeff(mu,k,j),gel(MCS,j)));
}
gel(B,k) = real_i(rnfscal(mth,gel(MCS,k),gel(MCS,k)));
if (check_0(gel(B,k))) goto PRECPB;
}
if (!RED(k, k-1, h, mu, MC, nf, I, &Ik_inv)) goto PRECPB;
if (do_SWAP(I,MC,MCS,h,mu,B,kmax,k,alpha, r1))
{
if (!B[k]) goto PRECPB;
if (k > 2) k--;
}
else
{
for (l=k-2; l; l--)
if (!RED(k, l, h, mu, MC, nf, I, &Ik_inv)) goto PRECPB;
k++;
}
if (gc_needed(av,2))
{
if(DEBUGMEM>1) pari_warn(warnmem,"rnflllgram");
gerepileall(av, H?10:9, &nf,&mth,&h,&MPOL,&B,&MC,&MCS,&mu,&I,&H);
}
}
while (k < lx);
MPOL = gmul(MPOL,h);
if (H) h = gmul(H, h);
if (DEBUGLEVEL) err_printf("\n");
MPOL = RgM_to_nfM(nf,MPOL);
h = RgM_to_nfM(nf,h);
return gerepilecopy(av, mkvec2(mkvec2(MPOL,I), h));
}
GEN
rnfpolred(GEN nf, GEN pol, long prec)
{
pari_sp av = avma;
long i, j, n, v = varn(pol);
GEN id, w, I, O, bnf, nfpol;
if (typ(pol)!=t_POL) pari_err_TYPE("rnfpolred",pol);
bnf = nf; nf = checknf(bnf);
bnf = (nf == bnf)? NULL: checkbnf(bnf);
if (degpol(pol) <= 1) { w = cgetg(2, t_VEC); gel(w,1) = pol_x(v); return w; }
nfpol = nf_get_pol(nf);
id = rnfpseudobasis(nf,pol);
if (bnf && is_pm1( bnf_get_no(bnf) ))
{
GEN newI, newO;
O = gel(id,1);
I = gel(id,2); n = lg(I)-1;
newI = cgetg(n+1,t_VEC);
newO = cgetg(n+1,t_MAT);
for (j=1; j<=n; j++)
{
GEN al = gen_if_principal(bnf,gel(I,j));
gel(newI,j) = gen_1;
gel(newO,j) = nfC_nf_mul(nf, gel(O,j), al);
}
id = mkvec2(newO, newI);
}
id = gel(rnflllgram(nf,pol,id,prec),1);
O = gel(id,1);
I = gel(id,2); n = lg(I)-1;
w = cgetg(n+1,t_VEC);
pol = lift_shallow(pol);
for (j=1; j<=n; j++)
{
GEN newpol, L, a, Ij = gel(I,j);
a = RgC_Rg_mul(gel(O,j), (typ(Ij) == t_MAT)? gcoeff(Ij,1,1): Ij);
for (i=n; i; i--) gel(a,i) = nf_to_scalar_or_alg(nf, gel(a,i));
a = RgV_to_RgX(a, v);
newpol = RgXQX_red(RgXQ_charpoly(a, pol, v), nfpol);
newpol = Q_primpart(newpol);
(void)nfgcd_all(newpol, RgX_deriv(newpol), nfpol, nf_get_index(nf), &newpol);
L = leading_coeff(newpol);
gel(w,j) = (typ(L) == t_POL)? RgXQX_div(newpol, L, nfpol)
: RgX_Rg_div(newpol, L);
}
return gerepilecopy(av,w);
}
GEN
nfreduce(GEN nf, GEN x, GEN I)
{
pari_sp av = avma;
GEN aI;
x = nf_to_scalar_or_basis(checknf(nf), x);
if (idealtyp(&I,&aI) != id_MAT || lg(I)==1) pari_err_TYPE("nfreduce",I);
if (typ(x) != t_COL) x = scalarcol( gmod(x, gcoeff(I,1,1)), lg(I)-1 );
else x = reducemodinvertible(x, I);
return gerepileupto(av, x);
}
static GEN
element_close(GEN nf, GEN x, GEN ideal)
{
pari_sp av = avma;
GEN y = gcoeff(ideal,1,1);
x = nf_to_scalar_or_basis(nf, x);
if (typ(y) == t_INT && is_pm1(y)) return ground(x);
if (typ(x) == t_COL)
x = closemodinvertible(x, ideal);
else
x = gmul(y, gdivround(x,y));
return gerepileupto(av, x);
}
static GEN
colcomb1(GEN nf, GEN v, GEN A, GEN B)
{
if (isintzero(v)) return A;
return RgC_to_nfC(nf, RgC_add(A, nfC_nf_mul(nf,B,v)));
}
static GEN
colcomb(GEN nf, GEN u, GEN v, GEN A, GEN B)
{
if (isintzero(u)) return nfC_nf_mul(nf,B,v);
if (u != gen_1) A = nfC_nf_mul(nf,A,u);
return colcomb1(nf, v, A, B);
}
static GEN
element_mulvecrow(GEN nf, GEN x, GEN m, long i, long lim)
{
long j, l = minss(lg(m), lim+1);
GEN dx, y = cgetg(l, t_VEC);
x = nf_to_scalar_or_basis(nf, x);
if (typ(x) == t_COL)
{
x = zk_multable(nf, Q_remove_denom(x, &dx));
for (j=1; j<l; j++)
{
GEN t = gcoeff(m,i,j);
if (!isintzero(t))
{
if (typ(t) == t_COL)
t = RgM_RgC_mul(x, t);
else
t = ZC_Q_mul(gel(x,1), t);
if (dx) t = gdiv(t, dx);
t = nf_to_scalar_or_basis(nf,t);
}
gel(y,j) = t;
}
}
else
{
for (j=1; j<l; j++) gel(y,j) = gmul(x, gcoeff(m,i,j));
}
return y;
}
static GEN
rowcomb(GEN nf, GEN u, GEN v, long s, long t, GEN Z, long lim)
{
GEN z;
if (gequal0(u))
z = element_mulvecrow(nf,v,Z,t, lim);
else
{
z = element_mulvecrow(nf,u,Z,s, lim);
if (!gequal0(v)) z = gadd(z, element_mulvecrow(nf,v,Z,t, lim));
}
return z;
}
static GEN
zero_nfbezout(GEN nf,GEN bB, GEN b, GEN A,GEN B,GEN *u,GEN *v,GEN *w,GEN *di)
{
GEN d;
if (isint1(b))
{
*v = gen_1;
*w = A;
d = B;
*di = idealinv(nf,d);
}
else
{
*v = nfinv(nf,b);
*w = idealmul(nf,A,*v);
d = bB? bB: idealmul(nf,b,B);
*di = idealHNF_inv(nf,d);
}
*u = gen_0; return d;
}
static GEN
nfbezout(GEN nf,GEN a,GEN b, GEN A,GEN B, GEN *pu,GEN *pv,GEN *pw,GEN *pdi,
int red)
{
GEN w, u, v, d, di, aA, bB;
if (isintzero(a)) return zero_nfbezout(nf,NULL,b,A,B,pu,pv,pw,pdi);
if (isintzero(b)) return zero_nfbezout(nf,NULL,a,B,A,pv,pu,pw,pdi);
if (a != gen_1)
{
a = nf_to_scalar_or_basis(nf,a);
if (isint1(a)) a = gen_1;
}
aA = (a == gen_1)? idealhnf_shallow(nf,A): idealmul(nf,a,A);
bB = idealmul(nf,b,B);
d = idealadd(nf,aA,bB);
if (gequal(aA, d)) return zero_nfbezout(nf,d, a,B,A,pv,pu,pw,pdi);
if (gequal(bB, d)) return zero_nfbezout(nf,d, b,A,B,pu,pv,pw,pdi);
di = idealHNF_inv(nf,d);
aA = idealmul(nf,aA,di);
bB = idealmul(nf,bB,di);
u = red? idealaddtoone_i(nf, aA, bB): idealaddtoone_raw(nf, aA, bB);
w = idealmul(nf,aA,B);
v = nfdiv(nf, nfsub(nf, gen_1, u), b);
if (a != gen_1)
{
GEN inva = nfinv(nf, a);
u = nfmul(nf,u,inva);
w = idealmul(nf, inva, w);
}
*pu = u; *pv = v; *pw = w; *pdi = di; return d;
}
static void
idV_simplify(GEN v)
{
long i, l = lg(v);
for (i = 1; i < l; i++)
{
GEN M = gel(v,i);
if (typ(M)==t_MAT && RgM_isscalar(M,NULL))
gel(v,i) = Q_abs_shallow(gcoeff(M,1,1));
}
}
GEN
nfhnf0(GEN nf, GEN x, long flag)
{
long i, j, def, idef, m, n;
pari_sp av0 = avma, av;
GEN y, A, I, J, U;
nf = checknf(nf);
check_ZKmodule(x, "nfhnf");
A = gel(x,1); RgM_dimensions(A, &m, &n);
I = gel(x,2);
if (!n) {
if (!flag) return gcopy(x);
retmkvec2(gcopy(x), cgetg(1,t_MAT));
}
U = flag? matid(n): NULL;
idef = (n < m)? m-n : 0;
av = avma;
A = RgM_to_nfM(nf,A);
I = leafcopy(I);
J = zerovec(n); def = n;
for (i=m; i>idef; i--)
{
GEN d, di = NULL;
j=def; while (j>=1 && isintzero(gcoeff(A,i,j))) j--;
if (!j)
{
if (idef) idef--;
continue;
}
if (j==def) j--;
else {
swap(gel(A,j), gel(A,def));
swap(gel(I,j), gel(I,def));
if (U) swap(gel(U,j), gel(U,def));
}
for ( ; j; j--)
{
GEN a,b, u,v,w, S, T, S0, T0 = gel(A,j);
b = gel(T0,i); if (isintzero(b)) continue;
S0 = gel(A,def); a = gel(S0,i);
d = nfbezout(nf, a,b, gel(I,def),gel(I,j), &u,&v,&w,&di,1);
S = colcomb(nf, u,v, S0,T0);
T = colcomb(nf, a,gneg(b), T0,S0);
gel(A,def) = S; gel(A,j) = T;
gel(I,def) = d; gel(I,j) = w;
if (U)
{
S0 = gel(U,def);
T0 = gel(U,j);
gel(U,def) = colcomb(nf, u,v, S0,T0);
gel(U,j) = colcomb(nf, a,gneg(b), T0,S0);
}
}
y = gcoeff(A,i,def);
if (!isint1(y))
{
GEN yi = nfinv(nf,y);
gel(A,def) = nfC_nf_mul(nf, gel(A,def), yi);
gel(I,def) = idealmul(nf, y, gel(I,def));
if (U) gel(U,def) = nfC_nf_mul(nf, gel(U,def), yi);
di = NULL;
}
if (!di) di = idealinv(nf,gel(I,def));
d = gel(I,def);
gel(J,def) = di;
for (j=def+1; j<=n; j++)
{
GEN mc, c = gcoeff(A,i,j); if (isintzero(c)) continue;
c = element_close(nf, c, idealmul(nf,d,gel(J,j)));
mc = gneg(c);
gel(A,j) = colcomb1(nf, mc, gel(A,j),gel(A,def));
if (U) gel(U,j) = colcomb1(nf, mc, gel(U,j),gel(U,def));
}
def--;
if (gc_needed(av,2))
{
if(DEBUGMEM>1) pari_warn(warnmem,"nfhnf, i = %ld", i);
gerepileall(av,U?4:3, &A,&I,&J,&U);
}
}
n -= def;
A += def; A[0] = evaltyp(t_MAT)|evallg(n+1);
I += def; I[0] = evaltyp(t_VEC)|evallg(n+1);
idV_simplify(I);
x = mkvec2(A,I);
if (U) x = mkvec2(x,U);
return gerepilecopy(av0, x);
}
GEN
nfhnf(GEN nf, GEN x) { return nfhnf0(nf, x, 0); }
static GEN
RgV_find_denom(GEN x)
{
long i, l = lg(x);
for (i = 1; i < l; i++)
if (Q_denom(gel(x,i)) != gen_1) return gel(x,i);
return NULL;
}
GEN
nfsnf0(GEN nf, GEN x, long flag)
{
long i, j, k, l, n, m;
pari_sp av;
GEN z,u,v,w,d,dinv,A,I,J, U,V;
nf = checknf(nf);
if (typ(x)!=t_VEC || lg(x)!=4) pari_err_TYPE("nfsnf",x);
A = gel(x,1);
I = gel(x,2);
J = gel(x,3);
if (typ(A)!=t_MAT) pari_err_TYPE("nfsnf",A);
n = lg(A)-1;
if (typ(I)!=t_VEC) pari_err_TYPE("nfsnf",I);
if (typ(J)!=t_VEC) pari_err_TYPE("nfsnf",J);
if (lg(I)!=n+1 || lg(J)!=n+1) pari_err_DIM("nfsnf");
RgM_dimensions(A, &m, &n);
if (!n || n != m) pari_err_IMPL("nfsnf for empty or non square matrices");
av = avma;
if (!flag) U = V = NULL;
else
{
U = matid(m);
V = matid(n);
}
A = RgM_to_nfM(nf, A);
I = leafcopy(I);
J = leafcopy(J);
for (i = 1; i <= n; i++) gel(J,i) = idealinv(nf, gel(J,i));
z = zerovec(n);
for (i=n; i>=1; i--)
{
GEN Aii, a, b, db;
long c = 0;
for (j=i-1; j>=1; j--)
{
GEN S, T, S0, T0 = gel(A,j);
b = gel(T0,i); if (gequal0(b)) continue;
S0 = gel(A,i); a = gel(S0,i);
d = nfbezout(nf, a,b, gel(J,i),gel(J,j), &u,&v,&w,&dinv,1);
S = colcomb(nf, u,v, S0,T0);
T = colcomb(nf, a,gneg(b), T0,S0);
gel(A,i) = S; gel(A,j) = T;
gel(J,i) = d; gel(J,j) = w;
if (V)
{
T0 = gel(V,j);
S0 = gel(V,i);
gel(V,i) = colcomb(nf, u,v, S0,T0);
gel(V,j) = colcomb(nf, a,gneg(b), T0,S0);
}
}
for (j=i-1; j>=1; j--)
{
GEN ri, rj;
b = gcoeff(A,j,i); if (gequal0(b)) continue;
a = gcoeff(A,i,i);
d = nfbezout(nf, a,b, gel(I,i),gel(I,j), &u,&v,&w,&dinv,1);
ri = rowcomb(nf, u,v, i,j, A, i);
rj = rowcomb(nf, a,gneg(b), j,i, A, i);
for (k=1; k<=i; k++) {
gcoeff(A,j,k) = gel(rj,k);
gcoeff(A,i,k) = gel(ri,k);
}
if (U)
{
ri = rowcomb(nf, u,v, i,j, U, m);
rj = rowcomb(nf, a,gneg(b), j,i, U, m);
for (k=1; k<=m; k++) {
gcoeff(U,j,k) = gel(rj,k);
gcoeff(U,i,k) = gel(ri,k);
}
}
gel(I,i) = d; gel(I,j) = w; c = 1;
}
if (c) { i++; continue; }
Aii = gcoeff(A,i,i); if (gequal0(Aii)) continue;
gel(J,i) = idealmul(nf, gel(J,i), Aii);
gcoeff(A,i,i) = gen_1;
if (V) gel(V,i) = nfC_nf_mul(nf, gel(V,i), nfinv(nf,Aii));
gel(z,i) = idealmul(nf,gel(J,i),gel(I,i));
b = Q_remove_denom(gel(z,i), &db);
for (k=1; k<i; k++)
for (l=1; l<i; l++)
{
GEN d, D, p1, p2, p3, Akl = gcoeff(A,k,l);
long t;
if (gequal0(Akl)) continue;
p1 = idealmul(nf,Akl,gel(J,l));
p3 = idealmul(nf, p1, gel(I,k));
if (db) p3 = RgM_Rg_mul(p3, db);
if (RgM_is_ZM(p3) && hnfdivide(b, p3)) continue;
D = idealdiv(nf,gel(I,k),gel(I,i));
p2 = idealdiv(nf,gel(J,i), p1);
d = RgV_find_denom( RgM_solve(p2, D) );
if (!d) pari_err_BUG("nfsnf");
p1 = element_mulvecrow(nf,d,A,k,i);
for (t=1; t<=i; t++) gcoeff(A,i,t) = gadd(gcoeff(A,i,t),gel(p1,t));
if (U)
{
p1 = element_mulvecrow(nf,d,U,k,i);
for (t=1; t<=i; t++) gcoeff(U,i,t) = gadd(gcoeff(U,i,t),gel(p1,t));
}
k = i; c = 1; break;
}
if (gc_needed(av,1))
{
if(DEBUGMEM>1) pari_warn(warnmem,"nfsnf");
gerepileall(av,U?6:4, &A,&I,&J,&z,&U,&V);
}
if (c) i++;
}
if (U) z = mkvec3(z,U,V);
return gerepilecopy(av, z);
}
GEN
nfsnf(GEN nf, GEN x) { return nfsnf0(nf,x,0); }
GEN
nfdetint(GEN nf, GEN x)
{
GEN pass,c,v,det1,piv,pivprec,vi,p1,A,I,id,idprod;
long i, j, k, rg, n, m, m1, cm=0, N;
pari_sp av = avma, av1;
nf = checknf(nf); N = nf_get_degree(nf);
check_ZKmodule(x, "nfdetint");
A = gel(x,1);
I = gel(x,2);
n = lg(A)-1; if (!n) return gen_1;
m1 = lgcols(A); m = m1-1;
id = matid(N);
c = new_chunk(m1); for (k=1; k<=m; k++) c[k] = 0;
piv = pivprec = gen_1;
av1 = avma;
det1 = idprod = gen_0;
pass = cgetg(m1,t_MAT);
v = cgetg(m1,t_COL);
for (j=1; j<=m; j++)
{
gel(pass,j) = zerocol(m);
gel(v,j) = gen_0;
}
for (rg=0,k=1; k<=n; k++)
{
long t = 0;
for (i=1; i<=m; i++)
if (!c[i])
{
vi=nfmul(nf,piv,gcoeff(A,i,k));
for (j=1; j<=m; j++)
if (c[j]) vi=gadd(vi,nfmul(nf,gcoeff(pass,i,j),gcoeff(A,j,k)));
gel(v,i) = vi; if (!t && !gequal0(vi)) t=i;
}
if (t)
{
pivprec = piv;
if (rg == m-1)
{
if (!cm)
{
cm=1; idprod = id;
for (i=1; i<=m; i++)
if (i!=t)
idprod = (idprod==id)? gel(I,c[i])
: idealmul(nf,idprod,gel(I,c[i]));
}
p1 = idealmul(nf,gel(v,t),gel(I,k)); c[t]=0;
det1 = (typ(det1)==t_INT)? p1: idealadd(nf,p1,det1);
}
else
{
rg++; piv=gel(v,t); c[t]=k;
for (i=1; i<=m; i++)
if (!c[i])
{
for (j=1; j<=m; j++)
if (c[j] && j!=t)
{
p1 = gsub(nfmul(nf,piv,gcoeff(pass,i,j)),
nfmul(nf,gel(v,i),gcoeff(pass,t,j)));
gcoeff(pass,i,j) = rg>1? nfdiv(nf,p1,pivprec)
: p1;
}
gcoeff(pass,i,t) = gneg(gel(v,i));
}
}
}
if (gc_needed(av1,1))
{
if(DEBUGMEM>1) pari_warn(warnmem,"nfdetint");
gerepileall(av1,6, &det1,&piv,&pivprec,&pass,&v,&idprod);
}
}
if (!cm) { avma = av; return cgetg(1,t_MAT); }
return gerepileupto(av, idealmul(nf,idprod,det1));
}
static void
nfcleanmod(GEN nf, GEN x, long lim, GEN D)
{
GEN DZ, DZ2, dD;
long i;
D = Q_remove_denom(D, &dD);
DZ = gcoeff(D,1,1); DZ2 = shifti(DZ, -1);
for (i = 1; i <= lim; i++)
{
GEN c = nf_to_scalar_or_basis(nf, gel(x,i));
switch(typ(c))
{
case t_INT:
if (!signe(c)) break;
if (dD) c = mulii(c, dD);
c = centermodii(c, DZ, DZ2);
if (dD) c = Qdivii(c,dD);
break;
case t_FRAC: {
GEN dc = gel(c,2), nc = gel(c,1), N = mulii(DZ, dc);
if (dD) nc = mulii(nc, dD);
c = centermodii(nc, N, shifti(N,-1));
c = Qdivii(c, dD ? mulii(dc,dD): dc);
break;
}
case t_COL: {
GEN dc;
c = Q_remove_denom(c, &dc);
if (dD) c = ZC_Z_mul(c, dD);
c = ZC_hnfrem(c, dc? ZM_Z_mul(D,dc): D);
dc = mul_content(dc, dD);
if (ZV_isscalar(c))
{
c = gel(c,1);
if (dc) c = Qdivii(c,dc);
}
else
if (dc) c = RgC_Rg_div(c, dc);
break;
}
}
gel(x,i) = c;
}
}
GEN
nfhnfmod(GEN nf, GEN x, GEN D)
{
long li, co, i, j, def, ldef;
pari_sp av0=avma, av;
GEN dA, dI, d0, w, p1, d, u, v, A, I, J, di;
nf = checknf(nf);
check_ZKmodule(x, "nfhnfmod");
A = gel(x,1);
I = gel(x,2);
co = lg(A); if (co==1) return cgetg(1,t_MAT);
li = lgcols(A);
if (typ(D)!=t_MAT) D = idealhnf_shallow(nf, D);
D = Q_remove_denom(D, NULL);
RgM_check_ZM(D, "nfhnfmod");
av = avma;
A = RgM_to_nfM(nf, A);
A = Q_remove_denom(A, &dA);
I = Q_remove_denom(leafcopy(I), &dI);
dA = mul_denom(dA,dI);
if (dA) D = ZM_Z_mul(D, powiu(dA, minss(li,co)));
def = co; ldef = (li>co)? li-co+1: 1;
for (i=li-1; i>=ldef; i--)
{
def--; j=def; while (j>=1 && isintzero(gcoeff(A,i,j))) j--;
if (!j) continue;
if (j==def) j--;
else {
swap(gel(A,j), gel(A,def));
swap(gel(I,j), gel(I,def));
}
for ( ; j; j--)
{
GEN a, b, S, T, S0, T0 = gel(A,j);
b = gel(T0,i); if (isintzero(b)) continue;
S0 = gel(A,def); a = gel(S0,i);
d = nfbezout(nf, a,b, gel(I,def),gel(I,j), &u,&v,&w,&di,0);
S = colcomb(nf, u,v, S0,T0);
T = colcomb(nf, a,gneg(b), T0,S0);
if (u != gen_0 && v != gen_0)
nfcleanmod(nf, S, i, idealmul(nf,D,di));
nfcleanmod(nf, T, i, idealdiv(nf,D,w));
gel(A,def) = S; gel(A,j) = T;
gel(I,def) = d; gel(I,j) = w;
}
if (gc_needed(av,2))
{
if(DEBUGMEM>1) pari_warn(warnmem,"[1]: nfhnfmod, i = %ld", i);
gerepileall(av,dA? 4: 3, &A,&I,&D,&dA);
}
}
def--; d0 = D;
A += def; A[0] = evaltyp(t_MAT)|evallg(li);
I += def; I[0] = evaltyp(t_VEC)|evallg(li);
J = cgetg(li,t_VEC);
for (i=li-1; i>=1; i--)
{
GEN b = gcoeff(A,i,i);
d = nfbezout(nf, gen_1,b, d0,gel(I,i), &u,&v,&w,&di,0);
p1 = nfC_nf_mul(nf,gel(A,i),v);
if (i > 1)
{
d0 = idealmul(nf,d0,di);
nfcleanmod(nf, p1, i, d0);
}
gel(A,i) = p1; gel(p1,i) = gen_1;
gel(I,i) = d;
gel(J,i) = di;
}
for (i=li-2; i>=1; i--)
{
d = gel(I,i);
for (j=i+1; j<li; j++)
{
GEN c = gcoeff(A,i,j); if (isintzero(c)) continue;
c = element_close(nf, c, idealmul(nf,d,gel(J,j)));
gel(A,j) = colcomb1(nf, gneg(c), gel(A,j),gel(A,i));
}
if (gc_needed(av,2))
{
if(DEBUGMEM>1) pari_warn(warnmem,"[2]: nfhnfmod, i = %ld", i);
gerepileall(av,dA? 4: 3, &A,&I,&J,&dA);
}
}
idV_simplify(I);
if (dA) I = gdiv(I,dA);
return gerepilecopy(av0, mkvec2(A, I));
}