File: ipr.py

package info (click to toggle)
p4vasp 0.3.29%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 6,260 kB
  • ctags: 6,736
  • sloc: python: 34,304; cpp: 13,882; ansic: 2,076; makefile: 379; sql: 207; xml: 102; sh: 11
file content (50 lines) | stat: -rw-r--r-- 1,711 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
#!/usr/bin/python
#  p4vasp is a GUI-program and a library for processing outputs of the
#  Vienna Ab-inition Simulation Package (VASP)
#  (see http://cms.mpi.univie.ac.at/vasp/Welcome.html)
#
#  Copyright (C) 2003  Orest Dubay <dubay@danubiananotech.com>
#
#  This program is free software; you can redistribute it and/or modify
#  it under the terms of the GNU General Public License as published by
#  the Free Software Foundation; either version 2 of the License, or
#  (at your option) any later version.
#
#  This program is distributed in the hope that it will be useful,
#  but WITHOUT ANY WARRANTY; without even the implied warranty of
#  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#  GNU General Public License for more details.
#
#   You should have received a copy of the GNU General Public License
#  along with this program; if not, write to the Free Software
#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

import sys
from p4vasp.SystemPM import *

path="vasprun.xml"
out="ipr.dat"
if len(sys.argv)>1:
    path=sys.argv[1]
    if len(sys.argv)>2:
        out=sys.argv[2]

s=XMLSystemPM(path)
occ=s.PROJECTED_EIGENVALUES_L
energies=s.PROJECTED_EIGENVALUES_ENERGIES_L
fermi=s.E_FERMI

nions=len(occ[0][0][0])
f=open(out,"w")
for spin in range(len(occ)):
    for k in range(len(occ[spin])):
        for band in range(len(occ[spin][k])):
            sum1=0.0
            sum2=0.0
            for ion in range(len(occ[spin][k][band])):
                x=sum(occ[spin][k][band][ion])
                sum1+=x
                sum2+=x*x
            res=sum1*sum1/sum2/nions
            f.write("%f %f\n"%(energies[spin][k][band][0]-fermi,1/res))
f.close()