import numpy as np 
import matplotlib.pyplot as plt
import glob


i=0
tx_matrix = np.zeros((len(glob.glob("*.dat")), len(np.genfromtxt(glob.glob("*.dat")[0])[:,0]))  )

filelist =glob.glob("*.dat")

filelist.sort()
freqlist= np.zeros(len(filelist))
Blist= np.zeros(len(filelist))

outfile = open('tx_matrix','w')


for file in filelist:
	data_fromfile = np.genfromtxt(file)
	freqlist = data_fromfile[:,0]
	tx = data_fromfile[:,1]*0.01
	tx_matrix[i,:] =  tx
	Blist[i] = float(file[0:5])
	i=i+1
	for item in tx:
  		outfile.write("%s " % item)
	outfile.write("\n")
outfile.close()	    

freqmin = min(freqlist)
freqmax = max(freqlist)
#freqticks = np.arange(int(freqmin),freqmax, int( (freqmax-freqmin)/3))
#freqticks = [40,50]
#print(Blist)
Blist = Blist*13.3/9
extent = [freqmin, freqmax,max(Blist),min(Blist) ]

plt.figure()
plt.imshow(tx_matrix, extent=extent, cmap='gnuplot2',interpolation='none',vmin=0.77, vmax=1 )
#plt.xticks(freqticks, fontsize=9)
plt.colorbar(orientation='vertical')
plt.title('extinction over frequency and magnetic field')
plt.xlabel('probe frequency (MHz)')
plt.ylabel('magnetic field (Gauss)')


#plt.figure()
#plt.imshow(ac_scat_matrix.transpose(), extent=extent, cmap='gnuplot2',interpolation='none')
#plt.xticks(freqticks, fontsize=9)
#plt.colorbar(orientation='vertical')


# extract tx min and resonance freq from tx_matrix_scat
#tx_min = np.zeros(int(max_scat/scat_step_size))
#freq_tx_min = np.zeros(int(max_scat/scat_step_size))
#for k in range(0,int(max_scat/scat_step_size)):
#	tx_min[k] = (1-tx_matrix_scat[:,k]).max()
#	freq_tx_min[k] = (1-tx_matrix_scat[:,k]).argmax()*0.6 +freqmin
#plt.figure()
#plt.plot(tx_min,'o')
#plt.figure()
#plt.plot(freq_tx_min,'o')
###########


#print('extinction:', tx_matrix[:,0].min())

plt.show()