load('ZetaFunctionsNewtonND.py')
load('zetas_triples_general.py')
R3.<x, y, z> = PolynomialRing(QQ, order='negdeglex')
R2.<u, v> = PolynomialRing(QQ, order='negdeglex')
def monomios(a,b):
    f = (u^a * v^b)^2 + u^(2 * (a + b) + 1) + v^(2 * (a + b) + 1)
    g = (u^a * v^b)^3 + u^(3 * (a + b) + 1) + v^(3 * (a + b) + 1)
    F = z^3 - 3 * f(u=x, v=y) * z + 2 * g(u=x, v=y)
    A = -2 * identity_matrix(QQ, 3 * (a + b) + 1)
    A[0, 0] = -(3 * (a + b) + 1)
    A[3 * a, 3 * a] = A[3 * (a + b), 3 * (a + b)] = -1
    A[0, 3 * a] = A[3 * a, 0] = A[3 * (a + b), 0] = A[0, 3 * (a + b)] = 1
    for i in (1 .. 3 * a - 1):
        A[i, i + 1] = A[i + 1, i] = 1
    for i in (3 * a + 1  .. 3 * (a + b) - 1):
        A[i, i + 1] = A[i + 1, i] = 1
    Vf, Vg, Vd = [vector(QQ, 3 * (a + b) + 1) for i in (1, 2, 3)]
    Vf[2 * a] = Vf[3 * a + 2 * b] = 1
    Vg[3 * a] = Vg[3 * (a + b)] = 1
    Vd[3 * a] = Vd[3 * (a + b)] = 1
    Vd[0] = 3 * (a + b) + 1
    n, Nf, Ng, Nd, Nhs, Nks, euler, nus, pols = trinomiales_general(A, Vf, Vg, (), (), Vd, total=False)
    zetatop, zetamon, dic = zetas(n, pols, info=True)
    return zetatop, dic
res = {}
for a in (1, .., 100):
    for b in (a, .., 100):
        res[a, b] = monomios(a, b)
save(res,'zetas')