#!/usr/bin/python3
import numpy as np
# from lib.vasp import system_from_poscar
def _convert_dict(idict):
lines = []
for key in idict.keys():
if isinstance(idict[key], bool):
if idict[key]:
ws = ".TRUE."
else:
ws = ".FALSE."
elif isinstance(idict[key], str):
ws = "'" + idict[key] + "'"
else:
ws = str(idict[key])
lines.append(f"{key}={ws},")
return lines
[docs]
def make_pwscf_01_runctrl_dict(sys_data, idict):
tot_natoms = sum(sys_data["atom_numbs"])
ntypes = len(sys_data["atom_names"])
lines = []
lines.append("&control")
lines += _convert_dict(idict["control"])
lines.append("pseudo_dir='./',")
lines.append("/")
lines.append("&system")
lines += _convert_dict(idict["system"])
lines.append("ibrav=0,")
lines.append("nat=%d," % tot_natoms)
lines.append("ntyp=%d," % ntypes)
lines.append("/")
if "electrons" in idict:
lines.append("&electrons")
lines += _convert_dict(idict["electrons"])
lines.append("/")
lines.append("")
return "\n".join(lines)
def _make_pwscf_01_runctrl(sys_data, ecut, ediff, smearing, degauss):
tot_natoms = sum(sys_data["atom_numbs"])
ntypes = len(sys_data["atom_names"])
ret = ""
ret += "&control\n"
ret += "calculation='scf',\n"
ret += "restart_mode='from_scratch',\n"
ret += "outdir='./OUT',\n"
ret += "tprnfor=.TRUE.,\n"
ret += "tstress=.TRUE.,\n"
ret += "disk_io='none',\n"
ret += "pseudo_dir='./',\n"
ret += "/\n"
ret += "&system\n"
ret += "vdw_corr='TS',\n"
ret += f"ecutwfc={str(ecut)},\n"
ret += f"ts_vdw_econv_thr={str(ediff)},\n"
ret += "nosym=.TRUE.,\n"
ret += "ibrav=0,\n"
ret += "nat=%d,\n" % tot_natoms
ret += "ntyp=%d,\n" % ntypes
if degauss is not None:
ret += f"degauss={degauss:f},\n"
if smearing is not None:
ret += f"smearing='{smearing.lower()}',\n"
ret += "/\n"
ret += "&electrons\n"
ret += f"conv_thr={str(ediff)},\n"
ret += "/\n"
return ret
def _make_pwscf_02_species(sys_data, pps):
atom_names = sys_data["atom_names"]
if "atom_masses" in sys_data:
atom_masses = sys_data["atom_masses"]
else:
atom_masses = [1 for ii in atom_names]
ret = ""
ret += "ATOMIC_SPECIES\n"
ntypes = len(atom_names)
assert ntypes == len(atom_names)
assert ntypes == len(atom_masses)
assert ntypes == len(pps)
for ii in range(ntypes):
ret += f"{atom_names[ii]} {atom_masses[ii]} {pps[ii]}\n"
return ret
def _make_pwscf_03_config(sys_data):
cell = sys_data["cells"][0]
cell = np.reshape(cell, [3, 3])
coordinates = sys_data["coords"][0]
atom_names = sys_data["atom_names"]
atom_numbs = sys_data["atom_numbs"]
ntypes = len(atom_names)
ret = ""
ret += "CELL_PARAMETERS { angstrom }\n"
for ii in range(3):
for jj in range(3):
ret += f"{cell[ii][jj]:f} "
ret += "\n"
ret += "\n"
ret += "ATOMIC_POSITIONS { angstrom }\n"
cc = 0
for ii in range(ntypes):
for jj in range(atom_numbs[ii]):
ret += f"{atom_names[ii]} {coordinates[cc][0]:f} {coordinates[cc][1]:f} {coordinates[cc][2]:f}\n"
cc += 1
return ret
def _kshift(nkpt):
if (nkpt // 2) * 2 == nkpt:
return 1
else:
return 0
def _make_pwscf_04_kpoints(sys_data, kspacing):
cell = sys_data["cells"][0]
cell = np.reshape(cell, [3, 3])
rcell = np.linalg.inv(cell)
rcell = rcell.T
kpoints = [
(np.ceil(2 * np.pi * np.linalg.norm(ii) / kspacing).astype(int)) for ii in rcell
]
ret = ""
if kpoints == [1, 1, 1]:
ret += "K_POINTS gamma"
else:
ret += "K_POINTS { automatic }\n"
for ii in range(3):
ret += "%d " % kpoints[ii]
for ii in range(3):
ret += "%d " % _kshift(kpoints[ii])
ret += "\n"
return ret
def _make_smearing(fp_params):
smearing = None
degauss = None
if "smearing" in fp_params:
smearing = (fp_params["smearing"]).lower()
if "sigma" in fp_params:
degauss = fp_params["sigma"]
if (smearing is not None) and (smearing.split(":")[0] == "mp"):
smearing = "mp"
if smearing not in [None, "gauss", "mp", "fd"]:
raise RuntimeError("unknow smearing method " + smearing)
return smearing, degauss
ry2ev = 13.605693009
bohr2ang = 0.52917721067
kbar2evperang3 = 1
[docs]
def get_block(lines, keyword, skip=0):
ret = []
for idx, ii in enumerate(lines):
if keyword in ii:
blk_idx = idx + 1 + skip
while len(lines[blk_idx]) != 0 and blk_idx != len(lines):
ret.append(lines[blk_idx])
blk_idx += 1
break
return ret
[docs]
def get_types(lines):
ret = []
blk = get_block(lines, "ATOMIC_SPECIES")
for ii in blk:
ret.append(ii.split()[0])
return ret
[docs]
def get_cell(lines):
ret = []
blk = get_block(lines, "CELL_PARAMETERS")
for ii in blk:
ret.append([float(jj) for jj in ii.split()[0:3]])
ret = np.array([ret])
return ret
[docs]
def get_coords(lines):
ret = []
blk = get_block(lines, "ATOMIC_POSITIONS")
for ii in blk:
ret.append([float(jj) for jj in ii.split()[1:4]])
ret = np.array([ret])
return ret
[docs]
def get_natoms(lines):
types = get_types(lines)
names = []
blk = get_block(lines, "ATOMIC_POSITIONS")
for ii in blk:
names.append(ii.split()[0])
natoms = []
for ii in types:
natoms.append(names.count(ii))
# return np.array(natoms, dtype = int)
return natoms
[docs]
def get_atom_types(lines):
types = get_types(lines)
names = []
blk = get_block(lines, "ATOMIC_POSITIONS")
for ii in blk:
names.append(ii.split()[0])
ret = []
for ii in names:
ret.append(types.index(ii))
return np.array(ret, dtype=int)
[docs]
def get_energy(lines):
for ii in lines:
if "! total energy" in ii:
return np.array([ry2ev * float(ii.split("=")[1].split()[0])])
return None
[docs]
def get_force(lines):
blk = get_block(lines, "Forces acting on atoms", skip=1)
ret = []
for ii in blk:
ret.append([float(jj) for jj in ii.split("=")[1].split()])
ret = np.array([ret])
ret *= ry2ev / bohr2ang
return ret
[docs]
def get_stress(lines, cells):
vols = []
for ii in cells:
vols.append(np.linalg.det(ii.reshape([3, 3])))
blk = get_block(lines, "total stress")
ret = []
for ii in blk:
ret.append([float(jj) for jj in ii.split()[3:6]])
ret = np.array([ret])
for idx, ii in enumerate(ret):
ii *= vols[idx] * 1e3 / 1.602176621e6
return ret
[docs]
def cvt_1frame(fin, fout):
with open(fout) as fp:
outlines = fp.read().split("\n")
with open(fin) as fp:
inlines = fp.read().split("\n")
# outlines = open(fout, 'r').read().split('\n')
# inlines = open(fin, 'r').read().split('\n')
data = {}
data["orig"] = np.array([0, 0, 0])
data["atom_names"] = get_types(inlines)
data["atom_numbs"] = get_natoms(inlines)
data["atom_types"] = get_atom_types(inlines)
data["coords"] = get_coords(inlines)
data["cells"] = get_cell(inlines)
data["energies"] = get_energy(outlines)
data["forces"] = get_force(outlines)
data["virials"] = get_stress(outlines, data["cells"])
return data
