#read the transmission spectrum
transmission = nlread('mgo_para.nc',TransmissionSpectrum)[0]
#get the transmission spectrum
trans_coeff = transmission.transmission()
#select spin up component
T_uu = trans_coeff[0]
#select spin down component
T_dd = trans_coeff[1]
# shape of the components are [energy_points, k_points]
(n_e,n_k) = T_uu.shape
# assume equal many k_points in A and B directions
n_A=n_B=numpy.sqrt(n_k)
#reshape the arrays for plotting, assume only one energy point
T_uu = T_uu.reshape(n_A,n_B)
T_dd = T_dd.reshape(n_A,n_B)

#get the k-points, x component
K_A = transmission.kpoints()[:,0]
K_A = K_A.reshape(n_A,n_B)
#get the k-points, y component
K_B = transmission.kpoints()[:,1]
K_B = K_B.reshape(n_A,n_B)

import pylab
#plot the transmission majority spin
pylab.figure()
pylab.xlabel("$k_A \ (2\pi/a)$",fontsize=12,family='sans-serif')
pylab.ylabel("$k_B \ (2\pi/a)$",fontsize=12,family='sans-serif')
pylab.contourf(K_A,K_B,T_uu,40,cmap=pylab.cm.Spectral)
pylab.colorbar()
pylab.axis([-.5,.5,-.5,.5])
pylab.yticks(numpy.arange(-0.5,0.51,0.1),fontsize=10)
pylab.xticks(numpy.arange(-0.5,0.51,0.1),fontsize=10)
pylab.title('mgo_para, majority spin')
pylab.savefig('mgo_para_majority.png',dpi=100)

#plot the transmission minority spin
pylab.figure()
pylab.xlabel("$k_A \ (2\pi/a)$",fontsize=12,family='sans-serif')
pylab.ylabel("$k_B \ (2\pi/a)$",fontsize=12,family='sans-serif')
pylab.contourf(K_A,K_B,T_dd,40,cmap=pylab.cm.Spectral)
pylab.colorbar()
pylab.axis([-.5,.5,-.5,.5])
pylab.yticks(numpy.arange(-0.5,0.51,0.1),fontsize=10)
pylab.xticks(numpy.arange(-0.5,0.51,0.1),fontsize=10)
pylab.title('mgo_para, minority spin')
pylab.savefig('mgo_para_minority.png',dpi=100)
pylab.show()