myPlotter = Plotter()
kVec = arange(-3,3,0.1)
myPlotter.plotEigenvalues(BigPotWellSolverGaAsX, kVec, 6)
myPlotter.savePlots("GaAsBigXE.png")
myPlotter.plotEigenvalues(BigPotWellSolverGaAsZ, kVec, 6)
myPlotter.savePlots("GaAsBigZE.png")
myPlotter.plotEigenvalues(SmallPotWellSolverGaAsX, kVec, 4)
myPlotter.savePlots("GaAsSmallXE.png")
myPlotter.plotEigenvalues(SmallPotWellSolverGaAsZ, kVec, 4)
myPlotter.savePlots("GaAsSmallZE.png")
myPlotter.plotEigenvalues(BigPotWellSolverGaAsZ, kVec, 4)
myPlotter.plotEigenvalues(SmallPotWellSolverGaAsZ, kVec, 4)
myPlotter.savePlots("GaAsBigANDSmallE.png")
myPlotter.plotMixing(BigPotWellSolverGaAsZ, kVec)
myPlotter.savePlots("GaAsMixingZ.png")
myPlotter.plotMixing(BigPotWellSolverGaAsX, kVec)
myPlotter.savePlots("GaAsMixingX.png")
myPlotter.plotEigenvectors(BigPotWellSolverGaAsX)
myPlotter.savePlots("GaAsWaveX.png")
myPlotter.plotEigenvectors(BigPotWellSolverGaAsZ)
myPlotter.savePlots("GaAsWaveZ.png")
myPlotter.plotRotatedMixing(SmallPotWellSolverGaAsZ, kVec)
myPlotter.plotMixing(SmallPotWellSolverGaAsX, kVec)
myPlotter.savePlots("GaAsMixingRotationOfZ.png")
myPlotter.plotRotatedMixing(SmallPotWellSolverGaAsX, kVec)
myPlotter.plotMixing(SmallPotWellSolverGaAsZ, kVec)
myPlotter.savePlots("GaAsMixingRotationOfX.png")
myPlotter.plotEigenvectors(SmallPotWellSolverGaAsX, 0)
myPlotter.savePlots("GaAsXWavek0.png")
for i in xrange(matrixDim):
splitVectorsX[:,i] = np.squeeze(np.array(EigenvectorsX[i*nGridPointsX:(i+1)*nGridPointsX]))
H1rot = np.sqrt(2)/4 * (splitVectorsX[:,0] + splitVectorsX[:,3]) + np.sqrt(6)/4 * (splitVectorsX[:,1] + splitVectorsX[:,2])
L1rot = np.sqrt(6)/4 * (splitVectorsX[:,3] - splitVectorsX[:,0]) + np.sqrt(2)/4 * (-splitVectorsX[:,1] - splitVectorsX[:,2])
L2rot = np.sqrt(6)/4 * (splitVectorsX[:,0] + splitVectorsX[:,3]) + np.sqrt(2)/4 * (-splitVectorsX[:,1] - splitVectorsX[:,2])
H2rot = np.sqrt(2)/4 * (splitVectorsX[:,3] - splitVectorsX[:,0]) + np.sqrt(6)/4 * (splitVectorsX[:,1] - splitVectorsX[:,2])
H1normsq = norm(H1rot)**2
H2normsq = norm(H2rot)**2
L1normsq = norm(L1rot)**2
L2normsq = norm(L2rot)**2
total = H1normsq+ H2normsq + L1normsq + L2normsq
HHnormsq = H1normsq + H2normsq
LHnormsq = L1normsq + L2normsq
fractionsHH1.append(H1normsq/total)
fractionsHH2.append(H2normsq/total)
fractionsLH1.append(LHnormsq/total)
fractionsHHXinZ.append(HHnormsq/total)
plot(kVec, fractionsLH1, '*')
plot(kVec, fractionsHHXinZ, '*')
myPlotter.plotMixing(SmallPotWellSolverGaAsX, kVec)
show()
