#!/usr/bin/python import numpy as np import matplotlib.pyplot as plt import pyramids.io.result as dP import pyramids.plot.setting as ma import pyramids.plot.PlotUtility as pu #------------------------------------------------------------------------------ time, exe = dP.getProjectedPartition() time, eigen = dP.getAdiabaticEigenvalue() #print eigen[:,1,:] #h**o = dP.getHomo() #print h**o print exe.shape, eigen.shape c = ma.getColors(exe.shape[2], cmap='brg') norm = 50.0 #/np.max(exe[:,:,:] - exe[0,:,:]) #print norm kpts = [0] for kpt in kpts: fig, ax = plt.subplots(1, 1, sharex=True, sharey=False, figsize=(8, 6)) SaveName = __file__.split('/')[-1].split('.')[0] + str(kpt) h**o = 0 for i, band in enumerate(eigen[0, kpt, :]): #print band if band < 0: h**o = i evolvingBands = range(0, h**o + 40) X, Y = np.meshgrid(time, eigen[0, kpt, evolvingBands]) Z = np.transpose(exe[0, kpt, evolvingBands] - exe[:, kpt, evolvingBands])
nRow = len(ls.selectedTimeStep)/nCol fig, ax = plt.subplots(1,1,sharex=False,sharey=True,figsize=(8,6)) SaveName = __file__.split('/')[-1].split('.')[0] kpath = ls.kpath cut = ls.cut print kpath specialKPoints = ls.specialKPoints x = ls.x h**o = dP.getHomo() evolvingBands = range(0, h**o + 7) time, exe = dP.getProjectedPartition() time, eigen = dP.getAdiabaticEigenvalue() colors = ma.getColors(5,cmap='gnuplot') #for index, step in enumerate(ls.selectedTimeStep): excited = np.abs(exe[0,kpath,:] - exe[0,kpath,:]) norm = ls.norm/np.max(exe[:,kpath,:] - exe[0,kpath,:]) eigenvalue = eigen[0,kpath,:] line = [] scatter = [] for i in evolvingBands: if i < h**o: part = excited[:,i] if ls.drawfill: s = ax.fill_between(x, eigenvalue[:,i] - norm*part, eigenvalue[:,i] + norm*part, lw=0.0,color='b',alpha=0.7)