1import copy 

2 

3_setups_defaults = { 

4 'minimal': { 

5 'K': '_pv', 

6 'Ca': '_pv', 

7 'Rb': '_pv', 

8 'Sr': '_sv', 

9 'Y': '_sv', 

10 'Zr': '_sv', 

11 'Nb': '_pv', 

12 'Cs': '_sv', 

13 'Ba': '_sv', 

14 'Fr': '_sv', 

15 'Ra': '_sv', 

16 'Sc': '_sv' 

17 }, 

18 

19 # VASP recommended 

20 'recommended': { 

21 'Li': '_sv', 

22 'Na': '_pv', 

23 'K': '_sv', 

24 'Ca': '_sv', 

25 'Sc': '_sv', 

26 'Ti': '_sv', 

27 'V': '_sv', 

28 'Cr': '_pv', 

29 'Mn': '_pv', 

30 'Ga': '_d', 

31 'Ge': '_d', 

32 'Rb': '_sv', 

33 'Sr': '_sv', 

34 'Y': '_sv', 

35 'Zr': '_sv', 

36 'Nb': '_sv', 

37 'Mo': '_sv', 

38 'Tc': '_pv', 

39 'Ru': '_pv', 

40 'Rh': '_pv', 

41 'In': '_d', 

42 'Sn': '_d', 

43 'Cs': '_sv', 

44 'Ba': '_sv', 

45 'Pr': '_3', 

46 'Nd': '_3', 

47 'Pm': '_3', 

48 'Sm': '_3', 

49 'Eu': '_2', 

50 'Gd': '_3', 

51 'Tb': '_3', 

52 'Dy': '_3', 

53 'Ho': '_3', 

54 'Er': '_3', 

55 'Tm': '_3', 

56 'Yb': '_2', 

57 'Lu': '_3', 

58 'Hf': '_pv', 

59 'Ta': '_pv', 

60 'W': '_pv', 

61 'Tl': '_d', 

62 'Pb': '_d', 

63 'Bi': '_d', 

64 'Po': '_d', 

65 'At': '_d', 

66 'Fr': '_sv', 

67 'Ra': '_sv' 

68 }, 

69 

70 # https://wiki.materialsproject.org/Pseudopotentials_Choice 

71 'materialsproject': 

72 # Alkali and alkali-earth 

73 { 

74 'Li': '_sv', 

75 'Na': '_pv', 

76 'K': '_sv', 

77 'Cs': '_sv', 

78 'Rb': '_sv', 

79 'Be': '_sv', 

80 'Mg': '_pv', 

81 'Ca': '_sv', 

82 'Sr': '_sv', 

83 'Ba': '_sv', 

84 # d-elements, transition metals 

85 'Sc': '_sv', 

86 'Y': '_sv', 

87 'Ti': '_pv', 

88 'Zr': '_sv', 

89 'Hf': '_pv', 

90 'V': '_sv', 

91 'Nb': '_pv', 

92 'Ta': '_pv', 

93 'Cr': '_pv', 

94 'Mo': '_pv', 

95 'W': '_sv', # updated from W_pv 

96 'Mn': '_pv', 

97 'Tc': '_pv', 

98 'Re': '_pv', 

99 'Fe': '_pv', 

100 'Co': '', 

101 'Ni': '_pv', 

102 'Cu': '_pv', 

103 'Zn': '', 

104 'Ru': '_pv', 

105 'Rh': '_pv', 

106 'Pd': '', 

107 'Ag': '', 

108 'Cd': '', 

109 'Hg': '', 

110 'Ir': '', 

111 'Pt': '', 

112 'Os': '_pv', 

113 # Main group 

114 'Ga': '_d', 

115 'Ge': '_d', 

116 'Al': '', 

117 'As': '', 

118 'Se': '', 

119 'Br': '', 

120 'In': '_d', 

121 'Sn': '_d', 

122 'Tl': '_d', 

123 'Pb': '_d', 

124 'Bi': '_d', # ? 

125 'Po': '', 

126 'At': '_d', 

127 # Rare-earth, f-electrons 

128 'La': '', 

129 'Ce': '', 

130 'Pr': '_3', 

131 'Nd': '_3', 

132 'Pm': '_3', 

133 'Sm': '_3', 

134 'Eu': '', 

135 'Gd': '', 

136 'Tb': '_3', 

137 'Dy': '_3', 

138 'Ho': '_3', 

139 'Er': '_3', 

140 'Tm': '_3', 

141 'Yb': '', 

142 'Lu': '_3' 

143 }, 

144 

145 # 

146 'gw': { 

147 'H': '_GW', 

148 'He': '_GW', 

149 'Li': '_sv_GW', 

150 'Be': '_sv_GW', 

151 'B': '_GW', 

152 'C': '_GW', 

153 'N': '_GW', 

154 'O': '_GW', 

155 'F': '_GW', 

156 'Ne': '_GW', 

157 'Na': '_sv_GW', 

158 'Mg': '_sv_GW', 

159 'Al': '_GW', 

160 'Si': '_GW', 

161 'P': '_GW', 

162 'S': '_GW', 

163 'Cl': '_GW', 

164 'Ar': '_GW', 

165 'K': '_sv_GW', 

166 'Ca': '_sv_GW', 

167 'Sc': '_sv_GW', 

168 'Ti': '_sv_GW', 

169 'V': '_sv_GW', 

170 'Cr': '_sv_GW', 

171 'Mn': '_sv_GW', 

172 'Fe': '_sv_GW', 

173 'Co': '_sv_GW', 

174 'Ni': '_sv_GW', 

175 'Cu': '_sv_GW', 

176 'Zn': '_sv_GW', 

177 'Ga': '_d_GW', 

178 'Ge': '_d_GW', 

179 'As': '_GW', 

180 'Se': '_GW', 

181 'Br': '_GW', 

182 'Kr': '_GW', 

183 'Rb': '_sv_GW', 

184 'Sr': '_sv_GW', 

185 'Y': '_sv_GW', 

186 'Zr': '_sv_GW', 

187 'Nb': '_sv_GW', 

188 'Mo': '_sv_GW', 

189 'Tc': '_sv_GW', 

190 'Ru': '_sv_GW', 

191 'Rh': '_sv_GW', 

192 'Pd': '_sv_GW', 

193 'Ag': '_sv_GW', 

194 'Cd': '_sv_GW', 

195 'In': '_d_GW', 

196 'Sn': '_d_GW', 

197 'Sb': '_d_GW', 

198 'Te': '_GW', 

199 'I': '_GW', 

200 'Xe': '_GW', 

201 'Cs': '_sv_GW', 

202 'Ba': '_sv_GW', 

203 'La': '_GW', 

204 'Ce': '_GW', 

205 'Hf': '_sv_GW', 

206 'Ta': '_sv_GW', 

207 'W': '_sv_GW', 

208 'Re': '_sv_GW', 

209 'Os': '_sv_GW', 

210 'Ir': '_sv_GW', 

211 'Pt': '_sv_GW', 

212 'Au': '_sv_GW', 

213 'Hg': '_sv_GW', 

214 'Tl': '_d_GW', 

215 'Pb': '_d_GW', 

216 'Bi': '_d_GW', 

217 'Po': '_d_GW', 

218 'At': '_d_GW', 

219 'Rn': '_d_GW' 

220 } 

221} 

222 

223 

224def get_default_setups(): 

225 # Avoid mutating the module dictionary, so we use a copy instead 

226 # Note, it is a nested dict, so a regular copy is not enough 

227 return copy.deepcopy(_setups_defaults) 

228 

229 

230# Compatibility 

231setups_defaults = get_default_setups()