【獎勵】本帖被評價13次，作者銳利的碎片增加金幣 10 個

銳利的碎片

木蟲 (正式寫手)

1ST強帖: 1
應助: 136 (高中生)
金幣: 3637.1
帖子: 988
在線: 1224.9小時
蟲號: 961933

[資源] 【原創(chuàng)】處理elk能帶的程序

自己寫的用python處理elk能帶的程序

回復此樓

1樓 2011-01-06 14:46:10

已閱回復此樓關注TA 給TA發(fā)消息送TA紅花 TA的回帖

lixiao85

銀蟲 (小有名氣)

應助: 1 (幼兒園)
金幣: 371.2
帖子: 132
在線: 75.5小時
蟲號: 1142488

★★★★★ 五星級,優(yōu)秀推薦

或許以后能用上

回復此樓

5樓2011-02-26 14:49:39

已閱回復此樓關注TA 給TA發(fā)消息送TA紅花 TA的回帖

查看全部 17 個回答

銳利的碎片

木蟲 (正式寫手)

1ST強帖: 1
應助: 136 (高中生)
金幣: 3637.1
帖子: 988
在線: 1224.9小時
蟲號: 961933

用面向對象重新寫了一遍程序，現(xiàn)在只寫了elk的類，處理其他程序（比如vasp）結果的程序可以很方便的加進來。主程序部分沒寫，這個應該沒有多難。

CODE:

class Band():

    def write_data(self):

        if self.nspin==1:

            fout=file('band.data','w')

            for i in range(self.ndiv):

                fout.write("%12.8f "%self.band[0][i])

                for j in range(self.nband):

                    fout.write("%12.8f "%self.band[j+1][i])

                fout.write('\n')

            fout.close()

        elif self.nspin==2:

            foutup=file('bandup.data','w')

            foutdn=file('banddn.data','w')

            for i in range(self.ndiv):

                foutup.write("%12.8f "%self.band[0][i])

                foutdn.write("%12.8f "%self.band[0][i])

                for j in range(self.nband/2):

                    foutup.write("%12.8f "%self.band[j+1][i])

                foutup.write('\n')

                for j in range(self.nband/2,self.nband):

                    foutdn.write("%12.8f "%self.band[j+1][i])

                foutdn.write('\n')

            foutup.close()

            foutdn.close()

    def view_band(self,emin,emax,title,xlabel,ylabel):

        import numpy as np

        import matplotlib.pyplot as plt

        #plot setup

        plt.title(title)

        plt.xlabel(xlabel)

        plt.ylabel(ylabel)

        plt.xlim(self.kpos[0],self.kpos[-1])

        plt.ylim(emin,emax)

        plt.xticks(self.kpos,self.kpname)

        plt.grid(linestyle='-')

        plt.gca().xaxis.grid(True)

        plt.gca().yaxis.grid(False)

        #plot bands

        for i in range(self.nband):

            x=self.band[0]

            y=self.band[i+1]

            if np.min(y)emax:

                continue

            else:

                if self.nspin==1:

                    plt.plot(x,y,'r-')

                elif self.nspin==2:

                    if i
                        plt.plot(x,y,'r-')

                    else:

                        plt.plot(x,y,'b--')

        plt.plot([self.kpos[0],self.kpos[-1]],[0.0,0.0],\

                 color='black',linestyle='--')

        plt.text(0.0,0.1,r'$E_F$=%6.3f eV'%self.fermi)

        plt.show()

class ElkBand(Band):

    def __init__(self):

        import sys

        import numpy as np

        try:

            fmain=file('elk.in','r')

            fpos=file('BANDLINES.OUT','r')

            fband=file('BAND.OUT','r')

            ffermi=file('EFERMI.OUT','r')

        except IOError:

            print 'Error in open nessary inputs. Check your input file'

            sys.exit()

        # initalize

        self.kpname=[] #names of kpoints

        self.kpos=[] #positions of special kpoints in G space

        self.nspin=1 #number of spins

        self.nband=0 #number of band

        self.band=[] #banddata

        # get self.nspin self.kpname self.nkp(number of special kpoints)

        # self.ndiv

        while True:

            tmp=fmain.readline()

            if len(tmp)==0:

                break

            if tmp.strip()=='spinpol':

                if fmain.readline().strip()=='.true.':

                    self.nspin=2

            if tmp.strip()=='plot1d':

                self.nkp,self.ndiv=fmain.readline().split()[0:2]

                self.nkp=int(self.nkp)

                self.ndiv=int(self.ndiv)

                for i in range(self.nkp):

                    self.kpname.append('$'+str(fmain.readline().split()[3])+'$')

        # get self.kpos

        while True:

            tmp=fpos.readline()

            if len(tmp)==0:

                break

            self.kpos.append(float(tmp.split()[0]))

            fpos.readline()

            fpos.readline()

        # get self.fermi

        self.fermi=27.2114*float(ffermi.readline().strip())

        # get self.nband self.band

        while True:

            tmp=fband.readline()

            if len(tmp)==0:

                break

            if len(tmp.strip())==0 and len(tmp)!=0:

                self.nband=self.nband+1

        rawband=np.loadtxt('BAND.OUT')

        kp=rawband[0:self.ndiv,0]

        self.band.append(kp)

        for i in range(self.nband):

            eig=27.2114*rawband[i*self.ndiv:(i+1)*self.ndiv,1]

            self.band.append(eig)

        #close files

        fmain.close()

        fpos.close()

        fband.close()

        ffermi.close()

    def write_data(self):

        Band.write_data(self)

    def view_band(self,emin=-10.0,emax=15.0,title='Band Plot',\

                 xlabel='K-Path',ylabel='Energy / eV'):

        Band.view_band(self,emin,emax,title,xlabel,ylabel)

[ Last edited by 銳利的碎片 on 2011-2-26 at 12:17 ]

贊一下

回復此樓

2樓2011-02-26 09:15:54

已閱回復此樓關注TA 給TA發(fā)消息送TA紅花 TA的回帖

guohuazhong

至尊木蟲 (職業(yè)作家)

支持原創(chuàng)，贊一個，頂��！

回復此樓

4樓2011-02-26 13:26:03

已閱回復此樓關注TA 給TA發(fā)消息送TA紅花 TA的回帖

zzy870720z

榮譽版主 (文壇精英)

★★★★★ 五星級,優(yōu)秀推薦

★★★★★ 五星級

贊一下

回復此樓

6樓2011-02-26 16:30:45

已閱回復此樓關注TA 給TA發(fā)消息送TA紅花 TA的回帖

簡單回復

xh5128樓

2012-01-06 07:17 回復

五星好評頂一下，感謝分享！

tanloer9樓

2012-02-13 15:28 回復

五星好評頂一下，感謝分享！

mink10樓

2014-06-18 07:40 回復

五星好評頂一下，感謝分享！

102424790011樓

2014-07-03 21:32 回復

五星好評頂一下，感謝分享！

zwynu12樓

2014-09-24 21:57 回復

五星好評頂一下，感謝分享！

Toapollo13樓

2015-03-28 20:58 回復

三星好評頂一下，感謝分享！

最具人氣熱帖推薦 [查看全部]	作者	回/看	最后發(fā)表

[考研] 一志愿北京科技材料科學與工程288分，求調(diào)劑 +8	是辰啊 2026-04-02	8/400	2026-04-02 11:33 by Sammy2
[考研] 一志愿北京科技，085601總分305求調(diào)劑 +9	半生瓜！ 2026-04-01	11/550	2026-04-02 08:28 by Wang200018
[考研] 302求調(diào)劑一志愿北航070300，本科鄭大化學 +8	圣日耳曼條 2026-04-01	11/550	2026-04-02 07:40 by chemdavid
[考研] 085410 一志愿211 22408分數(shù)359求調(diào)劑 +3	123456789qw 2026-03-31	4/200	2026-04-02 00:06 by 義文wang
[考研] 291求調(diào)劑 +20	Y-cap 2026-03-29	25/1250	2026-04-01 23:49 by 欣喜777
[考研] 292求調(diào)劑 +17	木蟲er12138 2026-04-01	17/850	2026-04-01 21:37 by 七度不信任
[考研] 310分求調(diào)劑 +4	成功上岸wang 2026-04-01	4/200	2026-04-01 20:35 by liu823948201
[考研] 265求調(diào)劑 +11	yelck 2026-04-01	12/600	2026-04-01 19:12 by 549790059
[考研] 材料調(diào)劑 +11	一樣YWY 2026-03-31	11/550	2026-04-01 11:35 by wangjy2002
[考研] 085410人工智能初試316分求調(diào)劑 +3	殘星拂曙 2026-03-31	3/150	2026-04-01 11:09 by 小熊raider
[考研] 求0861交通運輸專碩or材料專碩調(diào)劑 +4	勒布朗@ 2026-03-31	4/200	2026-04-01 09:54 by 一只好果子?
[考研] 土木304求調(diào)劑 +5	頂級擦擦 2026-03-31	5/250	2026-04-01 08:15 by fdcxdystjk￥
[考研] 262求調(diào)劑 +7	ZZ..000 2026-03-30	8/400	2026-03-31 10:05 by cal0306
[考研] 22408 359分調(diào)劑 +4	Qshers 2026-03-27	8/400	2026-03-31 08:53 by Qshers
[考研] 哈爾濱工業(yè)大學材料與化工專碩378求調(diào)劑 +3	塔比烏斯 2026-03-30	3/150	2026-03-30 22:55 by 無際的草原
[考研] 317分一志愿南理工材料工程本科湖工大求調(diào)劑 +12	芋泥小鈴鐺 2026-03-28	12/600	2026-03-30 17:06 by wangjy2002
[考研] 332求調(diào)劑 +6	@MZB382400 2026-03-28	6/300	2026-03-30 16:57 by 無際的草原
[考研] 本科雙非材料，跨考一志愿華電085801電氣，283求調(diào)劑，任何專業(yè)都可以 +6	芝士雪baoo 2026-03-28	8/400	2026-03-29 08:16 by 松花缸1201
[考研] 352分化工與材料 +5	海納百川Ly 2026-03-27	5/250	2026-03-28 03:39 by fmesaito
[考研] 266求調(diào)劑 +11	陽陽哇塞 2026-03-27	12/600	2026-03-27 17:56 by yu221