/*
  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([41, 13, -12, -2, 1])
F.<w> = NumberField(g)
ZF = F.ring_of_integers()

NN = ZF.ideal([16, 2, 2])

primes_array = [
[9, 3, -w^3 + 3*w^2 + 5*w - 15],\
[9, 3, -w^3 + 8*w + 8],\
[16, 2, 2],\
[19, 19, w + 1],\
[19, 19, -w^2 + 6],\
[19, 19, -w^2 + 2*w + 5],\
[19, 19, -w + 2],\
[25, 5, 2*w^2 - 2*w - 13],\
[29, 29, -w^2 + 9],\
[29, 29, -w^2 + 2*w + 6],\
[29, 29, w^2 - 7],\
[29, 29, -w^2 + 2*w + 8],\
[31, 31, -2*w^2 + w + 12],\
[31, 31, 2*w^2 - 3*w - 11],\
[41, 41, -w],\
[41, 41, -w + 1],\
[49, 7, w^3 + 2*w^2 - 10*w - 20],\
[49, 7, w^3 - 5*w^2 - 3*w + 27],\
[61, 61, 2*w^2 - 3*w - 14],\
[61, 61, 2*w^2 - w - 15],\
[71, 71, -w^3 + w^2 + 7*w - 5],\
[71, 71, w^3 - 2*w^2 - 6*w + 2],\
[79, 79, w^3 - 2*w^2 - 6*w + 4],\
[79, 79, 3*w^2 - 2*w - 18],\
[89, 89, 2*w^3 - 6*w^2 - 11*w + 33],\
[89, 89, -4*w^2 + 3*w + 25],\
[101, 101, w^3 + w^2 - 9*w - 12],\
[101, 101, w^3 - 4*w^2 - 4*w + 19],\
[109, 109, -w^3 + 3*w^2 + 6*w - 19],\
[109, 109, -2*w^2 + w + 18],\
[121, 11, -3*w^2 + 3*w + 19],\
[121, 11, 3*w^2 - 3*w - 20],\
[131, 131, 2*w^2 - 3*w - 15],\
[131, 131, 2*w^2 - w - 16],\
[139, 139, 3*w^2 - 4*w - 21],\
[139, 139, -w^3 - w^2 + 7*w + 13],\
[151, 151, 4*w^2 - 3*w - 27],\
[151, 151, 2*w^3 - w^2 - 13*w - 8],\
[151, 151, w^3 + 3*w^2 - 13*w - 27],\
[151, 151, 4*w^2 - 5*w - 26],\
[179, 179, w^3 - 3*w^2 - 5*w + 11],\
[179, 179, 3*w^3 - 11*w^2 - 14*w + 61],\
[179, 179, 2*w^3 - 2*w^2 - 12*w + 3],\
[179, 179, w^3 - 8*w - 4],\
[181, 181, -w - 4],\
[181, 181, w^3 - 5*w^2 - 3*w + 25],\
[181, 181, w^3 + 2*w^2 - 10*w - 18],\
[181, 181, w - 5],\
[229, 229, w^2 - 2*w - 11],\
[229, 229, w^2 - 12],\
[239, 239, 6*w^2 - 7*w - 42],\
[239, 239, 2*w^3 - 2*w^2 - 14*w - 3],\
[241, 241, w^3 - 4*w^2 - 4*w + 18],\
[241, 241, 3*w^2 - 2*w - 16],\
[269, 269, 3*w^2 - 4*w - 22],\
[269, 269, -w^3 + 6*w + 8],\
[281, 281, w^3 - w^2 - 9*w + 4],\
[281, 281, 2*w^3 - 9*w^2 - 8*w + 49],\
[311, 311, -2*w^3 - w^2 + 17*w + 19],\
[311, 311, 2*w^3 - 7*w^2 - 9*w + 33],\
[331, 331, w^3 + 2*w^2 - 9*w - 21],\
[331, 331, w^3 - 5*w^2 - 2*w + 27],\
[349, 349, 2*w^3 - w^2 - 15*w - 4],\
[349, 349, -2*w^3 + 5*w^2 + 11*w - 18],\
[359, 359, -w^3 + 6*w^2 - 31],\
[359, 359, -w^3 - 3*w^2 + 9*w + 26],\
[379, 379, 2*w^3 - 5*w^2 - 11*w + 20],\
[379, 379, 2*w^3 - 14*w - 11],\
[401, 401, w^3 - 3*w^2 - 4*w + 16],\
[401, 401, -2*w^3 + 4*w^2 + 11*w - 16],\
[401, 401, 2*w^3 - 2*w^2 - 13*w - 3],\
[401, 401, w^3 - 7*w - 10],\
[409, 409, w^3 - 4*w^2 - 4*w + 17],\
[409, 409, 2*w^3 - 6*w^2 - 9*w + 22],\
[409, 409, -2*w^3 + 15*w + 9],\
[409, 409, -w^3 + 3*w^2 + 6*w - 12],\
[419, 419, -w^3 + 5*w^2 + 3*w - 24],\
[419, 419, w^3 + 2*w^2 - 10*w - 17],\
[421, 421, w^3 + 3*w^2 - 12*w - 25],\
[421, 421, -w^3 - w^2 + 9*w + 7],\
[431, 431, -w^3 - 2*w^2 + 12*w + 20],\
[431, 431, w^3 + 3*w^2 - 10*w - 28],\
[431, 431, -w^3 + 6*w^2 + w - 34],\
[431, 431, -w^3 + 5*w^2 + 5*w - 29],\
[439, 439, w^3 - 7*w^2 - 2*w + 39],\
[439, 439, w^3 + 4*w^2 - 13*w - 31],\
[449, 449, -w^3 - 4*w^2 + 13*w + 34],\
[449, 449, 6*w^2 - 5*w - 36],\
[461, 461, -w^3 + 5*w^2 + w - 25],\
[461, 461, 2*w^3 - 15*w - 12],\
[461, 461, -2*w^3 + 6*w^2 + 9*w - 25],\
[461, 461, w^3 + 2*w^2 - 8*w - 20],\
[479, 479, -w^3 - 4*w^2 + 12*w + 31],\
[479, 479, 2*w^3 - 15*w - 16],\
[479, 479, -2*w^3 + 6*w^2 + 9*w - 29],\
[479, 479, w^3 - 7*w^2 - w + 38],\
[491, 491, w^2 - 3*w - 7],\
[491, 491, -2*w^3 + 3*w^2 + 11*w - 2],\
[491, 491, 4*w^2 - 2*w - 29],\
[491, 491, w^2 + w - 9],\
[499, 499, 5*w^2 - 6*w - 33],\
[499, 499, 5*w^2 - 4*w - 34],\
[509, 509, 2*w - 3],\
[509, 509, -2*w - 1],\
[541, 541, -w^3 - 4*w^2 + 12*w + 29],\
[541, 541, -w^3 + 7*w^2 + w - 36],\
[599, 599, -w^3 - w^2 + 8*w + 7],\
[599, 599, w^3 - 4*w^2 - 3*w + 13],\
[619, 619, w^2 - 3*w - 6],\
[619, 619, w^2 + w - 8],\
[641, 641, -2*w^3 + 16*w + 11],\
[641, 641, -2*w^3 + 6*w^2 + 10*w - 25],\
[659, 659, 2*w^3 + w^2 - 16*w - 16],\
[659, 659, 2*w^3 - 7*w^2 - 8*w + 29],\
[691, 691, -w^3 + 5*w^2 + 3*w - 23],\
[691, 691, -w^3 + 3*w^2 + 6*w - 11],\
[701, 701, w^3 - w^2 - 9*w + 3],\
[701, 701, 2*w^3 - w^2 - 14*w - 5],\
[709, 709, 2*w^3 - 3*w^2 - 11*w + 6],\
[719, 719, w^2 - 2*w - 12],\
[719, 719, -w^2 + 13],\
[739, 739, -2*w^3 + 8*w^2 + 7*w - 36],\
[739, 739, w^2 + 2*w - 13],\
[751, 751, -2*w^3 + 5*w^2 + 10*w - 20],\
[751, 751, 2*w^3 - w^2 - 14*w - 7],\
[761, 761, w^3 + 4*w^2 - 12*w - 32],\
[761, 761, -w^3 + 7*w^2 + w - 39],\
[769, 769, -2*w^3 - 4*w^2 + 23*w + 45],\
[769, 769, 2*w^3 - 5*w^2 - 9*w + 21],\
[769, 769, 2*w^3 + 2*w^2 - 18*w - 31],\
[769, 769, 2*w^3 - 10*w^2 - 9*w + 62],\
[811, 811, -w^3 + 10*w + 5],\
[811, 811, -5*w^2 + 8*w + 30],\
[811, 811, w^3 - 8*w^2 + 2*w + 43],\
[811, 811, 2*w^3 - 2*w^2 - 13*w + 2],\
[829, 829, -2*w^3 - 2*w^2 + 18*w + 25],\
[829, 829, -2*w^3 + 8*w^2 + 8*w - 39],\
[859, 859, -w^3 - 4*w^2 + 12*w + 33],\
[859, 859, -w^3 + 7*w^2 + w - 40],\
[919, 919, w^3 - 4*w^2 - 3*w + 24],\
[919, 919, -w^3 + 5*w^2 + w - 26],\
[929, 929, 5*w^2 - 7*w - 28],\
[929, 929, -w^3 + 2*w^2 + 6*w - 13],\
[929, 929, w^3 - w^2 - 7*w - 6],\
[929, 929, -2*w^3 + 2*w^2 + 14*w - 5],\
[941, 941, -2*w^3 + 3*w^2 + 11*w - 1],\
[941, 941, 2*w^3 - 3*w^2 - 11*w + 11],\
[961, 31, 5*w^2 - 5*w - 33],\
[971, 971, -w^3 - 3*w^2 + 11*w + 22],\
[971, 971, 7*w^2 - 5*w - 45],\
[971, 971, 7*w^2 - 9*w - 43],\
[971, 971, w^3 - 6*w^2 - 2*w + 29],\
[991, 991, w^3 - 3*w^2 - 3*w + 15],\
[991, 991, -w^3 + 5*w^2 + 2*w - 29]]
primes = [ZF.ideal(I) for I in primes_array]

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

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

AL_eigenvalues = {}
AL_eigenvalues[ZF.ideal([16, 2, 2])] = 1

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