forked from itamblyn/scripts
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pdos_us.py
executable file
·126 lines (92 loc) · 3.82 KB
/
pdos_us.py
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
#!/usr/bin/env python
import sys
import commands
import numpy
try:
input_filename = sys.argv[1]
except IndexError:
print '\nusage: ' + sys.argv[0] + ' input_filename'
print '\nexiting...\n'
sys.exit(0)
core_valence_divide = -1000 # <- all valence
command_line_counter = commands.getoutput('wc -l ' + input_filename).split()
if len(command_line_counter) != 2:
print 'Error determining file size'
else:
number_of_lines = int(command_line_counter[0])
outputFile_dos = open('dos.dat', 'w')
outputFile_pdos = open('pdos.dat', 'w')
inputFile = open(input_filename, 'r')
natom = int(inputFile.readline().split()[0])
natom = 16
for i in range(4):
inputFile.readline()
inline_total = inputFile.readline()
emin_total = float(inline_total.split()[1])
emax_total = float(inline_total.split()[0])
enum_total = int(inline_total.split()[2])
efermi_total = float(inline_total.split()[3])
dos = numpy.zeros((enum_total, 3), dtype = numpy.float) # E, dos, idos
for i in range(enum_total):
inline_total = inputFile.readline().split()
# dos[i][0] = float(inline_total[0])
dos[i][0] = float(inline_total[0]) - efermi_total
dos[i][1], dos[i][2] = float(inline_total[1]), float(inline_total[2])
for row in dos:
for element in row:
outputFile_dos.write(str(element) + ' ')
outputFile_dos.write('\n')
############
enum_project = enum_total
spacing = (emax_total - emin_total)/enum_total
efermi_bin = int((efermi_total - emin_total)/spacing)
pdos = numpy.zeros((enum_project, 1 + 3*2), dtype = numpy.float) # E, s, p, d
for i in range(len(pdos)):
pdos[i][0] = dos[i][0] # fills in the E column of pdos
for atoms in range(natom):
inline_project = inputFile.readline()
for i in range(enum_project):
inline_project = inputFile.readline().split()
pdos[i][1] += float(inline_project[1]) # s
pdos[i][2] += float(inline_project[2]) + float(inline_project[3]) + float(inline_project[4]) # py pz px
pdos[i][3] += float(inline_project[5]) + float(inline_project[6]) + float(inline_project[7]) + float(inline_project[8]) + float(inline_project[9]) # dxy ...
for i in numpy.arange(1,enum_project):
pdos[i][4] = pdos[i - 1][4] + pdos[i][1]
pdos[i][5] = pdos[i - 1][5] + pdos[i][2]
pdos[i][6] = pdos[i - 1][6] + pdos[i][3]
energies = numpy.transpose(pdos)[0]
s_project = numpy.transpose(pdos)[1]
p_project = numpy.transpose(pdos)[2]
d_project = numpy.transpose(pdos)[3]
#energies_core = energies[0:core_valence_divide]
#s_project_core = s_project[0:core_valence_divide]
#p_project_core = p_project[0:core_valence_divide]
#d_project_core = d_project[0:core_valence_divide]
energies_valence = energies[core_valence_divide:efermi_bin]
s_project_valence = s_project[core_valence_divide:efermi_bin]
p_project_valence = p_project[core_valence_divide:efermi_bin]
d_project_valence = d_project[core_valence_divide:efermi_bin]
#spd_sum_core = sum(s_project_core) + sum(p_project_core) + sum(d_project_core)
spd_sum_valence = sum(s_project_valence) + sum(p_project_valence) + sum(d_project_valence)
#s_fraction_core = sum(s_project_core)/spd_sum_core
#p_fraction_core = sum(p_project_core)/spd_sum_core
s_fraction_valence = sum(s_project_valence)/spd_sum_valence
p_fraction_valence = sum(p_project_valence)/spd_sum_valence
d_fraction_valence = sum(d_project_valence)/spd_sum_valence
#print
#print "%s core", s_fraction_core*100
#print "%p core", p_fraction_core*100
print
print "%s valence", s_fraction_valence*100
print "%p valence", p_fraction_valence*100
print "%d valence", d_fraction_valence*100
print
print "norm = ", str(s_fraction_valence + p_fraction_valence + d_fraction_valence)
print
print "efermi_total = ", efermi_total
print "spacing = ", spacing
print
for row in pdos:
for element in row:
outputFile_pdos.write(str(element) + ' ')
outputFile_pdos.write('\n')