1# flake8: noqa 

2 

3 

4 

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

6# DOSCAR file based on the number of columns found 

7PDOS_orbital_names_and_DOSCAR_column = { 

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

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

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

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

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

13 9 : { 

14 's+': 1, 

15 's-up': 1, 

16 's-': 2, 

17 's-down': 2, 

18 'p+': 3, 

19 'p-up': 3, 

20 'p-': 4, 

21 'p-down': 4, 

22 'd+': 5, 

23 'd-up': 5, 

24 'd-': 6, 

25 'd-down': 6, 

26 'f+': 7, 

27 'f-up': 7, 

28 'f-': 8, 

29 'f-down': 8}, 

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

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

32 'dx2': 9}, 

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

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

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

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

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

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

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

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

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

42 17 : { 

43 's': 1, 

44 'py': 2, 

45 'pz': 3, 

46 'px': 4, 

47 'dxy': 5, 

48 'dyz': 6, 

49 'dz2': 7, 

50 'dxz': 8, 

51 'dx2': 9, 

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

53 'fxyz': 11, 

54 'fyz2': 12, 

55 'fz3': 13, 

56 'fxz2': 14, 

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

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

59 19 : { 

60 's+': 1, 

61 's-up': 1, 

62 's-': 2, 

63 's-down': 2, 

64 'py+': 3, 

65 'py-up': 3, 

66 'py-': 4, 

67 'py-down': 4, 

68 'pz+': 5, 

69 'pz-up': 5, 

70 'pz-': 6, 

71 'pz-down': 6, 

72 'px+': 7, 

73 'px-up': 7, 

74 'px-': 8, 

75 'px-down': 8, 

76 'dxy+': 9, 

77 'dxy-up': 9, 

78 'dxy-': 10, 

79 'dxy-down': 10, 

80 'dyz+': 11, 

81 'dyz-up': 11, 

82 'dyz-': 12, 

83 'dyz-down': 12, 

84 'dz2+': 13, 

85 'dz2-up': 13, 

86 'dz2-': 14, 

87 'dz2-down': 14, 

88 'dxz+': 15, 

89 'dxz-up': 15, 

90 'dxz-': 16, 

91 'dxz-down': 16, 

92 'dx2+': 17, 

93 'dx2-up': 17, 

94 'dx2-': 18, 

95 'dx2-down': 18}, 

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

97 # vasp reports totals for each orbital first 

98 37 : { 

99 's' : 1, 

100 's-x' : 2, 

101 's-y' : 3, 

102 's-z' : 4, 

103 'py' : 5, 

104 'py-x' : 6, 

105 'py-y' : 7, 

106 'py-z' : 8, 

107 'pz' : 9, 

108 'pz-x' : 10, 

109 'pz-y' : 11, 

110 'pz-z' : 12, 

111 'px' : 13, 

112 'px-x' : 14, 

113 'px-y' : 15, 

114 'px-z' : 16, 

115 'dxy' : 17, 

116 'dxy-x': 18, 

117 'dxy-y': 19, 

118 'dxy-z': 20, 

119 'dyz' : 21, 

120 'dyz-x': 22, 

121 'dyz-y': 23, 

122 'dyz-z': 24, 

123 'dz2' : 25, 

124 'dz2-x': 26, 

125 'dz2-y': 27, 

126 'dz2-z': 28, 

127 'dxz' : 29, 

128 'dxz-x': 30, 

129 'dxz-y': 31, 

130 'dxz-z': 32, 

131 'dx2' : 33, 

132 'dx2-x': 34, 

133 'dx2-y': 35, 

134 'dx2-z': 36}, 

135 33 : { 

136 's+': 1, 

137 's-up': 1, 

138 's-': 2, 

139 's-down': 2, 

140 'py+': 3, 

141 'py-up': 3, 

142 'py-': 4, 

143 'py-down': 4, 

144 'pz+': 5, 

145 'pz-up': 5, 

146 'pz-': 6, 

147 'pz-down': 6, 

148 'px+': 7, 

149 'px-up': 7, 

150 'px-': 8, 

151 'px-down': 8, 

152 'dxy+': 9, 

153 'dxy-up': 9, 

154 'dxy-': 10, 

155 'dxy-down': 10, 

156 'dyz+': 11, 

157 'dyz-up': 11, 

158 'dyz-': 12, 

159 'dyz-down': 12, 

160 'dz2+': 13, 

161 'dz2-up': 13, 

162 'dz2-': 14, 

163 'dz2-down': 14, 

164 'dxz+': 15, 

165 'dxz-up': 15, 

166 'dxz-': 16, 

167 'dxz-down': 16, 

168 'dx2+': 17, 

169 'dx2-up': 17, 

170 'dx2-': 18, 

171 'dx2-down': 18, 

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

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

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

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

176 'fxyz+': 21, 

177 'fxyz-up': 21, 

178 'fxyz-': 22, 

179 'fxyz-down': 22, 

180 'fyz2+': 23, 

181 'fyz2-up': 23, 

182 'fyz2-': 24, 

183 'fyz2-down': 24, 

184 'fz3+': 25, 

185 'fz3-up': 25, 

186 'fz3-': 26, 

187 'fz3-down': 26, 

188 'fxz2+': 27, 

189 'fxz2-up': 27, 

190 'fxz2-': 28, 

191 'fxz2-down': 28, 

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

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

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

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

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

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

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

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

200 } 

201