def SCF (r = 1.4632, Z=[1,1], b1 = hfscf.GTO["H"], b2 = hfscf.GTO["H"], b = 3):
R = [0, r] # Задаем центры атомов
s_scf = hfscf.S(R, b1, b2, b) # Генерация матрицы перекрывания S
h_scf = hfscf.H(R, Z, b1, b2, b) # Генерация гамильтониана H для оператора Фока (core-Hamiltonian operator)
X = hfscf.diagon(m=s_scf) # Диагонализированная матрицы S
Xa = X.getH() # транспонированная матрица
p_scf = np.matrix([[0,0],[0,0]], dtype=np.float64) # матрица плотности P
for iteracion in range(50): # проверка сходимости SCF итерациями (заполняем матрицу F, диагонализируем, проверяем разницу)
# Расчет двух электронной части матрицы Фока (G)
# Генерация матрицы Фока: F = H + G
g_scf = hfscf.G(r, p_scf, b1, b2, b)
f_scf = h_scf + g_scf
# Создание матрицы F' = X_adj * F * X
f_tra = Xa * f_scf * X
# Диагонализация F' и генерация C'
c_tra = hfscf.diagon2(m=f_tra)
# Построение матрицы коэффициентов C C = X * C'
c_scf = X * c_tra
# Из матрицы C строим матрицу Р и пересчитываем матрицы плотности P
p_temp = hfscf.P(C=c_scf)
print("Окончена " + str(iteracion + 1) + " итерация.\n")
# Проверка сходимости
if np.linalg.norm(p_temp - p_scf) < 1E-4:
print("SCF сошлось!")
return {"S":s_scf,"H":h_scf,"X": X,"F":f_scf,"C":c_scf,"P":p_temp}
else:
p_scf = p_temp
print("SCF не сошлось.")
return {"S":s_scf,"H":h_scf,"X": X,"F":f_scf,"C":c_scf,"P":p_temp}