def RKS(mol, xc='LDA,VWN'):
if mol.spin == 0:
if not mol.symmetry or mol.groupname == 'C1':
return rks.RKS(mol, xc)
else:
return rks_symm.RKS(mol, xc)
else:
return ROKS(mol, xc)
def RKS(mol, xc='LDA,VWN'):
if mol.nelectron == 1:
return uks.UKS(mol)
elif not mol.symmetry or mol.groupname == 'C1':
if mol.spin > 0:
return roks.ROKS(mol, xc)
else:
return rks.RKS(mol, xc)
else:
if mol.spin > 0:
return rks_symm.ROKS(mol, xc)
else:
return rks_symm.RKS(mol, xc)
def RKS(mol, *args):
if mol.nelectron == 1:
return uks.UKS(mol)
elif not mol.symmetry or mol.groupname is 'C1':
if mol.spin > 0:
return roks.ROKS(mol, *args)
else:
return rks.RKS(mol, *args)
else:
if mol.spin > 0:
return rks_symm.ROKS(mol, *args)
else:
return rks_symm.RKS(mol, *args)
def test_ks(pseudo=None):
# The molecular calculation
mol = gto.Mole()
mol.unit = 'B'
L = 10
mol.atom.extend([
['He', (L / 2., L / 2., L / 2.)],
])
# these are some exponents which are not hard to integrate
mol.basis = {'He': [[0, (0.8, 1.0)], [0, (1.0, 1.0)], [0, (1.2, 1.0)]]}
mol.build()
m = rks.RKS(mol)
m.xc = 'LDA,VWN_RPA'
#m.xc = 'B88,LYP'
print "Molecular DFT energy"
print(m.scf())
# LDA,VWN_RPA
# -2.64096172441
# BLYP
# -2.66058401340308
# The periodic calculation
cell = pbcgto.Cell()
cell.unit = 'B'
cell.h = np.diag([L, L, L])
cell.gs = np.array([80, 80, 80])
cell.nimgs = [1, 1, 1]
cell.atom = mol.atom
cell.basis = mol.basis
cell.pseudo = pseudo
cell.build()
mf = pbcrks.RKS(cell)
mf.xc = 'LDA,VWN_RPA'
# mf.xc = 'B88,LYP'
print(mf.scf())
def test_components(pseudo=None):
# The molecular calculation
mol = gto.Mole()
mol.unit = 'B'
L = 60
mol.atom.extend([
['He', (L / 2., L / 2., L / 2.)],
])
mol.basis = 'sto-3g'
mol.build()
m = rks.RKS(mol)
m.xc = 'LDA,VWN_RPA'
print(m.scf()) # -2.90705411168
dm = m.make_rdm1()
# The periodic calculation
cell = pbcgto.Cell()
cell.unit = 'B'
cell.a = np.diag([L, L, L])
cell.gs = np.array([80, 80, 80])
cell.atom = mol.atom
cell.basis = mol.basis
cell.pseudo = pseudo
cell.build()
# These should match reasonably well (roughly with accuracy of normalization)
print "Kinetic energy"
tao = pbchf.get_t(cell)
tao2 = mol.intor_symmetric('cint1e_kin_sph')
print np.dot(np.ravel(tao), np.ravel(dm)) # 2.82793077196
print np.dot(np.ravel(tao2), np.ravel(dm)) # 2.82352636524
print "Overlap"
sao = pbchf.get_ovlp(cell)
print np.dot(np.ravel(sao), np.ravel(dm)) # 1.99981725342
print np.dot(np.ravel(m.get_ovlp()), np.ravel(dm)) # 2.0
# The next two entries should *not* match, since G=0 component is removed
print "Coulomb (G!=0)"
jao = pbchf.get_j(cell, dm)
print np.dot(np.ravel(dm), np.ravel(jao)) # 4.03425518427
print np.dot(np.ravel(dm), np.ravel(m.get_j(dm))) # 4.22285177049
# The next two entries should *not* match, since G=0 component is removed
print "Nuc-el (G!=0)"
if cell.pseudo:
vppao = pbchf.get_pp(cell)
print np.dot(np.ravel(dm), np.ravel(vppao))
else:
neao = pbchf.get_nuc(cell)
print np.dot(np.ravel(dm), np.ravel(neao)) # -6.50203360062
vne = mol.intor_symmetric('cint1e_nuc_sph')
print np.dot(np.ravel(dm), np.ravel(vne)) # -6.68702326551
print "Normalization"
coords = gen_uniform_grids(cell)
aoR = eval_ao(cell, coords)
rhoR = eval_rho(cell, aoR, dm)
print cell.vol / len(rhoR) * np.sum(rhoR) # 1.99981725342 (should be 2.0)
print "(Hartree + vne) * DM"
print np.dot(np.ravel(dm), np.ravel(m.get_j(dm))) + np.dot(
np.ravel(dm), np.ravel(vne))
if cell.pseudo:
print np.einsum("ij,ij", dm, vppao + jao + pbchf.get_jvloc_G0(cell))
else:
print np.einsum("ij,ij", dm, neao + jao)
ew_cut = (40, 40, 40)
ew_eta = 0.05
for ew_eta in [0.1, 0.5, 1.]:
ew = pbchf.ewald(cell, ew_eta, ew_cut)
print "Ewald (eta, energy)", ew_eta, ew # should be same for all eta
print "Ewald divergent terms summation", ew
# These two should now match if the box is reasonably big to
# remove images, and ngs is big.
print "Total coulomb (analytic) ",
print(.5 * np.dot(np.ravel(dm), np.ravel(m.get_j(dm))) +
np.dot(np.ravel(dm), np.ravel(vne))) # -4.57559738004
if not cell.pseudo:
print "Total coulomb (fft coul + ewald)",
print np.einsum("ij,ij", dm, neao + .5 * jao) + ew # -4.57948259115
# Exc
cell.ew_eta, cell.ew_cut = ew_eta, ew_cut
mf = pbcdft.RKS(cell)
mf.xc = 'LDA,VWN_RPA'
rks.get_veff(mf, cell, dm)
print "Exc", mf._exc # -1.05967570089
