import numpy as np
import h5py
import matplotlib.pyplot as plt
import argparse
U_MEV   = 1.60217657e-6

parser = argparse.ArgumentParser(description="Read neutrino energy files")
parser.add_argument("--mass",help="progenitor mass to use as input, default 10",type=str,default='12')
parser.add_argument("--neutrino",help="neutrino species to use as input, default 0",type=str,default='0')
args = parser.parse_args()

lum_file = h5py.File("lum_spec_{}M_r10000_dat.h5".format(args.mass),"r")


#plot the angle-averaged neutrino luminosity for electron-neutrinos, summed over all bins,
time = lum_file["nu0"].attrs["time"]
de = np.array(lum_file["nu{}".format(args.neutrino)]["degroup"])
dlum=np.zeros((4704,len(de[0])))
for i in range(12):
    dlum[:,i]=lum_file["nu{}".format(args.neutrino)]["g{}".format(i)]
lum0 = np.sum(dlum*de,axis=1)
plt.figure()
plt.plot(time,lum0/100.)
plt.xlabel(r'Time after bounce [s]')
plt.ylabel(r'Average Neutrino Luminosity [10$^{52}$ erg s$^{-1}$ MeV $^{-1}$]')
plt.xlim(0,.300)
plt.savefig(r'lum0_{}M.pdf'.format(args.mass))
plt.close()