Coverage for /builds/kinetik161/ase/ase/calculators/vasp/vasp_data.py: 100.00%

1# flake8: noqa 

2 

3 

4# this returns a look up dict for the orbital name -> col number in the 

5# DOSCAR file based on the number of columns found 

6PDOS_orbital_names_and_DOSCAR_column = { 

7 4: {'s': 1, 'p': 2, 'd': 3}, 

8 5: {'s': 1, 'p': 2, 'd': 3, 'f': 4}, 

9 7: {'s+': 1, 's-up': 1, 's-': 2, 's-down': 2, 

10 'p+': 3, 'p-up': 3, 'p-': 4, 'p-down': 4, 

11 'd+': 5, 'd-up': 5, 'd-': 6, 'd-down': 6}, 

12 9: { 

13 's+': 1, 

14 's-up': 1, 

15 's-': 2, 

16 's-down': 2, 

17 'p+': 3, 

18 'p-up': 3, 

19 'p-': 4, 

20 'p-down': 4, 

21 'd+': 5, 

22 'd-up': 5, 

23 'd-': 6, 

24 'd-down': 6, 

25 'f+': 7, 

26 'f-up': 7, 

27 'f-': 8, 

28 'f-down': 8}, 

29 10: {'s': 1, 'py': 2, 'pz': 3, 'px': 4, 

30 'dxy': 5, 'dyz': 6, 'dz2': 7, 'dxz': 8, 

31 'dx2': 9}, 

32 19: {'s+': 1, 's-up': 1, 's-': 2, 's-down': 2, 

33 'py+': 3, 'py-up': 3, 'py-': 4, 'py-down': 4, 

34 'pz+': 5, 'pz-up': 5, 'pz-': 6, 'pz-down': 6, 

35 'px+': 7, 'px-up': 7, 'px-': 8, 'px-down': 8, 

36 'dxy+': 9, 'dxy-up': 9, 'dxy-': 10, 'dxy-down': 10, 

37 'dyz+': 11, 'dyz-up': 11, 'dyz-': 12, 'dyz-down': 12, 

38 'dz2+': 13, 'dz2-up': 13, 'dz2-': 14, 'dz2-down': 14, 

39 'dxz+': 15, 'dxz-up': 15, 'dxz-': 16, 'dxz-down': 16, 

40 'dx2+': 17, 'dx2-up': 17, 'dx2-': 18, 'dx2-down': 18}, 

41 17: { 

42 's': 1, 

43 'py': 2, 

44 'pz': 3, 

45 'px': 4, 

46 'dxy': 5, 

47 'dyz': 6, 

48 'dz2': 7, 

49 'dxz': 8, 

50 'dx2': 9, 

51 'fy(3x2-y2)': 10, 

52 'fxyz': 11, 

53 'fyz2': 12, 

54 'fz3': 13, 

55 'fxz2': 14, 

56 'fz(x2-y2)': 15, 

57 'fx(x2-3y2)': 16}, 

58 19: { 

59 's+': 1, 

60 's-up': 1, 

61 's-': 2, 

62 's-down': 2, 

63 'py+': 3, 

64 'py-up': 3, 

65 'py-': 4, 

66 'py-down': 4, 

67 'pz+': 5, 

68 'pz-up': 5, 

69 'pz-': 6, 

70 'pz-down': 6, 

71 'px+': 7, 

72 'px-up': 7, 

73 'px-': 8, 

74 'px-down': 8, 

75 'dxy+': 9, 

76 'dxy-up': 9, 

77 'dxy-': 10, 

78 'dxy-down': 10, 

79 'dyz+': 11, 

80 'dyz-up': 11, 

81 'dyz-': 12, 

82 'dyz-down': 12, 

83 'dz2+': 13, 

84 'dz2-up': 13, 

85 'dz2-': 14, 

86 'dz2-down': 14, 

87 'dxz+': 15, 

88 'dxz-up': 15, 

89 'dxz-': 16, 

90 'dxz-down': 16, 

91 'dx2+': 17, 

92 'dx2-up': 17, 

93 'dx2-': 18, 

94 'dx2-down': 18}, 

95 # this is Non-collinear. -x, -y, -z are magnetic moment 

96 # vasp reports totals for each orbital first 

97 37: { 

98 's': 1, 

99 's-x': 2, 

100 's-y': 3, 

101 's-z': 4, 

102 'py': 5, 

103 'py-x': 6, 

104 'py-y': 7, 

105 'py-z': 8, 

106 'pz': 9, 

107 'pz-x': 10, 

108 'pz-y': 11, 

109 'pz-z': 12, 

110 'px': 13, 

111 'px-x': 14, 

112 'px-y': 15, 

113 'px-z': 16, 

114 'dxy': 17, 

115 'dxy-x': 18, 

116 'dxy-y': 19, 

117 'dxy-z': 20, 

118 'dyz': 21, 

119 'dyz-x': 22, 

120 'dyz-y': 23, 

121 'dyz-z': 24, 

122 'dz2': 25, 

123 'dz2-x': 26, 

124 'dz2-y': 27, 

125 'dz2-z': 28, 

126 'dxz': 29, 

127 'dxz-x': 30, 

128 'dxz-y': 31, 

129 'dxz-z': 32, 

130 'dx2': 33, 

131 'dx2-x': 34, 

132 'dx2-y': 35, 

133 'dx2-z': 36}, 

134 33: { 

135 's+': 1, 

136 's-up': 1, 

137 's-': 2, 

138 's-down': 2, 

139 'py+': 3, 

140 'py-up': 3, 

141 'py-': 4, 

142 'py-down': 4, 

143 'pz+': 5, 

144 'pz-up': 5, 

145 'pz-': 6, 

146 'pz-down': 6, 

147 'px+': 7, 

148 'px-up': 7, 

149 'px-': 8, 

150 'px-down': 8, 

151 'dxy+': 9, 

152 'dxy-up': 9, 

153 'dxy-': 10, 

154 'dxy-down': 10, 

155 'dyz+': 11, 

156 'dyz-up': 11, 

157 'dyz-': 12, 

158 'dyz-down': 12, 

159 'dz2+': 13, 

160 'dz2-up': 13, 

161 'dz2-': 14, 

162 'dz2-down': 14, 

163 'dxz+': 15, 

164 'dxz-up': 15, 

165 'dxz-': 16, 

166 'dxz-down': 16, 

167 'dx2+': 17, 

168 'dx2-up': 17, 

169 'dx2-': 18, 

170 'dx2-down': 18, 

171 'fy(3x2-y2)+': 19, 

172 'fy(3x2-y2)-up': 19, 

173 'fy(3x2-y2)-': 20, 

174 'fy(3x2-y2)-down': 20, 

175 'fxyz+': 21, 

176 'fxyz-up': 21, 

177 'fxyz-': 22, 

178 'fxyz-down': 22, 

179 'fyz2+': 23, 

180 'fyz2-up': 23, 

181 'fyz2-': 24, 

182 'fyz2-down': 24, 

183 'fz3+': 25, 

184 'fz3-up': 25, 

185 'fz3-': 26, 

186 'fz3-down': 26, 

187 'fxz2+': 27, 

188 'fxz2-up': 27, 

189 'fxz2-': 28, 

190 'fxz2-down': 28, 

191 'fz(x2-y2)+': 29, 

192 'fz(x2-y2)-up': 29, 

193 'fz(x2-y2)-': 30, 

194 'fz(x2-y2)-down': 30, 

195 'fx(x2-3y2)+': 31, 

196 'fx(x2-3y2)-up': 31, 

197 'fx(x2-3y2)-': 32, 

198 'fx(x2-3y2)-down': 32}, 

199}