/*
  This code can be loaded, or copied and paste using cpaste, into Sage.
  It will load the data associated to the HMF, including
  the field, level, and Hecke and Atkin-Lehner eigenvalue data.
*/

P.<x> = PolynomialRing(QQ)
g = P([1, -2, -5, 0, 1])
F.<w> = NumberField(g)
ZF = F.ring_of_integers()

NN = ZF.ideal([31, 31, -w^2 + 2*w + 3])

primes_array = [
[4, 2, -w^3 + 4*w + 1],\
[5, 5, w - 1],\
[7, 7, -w^2 + w + 2],\
[13, 13, -w^3 + w^2 + 4*w],\
[13, 13, -w^2 + 3],\
[17, 17, -w^2 + 2],\
[19, 19, -w^3 + 5*w],\
[31, 31, -w^2 + 2*w + 3],\
[37, 37, -2*w^3 + w^2 + 8*w - 1],\
[43, 43, -w - 3],\
[53, 53, -w^3 + 2*w^2 + 3*w - 2],\
[53, 53, w^3 - 6*w - 2],\
[59, 59, 2*w^3 - w^2 - 10*w - 2],\
[61, 61, 2*w^3 - w^2 - 10*w],\
[61, 61, 2*w^3 - w^2 - 8*w],\
[71, 71, 2*w^3 - 9*w - 2],\
[73, 73, -w^3 - w^2 + 6*w + 3],\
[81, 3, -3],\
[83, 83, -w^3 + 2*w^2 + 3*w - 3],\
[101, 101, 2*w^3 - 8*w - 3],\
[125, 5, w^3 + w^2 - 4*w - 6],\
[127, 127, 2*w^3 - w^2 - 7*w - 2],\
[127, 127, w^3 - 7*w - 7],\
[131, 131, w^3 - 3*w - 4],\
[131, 131, 3*w^3 - 13*w - 7],\
[137, 137, w^2 + w - 4],\
[137, 137, -w^3 + 2*w^2 + 4*w - 4],\
[149, 149, 3*w + 5],\
[157, 157, -2*w^3 - w^2 + 7*w + 2],\
[157, 157, -2*w^3 + 2*w^2 + 9*w - 1],\
[163, 163, -2*w^3 + w^2 + 7*w],\
[163, 163, -3*w^3 + w^2 + 15*w + 3],\
[163, 163, 2*w^3 - 7*w - 1],\
[163, 163, 2*w^3 - 11*w - 5],\
[167, 167, -2*w^3 + w^2 + 9*w - 4],\
[167, 167, w^3 - w^2 - 7*w + 1],\
[167, 167, w^3 - w^2 - 7*w - 1],\
[167, 167, w^2 - 2*w - 5],\
[169, 13, w^3 + w^2 - 6*w - 2],\
[173, 173, -3*w^3 + 13*w + 3],\
[181, 181, 2*w^2 + w - 4],\
[191, 191, 2*w^3 - 10*w - 1],\
[191, 191, -2*w^3 + 2*w^2 + 6*w - 3],\
[193, 193, 3*w^3 - 13*w - 6],\
[197, 197, -w^3 + w^2 + 2*w - 3],\
[197, 197, w^3 + w^2 - 7*w - 7],\
[199, 199, -w^2 + 2*w + 6],\
[223, 223, -2*w^3 + w^2 + 11*w + 2],\
[227, 227, -3*w^3 + 2*w^2 + 15*w - 1],\
[229, 229, -w^3 + w^2 + 6*w - 5],\
[233, 233, 2*w^3 - 11*w - 4],\
[239, 239, -2*w^3 + 11*w + 3],\
[251, 251, 2*w^3 - w^2 - 10*w - 4],\
[257, 257, -2*w^3 + w^2 + 6*w + 1],\
[257, 257, w^3 - 7*w - 2],\
[263, 263, 2*w^3 - 9*w],\
[269, 269, w^3 - 4*w - 6],\
[269, 269, -2*w^3 + 8*w - 3],\
[271, 271, 2*w^2 - w - 5],\
[277, 277, -3*w^3 + 2*w^2 + 11*w - 2],\
[277, 277, 3*w^3 - w^2 - 15*w - 5],\
[281, 281, 2*w^2 - 5],\
[283, 283, -3*w^3 + 2*w^2 + 12*w + 2],\
[283, 283, -2*w^3 + w^2 + 6*w + 6],\
[293, 293, -2*w^3 - w^2 + 12*w + 7],\
[293, 293, -2*w^3 + 9*w - 1],\
[311, 311, -3*w^3 + w^2 + 12*w + 4],\
[313, 313, -w^3 + w^2 + 2*w - 4],\
[313, 313, -4*w^3 + 2*w^2 + 16*w - 3],\
[331, 331, 3*w^3 - w^2 - 16*w - 5],\
[331, 331, 2*w^3 - 2*w^2 - 8*w - 1],\
[343, 7, 2*w^3 - 3*w^2 - 9*w + 6],\
[359, 359, -2*w^3 + 10*w - 1],\
[367, 367, -w^3 + 2*w^2 + w - 4],\
[367, 367, 3*w^3 - 11*w - 1],\
[373, 373, -w^3 + w^2 + 2*w - 5],\
[373, 373, 2*w^3 - 2*w^2 - 11*w - 1],\
[389, 389, -2*w^3 + 3*w^2 + 8*w - 6],\
[409, 409, 3*w^3 - 2*w^2 - 13*w - 1],\
[419, 419, 3*w^3 - w^2 - 12*w - 1],\
[419, 419, w^3 - w^2 - 4*w - 4],\
[439, 439, w^2 + 2*w - 4],\
[439, 439, 3*w^3 + w^2 - 11*w - 5],\
[443, 443, 2*w^3 + 2*w^2 - 7*w - 8],\
[449, 449, 4*w^3 - w^2 - 16*w - 8],\
[457, 457, 2*w^3 + w^2 - 11*w - 4],\
[461, 461, -5*w^3 + 2*w^2 + 24*w + 2],\
[463, 463, -w^3 + 3*w^2 + 6*w - 5],\
[467, 467, w^3 - 3*w^2 - 6*w + 4],\
[467, 467, 3*w^3 - w^2 - 12*w - 2],\
[479, 479, -w^3 + w^2 + 4*w - 6],\
[479, 479, -w^3 + 2*w + 8],\
[491, 491, -w^3 + 2*w^2 + 2*w - 6],\
[491, 491, w - 5],\
[499, 499, 2*w^3 + w^2 - 10*w - 4],\
[509, 509, -4*w^3 + w^2 + 20*w + 5],\
[521, 521, w^3 - w + 6],\
[523, 523, -4*w^3 + 18*w + 5],\
[523, 523, -2*w^3 - 2*w^2 + 9*w + 12],\
[557, 557, w^2 - 3*w - 6],\
[571, 571, 3*w^3 - 11*w - 8],\
[577, 577, 2*w^3 + 2*w^2 - 11*w - 7],\
[577, 577, -3*w^3 + 4*w^2 + 11*w - 6],\
[587, 587, -w^3 - w^2 + 7*w + 11],\
[593, 593, w^3 + w^2 - 8*w - 8],\
[593, 593, -2*w^3 + 4*w^2 + 6*w - 9],\
[599, 599, 4*w^3 - 17*w - 9],\
[599, 599, w^3 + w^2 - 6*w - 10],\
[601, 601, 3*w^2 - 10],\
[607, 607, 4*w^3 - w^2 - 18*w - 8],\
[619, 619, w^2 - w - 8],\
[641, 641, -w^3 + 3*w^2 + 2*w - 10],\
[643, 643, 3*w^3 - w^2 - 16*w + 1],\
[653, 653, -2*w^3 + 2*w^2 + 9*w + 3],\
[677, 677, w^3 + w^2 - 8*w - 11],\
[677, 677, 2*w^3 - w^2 - 12*w - 3],\
[683, 683, 3*w^3 - 2*w^2 - 13*w - 2],\
[683, 683, 3*w^3 + 2*w^2 - 16*w - 10],\
[691, 691, -2*w^3 - 2*w^2 + 12*w + 15],\
[691, 691, 3*w^3 - 2*w^2 - 15*w - 2],\
[691, 691, w^3 - 2*w^2 - 6*w + 6],\
[691, 691, -4*w^3 + 3*w^2 + 15*w - 2],\
[719, 719, 2*w^3 - 7*w - 8],\
[719, 719, w^2 + 2*w - 5],\
[727, 727, -w^3 + 3*w^2 + 4*w - 8],\
[727, 727, 2*w^3 - 9*w - 9],\
[733, 733, w^3 - 8*w - 2],\
[739, 739, -w^3 - 2*w^2 + 4*w + 10],\
[743, 743, w^3 - w^2 - 2*w + 8],\
[743, 743, 4*w^3 - w^2 - 17*w - 4],\
[751, 751, -2*w^3 + 2*w^2 + 6*w - 5],\
[751, 751, w^3 - 5*w - 7],\
[769, 769, 3*w^3 - 2*w^2 - 10*w],\
[787, 787, -2*w^3 + 3*w^2 + 8*w],\
[787, 787, 3*w^3 - w^2 - 10*w - 4],\
[809, 809, 3*w^3 + w^2 - 16*w - 6],\
[811, 811, 3*w^3 - 14*w - 2],\
[811, 811, -w^3 + 3*w^2 + 5*w - 5],\
[811, 811, w^3 - 2*w - 6],\
[811, 811, 3*w^3 - w^2 - 11*w - 3],\
[823, 823, w^3 + 2*w^2 - 3*w - 8],\
[827, 827, -w^3 + 3*w^2 + 3*w - 7],\
[829, 829, w^3 + w^2 - 3*w - 7],\
[853, 853, 3*w^3 + w^2 - 14*w - 6],\
[857, 857, 3*w^3 + w^2 - 13*w - 5],\
[857, 857, 3*w^3 - 3*w^2 - 11*w - 1],\
[859, 859, 4*w^3 - 2*w^2 - 18*w - 3],\
[859, 859, 3*w^3 - 11*w - 3],\
[863, 863, 3*w^3 + w^2 - 17*w - 9],\
[877, 877, -w^3 - 2*w^2 + 9*w + 1],\
[877, 877, 5*w^3 + w^2 - 23*w - 11],\
[911, 911, w^2 - 4*w - 6],\
[911, 911, -2*w^3 + 2*w^2 + 10*w + 3],\
[919, 919, -4*w^3 + 3*w^2 + 18*w - 1],\
[929, 929, 2*w^3 - 13*w - 12],\
[929, 929, -w^3 + 2*w^2 + 7*w - 5],\
[953, 953, -3*w^3 + 3*w^2 + 13*w - 1],\
[967, 967, -3*w^3 + 4*w^2 + 13*w - 7],\
[971, 971, 2*w^2 - 3*w - 12],\
[971, 971, 3*w^3 - w^2 - 16*w - 3]]
primes = [ZF.ideal(I) for I in primes_array]

heckePol = x^2 - 2*x - 7
K.<e> = NumberField(heckePol)

hecke_eigenvalues_array = [e, -2, 0, e - 3, -2, 2, e + 3, -1, 3*e - 5, -2*e + 6, -2, -2*e + 8, 2*e + 2, 6, e + 5, 2*e - 10, -e - 5, e + 9, -e - 11, -e - 1, -2*e - 8, -2*e - 6, 2*e + 6, -4*e + 8, -4, 2*e - 8, 2, -4*e + 2, 2*e - 12, -4*e + 10, 2*e - 14, 5*e - 1, -2*e - 10, -6*e + 2, -3*e + 3, -3*e + 19, -5*e + 5, 2*e - 2, -6*e + 8, -4*e + 10, -2, -3*e + 11, 5*e + 3, 10, -e - 9, 4*e - 6, -4*e + 12, 2*e + 6, -2*e - 18, -2*e - 16, -3*e + 21, -4*e - 12, e - 21, -2*e + 20, 8*e - 14, 3*e - 3, e + 13, 4*e + 2, 3*e - 11, -8*e + 14, -2*e + 16, -4*e - 2, 2*e + 10, -4*e, 4*e - 6, 4*e - 22, -4*e + 4, -4*e + 14, -2*e + 20, 4*e + 8, 9*e - 13, -7*e + 23, 4*e + 12, 8*e - 8, -4*e - 4, 5*e + 1, 10*e - 4, -4*e + 2, -7*e - 7, -6*e + 26, e + 3, 8, -6*e + 22, -10*e + 22, e + 17, -5*e + 15, -11*e + 9, 8, -e + 5, 3*e + 17, 3*e + 5, -8*e + 16, -6*e - 14, 8*e + 4, 2*e - 14, -8*e + 30, 4*e + 6, -8*e - 4, 2*e - 22, 6*e - 16, -3*e + 31, -14, 26, -5*e + 1, 3*e + 7, -2*e + 28, -8*e + 8, 3*e + 13, 8*e - 22, 3*e - 11, 28, 2, -2*e - 18, -8*e - 2, 5*e - 15, -e + 23, -9*e + 5, -6*e + 2, 4*e, -4*e - 8, 8*e + 12, -10*e + 30, 7*e - 15, -2*e - 6, e - 33, -3*e + 35, -18, 12*e - 8, -4*e + 12, e + 15, -6*e + 38, -2*e + 2, -12*e + 14, e + 3, 10*e - 22, 6*e + 12, 15*e - 3, 6*e + 6, -6*e + 18, -6*e - 14, e - 1, 12*e, 14*e - 8, -19*e + 17, -5*e - 1, 10*e - 16, 10*e - 30, 4*e - 16, 2*e - 2, 12*e + 10, 14*e - 8, -6*e - 10, 8*e - 8, 10*e - 2, 12*e - 26, -6*e + 8, -14*e + 8, 4*e - 36, -17*e + 29, 8*e - 20]
hecke_eigenvalues = {}
for i in range(len(hecke_eigenvalues_array)):
    hecke_eigenvalues[primes[i]] = hecke_eigenvalues_array[i]

AL_eigenvalues = {}
AL_eigenvalues[ZF.ideal([31, 31, -w^2 + 2*w + 3])] = 1

# EXAMPLE:
# pp = ZF.ideal(2).factor()[0][0]
# hecke_eigenvalues[pp]