mirror of
https://github.com/LCPQ/quantum_package
synced 2024-11-19 04:22:36 +01:00
264 lines
5.8 KiB
Python
Executable File
264 lines
5.8 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
"""
|
|
Create the pseudo potential for a given atom
|
|
|
|
Usage:
|
|
put_pseudo_in_ezfio.py <ezfio_path>
|
|
|
|
Help:
|
|
atom is the Abreviation of the atom
|
|
"""
|
|
|
|
|
|
import os
|
|
import sys
|
|
from docopt import docopt
|
|
|
|
from subprocess import Popen, PIPE
|
|
|
|
qpackage_root = os.environ['QPACKAGE_ROOT']
|
|
|
|
EZFIO = "{0}/EZFIO".format(qpackage_root)
|
|
sys.path = [EZFIO + "/Python"] + sys.path
|
|
|
|
from ezfio import ezfio
|
|
|
|
import re
|
|
p = re.compile(ur'\|(\d+)><\d+\|')
|
|
|
|
|
|
def get_pseudo_str(l_atom):
|
|
"""
|
|
Run EMSL_local for geting the str of the speudo potential
|
|
|
|
str_ele :
|
|
Element Symbol: Na
|
|
Number of replaced protons: 10
|
|
Number of projectors: 2
|
|
|
|
Pseudopotential data:
|
|
|
|
Local component:
|
|
Coeff. r^n Exp.
|
|
1.00000000 -1 5.35838717
|
|
5.35838717 1 3.67918975
|
|
-2.07764789 0 1.60507673
|
|
|
|
Non-local component:
|
|
Coeff. r^n Exp. Proj.
|
|
10.69640234 0 1.32389367 |0><0|
|
|
10.11238853 0 1.14052020 |1><1|
|
|
"""
|
|
|
|
EMSL_root = "{0}/EMSL_Basis/".format(qpackage_root)
|
|
EMSL_path = "{0}/EMSL_api.py".format(EMSL_root)
|
|
db_path = "{0}/db/Pseudo.db".format(EMSL_root)
|
|
|
|
l_cmd_atom = []
|
|
for a in l_atom:
|
|
l_cmd_atom += ["--atom", a]
|
|
|
|
l_cmd_head = [EMSL_path, "get_basis_data",
|
|
"--db_path", db_path,
|
|
"--basis", "BFD-Pseudo"]
|
|
|
|
process = Popen(l_cmd_head + l_cmd_atom, stdout=PIPE, stderr=PIPE)
|
|
|
|
stdout, _ = process.communicate()
|
|
return stdout.strip()
|
|
|
|
|
|
def get_v_n_dz_local(str_ele):
|
|
"""
|
|
From a str_ele of the pseudo (aka only one ele in the str)
|
|
get the list ussefull for the Local potential : v_k n_k and dz_k
|
|
"""
|
|
l_v_k = []
|
|
l_n_k = []
|
|
l_dz_k = []
|
|
|
|
for l in str_ele.splitlines():
|
|
try:
|
|
v, n, dz = l.split()
|
|
v = float(v)
|
|
n = int(n)
|
|
dz = float(dz)
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
l_v_k.append(v)
|
|
l_n_k.append(n)
|
|
l_dz_k.append(dz)
|
|
|
|
return l_v_k, l_n_k, l_dz_k
|
|
|
|
|
|
def get_v_n_dz_l_nonlocal(str_ele):
|
|
"""
|
|
From a str_ele of the pseudo (aka only one ele in the str)
|
|
get the list ussefull for the non Local potential
|
|
v_kl (v, l)
|
|
n_k (v, l)
|
|
dz_k (dz ,l)
|
|
"""
|
|
l_v_kl = []
|
|
l_n_kl = []
|
|
l_dz_kl = []
|
|
|
|
for l in str_ele.splitlines():
|
|
try:
|
|
v, n, dz, proj = l.split()
|
|
v = float(v)
|
|
n = int(n)
|
|
dz = float(dz)
|
|
l = int(p.match(proj).group(1))
|
|
|
|
except ValueError:
|
|
pass
|
|
else:
|
|
l_v_kl.append([v])
|
|
l_n_kl.append([n])
|
|
l_dz_kl.append([dz])
|
|
|
|
if not l_v_kl:
|
|
l_v_kl.append([0.])
|
|
l_n_kl.append([0])
|
|
l_dz_kl.append([0.])
|
|
|
|
return l_v_kl, l_n_kl, l_dz_kl
|
|
|
|
|
|
def get_zeff_alpha_beta(str_ele):
|
|
"""
|
|
Return the the zeff, alpha num elec and beta num elec
|
|
Assert ezfio_set_file alredy defined
|
|
"""
|
|
|
|
import re
|
|
|
|
# ___
|
|
# | ._ o _|_
|
|
# _|_ | | | |_
|
|
#
|
|
|
|
# ~#~#~#~#~#~#~ #
|
|
# s t r _ e l e #
|
|
# ~#~#~#~#~#~#~ #
|
|
|
|
m = re.search('Element Symbol: ([a-zA-Z]+)', str_ele)
|
|
name = m.group(1).capitalize()
|
|
|
|
m = re.search('Number of replaced protons: (\d+)', str_ele)
|
|
z_remove = int(m.group(1))
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~ #
|
|
# F r o m _ e z f i o #
|
|
# ~#~#~#~#~#~#~#~#~#~ #
|
|
|
|
alpha = ezfio.get_electrons_elec_alpha_num()
|
|
beta = ezfio.get_electrons_elec_beta_num()
|
|
|
|
# _
|
|
# |_) _. ._ _ _
|
|
# | (_| | _> (/_
|
|
#
|
|
|
|
from elts_num_ele import name_to_elec
|
|
z = name_to_elec[name]
|
|
|
|
z_eff = z - z_remove
|
|
|
|
alpha = alpha - (z_remove / 2)
|
|
beta = beta - (z_remove / 2)
|
|
|
|
# _
|
|
# |_) _ _|_ ._ ._
|
|
# | \ (/_ |_ |_| | | |
|
|
#
|
|
|
|
return [z_eff, alpha, beta]
|
|
|
|
if __name__ == "__main__":
|
|
arguments = docopt(__doc__)
|
|
# ___
|
|
# | ._ o _|_
|
|
# _|_ | | | |_
|
|
#
|
|
|
|
# ~#~#~#~#~ #
|
|
# E Z F I O #
|
|
# ~#~#~#~#~ #
|
|
|
|
ezfio_path = arguments["<ezfio_path>"]
|
|
ezfio_path = os.path.expanduser(ezfio_path)
|
|
ezfio_path = os.path.expandvars(ezfio_path)
|
|
ezfio_path = os.path.abspath(ezfio_path)
|
|
|
|
ezfio.set_file("{0}".format(ezfio_path))
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~ #
|
|
# P s e u d o _ d a t a #
|
|
# ~#~#~#~#~#~#~#~#~#~#~ #
|
|
|
|
l_ele = ezfio.get_nuclei_nucl_label()
|
|
str_ = get_pseudo_str(l_ele)
|
|
|
|
# _
|
|
# |_) _. ._ _ _
|
|
# | (_| | _> (/_
|
|
#
|
|
|
|
l_str_ele = [str_ele for str_ele in str_.split("Element Symbol: ")
|
|
if str_ele]
|
|
|
|
l_zeff = []
|
|
alpha_tot = 0
|
|
beta_tot = 0
|
|
|
|
for str_ele in l_str_ele:
|
|
|
|
# ~#~#~#~#~ #
|
|
# S p l i t #
|
|
# ~#~#~#~#~ #
|
|
|
|
l = str_ele.find("Local component:")
|
|
nl = str_ele.find("Non-local component")
|
|
|
|
# ~#~#~#~#~ #
|
|
# L o c a l #
|
|
# ~#~#~#~#~ #
|
|
|
|
l_v, l_n, l_dz = get_v_n_dz_local(str_ele[l:nl])
|
|
|
|
ezfio.pseudo_klocmax = len(l_v)
|
|
ezfio.pseudo_v_k = l_v
|
|
ezfio.pseudo_n_k = l_n
|
|
ezfio.pseudo_dz_k = l_dz
|
|
|
|
# ~#~#~#~#~#~#~#~#~ #
|
|
# N o n _ L o c a l #
|
|
# ~#~#~#~#~#~#~#~#~ #
|
|
|
|
l_v_kl, l_n_kl, l_dz_kl = get_v_n_dz_l_nonlocal(str_ele[nl:])
|
|
|
|
ezfio.pseudo_lmaxpo = len(l_v_kl)
|
|
ezfio.pseudo_kmax = len(l_v_kl[0])
|
|
ezfio.pseudo_v_kl = l_v_kl
|
|
ezfio.pseudo_n_kl = l_n_kl
|
|
ezfio.pseudo_dz_kl = l_dz_kl
|
|
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~ #
|
|
# Z _ e f f , a l p h a / b e t a _ e l e c #
|
|
# ~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~#~ #
|
|
|
|
zeff, alpha, beta = get_zeff_alpha_beta(str_)
|
|
|
|
alpha_tot += alpha
|
|
beta_tot += beta
|
|
l_zeff.append(zeff)
|
|
|
|
ezfio.electrons_elec_alpha_num = alpha_tot
|
|
ezfio.electrons_elec_beta_num = beta_tot
|
|
ezfio.nuclei_nucl_charge = l_zeff
|